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
  74 #define IOSOCKETFLAG_SHUTDOWN         0x2000 /* disconnect pending */
  75 #define IOSOCKETFLAG_CONNECTING       0x4000
  76 #define IOSOCKETFLAG_INCOMING         0x8000 /* incoming (accepted) connection */
  77 #define IOSOCKETFLAG_DEAD            0x10000
  78
  79 struct IOSocketDNSLookup {
  80         unsigned int bindlookup : 1;
  81         char hostname[256];
  82         struct _IOSocket *iosocket;
  83         struct _IODNSQuery *query;
  84         struct IODNSResult *result;
  85 };
  86
  87 struct _IOSocket {
  88     int fd;
  89
  90         unsigned int socket_flags : 24;
  91
  92         union {
  93                 struct IODNSAddress addr;
  94                 struct IOSocketDNSLookup *addrlookup;
  95         } bind;
  96         union {
  97                 struct IODNSAddress addr;
  98                 struct IOSocketDNSLookup *addrlookup;
  99         } dest;
 100
 101         unsigned int port : 16;
 102
 103         struct IOSocketBuffer readbuf;
 104     struct IOSocketBuffer writebuf;
 105
 106         struct IOSSLDescriptor *sslnode;
 107
 108         void *parent;
 109
 110         struct _IOSocket *next, *prev;
 111 };
 112
 113 void _init_sockets();
 114 void iosocket_loop(int usec);
 115 void iosocket_lookup_callback(struct IOSocketDNSLookup *lookup, struct IODNSEvent *event);
 116 void iosocket_events_callback(struct _IOSocket *iosock, int readable, int writeable);
 117
 118 #define iosocket_wants_writes(IOSOCK) (IOSOCK->writebuf.bufpos || (IOSOCK->socket_flags & (IOSOCKETFLAG_CONNECTING | IOSOCKETFLAG_SSL_WANTWRITE)))
 119
 120 #endif
 121
 122 struct IOSocketEvent;
 123
 124 #define IOSOCKET_CALLBACK(NAME) void NAME(struct IOSocketEvent *event)
 125 typedef IOSOCKET_CALLBACK(iosocket_callback);
 126
 127 enum IOSocketStatus {
 128     IOSOCKET_CLOSED, /* descriptor is dead (socket waiting for removal or timer) */
 129     IOSOCKET_LISTENING, /* descriptor is waiting for connections (server socket) */
 130     IOSOCKET_CONNECTING, /* descriptor is waiting for connection approval (connecting client socket) */
 131     IOSOCKET_CONNECTED /* descriptor is connected (connected client socket) */
 132 };
 133
 134 enum IOSocketEventType {
 135     IOSOCKETEVENT_IGNORE,
 136     IOSOCKETEVENT_RECV, /* client socket received something (read_lines == 1  =>  recv_str valid;  read_lines == 0  =>  recv_buf valid) */
 137     IOSOCKETEVENT_CONNECTED, /* client socket connected successful */
 138     IOSOCKETEVENT_NOTCONNECTED, /* client socket could not connect (errid valid) */
 139     IOSOCKETEVENT_CLOSED, /* client socket lost connection (errid valid) */
 140     IOSOCKETEVENT_ACCEPT, /* server socket accepted new connection (accept_socket valid) */
 141     IOSOCKETEVENT_SSLFAILED, /* failed to initialize SSL session */
 142         IOSOCKETEVENT_DNSFAILED /* failed to lookup DNS information */
 143 };
 144
 145 struct IOSocket {
 146         void *iosocket;
 147
 148         enum IOSocketStatus status;
 149         int listening : 1;
 150         int ssl : 1;
 151         int parse_delimiter : 1;
 152         int parse_empty : 1; /* parse "empty" lines (only if parse_delimiter is set) */
 153         unsigned char delimiters[IOSOCKET_PARSE_DELIMITERS_COUNT];
 154
 155         void *data;
 156         iosocket_callback *callback;
 157 };
 158
 159 struct IOSocketEvent {
 160     enum IOSocketEventType type;
 161     struct IOSocket *socket;
 162     union {
 163         char *recv_str;
 164                 struct IOSocketBuffer *recv_buf;
 165         int errid;
 166         struct IOSocket *accept_socket;
 167     } data;
 168 };
 169
 170
 171 #define IOSOCKET_ADDR_IPV4 0x01
 172 #define IOSOCKET_ADDR_IPV6 0x02 /* overrides IOSOCKET_ADDR_IPV4 */
 173 #define IOSOCKET_PROTO_UDP 0x04
 174
 175 struct IOSocket *iosocket_connect(const char *hostname, unsigned int port, int ssl, const char *bindhost, iosocket_callback *callback);
 176 struct IOSocket *iosocket_connect_flags(const char *hostname, unsigned int port, int ssl, const char *bindhost, iosocket_callback *callback, int flags);
 177 struct IOSocket *iosocket_listen(const char *hostname, unsigned int port, iosocket_callback *callback);
 178 struct IOSocket *iosocket_listen_flags(const char *hostname, unsigned int port, iosocket_callback *callback, int flags);
 179 struct IOSocket *iosocket_listen_ssl(const char *hostname, unsigned int port, const char *certfile, const char *keyfile, iosocket_callback *callback);
 180 struct IOSocket *iosocket_listen_ssl_flags(const char *hostname, unsigned int port, const char *certfile, const char *keyfile, iosocket_callback *callback, int flags);
 181 void iosocket_write(struct IOSocket *iosocket, const char *line);
 182 void iosocket_send(struct IOSocket *iosocket, const char *data, size_t datalen);
 183 void iosocket_printf(struct IOSocket *iosocket, const char *text, ...);
 184 void iohandler_close(struct IOSocket *iosocket);
 185
 186 #endif