Add chroot capability

CMakeLists.txt

 
 option(GENERATE_SRS_SECRET "Generate a random SRS secret if none exists during install" ON)
 
+set(CHROOT_DIR "${CMAKE_INSTALL_PREFIX}/lib/${PROJECT_NAME}" CACHE PATH "Chroot jail for daemon")
+
 find_program(HELP2MAN help2man DOC "path to help2man executable")
 find_program(DD dd DOC "path to dd executable")
 find_program(BASE64 base64 DOC "path to base64 executable")

postinstall.cmake.in

 set(GENERATE_SRS_SECRET "@GENERATE_SRS_SECRET@")
+set(CHROOT_DIR "@CHROOT_DIR@")
 set(INIT_FLAVOR "@INIT_FLAVOR@")
 set(SECRET_FILE "@PROJECT_NAME@.secret")
 set(DD "@DD@")
 set(BASE64 "@BASE64@")
 set(INSSERV "@INSSERV@")
 
+if(CHROOT_DIR AND NOT EXISTS "$ENV{DESTDIR}${CHROOT_DIR}")
+	message(STATUS "Chroot jail: $ENV{DESTDIR}${CHROOT_DIR}")
+	file(MAKE_DIRECTORY "$ENV{DESTDIR}${CHROOT_DIR}")
+endif()
+
 if(INIT_FLAVOR AND NOT EXISTS "$ENV{DESTDIR}/etc/default/@PROJECT_NAME@")
 	file(INSTALL "@CMAKE_CURRENT_SOURCE_DIR@/@PROJECT_NAME@.default" DESTINATION "/etc/default" RENAME "@PROJECT_NAME@")
 endif()

postsrsd.c

     "   -f<port>       set port for the forward SRS lookup (default: 10001)\n"
     "   -r<port>       set port for the reverse SRS lookup (default: 10002)\n"
     "   -p<pidfile>    write process ID to pidfile (default: none)\n"
+    "   -c<dir>        chroot to <dir> (default: none)\n"
     "   -u<user>       switch user id after port bind (default: none)\n"
     "   -t<seconds>    timeout for idle client connections (default: 1800)\n"
     "   -D             fork into background\n"
   int opt, timeout = 1800, family = AF_UNSPEC;
   int daemonize = FALSE;
   char *forward_service = NULL, *reverse_service = NULL,
-       *user = NULL, *domain = NULL;
+       *user = NULL, *domain = NULL, *chroot_dir = NULL;
   int forward_sock, reverse_sock;
   char *secret_file = NULL, *pid_file = NULL;
   FILE *pf = NULL;
+  struct passwd *pwd = NULL;
   char secret[1024];
   char *tmp;
   srs_t *srs;
   tmp = strrchr(argv[0], '/');
   if (tmp) self = strdup(tmp + 1); else self = strdup(argv[0]);
 
-  while ((opt = getopt(argc, argv, "46d:f:r:s:u:t:p:Dhv")) != -1) {
+  while ((opt = getopt(argc, argv, "46d:f:r:s:u:t:p:c:Dhv")) != -1) {
     switch (opt) {
       case '?':
         return EXIT_FAILURE;
       case 'u':
         user = strdup(optarg);
         break;
+      case 'c':
+	chroot_dir = strdup(optarg);
+	break;
       case 'D':
         daemonize = TRUE;
         break;
 	return EXIT_SUCCESS;
     }
   }
+  if (domain == NULL) {
+    fprintf (stderr, "%s: You must set a home domain (-d)\n", self);
+    show_help();
+    return EXIT_FAILURE;
+  }
+
+  /* The stuff we do first may not be possible from within chroot or without privileges */
+
+  /* Open pid file for writing (the actual process ID is filled in later) */
   if (pid_file) {
     pf = fopen (pid_file, "w");
     if (pf == NULL) {
       return EXIT_FAILURE;
     }
   }
-  if (domain == NULL) {
-    fprintf (stderr, "%s: You must set a home domain (-d)\n", self);
-    show_help();
-    return EXIT_FAILURE;
-  }
+  /* Read secret. The default installation makes this root accessible only. */
   if (secret_file != NULL) {
     size_t len;
     FILE *fp = fopen(secret_file, "rb");
     show_help();
     return EXIT_FAILURE;
   }
+  /* Bind ports. May require privileges if the config specifies ports below 1024 */
   if (forward_service != NULL) {
     forward_sock = bind_service(forward_service, family);
     free (forward_service);
     reverse_sock = bind_service("10002", family);
   }
   if (reverse_sock < 0) return EXIT_FAILURE;
+
+  /* Open syslog now (NDELAY), because it may no longer reachable from chroot */
+  openlog (self, LOG_PID | LOG_NDELAY, LOG_MAIL);
+  /* We also have to lookup the uid of the unprivileged user for the same reason. */
   if (user) {
-    struct passwd *pwd;
     errno = 0;
     pwd = getpwnam(user);
     if (pwd == NULL) {
         fprintf (stderr, "%s: No such user: %s\n", self, user);
       return EXIT_FAILURE;
     }
+  }
+  /* Now we can chroot, which again requires root privileges */
+  if (chroot_dir) {
+    if (chdir(chroot_dir) < 0) {
+      fprintf (stderr, "%s: Cannot change to chroot: %s\n", self, strerror(errno));
+      return EXIT_FAILURE;
+    }
+    if (chroot(chroot_dir) < 0) {
+      fprintf (stderr, "%s: Failed to enable chroot: %s\n", self, strerror(errno));
+      return EXIT_FAILURE;
+    }
+  }
+  /* Finally, we revert to the unprivileged user */
+  if (pwd) {
     if (setuid(pwd->pw_uid) < 0) {
       fprintf (stderr, "%s: Failed to switch user id: %s\n", self, strerror(errno));
       return EXIT_FAILURE;
     }
   }
+  /* Standard double fork technique to disavow all knowledge about the controlling terminal */
   if (daemonize) {
     close(0); close(1); close(2);
     if (fork() != 0) return EXIT_SUCCESS;
     setsid();
     if (fork() != 0) return EXIT_SUCCESS;
   }
+  /* Make note of our actual process ID */
   if (pf) {
     fprintf (pf, "%d", (int)getpid());
     fclose (pf);
   }
 
-  openlog ("postsrsd", LOG_PID, LOG_MAIL);
   srs = srs_new();
   srs_add_secret (srs, secret);
   srs_set_separator (srs, '+');
     char keybuf[1024], *key;
 
     if (poll(fds, 2, 1000) < 0) {
-      if (!daemonize) fprintf (stderr, "%s: Poll failure: %s\n", self, strerror(errno));
+      if (daemonize)
+        syslog (LOG_MAIL | LOG_ERR, "Poll failure: %s", strerror(errno));
+      else
+        fprintf (stderr, "%s: Poll failure: %s\n", self, strerror(errno));
       return EXIT_FAILURE;
     }
     for (i = 0; i < 2; ++i) {

postsrsd.init.in

 SRS_REVERSE_PORT=10002
 SRS_SECRET=/etc/@PROJECT_NAME@.secret
 RUN_AS=nobody
+CHROOT=@CHROOT_DIR@
 
 # Read config file if it is present.
 if [ -r /etc/default/$NAME ]
     . /etc/default/$NAME
 fi
 
-POSTSRS_OPTS="-4 -f$SRS_FORWARD_PORT -r$SRS_REVERSE_PORT -d$SRS_DOMAIN -s$SRS_SECRET -u$RUN_AS -p$PIDFILE -D"
+POSTSRS_OPTS="-4 -f$SRS_FORWARD_PORT -r$SRS_REVERSE_PORT -d$SRS_DOMAIN -s$SRS_SECRET -u$RUN_AS -p$PIDFILE -c$CHROOT -D"
 
 test -r "$SRS_SECRET" -a -n "$SRS_DOMAIN" || exit 0
 
 esac
 
 exit $ret
+

postsrsd.upstart.in

 	SRS_REVERSE_PORT=10002
 	SRS_SECRET=/etc/@PROJECT_NAME@.secret
 	RUN_AS=nobody
+	CHROOT=@CHROOT_DIR@
 	if [ -r "$DEFAULTFILE" ]; then
 		. "$DEFAULTFILE"
 	fi
-	exec @CMAKE_INSTALL_PREFIX@/sbin/@POSTSRSD@ -4 -f$SRS_FORWARD_PORT -r$SRS_REVERSE_PORT -d$SRS_DOMAIN -s$SRS_SECRET -u$RUN_AS
+	exec @CMAKE_INSTALL_PREFIX@/sbin/@POSTSRSD@ -4 -f$SRS_FORWARD_PORT -r$SRS_REVERSE_PORT -d$SRS_DOMAIN -s$SRS_SECRET -u$RUN_AS -c$CHROOT
 end script
+