opr: Add new softsig implementation
[openafs.git] / tests / opr / softsig-helper.c
diff --git a/tests/opr/softsig-helper.c b/tests/opr/softsig-helper.c
new file mode 100644 (file)
index 0000000..cffb997
--- /dev/null
@@ -0,0 +1,140 @@
+/*
+ * Copyright (c) 2012 Your File System Inc. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR `AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+/* A test helper for the softsig code. This just sits and reports what signals
+ * it has received.
+ */
+
+#include <afsconfig.h>
+#include <afs/param.h>
+
+#include <roken.h>
+#include <pthread.h>
+#include <sys/mman.h>
+
+#include <afs/opr.h>
+#include <opr/softsig.h>
+
+#define SIGTABLEENTRY(name)    { #name, SIG##name }
+
+static struct sigtable {
+       char *name;
+       int signo;
+} sigtable[] = {
+       SIGTABLEENTRY(INT),
+       SIGTABLEENTRY(HUP),
+       SIGTABLEENTRY(QUIT),
+       SIGTABLEENTRY(ALRM),
+       SIGTABLEENTRY(TERM),
+       SIGTABLEENTRY(TSTP),
+       SIGTABLEENTRY(USR1),
+       SIGTABLEENTRY(USR2)
+       };
+
+static char *signame(int signo) {
+    int i;
+
+    for (i = 0; i < sizeof(sizeof(sigtable) / sizeof(sigtable[0])); ++i) {
+       if (sigtable[i].signo == signo) {
+           return sigtable[i].name;
+       }
+    }
+
+    return "UNK";
+}
+
+static void
+handler(int signal) {
+    printf("Received %s\n", signame(signal));
+
+    fflush(stdout);
+}
+
+static void *
+mythread(void *arg) {
+    while (1) {
+       sleep(60);
+    }
+    return NULL;
+}
+
+static void *
+crashingThread(void *arg) {
+    *(char *)1 = 'a';  /* raises SIGSEGV */
+    return NULL; /* Ha! */
+}
+
+static void *
+thrownUnderTheBusThread(void *arg) {
+    char *path = arg;
+    int fd = open(path, O_CREAT | O_APPEND, 0660);
+    char *m = mmap(NULL, 10, PROT_WRITE, MAP_PRIVATE, fd, 0);
+    *(m + 11) = 'a';  /* raises SIGBUS */
+    return NULL;
+}
+
+
+int
+main(int argvc, char **argv)
+{
+    char *path;
+    int i;
+    int threads = 10;
+
+    opr_softsig_Init();
+
+    opr_Verity(opr_softsig_Register(SIGINT, handler) == 0);
+    opr_Verity(opr_softsig_Register(SIGHUP, handler) == 0);
+    opr_Verity(opr_softsig_Register(SIGQUIT, handler) == 0);
+    opr_Verity(opr_softsig_Register(SIGALRM, handler) == 0);
+    opr_Verity(opr_softsig_Register(SIGTERM, handler) == 0);
+    opr_Verity(opr_softsig_Register(SIGTSTP, handler) == 0);
+    opr_Verity(opr_softsig_Register(SIGUSR1, handler) == 0);
+    opr_Verity(opr_softsig_Register(SIGUSR2, handler) == 0);
+
+    for (i=0; i<threads; i++) {
+       pthread_t id;
+       opr_Verify(pthread_create(&id, NULL, mythread, NULL) == 0);
+    }
+
+    if (argvc>1 && strcmp(argv[1], "-crash") == 0) {
+       pthread_t id;
+       opr_Verify(pthread_create(&id, NULL, crashingThread, NULL) == 0);
+    } else if (argvc>1 && strcmp(argv[1], "-buserror") == 0) {
+       if (argvc > 2)
+           path = argv[2];
+       else
+           opr_abort();
+       pthread_t id;
+       opr_Verify(pthread_create(&id, NULL, thrownUnderTheBusThread,
+                                 path) == 0);
+    } else {
+       printf("Ready\n");
+       fflush(stdout);
+    }
+
+    mythread(NULL);
+
+    return 0;
+}