]> git.ozlabs.org Git - ccan-lca-2011.git/commitdiff
lca2011: dump on SIGHUP.
authorRusty Russell <rusty@rustcorp.com.au>
Tue, 25 Jan 2011 00:31:48 +0000 (11:01 +1030)
committerRusty Russell <rusty@rustcorp.com.au>
Tue, 25 Jan 2011 00:31:48 +0000 (11:01 +1030)
ccan/oserver/_info
ccan/oserver/oserver.c
ccan/oserver/oserver.h
ccan/oserver/oserver_types.h
ccan/oserver/test/run-one-client-disconnect.c
ccan/oserver/test/run.c

index 4f54687d7d677ed06533e2d55e57281795040126..9107e66323806facf8cdfde91320f9509ef68c27 100644 (file)
@@ -30,7 +30,7 @@
  *             if (argc != 1)
  *                     opt_log_stderr_exit("Unknown extra arguments");
  *
- *             if (!oserver_setup(ev, port))
+ *             if (!oserver_setup(ev, port, "/var/run/oserver/dump"))
  *                     err(1, "Failed to set up server");
  *
  *             while (tevent_loop_wait(ev) == 0);
index 0ec1b6840ee40ea1ae548e9e300590e44f7d1f3d..8a5fdceecf1a0c2c4f7ddf1ac278ad8c61f7e793 100644 (file)
@@ -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);
 
index a465b2d3bbba175d6382ebe60c2b795682eab77f..3342ba83e61850def0eb60109c068b501b2312f8 100644 (file)
@@ -7,6 +7,7 @@
  * oserver_setup - set up an oserver
  * @ev: tevent context to use.
  * @port: port to use (usually OSERVER_PORT)
+ * @dumpfile: file to save state to on SIGHUP (if non-NULL).
  *
  * Opens a socket and binds it to @port, then sets it up to listen
  * for connections.  talloc_free() the pointer returned to shut it down
  *     struct tevent_context *ev;
  *
  *     ev = tevent_context_init(NULL);
- *     oserver = oserver_setup(ev, OSERVER_PORT);
+ *     oserver = oserver_setup(ev, OSERVER_PORT, "oserver.dump");
  *     if (!oserver)
  *             err(1, "Failed to set up server");
  *
  *     while (tevent_loop_wait(ev) == 0);
  *     err(1, "Event loop failed");
  */
-struct oserver *oserver_setup(struct tevent_context *ev, unsigned short port);
+struct oserver *oserver_setup(struct tevent_context *ev, unsigned short port,
+                             const char *dumpfile);
 
 #define OSERVER_PORT 2828
 #endif /* CCAN_OSERVER_H */
index 72717e89733ddf1793fd514988f7976b8740e4bc..8e998923314112a4c033768c2771d9ffd9cd041c 100644 (file)
@@ -41,5 +41,6 @@ CDUMP_SAVED struct oserver {
        struct client *clients[5];
        int fd;
        struct tevent_fd *fde CDUMP_IGNORE;
+       const char *dumpfile;
 };
 #endif /* CCAN_OSERVER_TYPES_H */
index 8367aad68dc4876528ede746fec5418ae9fa61b0..6023f784b17a2fd97e01fa3d513e9f05abe7751c 100644 (file)
@@ -1,4 +1,5 @@
 #include <ccan/oserver/oserver.c>
+#include <ccan/oserver/oserver_cdump.c>
 #include <ccan/oserver/oserver.h>
 #include <ccan/str/str.h>
 #include <ccan/tap/tap.h>
@@ -104,7 +105,7 @@ int main(int argc, char *argv[])
        /* This is how many tests you plan to run */
        plan_tests(2);
 
-       oserver = oserver_setup(ev, OSERVER_PORT);
+       oserver = oserver_setup(ev, OSERVER_PORT, NULL);
        if (!oserver)
                err(1, "Failed to set up server");
 
index 03180c9bec8ccccb6bcc524562aa2134fe7eb4ff..a8eb6f24391042b064cddaa243fe2c3ac80c2762 100644 (file)
@@ -1,4 +1,5 @@
 #include <ccan/oserver/oserver.c>
+#include <ccan/oserver/oserver_cdump.c>
 #include <ccan/oserver/oserver.h>
 #include <ccan/str/str.h>
 #include <ccan/tap/tap.h>
@@ -19,7 +20,7 @@ static void run_server(int readyfd, int exitfd)
 {
        struct tevent_context *ev = tevent_context_init(NULL);
 
-       if (oserver_setup(ev, OSERVER_PORT) == NULL)
+       if (oserver_setup(ev, OSERVER_PORT, NULL) == NULL)
                exit(1);
 
        tevent_add_fd(ev, ev, exitfd, TEVENT_FD_READ, exit_quietly, NULL);