/* $NetBSD: if_srt.c,v 1.5 2007/03/04 06:03:17 christos Exp $ */
/* This file is in the public domain. */

#include "opt_inet.h"

#if !defined(INET) && !defined(INET6)
#error "srt without INET/INET6?"
#endif

#ifndef SRT_MAXUNIT
#define SRT_MAXUNIT 255
#endif

/* include-file bug workarounds */
#include <sys/types.h> /* sys/conf.h */
#include <sys/resource.h> /* sys/resourcevar.h (uvm/uvm_param.h, sys/mbuf.h) */
#include <netinet/in.h> /* netinet/ip.h */
#include <sys/param.h> /* sys/mbuf.h */
#include <netinet/in_systm.h> /* netinet/ip.h */

#include <sys/conf.h>
#include <sys/mbuf.h>
#include <sys/errno.h>
#include <sys/fcntl.h>
#include <sys/param.h>
#include <sys/ioctl.h>
#include <netinet/ip.h>
#include <netinet/ip6.h>
#include <net/if_types.h>
#include <machine/stdarg.h>

#include "if_srt.h"
#include "bpfilter.h"

/* until we know what to pass to bpfattach.... */
#undef NBPFILTER
#define NBPFILTER 0

typedef struct srt_rt RT;
typedef struct softc SOFTC;

struct softc {
  struct ifnet intf;	/* XXX interface botch */
  int unit;
  int nrt;
  RT **rts;
  unsigned int flags;	/* SSF_* values from if_srt.h */
#define SSF_UCHG (SSF_MTULOCK) /* userland-changeable bits */
  unsigned int kflags;	/* bits private to this file */
#define SKF_CDEVOPEN 0x00000001
  } ;

static SOFTC *softcv[SRT_MAXUNIT+1];
static unsigned int global_flags;

/* Internal routines. */

static unsigned int ipv4_masks[33]
 = { 0x00000000, /* /0 */
     0x80000000, 0xc0000000, 0xe0000000, 0xf0000000, /* /1 - /4 */
     0xf8000000, 0xfc000000, 0xfe000000, 0xff000000, /* /5 - /8 */
     0xff800000, 0xffc00000, 0xffe00000, 0xfff00000, /* /9 - /12 */
     0xfff80000, 0xfffc0000, 0xfffe0000, 0xffff0000, /* /13 - /16 */
     0xffff8000, 0xffffc000, 0xffffe000, 0xfffff000, /* /17 - /20 */
     0xfffff800, 0xfffffc00, 0xfffffe00, 0xffffff00, /* /21 - /24 */
     0xffffff80, 0xffffffc0, 0xffffffe0, 0xfffffff0, /* /25 - /28 */
     0xfffffff8, 0xfffffffc, 0xfffffffe, 0xffffffff  /* /29 - /32 */ };

static void update_mtu(SOFTC *sc)
{
 int mtu;
 int i;
 RT *r;

 if (sc->flags & SSF_MTULOCK) return;
 mtu = 65535;
 for (i=sc->nrt-1;i>=0;i--)
  { r = sc->rts[i];
    if (r->u.dstifp->if_mtu < mtu) mtu = r->u.dstifp->if_mtu;
  }
 sc->intf.if_mtu = mtu;
}

static RT *find_rt(SOFTC *sc, int af, ...)
{
 int i;
 RT *r;
 struct in_addr ia;
 struct in6_addr ia6;
 va_list ap;

 ia.s_addr = 0; ia6.s6_addr[0] = 0; /* shut up incorrect -Wuninitialized */
 va_start(ap,af);
 switch (af)
  { case AF_INET:
       ia = va_arg(ap,struct in_addr);
       break;
    case AF_INET6:
       ia6 = va_arg(ap,struct in6_addr);
       break;
    default:
       panic("if_srt find_rt: impossible address family");
       break;
  }
 va_end(ap);
 for (i=0;i<sc->nrt;i++)
  { r = sc->rts[i];
    if (r->af != af) continue;
    switch (af)
     { case AF_INET:
	  if ((ia.s_addr & ipv4_masks[r->srcmask]) == r->srcmatch.v4.s_addr) return(r);
	  break;
       case AF_INET6:
	  if ((r->srcmask >= 8) && bcmp(&ia6,&r->srcmatch.v6,r->srcmask/8)) continue;
	  if ( (r->srcmask % 8) &&
	       ( ( ia6.s6_addr[r->srcmask/8] ^
		   r->srcmatch.v6.s6_addr[r->srcmask/8] ) &
		 0xff & (0xff00 >> (r->srcmask%8)) ) ) continue;
	  return(r);
	  break;
       default:
	  panic("if_srt find_rt: impossible address family 2");
	  break;
     }
  }
 return(0);
}

/* Network device interface. */

static int srt_if_ioctl(struct ifnet *intf, u_long cmd, void *data)
{
 struct ifaddr *ifa;
 struct ifreq *ifr;
 int s;
 int err;

 err = 0;
 s = splnet();
 switch (cmd)
  { case SIOCSIFADDR:
    case SIOCSIFDSTADDR:
       ifa = (void *) data;
       switch (ifa->ifa_addr->sa_family)
	{
#ifdef INET
	  case AF_INET:
#endif
#ifdef INET6
	  case AF_INET6:
#endif
	     break;
	  default:
	     err = EAFNOSUPPORT;
	     break;
	}
       /* XXX do we need to do more here for either of these? */
       break;
    case SIOCSIFMTU:
       ifr = (void *) data;
       ((SOFTC *)intf->if_softc)->intf.if_mtu = ifr->ifr_mtu;
       break;
    case SIOCGIFMTU:
       ifr = (void *) data;
       ifr->ifr_mtu = intf->if_mtu;
       break;
    default:
       err = EINVAL;
       break;
  }
 splx(s);
 return(err);
}

static int srt_if_output(
	struct ifnet *intf,
	struct mbuf *m,
	const struct sockaddr *to,
	struct rtentry *rtp )
{
 SOFTC *sc;
 RT *r;

 sc = intf->if_softc;
 if (! (intf->if_flags & IFF_UP))
  { m_freem(m);
    return(ENETDOWN);
  }
 switch (to->sa_family)
  {
#ifdef INET
    case AF_INET:
#endif
	{ struct ip *ip;
	  ip = mtod(m,struct ip *);
	  r = find_rt(sc,AF_INET,ip->ip_src);
	}
       break;
#ifdef INET6
    case AF_INET6:
#endif
	{ struct ip6_hdr *ip;
	  ip = mtod(m,struct ip6_hdr *);
	  r = find_rt(sc,AF_INET6,ip->ip6_src);
	}
       break;
    default:
       IF_DROP(&intf->if_snd);
       m_freem(m);
       return(EAFNOSUPPORT);
       break;
  }
 /* XXX Do we need to bpf_tap?  Or do higher layers now handle that? */
 /* if_gif.c seems to imply the latter. */
 intf->if_opackets ++;
 if (! r)
  { intf->if_oerrors ++;
    m_freem(m);
    return(0);
  }
 if (! (m->m_flags & M_PKTHDR))
  { printf("srt_if_output no PKTHDR\n");
    m_freem(m);
    return(0);
  }
 intf->if_obytes += m->m_pkthdr.len;
 if (! (r->u.dstifp->if_flags & IFF_UP))
  { m_freem(m);
    return(0); /* XXX ENETDOWN? */
  }
 /* XXX is 0 the right last arg here? */
 return((*r->u.dstifp->if_output)(r->u.dstifp,m,&r->dst.sa,0));
}

static int srt_clone_create(struct if_clone *cl, int unit)
{
 SOFTC *sc;

 if ((unit < 0) || (unit > SRT_MAXUNIT)) return(ENXIO);
 if (softcv[unit]) return(EBUSY);
 sc = malloc(sizeof(SOFTC),M_DEVBUF,M_WAIT);
 bzero(&sc->intf,sizeof(sc->intf)); /* XXX */
 sc->unit = unit;
 sc->nrt = 0;
 sc->rts = 0;
 sc->flags = 0;
 sc->kflags = 0;
 snprintf(&sc->intf.if_xname[0],sizeof(sc->intf.if_xname),"%s%d",cl->ifc_name,unit);
 sc->intf.if_softc = sc;
 sc->intf.if_mtu = 65535;
 sc->intf.if_flags = IFF_POINTOPOINT;
 sc->intf.if_type = IFT_OTHER;
 sc->intf.if_ioctl = &srt_if_ioctl;
 sc->intf.if_output = &srt_if_output;
 sc->intf.if_dlt = DLT_RAW;
 if_attach(&sc->intf);
 if_alloc_sadl(&sc->intf);
#if NBPFILTER > 0 /* see comment near top */
 bpfattach(&sc->intf,0/*???*/,0/*???*/);
#endif
 softcv[unit] = sc;
 return(0);
}

static int srt_clone_destroy(struct ifnet *intf)
{
 SOFTC *sc;

 sc = intf->if_softc;
 if ((intf->if_flags & IFF_UP) || (sc->kflags & SKF_CDEVOPEN)) return(EBUSY);
#if NBPFILTER > 0
 bpfdetach(intf);
#endif
 if_detach(intf);
 if ((sc->unit < 0) || (sc->unit > SRT_MAXUNIT))
  { panic("srt_clone_destroy: impossible unit %d\n",sc->unit);
  }
 if (softcv[sc->unit] != sc)
  { panic("srt_clone_destroy: bad backpointer ([%d]=%p not %p)\n",
			sc->unit,(void *)softcv[sc->unit],(void *)sc);
  }
 softcv[sc->unit] = 0;
 free(sc,M_DEVBUF);
 return(0);
}

struct if_clone srt_clone =
    IF_CLONE_INITIALIZER("srt",&srt_clone_create,&srt_clone_destroy);

extern void srtattach(void);
void srtattach(void)
{
 int i;

 for (i=SRT_MAXUNIT;i>=0;i--) softcv[i] = 0;
 global_flags = 0;
 if_clone_attach(&srt_clone);
}

/* Special-device interface. */

static int srt_open(dev_t dev, int flag, int mode, struct lwp *l)
{
 int unit;
 SOFTC *sc;

 unit = minor(dev);
 if ((unit < 0) || (unit > SRT_MAXUNIT)) return(ENXIO);
 sc = softcv[unit];
 if (! sc) return(ENXIO);
 sc->kflags |= SKF_CDEVOPEN;
 return(0);
}

static int srt_close(dev_t dev, int flag, int mode, struct lwp *l)
{
 int unit;
 SOFTC *sc;

 unit = minor(dev);
 if ((unit < 0) || (unit > SRT_MAXUNIT)) return(ENXIO);
 sc = softcv[unit];
 if (! sc) return(ENXIO);
 sc->kflags &= ~SKF_CDEVOPEN;
 return(0);
}

static int srt_ioctl(
	dev_t dev,
	u_long cmd,
	void *data,
	int flag,
	struct lwp *l )
{
 SOFTC *sc;
 RT *dr;
 RT *scr;
 struct ifnet *intf;
 char nbuf[IFNAMSIZ];

 sc = softcv[minor(dev)];
 if (! sc) panic("srt_ioctl: softc disappeared");
 switch (cmd)
  { case SRT_GETNRT:
       if (! (flag & FREAD)) return(EBADF);
       *(unsigned int *)data = sc->nrt;
       return(0);
       break;
    case SRT_GETRT:
       if (! (flag & FREAD)) return(EBADF);
       dr = (RT *) data;
       if (dr->inx >= sc->nrt) return(EDOM);
       scr = sc->rts[dr->inx];
       dr->af = scr->af;
       dr->srcmatch = scr->srcmatch;
       dr->srcmask = scr->srcmask;
       strncpy(&dr->u.dstifn[0],&scr->u.dstifp->if_xname[0],IFNAMSIZ);
       memcpy(&dr->dst,&scr->dst,scr->dst.sa.sa_len);
       return(0);
       break;
    case SRT_SETRT:
       if (! (flag & FWRITE)) return(EBADF);
       dr = (RT *) data;
       if (dr->inx > sc->nrt) return(EDOM);
       strncpy(&nbuf[0],&dr->u.dstifn[0],IFNAMSIZ);
       nbuf[IFNAMSIZ-1] = '\0';
       if (dr->dst.sa.sa_family != dr->af) return(EIO);
       switch (dr->af)
	{
#ifdef INET
	  case AF_INET:
	     if (dr->dst.sa.sa_len != sizeof(dr->dst.sin)) return(EIO);
	     if (dr->srcmask > 32) return(EIO);
	     break;
#endif
#ifdef INET6
	  case AF_INET6:
	     if (dr->dst.sa.sa_len != sizeof(dr->dst.sin6)) return(EIO);
	     if (dr->srcmask > 128) return(EIO);
	     break;
#endif
	     break;
	  default:
	     return(EAFNOSUPPORT);
	     break;
	}
       intf = ifunit(&nbuf[0]);
       if (intf == 0) return(ENXIO); /* needs translation */
       if (dr->inx == sc->nrt)
	{ RT **tmp;
	  tmp = malloc((sc->nrt+1)*sizeof(*tmp),M_DEVBUF,M_WAITOK);
	  if (tmp == 0) return(ENOBUFS);
	  tmp[sc->nrt] = 0;
	  if (sc->nrt > 0)
	   { memcpy(tmp,sc->rts,sc->nrt*sizeof(*tmp));
	     free(sc->rts,M_DEVBUF);
	   }
	  sc->rts = tmp;
	  sc->nrt ++;
	}
       scr = sc->rts[dr->inx];
       if (scr == 0)
	{ scr = malloc(sizeof(RT),M_DEVBUF,M_WAITOK);
	  if (scr == 0) return(ENOBUFS);
	  scr->inx = dr->inx;
	  scr->af = AF_UNSPEC;
	  sc->rts[dr->inx] = scr;
	}
       scr->af = dr->af;
       scr->srcmatch = dr->srcmatch;
       scr->srcmask = dr->srcmask;
       scr->u.dstifp = intf;
       memcpy(&scr->dst,&dr->dst,dr->dst.sa.sa_len);
       update_mtu(sc);
       return(0);
       break;
    case SRT_DELRT:
	{ unsigned int i;
	  if (! (flag & FWRITE)) return(EBADF);
	  i = *(unsigned int *)data;
	  if (i >= sc->nrt) return(EDOM);
	  scr = sc->rts[i];
	  sc->rts[i] = 0;
	  free(scr,M_DEVBUF);
	  sc->nrt --;
	  if (i < sc->nrt) memcpy(sc->rts+i,sc->rts+i+1,(sc->nrt-i)*sizeof(*sc->rts));
	  if (sc->nrt == 0)
	   { free(sc->rts,M_DEVBUF);
	     sc->rts = 0;
	     sc->intf.if_flags &= ~IFF_UP;
	   }
	}
       update_mtu(sc);
       return(0);
       break;
    case SRT_SFLAGS:
	{ unsigned int f;
	  if (! (flag & FWRITE)) return(EBADF);
	  f = *(unsigned int *)data & SSF_UCHG;
	  global_flags = (global_flags & ~SSF_UCHG) | (f & SSF_GLOBAL);
	  sc->flags = (sc->flags & ~SSF_UCHG) | (f & ~SSF_GLOBAL);
	}
       return(0);
       break;
    case SRT_GFLAGS:
       if (! (flag & FREAD)) return(EBADF);
       *(unsigned int *)data = sc->flags | global_flags;
       return(0);
       break;
    case SRT_SGFLAGS:
	{ unsigned int o;
	  unsigned int n;
	  if ((flag & (FWRITE|FREAD)) != (FWRITE|FREAD)) return(EBADF);
	  o = sc->flags | global_flags;
	  n = *(unsigned int *)data & SSF_UCHG;
	  global_flags = (global_flags & ~SSF_UCHG) | (n & SSF_GLOBAL);
	  sc->flags = (sc->flags & ~SSF_UCHG) | (n & ~SSF_GLOBAL);
	  *(unsigned int *)data = o;
	}
       return(0);
       break;
    case SRT_DEBUG:
       return(0);
       break;
  }
 return(ENOTTY);
}

const struct cdevsw srt_cdevsw
 = { &srt_open,
     &srt_close,
     nullread,
     nullwrite,
     &srt_ioctl,
     nullstop,
     notty,
     nullpoll,
     nommap,
     nullkqfilter,
     D_OTHER };