chiark / gitweb /
Signal handling integrated into I/O system.
[mLib] / sig.c
diff --git a/sig.c b/sig.c
new file mode 100644 (file)
index 0000000..1bfddbf
--- /dev/null
+++ b/sig.c
@@ -0,0 +1,270 @@
+/* -*-c-*-
+ *
+ * $Id: sig.c,v 1.1 1999/07/26 23:16:26 mdw Exp $
+ *
+ * Signal handling
+ *
+ * (c) 1999 Straylight/Edgeware
+ */
+
+/*----- Licensing notice --------------------------------------------------* 
+ *
+ * This file is part of the mLib utilities library.
+ *
+ * mLib is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Library General Public License as
+ * published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version.
+ * 
+ * mLib is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Library General Public License for more details.
+ * 
+ * You should have received a copy of the GNU Library General Public
+ * License along with mLib; if not, write to the Free
+ * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
+ * MA 02111-1307, USA.
+ */
+
+/*----- Revision history --------------------------------------------------* 
+ *
+ * $Log: sig.c,v $
+ * Revision 1.1  1999/07/26 23:16:26  mdw
+ * Signal handling integrated into I/O system.
+ *
+ */
+
+/*----- Header files ------------------------------------------------------*/
+
+#include <errno.h>
+#include <signal.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include <fcntl.h>
+
+#include "alloc.h"
+#include "fdflags.h"
+#include "report.h"
+#include "sel.h"
+#include "sig.h"
+
+/*----- Data structures ---------------------------------------------------*/
+
+typedef struct sigstate {
+  struct sigaction sa;
+  sig *list;
+} sigstate;
+
+/*----- Static variables --------------------------------------------------*/
+
+static sel_file sigsel;
+static int sigfd;
+static sigstate *sigs;
+
+/*----- Main code ---------------------------------------------------------*/
+
+/* --- @sig_handler@ --- *
+ *
+ * Arguments:  @int n@ = signal number
+ *
+ * Returns:    ---
+ *
+ * Use:                Generic signal handler.  Just writes a single byte to the
+ *             signal pipe.  The byte contains the signal number.
+ */
+
+static void sig_handler(int n)
+{
+  unsigned char sch = (unsigned char)n;
+  write(sigfd, &sch, 1);
+}
+
+/* --- @sig_read@ --- *
+ *
+ * Arguments:  @int fd@ = file descriptor to read
+ *             @unsigned mode@ = thing to do with descriptor
+ *             @void *p@ = uninteresting argument
+ *
+ * Returns:    ---
+ *
+ * Use:                Dispatches signals to their handlers safely.
+ */
+
+static void sig_read(int fd, unsigned mode, void *p)
+{
+  sigset_t ss;
+  unsigned char buf[256];
+  int r;
+
+  sigemptyset(&ss);
+  while ((r = read(fd, buf, sizeof(buf))) > 0) {
+    unsigned char *p, *q;
+    for (p = buf, q = buf + r; p < q; p++) {
+      sig *s;
+      int n = *p;
+      if (sigismember(&ss, n))
+       continue;
+      sigaddset(&ss, n);
+      s = sigs[n].list;
+      while (s) {
+       sig *ss = s;
+       s = s->next;
+       ss->proc(n, ss->p);
+      }
+    }
+  }
+}  
+
+/* --- @sig_add@ --- *
+ *
+ * Arguments:  @sig *s@ = pointer to signal handler block
+ *             @int n@ = number of the signal
+ *             @void (*proc)(int n, void *p)@ = signal handler function
+ *             @void *p@ = argument to pass to handler
+ *
+ * Returns:    ---
+ *
+ * Use:                Adds a signal handler.
+ */
+
+void sig_add(sig *s, int n, void (*proc)(int /*n*/, void */*p*/), void *p)
+{
+  sigstate *ss = &sigs[n];
+
+  /* --- Initialize the block --- */
+
+  s->proc = proc;
+  s->p = p;
+  s->sig = n;
+  s->next = ss->list;
+  s->prev = 0;
+
+  /* --- Engage a signal handler, maybe --- */
+
+  if (!ss->list) {
+    struct sigaction sa;
+    sa.sa_handler = sig_handler;
+    sa.sa_flags = SA_NOCLDSTOP;
+#ifdef SA_RESTART
+    sa.sa_flags |= SA_RESTART;
+#endif
+    sigemptyset(&sa.sa_mask);
+    sigaction(n, &sa, &ss->sa);
+  }
+
+  /* --- Link the block into the list --- */
+
+  if (ss->list)
+    ss->list->prev = s;
+  ss->list = s;
+}
+
+/* --- @sig_remove@ --- *
+ *
+ * Arguments:  @sig *s@ = pointer to signal handler block
+ *
+ * Returns:    ---
+ *
+ * Use:                Removes the signal handler from the list.
+ */
+
+void sig_remove(sig *s)
+{
+  sigstate *ss = &sigs[s->sig];
+
+  /* --- Unlink the handler block --- */
+
+  if (s->next)
+    s->next->prev = s->prev;
+  if (s->prev)
+    s->prev->next = s->next;
+  else
+    ss->list = s->next;
+
+  /* --- Maybe remove the handler --- */
+
+  if (!ss->list)
+    sigaction(s->sig, &ss->sa, 0);
+}
+
+/* --- @sig_init@ --- *
+ *
+ * Arguments:  @sel_state *s@ = pointer to select state
+ *
+ * Returns:    ---
+ *
+ * Use:                Initializes the signal handling system ready for use.
+ */
+
+void sig_init(sel_state *s)
+{
+  int fd[2];
+  unsigned nsig;
+
+  /* --- Work out how many signals there are --- */
+
+  {
+    sigset_t ss;
+    unsigned min = 0, max = 0;
+
+    sigemptyset(&ss);
+
+    /* --- Get a good upper bound --- *
+     *
+     * Cap the search at 256.  I'll be sending signal numbers as bytes.
+     */
+
+    nsig = 1;
+    while (sigaddset(&ss, nsig) == 0) {
+      if (nsig == 256)
+       goto counted;
+      nsig <<= 1;
+    }
+
+    /* --- Now binary search until I find the actual limit --- */
+
+    min = nsig >> 1;
+    max = nsig;
+    for (;;) {
+      nsig = (min + max) >> 1;
+      if (nsig == min)
+       break;
+      if (sigaddset(&ss, nsig))
+       max = nsig;
+      else
+       min = nsig;
+    }
+  counted:;
+  }
+
+  /* --- Initialize the signal state table --- */
+
+  {
+    unsigned i;
+
+    sigs = xmalloc(nsig * sizeof(*sigs));
+    for (i = 0; i < nsig; i++)
+      sigs[i].list = 0;
+  }
+
+  /* --- Create the signal pipe --- */
+
+  if (pipe(fd))
+    die(1, "couldn't create pipe for signal handling");
+
+  /* --- Set both ends to nonblocking and close-on-exec --- */
+
+  fdflags(fd[0], O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
+  fdflags(fd[1], O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
+
+  /* --- Set everything up for the future --- */
+    
+  sigfd = fd[1];
+  sel_initfile(s, &sigsel, fd[0], SEL_READ, sig_read, 0);
+  sel_addfile(&sigsel);
+}
+
+/*----- That's all, folks -------------------------------------------------*/