123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605 |
- /*
- Copyright (c) 2003-2006 by Juliusz Chroboczek
- Permission is hereby granted, free of charge, to any person obtaining a copy
- of this software and associated documentation files (the "Software"), to deal
- in the Software without restriction, including without limitation the rights
- to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
- copies of the Software, and to permit persons to whom the Software is
- furnished to do so, subject to the following conditions:
- The above copyright notice and this permission notice shall be included in
- all copies or substantial portions of the Software.
- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
- OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
- THE SOFTWARE.
- */
- #include "polipo.h"
- #ifdef NO_SOCKS
- AtomPtr socksParentProxy = NULL;
- void
- preinitSocks()
- {
- return;
- }
- void
- initSocks()
- {
- return;
- }
- int
- do_socks_connect(char *name, int port,
- int (*handler)(int, SocksRequestPtr),
- void *data)
- {
- SocksRequestRec request;
- request.name = internAtomLowerN(name, strlen(name));
- request.port = port;
- request.handler = handler;
- request.buf = NULL;
- request.data = data;
- handler(-ENOSYS, &request);
- releaseAtom(request.name);
- return 1;
- }
- #else
- int authed = -1;
- AtomPtr socksParentProxy = NULL;
- AtomPtr socksProxyHost = NULL;
- int socksProxyPort = -1;
- AtomPtr socksProxyAddress = NULL;
- int socksProxyAddressIndex = -1;
- AtomPtr socksUserName = NULL;
- AtomPtr socksPassWord = NULL;
- AtomPtr socksProxyType = NULL;
- AtomPtr aSocks4a, aSocks5;
- static int socksParentProxySetter(ConfigVariablePtr, void*);
- static int socksProxyTypeSetter(ConfigVariablePtr, void*);
- static int do_socks_connect_common(SocksRequestPtr);
- static int socksDnsHandler(int, GethostbynameRequestPtr);
- static int socksConnectHandler(int, FdEventHandlerPtr, ConnectRequestPtr);
- static int socksWriteHandler(int, FdEventHandlerPtr, StreamRequestPtr);
- static int socksReadHandler(int, FdEventHandlerPtr, StreamRequestPtr);
- static int socks5ReadHandler(int, FdEventHandlerPtr, StreamRequestPtr);
- static int socks5ReadHandlerAuth(int, FdEventHandlerPtr, StreamRequestPtr);
- static int socks5WriteHandler(int, FdEventHandlerPtr, StreamRequestPtr);
- static int socks5ReadHandler2(int, FdEventHandlerPtr, StreamRequestPtr);
- void
- preinitSocks()
- {
- AtomPtr socksAuthCredentials = internAtom("");
- aSocks4a = internAtom("socks4a");
- aSocks5 = internAtom("socks5");
- socksProxyType = retainAtom(aSocks5);
- socksUserName = internAtom("");
- socksPassWord = internAtom("");
- CONFIG_VARIABLE_SETTABLE(socksParentProxy, CONFIG_ATOM_LOWER,
- socksParentProxySetter,
- "SOCKS parent proxy (host:port)");
- CONFIG_VARIABLE_SETTABLE(socksAuthCredentials, CONFIG_PASSWORD,
- configAtomSetter,
- "SOCKS4a (or SOCKS5) credentials username:password");
- CONFIG_VARIABLE_SETTABLE(socksProxyType, CONFIG_ATOM_LOWER,
- socksProxyTypeSetter,
- "One of socks4a or socks5");
- // infer username and password from credentials
- int rc = atomSplit(socksAuthCredentials, ':', &socksUserName, &socksPassWord);
- if (rc < 0) {
- do_log(L_ERROR, "Error splitting credentials");
- exit(1);
- } else if (rc == 0) {
- // separator ':' not found
- socksUserName = socksAuthCredentials;
- releaseAtom(socksPassWord);
- socksPassWord = NULL;
- } else {
- // split successfull: free memory
- releaseAtom(socksAuthCredentials);
- }
- }
- static int
- socksParentProxySetter(ConfigVariablePtr var, void *value)
- {
- configAtomSetter(var, value);
- initSocks();
- return 1;
- }
- static int
- socksProxyTypeSetter(ConfigVariablePtr var, void *value)
- {
- if(*var->value.a != aSocks4a && *var->value.a != aSocks5) {
- do_log(L_ERROR, "Unknown socksProxyType %s\n", (*var->value.a)->string);
- return -1;
- }
- return configAtomSetter(var, value);
- }
- void
- initSocks()
- {
- int port = -1;
- AtomPtr host = NULL, port_atom;
- int rc;
- if(socksParentProxy != NULL && socksParentProxy->length == 0) {
- releaseAtom(socksParentProxy);
- socksParentProxy = NULL;
- }
- if(socksParentProxy) {
- rc = atomSplit(socksParentProxy, ':', &host, &port_atom);
- if(rc <= 0) {
- do_log(L_ERROR, "Couldn't parse socksParentProxy");
- exit(1);
- }
- port = atoi(port_atom->string);
- releaseAtom(port_atom);
- }
- if(socksProxyHost)
- releaseAtom(socksProxyHost);
- socksProxyHost = host;
- socksProxyPort = port;
- if(socksProxyAddress)
- releaseAtom(socksProxyAddress);
- socksProxyAddress = NULL;
- socksProxyAddressIndex = -1;
- if(socksProxyType != aSocks4a && socksProxyType != aSocks5) {
- do_log(L_ERROR, "Unknown socksProxyType %s\n", socksProxyType->string);
- exit(1);
- }
- }
- static void
- destroySocksRequest(SocksRequestPtr request)
- {
- releaseAtom(request->name);
- if(request->buf)
- free(request->buf);
- free(request);
- }
- int
- do_socks_connect(char *name, int port,
- int (*handler)(int, SocksRequestPtr),
- void *data)
- {
- SocksRequestPtr request = malloc(sizeof(SocksRequestRec));
- SocksRequestRec request_nomem;
- if(request == NULL)
- goto nomem;
- request->name = internAtomLowerN(name, strlen(name));
- if(request->name == NULL) {
- free(request);
- goto nomem;
- }
- request->port = port;
- request->fd = -1;
- request->handler = handler;
- request->buf = NULL;
- request->data = data;
- if(socksProxyAddress == NULL) {
- do_gethostbyname(socksProxyHost->string, 0,
- socksDnsHandler,
- request);
- return 1;
- }
- return do_socks_connect_common(request);
- nomem:
- request_nomem.name = internAtomLowerN(name, strlen(name));
- request_nomem.port = port;
- request_nomem.handler = handler;
- request_nomem.buf = NULL;
- request_nomem.data = data;
- handler(-ENOMEM, &request_nomem);
- releaseAtom(request_nomem.name);
- return 1;
- }
- static int
- do_socks_connect_common(SocksRequestPtr request)
- {
- assert(socksProxyAddressIndex >= 0);
- do_connect(retainAtom(socksProxyAddress),
- socksProxyAddressIndex,
- socksProxyPort,
- socksConnectHandler, request);
- return 1;
- }
- static int
- socksDnsHandler(int status, GethostbynameRequestPtr grequest)
- {
- SocksRequestPtr request = grequest->data;
- if(status <= 0) {
- request->handler(status, request);
- destroySocksRequest(request);
- return 1;
- }
- if(grequest->addr->string[0] == DNS_CNAME) {
- if(grequest->count > 10) {
- do_log(L_ERROR, "DNS CNAME loop.\n");
- request->handler(-EDNS_CNAME_LOOP, request);
- destroySocksRequest(request);
- return 1;
- }
- do_gethostbyname(grequest->addr->string + 1, grequest->count + 1,
- socksDnsHandler, request);
- return 1;
- }
- socksProxyAddress = retainAtom(grequest->addr);
- socksProxyAddressIndex = 0;
- do_socks_connect_common(request);
- return 1;
- }
- static int
- socksConnectHandler(int status,
- FdEventHandlerPtr event,
- ConnectRequestPtr crequest)
- {
- SocksRequestPtr request = crequest->data;
- int rc;
- if(status < 0) {
- request->handler(status, request);
- destroySocksRequest(request);
- return 1;
- }
- assert(request->fd < 0);
- request->fd = crequest->fd;
- socksProxyAddressIndex = crequest->index;
- rc = setNodelay(request->fd, 1);
- if(rc < 0)
- do_log_error(L_WARN, errno, "Couldn't disable Nagle's algorithm");
- if(socksProxyType == aSocks4a) {
- request->buf = malloc(8 +
- socksUserName->length + 1 +
- request->name->length + 1);
- if(request->buf == NULL) {
- CLOSE(request->fd);
- request->fd = -1;
- request->handler(-ENOMEM, request);
- destroySocksRequest(request);
- return 1;
- }
- request->buf[0] = 4; /* VN */
- request->buf[1] = 1; /* CD = REQUEST */
- request->buf[2] = (request->port >> 8) & 0xFF;
- request->buf[3] = request->port & 0xFF;
- request->buf[4] = request->buf[5] = request->buf[6] = 0;
- request->buf[7] = 3;
- memcpy(request->buf + 8, socksUserName->string, socksUserName->length);
- request->buf[8 + socksUserName->length] = '\0';
- memcpy(request->buf + 8 + socksUserName->length + 1,
- request->name->string, request->name->length);
- request->buf[8 + socksUserName->length + 1 + request->name->length] =
- '\0';
- do_stream(IO_WRITE, request->fd, 0, request->buf,
- 8 + socksUserName->length + 1 + request->name->length + 1,
- socksWriteHandler, request);
- } else if(socksProxyType == aSocks5) {
- request->buf = malloc(8); /* 8 needed for the subsequent read */
- if(request->buf == NULL) {
- CLOSE(request->fd);
- request->fd = -1;
- request->handler(-ENOMEM, request);
- destroySocksRequest(request);
- return 1;
- }
- request->buf[0] = 5; /* ver */
- request->buf[1] = 1; /* nmethods */
- if (socksPassWord == NULL) {
- request->buf[2] = 0; /* no authentication required */
- } else {
- request->buf[2] = 2; /* username/password */
- }
- do_stream(IO_WRITE, request->fd, 0, request->buf, 3,
- socksWriteHandler, request);
- } else {
- request->handler(-EUNKNOWN, request);
- }
- return 1;
- }
- static int
- socksWriteHandler(int status,
- FdEventHandlerPtr event,
- StreamRequestPtr srequest)
- {
- SocksRequestPtr request = srequest->data;
- if(status < 0)
- goto error;
- if(!streamRequestDone(srequest)) {
- if(status) {
- status = -ESOCKS_PROTOCOL;
- goto error;
- }
- return 0;
- }
- int (*readHandler)(int, FdEventHandlerPtr, StreamRequestPtr) = NULL;
- if (socksProxyType == aSocks4a){
- readHandler = socksReadHandler;
- } else if (socksPassWord == NULL) {
- readHandler = socks5ReadHandler;
- } else if (socksPassWord != NULL) {
- readHandler = socks5ReadHandlerAuth;
- }
- do_stream(IO_READ | IO_NOTNOW, request->fd, 0, request->buf, 8, readHandler, request);
- return 1;
- error:
- CLOSE(request->fd);
- request->fd = -1;
- request->handler(status, request);
- destroySocksRequest(request);
- return 1;
- }
- static int
- socksReadHandler(int status,
- FdEventHandlerPtr event,
- StreamRequestPtr srequest)
- {
- SocksRequestPtr request = srequest->data;
- if(status < 0)
- goto error;
- if(srequest->offset < 8) {
- if(status) {
- status = -ESOCKS_PROTOCOL;
- goto error;
- }
- return 0;
- }
- if(request->buf[0] != 0 || request->buf[1] != 90) {
- if(request->buf[1] >= 91 && request->buf[1] <= 93)
- status = -(ESOCKS_PROTOCOL + request->buf[1] - 90);
- else
- status = -ESOCKS_PROTOCOL;
- goto error;
- }
- request->handler(1, request);
- destroySocksRequest(request);
- return 1;
- error:
- CLOSE(request->fd);
- request->fd = -1;
- request->handler(status, request);
- destroySocksRequest(request);
- return 1;
- }
- static int
- socks5ReadHandlerAuth(int status,
- FdEventHandlerPtr event,
- StreamRequestPtr srequest)
- {
- SocksRequestPtr request = srequest->data;
- if(status < 0)
- goto error;
- if(srequest->offset < 2) {
- if(status) {
- status = -ESOCKS_PROTOCOL;
- goto error;
- }
- return 0;
- }
- if(request->buf[0] != 5 || request->buf[1] != 2) {
- status = -ESOCKS_PROTOCOL;
- goto error;
- }
- free(request->buf);
- request->buf = malloc(5 + socksUserName->length + socksPassWord->length);
- if(request->buf == NULL) {
- status = -ENOMEM;
- goto error;
- }
- request->buf[0] = 1; /* ver */
- request->buf[1] = socksUserName->length; /* username length */
- memcpy(request->buf + 2, socksUserName->string, socksUserName->length);
- request->buf[2 + socksUserName->length] = socksPassWord->length; /* password length */
- memcpy(request->buf + 3 + socksUserName->length, socksPassWord->string, socksPassWord->length);
- do_stream(IO_WRITE, request->fd, 0,
- request->buf, 3 + socksUserName->length + socksPassWord->length,
- socks5WriteHandler, request);
- return 1;
- error:
- CLOSE(request->fd);
- request->fd = -1;
- request->handler(status, request);
- destroySocksRequest(request);
- return 1;
- }
- static int
- socks5ReadHandler(int status,
- FdEventHandlerPtr event,
- StreamRequestPtr srequest)
- {
- SocksRequestPtr request = srequest->data;
- if(status < 0)
- goto error;
- if(srequest->offset < 2) {
- if(status) {
- status = -ESOCKS_PROTOCOL;
- goto error;
- }
- return 0;
- }
- if(request->buf[1] != 0 ||
- (socksPassWord != NULL && request->buf[0] != 1) || // user/pass: need ver 1
- (socksPassWord == NULL && request->buf[0] != 5)) // no-auth: need ver 5
- {
- status = -ESOCKS_PROTOCOL;
- goto error;
- }
- authed = 1;
- free(request->buf);
- request->buf = malloc(5 + request->name->length + 2);
- if(request->buf == NULL) {
- status = -ENOMEM;
- goto error;
- }
- request->buf[0] = 5; /* ver */
- request->buf[1] = 1; /* cmd */
- request->buf[2] = 0; /* rsv */
- request->buf[3] = 3; /* atyp */
- request->buf[4] = request->name->length;
- memcpy(request->buf + 5, request->name->string, request->name->length);
- request->buf[5 + request->name->length] = (request->port >> 8) & 0xFF;
- request->buf[5 + request->name->length + 1] = request->port & 0xFF;
- do_stream(IO_WRITE, request->fd, 0,
- request->buf, 5 + request->name->length + 2,
- socks5WriteHandler, request);
- return 1;
- error:
- CLOSE(request->fd);
- request->fd = -1;
- request->handler(status, request);
- destroySocksRequest(request);
- return 1;
- }
- static int
- socks5WriteHandler(int status,
- FdEventHandlerPtr event,
- StreamRequestPtr srequest)
- {
- SocksRequestPtr request = srequest->data;
- if(status < 0)
- goto error;
- if(!streamRequestDone(srequest)) {
- if(status) {
- status = -ESOCKS_PROTOCOL;
- goto error;
- }
- return 0;
- }
- do_stream(IO_READ | IO_NOTNOW, request->fd, 0, request->buf, 10,
- ((socksPassWord != NULL) && (authed == -1) ? socks5ReadHandler : socks5ReadHandler2),
- request);
- return 1;
- error:
- request->handler(status, request);
- destroySocksRequest(request);
- return 1;
- }
- static int
- socks5ReadHandler2(int status,
- FdEventHandlerPtr event,
- StreamRequestPtr srequest)
- {
- SocksRequestPtr request = srequest->data;
- authed = -1;
- if(status < 0)
- goto error;
- if(srequest->offset < 4) {
- if(status) {
- status = -ESOCKS_PROTOCOL;
- goto error;
- }
- return 0;
- }
- if(request->buf[0] != 5) {
- status = -ESOCKS_PROTOCOL;
- goto error;
- }
- if(request->buf[1] != 0) {
- status = -(ESOCKS5_BASE + request->buf[1]);
- goto error;
- }
- if(request->buf[3] != 1) {
- status = -ESOCKS_PROTOCOL;
- goto error;
- }
- if(srequest->offset < 10)
- return 0;
- request->handler(1, request);
- destroySocksRequest(request);
- return 1;
- error:
- CLOSE(request->fd);
- request->fd = -1;
- request->handler(status, request);
- destroySocksRequest(request);
- return 1;
- }
- #endif
|