static int set_details() { #if (!TCPCOPY_PCAP) int rand_port; #else int len, rand_port; #endif unsigned int seed; struct timeval tp; /* generate a random port number for avoiding port conflicts */ gettimeofday(&tp, NULL); seed = tp.tv_usec; rand_port = (int) ((rand_r(&seed)/(RAND_MAX + 1.0))*512); clt_settings.rand_port_shifted = rand_port; /* set the ip port pair mapping according to settings */ if (retrieve_target_addresses(clt_settings.raw_transfer, &clt_settings.transfer) == -1) { return -1; } if (clt_settings.percentage < 0 || clt_settings.percentage > 99) { clt_settings.percentage = 0; } if (clt_settings.par_connections <= 0) { clt_settings.par_connections = 1; } else if (clt_settings.par_connections > MAX_CONNECTION_NUM) { clt_settings.par_connections = MAX_CONNECTION_NUM; } tc_log_info(LOG_NOTICE, 0, "parallel connections per target:%d", clt_settings.par_connections); #if (TCPCOPY_OFFLINE) if (clt_settings.pcap_file == NULL) { tc_log_info(LOG_ERR, 0, "it must have -i argument for offline"); fprintf(stderr, "no -i argument\n"); return -1; } if (clt_settings.accelerated_times < 1) { clt_settings.accelerated_times = 1; } tc_log_info(LOG_NOTICE, 0, "accelerated %d times,interval:%llu ms", clt_settings.accelerated_times, clt_settings.interval); if (clt_settings.interval > 0) { clt_settings.interval = clt_settings.interval * 1000; } #endif #if (TCPCOPY_PCAP_SEND) if (clt_settings.output_if_name != NULL) { tc_log_info(LOG_NOTICE, 0, "output device:%s", clt_settings.output_if_name); } else { tc_log_info(LOG_ERR, 0, "output device is null"); return -1; } #endif #if (TCPCOPY_PCAP) if (clt_settings.raw_device != NULL) { tc_log_info(LOG_NOTICE, 0, "device:%s", clt_settings.raw_device); if (strcmp(clt_settings.raw_device, DEFAULT_DEVICE) == 0) { clt_settings.raw_device = NULL; } else { retrieve_devices(clt_settings.raw_device, &(clt_settings.devices)); } } if (clt_settings.user_filter != NULL) { tc_log_info(LOG_NOTICE, 0, "user filter:%s", clt_settings.user_filter); len = strlen(clt_settings.user_filter); if (len >= MAX_FILTER_LENGH) { tc_log_info(LOG_ERR, 0, "user filter is too long"); return -1; } memcpy(clt_settings.filter, clt_settings.user_filter, len); } else { extract_filter(); } #endif #if (TCPCOPY_MYSQL_ADVANCED) if (clt_settings.user_pwd != NULL) { tc_log_info(LOG_NOTICE, 0, "-u argument:%s",clt_settings.user_pwd); if (retrieve_mysql_user_pwd_info(clt_settings.user_pwd) == -1) { tc_log_info(LOG_ERR, 0, "wrong -u argument"); return -1; } } else { tc_log_info(LOG_ERR, 0, "it must have -u argument"); fprintf(stderr, "no -u argument\n"); return -1; } #endif #if (TCPCOPY_DR) /* retrieve real server ip addresses */ if (clt_settings.raw_rs_ip_list != NULL) { tc_log_info(LOG_NOTICE, 0, "s parameter:%s", clt_settings.raw_rs_ip_list); retrieve_real_servers(); } else { tc_log_info(LOG_WARN, 0, "no real server ip addresses"); return -1; } #endif /* daemonize */ if (clt_settings.do_daemonize) { if (sigignore(SIGHUP) == -1) { tc_log_info(LOG_ERR, errno, "Failed to ignore SIGHUP"); } if (daemonize() == -1) { fprintf(stderr, "failed to daemonize() in order to daemonize\n"); return -1; } } return 0; }
static int set_details() { int rand_port; unsigned int seed; struct timeval tp; /* generate a random port number for avoiding port conflicts */ gettimeofday(&tp, NULL); seed = tp.tv_usec; rand_port = (int) ((rand_r(&seed)/(RAND_MAX + 1.0))*512); clt_settings.rand_port_shifted = rand_port; /* set the ip port pair mapping according to settings */ if (retrieve_target_addresses(clt_settings.raw_transfer, &clt_settings.transfer) == -1) { return -1; } if (clt_settings.percentage < 0 && clt_settings.percentage >99) { clt_settings.percentage = 0; } #if (TCPCOPY_OFFLINE) if (clt_settings.pcap_file == NULL) { tc_log_info(LOG_ERR, 0, "it must have -i argument for offline"); fprintf(stderr, "no -i argument\n"); return -1; } #endif #if (TCPCOPY_PCAP) if (clt_settings.raw_device != NULL) { tc_log_info(LOG_NOTICE, 0, "device:%s", clt_settings.raw_device); if (strcmp(clt_settings.raw_device, DEFAULT_DEVICE) == 0) { clt_settings.raw_device = NULL; } else { retrieve_devices(); } } #endif #if (TCPCOPY_MYSQL_ADVANCED) if (clt_settings.user_pwd != NULL) { tc_log_info(LOG_NOTICE, 0, "-u argument:%s",clt_settings.user_pwd); if (retrieve_mysql_user_pwd_info(clt_settings.user_pwd) == -1) { tc_log_info(LOG_ERR, 0, "wrong -u argument"); return -1; } } else { tc_log_info(LOG_ERR, 0, "it must have -u argument"); fprintf(stderr, "no -u argument\n"); return -1; } #endif #if (TCPCOPY_DR) /* retrieve real server ip addresses */ if (clt_settings.raw_rs_ip_list != NULL) { tc_log_info(LOG_NOTICE, 0, "s parameter:%s", clt_settings.raw_rs_ip_list); retrieve_real_servers(); } else { tc_log_info(LOG_WARN, 0, "no real server ip addresses"); return -1; } #endif /* daemonize */ if (clt_settings.do_daemonize) { if (sigignore(SIGHUP) == -1) { tc_log_info(LOG_ERR, errno, "Failed to ignore SIGHUP"); } if (daemonize(clt_settings.pid_file) == -1) { fprintf(stderr, "failed to daemonize() in order to daemonize\n"); return -1; } } #if (TCPCOPY_OFFLINE) if (tc_time_set_timer(TIMER_INTERVAL) == TC_ERROR) { #else if (tc_time_set_timer(10) == TC_ERROR) { #endif tc_log_info(LOG_ERR, 0, "set timer error"); return -1; } return 0; } /* set default values for TCPCopy client */ static void settings_init() { /* init values */ clt_settings.mtu = DEFAULT_MTU; clt_settings.mss = DEFAULT_MSS; clt_settings.max_rss = MAX_MEMORY_SIZE; clt_settings.srv_port = SERVER_PORT; clt_settings.percentage = 0; clt_settings.session_timeout = DEFAULT_SESSION_TIMEOUT; tc_raw_socket_out = TC_INVALID_SOCKET; set_signal_handler(); }
static int set_details() { int mtu_list[] = {576, 1492, 1500, 0}; int i, len, index, offset, rand_port; unsigned int seed; unsigned char value; struct timeval tp; for (i = tc_pagesize; i >>= 1; tc_pagesize_shift++) { /* void */ } /* generate a random port number for avoiding port conflicts */ gettimeofday(&tp, NULL);\ seed = tp.tv_usec; #if (MINGW32) srand(seed); rand_port = (int) ((rand() / (RAND_MAX + 1.0)) * 512); #else rand_port = (int) ((rand_r(&seed) / (RAND_MAX + 1.0)) * 512); #endif clt_settings.rand_port_shifted = rand_port; if (clt_settings.sess_timeout < 0) { clt_settings.sess_timeout = DEFAULT_SESS_TIMEOUT; } tc_log_info(LOG_NOTICE, 0, "session timeout:%d", clt_settings.sess_timeout); if (clt_settings.sess_keepalive_timeout <= 0) { clt_settings.sess_keepalive_timeout = clt_settings.sess_timeout + SESS_KEEPLIVE_ADD; } tc_log_info(LOG_NOTICE, 0, "keepalive timeout:%d", clt_settings.sess_keepalive_timeout); #if (!TC_UDP) if (clt_settings.s_pool_size == 0) { clt_settings.s_pool_size = TC_DEFAULT_UPOOL_SIZE; } tc_log_info(LOG_NOTICE, 0, "min sess pool size:%d", TC_MIN_SESS_POOL_SIZE); tc_log_info(LOG_NOTICE, 0, "sess pool size:%d", clt_settings.s_pool_size); if (clt_settings.s_pool_size < TC_MIN_SESS_POOL_SIZE) { tc_log_info(LOG_NOTICE, 0, "sess pool size is too small"); } #endif if (clt_settings.replica_num > 1) { tc_log_info(LOG_NOTICE, 0, "repl num:%d", clt_settings.replica_num); } if (clt_settings.gradully) { tc_log_info(LOG_NOTICE, 0, "gradully replay"); } /* set the ip port pair mapping according to settings */ if (retr_target_addrs(clt_settings.raw_tf, &clt_settings.transfer) == -1) { return -1; } if (clt_settings.raw_clt_tf_ip != NULL) { /* print out raw_clt_tf_ip */ tc_log_info(LOG_NOTICE, 0, "raw_clt_tf_ip:%s", clt_settings.raw_clt_tf_ip); retrieve_clt_tf_ips(); } if (clt_settings.percentage > 99) { clt_settings.percentage = 0; } #if (!TC_UDP) if (sizeof(tc_sess_t) > TC_UPOOL_MAXV) { tc_log_info(LOG_NOTICE, 0, "TC_UPOOL_MAXV is too small"); } #endif if (clt_settings.par_conns <= 0) { clt_settings.par_conns = 1; } else if (clt_settings.par_conns > MAX_CONN_NUM) { clt_settings.par_conns = MAX_CONN_NUM; } tc_log_info(LOG_NOTICE, 0, "parallel connections per target:%d", clt_settings.par_conns); len = sizeof(mtu_list) / sizeof(int) - 1; for (i = 0; i < len; i++) { if (mtu_list[i] == clt_settings.mtu) { break; } } if (i == len) { mtu_list[len++] = clt_settings.mtu; } for (i = 0; i < len; i++) { index = mtu_list[i] >> 3; offset = mtu_list[i] - (index << 3); value = clt_settings.candidate_mtu[index]; value = value | (1 << offset); clt_settings.candidate_mtu[index] = value; } #if (TC_OFFLINE) if (clt_settings.pcap_file == NULL) { tc_log_info(LOG_ERR, 0, "it must have -i argument for offline"); fprintf(stderr, "no -i argument\n"); return -1; } if (clt_settings.accelerated_times < 1) { clt_settings.accelerated_times = 1; } tc_log_info(LOG_NOTICE, 0, "accelerated %d times,interval:%llu ms", clt_settings.accelerated_times, clt_settings.interval); if (clt_settings.interval > 0) { clt_settings.interval = clt_settings.interval * 1000; } #endif #if (TC_PCAP_SND) if (clt_settings.output_if_name != NULL) { tc_log_info(LOG_NOTICE, 0, "output device:%s", clt_settings.output_if_name); } else { tc_log_info(LOG_ERR, 0, "no -o argument"); fprintf(stderr, "no -o argument\n"); return -1; } #endif #if (TC_PCAP) if (clt_settings.raw_device != NULL) { tc_log_info(LOG_NOTICE, 0, "device:%s", clt_settings.raw_device); if (strcmp(clt_settings.raw_device, DEFAULT_DEVICE) == 0) { clt_settings.raw_device = NULL; } else { retrieve_devices(clt_settings.raw_device, &(clt_settings.devices)); } } if (clt_settings.user_filter != NULL) { tc_log_info(LOG_NOTICE, 0, "user filter:%s", clt_settings.user_filter); len = strlen(clt_settings.user_filter); if (len >= MAX_FILTER_LENGH) { tc_log_info(LOG_ERR, 0, "user filter is too long"); return -1; } memcpy(clt_settings.filter, clt_settings.user_filter, len); } else { extract_filter(); } if (clt_settings.snaplen > PCAP_RCV_BUF_SIZE) { clt_settings.snaplen = PCAP_RCV_BUF_SIZE; } #endif /* retrieve real server ip addresses */ if (clt_settings.raw_rs_list != NULL) { tc_log_info(LOG_NOTICE, 0, "s parameter:%s", clt_settings.raw_rs_list); retrieve_real_servers(); } else { tc_log_info(LOG_WARN, 0, "no -s parameter(intercept addresses)"); fprintf(stderr, "tcpcopy needs -s paramter(intercept addresses)\n"); return -1; } #if (TC_PLUGIN) /* support only one additional module*/ clt_settings.plugin = tc_modules[0]; #endif if (read_conf_file() != TC_OK) { return -1; } if (clt_settings.do_daemonize) { /* ignore if (sigignore(SIGHUP) == -1) { tc_log_info(LOG_ERR, errno, "Failed to ignore SIGHUP"); } if (daemonize() == -1) { fprintf(stderr, "failed to daemonize() in order to daemonize\n"); return -1; } */ } return 0; }
static int set_details() { clt_settings.session_keepalive_timeout = clt_settings.session_timeout; tc_log_info(LOG_NOTICE, 0, "keepalive timeout:%d", clt_settings.session_keepalive_timeout); if (clt_settings.users <= 0) { tc_log_info(LOG_ERR, 0, "please set the -u parameter"); return -1; } if (clt_settings.raw_clt_ips == NULL) { tc_log_info(LOG_ERR, 0, "please set the -c parameter"); return -1; } retrieve_client_ips(); /* set the ip port pair mapping according to settings */ if (retrieve_target_addresses(clt_settings.raw_transfer, &clt_settings.transfer) == -1) { return -1; } if (clt_settings.par_connections <= 0) { clt_settings.par_connections = 1; } else if (clt_settings.par_connections > MAX_CONNECTION_NUM) { clt_settings.par_connections = MAX_CONNECTION_NUM; } tc_log_info(LOG_NOTICE, 0, "parallel connections per target:%d", clt_settings.par_connections); if (clt_settings.raw_pcap_files == NULL) { tc_log_info(LOG_ERR, 0, "it must have -f argument for gryphon"); fprintf(stderr, "no -f argument\n"); return -1; } retrieve_pcap_files(); if (clt_settings.throughput_factor < 1) { clt_settings.throughput_factor = 1; } tc_log_info(LOG_NOTICE, 0, "throughput factor: %d,interval:%llu ms", clt_settings.throughput_factor, clt_settings.interval); if (clt_settings.interval > 0) { clt_settings.interval = clt_settings.interval * 1000; } #if (GRYPHON_PCAP_SEND) if (clt_settings.output_if_name != NULL) { tc_log_info(LOG_NOTICE, 0, "output device:%s", clt_settings.output_if_name); } else { tc_log_info(LOG_ERR, 0, "output device is null"); return -1; } #endif /* retrieve real server ip addresses */ if (clt_settings.raw_rs_list != NULL) { tc_log_info(LOG_NOTICE, 0, "s parameter:%s", clt_settings.raw_rs_list); retrieve_real_servers(); } else { tc_log_info(LOG_WARN, 0, "no real server ip addresses"); return -1; } /* daemonize */ if (clt_settings.do_daemonize) { if (sigignore(SIGHUP) == -1) { tc_log_info(LOG_ERR, errno, "Failed to ignore SIGHUP"); } if (daemonize() == -1) { fprintf(stderr, "failed to daemonize() in order to daemonize\n"); return -1; } } return 0; }