summaryrefslogtreecommitdiffstats
path: root/utils/nfulnl_test.c
blob: e5f625bc09dd9925fd9109035dd7f23f3f6d2abd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <netinet/in.h>

#include <libnetfilter_log/libnetfilter_log.h>

static int print_pkt(struct nflog_data *nfad)
{
	struct nfulnl_msg_packet_hdr *ph;
	uint32_t outdev, indev;
	int payload_len;
	char *payload;
	char *prefix;

	ph = nflog_get_msg_packet_hdr(nfad);
	if (ph) {
		printf("hw_protocol=0x%04x hook=%u ",
			ntohs(ph->hw_protocol), ph->hook);
	}

	printf("mark=%u ", nflog_get_nfmark(nfad));

	indev = nflog_get_indev(nfad);
	if (indev > 0)
		printf("indev=%u ", indev);

	outdev = nflog_get_outdev(nfad);
	if (outdev > 0)
		printf("outdev=%u ", outdev);


	prefix = nflog_get_prefix(nfad);
	if (prefix)
		printf("prefix=\"%s\" ", prefix);

	payload_len = nflog_get_payload(nfad, &payload);
	if (payload_len >= 0)
		printf("payload_len=%d ", payload_len);

	fputc('\n', stdout);
	return 0;
}

static int cb(struct nflog_g_handle *gh, struct nfgenmsg *nfmsg,
		struct nflog_data *nfa, void *data)
{
	print_pkt(nfa);
	return 0;
}


int main(int argc, char **argv)
{
	struct nflog_handle *h;
	struct nflog_g_handle *gh;
	struct nflog_g_handle *gh100;
	int rv, fd;
	char buf[4096];

	h = nflog_open();
	if (!h) {
		perror("nflog_open");
		exit(1);
	}

	printf("unbinding existing nf_log handler for AF_INET (if any)\n");
	if (nflog_unbind_pf(h, AF_INET) < 0) {
		perror("nflog_unbind_pf");
		exit(1);
	}

	printf("binding nfnetlink_log to AF_INET\n");
	if (nflog_bind_pf(h, AF_INET) < 0) {
		perror("nflog_bind_pf");
		exit(1);
	}
	printf("binding this socket to group 0\n");
	gh = nflog_bind_group(h, 0);
	if (!gh) {
		perror("nflog_bind_group 0");
		exit(1);
	}

	printf("binding this socket to group 100\n");
	gh100 = nflog_bind_group(h, 100);
	if (!gh100) {
		perror("nflog_bind_group 100");
		exit(1);
	}

	printf("setting copy_packet mode\n");
	if (nflog_set_mode(gh, NFULNL_COPY_PACKET, 0xffff) < 0) {
		perror("nflog_set_mode NFULNL_COPY_PACKET");
		exit(1);
	}

	fd = nflog_fd(h);

	printf("registering callback for group 0\n");
	nflog_callback_register(gh, &cb, NULL);

	printf("going into main loop\n");
	while ((rv = recv(fd, buf, sizeof(buf), 0)) && rv >= 0) {
		printf("pkt received (len=%u)\n", rv);

		/* handle messages in just-received packet */
		nflog_handle_packet(h, buf, rv);
	}

	printf("unbinding from group 100\n");
	nflog_unbind_group(gh100);
	printf("unbinding from group 0\n");
	nflog_unbind_group(gh);

#ifdef INSANE
	/* norally, applications SHOULD NOT issue this command,
	 * since it detaches other programs/sockets from AF_INET, too ! */
	printf("unbinding from AF_INET\n");
	nflog_unbind_pf(h, AF_INET);
#endif

	printf("closing handle\n");
	nflog_close(h);

	return EXIT_SUCCESS;
}