[ViewVC] Diff of: cvs/rxvt-unicode/src/rxvtd.C

Comparing rxvt-unicode/src/rxvtd.C (file contents):
Revision 1.60 by root, Tue Jun 12 10:45:53 2012 UTC vs.
Revision 1.63 by sf-exg, Sun May 9 05:53:12 2021 UTC

  * All portions of code are copyright by their respective author/s.
  * Copyright (c) 2003-2007 Marc Lehmann <schmorp@schmorp.de>
  *
  * 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
+ * the Free Software Foundation; either version 3 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
 struct unix_listener {
   int fd;
   void accept_cb (ev::io &w, int revents); ev::io accept_ev;
+  unix_listener (int fd);
-  unix_listener (const char *sockname);
+  static int open (const char *sockname);
 };
-unix_listener::unix_listener (const char *sockname)
+unix_listener::unix_listener (int fd) : fd (fd)
 {
   accept_ev.set<unix_listener, &unix_listener::accept_cb> (this);
-  sockaddr_un sa;
-  if (strlen (sockname) >= sizeof(sa.sun_path))
-    {
-      fputs ("socket name too long, aborting.\n", stderr);
-      exit (EXIT_FAILURE);
-    }
-  if ((fd = socket (AF_UNIX, SOCK_STREAM, 0)) < 0)
-    {
-      perror ("unable to create listening socket");
-      exit (EXIT_FAILURE);
-    }
   fcntl (fd, F_SETFD, FD_CLOEXEC);
   fcntl (fd, F_SETFL, O_NONBLOCK);
+  accept_ev.start (fd, ev::READ);
+}
+int unix_listener::open (const char *sockname)
+{
+  sockaddr_un sa;
+  if (strlen (sockname) >= sizeof(sa.sun_path))
+    {
+      fputs ("socket name too long, aborting.\n", stderr);
+      exit (EXIT_FAILURE);
+    }
+  int fd;
+  if ((fd = socket (AF_UNIX, SOCK_STREAM, 0)) < 0)
+    {
+      perror ("unable to create listening socket");
+      exit (EXIT_FAILURE);
+    }
   sa.sun_family = AF_UNIX;
   strcpy (sa.sun_path, sockname);
   unlink (sockname);
     {
       perror ("unable to listen on socket");
       exit (EXIT_FAILURE);
     }
-  accept_ev.start (fd, ev::READ);
+  return fd;
 }
 void unix_listener::accept_cb (ev::io &w, int revents)
 {
   int fd2 = accept (fd, 0, 0);
               term->destroy ();
             send ("END"); send (success ? 1 : 0);
           }
         }
+      else if (!strcmp (tok, "QUIT"))
+        _exit (0);
       else
         return err ("protocol error: request '%s' unsupported.\n", (char *)tok);
     }
   else
     return err ();
 }
+#ifdef ENABLE_FRILLS
+enum {
+  SD_LISTEN_FDS_START = 3,
+};
+// https://www.freedesktop.org/software/systemd/man/sd_listen_fds.html
+// https://github.com/systemd/systemd/blob/main/src/libsystemd/sd-daemon/sd-daemon.c
+static int get_listen_fds ()
+{
+  const char *listen_pid = getenv ("LISTEN_PID");
+  if (!listen_pid)
+    return 0;
+  char *end;
+  errno = 0;
+  long pid = strtol (listen_pid, &end, 10);
+  if (errno || end == listen_pid || *end)
+    return -1;
+  if (getpid () != pid)
+    return 0;
+  const char *listen_fds = getenv ("LISTEN_FDS");
+  if (!listen_fds)
+    return 0;
+  errno = 0;
+  long n = strtol (listen_fds, &end, 10);
+  if (errno || end == listen_fds || *end)
+    return -1;
+  if (n <= 0 || n > INT_MAX - SD_LISTEN_FDS_START)
+    return -1;
+  unsetenv ("LISTEN_PID");
+  unsetenv ("LISTEN_FDS");
+  unsetenv ("LISTEN_FDNAMES");
+  return n;
+}
+#endif
 int
 main (int argc, char *argv[])
 {
   ptytty::init ();
   // optionally open display and never release it.
   if (opt_opendisplay)
     if (const char *dpy = getenv ("DISPLAY"))
       displays.get (dpy ? dpy : ":0"); // move string logic into rxvt_display maybe?
+  int fd;
+#ifdef ENABLE_FRILLS
+  int n = get_listen_fds ();
+  if (n > 1)
+    {
+      fputs ("received multiple file descriptors, aborting.\n", stderr);
+      exit (EXIT_FAILURE);
+    }
+  else if (n == 1)
+    {
+      fd = SD_LISTEN_FDS_START;
+      if (!opt_quiet)
+        {
+          printf ("rxvt-unicode daemon listening on fd.\n");
+          fflush (stdout);
+        }
+    }
+  else
+#endif
+    {
-  char *sockname = rxvt_connection::unix_sockname ();
+      char *sockname = rxvt_connection::unix_sockname ();
+      fd = unix_listener::open (sockname);
+      if (!opt_quiet)
+        {
+          printf ("rxvt-unicode daemon listening on %s.\n", sockname);
+          fflush (stdout);
+        }
+      free (sockname);
+    }
-  unix_listener l (sockname);
+  unix_listener l (fd);
   chdir ("/");
-  if (!opt_quiet)
-    {
-      printf ("rxvt-unicode daemon listening on %s.\n", sockname);
-      fflush (stdout);
-    }
-  free (sockname);
   pid_t pid = 0;
   if (opt_fork)
     {
       pid = fork ();

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing rxvt-unicode/src/rxvtd.C (file contents): Revision 1.60 by root, Tue Jun 12 10:45:53 2012 UTC vs. Revision 1.63 by sf-exg, Sun May 9 05:53:12 2021 UTC

Diff Legend

Comparing rxvt-unicode/src/rxvtd.C (file contents):
Revision 1.60 by root, Tue Jun 12 10:45:53 2012 UTC vs.
Revision 1.63 by sf-exg, Sun May 9 05:53:12 2021 UTC