sock.c revision 180778 
1/* $FreeBSD: head/contrib/ipfilter/ipsend/sock.c 180778 2008-07-24 12:35:05Z darrenr $ */
2/*
3 * sock.c (C) 1995-1998 Darren Reed
4 *
5 * See the IPFILTER.LICENCE file for details on licencing.
6 *
7 */
8#if !defined(lint)
9static const char sccsid[] = "@(#)sock.c	1.2 1/11/96 (C)1995 Darren Reed";
10static const char rcsid[] = "@(#)$Id: sock.c,v 2.8.4.7 2007/09/13 07:19:34 darrenr Exp $";
11#endif
12#include <sys/param.h>
13#include <sys/types.h>
14#include <sys/time.h>
15#include <sys/stat.h>
16#if defined(__NetBSD__) && defined(__vax__)
17/*
18 * XXX need to declare boolean_t for _KERNEL <sys/files.h>
19 * which ends up including <sys/device.h> for vax.  See PR#32907
20 * for further details.
21 */
22typedef int     boolean_t;
23#endif
24#ifndef	ultrix
25#include <fcntl.h>
26#endif
27#if (__FreeBSD_version >= 300000)
28# include <sys/dirent.h>
29#else
30# include <sys/dir.h>
31#endif
32#if !defined(__osf__)
33# ifdef __NetBSD__
34#  include <machine/lock.h>
35#  include <machine/mutex.h>
36# endif
37# ifdef __FreeBSD__
38#  define _WANT_FILE
39# else
40#  define _KERNEL
41#  define	KERNEL
42# endif
43# ifdef	ultrix
44#  undef	LOCORE
45#  include <sys/smp_lock.h>
46# endif
47# include <sys/file.h>
48# ifdef __FreeBSD__
49#  undef _WANT_FILE
50# else
51#  undef  _KERNEL
52#  undef  KERNEL
53# endif
54#endif
55#include <nlist.h>
56#include <sys/user.h>
57#include <sys/socket.h>
58#include <sys/socketvar.h>
59#include <sys/proc.h>
60#if !defined(ultrix) && !defined(hpux) && !defined(__osf__)
61# include <kvm.h>
62#endif
63#ifdef sun
64#include <sys/systm.h>
65#include <sys/session.h>
66#endif
67#if BSD >= 199103
68#include <sys/sysctl.h>
69#include <sys/filedesc.h>
70#include <paths.h>
71#endif
72#include <math.h>
73#include <netinet/in.h>
74#include <netinet/in_systm.h>
75#include <netinet/ip.h>
76#include <netinet/tcp.h>
77#include <net/if.h>
78#if defined(__FreeBSD__)
79# include "radix_ipf.h"
80#endif
81#ifndef __osf__
82# include <net/route.h>
83#endif
84#include <netinet/ip_var.h>
85#include <netinet/in_pcb.h>
86#include <netinet/tcp_timer.h>
87#include <netinet/tcp_var.h>
88#include <stdio.h>
89#include <unistd.h>
90#include <string.h>
91#include <stdlib.h>
92#include <stddef.h>
93#include <pwd.h>
94#include "ipsend.h"
95
96
97int	nproc;
98struct	proc	*proc;
99
100#ifndef	KMEM
101# ifdef	_PATH_KMEM
102#  define	KMEM	_PATH_KMEM
103# endif
104#endif
105#ifndef	KERNEL
106# ifdef	_PATH_UNIX
107#  define	KERNEL	_PATH_UNIX
108# endif
109#endif
110#ifndef	KMEM
111# define	KMEM	"/dev/kmem"
112#endif
113#ifndef	KERNEL
114# define	KERNEL	"/vmunix"
115#endif
116
117
118#if BSD < 199103
119static	struct	proc	*getproc __P((void));
120#else
121static	struct	kinfo_proc	*getproc __P((void));
122#endif
123
124
125int	kmemcpy(buf, pos, n)
126char	*buf;
127void	*pos;
128int	n;
129{
130	static	int	kfd = -1;
131	off_t	offset = (u_long)pos;
132
133	if (kfd == -1)
134		kfd = open(KMEM, O_RDONLY);
135
136	if (lseek(kfd, offset, SEEK_SET) == -1)
137	    {
138		perror("lseek");
139		return -1;
140	    }
141	if (read(kfd, buf, n) == -1)
142	    {
143		perror("read");
144		return -1;
145	    }
146	return n;
147}
148
149struct	nlist	names[4] = {
150	{ "_proc" },
151	{ "_nproc" },
152#ifdef	ultrix
153	{ "_u" },
154#else
155	{ NULL },
156#endif
157	{ NULL }
158	};
159
160#if BSD < 199103
161static struct proc *getproc()
162{
163	struct	proc	*p;
164	pid_t	pid = getpid();
165	int	siz, n;
166
167	n = nlist(KERNEL, names);
168	if (n != 0)
169	    {
170		fprintf(stderr, "nlist(%#x) == %d\n", names, n);
171		return NULL;
172	    }
173	if (KMCPY(&nproc, names[1].n_value, sizeof(nproc)) == -1)
174	    {
175		fprintf(stderr, "read nproc (%#x)\n", names[1].n_value);
176		return NULL;
177	    }
178	siz = nproc * sizeof(struct proc);
179	if (KMCPY(&p, names[0].n_value, sizeof(p)) == -1)
180	    {
181		fprintf(stderr, "read(%#x,%#x,%d) proc\n",
182			names[0].n_value, &p, sizeof(p));
183		return NULL;
184	    }
185	proc = (struct proc *)malloc(siz);
186	if (KMCPY(proc, p, siz) == -1)
187	    {
188		fprintf(stderr, "read(%#x,%#x,%d) proc\n",
189			p, proc, siz);
190		return NULL;
191	    }
192
193	p = proc;
194
195	for (n = nproc; n; n--, p++)
196		if (p->p_pid == pid)
197			break;
198	if (!n)
199		return NULL;
200
201	return p;
202}
203
204
205struct	tcpcb	*find_tcp(fd, ti)
206int	fd;
207struct	tcpiphdr *ti;
208{
209	struct	tcpcb	*t;
210	struct	inpcb	*i;
211	struct	socket	*s;
212	struct	user	*up;
213	struct	proc	*p;
214	struct	file	*f, **o;
215
216	if (!(p = getproc()))
217		return NULL;
218	up = (struct user *)malloc(sizeof(*up));
219#ifndef	ultrix
220	if (KMCPY(up, p->p_uarea, sizeof(*up)) == -1)
221	    {
222		fprintf(stderr, "read(%#x,%#x) failed\n", p, p->p_uarea);
223		return NULL;
224	    }
225#else
226	if (KMCPY(up, names[2].n_value, sizeof(*up)) == -1)
227	    {
228		fprintf(stderr, "read(%#x,%#x) failed\n", p, names[2].n_value);
229		return NULL;
230	    }
231#endif
232
233	o = (struct file **)calloc(1, sizeof(*o) * (up->u_lastfile + 1));
234	if (KMCPY(o, up->u_ofile, (up->u_lastfile + 1) * sizeof(*o)) == -1)
235	    {
236		fprintf(stderr, "read(%#x,%#x,%d) - u_ofile - failed\n",
237			up->u_ofile, o, sizeof(*o));
238		return NULL;
239	    }
240	f = (struct file *)calloc(1, sizeof(*f));
241	if (KMCPY(f, o[fd], sizeof(*f)) == -1)
242	    {
243		fprintf(stderr, "read(%#x,%#x,%d) - o[fd] - failed\n",
244			up->u_ofile[fd], f, sizeof(*f));
245		return NULL;
246	    }
247
248	s = (struct socket *)calloc(1, sizeof(*s));
249	if (KMCPY(s, f->f_data, sizeof(*s)) == -1)
250	    {
251		fprintf(stderr, "read(%#x,%#x,%d) - f_data - failed\n",
252			o[fd], s, sizeof(*s));
253		return NULL;
254	    }
255
256	i = (struct inpcb *)calloc(1, sizeof(*i));
257	if (KMCPY(i, s->so_pcb, sizeof(*i)) == -1)
258	    {
259		fprintf(stderr, "kvm_read(%#x,%#x,%d) - so_pcb - failed\n",
260			s->so_pcb, i, sizeof(*i));
261		return NULL;
262	    }
263
264	t = (struct tcpcb *)calloc(1, sizeof(*t));
265	if (KMCPY(t, i->inp_ppcb, sizeof(*t)) == -1)
266	    {
267		fprintf(stderr, "read(%#x,%#x,%d) - inp_ppcb - failed\n",
268			i->inp_ppcb, t, sizeof(*t));
269		return NULL;
270	    }
271	return (struct tcpcb *)i->inp_ppcb;
272}
273#else
274static struct kinfo_proc *getproc()
275{
276	static	struct	kinfo_proc kp;
277	pid_t	pid = getpid();
278	int	mib[4];
279	size_t	n;
280
281	mib[0] = CTL_KERN;
282	mib[1] = KERN_PROC;
283	mib[2] = KERN_PROC_PID;
284	mib[3] = pid;
285
286	n = sizeof(kp);
287	if (sysctl(mib, 4, &kp, &n, NULL, 0) == -1)
288	    {
289		perror("sysctl");
290		return NULL;
291	    }
292	return &kp;
293}
294
295
296struct	tcpcb	*find_tcp(tfd, ti)
297int	tfd;
298struct	tcpiphdr *ti;
299{
300	struct	tcpcb	*t;
301	struct	inpcb	*i;
302	struct	socket	*s;
303	struct	filedesc	*fd;
304	struct	kinfo_proc	*p;
305	struct	file	*f, **o;
306
307	if (!(p = getproc()))
308		return NULL;
309
310	fd = (struct filedesc *)malloc(sizeof(*fd));
311	if (fd == NULL)
312		return NULL;
313#if defined( __FreeBSD_version) && __FreeBSD_version >= 500013
314	if (KMCPY(fd, p->ki_fd, sizeof(*fd)) == -1)
315	    {
316		fprintf(stderr, "read(%#lx,%#lx) failed\n",
317			(u_long)p, (u_long)p->ki_fd);
318		free(fd);
319		return NULL;
320	    }
321#else
322	if (KMCPY(fd, p->kp_proc.p_fd, sizeof(*fd)) == -1)
323	    {
324		fprintf(stderr, "read(%#lx,%#lx) failed\n",
325			(u_long)p, (u_long)p->kp_proc.p_fd);
326		free(fd);
327		return NULL;
328	    }
329#endif
330
331	o = NULL;
332	f = NULL;
333	s = NULL;
334	i = NULL;
335	t = NULL;
336
337	o = (struct file **)calloc(1, sizeof(*o) * (fd->fd_lastfile + 1));
338	if (KMCPY(o, fd->fd_ofiles, (fd->fd_lastfile + 1) * sizeof(*o)) == -1)
339	    {
340		fprintf(stderr, "read(%#lx,%#lx,%lu) - u_ofile - failed\n",
341			(u_long)fd->fd_ofiles, (u_long)o, (u_long)sizeof(*o));
342		goto finderror;
343	    }
344	f = (struct file *)calloc(1, sizeof(*f));
345	if (KMCPY(f, o[tfd], sizeof(*f)) == -1)
346	    {
347		fprintf(stderr, "read(%#lx,%#lx,%lu) - o[tfd] - failed\n",
348			(u_long)o[tfd], (u_long)f, (u_long)sizeof(*f));
349		goto finderror;
350	    }
351
352	s = (struct socket *)calloc(1, sizeof(*s));
353	if (KMCPY(s, f->f_data, sizeof(*s)) == -1)
354	    {
355		fprintf(stderr, "read(%#lx,%#lx,%lu) - f_data - failed\n",
356			(u_long)f->f_data, (u_long)s, (u_long)sizeof(*s));
357		goto finderror;
358	    }
359
360	i = (struct inpcb *)calloc(1, sizeof(*i));
361	if (KMCPY(i, s->so_pcb, sizeof(*i)) == -1)
362	    {
363		fprintf(stderr, "kvm_read(%#lx,%#lx,%lu) - so_pcb - failed\n",
364			(u_long)s->so_pcb, (u_long)i, (u_long)sizeof(*i));
365		goto finderror;
366	    }
367
368	t = (struct tcpcb *)calloc(1, sizeof(*t));
369	if (KMCPY(t, i->inp_ppcb, sizeof(*t)) == -1)
370	    {
371		fprintf(stderr, "read(%#lx,%#lx,%lu) - inp_ppcb - failed\n",
372			(u_long)i->inp_ppcb, (u_long)t, (u_long)sizeof(*t));
373		goto finderror;
374	    }
375	return (struct tcpcb *)i->inp_ppcb;
376
377finderror:
378	if (o != NULL)
379		free(o);
380	if (f != NULL)
381		free(f);
382	if (s != NULL)
383		free(s);
384	if (i != NULL)
385		free(i);
386	if (t != NULL)
387		free(t);
388	return NULL;
389}
390#endif /* BSD < 199301 */
391
392int	do_socket(dev, mtu, ti, gwip)
393char	*dev;
394int	mtu;
395struct	tcpiphdr *ti;
396struct	in_addr	gwip;
397{
398	struct	sockaddr_in	rsin, lsin;
399	struct	tcpcb	*t, tcb;
400	int	fd, nfd;
401	socklen_t len;
402
403	printf("Dest. Port: %d\n", ti->ti_dport);
404
405	fd = socket(AF_INET, SOCK_STREAM, 0);
406	if (fd == -1)
407	    {
408		perror("socket");
409		return -1;
410	    }
411
412	if (fcntl(fd, F_SETFL, FNDELAY) == -1)
413	    {
414		perror("fcntl");
415		return -1;
416	    }
417
418	bzero((char *)&lsin, sizeof(lsin));
419	lsin.sin_family = AF_INET;
420	bcopy((char *)&ti->ti_src, (char *)&lsin.sin_addr,
421	      sizeof(struct in_addr));
422	if (bind(fd, (struct sockaddr *)&lsin, sizeof(lsin)) == -1)
423	    {
424		perror("bind");
425		return -1;
426	    }
427	len = sizeof(lsin);
428	(void) getsockname(fd, (struct sockaddr *)&lsin, &len);
429	ti->ti_sport = lsin.sin_port;
430	printf("sport %d\n", ntohs(lsin.sin_port));
431
432	nfd = initdevice(dev, 1);
433	if (nfd == -1)
434		return -1;
435
436	if (!(t = find_tcp(fd, ti)))
437		return -1;
438
439	bzero((char *)&rsin, sizeof(rsin));
440	rsin.sin_family = AF_INET;
441	bcopy((char *)&ti->ti_dst, (char *)&rsin.sin_addr,
442	      sizeof(struct in_addr));
443	rsin.sin_port = ti->ti_dport;
444	if (connect(fd, (struct sockaddr *)&rsin, sizeof(rsin)) == -1 &&
445	    errno != EINPROGRESS)
446	    {
447		perror("connect");
448		return -1;
449	    }
450	KMCPY(&tcb, t, sizeof(tcb));
451	ti->ti_win = tcb.rcv_adv;
452	ti->ti_seq = tcb.snd_nxt - 1;
453	ti->ti_ack = tcb.rcv_nxt;
454
455	if (send_tcp(nfd, mtu, (ip_t *)ti, gwip) == -1)
456		return -1;
457	(void)write(fd, "Hello World\n", 12);
458	sleep(2);
459	close(fd);
460	return 0;
461}
462