summaryrefslogtreecommitdiff
path: root/sample/test_ipv4_options.c
diff options
context:
space:
mode:
Diffstat (limited to 'sample/test_ipv4_options.c')
-rw-r--r--sample/test_ipv4_options.c252
1 files changed, 252 insertions, 0 deletions
diff --git a/sample/test_ipv4_options.c b/sample/test_ipv4_options.c
new file mode 100644
index 0000000..71b5335
--- /dev/null
+++ b/sample/test_ipv4_options.c
@@ -0,0 +1,252 @@
+/*
+ * Regression test for bugs in ipv4 ip_offset and h_len handling, such as
+ * http://bugs.debian.org/cgi-bin/bugreport.cgi?bug=418975
+ *
+ * Copyright (c) 2009 Sam Roberts <sroberts@wurldtech.com>
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ *
+ */
+#if (HAVE_CONFIG_H)
+#include "../include/config.h"
+#endif
+#include "./libnet_test.h"
+
+#include <assert.h>
+
+static void assert_eq_(long have, long want, const char* file, int line) {
+ if(have != want) {
+ printf("%s:%d: fail - have %ld want %ld\n", file, line, have, want);
+ abort();
+ }
+}
+#define assert_eq(have, want) assert_eq_(have, want, __FILE__, __LINE__)
+
+
+static void print_pblocks(libnet_t* l)
+{
+ libnet_pblock_t* p = l->protocol_blocks;
+
+ while(p) {
+ printf(" tag %2d flags %d type %20s/%#x buf %p b_len %2u h_len %2u copied %2u\n",
+ p->ptag, p->flags,
+ libnet_diag_dump_pblock_type(p->type), p->type,
+ p->buf, p->b_len, p->h_len, p->copied);
+ p = p->next;
+ }
+ printf(" link_offset %d aligner %d total_size %u nblocks %d\n",
+ l->link_offset, l->aligner, l->total_size, l->n_pblocks);
+
+}
+
+static void ptag_error(libnet_t* l, int ptag)
+{
+ if(ptag <= 0) {
+ printf("error: %s\n", libnet_geterror(l));
+ }
+ assert(ptag > 0);
+}
+
+static int build_ipo(libnet_t* l, libnet_ptag_t ptag, int payload_s)
+{
+ uint8_t* payload = malloc(payload_s);
+ assert(payload);
+ memset(payload, '\x88', payload_s);
+
+ ptag = libnet_build_ipv4_options(payload, payload_s, l, ptag);
+
+ ptag_error(l, ptag);
+
+ free(payload);
+
+ return ptag;
+}
+
+static int build_ipv4(libnet_t* l, libnet_ptag_t ptag, int payload_s, int ip_len)
+{
+ u_long src_ip = 0xf101f1f1;
+ u_long dst_ip = 0xf102f1f1;
+ uint8_t* payload = malloc(payload_s);
+ assert(payload);
+ memset(payload, '\x99', payload_s);
+
+ if(!ip_len) {
+ ip_len = LIBNET_IPV4_H + payload_s;
+ }
+
+ ptag = libnet_build_ipv4(
+ ip_len, /* length */
+ 0, /* TOS */
+ 0xbbbb, /* IP ID */
+ 0, /* IP Frag */
+ 0xcc, /* TTL */
+ IPPROTO_UDP, /* protocol */
+ 0, /* checksum */
+ src_ip, /* source IP */
+ dst_ip, /* destination IP */
+ payload_s ? payload : NULL, /* payload */
+ payload_s, /* payload size */
+ l, /* libnet handle */
+ ptag); /* libnet id */
+
+ ptag_error(l, ptag);
+
+ free(payload);
+
+ return ptag;
+}
+
+static int build_ethernet(libnet_t* l, libnet_ptag_t ptag)
+{
+ uint8_t enet_src[6] = {0x11, 0x11, 0x11, 0x11, 0x11, 0x11};
+ uint8_t enet_dst[6] = {0x22, 0x22, 0x22, 0x22, 0x22, 0x22};
+
+ ptag = libnet_build_ethernet(
+ enet_dst, /* ethernet destination */
+ enet_src, /* ethernet source */
+ ETHERTYPE_IP, /* protocol type */
+ NULL, /* payload */
+ 0, /* payload size */
+ l, /* libnet handle */
+ ptag); /* libnet id */
+
+ ptag_error(l, ptag);
+
+ return ptag;
+}
+
+static
+void assert_lengths(libnet_t* l, int ip_len, int ip_ihl, int payload_s)
+{
+ uint8_t* pkt1 = NULL;
+ uint32_t pkt1_sz = 0;
+ struct libnet_ipv4_hdr* h1;
+ uint8_t* payload = NULL;
+
+
+ int r = libnet_pblock_coalesce(l, &pkt1, &pkt1_sz);
+ assert(r >= 0);
+
+ print_pblocks(l);
+
+ libnet_diag_dump_hex(pkt1, 14, 1, stdout);
+ libnet_diag_dump_hex(pkt1+14, pkt1_sz-14, 1, stdout);
+
+ /* check ip IHL value, total ip pkt length, and options value */
+ h1 = (struct libnet_ipv4_hdr*) (pkt1+14);
+ assert_eq(h1->ip_hl, ip_ihl);
+ assert_eq(ntohs(h1->ip_len), ip_len);
+
+ payload = ((uint8_t*) h1) + ip_ihl * 4;
+ if(payload_s > 0) {
+ assert(payload[0] == (uint8_t)'\x99');
+ assert(payload[payload_s-1] == (uint8_t)'\x99');
+ }
+}
+
+int
+main(int argc, char *argv[])
+{
+ libnet_t *l;
+ char *device = "eth0";
+ char errbuf[LIBNET_ERRBUF_SIZE];
+ libnet_ptag_t ipo_ptag = 0;
+ libnet_ptag_t ip_ptag = 0;
+ libnet_ptag_t eth_ptag = 0;
+ int ip_len = 0;
+
+ l = libnet_init( LIBNET_LINK, device, errbuf);
+
+ assert(l);
+
+ printf("Packet: options=4, payload=0\n");
+
+ ip_len = 20 + 4 + 0; /* ip + options + payload */
+ ipo_ptag = build_ipo(l, ipo_ptag, 4);
+ ip_ptag = build_ipv4(l, ip_ptag, 0, 24);
+ eth_ptag = build_ethernet(l, eth_ptag);
+
+ assert_lengths(l, 24, 6, 0);
+
+ ipo_ptag = ip_ptag = eth_ptag = 0;
+
+ libnet_clear_packet(l);
+
+ printf("Packet: options=3, payload=1\n");
+
+ ip_len = 20 + 4 + 1; /* ip + options + payload */
+ ipo_ptag = build_ipo(l, ipo_ptag, 3);
+ ip_ptag = build_ipv4(l, ip_ptag, 1, 25);
+ eth_ptag = build_ethernet(l, eth_ptag);
+
+ assert_lengths(l, 25, 6, 1);
+
+ ipo_ptag = ip_ptag = eth_ptag = 0;
+
+ libnet_clear_packet(l);
+
+ printf("Packet: options=3, payload=1\n");
+
+ ip_len = 20 + 4 + 1; /* ip + options + payload */
+ ipo_ptag = build_ipo(l, ipo_ptag, 3);
+ ip_ptag = build_ipv4(l, ip_ptag, 1, ip_len);
+ eth_ptag = build_ethernet(l, eth_ptag);
+
+ assert_lengths(l, 25, 6, 1);
+
+ printf("... modify -> options=40\n");
+
+ ip_len = 20 + 40 + 1; /* ip + options + payload */
+ ipo_ptag = build_ipo(l, ipo_ptag, 40);
+
+ assert_lengths(l, ip_len, 15, 1);
+
+ printf("... modify -> options=0\n");
+
+ ip_len = 20 + 0 + 1; /* ip + options + payload */
+ ipo_ptag = build_ipo(l, ipo_ptag, 0);
+
+ assert_lengths(l, ip_len, 5, 1);
+
+ printf("... modify -> options=5\n");
+
+ ip_len = 20 + 8 + 1; /* ip + options + payload */
+ ipo_ptag = build_ipo(l, ipo_ptag, 5);
+
+ assert_lengths(l, ip_len, 7, 1);
+
+ printf("... modify -> ip_payload=5\n");
+
+ ip_len = 20 + 8 + 5; /* ip + options + payload */
+ ip_ptag = build_ipv4(l, ip_ptag, 5, ip_len);
+
+ assert_lengths(l, ip_len, 7, 1);
+
+ ipo_ptag = ip_ptag = eth_ptag = 0;
+
+ libnet_clear_packet(l);
+
+
+ return (EXIT_SUCCESS);
+}
+
generated by cgit v1.2.3 (git 2.25.1) at 2024-06-25 23:02:17 +0000