1 #include "test_ip4.h" 2 3 #include "lwip/ip4.h" 4 #include "lwip/inet_chksum.h" 5 #include "lwip/stats.h" 6 #include "lwip/prot/ip.h" 7 #include "lwip/prot/ip4.h" 8 9 #include "lwip/tcpip.h" 10 11 #if !LWIP_IPV4 || !IP_REASSEMBLY || !MIB2_STATS || !IPFRAG_STATS 12 #error "This tests needs LWIP_IPV4, IP_REASSEMBLY; MIB2- and IPFRAG-statistics enabled" 13 #endif 14 15 /* Helper functions */ 16 static void 17 create_ip4_input_fragment(u16_t ip_id, u16_t start, u16_t len, int last) 18 { 19 struct pbuf *p; 20 struct netif *input_netif = netif_list; /* just use any netif */ 21 fail_unless((start & 7) == 0); 22 fail_unless(((len & 7) == 0) || last); 23 fail_unless(input_netif != NULL); 24 25 p = pbuf_alloc(PBUF_RAW, len + sizeof(struct ip_hdr), PBUF_RAM); 26 fail_unless(p != NULL); 27 if (p != NULL) { 28 err_t err; 29 struct ip_hdr *iphdr = (struct ip_hdr *)p->payload; 30 IPH_VHL_SET(iphdr, 4, sizeof(struct ip_hdr) / 4); 31 IPH_TOS_SET(iphdr, 0); 32 IPH_LEN_SET(iphdr, lwip_htons(p->tot_len)); 33 IPH_ID_SET(iphdr, lwip_htons(ip_id)); 34 if (last) { 35 IPH_OFFSET_SET(iphdr, lwip_htons(start / 8)); 36 } else { 37 IPH_OFFSET_SET(iphdr, lwip_htons((start / 8) | IP_MF)); 38 } 39 IPH_TTL_SET(iphdr, 5); 40 IPH_PROTO_SET(iphdr, IP_PROTO_UDP); 41 IPH_CHKSUM_SET(iphdr, 0); 42 ip4_addr_copy(iphdr->src, *netif_ip4_addr(input_netif)); 43 iphdr->src.addr = lwip_htonl(lwip_htonl(iphdr->src.addr) + 1); 44 ip4_addr_copy(iphdr->dest, *netif_ip4_addr(input_netif)); 45 IPH_CHKSUM_SET(iphdr, inet_chksum(iphdr, sizeof(struct ip_hdr))); 46 47 err = ip4_input(p, input_netif); 48 if (err != ERR_OK) { 49 pbuf_free(p); 50 } 51 fail_unless(err == ERR_OK); 52 } 53 } 54 55 /* Setups/teardown functions */ 56 57 static void 58 ip4_setup(void) 59 { 60 lwip_check_ensure_no_alloc(SKIP_POOL(MEMP_SYS_TIMEOUT)); 61 } 62 63 static void 64 ip4_teardown(void) 65 { 66 if (netif_list->loop_first != NULL) { 67 pbuf_free(netif_list->loop_first); 68 netif_list->loop_first = NULL; 69 } 70 netif_list->loop_last = NULL; 71 /* poll until all memory is released... */ 72 tcpip_thread_poll_one(); 73 lwip_check_ensure_no_alloc(SKIP_POOL(MEMP_SYS_TIMEOUT)); 74 } 75 76 77 /* Test functions */ 78 79 START_TEST(test_ip4_reass) 80 { 81 const u16_t ip_id = 128; 82 LWIP_UNUSED_ARG(_i); 83 84 memset(&lwip_stats.mib2, 0, sizeof(lwip_stats.mib2)); 85 86 create_ip4_input_fragment(ip_id, 8*200, 200, 1); 87 fail_unless(lwip_stats.ip_frag.recv == 1); 88 fail_unless(lwip_stats.ip_frag.err == 0); 89 fail_unless(lwip_stats.ip_frag.memerr == 0); 90 fail_unless(lwip_stats.ip_frag.drop == 0); 91 fail_unless(lwip_stats.mib2.ipreasmoks == 0); 92 93 create_ip4_input_fragment(ip_id, 0*200, 200, 0); 94 fail_unless(lwip_stats.ip_frag.recv == 2); 95 fail_unless(lwip_stats.ip_frag.err == 0); 96 fail_unless(lwip_stats.ip_frag.memerr == 0); 97 fail_unless(lwip_stats.ip_frag.drop == 0); 98 fail_unless(lwip_stats.mib2.ipreasmoks == 0); 99 100 create_ip4_input_fragment(ip_id, 1*200, 200, 0); 101 fail_unless(lwip_stats.ip_frag.recv == 3); 102 fail_unless(lwip_stats.ip_frag.err == 0); 103 fail_unless(lwip_stats.ip_frag.memerr == 0); 104 fail_unless(lwip_stats.ip_frag.drop == 0); 105 fail_unless(lwip_stats.mib2.ipreasmoks == 0); 106 107 create_ip4_input_fragment(ip_id, 2*200, 200, 0); 108 fail_unless(lwip_stats.ip_frag.recv == 4); 109 fail_unless(lwip_stats.ip_frag.err == 0); 110 fail_unless(lwip_stats.ip_frag.memerr == 0); 111 fail_unless(lwip_stats.ip_frag.drop == 0); 112 fail_unless(lwip_stats.mib2.ipreasmoks == 0); 113 114 create_ip4_input_fragment(ip_id, 3*200, 200, 0); 115 fail_unless(lwip_stats.ip_frag.recv == 5); 116 fail_unless(lwip_stats.ip_frag.err == 0); 117 fail_unless(lwip_stats.ip_frag.memerr == 0); 118 fail_unless(lwip_stats.ip_frag.drop == 0); 119 fail_unless(lwip_stats.mib2.ipreasmoks == 0); 120 121 create_ip4_input_fragment(ip_id, 4*200, 200, 0); 122 fail_unless(lwip_stats.ip_frag.recv == 6); 123 fail_unless(lwip_stats.ip_frag.err == 0); 124 fail_unless(lwip_stats.ip_frag.memerr == 0); 125 fail_unless(lwip_stats.ip_frag.drop == 0); 126 fail_unless(lwip_stats.mib2.ipreasmoks == 0); 127 128 create_ip4_input_fragment(ip_id, 7*200, 200, 0); 129 fail_unless(lwip_stats.ip_frag.recv == 7); 130 fail_unless(lwip_stats.ip_frag.err == 0); 131 fail_unless(lwip_stats.ip_frag.memerr == 0); 132 fail_unless(lwip_stats.ip_frag.drop == 0); 133 fail_unless(lwip_stats.mib2.ipreasmoks == 0); 134 135 create_ip4_input_fragment(ip_id, 6*200, 200, 0); 136 fail_unless(lwip_stats.ip_frag.recv == 8); 137 fail_unless(lwip_stats.ip_frag.err == 0); 138 fail_unless(lwip_stats.ip_frag.memerr == 0); 139 fail_unless(lwip_stats.ip_frag.drop == 0); 140 fail_unless(lwip_stats.mib2.ipreasmoks == 0); 141 142 create_ip4_input_fragment(ip_id, 5*200, 200, 0); 143 fail_unless(lwip_stats.ip_frag.recv == 9); 144 fail_unless(lwip_stats.ip_frag.err == 0); 145 fail_unless(lwip_stats.ip_frag.memerr == 0); 146 fail_unless(lwip_stats.ip_frag.drop == 0); 147 fail_unless(lwip_stats.mib2.ipreasmoks == 1); 148 } 149 END_TEST 150 151 152 /** Create the suite including all tests for this module */ 153 Suite * 154 ip4_suite(void) 155 { 156 testfunc tests[] = { 157 TESTFUNC(test_ip4_reass), 158 }; 159 return create_suite("IPv4", tests, sizeof(tests)/sizeof(testfunc), ip4_setup, ip4_teardown); 160 } 161