src/IOHandler/IOSockets.c

   1 /* IOSockets.c - 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 #define _IOHandler_internals
  18 #include "IOInternal.h"
  19 #include "IOHandler.h"
  20 #include "IOSockets.h"
  21 #include "IOLog.h"
  22 #include "IODNSLookup.h"
  23 #include "IOSSLBackend.h"
  24
  25 #ifdef WIN32
  26 #ifdef _WIN32_WINNT
  27 #undef _WIN32_WINNT
  28 #endif
  29 #define _WIN32_WINNT 0x501
  30 #include <winsock2.h>
  31 #include <windows.h>
  32 #include <ws2tcpip.h>
  33 #else
  34 #include <sys/socket.h>
  35 #include <netinet/in.h>
  36 #include <netinet/ip.h>
  37 #include <arpa/inet.h>
  38 #include <unistd.h>
  39 #include <fcntl.h>
  40 #endif
  41 #include "compat/inet.h"
  42 #include <stdio.h>
  43 #include <stdlib.h>
  44 #include <string.h>
  45 #include <signal.h>
  46 #include <errno.h>
  47 #include <stdarg.h>
  48
  49 #ifndef EWOULDBLOCK
  50 #define EWOULDBLOCK EAGAIN
  51 #endif
  52
  53 struct _IOSocket *iosocket_first = NULL;
  54 struct _IOSocket *iosocket_last = NULL;
  55
  56 struct IOEngine *engine = NULL;
  57
  58 static void iosocket_increase_buffer(struct IOSocketBuffer *iobuf, size_t required);
  59 static int iosocket_parse_address(const char *hostname, struct IODNSAddress *addr, int records);
  60 static int iosocket_lookup_hostname(struct _IOSocket *iosock, const char *hostname, int records, int bindaddr);
  61 static int iosocket_lookup_apply(struct _IOSocket *iosock, int noip6);
  62 static void socket_lookup_clear(struct _IOSocket *iosock);
  63 static void iosocket_connect_finish(struct _IOSocket *iosock);
  64 static void iosocket_listen_finish(struct _IOSocket *iosock);
  65 static int iosocket_try_write(struct _IOSocket *iosock);
  66 static void iosocket_trigger_event(struct IOSocketEvent *event);
  67
  68 #ifdef WIN32
  69 static int close(int fd) {
  70         return closesocket(fd);
  71 }
  72 #endif
  73
  74 static void iosockets_init_engine() {
  75         //try other engines
  76         if(!engine && engine_kevent.init && engine_kevent.init())
  77                 engine = &engine_kevent;
  78         if(!engine && engine_epoll.init && engine_epoll.init())
  79                 engine = &engine_epoll;
  80         if(!engine && engine_win32.init && engine_win32.init())
  81                 engine = &engine_win32;
  82
  83         if (!engine) {
  84                 if(engine_select.init())
  85                         engine = &engine_select;
  86                 else {
  87                         iolog_trigger(IOLOG_FATAL, "found no useable IO engine");
  88                         return;
  89                 }
  90         }
  91         iolog_trigger(IOLOG_DEBUG, "using %s IOSockets engine", engine->name);
  92 }
  93
  94 void _init_sockets() {
  95         #ifdef WIN32
  96         WSADATA wsaData;
  97         int iResult;
  98         //Initialize Winsock
  99         iResult = WSAStartup(MAKEWORD(2,2), &wsaData);
 100         if(iResult != 0){
 101                 iolog_trigger(IOLOG_ERROR, "WSAStartup returned error code: %d", iResult);
 102         }
 103         #endif
 104
 105         iosockets_init_engine();\r
 106         iossl_init();
 107 }
 108
 109
 110 struct _IOSocket *_create_socket() {
 111         struct _IOSocket *iosock = calloc(1, sizeof(*iosock));
 112         if(!iosock) {
 113                 iolog_trigger(IOLOG_ERROR, "could not allocate memory for _IOSocket in %s:%d", __FILE__, __LINE__);
 114                 return NULL;
 115         }
 116         if(iosocket_last)
 117                 iosocket_last->next = iosock;
 118         else
 119                 iosocket_first = iosock;
 120         iosock->prev = iosocket_last;
 121         iosocket_last = iosock;
 122         return iosock;
 123 }
 124
 125 void _free_socket(struct _IOSocket *iosock) {
 126         iosocket_deactivate(iosock);
 127         if(iosock->prev)
 128                 iosock->prev->next = iosock->next;
 129         else
 130                 iosocket_first = iosock->next;
 131         if(iosock->next)
 132                 iosock->next->prev = iosock->prev;
 133         else
 134                 iosocket_last = iosock->prev;
 135
 136         if(iosock->bind.addr.addresslen)
 137                 free(iosock->bind.addr.address);
 138         if(iosock->dest.addr.addresslen)
 139                 free(iosock->dest.addr.address);
 140         if(iosock->bind.addrlookup || iosock->dest.addrlookup)
 141                 socket_lookup_clear(iosock);
 142         if(iosock->readbuf.buffer)
 143                 free(iosock->readbuf.buffer);
 144         if(iosock->writebuf.buffer)
 145                 free(iosock->writebuf.buffer);
 146         if((iosock->socket_flags & IOSOCKETFLAG_SSLSOCKET))\r
 147                 iossl_disconnect(iosock);\r
 148
 149         free(iosock);
 150 }
 151
 152 void iosocket_activate(struct _IOSocket *iosock) {
 153         if((iosock->socket_flags & IOSOCKETFLAG_ACTIVE))
 154                 return;
 155         iosock->socket_flags |= IOSOCKETFLAG_ACTIVE;
 156         engine->add(iosock);
 157 }
 158
 159 void iosocket_deactivate(struct _IOSocket *iosock) {
 160         if(!(iosock->socket_flags & IOSOCKETFLAG_ACTIVE))
 161                 return;
 162         iosock->socket_flags &= ~IOSOCKETFLAG_ACTIVE;
 163         engine->remove(iosock);
 164 }
 165
 166 void iosocket_update(struct _IOSocket *iosock) {
 167         if(!(iosock->socket_flags & IOSOCKETFLAG_ACTIVE))
 168                 return;
 169         engine->update(iosock);
 170 }
 171
 172 static void iosocket_increase_buffer(struct IOSocketBuffer *iobuf, size_t required) {
 173         if(iobuf->buflen >= required) return;
 174         char *new_buf;
 175         if(iobuf->buffer)
 176                 new_buf = realloc(iobuf->buffer, required + 2);
 177         else
 178                 new_buf = malloc(required + 2);
 179         if(new_buf) {
 180                 iobuf->buffer = new_buf;
 181                 iobuf->buflen = required;
 182         }
 183 }
 184
 185 static int iosocket_parse_address(const char *hostname, struct IODNSAddress *addr, int records) {
 186         int ret;
 187         if((records & IOSOCKET_ADDR_IPV4)) {
 188                 struct sockaddr_in ip4addr;
 189                 ret = inet_pton(AF_INET, hostname, &(ip4addr.sin_addr));
 190                 if(ret == 1) {
 191                         addr->addresslen = sizeof(ip4addr);
 192                         addr->address = malloc(addr->addresslen);
 193                         if(!addr->address) {
 194                                 iolog_trigger(IOLOG_ERROR, "could not allocate memory for sockaddr in %s:%d", __FILE__, __LINE__);
 195                                 return -1;
 196                         }
 197                         memcpy(addr->address, &ip4addr, sizeof(ip4addr));
 198                         return 1;
 199                 }
 200         }
 201         if((records & IOSOCKET_ADDR_IPV6)) {
 202                 struct sockaddr_in6 ip6addr;
 203                 ret = inet_pton(AF_INET6, hostname, &(ip6addr.sin6_addr));
 204                 if(ret == 1) {
 205                         addr->addresslen = sizeof(ip6addr);
 206                         addr->address = malloc(addr->addresslen);
 207                         if(!addr->address) {
 208                                 iolog_trigger(IOLOG_ERROR, "could not allocate memory for sockaddr in %s:%d", __FILE__, __LINE__);
 209                                 return -1;
 210                         }
 211                         memcpy(addr->address, &ip6addr, sizeof(ip6addr));
 212                         return 1;
 213                 }
 214         }
 215         return 0;
 216 }
 217
 218 static int iosocket_lookup_hostname(struct _IOSocket *iosock, const char *hostname, int records, int bindaddr) {
 219         struct IOSocketDNSLookup *lookup = calloc(1, sizeof(*lookup));
 220         if(!lookup) {
 221                 iolog_trigger(IOLOG_ERROR, "could not allocate memory for IOSocketDNSLookup in %s:%d", __FILE__, __LINE__);
 222                 return 0;
 223         }
 224
 225         struct _IODNSQuery *query = _create_dnsquery();
 226         if(!query) {
 227                 free(lookup);
 228                 return 0;
 229         }
 230
 231         query->parent = lookup;
 232         query->flags |= IODNSFLAG_PARENT_SOCKET;
 233         lookup->iosocket = iosock;
 234         lookup->query = query;
 235         strncpy(lookup->hostname, hostname, sizeof(lookup->hostname));
 236         lookup->hostname[sizeof(lookup->hostname)-1] = 0;
 237         if(bindaddr) {
 238                 lookup->bindlookup = 1;
 239                 iosock->bind.addrlookup = lookup;
 240         } else {
 241                 lookup->bindlookup = 0;
 242                 iosock->dest.addrlookup = lookup;
 243         }
 244
 245         int dnsrecords = 0;
 246         if((records & IOSOCKET_ADDR_IPV4))
 247                 dnsrecords |= IODNS_RECORD_A;
 248         if((records & IOSOCKET_ADDR_IPV6))
 249                 dnsrecords |= IODNS_RECORD_AAAA;
 250
 251         query->request.host = strdup(hostname);
 252         query->type = (dnsrecords & IODNS_FORWARD);
 253
 254         _start_dnsquery(query);
 255         return 1;
 256 }
 257
 258 void iosocket_lookup_callback(struct IOSocketDNSLookup *lookup, struct IODNSEvent *event) {
 259         lookup->query = NULL;
 260         struct _IOSocket *iosock = lookup->iosocket;
 261         if(iosock == NULL)
 262                 return;
 263
 264         if(event->type == IODNSEVENT_SUCCESS)
 265                 lookup->result = event->result;
 266         else
 267                 lookup->result = NULL;
 268
 269         if(lookup->bindlookup) {
 270                 iosock->socket_flags &= ~IOSOCKETFLAG_PENDING_BINDDNS;
 271                 iosock->socket_flags |= IOSOCKETFLAG_DNSDONE_BINDDNS;
 272         } else {
 273                 iosock->socket_flags &= ~IOSOCKETFLAG_PENDING_DESTDNS;
 274                 iosock->socket_flags |= IOSOCKETFLAG_DNSDONE_DESTDNS;
 275         }
 276
 277         int dns_finished = 0;
 278         if((iosock->socket_flags & (IOSOCKETFLAG_PENDING_BINDDNS | IOSOCKETFLAG_PENDING_DESTDNS)) == 0)
 279                 dns_finished = 1;
 280
 281         if(dns_finished) {
 282                 int ret;
 283                 ret = iosocket_lookup_apply(iosock, 0);
 284                 if(ret) { //if ret=0 an error occured in iosocket_lookup_apply and we should stop here.
 285                         if((iosock->socket_flags & IOSOCKETFLAG_LISTENING)) {
 286                                 socket_lookup_clear(iosock);
 287                                 iosocket_listen_finish(iosock);
 288                         } else
 289                                 iosocket_connect_finish(iosock);
 290                 }
 291         }
 292 }
 293
 294 static int iosocket_lookup_apply(struct _IOSocket *iosock, int noip6) {
 295         char errbuf[512];
 296         struct IOSocketDNSLookup *bind_lookup = ((iosock->socket_flags & IOSOCKETFLAG_DNSDONE_BINDDNS) ? iosock->bind.addrlookup : NULL);
 297         struct IOSocketDNSLookup *dest_lookup = ((iosock->socket_flags & IOSOCKETFLAG_DNSDONE_DESTDNS) ? iosock->dest.addrlookup : NULL);
 298
 299         iolog_trigger(IOLOG_DEBUG, "all pending lookups finished. trying to apply lookup results...");
 300
 301         if(!bind_lookup && !dest_lookup) {
 302                 iosock->socket_flags |= IOSOCKETFLAG_DNSERROR;
 303                 sprintf(errbuf, "Internal Error");
 304                 iolog_trigger(IOLOG_ERROR, "trying to apply lookup results without any lookups processed in %s:%d", __FILE__, __LINE__);
 305                 goto iosocket_lookup_apply_end;
 306         }
 307
 308         struct IODNSResult *result;
 309         int bind_numip4 = 0, bind_numip6 = 0;
 310         int dest_numip4 = 0, dest_numip6 = 0;
 311
 312         if(bind_lookup) {
 313                 for(result = bind_lookup->result; result; result = result->next) {
 314                         if((result->type & IODNS_RECORD_A))
 315                                 bind_numip4++;
 316                         if((result->type & IODNS_RECORD_AAAA))
 317                                 bind_numip6++;
 318                 }
 319         }
 320         if(dest_lookup) {
 321                 for(result = dest_lookup->result; result; result = result->next) {
 322                         if((result->type & IODNS_RECORD_A))
 323                                 dest_numip4++;
 324                         if((result->type & IODNS_RECORD_AAAA))
 325                                 dest_numip6++;
 326                 }
 327         }
 328         int useip6 = 0;
 329         int useip4 = 0;
 330         if(bind_lookup && (bind_numip6 == 0 && bind_numip4 == 0)) {
 331                 iosock->socket_flags |= IOSOCKETFLAG_DNSERROR;
 332                 sprintf(errbuf, "could not lookup bind address (%s)", bind_lookup->hostname);
 333                 goto iosocket_lookup_apply_end;
 334         } else if(dest_lookup && (dest_numip6 == 0 && dest_numip4 == 0)) {
 335                 iosock->socket_flags |= IOSOCKETFLAG_DNSERROR;
 336                 sprintf(errbuf, "could not lookup destination address (%s)", dest_lookup->hostname);
 337                 goto iosocket_lookup_apply_end;
 338         } else if(bind_lookup && dest_lookup) {
 339                 if(bind_numip6 > 0 && dest_numip6 > 0)
 340                         useip6 = 1;
 341                 if(bind_numip4 > 0 && dest_numip4 > 0)
 342                         useip4 = 1;
 343         } else if(bind_lookup) {
 344                 if(bind_numip6)
 345                         useip6 = 1;
 346                 if(bind_numip4)
 347                         useip4 = 1;
 348         } else if(dest_lookup) {
 349                 if(dest_numip6)
 350                         useip6 = 1;
 351                 if(dest_numip4)
 352                         useip4 = 1;
 353         }
 354
 355         int usetype = 0;
 356         if(useip6 && !noip6) {
 357                 usetype = IODNS_RECORD_AAAA;
 358                 iosock->socket_flags |= IOSOCKETFLAG_IPV6SOCKET;
 359                 if(useip4)
 360                         iosock->socket_flags |= IOSOCKETFLAG_RECONNECT_IPV4;
 361         } else if(useip4) {
 362                 usetype = IODNS_RECORD_A;
 363                 iosock->socket_flags &= ~(IOSOCKETFLAG_IPV6SOCKET | IOSOCKETFLAG_RECONNECT_IPV4);
 364         } else {
 365                 iosock->socket_flags |= IOSOCKETFLAG_DNSERROR;
 366                 sprintf(errbuf, "could not lookup adresses of the same IP family for bind and destination host. (bind: %d ip4, %d ip6 | dest: %d ip4, %d ip6)", bind_numip4, bind_numip6, dest_numip4, dest_numip6);
 367                 goto iosocket_lookup_apply_end;
 368         }
 369
 370         #define IOSOCKET_APPLY_COPYADDR(type) \
 371         iosock->type.addr.addresslen = result->result.addr.addresslen; \
 372         iosock->type.addr.address = malloc(result->result.addr.addresslen); \
 373         if(!iosock->type.addr.address) { \
 374                 iolog_trigger(IOLOG_ERROR, "could not allocate memory for sockaddr in %s:%d", __FILE__, __LINE__); \
 375                 iosock->type.addr.addresslen = 0; \
 376                 iosock->socket_flags |= IOSOCKETFLAG_DNSERROR; \
 377                 sprintf(errbuf, "could not allocate memory for dns information"); \
 378                 goto iosocket_lookup_apply_end; \
 379         } \
 380         memcpy(iosock->type.addr.address, result->result.addr.address, result->result.addr.addresslen);
 381
 382
 383         if(bind_lookup) {
 384                 int usenum = ((usetype == IODNS_RECORD_AAAA) ? bind_numip6 : bind_numip4);
 385                 usenum = rand() % usenum;
 386                 for(result = bind_lookup->result; result; result = result->next) {
 387                         if((result->type & usetype)) {
 388                                 if(usenum == 0) {
 389                                         inet_ntop(((usetype == IODNS_RECORD_AAAA) ? AF_INET6 : AF_INET), ((usetype == IODNS_RECORD_AAAA) ? (void *)(&((struct sockaddr_in6 *)result->result.addr.address)->sin6_addr) : (void *)(&((struct sockaddr_in *)result->result.addr.address)->sin_addr)), errbuf, sizeof(errbuf));
 390                                         iolog_trigger(IOLOG_DEBUG, "using IPv%s Address (%s) as bind address", ((usetype == IODNS_RECORD_AAAA) ? "6" : "4"), errbuf);
 391                                         IOSOCKET_APPLY_COPYADDR(bind)
 392                                         break;
 393                                 }
 394                                 usenum--;
 395                         }
 396                 }
 397         } else
 398                 iosock->bind.addr.addresslen = 0;
 399
 400         if(dest_lookup) {
 401                 int usenum = ((usetype == IODNS_RECORD_AAAA) ? dest_numip6 : dest_numip4);
 402                 usenum = rand() % usenum;
 403                 for(result = dest_lookup->result; result; result = result->next) {
 404                         if((result->type & usetype)) {
 405                                 if(usenum == 0) {
 406                                         inet_ntop(((usetype == IODNS_RECORD_AAAA) ? AF_INET6 : AF_INET), ((usetype == IODNS_RECORD_AAAA) ? (void *)(&((struct sockaddr_in6 *)result->result.addr.address)->sin6_addr) : (void *)(&((struct sockaddr_in *)result->result.addr.address)->sin_addr)), errbuf, sizeof(errbuf));
 407                                         iolog_trigger(IOLOG_DEBUG, "using IPv%s Address (%s) as dest address", ((usetype == IODNS_RECORD_AAAA) ? "6" : "4"), errbuf);
 408                                         IOSOCKET_APPLY_COPYADDR(dest)
 409                                         break;
 410                                 }
 411                                 usenum--;
 412                         }
 413                 }
 414         } else
 415                 iosock->dest.addr.addresslen = 0;
 416
 417         iosocket_lookup_apply_end:
 418
 419         if((iosock->socket_flags & IOSOCKETFLAG_DNSERROR)) {
 420                 // TODO: trigger error
 421                 iolog_trigger(IOLOG_ERROR, "error while trying to apply dns lookup information: %s", errbuf);
 422
 423                 if((iosock->socket_flags & IOSOCKETFLAG_PARENT_PUBLIC)) {
 424                         //trigger event
 425                         struct IOSocket *iosocket = iosock->parent;
 426
 427                         struct IOSocketEvent callback_event;
 428                         callback_event.type = IOSOCKETEVENT_DNSFAILED;
 429                         callback_event.socket = iosocket;
 430                         callback_event.data.recv_str = errbuf;
 431                         iosocket_trigger_event(&callback_event);
 432
 433                         iosocket_close(iosocket);
 434                 } else {
 435                         // TODO: IODNS Callback
 436                 }
 437                 return 0;
 438         } else
 439                 return 1;
 440 }
 441
 442 static void socket_lookup_clear(struct _IOSocket *iosock) {
 443         struct IOSocketDNSLookup *bind_lookup = ((iosock->socket_flags & IOSOCKETFLAG_DNSDONE_BINDDNS) ? iosock->bind.addrlookup : NULL);
 444         struct IOSocketDNSLookup *dest_lookup = ((iosock->socket_flags & IOSOCKETFLAG_DNSDONE_DESTDNS) ? iosock->dest.addrlookup : NULL);
 445         if(bind_lookup) {
 446                 if(bind_lookup->result)
 447                         iodns_free_result(bind_lookup->result);
 448                 free(bind_lookup);
 449                 iosock->bind.addrlookup = NULL;
 450         }
 451         if(dest_lookup) {
 452                 if(dest_lookup->result)
 453                         iodns_free_result(dest_lookup->result);
 454                 free(dest_lookup);
 455                 iosock->dest.addrlookup = NULL;
 456         }
 457 }
 458
 459 static void iosocket_prepare_fd(int sockfd) {
 460         // prevent SIGPIPE
 461         #ifndef WIN32
 462         #if defined(SO_NOSIGPIPE)
 463         {
 464                 int set = 1;
 465                 setsockopt(sockfd, SOL_SOCKET, SO_NOSIGPIPE, (void *)&set, sizeof(int));
 466         }
 467         #else
 468         signal(SIGPIPE, SIG_IGN);
 469         #endif
 470         #endif
 471
 472         // make sockfd unblocking
 473         #if defined(F_GETFL)
 474         {
 475                 int fcntl_flags;
 476                 fcntl_flags = fcntl(sockfd, F_GETFL);
 477                 fcntl(sockfd, F_SETFL, fcntl_flags|O_NONBLOCK);
 478                 fcntl_flags = fcntl(sockfd, F_GETFD);
 479                 fcntl(sockfd, F_SETFD, fcntl_flags|FD_CLOEXEC);
 480         }
 481         #elif defined(FIONBIO)
 482         {
 483                 unsigned long ulong = 1;
 484                 ioctlsocket(sockfd, FIONBIO, &ulong);
 485         }
 486         #endif
 487 }
 488
 489 static void iosocket_connect_finish(struct _IOSocket *iosock) {
 490         int sockfd;
 491         if((iosock->socket_flags & IOSOCKETFLAG_IPV6SOCKET))
 492                 sockfd = socket(AF_INET6, SOCK_STREAM, 0);
 493         else
 494                 sockfd = socket(AF_INET, SOCK_STREAM, 0);
 495         if(sockfd == -1) {
 496                 iolog_trigger(IOLOG_ERROR, "could not create socket in %s:%d", __FILE__, __LINE__);
 497                 // TODO: trigger error
 498
 499                 return;
 500         }
 501
 502         // set port and bind address
 503         if((iosock->socket_flags & IOSOCKETFLAG_IPV6SOCKET)) {
 504                 struct sockaddr_in6 *ip6 = (void*) iosock->dest.addr.address;
 505                 ip6->sin6_family = AF_INET6;
 506                 ip6->sin6_port = htons(iosock->port);
 507
 508                 if(iosock->bind.addr.addresslen) {
 509                         struct sockaddr_in6 *ip6bind = (void*) iosock->bind.addr.address;
 510                         ip6bind->sin6_family = AF_INET6;
 511                         ip6bind->sin6_port = htons(0);
 512
 513                         bind(sockfd, (struct sockaddr*)ip6bind, sizeof(*ip6bind));
 514                 }
 515         } else {
 516                 struct sockaddr_in *ip4 = (void*) iosock->dest.addr.address;
 517                 ip4->sin_family = AF_INET;
 518                 ip4->sin_port = htons(iosock->port);
 519
 520                 if(iosock->bind.addr.addresslen) {
 521                         struct sockaddr_in *ip4bind = (void*) iosock->bind.addr.address;
 522                         ip4bind->sin_family = AF_INET;
 523                         ip4bind->sin_port = htons(0);
 524
 525                         bind(sockfd, (struct sockaddr*)ip4bind, sizeof(*ip4bind));
 526                 }
 527         }
 528
 529         iosocket_prepare_fd(sockfd);
 530
 531         int ret = connect(sockfd, iosock->dest.addr.address, iosock->dest.addr.addresslen); //returns EINPROGRESS here (nonblocking)
 532         iolog_trigger(IOLOG_DEBUG, "connecting socket (connect: %d)", ret);
 533
 534         iosock->fd = sockfd;
 535         iosock->socket_flags |= IOSOCKETFLAG_CONNECTING;
 536
 537         iosocket_activate(iosock);
 538 }
 539
 540 static void iosocket_listen_finish(struct _IOSocket *iosock) {
 541         int sockfd;
 542         if((iosock->socket_flags & IOSOCKETFLAG_IPV6SOCKET))
 543                 sockfd = socket(AF_INET6, SOCK_STREAM, 0);
 544         else
 545                 sockfd = socket(AF_INET, SOCK_STREAM, 0);
 546         if(sockfd == -1) {
 547                 iolog_trigger(IOLOG_ERROR, "could not create socket in %s:%d", __FILE__, __LINE__);
 548                 // TODO: trigger error
 549
 550                 return;
 551         }
 552
 553         // set port and bind address
 554         if((iosock->socket_flags & IOSOCKETFLAG_IPV6SOCKET)) {
 555                 struct sockaddr_in6 *ip6bind = (void*) iosock->bind.addr.address;
 556                 ip6bind->sin6_family = AF_INET6;
 557                 ip6bind->sin6_port = htons(iosock->port);
 558
 559                 int opt = 1;
 560                 setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, (const char*)&opt, sizeof(opt));
 561
 562                 bind(sockfd, (struct sockaddr*)ip6bind, sizeof(*ip6bind));
 563         } else {
 564                 struct sockaddr_in *ip4bind = (void*) iosock->bind.addr.address;
 565                 ip4bind->sin_family = AF_INET;
 566                 ip4bind->sin_port = htons(iosock->port);
 567
 568                 int opt = 1;
 569                 setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, (const char*)&opt, sizeof(opt));
 570
 571                 bind(sockfd, (struct sockaddr*)ip4bind, sizeof(*ip4bind));
 572         }
 573
 574         iosocket_prepare_fd(sockfd);
 575
 576         listen(sockfd, 1);
 577         iosock->fd = sockfd;
 578
 579         iosocket_activate(iosock);
 580 }
 581
 582 struct _IOSocket *iosocket_accept_client(struct _IOSocket *iosock) {
 583         struct IOSocket *new_iosocket = calloc(1, sizeof(*new_iosocket));
 584         if(!new_iosocket) {
 585                 iolog_trigger(IOLOG_ERROR, "could not allocate memory for IOSocket in %s:%d", __FILE__, __LINE__);
 586                 close(accept(iosock->fd, NULL, 0)); // simply drop connection
 587                 return NULL;
 588         }
 589         struct _IOSocket *new_iosock = _create_socket();
 590         if(!new_iosock) {
 591                 free(new_iosocket);
 592                 close(accept(iosock->fd, NULL, 0)); // simply drop connection
 593                 return NULL;
 594         }
 595         new_iosocket->iosocket = new_iosock;
 596         new_iosocket->status = IOSOCKET_CONNECTED;
 597         new_iosocket->data = iosock;
 598         new_iosock->parent = new_iosocket;
 599         new_iosock->socket_flags |= IOSOCKETFLAG_PARENT_PUBLIC | IOSOCKETFLAG_INCOMING | (iosock->socket_flags & IOSOCKETFLAG_IPV6SOCKET);
 600
 601         struct sockaddr_storage addr;
 602         socklen_t addrlen = sizeof(addr);
 603
 604         //accept client
 605         new_iosock->fd = accept(iosock->fd, (struct sockaddr *)&addr, &addrlen);
 606
 607         //copy remote address
 608         new_iosock->dest.addr.address = malloc(addrlen);
 609         if(!new_iosock->dest.addr.address) {
 610                 close(new_iosock->fd);
 611                 free(new_iosock);
 612                 free(new_iosock);
 613                 iolog_trigger(IOLOG_ERROR, "could not allocate memory for sockaddr in %s:%d", __FILE__, __LINE__);
 614                 return NULL;
 615         }
 616         memcpy(new_iosock->dest.addr.address, &addr, addrlen);
 617         new_iosock->dest.addr.addresslen = addrlen;
 618
 619         //copy local address
 620         new_iosock->bind.addr.address = malloc(iosock->bind.addr.addresslen);
 621         if(!new_iosock->bind.addr.address) {
 622                 close(new_iosock->fd);
 623                 free(new_iosock);
 624                 free(new_iosock);
 625                 iolog_trigger(IOLOG_ERROR, "could not allocate memory for sockaddr in %s:%d", __FILE__, __LINE__);
 626                 return NULL;
 627         }
 628         memcpy(new_iosock->bind.addr.address, iosock->bind.addr.address, iosock->bind.addr.addresslen);
 629         new_iosock->bind.addr.addresslen = iosock->bind.addr.addresslen;
 630
 631         //prepare new socket fd
 632         iosocket_prepare_fd(new_iosock->fd);
 633
 634         if((iosock->socket_flags & IOSOCKETFLAG_SSLSOCKET)) {
 635                 new_iosocket->ssl = 1;
 636                 new_iosock->socket_flags |= IOSOCKETFLAG_SSLSOCKET;
 637                 \r
 638                 iossl_client_accepted(iosock, new_iosock);
 639         } else {
 640                 //initialize readbuf
 641                 iosocket_increase_buffer(&iosock->readbuf, 1024);
 642         }
 643
 644         iosocket_activate(new_iosock);
 645         return new_iosock;
 646 }
 647
 648 /* public functions */
 649
 650 struct IOSocket *iosocket_connect(const char *hostname, unsigned int port, int ssl, const char *bindhost, iosocket_callback *callback) {
 651         return iosocket_connect_flags(hostname, port, ssl, bindhost, callback, (IOSOCKET_ADDR_IPV4 | IOSOCKET_ADDR_IPV6));
 652 }
 653
 654 struct IOSocket *iosocket_connect_flags(const char *hostname, unsigned int port, int ssl, const char *bindhost, iosocket_callback *callback, int flags) {
 655         struct IOSocket *iodescriptor = calloc(1, sizeof(*iodescriptor));
 656         if(!iodescriptor) {
 657                 iolog_trigger(IOLOG_ERROR, "could not allocate memory for IOSocket in %s:%d", __FILE__, __LINE__);
 658                 return NULL;
 659         }
 660
 661         struct _IOSocket *iosock = _create_socket();
 662         if(!iosock) {
 663                 free(iodescriptor);
 664                 return NULL;
 665         }
 666
 667         iodescriptor->iosocket = iosock;
 668         iodescriptor->status = IOSOCKET_CONNECTING;
 669         iodescriptor->callback = callback;
 670         iosock->parent = iodescriptor;
 671         iosock->socket_flags |= IOSOCKETFLAG_PARENT_PUBLIC;
 672         iosock->port = port;
 673         if(ssl) {
 674                 iodescriptor->ssl = 1;
 675                 iosock->socket_flags |= IOSOCKETFLAG_SSLSOCKET;
 676         }
 677
 678         if(bindhost) {
 679                 switch(iosocket_parse_address(bindhost, &iosock->bind.addr, flags)) {
 680                 case -1:
 681                         free(iosock);
 682                         return NULL;
 683                 case 0:
 684                         /* start dns lookup */
 685                         iosock->socket_flags |= IOSOCKETFLAG_PENDING_BINDDNS;
 686                         iosocket_lookup_hostname(iosock, bindhost, flags, 1);
 687                         break;
 688                 case 1:
 689                         /* valid address */
 690                         break;
 691                 }
 692         }
 693         switch(iosocket_parse_address(hostname, &iosock->dest.addr, flags)) {
 694         case -1:
 695                 free(iosock);
 696                 return NULL;
 697         case 0:
 698                 /* start dns lookup */
 699                 iosock->socket_flags |= IOSOCKETFLAG_PENDING_DESTDNS;
 700                 iosocket_lookup_hostname(iosock, hostname, flags, 0);
 701                 break;
 702         case 1:
 703                 /* valid address */
 704                 break;
 705         }
 706         if((iosock->socket_flags & (IOSOCKETFLAG_PENDING_BINDDNS | IOSOCKETFLAG_PENDING_DESTDNS)) == 0) {
 707                 iosocket_connect_finish(iosock);
 708         }
 709         return iodescriptor;
 710 }
 711
 712 struct IOSocket *iosocket_listen(const char *hostname, unsigned int port, iosocket_callback *callback) {
 713         return iosocket_listen_flags(hostname, port, callback, (IOSOCKET_ADDR_IPV4 | IOSOCKET_ADDR_IPV6));
 714 }
 715
 716 struct IOSocket *iosocket_listen_flags(const char *hostname, unsigned int port, iosocket_callback *callback, int flags) {
 717         struct IOSocket *iodescriptor = calloc(1, sizeof(*iodescriptor));
 718         if(!iodescriptor) {
 719                 iolog_trigger(IOLOG_ERROR, "could not allocate memory for IOSocket in %s:%d", __FILE__, __LINE__);
 720                 return NULL;
 721         }
 722
 723         struct _IOSocket *iosock = _create_socket();
 724         if(!iosock) {
 725                 free(iodescriptor);
 726                 return NULL;
 727         }
 728
 729         iodescriptor->iosocket = iosock;
 730         iodescriptor->status = IOSOCKET_LISTENING;
 731         iodescriptor->listening = 1;
 732         iodescriptor->callback = callback;
 733         iosock->parent = iodescriptor;
 734         iosock->socket_flags |= IOSOCKETFLAG_PARENT_PUBLIC | IOSOCKETFLAG_LISTENING;
 735         iosock->port = port;
 736
 737         switch(iosocket_parse_address(hostname, &iosock->bind.addr, flags)) {
 738         case -1:
 739                 free(iosock);
 740                 return NULL;
 741         case 0:
 742                 /* start dns lookup */
 743                 iosock->socket_flags |= IOSOCKETFLAG_PENDING_BINDDNS;
 744                 iosocket_lookup_hostname(iosock, hostname, flags, 1);
 745                 break;
 746         case 1:
 747                 /* valid address */
 748                 break;
 749         }
 750         if((iosock->socket_flags & IOSOCKETFLAG_PENDING_BINDDNS) == 0) {
 751                 iosocket_listen_finish(iosock);
 752         }
 753         return iodescriptor;
 754 }
 755
 756 struct IOSocket *iosocket_listen_ssl(const char *hostname, unsigned int port, const char *certfile, const char *keyfile, iosocket_callback *callback) {
 757         return iosocket_listen_ssl_flags(hostname, port, certfile, keyfile, callback, (IOSOCKET_ADDR_IPV4 | IOSOCKET_ADDR_IPV6));
 758 }
 759
 760 struct IOSocket *iosocket_listen_ssl_flags(const char *hostname, unsigned int port, const char *certfile, const char *keyfile, iosocket_callback *callback, int flags) {
 761         struct IOSocket *iosocket = iosocket_listen_flags(hostname, port, callback, flags);\r
 762         struct _IOSocket *iosock = iosocket->iosocket;\r
 763         iosock->socket_flags |= IOSOCKETFLAG_SSLSOCKET;\r
 764         iossl_listen(iosock, certfile, keyfile);
 765         return iosocket;
 766 }
 767
 768 void iosocket_close(struct IOSocket *iosocket) {
 769         struct _IOSocket *iosock = iosocket->iosocket;
 770         if(iosock == NULL) {
 771                 iolog_trigger(IOLOG_WARNING, "called iosocket_close for destroyed IOSocket in %s:%d", __FILE__, __LINE__);
 772                 return;
 773         }
 774
 775         iosock->socket_flags |= IOSOCKETFLAG_SHUTDOWN;
 776
 777         if(iosock->writebuf.bufpos) {
 778                 //try to send everything before closing
 779 #if defined(F_GETFL)
 780                 {
 781                         int flags;
 782                         flags = fcntl(iosock->fd, F_GETFL);
 783                         fcntl(iosock->fd, F_SETFL, flags & ~O_NONBLOCK);
 784                         flags = fcntl(iosock->fd, F_GETFD);
 785                         fcntl(iosock->fd, F_SETFD, flags|FD_CLOEXEC);
 786                 }
 787 #else
 788                 iosocket_deactivate(iosock);
 789 #endif
 790                 iosocket_try_write(iosock);
 791         }
 792         //close IOSocket
 793         if((iosock->socket_flags & IOSOCKETFLAG_SSLSOCKET))
 794                 iossl_disconnect(iosock);
 795         if(iosock->fd)
 796                 close(iosock->fd);
 797         _free_socket(iosock);
 798         iosocket->iosocket = NULL;
 799         iosocket->status = IOSOCKET_CLOSED;
 800         iogc_add(iosocket);
 801 }
 802
 803 struct IODNSAddress *iosocket_get_remote_addr(struct IOSocket *iosocket) {
 804         struct _IOSocket *iosock = iosocket->iosocket;
 805         if(iosock == NULL) {
 806                 iolog_trigger(IOLOG_WARNING, "called iosocket_get_remote_addr for destroyed IOSocket in %s:%d", __FILE__, __LINE__);
 807                 return NULL;
 808         }
 809         if(iosock->socket_flags & IOSOCKETFLAG_PENDING_DESTDNS)
 810                 return NULL;
 811         if(!iosock->dest.addr.addresslen)
 812                 return NULL;
 813         return &iosock->dest.addr;
 814 }
 815
 816 struct IODNSAddress *iosocket_get_local_addr(struct IOSocket *iosocket) {
 817         struct _IOSocket *iosock = iosocket->iosocket;
 818         if(iosock == NULL) {
 819                 iolog_trigger(IOLOG_WARNING, "called iosocket_get_local_addr for destroyed IOSocket in %s:%d", __FILE__, __LINE__);
 820                 return NULL;
 821         }
 822         if(iosock->socket_flags & IOSOCKETFLAG_PENDING_BINDDNS)
 823                 return NULL;
 824         if(!iosock->bind.addr.addresslen)
 825                 return NULL;
 826         return &iosock->bind.addr;
 827 }
 828
 829 static int iosocket_try_write(struct _IOSocket *iosock) {
 830         if(!iosock->writebuf.bufpos && !(iosock->socket_flags & IOSOCKETFLAG_SSL_WRITEHS)) \r
 831                 return 0;
 832         iolog_trigger(IOLOG_DEBUG, "write writebuf (%d bytes) to socket (fd: %d)", iosock->writebuf.bufpos, iosock->fd);
 833         int res;
 834         if((iosock->socket_flags & IOSOCKETFLAG_SSLSOCKET))
 835                 res = iossl_write(iosock, iosock->writebuf.buffer, iosock->writebuf.bufpos);
 836         else
 837                 res = send(iosock->fd, iosock->writebuf.buffer, iosock->writebuf.bufpos, 0);
 838         if(res < 0) {
 839                 if (errno != EAGAIN && errno != EWOULDBLOCK)
 840                         iolog_trigger(IOLOG_ERROR, "could not write to socket (fd: %d): %d - %s", iosock->fd, errno, strerror(errno));
 841                 else
 842                         res = 0;
 843         } else {
 844                 iosock->writebuf.bufpos -= res;
 845                 if((iosock->socket_flags & (IOSOCKETFLAG_ACTIVE | IOSOCKETFLAG_SHUTDOWN)) == IOSOCKETFLAG_ACTIVE)
 846                         engine->update(iosock);
 847         }
 848         return res;
 849 }
 850
 851 void iosocket_send(struct IOSocket *iosocket, const char *data, size_t datalen) {
 852         struct _IOSocket *iosock = iosocket->iosocket;
 853         if(iosock == NULL) {
 854                 iolog_trigger(IOLOG_WARNING, "called iosocket_close for destroyed IOSocket in %s:%d", __FILE__, __LINE__);
 855                 return;
 856         }
 857         if(iosock->socket_flags & IOSOCKETFLAG_SHUTDOWN) {
 858                 iolog_trigger(IOLOG_ERROR, "could not write to socket (socket is closing)");
 859                 return;
 860         }
 861         iolog_trigger(IOLOG_DEBUG, "add %d to writebuf (fd: %d): %s", datalen, iosock->fd, data);
 862         if(iosock->writebuf.buflen < iosock->writebuf.bufpos + datalen) {
 863                 iolog_trigger(IOLOG_DEBUG, "increase writebuf (curr: %d) to %d (+%d bytes)", iosock->writebuf.buflen, iosock->writebuf.bufpos + datalen, (iosock->writebuf.bufpos + datalen - iosock->writebuf.buflen));
 864                 iosocket_increase_buffer(&iosock->writebuf, iosock->writebuf.bufpos + datalen);
 865                 if(iosock->writebuf.buflen < iosock->writebuf.bufpos + datalen) {
 866                         iolog_trigger(IOLOG_ERROR, "increase writebuf (curr: %d) to %d (+%d bytes) FAILED", iosock->writebuf.buflen, iosock->writebuf.bufpos + datalen, (iosock->writebuf.bufpos + datalen - iosock->writebuf.buflen));
 867                         return;
 868                 }
 869         }
 870         memcpy(iosock->writebuf.buffer + iosock->writebuf.bufpos, data, datalen);
 871         iosock->writebuf.bufpos += datalen;
 872         if((iosock->socket_flags & IOSOCKETFLAG_ACTIVE))
 873                 engine->update(iosock);
 874 }
 875
 876 void iosocket_write(struct IOSocket *iosocket, const char *line) {
 877         size_t linelen = strlen(line);
 878         iosocket_send(iosocket, line, linelen);
 879 }
 880
 881 void iosocket_printf(struct IOSocket *iosocket, const char *text, ...) {
 882         va_list arg_list;
 883         char sendBuf[IOSOCKET_PRINTF_LINE_LEN];
 884         int pos;
 885         sendBuf[0] = '\0';
 886         va_start(arg_list, text);
 887         pos = vsnprintf(sendBuf, IOSOCKET_PRINTF_LINE_LEN - 1, text, arg_list);
 888         va_end(arg_list);
 889         if (pos < 0 || pos > (IOSOCKET_PRINTF_LINE_LEN - 1)) pos = IOSOCKET_PRINTF_LINE_LEN - 1;
 890         sendBuf[pos] = '\0';
 891         iosocket_send(iosocket, sendBuf, pos);
 892 }
 893
 894
 895 int iosocket_wants_reads(struct _IOSocket *iosock) {
 896         if((iosock->socket_flags & (IOSOCKETFLAG_SSL_READHS | IOSOCKETFLAG_SSL_WRITEHS)))
 897                 return ((iosock->socket_flags & IOSOCKETFLAG_SSL_WANTWRITE) ? 0 : 1);
 898         if(!(iosock->socket_flags & IOSOCKETFLAG_OVERRIDE_WANT_RW))
 899                 return 1;
 900         else if((iosock->socket_flags & IOSOCKETFLAG_OVERRIDE_WANT_R))
 901                 return 1;
 902         return 0;
 903 }
 904 int iosocket_wants_writes(struct _IOSocket *iosock) {
 905         if((iosock->socket_flags & (IOSOCKETFLAG_SSL_READHS | IOSOCKETFLAG_SSL_WRITEHS)))
 906                 return ((iosock->socket_flags & IOSOCKETFLAG_SSL_WANTWRITE) ? 1 : 0);
 907         if(!(iosock->socket_flags & IOSOCKETFLAG_OVERRIDE_WANT_RW)) {
 908                 if(iosock->writebuf.bufpos || (iosock->socket_flags & IOSOCKETFLAG_CONNECTING))
 909                         return 1;
 910                 else
 911                         return 0;
 912         } else if((iosock->socket_flags & IOSOCKETFLAG_OVERRIDE_WANT_W))
 913                 return 1;
 914         return 0;
 915 }
 916
 917
 918 static void iosocket_trigger_event(struct IOSocketEvent *event) {
 919         if(!event->socket->callback)
 920                 return;
 921         iolog_trigger(IOLOG_DEBUG, "triggering event");
 922         event->socket->callback(event);
 923 }
 924
 925 void iosocket_events_callback(struct _IOSocket *iosock, int readable, int writeable) {
 926         if((iosock->socket_flags & IOSOCKETFLAG_PARENT_PUBLIC)) {
 927                 struct IOSocket *iosocket = iosock->parent;
 928                 struct IOSocketEvent callback_event;
 929                 callback_event.type = IOSOCKETEVENT_IGNORE;
 930                 callback_event.socket = iosocket;
 931
 932                 if((iosock->socket_flags & IOSOCKETFLAG_SSL_HANDSHAKE)) {
 933                         if(readable || writeable) {
 934                                 if((iosock->socket_flags & IOSOCKETFLAG_INCOMING))
 935                                         iossl_server_handshake(iosock);
 936                                 else
 937                                         iossl_client_handshake(iosock);
 938                                 engine->update(iosock);\r
 939                         } else if((iosock->socket_flags & IOSOCKETFLAG_LISTENING)) {\r
 940                                 //TODO: SSL init error\r
 941                         } else if((iosock->socket_flags & IOSOCKETFLAG_INCOMING)) {\r
 942                                 if((iosock->socket_flags & IOSOCKETFLAG_SSL_ESTABLISHED)) {\r
 943                                         //incoming SSL connection accepted\r
 944                                         iosock->socket_flags &= ~IOSOCKETFLAG_SSL_HANDSHAKE;\r
 945                                         callback_event.type = IOSOCKETEVENT_ACCEPT;\r
 946                                         callback_event.data.accept_socket = iosock->parent;
 947                                         struct _IOSocket *parent_socket = iosocket->data;
 948                                         callback_event.socket = parent_socket->parent;
 949
 950                                         //initialize readbuf
 951                                         iosocket_increase_buffer(&iosock->readbuf, 1024);\r
 952                                 } else {\r
 953                                         //incoming SSL connection failed, simply drop\r
 954                                         iosock->socket_flags |= IOSOCKETFLAG_DEAD;\r
 955                                         iolog_trigger(IOLOG_ERROR, "SSL Handshake failed for incoming connection. Dropping fd %d", iosock->fd);\r
 956                                 }
 957                         } else {
 958                                 // SSL Backend finished
 959                                 if((iosock->socket_flags & IOSOCKETFLAG_SSL_ESTABLISHED)) {\r
 960                                         iosocket->status = IOSOCKET_CONNECTED;\r
 961                                         iosock->socket_flags &= ~IOSOCKETFLAG_SSL_HANDSHAKE;\r
 962                                         callback_event.type = IOSOCKETEVENT_CONNECTED;\r
 963                                         engine->update(iosock);\r
 964                                         \r
 965                                         //initialize readbuf\r
 966                                         iosocket_increase_buffer(&iosock->readbuf, 1024);
 967                                 } else {
 968                                         callback_event.type = IOSOCKETEVENT_NOTCONNECTED;\r
 969                                         iosock->socket_flags |= IOSOCKETFLAG_DEAD;
 970                                 }
 971                         }
 972                 } else if((iosock->socket_flags & IOSOCKETFLAG_LISTENING)) {
 973                         if(readable) {
 974                                 //new client connected
 975                                 struct _IOSocket *new_iosock = iosocket_accept_client(iosock);
 976                                 if(!new_iosock)
 977                                         return;
 978
 979                                 if(!(new_iosock->socket_flags & IOSOCKETFLAG_SSLSOCKET)) {
 980                                         callback_event.type = IOSOCKETEVENT_ACCEPT;
 981                                         callback_event.data.accept_socket = new_iosock->parent;
 982                                 }
 983                         }
 984
 985                 } else if((iosock->socket_flags & IOSOCKETFLAG_CONNECTING)) {
 986                         if(readable) { //could not connect
 987                                 if((iosock->socket_flags & (IOSOCKETFLAG_IPV6SOCKET | IOSOCKETFLAG_RECONNECT_IPV4)) == (IOSOCKETFLAG_IPV6SOCKET | IOSOCKETFLAG_RECONNECT_IPV4)) {
 988                                         iolog_trigger(IOLOG_DEBUG, "connecting to IPv6 host (%s) failed. trying to connect using IPv4.", iosock->dest.addrlookup->hostname);
 989                                         iosocket_deactivate(iosock);
 990                                         if(iosocket_lookup_apply(iosock, 1)) { //if ret=0 an error occured in iosocket_lookup_apply and we should stop here.
 991                                                 iosocket_connect_finish(iosock);
 992                                                 socket_lookup_clear(iosock);
 993                                         }
 994                                 } else {
 995                                         callback_event.type = IOSOCKETEVENT_NOTCONNECTED;
 996                                         /*
 997                                         socklen_t arglen = sizeof(callback_event.data.errid);
 998                                         if (getsockopt(iosock->fd, SOL_SOCKET, SO_ERROR, &callback_event.data.errid, &arglen) < 0)
 999                                                 callback_event.data.errid = errno;
1000                                         */
1001                                         iosock->socket_flags |= IOSOCKETFLAG_DEAD;
1002                                 }
1003                         } else if(writeable) { //connection established
1004                                 iosock->socket_flags &= ~IOSOCKETFLAG_CONNECTING;
1005                                 socket_lookup_clear(iosock);
1006                                 if((iosock->socket_flags & IOSOCKETFLAG_SSLSOCKET)) {\r
1007                                         iolog_trigger(IOLOG_DEBUG, "SSL client socket connected. Stating SSL handshake...");
1008                                         iossl_connect(iosock);
1009                                         engine->update(iosock);
1010                                         return;
1011                                 }
1012                                 iosocket->status = IOSOCKET_CONNECTED;
1013
1014                                 callback_event.type = IOSOCKETEVENT_CONNECTED;
1015                                 engine->update(iosock);
1016
1017                                 //initialize readbuf
1018                                 iosocket_increase_buffer(&iosock->readbuf, 1024);
1019                         }
1020                 } else {\r
1021                         int ssl_rehandshake = 0;\r
1022                         if((iosock->socket_flags & IOSOCKETFLAG_SSLSOCKET)) {\r
1023                                 if((iosock->socket_flags & IOSOCKETFLAG_SSL_READHS))\r
1024                                         ssl_rehandshake = 1;\r
1025                                 else if((iosock->socket_flags & IOSOCKETFLAG_SSL_WRITEHS))\r
1026                                         ssl_rehandshake = 2;\r
1027                         }
1028                         iosocketevents_callback_retry_read:
1029                         if((readable && ssl_rehandshake == 0) || ssl_rehandshake == 1) {
1030                                 int bytes;
1031                                 if(iosock->readbuf.buflen - iosock->readbuf.bufpos <= 128) {
1032                                         int addsize;
1033                                         if(iosock->readbuf.buflen >= 2048)
1034                                                 addsize = 1024;
1035                                         else
1036                                                 addsize = iosock->readbuf.buflen;
1037                                         if(addsize == 0) {
1038                                                 iolog_trigger(IOLOG_WARNING, "readbuf length is 0 when trying to read from fd %d", iosock->fd);
1039                                                 addsize = 512;
1040                                         }
1041                                         iosocket_increase_buffer(&iosock->readbuf, iosock->readbuf.buflen + addsize);
1042                                 }
1043                                 if((iosock->socket_flags & IOSOCKETFLAG_SSLSOCKET))
1044                                         bytes = iossl_read(iosock, iosock->readbuf.buffer + iosock->readbuf.bufpos, iosock->readbuf.buflen - iosock->readbuf.bufpos);
1045                                 else
1046                                         bytes = recv(iosock->fd, iosock->readbuf.buffer + iosock->readbuf.bufpos, iosock->readbuf.buflen - iosock->readbuf.bufpos, 0);
1047
1048                                 if(bytes <= 0) {\r
1049                                         if((iosock->socket_flags & (IOSOCKETFLAG_SSLSOCKET | IOSOCKETFLAG_SSL_READHS)) == (IOSOCKETFLAG_SSLSOCKET | IOSOCKETFLAG_SSL_READHS)) {\r
1050                                                 ssl_rehandshake = 1;\r
1051                                         } else if (errno != EAGAIN || errno != EWOULDBLOCK) {
1052                                                 iosock->socket_flags |= IOSOCKETFLAG_DEAD;
1053
1054                                                 callback_event.type = IOSOCKETEVENT_CLOSED;
1055                                                 callback_event.data.errid = errno;
1056                                         }
1057                                 } else {
1058                                         int i;
1059                                         iolog_trigger(IOLOG_DEBUG, "received %d bytes (fd: %d). readbuf position: %d", bytes, iosock->fd, iosock->readbuf.bufpos);
1060                                         iosock->readbuf.bufpos += bytes;
1061                                         int retry_read = (iosock->readbuf.bufpos == iosock->readbuf.buflen);
1062                                         callback_event.type = IOSOCKETEVENT_RECV;
1063
1064                                         if(iosocket->parse_delimiter) {
1065                                                 int j, used_bytes = 0;
1066                                                 for(i = 0; i < iosock->readbuf.bufpos; i++) {
1067                                                         int is_delimiter = 0;
1068                                                         for(j = 0; j < IOSOCKET_PARSE_DELIMITERS_COUNT; j++) {
1069                                                                 if(iosock->readbuf.buffer[i] == iosocket->delimiters[j]) {
1070                                                                         is_delimiter = 1;
1071                                                                         break;
1072                                                                 }
1073                                                         }
1074                                                         if(is_delimiter) {
1075                                                                 iosock->readbuf.buffer[i] = 0;
1076                                                                 callback_event.data.recv_str = iosock->readbuf.buffer + used_bytes;
1077                                                                 iolog_trigger(IOLOG_DEBUG, "parsed line (%d bytes): %s", i - used_bytes, iosock->readbuf.buffer + used_bytes);
1078                                                                 used_bytes = i+1;
1079                                                                 if(iosock->readbuf.buffer[i-1] != 0 || iosocket->parse_empty)
1080                                                                         iosocket_trigger_event(&callback_event);
1081                                                         }
1082                                                         #ifdef IOSOCKET_PARSE_LINE_LIMIT
1083                                                         else if(i + 1 - used_bytes >= IOSOCKET_PARSE_LINE_LIMIT) {
1084                                                                 iosock->readbuf.buffer[i] = 0;
1085                                                                 callback_event.data.recv_str = iosock->readbuf.buffer + used_bytes;
1086                                                                 iolog_trigger(IOLOG_DEBUG, "parsed and stripped line (%d bytes): %s", i - used_bytes, iosock->readbuf.buffer + used_bytes);
1087                                                                 for(; i < iosock->readbuf.bufpos; i++) { //skip the rest of the line
1088                                                                         is_delimiter = 0;
1089                                                                         if(iosock->readbuf.buffer[i] == iosocket->delimiters[j])
1090                                                                                 break;
1091                                                                 }
1092                                                                 used_bytes = i+1;
1093                                                                 iosocket_trigger_event(&callback_event);
1094                                                         }
1095                                                         #endif
1096                                                 }
1097                                                 if(used_bytes) {
1098                                                         if(used_bytes == iosock->readbuf.bufpos) {
1099                                                                 iosock->readbuf.bufpos = 0;
1100                                                                 iolog_trigger(IOLOG_DEBUG, "readbuf fully processed (set buffer position to 0)");
1101                                                         } else {
1102                                                                 iolog_trigger(IOLOG_DEBUG, "readbuf rest: %d bytes (used %d bytes)", iosock->readbuf.bufpos - used_bytes, used_bytes);
1103                                                                 memmove(iosock->readbuf.buffer, iosock->readbuf.buffer + used_bytes, iosock->readbuf.bufpos - used_bytes);
1104                                                                 iosock->readbuf.bufpos -= used_bytes;
1105                                                         }
1106                                                 }
1107                                                 callback_event.type = IOSOCKETEVENT_IGNORE;
1108                                         } else
1109                                                 callback_event.data.recv_buf = &iosock->readbuf;
1110                                         if(retry_read)
1111                                                 goto iosocketevents_callback_retry_read;
1112                                 }
1113                         }
1114                         if((writeable && ssl_rehandshake == 0) || ssl_rehandshake == 2) {
1115                                 int bytes;
1116                                 bytes = iosocket_try_write(iosock);
1117                                 if(bytes < 0) {\r
1118                                         if((iosock->socket_flags & (IOSOCKETFLAG_SSLSOCKET | IOSOCKETFLAG_SSL_WRITEHS)) == (IOSOCKETFLAG_SSLSOCKET | IOSOCKETFLAG_SSL_WRITEHS)) {\r
1119                                                 ssl_rehandshake = 1;\r
1120                                         } else {
1121                                                 iosock->socket_flags |= IOSOCKETFLAG_DEAD;
1122
1123                                                 callback_event.type = IOSOCKETEVENT_CLOSED;
1124                                                 callback_event.data.errid = errno;\r
1125                                         }
1126                                 }
1127                         }\r
1128                         if(ssl_rehandshake) {
1129                                 engine->update(iosock);\r
1130                         }
1131                 }
1132                 if(callback_event.type != IOSOCKETEVENT_IGNORE)
1133                         iosocket_trigger_event(&callback_event);
1134                 if((iosock->socket_flags & IOSOCKETFLAG_DEAD))
1135                         iosocket_close(iosocket);
1136
1137         } else if((iosock->socket_flags & IOSOCKETFLAG_PARENT_DNSENGINE)) {
1138                 iodns_socket_callback(iosock, readable, writeable);
1139         }
1140 }
1141
1142 void iosocket_loop(int usec) {
1143         struct timeval timeout;
1144         timeout.tv_sec = usec / 1000000;
1145         timeout.tv_usec = usec % 1000000;
1146         engine->loop(&timeout);
1147 }