--- rxvt-unicode/src/rxvtc.C 2004/08/22 01:40:31 1.13 +++ rxvt-unicode/src/rxvtc.C 2006/02/01 18:07:18 1.23 @@ -21,29 +21,34 @@ *----------------------------------------------------------------------*/ #include "../config.h" -#include "rxvtdaemon.h" - -#include "rxvt.h" - #include #include #include +#include #include -#include #include #include +#include "rxvtdaemon.h" +#include "libptytty.h" + +#include "rxvt.h" + +#define STATUS_SUCCESS 0 +#define STATUS_FAILURE 1 +#define STATUS_CONNECTION_FAILED 2 + struct client : rxvt_connection { client (); }; client::client () { - if ((fd = socket (PF_UNIX, SOCK_STREAM, 0)) < 0) + if ((fd = socket (AF_UNIX, SOCK_STREAM, 0)) < 0) { perror ("unable to create communications socket"); - exit (EXIT_FAILURE); + exit (STATUS_FAILURE); } char *sockname = rxvt_connection::unix_sockname (); @@ -55,7 +60,7 @@ if (connect (fd, (sockaddr *)&sa, sizeof (sa))) { perror ("unable to connect to the rxvt-unicode daemon"); - exit (EXIT_FAILURE); + exit (STATUS_CONNECTION_FAILED); } } @@ -65,8 +70,6 @@ main (int argc, const char *const *argv) { client c; - char buf[PATH_MAX]; - { sigset_t ss; @@ -76,11 +79,20 @@ } c.send ("NEW"); + // instead of getcwd we could opendir (".") and pass the fd for fchdir *g* - c.send ("CWD"), c.send (getcwd (buf, sizeof (buf))); + char cwd[PATH_MAX]; + + if (!getcwd (cwd, sizeof (cwd))) + { + perror ("unable to determine current working directory"); + exit (STATUS_FAILURE); + } - for (char **var = environ; *environ; environ++) - c.send ("ENV"), c.send (*environ); + c.send ("CWD"), c.send (cwd); + + for (char **var = environ; *var; var++) + c.send ("ENV"), c.send (*var); const char *base = strrchr (argv[0], '/'); base = base ? base + 1 : argv[0]; @@ -92,6 +104,7 @@ c.send ("END"); auto_str tok; + int cint; for (;;) if (!c.recv (tok)) @@ -101,18 +114,27 @@ } else if (!strcmp (tok, "MSG") && c.recv (tok)) fprintf (stderr, "%s", (const char *)tok); + else if (!strcmp (tok, "GETFD") && c.recv (cint)) + { + if (!ptytty::send_fd (c.fd, cint)) + { + fprintf (stderr, "unable to send fd %d: ", cint); perror (0); + exit (STATUS_FAILURE); + } + } else if (!strcmp (tok, "END")) { int success; + if (c.recv (success)) - exit (success ? EXIT_SUCCESS : EXIT_FAILURE); + exit (success ? STATUS_SUCCESS : STATUS_FAILURE); } else { - fprintf (stderr, "protocol error: received illegal token '%s'.\n", (const char *)tok); + fprintf (stderr, "protocol error: received unsupported token '%s'.\n", (const char *)tok); break; } - return EXIT_FAILURE; + return STATUS_FAILURE; }