common.c 22 KB

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