1 /*	$NetBSD: t_pr.c,v 1.8 2017/01/13 21:30:42 christos Exp $	*/
2 
3 /*-
4  * Copyright (c) 2010 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND
17  * CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
18  * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
19  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20  * IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY
21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
23  * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
25  * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
26  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
27  * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28  */
29 
30 #include <sys/cdefs.h>
31 #ifndef lint
32 __RCSID("$NetBSD: t_pr.c,v 1.8 2017/01/13 21:30:42 christos Exp $");
33 #endif /* not lint */
34 
35 #include <sys/types.h>
36 #include <sys/socket.h>
37 #include <sys/sysctl.h>
38 
39 #include <netinet/in.h>
40 #include <net/route.h>
41 
42 #include <rump/rump.h>
43 #include <rump/rump_syscalls.h>
44 
45 #include <atf-c.h>
46 #include <errno.h>
47 #include <stdio.h>
48 #include <stdlib.h>
49 #include <string.h>
50 #include <unistd.h>
51 
52 #include "../config/netconfig.c"
53 #include "h_macros.h"
54 
55 /*
56  * Prepare rump, configure interface and route to cause fragmentation
57  */
58 static void
59 setup(void)
60 {
61 	char ifname[IFNAMSIZ];
62 	struct {
63 		struct rt_msghdr m_rtm;
64 		struct sockaddr_in m_sin;
65 	} m_rtmsg;
66 #define rtm m_rtmsg.m_rtm
67 #define rsin m_rtmsg.m_sin
68 	struct ifreq ifr;
69 	int s;
70 
71 	rump_init();
72 
73 	/* first, config lo0 & route */
74 	strcpy(ifname, "lo0");
75 	netcfg_rump_if(ifname, "127.0.0.1", "255.0.0.0");
76 	netcfg_rump_route("127.0.0.1", "255.0.0.0", "127.0.0.1");
77 
78 	if ((s = rump_sys_socket(PF_ROUTE, SOCK_RAW, 0)) == -1)
79 		atf_tc_fail_errno("routing socket");
80 
81 	/*
82 	 * set MTU for interface so that route MTU doesn't
83 	 * get overridden by it.
84 	 */
85 	memset(&ifr, 0, sizeof(ifr));
86 	strcpy(ifr.ifr_name, "lo0");
87 	ifr.ifr_mtu = 1300;
88 	if (rump_sys_ioctl(s, SIOCSIFMTU, &ifr) == -1)
89 		atf_tc_fail_errno("set mtu");
90 
91 	/* change route MTU to 100 */
92 	memset(&m_rtmsg, 0, sizeof(m_rtmsg));
93 	rtm.rtm_type = RTM_CHANGE;
94 	rtm.rtm_flags = RTF_STATIC;
95 	rtm.rtm_version = RTM_VERSION;
96 	rtm.rtm_seq = 3;
97 	rtm.rtm_inits = RTV_MTU;
98 	rtm.rtm_addrs = RTA_DST;
99 	rtm.rtm_rmx.rmx_mtu = 100;
100 	rtm.rtm_msglen = sizeof(m_rtmsg);
101 
102 	memset(&rsin, 0, sizeof(rsin));
103 	rsin.sin_family = AF_INET;
104 	rsin.sin_len = sizeof(rsin);
105 	rsin.sin_addr.s_addr = inet_addr("127.0.0.1");
106 
107 	if (rump_sys_write(s, &m_rtmsg, sizeof(m_rtmsg)) != sizeof(m_rtmsg))
108 		atf_tc_fail_errno("set route mtu");
109 	rump_sys_close(s);
110 }
111 
112 /*
113  * Turn on checksums on loopback interfaces
114  */
115 static int
116 enable_locsums(void)
117 {
118 	struct sysctlnode q, ans[256];
119 	int mib[5], enable;
120 	size_t alen;
121 	unsigned i;
122 
123 	mib[0] = CTL_NET;
124 	mib[1] = PF_INET;
125 	mib[2] = IPPROTO_IP;
126 	mib[3] = CTL_QUERY;
127 	alen = sizeof(ans);
128 
129 	memset(&q, 0, sizeof(q));
130 	q.sysctl_flags = SYSCTL_VERSION;
131 
132 	if (rump_sys___sysctl(mib, 4, ans, &alen, &q, sizeof(q)) == -1)
133 		return -1;
134 
135 	for (i = 0; i < __arraycount(ans); i++)
136 		if (strcmp("do_loopback_cksum", ans[i].sysctl_name) == 0)
137 			break;
138 	if (i == __arraycount(ans)) {
139 		errno = ENOENT;
140 		return -1;
141 	}
142 
143 	mib[3] = ans[i].sysctl_num;
144 
145 	enable = 1;
146 	if (rump_sys___sysctl(mib, 4, NULL, NULL, &enable,
147 	    sizeof(enable)) == -1)
148 		return errno;
149 
150 	return 0;
151 }
152 
153 ATF_TC(loopmtu);
154 ATF_TC_HEAD(loopmtu, tc)
155 {
156 
157 	atf_tc_set_md_var(tc, "descr",
158 	    "test lo0 fragmentation (PR kern/43664)");
159 }
160 
161 ATF_TC_BODY(loopmtu, tc)
162 {
163 	struct sockaddr_in sin;
164 	char data[2000];
165 	int s;
166 
167 	setup();
168 
169 	/* open raw socket */
170 	s = rump_sys_socket(PF_INET, SOCK_RAW, 0);
171 	if (s == -1)
172 		atf_tc_fail_errno("raw socket");
173 
174 	/* then, send data */
175 	memset(&sin, 0, sizeof(sin));
176 	sin.sin_family = AF_INET;
177 	sin.sin_len = sizeof(sin);
178 	sin.sin_port = htons(12345);
179 	sin.sin_addr.s_addr = inet_addr("127.0.0.1");
180 
181 	if (rump_sys_sendto(s, data, sizeof(data), 0,
182 	    (struct sockaddr *)&sin, sizeof(sin)) == -1)
183 		atf_tc_fail_errno("sendto failed");
184 }
185 
186 ATF_TC(loopmtu_csum);
187 ATF_TC_HEAD(loopmtu_csum, tc)
188 {
189 
190 	atf_tc_set_md_var(tc, "descr",
191 	    "test lo0 fragmentation with checksums (PR kern/43664)");
192 }
193 
194 ATF_TC_BODY(loopmtu_csum, tc)
195 {
196 	struct sockaddr_in sin;
197 	char data[2000];
198 	int s;
199 
200 	setup();
201 
202 	ATF_CHECK(enable_locsums() == 0);
203 
204 	/* open raw socket */
205 	s = rump_sys_socket(PF_INET, SOCK_RAW, 0);
206 	if (s == -1)
207 		atf_tc_fail_errno("raw socket");
208 
209 	/* then, send data */
210 	memset(&sin, 0, sizeof(sin));
211 	sin.sin_family = AF_INET;
212 	sin.sin_len = sizeof(sin);
213 	sin.sin_port = htons(12345);
214 	sin.sin_addr.s_addr = inet_addr("127.0.0.1");
215 
216 	if (rump_sys_sendto(s, data, sizeof(data), 0,
217 	    (struct sockaddr *)&sin, sizeof(sin)) == -1)
218 		atf_tc_fail_errno("sendto failed");
219 }
220 
221 ATF_TP_ADD_TCS(tp)
222 {
223 
224 	ATF_TP_ADD_TC(tp, loopmtu);
225 	ATF_TP_ADD_TC(tp, loopmtu_csum);
226 
227 	return atf_no_error();
228 }
229 
230