common.c 21 KB

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