开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 代理服务器 > 查看源码
udp.c
资源名称：socks5.zip　[点击查看]
上传用户：sddyfurun
上传日期：2007-01-04
资源大小：525k
文件大小：24k
源码类别：

代理服务器

开发平台：

Unix_Linux

udp.c：源码内容
  1. /* Copyright (c) 1995,1996,1997 NEC Corporation.  All rights reserved.       */
  2. /*                                                                           */
  3. /* The redistribution, use and modification in source or binary forms of     */
  4. /* this software is subject to the conditions set forth in the copyright     */
  5. /* document ("Copyright") included with this distribution.                   */
  7. /*
  8.  * $Id: udp.c,v 1.80.2.1.2.7 1998/11/04 00:03:41 steve Exp $
  9.  */
  11. /* This file has all the main functions for handling udp connections.  In    */
  12. /* particular it only one external function which is the main udp function,  */
  13. /* HandleUdpConnection().                                                    */
  14. #include "socks5p.h"
  15. #include "threads.h"
  16. #include "daemon.h"
  17. #include "protocol.h"
  18. #include "validate.h"
  19. #include "udputil.h"
  20. #include "msgids.h"
  21. #include "info.h"
  22. #include "msg.h"
  23. #include "s2s.h"
  24. #include "udp.h"
  25. #include "log.h"
  27. #ifndef UDPTIMEOUT
  28. #define UDPTIMEOUT 15*60
  29. #endif
  31. #define CODEABLE(x)          ((x) && (x)->auth.encode)
  32. #define CODEBUFF(a, b, c, d) ((a)->auth.encode(b, c, d, (a)->auth.context))
  33.     
  34. #define NEXTADDR(x) (((x)->nextVersion)?&((x)->sckAddr)  :&((x)->dstAddr))
  35. #define NEXTNAME(x) (((x)->nextVersion)? ((x)->sckName): ((x)->dstName))
  37. /* Close all the file descriptor, clean up the caches, log the exit error,   */
  38. /* and quit...                                                               */
  39. static void UdpCleanup(S5LinkInfo *pri, UdpInfo *u) {
  40.     S5IOHandle i;
  41.     void *tmp;
  43.     if (!u) return;
  45.     for (; u->scache ; u->scache = (struct scache *)tmp) {
  46. S5BufCleanContext(&u->scache->cinfo);
  47. RemoveIdentEntry(u->scache->idtentry);
  48. tmp = (void *)u->scache->next;
  49. free(u->scache);
  50.     }
  52.     S5BufCleanContext(&u->iio);
  54.     for (i = 0; i < u->maxfd; i++) {
  55. if (FD_ISSET(i, &u->myfds)) CLOSESOCKET(i);
  56.     }
  58.     if (u->relay != S5InvalidIOHandle) CLOSESOCKET(u->relay);
  60.     for (; u->acache ; u->acache = (struct acache *)tmp) {
  61. tmp = (void *)u->acache->next;
  62. free(u->acache);
  63.     }
  65.     for (; u->rcache ; u->rcache = (struct aroute *)tmp) {
  66. tmp = (void *)u->rcache->next;
  67. free(u->rcache);
  68.     }
  70.     for (; u->hcache; u->hcache = (struct hinfo *)tmp) {
  71. tmp = (void *)u->hcache->next;
  72. free(u->hcache);
  73.     }
  75.     tmp = (void *)u->sdring->next;
  76.     u->sdring->next = NULL;
  77.     u->sdring = (UdpSdRing *)tmp;
  79.     for (; u->sdring; u->sdring = (UdpSdRing *)tmp) {
  80.         tmp = (void *)u->sdring->next;
  81.         free(u->sdring);
  82.     }
  83.     
  84.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_INFO, MSGID_SERVER_UDP_END, "UDP Proxy Termination: (%s:%d) for user %s; %d bytes out %d bytes in",
  85.     pri->srcName, ntohs(lsAddr2Port(&pri->srcAddr)), pri->srcUser, pri->outbc, pri->inbc);
  87.     free(u);
  88.     u = NULL;
  89. }
  91. /* Process udp use_client_port sub-command.                                  */
  92. static void UdpUseClientPort(S5LinkInfo *pri, UdpInfo *u, u_char *err) {
  93.     /* do we need this ????            */
  94.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "UDP Command: Doing USECLIENTPORT command");
  96.     if (!lsAddrAddrComp(&pri->dstAddr, &pri->srcAddr) ||
  97.     pri->dstAddr.sin.sin_addr.s_addr == INADDR_ANY ||
  98.     pri->dstAddr.sin.sin_addr.s_addr == htonl(INADDR_LOOPBACK)) {
  99.      pri->clientPort = lsAddr2Port(&pri->dstAddr);
  100.      *err = SOCKS5_NOERR;
  101.     }
  102. }
  104. /* Process udp bind sub-command.                                             */
  105. static void UdpBind(S5LinkInfo *pri, UdpInfo *u, u_char *err) {
  106.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "UDP Command: Doing BIND command");
  108.     if (ResolveNames(pri) < 0) return;
  110.     if (!lsAddrIsNull(&pri->dstAddr)) {
  111. *err = SOCKS5_BADADDR;
  112. return;
  113.     }
  115.     if (CheckIfCached(u, pri, 1) == 0) {
  116.     if (u->curs != NULL) {
  117.     if (S5IOCheck(u->curs->cinfo.fd) < 0) return; /* should it re-establish???    */
  119.     if (u->curs->reserved != S5UDP_USECTRL) {
  120. lsAddrCopy(&pri->intAddr, &u->curr->raddr, sizeof(S5NetAddr));
  121.      *err = SOCKS5_NOERR;
  122.   return;
  123.     } else {
  124.      if (u->cura->bounded) lsAddrCopy(&pri->intAddr, &u->cura->baddr, sizeof(S5NetAddr));
  125.      else if (S5SExchgUdpCmd(u->curs->cinfo.fd, &u->curs->cinfo, pri, SOCKS5_VERSION, S5UDP_BIND, err) < 0) return;
  126.     }
  127. } else lsAddrCopy(&pri->intAddr, &u->curr->raddr, sizeof(S5NetAddr));
  128.     } else {
  129.      if (CheckIfAllowed(u, pri) < 0) {
  130.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "UDP Command: Bind failed: permission denied");
  131.     *err = SOCKS5_AUTHORIZE;
  132.     return;
  133.         }
  135.         if (pri->nextVersion == SOCKS5_VERSION && FindProxyEntry(u, pri, &pri->sckAddr, 1) < 0) {
  136.             S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Command: Couldn't connect to proxy: %s:%d", ADDRANDPORT(&pri->sckAddr));
  137.     return;
  138.         } else if (pri->nextVersion == SOCKS5_VERSION) {
  139.     if (!(pri->nextReserved & S5UDP_USECTRL)) {
  140. /*
  141.                 S5NetAddr na;
  142.              char msg = '';
  144.           lsAddrCopy(&na, &pri->dstAddr, lsAddrSize(&pri->dstAddr));
  145.              lsAddrSetPort(&na, 9);
  146.              if (lsUdpProtoSend(pcon->fd, pri, &msg, 1, 0, &na.sa, lsAddrSize(&na)) < 0) return;
  148.    lsAddrCopy(&pri->intAddr, &u->curr->raddr, sizeof(S5NetAddr));
  149. *err = SOCKS5_NOERR;
  150. */
  151.   return;
  152.          } else if (S5SExchgUdpCmd(u->curs->cinfo.fd, &u->curs->cinfo, pri, SOCKS5_VERSION, S5UDP_BIND, err) < 0) return;
  153.         } else if (pri->nextVersion != SOCKS5_VERSION) {
  154.     if (FindOutSocket(u, pri, &pri->dstAddr, pri->dstName) == S5InvalidIOHandle) {
  155.                 S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Command: Couldn't make out socket to destination: %s:%d", ADDRANDPORT(&pri->dstAddr));
  156.                 return;
  157.          }
  158.         }
  159.     }
  161.     u->cura->bounded = 1;
  162.     lsAddrCopy(&u->cura->baddr, &pri->intAddr, sizeof(S5NetAddr));
  163.     *err = SOCKS5_NOERR;
  164. }
  166. /* Process udp getsockname sub-command.                                      */
  167. static void UdpGetsockname(S5LinkInfo *pri, UdpInfo *u, u_char *err) {
  169.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "UDP Command: Doing GETSOCKNAME command");
  171.     if (ResolveNames(pri) < 0) return;
  173.     if (!lsAddrIsNull(&pri->dstAddr)) {
  174. *err = SOCKS5_BADADDR;
  175. return;
  176.     }
  178.     if (CheckIfCached(u, pri, 1) < 0) return;
  180.     if (u->curs != NULL) {
  181. if (u->curs->reserved != S5UDP_USECTRL) return;
  182. if (S5IOCheck(u->curs->cinfo.fd) < 0) return;
  184.      if (u->cura->bounded) lsAddrCopy(&pri->intAddr, &u->cura->baddr, sizeof(S5NetAddr));
  185.      else if (S5SExchgUdpCmd(u->curs->cinfo.fd, &u->curs->cinfo, pri, SOCKS5_VERSION, S5UDP_GETSOCKNAME, err) < 0) return;
  186.     } else lsAddrCopy(&pri->intAddr, &u->curr->raddr, sizeof(S5NetAddr));
  188.     u->cura->bounded = 1;
  189.     lsAddrCopy(&u->cura->baddr, &pri->intAddr, sizeof(S5NetAddr));
  190.     *err = SOCKS5_NOERR;
  191. }
  193. /* Process udp sub-commands received on the tcp control channel.             */
  194. /* Current implementation supports three sub-commands. People should be      */
  195. /* able to add new sub-commands                                              */
  196. static int HandleCommand(S5LinkInfo *pri, UdpInfo *u) {
  197.     u_char cmd = 0, res = 0, s5err = SOCKS5_BADCMND;
  199.     memset((char *)&pri->intAddr, 0, sizeof(S5NetAddr));
  200.     pri->intAddr.sa.sa_family = AF_INET;
  202.     if (lsReadRequest(u->iio.fd, &u->iio, &pri->dstAddr, &pri->peerVersion, &cmd, &res) < 0) {
  203.      S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "UDP Command: Read request command failed");
  204.         return -1;
  205.     }
  207.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "UDP Command: Read command (%d) request (%s:%d)", (int)cmd, ADDRANDPORT(&pri->dstAddr));
  209.     switch (cmd) {
  210. case S5UDP_USECLIENTPORT:
  211.     UdpUseClientPort(pri, u, &s5err);
  212.     break;
  213. case S5UDP_BIND:
  214.     UdpBind(pri, u, &s5err);
  215.     break;
  216. case S5UDP_GETSOCKNAME:
  217.     UdpGetsockname(pri, u, &s5err);
  218.     break;
  219. default:
  220.          S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "UDP Command: Unknown command");
  221.     break;
  222.     }
  224.     return lsSendResponse(u->iio.fd, &u->iio, &pri->intAddr, pri->peerVersion, s5err, 0, NULL);
  225. }
  227. /* Proxy a request for the client.  This basically involves checking to see  */
  228. /* that the information is correct, then checking to see if it is allowed to */
  229. /* send to the destination (in FindOutDestination) as well as finding the    */
  230. /* intermediate destination...If the destination turns out to be directly    */
  231. /* reachable, then we may reassemble a fragment and send that instead.       */
  232. /*                                                                           */
  233. /* Arguments: sin -- the address of the person who sent the message to us    */
  234. static int RecvFromClient(S5Packet *packet, S5LinkInfo *pri, UdpInfo *u, S5NetAddr *sender) {
  235.     S5Packet buf[2];
  237.     /* Make sure the client really sent this message.  If not, assume it is  */
  238.     /* a forgery.                                                            */
  239.     if (lsAddrComp(&pri->srcAddr, sender)) {
  240.         S5LogUpdate(S5LogDefaultHandle, S5_LOG_WARNING, 0, "UDP Client Receive: Received message from non-client (%s:%d) on client socket", ADDRANDPORT(sender));
  241.         return -1;
  242.     }
  244.     /* Decode the message and copy the decode message into u->obuf.          */
  245.     if (CODEABLE(&u->iio)) {
  246. buf[0].data = u->obuf; buf[0].len = u->obuflen; buf[0].off = u->obuflen;
  247. buf[1].data = NULL;    buf[1].len = 0; buf[1].off = 0;
  249. if (CODEBUFF(&u->iio, &buf[0], &buf[1], S5_DECODE) < 0 || buf[1].len > UDP_MAX_PAYLOAD) {
  250.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_WARNING, 0, "UDP Client Receive: Error decoding message");
  251.     return -1;
  252. } else {
  253.     memcpy(u->obuf, buf[1].data, buf[1].len);
  254.     u->obuflen = buf[1].len;
  255.     free(buf[1].data);
  256. }
  257.     }
  259.     /* A real message would have to have a length greater than the header    */
  260.     /* size.  Drop the message if it does not.                               */
  261.     if (u->obuflen <= MINHDRSIZE) {
  262. S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Client Receive: Received invalid message from client: too short");
  263. return -1;
  264.     }
  266.     /* Extract the destination address and it's name.                        */
  267.     if (lsGetProtoAddr(SOCKS5_VERSION, u->obuf, &pri->dstAddr) < 0) {
  268. S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Client Receive: Received invalid message from client: invalid address");
  269. return -1;
  270.     } 
  272.     if (ResolveNames(pri) < 0) {
  273. return -1;
  274.     }
  276.     /* Make sure we've either talked to this destination before, or we are   */
  277.     /* allowed to talk to it...                                              */
  278.     if (CheckIfCached(u, pri, 1) < 0 && CheckIfAllowed(u, pri) < 0) {
  279. S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Client Receive: send failed: permission denied");
  280. return -1;
  281.     }
  283.     packet->data = u->obuf    + HDRSIZE(u->obuf);
  284.     packet->off  = u->obuflen - HDRSIZE(u->obuf);
  285.     packet->len  = UDP_MAX_PAYLOAD;
  287.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "UDP Client Receive: Received valid message of length %d", u->obuflen-HDRSIZE(u->obuf));
  288.     return u->obuflen - HDRSIZE(u->obuf);
  289. }
  291. /* Proxy a request back from a server. This basically involves checking to   */
  292. /* see if this is from a server we've talked to (FindCachedSocks) in which   */
  293. /* case we forward it on... If not, it was from the application server.      */
  294. /* To handle this case, we first check that this server is allowed (cache?)  */
  295. /* then, if it is, package it up (possibly fragmenting it) and ship it back  */
  296. /* to the client.                                                            */
  297. /*                                                                           */
  298. /* Arguments: sin -- the address of the person who sent the message to us    */
  299. static int RecvFromServer(S5Packet *packet, S5LinkInfo *pri, UdpInfo *u, S5NetAddr *sender) {
  300.     S5Packet buf[2];
  301.     
  302.     /* If we don't find a proxy entry, zero out the appropriate structures   */
  303.     /* so that when we look at them later, we don't do something wrong.  If  */
  304.     /* we did find one, go ahead and look up its name for logging...         */
  305.     pri->dstAddr = *sender;
  306.     if (FindProxyEntry(u, pri, &pri->dstAddr, 0) < 0) {
  307. S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Server Receive: Received message from dead server");
  308. return -1;
  309.     }
  311.     /* If we have a decode function for the server, go ahead and decode the  */
  312.     /* message and move it into the appropriate place                        */
  313.     if (CODEABLE(u->oiop)) {
  314. buf[0].data = u->obuf; buf[0].len = u->obuflen; buf[0].off = u->obuflen;
  315. buf[1].data = NULL;    buf[1].len = 0; buf[1].off = 0;
  317. if (CODEBUFF(u->oiop, &buf[0], &buf[1], S5_DECODE) < 0 || buf[1].len > UDP_MAX_PAYLOAD) {
  318.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Server Receive: Error decoding message from server");
  319.     return -1;
  320. } else {
  321.     memcpy(u->obuf, buf[1].data, buf[1].len);
  322.     u->obuflen = buf[1].len;
  323.     free(buf[1].data);
  324. }
  325.     }
  327.     /* If FindProxyEntry told us this was from another socks server, strip   */
  328.     /* off the header that it put on the message...                          */
  329.     if (pri->nextVersion == SOCKS5_VERSION) {
  330. if (lsGetProtoAddr(SOCKS5_VERSION, u->obuf, &pri->dstAddr) < 0) {
  331.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Server Receive: Received invalid message from server: invalid address");
  332.     return -1;
  333. } 
  335. u->obuflen -= HDRSIZE(u->obuf);
  336. u->obuf    += HDRSIZE(u->obuf);
  337.     }
  339.     GetName(pri->dstName, &pri->dstAddr);
  341.     /* Make sure we've talked to the sender before.  If not, we will ignore  */
  342.     /* this reply, assuming that this is a forgery.                          */
  343.     if (CheckIfCached(u, pri, 1) < 0 && CheckIfAllowed(u, pri) < 0) {
  344. S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Server Receive: recv failed: permission denied");
  345.         return -1;
  346.     }
  348.     packet->data = u->obuf;
  349.     packet->len  = UDP_MAX_PAYLOAD;
  350.     packet->off  = u->obuflen;
  352.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(12), 0, "UDP Server Receive: Received valid message of length %d", u->obuflen);
  353.     return u->obuflen;
  354. }
  356. int UdpRecvPkt(S5Packet *packet, S5LinkInfo *pri, UdpInfo *u, int *dir) {
  357.     S5IOHandle maxfd;
  358.     fd_set fds;
  359.     UdpSdRing *sr;
  360.     int rval;
  361.     
  362.     for (;;) {
  363. struct timeval inact = { UDPTIMEOUT, 0 };
  365. if (*dir & S5_DIRECTION_IN) {
  366.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(15), 0, "UDP Receive: Selecting on outer sockets...");
  367.     maxfd = u->maxfd;
  368.     fds = u->myfds;
  369. } else {
  370.     maxfd = 0;
  371.     FD_ZERO(&fds);
  372. }
  374. if (*dir & S5_DIRECTION_OUT) {
  375.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(15), 0, "UDP Receive: Selecting on inner socket...");
  376.     FD_SET(u->relay, &fds);
  377.     maxfd = MAX(maxfd, u->relay);
  378. }
  380. FD_SET(u->iio.fd, &fds);    
  381. maxfd = MAX(maxfd, u->iio.fd);
  383. S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(12), 0, "UDP Receive: Selecting...");
  385. switch (select(maxfd+1, &fds, NULL, NULL, &inact)) {
  386.     case 0:
  387. S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Receive: Inactivity timeout expired");
  388. rval = -1;
  389. goto end;
  390.     case -1:
  391. if (ISSOCKETERROR(EINTR)) continue;
  392. S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Receive: select failed: %m");
  393. rval = -1;
  394. goto end;
  395. }
  397. if (FD_ISSET(u->iio.fd, &fds)) {
  398.             if (S5IOCheck(u->iio.fd) < 0) rval = 0;
  399.     else if ((pri->peerReserved & S5UDP_USECTRL) && !HandleCommand(pri, u)) continue;
  400.     else {
  401.         S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Receive: client closed connection");
  402.         rval = -1;
  403.     }
  405.     goto end;
  406. }
  407.     
  408.         if (u->sdring) u->sdring = u->sdring->next;
  409.         for (sr = u->sdring;sr;sr = (u->sdring==sr->next)?NULL:sr->next) {
  410.             int fromlen = sizeof(S5NetAddr);
  411.             S5NetAddr sender;
  412.     
  413.     if (!FD_ISSET(sr->sd, &fds)) continue;
  415.     u->obuf    = u->mbuf         + MAXHDRSIZE;
  416.     u->obuflen = sizeof(u->mbuf) - MAXHDRSIZE;
  417.     
  418.     while ((u->obuflen = recvfrom(sr->sd, u->obuf, u->obuflen, 0, (ss *)&sender, &fromlen)) < 0 && ISSOCKETERROR(EINTR));
  420.     if (u->obuflen < 0) {
  421. S5LogUpdate(S5LogDefaultHandle, S5_LOG_WARNING, 0, "UDP Receive: real recvfrom failed: %m");
  422. rval = -1;
  423. goto end;
  424.     }
  425.     
  426.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "UDP Receive: received a message from %s:%d", ADDRANDPORT(&sender));
  428.     if (sr->sd == u->relay) {
  429. if ((rval = RecvFromClient(packet, pri, u, &sender)) <= 0) continue; 
  430. *dir = S5_DIRECTION_OUT;
  431.     } else {
  432.      if ((rval = RecvFromServer(packet, pri, u, &sender)) <= 0) continue;
  433.      *dir = S5_DIRECTION_IN;
  434.     }
  436.     goto end;
  437. }
  438.     }
  440.   end:
  441.     if (rval <= 0) {
  442. packet->data = NULL;
  443. packet->off  = 0;
  444. packet->len  = 0;
  445.     }
  446.     
  447.     return rval;
  448. }
  449.     
  450. int UdpSendPkt(S5Packet *packet, S5LinkInfo *pri, UdpInfo *u, int *dir) {
  451.     S5Packet buf[2];
  452.     S5NetAddr *dst;
  453.     S5IOInfo *info;
  454.     S5IOHandle fd;
  455.     int alen;
  457.     if (packet->data < u->mbuf + MAXHDRSIZE ||
  458. packet->data > u->mbuf + MAXHDRSIZE + UDP_MAX_PAYLOAD ) {
  459. char *tmp;
  461. if ((tmp = (char *)malloc(packet->off)) == NULL) {
  462.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_WARNING, 0, "UDP send: Malloc failed.");
  463.     u->obuflen = -1;
  464.     goto end;
  465. }
  467. memcpy(tmp, packet->data, packet->off);
  468. u->obuf    = u->mbuf      + MAXHDRSIZE;
  469. u->obuflen = packet->off;
  470. memcpy(u->obuf, tmp, u->obuflen);
  471. free(tmp);
  472.     } else {
  473. u->obuf    = packet->data;
  474. u->obuflen = packet->off;
  475.     }
  476.     
  477.     if (*dir == S5_DIRECTION_IN) {
  478. dst  = &pri->srcAddr;
  479. fd   = u->relay;
  480. info = &u->iio;
  481.     } else {
  482. /* If we'll be using a proxy and we haven't talked to it yet,        */
  483. /* initialize the connection and add it it list of know proxies...   */
  484. if (pri->nextVersion == SOCKS5_VERSION && FindProxyEntry(u, pri, &pri->sckAddr, 1) < 0) {
  485.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP send: Couldn't connect to %s:%d", ADDRANDPORT(&pri->sckAddr));
  486.     u->obuflen = -1;
  487.     goto end;
  488. }
  490. /* Find out how we should go out, update the byte count, and dump    */
  491. /* the packet if need be.                                            */
  492. fd   = FindOutSocket(u, pri, NEXTADDR(pri), NEXTNAME(pri));
  493. dst  = NEXTADDR(pri);
  494. info = u->oiop;
  495.     }
  497.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "UDP Send: Sending to %s:%d", ADDRANDPORT(dst));
  499.     if (*dir == S5_DIRECTION_IN || pri->nextVersion == SOCKS5_VERSION) {
  500. if ((alen = lsGetProtoAddrLenFromAddr(SOCKS5_VERSION, &pri->dstAddr)) < 0) {
  501.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Send: Error filling in address in packet header");
  502.     u->obuflen = -1;
  503.     goto end;
  504. }
  506. u->obuf -= alen;
  507. memset(u->obuf, 0, 3);
  508. lsSetProtoAddr(SOCKS5_VERSION, u->obuf, &pri->dstAddr);
  509. u->obuflen += HDRSIZE(u->obuf);
  510.     }
  511.     
  512.     /* Encode the packet if necessary... This includes changing obuf to      */
  513.     /* point to the buffer that the encoding creates.  We'll free that       */
  514.     /* buffer later if need be.                                              */
  515.     if (CODEABLE(info)) {
  517. buf[0].data = u->obuf;
  518. buf[0].len  = u->obuflen;
  519. buf[0].off  = 0;
  521. buf[1].data = NULL;
  522. buf[1].len = 0;
  523. buf[1].off = 0;
  525. if (CODEBUFF(info, &buf[0], &buf[1], S5_ENCODE) < 0) {
  526.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Send: Error encoding message for client");
  527.     u->obuflen = -1;
  528.     goto end;
  529. } else {
  530.     u->obuf    = buf[1].data;
  531.     u->obuflen = buf[1].len;
  532. }
  533.     }
  535.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "UDP Send: Sending a message of length %d", u->obuflen);
  537.     /* Send the message and log the fact that we have.                       */
  538.     if (sendto(fd, u->obuf, u->obuflen, 0, &dst->sa, sizeof(ssi)) < 0) {
  539. S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(0), 0, "UDP Send: Sendto failed: %m");
  540.     } else if (*dir == S5_DIRECTION_OUT) {
  541. S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "UDP Send: Client Sent Message to %s:%d", ADDRANDPORT(dst));
  542.     } else {
  543. S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "UDP Send: Client Recveived Message from %s:%d", ADDRANDPORT(&pri->dstAddr));
  544.     }
  546.     /* Free any space we had to allocate because of the encoding             */
  547.     if (CODEABLE(info)) {
  548. if (buf[1].data) free(buf[1].data);
  549.     }
  551.   end:
  552.     packet->data = NULL;
  553.     packet->off  = 0;
  554.     packet->len  = 0;
  555.     return u->obuflen;
  556. }
  558. int UdpSetup(S5IOInfo *ioinfo, S5LinkInfo *pri, S5CommandInfo *cmdinfo) {
  559.     int len = sizeof(S5NetAddr), rval = EXIT_ERR;
  560.     S5NetAddr route;
  561.     u_char s5err = SOCKS5_FAIL;
  562.     UdpInfo *u = NULL;
  563.     
  564.     /* The destination is really the port we'll be receiving request from    */
  565.     /* it is allowed to be 0, but we take that to mean we'll be recv'ing it  */
  566.     /* from the same place the tcp connection came from, different port      */
  567.     if (lsAddrAddrComp(&pri->dstAddr, &pri->srcAddr) != 0) {
  568. if (pri->dstAddr.sin.sin_addr.s_addr != INADDR_ANY && pri->dstAddr.sin.sin_addr.s_addr != htonl(INADDR_LOOPBACK)) {
  569.     s5err = SOCKS5_BADADDR;
  570.     goto cleanup;
  571. }
  572.     }
  574.     /* Allocate space for the udp specific options we'll be using...         */
  575.     if (!(cmdinfo->option = (void *)(u = (UdpInfo *)calloc(1, sizeof(UdpInfo))))) {
  576. S5LogUpdate(S5LogDefaultHandle, S5_LOG_WARNING, 0, "UDP Setup: Malloc() failed");
  577. goto cleanup;
  578.     }
  580.     u->sdring = NULL;
  582.     u->iio  = *ioinfo;
  583.     memset(&u->myfds, 0, sizeof(fd_set));
  585.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_INFO, MSGID_SERVER_UDP_START, "UDP Proxy Request: (%s:%d) for user %s", pri->srcName, ntohs(lsAddr2Port(&pri->srcAddr)), pri->srcUser);
  587.     /* if the destination port is 0, we will assume the same port as tcp's    */
  588.     if (lsAddr2Port(&pri->dstAddr) != (u_short)0) lsAddrSetPort(&pri->srcAddr, lsAddr2Port(&pri->dstAddr));
  590.     /* We set the destination address same as the source for Authorize        */
  591.     /* function to distinguish the UDP request and UDP messages ...           */
  592.     lsAddrCopy(&pri->dstAddr, &pri->srcAddr, lsAddrSize(&pri->srcAddr));
  593.     if (Authorize(pri, 0) != AUTH_OK) {
  594. S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "UDP Setup: Setup failed: permission denied");
  595. s5err = SOCKS5_AUTHORIZE;
  596. rval  = EXIT_AUTH;
  597. goto cleanup;
  598.     }
  600.     lsAddrCopy(&route, &pri->bndAddr, lsAddrSize(&pri->bndAddr));
  601.     lsAddrSetPort(&route, (u_short)0);
  603.     if ((u->relay = MakeOutSocket(u, &route, (u_short)0)) == S5InvalidIOHandle) {
  604. S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "UDP Setup: Couldn't make main udp socket: %m");
  605. goto cleanup;
  606.     }
  608.     if (getsockname(u->relay, (ss *)&route.sa, &len) < 0) {
  609. S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "UDP Setup: Getsockname failed on main udp socket: %m");
  610. goto cleanup;
  611.     }
  613.     lsAddrSetPort(&pri->bndAddr, lsAddr2Port(&route));
  615.     if (lsSendResponse(u->iio.fd, &u->iio, &pri->bndAddr, pri->peerVersion, SOCKS5_RESULT, (pri->peerReserved & S5UDP_USECTRL)?S5UDP_USECTRL:0, NULL) < 0) {
  616. S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "UDP Setup: couldn't send response");
  617. rval = EXIT_NETERR;
  618. goto cleanup;
  619.     }
  621.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_INFO, MSGID_SERVER_UDP_ESTAB, "UDP Proxy Established: (%s:%d) for user %s",
  622.     pri->srcName, ntohs(lsAddr2Port(&pri->srcAddr)), pri->srcUser);
  624.     cmdinfo->recvpkt  = (int (*)(S5Packet *, S5LinkInfo *, void *, int *))UdpRecvPkt;
  625.     cmdinfo->sendpkt  = (int (*)(S5Packet *, S5LinkInfo *, void *, int *))UdpSendPkt;
  626.     cmdinfo->clean    = (int (*)(S5LinkInfo *, void *))UdpCleanup;
  627.     return EXIT_OK;
  629.   cleanup:
  630.     if (rval != EXIT_NETERR) lsSendResponse(ioinfo->fd, ioinfo, &pri->bndAddr, pri->peerVersion, s5err, 0, NULL);
  632.     if (u != NULL) UdpCleanup(pri, u);
  633.     else {
  634.      S5LogUpdate(S5LogDefaultHandle, S5_LOG_INFO, MSGID_SERVER_UDP_END, "UDP Proxy Termination: (%s:%d) for user %s; %d bytes out %d bytes in",
  635.              pri->srcName, ntohs(lsAddr2Port(&pri->srcAddr)), pri->srcUser, pri->outbc, pri->inbc);
  637.      S5BufCleanContext(ioinfo);
  638.     }
  640.     /* Prevent any further problems from being seg faults.                   */
  641.     cmdinfo->option = NULL;
  642.     if (rval == EXIT_OK) rval = EXIT_ERR;
  643.     return rval;
  644. }