ucw/main-test.c

   1 /*
   2  *      UCW Library -- Main Loop: Testing
   3  *
   4  *      (c) 2004--2011 Martin Mares <mj@ucw.cz>
   5  *
   6  *      This software may be freely distributed and used according to the terms
   7  *      of the GNU Lesser General Public License.
   8  */
   9
  10 #include "ucw/lib.h"
  11 #include "ucw/mainloop.h"
  12
  13 #include <stdio.h>
  14 #include <stdlib.h>
  15 #include <unistd.h>
  16
  17 #ifdef TEST
  18
  19 static struct main_process mp;
  20 static struct main_block_io fin, fout;
  21 static struct main_hook hook;
  22 static struct main_timer tm;
  23 static struct main_signal sg;
  24 static int sig_counter;
  25
  26 static byte rb[16];
  27
  28 static void dread(struct main_block_io *bio)
  29 {
  30   if (bio->rpos < bio->rlen)
  31     {
  32       msg(L_INFO, "Read EOF");
  33       block_io_del(bio);
  34     }
  35   else
  36     {
  37       msg(L_INFO, "Read done");
  38       block_io_read(bio, rb, sizeof(rb));
  39     }
  40 }
  41
  42 static void derror(struct main_block_io *bio, int cause)
  43 {
  44   msg(L_INFO, "Error: %m !!! (cause %d)", cause);
  45   block_io_del(bio);
  46 }
  47
  48 static void dwrite(struct main_block_io *bio UNUSED)
  49 {
  50   msg(L_INFO, "Write done");
  51 }
  52
  53 static int dhook(struct main_hook *ho UNUSED)
  54 {
  55   msg(L_INFO, "Hook called");
  56   if (sig_counter >= 3)
  57     return HOOK_SHUTDOWN;
  58   return 0;
  59 }
  60
  61 static void dtimer(struct main_timer *tm)
  62 {
  63   msg(L_INFO, "Timer tick");
  64   timer_add_rel(tm, 11000);
  65   timer_add_rel(tm, 10000);
  66 }
  67
  68 static void dentry(void)
  69 {
  70   log_fork();
  71   main_teardown();
  72   msg(L_INFO, "*** SUBPROCESS START ***");
  73   sleep(2);
  74   msg(L_INFO, "*** SUBPROCESS FINISH ***");
  75   exit(0);
  76 }
  77
  78 static void dexit(struct main_process *pr)
  79 {
  80   msg(L_INFO, "Subprocess %d exited with status %x", pr->pid, pr->status);
  81 }
  82
  83 static void dsignal(struct main_signal *sg UNUSED)
  84 {
  85   msg(L_INFO, "SIGINT received (send 3 times to really quit, or use Ctrl-\\)");
  86   sig_counter++;
  87 }
  88
  89 int
  90 main(void)
  91 {
  92   log_init(NULL);
  93   main_init();
  94
  95   fin.read_done = dread;
  96   fin.error_handler = derror;
  97   block_io_add(&fin, 0);
  98   block_io_read(&fin, rb, sizeof(rb));
  99
 100   fout.write_done = dwrite;
 101   fout.error_handler = derror;
 102   block_io_add(&fout, 1);
 103   block_io_write(&fout, "Hello, world!\n", 14);
 104
 105   hook.handler = dhook;
 106   hook_add(&hook);
 107
 108   tm.handler = dtimer;
 109   timer_add_rel(&tm,  1000);
 110
 111   sg.signum = SIGINT;
 112   sg.handler = dsignal;
 113   signal_add(&sg);
 114
 115   mp.handler = dexit;
 116   if (!process_fork(&mp))
 117     dentry();
 118
 119   main_debug();
 120
 121   main_loop();
 122   msg(L_INFO, "Finished.");
 123
 124   block_io_del(&fin);
 125   block_io_del(&fout);
 126   hook_del(&hook);
 127   signal_del(&sg);
 128   timer_del(&tm);
 129   main_cleanup();
 130   return 0;
 131 }
 132
 133 #endif