common.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785
  1. /*
  2. 3APA3A simpliest proxy server
  3. (c) 2002-2008 by ZARAZA <3APA3A@security.nnov.ru>
  4. please read License Agreement
  5. $Id: common.c,v 1.94 2014-04-07 21:24:45 vlad Exp $
  6. */
  7. #include "proxy.h"
  8. char * copyright = COPYRIGHT;
  9. int randomizer = 1;
  10. #ifndef _WIN32
  11. pthread_attr_t pa;
  12. #endif
  13. unsigned char **stringtable = NULL;
  14. int myinet_ntoa(struct in_addr in, char * buf){
  15. unsigned u = ntohl(in.s_addr);
  16. return sprintf(buf, "%u.%u.%u.%u",
  17. ((u&0xFF000000)>>24),
  18. ((u&0x00FF0000)>>16),
  19. ((u&0x0000FF00)>>8),
  20. ((u&0x000000FF)));
  21. }
  22. char *rotations[] = {
  23. "",
  24. "/min",
  25. "/hour",
  26. "/day",
  27. "/week",
  28. "/month",
  29. "/year",
  30. "",
  31. };
  32. struct extparam conf = {
  33. {1, 5, 30, 60, 180, 1800, 15, 60, 0, 0},
  34. NULL,
  35. NULL,
  36. NULL, NULL,
  37. NULL,
  38. NULL,
  39. 0, -1, 0, 0, 0, 0, 0, 500, 0, 0, 0,
  40. 6, 600,
  41. 1048576,
  42. NULL, NULL,
  43. NONE, NONE,
  44. NULL,
  45. {AF_INET},
  46. INADDR_ANY,
  47. 0,
  48. NULL,
  49. NULL,
  50. doconnect,
  51. lognone,
  52. NULL,
  53. NULL,
  54. NULL, NULL,
  55. NULL,
  56. NULL,
  57. NULL,
  58. NULL,
  59. NULL,
  60. NULL,
  61. (time_t)0, (time_t)0,
  62. 0,0,
  63. '@'
  64. };
  65. int myrand(void * entropy, int len){
  66. int i;
  67. unsigned short init;
  68. init = randomizer;
  69. for(i=0; i < len/2; i++){
  70. init ^= ((unsigned short *)entropy)[i];
  71. }
  72. srand(init);
  73. randomizer = rand();
  74. return rand();
  75. }
  76. #ifndef WITH_POLL
  77. int
  78. #ifdef _WIN32
  79. WINAPI
  80. #endif
  81. mypoll(struct mypollfd *fds, unsigned int nfds, int timeout){
  82. fd_set readfd;
  83. fd_set writefd;
  84. fd_set oobfd;
  85. struct timeval tv;
  86. unsigned i;
  87. int num;
  88. SOCKET maxfd = 0;
  89. tv.tv_sec = timeout/1000;
  90. tv.tv_usec = (timeout%1000)*1000;
  91. FD_ZERO(&readfd);
  92. FD_ZERO(&writefd);
  93. FD_ZERO(&oobfd);
  94. for(i=0; i<nfds; i++){
  95. if((fds[i].events&POLLIN))FD_SET(fds[i].fd, &readfd);
  96. if((fds[i].events&POLLOUT))FD_SET(fds[i].fd, &writefd);
  97. if((fds[i].events&POLLPRI))FD_SET(fds[i].fd, &oobfd);
  98. fds[i].revents = 0;
  99. if(fds[i].fd > maxfd) maxfd = fds[i].fd;
  100. }
  101. if((num = select(((int)(maxfd))+1, &readfd, &writefd, &oobfd, &tv)) < 1) return num;
  102. for(i=0; i<nfds; i++){
  103. if(FD_ISSET(fds[i].fd, &readfd)) fds[i].revents |= POLLIN;
  104. if(FD_ISSET(fds[i].fd, &writefd)) fds[i].revents |= POLLOUT;
  105. if(FD_ISSET(fds[i].fd, &oobfd)) fds[i].revents |= POLLPRI;
  106. }
  107. return num;
  108. }
  109. #endif
  110. struct sockfuncs so = {
  111. socket,
  112. accept,
  113. bind,
  114. listen,
  115. connect,
  116. getpeername,
  117. getsockname,
  118. getsockopt,
  119. setsockopt,
  120. #ifdef WITH_POLL
  121. poll,
  122. #else
  123. mypoll,
  124. #endif
  125. send,
  126. sendto,
  127. recv,
  128. recvfrom,
  129. shutdown,
  130. #ifdef _WIN32
  131. closesocket
  132. #else
  133. close
  134. #endif
  135. };
  136. #ifdef _WINCE
  137. static char cebuf[1024];
  138. static char ceargbuf[256];
  139. char * ceargv[32];
  140. char * CEToUnicode (const char *str){
  141. int i;
  142. for(i=0; i<510 && str[i]; i++){
  143. cebuf[(i*2)] = str[i];
  144. cebuf[(i*2)+1] = 0;
  145. }
  146. cebuf[(i*2)] = 0;
  147. cebuf[(i*2)+1] = 0;
  148. return cebuf;
  149. };
  150. int cesystem(const char *str){
  151. STARTUPINFO startupInfo = {0};
  152. startupInfo.cb = sizeof(startupInfo);
  153. PROCESS_INFORMATION processInformation;
  154. return CreateProcessW((LPWSTR)CEToUnicode(str), NULL, NULL, NULL, FALSE, NORMAL_PRIORITY_CLASS, NULL, NULL, &startupInfo, &processInformation);
  155. }
  156. int ceparseargs(const char *str){
  157. int argc = 0, i;
  158. int space = 1;
  159. for(i=0; i<250 && argc<30 && str[2*i]; i++){
  160. ceargbuf[i] = str[2*i];
  161. if(space && ceargbuf[i]!=' '&& ceargbuf[i]!='\t'&& ceargbuf[i]!='\r'&& ceargbuf[i]!='\n'){
  162. ceargv[argc++] = ceargbuf + i;
  163. space = 0;
  164. }
  165. else if(!space && (ceargbuf[i]==' ' || ceargbuf[i]=='\t' || ceargbuf[i]=='\r' || ceargbuf[i]=='\n')){
  166. ceargbuf[i] = 0;
  167. space = 1;
  168. }
  169. }
  170. return argc;
  171. }
  172. #endif
  173. int parsehostname(char *hostname, struct clientparam *param, unsigned short port){
  174. char *sp;
  175. if(!hostname || !*hostname)return 1;
  176. if ( (sp = strchr(hostname, ':')) ) *sp = 0;
  177. if(hostname != param->hostname){
  178. if(param->hostname) myfree(param->hostname);
  179. param->hostname = (unsigned char *)mystrdup(hostname);
  180. }
  181. if(sp){
  182. port = atoi(sp+1);
  183. *sp = ':';
  184. }
  185. param->req.sin_port=htons(port);
  186. param->req.sin_addr.s_addr = getip(param->hostname);
  187. param->sins.sin_addr.s_addr = 0;
  188. param->sins.sin_port = 0;
  189. return 0;
  190. }
  191. int parseusername(char *username, struct clientparam *param, int extpasswd){
  192. char *sb = NULL, *se = NULL, *sp = NULL;
  193. if(!username || !*username) return 1;
  194. if(!param->srv->nouser && (sb = strchr(username, ':')) && (se = strchr(sb + 1, ':')) && (!extpasswd || (sp = strchr(se + 1, ':')))){
  195. *sb = 0;
  196. *se = 0;
  197. if(sp) *sp = 0;
  198. if(*(sb+1)) {
  199. if(param->password) myfree(param->password);
  200. param->password = (unsigned char *)mystrdup(sb+1);
  201. }
  202. if(*username) {
  203. if(param->username) myfree(param->username);
  204. param->username = (unsigned char *)mystrdup(username);
  205. }
  206. username = se+1;
  207. }
  208. if(extpasswd){
  209. if(!sp) sp = strchr(username, ':');
  210. if(sp){
  211. *sp = 0;
  212. if(param->extpassword) myfree(param->extpassword);
  213. param->extpassword = (unsigned char *) mystrdup(sp+1);
  214. }
  215. }
  216. if(param->extusername) myfree(param->extusername);
  217. param->extusername = (unsigned char *)mystrdup(username);
  218. if(sb) *sb = ':';
  219. if(se) *se = ':';
  220. if(sp) *sp = ':';
  221. return 0;
  222. }
  223. int parseconnusername(char *username, struct clientparam *param, int extpasswd, unsigned short port){
  224. char *sb, *se;
  225. if(!username || !*username) return 1;
  226. if ((sb=strchr(username, conf.delimchar)) == NULL){
  227. if(!param->hostname && param->remsock == INVALID_SOCKET) return 2;
  228. return parseusername(username, param, extpasswd);
  229. }
  230. while ((se=strchr(sb+1, conf.delimchar)))sb=se;
  231. *(sb) = 0;
  232. if(parseusername(username, param, extpasswd)) return 3;
  233. *(sb) = conf.delimchar;
  234. if(parsehostname(sb+1, param, port)) return 4;
  235. return 0;
  236. }
  237. void clearstat(struct clientparam * param) {
  238. #ifdef _WIN32
  239. struct timeb tb;
  240. ftime(&tb);
  241. param->time_start = (time_t)tb.time;
  242. param->msec_start = (unsigned)tb.millitm;
  243. #else
  244. struct timeval tv;
  245. struct timezone tz;
  246. gettimeofday(&tv, &tz);
  247. param->time_start = (time_t)tv.tv_sec;
  248. param->msec_start = (tv.tv_usec / 1000);
  249. #endif
  250. param->statscli64 = param->statssrv64 = param->nreads = param->nwrites =
  251. param->nconnects = 0;
  252. }
  253. char months[12][4] = {
  254. "Jan", "Feb", "Mar", "Apr", "May", "Jun",
  255. "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
  256. };
  257. int dobuf2(struct clientparam * param, unsigned char * buf, const unsigned char *s, const unsigned char * doublec, struct tm* tm, char * format){
  258. int i, j;
  259. int len;
  260. time_t sec;
  261. unsigned msec;
  262. long timezone;
  263. unsigned delay;
  264. struct in_addr tmpia;
  265. #ifdef _WIN32
  266. struct timeb tb;
  267. ftime(&tb);
  268. sec = (time_t)tb.time;
  269. msec = (unsigned)tb.millitm;
  270. timezone = tm->tm_isdst*60 - tb.timezone;
  271. #else
  272. struct timeval tv;
  273. struct timezone tz;
  274. gettimeofday(&tv, &tz);
  275. sec = (time_t)tv.tv_sec;
  276. msec = tv.tv_usec / 1000;
  277. #ifdef _SOLARIS
  278. timezone = -altzone / 60;
  279. #else
  280. timezone = tm->tm_gmtoff / 60;
  281. #endif
  282. #endif
  283. delay = param->time_start?((unsigned) ((sec - param->time_start))*1000 + msec) - param->msec_start : 0;
  284. *buf = 0;
  285. for(i=0, j=0; format[j] && i < 4040; j++){
  286. if(format[j] == '%' && format[j+1]){
  287. j++;
  288. switch(format[j]){
  289. case '%':
  290. buf[i++] = '%';
  291. break;
  292. case 'y':
  293. sprintf((char *)buf+i, "%.2d", tm->tm_year%100);
  294. i+=2;
  295. break;
  296. case 'Y':
  297. sprintf((char *)buf+i, "%.4d", tm->tm_year+1900);
  298. i+=4;
  299. break;
  300. case 'm':
  301. sprintf((char *)buf+i, "%.2d", tm->tm_mon+1);
  302. i+=2;
  303. break;
  304. case 'o':
  305. sprintf((char *)buf+i, "%s", months[tm->tm_mon]);
  306. i+=3;
  307. break;
  308. case 'd':
  309. sprintf((char *)buf+i, "%.2d", tm->tm_mday);
  310. i+=2;
  311. break;
  312. case 'H':
  313. sprintf((char *)buf+i, "%.2d", tm->tm_hour);
  314. i+=2;
  315. break;
  316. case 'M':
  317. sprintf((char *)buf+i, "%.2d", tm->tm_min);
  318. i+=2;
  319. break;
  320. case 'S':
  321. sprintf((char *)buf+i, "%.2d", tm->tm_sec);
  322. i+=2;
  323. break;
  324. case 't':
  325. sprintf((char *)buf+i, "%.10u", (unsigned)sec);
  326. i+=10;
  327. break;
  328. case 'b':
  329. i+=sprintf((char *)buf+i, "%u", delay?(unsigned)(param->statscli64 * 1000./delay):0);
  330. break;
  331. case 'B':
  332. i+=sprintf((char *)buf+i, "%u", delay?(unsigned)(param->statssrv64 * 1000./delay):0);
  333. break;
  334. case 'D':
  335. i+=sprintf((char *)buf+i, "%u", delay);
  336. break;
  337. case '.':
  338. sprintf((char *)buf+i, "%.3u", msec);
  339. i+=3;
  340. break;
  341. case 'z':
  342. sprintf((char *)buf+i, "%+.2ld%.2u", timezone / 60, (unsigned)(timezone%60));
  343. i+=5;
  344. break;
  345. case 'U':
  346. if(param->username && *param->username){
  347. for(len = 0; i< 4000 && param->username[len]; len++){
  348. buf[i] = param->username[len];
  349. if(param->srv->nonprintable && (buf[i] < 0x20 || strchr((char *)param->srv->nonprintable, buf[i]))) buf[i] = param->srv->replace;
  350. if(doublec && strchr((char *)doublec, buf[i])) {
  351. buf[i+1] = buf[i];
  352. i++;
  353. }
  354. i++;
  355. }
  356. }
  357. else {
  358. buf[i++] = '-';
  359. }
  360. break;
  361. case 'n':
  362. len = param->hostname? (int)strlen((char *)param->hostname) : 0;
  363. if (len > 0) for(len = 0; param->hostname[len] && i < 4000; len++, i++){
  364. buf[i] = param->hostname[len];
  365. if(param->srv->nonprintable && (buf[i] < 0x20 || strchr((char *)param->srv->nonprintable, buf[i]))) buf[i] = param->srv->replace;
  366. if(doublec && strchr((char *)doublec, buf[i])) {
  367. buf[i+1] = buf[i];
  368. i++;
  369. }
  370. }
  371. else i += myinet_ntoa(param->sins.sin_addr, (char *)buf + i);
  372. break;
  373. case 'N':
  374. if(param->service >=0 && param->service < 15) {
  375. len = (conf.stringtable)? (int)strlen((char *)conf.stringtable[SERVICES + param->service]) : 0;
  376. if(len > 20) len = 20;
  377. memcpy(buf+i, (len)?conf.stringtable[SERVICES + param->service]:(unsigned char*)"-", (len)?len:1);
  378. i += (len)?len:1;
  379. }
  380. break;
  381. case 'E':
  382. sprintf((char *)buf+i, "%.05d", param->res);
  383. i += 5;
  384. break;
  385. case 'T':
  386. if(s){
  387. for(len = 0; i<4000 && s[len]; len++){
  388. buf[i] = s[len];
  389. if(param->srv->nonprintable && (buf[i] < 0x20 || strchr((char *)param->srv->nonprintable, buf[i]))) buf[i] = param->srv->replace;
  390. if(doublec && strchr((char *)doublec, buf[i])) {
  391. buf[i+1] = buf[i];
  392. i++;
  393. }
  394. i++;
  395. }
  396. }
  397. break;
  398. case 'e':
  399. tmpia.s_addr = param->extip;
  400. i += myinet_ntoa(tmpia, (char *)buf + i);
  401. break;
  402. case 'C':
  403. i += myinet_ntoa(param->sinc.sin_addr, (char *)buf + i);
  404. break;
  405. case 'R':
  406. i += myinet_ntoa(param->sins.sin_addr, (char *)buf + i);
  407. break;
  408. case 'Q':
  409. i += myinet_ntoa(param->req.sin_addr, (char *)buf + i);
  410. break;
  411. case 'p':
  412. sprintf((char *)buf+i, "%hu", ntohs(*SAPORT(&param->srv->intsa)));
  413. i += (int)strlen((char *)buf+i);
  414. break;
  415. case 'c':
  416. sprintf((char *)buf+i, "%hu", ntohs(param->sinc.sin_port));
  417. i += (int)strlen((char *)buf+i);
  418. break;
  419. case 'r':
  420. sprintf((char *)buf+i, "%hu", ntohs(param->sins.sin_port));
  421. i += (int)strlen((char *)buf+i);
  422. break;
  423. case 'q':
  424. sprintf((char *)buf+i, "%hu", ntohs(param->req.sin_port));
  425. i += (int)strlen((char *)buf+i);
  426. break;
  427. case 'I':
  428. sprintf((char *)buf+i, "%"PRINTF_INT64_MODIFIER"u", param->statssrv64);
  429. i += (int)strlen((char *)buf+i);
  430. break;
  431. case 'O':
  432. sprintf((char *)buf+i, "%"PRINTF_INT64_MODIFIER"u", param->statscli64);
  433. i += (int)strlen((char *)buf+i);
  434. break;
  435. case 'h':
  436. sprintf((char *)buf+i, "%d", param->redirected);
  437. i += (int)strlen((char *)buf+i);
  438. break;
  439. case '1':
  440. case '2':
  441. case '3':
  442. case '4':
  443. case '5':
  444. case '6':
  445. case '7':
  446. case '8':
  447. case '9':
  448. {
  449. int k, pmin=0, pmax=0;
  450. for (k = j; isnumber(format[k]); k++);
  451. if(format[k] == '-' && isnumber(format[k+1])){
  452. pmin = atoi(format + j) - 1;
  453. k++;
  454. pmax = atoi(format + k) -1;
  455. for (; isnumber(format[k]); k++);
  456. j = k;
  457. }
  458. if(!s || format[k]!='T') break;
  459. for(k = 0, len = 0; s[len] && i < 4000; len++){
  460. if(isspace(s[len])){
  461. k++;
  462. while(isspace(s[len+1]))len++;
  463. if(k == pmin) continue;
  464. }
  465. if(k>=pmin && k<=pmax) {
  466. buf[i] = s[len];
  467. if(param->srv->nonprintable && (buf[i] < 0x20 || strchr((char *)param->srv->nonprintable, buf[i]))) buf[i] = param->srv->replace;
  468. if(doublec && strchr((char *)doublec, buf[i])) {
  469. buf[i+1] = buf[i];
  470. i++;
  471. }
  472. i++;
  473. }
  474. }
  475. break;
  476. }
  477. default:
  478. buf[i++] = format[j];
  479. }
  480. }
  481. else buf[i++] = format[j];
  482. }
  483. buf[i] = 0;
  484. return i;
  485. }
  486. int dobuf(struct clientparam * param, unsigned char * buf, const unsigned char *s, const unsigned char * doublec){
  487. struct tm* tm;
  488. int i;
  489. char * format;
  490. time_t t;
  491. time(&t);
  492. if(!param) return 0;
  493. if(param->trafcountfunc)(*param->trafcountfunc)(param);
  494. format = (char *)param->srv->logformat;
  495. if(!format) format = "G%y%m%d%H%M%S.%. %p %E %U %C:%c %R:%r %O %I %h %T";
  496. tm = (*format == 'G' || *format == 'g')?
  497. gmtime(&t) : localtime(&t);
  498. i = dobuf2(param, buf, s, doublec, tm, format + 1);
  499. clearstat(param);
  500. return i;
  501. }
  502. void lognone(struct clientparam * param, const unsigned char *s) {
  503. if(param->trafcountfunc)(*param->trafcountfunc)(param);
  504. clearstat(param);
  505. }
  506. void logstdout(struct clientparam * param, const unsigned char *s) {
  507. unsigned char buf[4096];
  508. FILE *log;
  509. log = param->srv->stdlog?param->srv->stdlog:conf.stdlog?conf.stdlog:stdout;
  510. dobuf(param, buf, s, NULL);
  511. if(!param->nolog)if(fprintf(log, "%s\n", buf) < 0) {
  512. perror("printf()");
  513. };
  514. if(log != conf.stdlog)fflush(log);
  515. }
  516. #ifndef _WIN32
  517. void logsyslog(struct clientparam * param, const unsigned char *s) {
  518. unsigned char buf[4096];
  519. dobuf(param, buf, s, NULL);
  520. if(!param->nolog)syslog(LOG_INFO, "%s", buf);
  521. }
  522. #endif
  523. int doconnect(struct clientparam * param){
  524. SASIZETYPE size = sizeof(param->sins);
  525. struct sockaddr_in bindsa;
  526. if (param->operation == ADMIN || param->operation == DNSRESOLVE || param->operation == BIND || param->operation == UDPASSOC)
  527. return 0;
  528. if (param->remsock != INVALID_SOCKET){
  529. if(so._getpeername(param->remsock, (struct sockaddr *)&param->sins, &size)==-1) {return (15);}
  530. }
  531. else {
  532. struct linger lg;
  533. if(!param->sins.sin_addr.s_addr)
  534. if(!(param->sins.sin_addr.s_addr = param->req.sin_addr.s_addr)) return 100;
  535. if(!param->sins.sin_port)param->sins.sin_port = param->req.sin_port;
  536. if ((param->remsock=so._socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == INVALID_SOCKET) {return (11);}
  537. so._setsockopt(param->remsock, SOL_SOCKET, SO_LINGER, (unsigned char *)&lg, sizeof(lg));
  538. memset(&bindsa, 0, sizeof(bindsa));
  539. bindsa.sin_family = AF_INET;
  540. bindsa.sin_port = param->extport;
  541. bindsa.sin_addr.s_addr = param->extip;
  542. if (param->srv->targetport && !bindsa.sin_port && ntohs(param->sinc.sin_port) > 1023) bindsa.sin_port = param->sinc.sin_port;
  543. if(so._bind(param->remsock, (struct sockaddr*)&bindsa, sizeof(bindsa))==-1) {
  544. memset(&bindsa, 0, sizeof(bindsa));
  545. bindsa.sin_family = AF_INET;
  546. bindsa.sin_addr.s_addr = param->extip;
  547. bindsa.sin_port = 0;
  548. if(so._bind(param->remsock, (struct sockaddr*)&bindsa, sizeof(bindsa))==-1) {
  549. return 12;
  550. }
  551. }
  552. param->sins.sin_family = AF_INET;
  553. if(param->operation >= 256 || (param->operation & CONNECT)){
  554. #ifdef _WIN32
  555. unsigned long ul = 1;
  556. #endif
  557. if(so._connect(param->remsock,(struct sockaddr *)&param->sins,sizeof(param->sins))) {return (13);}
  558. param->nconnects++;
  559. #ifdef _WIN32
  560. ioctlsocket(param->remsock, FIONBIO, &ul);
  561. #else
  562. fcntl(param->remsock,F_SETFL,O_NONBLOCK);
  563. #endif
  564. if(so._getsockname(param->remsock, (struct sockaddr *)&bindsa, &size)==-1) {return (15);}
  565. param->extip = bindsa.sin_addr.s_addr;
  566. }
  567. else {
  568. if(so._getsockname(param->remsock, (struct sockaddr *)&param->sins, &size)==-1) {return (15);}
  569. }
  570. }
  571. return 0;
  572. }
  573. int scanaddr(const unsigned char *s, unsigned long * ip, unsigned long * mask) {
  574. unsigned d1, d2, d3, d4, m;
  575. int res;
  576. if ((res = sscanf((char *)s, "%u.%u.%u.%u/%u", &d1, &d2, &d3, &d4, &m)) < 4) return 0;
  577. if(mask && res == 4) *mask = 0xFFFFFFFF;
  578. else if (mask) *mask = htonl(0xFFFFFFFF << (32 - m));
  579. *ip = htonl ((d1<<24) ^ (d2<<16) ^ (d3<<8) ^ d4);
  580. return res;
  581. }
  582. RESOLVFUNC resolvfunc = NULL;
  583. #ifndef _WIN32
  584. pthread_mutex_t gethostbyname_mutex;
  585. int ghbn_init = 0;
  586. #endif
  587. #ifdef GETHOSTBYNAME_R
  588. struct hostent * my_gethostbyname(char *name, char *buf, struct hostent *hp){
  589. struct hostent *result;
  590. int gherrno;
  591. #ifdef _SOLARIS
  592. return gethostbyname_r(name, hp, buf, 1024, &gherrno);
  593. #else
  594. if(gethostbyname_r(name, hp, buf, 1024, &result, &gherrno) != 0)
  595. return NULL;
  596. return result;
  597. #endif
  598. }
  599. #endif
  600. unsigned long getip(unsigned char *name){
  601. unsigned long retval;
  602. int i;
  603. int ndots = 0;
  604. struct hostent *hp=NULL;
  605. #ifdef GETHOSTBYNAME_R
  606. struct hostent he;
  607. char ghbuf[1024];
  608. #define gethostbyname(NAME) my_gethostbyname(NAME, ghbuf, &he)
  609. #endif
  610. if(strlen((char *)name)>255)name[255] = 0;
  611. for(i=0; name[i]; i++){
  612. if(name[i] == '.'){
  613. if(++ndots > 3) break;
  614. continue;
  615. }
  616. if(name[i] <'0' || name[i] >'9') break;
  617. }
  618. if(!name[i] && ndots == 3){
  619. if(scanaddr(name, &retval, NULL) == 4){
  620. return retval;
  621. }
  622. }
  623. if(resolvfunc){
  624. if((retval = (*resolvfunc)(name))) return retval;
  625. if(conf.demanddialprog) system(conf.demanddialprog);
  626. return (*resolvfunc)(name);
  627. }
  628. #if !defined(_WIN32) && !defined(GETHOSTBYNAME_R)
  629. if(!ghbn_init){
  630. pthread_mutex_init(&gethostbyname_mutex, NULL);
  631. ghbn_init++;
  632. }
  633. pthread_mutex_lock(&gethostbyname_mutex);
  634. #endif
  635. hp=gethostbyname((char *)name);
  636. if (!hp && conf.demanddialprog) {
  637. system(conf.demanddialprog);
  638. hp=gethostbyname((char *)name);
  639. }
  640. retval = hp?*(unsigned long *)hp->h_addr:0;
  641. #if !defined(_WIN32) && !defined(GETHOSTBYNAME_R)
  642. pthread_mutex_unlock(&gethostbyname_mutex);
  643. #endif
  644. #ifdef GETHOSTBYNAME_R
  645. #undef gethostbyname
  646. #endif
  647. return retval;
  648. }
  649. unsigned long getip46(int family, unsigned char *name, struct sockaddr *sa){
  650. #ifndef NOIPV6
  651. int ndots=0, ncols=0, nhex=0;
  652. struct addrinfo *ai, *iter;
  653. struct sockaddr *sa4=NULL, *sa6=NULL;
  654. int i;
  655. if(!sa) return 0;
  656. if(!family) {
  657. #endif
  658. ((struct sockaddr_in *)sa)->sin_family = AF_INET;
  659. return (((struct sockaddr_in *)sa)->sin_addr.s_addr = getip(name))? AF_INET:0;
  660. #ifndef NOIPV6
  661. }
  662. for(i=0; name[i]; i++){
  663. if(name[i] == '.'){
  664. if(++ndots > 3) {
  665. break;
  666. }
  667. }
  668. else if(name[i] == ':'){
  669. if(++ncols > 7) {
  670. break;
  671. }
  672. }
  673. else if(name[i] == '%' || (name[i] >= 'a' && name[i] <= 'f') || (name[i] >= 'A' && name[i] <= 'F')){
  674. nhex++;
  675. }
  676. else if(name[i] <'0' || name[i] >'9') {
  677. break;
  678. }
  679. }
  680. if(!name[i]){
  681. if(ndots == 3 && ncols == 0 && nhex == 0){
  682. *SAFAMILY(sa)=AF_INET;
  683. return inet_pton(AF_INET, name, SAADDR(sa))? AF_INET : 0;
  684. }
  685. if(ncols >= 2) {
  686. *SAFAMILY(sa)=AF_INET6;
  687. return inet_pton(AF_INET6, name, SAADDR(sa))? AF_INET6 : 0;
  688. }
  689. }
  690. if (getaddrinfo(name, NULL, NULL, &ai)) return 0;
  691. for(iter = ai; iter; iter = iter->ai_next){
  692. if(!sa4 && iter->ai_addr->sa_family == AF_INET) sa4 = iter->ai_addr;
  693. if(!sa6 && iter->ai_addr->sa_family == AF_INET6) sa6 = iter->ai_addr;
  694. }
  695. if(sa6 && ((family == 6) || (family == 64) || (family == 46 && !sa4))){
  696. *SAFAMILY(sa)=AF_INET6;
  697. memcpy(SAADDR(sa), SAADDR(sa6), SAADDRLEN(sa));
  698. freeaddrinfo(ai);
  699. return AF_INET6;
  700. }
  701. else if(sa4 && family != 6){
  702. *SAFAMILY(sa)=AF_INET;
  703. memcpy(SAADDR(sa), SAADDR(sa4), SAADDRLEN(sa));
  704. freeaddrinfo(ai);
  705. return AF_INET;
  706. }
  707. freeaddrinfo(ai);
  708. return 0;
  709. #endif
  710. }