1 /*
2  * Copyright (c) 2017 JingPiao Chen <chenjingpiao@gmail.com>
3  * Copyright (c) 2017 The strace developers.
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  * 3. The name of the author may not be used to endorse or promote products
15  *    derived from this software without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  */
28 
29 #include "tests.h"
30 
31 #ifdef HAVE_LINUX_NETFILTER_NFNETLINK_CTTIMEOUT_H
32 
33 # include <stdio.h>
34 # include <sys/socket.h>
35 # include "netlink.h"
36 # include <linux/netfilter/nfnetlink.h>
37 # include <linux/netfilter/nfnetlink_cttimeout.h>
38 
39 static void
test_nlmsg_type(const int fd)40 test_nlmsg_type(const int fd)
41 {
42 	long rc;
43 	struct nlmsghdr nlh = {
44 		.nlmsg_len = sizeof(nlh),
45 		.nlmsg_flags = NLM_F_REQUEST,
46 	};
47 
48 	nlh.nlmsg_type =
49 		NFNL_SUBSYS_CTNETLINK_TIMEOUT << 8 | IPCTNL_MSG_TIMEOUT_NEW;
50 	rc = sendto(fd, &nlh, nlh.nlmsg_len, MSG_DONTWAIT, NULL, 0);
51 	printf("sendto(%d, {len=%u"
52 	       ", type=NFNL_SUBSYS_CTNETLINK_TIMEOUT<<8|IPCTNL_MSG_TIMEOUT_NEW"
53 	       ", flags=NLM_F_REQUEST, seq=0, pid=0}"
54 	       ", %u, MSG_DONTWAIT, NULL, 0) = %s\n",
55 	       fd, nlh.nlmsg_len, nlh.nlmsg_len, sprintrc(rc));
56 
57 	nlh.nlmsg_type = NFNL_SUBSYS_CTNETLINK_TIMEOUT << 8 | 0xff;
58 	rc = sendto(fd, &nlh, nlh.nlmsg_len, MSG_DONTWAIT, NULL, 0);
59 	printf("sendto(%d, {len=%u"
60 	       ", type=NFNL_SUBSYS_CTNETLINK_TIMEOUT<<8|0xff /* IPCTNL_MSG_TIMEOUT_??? */"
61 	       ", flags=NLM_F_REQUEST, seq=0, pid=0}"
62 	       ", %u, MSG_DONTWAIT, NULL, 0) = %s\n",
63 	       fd, nlh.nlmsg_len, nlh.nlmsg_len, sprintrc(rc));
64 }
65 
66 static void
test_nlmsg_flags(const int fd)67 test_nlmsg_flags(const int fd)
68 {
69 	long rc;
70 	struct nlmsghdr nlh = {
71 		.nlmsg_len = sizeof(nlh),
72 	};
73 
74 	nlh.nlmsg_type =
75 		NFNL_SUBSYS_CTNETLINK_TIMEOUT << 8 | IPCTNL_MSG_TIMEOUT_NEW;
76 	nlh.nlmsg_flags = NLM_F_REQUEST | NLM_F_APPEND;
77 	rc = sendto(fd, &nlh, nlh.nlmsg_len, MSG_DONTWAIT, NULL, 0);
78 	printf("sendto(%d, {len=%u"
79 	       ", type=NFNL_SUBSYS_CTNETLINK_TIMEOUT<<8|IPCTNL_MSG_TIMEOUT_NEW"
80 	       ", flags=NLM_F_REQUEST|NLM_F_APPEND, seq=0, pid=0}"
81 	       ", %u, MSG_DONTWAIT, NULL, 0) = %s\n",
82 	       fd, nlh.nlmsg_len, nlh.nlmsg_len, sprintrc(rc));
83 
84 	nlh.nlmsg_type =
85 		NFNL_SUBSYS_CTNETLINK_TIMEOUT << 8 | IPCTNL_MSG_TIMEOUT_GET;
86 	nlh.nlmsg_flags = NLM_F_REQUEST | NLM_F_ATOMIC;
87 	rc = sendto(fd, &nlh, nlh.nlmsg_len, MSG_DONTWAIT, NULL, 0);
88 	printf("sendto(%d, {len=%u"
89 	       ", type=NFNL_SUBSYS_CTNETLINK_TIMEOUT<<8|IPCTNL_MSG_TIMEOUT_GET"
90 	       ", flags=NLM_F_REQUEST|NLM_F_ATOMIC, seq=0, pid=0}"
91 	       ", %u, MSG_DONTWAIT, NULL, 0) = %s\n",
92 	       fd, nlh.nlmsg_len, nlh.nlmsg_len, sprintrc(rc));
93 
94 	nlh.nlmsg_type =
95 		NFNL_SUBSYS_CTNETLINK_TIMEOUT << 8 | IPCTNL_MSG_TIMEOUT_DELETE;
96 	nlh.nlmsg_flags = NLM_F_REQUEST | NLM_F_NONREC;
97 	rc = sendto(fd, &nlh, nlh.nlmsg_len, MSG_DONTWAIT, NULL, 0);
98 	printf("sendto(%d, {len=%u"
99 	       ", type=NFNL_SUBSYS_CTNETLINK_TIMEOUT<<8|IPCTNL_MSG_TIMEOUT_DELETE"
100 	       ", flags=NLM_F_REQUEST|NLM_F_NONREC, seq=0, pid=0}"
101 	       ", %u, MSG_DONTWAIT, NULL, 0) = %s\n",
102 	       fd, nlh.nlmsg_len, nlh.nlmsg_len, sprintrc(rc));
103 }
104 
105 int
main(void)106 main(void)
107 {
108 	skip_if_unavailable("/proc/self/fd/");
109 
110 	int fd = create_nl_socket(NETLINK_NETFILTER);
111 
112 	test_nlmsg_type(fd);
113 	test_nlmsg_flags(fd);
114 
115 	puts("+++ exited with 0 +++");
116 
117 	return 0;
118 }
119 
120 #else
121 
122 SKIP_MAIN_UNDEFINED("HAVE_LINUX_NETFILTER_NFNETLINK_CTTIMEOUT_H")
123 
124 #endif
125