#include #include #include "packet_def.h" #include "packet_parse.h" #include "packet_build.h" #include "packet_dump.h" #define PRINT_GREEN(fmt, ...) printf("\033[0;32m" fmt "\033[0m\n", ##__VA_ARGS__) #define PRINT_RED(fmt, ...) printf("\033[0;31m" fmt "\033[0m\n", ##__VA_ARGS__) // https://dox.ipxe.org/tcp_8h_source.html #define TCP_OPTION_TS 8 #define TCP_OPTION_NOP 1 struct tcp_timestamp_option { uint8_t kind; uint8_t length; uint32_t tsval; uint32_t tsecr; } __attribute__((packed)); struct tcp_timestamp_padded_option { uint8_t nop[2]; struct tcp_timestamp_option tsopt; } __attribute__((packed)); struct tcp_timestamp_padded_option ts_pad_opt = { .nop = {TCP_OPTION_NOP, TCP_OPTION_NOP}, .tsopt = { .kind = TCP_OPTION_TS, .length = 10, .tsval = 0x12345678, .tsecr = 0x87654321, }, }; /****************************************************************************** * [Protocols in frame: eth:ethertype:ip:tcp] ****************************************************************************** * * Frame 1: 60 bytes on wire (480 bits), 60 bytes captured (480 bits) * Ethernet II, Src: 52:54:00:94:27:9b (52:54:00:94:27:9b), Dst: 52:54:00:19:8f:63 (52:54:00:19:8f:63) * Destination: 52:54:00:19:8f:63 (52:54:00:19:8f:63) * Source: 52:54:00:94:27:9b (52:54:00:94:27:9b) * Type: IPv4 (0x0800) * Padding: 000000000000 * Internet Protocol Version 4, Src: 192.168.122.202, Dst: 192.168.122.100 * 0100 .... = Version: 4 * .... 0101 = Header Length: 20 bytes (5) * Differentiated Services Field: 0x00 (DSCP: CS0, ECN: Not-ECT) * 0000 00.. = Differentiated Services Codepoint: Default (0) * .... ..00 = Explicit Congestion Notification: Not ECN-Capable Transport (0) * Total Length: 40 * Identification: 0x0c5e (3166) * 010. .... = Flags: 0x2, Don't fragment * 0... .... = Reserved bit: Not set * .1.. .... = Don't fragment: Set * ..0. .... = More fragments: Not set * ...0 0000 0000 0000 = Fragment Offset: 0 * Time to Live: 64 * Protocol: TCP (6) * Header Checksum: 0xb7f2 [correct] * [Header checksum status: Good] * [Calculated Checksum: 0xb7f2] * Source Address: 192.168.122.202 * Destination Address: 192.168.122.100 * Transmission Control Protocol, Src Port: 1080, Dst Port: 62395, Seq: 1457975085, Ack: 1047768425, Len: 0 * Source Port: 1080 * Destination Port: 62395 * [Stream index: 0] * [Conversation completeness: Incomplete (4)] * ..0. .... = RST: Absent * ...0 .... = FIN: Absent * .... 0... = Data: Absent * .... .1.. = ACK: Present * .... ..0. = SYN-ACK: Absent * .... ...0 = SYN: Absent * [Completeness Flags: ···A··] * [TCP Segment Len: 0] * Sequence Number: 1457975085 * [Next Sequence Number: 1457975085] * Acknowledgment Number: 1047768425 * 0101 .... = Header Length: 20 bytes (5) * Flags: 0x010 (ACK) * 000. .... .... = Reserved: Not set * ...0 .... .... = Accurate ECN: Not set * .... 0... .... = Congestion Window Reduced: Not set * .... .0.. .... = ECN-Echo: Not set * .... ..0. .... = Urgent: Not set * .... ...1 .... = Acknowledgment: Set * .... .... 0... = Push: Not set * .... .... .0.. = Reset: Not set * .... .... ..0. = Syn: Not set * .... .... ...0 = Fin: Not set * [TCP Flags: ·······A····] * Window: 457 * [Calculated window size: 457] * [Window size scaling factor: -1 (unknown)] * Checksum: 0x0da7 [correct] * [Calculated Checksum: 0x0da7] * [Checksum Status: Good] * Urgent Pointer: 0 * [Timestamps] * [Time since first frame in this TCP stream: 0.000000000 seconds] * [Time since previous frame in this TCP stream: 0.000000000 seconds] */ unsigned char data1[] = { 0x52, 0x54, 0x00, 0x19, 0x8f, 0x63, 0x52, 0x54, 0x00, 0x94, 0x27, 0x9b, 0x08, 0x00, 0x45, 0x00, 0x00, 0x28, 0x0c, 0x5e, 0x40, 0x00, 0x40, 0x06, 0xb7, 0xf2, 0xc0, 0xa8, 0x7a, 0xca, 0xc0, 0xa8, 0x7a, 0x64, 0x04, 0x38, 0xf3, 0xbb, 0x56, 0xe6, 0xef, 0x2d, 0x3e, 0x73, 0xad, 0x69, 0x50, 0x10, 0x01, 0xc9, 0x0d, 0xa7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00}; /* * imitate_tcp_packet() * -> ETH->IPv4->TCP * -> with TCP options * -> with TCP payload */ #if 1 TEST(PACKET_BUILD, ETH_IP4_TCP) { struct packet orig_pkt; memset(&orig_pkt, 0, sizeof(orig_pkt)); packet_parse(&orig_pkt, (const char *)data1, sizeof(data1)); PRINT_GREEN("origin packet:"); packet_print(&orig_pkt); struct packet *new_pkt = imitate_tcp_packet(&orig_pkt, 1, 2, TH_ACK, (const char *)&ts_pad_opt, sizeof(ts_pad_opt), "Hello", 5); EXPECT_TRUE(new_pkt != nullptr); PRINT_GREEN("new packet:"); packet_print(new_pkt); packet_dump_hex(new_pkt, STDOUT_FILENO); packet_dump_pcap(new_pkt, "imitate-eth-ipv4-tcp.pcap"); const char *orig_pkt_data = packet_get_raw_data(&orig_pkt); uint16_t orig_pkt_len = packet_get_raw_len(&orig_pkt); const char *new_pkt_data = packet_get_raw_data(new_pkt); uint16_t new_pkt_len = packet_get_raw_len(new_pkt); EXPECT_TRUE(orig_pkt_len - 6 == // trim Eth padding: 000000000000 new_pkt_len - 12 - 5 // trim TCP options, TCP payload ); for (uint16_t i = 0; i < new_pkt_len - 12 - 5; i++) { if ((16 <= i && i <= 17) || // skip IPv4 total length (18 <= i && i <= 19) || // skip IPv4 identification i == 22 || // skip IPv4 TTL (24 <= i && i <= 25) || // skip IPv4 checksum (38 <= i && i <= 41) || // skip TCP seq (42 <= i && i <= 45) || // skip TCP ack i == 46 || // skip TCP data offset i == 47 || // skip TCP flags (48 <= i && i <= 49) || // skip TCP window (50 <= i && i <= 51) // skip TCP checksum ) { continue; } // printf(("idx: %d, orig: %02x, new: %02x\n"), i, orig_pkt_data[i], new_pkt_data[i]); EXPECT_TRUE(orig_pkt_data[i] == new_pkt_data[i]); } } #endif int main(int argc, char **argv) { ::testing::InitGoogleTest(&argc, argv); return RUN_ALL_TESTS(); }