xref: /freebsd-12.1/contrib/ipfilter/ipsend/ip.c (revision af02ca25)
1 /*	$FreeBSD$	*/
2 
3 /*
4  * ip.c (C) 1995-1998 Darren Reed
5  *
6  * See the IPFILTER.LICENCE file for details on licencing.
7  */
8 #if !defined(lint)
9 static const char sccsid[] = "%W% %G% (C)1995";
10 static const char rcsid[] = "@(#)$Id$";
11 #endif
12 #include <sys/param.h>
13 #include <sys/types.h>
14 #include <netinet/in_systm.h>
15 #include <sys/socket.h>
16 #include <net/if.h>
17 #include <netinet/in.h>
18 #include <netinet/ip.h>
19 #include <sys/param.h>
20 #ifndef	linux
21 # include <net/route.h>
22 # include <netinet/if_ether.h>
23 # include <netinet/ip_var.h>
24 #endif
25 #include <errno.h>
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <unistd.h>
29 #include <string.h>
30 #include "ipsend.h"
31 
32 
33 static	char	*ipbuf = NULL, *ethbuf = NULL;
34 
35 
36 u_short	chksum(buf,len)
37 	u_short	*buf;
38 	int	len;
39 {
40 	u_long	sum = 0;
41 	int	nwords = len >> 1;
42 
43 	for(; nwords > 0; nwords--)
44 		sum += *buf++;
45 	sum = (sum>>16) + (sum & 0xffff);
46 	sum += (sum >>16);
47 	return (~sum);
48 }
49 
50 
51 int	send_ether(nfd, buf, len, gwip)
52 	int	nfd, len;
53 	char	*buf;
54 	struct	in_addr	gwip;
55 {
56 	static	struct	in_addr	last_gw;
57 	static	char	last_arp[6] = { 0, 0, 0, 0, 0, 0};
58 	ether_header_t	*eh;
59 	char	*s;
60 	int	err;
61 
62 	if (!ethbuf)
63 		ethbuf = (char *)calloc(1, 65536+1024);
64 	s = ethbuf;
65 	eh = (ether_header_t *)s;
66 
67 	bcopy((char *)buf, s + sizeof(*eh), len);
68 	if (gwip.s_addr == last_gw.s_addr)
69 	    {
70 		bcopy(last_arp, (char *)A_A eh->ether_dhost, 6);
71 	    }
72 	else if (arp((char *)&gwip, (char *)A_A eh->ether_dhost) == -1)
73 	    {
74 		perror("arp");
75 		return -2;
76 	    }
77 	eh->ether_type = htons(ETHERTYPE_IP);
78 	last_gw.s_addr = gwip.s_addr;
79 	err = sendip(nfd, s, sizeof(*eh) + len);
80 	return err;
81 }
82 
83 
84 /*
85  */
86 int	send_ip(nfd, mtu, ip, gwip, frag)
87 	int	nfd, mtu;
88 	ip_t	*ip;
89 	struct	in_addr	gwip;
90 	int	frag;
91 {
92 	static	struct	in_addr	last_gw, local_ip;
93 	static	char	local_arp[6] = { 0, 0, 0, 0, 0, 0};
94 	static	char	last_arp[6] = { 0, 0, 0, 0, 0, 0};
95 	static	u_short	id = 0;
96 	ether_header_t	*eh;
97 	ip_t	ipsv;
98 	int	err, iplen;
99 
100 	if (!ipbuf)
101 	  {
102 		ipbuf = (char *)malloc(65536);
103 		if (!ipbuf)
104 		  {
105 			perror("malloc failed");
106 			return -2;
107 		  }
108 	  }
109 
110 	eh = (ether_header_t *)ipbuf;
111 
112 	bzero((char *)A_A eh->ether_shost, sizeof(eh->ether_shost));
113 	if (last_gw.s_addr && (gwip.s_addr == last_gw.s_addr))
114 	    {
115 		bcopy(last_arp, (char *)A_A eh->ether_dhost, 6);
116 	    }
117 	else if (arp((char *)&gwip, (char *)A_A eh->ether_dhost) == -1)
118 	    {
119 		perror("arp");
120 		return -2;
121 	    }
122 	bcopy((char *)A_A eh->ether_dhost, last_arp, sizeof(last_arp));
123 	eh->ether_type = htons(ETHERTYPE_IP);
124 
125 	bcopy((char *)ip, (char *)&ipsv, sizeof(*ip));
126 	last_gw.s_addr = gwip.s_addr;
127 	iplen = ip->ip_len;
128 	ip->ip_len = htons(iplen);
129 	if (!(frag & 2)) {
130 		if (!IP_V(ip))
131 			IP_V_A(ip, IPVERSION);
132 		if (!ip->ip_id)
133 			ip->ip_id  = htons(id++);
134 		if (!ip->ip_ttl)
135 			ip->ip_ttl = 60;
136 	}
137 
138 	if (ip->ip_src.s_addr != local_ip.s_addr) {
139 		(void) arp((char *)&ip->ip_src, (char *)A_A local_arp);
140 		bcopy(local_arp, (char *)A_A eh->ether_shost,sizeof(last_arp));
141 		local_ip = ip->ip_src;
142 	} else
143 		bcopy(local_arp, (char *)A_A eh->ether_shost, 6);
144 
145 	if (!frag || (sizeof(*eh) + iplen < mtu))
146 	    {
147 		ip->ip_sum = 0;
148 		ip->ip_sum = chksum((u_short *)ip, IP_HL(ip) << 2);
149 
150 		bcopy((char *)ip, ipbuf + sizeof(*eh), iplen);
151 		err =  sendip(nfd, ipbuf, sizeof(*eh) + iplen);
152 	    }
153 	else
154 	    {
155 		/*
156 		 * Actually, this is bogus because we're putting all IP
157 		 * options in every packet, which isn't always what should be
158 		 * done.  Will do for now.
159 		 */
160 		ether_header_t	eth;
161 		char	optcpy[48], ol;
162 		char	*s;
163 		int	i, sent = 0, ts, hlen, olen;
164 
165 		hlen = IP_HL(ip) << 2;
166 		if (mtu < (hlen + 8)) {
167 			fprintf(stderr, "mtu (%d) < ip header size (%d) + 8\n",
168 				mtu, hlen);
169 			fprintf(stderr, "can't fragment data\n");
170 			return -2;
171 		}
172 		ol = (IP_HL(ip) << 2) - sizeof(*ip);
173 		for (i = 0, s = (char*)(ip + 1); ol > 0; )
174 			if (*s == IPOPT_EOL) {
175 				optcpy[i++] = *s;
176 				break;
177 			} else if (*s == IPOPT_NOP) {
178 				s++;
179 				ol--;
180 			} else
181 			    {
182 				olen = (int)(*(u_char *)(s + 1));
183 				ol -= olen;
184 				if (IPOPT_COPIED(*s))
185 				    {
186 					bcopy(s, optcpy + i, olen);
187 					i += olen;
188 					s += olen;
189 				    }
190 			    }
191 		if (i)
192 		    {
193 			/*
194 			 * pad out
195 			 */
196 			while ((i & 3) && (i & 3) != 3)
197 				optcpy[i++] = IPOPT_NOP;
198 			if ((i & 3) == 3)
199 				optcpy[i++] = IPOPT_EOL;
200 		    }
201 
202 		bcopy((char *)eh, (char *)&eth, sizeof(eth));
203 		s = (char *)ip + hlen;
204 		iplen = ntohs(ip->ip_len) - hlen;
205 		ip->ip_off |= htons(IP_MF);
206 
207 		while (1)
208 		    {
209 			if ((sent + (mtu - hlen)) >= iplen)
210 			    {
211 				ip->ip_off ^= htons(IP_MF);
212 				ts = iplen - sent;
213 			    }
214 			else
215 				ts = (mtu - hlen);
216 			ip->ip_off &= htons(0xe000);
217 			ip->ip_off |= htons(sent >> 3);
218 			ts += hlen;
219 			ip->ip_len = htons(ts);
220 			ip->ip_sum = 0;
221 			ip->ip_sum = chksum((u_short *)ip, hlen);
222 			bcopy((char *)ip, ipbuf + sizeof(*eh), hlen);
223 			bcopy(s + sent, ipbuf + sizeof(*eh) + hlen, ts - hlen);
224 			err =  sendip(nfd, ipbuf, sizeof(*eh) + ts);
225 
226 			bcopy((char *)&eth, ipbuf, sizeof(eth));
227 			sent += (ts - hlen);
228 			if (!(ntohs(ip->ip_off) & IP_MF))
229 				break;
230 			else if (!(ip->ip_off & htons(0x1fff)))
231 			    {
232 				hlen = i + sizeof(*ip);
233 				IP_HL_A(ip, (sizeof(*ip) + i) >> 2);
234 				bcopy(optcpy, (char *)(ip + 1), i);
235 			    }
236 		    }
237 	    }
238 
239 	bcopy((char *)&ipsv, (char *)ip, sizeof(*ip));
240 	return err;
241 }
242 
243 
244 /*
245  * send a tcp packet.
246  */
247 int	send_tcp(nfd, mtu, ip, gwip)
248 	int	nfd, mtu;
249 	ip_t	*ip;
250 	struct	in_addr	gwip;
251 {
252 	static	tcp_seq	iss = 2;
253 	tcphdr_t *t, *t2;
254 	int	thlen, i, iplen, hlen;
255 	u_32_t	lbuf[20];
256 	ip_t	*ip2;
257 
258 	iplen = ip->ip_len;
259 	hlen = IP_HL(ip) << 2;
260 	t = (tcphdr_t *)((char *)ip + hlen);
261 	ip2 = (struct ip *)lbuf;
262 	t2 = (tcphdr_t *)((char *)ip2 + hlen);
263 	thlen = TCP_OFF(t) << 2;
264 	if (!thlen)
265 		thlen = sizeof(tcphdr_t);
266 	bzero((char *)ip2, sizeof(*ip2) + sizeof(*t2));
267 	ip->ip_p = IPPROTO_TCP;
268 	ip2->ip_p = ip->ip_p;
269 	ip2->ip_src = ip->ip_src;
270 	ip2->ip_dst = ip->ip_dst;
271 	bcopy((char *)ip + hlen, (char *)t2, thlen);
272 
273 	if (!t2->th_win)
274 		t2->th_win = htons(4096);
275 	iss += 63;
276 
277 	i = sizeof(struct tcpiphdr) / sizeof(long);
278 
279 	if ((t2->th_flags == TH_SYN) && !ntohs(ip->ip_off) &&
280 	    (lbuf[i] != htonl(0x020405b4))) {
281 		lbuf[i] = htonl(0x020405b4);
282 		bcopy((char *)ip + hlen + thlen, (char *)ip + hlen + thlen + 4,
283 		      iplen - thlen - hlen);
284 		thlen += 4;
285 	    }
286 	TCP_OFF_A(t2, thlen >> 2);
287 	ip2->ip_len = htons(thlen);
288 	ip->ip_len = hlen + thlen;
289 	t2->th_sum = 0;
290 	t2->th_sum = chksum((u_short *)ip2, thlen + sizeof(ip_t));
291 
292 	bcopy((char *)t2, (char *)ip + hlen, thlen);
293 	return send_ip(nfd, mtu, ip, gwip, 1);
294 }
295 
296 
297 /*
298  * send a udp packet.
299  */
300 int	send_udp(nfd, mtu, ip, gwip)
301 	int	nfd, mtu;
302 	ip_t	*ip;
303 	struct	in_addr	gwip;
304 {
305 	struct	tcpiphdr *ti;
306 	int	thlen;
307 	u_long	lbuf[20];
308 
309 	ti = (struct tcpiphdr *)lbuf;
310 	bzero((char *)ti, sizeof(*ti));
311 	thlen = sizeof(udphdr_t);
312 	ti->ti_pr = ip->ip_p;
313 	ti->ti_src = ip->ip_src;
314 	ti->ti_dst = ip->ip_dst;
315 	bcopy((char *)ip + (IP_HL(ip) << 2),
316 	      (char *)&ti->ti_sport, sizeof(udphdr_t));
317 
318 	ti->ti_len = htons(thlen);
319 	ip->ip_len = (IP_HL(ip) << 2) + thlen;
320 	ti->ti_sum = 0;
321 	ti->ti_sum = chksum((u_short *)ti, thlen + sizeof(ip_t));
322 
323 	bcopy((char *)&ti->ti_sport,
324 	      (char *)ip + (IP_HL(ip) << 2), sizeof(udphdr_t));
325 	return send_ip(nfd, mtu, ip, gwip, 1);
326 }
327 
328 
329 /*
330  * send an icmp packet.
331  */
332 int	send_icmp(nfd, mtu, ip, gwip)
333 	int	nfd, mtu;
334 	ip_t	*ip;
335 	struct	in_addr	gwip;
336 {
337 	struct	icmp	*ic;
338 
339 	ic = (struct icmp *)((char *)ip + (IP_HL(ip) << 2));
340 
341 	ic->icmp_cksum = 0;
342 	ic->icmp_cksum = chksum((u_short *)ic, sizeof(struct icmp));
343 
344 	return send_ip(nfd, mtu, ip, gwip, 1);
345 }
346 
347 
348 int	send_packet(nfd, mtu, ip, gwip)
349 	int	nfd, mtu;
350 	ip_t	*ip;
351 	struct	in_addr	gwip;
352 {
353         switch (ip->ip_p)
354         {
355         case IPPROTO_TCP :
356                 return send_tcp(nfd, mtu, ip, gwip);
357         case IPPROTO_UDP :
358                 return send_udp(nfd, mtu, ip, gwip);
359         case IPPROTO_ICMP :
360                 return send_icmp(nfd, mtu, ip, gwip);
361         default :
362                 return send_ip(nfd, mtu, ip, gwip, 1);
363         }
364 }
365