src/IOHandler/IOSockets.h

   1 /* IOSockets.h - IOMultiplexer v2
   2  * Copyright (C) 2014  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 _IOSockets_h
  18 #define _IOSockets_h
  19 #include <sys/time.h>
  20 #include <stddef.h>
  21 #include "IODNSAddress.struct.h"
  22
  23 struct IOSocketBuffer {
  24     char *buffer;
  25     size_t bufpos, buflen;
  26 };
  27
  28 #ifndef _IOHandler_internals
  29 #include "IOHandler.h"
  30 #else
  31
  32 struct _IOSocket;
  33 struct IOSocket;
  34 struct IOSocketBuffer;
  35 struct IOSSLDescriptor;
  36 struct _IODNSQuery;
  37 struct IODNSEvent;
  38
  39 struct IOEngine {
  40     const char *name;
  41     int (*init)(void);
  42     void (*add)(struct _IOSocket *iosock);
  43     void (*remove)(struct _IOSocket *iosock);
  44     void (*update)(struct _IOSocket *iosock);
  45     void (*loop)(struct timeval *timeout);
  46     void (*cleanup)(void);
  47 };
  48
  49 /* IO Engines */
  50 extern struct IOEngine engine_select; /* select system call (should always be useable) */
  51 extern struct IOEngine engine_kevent;
  52 extern struct IOEngine engine_epoll;
  53 extern struct IOEngine engine_win32;
  54
  55
  56 /* _IOSocket linked list */
  57 extern struct _IOSocket *iosocket_first;
  58 extern struct _IOSocket *iosocket_last;
  59
  60 /* _IOSocket socket_flags */
  61 #define IOSOCKETFLAG_ACTIVE           0x0001
  62 #define IOSOCKETFLAG_LISTENING        0x0002
  63 #define IOSOCKETFLAG_PENDING_BINDDNS  0x0004
  64 #define IOSOCKETFLAG_PENDING_DESTDNS  0x0008
  65 #define IOSOCKETFLAG_DNSDONE_BINDDNS  0x0010
  66 #define IOSOCKETFLAG_DNSDONE_DESTDNS  0x0020
  67 #define IOSOCKETFLAG_DNSERROR         0x0040
  68 #define IOSOCKETFLAG_IPV6SOCKET       0x0080
  69 #define IOSOCKETFLAG_PARENT_PUBLIC    0x0100
  70 #define IOSOCKETFLAG_PARENT_DNSENGINE 0x0200
  71 #define IOSOCKETFLAG_SSLSOCKET        0x0400 /* use ssl after connecting */
  72 #define IOSOCKETFLAG_SSL_HANDSHAKE    0x0800 /* SSL Handshake in progress */
  73 #define IOSOCKETFLAG_SSL_WANTWRITE    0x1000\r
  74 #define IOSOCKETFLAG_SSL_READHS       0x2000 /* ssl read rehandshake */\r
  75 #define IOSOCKETFLAG_SSL_WRITEHS      0x4000 /* ssl write rehandshake */
  76 #define IOSOCKETFLAG_SSL_ESTABLISHED  0x8000
  77 #define IOSOCKETFLAG_SHUTDOWN        0x10000 /* disconnect pending */
  78 #define IOSOCKETFLAG_CONNECTING      0x20000
  79 #define IOSOCKETFLAG_INCOMING        0x40000 /* incoming (accepted) connection */
  80 #define IOSOCKETFLAG_DEAD            0x80000
  81 #define IOSOCKETFLAG_RECONNECT_IPV4 0x100000 /* possible fallback to ipv4 connect if ipv6 fails */
  82
  83 struct IOSocketDNSLookup {
  84         unsigned int bindlookup : 1;
  85         char hostname[256];
  86         struct _IOSocket *iosocket;
  87         struct _IODNSQuery *query;
  88         struct IODNSResult *result;
  89 };
  90
  91 struct _IOSocket {
  92     int fd;
  93
  94         unsigned int socket_flags : 24;
  95
  96         struct {
  97                 struct IODNSAddress addr;
  98                 struct IOSocketDNSLookup *addrlookup;
  99         } bind;
 100         struct {
 101                 struct IODNSAddress addr;
 102                 struct IOSocketDNSLookup *addrlookup;
 103         } dest;
 104
 105         unsigned int port : 16;
 106
 107         struct IOSocketBuffer readbuf;
 108     struct IOSocketBuffer writebuf;
 109
 110         struct IOSSLDescriptor *sslnode;
 111
 112         void *parent;
 113
 114         struct _IOSocket *next, *prev;
 115 };
 116
 117 void _init_sockets();
 118 void iosocket_loop(int usec);
 119 void iosocket_lookup_callback(struct IOSocketDNSLookup *lookup, struct IODNSEvent *event);
 120 void iosocket_events_callback(struct _IOSocket *iosock, int readable, int writeable);
 121 \r
 122 #define iosocket_wants_writes(IOSOCK) ((IOSOCK->writebuf.bufpos && !(IOSOCK->socket_flags & (IOSOCKETFLAG_SSL_READHS | IOSOCKETFLAG_SSL_WRITEHS))) || (IOSOCK->socket_flags & (IOSOCKETFLAG_CONNECTING | IOSOCKETFLAG_SSL_WANTWRITE)))
 123
 124 #endif
 125
 126 struct IOSocketEvent;
 127
 128 #define IOSOCKET_CALLBACK(NAME) void NAME(struct IOSocketEvent *event)
 129 typedef IOSOCKET_CALLBACK(iosocket_callback);
 130
 131 enum IOSocketStatus {
 132     IOSOCKET_CLOSED, /* descriptor is dead (socket waiting for removal or timer) */
 133     IOSOCKET_LISTENING, /* descriptor is waiting for connections (server socket) */
 134     IOSOCKET_CONNECTING, /* descriptor is waiting for connection approval (connecting client socket) */
 135     IOSOCKET_CONNECTED, /* descriptor is connected (connected client socket) */
 136         IOSOCKET_SSLHANDSHAKE /* descriptor is waiting for ssl (handshake) */
 137 };
 138
 139 enum IOSocketEventType {
 140     IOSOCKETEVENT_IGNORE,
 141     IOSOCKETEVENT_RECV, /* client socket received something (read_lines == 1  =>  recv_str valid;  read_lines == 0  =>  recv_buf valid) */
 142     IOSOCKETEVENT_CONNECTED, /* client socket connected successful */
 143     IOSOCKETEVENT_NOTCONNECTED, /* client socket could not connect (errid valid) */
 144     IOSOCKETEVENT_CLOSED, /* client socket lost connection (errid valid) */
 145     IOSOCKETEVENT_ACCEPT, /* server socket accepted new connection (accept_socket valid) */
 146         IOSOCKETEVENT_DNSFAILED /* failed to lookup DNS information (recv_str contains error message) */
 147 };
 148
 149 struct IOSocket {
 150         void *iosocket;
 151
 152         enum IOSocketStatus status;
 153         int listening : 1;
 154         int ssl : 1;
 155         int parse_delimiter : 1;
 156         int parse_empty : 1; /* parse "empty" lines (only if parse_delimiter is set) */
 157         unsigned char delimiters[IOSOCKET_PARSE_DELIMITERS_COUNT];
 158
 159         void *data;
 160         iosocket_callback *callback;
 161 };
 162
 163 struct IOSocketEvent {
 164     enum IOSocketEventType type;
 165     struct IOSocket *socket;
 166     union {
 167         char *recv_str;
 168                 struct IOSocketBuffer *recv_buf;
 169         int errid;
 170         struct IOSocket *accept_socket;
 171     } data;
 172 };
 173
 174
 175 #define IOSOCKET_ADDR_IPV4 0x01
 176 #define IOSOCKET_ADDR_IPV6 0x02 /* overrides IOSOCKET_ADDR_IPV4 */
 177 #define IOSOCKET_PROTO_UDP 0x04
 178
 179 struct IOSocket *iosocket_connect(const char *hostname, unsigned int port, int ssl, const char *bindhost, iosocket_callback *callback);
 180 struct IOSocket *iosocket_connect_flags(const char *hostname, unsigned int port, int ssl, const char *bindhost, iosocket_callback *callback, int flags);
 181 struct IOSocket *iosocket_listen(const char *hostname, unsigned int port, iosocket_callback *callback);
 182 struct IOSocket *iosocket_listen_flags(const char *hostname, unsigned int port, iosocket_callback *callback, int flags);
 183 struct IOSocket *iosocket_listen_ssl(const char *hostname, unsigned int port, const char *certfile, const char *keyfile, iosocket_callback *callback);
 184 struct IOSocket *iosocket_listen_ssl_flags(const char *hostname, unsigned int port, const char *certfile, const char *keyfile, iosocket_callback *callback, int flags);
 185 void iosocket_write(struct IOSocket *iosocket, const char *line);
 186 void iosocket_send(struct IOSocket *iosocket, const char *data, size_t datalen);
 187 void iosocket_printf(struct IOSocket *iosocket, const char *text, ...);
 188 void iosocket_close(struct IOSocket *iosocket);
 189
 190 #endif