network.h 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135
  1. /*
  2. * network.h
  3. *
  4. * Created on: Jul 29, 2017
  5. * Author: wangyu
  6. */
  7. #ifndef UDP2RAW_NETWORK_H_
  8. #define UDP2RAW_NETWORK_H_
  9. extern int raw_recv_fd;
  10. extern int raw_send_fd;
  11. extern int seq_mode;
  12. extern int max_seq_mode;
  13. extern int filter_port;
  14. extern u32_t bind_address_uint32;
  15. extern int disable_bpf_filter;
  16. extern int lower_level;
  17. extern int lower_level_manual;
  18. extern char if_name[100];
  19. extern unsigned char dest_hw_addr[];
  20. extern int ifindex;
  21. struct icmphdr
  22. {
  23. uint8_t type;
  24. uint8_t code;
  25. uint16_t check_sum;
  26. uint16_t id;
  27. uint16_t seq;
  28. };
  29. struct pseudo_header {
  30. u_int32_t source_address;
  31. u_int32_t dest_address;
  32. u_int8_t placeholder;
  33. u_int8_t protocol;
  34. u_int16_t tcp_length;
  35. };
  36. struct packet_info_t //todo change this to union
  37. {
  38. uint8_t protocol;
  39. //ip_part:
  40. u32_t src_ip;
  41. uint16_t src_port;
  42. u32_t dst_ip;
  43. uint16_t dst_port;
  44. //tcp_part:
  45. bool syn,ack,psh,rst;
  46. u32_t seq,ack_seq;
  47. u32_t ack_seq_counter;
  48. u32_t ts,ts_ack;
  49. uint16_t icmp_seq;
  50. bool has_ts;
  51. sockaddr_ll addr_ll;
  52. i32_t data_len;
  53. packet_info_t();
  54. };
  55. struct raw_info_t
  56. {
  57. packet_info_t send_info;
  58. packet_info_t recv_info;
  59. //int last_send_len;
  60. //int last_recv_len;
  61. u32_t reserved_send_seq;
  62. //uint32_t first_seq,first_ack_seq;
  63. };//g_raw_info;
  64. int init_raw_socket();
  65. void init_filter(int port);
  66. void remove_filter();
  67. int init_ifindex(const char * if_name,int &index);
  68. int find_lower_level_info(u32_t ip,u32_t &dest_ip,string &if_name,string &hw);
  69. int get_src_adress(u32_t &ip,u32_t remote_ip_uint32,int remote_port); //a trick to get src adress for a dest adress,so that we can use the src address in raw socket as source ip
  70. int try_to_list_and_bind(int bind_fd,u32_t local_ip_uint32,int port); //try to bind to a port,may fail.
  71. int client_bind_to_a_new_port(int bind_fd,u32_t local_ip_uint32);//find a free port and bind to it.
  72. int send_raw_ip(raw_info_t &raw_info,const char * payload,int payloadlen);
  73. int peek_raw(packet_info_t &peek_info);
  74. int recv_raw_ip(raw_info_t &raw_info,char * &payload,int &payloadlen);
  75. int send_raw_icmp(raw_info_t &raw_info, const char * payload, int payloadlen);
  76. int send_raw_udp(raw_info_t &raw_info, const char * payload, int payloadlen);
  77. int send_raw_tcp(raw_info_t &raw_info,const char * payload, int payloadlen);
  78. int recv_raw_icmp(raw_info_t &raw_info, char *&payload, int &payloadlen);
  79. int recv_raw_udp(raw_info_t &raw_info, char *&payload, int &payloadlen);
  80. int recv_raw_tcp(raw_info_t &raw_info,char * &payload,int &payloadlen);
  81. //int send_raw(raw_info_t &raw_info,const char * payload,int payloadlen);
  82. //int recv_raw(raw_info_t &raw_info,char * &payload,int &payloadlen);
  83. int send_raw0(raw_info_t &raw_info,const char * payload,int payloadlen);
  84. int recv_raw0(raw_info_t &raw_info,char * &payload,int &payloadlen);
  85. int after_send_raw0(raw_info_t &raw_info);
  86. int after_recv_raw0(raw_info_t &raw_info);
  87. #endif /* NETWORK_H_ */