Changeset 2743

Timestamp:

Aug 14, 2007, 4:02:50 AM (15 years ago)

Author:

joshe

Message:

Add a pidfile command-line argument.

Location:

trunk/daemon

Files:

• daemon.c (modified) (13 diffs)
• transmission-daemon.1 (modified) (2 diffs)

trunk/daemon/daemon.c

@@ -51 +51 @@
 
 static void usage       ( const char *, ... );
-static void readargs    ( int, char **, int *, int *, char ** );
+static void readargs    ( int, char **, int *, int *, char **, char ** );
 static int  trylocksock ( const char * );
 static int  getlock     ( const char * );
@@ -58 +58 @@
 static void setupsigs   ( struct event_base * );
 static void gotsig      ( int, short, void * );
+static int  savepid     ( const char * );
 
 static int  gl_lockfd               = -1;
@@ -63 +64 @@
 static int  gl_sockfd               = -1;
 static char gl_sockpath[MAXPATHLEN] = "";
+static char gl_pidfile[MAXPATHLEN]  = "";
 
 int
@@ -69 +71 @@
     struct event_base * evbase;
     int                 nofork, debug, sockfd;
-    char              * sockpath;
+    char              * sockpath, * pidfile;
 
     setmyname( argv[0] );
-    readargs( argc, argv, &nofork, &debug, &sockpath );
+    readargs( argc, argv, &nofork, &debug, &sockpath, &pidfile );
 
     if( !nofork )
@@ -84 +86 @@
     }
 
+    if( 0 > savepid( pidfile ) )
+        exit( 1 );
     atexit( exitcleanup );
     sockfd = trylocksock( sockpath );
@@ -119 +123 @@
 
     printf(
-  "usage: %s [-dfh]\n"
+  "usage: %s [-dfh] [-p file] [-s file]\n"
   "\n"
   "Transmission %s http://transmission.m0k.org/\n"
@@ -127 +131 @@
   "  -f --foreground           Run in the foreground and log to stderr\n"
   "  -h --help                 Display this message and exit\n"
+  "  -p --pidfile <path>       Save the process id in a file at <path>\n"
   "  -s --socket <path>        Place the socket file at <path>\n"
   "\n"
@@ -135 +140 @@
 
 void
-readargs( int argc, char ** argv, int * nofork, int * debug, char ** sock )
-{
-    char optstr[] = "dfhs:";
+readargs( int argc, char ** argv, int * nofork, int * debug, char ** sock,
+          char ** pidfile )
+{
+    char optstr[] = "dfhp:s:";
     struct option longopts[] =
     {
@@ -143 +149 @@
         { "foreground",         no_argument,       NULL, 'f' },
         { "help",               no_argument,       NULL, 'h' },
+        { "pidfile",            required_argument, NULL, 'p' },
         { "socket",             required_argument, NULL, 's' },
         { NULL, 0, NULL, 0 }
@@ -148 +155 @@
     int opt;
 
-    *nofork = 0;
-    *debug  = 0;
-    *sock   = NULL;
+    *nofork    = 0;
+    *debug     = 0;
+    *sock      = NULL;
+    *pidfile   = NULL;
 
     while( 0 <= ( opt = getopt_long( argc, argv, optstr, longopts, NULL ) ) )
@@ -161 +169 @@
             case 'f':
                 *nofork = 1;
+                break;
+            case 'p':
+                *pidfile = optarg;
                 break;
             case 's':
@@ -295 +306 @@
         close( gl_lockfd );
     }
+    if( 0 != gl_pidfile[0] )
+    {
+        unlink( gl_pidfile );
+    }
 }
 
@@ -339 +354 @@
     }
 }
+
+int
+savepid( const char * file )
+{
+    FILE * pid;
+
+    pid = fopen( file, "wb" );
+    if( NULL == pid )
+    {
+        errnomsg( "failed to open pid file: %s", file );
+        return -1;
+    }
+
+    if( 0 > fprintf( pid, "%d\n", (int) getpid() ) )
+    {
+        errnomsg( "failed to write pid to file: %s", file );
+        fclose( pid );
+        unlink( file );
+        return -1;
+    }
+
+    fclose( pid );
+    strlcpy( gl_pidfile, file, sizeof gl_pidfile );
+
+    return 0;
+}

trunk/daemon/transmission-daemon.1

@@ -33 +33 @@
 .Nm
 .Op Fl df
+.Op Fl p Ar pid-file
 .Op Fl s Ar socket-file
 .Ek
@@ -50 +51 @@
 .It Fl h Fl -help
 Print command-line option descriptions.
+.It Fl p Fl -pidfile Ar pid-file
+Save the daemon's process ID in
+.Ar pid-file .
+.It Fl s Fl -socket Ar socket-file
+Create the unix-domain socket file at
+.Ar socket-file
+instead of the platform-dependent default location.
 .El
 .Sh FILES