#include "gtest_utils.h"

/*
 * Frame 4: 60 bytes on wire (480 bits), 60 bytes captured (480 bits)
 * Ethernet II, Src: Fortinet_cc:87:22 (e8:1c:ba:cc:87:22), Dst: EvocInte_2f:35:b8 (00:22:46:2f:35:b8)
 *     Destination: EvocInte_2f:35:b8 (00:22:46:2f:35:b8)
 *         Address: EvocInte_2f:35:b8 (00:22:46:2f:35:b8)
 *         .... ..0. .... .... .... .... = LG bit: Globally unique address (factory default)
 *         .... ...0 .... .... .... .... = IG bit: Individual address (unicast)
 *     Source: Fortinet_cc:87:22 (e8:1c:ba:cc:87:22)
 *         Address: Fortinet_cc:87:22 (e8:1c:ba:cc:87:22)
 *         .... ..0. .... .... .... .... = LG bit: Globally unique address (factory default)
 *         .... ...0 .... .... .... .... = IG bit: Individual address (unicast)
 *     Type: IPv4 (0x0800)
 *     Padding: 0000
 * Internet Protocol Version 4, Src: 192.168.36.103, Dst: 192.168.40.137
 *     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: 44
 *     Identification: 0xffff (65535)
 *     001. .... = Flags: 0x1, More fragments
 *         0... .... = Reserved bit: Not set
 *         .0.. .... = Don't fragment: Not set
 *         ..1. .... = More fragments: Set
 *     ...0 0000 0000 0000 = Fragment Offset: 0
 *     Time to Live: 127
 *     Protocol: TCP (6)
 *     Header Checksum: 0x4d8b [correct]
 *     [Header checksum status: Good]
 *     [Calculated Checksum: 0x4d8b]
 *     Source Address: 192.168.36.103
 *     Destination Address: 192.168.40.137
 *     [Reassembled IPv4 in frame: 5]
 * Data (24 bytes)
 *     Data: f4a5270f9107248703d518e75018ff005e9200003132330a
 *     [Length: 24]
 */

unsigned char frag1[] = {
    /* ETH */
    0x00, 0x22, 0x46, 0x2f, 0x35, 0xb8, 0xe8, 0x1c, 0xba, 0xcc, 0x87, 0x22, 0x08, 0x00,
    /* IP */
    0x45, 0x00,             // version & hdr len & tos
    0x00, 0x2c,             // total length
    0xff, 0xff,             // ip id
    0x20, 0x00,             // flags & frag offset
    0x7f, 0x06,             // ttl & protocol
    0x4d, 0x8b,             // checksum
    0xc0, 0xa8, 0x24, 0x67, // src addr
    0xc0, 0xa8, 0x28, 0x89, // dst addr
    /* TCP */
    0xf4, 0xa5,             // src port
    0x27, 0x0f,             // dst port
    0x91, 0x07, 0x24, 0x87, // seq
    0x03, 0xd5, 0x18, 0xe7, // ack
    0x50, 0x18,             // data offset & flags
    0xff, 0x00,             // windows
    0x5e, 0x92,             // checksum
    0x00, 0x00,             // urgent pointer
    /* DATA */
    0x31, 0x32, 0x33, 0x0a,
    /* Padding */
    0x00, 0x00};

/*
 * Frame 5: 60 bytes on wire (480 bits), 60 bytes captured (480 bits)
 * Ethernet II, Src: Fortinet_cc:87:22 (e8:1c:ba:cc:87:22), Dst: EvocInte_2f:35:b8 (00:22:46:2f:35:b8)
 *     Destination: EvocInte_2f:35:b8 (00:22:46:2f:35:b8)
 *         Address: EvocInte_2f:35:b8 (00:22:46:2f:35:b8)
 *         .... ..0. .... .... .... .... = LG bit: Globally unique address (factory default)
 *         .... ...0 .... .... .... .... = IG bit: Individual address (unicast)
 *     Source: Fortinet_cc:87:22 (e8:1c:ba:cc:87:22)
 *         Address: Fortinet_cc:87:22 (e8:1c:ba:cc:87:22)
 *         .... ..0. .... .... .... .... = LG bit: Globally unique address (factory default)
 *         .... ...0 .... .... .... .... = IG bit: Individual address (unicast)
 *     Type: IPv4 (0x0800)
 *     Padding: 0000
 * Internet Protocol Version 4, Src: 192.168.36.103, Dst: 192.168.40.137
 *     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: 44
 *     Identification: 0xffff (65535)
 *     000. .... = Flags: 0x0
 *         0... .... = Reserved bit: Not set
 *         .0.. .... = Don't fragment: Not set
 *         ..0. .... = More fragments: Not set
 *     ...0 0000 0000 0011 = Fragment Offset: 24
 *     Time to Live: 127
 *     Protocol: TCP (6)
 *     Header Checksum: 0x6d88 [correct]
 *     [Header checksum status: Good]
 *     [Calculated Checksum: 0x6d88]
 *     Source Address: 192.168.36.103
 *     Destination Address: 192.168.40.137
 *     [2 IPv4 Fragments (48 bytes): #4(24), #5(24)]
 *         [Frame: 4, payload: 0-23 (24 bytes)]
 *         [Frame: 5, payload: 24-47 (24 bytes)]
 *         [Fragment count: 2]
 *         [Reassembled IPv4 length: 48]
 *         [Reassembled IPv4 data: f4a5270f9107248703d518e75018ff005e9200003132330af4a5270f9107248b03d518e7…]
 * Transmission Control Protocol, Src Port: 62629, Dst Port: 9999, Seq: 1, Ack: 1, Len: 28
 *     Source Port: 62629
 *     Destination Port: 9999
 *     [Stream index: 0]
 *     [Conversation completeness: Complete, WITH_DATA (31)]
 *     [TCP Segment Len: 28]
 *     Sequence Number: 1    (relative sequence number)
 *     Sequence Number (raw): 2433164423
 *     [Next Sequence Number: 29    (relative sequence number)]
 *     Acknowledgment Number: 1    (relative ack number)
 *     Acknowledgment number (raw): 64297191
 *     0101 .... = Header Length: 20 bytes (5)
 *     Flags: 0x018 (PSH, 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
 *         .... .... 1... = Push: Set
 *         .... .... .0.. = Reset: Not set
 *         .... .... ..0. = Syn: Not set
 *         .... .... ...0 = Fin: Not set
 *         [TCP Flags: ·······AP···]
 *     Window: 65280
 *     [Calculated window size: 65280]
 *     [Window size scaling factor: -2 (no window scaling used)]
 *     Checksum: 0x5e92 [correct]
 *     [Checksum Status: Good]
 *     [Calculated Checksum: 0x5e92]
 *     Urgent Pointer: 0
 *     [Timestamps]
 *         [Time since first frame in this TCP stream: 6.525912000 seconds]
 *         [Time since previous frame in this TCP stream: 6.525083000 seconds]
 *     [SEQ/ACK analysis]
 *         [iRTT: 0.000829000 seconds]
 *         [Bytes in flight: 28]
 *         [Bytes sent since last PSH flag: 28]
 *     TCP payload (28 bytes)
 * Data (28 bytes)
 *     Data: 3132330af4a5270f9107248b03d518e75018ff00301600006162630a
 *     [Length: 28]
 */

unsigned char frag2[] = {
    /* ETH */
    0x00, 0x22, 0x46, 0x2f, 0x35, 0xb8, 0xe8, 0x1c, 0xba, 0xcc, 0x87, 0x22, 0x08, 0x00,
    /* IP */
    0x45, 0x00,             // version & hdr len & tos
    0x00, 0x2c,             // total length
    0xff, 0xff,             // ip id
    0x00, 0x03,             // flags & frag offset
    0x7f, 0x06,             // ttl & protocol
    0x6d, 0x88,             // checksum
    0xc0, 0xa8, 0x24, 0x67, // src addr
    0xc0, 0xa8, 0x28, 0x89, // dst addr
    /* DATA */
    0xf4, 0xa5, 0x27, 0x0f, 0x91, 0x07, 0x24, 0x8b, 0x03, 0xd5, 0x18, 0xe7, 0x50, 0x18, 0xff, 0x00, 0x30, 0x16, 0x00, 0x00, 0x61, 0x62, 0x63, 0x0a,
    /* Padding */
    0x00, 0x00};

unsigned char expect[] = {
    /* ETH */
    0x00, 0x22, 0x46, 0x2f, 0x35, 0xb8, 0xe8, 0x1c, 0xba, 0xcc, 0x87, 0x22, 0x08, 0x00,
    /* IP */
    0x45, 0x00,             // version & hdr len & tos
    0x00, 0x44,             // total length         --------- need update ---------
    0xff, 0xff,             // ip id
    0x00, 0x00,             // flags & frag offset  --------- need update ---------
    0x7f, 0x06,             // ttl & protocol
    0x6d, 0x73,             // checksum             --------- need update ---------
    0xc0, 0xa8, 0x24, 0x67, // src addr
    0xc0, 0xa8, 0x28, 0x89, // dst addr
    /* TCP */
    0xf4, 0xa5,             // src port
    0x27, 0x0f,             // dst port
    0x91, 0x07, 0x24, 0x87, // seq
    0x03, 0xd5, 0x18, 0xe7, // ack
    0x50, 0x18,             // data offset & flags
    0xff, 0x00,             // windows
    0x5e, 0x92,             // checksum
    0x00, 0x00,             // urgent pointer
    /* DATA */
    0x31, 0x32, 0x33, 0x0a,                                                                                                                        // data from frag1
    0xf4, 0xa5, 0x27, 0x0f, 0x91, 0x07, 0x24, 0x8b, 0x03, 0xd5, 0x18, 0xe7, 0x50, 0x18, 0xff, 0x00, 0x30, 0x16, 0x00, 0x00, 0x61, 0x62, 0x63, 0x0a // data from frag2
};

#if 1
TEST(IPV4_REASSEMBLE, PADDING_ORDER)
{
    struct packet pkt;
    struct packet *new_pkt;
    const struct packet_layer *layer;
    struct ip_reassembly *assy;
    struct ip_reassembly_options opts = {
        .enable = true,
        .timeout = 1,
        .bucket_entries = 16,
        .bucket_num = 8,
    };

    assy = ip_reassembly_new(&opts);
    EXPECT_TRUE(assy != NULL);

    check_stat(ip_reassembly_stat(assy),
               0, 0, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // frag1
    packet_parse(&pkt, (const char *)frag1, sizeof(frag1));
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt == NULL);

    check_stat(ip_reassembly_stat(assy),
               1, 1, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // frag2
    packet_parse(&pkt, (const char *)frag2, sizeof(frag2));
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt);

    check_stat(ip_reassembly_stat(assy),
               2, 1, 1, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // check packet
    // packet_print_str(new_pkt);
    EXPECT_TRUE(new_pkt->data_len == 14 /* ETH */ + 20 /* IPv4 */ + 20 /* TCP */ + 28 /* DATA */);
    EXPECT_TRUE(new_pkt->data_len == sizeof(expect));
    EXPECT_TRUE(memcmp(new_pkt->data_ptr, expect, new_pkt->data_len) == 0);

    // check IPv4
    layer = packet_get_innermost_layer(new_pkt, LAYER_TYPE_IPV4);
    EXPECT_TRUE(layer);
    struct ip *hdr = (struct ip *)layer->hdr_ptr;
    EXPECT_TRUE(ipv4_hdr_get_version(hdr) == 4);
    EXPECT_TRUE(ipv4_hdr_get_hdr_len(hdr) == 20 /* IPv4 */);
    EXPECT_TRUE(ipv4_hdr_get_tos(hdr) == 0);
    EXPECT_TRUE(ipv4_hdr_get_total_len(hdr) == 20 /* IPv4 */ + 20 /* TCP */ + 28 /* DATA */);
    EXPECT_TRUE(ipv4_hdr_get_ipid(hdr) == 0xffff);
    EXPECT_TRUE(ipv4_hdr_get_flags(hdr) == 0x0);
    EXPECT_TRUE(ipv4_hdr_get_frag_offset(hdr) == 0);
    EXPECT_TRUE(ipv4_hdr_get_ttl(hdr) == 127);
    EXPECT_TRUE(ipv4_hdr_get_proto(hdr) == 6);
    EXPECT_TRUE(ipv4_hdr_get_checksum(hdr) == 0x6d73); // NOTE this is correct checksum
    EXPECT_TRUE(ipv4_hdr_get_src_addr(hdr) == 0xc0a82467);
    EXPECT_TRUE(ipv4_hdr_get_dst_addr(hdr) == 0xc0a82889);
    EXPECT_TRUE(ipv4_hdr_get_opt_len(hdr) == 0);
    EXPECT_TRUE(ipv4_hdr_get_opt_data(hdr) == NULL);

    // check TCP
    layer = packet_get_innermost_layer(new_pkt, LAYER_TYPE_TCP);
    EXPECT_TRUE(layer);
    struct tcphdr *tcp_hdr = (struct tcphdr *)layer->hdr_ptr;
    EXPECT_TRUE(tcp_hdr_get_src_port(tcp_hdr) == 62629);
    EXPECT_TRUE(tcp_hdr_get_dst_port(tcp_hdr) == 9999);
    EXPECT_TRUE(tcp_hdr_get_seq(tcp_hdr) == 2433164423);
    EXPECT_TRUE(tcp_hdr_get_ack(tcp_hdr) == 64297191);
    EXPECT_TRUE(tcp_hdr_get_hdr_len(tcp_hdr) == 20);
    EXPECT_TRUE(tcp_hdr_get_flags(tcp_hdr) == 0x018);
    EXPECT_TRUE(tcp_hdr_get_window(tcp_hdr) == 65280);
    EXPECT_TRUE(tcp_hdr_get_checksum(tcp_hdr) == 0x5e92);
    EXPECT_TRUE(tcp_hdr_get_urg_ptr(tcp_hdr) == 0);
    EXPECT_TRUE(tcp_hdr_get_opt_len(tcp_hdr) == 0);
    EXPECT_TRUE(tcp_hdr_get_opt_data(tcp_hdr) == NULL);

    // free packet
    packet_free(new_pkt);

    ip_reassembly_free(assy);
}
#endif

#if 1
TEST(IPV4_REASSEMBLE, PADDING_UNORDER)
{
    struct packet pkt;
    struct packet *new_pkt;
    const struct packet_layer *layer;
    struct ip_reassembly *assy;
    struct ip_reassembly_options opts = {
        .enable = true,
        .timeout = 1,
        .bucket_entries = 16,
        .bucket_num = 8,
    };

    assy = ip_reassembly_new(&opts);
    EXPECT_TRUE(assy != NULL);

    check_stat(ip_reassembly_stat(assy),
               0, 0, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // frag2
    packet_parse(&pkt, (const char *)frag2, sizeof(frag2));
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt == NULL);

    check_stat(ip_reassembly_stat(assy),
               1, 1, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // frag1
    packet_parse(&pkt, (const char *)frag1, sizeof(frag1));
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt);

    check_stat(ip_reassembly_stat(assy),
               2, 1, 1, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // check packet
    // packet_print_str(new_pkt);
    EXPECT_TRUE(new_pkt->data_len == 14 /* ETH */ + 20 /* IPv4 */ + 20 /* TCP */ + 28 /* DATA */);
    EXPECT_TRUE(new_pkt->data_len == sizeof(expect));
    EXPECT_TRUE(memcmp(new_pkt->data_ptr, expect, new_pkt->data_len) == 0);

    // check IPv4
    layer = packet_get_innermost_layer(new_pkt, LAYER_TYPE_IPV4);
    EXPECT_TRUE(layer);
    struct ip *hdr = (struct ip *)layer->hdr_ptr;
    EXPECT_TRUE(ipv4_hdr_get_version(hdr) == 4);
    EXPECT_TRUE(ipv4_hdr_get_hdr_len(hdr) == 20 /* IPv4 */);
    EXPECT_TRUE(ipv4_hdr_get_tos(hdr) == 0);
    EXPECT_TRUE(ipv4_hdr_get_total_len(hdr) == 20 /* IPv4 */ + 20 /* TCP */ + 28 /* DATA */);
    EXPECT_TRUE(ipv4_hdr_get_ipid(hdr) == 0xffff);
    EXPECT_TRUE(ipv4_hdr_get_flags(hdr) == 0x0);
    EXPECT_TRUE(ipv4_hdr_get_frag_offset(hdr) == 0);
    EXPECT_TRUE(ipv4_hdr_get_ttl(hdr) == 127);
    EXPECT_TRUE(ipv4_hdr_get_proto(hdr) == 6);
    EXPECT_TRUE(ipv4_hdr_get_checksum(hdr) == 0x6d73); // NOTE this is correct checksum
    EXPECT_TRUE(ipv4_hdr_get_src_addr(hdr) == 0xc0a82467);
    EXPECT_TRUE(ipv4_hdr_get_dst_addr(hdr) == 0xc0a82889);
    EXPECT_TRUE(ipv4_hdr_get_opt_len(hdr) == 0);
    EXPECT_TRUE(ipv4_hdr_get_opt_data(hdr) == NULL);

    // check TCP
    layer = packet_get_innermost_layer(new_pkt, LAYER_TYPE_TCP);
    EXPECT_TRUE(layer);
    struct tcphdr *tcp_hdr = (struct tcphdr *)layer->hdr_ptr;
    EXPECT_TRUE(tcp_hdr_get_src_port(tcp_hdr) == 62629);
    EXPECT_TRUE(tcp_hdr_get_dst_port(tcp_hdr) == 9999);
    EXPECT_TRUE(tcp_hdr_get_seq(tcp_hdr) == 2433164423);
    EXPECT_TRUE(tcp_hdr_get_ack(tcp_hdr) == 64297191);
    EXPECT_TRUE(tcp_hdr_get_hdr_len(tcp_hdr) == 20);
    EXPECT_TRUE(tcp_hdr_get_flags(tcp_hdr) == 0x018);
    EXPECT_TRUE(tcp_hdr_get_window(tcp_hdr) == 65280);
    EXPECT_TRUE(tcp_hdr_get_checksum(tcp_hdr) == 0x5e92);
    EXPECT_TRUE(tcp_hdr_get_urg_ptr(tcp_hdr) == 0);
    EXPECT_TRUE(tcp_hdr_get_opt_len(tcp_hdr) == 0);
    EXPECT_TRUE(tcp_hdr_get_opt_data(tcp_hdr) == NULL);

    // free packet
    packet_free(new_pkt);

    ip_reassembly_free(assy);
}
#endif

#if 1
TEST(IPV4_REASSEMBLE, EXPIRE)
{
    struct packet pkt;
    struct packet *new_pkt;
    struct ip_reassembly *assy;
    struct ip_reassembly_options opts = {
        .enable = true,
        .timeout = 1,
        .bucket_entries = 16,
        .bucket_num = 8,
    };

    assy = ip_reassembly_new(&opts);
    EXPECT_TRUE(assy != NULL);

    check_stat(ip_reassembly_stat(assy),
               0, 0, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // frag1
    packet_parse(&pkt, (const char *)frag1, sizeof(frag1));
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt == NULL);

    check_stat(ip_reassembly_stat(assy),
               1, 1, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // frag2
    packet_parse(&pkt, (const char *)frag2, sizeof(frag2));
    new_pkt = ip_reassembly_packet(assy, &pkt, 2);
    EXPECT_TRUE(new_pkt == NULL);

    check_stat(ip_reassembly_stat(assy),
               2, 2, 1, 1,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // free packet
    packet_free(new_pkt);

    ip_reassembly_free(assy);
}
#endif

#if 1
TEST(IPV4_REASSEMBLE, DUP_FIRST_FRAG)
{
    struct packet pkt;
    struct packet *new_pkt;
    const struct packet_layer *layer;
    struct ip_reassembly *assy;
    struct ip_reassembly_options opts = {
        .enable = true,
        .timeout = 1,
        .bucket_entries = 16,
        .bucket_num = 8,
    };

    assy = ip_reassembly_new(&opts);
    EXPECT_TRUE(assy != NULL);

    check_stat(ip_reassembly_stat(assy),
               0, 0, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // frag1
    packet_parse(&pkt, (const char *)frag1, sizeof(frag1));
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt == NULL);

    check_stat(ip_reassembly_stat(assy),
               1, 1, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // frag1
    packet_parse(&pkt, (const char *)frag1, sizeof(frag1));
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt == NULL);

    check_stat(ip_reassembly_stat(assy),
               2, 1, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 1, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // frag2
    packet_parse(&pkt, (const char *)frag2, sizeof(frag2));
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt);

    check_stat(ip_reassembly_stat(assy),
               3, 1, 1, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 1, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // check packet
    // packet_print_str(new_pkt);
    EXPECT_TRUE(new_pkt->data_len == 14 /* ETH */ + 20 /* IPv4 */ + 20 /* TCP */ + 28 /* DATA */);
    EXPECT_TRUE(new_pkt->data_len == sizeof(expect));
    EXPECT_TRUE(memcmp(new_pkt->data_ptr, expect, new_pkt->data_len) == 0);

    // check IPv4
    layer = packet_get_innermost_layer(new_pkt, LAYER_TYPE_IPV4);
    EXPECT_TRUE(layer);
    struct ip *hdr = (struct ip *)layer->hdr_ptr;
    EXPECT_TRUE(ipv4_hdr_get_version(hdr) == 4);
    EXPECT_TRUE(ipv4_hdr_get_hdr_len(hdr) == 20 /* IPv4 */);
    EXPECT_TRUE(ipv4_hdr_get_tos(hdr) == 0);
    EXPECT_TRUE(ipv4_hdr_get_total_len(hdr) == 20 /* IPv4 */ + 20 /* TCP */ + 28 /* DATA */);
    EXPECT_TRUE(ipv4_hdr_get_ipid(hdr) == 0xffff);
    EXPECT_TRUE(ipv4_hdr_get_flags(hdr) == 0x0);
    EXPECT_TRUE(ipv4_hdr_get_frag_offset(hdr) == 0);
    EXPECT_TRUE(ipv4_hdr_get_ttl(hdr) == 127);
    EXPECT_TRUE(ipv4_hdr_get_proto(hdr) == 6);
    EXPECT_TRUE(ipv4_hdr_get_checksum(hdr) == 0x6d73); // NOTE this is correct checksum
    EXPECT_TRUE(ipv4_hdr_get_src_addr(hdr) == 0xc0a82467);
    EXPECT_TRUE(ipv4_hdr_get_dst_addr(hdr) == 0xc0a82889);
    EXPECT_TRUE(ipv4_hdr_get_opt_len(hdr) == 0);
    EXPECT_TRUE(ipv4_hdr_get_opt_data(hdr) == NULL);

    // check TCP
    layer = packet_get_innermost_layer(new_pkt, LAYER_TYPE_TCP);
    EXPECT_TRUE(layer);
    struct tcphdr *tcp_hdr = (struct tcphdr *)layer->hdr_ptr;
    EXPECT_TRUE(tcp_hdr_get_src_port(tcp_hdr) == 62629);
    EXPECT_TRUE(tcp_hdr_get_dst_port(tcp_hdr) == 9999);
    EXPECT_TRUE(tcp_hdr_get_seq(tcp_hdr) == 2433164423);
    EXPECT_TRUE(tcp_hdr_get_ack(tcp_hdr) == 64297191);
    EXPECT_TRUE(tcp_hdr_get_hdr_len(tcp_hdr) == 20);
    EXPECT_TRUE(tcp_hdr_get_flags(tcp_hdr) == 0x018);
    EXPECT_TRUE(tcp_hdr_get_window(tcp_hdr) == 65280);
    EXPECT_TRUE(tcp_hdr_get_checksum(tcp_hdr) == 0x5e92);
    EXPECT_TRUE(tcp_hdr_get_urg_ptr(tcp_hdr) == 0);
    EXPECT_TRUE(tcp_hdr_get_opt_len(tcp_hdr) == 0);
    EXPECT_TRUE(tcp_hdr_get_opt_data(tcp_hdr) == NULL);

    // free packet
    packet_free(new_pkt);

    ip_reassembly_free(assy);
}
#endif

#if 1
TEST(IPV4_REASSEMBLE, DUP_LAST_FRAG)
{
    struct packet pkt;
    struct packet *new_pkt;
    const struct packet_layer *layer;
    struct ip_reassembly *assy;
    struct ip_reassembly_options opts = {
        .enable = true,
        .timeout = 1,
        .bucket_entries = 16,
        .bucket_num = 8,
    };

    assy = ip_reassembly_new(&opts);
    EXPECT_TRUE(assy != NULL);

    check_stat(ip_reassembly_stat(assy),
               0, 0, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // frag2
    packet_parse(&pkt, (const char *)frag2, sizeof(frag2));
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt == NULL);

    check_stat(ip_reassembly_stat(assy),
               1, 1, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // frag2
    packet_parse(&pkt, (const char *)frag2, sizeof(frag2));
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt == NULL);

    check_stat(ip_reassembly_stat(assy),
               2, 1, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 1,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // frag1
    packet_parse(&pkt, (const char *)frag1, sizeof(frag1));
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt);

    check_stat(ip_reassembly_stat(assy),
               3, 1, 1, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 1,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // check packet
    // packet_print_str(new_pkt);
    EXPECT_TRUE(new_pkt->data_len == 14 /* ETH */ + 20 /* IPv4 */ + 20 /* TCP */ + 28 /* DATA */);
    EXPECT_TRUE(new_pkt->data_len == sizeof(expect));
    EXPECT_TRUE(memcmp(new_pkt->data_ptr, expect, new_pkt->data_len) == 0);

    // check IPv4
    layer = packet_get_innermost_layer(new_pkt, LAYER_TYPE_IPV4);
    EXPECT_TRUE(layer);
    struct ip *hdr = (struct ip *)layer->hdr_ptr;
    EXPECT_TRUE(ipv4_hdr_get_version(hdr) == 4);
    EXPECT_TRUE(ipv4_hdr_get_hdr_len(hdr) == 20 /* IPv4 */);
    EXPECT_TRUE(ipv4_hdr_get_tos(hdr) == 0);
    EXPECT_TRUE(ipv4_hdr_get_total_len(hdr) == 20 /* IPv4 */ + 20 /* TCP */ + 28 /* DATA */);
    EXPECT_TRUE(ipv4_hdr_get_ipid(hdr) == 0xffff);
    EXPECT_TRUE(ipv4_hdr_get_flags(hdr) == 0x0);
    EXPECT_TRUE(ipv4_hdr_get_frag_offset(hdr) == 0);
    EXPECT_TRUE(ipv4_hdr_get_ttl(hdr) == 127);
    EXPECT_TRUE(ipv4_hdr_get_proto(hdr) == 6);
    EXPECT_TRUE(ipv4_hdr_get_checksum(hdr) == 0x6d73); // NOTE this is correct checksum
    EXPECT_TRUE(ipv4_hdr_get_src_addr(hdr) == 0xc0a82467);
    EXPECT_TRUE(ipv4_hdr_get_dst_addr(hdr) == 0xc0a82889);
    EXPECT_TRUE(ipv4_hdr_get_opt_len(hdr) == 0);
    EXPECT_TRUE(ipv4_hdr_get_opt_data(hdr) == NULL);

    // check TCP
    layer = packet_get_innermost_layer(new_pkt, LAYER_TYPE_TCP);
    EXPECT_TRUE(layer);
    struct tcphdr *tcp_hdr = (struct tcphdr *)layer->hdr_ptr;
    EXPECT_TRUE(tcp_hdr_get_src_port(tcp_hdr) == 62629);
    EXPECT_TRUE(tcp_hdr_get_dst_port(tcp_hdr) == 9999);
    EXPECT_TRUE(tcp_hdr_get_seq(tcp_hdr) == 2433164423);
    EXPECT_TRUE(tcp_hdr_get_ack(tcp_hdr) == 64297191);
    EXPECT_TRUE(tcp_hdr_get_hdr_len(tcp_hdr) == 20);
    EXPECT_TRUE(tcp_hdr_get_flags(tcp_hdr) == 0x018);
    EXPECT_TRUE(tcp_hdr_get_window(tcp_hdr) == 65280);
    EXPECT_TRUE(tcp_hdr_get_checksum(tcp_hdr) == 0x5e92);
    EXPECT_TRUE(tcp_hdr_get_urg_ptr(tcp_hdr) == 0);
    EXPECT_TRUE(tcp_hdr_get_opt_len(tcp_hdr) == 0);
    EXPECT_TRUE(tcp_hdr_get_opt_data(tcp_hdr) == NULL);

    // free packet
    packet_free(new_pkt);

    ip_reassembly_free(assy);
}
#endif

#if 1
TEST(IPV4_REASSEMBLE, FULL)
{
    struct packet pkt;
    struct packet *new_pkt;
    struct ip_reassembly *assy;
    struct ip_reassembly_options opts = {
        .enable = true,
        .timeout = 1,
        .bucket_entries = 1,
        .bucket_num = 1,
    };

    assy = ip_reassembly_new(&opts);
    EXPECT_TRUE(assy != NULL);

    check_stat(ip_reassembly_stat(assy),
               0, 0, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    char dup_frag[sizeof(frag1)] = {0};
    memcpy(dup_frag, frag1, sizeof(frag1));
    packet_parse(&pkt, (const char *)dup_frag, sizeof(dup_frag));

    // flow1
    packet_set_ipv4_src_addr(&pkt, 1);
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt == NULL);

    check_stat(ip_reassembly_stat(assy),
               1, 1, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // flow2
    packet_set_ipv4_src_addr(&pkt, 2);
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt == NULL);

    check_stat(ip_reassembly_stat(assy),
               2, 2, 0, 0,        // ip4: find, add, del, timeout
               0, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // flow3
    packet_set_ipv4_src_addr(&pkt, 3);
    new_pkt = ip_reassembly_packet(assy, &pkt, 1);
    EXPECT_TRUE(new_pkt == NULL);

    check_stat(ip_reassembly_stat(assy),
               3, 2, 0, 0,        // ip4: find, add, del, timeout
               1, 0, 0, 0, 0, 0,  // ip4: nospace, overlap, many frag, invalid length, dup first frag, dup last frag
               0, 0, 0, 0,        // ip6: find, add, del, timeout
               0, 0, 0, 0, 0, 0); // ip6: nospace, overlap, many frag, invalid length, dup first frag, dup last frag

    // free packet
    packet_free(new_pkt);

    ip_reassembly_free(assy);
}
#endif

#if 1
TEST(IPV4_REASSEMBLE, OVERLAP)
{
    // TEST ON IPv6
}
#endif

int main(int argc, char **argv)
{
    ::testing::InitGoogleTest(&argc, argv);
    return RUN_ALL_TESTS();
}