log.c 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080
  1. /*
  2. 3APA3A simpliest proxy server
  3. (c) 2002-2020 by Vladimir Dubrovin <3proxy@3proxy.ru>
  4. please read License Agreement
  5. */
  6. #include "proxy.h"
  7. pthread_mutex_t log_mutex;
  8. #ifdef _WIN32
  9. HANDLE log_sem;
  10. #else
  11. sem_t log_sem;
  12. #endif
  13. #define MAXLOG 1024
  14. #define EVENTSIZE (4096 - sizeof(void *))
  15. #define LOGBUFSIZE (EVENTSIZE - sizeof(struct logevent))
  16. #define MAX_SEM_COUNT 256
  17. typedef enum {
  18. REGISTER,
  19. UNREGISTER,
  20. LOG,
  21. FLUSH,
  22. FREEPARAM
  23. } EVENTTYPE;
  24. struct clientparam logparam;
  25. struct srvparam logsrv;
  26. struct LOGGER;
  27. void(*prelog)(struct clientparam * param) = NULL;
  28. struct logevent {
  29. struct logevent *next;
  30. struct LOGGER *log;
  31. EVENTTYPE event;
  32. int inbuf;
  33. struct clientparam *param;
  34. char * logstring;
  35. char buf[1];
  36. } *logtail=NULL, *loghead=NULL;
  37. static void delayflushlogs(void);
  38. static void delayunregisterlog (struct LOGGER * log);
  39. static void delayregisterlog (struct LOGGER * log);
  40. static void delaydolog(struct logevent *evt);
  41. static void delayfreeparam(struct clientparam *param);
  42. void logpush(struct logevent *evt);
  43. #ifdef WITHMAIN
  44. #define HAVERADIUS 0
  45. #define HAVESQL 0
  46. #else
  47. int raddobuf(struct clientparam * param, unsigned char * buf, int bufsize, const unsigned char *s);
  48. void logradius(const char * buf, int len, struct LOGGER *logger);
  49. #define HAVERADIUS 1
  50. #ifndef NOODBC
  51. #define HAVESQL 1
  52. static int sqlinit(struct LOGGER *logger);
  53. static int sqldobuf(struct clientparam * param, unsigned char * buf, int bufsize, const unsigned char *s);
  54. static void sqllog(const char * buf, int len, struct LOGGER *logger);
  55. static void sqlrotate(struct LOGGER *logger);
  56. static void sqlclose(struct LOGGER *logger);
  57. #else
  58. #define HAVESQL 0
  59. #endif
  60. #endif
  61. #ifdef _WIN32
  62. #define HAVESYSLOG 0
  63. #else
  64. #define HAVESYSLOG 1
  65. static int sysloginit(struct LOGGER *logger);
  66. static void logsyslog(const char * buf, int len, struct LOGGER *logger);
  67. static void syslogrotate(struct LOGGER *logger);
  68. static void syslogclose(struct LOGGER *logger);
  69. #endif
  70. static int stdloginit(struct LOGGER *logger);
  71. static int stddobuf(struct clientparam * param, unsigned char * buf, int bufsize, const unsigned char *s);
  72. static void stdlog(const char * buf, int len, struct LOGGER *logger);
  73. static void stdlogrotate(struct LOGGER *logger);
  74. static void stdlogclose(struct LOGGER *logger);
  75. static void stdlogflush(struct LOGGER *logger);
  76. struct LOGFUNC stdlogfuncs[] = {
  77. #if HAVESYSLOG > 0
  78. {stdlogfuncs+1, sysloginit, stddobuf, logsyslog, syslogrotate, NULL, syslogclose, "@"},
  79. #endif
  80. #if HAVERADIUS > 0
  81. {stdlogfuncs+1+HAVESYSLOG, NULL, raddobuf, logradius, NULL, NULL, NULL, "radius"},
  82. #endif
  83. #if HAVESQL > 0
  84. {stdlogfuncs+1+HAVESYSLOG+HAVERADIUS, sqlinit, sqldobuf, sqllog, sqlrotate, NULL, sqlclose, "&"},
  85. #endif
  86. {NULL, stdloginit, stddobuf, stdlog, stdlogrotate, stdlogflush, stdlogclose, ""}
  87. };
  88. struct LOGFUNC *logfuncs = stdlogfuncs;
  89. struct stdlogdata{
  90. FILE *fp;
  91. } errld;
  92. struct LOGGER errlogger = {NULL, NULL, "stderr", &errld, stdlogfuncs+1+HAVESYSLOG+HAVERADIUS+HAVESQL, 0, 0, 1};
  93. struct LOGGER *loggers = &errlogger;
  94. static void delayflushlogs(void){
  95. struct LOGGER *log;
  96. for(log = loggers; log; log=log->next){
  97. if(log->logfunc && log->logfunc->flush)log->logfunc->flush(log);
  98. }
  99. }
  100. void flushlogs(void){
  101. struct logevent * evt;
  102. evt = malloc(sizeof(struct logevent));
  103. evt->event = FLUSH;
  104. logpush(evt);
  105. }
  106. void delayregisterlog(struct LOGGER *log){
  107. struct LOGFUNC *funcs;
  108. if(log->logfunc) return;
  109. for(funcs = logfuncs; funcs; funcs=funcs->next){
  110. if(!strncmp(log->selector, funcs->prefix, strlen(funcs->prefix))){
  111. if(funcs->init && funcs->init(log)) break;
  112. log->logfunc = funcs;
  113. return;
  114. }
  115. }
  116. }
  117. struct LOGGER * registerlog(const char * logstring, int logtype){
  118. struct LOGGER *log;
  119. if(!logstring || !strcmp(logstring, "NUL") || !strcmp(logstring, "/dev/null")) return NULL;
  120. pthread_mutex_lock(&log_mutex);
  121. for(log = loggers; log; log=log->next){
  122. if(!strcmp(logstring, log->selector)){
  123. if(logtype >= 0) log->rotate = logtype;
  124. log->registered++;
  125. pthread_mutex_unlock(&log_mutex);
  126. return log;
  127. }
  128. }
  129. log = malloc(sizeof(struct LOGGER));
  130. if(!log) {
  131. pthread_mutex_unlock(&log_mutex);
  132. return NULL;
  133. }
  134. memset (log, 0, sizeof(struct LOGGER));
  135. log->selector = mystrdup(logstring);
  136. if(log->selector){
  137. struct logevent *evt;
  138. if(logtype)log->rotate = logtype;
  139. log->registered++;
  140. log->next = loggers;
  141. if (log->next)log->next->prev = log;
  142. loggers = log;
  143. pthread_mutex_unlock(&log_mutex);
  144. evt = malloc(sizeof(struct logevent));
  145. evt->event = REGISTER;
  146. evt->log = log;
  147. logpush(evt);
  148. return log;
  149. }
  150. pthread_mutex_unlock(&log_mutex);
  151. myfree(log);
  152. return NULL;
  153. }
  154. static void delayunregisterlog (struct LOGGER * log){
  155. if(log){
  156. pthread_mutex_lock(&log_mutex);
  157. log->registered--;
  158. if(!log->registered){
  159. if(log->prev)log->prev->next = log->next;
  160. else loggers = log->next;
  161. if(log->next)log->next->prev = log->prev;
  162. pthread_mutex_unlock(&log_mutex);
  163. if(log->logfunc){
  164. if(log->logfunc->flush) log->logfunc->flush(log);
  165. if(log->logfunc->close) log->logfunc->close(log);
  166. }
  167. myfree(log->selector);
  168. myfree(log);
  169. }
  170. else pthread_mutex_unlock(&log_mutex);
  171. }
  172. }
  173. void unregisterlog (struct LOGGER * log){
  174. struct logevent *evt;
  175. if(!log) return;
  176. evt = malloc(sizeof(struct logevent));
  177. evt->event = UNREGISTER;
  178. evt->log = log;
  179. logpush(evt);
  180. }
  181. #ifdef _WIN32
  182. DWORD WINAPI logthreadfunc(LPVOID p) {
  183. #else
  184. void * logthreadfunc (void *p) {
  185. #endif
  186. for(;;){
  187. struct logevent *evt;
  188. #ifdef _WIN32
  189. WaitForSingleObject(log_sem, INFINITE);
  190. #else
  191. sem_wait(&log_sem);
  192. #endif
  193. while(loghead){
  194. pthread_mutex_lock(&log_mutex);
  195. evt = loghead;
  196. loghead = evt->next;
  197. if(!loghead)logtail = NULL;
  198. pthread_mutex_unlock(&log_mutex);
  199. switch(evt->event){
  200. case REGISTER:
  201. delayregisterlog(evt->log);
  202. break;
  203. case UNREGISTER:
  204. delayunregisterlog(evt->log);
  205. break;
  206. case FLUSH:
  207. delayflushlogs();
  208. break;
  209. case LOG:
  210. delaydolog(evt);
  211. break;
  212. case FREEPARAM:
  213. delayfreeparam(evt->param);
  214. break;
  215. default:
  216. break;
  217. }
  218. myfree(evt);
  219. }
  220. }
  221. return 0;
  222. }
  223. void logpush(struct logevent *evt){
  224. pthread_mutex_lock(&log_mutex);
  225. if(logtail) logtail->next = evt;
  226. logtail = evt;
  227. evt->next = NULL;
  228. if(!loghead)loghead = evt;
  229. pthread_mutex_unlock(&log_mutex);
  230. #ifdef _WIN32
  231. ReleaseSemaphore(log_sem, 1, NULL);
  232. #else
  233. sem_post(&log_sem);
  234. #endif
  235. }
  236. void initlog(void){
  237. pthread_t thread;
  238. srvinit(&logsrv, &logparam);
  239. pthread_mutex_init(&log_mutex, NULL);
  240. errld.fp = stdout;
  241. #ifdef _WIN32
  242. {
  243. HANDLE h;
  244. if(!(log_sem = CreateSemaphore(NULL, 0, MAX_SEM_COUNT, NULL))) exit(11);
  245. #ifndef _WINCE
  246. h = (HANDLE)_beginthreadex((LPSECURITY_ATTRIBUTES )NULL, 65536, (void *)logthreadfunc, NULL, 0, &thread);
  247. #else
  248. h = (HANDLE)CreateThread((LPSECURITY_ATTRIBUTES )NULL, 65536, (void *)logthreadfunc, NULL, 0, &thread);
  249. #endif
  250. if (h) {
  251. CloseHandle(h);
  252. }
  253. else {
  254. exit(10);
  255. }
  256. }
  257. #else
  258. {
  259. pthread_attr_t pa;
  260. pthread_attr_init(&pa);
  261. pthread_attr_setstacksize(&pa,PTHREAD_STACK_MIN + 1024*256);
  262. pthread_attr_setdetachstate(&pa,PTHREAD_CREATE_DETACHED);
  263. if(sem_init(&log_sem, 0, 0)) exit(11);
  264. if(pthread_create(&thread, &pa, logthreadfunc, NULL)) exit(10);
  265. }
  266. #endif
  267. }
  268. static void delaydolog(struct logevent *evt){
  269. if(!evt->log->logfunc || !evt->log->logfunc->log) return;
  270. if(evt->inbuf){
  271. evt->log->logfunc->log(evt->buf, evt->inbuf, evt->log);
  272. }
  273. else if(evt->param && evt->log->logfunc->dobuf){
  274. char buf[LOGBUFSIZE];
  275. evt->log->logfunc->log(buf, evt->log->logfunc->dobuf(evt->param, buf, LOGBUFSIZE, evt->logstring), evt->log);
  276. }
  277. }
  278. void dolog(struct clientparam * param, const unsigned char *s){
  279. static int init = 0;
  280. if(!param || !param->srv){
  281. stdlog(s, strlen(s), &errlogger);
  282. return;
  283. }
  284. if(prelog)prelog(param);
  285. if(!param->nolog && param->srv->log) {
  286. struct logevent *evt;
  287. if(!param->srv->log->logfunc) {
  288. int slen =0, hlen=0, ulen=0;
  289. slen = s?strlen(s)+1 : 0;
  290. hlen = param->hostname? strlen(param->hostname)+1 : 0;
  291. ulen = param->username? strlen(param->username)+1 : 0;
  292. if(!(evt = malloc(sizeof(struct logevent) + slen + ulen + hlen))) return;
  293. evt->inbuf = 0;
  294. evt->param=param;
  295. evt->logstring = NULL;
  296. if(slen){
  297. memcpy(evt->buf,s, slen);
  298. evt->logstring = evt->buf;
  299. }
  300. if(hlen){
  301. memcpy(evt->buf+slen,param->hostname, hlen);
  302. param->hostname = evt->buf + slen;
  303. }
  304. if(ulen){
  305. memcpy(evt->buf+slen+hlen,param->username, ulen);
  306. param->username = evt->buf + slen + hlen;
  307. }
  308. evt->event = LOG;
  309. evt->log = param->srv->log;
  310. logpush(evt);
  311. }
  312. else if (param->srv->log->logfunc->log){
  313. if(!(evt = malloc(param->srv->log->logfunc->dobuf?EVENTSIZE:sizeof(struct logevent)))) return;
  314. evt->inbuf = 0;
  315. evt->param = NULL;
  316. evt->logstring = NULL;
  317. if(param->srv->log->logfunc->dobuf){
  318. evt->inbuf = param->srv->log->logfunc->dobuf(param, evt->buf, LOGBUFSIZE, s);
  319. }
  320. evt->event = LOG;
  321. evt->log = param->srv->log;
  322. logpush(evt);
  323. }
  324. }
  325. if(param->trafcountfunc)(*param->trafcountfunc)(param);
  326. clearstat(param);
  327. }
  328. static void delayfreeparam(struct clientparam * param) {
  329. if(param->res == 2) return;
  330. if(param->ctrlsocksrv != INVALID_SOCKET && param->ctrlsocksrv != param->remsock) {
  331. so._shutdown(param->ctrlsocksrv, SHUT_RDWR);
  332. so._closesocket(param->ctrlsocksrv);
  333. }
  334. if(param->ctrlsock != INVALID_SOCKET && param->ctrlsock != param->clisock) {
  335. so._shutdown(param->ctrlsock, SHUT_RDWR);
  336. so._closesocket(param->ctrlsock);
  337. }
  338. if(param->remsock != INVALID_SOCKET) {
  339. so._shutdown(param->remsock, SHUT_RDWR);
  340. so._closesocket(param->remsock);
  341. }
  342. if(param->clisock != INVALID_SOCKET) {
  343. so._shutdown(param->clisock, SHUT_RDWR);
  344. so._closesocket(param->clisock);
  345. }
  346. myfree(param->clibuf);
  347. myfree(param->srvbuf);
  348. if(param->datfilterssrv) myfree(param->datfilterssrv);
  349. #ifndef STDMAIN
  350. if(param->reqfilters) myfree(param->reqfilters);
  351. if(param->hdrfilterscli) myfree(param->hdrfilterscli);
  352. if(param->hdrfilterssrv) myfree(param->hdrfilterssrv);
  353. if(param->predatfilters) myfree(param->predatfilters);
  354. if(param->datfilterscli) myfree(param->datfilterscli);
  355. if(param->filters){
  356. if(param->nfilters)while(param->nfilters--){
  357. if(param->filters[param->nfilters].filter->filter_clear)
  358. (*param->filters[param->nfilters].filter->filter_clear)(param->filters[param->nfilters].data);
  359. }
  360. myfree(param->filters);
  361. }
  362. if(conf.connlimiter && (param->res != 95 || param->remsock != INVALID_SOCKET)) stopconnlims(param);
  363. #endif
  364. if(param->srv){
  365. pthread_mutex_lock(&param->srv->counter_mutex);
  366. if(param->prev){
  367. param->prev->next = param->next;
  368. }
  369. else
  370. param->srv->child = param->next;
  371. if(param->next){
  372. param->next->prev = param->prev;
  373. }
  374. (param->srv->childcount)--;
  375. if(param->srv->service == S_ZOMBIE && !param->srv->child)srvpostfree(param->srv);
  376. pthread_mutex_unlock(&param->srv->counter_mutex);
  377. }
  378. if(param->hostname) myfree(param->hostname);
  379. if(param->username) myfree(param->username);
  380. if(param->password) myfree(param->password);
  381. if(param->extusername) myfree(param->extusername);
  382. if(param->extpassword) myfree(param->extpassword);
  383. myfree(param);
  384. }
  385. void freeparam(struct clientparam * param) {
  386. struct logevent *evt;
  387. evt = malloc(sizeof(struct logevent));
  388. evt->event = FREEPARAM;
  389. evt->param = param;
  390. logpush(evt);
  391. }
  392. void clearstat(struct clientparam * param) {
  393. #ifdef _WIN32
  394. struct timeb tb;
  395. ftime(&tb);
  396. param->time_start = (time_t)tb.time;
  397. param->msec_start = (unsigned)tb.millitm;
  398. #else
  399. struct timeval tv;
  400. struct timezone tz;
  401. gettimeofday(&tv, &tz);
  402. param->time_start = (time_t)tv.tv_sec;
  403. param->msec_start = (tv.tv_usec / 1000);
  404. #endif
  405. param->statscli64 = param->statssrv64 = param->nreads = param->nwrites =
  406. param->nconnects = 0;
  407. }
  408. char months[12][4] = {
  409. "Jan", "Feb", "Mar", "Apr", "May", "Jun",
  410. "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
  411. };
  412. unsigned char * dologname (unsigned char *buf, int bufsize, unsigned char *name, const unsigned char *ext, ROTATION lt, time_t t) {
  413. struct tm *ts;
  414. ts = localtime(&t);
  415. if(strchr((char *)name, '%')){
  416. struct clientparam fakecli;
  417. memset(&fakecli, 0, sizeof(fakecli));
  418. dobuf2(&fakecli, buf, bufsize - strlen(ext), NULL, NULL, ts, (char *)name);
  419. }
  420. else switch(lt){
  421. case NONE:
  422. sprintf((char *)buf, "%s", name);
  423. break;
  424. case ANNUALLY:
  425. sprintf((char *)buf, "%s.%04d", name, ts->tm_year+1900);
  426. break;
  427. case MONTHLY:
  428. sprintf((char *)buf, "%s.%04d.%02d", name, ts->tm_year+1900, ts->tm_mon+1);
  429. break;
  430. case WEEKLY:
  431. t = t - (ts->tm_wday * (60*60*24));
  432. ts = localtime(&t);
  433. sprintf((char *)buf, "%s.%04d.%02d.%02d", name, ts->tm_year+1900, ts->tm_mon+1, ts->tm_mday);
  434. break;
  435. case DAILY:
  436. sprintf((char *)buf, "%s.%04d.%02d.%02d", name, ts->tm_year+1900, ts->tm_mon+1, ts->tm_mday);
  437. break;
  438. case HOURLY:
  439. sprintf((char *)buf, "%s.%04d.%02d.%02d-%02d", name, ts->tm_year+1900, ts->tm_mon+1, ts->tm_mday, ts->tm_hour);
  440. break;
  441. case MINUTELY:
  442. sprintf((char *)buf, "%s.%04d.%02d.%02d-%02d.%02d", name, ts->tm_year+1900, ts->tm_mon+1, ts->tm_mday, ts->tm_hour, ts->tm_min);
  443. break;
  444. default:
  445. break;
  446. }
  447. if(ext){
  448. strcat((char *)buf, ".");
  449. strcat((char *)buf, (char *)ext);
  450. }
  451. return buf;
  452. }
  453. int dobuf2(struct clientparam * param, unsigned char * buf, int bufsize, const unsigned char *s, const unsigned char * doublec, struct tm* tm, char * format){
  454. int i, j;
  455. int len;
  456. time_t sec;
  457. unsigned msec;
  458. long timezone;
  459. unsigned delay;
  460. #ifdef _WIN32
  461. struct timeb tb;
  462. ftime(&tb);
  463. sec = (time_t)tb.time;
  464. msec = (unsigned)tb.millitm;
  465. timezone = tm->tm_isdst*60 - tb.timezone;
  466. #else
  467. struct timeval tv;
  468. struct timezone tz;
  469. gettimeofday(&tv, &tz);
  470. sec = (time_t)tv.tv_sec;
  471. msec = tv.tv_usec / 1000;
  472. #ifdef _SOLARIS
  473. timezone = -altzone / 60;
  474. #else
  475. timezone = tm->tm_gmtoff / 60;
  476. #endif
  477. #endif
  478. delay = param->time_start?((unsigned) ((sec - param->time_start))*1000 + msec) - param->msec_start : 0;
  479. *buf = 0;
  480. for(i=0, j=0; format[j] && i < (bufsize-70); j++){
  481. if(format[j] == '%' && format[j+1]){
  482. j++;
  483. switch(format[j]){
  484. case '%':
  485. buf[i++] = '%';
  486. break;
  487. case 'y':
  488. sprintf((char *)buf+i, "%.2d", tm->tm_year%100);
  489. i+=2;
  490. break;
  491. case 'Y':
  492. sprintf((char *)buf+i, "%.4d", tm->tm_year+1900);
  493. i+=4;
  494. break;
  495. case 'm':
  496. sprintf((char *)buf+i, "%.2d", tm->tm_mon+1);
  497. i+=2;
  498. break;
  499. case 'o':
  500. sprintf((char *)buf+i, "%s", months[tm->tm_mon]);
  501. i+=3;
  502. break;
  503. case 'd':
  504. sprintf((char *)buf+i, "%.2d", tm->tm_mday);
  505. i+=2;
  506. break;
  507. case 'H':
  508. sprintf((char *)buf+i, "%.2d", tm->tm_hour);
  509. i+=2;
  510. break;
  511. case 'M':
  512. sprintf((char *)buf+i, "%.2d", tm->tm_min);
  513. i+=2;
  514. break;
  515. case 'S':
  516. sprintf((char *)buf+i, "%.2d", tm->tm_sec);
  517. i+=2;
  518. break;
  519. case 't':
  520. sprintf((char *)buf+i, "%.10u", (unsigned)sec);
  521. i+=10;
  522. break;
  523. case 'b':
  524. i+=sprintf((char *)buf+i, "%u", delay?(unsigned)(param->statscli64 * 1000./delay):0);
  525. break;
  526. case 'B':
  527. i+=sprintf((char *)buf+i, "%u", delay?(unsigned)(param->statssrv64 * 1000./delay):0);
  528. break;
  529. case 'D':
  530. i+=sprintf((char *)buf+i, "%u", delay);
  531. break;
  532. case '.':
  533. sprintf((char *)buf+i, "%.3u", msec);
  534. i+=3;
  535. break;
  536. case 'z':
  537. sprintf((char *)buf+i, "%+.2ld%.2u", timezone / 60, (unsigned)(timezone%60));
  538. i+=5;
  539. break;
  540. case 'U':
  541. if(param->username && *param->username){
  542. for(len = 0; i< (bufsize - 3) && param->username[len]; len++){
  543. buf[i] = param->username[len];
  544. if(param->srv->nonprintable && (buf[i] < 0x20 || strchr((char *)param->srv->nonprintable, buf[i]))) buf[i] = param->srv->replace;
  545. if(doublec && strchr((char *)doublec, buf[i])) {
  546. buf[i+1] = buf[i];
  547. i++;
  548. }
  549. i++;
  550. }
  551. }
  552. else {
  553. buf[i++] = '-';
  554. }
  555. break;
  556. case 'n':
  557. len = param->hostname? (int)strlen((char *)param->hostname) : 0;
  558. if (len > 0 && !strchr((char *)param->hostname, ':')) for(len = 0; param->hostname[len] && i < (bufsize-3); len++, i++){
  559. buf[i] = param->hostname[len];
  560. if(param->srv->nonprintable && (buf[i] < 0x20 || strchr((char *)param->srv->nonprintable, buf[i]))) buf[i] = param->srv->replace;
  561. if(doublec && strchr((char *)doublec, buf[i])) {
  562. buf[i+1] = buf[i];
  563. i++;
  564. }
  565. }
  566. else {
  567. buf[i++] = '[';
  568. i += myinet_ntop(*SAFAMILY(&param->req), SAADDR(&param->req), (char *)buf + i, 64);
  569. buf[i++] = ']';
  570. }
  571. break;
  572. case 'N':
  573. if(param->service < 15) {
  574. len = (conf.stringtable)? (int)strlen((char *)conf.stringtable[SERVICES + param->service]) : 0;
  575. if(len > 20) len = 20;
  576. memcpy(buf+i, (len)?conf.stringtable[SERVICES + param->service]:(unsigned char*)"-", (len)?len:1);
  577. i += (len)?len:1;
  578. }
  579. break;
  580. case 'E':
  581. sprintf((char *)buf+i, "%.05d", param->res);
  582. i += 5;
  583. break;
  584. case 'T':
  585. if(s){
  586. for(len = 0; i < (bufsize-3) && s[len]; len++){
  587. buf[i] = s[len];
  588. if(param->srv->nonprintable && (buf[i] < 0x20 || strchr((char *)param->srv->nonprintable, buf[i]))) buf[i] = param->srv->replace;
  589. if(doublec && strchr((char *)doublec, buf[i])) {
  590. buf[i+1] = buf[i];
  591. i++;
  592. }
  593. i++;
  594. }
  595. }
  596. break;
  597. case 'e':
  598. i += myinet_ntop(*SAFAMILY(&param->sinsl), SAADDR(&param->sinsl), (char *)buf + i, 64);
  599. break;
  600. case 'i':
  601. i += myinet_ntop(*SAFAMILY(&param->sincl), SAADDR(&param->sincl), (char *)buf + i, 64);
  602. break;
  603. case 'C':
  604. i += myinet_ntop(*SAFAMILY(&param->sincr), SAADDR(&param->sincr), (char *)buf + i, 64);
  605. break;
  606. case 'R':
  607. i += myinet_ntop(*SAFAMILY(&param->sinsr), SAADDR(&param->sinsr), (char *)buf + i, 64);
  608. break;
  609. case 'Q':
  610. i += myinet_ntop(*SAFAMILY(&param->req), SAADDR(&param->req), (char *)buf + i, 64);
  611. break;
  612. case 'p':
  613. sprintf((char *)buf+i, "%hu", ntohs(*SAPORT(&param->srv->intsa)));
  614. i += (int)strlen((char *)buf+i);
  615. break;
  616. case 'c':
  617. sprintf((char *)buf+i, "%hu", ntohs(*SAPORT(&param->sincr)));
  618. i += (int)strlen((char *)buf+i);
  619. break;
  620. case 'r':
  621. sprintf((char *)buf+i, "%hu", ntohs(*SAPORT(&param->sinsr)));
  622. i += (int)strlen((char *)buf+i);
  623. break;
  624. case 'q':
  625. sprintf((char *)buf+i, "%hu", ntohs(*SAPORT(&param->req)));
  626. i += (int)strlen((char *)buf+i);
  627. break;
  628. case 'L':
  629. sprintf((char *)buf+i, "%"PRIu64, param->cycles);
  630. i += (int)strlen((char *)buf+i);
  631. break;
  632. case 'I':
  633. sprintf((char *)buf+i, "%"PRIu64, param->statssrv64);
  634. i += (int)strlen((char *)buf+i);
  635. break;
  636. case 'O':
  637. sprintf((char *)buf+i, "%"PRIu64, param->statscli64);
  638. i += (int)strlen((char *)buf+i);
  639. break;
  640. case 'h':
  641. sprintf((char *)buf+i, "%d", param->redirected);
  642. i += (int)strlen((char *)buf+i);
  643. break;
  644. case '1':
  645. case '2':
  646. case '3':
  647. case '4':
  648. case '5':
  649. case '6':
  650. case '7':
  651. case '8':
  652. case '9':
  653. {
  654. int k, pmin=0, pmax=0;
  655. for (k = j; isnumber(format[k]); k++);
  656. if(format[k] == '-' && isnumber(format[k+1])){
  657. pmin = atoi(format + j) - 1;
  658. k++;
  659. pmax = atoi(format + k) -1;
  660. for (; isnumber(format[k]); k++);
  661. j = k;
  662. }
  663. if(!s || format[k]!='T') break;
  664. for(k = 0, len = 0; s[len]; len++){
  665. if(isspace(s[len])){
  666. k++;
  667. while(isspace(s[len+1]))len++;
  668. if(k == pmin) continue;
  669. }
  670. if(k>=pmin && k<=pmax && i < (bufsize-3)) {
  671. buf[i] = s[len];
  672. if(param->srv->nonprintable && (buf[i] < 0x20 || strchr((char *)param->srv->nonprintable, buf[i]))) buf[i] = param->srv->replace;
  673. if(doublec && strchr((char *)doublec, buf[i])) {
  674. buf[i+1] = buf[i];
  675. i++;
  676. }
  677. i++;
  678. }
  679. }
  680. break;
  681. }
  682. default:
  683. buf[i++] = format[j];
  684. }
  685. }
  686. else buf[i++] = format[j];
  687. }
  688. buf[i] = 0;
  689. return i;
  690. }
  691. int dobuf(struct clientparam * param, unsigned char * buf, int bufsize, const unsigned char *s, const unsigned char * doublec){
  692. struct tm* tm;
  693. int i;
  694. char * format;
  695. time_t t;
  696. time(&t);
  697. if(!param) return 0;
  698. format = param->srv->logformat?(char *)param->srv->logformat : DEFLOGFORMAT;
  699. tm = (*format == 'G' || *format == 'g')?
  700. gmtime(&t) : localtime(&t);
  701. i = dobuf2(param, buf, bufsize, s, doublec, tm, format + 1);
  702. return i;
  703. }
  704. static int stdloginit(struct LOGGER *logger){
  705. char tmpbuf[1024];
  706. struct stdlogdata *lp;
  707. lp = myalloc(sizeof(struct stdlogdata));
  708. if(!lp) return 1;
  709. logger->data = lp;
  710. if(!*logger->selector || !strcmp(logger->selector, "stdout")){
  711. logger->rotate = NONE;
  712. lp->fp = stdout;
  713. }
  714. else if(!strcmp(logger->selector,"stderr")){
  715. logger->rotate = NONE;
  716. lp->fp = stderr;
  717. }
  718. else {
  719. lp->fp = fopen((char *)dologname (tmpbuf, sizeof(tmpbuf) - 1, logger->selector, NULL, logger->rotate, time(NULL)), "a");
  720. if(!lp->fp){
  721. myfree(lp);
  722. return(2);
  723. }
  724. }
  725. return 0;
  726. }
  727. static int stddobuf(struct clientparam * param, unsigned char * buf, int bufsize, const unsigned char *s){
  728. return dobuf(param, buf, bufsize, s, NULL);
  729. }
  730. static void stdlog(const char * buf, int len, struct LOGGER *logger) {
  731. FILE *log = ((struct stdlogdata *)logger->data)->fp;
  732. fprintf(log, "%s\n", buf);
  733. }
  734. static void stdlogrotate(struct LOGGER *logger){
  735. char tmpbuf[1024];
  736. struct stdlogdata *lp = (struct stdlogdata *)logger->data;
  737. if(lp->fp) lp->fp = freopen((char *)dologname (tmpbuf, sizeof(tmpbuf) -1, logger->selector, NULL, logger->rotate, conf.time), "a", lp->fp);
  738. else lp->fp = fopen((char *)dologname (tmpbuf, sizeof(tmpbuf) -1, logger->selector, NULL, logger->rotate, conf.time), "a");
  739. logger->rotated = conf.time;
  740. if(logger->rotate) {
  741. int t;
  742. t = 1;
  743. switch(logger->rotate){
  744. case ANNUALLY:
  745. t = t * 12;
  746. case MONTHLY:
  747. t = t * 4;
  748. case WEEKLY:
  749. t = t * 7;
  750. case DAILY:
  751. t = t * 24;
  752. case HOURLY:
  753. t = t * 60;
  754. case MINUTELY:
  755. t = t * 60;
  756. default:
  757. break;
  758. }
  759. /*
  760. FIXME: move archiver to thread
  761. */
  762. dologname (tmpbuf, sizeof(tmpbuf) -1, logger->selector, (conf.archiver)?conf.archiver[1]:NULL, logger->rotate, (logger->rotated - t * conf.rotate));
  763. remove ((char *) tmpbuf);
  764. if(conf.archiver) {
  765. int i;
  766. *tmpbuf = 0;
  767. for(i = 2; i < conf.archiverc && strlen((char *)tmpbuf) < 512; i++){
  768. strcat((char *)tmpbuf, " ");
  769. if(!strcmp((char *)conf.archiver[i], "%A")){
  770. strcat((char *)tmpbuf, "\"");
  771. dologname (tmpbuf + strlen((char *)tmpbuf), sizeof(tmpbuf) - (strlen((char *)tmpbuf) + 1), logger->selector, conf.archiver[1], logger->rotate, (logger->rotated - t));
  772. strcat((char *)tmpbuf, "\"");
  773. }
  774. else if(!strcmp((char *)conf.archiver[i], "%F")){
  775. strcat((char *)tmpbuf, "\"");
  776. dologname (tmpbuf+strlen((char *)tmpbuf), sizeof(tmpbuf) - (strlen((char *)tmpbuf) + 1), logger->selector, NULL, logger->rotate, (logger->rotated-t));
  777. strcat((char *)tmpbuf, "\"");
  778. }
  779. else
  780. strcat((char *)tmpbuf, (char *)conf.archiver[i]);
  781. }
  782. system((char *)tmpbuf+1);
  783. }
  784. }
  785. }
  786. static void stdlogflush(struct LOGGER *logger){
  787. fflush(((struct stdlogdata *)logger->data)->fp);
  788. }
  789. static void stdlogclose(struct LOGGER *logger){
  790. if(((struct stdlogdata *)logger->data)->fp != stdout && ((struct stdlogdata *)logger->data)->fp != stderr)
  791. fclose(((struct stdlogdata *)logger->data)->fp);
  792. myfree(logger->data);
  793. }
  794. #if HAVESYSLOG > 0
  795. static int sysloginit(struct LOGGER *logger){
  796. openlog(logger->selector, LOG_PID, LOG_DAEMON);
  797. return 0;
  798. }
  799. static void logsyslog(const char * buf, int len, struct LOGGER *logger) {
  800. syslog(LOG_INFO, "%s", buf);
  801. }
  802. static void syslogrotate(struct LOGGER *logger){
  803. closelog();
  804. openlog(logger->selector+1, LOG_PID, LOG_DAEMON);
  805. }
  806. static void syslogclose(struct LOGGER *logger){
  807. closelog();
  808. }
  809. #endif
  810. #if HAVESQL > 0
  811. struct sqldata {
  812. SQLHENV henv;
  813. SQLHSTMT hstmt;
  814. SQLHDBC hdbc;
  815. int attempt;
  816. time_t attempt_time;
  817. };
  818. static int sqlinit2(struct sqldata * sd, char * source){
  819. SQLRETURN retcode;
  820. char * datasource;
  821. char * username;
  822. char * password;
  823. char * string;
  824. int ret = 0;
  825. retcode = SQLAllocHandle(SQL_HANDLE_ENV, SQL_NULL_HANDLE, &sd->henv);
  826. if (!sd->henv || (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO)){
  827. return 1;
  828. }
  829. retcode = SQLSetEnvAttr(sd->henv, SQL_ATTR_ODBC_VERSION, (void*)SQL_OV_ODBC3, 0);
  830. if (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO) {
  831. ret = 2;
  832. goto CLOSEENV;
  833. }
  834. retcode = SQLAllocHandle(SQL_HANDLE_DBC, sd->henv, &sd->hdbc);
  835. if (!sd->hdbc || (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO)) {
  836. ret = 3;
  837. goto CLOSEENV;
  838. }
  839. SQLSetConnectAttr(sd->hdbc, SQL_LOGIN_TIMEOUT, (void*)15, 0);
  840. string = mystrdup(source);
  841. if(!string) goto CLOSEHDBC;
  842. datasource = strtok(string, ",");
  843. username = strtok(NULL, ",");
  844. password = strtok(NULL, ",");
  845. /* Connect to data source */
  846. retcode = SQLConnect(sd->hdbc, (SQLCHAR*) datasource, (SQLSMALLINT)strlen(datasource),
  847. (SQLCHAR*) username, (SQLSMALLINT)((username)?strlen(username):0),
  848. (SQLCHAR*) password, (SQLSMALLINT)((password)?strlen(password):0));
  849. myfree(string);
  850. if (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO){
  851. ret = 4;
  852. goto CLOSEHDBC;
  853. }
  854. retcode = SQLAllocHandle(SQL_HANDLE_STMT, sd->hdbc, &sd->hstmt);
  855. if (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO){
  856. sd->hstmt = 0;
  857. ret = 5;
  858. goto CLOSEHDBC;
  859. }
  860. return 0;
  861. CLOSEHDBC:
  862. SQLFreeHandle(SQL_HANDLE_DBC, sd->hdbc);
  863. sd->hdbc = 0;
  864. CLOSEENV:
  865. SQLFreeHandle(SQL_HANDLE_ENV, sd->henv);
  866. sd->henv = 0;
  867. return ret;
  868. }
  869. static int sqlinit(struct LOGGER *logger){
  870. struct sqldata *sd;
  871. int res;
  872. sd = (struct sqldata *)myalloc(sizeof(struct sqldata));
  873. memset(sd, 0, sizeof(struct sqldata));
  874. logger->data = sd;
  875. if((res = sqlinit2(sd, logger->selector))) {
  876. myfree(sd);
  877. return res;
  878. }
  879. return 0;
  880. }
  881. static int sqldobuf(struct clientparam * param, unsigned char * buf, int bufsize, const unsigned char *s){
  882. return dobuf(param, buf, bufsize, s, (unsigned char *)"\'");
  883. }
  884. static void sqllog(const char * buf, int len, struct LOGGER *logger){
  885. SQLRETURN ret;
  886. struct sqldata *sd = (struct sqldata *)logger->data;
  887. if(sd->attempt > 5){
  888. if (conf.time - sd->attempt_time < 180){
  889. return;
  890. }
  891. }
  892. if(sd->attempt){
  893. sd->attempt++;
  894. sqlrotate(logger);
  895. if(!sd->hstmt){
  896. sd->attempt_time=conf.time;
  897. return;
  898. }
  899. }
  900. ret = SQLExecDirect(sd->hstmt, (SQLCHAR *)buf, (SQLINTEGER)len);
  901. if(ret != SQL_SUCCESS && ret != SQL_SUCCESS_WITH_INFO){
  902. sqlrotate(logger);
  903. if(sd->hstmt) {
  904. ret = SQLExecDirect(sd->hstmt, (SQLCHAR *)buf, (SQLINTEGER)len);
  905. if(ret != SQL_SUCCESS && ret != SQL_SUCCESS_WITH_INFO){
  906. sd->attempt++;
  907. sd->attempt_time=conf.time;
  908. return;
  909. }
  910. }
  911. }
  912. sd->attempt=0;
  913. }
  914. static void sqlrotate(struct LOGGER *logger){
  915. struct sqldata * sd;
  916. sqlclose(logger);
  917. sd = (struct sqldata *)myalloc(sizeof(struct sqldata));
  918. memset(sd, 0, sizeof(struct sqldata));
  919. logger->data = sd;
  920. sqlinit2(sd, logger->selector+1);
  921. }
  922. static void sqlclose(struct LOGGER *logger){
  923. struct sqldata *sd = (struct sqldata *)logger->data;
  924. if(sd->hstmt) {
  925. SQLFreeHandle(SQL_HANDLE_STMT, sd->hstmt);
  926. sd->hstmt = NULL;
  927. }
  928. if(sd->hdbc){
  929. SQLDisconnect(sd->hdbc);
  930. SQLFreeHandle(SQL_HANDLE_DBC, sd->hdbc);
  931. sd->hdbc = NULL;
  932. }
  933. if(sd->henv) {
  934. SQLFreeHandle(SQL_HANDLE_ENV, sd->henv);
  935. sd->henv = NULL;
  936. }
  937. myfree(sd);
  938. }
  939. #endif