1

#define _GNU_SOURCE

     2

#include <signal.h>

     3

#include <string.h>

     4

#include <stdlib.h>

     5

#include <assert.h>

     6

     7

#include "signals.h"

     8

#include "common.h"

     9

    10

struct signals {

    11

    struct event_base *ev_base;

    12

    13

    struct signal {

    14

        struct event *ev;

    15

    } sig_list[MAX_SIGNALS];

    16

    17

    int sig_count;

    18

};

    19

    20

void signals_loopexit (int signal, short event, void *arg) {

    21

    struct signals *signals = arg;

    22

    23

    INFO("[signal] caught %s: exiting the event loop", strsignal(signal));

    24

    25

    if (event_base_loopexit(signals->ev_base, NULL))

    26

        FATAL("event_base_loopexit");

    27

}

    28

    29

void signals_ignore (int signal, short event, void *arg) {

    30

    struct signals *signals = arg;

    31

    32

    (void) signals;

    33

    34

    /* ignore */

    35

}

    36

    37

struct signals *signals_alloc (struct event_base *ev_base) {

    38

    struct signals *signals = NULL;

    39

    40

    if ((signals = calloc(1, sizeof(*signals))) == NULL)

    41

        ERROR("calloc");

    42

    43

    // simple attributes

    44

    signals->ev_base = ev_base;

    45

    46

    // done

    47

    return signals;

    48

    49

error:

    50

    return NULL;

    51

}

    52

    53

int signals_add (struct signals *signals, int sigval, void (*sig_handler)(evutil_socket_t, short, void *)) {

    54

    struct signal *sig_info;

    55

    56

    // find our sig_info

    57

    assert(signals->sig_count < MAX_SIGNALS);

    58

    sig_info = &signals->sig_list[signals->sig_count++];

    59

    60

    // set up the libevent signal events

    61

    if ((sig_info->ev = signal_new(signals->ev_base, sigval, sig_handler, signals)) == NULL)

    62

        PERROR("signal_new");

    63

    64

    if (signal_add(sig_info->ev, NULL))

    65

        PERROR("signal_add");

    66

    67

    // success

    68

    return 0;

    69

    70

error:

    71

    return -1;

    72

}

    73

    74

struct signals *signals_default (struct event_base *ev_base) {

    75

    struct signals *signals = NULL;

    76

    77

    // alloc signals

    78

    if ((signals = signals_alloc(ev_base)) == NULL)

    79

        return NULL;

    80

    81

    // add the set of default signals

    82

    if (    signals_add(signals,    SIGPIPE,    &signals_ignore)

    83

        ||  signals_add(signals,    SIGINT,     &signals_loopexit)

    84

    ) ERROR("signals_add");

    85

    86

    // success

    87

    return signals;

    88

    89

error:

    90

    if (signals)

    91

        signals_free(signals);

    92

    93

    return NULL;

    94

}   

    95

    96

void signals_free (struct signals *signals) {

    97

    int i;

    98

    99

    // free all events

   100

    for (i = 0; i < signals->sig_count; i++) {

   101

        if (signal_del(signals->sig_list[i].ev))

   102

            PWARNING("signal_del");

   103

   104

    }

   105

   106

    // free the info itself

   107

    free(signals);

   108

}

   109