Provide a `--pidfile' option in `fw'.

[fwd] / fw.c

diff --git a/fw.c b/fw.c
index 5a5db0f..85e6573 100644 (file)
--- a/fw.c
+++ b/fw.c
@@ -1,6 +1,6 @@
 /* -*-c-*-
  *
- * $Id: fw.c,v 1.15 2003/11/25 14:46:50 mdw Exp $
+ * $Id$
  *
  * Port forwarding thingy
  *
@@ -26,60 +26,6 @@
  * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  */
 
-/*----- Revision history --------------------------------------------------* 
- *
- * $Log: fw.c,v $
- * Revision 1.15  2003/11/25 14:46:50  mdw
- * Update docco for new options.
- *
- * Revision 1.14  2003/01/24 20:12:40  mdw
- * Correctly cast uid and gid sentinel values.
- *
- * Revision 1.13  2002/02/22 23:45:20  mdw
- * Add option to change the listen(2) parameter.  Receive `fw'-specific
- * code from `conf.c'.
- *
- * Revision 1.12  2002/01/13 14:49:17  mdw
- * Track @dstr_vputf@ change.
- *
- * Revision 1.11  2001/02/03 20:33:26  mdw
- * Fix flags to be unsigned.
- *
- * Revision 1.10  2001/02/03 20:30:03  mdw
- * Support re-reading config files on SIGHUP.
- *
- * Revision 1.9  2001/01/20 11:55:17  mdw
- * Handle select errors more robustly.
- *
- * Revision 1.8  2000/03/23 23:19:19  mdw
- * Fix changed options in parser table.
- *
- * Revision 1.7  2000/03/23 00:37:33  mdw
- * Add option to change user and group after initialization.  Naughtily
- * reassign short equivalents of --grammar and --options.
- *
- * Revision 1.6  1999/12/22 15:44:10  mdw
- * Make syslog a separate option, and do it better.
- *
- * Revision 1.5  1999/10/22 22:47:50  mdw
- * Grammar changes.  Also, don't enable SIGINT if it's currently ignored.
- *
- * Revision 1.4  1999/10/10 16:46:12  mdw
- * New resolver to initialize.  Also, include options for grammar and
- * options references.
- *
- * Revision 1.3  1999/07/26 23:30:42  mdw
- * Major reconstruction work for new design.
- *
- * Revision 1.2  1999/07/03 13:55:17  mdw
- * Various changes.  Add configuration grammar to help text.  Change to
- * root directory and open syslog when forking into background.
- *
- * Revision 1.1.1.1  1999/07/01 08:56:23  mdw
- * Initial revision.
- *
- */
-
 /*----- Header files ------------------------------------------------------*/
 
 #include "config.h"
@@ -115,6 +61,7 @@
 #include "fattr.h"
 #include "file.h"
 #include "fw.h"
+#include "privconn.h"
 #include "scan.h"
 #include "socket.h"
 #include "source.h"
@@ -247,6 +194,8 @@ void parse(scanner *sc)
       /* --- Combine the source and target --- */
 
       s->ops->attach(s, sc, t);
+      if (t->ops->confirm)
+       t->ops->confirm(t);
     }
 
     /* --- Include configuration from a file --- *
@@ -537,7 +486,7 @@ File source and target\n\
        FILE ::= `file' [`.'] FSPEC [`,' FSPEC]\n\
        FSPEC ::= FD-SPEC | NAME-SPEC | NULL-SPEC\n\
        FD-SPEC ::= [[`:']`fd'[`:']] NUMBER|`stdin'|`stdout'\n\
-       NAME-SPEC ::= [[`:']`file'[`:']] FILE-NAME\n\
+       NAME-SPEC ::= [[`:']`name'[`:']] FILE-NAME\n\
        FILE-NAME ::= PATH-SEQ | [ PATH-SEQ ]\n\
        PATH-SEQ ::= PATH-ELT | PATH-SEQ PATH-ELT\n\
        PATH-ELT ::= `/' | WORD\n\
@@ -604,6 +553,7 @@ Socket options\n\
        socket.inet.source.[allow|deny] priv-port\n\
        socket.inet.source.addr [=] any|ADDR\n\
        socket.inet.dest.addr [=] any|ADDR\n\
+       socket.inet.dest.priv-port [=] yes|no\n\
 \n\
        socket.unix.fattr.*\n\
 ");
@@ -627,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
@@ -670,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' },
@@ -682,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;
@@ -725,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;
@@ -805,6 +760,8 @@ int main(int argc, char *argv[])
 
   /* --- Drop privileges --- */
 
+  if (drop != (uid_t)-1)
+    privconn_split(sel);
 #ifdef HAVE_SETGROUPS
   if ((dropg != (gid_t)-1 && (setgid(dropg) || setgroups(1, &dropg))) ||
       (drop != (uid_t)-1 && setuid(drop)))
@@ -834,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;