network.h 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. /*
  2. * network.h
  3. *
  4. * Created on: Jul 29, 2017
  5. * Author: wangyu
  6. */
  7. #ifndef NETWORK_H_
  8. #define NETWORK_H_
  9. extern int raw_recv_fd;
  10. extern int raw_send_fd;
  11. extern int seq_mode;
  12. extern int filter_port;
  13. extern u32_t bind_address_uint32;
  14. extern int disable_bpf_filter;
  15. struct icmphdr
  16. {
  17. uint8_t type;
  18. uint8_t code;
  19. uint16_t check_sum;
  20. uint16_t id;
  21. uint16_t seq;
  22. };
  23. struct pseudo_header {
  24. u_int32_t source_address;
  25. u_int32_t dest_address;
  26. u_int8_t placeholder;
  27. u_int8_t protocol;
  28. u_int16_t tcp_length;
  29. };
  30. struct packet_info_t //todo change this to union
  31. {
  32. uint8_t protocol;
  33. //ip_part:
  34. u32_t src_ip;
  35. uint16_t src_port;
  36. u32_t dst_ip;
  37. uint16_t dst_port;
  38. //tcp_part:
  39. bool syn,ack,psh,rst;
  40. u32_t seq,ack_seq;
  41. u32_t ts,ts_ack;
  42. uint16_t icmp_seq;
  43. bool has_ts;
  44. packet_info_t();
  45. };
  46. struct raw_info_t
  47. {
  48. packet_info_t send_info;
  49. packet_info_t recv_info;
  50. int last_send_len;
  51. int last_recv_len;
  52. u32_t reserved_seq;
  53. //uint32_t first_seq,first_ack_seq;
  54. };//g_raw_info;
  55. int init_raw_socket();
  56. void init_filter(int port);
  57. void remove_filter();
  58. int send_raw_ip(raw_info_t &raw_info,const char * payload,int payloadlen);
  59. int peek_raw(packet_info_t &peek_info);
  60. int recv_raw_ip(raw_info_t &raw_info,char * &payload,int &payloadlen);
  61. int send_raw_icmp(raw_info_t &raw_info, const char * payload, int payloadlen);
  62. int send_raw_udp(raw_info_t &raw_info, const char * payload, int payloadlen);
  63. int send_raw_tcp(raw_info_t &raw_info,const char * payload, int payloadlen);
  64. int recv_raw_icmp(raw_info_t &raw_info, char *&payload, int &payloadlen);
  65. int recv_raw_udp(raw_info_t &raw_info, char *&payload, int &payloadlen);
  66. int recv_raw_tcp(raw_info_t &raw_info,char * &payload,int &payloadlen);
  67. //int send_raw(raw_info_t &raw_info,const char * payload,int payloadlen);
  68. //int recv_raw(raw_info_t &raw_info,char * &payload,int &payloadlen);
  69. int send_raw0(raw_info_t &raw_info,const char * payload,int payloadlen);
  70. int recv_raw0(raw_info_t &raw_info,char * &payload,int &payloadlen);
  71. int after_send_raw0(raw_info_t &raw_info);
  72. int after_recv_raw0(raw_info_t &raw_info);
  73. #endif /* NETWORK_H_ */