--- rxvt-unicode/src/rxvtd.C	2003/11/24 17:28:08	1.1
+++ rxvt-unicode/src/rxvtd.C	2004/08/15 02:17:32	1.16
@@ -1,57 +1,99 @@
-#include "rxvtlib.h"
+/*--------------------------------*-C-*---------------------------------*
+ * File:	rxvtd.C
+ *----------------------------------------------------------------------*
+ *
+ * All portions of code are copyright by their respective author/s.
+ * Copyright (c) 2003-2004 Marc Lehmann <pcg@goof.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ *----------------------------------------------------------------------*/
+
+#include "../config.h"
+#include "rxvt.h"
 #include "rxvtdaemon.h"
 #include "iom.h"
 
 #include <cstdio>
 #include <cstdlib>
+#include <cstdarg>
+#include <cstring>
 
 #include <unistd.h>
+#include <fcntl.h>
+#include <sys/types.h>
+#include <sys/stat.h>
 #include <sys/socket.h>
 #include <sys/un.h>
 
+#include <cerrno>
+
+extern char **environ;
+
 struct server : rxvt_connection {
+  log_callback log_cb;
+
   void read_cb (io_watcher &w, short revents); io_watcher read_ev;
+  void log_msg (const char *msg);
 
   server (int fd)
-  : read_ev (this, &server::read_cb)
+  : read_ev (this, &server::read_cb),
+    log_cb (this, &server::log_msg)
   {
     this->fd = fd;
     read_ev.start (fd, EVENT_READ);
   }
 
-  void err ();
+  void err (const char *format = 0, ...);
 };
 
-struct listener {
+struct unix_listener {
   int fd;
 
   void accept_cb (io_watcher &w, short revents); io_watcher accept_ev;
 
-  listener ();
+  unix_listener (const char *sockname);
 };
 
-listener::listener ()
-: accept_ev (this, &listener::accept_cb)
+unix_listener::unix_listener (const char *sockname)
+: accept_ev (this, &unix_listener::accept_cb)
 {
-  if ((fd = socket (PF_LOCAL, SOCK_STREAM, 0)) < 0)
+  if ((fd = socket (PF_UNIX, SOCK_STREAM, 0)) < 0)
     {
       perror ("unable to create listening socket");
       exit (EXIT_FAILURE);
     }
 
+  fcntl (fd, F_SETFD, FD_CLOEXEC);
+
   sockaddr_un sa;
 
   sa.sun_family = AF_UNIX;
-  strcpy (sa.sun_path, rxvt_connection::unix_sockname ());
+  strcpy (sa.sun_path, sockname);
 
   unlink (rxvt_connection::unix_sockname ());
 
+  mode_t omask = umask (0077);
+
   if (bind (fd, (sockaddr *)&sa, sizeof (sa)))
     {
       perror ("unable to bind listening socket");
       exit (EXIT_FAILURE);
     }
 
+  umask (omask);
+
   if (listen (fd, 5))
     {
       perror ("unable to listen on socket");
@@ -61,48 +103,125 @@
   accept_ev.start (fd, EVENT_READ);
 }
 
-void listener::accept_cb (io_watcher &w, short revents)
+void unix_listener::accept_cb (io_watcher &w, short revents)
 {
   int fd2 = accept (fd, 0, 0);
 
   if (fd2 >= 0)
-    new server (fd2);
+    {
+      fcntl (fd2, F_SETFD, FD_CLOEXEC);
+      new server (fd2);
+    }
+}
+
+void server::log_msg (const char *msg)
+{
+  send ("MSG"), send (msg);
 }
 
-void server::err ()
+void server::err (const char *format, ...)
 {
+  if (format)
+    {
+      char err[1024];
+
+      va_list ap;
+      va_start (ap, format);
+      vsnprintf (err, 1024, format, ap);
+      va_end (ap);
+
+      send ("MSG"), send (err);
+    }
+
+  send ("END", 0);
   close (fd);
   delete this;
 }
 
 void server::read_cb (io_watcher &w, short revents)
 {
-  token cmd;
+  auto_str tok;
 
-  if (recv (cmd))
+  if (recv (tok))
     {
-      if (!strcmp (cmd, "NEW"))
+      if (!strcmp (tok, "NEW"))
         {
+          stringvec *argv = new stringvec;
+          stringvec *envv = new stringvec;
+            
+          for (;;)
+            {
+              if (!recv (tok))
+                return err ();
+
+              if (!strcmp (tok, "END"))
+                break;
+              else if (!strcmp (tok, "ENV") && recv (tok))
+                envv->push_back (tok.get ());
+              else if (!strcmp (tok, "CWD") && recv (tok))
+                {
+                  if (chdir (tok))
+                    err ("unable to change to working directory to '%s': %s",
+                         (char *)tok, strerror (errno));
+                }
+              else if (!strcmp (tok, "ARG") && recv (tok))
+                argv->push_back (tok.get ());
+              else
+                return err ("protocol error: unexpected NEW token");
+            }
+
+          envv->push_back (0);
+
+          {
+            char **old_environ = environ;
+            environ = envv->begin ();
+
+            rxvt_term *term = new rxvt_term;
+            
+            term->log_hook = &log_cb;
+            term->argv = argv;
+            term->envv = envv;
+
+            bool success;
+            
+            try
+              {
+                success = term->init (argv->size (), argv->begin ());
+              }
+            catch (const class rxvt_failure_exception &e)
+              {
+                success = false;
+              }
+
+            term->log_hook = 0;
+
+            environ = old_environ;
+
+            if (!success)
+              term->destroy ();
+
+            send ("END"); send (success ? 1 : 0);
+          }
         }
       else
-        err ();
+        return err ("protocol error: request '%s' unsupported.", (char *)tok);
     }
   else
-    err ();
+    return err ();
 }
 
 int
-main(int argc, const char *const *argv)
+main (int argc, const char *const *argv)
 {
-  listener l;
+  rxvt_init ();
+
+  char *sockname = rxvt_connection::unix_sockname ();
+  unix_listener l (sockname);
+  printf ("rxvtd listening on %s.\n", sockname);
+  free (sockname);
+
   iom.loop ();
 
-#if 0
-  if (rxvt_init(argc, argv) == NULL)
-      return EXIT_FAILURE;
-
-  dR;
-  rxvt_main_loop(aR);	/* main processing loop */
-#endif
   return EXIT_SUCCESS;
 }
+