src/rxvtc.C

   1 /*--------------------------------*-C-*---------------------------------*
   2  * File:        rxvtc.C
   3  *----------------------------------------------------------------------*
   4  *
   5  * All portions of code are copyright by their respective author/s.
   6  * Copyright (c) 2003-2004 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 "rxvtdaemon.h"
  25
  26 #include "rxvt.h"
  27
  28 #include <cstdio>
  29 #include <cstdlib>
  30 #include <cstring>
  31 #include <csignal>
  32
  33 #include <unistd.h>
  34 #include <sys/socket.h>
  35 #include <sys/un.h>
  36
  37 struct client : rxvt_connection {
  38   client ();
  39 };
  40
  41 client::client ()
  42 {
  43   if ((fd = socket (PF_UNIX, SOCK_STREAM, 0)) < 0)
  44     {
  45       perror ("unable to create communications socket");
  46       exit (EXIT_FAILURE);
  47     }
  48
  49   char *sockname = rxvt_connection::unix_sockname ();
  50   sockaddr_un sa;
  51   sa.sun_family = AF_UNIX;
  52   strcpy (sa.sun_path, sockname);
  53   free (sockname);
  54
  55   if (connect (fd, (sockaddr *)&sa, sizeof (sa)))
  56     {
  57       perror ("unable to connect to the rxvt-unicode daemon");
  58       exit (EXIT_FAILURE);
  59     }
  60 }
  61
  62 extern char **environ;
  63
  64 int
  65 main (int argc, const char *const *argv)
  66 {
  67   client c;
  68   char buf[PATH_MAX];
  69
  70   {
  71     sigset_t ss;
  72
  73     sigemptyset (&ss);
  74     sigaddset (&ss, SIGHUP);
  75     sigprocmask (SIG_BLOCK, &ss, 0);
  76   }
  77
  78   c.send ("NEW");
  79   // instead of getcwd we could opendir (".") and pass the fd for fchdir *g*
  80   c.send ("CWD"), c.send (getcwd (buf, sizeof (buf)));
  81
  82   for (char **var = environ; *environ; environ++)
  83     c.send ("ENV"), c.send (*environ);
  84
  85   const char *base = strrchr (argv[0], '/');
  86   base = base ? base + 1 : argv[0];
  87   c.send ("ARG"), c.send (strcmp (base, RXVTNAME "c") ? base : RXVTNAME);
  88
  89   for (int i = 1; i < argc; i++)
  90     c.send ("ARG"), c.send (argv[i]);
  91
  92   c.send ("END");
  93
  94   auto_str tok;
  95
  96   for (;;)
  97     if (!c.recv (tok))
  98       {
  99         fprintf (stderr, "protocol error: unexpected eof from server.\n");
 100         break;
 101       }
 102     else if (!strcmp (tok, "MSG") && c.recv (tok))
 103       fprintf (stderr, "%s", (const char *)tok);
 104     else if (!strcmp (tok, "END"))
 105       {
 106         int success;
 107         if (c.recv (success))
 108           exit (success ? EXIT_SUCCESS : EXIT_FAILURE);
 109       }
 110     else
 111       {
 112         fprintf (stderr, "protocol error: received illegal token '%s'.\n", (const char *)tok);
 113         break;
 114       }
 115
 116   return EXIT_FAILURE;
 117 }
 118