state.c

   1 #include "state.h"
   2 #include "hash.h"
   3 #include "utils.h"
   4 #include <stdlib.h>
   5
   6 #define HOST_STATES_HASH_BUCKETS 4096
   7 #define SERVICE_STATES_HASH_BUCKETS (HOST_STATES_HASH_BUCKETS * 4)
   8
   9 /*
  10  * The hooks are called from broker.c in Nagios.
  11  */
  12 static hash_table *host_states;
  13 static hash_table *svc_states;
  14
  15 int state_init(void)
  16 {
  17         host_states = hash_init(HOST_STATES_HASH_BUCKETS);
  18         if (!host_states)
  19                 return -1;
  20
  21         svc_states = hash_init(SERVICE_STATES_HASH_BUCKETS);
  22         if (!svc_states) {
  23                 free(host_states);
  24                 host_states = NULL;
  25                 return -1;
  26         }
  27
  28         return prime_initial_states(host_states, svc_states);
  29 }
  30
  31 static inline int has_state_change(int *old, int state, int type)
  32 {
  33         /*
  34          * A state change is considered to consist of a change
  35          * to either state_type or state, so we OR the two
  36          * together to form a complete state. This will make
  37          * the module log as follows:
  38          *    service foo;poo is HARD OK initially
  39          *    service foo;poo goes to SOFT WARN, attempt 1   (logged)
  40          *    service foo;poo goes to SOFT WARN, attempt 2   (not logged)
  41          *    service foo;poo goes to HARD WARN              (logged)
  42          */
  43         state = CAT_STATE(state, type);
  44
  45         if (*old == state)
  46                 return 0;
  47
  48         *old = state;
  49         return 1;
  50 }
  51
  52 int host_has_new_state(nebstruct_host_check_data *ds)
  53 {
  54         int *old_state;
  55
  56         old_state = hash_find(host_states, ds->host_name);
  57         if (!old_state) {
  58                 int *state;
  59
  60                 state = malloc(sizeof(*state));
  61                 *state = CAT_STATE(ds->state, ds->state_type);
  62                 hash_add(host_states, ds->host_name, state);
  63                 return 1;
  64         }
  65
  66         return has_state_change(old_state, ds->state, ds->state_type);
  67 }
  68
  69 int service_has_new_state(nebstruct_service_check_data *ds)
  70 {
  71         int *old_state;
  72
  73         old_state = hash_find2(svc_states, ds->host_name, ds->service_description);
  74         if (!old_state) {
  75                 int *state;
  76
  77                 state = malloc(sizeof(*state));
  78                 *state = CAT_STATE(ds->state, ds->state_type);
  79                 hash_add2(svc_states, ds->host_name, ds->service_description, state);
  80                 return 1;
  81         }
  82
  83         return has_state_change(old_state, ds->state, ds->state_type);
  84 }