1/* $OpenBSD: accept.c,v 1.7 2016/05/23 19:11:42 renato Exp $ */ 2 3/* 4 * Copyright (c) 2012 Claudio Jeker <claudio@openbsd.org> 5 * 6 * Permission to use, copy, modify, and distribute this software for any 7 * purpose with or without fee is hereby granted, provided that the above 8 * copyright notice and this permission notice appear in all copies. 9 * 10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES 11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF 12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR 13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF 16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 17 */ 18 19#include <sys/types.h> 20#include <stdlib.h> 21 22#include "ldpd.h" 23#include "ldpe.h" 24#include "log.h" 25 26struct accept_ev { 27 LIST_ENTRY(accept_ev) entry; 28 struct event ev; 29 void (*accept_cb)(int, short, void *); 30 void *arg; 31 int fd; 32}; 33 34struct { 35 LIST_HEAD(, accept_ev) queue; 36 struct event evt; 37} accept_queue; 38 39static void accept_arm(void); 40static void accept_unarm(void); 41static void accept_cb(int, short, void *); 42static void accept_timeout(int, short, void *); 43 44void 45accept_init(void) 46{ 47 LIST_INIT(&accept_queue.queue); 48 evtimer_set(&accept_queue.evt, accept_timeout, NULL); 49} 50 51int 52accept_add(int fd, void (*cb)(int, short, void *), void *arg) 53{ 54 struct accept_ev *av; 55 56 if ((av = calloc(1, sizeof(*av))) == NULL) 57 return (-1); 58 av->fd = fd; 59 av->accept_cb = cb; 60 av->arg = arg; 61 LIST_INSERT_HEAD(&accept_queue.queue, av, entry); 62 63 event_set(&av->ev, av->fd, EV_READ, accept_cb, av); 64 event_add(&av->ev, NULL); 65 66 log_debug("%s: accepting on fd %d", __func__, fd); 67 68 return (0); 69} 70 71void 72accept_del(int fd) 73{ 74 struct accept_ev *av; 75 76 LIST_FOREACH(av, &accept_queue.queue, entry) 77 if (av->fd == fd) { 78 log_debug("%s: %d removed from queue", __func__, fd); 79 event_del(&av->ev); 80 LIST_REMOVE(av, entry); 81 free(av); 82 return; 83 } 84} 85 86void 87accept_pause(void) 88{ 89 struct timeval evtpause = { 1, 0 }; 90 91 log_debug(__func__); 92 accept_unarm(); 93 evtimer_add(&accept_queue.evt, &evtpause); 94} 95 96void 97accept_unpause(void) 98{ 99 if (evtimer_pending(&accept_queue.evt, NULL)) { 100 log_debug(__func__); 101 evtimer_del(&accept_queue.evt); 102 accept_arm(); 103 } 104} 105 106static void 107accept_arm(void) 108{ 109 struct accept_ev *av; 110 LIST_FOREACH(av, &accept_queue.queue, entry) 111 event_add(&av->ev, NULL); 112} 113 114static void 115accept_unarm(void) 116{ 117 struct accept_ev *av; 118 LIST_FOREACH(av, &accept_queue.queue, entry) 119 event_del(&av->ev); 120} 121 122static void 123accept_cb(int fd, short event, void *arg) 124{ 125 struct accept_ev *av = arg; 126 event_add(&av->ev, NULL); 127 av->accept_cb(fd, event, av->arg); 128} 129 130static void 131accept_timeout(int fd, short event, void *bula) 132{ 133 log_debug(__func__); 134 accept_arm(); 135} 136