sbin/ipf/ipsend/sock.c

   1 /* $FreeBSD$ */
   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)
   9 static const char sccsid[] = "@(#)sock.c        1.2 1/11/96 (C)1995 Darren Reed";
  10 static const char rcsid[] = "@(#)$Id$";
  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  */
  22 typedef int     boolean_t;
  23 #endif
  24 #include <fcntl.h>
  25 # include <sys/dirent.h>
  26 # ifdef __NetBSD__
  27 #  include <machine/lock.h>
  28 # endif
  29 # ifdef __FreeBSD__
  30 #  define _WANT_FILE
  31 # else
  32 #  define _KERNEL
  33 #  define       KERNEL
  34 # endif
  35 # include <sys/file.h>
  36 # ifdef __FreeBSD__
  37 #  undef _WANT_FILE
  38 # else
  39 #  undef  _KERNEL
  40 #  undef  KERNEL
  41 # endif
  42 #include <nlist.h>
  43 #include <sys/user.h>
  44 #include <sys/socket.h>
  45 #define _WANT_SOCKET
  46 #include <sys/socketvar.h>
  47 #include <sys/proc.h>
  48 # include <kvm.h>
  49 #ifdef sun
  50 #include <sys/systm.h>
  51 #include <sys/session.h>
  52 #endif
  53 #include <sys/sysctl.h>
  54 #include <sys/filedesc.h>
  55 #include <paths.h>
  56 #include <math.h>
  57 #include <netinet/in.h>
  58 #include <netinet/in_systm.h>
  59 #include <netinet/ip.h>
  60 #include <netinet/tcp.h>
  61 #include <net/if.h>
  62 # include <net/route.h>
  63 #include <netinet/ip_var.h>
  64 #define _WANT_INPCB
  65 #include <netinet/in_pcb.h>
  66 #include <netinet/tcp_timer.h>
  67 #define _WANT_TCPCB
  68 #include <netinet/tcp_var.h>
  69 #include <stdio.h>
  70 #include <unistd.h>
  71 #include <string.h>
  72 #include <stdlib.h>
  73 #include <stddef.h>
  74 #include <pwd.h>
  75 #include "ipsend.h"
  76
  77
  78 int     nproc;
  79 struct  proc    *proc;
  80
  81 #ifndef KMEM
  82 # ifdef _PATH_KMEM
  83 #  define       KMEM    _PATH_KMEM
  84 # endif
  85 #endif
  86 #ifndef KERNEL
  87 # ifdef _PATH_UNIX
  88 #  define       KERNEL  _PATH_UNIX
  89 # endif
  90 #endif
  91 #ifndef KMEM
  92 # define        KMEM    "/dev/kmem"
  93 #endif
  94 #ifndef KERNEL
  95 # define        KERNEL  "/vmunix"
  96 #endif
  97
  98
  99 static  struct  kinfo_proc      *getproc(void);
 100
 101
 102 int
 103 kmemcpy(char *buf, void *pos, int n)
 104 {
 105         static  int     kfd = -1;
 106         off_t   offset = (u_long)pos;
 107
 108         if (kfd == -1)
 109                 kfd = open(KMEM, O_RDONLY);
 110
 111         if (lseek(kfd, offset, SEEK_SET) == -1)
 112             {
 113                 perror("lseek");
 114                 return (-1);
 115             }
 116         if (read(kfd, buf, n) == -1)
 117             {
 118                 perror("read");
 119                 return (-1);
 120             }
 121         return (n);
 122 }
 123
 124 struct  nlist   names[4] = {
 125         { "_proc" },
 126         { "_nproc" },
 127         { NULL },
 128         { NULL }
 129         };
 130
 131 static struct
 132 kinfo_proc *getproc(void)
 133 {
 134         static  struct  kinfo_proc kp;
 135         pid_t   pid = getpid();
 136         int     mib[4];
 137         size_t  n;
 138
 139         mib[0] = CTL_KERN;
 140         mib[1] = KERN_PROC;
 141         mib[2] = KERN_PROC_PID;
 142         mib[3] = pid;
 143
 144         n = sizeof(kp);
 145         if (sysctl(mib, 4, &kp, &n, NULL, 0) == -1)
 146             {
 147                 perror("sysctl");
 148                 return (NULL);
 149             }
 150         return (&kp);
 151 }
 152
 153
 154 struct tcpcb *
 155 find_tcp(int tfd, struct  tcpiphdr *ti)
 156 {
 157         struct  tcpcb   *t;
 158         struct  inpcb   *i;
 159         struct  socket  *s;
 160         struct  filedesc        *fd;
 161         struct  kinfo_proc      *p;
 162         struct  file    *f, **o;
 163
 164         if (!(p = getproc()))
 165                 return (NULL);
 166
 167         fd = (struct filedesc *)malloc(sizeof(*fd));
 168         if (fd == NULL)
 169                 return (NULL);
 170 #if defined( __FreeBSD__)
 171         if (KMCPY(fd, p->ki_fd, sizeof(*fd)) == -1)
 172             {
 173                 fprintf(stderr, "read(%#lx,%#lx) failed\n",
 174                         (u_long)p, (u_long)p->ki_fd);
 175                 free(fd);
 176                 return (NULL);
 177             }
 178 #else
 179         if (KMCPY(fd, p->kp_proc.p_fd, sizeof(*fd)) == -1)
 180             {
 181                 fprintf(stderr, "read(%#lx,%#lx) failed\n",
 182                         (u_long)p, (u_long)p->kp_proc.p_fd);
 183                 free(fd);
 184                 return (NULL);
 185             }
 186 #endif
 187
 188         o = NULL;
 189         f = NULL;
 190         s = NULL;
 191         i = NULL;
 192         t = NULL;
 193
 194         o = (struct file **)calloc(fd->fd_lastfile + 1, sizeof(*o));
 195         if (KMCPY(o, fd->fd_ofiles, (fd->fd_lastfile + 1) * sizeof(*o)) == -1)
 196             {
 197                 fprintf(stderr, "read(%#lx,%#lx,%lu) - u_ofile - failed\n",
 198                         (u_long)fd->fd_ofiles, (u_long)o, (u_long)sizeof(*o));
 199                 goto finderror;
 200             }
 201         f = (struct file *)calloc(1, sizeof(*f));
 202         if (KMCPY(f, o[tfd], sizeof(*f)) == -1)
 203             {
 204                 fprintf(stderr, "read(%#lx,%#lx,%lu) - o[tfd] - failed\n",
 205                         (u_long)o[tfd], (u_long)f, (u_long)sizeof(*f));
 206                 goto finderror;
 207             }
 208
 209         s = (struct socket *)calloc(1, sizeof(*s));
 210         if (KMCPY(s, f->f_data, sizeof(*s)) == -1)
 211             {
 212                 fprintf(stderr, "read(%#lx,%#lx,%lu) - f_data - failed\n",
 213                         (u_long)f->f_data, (u_long)s, (u_long)sizeof(*s));
 214                 goto finderror;
 215             }
 216
 217         i = (struct inpcb *)calloc(1, sizeof(*i));
 218         if (KMCPY(i, s->so_pcb, sizeof(*i)) == -1)
 219             {
 220                 fprintf(stderr, "kvm_read(%#lx,%#lx,%lu) - so_pcb - failed\n",
 221                         (u_long)s->so_pcb, (u_long)i, (u_long)sizeof(*i));
 222                 goto finderror;
 223             }
 224
 225         t = (struct tcpcb *)calloc(1, sizeof(*t));
 226         if (KMCPY(t, i->inp_ppcb, sizeof(*t)) == -1)
 227             {
 228                 fprintf(stderr, "read(%#lx,%#lx,%lu) - inp_ppcb - failed\n",
 229                         (u_long)i->inp_ppcb, (u_long)t, (u_long)sizeof(*t));
 230                 goto finderror;
 231             }
 232         return (struct tcpcb *)i->inp_ppcb;
 233
 234 finderror:
 235         if (o != NULL)
 236                 free(o);
 237         if (f != NULL)
 238                 free(f);
 239         if (s != NULL)
 240                 free(s);
 241         if (i != NULL)
 242                 free(i);
 243         if (t != NULL)
 244                 free(t);
 245         return (NULL);
 246 }
 247
 248 int
 249 do_socket(char *dev, int mtu, struct  tcpiphdr *ti, struct  in_addr gwip)
 250 {
 251         struct  sockaddr_in     rsin, lsin;
 252         struct  tcpcb   *t, tcb;
 253         int     fd, nfd;
 254         socklen_t len;
 255
 256         printf("Dest. Port: %d\n", ti->ti_dport);
 257
 258         fd = socket(AF_INET, SOCK_STREAM, 0);
 259         if (fd == -1)
 260             {
 261                 perror("socket");
 262                 return (-1);
 263             }
 264
 265         if (fcntl(fd, F_SETFL, FNDELAY) == -1)
 266             {
 267                 perror("fcntl");
 268                 return (-1);
 269             }
 270
 271         bzero((char *)&lsin, sizeof(lsin));
 272         lsin.sin_family = AF_INET;
 273         bcopy((char *)&ti->ti_src, (char *)&lsin.sin_addr,
 274               sizeof(struct in_addr));
 275         if (bind(fd, (struct sockaddr *)&lsin, sizeof(lsin)) == -1)
 276             {
 277                 perror("bind");
 278                 return (-1);
 279             }
 280         len = sizeof(lsin);
 281         (void) getsockname(fd, (struct sockaddr *)&lsin, &len);
 282         ti->ti_sport = lsin.sin_port;
 283         printf("sport %d\n", ntohs(lsin.sin_port));
 284
 285         nfd = initdevice(dev, 1);
 286         if (nfd == -1)
 287                 return (-1);
 288
 289         if (!(t = find_tcp(fd, ti)))
 290                 return (-1);
 291
 292         bzero((char *)&rsin, sizeof(rsin));
 293         rsin.sin_family = AF_INET;
 294         bcopy((char *)&ti->ti_dst, (char *)&rsin.sin_addr,
 295               sizeof(struct in_addr));
 296         rsin.sin_port = ti->ti_dport;
 297         if (connect(fd, (struct sockaddr *)&rsin, sizeof(rsin)) == -1 &&
 298             errno != EINPROGRESS)
 299             {
 300                 perror("connect");
 301                 return (-1);
 302             }
 303         KMCPY(&tcb, t, sizeof(tcb));
 304         ti->ti_win = tcb.rcv_adv;
 305         ti->ti_seq = tcb.snd_nxt - 1;
 306         ti->ti_ack = tcb.rcv_nxt;
 307
 308         if (send_tcp(nfd, mtu, (ip_t *)ti, gwip) == -1)
 309                 return (-1);
 310         (void)write(fd, "Hello World\n", 12);
 311         sleep(2);
 312         close(fd);
 313         return (0);
 314 }