diff options
Diffstat (limited to 'ratatox.c')
| -rw-r--r-- | ratatox.c | 361 |
1 files changed, 320 insertions, 41 deletions
@@ -3,6 +3,7 @@ #include <sys/stat.h> #include <sys/types.h> +#include <ctype.h> #include <errno.h> #include <fcntl.h> #include <limits.h> @@ -35,9 +36,10 @@ enum { static struct fifo { const char *name; + int flags; mode_t mode; } fifos[] = { - { "text_in", 0644 }, + { .name = "text_in", .flags = O_RDONLY | O_NONBLOCK, .mode = 0644 }, }; struct friend { @@ -51,28 +53,145 @@ struct friend { TAILQ_ENTRY(friend) entry; }; +struct request { + uint8_t id[TOX_CLIENT_ID_SIZE]; + /* null terminated id */ + uint8_t idstr[2 * TOX_CLIENT_ID_SIZE + 1]; + /* null terminated friend request message */ + uint8_t *msgstr; + TAILQ_ENTRY(request) entry; +}; + static TAILQ_HEAD(friendhead, friend) friendhead = TAILQ_HEAD_INITIALIZER(friendhead); +static TAILQ_HEAD(reqhead, request) reqhead = TAILQ_HEAD_INITIALIZER(reqhead); static Tox *tox; + +static void cb_conn_status(Tox *, int32_t, uint8_t, void *); +static void cb_friend_message(Tox *, int32_t, const uint8_t *, uint16_t, void *); +static void cb_friend_request(Tox *, const uint8_t *, const uint8_t *, uint16_t, void *); +static void cb_name_change(Tox *, int32_t, const uint8_t *, uint16_t, void *); +static void cb_status_message(Tox *, int32_t, const uint8_t *, uint16_t, void *); +static void send_friend_text(struct friend *); static void dataload(void); static void datasave(void); +static void toxrestore(void); +static int toxinit(void); +static int toxconnect(void); +static void id2str(uint8_t *, uint8_t *); +static void str2id(uint8_t *, uint8_t *); static void friendcreate(int32_t); +static void friendload(void); +static int cmdrun(void); +static int doaccept(char *, size_t); +static int dofriend(char *, size_t); +static int dohelp(char *, size_t); +static void loop(void); + +static char qsep[] = " \t\r\n"; + +/* tokenization routines taken from Plan9 */ +static char * +qtoken(char *s, char *sep) +{ + int quoting; + char *t; + + quoting = 0; + t = s; /* s is output string, t is input string */ + while(*t!='\0' && (quoting || strchr(sep, *t)==NULL)) { + if(*t != '\'') { + *s++ = *t++; + continue; + } + /* *t is a quote */ + if(!quoting) { + quoting = 1; + t++; + continue; + } + /* quoting and we're on a quote */ + if(t[1] != '\'') { + /* end of quoted section; absorb closing quote */ + t++; + quoting = 0; + continue; + } + /* doubled quote; fold one quote into two */ + t++; + *s++ = *t++; + } + if(*s != '\0') { + *s = '\0'; + if(t == s) + t++; + } + return t; +} -static void -masterout(const char *fmt, ...) +static char * +etoken(char *t, char *sep) { - FILE *fp; - va_list ap; + int quoting; + + /* move to end of next token */ + quoting = 0; + while(*t!='\0' && (quoting || strchr(sep, *t)==NULL)) { + if(*t != '\'') { + t++; + continue; + } + /* *t is a quote */ + if(!quoting) { + quoting = 1; + t++; + continue; + } + /* quoting and we're on a quote */ + if(t[1] != '\'') { + /* end of quoted section; absorb closing quote */ + t++; + quoting = 0; + continue; + } + /* doubled quote; fold one quote into two */ + t += 2; + } + return t; +} - fp = fopen("master_out", "a"); - if (!fp) { - perror("fopen"); - exit(1); +static int +gettokens(char *s, char **args, int maxargs, char *sep) +{ + int nargs; + + for(nargs=0; nargs<maxargs; nargs++) { + while(*s!='\0' && strchr(sep, *s)!=NULL) + *s++ = '\0'; + if(*s == '\0') + break; + args[nargs] = s; + s = etoken(s, sep); } - va_start(ap, fmt); - vfprintf(fp, fmt, ap); - va_end(ap); - fclose(fp); + + return nargs; +} + +static int +tokenize(char *s, char **args, int maxargs) +{ + int nargs; + + for(nargs=0; nargs<maxargs; nargs++) { + while(*s!='\0' && strchr(qsep, *s)!=NULL) + s++; + if(*s == '\0') + break; + args[nargs] = s; + s = qtoken(s, qsep); + } + + return nargs; } static void @@ -92,14 +211,14 @@ cb_conn_status(Tox *tox, int32_t fid, uint8_t status, void *udata) if (n == 0) { if (status == 0) - masterout("Anonymous went offline\n"); + printf("Anonymous went offline\n"); else - masterout("Anonymous came online\n"); + printf("Anonymous came online\n"); } else { if (status == 0) - masterout("%s went offline\n", name); + printf("%s went offline\n", name); else - masterout("%s came online\n", name); + printf("%s came online\n", name); } TAILQ_FOREACH(f, &friendhead, entry) @@ -139,16 +258,30 @@ cb_friend_message(Tox *tox, int32_t fid, const uint8_t *data, uint16_t len, void static void cb_friend_request(Tox *tox, const uint8_t *id, const uint8_t *data, uint16_t len, void *udata) { - uint8_t msg[len + 1]; + struct request *req; - memcpy(msg, data, len); - msg[len] = '\0'; - tox_add_friend_norequest(tox, id); - if (len > 0) - masterout("Accepted friend request with msg: %s\n", msg); - else - masterout("Accepted friend request\n"); - datasave(); + req = calloc(1, sizeof(*req)); + if (!req) { + perror("calloc"); + exit(1); + } + memcpy(req->id, id, TOX_CLIENT_ID_SIZE); + id2str(req->id, req->idstr); + + if (len > 0) { + req->msgstr = malloc(len + 1); + if (!req->msgstr) { + perror("malloc"); + exit(1); + } + memcpy(req->msgstr, data, len); + req->msgstr[len] = '\0'; + } + + TAILQ_INSERT_TAIL(&reqhead, req, entry); + + printf("Pending request from %s with message: %s\n", + req->idstr, req->msgstr); } static void @@ -176,9 +309,9 @@ cb_name_change(Tox *m, int32_t fid, const uint8_t *data, uint16_t len, void *use if (memcmp(f->namestr, name, len + 1) == 0) break; if (f->namestr[0] == '\0') { - masterout("%s -> %s\n", "Anonymous", name); + printf("%s -> %s\n", "Anonymous", name); } else { - masterout("%s -> %s\n", f->namestr, name); + printf("%s -> %s\n", f->namestr, name); } memcpy(f->namestr, name, len + 1); break; @@ -209,7 +342,7 @@ cb_status_message(Tox *m, int32_t fid, const uint8_t *data, uint16_t len, void * fputs(status, fp); fputc('\n', fp); fclose(fp); - masterout("%s current status to %s\n", f->namestr, status); + printf("%s current status to %s\n", f->namestr, status); break; } } @@ -318,10 +451,10 @@ toxinit(void) tox_set_user_status(tox, TOX_USERSTATUS_NONE); tox_get_address(tox, address); - masterout("ID: "); + printf("ID: "); for (i = 0; i < TOX_FRIEND_ADDRESS_SIZE; i++) - masterout("%02x", address[i]); - masterout("\n"); + printf("%02x", address[i]); + printf("\n"); return 0; } @@ -339,7 +472,7 @@ toxconnect(void) return 0; } -static int +static void id2str(uint8_t *id, uint8_t *idstr) { uint8_t hex[] = "0123456789abcdef"; @@ -353,6 +486,16 @@ id2str(uint8_t *id, uint8_t *idstr) } static void +str2id(uint8_t *idstr, uint8_t *id) +{ + size_t i, len = strlen(idstr) / 2; + char *p = idstr; + + for (i = 0; i < len; ++i, p += 2) + sscanf(p, "%2hhx", &id[i]); +} + +static void friendcreate(int32_t fid) { struct friend *f; @@ -362,7 +505,7 @@ friendcreate(int32_t fid) f = calloc(1, sizeof(*f)); if (!f) { - perror("malloc"); + perror("calloc"); exit(1); } @@ -391,7 +534,7 @@ friendcreate(int32_t fid) perror("mkfifo"); exit(1); } - r = open(path, O_RDONLY | O_NONBLOCK, 0); + r = open(path, fifos[i].flags, 0); if (r < 0) { perror("open"); exit(1); @@ -423,6 +566,134 @@ friendload(void) friendcreate(fids[i]); } +struct cmd { + const char *cmd; + int (*cb)(char *, size_t); + const char *helpstr; +} cmds[] = { + { .cmd = "a", .cb = doaccept, .helpstr = "usage: a [ID]\tAccept or list pending requests\n" }, + { .cmd = "f", .cb = dofriend, .helpstr = "usage: f ID\tSend friend request to ID\n" }, + { .cmd = "h", .cb = dohelp, .helpstr = NULL }, +}; + +static int +doaccept(char *cmd, size_t sz) +{ + struct request *req, *tmp; + char *args[2]; + int n; + int found = 0; + + n = tokenize(cmd, args, 2); + + if (n == 1) { + TAILQ_FOREACH(req, &reqhead, entry) { + printf("Pending request from %s with message: %s\n", + req->idstr, req->msgstr); + found = 1; + } + if (found == 0) + printf("No pending requests\n"); + } else { + for (req = TAILQ_FIRST(&reqhead); req; req = tmp) { + tmp = TAILQ_NEXT(req, entry); + if (strcmp(req->idstr, args[1]) == 0) { + tox_add_friend_norequest(tox, req->id); + printf("Accepted friend request for %s\n", req->idstr); + TAILQ_REMOVE(&reqhead, req, entry); + free(req->msgstr); + free(req); + break; + } + } + } + + return 0; +} + +static int +dofriend(char *cmd, size_t sz) +{ + char *args[2]; + uint8_t id[TOX_FRIEND_ADDRESS_SIZE]; + uint8_t idstr[2 * TOX_FRIEND_ADDRESS_SIZE + 1]; + char *msgstr = "ratatox is awesome!"; + int r; + + r = tokenize(cmd, args, 2); + if (r != 2) { + printf("Command error, type h for help\n"); + return -1; + } + str2id(args[1], id); + + r = tox_add_friend(tox, id, msgstr, strlen(msgstr)); + switch (r) { + case TOX_FAERR_TOOLONG: + printf("Message is too long\n"); + break; + case TOX_FAERR_NOMESSAGE: + printf("Please add a message to your request\n"); + break; + case TOX_FAERR_OWNKEY: + printf("That appears to be your own ID\n"); + break; + case TOX_FAERR_ALREADYSENT: + printf("Friend request already sent\n"); + break; + case TOX_FAERR_UNKNOWN: + printf("Unknown error while sending your request\n"); + break; + default: + printf("Friend request sent\n"); + break; + } + return 0; +} + +static int +dohelp(char *cmd, size_t sz) +{ + size_t i; + + for (i = 0; i < LEN(cmds); i++) + if (cmds[i].helpstr) + printf("%s", cmds[i].helpstr); + return 0; +} + +static int +cmdrun(void) +{ + char cmd[BUFSIZ]; + ssize_t n; + size_t i; + +again: + n = read(STDIN_FILENO, cmd, sizeof(cmd) - 1); + if (n < 0) { + if (errno == EINTR) + goto again; + perror("read"); + exit(1); + } + if (n == 0) + return 0; + cmd[n] = '\0'; + if (cmd[strlen(cmd) - 1] == '\n') + cmd[strlen(cmd) - 1] = '\0'; + if (cmd[0] == '\0') + return 0; + + for (i = 0; i < LEN(cmds); i++) + if (cmd[0] == cmds[i].cmd[0]) + if (cmd[1] == '\0' || isspace((int)cmd[1])) + return (*cmds[i].cb)(cmd, strlen(cmd)); + + fprintf(stderr, "unknown command: %s\n", cmd); + return -1; +} + static void loop(void) { @@ -430,47 +701,55 @@ loop(void) time_t t0, t1; int connected = 0; int i, n; - int fdmax = 0; + int fdmax; fd_set rfds; struct timeval tv; t0 = time(NULL); + printf("Connecting to DHT...\n"); toxconnect(); while (1) { if (tox_isconnected(tox) == 1) { if (connected == 0) { - masterout("Connected to DHT\n"); + printf("Connected to DHT\n"); connected = 1; } } else { t1 = time(NULL); if (t1 > t0 + 5) { t0 = time(NULL); - masterout("Connecting to DHT...\n"); + printf("Connecting to DHT...\n"); toxconnect(); } } tox_do(tox); FD_ZERO(&rfds); + FD_SET(STDIN_FILENO, &rfds); + fdmax = STDIN_FILENO; + TAILQ_FOREACH(f, &friendhead, entry) { for (i = 0; i < NR_FIFOS; i++) { + FD_SET(f->fd[i], &rfds); if (f->fd[i] > fdmax) fdmax = f->fd[i]; - FD_SET(f->fd[i], &rfds); } } tv.tv_sec = 0; tv.tv_usec = tox_do_interval(tox) * 1000; - n = select(fdmax + 1, &rfds, NULL, NULL, - &tv); + n = select(fdmax + 1, &rfds, NULL, NULL, &tv); if (n < 0) { if (errno == EINTR) continue; perror("select"); exit(1); } + if (n == 0) + continue; + + if (FD_ISSET(STDIN_FILENO, &rfds) != 0) + cmdrun(); TAILQ_FOREACH(f, &friendhead, entry) { for (i = 0; i < NR_FIFOS; i++) { |