common.cpp 25 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300
  1. /*
  2. * comm.cpp
  3. *
  4. * Created on: Jul 29, 2017
  5. * Author: wangyu
  6. */
  7. #include "common.h"
  8. #include "log.h"
  9. #include "misc.h"
  10. #include <random>
  11. #include <cmath>
  12. //static int random_number_fd=-1;
  13. int force_socket_buf=0;
  14. int address_t::from_str(char *str)
  15. {
  16. clear();
  17. char ip_addr_str[100];u32_t port;
  18. mylog(log_info,"parsing address: %s\n",str);
  19. int is_ipv6=0;
  20. if(sscanf(str, "[%[^]]]:%u", ip_addr_str,&port)==2)
  21. {
  22. mylog(log_info,"its an ipv6 adress\n");
  23. inner.ipv6.sin6_family=AF_INET6;
  24. is_ipv6=1;
  25. }
  26. else if(sscanf(str, "%[^:]:%u", ip_addr_str,&port)==2)
  27. {
  28. mylog(log_info,"its an ipv4 adress\n");
  29. inner.ipv4.sin_family=AF_INET;
  30. }
  31. else
  32. {
  33. mylog(log_error,"failed to parse\n");
  34. myexit(-1);
  35. }
  36. mylog(log_info,"ip_address is {%s}, port is {%u}\n",ip_addr_str,port);
  37. if(port>65535)
  38. {
  39. mylog(log_error,"invalid port: %d\n",port);
  40. myexit(-1);
  41. }
  42. int ret=-100;
  43. if(is_ipv6)
  44. {
  45. ret=inet_pton(AF_INET6, ip_addr_str,&(inner.ipv6.sin6_addr));
  46. inner.ipv6.sin6_port=htons(port);
  47. if(ret==0) // 0 if address type doesnt match
  48. {
  49. mylog(log_error,"ip_addr %s is not an ipv6 address, %d\n",ip_addr_str,ret);
  50. myexit(-1);
  51. }
  52. else if(ret==1) // inet_pton returns 1 on success
  53. {
  54. //okay
  55. }
  56. else
  57. {
  58. mylog(log_error,"ip_addr %s is invalid, %d\n",ip_addr_str,ret);
  59. myexit(-1);
  60. }
  61. }
  62. else
  63. {
  64. ret=inet_pton(AF_INET, ip_addr_str,&(inner.ipv4.sin_addr));
  65. inner.ipv4.sin_port=htons(port);
  66. if(ret==0)
  67. {
  68. mylog(log_error,"ip_addr %s is not an ipv4 address, %d\n",ip_addr_str,ret);
  69. myexit(-1);
  70. }
  71. else if(ret==1)
  72. {
  73. //okay
  74. }
  75. else
  76. {
  77. mylog(log_error,"ip_addr %s is invalid, %d\n",ip_addr_str,ret);
  78. myexit(-1);
  79. }
  80. }
  81. return 0;
  82. }
  83. int address_t::from_str_ip_only(char * str)
  84. {
  85. clear();
  86. u32_t type;
  87. if(strchr(str,':')==NULL)
  88. type=AF_INET;
  89. else
  90. type=AF_INET6;
  91. ((sockaddr*)&inner)->sa_family=type;
  92. int ret;
  93. if(type==AF_INET)
  94. {
  95. ret=inet_pton(type, str,&inner.ipv4.sin_addr);
  96. }
  97. else
  98. {
  99. ret=inet_pton(type, str,&inner.ipv6.sin6_addr);
  100. }
  101. if(ret==0) // 0 if address type doesnt match
  102. {
  103. mylog(log_error,"confusion in parsing %s, %d\n",str,ret);
  104. myexit(-1);
  105. }
  106. else if(ret==1) // inet_pton returns 1 on success
  107. {
  108. //okay
  109. }
  110. else
  111. {
  112. mylog(log_error,"ip_addr %s is invalid, %d\n",str,ret);
  113. myexit(-1);
  114. }
  115. return 0;
  116. }
  117. char * address_t::get_str()
  118. {
  119. static char res[max_addr_len];
  120. to_str(res);
  121. return res;
  122. }
  123. void address_t::to_str(char * s)
  124. {
  125. //static char res[max_addr_len];
  126. char ip_addr[max_addr_len];
  127. u32_t port;
  128. const char * ret=0;
  129. if(get_type()==AF_INET6)
  130. {
  131. ret=inet_ntop(AF_INET6, &inner.ipv6.sin6_addr, ip_addr,max_addr_len);
  132. port=inner.ipv6.sin6_port;
  133. }
  134. else if(get_type()==AF_INET)
  135. {
  136. ret=inet_ntop(AF_INET, &inner.ipv4.sin_addr, ip_addr,max_addr_len);
  137. port=inner.ipv4.sin_port;
  138. }
  139. else
  140. {
  141. assert(0==1);
  142. }
  143. if(ret==0) //NULL on failure
  144. {
  145. mylog(log_error,"inet_ntop failed\n");
  146. myexit(-1);
  147. }
  148. port=ntohs(port);
  149. ip_addr[max_addr_len-1]=0;
  150. if(get_type()==AF_INET6)
  151. {
  152. sprintf(s,"[%s]:%u",ip_addr,(u32_t)port);
  153. }else
  154. {
  155. sprintf(s,"%s:%u",ip_addr,(u32_t)port);
  156. }
  157. //return res;
  158. }
  159. char* address_t::get_ip()
  160. {
  161. char ip_addr[max_addr_len];
  162. static char s[max_addr_len];
  163. const char * ret=0;
  164. if(get_type()==AF_INET6)
  165. {
  166. ret=inet_ntop(AF_INET6, &inner.ipv6.sin6_addr, ip_addr,max_addr_len);
  167. }
  168. else if(get_type()==AF_INET)
  169. {
  170. ret=inet_ntop(AF_INET, &inner.ipv4.sin_addr, ip_addr,max_addr_len);
  171. }
  172. else
  173. {
  174. assert(0==1);
  175. }
  176. if(ret==0) //NULL on failure
  177. {
  178. mylog(log_error,"inet_ntop failed\n");
  179. myexit(-1);
  180. }
  181. ip_addr[max_addr_len-1]=0;
  182. if(get_type()==AF_INET6)
  183. {
  184. sprintf(s,"%s",ip_addr);
  185. }else
  186. {
  187. sprintf(s,"%s",ip_addr);
  188. }
  189. return s;
  190. }
  191. int address_t::from_sockaddr(sockaddr * addr,socklen_t slen)
  192. {
  193. clear();
  194. //memset(&inner,0,sizeof(inner));
  195. if(addr->sa_family==AF_INET6)
  196. {
  197. assert(slen==sizeof(sockaddr_in6));
  198. //inner.ipv6= *( (sockaddr_in6*) addr );
  199. memcpy(&inner,addr,slen);
  200. }
  201. else if(addr->sa_family==AF_INET)
  202. {
  203. assert(slen==sizeof(sockaddr_in));
  204. //inner.ipv4= *( (sockaddr_in*) addr );
  205. memcpy(&inner,addr,slen);
  206. }
  207. else
  208. {
  209. assert(0==1);
  210. }
  211. return 0;
  212. }
  213. int address_t::new_connected_udp_fd()
  214. {
  215. int new_udp_fd;
  216. new_udp_fd = socket(get_type(), SOCK_DGRAM, IPPROTO_UDP);
  217. if (new_udp_fd < 0) {
  218. mylog(log_warn, "create udp_fd error\n");
  219. return -1;
  220. }
  221. setnonblocking(new_udp_fd);
  222. set_buf_size(new_udp_fd,socket_buf_size);
  223. mylog(log_debug, "created new udp_fd %d\n", new_udp_fd);
  224. int ret = connect(new_udp_fd, (struct sockaddr *) &inner, get_len());
  225. if (ret != 0) {
  226. mylog(log_warn, "udp fd connect fail %d %s\n",ret,strerror(errno) );
  227. //sock_close(new_udp_fd);
  228. close(new_udp_fd);
  229. return -1;
  230. }
  231. return new_udp_fd;
  232. }
  233. bool my_ip_t::equal (const my_ip_t &b) const
  234. {
  235. //extern int raw_ip_version;
  236. if(raw_ip_version==AF_INET)
  237. {
  238. return v4==b.v4;
  239. }else if(raw_ip_version==AF_INET6)
  240. {
  241. return memcmp(&v6,&b.v6,sizeof(v6))==0;
  242. }
  243. assert(0==1);
  244. return 0;
  245. }
  246. char * my_ip_t::get_str1() const
  247. {
  248. static char res[max_addr_len];
  249. if(raw_ip_version==AF_INET6)
  250. {
  251. assert(inet_ntop(AF_INET6, &v6, res,max_addr_len)!=0);
  252. }
  253. else
  254. {
  255. assert(raw_ip_version==AF_INET);
  256. assert(inet_ntop(AF_INET, &v4, res,max_addr_len)!=0);
  257. }
  258. return res;
  259. }
  260. char * my_ip_t::get_str2() const
  261. {
  262. static char res[max_addr_len];
  263. if(raw_ip_version==AF_INET6)
  264. {
  265. assert(inet_ntop(AF_INET6, &v6, res,max_addr_len)!=0);
  266. }
  267. else
  268. {
  269. assert(raw_ip_version==AF_INET);
  270. assert(inet_ntop(AF_INET, &v4, res,max_addr_len)!=0);
  271. }
  272. return res;
  273. }
  274. int my_ip_t::from_address_t(address_t tmp_addr)
  275. {
  276. if(tmp_addr.get_type()==raw_ip_version&&raw_ip_version==AF_INET)
  277. {
  278. v4=tmp_addr.inner.ipv4.sin_addr.s_addr;
  279. }
  280. else if(tmp_addr.get_type()==raw_ip_version&&raw_ip_version==AF_INET6)
  281. {
  282. v6=tmp_addr.inner.ipv6.sin6_addr;
  283. }
  284. else
  285. {
  286. assert(0==1);
  287. }
  288. return 0;
  289. }
  290. /*
  291. int my_ip_t::from_str(char * str)
  292. {
  293. u32_t type;
  294. if(strchr(str,':')==NULL)
  295. type=AF_INET;
  296. else
  297. type=AF_INET6;
  298. int ret;
  299. ret=inet_pton(type, str,this);
  300. if(ret==0) // 0 if address type doesnt match
  301. {
  302. mylog(log_error,"confusion in parsing %s, %d\n",str,ret);
  303. myexit(-1);
  304. }
  305. else if(ret==1) // inet_pton returns 1 on success
  306. {
  307. //okay
  308. }
  309. else
  310. {
  311. mylog(log_error,"ip_addr %s is invalid, %d\n",str,ret);
  312. myexit(-1);
  313. }
  314. return 0;
  315. }*/
  316. int init_ws()
  317. {
  318. #if defined(__MINGW32__)
  319. WORD wVersionRequested;
  320. WSADATA wsaData;
  321. int err;
  322. /* Use the MAKEWORD(lowbyte, highbyte) macro declared in Windef.h */
  323. wVersionRequested = MAKEWORD(2, 2);
  324. err = WSAStartup(wVersionRequested, &wsaData);
  325. if (err != 0) {
  326. /* Tell the user that we could not find a usable */
  327. /* Winsock DLL. */
  328. printf("WSAStartup failed with error: %d\n", err);
  329. exit(-1);
  330. }
  331. /* Confirm that the WinSock DLL supports 2.2.*/
  332. /* Note that if the DLL supports versions greater */
  333. /* than 2.2 in addition to 2.2, it will still return */
  334. /* 2.2 in wVersion since that is the version we */
  335. /* requested. */
  336. if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2) {
  337. /* Tell the user that we could not find a usable */
  338. /* WinSock DLL. */
  339. printf("Could not find a usable version of Winsock.dll\n");
  340. WSACleanup();
  341. exit(-1);
  342. }
  343. else
  344. {
  345. printf("The Winsock 2.2 dll was found okay");
  346. }
  347. int tmp[]={0,100,200,300,500,800,1000,2000,3000,4000,-1};
  348. int succ=0;
  349. for(int i=1;tmp[i]!=-1;i++)
  350. {
  351. if(_setmaxstdio(100)==-1) break;
  352. else succ=i;
  353. }
  354. printf(", _setmaxstdio() was set to %d\n",tmp[succ]);
  355. #endif
  356. return 0;
  357. }
  358. #if defined(__MINGW32__)
  359. int inet_pton(int af, const char *src, void *dst)
  360. {
  361. struct sockaddr_storage ss;
  362. int size = sizeof(ss);
  363. char src_copy[max_addr_len+1];
  364. ZeroMemory(&ss, sizeof(ss));
  365. /* stupid non-const API */
  366. strncpy (src_copy, src, max_addr_len+1);
  367. src_copy[max_addr_len] = 0;
  368. if (WSAStringToAddress(src_copy, af, NULL, (struct sockaddr *)&ss, &size) == 0) {
  369. switch(af) {
  370. case AF_INET:
  371. *(struct in_addr *)dst = ((struct sockaddr_in *)&ss)->sin_addr;
  372. return 1;
  373. case AF_INET6:
  374. *(struct in6_addr *)dst = ((struct sockaddr_in6 *)&ss)->sin6_addr;
  375. return 1;
  376. }
  377. }
  378. return 0;
  379. }
  380. const char *inet_ntop(int af, const void *src, char *dst, socklen_t size)
  381. {
  382. struct sockaddr_storage ss;
  383. unsigned long s = size;
  384. ZeroMemory(&ss, sizeof(ss));
  385. ss.ss_family = af;
  386. switch(af) {
  387. case AF_INET:
  388. ((struct sockaddr_in *)&ss)->sin_addr = *(struct in_addr *)src;
  389. break;
  390. case AF_INET6:
  391. ((struct sockaddr_in6 *)&ss)->sin6_addr = *(struct in6_addr *)src;
  392. break;
  393. default:
  394. return NULL;
  395. }
  396. /* cannot direclty use &size because of strict aliasing rules */
  397. return (WSAAddressToString((struct sockaddr *)&ss, sizeof(ss), NULL, dst, &s) == 0)?
  398. dst : NULL;
  399. }
  400. char *get_sock_error()
  401. {
  402. static char buf[1000];
  403. int e=WSAGetLastError();
  404. wchar_t *s = NULL;
  405. FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
  406. NULL, e,
  407. MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
  408. (LPWSTR)&s, 0, NULL);
  409. sprintf(buf, "%d:%S", e,s);
  410. int len=strlen(buf);
  411. while(len>0 && (buf[len-1]=='\r'||buf[len-1]=='\n' ))
  412. {
  413. len--;
  414. buf[len]=0;
  415. }
  416. LocalFree(s);
  417. return buf;
  418. }
  419. int get_sock_errno()
  420. {
  421. return WSAGetLastError();
  422. }
  423. #else
  424. char *get_sock_error()
  425. {
  426. static char buf[1000];
  427. sprintf(buf, "%d:%s", errno,strerror(errno));
  428. return buf;
  429. }
  430. int get_sock_errno()
  431. {
  432. return errno;
  433. }
  434. #endif
  435. u64_t get_current_time()
  436. {
  437. //timespec tmp_time;
  438. //clock_gettime(CLOCK_MONOTONIC, &tmp_time);
  439. //return ((u64_t)tmp_time.tv_sec)*1000llu+((u64_t)tmp_time.tv_nsec)/(1000*1000llu);
  440. return (u64_t)(ev_time()*1000);
  441. }
  442. u64_t pack_u64(u32_t a,u32_t b)
  443. {
  444. u64_t ret=a;
  445. ret<<=32u;
  446. ret+=b;
  447. return ret;
  448. }
  449. u32_t get_u64_h(u64_t a)
  450. {
  451. return a>>32u;
  452. }
  453. u32_t get_u64_l(u64_t a)
  454. {
  455. return (a<<32u)>>32u;
  456. }
  457. char * my_ntoa(u32_t ip)
  458. {
  459. in_addr a;
  460. a.s_addr=ip;
  461. return inet_ntoa(a);
  462. }
  463. /*
  464. void init_random_number_fd()
  465. {
  466. random_number_fd=open("/dev/urandom",O_RDONLY);
  467. if(random_number_fd==-1)
  468. {
  469. mylog(log_fatal,"error open /dev/urandom\n");
  470. myexit(-1);
  471. }
  472. setnonblocking(random_number_fd);
  473. }*/
  474. #if !defined(__MINGW32__)
  475. struct random_fd_t
  476. {
  477. int random_number_fd;
  478. random_fd_t()
  479. {
  480. random_number_fd=open("/dev/urandom",O_RDONLY);
  481. if(random_number_fd==-1)
  482. {
  483. mylog(log_fatal,"error open /dev/urandom\n");
  484. myexit(-1);
  485. }
  486. setnonblocking(random_number_fd);
  487. }
  488. int get_fd()
  489. {
  490. return random_number_fd;
  491. }
  492. }random_fd;
  493. #else
  494. struct my_random_t
  495. {
  496. std::random_device rd;
  497. std::mt19937 gen;
  498. std::uniform_int_distribution<u64_t> dis64;
  499. std::uniform_int_distribution<u32_t> dis32;
  500. std::uniform_int_distribution<unsigned char> dis8;
  501. my_random_t()
  502. {
  503. //std::mt19937 gen_tmp(rd()); //random device is broken on mingw
  504. timespec tmp_time;
  505. clock_gettime(CLOCK_MONOTONIC, &tmp_time);
  506. long long a=((u64_t)tmp_time.tv_sec)*1000000000llu+((u64_t)tmp_time.tv_nsec);
  507. std::mt19937 gen_tmp(a);
  508. gen=gen_tmp;
  509. gen.discard(700000); //magic
  510. }
  511. u64_t gen64()
  512. {
  513. return dis64(gen);
  514. }
  515. u32_t gen32()
  516. {
  517. return dis32(gen);
  518. }
  519. unsigned char gen8()
  520. {
  521. return dis8(gen);
  522. }
  523. /*int random_number_fd;
  524. random_fd_t()
  525. {
  526. random_number_fd=open("/dev/urandom",O_RDONLY);
  527. if(random_number_fd==-1)
  528. {
  529. mylog(log_fatal,"error open /dev/urandom\n");
  530. myexit(-1);
  531. }
  532. setnonblocking(random_number_fd);
  533. }
  534. int get_fd()
  535. {
  536. return random_number_fd;
  537. }*/
  538. }my_random;
  539. #endif
  540. u64_t get_true_random_number_64()
  541. {
  542. #if !defined(__MINGW32__)
  543. u64_t ret;
  544. int size=read(random_fd.get_fd(),&ret,sizeof(ret));
  545. if(size!=sizeof(ret))
  546. {
  547. mylog(log_fatal,"get random number failed %d\n",size);
  548. myexit(-1);
  549. }
  550. return ret;
  551. #else
  552. return my_random.gen64(); //fake random number
  553. #endif
  554. }
  555. u32_t get_true_random_number()
  556. {
  557. #if !defined(__MINGW32__)
  558. u32_t ret;
  559. int size=read(random_fd.get_fd(),&ret,sizeof(ret));
  560. if(size!=sizeof(ret))
  561. {
  562. mylog(log_fatal,"get random number failed %d\n",size);
  563. myexit(-1);
  564. }
  565. return ret;
  566. #else
  567. return my_random.gen32(); //fake random number
  568. #endif
  569. }
  570. u32_t get_true_random_number_nz() //nz for non-zero
  571. {
  572. u32_t ret=0;
  573. while(ret==0)
  574. {
  575. ret=get_true_random_number();
  576. }
  577. return ret;
  578. }
  579. inline int is_big_endian()
  580. {
  581. int i=1;
  582. return ! *((char *)&i);
  583. }
  584. u64_t ntoh64(u64_t a)
  585. {
  586. #ifdef UDP2RAW_LITTLE_ENDIAN
  587. u32_t h=get_u64_h(a);
  588. u32_t l=get_u64_l(a);
  589. return pack_u64(ntohl(l),ntohl(h));
  590. //return bswap_64( a);
  591. #else
  592. return a;
  593. #endif
  594. }
  595. u64_t hton64(u64_t a)
  596. {
  597. return ntoh64(a);
  598. }
  599. void write_u16(char * p,u16_t w)
  600. {
  601. *(unsigned char*)(p + 1) = (w & 0xff);
  602. *(unsigned char*)(p + 0) = (w >> 8);
  603. }
  604. u16_t read_u16(char * p)
  605. {
  606. u16_t res;
  607. res = *(const unsigned char*)(p + 0);
  608. res = *(const unsigned char*)(p + 1) + (res << 8);
  609. return res;
  610. }
  611. void write_u32(char * p,u32_t l)
  612. {
  613. *(unsigned char*)(p + 3) = (unsigned char)((l >> 0) & 0xff);
  614. *(unsigned char*)(p + 2) = (unsigned char)((l >> 8) & 0xff);
  615. *(unsigned char*)(p + 1) = (unsigned char)((l >> 16) & 0xff);
  616. *(unsigned char*)(p + 0) = (unsigned char)((l >> 24) & 0xff);
  617. }
  618. u32_t read_u32(char * p)
  619. {
  620. u32_t res;
  621. res = *(const unsigned char*)(p + 0);
  622. res = *(const unsigned char*)(p + 1) + (res << 8);
  623. res = *(const unsigned char*)(p + 2) + (res << 8);
  624. res = *(const unsigned char*)(p + 3) + (res << 8);
  625. return res;
  626. }
  627. void write_u64(char * s,u64_t a)
  628. {
  629. assert(0==1);
  630. }
  631. u64_t read_u64(char * s)
  632. {
  633. assert(0==1);
  634. return 0;
  635. }
  636. void setnonblocking(int sock) {
  637. #if !defined(__MINGW32__)
  638. int opts;
  639. opts = fcntl(sock, F_GETFL);
  640. if (opts < 0) {
  641. mylog(log_fatal,"fcntl(sock,GETFL)\n");
  642. //perror("fcntl(sock,GETFL)");
  643. myexit(1);
  644. }
  645. opts = opts | O_NONBLOCK;
  646. if (fcntl(sock, F_SETFL, opts) < 0) {
  647. mylog(log_fatal,"fcntl(sock,SETFL,opts)\n");
  648. //perror("fcntl(sock,SETFL,opts)");
  649. myexit(1);
  650. }
  651. #else
  652. int iResult;
  653. u_long iMode = 1;
  654. iResult = ioctlsocket(sock, FIONBIO, &iMode);
  655. if (iResult != NO_ERROR)
  656. printf("ioctlsocket failed with error: %d\n", iResult);
  657. #endif
  658. }
  659. /*
  660. Generic checksum calculation function
  661. */
  662. unsigned short csum(const unsigned short *ptr,int nbytes) {//works both for big and little endian
  663. long sum;
  664. unsigned short oddbyte;
  665. short answer;
  666. sum=0;
  667. while(nbytes>1) {
  668. sum+=*ptr++;
  669. nbytes-=2;
  670. }
  671. if(nbytes==1) {
  672. oddbyte=0;
  673. *((u_char*)&oddbyte)=*(u_char*)ptr;
  674. sum+=oddbyte;
  675. }
  676. sum = (sum>>16)+(sum & 0xffff);
  677. sum = sum + (sum>>16);
  678. answer=(short)~sum;
  679. return(answer);
  680. }
  681. unsigned short csum_with_header(char* header,int hlen,const unsigned short *ptr,int nbytes) {//works both for big and little endian
  682. long sum;
  683. unsigned short oddbyte;
  684. short answer;
  685. assert(hlen%2==0);
  686. sum=0;
  687. unsigned short * tmp= (unsigned short *)header;
  688. for(int i=0;i<hlen/2;i++)
  689. {
  690. sum+=*tmp++;
  691. }
  692. while(nbytes>1) {
  693. sum+=*ptr++;
  694. nbytes-=2;
  695. }
  696. if(nbytes==1) {
  697. oddbyte=0;
  698. *((u_char*)&oddbyte)=*(u_char*)ptr;
  699. sum+=oddbyte;
  700. }
  701. sum = (sum>>16)+(sum & 0xffff);
  702. sum = sum + (sum>>16);
  703. answer=(short)~sum;
  704. return(answer);
  705. }
  706. int set_buf_size(int fd,int socket_buf_size)
  707. {
  708. if(force_socket_buf)
  709. {
  710. mylog(log_fatal,"force_socket_buf not supported in this verion\n");
  711. myexit(-1);
  712. //assert(0==1);
  713. #if 0
  714. if(setsockopt(fd, SOL_SOCKET, SO_SNDBUFFORCE, &socket_buf_size, sizeof(socket_buf_size))<0)
  715. {
  716. mylog(log_fatal,"SO_SNDBUFFORCE fail socket_buf_size=%d errno=%s\n",socket_buf_size,strerror(errno));
  717. myexit(1);
  718. }
  719. if(setsockopt(fd, SOL_SOCKET, SO_RCVBUFFORCE, &socket_buf_size, sizeof(socket_buf_size))<0)
  720. {
  721. mylog(log_fatal,"SO_RCVBUFFORCE fail socket_buf_size=%d errno=%s\n",socket_buf_size,strerror(errno));
  722. myexit(1);
  723. }
  724. #endif
  725. }
  726. else
  727. {
  728. if(setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &socket_buf_size, sizeof(socket_buf_size))<0)
  729. {
  730. mylog(log_fatal,"SO_SNDBUF fail socket_buf_size=%d errno=%s\n",socket_buf_size,get_sock_error());
  731. myexit(1);
  732. }
  733. if(setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &socket_buf_size, sizeof(socket_buf_size))<0)
  734. {
  735. mylog(log_fatal,"SO_RCVBUF fail socket_buf_size=%d errno=%s\n",socket_buf_size,get_sock_error());
  736. myexit(1);
  737. }
  738. }
  739. return 0;
  740. }
  741. int numbers_to_char(my_id_t id1,my_id_t id2,my_id_t id3,char * &data,int &len)
  742. {
  743. static char buf[buf_len];
  744. data=buf;
  745. my_id_t tmp=htonl(id1);
  746. memcpy(buf,&tmp,sizeof(tmp));
  747. tmp=htonl(id2);
  748. memcpy(buf+sizeof(tmp),&tmp,sizeof(tmp));
  749. tmp=htonl(id3);
  750. memcpy(buf+sizeof(tmp)*2,&tmp,sizeof(tmp));
  751. len=sizeof(my_id_t)*3;
  752. return 0;
  753. }
  754. int char_to_numbers(const char * data,int len,my_id_t &id1,my_id_t &id2,my_id_t &id3)
  755. {
  756. if(len<int(sizeof(my_id_t)*3)) return -1;
  757. //id1=ntohl( *((id_t*)(data+0)) );
  758. memcpy(&id1,data+0,sizeof(id1));
  759. id1=ntohl(id1);
  760. //id2=ntohl( *((id_t*)(data+sizeof(id_t))) );
  761. memcpy(&id2,data+sizeof(my_id_t),sizeof(id2));
  762. id2=ntohl(id2);
  763. //id3=ntohl( *((id_t*)(data+sizeof(id_t)*2)) );
  764. memcpy(&id3,data+sizeof(my_id_t)*2,sizeof(id3));
  765. id3=ntohl(id3);
  766. return 0;
  767. }
  768. int hex_to_u32(const string & a,u32_t &output)
  769. {
  770. //string b="0x";
  771. //b+=a;
  772. if(sscanf(a.c_str(),"%x",&output)==1)
  773. {
  774. //printf("%s %x\n",a.c_str(),output);
  775. return 0;
  776. }
  777. mylog(log_error,"<%s> doesnt contain a hex\n",a.c_str());
  778. return -1;
  779. }
  780. int hex_to_u32_with_endian(const string & a,u32_t &output)
  781. {
  782. //string b="0x";
  783. //b+=a;
  784. if(sscanf(a.c_str(),"%x",&output)==1)
  785. {
  786. output=htonl(output);
  787. //printf("%s %x\n",a.c_str(),output);
  788. return 0;
  789. }
  790. mylog(log_error,"<%s> doesnt contain a hex\n",a.c_str());
  791. return -1;
  792. }
  793. bool larger_than_u32(u32_t a,u32_t b)
  794. {
  795. return ((i32_t(a-b)) >0);
  796. /*
  797. u32_t smaller,bigger;
  798. smaller=min(a,b);//smaller in normal sense
  799. bigger=max(a,b);
  800. u32_t distance=min(bigger-smaller,smaller+(0xffffffff-bigger+1));
  801. if(distance==bigger-smaller)
  802. {
  803. if(bigger==a)
  804. {
  805. return 1;
  806. }
  807. else
  808. {
  809. return 0;
  810. }
  811. }
  812. else
  813. {
  814. if(smaller==b)
  815. {
  816. return 0;
  817. }
  818. else
  819. {
  820. return 1;
  821. }
  822. }
  823. */
  824. }
  825. bool larger_than_u16(uint16_t a,uint16_t b)
  826. {
  827. return ((i16_t(a-b)) >0);
  828. /*
  829. uint16_t smaller,bigger;
  830. smaller=min(a,b);//smaller in normal sense
  831. bigger=max(a,b);
  832. uint16_t distance=min(bigger-smaller,smaller+(0xffff-bigger+1));
  833. if(distance==bigger-smaller)
  834. {
  835. if(bigger==a)
  836. {
  837. return 1;
  838. }
  839. else
  840. {
  841. return 0;
  842. }
  843. }
  844. else
  845. {
  846. if(smaller==b)
  847. {
  848. return 0;
  849. }
  850. else
  851. {
  852. return 1;
  853. }
  854. }
  855. */
  856. }
  857. void myexit(int a)
  858. {
  859. if(enable_log_color)
  860. printf("%s\n",RESET);
  861. /*
  862. if(keep_thread_running)
  863. {
  864. if(pthread_cancel(keep_thread))
  865. {
  866. mylog(log_warn,"pthread_cancel failed\n");
  867. }
  868. else
  869. {
  870. mylog(log_info,"pthread_cancel success\n");
  871. }
  872. }
  873. clear_iptables_rule();
  874. */
  875. exit(a);
  876. }
  877. vector<string> string_to_vec(const char * s,const char * sp) {
  878. vector<string> res;
  879. string str=s;
  880. char *p = strtok ((char *)str.c_str(),sp);
  881. while (p != NULL)
  882. {
  883. res.push_back(p);
  884. //printf ("%s\n",p);
  885. p = strtok(NULL, sp);
  886. }
  887. /* for(int i=0;i<(int)res.size();i++)
  888. {
  889. printf("<<%s>>\n",res[i].c_str());
  890. }*/
  891. return res;
  892. }
  893. vector< vector <string> > string_to_vec2(const char * s)
  894. {
  895. vector< vector <string> > res;
  896. vector<string> lines=string_to_vec(s,"\n");
  897. for(int i=0;i<int(lines.size());i++)
  898. {
  899. vector<string> tmp;
  900. tmp=string_to_vec(lines[i].c_str(),"\t ");
  901. res.push_back(tmp);
  902. }
  903. return res;
  904. }
  905. int read_file(const char * file,string &output)
  906. {
  907. const int max_len=3*1024*1024;
  908. // static char buf[max_len+100];
  909. string buf0;
  910. buf0.reserve(max_len+200);
  911. char * buf=(char *)buf0.c_str();
  912. buf[max_len]=0;
  913. //buf[sizeof(buf)-1]=0;
  914. int fd=open(file,O_RDONLY);
  915. if(fd==-1)
  916. {
  917. mylog(log_error,"read_file %s fail\n",file);
  918. return -1;
  919. }
  920. int len=read(fd,buf,max_len);
  921. if(len==max_len)
  922. {
  923. buf[0]=0;
  924. mylog(log_error,"%s too long,buf not large enough\n",file);
  925. return -2;
  926. }
  927. else if(len<0)
  928. {
  929. buf[0]=0;
  930. mylog(log_error,"%s read fail %d\n",file,len);
  931. return -3;
  932. }
  933. else
  934. {
  935. buf[len]=0;
  936. output=buf;
  937. }
  938. return 0;
  939. }
  940. int run_command(string command0,char * &output,int flag) {
  941. mylog(log_fatal,"run_command not supported in this version\n");
  942. myexit(-1);
  943. #if 0
  944. FILE *in;
  945. if((flag&show_log)==0) command0+=" 2>&1 ";
  946. const char * command=command0.c_str();
  947. int level= (flag&show_log)?log_warn:log_debug;
  948. if(flag&show_command)
  949. {
  950. mylog(log_info,"run_command %s\n",command);
  951. }
  952. else
  953. {
  954. mylog(log_debug,"run_command %s\n",command);
  955. }
  956. static __thread char buf[1024*1024+100];
  957. buf[sizeof(buf)-1]=0;
  958. if(!(in = popen(command, "r"))){
  959. mylog(level,"command %s popen failed,errno %s\n",command,strerror(errno));
  960. return -1;
  961. }
  962. int len =fread(buf, 1024*1024, 1, in);
  963. if(len==1024*1024)
  964. {
  965. buf[0]=0;
  966. mylog(level,"too long,buf not larger enough\n");
  967. return -2;
  968. }
  969. else
  970. {
  971. buf[len]=0;
  972. }
  973. int ret;
  974. if(( ret=ferror(in) ))
  975. {
  976. mylog(level,"command %s fread failed,ferror return value %d \n",command,ret);
  977. return -3;
  978. }
  979. //if(output!=0)
  980. output=buf;
  981. ret= pclose(in);
  982. int ret2=WEXITSTATUS(ret);
  983. if(ret!=0||ret2!=0)
  984. {
  985. mylog(level,"commnad %s ,pclose returned %d ,WEXITSTATUS %d,errnor :%s \n",command,ret,ret2,strerror(errno));
  986. return -4;
  987. }
  988. #endif
  989. return 0;
  990. }
  991. /*
  992. int run_command_no_log(string command0,char * &output) {
  993. FILE *in;
  994. command0+=" 2>&1 ";
  995. const char * command=command0.c_str();
  996. mylog(log_debug,"run_command_no_log %s\n",command);
  997. static char buf[1024*1024+100];
  998. buf[sizeof(buf)-1]=0;
  999. if(!(in = popen(command, "r"))){
  1000. mylog(log_debug,"command %s popen failed,errno %s\n",command,strerror(errno));
  1001. return -1;
  1002. }
  1003. int len =fread(buf, 1024*1024, 1, in);
  1004. if(len==1024*1024)
  1005. {
  1006. buf[0]=0;
  1007. mylog(log_debug,"too long,buf not larger enough\n");
  1008. return -2;
  1009. }
  1010. else
  1011. {
  1012. buf[len]=0;
  1013. }
  1014. int ret;
  1015. if(( ret=ferror(in) ))
  1016. {
  1017. mylog(log_debug,"command %s fread failed,ferror return value %d \n",command,ret);
  1018. return -3;
  1019. }
  1020. //if(output!=0)
  1021. output=buf;
  1022. ret= pclose(in);
  1023. int ret2=WEXITSTATUS(ret);
  1024. if(ret!=0||ret2!=0)
  1025. {
  1026. mylog(log_debug,"commnad %s ,pclose returned %d ,WEXITSTATUS %d,errnor :%s \n",command,ret,ret2,strerror(errno));
  1027. return -4;
  1028. }
  1029. return 0;
  1030. }*/
  1031. // Remove preceding and trailing characters
  1032. string trim(const string& str, char c) {
  1033. size_t first = str.find_first_not_of(c);
  1034. if(string::npos==first)
  1035. {
  1036. return "";
  1037. }
  1038. size_t last = str.find_last_not_of(c);
  1039. return str.substr(first,(last-first+1));
  1040. }
  1041. vector<string> parse_conf_line(const string& s0)
  1042. {
  1043. string s=s0;
  1044. s.reserve(s.length()+200);
  1045. char *buf=(char *)s.c_str();
  1046. //char buf[s.length()+200];
  1047. char *p=buf;
  1048. int i=int(s.length())-1;
  1049. int j;
  1050. vector<string>res;
  1051. strcpy(buf,(char *)s.c_str());
  1052. while(i>=0)
  1053. {
  1054. if(buf[i]==' ' || buf[i]== '\t')
  1055. buf[i]=0;
  1056. else break;
  1057. i--;
  1058. }
  1059. while(*p!=0)
  1060. {
  1061. if(*p==' ' || *p== '\t')
  1062. {
  1063. p++;
  1064. }
  1065. else break;
  1066. }
  1067. int new_len=strlen(p);
  1068. if(new_len==0)return res;
  1069. if(p[0]=='#') return res;
  1070. if(p[0]!='-')
  1071. {
  1072. mylog(log_fatal,"line :<%s> not begin with '-' ",s.c_str());
  1073. myexit(-1);
  1074. }
  1075. for(i=0;i<new_len;i++)
  1076. {
  1077. if(p[i]==' '||p[i]=='\t')
  1078. {
  1079. break;
  1080. }
  1081. }
  1082. if(i==new_len)
  1083. {
  1084. res.push_back(p);
  1085. return res;
  1086. }
  1087. j=i;
  1088. while(p[j]==' '||p[j]=='\t')
  1089. j++;
  1090. p[i]=0;
  1091. res.push_back(p);
  1092. res.push_back(p+j);
  1093. return res;
  1094. }
  1095. int create_fifo(char * file)
  1096. {
  1097. #if !defined(__MINGW32__)
  1098. if(mkfifo (file, 0666)!=0)
  1099. {
  1100. if(errno==EEXIST)
  1101. {
  1102. mylog(log_warn,"warning fifo file %s exist\n",file);
  1103. }
  1104. else
  1105. {
  1106. mylog(log_fatal,"create fifo file %s failed\n",file);
  1107. myexit(-1);
  1108. }
  1109. }
  1110. int fifo_fd=open (file, O_RDWR);
  1111. if(fifo_fd<0)
  1112. {
  1113. mylog(log_fatal,"create fifo file %s failed\n",file);
  1114. myexit(-1);
  1115. }
  1116. struct stat st;
  1117. if (fstat(fifo_fd, &st)!=0)
  1118. {
  1119. mylog(log_fatal,"fstat failed for fifo file %s\n",file);
  1120. myexit(-1);
  1121. }
  1122. if(!S_ISFIFO(st.st_mode))
  1123. {
  1124. mylog(log_fatal,"%s is not a fifo\n",file);
  1125. myexit(-1);
  1126. }
  1127. setnonblocking(fifo_fd);
  1128. return fifo_fd;
  1129. #else
  1130. mylog(log_fatal,"--fifo not supported in this version\n");
  1131. myexit(-1);
  1132. return 0;
  1133. #endif
  1134. }
  1135. void ip_port_t::from_u64(u64_t u64)
  1136. {
  1137. ip=get_u64_h(u64);
  1138. port=get_u64_l(u64);
  1139. }
  1140. u64_t ip_port_t::to_u64()
  1141. {
  1142. return pack_u64(ip,port);
  1143. }
  1144. char * ip_port_t::to_s()
  1145. {
  1146. static char res[40];
  1147. sprintf(res,"%s:%d",my_ntoa(ip),port);
  1148. return res;
  1149. }
  1150. void print_binary_chars(const char * a,int len)
  1151. {
  1152. for(int i=0;i<len;i++)
  1153. {
  1154. unsigned char b=a[i];
  1155. log_bare(log_debug,"<%02x>",(int)b);
  1156. }
  1157. log_bare(log_debug,"\n");
  1158. }
  1159. u32_t djb2(unsigned char *str,int len)
  1160. {
  1161. u32_t hash = 5381;
  1162. int c;
  1163. int i=0;
  1164. while(c = *str++,i++!=len)
  1165. {
  1166. hash = ((hash << 5) + hash)^c; /* (hash * 33) ^ c */
  1167. }
  1168. hash=htonl(hash);
  1169. return hash;
  1170. }
  1171. u32_t sdbm(unsigned char *str,int len)
  1172. {
  1173. u32_t hash = 0;
  1174. int c;
  1175. int i=0;
  1176. while(c = *str++,i++!=len)
  1177. {
  1178. hash = c + (hash << 6) + (hash << 16) - hash;
  1179. }
  1180. //hash=htonl(hash);
  1181. return hash;
  1182. }