| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809 |
- /*
- * net.c -- handles:
- * all raw network i/o
- *
- */
- #include <fcntl.h>
- #include "common.h"
- #include "net.h"
- #include "misc.h"
- #include "main.h"
- #include "debug.h"
- #include "dccutil.h"
- #include "crypt.h"
- #include "egg_timer.h"
- #include "traffic.h"
- #include <limits.h>
- #include <string.h>
- #include <netdb.h>
- #include <signal.h>
- #include <sys/types.h>
- #include <sys/socket.h>
- #include <setjmp.h>
- #if HAVE_SYS_SELECT_H
- # include <sys/select.h>
- #endif /* HAVE_SYS_SELECT_H */
- #include <netinet/in.h>
- #include <arpa/inet.h>
- #include <errno.h>
- #include <sys/stat.h>
- #if HAVE_UNISTD_H
- # include <unistd.h>
- #endif /* HAVE_UNITSTD_H */
- extern egg_traffic_t traffic;
- #ifdef HAVE_SSL
- SSL_CTX *ssl_c_ctx = NULL, *ssl_s_ctx = NULL;
- char *tls_rand_file = NULL;
- #endif /* HAVE_SSL */
- union sockaddr_union cached_myip4_so;
- #ifdef USE_IPV6
- union sockaddr_union cached_myip6_so;
- #endif /* USE_IPV6 */
- char firewall[121] = ""; /* Socks server for firewall */
- port_t firewallport = 1080; /* Default port of Sock4/5 firewalls */
- char botuser[21] = ""; /* Username of the user running the bot */
- int resolve_timeout = 10; /* hostname/address lookup timeout */
- sock_list *socklist = NULL; /* Enough to be safe */
- int MAXSOCKS = 0;
- jmp_buf alarmret; /* Env buffer for alarm() returns */
- /* Types of proxy */
- #define PROXY_SOCKS 1
- #define PROXY_SUN 2
- /* I need an UNSIGNED long for dcc type stuff
- */
- unsigned long my_atoul(char *s)
- {
- unsigned long ret = 0;
- while ((*s >= '0') && (*s <= '9')) {
- ret *= 10;
- ret += ((*s) - '0');
- s++;
- }
- return ret;
- }
- int hostprotocol(char *host)
- {
- #ifdef USE_IPV6
- struct hostent *he = NULL;
- # ifndef HAVE_GETHOSTBYNAME2
- int error_num;
- # endif /* !HAVE_GETHOSTBYNAME2 */
- if (!host || (host && !host[0]))
- return 0;
- if (!setjmp(alarmret)) {
- alarm(resolve_timeout);
- # ifdef HAVE_GETHOSTBYNAME2
- he = gethostbyname2(host, AF_INET6);
- # else
- he = getipnodebyname(host, AF_INET6, AI_DEFAULT, &error_num);
- # endif /* HAVE_GETHOSTBYNAME2 */
- alarm(0);
- } else
- he = NULL;
- if (!he)
- return AF_INET;
- return AF_INET6;
- #else
- return 0;
- #endif /* USE_IPV6 */
- }
- /* get the protocol used on a socket */
- int sockprotocol(int sock)
- {
- struct sockaddr sa;
- int i = sizeof(sa);
- if (getsockname(sock, &sa, &i))
- return -1;
- else
- return sa.sa_family;
- }
- /* AF_INET-independent resolving routine */
- int get_ip(char *hostname, union sockaddr_union *so)
- {
- #ifdef USE_IPV6
- struct addrinfo hints, *ai = NULL, *res = NULL;
- int error = 0;
- #else
- struct hostent *hp = NULL;
- #endif /* USE_IPV6 */
- egg_memset(so, 0, sizeof(union sockaddr_union));
- debug1("get_ip(%s)", hostname);
- if (!hostname || (hostname && !hostname[0]))
- return 1;
- #ifdef USE_IPV6
- egg_memset(&hints, 0, sizeof(struct addrinfo));
- hints.ai_socktype = SOCK_STREAM;
- if ((error = getaddrinfo(hostname, NULL, &hints, &res)))
- return error;
- error = 1;
- for (ai = res; ai != NULL; ai = ai->ai_next) {
- if ((ai->ai_family == AF_INET6) || (ai->ai_family == AF_INET)) {
- memcpy(so, ai->ai_addr, ai->ai_addrlen);
- error = 0;
- break;
- }
- }
- freeaddrinfo(res);
- return error;
- #else
- if (!(hp = gethostbyname(hostname)))
- return -1;
- memcpy(&so->sin.sin_addr, hp->h_addr, 4);
- so->sin.sin_family = AF_INET;
- return 0;
- #endif /* USE_IPV6 */
- }
- #ifdef HAVE_SSL
- int seed_PRNG(void)
- {
- char stackdata[1024] = "";
- static char rand_file[300] = "";
- FILE *fh = NULL;
- #if OPENSSL_VERSION_NUMBER >= 0x00905100
- if (RAND_status()) return 0;
- #endif /* OPENSSL_VERSION_NUMBER */
- if ((fh = fopen("/dev/urandom", "r"))) {
- fclose(fh);
- return 0;
- }
- if (RAND_file_name(rand_file, sizeof(rand_file)))
- tls_rand_file = rand_file;
- else
- return 1;
- if (!RAND_load_file(rand_file, 1024)) {
- unsigned int c;
- c = now;
- RAND_seed(&c, sizeof(c));
- c = getpid();
- RAND_seed(&c, sizeof(c));
- RAND_seed(stackdata, sizeof(stackdata));
- }
- #if OPENSSL_VERSION_NUMBER >= 0x00905100
- if (!RAND_status()) return 2;
- #endif /* OPENSSL_VERSION_NUMBER >= 0x00905100 */
- return 0;
- }
- #endif /* HAVE_SSL */
- /* Initialize the socklist
- */
- void init_net()
- {
- int i;
- for (i = 0; i < MAXSOCKS; i++) {
- egg_bzero(&socklist[i], sizeof(socklist[i]));
- #ifdef HAVE_SSL
- socklist[i].ssl=NULL;
- #endif /* HAVE_SSL */
- socklist[i].flags = SOCK_UNUSED;
- }
- #ifdef HAVE_SSL
- SSL_load_error_strings();
- OpenSSL_add_ssl_algorithms();
- ssl_c_ctx = SSL_CTX_new(SSLv23_client_method());
- ssl_s_ctx = SSL_CTX_new(SSLv23_server_method());
- if (!ssl_c_ctx || !ssl_s_ctx)
- fatal("SSL Inititlization failed", 0);
- if (seed_PRNG())
- fatal("SSL PRNG seeding failed!", 0);
- #endif /* HAVE_SSL */
- }
- #ifdef HAVE_SSL
- int ssl_cleanup() {
- if (ssl_c_ctx) {
- SSL_CTX_free(ssl_c_ctx);
- ssl_c_ctx = NULL;
- }
- if (ssl_s_ctx) {
- SSL_CTX_free(ssl_s_ctx);
- ssl_s_ctx = NULL;
- }
- if (tls_rand_file) RAND_write_file(tls_rand_file);
- return 0;
- }
- #endif /* HAVE_SSL */
- /* Get my ipv? ip
- */
- char *myipstr(int af_type)
- {
- #ifdef USE_IPV6
- if (af_type == 6) {
- static char s[UHOSTLEN + 1] = "";
- egg_inet_ntop(AF_INET6, &cached_myip6_so.sin6.sin6_addr, s, 119);
- s[120] = 0;
- return s;
- } else
- #endif /* USE_IPV6 */
- if (af_type == 4) {
- static char s[UHOSTLEN + 1] = "";
- egg_inet_ntop(AF_INET, &cached_myip4_so.sin.sin_addr, s, 119);
- s[120] = 0;
- return s;
- }
- return "";
- }
- /* Get my ip number
- */
- IP getmyip() {
- return (IP) cached_myip4_so.sin.sin_addr.s_addr;
- }
- /* see if it's necessary to set inaddr_any... because if we can't resolve, we die anyway */
- void cache_my_ip()
- {
- char s[121] = "";
- int error;
- #ifdef USE_IPV6
- int any = 0;
- #endif /* USE_IPV6 */
- debug0("cache_my_ip()");
- egg_memset(&cached_myip4_so, 0, sizeof(union sockaddr_union));
- #ifdef USE_IPV6
- egg_memset(&cached_myip6_so, 0, sizeof(union sockaddr_union));
- if (conf.bot->ip6) {
- sdprintf("ip6: %s", conf.bot->ip6);
- if (get_ip(conf.bot->ip6, &cached_myip6_so))
- any = 1;
- } else if (conf.bot->host6) {
- sdprintf("host6: %s", conf.bot->host6);
- if (get_ip(conf.bot->host6, &cached_myip6_so))
- any = 1;
- } else
- any = 1;
- if (any) {
- sdprintf("IPV6 addr_any is set.");
- cached_myip6_so.sin6.sin6_family = AF_INET6;
- cached_myip6_so.sin6.sin6_addr = in6addr_any;
- }
- #endif /* USE_IPV6 */
- error = 0;
- if (conf.bot->ip) {
- if (get_ip(conf.bot->ip, &cached_myip4_so))
- error = 1;
- } else if (conf.bot->host) {
- if (get_ip(conf.bot->host, &cached_myip4_so))
- error = 2;
- } else {
- gethostname(s, sizeof(s));
- if (get_ip(s, &cached_myip4_so)) {
- /* error = 3; */
- sdprintf("IPV4 addr_any is set.");
- cached_myip4_so.sin.sin_family = AF_INET;
- cached_myip4_so.sin.sin_addr.s_addr = INADDR_ANY;
- }
- }
- if (error) {
- putlog(LOG_DEBUG, "*", "Hostname self-lookup error: %d", error);
- fatal("Hostname self-lookup failed.", 0);
- }
- }
- void neterror(char *s)
- {
- switch (errno) {
- case EADDRINUSE:
- strcpy(s, "Address already in use");
- break;
- case EADDRNOTAVAIL:
- strcpy(s, "Address invalid on remote machine");
- break;
- case EAFNOSUPPORT:
- strcpy(s, "Address family not supported");
- break;
- case EALREADY:
- strcpy(s, "Socket already in use");
- break;
- case EBADF:
- strcpy(s, "Socket descriptor is bad");
- break;
- case ECONNREFUSED:
- strcpy(s, "Connection refused");
- break;
- case EFAULT:
- strcpy(s, "Namespace segment violation");
- break;
- case EINPROGRESS:
- strcpy(s, "Operation in progress");
- break;
- case EINTR:
- strcpy(s, "Timeout");
- break;
- case EINVAL:
- strcpy(s, "Invalid namespace");
- break;
- case EISCONN:
- strcpy(s, "Socket already connected");
- break;
- case ENETUNREACH:
- strcpy(s, "Network unreachable");
- break;
- case ENOTSOCK:
- strcpy(s, "File descriptor, not a socket");
- break;
- case ETIMEDOUT:
- strcpy(s, "Connection timed out");
- break;
- case ENOTCONN:
- strcpy(s, "Socket is not connected");
- break;
- case EHOSTUNREACH:
- strcpy(s, "Host is unreachable");
- break;
- case EPIPE:
- strcpy(s, "Broken pipe");
- break;
- #ifdef ECONNRESET
- case ECONNRESET:
- strcpy(s, "Connection reset by peer");
- break;
- #endif /* ECONNRESET */
- #ifdef EACCES
- case EACCES:
- strcpy(s, "Permission denied");
- break;
- #endif /* EACCESS */
- #ifdef EMFILE
- case EMFILE:
- strcpy(s, "Too many open files");
- break;
- #endif /* EMFILE */
- case 0:
- strcpy(s, "Error 0");
- break;
- default:
- sprintf(s, "Unforseen error %d", errno);
- break;
- }
- }
- /* Sets/Unsets options for a specific socket.
- *
- * Returns: 0 - on success
- * -1 - socket not found
- * -2 - illegal operation
- */
- int sockoptions(int sock, int operation, int sock_options)
- {
- int i;
- for (i = 0; i < MAXSOCKS; i++)
- if ((socklist[i].sock == sock) && !(socklist[i].flags & SOCK_UNUSED)) {
- if (operation == EGG_OPTION_SET)
- socklist[i].flags |= sock_options;
- else if (operation == EGG_OPTION_UNSET)
- socklist[i].flags &= ~sock_options;
- else
- return -2;
- return 0;
- }
- return -1;
- }
- /* Return a free entry in the socket entry
- */
- int allocsock(int sock, int options)
- {
- int i;
- for (i = 0; i < MAXSOCKS; i++) {
- if (socklist[i].flags & SOCK_UNUSED) {
- /* yay! there is table space */
- socklist[i].inbuf = socklist[i].outbuf = NULL;
- socklist[i].inbuflen = socklist[i].outbuflen = 0;
- #ifdef HAVE_SSL
- socklist[i].ssl = NULL;
- #endif /* HAVE_SSL */
- socklist[i].flags = options;
- socklist[i].sock = sock;
- socklist[i].encstatus = 0;
- socklist[i].gz = 0;
- egg_bzero(&socklist[i].okey, sizeof(socklist[i].okey));
- egg_bzero(&socklist[i].ikey, sizeof(socklist[i].ikey));
- socklist[i].okey[0] = 0;
- socklist[i].ikey[0] = 0;
- return i;
- }
- }
- fatal("Socket table is full!", 0);
- return -1; /* Never reached */
- }
- /* Request a normal socket for i/o
- */
- void setsock(int sock, int options)
- {
- int i = allocsock(sock, options), parm;
- if (((sock != STDOUT) || backgrd) && !(socklist[i].flags & SOCK_NONSOCK)) {
- parm = 1;
- setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE, (void *) &parm, sizeof(int));
- parm = 0;
- setsockopt(sock, SOL_SOCKET, SO_LINGER, (void *) &parm, sizeof(int));
- }
- if (options & SOCK_LISTEN) {
- /* Tris says this lets us grab the same port again next time */
- parm = 1;
- setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (void *) &parm, sizeof(int));
- }
- /* Yay async i/o ! */
- fcntl(sock, F_SETFL, O_NONBLOCK);
- }
- #ifdef USE_IPV6
- int real_getsock(int options, int af_def, char *fname, int line)
- {
- #else
- int real_getsock(int options, char *fname, int line)
- {
- int af_def = AF_INET;
- #endif /* USE_IPV6 */
- int sock;
- sock = socket(af_def, SOCK_STREAM, 0);
- if (sock >= 0)
- setsock(sock, options);
- else
- putlog(LOG_WARNING, "*", "Warning: Can't create new socket! (%s:%d)", fname, line);
- return sock;
- }
- void dropssl(register int sock)
- {
- #ifdef HAVE_SSL
- int i;
- if (sock < 0)
- return;
- for (i = 0; (i < MAXSOCKS); i++)
- if (socklist[i].sock == sock) break;
- if (socklist[i].ssl) {
- SSL_set_quiet_shutdown(socklist[i].ssl, 1);
- SSL_shutdown(socklist[i].ssl);
- usleep(1000 * 500);
- SSL_free(socklist[i].ssl);
- usleep(1000 * 500);
- socklist[i].ssl = NULL;
- }
- #endif /* HAVE_SSL */
- }
- /* Done with a socket
- */
- void real_killsock(register int sock, const char *file, int line)
- {
- register int i;
- /* Ignore invalid sockets. */
- if (sock < 0)
- return;
- for (i = 0; i < MAXSOCKS; i++) {
- if ((socklist[i].sock == sock) && !(socklist[i].flags & SOCK_UNUSED)) {
- dropssl(sock);
- close(socklist[i].sock);
- if (socklist[i].inbuf != NULL) {
- free(socklist[i].inbuf);
- socklist[i].inbuf = NULL;
- }
- if (socklist[i].outbuf != NULL) {
- free(socklist[i].outbuf);
- socklist[i].outbuf = NULL;
- socklist[i].outbuflen = 0;
- }
- egg_bzero(&socklist[i], sizeof(socklist[i]));
- socklist[i].flags = SOCK_UNUSED;
- return;
- }
- }
- putlog(LOG_MISC, "*", "Attempt to kill un-allocated socket %d %s:%d !!", sock, file, line);
- }
- /* Send connection request to proxy
- */
- static int proxy_connect(int sock, char *host, int port, int proxy)
- {
- #ifdef USE_IPV6
- unsigned char x[32] = "";
- int af_ty;
- #else
- unsigned char x[10] = "";
- #endif /* USE_IPV6 */
- struct hostent *hp = NULL;
- char s[256] = "";
- int i;
- #ifdef USE_IPV6
- af_ty = sockprotocol(sock);
- #endif /* USE_IPV6 */
- /* socks proxy */
- if (proxy == PROXY_SOCKS) {
- /* numeric IP? */
- #ifdef USE_IPV6
- if ((host[strlen(host) - 1] >= '0' && host[strlen(host) - 1] <= '9') && af_ty != AF_INET6) {
- #else
- if (host[strlen(host) - 1] >= '0' && host[strlen(host) - 1] <= '9') {
- #endif /* USE_IPV6 */
- IP ip = ((IP) inet_addr(host));
- egg_memcpy(x, &ip, 4);
- } else {
- /* no, must be host.domain */
- if (!setjmp(alarmret)) {
- #ifdef USE_IPV6
- alarm(resolve_timeout);
- if (af_ty == AF_INET6)
- hp = gethostbyname(host);
- else
- #endif /* USE_IPV6 */
- hp = gethostbyname(host);
- #ifdef USE_IPV6
- alarm(0);
- #endif /* USE_IPV6 */
- } else
- hp = NULL;
- if (hp == NULL) {
- killsock(sock);
- return -2;
- }
- egg_memcpy(x, hp->h_addr, hp->h_length);
- }
- for (i = 0; i < MAXSOCKS; i++)
- if (!(socklist[i].flags & SOCK_UNUSED) && socklist[i].sock == sock)
- socklist[i].flags |= SOCK_PROXYWAIT; /* drummer */
- #ifdef USE_IPV6
- if (af_ty == AF_INET6)
- egg_snprintf(s, sizeof s,
- "\004\001%c%c%c%c%c%c%c%c%c%c%c%c%c%c%c%c%c%c%s",
- (port >> 8) % 256, (port % 256), x[0], x[1], x[2], x[3],
- x[4], x[5], x[6], x[7], x[9], x[9], x[10], x[11], x[12],
- x[13], x[14], x[15], botuser);
- else
- #endif /* USE_IPV6 */
- egg_snprintf(s, sizeof s, "\004\001%c%c%c%c%c%c%s", (port >> 8) % 256,
- (port % 256), x[0], x[1], x[2], x[3], botuser);
- tputs(sock, s, strlen(botuser) + 9); /* drummer */
- } else if (proxy == PROXY_SUN) {
- egg_snprintf(s, sizeof s, "%s %d\n", host, port);
- tputs(sock, s, strlen(s)); /* drummer */
- }
- return sock;
- }
- /* Starts a connection attempt to a socket
- *
- * If given a normal hostname, this will be resolved to the corresponding
- * IP address first. PLEASE try to use the non-blocking dns functions
- * instead and then call this function with the IP address to avoid blocking.
- *
- * returns <0 if connection refused:
- * -1 neterror() type error
- * -2 can't resolve hostname
- */
- int open_telnet_raw(int sock, char *server, port_t sport)
- {
- union sockaddr_union so;
- char host[121] = "";
- int i, error = 0, rc;
- port_t port;
- volatile int proxy;
- /* firewall? use socks */
- if (firewall[0]) {
- if (firewall[0] == '!') {
- proxy = PROXY_SUN;
- strcpy(host, &firewall[1]);
- } else {
- proxy = PROXY_SOCKS;
- strcpy(host, firewall);
- }
- port = firewallport;
- } else {
- proxy = 0;
- strncpyz(host, server, sizeof host);
- port = sport;
- }
- error = 0;
- if (!setjmp(alarmret)) {
- alarm(resolve_timeout);
- if (!get_ip(host, &so)) {
- alarm(0);
- /* ok, we resolved it, bind an appropriate ip */
- #ifdef USE_IPV6
- if (so.sa.sa_family == AF_INET6) {
- if (bind(sock, &cached_myip6_so.sa, SIZEOF_SOCKADDR(cached_myip6_so)) < 0) {
- killsock(sock);
- return -1;
- }
- } else {
- #endif /* USE_IPV6 */
- if (bind(sock, &cached_myip4_so.sa, SIZEOF_SOCKADDR(cached_myip4_so)) < 0) {
- killsock(sock);
- return -3;
- }
- #ifdef USE_IPV6
- }
- if (so.sa.sa_family == AF_INET6)
- so.sin6.sin6_port = htons(port);
- else
- #endif /* USE_IPV6 */
- so.sin.sin_port = htons(port);
- } else {
- alarm(0);
- error = 1;
- }
- }
- /* I guess we broke something */
- if (error) {
- killsock(sock);
- return -2;
- }
- for (i = 0; i < MAXSOCKS; i++) {
- if (!(socklist[i].flags & SOCK_UNUSED) && (socklist[i].sock == sock))
- socklist[i].flags = (socklist[i].flags & ~SOCK_VIRTUAL) | SOCK_CONNECT;
- }
- rc = connect(sock, &so.sa, SIZEOF_SOCKADDR(so));
- if (rc < 0) { if (errno == EINPROGRESS) {
- /* Firewall? announce connect attempt to proxy */
- if (firewall[0])
- return proxy_connect(sock, server, sport, proxy);
- return sock; /* async success! */
- } else
- return -1;
- }
- /* Synchronous? :/ */
- if (firewall[0])
- return proxy_connect(sock, server, sport, proxy);
- return sock;
- }
- /* Ordinary non-binary connection attempt */
- int open_telnet(char *server, port_t port)
- {
- #ifdef USE_IPV6
- int sock = getsock(0, hostprotocol(server)) , ret = open_telnet_raw(sock, server, port);
- #else
- int sock = getsock(0) , ret = open_telnet_raw(sock, server, port);
- #endif /* USE_IPV6 */
- return ret;
- }
- /* Returns a socket number for a listening socket that will accept any
- * connection on a certain address -- port # is returned in port
- *
- * 'addr' is ignored if af_def is AF_INET6 -poptix (02/03/03)
- */
- #ifdef USE_IPV6
- int open_address_listen(IP addr, int af_def, port_t *port)
- #else
- int open_address_listen(IP addr, port_t int *port)
- #endif /* USE_IPV6 */
- {
- int sock = 0;
- unsigned int addrlen;
- struct sockaddr_in name;
- if (firewall[0]) {
- /* FIXME: can't do listen port thru firewall yet */
- putlog(LOG_MISC, "*", "!! Cant open a listen port (you are using a "
- "firewall)");
- return -1;
- }
- #ifdef USE_IPV6
- if (af_def == AF_INET6) {
- struct sockaddr_in6 name6;
- sock = getsock(SOCK_LISTEN, af_def);
- if (sock < 1)
- return -1;
- debug2("Opening listen socket on port %d with AF_INET6, sock: %d", *port, sock);
- egg_bzero((char *) &name6, sizeof(name6));
- name6.sin6_family = af_def;
- name6.sin6_port = htons(*port); /* 0 = just assign us a port */
- /* memcpy(&name6.sin6_addr, &in6addr_any, 16); */ /* this is the only way to get ipv6+ipv4 in 1 socket */
- memcpy(&name6.sin6_addr, &cached_myip6_so.sin6.sin6_addr, 16);
- if (bind(sock, (struct sockaddr *) &name6, sizeof(name6)) < 0) {
- killsock(sock);
- return -1;
- }
- addrlen = sizeof(name6);
- if (getsockname(sock, (struct sockaddr *) &name6, &addrlen) < 0) {
- killsock(sock);
- return -1;
- }
- *port = ntohs(name6.sin6_port);
- if (listen(sock, 1) < 0) {
- killsock(sock);
- return -1;
- }
- } else {
- sock = getsock(SOCK_LISTEN, AF_INET);
- #else
- sock = getsock(SOCK_LISTEN);
- #endif /* USE_IPV6 */
- if (sock < 1)
- return -1;
- debug2("Opening listen socket on port %d with AF_INET, sock: %d", *port, sock);
- egg_bzero((char *) &name, sizeof(struct sockaddr_in));
- name.sin_family = AF_INET;
- name.sin_port = htons(*port); /* 0 = just assign us a port */
- name.sin_addr.s_addr = addr;
- if (bind(sock, (struct sockaddr *) &name, sizeof(name)) < 0) {
- killsock(sock);
- return -1;
- }
- /* what port are we on? */
- addrlen = sizeof(name);
- if (getsockname(sock, (struct sockaddr *) &name, &addrlen) < 0) {
- killsock(sock);
- return -1;
- }
- *port = ntohs(name.sin_port);
- if (listen(sock, 1) < 0) {
- killsock(sock);
- return -1;
- }
- #ifdef USE_IPV6
- }
- #endif /* USE_IPV6 */
- return sock;
- }
- /* Returns a socket number for a listening socket that will accept any
- * connection -- port # is returned in port
- */
- __inline__ int open_listen(port_t *port)
- {
- #ifdef USE_IPV6
- return open_address_listen(conf.bot->ip ? getmyip() : INADDR_ANY, AF_INET, port);
- #else
- return open_address_listen(conf.bot->ip ? getmyip() : INADDR_ANY, port);
- #endif /* USE_IPV6 */
- }
- /* Same as above, except this one can be called with an AF_ type
- * the above is being left in for compatibility, and should NOT LONGER BE USED IN THE CORE CODE.
- */
- __inline__ int open_listen_by_af(port_t *port, int af_def)
- {
- #ifdef USE_IPV6
- return open_address_listen(conf.bot->ip ? getmyip() : INADDR_ANY, af_def, port);
- #else
- return 0;
- #endif /* USE_IPV6 */
- }
- int ssl_link(register int sock, int state)
- {
- #ifdef HAVE_SSL
- int err = 0, i = 0, errs = 0;
- debug2("ssl_link(%d, %d)", sock, state);
- for (i = 0; (i < MAXSOCKS); i++) {
- if (socklist[i].sock == sock) break;
- }
- if (socklist[i].ssl) {
- putlog(LOG_ERROR, "*", "Switching to SSL (%d,%d) - already active", state, sock);
- return 0;
- }
- if (state == CONNECT_SSL) {
- socklist[i].ssl = SSL_new(ssl_c_ctx);
- } else if (state == ACCEPT_SSL) {
- socklist[i].ssl = SSL_new(ssl_s_ctx);
- }
- if (!socklist[i].ssl) {
- putlog(LOG_ERROR, "*", "Switching to SSL (%d) - SSL_new(%d) failed", sock, state);
- return 0;
- }
- if (!SSL_set_fd(socklist[i].ssl, socklist[i].sock)) {
- putlog(LOG_ERROR, "*", "SSL_set_fd(%d) (%d) failed", state, socklist[i].sock);
- return 0;
- }
- if (state == CONNECT_SSL) {
- SSL_set_connect_state(socklist[i].ssl);
- } else if (state == ACCEPT_SSL) {
- SSL_set_accept_state(socklist[i].ssl);
- } else {
- putlog(LOG_DEBUG, "*", "ssl_link(%d, 0?) NO STATE?", sock);
- return 0;
- }
- if (state == CONNECT_SSL) {
- err = SSL_connect(socklist[i].ssl);
- } else if (state == ACCEPT_SSL) {
- err = SSL_accept(socklist[i].ssl);
- }
- if (!setjmp(alarmret)) {
- alarm(5); /* this is plenty of time */
- while ((err < 1) && (errno == EAGAIN)) {
- if (state == CONNECT_SSL) {
- err = SSL_connect(socklist[i].ssl);
- } else if (state == ACCEPT_SSL) {
- err = SSL_accept(socklist[i].ssl);
- }
- /* if ((errs!=SSL_ERROR_WANT_READ)&&(errs!=SSL_ERROR_WANT_WRITE)&& (errs!=SSL_ERROR_WANT_X509_LOOKUP)) */
- /* break; anything not one of these is a sufficient condition to break out... */
- }
- alarm(0);
- }
- errs = SSL_get_error(socklist[i].ssl, err);
- putlog(LOG_DEBUG, "*", "SSL_link(%d, %d) = %d, errs: %d (%d), %s", sock, state, err, errs, errno, (char *)ERR_error_string(ERR_get_error(), NULL));
- if (errno) putlog(LOG_DEBUG, "*", "errno %d: %s", errno, strerror(errno));
- if (err == 1) {
- putlog(LOG_ERROR, "*", "SSL_link(%d, %d) was successfull", sock, state);
- return 1;
- } else {
- putlog(LOG_ERROR, "*", "SSL_link(%d, %d) failed", sock, state);
- dropssl(socklist[i].sock);
- }
- #endif /* HAVE_SSL */
- return 0;
- }
- /* Given a network-style IP address, returns the hostname. The hostname
- * will be in the "##.##.##.##" format if there was an error.
- *
- * NOTE: This function is depreciated. Try using the async dns approach
- * instead.
- */
- char *hostnamefromip(IP ip)
- {
- struct hostent *hp = NULL;
- IP addr = ip;
- unsigned char *p = NULL;
- static char s[UHOSTLEN] = "";
- if (!setjmp(alarmret)) {
- alarm(resolve_timeout);
- hp = gethostbyaddr((char *) &addr, sizeof(addr), AF_INET);
- alarm(0);
- } else {
- hp = NULL;
- }
- if (hp == NULL) {
- p = (unsigned char *) &addr;
- sprintf(s, "%u.%u.%u.%u", p[0], p[1], p[2], p[3]);
- return s;
- }
- strncpyz(s, hp->h_name, sizeof s);
- return s;
- }
- /* Returns the given network byte order IP address in the
- * dotted format - "##.##.##.##"
- */
- char *iptostr(IP ip)
- {
- static char ipbuf[32];
- struct in_addr a;
- a.s_addr = ip;
- return (char *) egg_inet_ntop(AF_INET, &a, ipbuf, sizeof(ipbuf));
- }
- /* Short routine to answer a connect received on a socket made previously
- * by open_listen ... returns hostname of the caller & the new socket
- * does NOT dispose of old "public" socket!
- */
- int answer(int sock, char *caller, IP *ip, port_t *port, int binary)
- {
- int new_sock;
- unsigned int addrlen;
- struct sockaddr_in from;
- #ifdef USE_IPV6
- int af_ty = sockprotocol(sock);
- struct sockaddr_in6 from6;
- egg_bzero(&from6, sizeof(struct sockaddr_in6));
- if (af_ty == AF_INET6) {
- addrlen = sizeof(from6);
- new_sock = accept(sock, (struct sockaddr *) &from6, &addrlen);
- } else {
- #endif /* USE_IPV6 */
- addrlen = sizeof(struct sockaddr);
- new_sock = accept(sock, (struct sockaddr *) &from, &addrlen);
- #ifdef USE_IPV6
- }
- #endif /* USE_IPV6 */
-
- if (new_sock < 0)
- return -1;
- if (ip != NULL) {
- #ifdef USE_IPV6
- /* Detect IPv4 in IPv6 mapped address .... */
- if (af_ty == AF_INET6 && (!IN6_IS_ADDR_V4MAPPED(&from6.sin6_addr))) {
- egg_inet_ntop(AF_INET6, &from6.sin6_addr, caller, 119);
- caller[120] = 0;
- *ip = 0L;
- } else if (IN6_IS_ADDR_V4MAPPED(&from6.sin6_addr)) { /* ...and convert it to plain (AF_INET) IPv4 address (openssh) */
- struct sockaddr_in *from4 = (struct sockaddr_in *)&from6;
- struct in_addr addr;
- memcpy(&addr, ((char *)&from6.sin6_addr) + 12, sizeof(addr));
- egg_memset(&from, 0, sizeof(from));
- from4->sin_family = AF_INET;
- addrlen = sizeof(*from4);
- memcpy(&from4->sin_addr, &addr, sizeof(addr));
- *ip = from4->sin_addr.s_addr;
- strncpyz(caller, iptostr(*ip), 121);
- *ip = ntohl(*ip);
- } else {
- #endif /* USE_IPV6 */
- *ip = from.sin_addr.s_addr;
- /* This is now done asynchronously. We now only provide the IP address.
- *
- * strncpy(caller, hostnamefromip(*ip), 120);
- */
- strncpyz(caller, iptostr(*ip), 121);
- *ip = ntohl(*ip);
- #ifdef USE_IPV6
- }
- #endif /* USE_IPV6 */
- }
- if (port != NULL) {
- #ifdef USE_IPV6
- if (af_ty == AF_INET6)
- *port = ntohs(from6.sin6_port);
- else
- #endif /* USE_IPV6 */
- *port = ntohs(from.sin_port);
- }
- /* Set up all the normal socket crap */
- setsock(new_sock, (binary ? SOCK_BINARY : 0));
- return new_sock;
- }
- /* Like open_telnet, but uses server & port specifications of dcc
- */
- int open_telnet_dcc(int sock, char *server, char *port)
- {
- int p;
- unsigned long addr;
- char sv[500] = "";
- unsigned char c[4] = "";
- #ifdef DEBUG_IPV6
- debug1("open_telnet_dcc %s", server);
- #endif /* DEBUG_IPV6 */
- if (port != NULL)
- p = atoi(port);
- else
- p = 2000;
- #ifdef USE_IPV6
- if (sockprotocol(sock) == AF_INET6) {
- # ifdef DEBUG_IPV6
- debug0("open_telnet_dcc, af_inet6!");
- # endif /* DEBUG_IPV6 */
- strncpyz(sv, server, sizeof sv);
- debug2("%s should be %s",sv,server);
- } else {
- #endif /* USE_IPV6 */
- if (server != NULL)
- addr = my_atoul(server);
- else
- addr = 0L;
- if (addr < (1 << 24))
- return -3; /* fake address */
- c[0] = (addr >> 24) & 0xff;
- c[1] = (addr >> 16) & 0xff;
- c[2] = (addr >> 8) & 0xff;
- c[3] = addr & 0xff;
- sprintf(sv, "%u.%u.%u.%u", c[0], c[1], c[2], c[3]);
- #ifdef USE_IPV6
- }
- /* strcpy(sv,hostnamefromip(addr)); */
- # ifdef DEBUG_IPV6
- debug3("open_telnet_raw %s %d %d", sv, sock,p);
- # endif /* DEBUG_IPV6 */
- #endif /* USE_IPV6 */
- p = open_telnet_raw(sock, sv, p);
- return p;
- }
- /* Attempts to read from all the sockets in socklist
- * fills s with up to 511 bytes if available, and returns the array index
- *
- * on EOF: returns -1, with socket in len
- * on socket error: returns -2
- * if nothing is ready: returns -3
- */
- static int sockread(char *s, int *len)
- {
- fd_set fd;
- int fds = 0, i, x, fdtmp;
- struct timeval t;
- int grab = SGRAB + 1;
- egg_timeval_t howlong;
- if (timer_get_shortest(&howlong)) {
- /* No timer, default to 1 second. */
- t.tv_sec = 1;
- t.tv_usec = 0;
- }
- else {
- t.tv_sec = howlong.sec;
- t.tv_usec = howlong.usec;
- }
- FD_ZERO(&fd);
-
- for (i = 0; i < MAXSOCKS; i++) {
- if (!(socklist[i].flags & (SOCK_UNUSED | SOCK_VIRTUAL))) {
- if ((socklist[i].sock == STDOUT) && !backgrd)
- fdtmp = STDIN;
- else
- fdtmp = socklist[i].sock;
- if (fdtmp > fds)
- fds = fdtmp;
- FD_SET(fdtmp, &fd);
- }
- }
- fds++;
- x = select(fds, &fd, NULL, NULL, &t);
- if (x > 0) {
- /* Something happened */
- for (i = 0; i < MAXSOCKS; i++) {
- if ((!(socklist[i].flags & SOCK_UNUSED)) && ((FD_ISSET(socklist[i].sock, &fd)) ||
- #ifdef HAVE_SSL
- ((socklist[i].ssl) && (SSL_pending(socklist[i].ssl))) ||
- #endif /* HAVE_SSL */
- ((socklist[i].sock == STDOUT) && (!backgrd) && (FD_ISSET(STDIN, &fd))))) {
- if (socklist[i].flags & (SOCK_LISTEN | SOCK_CONNECT)) {
- /* Listening socket -- don't read, just return activity */
- /* Same for connection attempt */
- /* (for strong connections, require a read to succeed first) */
- if (socklist[i].flags & SOCK_PROXYWAIT) { /* drummer */
- /* Hang around to get the return code from proxy */
- grab = 10;
- } else if (!(socklist[i].flags & SOCK_STRONGCONN)) {
- debug1("net: connect! sock %d", socklist[i].sock);
- s[0] = 0;
- *len = 0;
- #ifdef HAVE_SSL
- /* debug0("CALLING SSL_LINK() FROM SOCKREAD");
- if (!ssl_link(socklist[i].sock))
- debug0("SSL_LINK FAILED");
- debug0("BACK FROM SSL_LINK()"); */
- #endif /* HAVE_SSL */
- return i;
- }
- } else if (socklist[i].flags & SOCK_PASS) {
- s[0] = 0;
- *len = 0;
- return i;
- }
- errno = 0;
- if ((socklist[i].sock == STDOUT) && !backgrd)
- x = read(STDIN, s, grab);
- else {
- #ifdef HAVE_SSL
- if (socklist[i].ssl) {
- x = SSL_read(socklist[i].ssl, s, grab);
- if (x < 0) {
- int err = SSL_get_error(socklist[i].ssl, x);
- x = -1;
- switch (err) {
- case SSL_ERROR_WANT_READ:
- errno = EAGAIN;
- break;
- case SSL_ERROR_WANT_WRITE:
- errno = EAGAIN;
- break;
- case SSL_ERROR_WANT_X509_LOOKUP:
- errno = EAGAIN;
- break;
- }
- }
- } else
- #endif /* HAVE_SSL */
- x = read(socklist[i].sock, s, grab);
- }
- if (x <= 0) { /* eof */
- if (errno != EAGAIN) { /* EAGAIN happens when the operation would block
- on a non-blocking socket, if the socket is going
- to die, it will die later, otherwise it will connect */
- *len = socklist[i].sock;
- socklist[i].flags &= ~SOCK_CONNECT;
- debug1("net: eof!(read) socket %d", socklist[i].sock);
- return -1;
- } else {
- debug3("sockread EAGAIN: %d %d (%s)", socklist[i].sock, errno, strerror(errno));
- continue; /* EAGAIN */
- }
- }
- s[x] = 0;
- *len = x;
- if (socklist[i].flags & SOCK_PROXYWAIT) {
- debug2("net: socket: %d proxy errno: %d", socklist[i].sock, s[1]);
- socklist[i].flags &= ~(SOCK_CONNECT | SOCK_PROXYWAIT);
- switch (s[1]) {
- case 90: /* Success */
- s[0] = 0;
- *len = 0;
- return i;
- case 91: /* Failed */
- errno = ECONNREFUSED;
- break;
- case 92: /* No identd */
- case 93: /* Identd said wrong username */
- /* A better error message would be "socks misconfigured"
- * or "identd not working" but this is simplest.
- */
- errno = ENETUNREACH;
- break;
- }
- *len = socklist[i].sock;
- return -1;
- }
- return i;
- }
- }
- } else if (x == -1)
- return -2; /* socket error */
- else {
- s[0] = 0;
- *len = 0;
- }
- return -3;
- }
- int prand(int *seed, int range)
- {
- long long i1;
- i1 = *seed;
- i1 = (i1 * 0x08088405 + 1) & 0xFFFFFFFF;
- *seed = i1;
- i1 = (i1 * range) >> 32;
- return i1;
- }
- char *botlink_decrypt(int snum, char *src)
- {
- char *line = NULL;
- int i;
- line = decrypt_string(socklist[snum].ikey, src);
- strcpy(src, line);
- free(line);
- if (socklist[snum].iseed) {
- for (i = 0; i <= 3; i++)
- *(dword *) & socklist[snum].ikey[i * 4] = prand(&socklist[snum].iseed, 0xFFFFFFFF);
- if (!socklist[snum].iseed)
- socklist[snum].iseed++;
- }
- return src;
- }
- char *botlink_encrypt(int snum, char *src, size_t *len)
- {
- char *srcbuf = NULL, *buf = NULL, *line = NULL, *eol = NULL, *eline = NULL;
- int bufpos = 0, i = 0;
- srcbuf = calloc(1, *len + 9 + 1);
- strcpy(srcbuf, src);
- line = srcbuf;
- if (!line) {
- free(srcbuf);
- return NULL;
- }
- eol = strchr(line, '\n');
- while (eol) {
- *eol++ = 0;
- eline = encrypt_string(socklist[snum].okey, line);
- if (socklist[snum].oseed) {
- for (i = 0; i <= 3; i++)
- *(dword *) & socklist[snum].okey[i * 4] = prand(&socklist[snum].oseed, 0xFFFFFFFF);
- if (!socklist[snum].oseed)
- socklist[snum].oseed++;
- }
- buf = realloc(buf, bufpos + strlen(eline) + 1 + 9);
- strcpy((char *) &buf[bufpos], eline);
- free(eline);
- strcat(buf, "\n");
- bufpos = strlen(buf);
- line = eol;
- eol = strchr(line, '\n');
- }
- if (line[0]) {
- eline = encrypt_string(socklist[snum].okey, line);
- if (socklist[snum].oseed) {
- for (i = 0; i <= 3; i++)
- *(dword *) & socklist[snum].okey[i * 4] = prand(&socklist[snum].oseed, 0xFFFFFFFF);
- if (!socklist[snum].oseed)
- socklist[snum].oseed++;
- }
- buf = realloc(buf, bufpos + strlen(eline) + 1 + 9);
- strcpy((char *) &buf[bufpos], eline);
- free(eline);
- strcat(buf, "\n");
- }
- free(srcbuf);
- *len = strlen(buf);
- return buf;
- }
- /* sockgets: buffer and read from sockets
- *
- * Attempts to read from all registered sockets for up to one second. if
- * after one second, no complete data has been received from any of the
- * sockets, 's' will be empty, 'len' will be 0, and sockgets will return -3.
- * if there is returnable data received from a socket, the data will be
- * in 's' (null-terminated if non-binary), the length will be returned
- * in len, and the socket number will be returned.
- * normal sockets have their input buffered, and each call to sockgets
- * will return one line terminated with a '\n'. binary sockets are not
- * buffered and return whatever coems in as soon as it arrives.
- * listening sockets will return an empty string when a connection comes in.
- * connecting sockets will return an empty string on a successful connect,
- * or EOF on a failed connect.
- * if an EOF is detected from any of the sockets, that socket number will be
- * put in len, and -1 will be returned.
- * the maximum length of the string returned is 512 (including null)
- *
- * Returns -4 if we handled something that shouldn't be handled by the
- * dcc functions. Simply ignore it.
- */
- int sockgets(char *s, int *len)
- {
- char xx[SGRAB + 4] = "", *p = NULL, *px = NULL;
- int ret, i, data = 0;
- for (i = 0; i < MAXSOCKS; i++) {
- /* Check for stored-up data waiting to be processed */
- if (!(socklist[i].flags & SOCK_UNUSED) && !(socklist[i].flags & SOCK_BUFFER) && (socklist[i].inbuf != NULL)) {
- if (!(socklist[i].flags & SOCK_BINARY)) {
- /* look for \r too cos windows can't follow RFCs */
- p = strchr(socklist[i].inbuf, '\n');
- if (p == NULL)
- p = strchr(socklist[i].inbuf, '\r');
- if (p != NULL) {
- *p = 0;
- if (strlen(socklist[i].inbuf) > SGRAB)
- socklist[i].inbuf[SGRAB] = 0;
- strcpy(s, socklist[i].inbuf);
- px = calloc(1, strlen(p + 1) + 1);
- strcpy(px, p + 1);
- free(socklist[i].inbuf);
- if (px[0])
- socklist[i].inbuf = px;
- else {
- free(px);
- socklist[i].inbuf = NULL;
- }
- /* Strip CR if this was CR/LF combo */
- if (s[strlen(s) - 1] == '\r')
- s[strlen(s) - 1] = 0;
- if (socklist[i].encstatus && s[0])
- botlink_decrypt(i, s);
- *len = strlen(s);
- return socklist[i].sock;
- }
- } else {
- /* i dont think any of this is *ever* called */
- /* Handling buffered binary data (must have been SOCK_BUFFER before). */
- if (socklist[i].inbuflen <= SGRAB) {
- *len = socklist[i].inbuflen;
- egg_memcpy(s, socklist[i].inbuf, socklist[i].inbuflen);
- free(socklist[i].inbuf);
- socklist[i].inbuf = NULL;
- socklist[i].inbuflen = 0;
- } else {
- /* Split up into chunks of SGRAB bytes. */
- *len = SGRAB;
- egg_memcpy(s, socklist[i].inbuf, *len);
- egg_memcpy(socklist[i].inbuf, socklist[i].inbuf + *len, *len);
- socklist[i].inbuflen -= *len;
- socklist[i].inbuf = realloc(socklist[i].inbuf, socklist[i].inbuflen);
- }
- return socklist[i].sock;
- }
- }
- /* Also check any sockets that might have EOF'd during write */
- if (!(socklist[i].flags & SOCK_UNUSED)
- && (socklist[i].flags & SOCK_EOFD)) {
- s[0] = 0;
- *len = socklist[i].sock;
- return -1;
- }
- }
- /* No pent-up data of any worth -- down to business */
- *len = 0;
- ret = sockread(xx, len);
- if (ret < 0) {
- s[0] = 0;
- return ret;
- }
- /* Binary, listening and passed on sockets don't get buffered. */
- if (socklist[ret].flags & SOCK_CONNECT) {
- if (socklist[ret].flags & SOCK_STRONGCONN) {
- socklist[ret].flags &= ~SOCK_STRONGCONN;
- /* Buffer any data that came in, for future read. */
- socklist[ret].inbuflen = *len;
- socklist[ret].inbuf = calloc(1, *len + 1);
- /* It might be binary data. You never know. */
- egg_memcpy(socklist[ret].inbuf, xx, *len);
- socklist[ret].inbuf[*len] = 0;
- }
- socklist[ret].flags &= ~SOCK_CONNECT;
- s[0] = 0;
- return socklist[ret].sock;
- }
- if (socklist[ret].flags & SOCK_BINARY) {
- egg_memcpy(s, xx, *len);
- return socklist[ret].sock;
- }
- if ((socklist[ret].flags & SOCK_LISTEN) || (socklist[ret].flags & SOCK_PASS))
- return socklist[ret].sock;
- if (socklist[ret].flags & SOCK_BUFFER) {
- socklist[ret].inbuf = (char *) realloc(socklist[ret].inbuf,
- socklist[ret].inbuflen + *len + 1);
- egg_memcpy(socklist[ret].inbuf + socklist[ret].inbuflen, xx, *len);
- socklist[ret].inbuflen += *len;
- /* We don't know whether it's binary data. Make sure normal strings
- will be handled properly later on too. */
- socklist[ret].inbuf[socklist[ret].inbuflen] = 0;
- return -4; /* Ignore this one. */
- }
- /* Might be necessary to prepend stored-up data! */
- if (socklist[ret].inbuf != NULL) {
- p = socklist[ret].inbuf;
- socklist[ret].inbuf = calloc(1, strlen(p) + strlen(xx) + 1);
- strcpy(socklist[ret].inbuf, p);
- strcat(socklist[ret].inbuf, xx);
- free(p);
- if (strlen(socklist[ret].inbuf) < (SGRAB + 2)) {
- strcpy(xx, socklist[ret].inbuf);
- free(socklist[ret].inbuf);
- socklist[ret].inbuf = NULL;
- socklist[ret].inbuflen = 0;
- } else {
- p = socklist[ret].inbuf;
- socklist[ret].inbuflen = strlen(p) - SGRAB;
- socklist[ret].inbuf = (char *) calloc(1, socklist[ret].inbuflen + 1);
- strcpy(socklist[ret].inbuf, p + SGRAB);
- *(p + SGRAB) = 0;
- strcpy(xx, p);
- free(p);
- /* (leave the rest to be post-pended later) */
- }
- }
- /* Look for EOL marker; if it's there, i have something to show */
- p = strchr(xx, '\n');
- if (p == NULL)
- p = strchr(xx, '\r');
- if (p != NULL) {
- *p = 0;
- strcpy(s, xx);
- /* strcpy(xx, p + 1); */
- sprintf(xx, "%s", p + 1);
- /* if (s[0] && strlen(s) && (s[strlen(s) - 1] == '\r')) */
- if (s[strlen(s) - 1] == '\r')
- s[strlen(s) - 1] = 0;
- data = 1; /* DCC_CHAT may now need to process a blank line */
- /* NO! */
- /* if (!s[0]) strcpy(s," "); */
- } else {
- s[0] = 0;
- if (strlen(xx) >= SGRAB) {
- /* String is too long, so just insert fake \n */
- strcpy(s, xx);
- xx[0] = 0;
- data = 1;
- }
- }
- if (socklist[ret].encstatus && s[0])
- botlink_decrypt(ret, s);
- *len = strlen(s);
- /* Anything left that needs to be saved? */
- if (!xx[0]) {
- if (data)
- return socklist[ret].sock;
- else
- return -3;
- }
- /* Prepend old data back */
- if (socklist[ret].inbuf != NULL) {
- p = socklist[ret].inbuf;
- socklist[ret].inbuflen = strlen(p) + strlen(xx);
- socklist[ret].inbuf = calloc(1, socklist[ret].inbuflen + 1);
- strcpy(socklist[ret].inbuf, xx);
- strcat(socklist[ret].inbuf, p);
- free(p);
- } else {
- socklist[ret].inbuflen = strlen(xx);
- socklist[ret].inbuf = calloc(1, socklist[ret].inbuflen + 1);
- strcpy(socklist[ret].inbuf, xx);
- }
- if (data) {
- return socklist[ret].sock;
- } else {
- return -3;
- }
- }
- /* Dump something to a socket
- *
- * NOTE: Do NOT put Contexts in here if you want DEBUG to be meaningful!!
- */
- void tputs(register int z, char *s, size_t len)
- {
- register int i, x, idx;
- char *p = NULL;
- static int inhere = 0;
- if (z < 0) /* um... HELLO?! sanity check please! */
- return;
- if (((z == STDOUT) || (z == STDERR)) && (!backgrd || use_stderr)) {
- write(z, s, len);
- return;
- }
- for (i = 0; i < MAXSOCKS; i++) {
- if (!(socklist[i].flags & SOCK_UNUSED) && (socklist[i].sock == z)) {
- for (idx = 0; idx < dcc_total; idx++) {
- if ((dcc[idx].sock == z) && dcc[idx].type && dcc[idx].type->name) {
- if (!strncmp(dcc[idx].type->name, "BOT", 3))
- traffic.out_today.bn += len;
- else if (!strcmp(dcc[idx].type->name, "SERVER"))
- traffic.out_today.irc += len;
- else if (!strncmp(dcc[idx].type->name, "CHAT", 4))
- traffic.out_today.dcc += len;
- else if (!strncmp(dcc[idx].type->name, "FILES", 5))
- traffic.out_today.filesys += len;
- else if (!strcmp(dcc[idx].type->name, "SEND"))
- traffic.out_today.trans += len;
- else if (!strncmp(dcc[idx].type->name, "GET", 3))
- traffic.out_today.trans += len;
- else
- traffic.out_today.unknown += len;
- break;
- }
- }
- if (socklist[i].encstatus && (len > 0))
- s = botlink_encrypt(i, s, &len); /* will modify len */
-
- if (socklist[i].outbuf != NULL) {
- /* Already queueing: just add it */
- p = (char *) realloc(socklist[i].outbuf, socklist[i].outbuflen + len);
- egg_memcpy(p + socklist[i].outbuflen, s, len);
- socklist[i].outbuf = p;
- socklist[i].outbuflen += len;
- if (socklist[i].encstatus && s)
- free(s);
- return;
- }
- /* Try. */
- #ifdef HAVE_SSL
- if (socklist[i].ssl) {
- x = SSL_write(socklist[i].ssl, s, len);
- if (x < 0) {
- int err = SSL_get_error(socklist[i].ssl, x);
- x = -1;
- switch (err) {
- case SSL_ERROR_WANT_READ:
- errno = EAGAIN;
- break;
- case SSL_ERROR_WANT_WRITE:
- errno = EAGAIN;
- break;
- case SSL_ERROR_WANT_X509_LOOKUP:
- errno = EAGAIN;
- break;
- }
- }
- } else
- #endif /* HAVE_SSL */
- #ifdef HAVE_ZLIB_H
- /*
- if (socklist[i].gz) {
- FILE *fp;
- fp = gzdopen(z, "wb0");
- x = gzwrite(fp, s, len);
-
- } else
- */
- #endif /* HAVE_ZLIB_H */
- x = write(z, s, len);
- if (x == -1)
- x = 0;
- if ((size_t) x < len) {
- /* Socket is full, queue it */
- socklist[i].outbuf = calloc(1, len - x);
- egg_memcpy(socklist[i].outbuf, &s[x], len - x);
- socklist[i].outbuflen = len - x;
- }
- if (socklist[i].encstatus && s)
- free(s);
- return;
- }
- }
- /* Make sure we don't cause a crash by looping here */
- if (!inhere) {
- inhere = 1;
- putlog(LOG_MISC, "*", "!!! writing to nonexistent socket: %d", z);
- s[strlen(s) - 1] = 0;
- putlog(LOG_MISC, "*", "!-> '%s'", s);
- inhere = 0;
- }
- /* if (socklist[i].encstatus > 0)
- free(s);
- */
- }
- int findanyidx(register int z)
- {
- register int j;
- if (z != -1)
- for (j = 0; j < dcc_total; j++)
- if (dcc[j].sock == z)
- return j;
- return -1;
- }
- /* tputs might queue data for sockets, let's dump as much of it as
- * possible.
- */
- void dequeue_sockets()
- {
- int i, x;
- int z = 0, fds = 0;
- fd_set wfds;
- struct timeval tv;
- /* ^-- start poptix test code, this should avoid writes to sockets not ready to be written to. */
- FD_ZERO(&wfds);
- tv.tv_sec = 0;
- tv.tv_usec = 0; /* we only want to see if it's ready for writing, no need to actually wait.. */
- for (i = 0; i < MAXSOCKS; i++) {
- if (!(socklist[i].flags & SOCK_UNUSED) && socklist[i].outbuf != NULL) {
- FD_SET(socklist[i].sock, &wfds);
- if (socklist[i].sock > fds)
- fds = socklist[i].sock;
- z = 1;
- }
- }
- if (!z)
- return; /* nothing to write */
- fds++;
- select(fds, NULL, &wfds, NULL, &tv);
- /* end poptix */
- for (i = 0; i < MAXSOCKS; i++) {
- if (!(socklist[i].flags & SOCK_UNUSED) &&
- (socklist[i].outbuf != NULL) && (FD_ISSET(socklist[i].sock, &wfds))) {
- /* Trick tputs into doing the work */
- errno = 0;
- #ifdef HAVE_SSL
- if (socklist[i].ssl) {
- x = write(socklist[i].sock, socklist[i].outbuf, socklist[i].outbuflen);
- if (x < 0) {
- int err = SSL_get_error(socklist[i].ssl, x);
- x = -1;
- switch (err) {
- case SSL_ERROR_WANT_READ:
- errno = EAGAIN;
- break;
- case SSL_ERROR_WANT_WRITE:
- errno = EAGAIN;
- break;
- case SSL_ERROR_WANT_X509_LOOKUP:
- errno = EAGAIN;
- break;
- }
- }
- } else
- #endif /* HAVE_SSL */
- x = write(socklist[i].sock, socklist[i].outbuf, socklist[i].outbuflen);
- if ((x < 0) && (errno != EAGAIN)
- #ifdef EBADSLT
- && (errno != EBADSLT)
- #endif /* EBADSLT */
- #ifdef ENOTCONN
- && (errno != ENOTCONN)
- #endif /* EBADSLT */
- ) {
- /* This detects an EOF during writing */
- debug3("net: eof!(write) socket %d (%s,%d)", socklist[i].sock,
- strerror(errno), errno);
- socklist[i].flags |= SOCK_EOFD;
- } else if ((size_t) x == socklist[i].outbuflen) {
- /* If the whole buffer was sent, nuke it */
- free(socklist[i].outbuf);
- socklist[i].outbuf = NULL;
- socklist[i].outbuflen = 0;
- } else if (x > 0) {
- char *p = socklist[i].outbuf;
- /* This removes any sent bytes from the beginning of the buffer */
- socklist[i].outbuf = (char *) calloc(1, socklist[i].outbuflen - x);
- egg_memcpy(socklist[i].outbuf, p + x, socklist[i].outbuflen - x);
- socklist[i].outbuflen -= x;
- free(p);
- } else {
- debug3("dequeue_sockets(): errno = %d (%s) on %d", errno,
- strerror(errno), socklist[i].sock);
- }
- /* All queued data was sent. Call handler if one exists and the
- * dcc entry wants it.
- */
- if (!socklist[i].outbuf) {
- int idx = findanyidx(socklist[i].sock);
- if (idx > 0 && dcc[idx].type && dcc[idx].type->outdone)
- dcc[idx].type->outdone(idx);
- }
- }
- }
- }
-
- /*
- * Debugging stuff
- */
- void tell_netdebug(int idx)
- {
- int i;
- char s[80] = "";
- dprintf(idx, "Open sockets:");
- for (i = 0; i < MAXSOCKS; i++) {
- if (!(socklist[i].flags & SOCK_UNUSED)) {
- sprintf(s, " %d", socklist[i].sock);
- if (socklist[i].flags & SOCK_BINARY)
- strcat(s, " (binary)");
- if (socklist[i].flags & SOCK_LISTEN)
- strcat(s, " (listen)");
- if (socklist[i].flags & SOCK_PASS)
- strcat(s, " (passed on)");
- if (socklist[i].flags & SOCK_CONNECT)
- strcat(s, " (connecting)");
- if (socklist[i].flags & SOCK_STRONGCONN)
- strcat(s, " (strong)");
- if (socklist[i].flags & SOCK_NONSOCK)
- strcat(s, " (file)");
- if (socklist[i].inbuf != NULL)
- sprintf(&s[strlen(s)], " (inbuf: %04X)", strlen(socklist[i].inbuf));
- if (socklist[i].outbuf != NULL)
- sprintf(&s[strlen(s)], " (outbuf: %06lX)", (unsigned long) socklist[i].outbuflen);
- strcat(s, ",");
- dprintf(idx, "%s", s);
- }
- }
- dprintf(idx, " done.\n");
- }
- /* Checks wether the referenced socket has data queued.
- *
- * Returns true if the incoming/outgoing (depending on 'type') queues
- * contain data, otherwise false.
- */
- int sock_has_data(int type, int sock)
- {
- int ret = 0, i;
- for (i = 0; i < MAXSOCKS; i++)
- if (!(socklist[i].flags & SOCK_UNUSED) && socklist[i].sock == sock)
- break;
- if (i < MAXSOCKS) {
- switch (type) {
- case SOCK_DATA_OUTGOING:
- ret = (socklist[i].outbuf != NULL);
- break;
- case SOCK_DATA_INCOMING:
- ret = (socklist[i].inbuf != NULL);
- break;
- }
- } else
- debug1("sock_has_data: could not find socket #%d, returning false.", sock);
- return ret;
- }
- /* flush_inbuf():
- * checks if there's data in the incoming buffer of an connection
- * and flushs the buffer if possible
- *
- * returns: -1 if the dcc entry wasn't found
- * -2 if dcc[idx].type->activity doesn't exist and the data couldn't
- * be handled
- * 0 if buffer was empty
- * otherwise length of flushed buffer
- */
- int flush_inbuf(int idx)
- {
- int i, len;
- char *inbuf = NULL;
- Assert((idx >= 0) && (idx < dcc_total));
- for (i = 0; i < MAXSOCKS; i++) {
- if ((dcc[idx].sock == socklist[i].sock)
- && !(socklist[i].flags & SOCK_UNUSED)) {
- len = socklist[i].inbuflen;
- if ((len > 0) && socklist[i].inbuf) {
- if (dcc[idx].type && dcc[idx].type->activity) {
- inbuf = socklist[i].inbuf;
- socklist[i].inbuf = NULL;
- dcc[idx].type->activity(idx, inbuf, len);
- free(inbuf);
- return len;
- } else
- return -2;
- } else
- return 0;
- }
- }
- return -1;
- }
|