misc/fsm/test_thread.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <pthread.h>
   4
   5 #include "fsm.h"
   6 #include "states.h"
   7 #include "types.h"
   8
   9 #define NUM_THREADS 3
  10
  11 /* function prototypes */
  12 void *thread_alpha_gen(void *arg);
  13 void *thread_num_gen(void *arg);
  14 void *thread_consumer(void *arg);
  15 void dief(const char *s);
  16 void diep(const char *s);
  17 void print_char(void *data);
  18
  19 struct mythread {
  20     void *(*th_pfunc)(void *);
  21     pthread_t th_id;
  22 } thtbl[] = {
  23     { thread_alpha_gen, 0 },
  24     { thread_num_gen, 0 },
  25     { thread_consumer, 0 }
  26 };
  27
  28 int main(void)
  29 {
  30     fsm_t *fsm;
  31     state_t *steadystate;
  32     int c, i;
  33
  34     /* Initialize fsm */
  35     fsm_init(&fsm, 2<<10, 5, 2);
  36
  37     /* Initialize state */
  38     if (state_init(&steadystate, 2<<10, 2) == ST_NOMEM) {
  39         fsm_free(fsm);
  40         dief("state_init(): ST_NOMEM");
  41     }
  42
  43     /* Add events to state
  44      *
  45      * We have only one state named "steady state",
  46      * which handles all events from 'A' to 'Z'
  47      * and '0' to '9'.
  48      */
  49     for (c = 'A'; c < 'Z'; c++) {
  50         if (state_add_evt(steadystate, c, "", print_char, steadystate) == ST_NOMEM) {
  51             dief("state_add_evt(): ST_NOMEM");
  52             fsm_free(fsm);
  53         }
  54     }
  55     for (c = '0'; c < '9'; c++) {
  56         if (state_add_evt(steadystate, c, "", print_char, steadystate) == ST_NOMEM) {
  57             dief("state_add_evt(): ST_NOMEM");
  58             fsm_free(fsm);
  59         }
  60     }
  61
  62     /* Add steady state to fsm */
  63     fsm_add_state(fsm, 0, steadystate);
  64
  65     /* Set initial state */
  66     fsm_set_state(fsm, 0);
  67
  68     /* Create threads (alpha generator, num generator and consumer) */
  69     for (i = 0; i < NUM_THREADS; i++) {
  70         if (pthread_create(&thtbl[i].th_id, NULL, thtbl[i].th_pfunc, (void *)fsm))
  71             diep("pthread_create");
  72     }
  73
  74     for (;;)
  75         ;
  76
  77     /* Free memory */
  78     fsm_free(fsm);
  79
  80     return EXIT_SUCCESS;
  81 }
  82
  83 void *thread_alpha_gen(void *arg)
  84 {
  85     fsm_t *fsm;
  86     size_t c;
  87
  88     /* Get a pointer to the fsm we are interested in */
  89     fsm = (fsm_t *)arg;
  90
  91     /* Initialize random number generator */
  92     srand(time(NULL));
  93
  94     /* Broadcast events */
  95     for (;;) {
  96         c = 'A' + rand() % 26;    /* from 'A' to 'Z' */
  97         fsm_queue_event(fsm, c, &c, 1, 0);
  98     }
  99
 100     pthread_exit(NULL);
 101 }
 102
 103 void *thread_num_gen(void *arg)
 104 {
 105     fsm_t *fsm;
 106     size_t c;
 107
 108     /* Get a pointer to the fsm we are interested in */
 109     fsm = (fsm_t *)arg;
 110
 111     /* Initialize random number generator */
 112     srand(time(NULL));
 113
 114     /* Broadcast events */
 115     for (;;) {
 116         c = '0' + rand() % 10;    /* from '0' to '9' */
 117         fsm_queue_event(fsm, c, &c, 1, 0);
 118     }
 119
 120     pthread_exit(NULL);
 121 }
 122
 123 void *thread_consumer(void *arg)
 124 {
 125     fsm_t *fsm;
 126
 127     /* Get a pointer to the fsm we are interested in */
 128     fsm = (fsm_t *)arg;
 129
 130     /* Thread acts as event consumer */
 131     for (;;)
 132         fsm_dequeue_event(fsm);
 133
 134     pthread_exit(NULL);
 135 }
 136
 137 void dief(const char *s)
 138 {
 139     fprintf(stderr, "error: %s\n", s);
 140     exit(EXIT_FAILURE);
 141 }
 142
 143 void diep(const char *s)
 144 {
 145     perror(s);
 146     exit(EXIT_FAILURE);
 147 }
 148
 149 void print_char(void *data)
 150 {
 151     printf("%c", *(char *)data);
 152 }