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