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

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

代理服务器

开发平台：

Unix_Linux

hostname.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: hostname.c,v 1.31.4.6 1998/11/18 17:39:12 wlu Exp $
  9.  */
  11. #include "socks5p.h"
  12. #include "protocol.h"
  13. #include "addr.h"
  14. #include "wrap.h"
  15. #include "log.h"
  17. #define S5_HOSTLIST_SIZE    256
  18. #define S5_HOSTALIASES_SIZE 16
  19. #define S5_FAKEHOSTFILE     "/.s5fakehost"
  20.  
  21. int lsInWrapHostname = 0;
  22. int fd = -1;
  24. /* local array of host address list and aliases list                         */
  25. static struct in_addr host_addr[S5_HOSTLIST_SIZE];
  26. static char host_aliases[S5_HOSTALIASES_SIZE][S5_HOSTNAME_SIZE];
  28. /* The address of unresolved host will be of the format "0.0.0.i", where i   */
  29. /* ranges from 1 to 254, pretty simple...                                    */
  30. #define FAKEPREFIX "0.0.0."
  32. static int SetReadLock(int lock) {
  33.     struct flock locks;
  35.     locks.l_type = (lock?F_RDLCK:F_UNLCK);
  36.     locks.l_start = 0;
  37.     locks.l_whence = SEEK_SET;
  38.     locks.l_len = 0;
  40.     return fcntl(fd, F_SETLKW, &locks);
  41. }
  43. static int SetWriteLock(int lock) {
  44.     struct flock locks;
  46.     locks.l_type = (lock?F_WRLCK:F_UNLCK);
  47.     locks.l_start = 0;
  48.     locks.l_whence = SEEK_SET;
  49.     locks.l_len = 0;
  51.     return fcntl(fd, F_SETLKW, &locks);
  52. }
  54. #define S_ISRW(mode) (((mode)&(S_IREAD | S_IWRITE)) == (S_IREAD | S_IWRITE))
  56. static void FakeHostInit(void) {
  57.     int i, j, flags;
  58.     struct stat sbuf;
  59.     char filename[1024], hostname[S5_HOSTNAME_SIZE], *tmp;
  61.     if (fd > 0) return;
  63.     if ((tmp = getenv("HOME"))) {
  64.         i = MIN(strlen(tmp)+1, sizeof(filename));
  65.         strncpy(filename, tmp, i);
  66.         filename[i-1] = '';
  67.     } else strcpy(filename, "/tmp");
  69.     j = strlen(filename);
  70.     i = MIN(strlen(S5_FAKEHOSTFILE)+1, sizeof(filename)-j);
  71.     strncat(filename, S5_FAKEHOSTFILE, i);
  72.     filename[i+j-1] = '';
  74.     if (!lstat(filename, &sbuf)) {
  75. if (sbuf.st_size != 255*sizeof(hostname)+sizeof(int) || !S_ISRW(sbuf.st_mode)) {
  76.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_WARNING, 0, "FakeHostInit: fakehost file (%s) has been changed", filename);
  77.     flags = O_RDWR | O_CREAT | O_TRUNC | O_EXCL;
  78. } else if (S_ISLNK(sbuf.st_mode) && geteuid() != sbuf.st_uid) {
  79.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_WARNING, 0, "FakeHostInit: fakehost file (%s) is a link", filename);
  80.     flags = O_RDWR | O_CREAT | O_TRUNC | O_EXCL;
  81. } else flags = O_RDWR;
  82.     } else flags = O_RDWR | O_CREAT | O_TRUNC | O_EXCL;
  84.     while ((fd = open(filename, flags, 0600)) < 0) {
  85. if (errno != EINTR && errno != EAGAIN) {
  86.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "FakeHostInit: open fakehost file (%s) failed %m", filename);
  87.     return;
  88. }
  89.     }
  91.     if (!(flags & O_CREAT)) return;
  93.     SetWriteLock(1);
  94.     i = 0;
  95.     j = sizeof(int);
  96.     lseek(fd, 0, SEEK_SET);
  97.     if (REAL(write)(fd, (char *)&i, j) != j) {
  98.         S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "FakeHostInit: write fakehost file failed %m");
  99.         REAL(close)(fd);
  100.         fd = -1;
  101.         return;
  102.     }
  104.     memset(hostname, 0, sizeof(hostname));
  105.     for (i = 0; i < 255; i++) {
  106.         if (REAL(write)(fd, hostname, sizeof(hostname)) != sizeof(hostname)) {
  107.             S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "FakeHostInit: write fakehost file failed %m");
  108.             REAL(close)(fd);
  109.             fd = -1;
  110.             return;
  111.         }
  112.     }
  113.     SetWriteLock(0);
  114. }
  116. static int GetFakeHost(const char *name) {
  117.     int i, j;
  118.     char hostname[S5_HOSTNAME_SIZE];
  120.     FakeHostInit();
  121.     if (fd <= 0) return -1;
  123.     SetWriteLock(1);
  124.     lseek(fd, 0, SEEK_SET);
  125.     i = 0;
  126.     if (REAL(read)(fd, (char *)&i, sizeof(int)) != sizeof(int)) {
  127.         S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "GetFakeHost: read table failed %m");
  128.         SetWriteLock(0);
  129.         return -1;
  130.     }
  132.     for (j = 1; j < 256; j++) {
  133.         if (REAL(read)(fd, hostname, sizeof(hostname)) != sizeof(hostname)) {
  134.             S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "GetFakeHost: read table failed %m");
  135.             SetWriteLock(0);
  136.             return -1;
  137.         }
  139.         if (hostname[0] == '' || !strcasecmp(name, hostname)) break;
  140.     }
  142.     if (j == 256 || hostname[0] == '') {
  143.         i++;
  144.         if (i == 256) {
  145.             S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(20), 0, "GetFakeHost: FAKE table is recycled");
  146.             i = 1;
  147.             j = i;
  148.         } else j = i;
  150.         lseek(fd, 0, SEEK_SET);
  151.         if (REAL(write)(fd, (char *)&i, sizeof(int)) != sizeof(int)) {
  152.             S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "GetFakeHost: write table failed %m");
  153.             SetWriteLock(0);
  154.             return -1;
  155.         }
  157.         memset(hostname, 0, sizeof(hostname));
  158.         strncpy(hostname, name, MIN(strlen(name), S5_HOSTNAME_SIZE-1));
  159.         hostname[MIN(strlen(name), S5_HOSTNAME_SIZE-1)] = '';
  161.         lseek(fd, (j-1)*S5_HOSTNAME_SIZE+sizeof(int), SEEK_SET);
  162.         if (REAL(write)(fd, hostname, sizeof(hostname)) != sizeof(hostname)) {
  163.             S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "GetFakeHost: write table failed %m");
  164.             SetWriteLock(0);
  165.             return -1;
  166.         }
  167.     }
  169.     SetWriteLock(0);
  170.     return j;
  171. }
  173. static void HostentCopy(struct in_addr **addr_list, char **aliases, const struct hostent *h) {
  174.     int i;
  176.     for (i = 0; i < S5_HOSTALIASES_SIZE; i++) {
  177. if (h->h_aliases[i] == NULL) break;
  179. strncpy(host_aliases[i], h->h_aliases[i], MIN(strlen(h->h_aliases[i]), S5_HOSTNAME_SIZE-1));
  180. host_aliases[i][MIN(strlen(h->h_aliases[i]), S5_HOSTNAME_SIZE-1)] = '';
  181. aliases[i] = (char *)host_aliases[i];
  182.     }
  183.     aliases[i] = NULL;
  185.     for (i = 0; i < S5_HOSTLIST_SIZE; i++) {
  186. if (h->h_addr_list[i] == NULL) break;
  188. memcpy((char *)&host_addr[i], h->h_addr_list[i], sizeof(struct in_addr));
  189. addr_list[i] = &host_addr[i];
  190.     }
  191.     addr_list[i] = NULL;
  192. }
  194. /* wrapper around the gethostbyname call.                                    */
  195. /* similar to gethostbyname() except for:                                    */
  196. /* *** if gethostbyname() fails, then it returns a pointer to a hostent      */
  197. /*     structure filled with a special value, so that SOCKSxxxxxx() will     */
  198. /*     realize that this host was unresolved and fill in the protocol        */
  199. /*     accordingly...                                                        */
  200. /*                                                                           */
  201. /* returns a pointer to a gethostent structure on success; NULL on failure   */
  202. struct hostent *LIBPREFIX(gethostbyname)(const char *name) {
  203.     static struct in_addr special_addr, *my_addr_list[S5_HOSTLIST_SIZE+1];
  204.     static char my_name[MAXNAMELEN], *my_aliases[S5_HOSTALIASES_SIZE+1];
  205.     static struct hostent h;
  206.     struct hostent *hp;
  207.     char *local, *fake;
  208.     int hlen, i;
  210. #ifdef FOR_SHARED_LIBRARY
  211.     if (lsInRLDFunctions || lsInWrapFunction || lsInWrapHostname) return REAL(gethostbyname)(name);
  212. #endif
  214.     lsInWrapFunction = 1;
  215.     lsInWrapHostname = 1;
  216.     LIBPREFIX2(init)("libsocks5");
  217.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "SOCKS gethostbyname: looking up %s", name);
  219.     fake  = getenv("SOCKS5_FAKEALLHOSTS");
  220.     local = getenv("SOCKS5_LOCALDNSONLY");
  222.     if (!fake && (hp = REAL(gethostbyname)(name)) != NULL) {
  223. S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "SOCKS gethostbyname: REAL: %s", inet_ntoa(*(struct in_addr *)hp->h_addr));
  225.         hlen = MIN(strlen(hp->h_name)+1, sizeof(my_name));
  226.         strncpy(my_name, hp->h_name, hlen);
  227.         if (hlen == sizeof(my_name)) my_name[hlen-1] = '';
  229. HostentCopy(my_addr_list, my_aliases, hp);
  231.      h.h_name      = my_name;
  232.      h.h_aliases   = my_aliases;
  233.      h.h_addrtype  = hp->h_addrtype;
  234.      h.h_length    = hp->h_length;
  235.      h.h_addr_list = (char **)my_addr_list;
  237.         lsInWrapFunction = 0;
  238.         lsInWrapHostname = 0;
  239. return &h;
  240.     }
  242.     /* If your DNS is the same as the socks server, don't fake a correct     */
  243.     /* lookup when you know it won't work...                                 */
  244.     if (local) {
  245. S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "SOCKS gethostbyname: REAL: Fake not configured");
  246.         lsInWrapFunction = 0;
  247.         lsInWrapHostname = 0;
  248. return NULL;
  249.     }
  251.     /* Fill in some UNRESOLVED values and let the daemon resolve it          */
  252.     if ((i = GetFakeHost(name)) <= 0) {
  253.         S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "SOCKS gethostbyname: Get fake host failed");
  254.         lsInWrapFunction = 0;
  255.         lsInWrapHostname = 0;
  256.         return NULL;
  257.     }
  259.     hlen = MIN(strlen(name)+1, sizeof(my_name));
  260.     strncpy(my_name, name, hlen);
  261.     if (hlen == sizeof(my_name)) my_name[hlen-1] = '';
  262.     my_aliases[0] = NULL;
  264.     special_addr.s_addr = htonl(i);
  265.     my_addr_list[0] = &special_addr;
  266.     my_addr_list[1] = NULL;
  268.     h.h_name      = my_name;
  269.     h.h_aliases   = my_aliases;
  270.     h.h_addrtype  = AF_INET;
  271.     h.h_length    = sizeof(struct in_addr);
  272.     h.h_addr_list = (char **)my_addr_list;
  273.     
  274.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "SOCKS gethostbyname: FAKE: %s",  inet_ntoa(*(struct in_addr *)h.h_addr_list[0]));
  275.     lsInWrapFunction = 0;
  276.     lsInWrapHostname = 0;
  277.     return &h;
  278. }
  280. int lsGetCachedAddress(const char *name, S5NetAddr *na) {
  281.     int i;
  282.     char hostname[S5_HOSTNAME_SIZE];
  284.     if (fd > 0) {
  285.         SetReadLock(1);
  286.         lseek(fd, sizeof(int), SEEK_SET);
  288.         for (i = 1; i < 256; i++) {
  289.             if (REAL(read)(fd, hostname, sizeof(hostname)) != sizeof(hostname)) {
  290.                 S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "lsGetCachedAddress: read fake table failed %m");
  291.                 SetReadLock(0);
  292.                 return -1;
  293.             }
  295.             if (hostname[0] == '') {
  296.                 SetReadLock(0);
  297.                 return -1;
  298.             }
  299.             if (strcasecmp(name, hostname)) continue;
  301.             SetReadLock(0);
  302.             S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(20), 0, "lsGetCachedAddress: Faked host #%d, name is: %s", i, hostname);
  303.             memset(&na->sin, 0, sizeof(ssi));
  304.             na->sin.sin_family      = AF_INET;
  305.             na->sin.sin_port        = 0;
  306.             na->sin.sin_addr.s_addr = htonl(i);
  307.             return 0;
  308.         }
  309.     }
  311.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "lsGetCachedAddress: Not a fake a hostname: %s", name);
  312.     return -1;
  313. }
  315. /* checks if the address is an unresolved-address and if so, returns the     */
  316. /* unresolved hostname else returns NULL                                     */
  317. int lsGetCachedHostname(const S5NetAddr *na, char *hostname, int len) {
  318.     int i;
  320.     if (!na || !hostname) return -1;
  321.     if ((i = (int)ntohl(na->sin.sin_addr.s_addr)) > 255 || i < 1) {
  322.         S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(20), 0, "lsGetCachedHostname: Not a fake hostname: %s", inet_ntoa(na->sin.sin_addr));
  323.         return -1;
  324.     }
  326.     if (fd > 0) {
  327.         SetReadLock(1);
  328.         lseek(fd, (i-1)*S5_HOSTNAME_SIZE+sizeof(int), SEEK_SET);
  330.         if (REAL(read)(fd, hostname, len) != len) {
  331.             S5LogUpdate(S5LogDefaultHandle, S5_LOG_ERROR, 0, "lsGetCachedHostname: read fake table failed %m");
  332.             SetReadLock(0);
  333.             return -1;
  334.         }
  335.         SetReadLock(0);
  337.         if (!*hostname) return -1;
  339.         hostname[len - 1] = '';
  340.         S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(20), 0, "lsGetCachedHostname: Faked host #%d, name is: %s", i, hostname);
  341.         return 0;
  342.     }
  344.     S5LogUpdate(S5LogDefaultHandle, S5_LOG_DEBUG(10), 0, "lsGetCachedHostname: Not a fake hostname: %s", inet_ntoa(na->sin.sin_addr));
  345.     return -1;
  346. }