src/event.c

   1 /*
   2     event.c -- event queue
   3     Copyright (C) 2002-2009 Guus Sliepen <guus@tinc-vpn.org>,
   4                   2002-2005 Ivo Timmermans
   5
   6     This program is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     This program is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License along
  17     with this program; if not, write to the Free Software Foundation, Inc.,
  18     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 */
  20
  21 #include "system.h"
  22
  23 #include "avl_tree.h"
  24 #include "event.h"
  25 #include "utils.h"
  26 #include "xalloc.h"
  27
  28 avl_tree_t *event_tree;
  29 extern time_t now;
  30
  31 static int id;
  32
  33 static int event_compare(const event_t *a, const event_t *b) {
  34         if(a->time > b->time)
  35                 return 1;
  36
  37         if(a->time < b->time)
  38                 return -1;
  39
  40         return a->id - b->id;
  41 }
  42
  43 void init_events(void) {
  44         event_tree = avl_alloc_tree((avl_compare_t) event_compare, (avl_action_t) free_event);
  45 }
  46
  47 void exit_events(void) {
  48         avl_delete_tree(event_tree);
  49 }
  50
  51 void expire_events(void) {
  52         avl_node_t *node;
  53         event_t *event;
  54         time_t diff;
  55
  56         /*
  57          * Make all events appear expired by substracting the difference between
  58          * the expiration time of the last event and the current time.
  59          */
  60
  61         if(!event_tree->tail)
  62                 return;
  63
  64         event = event_tree->tail->data;
  65         if(event->time <= now)
  66                 return;
  67
  68         diff = event->time - now;
  69
  70         for(node = event_tree->head; node; node = node->next) {
  71                 event = node->data;
  72                 event->time -= diff;
  73         }
  74 }
  75
  76 event_t *new_event(void) {
  77         return xmalloc_and_zero(sizeof(event_t));
  78 }
  79
  80 void free_event(event_t *event) {
  81         free(event);
  82 }
  83
  84 void event_add(event_t *event) {
  85         event->id = ++id;
  86         avl_insert(event_tree, event);
  87 }
  88
  89 void event_del(event_t *event) {
  90         avl_delete(event_tree, event);
  91 }
  92
  93 event_t *get_expired_event(void) {
  94         event_t *event;
  95
  96         if(event_tree->head) {
  97                 event = event_tree->head->data;
  98
  99                 if(event->time <= now) {
 100                         avl_node_t *node = event_tree->head;
 101                         avl_unlink_node(event_tree, node);
 102                         free(node);
 103                         return event;
 104                 }
 105         }
 106
 107         return NULL;
 108 }
 109
 110 event_t *peek_next_event(void) {
 111         if (event_tree->head)
 112                 return event_tree->head->data;
 113         return NULL;
 114 }