Permit multiple-argumented commands.

[inetdmx.git] / inetdmx.c

diff --git a/inetdmx.c b/inetdmx.c
index ebab3bf..13efc7f 100644 (file)
--- a/inetdmx.c
+++ b/inetdmx.c
@@ -72,7 +72,7 @@ static int opt_syslog;
 static int opt_stderr;
 static const char *opt_lock;
 static int opt_ignore_spawned_command_output;
-static const char *opt_command;
+static char *opt_command;
 
 
 static void fatal(const char *fmt,...) G_GNUC_PRINTF(1,2) G_GNUC_NORETURN;
@@ -135,6 +135,15 @@ int rc;
        return string;
 }
 
+static void *xmalloc(size_t size)
+{
+void *r;
+
+       if ((r=malloc(size)))
+               return r;
+       fatal("Error allocing %lu bytes",(unsigned long)size);
+}
+
 static void usage(void)
 {
        fprintf(stderr,"\
@@ -276,13 +285,6 @@ int fdi,fdo;
        pollfdi_name[0]=conn0_name;
        pollfdi_name[1]=conn1_name;
        for (;;) {
-               for (fdi=0;fdi<G_N_ELEMENTS(pollfdi);fdi++)
-                       if (pollfdi[fdi].events)
-                               break;
-               if (fdi>=G_N_ELEMENTS(pollfdi)) {
-                       lock_close();
-                       exit(EXIT_SUCCESS);
-                       }
                if (0>=poll(pollfdi,G_N_ELEMENTS(pollfdi),-1))
                        fatal("poll(%s socket,%s socket): %m",pollfdi_name[0],pollfdi_name[1]);
                for (fdi=0;fdi<G_N_ELEMENTS(pollfdi);fdi++)
@@ -307,8 +309,8 @@ int fdi,fdo;
                                        fatal("read(%s socket): %m",pollfdi_name[fdi]);
                                        }
                                if (got==0) {
-                                       pollfdi[fdi].events&=~POLLIN;
-                                       break;
+                                       lock_close();
+                                       exit(EXIT_SUCCESS);
                                        }
                                for (fdo=0;fdo<G_N_ELEMENTS(pollfdi);fdo++) {
                                        if (fdi==fdo)
@@ -486,6 +488,9 @@ int main(int argc,char **argv) G_GNUC_NORETURN;
 int main(int argc,char **argv)
 {
 char optc;
+size_t opt_command_len;
+int i;
+char *s;
 
        if ((program_name=strrchr(argv[0],'/')))
                program_name++;
@@ -556,9 +561,21 @@ char *endptr;
 
        if (optind>=argc)
                fatal("<start-server-command/stop-server-command> is a required argument");
-       if (optind+1<argc)
-               fatal("Too many arguments, <start-server-command/stop-server-command> may need quoting");
-       opt_command=argv[optind];
+       opt_command_len=0;
+       for (i=optind;i<argc;i++)
+               opt_command_len+=strlen(argv[i])+1;
+       opt_command=xmalloc(opt_command_len);
+       s=opt_command;
+       for (i=optind;i<argc;i++) {
+size_t argv_i_len=strlen(argv[i]);
+
+               if (s>opt_command)
+                       *s++=' ';
+               memcpy(s,argv[i],argv_i_len);
+               s+=argv_i_len;
+               }
+       *s++=0;
+       assert(s==opt_command+opt_command_len);
 
        if (!opt_syslog && !opt_stderr)
                opt_stderr=1;