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; if (clt_settings.session_timeout < 0) { clt_settings.session_timeout = DEFAULT_SESSION_TIMEOUT; } tc_log_info(LOG_NOTICE, 0, "session timeout:%d", clt_settings.session_timeout); if (clt_settings.session_keepalive_timeout <= 0) { clt_settings.session_keepalive_timeout = clt_settings.session_timeout + SESS_KEEPLIVE_ADD; } tc_log_info(LOG_NOTICE, 0, "keepalive timeout:%d", clt_settings.session_keepalive_timeout); 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(); } /* 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(isOfflineMode()) { 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; } } #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_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; } #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() { #if (INTERCEPT_ADVANCED && TCPCOPY_PCAP) int len; #endif #if (!INTERCEPT_ADVANCED) /* retrieve ip address */ if (srv_settings.raw_ip_list != NULL) { tc_log_info(LOG_NOTICE, 0, "-x parameter:%s", srv_settings.raw_ip_list); retrieve_ip_addr(); } #endif #if (INTERCEPT_ADVANCED) if (srv_settings.raw_targets != NULL) { tc_log_info(LOG_NOTICE, 0, "-o parameter:%s", srv_settings.raw_targets); retrieve_target_addresses(srv_settings.raw_targets, &(srv_settings.targets)); } else { #if (!TCPCOPY_PCAP) tc_log_info(LOG_WARN, 0, "no raw targets for advanced mode"); return -1; #else tc_log_info(LOG_NOTICE, 0, "no raw targets for advanced mode"); #endif } #if (TCPCOPY_PCAP) if (srv_settings.raw_device != NULL) { tc_log_info(LOG_NOTICE, 0, "device:%s", srv_settings.raw_device); if (strcmp(srv_settings.raw_device, DEFAULT_DEVICE) == 0) { srv_settings.raw_device = NULL; } else { retrieve_devices(srv_settings.raw_device, &(srv_settings.devices)); } } if (srv_settings.user_filter != NULL) { tc_log_info(LOG_NOTICE, 0, "user filter:%s", srv_settings.user_filter); len = strlen(srv_settings.user_filter); if (len >= MAX_FILTER_LENGH) { tc_log_info(LOG_ERR, 0, "user filter is too long"); return -1; } memcpy(srv_settings.filter, srv_settings.user_filter, len); } else { extract_filter(); } #endif #endif #if (INTERCEPT_NFQUEUE) if (srv_settings.max_queue_len <= 1024) { srv_settings.max_queue_len = -1; } #endif /* daemonize */ if (srv_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() == -1) { fprintf(stderr, "failed to daemon() 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(100) == 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() { 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; }