improve test

Author: haesbaert

quark-test.c

@@ -34,10 +34,13 @@
 #include "manpages.h"
 
 #define PATTERN "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890"
-//	"Nobody calls me Lebowski. You got the wrong guy. I'm the Dude, man."
 
-static int	bflag;	/* run bpf tests */
-static int	kflag;	/* run kprobe tests */
+struct udphdr {
+	u16 source;
+	u16 dest;
+	u16 len;
+	u16 check;
+};
 
 #define msleep(_x)	usleep((uint64_t)_x * 1000ULL)
 
@@ -47,6 +50,9 @@ enum {
 	GREEN
 };
 
+static int	bflag;	/* run bpf tests */
+static int	kflag;	/* run kprobe tests */
+
 static int
 fancy_tty(void)
 {
@@ -731,8 +737,12 @@ t_stats(const struct test *t, struct quark_queue_attr *qa)
 static int
 t_dns(const struct test *t, struct quark_queue_attr *qa)
 {
-	struct quark_queue	qq;
-	pid_t			child;
+	struct quark_queue		 qq;
+	const struct quark_event	*qev;
+	struct quark_packet		*packet;
+	pid_t				 child;
+	struct iphdr			 ip;
+	struct udphdr			 udp;
 
 	assert_localhost();
 
@@ -742,7 +752,44 @@ t_dns(const struct test *t, struct quark_queue_attr *qa)
 		err(1, "quark_queue_open");
 
 	child = fork_sock_write(53, SOCK_DGRAM);
-	drain_for_pid(&qq, child);
+
+	/* first is the fork, no agg */
+	qev = drain_for_pid(&qq, child);
+	assert(qev->packet == NULL);
+
+	/* egress */
+	qev = drain_for_pid(&qq, child);
+	packet = qev->packet;
+	assert(packet != NULL);
+	assert(packet->cap_len == 90);
+	assert(packet->orig_len == 90);
+	/* ip */
+	memcpy(&ip, packet->data, sizeof(ip));
+	assert(ip.protocol == IPPROTO_UDP);
+	assert(ip.saddr == htonl(INADDR_LOOPBACK));
+	assert(ip.daddr == htonl(INADDR_LOOPBACK));
+	/* udp */
+	memcpy(&udp, packet->data + sizeof(ip), sizeof(udp));
+	assert(udp.dest == htons(53));
+	/* dns  */
+	assert(!memcmp(packet->data + 28, PATTERN, packet->cap_len - 28));
+
+	/* ingress */
+	qev = drain_for_pid(&qq, child);
+	packet = qev->packet;
+	assert(packet != NULL);
+	assert(packet->cap_len == 90);
+	assert(packet->orig_len == 90);
+	/* ip */
+	memcpy(&ip, packet->data, sizeof(ip));
+	assert(ip.protocol == IPPROTO_UDP);
+	assert(ip.saddr == htonl(INADDR_LOOPBACK));
+	assert(ip.daddr == htonl(INADDR_LOOPBACK));
+	/* udp */
+	memcpy(&udp, packet->data + sizeof(ip), sizeof(udp));
+	assert(udp.dest == htons(53));
+	/* dns */
+	assert(!memcmp(packet->data + 28, PATTERN, packet->cap_len - 28));
 
 	quark_queue_close(&qq);