lca2011: dump on SIGHUP.

[ccan-lca-2011.git] / ccan / oserver / oserver.c

diff --git a/ccan/oserver/oserver.c b/ccan/oserver/oserver.c
index 0ec1b6840ee40ea1ae548e9e300590e44f7d1f3d..8a5fdceecf1a0c2c4f7ddf1ac278ad8c61f7e793 100644 (file)
--- a/ccan/oserver/oserver.c
+++ b/ccan/oserver/oserver.c
@@ -1,5 +1,6 @@
 #include <ccan/oserver/oserver.h>
 #include <ccan/oserver/oserver_types.h>
+#include <ccan/oserver/oserver_cdump.h>
 #include <ccan/read_write_all/read_write_all.h>
 #include <ccan/opt/opt.h>
 #include <ccan/tevent/tevent.h>
@@ -17,6 +18,9 @@
 #include <errno.h>
 #include <signal.h>
 #include <assert.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
 
 static uint16_t state_flag_map[] = {
        [SENDING_GREETING]              = TEVENT_FD_WRITE,
@@ -284,7 +288,26 @@ static void talloc_dump(struct tevent_context *ev,
        }
 }
 
-struct oserver *oserver_setup(struct tevent_context *ev, unsigned short port)
+static void dump(struct tevent_context *ev,
+                struct tevent_signal *se,
+                int signum,
+                int count,
+                void *siginfo,
+                void *_oserver)
+{
+       struct oserver *oserver = _oserver;
+       char *str;
+       int fd;
+
+       str = cdump_bundle(ev, cdump_struct_oserver, oserver);
+       fd = open(oserver->dumpfile, O_CREAT|O_TRUNC|O_WRONLY, 0600);
+       write(fd, str, strlen(str));
+       close(fd);
+       talloc_free(str);
+}
+
+struct oserver *oserver_setup(struct tevent_context *ev, unsigned short port,
+                             const char *dumpfile)
 {
        struct oserver *oserver;
        int one = 1;
@@ -327,6 +350,12 @@ struct oserver *oserver_setup(struct tevent_context *ev, unsigned short port)
                return NULL;
        }
 
+       oserver->dumpfile = dumpfile;
+       if (oserver->dumpfile) {
+               tevent_add_signal(ev, oserver, SIGHUP, SA_RESTART,
+                                 dump, oserver);
+       }
+
        /* Don't kill us if client dies. */
        signal(SIGPIPE, SIG_IGN);