misc/fsm/states.c

   1 #include "types.h"
   2 #include "states.h"
   3
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6 #include <string.h>
   7
   8 /* Callback functions prototypes */
   9 static unsigned int state_hashf(const void *key);
  10 static int state_cmpf(const void *arg1, const void *arg2);
  11 static void state_printf(const void *key, const void *data);
  12
  13 stret_t state_init(state_t **state, size_t size, unsigned int factor)
  14 {
  15     /* Allocate memory state's event table */
  16     if ((*state = malloc(sizeof **state)) == NULL)
  17         return ST_NOMEM;
  18
  19     if (((*state)->evttable = malloc(sizeof *(*state)->evttable)) == NULL) {
  20         free(*state);
  21         return ST_NOMEM;
  22     }
  23
  24     if (htable_init((*state)->evttable, size, factor,
  25                     state_hashf, state_cmpf, state_printf) == HT_NOMEM) {
  26         free((*state)->evttable);
  27         free(*state);
  28         return ST_NOMEM;
  29     }
  30
  31     /* Allocate memory for state's key */
  32     if (((*state)->st_key = malloc(sizeof *(*state)->st_key)) == NULL) {
  33         free((*state)->evttable);
  34         free(*state);
  35         return ST_NOMEM;
  36     }
  37
  38     return ST_OK;
  39 }
  40
  41 stret_t state_add_evt(state_t *state, unsigned int key, const char *desc, void (*actionf)(void *data), state_t *newstate)
  42 {
  43     event_t *pevt;
  44     unsigned int *pkey;
  45
  46     /* Allocate memory for new key */
  47     if ((pkey = malloc(sizeof *pkey)) == NULL)
  48         return ST_NOMEM;
  49
  50     *pkey = key;
  51
  52     /* Allocate memory for new event */
  53     if ((pevt = malloc(sizeof *pevt)) == NULL) {
  54         free(pkey);
  55         return ST_NOMEM;
  56     }
  57
  58     /* Fill in structure's members */
  59     strncpy(pevt->evt_desc, desc, MAX_EVT_DESC);
  60     pevt->evt_actionf = actionf;
  61     pevt->evt_newstate = newstate;
  62
  63     /* Insert event to hash table */
  64     if (htable_insert(state->evttable, pkey, pevt) == HT_EXISTS) {
  65         free(pkey);
  66         free(pevt);
  67         return ST_EXISTS;
  68     }
  69
  70     return ST_OK;
  71 }
  72
  73 stret_t state_rem_evt(state_t *state, unsigned int key)
  74 {
  75     if (htable_free_obj(state->evttable, &key, HT_FREEKEY | HT_FREEDATA) == HT_NOTFOUND)
  76         return ST_NOTFOUND;
  77
  78     return ST_OK;
  79 }
  80
  81 unsigned int state_get_key(state_t *state)
  82 {
  83     return *state->st_key;
  84 }
  85
  86 stret_t state_free(state_t *state)
  87 {
  88     htable_free_all_obj(state->evttable, HT_FREEKEY | HT_FREEDATA);
  89     htable_free(state->evttable);
  90     free(state->evttable);
  91     free(state);
  92
  93     return ST_OK;
  94 }
  95
  96 void state_print_evts(const state_t *state)
  97 {
  98     htable_print(state->evttable);
  99 }
 100
 101 /* Callback funtions */
 102 static unsigned int state_hashf(const void *key)
 103 {
 104     return *(const unsigned int *)key;
 105 }
 106
 107 static int state_cmpf(const void *arg1, const void *arg2)
 108 {
 109     unsigned int a = *(const unsigned int *)arg1;
 110     unsigned int b = *(const unsigned int *)arg2;
 111
 112     if (a > b)
 113         return -1;
 114     else if (a == b)
 115         return 0;
 116     else
 117         return 1;
 118 }
 119
 120 static void state_printf(const void *key, const void *data)
 121 {
 122     printf("key: %u\tdesc: %s ",
 123            *(const unsigned int *)key,
 124            ((const event_t *)data)->evt_desc);
 125 }