source3/lib/events.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Timed event library.
   4    Copyright (C) Andrew Tridgell 1992-1998
   5    Copyright (C) Volker Lendecke 2005
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #include "includes.h"
  22 #include <tevent_internal.h>
  23
  24 /*
  25  * Return if there's something in the queue
  26  */
  27
  28 bool event_add_to_select_args(struct tevent_context *ev,
  29                               const struct timeval *now,
  30                               fd_set *read_fds, fd_set *write_fds,
  31                               struct timeval *timeout, int *maxfd)
  32 {
  33         struct tevent_fd *fde;
  34         struct timeval diff;
  35         bool ret = false;
  36
  37         for (fde = ev->fd_events; fde; fde = fde->next) {
  38                 if (fde->flags & EVENT_FD_READ) {
  39                         FD_SET(fde->fd, read_fds);
  40                         ret = true;
  41                 }
  42                 if (fde->flags & EVENT_FD_WRITE) {
  43                         FD_SET(fde->fd, write_fds);
  44                         ret = true;
  45                 }
  46
  47                 if ((fde->flags & (EVENT_FD_READ|EVENT_FD_WRITE))
  48                     && (fde->fd > *maxfd)) {
  49                         *maxfd = fde->fd;
  50                 }
  51         }
  52
  53         if (ev->immediate_events != NULL) {
  54                 *timeout = timeval_zero();
  55                 return true;
  56         }
  57
  58         if (ev->timer_events == NULL) {
  59                 return ret;
  60         }
  61
  62         diff = timeval_until(now, &ev->timer_events->next_event);
  63         *timeout = timeval_min(timeout, &diff);
  64
  65         return true;
  66 }
  67
  68 bool run_events(struct tevent_context *ev,
  69                 int selrtn, fd_set *read_fds, fd_set *write_fds)
  70 {
  71         struct tevent_fd *fde;
  72         struct timeval now;
  73
  74         if (ev->signal_events &&
  75             tevent_common_check_signal(ev)) {
  76                 return true;
  77         }
  78
  79         if (ev->immediate_events &&
  80             tevent_common_loop_immediate(ev)) {
  81                 return true;
  82         }
  83
  84         GetTimeOfDay(&now);
  85
  86         if ((ev->timer_events != NULL)
  87             && (timeval_compare(&now, &ev->timer_events->next_event) >= 0)) {
  88                 /* this older events system did not auto-free timed
  89                    events on running them, and had a race condition
  90                    where the event could be called twice if the
  91                    talloc_free of the te happened after the callback
  92                    made a call which invoked the event loop. To avoid
  93                    this while still allowing old code which frees the
  94                    te, we need to create a temporary context which
  95                    will be used to ensure the te is freed. We also
  96                    remove the te from the timed event list before we
  97                    call the handler, to ensure we can't loop */
  98
  99                 struct tevent_timer *te = ev->timer_events;
 100                 TALLOC_CTX *tmp_ctx = talloc_new(ev);
 101
 102                 DEBUG(10, ("Running timed event \"%s\" %p\n",
 103                            ev->timer_events->handler_name, ev->timer_events));
 104
 105                 DLIST_REMOVE(ev->timer_events, te);
 106                 talloc_steal(tmp_ctx, te);
 107
 108                 te->handler(ev, te, now, te->private_data);
 109
 110                 talloc_free(tmp_ctx);
 111                 return true;
 112         }
 113
 114         if (selrtn == 0) {
 115                 /*
 116                  * No fd ready
 117                  */
 118                 return false;
 119         }
 120
 121         for (fde = ev->fd_events; fde; fde = fde->next) {
 122                 uint16 flags = 0;
 123
 124                 if (FD_ISSET(fde->fd, read_fds)) flags |= EVENT_FD_READ;
 125                 if (FD_ISSET(fde->fd, write_fds)) flags |= EVENT_FD_WRITE;
 126
 127                 if (flags & fde->flags) {
 128                         fde->handler(ev, fde, flags, fde->private_data);
 129                         return true;
 130                 }
 131         }
 132
 133         return false;
 134 }
 135
 136
 137 struct timeval *get_timed_events_timeout(struct tevent_context *ev,
 138                                          struct timeval *to_ret)
 139 {
 140         struct timeval now;
 141
 142         if ((ev->timer_events == NULL) && (ev->immediate_events == NULL)) {
 143                 return NULL;
 144         }
 145         if (ev->immediate_events != NULL) {
 146                 *to_ret = timeval_zero();
 147                 return to_ret;
 148         }
 149
 150         now = timeval_current();
 151         *to_ret = timeval_until(&now, &ev->timer_events->next_event);
 152
 153         DEBUG(10, ("timed_events_timeout: %d/%d\n", (int)to_ret->tv_sec,
 154                 (int)to_ret->tv_usec));
 155
 156         return to_ret;
 157 }
 158
 159 static int s3_event_loop_once(struct tevent_context *ev, const char *location)
 160 {
 161         struct timeval now, to;
 162         fd_set r_fds, w_fds;
 163         int maxfd = 0;
 164         int ret;
 165
 166         FD_ZERO(&r_fds);
 167         FD_ZERO(&w_fds);
 168
 169         to.tv_sec = 9999;       /* Max timeout */
 170         to.tv_usec = 0;
 171
 172         if (run_events(ev, 0, NULL, NULL)) {
 173                 return 0;
 174         }
 175
 176         GetTimeOfDay(&now);
 177
 178         if (!event_add_to_select_args(ev, &now, &r_fds, &w_fds, &to, &maxfd)) {
 179                 return -1;
 180         }
 181
 182         ret = sys_select(maxfd+1, &r_fds, &w_fds, NULL, &to);
 183
 184         if (ret == -1 && errno != EINTR) {
 185                 tevent_debug(ev, TEVENT_DEBUG_FATAL,
 186                              "sys_select() failed: %d:%s\n",
 187                              errno, strerror(errno));
 188                 return -1;
 189         }
 190
 191         run_events(ev, ret, &r_fds, &w_fds);
 192         return 0;
 193 }
 194
 195 void event_context_reinit(struct tevent_context *ev)
 196 {
 197         tevent_common_context_destructor(ev);
 198         return;
 199 }
 200
 201 static int s3_event_context_init(struct tevent_context *ev)
 202 {
 203         return 0;
 204 }
 205
 206 void dump_event_list(struct tevent_context *ev)
 207 {
 208         struct tevent_timer *te;
 209         struct tevent_fd *fe;
 210         struct timeval evt, now;
 211
 212         if (!ev) {
 213                 return;
 214         }
 215
 216         now = timeval_current();
 217
 218         DEBUG(10,("dump_event_list:\n"));
 219
 220         for (te = ev->timer_events; te; te = te->next) {
 221
 222                 evt = timeval_until(&now, &te->next_event);
 223
 224                 DEBUGADD(10,("Timed Event \"%s\" %p handled in %d seconds (at %s)\n",
 225                            te->handler_name,
 226                            te,
 227                            (int)evt.tv_sec,
 228                            http_timestring(talloc_tos(), te->next_event.tv_sec)));
 229         }
 230
 231         for (fe = ev->fd_events; fe; fe = fe->next) {
 232
 233                 DEBUGADD(10,("FD Event %d %p, flags: 0x%04x\n",
 234                            fe->fd,
 235                            fe,
 236                            fe->flags));
 237         }
 238 }
 239
 240 static const struct tevent_ops s3_event_ops = {
 241         .context_init           = s3_event_context_init,
 242         .add_fd                 = tevent_common_add_fd,
 243         .set_fd_close_fn        = tevent_common_fd_set_close_fn,
 244         .get_fd_flags           = tevent_common_fd_get_flags,
 245         .set_fd_flags           = tevent_common_fd_set_flags,
 246         .add_timer              = tevent_common_add_timer,
 247         .schedule_immediate     = tevent_common_schedule_immediate,
 248         .add_signal             = tevent_common_add_signal,
 249         .loop_once              = s3_event_loop_once,
 250         .loop_wait              = tevent_common_loop_wait,
 251 };
 252
 253 static bool s3_tevent_init(void)
 254 {
 255         static bool initialized;
 256         if (initialized) {
 257                 return true;
 258         }
 259         initialized = tevent_register_backend("s3", &s3_event_ops);
 260         tevent_set_default_backend("s3");
 261         return initialized;
 262 }
 263
 264 /*
 265   this is used to catch debug messages from events
 266 */
 267 static void s3_event_debug(void *context, enum tevent_debug_level level,
 268                            const char *fmt, va_list ap)  PRINTF_ATTRIBUTE(3,0);
 269
 270 static void s3_event_debug(void *context, enum tevent_debug_level level,
 271                            const char *fmt, va_list ap)
 272 {
 273         int samba_level = -1;
 274         char *s = NULL;
 275         switch (level) {
 276         case TEVENT_DEBUG_FATAL:
 277                 samba_level = 0;
 278                 break;
 279         case TEVENT_DEBUG_ERROR:
 280                 samba_level = 1;
 281                 break;
 282         case TEVENT_DEBUG_WARNING:
 283                 samba_level = 2;
 284                 break;
 285         case TEVENT_DEBUG_TRACE:
 286                 samba_level = 11;
 287                 break;
 288
 289         };
 290         if (vasprintf(&s, fmt, ap) == -1) {
 291                 return;
 292         }
 293         DEBUG(samba_level, ("s3_event: %s", s));
 294         free(s);
 295 }
 296
 297 struct tevent_context *s3_tevent_context_init(TALLOC_CTX *mem_ctx)
 298 {
 299         struct tevent_context *ev;
 300
 301         s3_tevent_init();
 302
 303         ev = tevent_context_init_byname(mem_ctx, "s3");
 304         if (ev) {
 305                 tevent_set_debug(ev, s3_event_debug, NULL);
 306         }
 307
 308         return ev;
 309 }
 310