X-Git-Url: https://git.distorted.org.uk/~mdw/fwd/blobdiff_plain/b69a615b311fb4081d97ccee6cd13777bcf4d5ca..4166ea7c41cac762e5e318567a4f993d8442d0a7:/fw.c

diff --git a/fw.c b/fw.c
index bf3e477..85e6573 100644
--- a/fw.c
+++ b/fw.c
@@ -577,6 +577,7 @@ int main(int argc, char *argv[])
   scanner sc;
   uid_t drop = -1;
   gid_t dropg = -1;
+  const char *pidfile = 0;
   conffile *cf, **cff = &conffiles;
 
 #define f_bogus 1u
@@ -620,6 +621,7 @@ int main(int argc, char *argv[])
       { "file",		OPTF_ARGREQ,	0,	'f' },
       { "fork",		0,		0,	'd' },
       { "daemon",	0,		0,	'd' },
+      { "pidfile",	OPTF_ARGREQ,	0,	'p' },
       { "syslog",	0,		0,	'l' },
       { "log",		0,		0,	'l' },
       { "quiet",	0,		0,	'q' },
@@ -632,7 +634,7 @@ int main(int argc, char *argv[])
 
       { 0,		0,		0,	0 }
     };
-    int i = mdwopt(argc, argv, "+hvu GO f:dls:g:", opts, 0, 0, 0);
+    int i = mdwopt(argc, argv, "+hvu" "GO" "f:dp:ls:g:", opts, 0, 0, 0);
 
     if (i < 0)
       break;
@@ -675,6 +677,9 @@ int main(int argc, char *argv[])
       case 'd':
 	f |= f_fork;
 	break;
+      case 'p':
+	pidfile = optarg;
+	break;
       case 'l':
 	f |= f_syslog;
 	break;
@@ -786,6 +791,18 @@ int main(int argc, char *argv[])
     if (kid != 0)
       _exit(0);
   }
+  if (pidfile) {
+    FILE *fp = fopen(pidfile, "w");
+    if (!fp) {
+      die(EXIT_FAILURE, "couldn't create pidfile `%s': %s",
+	  pidfile, strerror(errno));
+    }
+    fprintf(fp, "%lu\n", (unsigned long)getpid());
+    if (fflush(fp) || ferror(fp) || fclose(fp)) {
+      die(EXIT_FAILURE, "couldn't write pidfile `%s': %s",
+	  pidfile, strerror(errno));
+    }
+  }
 
   if (f & f_syslog) {
     flags |= FW_SYSLOG;