/*
 *	BIRD -- OSPF
 *
 *	(c) 1999--2005 Ondrej Filip <feela@network.cz>
 *	(c) 2009--2012 Ondrej Zajicek <santiago@crfreenet.org>
 *	(c) 2009--2012 CZ.NIC z.s.p.o.
 *
 *	Can be freely distributed and used under the terms of the GNU GPL.
 */

#include "ospf.h"
#include "nest/password.h"
#include "lib/md5.h"

void
ospf_pkt_fill_hdr(struct ospf_iface *ifa, void *buf, u8 h_type)
{
  struct proto_ospf *po = ifa->oa->po;
  struct ospf_packet *pkt;

  pkt = (struct ospf_packet *) buf;

  pkt->version = ospf_get_version(po);

  pkt->type = h_type;

  pkt->routerid = htonl(po->router_id);
  pkt->areaid = htonl(ifa->oa->areaid);
  pkt->checksum = 0;

  if (ospf_is_v2(po))
    ospf_pkt_set_autype(pkt, ifa->autype);
  else
    ospf_pkt_set_instance_id(pkt, ifa->instance_id);
}

unsigned
ospf_pkt_maxsize(struct ospf_iface *ifa)
{
  unsigned mtu = (ifa->type == OSPF_IT_VLINK) ? OSPF_VLINK_MTU : ifa->iface->mtu;
  unsigned headers = SIZE_OF_IP_HEADER;

  /* For OSPFv2 */
  if (ifa->autype == OSPF_AUTH_CRYPT)
    headers += OSPF_AUTH_CRYPT_SIZE;

  return mtu - headers;
}


/* We assume OSPFv2 in ospf_pkt_finalize() */
static void
ospf_pkt_finalize(struct ospf_iface *ifa, struct ospf_packet *pkt)
{
  struct password_item *passwd = NULL;
  union ospf_auth *auth = (void *) (pkt + 1);
  unsigned plen = ntohs(pkt->length);

  pkt->checksum = 0;
  ospf_pkt_set_autype(pkt, ifa->autype);
  bzero(auth, sizeof(union ospf_auth));

  /* Compatibility note: auth may contain anything if autype is
     none, but nonzero values do not work with Mikrotik OSPF */

  switch (ifa->autype)
  {
    case OSPF_AUTH_SIMPLE:
      passwd = password_find(ifa->passwords, 1);
      if (!passwd)
      {
        log(L_ERR "No suitable password found for authentication");
        return;
      }
      password_cpy(auth->password, passwd->password, sizeof(union ospf_auth));

    case OSPF_AUTH_NONE:
      {
	void *body = (void *) (auth + 1);
	unsigned blen = plen - sizeof(struct ospf_packet) - sizeof(union ospf_auth);
	pkt->checksum = ipsum_calculate(pkt, sizeof(struct ospf_packet), body, blen, NULL);
      }
      break;

    case OSPF_AUTH_CRYPT:
      passwd = password_find(ifa->passwords, 0);
      if (!passwd)
      {
        log( L_ERR "No suitable password found for authentication" );
        return;
      }

      /* Perhaps use random value to prevent replay attacks after
	 reboot when system does not have independent RTC? */
      if (!ifa->csn)
	{
	  ifa->csn = (u32) now;
	  ifa->csn_use = now;
	}

      /* We must have sufficient delay between sending a packet and increasing 
	 CSN to prevent reordering of packets (in a network) with different CSNs */
      if ((now - ifa->csn_use) > 1)
	ifa->csn++;

      ifa->csn_use = now;

      auth->md5.zero = 0;
      auth->md5.keyid = passwd->id;
      auth->md5.len = OSPF_AUTH_CRYPT_SIZE;
      auth->md5.csn = htonl(ifa->csn);

      {
	void *tail = ((void *) pkt) + plen;
	char password[OSPF_AUTH_CRYPT_SIZE];
	password_cpy(password, passwd->password, OSPF_AUTH_CRYPT_SIZE);

	struct MD5Context ctxt;
	MD5Init(&ctxt);
	MD5Update(&ctxt, (char *) pkt, plen);
	MD5Update(&ctxt, password, OSPF_AUTH_CRYPT_SIZE);
	MD5Final(tail, &ctxt);
      }
      break;

    default:
      bug("Unknown authentication type");
  }
}

/* We assume OSPFv2 in ospf_pkt_checkauth() */
static int
ospf_pkt_checkauth(struct ospf_neighbor *n, struct ospf_iface *ifa, struct ospf_packet *pkt, int size)
{
  struct proto_ospf *po = ifa->oa->po;
  union ospf_auth *auth = (void *) (pkt + 1);
  struct password_item *pass = NULL, *ptmp;
  char password[OSPF_AUTH_CRYPT_SIZE];

  unsigned plen = ntohs(pkt->length);
  u16 autype = ospf_pkt_get_autype(pkt);

  if (autype != ifa->autype)
  {
    OSPF_TRACE(D_PACKETS, "OSPF_auth: Method differs (%d)", autype);
    return 0;
  }

  switch (autype)
  {
    case OSPF_AUTH_NONE:
      return 1;
      break;

    case OSPF_AUTH_SIMPLE:
      pass = password_find(ifa->passwords, 1);
      if (!pass)
      {
        OSPF_TRACE(D_PACKETS, "OSPF_auth: no password found");
	return 0;
      }
      password_cpy(password, pass->password, sizeof(union ospf_auth));

      if (memcmp(auth->password, password, sizeof(union ospf_auth)))
      {
        char ppass[sizeof(union ospf_auth) + 1];
        bzero(ppass, (sizeof(union ospf_auth) + 1));
        memcpy(ppass, auth->password, sizeof(union ospf_auth));
        OSPF_TRACE(D_PACKETS, "OSPF_auth: different passwords (%s)", ppass);
	return 0;
      }
      return 1;
      break;

    case OSPF_AUTH_CRYPT:
      if (auth->md5.len != OSPF_AUTH_CRYPT_SIZE)
      {
        OSPF_TRACE(D_PACKETS, "OSPF_auth: wrong size of md5 digest");
        return 0;
      }

      if (plen + OSPF_AUTH_CRYPT_SIZE > size)
      {
        OSPF_TRACE(D_PACKETS, "OSPF_auth: size mismatch (%d vs %d)",
		   plen + OSPF_AUTH_CRYPT_SIZE, size);
        return 0;
      }

      if (n)
      {
	u32 rcv_csn = ntohl(auth->md5.csn);
	if(rcv_csn < n->csn)
	{
	  OSPF_TRACE(D_PACKETS, "OSPF_auth: lower sequence number (rcv %d, old %d)", rcv_csn, n->csn);
	  return 0;
	}

	n->csn = rcv_csn;
      }

      if (ifa->passwords)
      {
	WALK_LIST(ptmp, *(ifa->passwords))
	{
	  if (auth->md5.keyid != ptmp->id) continue;
	  if ((ptmp->accfrom > now_real) || (ptmp->accto < now_real)) continue;
	  pass = ptmp;
	  break;
	}
      }

      if (!pass)
      {
        OSPF_TRACE(D_PACKETS, "OSPF_auth: no suitable md5 password found");
        return 0;
      }

      password_cpy(password, pass->password, OSPF_AUTH_CRYPT_SIZE);

      {
	void *tail = ((void *) pkt) + plen;
	char md5sum[OSPF_AUTH_CRYPT_SIZE];

	struct MD5Context ctxt;
	MD5Init(&ctxt);
	MD5Update(&ctxt, (char *) pkt, plen);
	MD5Update(&ctxt, password, OSPF_AUTH_CRYPT_SIZE);
	MD5Final(md5sum, &ctxt);
	if (memcmp(md5sum, tail, OSPF_AUTH_CRYPT_SIZE))
	{
	  OSPF_TRACE(D_PACKETS, "OSPF_auth: wrong md5 digest");
	  return 0;
	}
      }
      return 1;
      break;

    default:
      OSPF_TRACE(D_PACKETS, "OSPF_auth: unknown auth type");
      return 0;
  }
}


/**
 * ospf_rx_hook
 * @sk: socket we received the packet.
 * @size: size of the packet
 *
 * This is the entry point for messages from neighbors. Many checks (like
 * authentication, checksums, size) are done before the packet is passed to
 * non generic functions.
 */
int
ospf_rx_hook(sock *sk, int size)
{
  char *mesg = "OSPF: Bad packet from ";

  /* We want just packets from sk->iface. Unfortunately, on BSD we
     cannot filter out other packets at kernel level and we receive
     all packets on all sockets */
  if (sk->lifindex != sk->iface->index)
    return 1;

  DBG("OSPF: RX hook called (iface %s, src %I, dst %I)\n",
      sk->iface->name, sk->faddr, sk->laddr);

  /* Initially, the packet is associated with the 'master' iface */
  struct ospf_iface *ifa = sk->data;
  struct proto_ospf *po = ifa->oa->po;
  // struct proto *p = &po->proto;

  int src_local, dst_local UNUSED, dst_mcast; 
  src_local = ipa_in_net(sk->faddr, ifa->addr->prefix, ifa->addr->pxlen);
  dst_local = ipa_equal(sk->laddr, ifa->addr->ip);
  dst_mcast = ipa_equal(sk->laddr, ifa->all_routers) || ipa_equal(sk->laddr, ifa->des_routers);

  if (ospf_is_v2(po))
  {
    /* First, we eliminate packets with strange address combinations.
     * In OSPFv2, they might be for other ospf_ifaces (with different IP
     * prefix) on the same real iface, so we don't log it. We enforce
     * that (src_local || dst_local), therefore we are eliminating all
     * such cases. 
     */
    if (dst_mcast && !src_local)
      return 1;
    if (!dst_mcast && !dst_local)
      return 1;

    /* Ignore my own broadcast packets */
    if (ifa->cf->real_bcast && ipa_equal(sk->faddr, ifa->addr->ip))
      return 1;
  }
  else
  {
    /* In OSPFv3, src_local and dst_local mean link-local. 
     * RFC 5340 says that local (non-vlink) packets use
     * link-local src address, but does not enforce it. Strange.
     */
    if (dst_mcast && !src_local)
      log(L_WARN "OSPF: Received multicast packet from %I (not link-local)", sk->faddr);
  }

  /* Second, we check packet size, checksum, and the protocol version */
  struct ospf_packet *pkt = (struct ospf_packet *) ip_skip_header(sk->rbuf, &size);

  if (pkt == NULL)
  {
    log(L_ERR "%s%I - bad IP header", mesg, sk->faddr);
    return 1;
  }

  if (ifa->check_ttl && (sk->ttl < 255))
  {
    log(L_ERR "%s%I - TTL %d (< 255)", mesg, sk->faddr, sk->ttl);
    return 1;
  }

  if ((unsigned) size < sizeof(struct ospf_packet))
  {
    log(L_ERR "%s%I - too short (%u bytes)", mesg, sk->faddr, size);
    return 1;
  }

  unsigned plen = ntohs(pkt->length);
  if (plen < sizeof(struct ospf_packet))
  {
    log(L_ERR "%s%I - too low value in size field (%u bytes)", mesg, sk->faddr, plen);
    return 1;
  }

  if ((plen > size) || ((plen % 4) != 0))
  {
    log(L_ERR "%s%I - size field does not match (%d/%d)", mesg, sk->faddr, plen, size);
    return 1;
  }

  if ((unsigned) size > sk->rbsize)
  {
    log(L_ERR "%s%I - too large (%d vs %d)", mesg, sk->faddr, size, sk->rbsize);
    return 1;
  }

  if (pkt->version != ospf_get_version(po))
  {
    log(L_ERR "%s%I - version %u", mesg, sk->faddr, pkt->version);
    return 1;
  }

  if (ospf_is_v2(po) && ospf_pkt_get_autype(pkt) != OSPF_AUTH_CRYPT)
  {
    unsigned hlen = sizeof(struct ospf_packet) - sizeof(union ospf_auth);
    unsigned blen = plen - hlen;
    void *body = ((void *) pkt) + hlen;

    if (! ipsum_verify(pkt, sizeof(struct ospf_packet), body, blen, NULL))
    {
      log(L_ERR "%s%I - bad checksum", mesg, sk->faddr);
      return 1;
    }
  }

  /* Third, we resolve associated iface and handle vlinks. */

  u32 areaid = ntohl(pkt->areaid);
  u32 rid = ntohl(pkt->routerid);
  u8 instance_id = ospf_is_v2(po) ? 0 : ospf_pkt_get_instance_id(pkt);

  if ((areaid == ifa->oa->areaid) &&
      (instance_id == ifa->instance_id))
  {
    /* It is real iface, source should be local (in OSPFv2) */
    if (ospf_is_v2(po) && !src_local)
      return 1;
  }
  else if (dst_mcast || (areaid != 0))
  {
    /* Obvious mismatch */

    /* We ignore mismatch in OSPFv3 when there might be
       another instance with a different instance ID */
    if (instance_id == ifa->instance_id)
      log(L_ERR "%s%I - area does not match (%R vs %R)",
	  mesg, sk->faddr, areaid, ifa->oa->areaid);
    return 1;
  }
  else
  {
    /* Some vlink? */
    struct ospf_iface *iff = NULL;

    WALK_LIST(iff, po->iface_list)
    {
      if ((iff->type == OSPF_IT_VLINK) && 
	  (iff->voa == ifa->oa) &&
	  (iff->instance_id == instance_id) &&
	  (iff->vid == rid))
	{
	  /* Vlink should be UP */
	  if (iff->state != OSPF_IS_PTP)
	    return 1;
	  
	  ifa = iff;
	  goto found;
	}
    }

    /* FIXME: this warning is strange - NBMA could trigger it too */
    if (ospf_is_v2(po))
      log(L_WARN "OSPF: Received packet for unknown vlink (ID %R, IP %I)", rid, sk->faddr);

    return 1;
  }

 found:
  if (ifa->stub)	    /* This shouldn't happen */
    return 1;

  if (ipa_equal(sk->laddr, ifa->des_routers) && (ifa->sk_dr == 0))
    return 1;

  if (rid == po->router_id)
  {
    log(L_ERR "%s%I - received my own router ID!", mesg, sk->faddr);
    return 1;
  }

  if (rid == 0)
  {
    log(L_ERR "%s%I - router id = 0.0.0.0", mesg, sk->faddr);
    return 1;
  }

  /* In OSPFv2, neighbors are identified by either IP or Router ID, base on network type */
  unsigned t = ifa->type;
  struct ospf_neighbor *n;
  if (ospf_is_v2(po) && ((t == OSPF_IT_BCAST) || (t == OSPF_IT_NBMA) || (t == OSPF_IT_PTMP)))
    n = find_neigh_by_ip(ifa, sk->faddr);
  else
    n = find_neigh(ifa, rid);

  if (!n && (pkt->type != HELLO_P))
  {
    log(L_WARN "OSPF: Received non-hello packet from unknown neighbor (src %I, iface %s)",
	sk->faddr, ifa->iface->name);
    return 1;
  }

  if (ospf_is_v2(po) && ! ospf_pkt_checkauth(n, ifa, pkt, size))
  {
    log(L_ERR "%s%I - authentication failed", mesg, sk->faddr);
    return 1;
  }

  /* Dump packet 
     pu8=(u8 *)(sk->rbuf+5*4);
     for(i=0;i<ntohs(pkt->length);i+=4)
     DBG("%s: received %u,%u,%u,%u\n",p->name, pu8[i+0], pu8[i+1], pu8[i+2],
     pu8[i+3]);
     DBG("%s: received size: %u\n",p->name,size);
   */

  switch (pkt->type)
  {
  case HELLO_P:
    DBG("%s: Hello received.\n", p->name);
    ospf_hello_receive(pkt, ifa, n, sk->faddr);
    break;
  case DBDES_P:
    DBG("%s: Database description received.\n", p->name);
    ospf_dbdes_receive(pkt, ifa, n);
    break;
  case LSREQ_P:
    DBG("%s: Link state request received.\n", p->name);
    ospf_lsreq_receive(pkt, ifa, n);
    break;
  case LSUPD_P:
    DBG("%s: Link state update received.\n", p->name);
    ospf_lsupd_receive(pkt, ifa, n);
    break;
  case LSACK_P:
    DBG("%s: Link state ack received.\n", p->name);
    ospf_lsack_receive(pkt, ifa, n);
    break;
  default:
    log(L_ERR "%s%I - wrong type %u", mesg, sk->faddr, pkt->type);
    return 1;
  };
  return 1;
}

void
ospf_tx_hook(sock * sk)
{
  struct ospf_iface *ifa= (struct ospf_iface *) (sk->data);
//  struct proto *p = (struct proto *) (ifa->oa->po);
  log(L_ERR "OSPF: TX hook called on %s", ifa->iface->name);
}

void
ospf_err_hook(sock * sk, int err)
{
  struct ospf_iface *ifa= (struct ospf_iface *) (sk->data);
//  struct proto *p = (struct proto *) (ifa->oa->po);
  log(L_ERR "OSPF: Socket error on %s: %M", ifa->iface->name, err);
}

void
ospf_send_to(struct ospf_iface *ifa, ip_addr dst)
{
  sock *sk = ifa->sk;
  struct ospf_packet *pkt = (struct ospf_packet *) sk->tbuf;
  int plen = ntohs(pkt->length);

  if (sk->tbuf != sk->tpos)
    log(L_ERR "Aiee, old packet was overwritten in TX buffer");

  if (ospf_is_v2(ifa->oa->po))
  {
    if (ifa->autype == OSPF_AUTH_CRYPT)
      plen += OSPF_AUTH_CRYPT_SIZE;

    ospf_pkt_finalize(ifa, pkt);
  }

  sk_send_to(sk, plen, dst, 0);
}

void
ospf_send_to_agt(struct ospf_iface *ifa, u8 state)
{
  struct ospf_neighbor *n;

  WALK_LIST(n, ifa->neigh_list)
    if (n->state >= state)
      ospf_send_to(ifa, n->ip);
}

void
ospf_send_to_bdr(struct ospf_iface *ifa)
{
  if (ipa_nonzero(ifa->drip))
    ospf_send_to(ifa, ifa->drip);
  if (ipa_nonzero(ifa->bdrip))
    ospf_send_to(ifa, ifa->bdrip);
}