src/IOHandler.h

   1 /* IOHandler.h - IOMultiplexer
   2  * Copyright (C) 2012  Philipp Kreil (pk910)
   3  *
   4  * This program is free software: you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation, either version 3 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program. If not, see <http://www.gnu.org/licenses/>.
  16  */
  17 #ifndef _IOHandler_h
  18 #define _IOHandler_h
  19 #include <stddef.h>
  20 #include <sys/time.h> /* struct timeval */
  21
  22 #define IO_READ_BUFLEN 1024
  23 #define IO_MAX_TIMEOUT 10
  24 #define IO_LINE_LEN    1024
  25
  26 struct timeval;
  27 struct IODescriptor;
  28 struct IOEvent;
  29 struct IOSSLNode;
  30
  31 enum IOLogType {
  32     IOLOG_DEBUG,
  33     IOLOG_WARNING,
  34     IOLOG_ERROR,
  35     IOLOG_FATAL
  36 };
  37
  38 #define IOHANDLER_CALLBACK(NAME) void NAME(struct IOEvent *event)
  39 typedef IOHANDLER_CALLBACK(iohandler_callback);
  40
  41 #define IOHANDLER_LOG_BACKEND(NAME) void NAME(enum IOLogType type, const char *line)
  42 typedef IOHANDLER_LOG_BACKEND(iohandler_log_callback);
  43
  44 extern iohandler_log_callback *iolog_backend;
  45
  46 enum IOType {
  47     IOTYPE_UNKNOWN, /* ignore descriptor (uninitialized) */
  48     IOTYPE_SERVER, /* server socket */
  49     IOTYPE_CLIENT, /* client socket */
  50     IOTYPE_STDIN, /* stdin */
  51     IOTYPE_TIMER /* timer */
  52 };
  53
  54 enum IOStatus {
  55     IO_CLOSED, /* descriptor is dead (socket waiting for removal or timer) */
  56     IO_LISTENING, /* descriptor is waiting for connections (server socket) */
  57     IO_CONNECTING, /* descriptor is waiting for connection approval (connecting client socket) */
  58     IO_CONNECTED, /* descriptor is connected (connected client socket) */
  59     IO_SSLWAIT /* waiting for SSL backend (e.g. handshake) */
  60 };
  61
  62 enum IOEventType {
  63     IOEVENT_IGNORE,
  64     IOEVENT_READABLE, /* socket is readable - not read anything yet, could also be disconnect notification */
  65     IOEVENT_RECV, /* client socket received something (recv_str valid) */
  66     IOEVENT_CONNECTED, /* client socket connected successful */
  67     IOEVENT_NOTCONNECTED, /* client socket could not connect (errid valid) */
  68     IOEVENT_CLOSED, /* client socket lost connection (errid valid) */
  69     IOEVENT_ACCEPT, /* server socket accepted new connection (accept_fd valid) */
  70     IOEVENT_SSLACCEPT, /* SSL server socket accepted new connection (accept_iofd valid) */
  71     IOEVENT_TIMEOUT, /* timer timed out */
  72     IOEVENT_SSLFAILED /* failed to initialize SSL session */
  73 };
  74
  75 struct IOBuffer {
  76     char *buffer;
  77     size_t bufpos, buflen;
  78 };
  79
  80 struct IODescriptor {
  81     int fd;
  82     enum IOType type;
  83     enum IOStatus state;
  84     struct timeval timeout;
  85     iohandler_callback *callback;
  86     struct IOBuffer readbuf;
  87     struct IOBuffer writebuf;
  88     void *data;
  89     int read_lines : 1;
  90     int ssl : 1;
  91     int ssl_server_hs : 1;
  92     int ssl_active : 1;
  93     int ssl_hs_read : 1;
  94     int ssl_hs_write : 1;
  95     struct IOSSLNode *sslnode;
  96
  97     struct IODescriptor *next, *prev;
  98 };
  99
 100 struct IOEvent {
 101     enum IOEventType type;
 102     struct IODescriptor *iofd;
 103     union {
 104         char *recv_str;
 105         int accept_fd;
 106         int errid;
 107         struct IODescriptor *accept_iofd;
 108     } data;
 109 };
 110
 111 #define IOHANDLER_LISTEN_IPV4 0x01
 112 #define IOHANDLER_LISTEN_IPV6 0x02 /* overrides IOHANDLER_LISTEN_IPV4 */
 113
 114 #define IOHANDLER_CONNECT_IPV4 0x01
 115 #define IOHANDLER_CONNECT_IPV6 0x02 /* overrides IOHANDLER_CONNECT_IPV4 */
 116
 117 #define IOHANDLER_SETTING_HIGH_PRECISION_TIMER 0x01
 118
 119 void iohandler_set(int setting, int value);
 120
 121 struct IODescriptor *iohandler_add(int sockfd, enum IOType type, struct timeval *timeout, iohandler_callback *callback);
 122 struct IODescriptor *iohandler_timer(struct timeval timeout, iohandler_callback *callback);
 123 struct IODescriptor *iohandler_connect(const char *hostname, unsigned int port, int ssl, const char *bind, iohandler_callback *callback);
 124 struct IODescriptor *iohandler_connect_flags(const char *hostname, unsigned int port, int ssl, const char *bindhost, iohandler_callback *callback, int flags);
 125 struct IODescriptor *iohandler_listen(const char *hostname, unsigned int port, iohandler_callback *callback);
 126 struct IODescriptor *iohandler_listen_flags(const char *hostname, unsigned int port, iohandler_callback *callback, int flags);
 127 struct IODescriptor *iohandler_listen_ssl(const char *hostname, unsigned int port, const char *certfile, const char *keyfile, iohandler_callback *callback);
 128 struct IODescriptor *iohandler_listen_ssl_flags(const char *hostname, unsigned int port, const char *certfile, const char *keyfile, iohandler_callback *callback, int flags);
 129 void iohandler_write(struct IODescriptor *iofd, const char *line);
 130 void iohandler_send(struct IODescriptor *iofd, const char *data, size_t datalen);
 131 void iohandler_printf(struct IODescriptor *iofd, const char *text, ...);
 132 void iohandler_close(struct IODescriptor *iofd);
 133 void iohandler_update(struct IODescriptor *iofd);
 134 void iohandler_set_timeout(struct IODescriptor *iofd, struct timeval *timeout);
 135
 136 void iohandler_poll();
 137 void iohandler_poll_timeout(struct timeval timeout);
 138
 139 #endif