src/rxvtc.C

   1 /*----------------------------------------------------------------------*
   2  * File:        rxvtc.C
   3  *----------------------------------------------------------------------*
   4  *
   5  * All portions of code are copyright by their respective author/s.
   6  * Copyright (c) 2003-2006 Marc Lehmann <pcg@goof.com>
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  21  *----------------------------------------------------------------------*/
  22
  23 #include "../config.h"
  24 #include <cstdio>
  25 #include <cstdlib>
  26 #include <cstring>
  27 #include <csignal>
  28
  29 #include <unistd.h>
  30 #include <sys/socket.h>
  31 #include <sys/un.h>
  32
  33 #include "rxvtdaemon.h"
  34 #include "libptytty.h"
  35
  36 #include "rxvt.h"
  37
  38 #define STATUS_SUCCESS           0
  39 #define STATUS_FAILURE           1
  40 #define STATUS_CONNECTION_FAILED 2
  41
  42 struct client : rxvt_connection
  43 {
  44   client ();
  45 };
  46
  47 client::client ()
  48 {
  49   sockaddr_un sa;
  50   char *sockname = rxvt_connection::unix_sockname ();
  51
  52   if (strlen (sockname) >= sizeof (sa.sun_path))
  53     {
  54       fputs ("socket name too long, aborting.\n", stderr);
  55       exit (STATUS_FAILURE);
  56     }
  57
  58   if ((fd = socket (AF_UNIX, SOCK_STREAM, 0)) < 0)
  59     {
  60       perror ("unable to create communications socket");
  61       exit (STATUS_FAILURE);
  62     }
  63
  64   sa.sun_family = AF_UNIX;
  65   strcpy (sa.sun_path, sockname);
  66   free (sockname);
  67
  68   if (connect (fd, (sockaddr *)&sa, sizeof (sa)))
  69     {
  70       perror ("unable to connect to the rxvt-unicode daemon");
  71       exit (STATUS_CONNECTION_FAILED);
  72     }
  73 }
  74
  75 extern char **environ;
  76
  77 int
  78 main (int argc, const char *const *argv)
  79 {
  80   // instead of getcwd we could opendir (".") and pass the fd for fchdir *g*
  81   char cwd[PATH_MAX];
  82
  83   if (!getcwd (cwd, sizeof (cwd)))
  84     {
  85       perror ("unable to determine current working directory");
  86       exit (STATUS_FAILURE);
  87     }
  88
  89   client c;
  90
  91   {
  92     sigset_t ss;
  93
  94     sigemptyset (&ss);
  95     sigaddset (&ss, SIGHUP);
  96     sigaddset (&ss, SIGPIPE);
  97     sigprocmask (SIG_BLOCK, &ss, 0);
  98   }
  99
 100   c.send ("NEW");
 101
 102   for (char **var = environ; *var; var++)
 103     c.send ("ENV"), c.send (*var);
 104
 105   const char *base = strrchr (argv[0], '/');
 106   base = base ? base + 1 : argv[0];
 107   c.send ("ARG"), c.send (strcmp (base, RXVTNAME "c") ? base : RXVTNAME);
 108
 109   c.send ("ARG"), c.send ("-cd");
 110   c.send ("ARG"), c.send (cwd);
 111
 112   for (int i = 1; i < argc; i++)
 113     c.send ("ARG"), c.send (argv[i]);
 114
 115   c.send ("END");
 116
 117   auto_str tok;
 118   int cint;
 119
 120   for (;;)
 121     if (!c.recv (tok))
 122       {
 123         fprintf (stderr, "protocol error: unexpected eof from server.\n");
 124         break;
 125       }
 126     else if (!strcmp (tok, "MSG") && c.recv (tok))
 127       fprintf (stderr, "%s", (const char *)tok);
 128     else if (!strcmp (tok, "GETFD") && c.recv (cint))
 129       {
 130         if (!ptytty::send_fd (c.fd, cint))
 131           {
 132             fprintf (stderr, "unable to send fd %d: ", cint); perror (0);
 133             exit (STATUS_FAILURE);
 134           }
 135       }
 136     else if (!strcmp (tok, "END"))
 137       {
 138         int success;
 139
 140         if (c.recv (success))
 141           exit (success ? STATUS_SUCCESS : STATUS_FAILURE);
 142       }
 143     else
 144       {
 145         fprintf (stderr, "protocol error: received unsupported token '%s'.\n", (const char *)tok);
 146         break;
 147       }
 148
 149   return STATUS_FAILURE;
 150 }
 151