/* * comm.cpp * * Created on: Jul 29, 2017 * Author: wangyu */ #include "common.h" #include "log.h" int about_to_exit=0; raw_mode_t raw_mode=mode_faketcp; unordered_map raw_mode_tostring = {{mode_faketcp, "faketcp"}, {mode_udp, "udp"}, {mode_icmp, "icmp"}}; int max_pending_packet=0; static int random_number_fd=-1; char iptables_rule[200]=""; //int is_client = 0, is_server = 0; program_mode_t program_mode=unset_mode;//0 unset; 1client 2server u64_t get_current_time() { timespec tmp_time; clock_gettime(CLOCK_MONOTONIC, &tmp_time); return tmp_time.tv_sec*1000+tmp_time.tv_nsec/(1000*1000l); } u64_t get_current_time_us() { timespec tmp_time; clock_gettime(CLOCK_MONOTONIC, &tmp_time); return (uint64_t(tmp_time.tv_sec))*1000llu*1000llu+ (uint64_t(tmp_time.tv_nsec))/1000llu; } u64_t pack_u64(u32_t a,u32_t b) { u64_t ret=a; ret<<=32u; ret+=b; return ret; } u32_t get_u64_h(u64_t a) { return a>>32u; } u32_t get_u64_l(u64_t a) { return (a<<32u)>>32u; } char * my_ntoa(u32_t ip) { in_addr a; a.s_addr=ip; return inet_ntoa(a); } int add_iptables_rule(char * s) { strcpy(iptables_rule,s); char buf[300]="iptables -I "; strcat(buf,s); if(system(buf)==0) { mylog(log_warn,"auto added iptables rule by: %s\n",buf); } else { mylog(log_fatal,"auto added iptables failed by: %s\n",buf); myexit(-1); } return 0; } int clear_iptables_rule() { if(iptables_rule[0]!=0) { char buf[300]="iptables -D "; strcat(buf,iptables_rule); if(system(buf)==0) { mylog(log_warn,"iptables rule cleared by: %s \n",buf); } else { mylog(log_error,"clear iptables failed by: %s\n",buf); } } return 0; } void init_random_number_fd() { random_number_fd=open("/dev/urandom",O_RDONLY); if(random_number_fd==-1) { mylog(log_fatal,"error open /dev/urandom\n"); myexit(-1); } setnonblocking(random_number_fd); } u64_t get_true_random_number_64() { u64_t ret; int size=read(random_number_fd,&ret,sizeof(ret)); if(size!=sizeof(ret)) { mylog(log_fatal,"get random number failed %d\n",size); myexit(-1); } return ret; } u32_t get_true_random_number() { u32_t ret; int size=read(random_number_fd,&ret,sizeof(ret)); if(size!=sizeof(ret)) { mylog(log_fatal,"get random number failed %d\n",size); myexit(-1); } return ret; } u32_t get_true_random_number_nz() //nz for non-zero { u32_t ret=0; while(ret==0) { ret=get_true_random_number(); } return ret; } u64_t ntoh64(u64_t a) { if(__BYTE_ORDER == __LITTLE_ENDIAN) { return __bswap_64( a); } else return a; } u64_t hton64(u64_t a) { if(__BYTE_ORDER == __LITTLE_ENDIAN) { return __bswap_64( a); } else return a; } void setnonblocking(int sock) { int opts; opts = fcntl(sock, F_GETFL); if (opts < 0) { mylog(log_fatal,"fcntl(sock,GETFL)\n"); //perror("fcntl(sock,GETFL)"); myexit(1); } opts = opts | O_NONBLOCK; if (fcntl(sock, F_SETFL, opts) < 0) { mylog(log_fatal,"fcntl(sock,SETFL,opts)\n"); //perror("fcntl(sock,SETFL,opts)"); myexit(1); } } /* Generic checksum calculation function */ unsigned short csum(const unsigned short *ptr,int nbytes) { register long sum; unsigned short oddbyte; register short answer; sum=0; while(nbytes>1) { sum+=*ptr++; nbytes-=2; } if(nbytes==1) { oddbyte=0; *((u_char*)&oddbyte)=*(u_char*)ptr; sum+=oddbyte; } sum = (sum>>16)+(sum & 0xffff); sum = sum + (sum>>16); answer=(short)~sum; return(answer); } int set_buf_size(int fd,int socket_buf_size,int force_socket_buf) { if(force_socket_buf) { if(setsockopt(fd, SOL_SOCKET, SO_SNDBUFFORCE, &socket_buf_size, sizeof(socket_buf_size))<0) { mylog(log_fatal,"SO_SNDBUFFORCE fail socket_buf_size=%d errno=%s\n",socket_buf_size,strerror(errno)); myexit(1); } if(setsockopt(fd, SOL_SOCKET, SO_RCVBUFFORCE, &socket_buf_size, sizeof(socket_buf_size))<0) { mylog(log_fatal,"SO_RCVBUFFORCE fail socket_buf_size=%d errno=%s\n",socket_buf_size,strerror(errno)); myexit(1); } } else { if(setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &socket_buf_size, sizeof(socket_buf_size))<0) { mylog(log_fatal,"SO_SNDBUF fail socket_buf_size=%d errno=%s\n",socket_buf_size,strerror(errno)); myexit(1); } if(setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &socket_buf_size, sizeof(socket_buf_size))<0) { mylog(log_fatal,"SO_RCVBUF fail socket_buf_size=%d errno=%s\n",socket_buf_size,strerror(errno)); myexit(1); } } return 0; } void myexit(int a) { if(enable_log_color) printf("%s\n",RESET); clear_iptables_rule(); exit(a); } void signal_handler(int sig) { about_to_exit=1; // myexit(0); } int numbers_to_char(id_t id1,id_t id2,id_t id3,char * &data,int &len) { static char buf[buf_len]; data=buf; id_t tmp=htonl(id1); memcpy(buf,&tmp,sizeof(tmp)); tmp=htonl(id2); memcpy(buf+sizeof(tmp),&tmp,sizeof(tmp)); tmp=htonl(id3); memcpy(buf+sizeof(tmp)*2,&tmp,sizeof(tmp)); len=sizeof(id_t)*3; return 0; } int char_to_numbers(const char * data,int len,id_t &id1,id_t &id2,id_t &id3) { if(lenb) { mylog(log_fatal,"min >max?? %d %d\n",a ,b); myexit(1); } if(a==b)return a; else return a+get_true_random_number()%(b+1-a); } int set_timer_ms(int epollfd,int &timer_fd,u32_t timer_interval) { int ret; epoll_event ev; itimerspec its; memset(&its,0,sizeof(its)); if((timer_fd=timerfd_create(CLOCK_MONOTONIC,TFD_NONBLOCK)) < 0) { mylog(log_fatal,"timer_fd create error\n"); myexit(1); } its.it_interval.tv_sec=(timer_interval/1000); its.it_interval.tv_nsec=(timer_interval%1000)*1000ll*1000ll; its.it_value.tv_nsec=1; //imidiately timerfd_settime(timer_fd,0,&its,0); ev.events = EPOLLIN; ev.data.fd = timer_fd; ret=epoll_ctl(epollfd, EPOLL_CTL_ADD, timer_fd, &ev); if (ret < 0) { mylog(log_fatal,"epoll_ctl return %d\n", ret); myexit(-1); } return 0; } /* int create_new_udp(int &new_udp_fd,int remote_address_uint32,int remote_port) { struct sockaddr_in remote_addr_in; socklen_t slen = sizeof(sockaddr_in); memset(&remote_addr_in, 0, sizeof(remote_addr_in)); remote_addr_in.sin_family = AF_INET; remote_addr_in.sin_port = htons(remote_port); remote_addr_in.sin_addr.s_addr = remote_address_uint32; new_udp_fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); if (new_udp_fd < 0) { mylog(log_warn, "create udp_fd error\n"); return -1; } setnonblocking(new_udp_fd); set_buf_size(new_udp_fd); mylog(log_debug, "created new udp_fd %d\n", new_udp_fd); int ret = connect(new_udp_fd, (struct sockaddr *) &remote_addr_in, slen); if (ret != 0) { mylog(log_warn, "udp fd connect fail %d %s\n",ret,strerror(errno)); close(new_udp_fd); return -1; } return 0; }*/