X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=blobdiff_plain;f=apps%2Fsympathy.c;h=367797836b14385fbf9065b3a9c7807383c59732;hb=HEAD;hp=41d0bbb6961841d740a9e7f8cf52c37715ac141a;hpb=ef427cbe3bc216800179b66b889b71fec843387c;p=sympathy.git diff --git a/apps/sympathy.c b/apps/sympathy.c index 41d0bbb..3677978 100644 --- a/apps/sympathy.c +++ b/apps/sympathy.c @@ -1,16 +1,91 @@ -/* +/* * sympathy.c: * - * Copyright (c) 2008 James McKenzie , + * Copyright (c) 2008 James McKenzie , * All rights reserved. * */ static char rcsid[] = - "$Id$"; + "$Id: sympathy.c,v 1.52 2010/07/27 14:49:34 james Exp $"; -/* - * $Log$ +/* + * $Log: sympathy.c,v $ + * Revision 1.52 2010/07/27 14:49:34 james + * add support for byte logging + * + * Revision 1.51 2010/07/16 11:04:10 james + * ignore tedious return values + * + * Revision 1.50 2008/05/09 12:56:28 staffcvs + * *** empty log message *** + * + * Revision 1.49 2008/05/09 12:43:49 staffcvs + * *** empty log message *** + * + * Revision 1.48 2008/05/09 12:35:57 james + * *** empty log message *** + * + * Revision 1.47 2008/05/09 12:26:58 staffcvs + * *** empty log message *** + * + * Revision 1.46 2008/05/09 12:19:18 james + * *** empty log message *** + * + * Revision 1.45 2008/03/15 01:44:31 james + * *** empty log message *** + * + * Revision 1.44 2008/03/12 01:30:23 james + * *** empty log message *** + * + * Revision 1.43 2008/03/12 01:26:56 james + * *** empty log message *** + * + * Revision 1.42 2008/03/11 15:02:52 james + * *** empty log message *** + * + * Revision 1.41 2008/03/10 11:49:32 james + * *** empty log message *** + * + * Revision 1.40 2008/03/07 14:16:44 james + * *** empty log message *** + * + * Revision 1.39 2008/03/07 14:13:40 james + * *** empty log message *** + * + * Revision 1.38 2008/03/07 13:56:39 james + * *** empty log message *** + * + * Revision 1.37 2008/03/07 13:16:02 james + * *** empty log message *** + * + * Revision 1.36 2008/03/06 21:34:09 james + * *** empty log message *** + * + * Revision 1.35 2008/03/06 21:33:02 james + * *** empty log message *** + * + * Revision 1.34 2008/03/06 16:49:39 james + * *** empty log message *** + * + * Revision 1.33 2008/03/06 16:49:05 james + * *** empty log message *** + * + * Revision 1.32 2008/03/03 06:04:42 james + * *** empty log message *** + * + * Revision 1.31 2008/03/03 06:04:18 james + * *** empty log message *** + * + * Revision 1.30 2008/03/02 10:38:18 james + * *** empty log message *** + * + * Revision 1.29 2008/03/02 10:37:56 james + * *** empty log message *** + * + * Revision 1.28 2008/03/02 10:27:24 james + * *** empty log message *** + * * Revision 1.27 2008/03/02 09:55:37 james * *** empty log message *** * @@ -105,14 +180,21 @@ static char rcsid[] = #include #include #include +#include +#include +#include + #include "mainloop.h" +int use_syslog = 0; + extern void usage (void); +extern char *expand (const char *, int *); static char hostname[1024]; char *socket_dirs[] = - { ".sympathy", "sympathy", "/etc/sympathy", "/var/sympathy", NULL }; + { "~/.sympathy", "~/sympathy", "/etc/sympathy", "/var/sympathy", NULL }; int safe_atoi (char *a) @@ -138,60 +220,62 @@ fatal_moan (char *fmt, ...) va_start (ap, fmt); - vfprintf (stderr, fmt, ap); + + if (use_syslog) + { + vsyslog (LOG_ERR, fmt, ap); + } + else + { + vfprintf (stderr, fmt, ap); + putc ('\n', stderr); + } va_end (ap); - putc ('\n', stderr); exit (1); } +static void +sigchld (int dummy) +{ + int status; + wait3 (&status, WNOHANG, NULL); +} +/* Dispell zombies, from, for example, log compression */ +void +garlic (void) +{ + struct sigaction sa; -/*make the path in fmt from home (hence the name) */ + sa.sa_handler = sigchld; + sa.sa_flags = SA_RESTART; + sigaction (SIGCHLD, &sa, NULL); +} char * -mome (char *fmt, ...) +teedious_snprintf (char *fmt, va_list ap) { - + va_list aq; + int size = 1024; + char *buf = malloc (size); int n; - int homelen; - char *buf, *home; - va_list ap; - int size; - - - home = getenv ("HOME"); - if (!home) - return NULL; - - homelen = strlen (home) + 1; - - size = 1024 + homelen; - - buf = malloc (size); if (!buf) fatal_moan ("malloc failed"); - strcpy (buf, home); - strcat (buf, "/"); - - if (!fmt) - return buf; - while (1) { + va_copy (aq, ap); + n = vsnprintf (buf, size, fmt, aq); + va_end (aq); - va_start (ap, fmt); - n = vsnprintf (buf + homelen, size - homelen, fmt, ap); - va_end (ap); - - if (n > -1 && n < (size - homelen)) + if (n > -1 && n < (size)) return buf; if (n > -1) /* glibc 2.1 */ - size = homelen + n + 1; + size = n + 1; else /* glibc 2.0 */ size *= 2; /* twice the old size */ @@ -200,6 +284,7 @@ mome (char *fmt, ...) if (!buf) fatal_moan ("malloc failed"); } + } char * @@ -207,7 +292,11 @@ gloo_paths (char *dir, char *leaf) { int i; char *ret, *ptr; - ret = ptr = malloc (strlen (dir) + strlen (leaf) + 2); + if (!dir) + dir = ""; + if (!leaf) + leaf = ""; + ret = ptr = xmalloc (strlen (dir) + strlen (leaf) + 2); while (*dir) *(ptr++) = *(dir++); @@ -220,6 +309,103 @@ gloo_paths (char *dir, char *leaf) } +/* make the path in fmt from home (hence the name) */ +char * +mome (char *fmt, ...) +{ + char *ret, *home, *leaf; + va_list ap; + + + home = getenv ("HOME"); + if (!home) + return NULL; + + if (fmt) + { + va_start (ap, fmt); + leaf = teedious_snprintf (fmt, ap); + va_end (ap); + } + else + { + leaf = NULL; + } + + ret = gloo_paths (home, leaf); + if (leaf) + free (leaf); + + return ret; +} + + + +Socket * +find_socket (char **retpath, char *fmt, ...) +{ + Socket *ret; + char *path, *leaf, *h, **ptr; + va_list ap; + + + if (fmt) + { + va_start (ap, fmt); + leaf = teedious_snprintf (fmt, ap); + va_end (ap); + } + else + { + leaf = NULL; + } + + + for (ptr = socket_dirs; *ptr; ptr++) + { + if (**ptr == '~') + { + h = mome (*ptr + 1); + } + else + { + h = *ptr; + } + + if (!h) + continue; + + + path = gloo_paths (h, leaf); + if (**ptr == '~') + free (h); + + ret = socket_connect (path); + + + if (ret) + { + if (retpath) + { + *retpath = path; + } + else + { + free (path); + } + free (leaf); + return ret; + } + free (path); + + } + + free (leaf); + return NULL; + +} + + int list_sockets_in_dir (char *sockdir) @@ -235,7 +421,7 @@ list_sockets_in_dir (char *sockdir) int hostname_len = strlen (hostname); if (!dir) - fatal_moan ("can't examine %s for sockets", sockdir); + return; rewinddir (dir); @@ -285,20 +471,28 @@ list_sockets (void) for (ptr = socket_dirs; *ptr; ptr++) { - h = mome (*ptr); - - if (h) + if (**ptr == '~') + { + h = mome (*ptr + 1); + } + else { - list_sockets_in_dir (h); - free (h); + h = *ptr; } + if (!h) + continue; + + list_sockets_in_dir (h); + + if (**ptr == '~') + free (h); + } return 0; } - void get_hostname (void) { @@ -315,34 +509,54 @@ get_hostname (void) strcat (hostname, "."); } +void +send_to_server (Socket * c, char *s) +{ + int n; + + s = expand (s, &n); + + if (!n) + return; + + while (n--) + { + ipc_msg_send_key (c, *(uint8_t *) s); + s++; + } + ipc_msg_send_killme (c); +} int main (int argc, char *argv[]) { + Context ctx_store = { 0 }, *ctx = &ctx_store; int c; extern char *optarg; extern int optind, opterr, optopt; CRT_Pos size = { VT102_COLS_80, VT102_ROWS_24 }; - int csnok_pipe[2] = { 0 }; - int csnok = 0; + ANSI *ansi = NULL; + + int cs_pipe[2] = { 0 }; + int cs = 0; int oflags[128]; char *oargs[128]; Socket *server_socket = NULL, *client_socket = NULL; - ANSI *ansi = NULL; - TTY *tty = NULL; - Log *log = NULL; int history = 200; int pid; + char *pid_file = NULL; + get_hostname (); memset (oflags, 0, sizeof (oflags)); memset (oargs, 0, sizeof (oargs)); - while ((c = getopt (argc, argv, "vw:utscr:lKHd:pb:fL:Fk:n:")) != EOF) + + while ((c = getopt (argc, argv, "BI:NCSRP:vw:utscr:lKHd:pb:fL:Fk:n:")) != EOF) { switch (c) { @@ -350,27 +564,39 @@ main (int argc, char *argv[]) case 'h': case '?': usage (); + case 'S': + use_syslog++; + openlog ("sympathy", LOG_PID | LOG_CONS, LOG_DAEMON); + /*fall through */ default: - if ((c > 64) && (c < 91)) + if ((c >= 'A') && (c <= 'Z')) { oflags[c]++; oargs[c] = optarg; } - else if ((c > 96) && (c < 123)) + else if ((c >= 'a') && (c <= 'z')) { oflags[c]++; oargs[c] = optarg; } else { - fprintf (stderr, "unknown option %c\n", c); + if (use_syslog) + { + syslog (LOG_ERR, "unknown option %c\n", c); + } + else + { + fprintf (stderr, "unknown option %c\n", c); + } usage (); } } } - /*Compatability for screen's ls */ + + /* Compatability for screen's ls */ if (oflags['l']) oflags['s'] = 0; @@ -382,10 +608,11 @@ main (int argc, char *argv[]) sum += oflags['r']; sum += oflags['l']; sum += oflags['v']; + sum += oflags['C']; if (!sum) { - /*If no mode is specified behave like screen */ + /* If no mode is specified behave like screen */ oflags['s']++; oflags['c']++; sum++; @@ -393,7 +620,7 @@ main (int argc, char *argv[]) if (sum != 1) fatal_moan - ("specifiy exactly one of ( -c and or -s ), -t, -r, -l and -v"); + ("specifiy exactly one of ( -c and or -s ), -t, -r, -C, -l and -v"); } if (oflags['v']) @@ -421,7 +648,7 @@ main (int argc, char *argv[]) fatal_moan ("agrument to -n must be greater than zero"); } - /*Fold -r implies -c */ + /* Fold -r implies -c */ if (oflags['r']) oflags['c']++; @@ -432,31 +659,33 @@ main (int argc, char *argv[]) if (oflags['c'] && oflags['s'] && oflags['F']) fatal_moan ("-F is incompatible with -c -s"); - /*implement server and client: this process forks. The parent */ - /*becomes the client and the child forks again to become the */ - /*server. If there's no -k argument the client (parent) needs */ - /*to find out the pid of the server, we use a pipe */ + if (oflags['H'] && oflags['N']) + fatal_moan ("-H is incompatible with -N"); + + /* implement server and client: this process forks. The parent */ + /* becomes the client and the child forks again to become the */ + /* server. If there's no -k argument the client (parent) needs */ + /* to find out the pid of the server, we use a pipe */ if (oflags['s'] && oflags['c']) { - if (!oflags['k']) - { - csnok++; - pipe (csnok_pipe); - } + cs++; + int result; + result = pipe (cs_pipe); switch (pid = fork ()) { - case 0: /*child becomes the server */ + case 0: /* child becomes the server */ oflags['c'] = 0; oflags['H'] = 0; + oflags['N'] = 0; + oflags['I'] = 0; - if (csnok) - close (csnok_pipe[0]); + close (cs_pipe[0]); break; case -1: fatal_moan ("fork failed"); - default: /*parent becomes client */ + default: /* parent becomes client */ oflags['s'] = 0; oflags['K'] = 0; oflags['d'] = 0; @@ -464,26 +693,31 @@ main (int argc, char *argv[]) oflags['b'] = 0; oflags['f'] = 0; oflags['L'] = 0; + oflags['R'] = 0; + oflags['B'] = 0; + oflags['P'] = 0; oflags['n'] = 0; oflags['w'] = 0; - /*Collect the child */ + /* Collect the child */ waitpid (pid, NULL, 0); - /*if there was no k argument we need to find the */ - /*pid of the server process so that we can work out */ - /*what the socket is called. The server tells us on */ - /*a pipe */ + /* if there was no k argument we need to find the */ + /* pid of the server process so that we can work out */ + /* what the socket is called. The server tells us on */ + /* a pipe. We do this even if k is specified to avoid */ + /* a race, the server writes to the pipe when the socket */ + /* is opened */ - if (csnok) - { - close (csnok_pipe[1]); + close (cs_pipe[1]); - if (read (csnok_pipe[0], &pid, sizeof (pid)) != sizeof (pid)) - fatal_moan ("Failed to receive pid of server process"); + if (read (cs_pipe[0], &pid, sizeof (pid)) != sizeof (pid)) + fatal_moan ("Failed to receive pid of server process"); - close (csnok_pipe[0]); + close (cs_pipe[0]); + if (!oflags['k']) + { oargs['k'] = mome ("/.sympathy/%s%d", hostname, pid); oflags['k']++; } @@ -491,12 +725,21 @@ main (int argc, char *argv[]) } + if (oflags['c'] && !oflags['k'] && !oflags['r']) fatal_moan ("-c requires a socket to be specified with -s or -k or -r"); + if ((oflags['H'] || oflags['N'] || oflags['I']) && oflags['s']) + fatal_moan ("-s is incompatible with -H, -N and -I"); + if ((oflags['p'] || oflags['d'] || oflags['K'] || oflags['b'] || oflags['f'] - || oflags['L']) && oflags['c']) - fatal_moan ("-c or -r are incompatible with -p, -d, -K, -b, -f or -L"); + || oflags['L'] || oflags['R'] || oflags['B'] || oflags['P']) && oflags['c']) + fatal_moan + ("-c or -r are incompatible with -p, -d, -K, -b, -f, -R, -P, -B or -L"); + + if (oflags['C'] && (!oflags['d'])) + fatal_moan ("-C requires -d"); + if (oflags['t'] || oflags['s']) { @@ -528,11 +771,13 @@ main (int argc, char *argv[]) if (oflags['s'] && !oflags['F']) { - /*nochdir incase socket is relative path, unlink then will fail */ - daemon (1, 0); + /* nochdir incase socket is relative path, unlink then will fail */ + int fish; + fish = daemon (1, 0); } + garlic (); if (oflags['s']) { @@ -552,13 +797,15 @@ main (int argc, char *argv[]) server_socket = socket_listen (oargs['k']); - /*Tell our parent's parent what our pid is */ - if (csnok) + /* Tell our parent's parent what our pid is */ + /* and that we've opened the server socket */ + if (cs) { pid = getpid (); - write (csnok_pipe[1], &pid, sizeof (pid)); - close (csnok_pipe[1]); + int fish; + fish = write (cs_pipe[1], &pid, sizeof (pid)); + close (cs_pipe[1]); } if (!server_socket) @@ -566,26 +813,46 @@ main (int argc, char *argv[]) } - if (oflags['s'] || oflags['t']) + if (oflags['s'] || oflags['t'] || oflags['C']) { + if (oflags['P']) + { + FILE *fp; + pid_file = oargs['P']; + fp = fopen (pid_file, "w"); + if (fp) + { + fprintf (fp, "%d", getpid ()); + fclose (fp); + } + } if (oflags['L']) { - log = file_log_new (oargs['L']); - if (!log) + ctx->l = file_log_new (oargs['L'], oflags['R']); + ctx->byte_logging=oflags['B']; + if (!ctx->l) fatal_moan ("unable to access log file %s", oargs['L']); } if (oflags['p']) { - tty = ptty_open (NULL, NULL, &size); - if (!tty) + if (optind < argc) + { + ctx->t = ptty_open (argv[optind], &argv[optind], &size); + } + else + { + ctx->t = ptty_open (NULL, NULL, &size); + } + + if (!ctx->t) fatal_moan ("unable to open a ptty"); } else { - /*HACK-- check that console=device does not occur in */ - /*/proc/cmdline */ + /* HACK-- check that console=device does not occur in */ + /* /proc/cmdline */ if (!oargs['d']) fatal_moan ("no argument to -d"); @@ -601,7 +868,8 @@ main (int argc, char *argv[]) strcat (search_string, ptr); fd = open ("/proc/cmdline", O_RDONLY); - read (fd, kernel_cmdline, sizeof (kernel_cmdline)); + int fish; + fish = read (fd, kernel_cmdline, sizeof (kernel_cmdline)); close (fd); kernel_cmdline[sizeof (kernel_cmdline) - 1] = 0; @@ -610,12 +878,18 @@ main (int argc, char *argv[]) fatal_moan ("/proc/cmdline contains %s", search_string); } - tty = + ctx->t = serial_open (oargs['d'], oflags['K'] ? SERIAL_LOCK_ACTIVE : SERIAL_LOCK_PASSIVE); - if (!tty) + if (!ctx->t) fatal_moan ("unable to open serial port %s", oargs['d']); + + if (oflags['C']) + { + ctx->t->close (ctx->t); + return 0; + } } if (oflags['b']) @@ -625,63 +899,97 @@ main (int argc, char *argv[]) if (baud < 0) fatal_moan ("Unable to parse baudrate %s", oargs['b']); - tty_set_baud (tty, baud); + tty_set_baud (ctx->t, baud); } - tty_set_flow (tty, oflags['f'] ? 1 : 0); - + tty_set_flow (ctx->t, oflags['f'] ? 1 : 0); } - FISH - { - char *id = oargs['r']; - if (id < 0) - fatal_moan ("cannot parse -r %s as an integer", oargs['r']); + if (oflags['r']) + { + char *id = oargs['r']; - oflags['k']++; - if (safe_atoi (id) > 0) - { - oargs['k'] = find_socket ("%s%d", hostname, safe_atoi (id)); - } - else - { - oargs['k'] = find_socket ("%s", id); - } - oflags['r'] = 0; - oflags['c']++; - } + if (!id) + fatal_moan ("-r requires an argument"); - if (oflags['c']) - { + if (safe_atoi (id) > 0) + { + client_socket = + find_socket (&oargs['k'], "%s%d", hostname, safe_atoi (id)); + } + else + { + client_socket = find_socket (&oargs['k'], "%s", id); + } + + if (!client_socket) + fatal_moan ("failed to find socket %s", oargs['r']); + } + else if (oflags['c']) + { client_socket = socket_connect (oargs['k']); + if (!client_socket) fatal_moan ("failed to connect to socket %s", oargs['k']); - } - - if (oflags['c'] || oflags['t']) + if (oflags['I']) + { + if (!client_socket) + fatal_moan ("-I requires either -c or -r", oargs['k']); + if (!oargs['I']) + fatal_moan ("-I requires an arugment"); + send_to_server (client_socket, oargs['I']); + } + else { + if (client_socket) + ipc_msg_send_initialize (client_socket); - if (oflags['H']) - { - ansi = ansi_new_html (stdout); - } - else + if (oflags['c'] || oflags['t']) { - terminal_register_handlers (); - ansi = - ansi_new_from_terminal (terminal_open (0, 1), - oflags['u'] ? 0 : 1); - ansi->reset (ansi, NULL); + if (oflags['N']) + { + ctx->r = rx_new_raw (0, 1); + ansi = ansi_new_raw (0, 1); + } + else if (oflags['H']) + { + ansi = ansi_new_html (stdout); + } + else + { + terminal_register_handlers (); + ansi = + ansi_new_from_terminal (terminal_open (0, 1), + oflags['u'] ? 0 : 1); + ansi->reset (ansi, NULL); + + } + + if (ansi->set_title) + { + if (oflags['c'] && oargs['k']) + { + ansi->set_title (ansi, oargs['k']); + } + else if ((ctx->t) && (ctx->t->name)) + { + ansi->set_title (ansi, ctx->t->name); + } + } + } } - mainloop (tty, server_socket, client_socket, ansi, log, history, &size); + ctx->v = vt102_new (&size); + ctx->h = history_new (history); + + mainloop (ctx, ansi, server_socket, client_socket); if (ansi) { @@ -689,17 +997,20 @@ main (int argc, char *argv[]) terminal_atexit (); } - if (tty) - tty->close (tty); + if (ctx->t) + ctx->t->close (ctx->t); - if (log) - log->close (log); + if (ctx->l) + ctx->l->close (ctx->l); if (server_socket) socket_free (server_socket); if (client_socket) socket_free (client_socket); - if (!oflags['H']) + if (pid_file) + unlink (pid_file); + + if (!oflags['H'] && !oflags['I']) printf ("you have now exited sympathy\n"); return 0; }