123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895 |
- #include "common.h"
- #include "network.h"
- #include "connection.h"
- #include "misc.h"
- #include "log.h"
- #include "lib/md5.h"
- #include "encrypt.h"
- #include "fd_manager.h"
- #ifdef UDP2RAW_MP
- u32_t detect_interval = 1500;
- u64_t laste_detect_time = 0;
- int use_udp_for_detection = 0;
- int use_tcp_for_detection = 1;
- extern pcap_t *pcap_handle;
- extern int pcap_captured_full_len;
- #endif
- int client_on_timer(conn_info_t &conn_info) // for client. called when a timer is ready in epoll
- {
- packet_info_t &send_info = conn_info.raw_info.send_info;
- packet_info_t &recv_info = conn_info.raw_info.recv_info;
- raw_info_t &raw_info = conn_info.raw_info;
- conn_info.blob->conv_manager.c.clear_inactive();
- mylog(log_trace, "timer!\n");
- mylog(log_trace, "roller my %d,oppsite %d,%lld\n", int(conn_info.my_roller), int(conn_info.oppsite_roller), conn_info.last_oppsite_roller_time);
- mylog(log_trace, "<client_on_timer,send_info.ts_ack= %u>\n", send_info.ts_ack);
- #ifdef UDP2RAW_MP
- // mylog(log_debug,"pcap cnt :%d\n",pcap_cnt);
- if (send_with_pcap && !pcap_header_captured) {
- if (get_current_time() - laste_detect_time > detect_interval) {
- laste_detect_time = get_current_time();
- } else {
- return 0;
- }
- /*
- struct sockaddr_in remote_addr_in={0};
- socklen_t slen = sizeof(sockaddr_in);
- int port=get_true_random_number()%65534+1;
- remote_addr_in.sin_family = AF_INET;
- remote_addr_in.sin_port = htons(port);
- remote_addr_in.sin_addr.s_addr = remote_ip_uint32;*/
- int port = get_true_random_number() % 65534 + 1;
- address_t tmp_addr = remote_addr;
- tmp_addr.set_port(port);
- if (use_udp_for_detection) {
- int new_udp_fd = socket(tmp_addr.get_type(), SOCK_DGRAM, IPPROTO_UDP);
- if (new_udp_fd < 0) {
- mylog(log_warn, "create new_udp_fd error\n");
- return -1;
- }
- setnonblocking(new_udp_fd);
- u64_t tmp = get_true_random_number();
- int ret = sendto(new_udp_fd, (char *)(&tmp), sizeof(tmp), 0, (struct sockaddr *)&tmp_addr.inner, tmp_addr.get_len());
- if (ret == -1) {
- mylog(log_warn, "sendto() failed\n");
- }
- sock_close(new_udp_fd);
- }
- if (use_tcp_for_detection) {
- static int last_tcp_fd = -1;
- int new_tcp_fd = socket(tmp_addr.get_type(), SOCK_STREAM, IPPROTO_TCP);
- if (new_tcp_fd < 0) {
- mylog(log_warn, "create new_tcp_fd error\n");
- return -1;
- }
- setnonblocking(new_tcp_fd);
- connect(new_tcp_fd, (struct sockaddr *)&tmp_addr.inner, tmp_addr.get_len());
- if (last_tcp_fd != -1)
- sock_close(last_tcp_fd);
- last_tcp_fd = new_tcp_fd;
- // close(new_tcp_fd);
- }
- mylog(log_info, "waiting for a use-able packet to be captured\n");
- return 0;
- }
- #endif
- if (raw_info.disabled) {
- conn_info.state.client_current_state = client_idle;
- conn_info.my_id = get_true_random_number_nz();
- mylog(log_info, "state back to client_idle\n");
- }
- if (conn_info.state.client_current_state == client_idle) {
- raw_info.rst_received = 0;
- raw_info.disabled = 0;
- fail_time_counter++;
- if (max_fail_time > 0 && fail_time_counter > max_fail_time) {
- mylog(log_fatal, "max_fail_time exceed\n");
- myexit(-1);
- }
- conn_info.blob->anti_replay.re_init();
- conn_info.my_id = get_true_random_number_nz(); /// todo no need to do this everytime
- address_t tmp_addr;
- // u32_t new_ip=0;
- if (!force_source_ip) {
- if (get_src_adress2(tmp_addr, remote_addr) != 0) {
- mylog(log_warn, "get_src_adress() failed\n");
- return -1;
- }
- // source_addr=new_addr;
- // source_addr.set_port(0);
- mylog(log_info, "source_addr is now %s\n", tmp_addr.get_ip());
- /*
- if(new_ip!=source_ip_uint32)
- {
- mylog(log_info,"source ip changed from %s to ",my_ntoa(source_ip_uint32));
- log_bare(log_info,"%s\n",my_ntoa(new_ip));
- source_ip_uint32=new_ip;
- send_info.src_ip=new_ip;
- }*/
- } else {
- tmp_addr = source_addr;
- }
- send_info.new_src_ip.from_address_t(tmp_addr);
- if (force_source_port == 0) {
- send_info.src_port = client_bind_to_a_new_port2(bind_fd, tmp_addr);
- } else {
- send_info.src_port = source_port;
- }
- if (raw_mode == mode_icmp) {
- send_info.dst_port = send_info.src_port;
- }
- mylog(log_info, "using port %d\n", send_info.src_port);
- init_filter(send_info.src_port);
- if (raw_mode == mode_icmp || raw_mode == mode_udp) {
- conn_info.state.client_current_state = client_handshake1;
- mylog(log_info, "state changed from client_idle to client_pre_handshake\n");
- }
- if (raw_mode == mode_faketcp) {
- if (use_tcp_dummy_socket) {
- setnonblocking(bind_fd);
- int ret = connect(bind_fd, (struct sockaddr *)&remote_addr.inner, remote_addr.get_len());
- mylog(log_debug, "ret=%d,errno=%s, %d %s\n", ret, get_sock_error(), bind_fd, remote_addr.get_str());
- // mylog(log_info,"ret=%d,errno=,%d %s\n",ret,bind_fd,remote_addr.get_str());
- conn_info.state.client_current_state = client_tcp_handshake_dummy;
- mylog(log_info, "state changed from client_idle to client_tcp_handshake_dummy\n");
- } else {
- conn_info.state.client_current_state = client_tcp_handshake;
- mylog(log_info, "state changed from client_idle to client_tcp_handshake\n");
- }
- }
- conn_info.last_state_time = get_current_time();
- conn_info.last_hb_sent_time = 0;
- // dont return;
- }
- if (conn_info.state.client_current_state == client_tcp_handshake) // send and resend syn
- {
- assert(raw_mode == mode_faketcp);
- if (get_current_time() - conn_info.last_state_time > client_handshake_timeout) {
- conn_info.state.client_current_state = client_idle;
- mylog(log_info, "state back to client_idle from client_tcp_handshake\n");
- return 0;
- } else if (get_current_time() - conn_info.last_hb_sent_time > client_retry_interval) {
- if (raw_mode == mode_faketcp) {
- if (conn_info.last_hb_sent_time == 0) {
- send_info.psh = 0;
- send_info.syn = 1;
- send_info.ack = 0;
- send_info.ts_ack = 0;
- send_info.seq = get_true_random_number();
- send_info.ack_seq = get_true_random_number();
- }
- }
- send_raw0(raw_info, 0, 0);
- conn_info.last_hb_sent_time = get_current_time();
- mylog(log_info, "(re)sent tcp syn\n");
- return 0;
- } else {
- return 0;
- }
- return 0;
- } else if (conn_info.state.client_current_state == client_tcp_handshake_dummy) {
- assert(raw_mode == mode_faketcp);
- if (get_current_time() - conn_info.last_state_time > client_handshake_timeout) {
- conn_info.state.client_current_state = client_idle;
- mylog(log_info, "state back to client_idle from client_tcp_handshake_dummy\n");
- return 0;
- }
- } else if (conn_info.state.client_current_state == client_handshake1) // send and resend handshake1
- {
- if (get_current_time() - conn_info.last_state_time > client_handshake_timeout) {
- conn_info.state.client_current_state = client_idle;
- mylog(log_info, "state back to client_idle from client_handshake1\n");
- return 0;
- } else if (get_current_time() - conn_info.last_hb_sent_time > client_retry_interval) {
- if (raw_mode == mode_faketcp) {
- if (conn_info.last_hb_sent_time == 0) {
- send_info.seq++;
- send_info.ack_seq = recv_info.seq + 1;
- send_info.ts_ack = recv_info.ts;
- raw_info.reserved_send_seq = send_info.seq;
- }
- send_info.seq = raw_info.reserved_send_seq;
- send_info.psh = 0;
- send_info.syn = 0;
- send_info.ack = 1;
- if (!use_tcp_dummy_socket)
- send_raw0(raw_info, 0, 0);
- send_handshake(raw_info, conn_info.my_id, 0, const_id);
- send_info.seq += raw_info.send_info.data_len;
- } else {
- send_handshake(raw_info, conn_info.my_id, 0, const_id);
- if (raw_mode == mode_icmp)
- send_info.my_icmp_seq++;
- }
- conn_info.last_hb_sent_time = get_current_time();
- mylog(log_info, "(re)sent handshake1\n");
- return 0;
- } else {
- return 0;
- }
- return 0;
- } else if (conn_info.state.client_current_state == client_handshake2) {
- if (get_current_time() - conn_info.last_state_time > client_handshake_timeout) {
- conn_info.state.client_current_state = client_idle;
- mylog(log_info, "state back to client_idle from client_handshake2\n");
- return 0;
- } else if (get_current_time() - conn_info.last_hb_sent_time > client_retry_interval) {
- if (raw_mode == mode_faketcp) {
- if (conn_info.last_hb_sent_time == 0) {
- send_info.ack_seq = recv_info.seq + raw_info.recv_info.data_len;
- send_info.ts_ack = recv_info.ts;
- raw_info.reserved_send_seq = send_info.seq;
- }
- send_info.seq = raw_info.reserved_send_seq;
- send_handshake(raw_info, conn_info.my_id, conn_info.oppsite_id, const_id);
- send_info.seq += raw_info.send_info.data_len;
- } else {
- send_handshake(raw_info, conn_info.my_id, conn_info.oppsite_id, const_id);
- if (raw_mode == mode_icmp)
- send_info.my_icmp_seq++;
- }
- conn_info.last_hb_sent_time = get_current_time();
- mylog(log_info, "(re)sent handshake2\n");
- return 0;
- } else {
- return 0;
- }
- return 0;
- } else if (conn_info.state.client_current_state == client_ready) {
- fail_time_counter = 0;
- mylog(log_trace, "time %llu,%llu\n", get_current_time(), conn_info.last_state_time);
- if (get_current_time() - conn_info.last_hb_recv_time > client_conn_timeout) {
- conn_info.state.client_current_state = client_idle;
- conn_info.my_id = get_true_random_number_nz();
- mylog(log_info, "state back to client_idle from client_ready bc of server-->client direction timeout\n");
- return 0;
- }
- if (get_current_time() - conn_info.last_oppsite_roller_time > client_conn_uplink_timeout) {
- conn_info.state.client_current_state = client_idle;
- conn_info.my_id = get_true_random_number_nz();
- mylog(log_info, "state back to client_idle from client_ready bc of client-->server direction timeout\n");
- }
- if (get_current_time() - conn_info.last_hb_sent_time < heartbeat_interval) {
- return 0;
- }
- mylog(log_debug, "heartbeat sent <%x,%x>\n", conn_info.oppsite_id, conn_info.my_id);
- if (hb_mode == 0)
- send_safer(conn_info, 'h', hb_buf, 0); /////////////send
- else
- send_safer(conn_info, 'h', hb_buf, hb_len);
- conn_info.last_hb_sent_time = get_current_time();
- return 0;
- } else {
- mylog(log_fatal, "unknown state,this shouldnt happen.\n");
- myexit(-1);
- }
- return 0;
- }
- int client_on_raw_recv_hs2_or_ready(conn_info_t &conn_info, char type, char *data, int data_len) {
- packet_info_t &send_info = conn_info.raw_info.send_info;
- packet_info_t &recv_info = conn_info.raw_info.recv_info;
- if (!recv_info.new_src_ip.equal(send_info.new_dst_ip) || recv_info.src_port != send_info.dst_port) {
- mylog(log_warn, "unexpected adress %s %s %d %d,this shouldnt happen.\n", recv_info.new_src_ip.get_str1(), send_info.new_dst_ip.get_str2(), recv_info.src_port, send_info.dst_port);
- return -1;
- }
- if (conn_info.state.client_current_state == client_handshake2) {
- mylog(log_info, "changed state from to client_handshake2 to client_ready\n");
- conn_info.state.client_current_state = client_ready;
- conn_info.last_hb_sent_time = 0;
- conn_info.last_hb_recv_time = get_current_time();
- conn_info.last_oppsite_roller_time = conn_info.last_hb_recv_time;
- client_on_timer(conn_info);
- }
- if (data_len >= 0 && type == 'h') {
- mylog(log_debug, "[hb]heart beat received,oppsite_roller=%d\n", int(conn_info.oppsite_roller));
- conn_info.last_hb_recv_time = get_current_time();
- return 0;
- } else if (data_len >= int(sizeof(u32_t)) && type == 'd') {
- mylog(log_trace, "received a data from fake tcp,len:%d\n", data_len);
- if (hb_mode == 0)
- conn_info.last_hb_recv_time = get_current_time();
- u32_t tmp_conv_id;
- memcpy(&tmp_conv_id, &data[0], sizeof(tmp_conv_id));
- tmp_conv_id = ntohl(tmp_conv_id);
- if (!conn_info.blob->conv_manager.c.is_conv_used(tmp_conv_id)) {
- mylog(log_info, "unknow conv %d,ignore\n", tmp_conv_id);
- return 0;
- }
- conn_info.blob->conv_manager.c.update_active_time(tmp_conv_id);
- // u64_t u64=conn_info.blob->conv_manager.c.find_data_by_conv(tmp_conv_id);
- address_t tmp_addr = conn_info.blob->conv_manager.c.find_data_by_conv(tmp_conv_id);
- // sockaddr_in tmp_sockaddr={0};
- // tmp_sockaddr.sin_family = AF_INET;
- // tmp_sockaddr.sin_addr.s_addr=(u64>>32u);
- // tmp_sockaddr.sin_port= htons(uint16_t((u64<<32u)>>32u));
- int ret = sendto(udp_fd, data + sizeof(u32_t), data_len - (sizeof(u32_t)), 0, (struct sockaddr *)&tmp_addr.inner, tmp_addr.get_len());
- if (ret < 0) {
- mylog(log_warn, "sento returned %d,%s,%02x,%s\n", ret, get_sock_error(), int(tmp_addr.get_type()), tmp_addr.get_str());
- // perror("ret<0");
- }
- } else {
- mylog(log_warn, "unknown packet,this shouldnt happen.\n");
- return -1;
- }
- return 0;
- }
- int client_on_raw_recv(conn_info_t &conn_info) // called when raw fd received a packet.
- {
- char *data;
- int data_len;
- packet_info_t &send_info = conn_info.raw_info.send_info;
- packet_info_t &recv_info = conn_info.raw_info.recv_info;
- raw_info_t &raw_info = conn_info.raw_info;
- mylog(log_trace, "<client_on_raw_recv,send_info.ts_ack= %u>\n", send_info.ts_ack);
- #ifdef UDP2RAW_LINUX
- if (pre_recv_raw_packet() < 0) return -1;
- #endif
- if (conn_info.state.client_current_state == client_idle) {
- discard_raw_packet();
- // recv(raw_recv_fd, 0,0, 0 );
- } else if (conn_info.state.client_current_state == client_tcp_handshake || conn_info.state.client_current_state == client_tcp_handshake_dummy) // received syn ack
- {
- assert(raw_mode == mode_faketcp);
- if (recv_raw0(raw_info, data, data_len) < 0) {
- return -1;
- }
- if (data_len >= max_data_len + 1) {
- mylog(log_debug, "data_len=%d >= max_data_len+1,ignored", data_len);
- return -1;
- }
- if (!recv_info.new_src_ip.equal(send_info.new_dst_ip) || recv_info.src_port != send_info.dst_port) {
- mylog(log_debug, "unexpected adress %s %s %d %d\n", recv_info.new_src_ip.get_str1(), send_info.new_dst_ip.get_str2(), recv_info.src_port, send_info.dst_port);
- return -1;
- }
- if (data_len == 0 && raw_info.recv_info.syn == 1 && raw_info.recv_info.ack == 1) {
- if (conn_info.state.client_current_state == client_tcp_handshake) {
- if (recv_info.ack_seq != send_info.seq + 1) {
- mylog(log_debug, "seq ack_seq mis match\n");
- return -1;
- }
- mylog(log_info, "state changed from client_tcp_handshake to client_handshake1\n");
- } else {
- send_info.seq = recv_info.ack_seq - 1;
- mylog(log_info, "state changed from client_tcp_dummy to client_handshake1\n");
- // send_info.ack_seq=recv_info.seq+1;
- }
- conn_info.state.client_current_state = client_handshake1;
- conn_info.last_state_time = get_current_time();
- conn_info.last_hb_sent_time = 0;
- client_on_timer(conn_info);
- return 0;
- } else {
- mylog(log_debug, "unexpected packet type,expected:syn ack\n");
- return -1;
- }
- } else if (conn_info.state.client_current_state == client_handshake1) // recevied respond of handshake1
- {
- if (recv_bare(raw_info, data, data_len) != 0) {
- mylog(log_debug, "recv_bare failed!\n");
- return -1;
- }
- if (!recv_info.new_src_ip.equal(send_info.new_dst_ip) || recv_info.src_port != send_info.dst_port) {
- mylog(log_debug, "unexpected adress %s %s %d %d\n", recv_info.new_src_ip.get_str1(), send_info.new_dst_ip.get_str2(), recv_info.src_port, send_info.dst_port);
- return -1;
- }
- if (data_len < int(3 * sizeof(my_id_t))) {
- mylog(log_debug, "too short to be a handshake\n");
- return -1;
- }
- my_id_t tmp_oppsite_id;
- memcpy(&tmp_oppsite_id, &data[0], sizeof(tmp_oppsite_id));
- tmp_oppsite_id = ntohl(tmp_oppsite_id);
- my_id_t tmp_my_id;
- memcpy(&tmp_my_id, &data[sizeof(my_id_t)], sizeof(tmp_my_id));
- tmp_my_id = ntohl(tmp_my_id);
- my_id_t tmp_oppsite_const_id;
- memcpy(&tmp_oppsite_const_id, &data[sizeof(my_id_t) * 2], sizeof(tmp_oppsite_const_id));
- tmp_oppsite_const_id = ntohl(tmp_oppsite_const_id);
- if (tmp_my_id != conn_info.my_id) {
- mylog(log_debug, "tmp_my_id doesnt match\n");
- return -1;
- }
- if (raw_mode == mode_faketcp) {
- if (recv_info.ack_seq != send_info.seq) {
- mylog(log_debug, "seq ack_seq mis match\n");
- return -1;
- }
- if (recv_info.seq != send_info.ack_seq) {
- mylog(log_debug, "seq ack_seq mis match\n");
- return -1;
- }
- }
- conn_info.oppsite_id = tmp_oppsite_id;
- mylog(log_info, "changed state from to client_handshake1 to client_handshake2,my_id is %x,oppsite id is %x\n", conn_info.my_id, conn_info.oppsite_id);
- conn_info.state.client_current_state = client_handshake2;
- conn_info.last_state_time = get_current_time();
- conn_info.last_hb_sent_time = 0;
- client_on_timer(conn_info);
- return 0;
- } else if (conn_info.state.client_current_state == client_handshake2 || conn_info.state.client_current_state == client_ready) // received heartbeat or data
- {
- vector<char> type_vec;
- vector<string> data_vec;
- recv_safer_multi(conn_info, type_vec, data_vec);
- if (data_vec.empty()) {
- mylog(log_debug, "recv_safer failed!\n");
- return -1;
- }
- for (int i = 0; i < (int)type_vec.size(); i++) {
- char type = type_vec[i];
- char *data = (char *)data_vec[i].c_str(); // be careful, do not append data to it
- int data_len = data_vec[i].length();
- client_on_raw_recv_hs2_or_ready(conn_info, type, data, data_len);
- }
- return 0;
- } else {
- mylog(log_fatal, "unknown state,this shouldnt happen.\n");
- myexit(-1);
- }
- return 0;
- }
- int client_on_udp_recv(conn_info_t &conn_info) {
- int recv_len;
- char buf[buf_len];
- address_t::storage_t udp_new_addr_in = {{0}};
- socklen_t udp_new_addr_len = sizeof(address_t::storage_t);
- if ((recv_len = recvfrom(udp_fd, buf, max_data_len + 1, 0,
- (struct sockaddr *)&udp_new_addr_in, &udp_new_addr_len)) == -1) {
- mylog(log_debug, "recv_from error,%s\n", get_sock_error());
- return -1;
- // myexit(1);
- };
- if (recv_len == max_data_len + 1) {
- mylog(log_warn, "huge packet, data_len > %d,dropped\n", max_data_len);
- return -1;
- }
- if (recv_len >= mtu_warn) {
- mylog(log_warn, "huge packet,data len=%d (>=%d).strongly suggested to set a smaller mtu at upper level,to get rid of this warn\n ", recv_len, mtu_warn);
- }
- address_t tmp_addr;
- tmp_addr.from_sockaddr((sockaddr *)&udp_new_addr_in, udp_new_addr_len);
- u32_t conv;
- if (!conn_info.blob->conv_manager.c.is_data_used(tmp_addr)) {
- if (conn_info.blob->conv_manager.c.get_size() >= max_conv_num) {
- mylog(log_warn, "ignored new udp connect bc max_conv_num exceed\n");
- return -1;
- }
- conv = conn_info.blob->conv_manager.c.get_new_conv();
- conn_info.blob->conv_manager.c.insert_conv(conv, tmp_addr);
- mylog(log_info, "new packet from %s,conv_id=%x\n", tmp_addr.get_str(), conv);
- } else {
- conv = conn_info.blob->conv_manager.c.find_conv_by_data(tmp_addr);
- }
- conn_info.blob->conv_manager.c.update_active_time(conv);
- if (conn_info.state.client_current_state == client_ready) {
- send_data_safer(conn_info, buf, recv_len, conv);
- }
- return 0;
- }
- void udp_accept_cb(struct ev_loop *loop, struct ev_io *watcher, int revents) {
- conn_info_t &conn_info = *((conn_info_t *)watcher->data);
- client_on_udp_recv(conn_info);
- }
- void raw_recv_cb(struct ev_loop *loop, struct ev_io *watcher, int revents) {
- if (is_udp2raw_mp) assert(0 == 1);
- conn_info_t &conn_info = *((conn_info_t *)watcher->data);
- client_on_raw_recv(conn_info);
- }
- #ifdef UDP2RAW_MP
- void async_cb(struct ev_loop *loop, struct ev_async *watcher, int revents) {
- conn_info_t &conn_info = *((conn_info_t *)watcher->data);
- if (send_with_pcap && !pcap_header_captured) {
- int empty = 0;
- char *p;
- int len;
- pthread_mutex_lock(&queue_mutex);
- empty = my_queue.empty();
- if (!empty) {
- my_queue.peek_front(p, len);
- my_queue.pop_front();
- }
- pthread_mutex_unlock(&queue_mutex);
- if (empty) return;
- pcap_header_captured = 1;
- assert(pcap_link_header_len != -1);
- memcpy(pcap_header_buf, p, max_data_len);
- log_bare(log_info, "link level header captured:\n");
- unsigned char *tmp = (unsigned char *)pcap_header_buf;
- pcap_captured_full_len = len;
- for (int i = 0; i < pcap_link_header_len; i++)
- log_bare(log_info, "<%x>", (u32_t)tmp[i]);
- log_bare(log_info, "\n");
- return;
- }
- // mylog(log_info,"async_cb called\n");
- while (1) {
- int empty = 0;
- char *p;
- int len;
- pthread_mutex_lock(&queue_mutex);
- empty = my_queue.empty();
- if (!empty) {
- my_queue.peek_front(p, len);
- my_queue.pop_front();
- }
- pthread_mutex_unlock(&queue_mutex);
- if (empty) break;
- if (g_fix_gro == 0 && len > max_data_len) {
- mylog(log_warn, "huge packet %d > %d, dropped. maybe you need to turn down mtu at upper level, or maybe you need the --fix-gro option\n", len, max_data_len);
- break;
- }
- int new_len = len - pcap_link_header_len;
- memcpy(g_packet_buf, p + pcap_link_header_len, new_len);
- g_packet_buf_len = new_len;
- assert(g_packet_buf_cnt == 0);
- g_packet_buf_cnt++;
- client_on_raw_recv(conn_info);
- }
- }
- #endif
- void clear_timer_cb(struct ev_loop *loop, struct ev_timer *watcher, int revents) {
- conn_info_t &conn_info = *((conn_info_t *)watcher->data);
- client_on_timer(conn_info);
- }
- void fifo_cb(struct ev_loop *loop, struct ev_io *watcher, int revents) {
- conn_info_t &conn_info = *((conn_info_t *)watcher->data);
- char buf[buf_len];
- int fifo_fd = watcher->fd;
- int len = read(fifo_fd, buf, sizeof(buf));
- if (len < 0) {
- mylog(log_warn, "fifo read failed len=%d,errno=%s\n", len, get_sock_error());
- return;
- }
- buf[len] = 0;
- while (len >= 1 && buf[len - 1] == '\n')
- buf[len - 1] = 0;
- mylog(log_info, "got data from fifo,len=%d,s=[%s]\n", len, buf);
- if (strcmp(buf, "reconnect") == 0) {
- mylog(log_info, "received command: reconnect\n");
- conn_info.state.client_current_state = client_idle;
- conn_info.my_id = get_true_random_number_nz();
- } else {
- mylog(log_info, "unknown command\n");
- }
- }
- int client_event_loop() {
- char buf[buf_len];
- conn_info_t conn_info;
- conn_info.my_id = get_true_random_number_nz();
- conn_info.prepare();
- packet_info_t &send_info = conn_info.raw_info.send_info;
- packet_info_t &recv_info = conn_info.raw_info.recv_info;
- #ifdef UDP2RAW_LINUX
- if (lower_level) {
- if (lower_level_manual) {
- int index;
- init_ifindex(if_name, raw_send_fd, index);
- // init_ifindex(if_name);
- memset(&send_info.addr_ll, 0, sizeof(send_info.addr_ll));
- send_info.addr_ll.sll_family = AF_PACKET;
- send_info.addr_ll.sll_ifindex = index;
- send_info.addr_ll.sll_halen = ETHER_ADDR_LEN;
- send_info.addr_ll.sll_protocol = htons(ETH_P_IP);
- memcpy(&send_info.addr_ll.sll_addr, dest_hw_addr, ETHER_ADDR_LEN);
- mylog(log_info, "we are running at lower-level (manual) mode\n");
- } else {
- u32_t dest_ip;
- string if_name_string;
- string hw_string;
- assert(remote_addr.get_type() == AF_INET);
- if (retry_on_error == 0) {
- if (find_lower_level_info(remote_addr.inner.ipv4.sin_addr.s_addr, dest_ip, if_name_string, hw_string) != 0) {
- mylog(log_fatal, "auto detect lower-level info failed for %s,specific it manually\n", remote_addr.get_ip());
- myexit(-1);
- }
- } else {
- int ok = 0;
- while (!ok) {
- if (find_lower_level_info(remote_addr.inner.ipv4.sin_addr.s_addr, dest_ip, if_name_string, hw_string) != 0) {
- mylog(log_warn, "auto detect lower-level info failed for %s,retry in %d seconds\n", remote_addr.get_ip(), retry_on_error_interval);
- sleep(retry_on_error_interval);
- } else {
- ok = 1;
- }
- }
- }
- mylog(log_info, "we are running at lower-level (auto) mode,%s %s %s\n", my_ntoa(dest_ip), if_name_string.c_str(), hw_string.c_str());
- u32_t hw[6];
- memset(hw, 0, sizeof(hw));
- sscanf(hw_string.c_str(), "%x:%x:%x:%x:%x:%x", &hw[0], &hw[1], &hw[2],
- &hw[3], &hw[4], &hw[5]);
- mylog(log_warn,
- "make sure this is correct: if_name=<%s> dest_mac_adress=<%02x:%02x:%02x:%02x:%02x:%02x> \n",
- if_name_string.c_str(), hw[0], hw[1], hw[2], hw[3], hw[4], hw[5]);
- for (int i = 0; i < 6; i++) {
- dest_hw_addr[i] = uint8_t(hw[i]);
- }
- // mylog(log_fatal,"--lower-level auto for client hasnt been implemented\n");
- int index;
- init_ifindex(if_name_string.c_str(), raw_send_fd, index);
- memset(&send_info.addr_ll, 0, sizeof(send_info.addr_ll));
- send_info.addr_ll.sll_family = AF_PACKET;
- send_info.addr_ll.sll_ifindex = index;
- send_info.addr_ll.sll_halen = ETHER_ADDR_LEN;
- send_info.addr_ll.sll_protocol = htons(ETH_P_IP);
- memcpy(&send_info.addr_ll.sll_addr, dest_hw_addr, ETHER_ADDR_LEN);
- // mylog(log_info,"we are running at lower-level (manual) mode\n");
- }
- }
- #endif
- #ifdef UDP2RAW_MP
- address_t tmp_addr;
- if (get_src_adress2(tmp_addr, remote_addr) != 0) {
- mylog(log_error, "get_src_adress() failed\n");
- myexit(-1);
- }
- if (strcmp(dev, "") == 0) {
- mylog(log_info, "--dev have not been set, trying to detect automatically, available devices:\n");
- mylog(log_info, "available device(device name: ip address ; description):\n");
- char errbuf[PCAP_ERRBUF_SIZE];
- int found = 0;
- pcap_if_t *interfaces, *d;
- if (pcap_findalldevs(&interfaces, errbuf) == -1) {
- mylog(log_fatal, "error in pcap_findalldevs(),%s\n", errbuf);
- myexit(-1);
- }
- for (pcap_if_t *d = interfaces; d != NULL; d = d->next) {
- log_bare(log_warn, "%s:", d->name);
- int cnt = 0;
- for (pcap_addr_t *a = d->addresses; a != NULL; a = a->next) {
- if (a->addr == NULL) {
- log_bare(log_debug, " [a->addr==NULL]");
- continue;
- }
- if (a->addr->sa_family == AF_INET || a->addr->sa_family == AF_INET6) {
- cnt++;
- if (a->addr->sa_family == AF_INET) {
- char s[max_addr_len];
- inet_ntop(AF_INET, &((struct sockaddr_in *)a->addr)->sin_addr, s, max_addr_len);
- log_bare(log_warn, " [%s]", s);
- if (a->addr->sa_family == raw_ip_version) {
- if (((struct sockaddr_in *)a->addr)->sin_addr.s_addr == tmp_addr.inner.ipv4.sin_addr.s_addr) {
- found++;
- strcpy(dev, d->name);
- }
- }
- } else {
- assert(a->addr->sa_family == AF_INET6);
- char s[max_addr_len];
- inet_ntop(AF_INET6, &((struct sockaddr_in6 *)a->addr)->sin6_addr, s, max_addr_len);
- log_bare(log_warn, " [%s]", s);
- if (a->addr->sa_family == raw_ip_version) {
- if (memcmp(&((struct sockaddr_in6 *)a->addr)->sin6_addr, &tmp_addr.inner.ipv6.sin6_addr, sizeof(struct in6_addr)) == 0) {
- found++;
- strcpy(dev, d->name);
- }
- }
- }
- } else {
- log_bare(log_debug, " [unknow:%d]", int(a->addr->sa_family));
- }
- }
- if (cnt == 0) log_bare(log_warn, " [no ip found]");
- if (d->description == 0) {
- log_bare(log_warn, "; (no description available)");
- } else {
- log_bare(log_warn, "; %s", d->description);
- }
- log_bare(log_warn, "\n");
- }
- if (found == 0) {
- mylog(log_fatal, "no matched device found for ip: [%s]\n", tmp_addr.get_ip());
- myexit(-1);
- } else if (found == 1) {
- mylog(log_info, "using device:[%s], ip: [%s]\n", dev, tmp_addr.get_ip());
- } else {
- mylog(log_fatal, "more than one devices found for ip: [%s] , you need to use --dev manually\n", tmp_addr.get_ip());
- myexit(-1);
- }
- } else {
- mylog(log_info, "--dev has been manually set, using device:[%s]\n", dev);
- }
- #endif
- send_info.src_port = 0;
- memset(&send_info.new_src_ip, 0, sizeof(send_info.new_src_ip));
- int i, j, k;
- int ret;
- send_info.new_dst_ip.from_address_t(remote_addr);
- send_info.dst_port = remote_addr.get_port();
- udp_fd = socket(local_addr.get_type(), SOCK_DGRAM, IPPROTO_UDP);
- set_buf_size(udp_fd, socket_buf_size);
- if (::bind(udp_fd, (struct sockaddr *)&local_addr.inner, local_addr.get_len()) == -1) {
- mylog(log_fatal, "socket bind error\n");
- // perror("socket bind error");
- myexit(1);
- }
- setnonblocking(udp_fd);
- // epollfd = epoll_create1(0);
- // const int max_events = 4096;
- // struct epoll_event ev, events[max_events];
- // if (epollfd < 0) {
- // mylog(log_fatal,"epoll return %d\n", epollfd);
- // myexit(-1);
- // }
- struct ev_loop *loop = ev_default_loop(0);
- assert(loop != NULL);
- // ev.events = EPOLLIN;
- // ev.data.u64 = udp_fd;
- // ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, udp_fd, &ev);
- // if (ret!=0) {
- // mylog(log_fatal,"add udp_listen_fd error\n");
- // myexit(-1);
- // }
- struct ev_io udp_accept_watcher;
- udp_accept_watcher.data = &conn_info;
- ev_io_init(&udp_accept_watcher, udp_accept_cb, udp_fd, EV_READ);
- ev_io_start(loop, &udp_accept_watcher);
- // ev.events = EPOLLIN;
- // ev.data.u64 = raw_recv_fd;
- // ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, raw_recv_fd, &ev);
- // if (ret!= 0) {
- // mylog(log_fatal,"add raw_fd error\n");
- // myexit(-1);
- // }
- #ifdef UDP2RAW_LINUX
- struct ev_io raw_recv_watcher;
- raw_recv_watcher.data = &conn_info;
- ev_io_init(&raw_recv_watcher, raw_recv_cb, raw_recv_fd, EV_READ);
- ev_io_start(loop, &raw_recv_watcher);
- #endif
- #ifdef UDP2RAW_MP
- g_default_loop = loop;
- async_watcher.data = &conn_info;
- ev_async_init(&async_watcher, async_cb);
- ev_async_start(loop, &async_watcher);
- init_raw_socket(); // must be put after dev detection
- #endif
- // set_timer(epollfd,timer_fd);
- struct ev_timer clear_timer;
- clear_timer.data = &conn_info;
- ev_timer_init(&clear_timer, clear_timer_cb, 0, timer_interval / 1000.0);
- ev_timer_start(loop, &clear_timer);
- mylog(log_debug, "send_raw : from %s %d to %s %d\n", send_info.new_src_ip.get_str1(), send_info.src_port, send_info.new_dst_ip.get_str2(), send_info.dst_port);
- int fifo_fd = -1;
- struct ev_io fifo_watcher;
- fifo_watcher.data = &conn_info;
- if (fifo_file[0] != 0) {
- fifo_fd = create_fifo(fifo_file);
- ev_io_init(&fifo_watcher, fifo_cb, fifo_fd, EV_READ);
- ev_io_start(loop, &fifo_watcher);
- mylog(log_info, "fifo_file=%s\n", fifo_file);
- }
- ev_run(loop, 0);
- return 0;
- }
|