/* initiate TCPCopy client */ int tcp_copy_init(tc_event_loop_t *event_loop) { int i, fd; uint32_t target_ip; ip_port_pair_mapping_t *pair, **mappings; /* register some timer */ tc_event_timer_add(event_loop, 60000, check_resource_usage); tc_event_timer_add(event_loop, 5000, tc_interval_dispose); /* init session table */ init_for_sessions(); address_init(); /* add connections to the tested server for exchanging info */ mappings = clt_settings.transfer.mappings; for (i = 0; i < clt_settings.transfer.num; i++) { pair = mappings[i]; target_ip = pair->target_ip; fd = tc_message_init(event_loop, target_ip, clt_settings.srv_port); if (fd == TC_INVALID_SOCKET) { return TC_ERROR; } address_add_sock(pair->online_ip, pair->online_port, fd); tc_log_info(LOG_NOTICE, 0, "add a tunnel for exchanging info:%u:%u", ntohl(target_ip), clt_settings.srv_port); } /* init packets for processing */ #if (TCPCOPY_OFFLINE) if (tc_offline_init(event_loop, clt_settings.pcap_file) == TC_ERROR) { return TC_ERROR; } #else if (tc_packets_init(event_loop) == TC_ERROR) { return TC_ERROR; } #endif return TC_OK; }
/* initiate TCPCopy client */ int tcp_copy_init(tc_event_loop_t *event_loop) { /* register some timer */ tc_event_timer_add(event_loop, 60000, check_resource_usage); tc_event_timer_add(event_loop, 5000, tc_interval_dispose); #if (TCPCOPY_DR) if (clt_settings.lonely) { tc_event_timer_add(event_loop, 10000, restore_work); } #endif /* init session table */ init_for_sessions(); #if (!TCPCOPY_DR) address_init(); #endif if (connect_to_server(event_loop) == TC_ERROR) { return TC_ERROR; } /* init packets for processing */ #if (TCPCOPY_OFFLINE) if (tc_offline_init(event_loop, clt_settings.pcap_file) == TC_ERROR) { return TC_ERROR; } #else if (tc_packets_init(event_loop) == TC_ERROR) { return TC_ERROR; } #endif return TC_OK; }
/* Initiate tcpcopy client */ int tcp_copy_init(cpy_event_loop_t *event_loop) { int i; #if (TCPCOPY_OFFLINE) char *pcap_file; char ebuf[PCAP_ERRBUF_SIZE]; #endif uint16_t online_port, target_port; uint32_t target_ip; ip_port_pair_mapping_t *pair; ip_port_pair_mapping_t **mappings; cpy_event_t *raw_socket_event; /* keep it temporarily */ select_server_set_callback(dispose_event); /* Init session table*/ init_for_sessions(); localhost = inet_addr("127.0.0.1"); /* Init output raw socket info */ send_init(); /* Add connections to the tested server for exchanging info */ mappings = clt_settings.transfer.mappings; for(i = 0; i < clt_settings.transfer.num; i++){ pair = mappings[i]; online_port = pair->online_port; target_ip = pair->target_ip; target_port = pair->target_port; if(address_add_msg_conn(event_loop, online_port, target_ip, clt_settings.srv_port)) { return FAILURE; } log_info(LOG_NOTICE, "add a tunnel for exchanging info:%u", ntohs(target_port)); } #if (!TCPCOPY_OFFLINE) /* Init input raw socket info */ raw_sock = init_input_raw_socket(); #endif if(raw_sock != -1){ /* Add the input raw socket to select */ raw_socket_event = cpy_event_create(raw_sock, dispose_event_wrapper, NULL); if (raw_socket_event == NULL) { return FAILURE; } if (cpy_event_add(event_loop, raw_socket_event, CPY_EVENT_READ) == CPY_EVENT_ERROR) { log_info(LOG_ERR, "add raw socket(%d) to event loop failed.", raw_socket_event->fd); return FAILURE; } /* Init output raw socket info */ send_init(); /* Add connections to the tested server for exchanging info */ mappings = clt_settings.transfer.mappings; for(i = 0; i < clt_settings.transfer.num; i++){ pair = mappings[i]; online_port = pair->online_port; target_ip = pair->target_ip; target_port = pair->target_port; if (address_add_msg_conn(event_loop, online_port, target_ip, clt_settings.srv_port) == -1) { return FAILURE; } log_info(LOG_NOTICE, "add a tunnel for exchanging info:%u", ntohs(target_port)); } return SUCCESS; }else{ #if (TCPCOPY_OFFLINE) select_offline_set_callback(send_packets_from_pcap); pcap_file = clt_settings.pcap_file; if(pcap_file != NULL){ if ((pcap = pcap_open_offline(pcap_file, ebuf)) == NULL){ log_info(LOG_ERR, "open %s" , ebuf); fprintf(stderr, "open %s\n", ebuf); return FAILURE; }else{ gettimeofday(&base_time, NULL); log_info(LOG_NOTICE, "open pcap success:%s", pcap_file); log_info(LOG_NOTICE, "send the first packets here"); send_packets_from_pcap(1); } }else{ return FAILURE; } #else return FAILURE; #endif } return SUCCESS; }
/* initiate TCPCopy client */ int tcp_copy_init(tc_event_loop_t *event_loop) { int i, fd; #if (TCPCOPY_PCAP) int j, filter_port_num = 0; char *pt; uint16_t filter_port[MAX_FILTER_PORTS]; #endif uint32_t target_ip; #if (!TCPCOPY_DR) ip_port_pair_mapping_t *pair, **mappings; #endif /* register some timer */ tc_event_timer_add(event_loop, 60000, check_resource_usage); tc_event_timer_add(event_loop, 5000, tc_interval_dispose); /* init session table */ init_for_sessions(); #if (TCPCOPY_PCAP) memset((void *) filter_port, 0, MAX_FILTER_PORTS << 1); #endif #if (TCPCOPY_DR) /* * add connections to the real servers for sending router info * and receiving response packet */ for (i = 0; i < clt_settings.real_servers.num; i++) { target_ip = clt_settings.real_servers.ips[i]; fd = tc_message_init(event_loop, target_ip, clt_settings.srv_port); if (fd == TC_INVALID_SOCKET) { return TC_ERROR; } clt_settings.real_servers.active_num++; clt_settings.real_servers.active[i] = 1; clt_settings.real_servers.fds[i] = fd; tc_log_info(LOG_NOTICE, 0, "add a tunnel for exchanging info:%u:%u", ntohl(target_ip), clt_settings.srv_port); } #else address_init(); /* add connections to the tested server for exchanging info */ mappings = clt_settings.transfer.mappings; for (i = 0; i < clt_settings.transfer.num; i++) { pair = mappings[i]; target_ip = pair->target_ip; #if (TCPCOPY_PCAP) for (j = 0; j < MAX_FILTER_PORTS; j++) { if (filter_port[j] == 0) { filter_port[j] = pair->online_port; filter_port_num++; break; } else if (filter_port[j] == pair->online_port) { break; } } #endif fd = tc_message_init(event_loop, target_ip, clt_settings.srv_port); if (fd == TC_INVALID_SOCKET) { return TC_ERROR; } address_add_sock(pair->online_ip, pair->online_port, fd); tc_log_info(LOG_NOTICE, 0, "add a tunnel for exchanging info:%u:%u", ntohl(target_ip), clt_settings.srv_port); } #endif #if (TCPCOPY_PCAP) if (filter_port_num == 0) { tc_log_info(LOG_ERR, 0, "filter_port_num is zero"); return TC_ERROR; } pt = clt_settings.filter; strcpy(pt, "tcp dst port "); pt = pt + strlen(pt); for (i = 0; i < filter_port_num -1; i++) { sprintf(pt, "%d or ", ntohs(filter_port[i])); pt = pt + strlen(pt); } sprintf(pt, "%d", ntohs(filter_port[i])); tc_log_info(LOG_NOTICE, 0, "filter = %s", clt_settings.filter); #endif /* init packets for processing */ #if (TCPCOPY_OFFLINE) if (tc_offline_init(event_loop, clt_settings.pcap_file) == TC_ERROR) { return TC_ERROR; } #else if (tc_packets_init(event_loop) == TC_ERROR) { return TC_ERROR; } #endif return TC_OK; }