/* 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;
}
