mirror of https://github.com/dirtbags/moth.git
97 lines
2.1 KiB
C
97 lines
2.1 KiB
C
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <sysexits.h>
|
|
#include <stdio.h>
|
|
#include "arc4.h"
|
|
|
|
/* I don't feel compelled to put all the TCP client code in here
|
|
* when it's so simple to run this with netcat or ucspi. Plus, using
|
|
* stdin and stdout makes it simpler to test.
|
|
*/
|
|
|
|
int
|
|
read_key(char *filename, uint8_t *key, size_t *keylen)
|
|
{
|
|
int fd = open(filename, O_RDONLY);
|
|
int len;
|
|
|
|
if (-1 == fd) {
|
|
perror("open");
|
|
return EX_NOINPUT;
|
|
}
|
|
|
|
len = read(fd, key, *keylen);
|
|
if (-1 == len) {
|
|
perror("read");
|
|
return EX_NOINPUT;
|
|
}
|
|
*keylen = (size_t)len;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[]) {
|
|
uint8_t skey[200];
|
|
size_t skeylen = sizeof(skey);
|
|
char token[200];
|
|
size_t tokenlen;
|
|
int ret;
|
|
|
|
if (argc != 3) {
|
|
fprintf(stderr, "Usage: %s SERVICE SERVICEKEY 3>TOKENFILE\n", argv[0]);
|
|
fprintf(stderr, "\n");
|
|
fprintf(stderr, "SERVICEKEY is a filename.\n");
|
|
fprintf(stderr, "Server chatter happens over stdin and stdout.\n");
|
|
fprintf(stderr, "Tokens are written to file descriptor 3.\n");
|
|
fprintf(stderr, "\n");
|
|
fprintf(stderr, "To run with netcat:\n");
|
|
fprintf(stderr, " nc server 1 -e tokencli cat cat.key 3> tokenfile\n");
|
|
return EX_USAGE;
|
|
}
|
|
|
|
/* read in keys */
|
|
ret = read_key(argv[2], skey, &skeylen);
|
|
if (0 != ret) return ret;
|
|
|
|
/* write service name */
|
|
write(1, argv[1], strlen(argv[1]));
|
|
|
|
/* read nonce, send back encrypted version */
|
|
{
|
|
uint8_t nonce[80];
|
|
int noncelen;
|
|
|
|
noncelen = read(0, nonce, sizeof(nonce));
|
|
if (0 >= noncelen) {
|
|
perror("read");
|
|
return EX_IOERR;
|
|
}
|
|
arc4_crypt_buffer(skey, skeylen, nonce, (size_t)noncelen);
|
|
write(1, nonce, (size_t)noncelen);
|
|
}
|
|
|
|
/* read token */
|
|
{
|
|
int len;
|
|
|
|
len = read(0, token, sizeof(token));
|
|
if (0 >= len) {
|
|
perror("read");
|
|
return EX_IOERR;
|
|
}
|
|
tokenlen = (size_t)len;
|
|
}
|
|
|
|
/* decrypt it */
|
|
arc4_crypt_buffer(skey, skeylen, (uint8_t *)token, tokenlen);
|
|
|
|
/* write it to fd 3 */
|
|
write(3, token, tokenlen);
|
|
|
|
return 0;
|
|
}
|