From: Pierre Habouzit <madcoder@debian.org>
Date: Thu, 30 Aug 2007 22:00:31 +0000 (+0200)
Subject: Fix typo, and use the privileges drop.
X-Git-Url: http://git.madism.org/?a=commitdiff_plain;h=52b485c0f261b6b178a35b4737b3f625126b5c74;p=apps%2Fpfixtools.git

Fix typo, and use the privileges drop.

Signed-off-by: Pierre Habouzit <madcoder@debian.org>
---

diff --git a/daemon.c b/daemon.c
index 1c74ed3..ebda967 100644
--- a/daemon.c
+++ b/daemon.c
@@ -152,7 +152,7 @@ int daemon_detach(void)
     return 0;
 }
 
-int drop_privilegies(const char *user, const char *group)
+int drop_privileges(const char *user, const char *group)
 {
     if (!geteuid()) {
         struct passwd *pw;
diff --git a/daemon.h b/daemon.h
index 5bc2cfb..5734e8f 100644
--- a/daemon.h
+++ b/daemon.h
@@ -40,6 +40,6 @@ int tcp_listen_nonblock(const struct sockaddr *addr, socklen_t len);
 int accept_nonblock(int fd);
 
 int daemon_detach(void);
-int drop_privilegies(const char *user, const char *group);
+int drop_privileges(const char *user, const char *group);
 
 #endif
diff --git a/srsd.c b/srsd.c
index a09befb..0958bfd 100644
--- a/srsd.c
+++ b/srsd.c
@@ -48,6 +48,9 @@
 #define DAEMON_NAME             "pfix-srsd"
 #define DEFAULT_ENCODER_PORT    10000
 #define DEFAULT_DECODER_PORT    10001
+#define RUNAS_USER              "nobody"
+#define RUNAS_GROUP             "nogroup"
+
 #define __tostr(x)  #x
 #define STR(x)      __tostr(x)
 
@@ -221,6 +224,7 @@ void usage(void)
           "    -d <port>    port to listen to for decoding requests\n"
           "                 (default: "STR(DEFAULT_DECODER_PORT)")\n"
           "    -p <pidfile> file to write our pid to\n"
+          "    -u           unsafe mode: don't drop privilegies\n"
          , stderr);
 }
 
@@ -375,6 +379,7 @@ static srs_t *srs_read_secrets(const char *sfile)
 
 int main(int argc, char *argv[])
 {
+    bool unsafe  = false;
     int port_enc = DEFAULT_ENCODER_PORT;
     int port_dec = DEFAULT_DECODER_PORT;
     const char *pidfile = NULL;
@@ -389,7 +394,7 @@ int main(int argc, char *argv[])
     }
     common_initialize();
 
-    for (int c = 0; (c = getopt(argc, argv, "he:d:p:")) >= 0; ) {
+    for (int c = 0; (c = getopt(argc, argv, "he:d:p:u")) >= 0; ) {
         switch (c) {
           case 'e':
             port_enc = atoi(optarg);
@@ -400,6 +405,9 @@ int main(int argc, char *argv[])
           case 'p':
             pidfile = optarg;
             break;
+          case 'u':
+            unsafe = true;
+            break;
           default:
             usage();
             return EXIT_FAILURE;
@@ -424,20 +432,29 @@ int main(int argc, char *argv[])
         fprintf(f, "%d\n", getpid());
         fflush(f);
     }
+
+    if (!unsafe && drop_privileges(RUNAS_USER, RUNAS_GROUP) < 0) {
+        syslog(LOG_CRIT, "unable to drop privileges");
+        return EXIT_FAILURE;
+    }
+
     if (daemon_detach() < 0) {
         syslog(LOG_CRIT, "unable to fork");
         return EXIT_FAILURE;
     }
+
     if (f) {
         rewind(f);
         ftruncate(fileno(f), 0);
         fprintf(f, "%d\n", getpid());
-        fclose(f);
-        f = NULL;
+        fflush(f);
     }
     res = main_loop(srs, argv[optind], port_enc, port_dec);
-    if (pidfile) {
-        unlink(pidfile);
+    if (f) {
+        rewind(f);
+        ftruncate(fileno(f), 0);
+        fclose(f);
+        f = NULL;
     }
     return res;
 }