common.c 22 KB

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