/*
 * Copyright (c) 2003 Bruce M. Simpson <bms@spc.org>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *        This product includes software developed by Bruce M. Simpson.
 * 4. Neither the name of Bruce M. Simpson nor the names of co-
 *    contributors may be used to endorse or promote products derived
 *    from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY Bruce M. Simpson AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL Bruce M. Simpson OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include <unistd.h>
#include <stdlib.h>
#include <stddef.h>

#include <argus_compat.h>
#include <rabins.h>
#include <argus_util.h>
#include <argus_client.h>
#include <argus_main.h>
#include <argus_filter.h>

#include <signal.h>
#include <ctype.h>
#include <argus/extract.h>

extern u_char *snapend;

#include "interface.h"
#include "aodv.h"

extern char ArgusBuf[];


static void
aodv_extension(const struct aodv_ext *ep, u_int length)
{
   u_int i;
   const struct aodv_hello *ah;

   switch (ep->type) {
   case AODV_EXT_HELLO:
      if (snapend < (u_char *) ep) {
         sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|hello]");
         return;
      }
      i = min(length, (u_int)(snapend - (u_char *)ep));
      if (i < sizeof(struct aodv_hello)) {
         sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|hello]");
         return;
      }
      i -= sizeof(struct aodv_hello);
      ah = (void *)ep;
      sprintf(&ArgusBuf[strlen(ArgusBuf)],"\n\text HELLO %ld ms",
          (unsigned long)EXTRACT_32BITS(&ah->interval));
      break;

   default:
      sprintf(&ArgusBuf[strlen(ArgusBuf)],"\n\text %u %u", ep->type, ep->length);
      break;
   }
}

static void
aodv_rreq(const union aodv *ap, const u_char *dat, u_int length)
{
   u_int i;

   if (snapend < dat) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|aodv]");
      return;
   }
   i = min(length, (u_int)(snapend - dat));
   if (i < sizeof(ap->rreq)) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|rreq]");
      return;
   }
   i -= sizeof(ap->rreq);
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," rreq %u %s%s%s%s%shops %u id 0x%08lx\n"
       "\tdst %s seq %lu src %s seq %lu", length,
       ap->rreq.rreq_type & RREQ_JOIN ? "[J]" : "",
       ap->rreq.rreq_type & RREQ_REPAIR ? "[R]" : "",
       ap->rreq.rreq_type & RREQ_GRAT ? "[G]" : "",
       ap->rreq.rreq_type & RREQ_DEST ? "[D]" : "",
       ap->rreq.rreq_type & RREQ_UNKNOWN ? "[U] " : " ",
       ap->rreq.rreq_hops,
       (unsigned long)EXTRACT_32BITS(&ap->rreq.rreq_id),
       ipaddr_string(&ap->rreq.rreq_da),
       (unsigned long)EXTRACT_32BITS(&ap->rreq.rreq_ds),
       ipaddr_string(&ap->rreq.rreq_oa),
       (unsigned long)EXTRACT_32BITS(&ap->rreq.rreq_os));
   if (i >= sizeof(struct aodv_ext))
      aodv_extension((void *)(&ap->rreq + 1), i);
}

static void
aodv_rrep(const union aodv *ap, const u_char *dat, u_int length)
{
   u_int i;

   if (snapend < dat) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|aodv]");
      return;
   }
   i = min(length, (u_int)(snapend - dat));
   if (i < sizeof(ap->rrep)) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|rrep]");
      return;
   }
   i -= sizeof(ap->rrep);
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," rrep %u %s%sprefix %u hops %u\n"
       "\tdst %s dseq %lu src %s %lu ms", length,
       ap->rrep.rrep_type & RREP_REPAIR ? "[R]" : "",
       ap->rrep.rrep_type & RREP_ACK ? "[A] " : " ",
       ap->rrep.rrep_ps & RREP_PREFIX_MASK,
       ap->rrep.rrep_hops,
       ipaddr_string(&ap->rrep.rrep_da),
       (unsigned long)EXTRACT_32BITS(&ap->rrep.rrep_ds),
       ipaddr_string(&ap->rrep.rrep_oa),
       (unsigned long)EXTRACT_32BITS(&ap->rrep.rrep_life));
   if (i >= sizeof(struct aodv_ext))
      aodv_extension((void *)(&ap->rrep + 1), i);
}

static void
aodv_rerr(const union aodv *ap, const u_char *dat, u_int length)
{
   u_int i;
   const struct rerr_unreach *dp = NULL;
   int n, trunc;

   if (snapend < dat) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|aodv]");
      return;
   }
   i = min(length, (u_int)(snapend - dat));
   if (i < offsetof(struct aodv_rerr, r)) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|rerr]");
      return;
   }
   i -= offsetof(struct aodv_rerr, r);
   dp = &ap->rerr.r.dest[0];
   n = ap->rerr.rerr_dc * sizeof(ap->rerr.r.dest[0]);
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," rerr %s [items %u] [%u]:",
       ap->rerr.rerr_flags & RERR_NODELETE ? "[D]" : "",
       ap->rerr.rerr_dc, length);
   trunc = n - (i/sizeof(ap->rerr.r.dest[0]));
   for (; i >= sizeof(ap->rerr.r.dest[0]);
       ++dp, i -= sizeof(ap->rerr.r.dest[0])) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," {%s}(%ld)", ipaddr_string(&dp->u_da),
          (unsigned long)EXTRACT_32BITS(&dp->u_ds));
   }
   if (trunc)
      sprintf(&ArgusBuf[strlen(ArgusBuf)],"[|rerr]");
}

static void
#ifdef INET6
aodv_v6_rreq(const union aodv *ap, const u_char *dat, u_int length)
#else
aodv_v6_rreq(const union aodv *ap, const u_char *dat, u_int length)
#endif
{
#ifdef INET6
   u_int i;

   if (snapend < dat) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|aodv]");
      return;
   }
   i = min(length, (u_int)(snapend - dat));
   if (i < sizeof(ap->rreq6)) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|rreq6]");
      return;
   }
   i -= sizeof(ap->rreq6);
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," v6 rreq %u %s%s%s%s%shops %u id 0x%08lx\n"
       "\tdst %s seq %lu src %s seq %lu", length,
       ap->rreq6.rreq_type & RREQ_JOIN ? "[J]" : "",
       ap->rreq6.rreq_type & RREQ_REPAIR ? "[R]" : "",
       ap->rreq6.rreq_type & RREQ_GRAT ? "[G]" : "",
       ap->rreq6.rreq_type & RREQ_DEST ? "[D]" : "",
       ap->rreq6.rreq_type & RREQ_UNKNOWN ? "[U] " : " ",
       ap->rreq6.rreq_hops,
       (unsigned long)EXTRACT_32BITS(&ap->rreq6.rreq_id),
       ip6addr_string(&ap->rreq6.rreq_da),
       (unsigned long)EXTRACT_32BITS(&ap->rreq6.rreq_ds),
       ip6addr_string(&ap->rreq6.rreq_oa),
       (unsigned long)EXTRACT_32BITS(&ap->rreq6.rreq_os));
   if (i >= sizeof(struct aodv_ext))
      aodv_extension((void *)(&ap->rreq6 + 1), i);
#else
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," v6 rreq %u", length);
#endif
}

static void
#ifdef INET6
aodv_v6_rrep(const union aodv *ap, const u_char *dat, u_int length)
#else
aodv_v6_rrep(const union aodv *ap, const u_char *dat, u_int length)
#endif
{
#ifdef INET6
   u_int i;

   if (snapend < dat) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|aodv]");
      return;
   }
   i = min(length, (u_int)(snapend - dat));
   if (i < sizeof(ap->rrep6)) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|rrep6]");
      return;
   }
   i -= sizeof(ap->rrep6);
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," rrep %u %s%sprefix %u hops %u\n"
      "\tdst %s dseq %lu src %s %lu ms", length,
       ap->rrep6.rrep_type & RREP_REPAIR ? "[R]" : "",
       ap->rrep6.rrep_type & RREP_ACK ? "[A] " : " ",
       ap->rrep6.rrep_ps & RREP_PREFIX_MASK,
       ap->rrep6.rrep_hops,
       ip6addr_string(&ap->rrep6.rrep_da),
       (unsigned long)EXTRACT_32BITS(&ap->rrep6.rrep_ds),
       ip6addr_string(&ap->rrep6.rrep_oa),
       (unsigned long)EXTRACT_32BITS(&ap->rrep6.rrep_life));
   if (i >= sizeof(struct aodv_ext))
      aodv_extension((void *)(&ap->rrep6 + 1), i);
#else
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," rrep %u", length);
#endif
}

static void
#ifdef INET6
aodv_v6_rerr(const union aodv *ap, u_int length)
#else
aodv_v6_rerr(const union aodv *ap, u_int length)
#endif
{
#ifdef INET6
   const struct rerr_unreach6 *dp6 = NULL;
   int i, j, n, trunc;

   i = length - offsetof(struct aodv_rerr, r);
   j = sizeof(ap->rerr.r.dest6[0]);
   dp6 = &ap->rerr.r.dest6[0];
   n = ap->rerr.rerr_dc * j;
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," rerr %s [items %u] [%u]:",
       ap->rerr.rerr_flags & RERR_NODELETE ? "[D]" : "",
       ap->rerr.rerr_dc, length);
   trunc = n - (i/j);
   for (; i -= j >= 0; ++dp6) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," {%s}(%ld)", ip6addr_string(&dp6->u_da),
          (unsigned long)EXTRACT_32BITS(&dp6->u_ds));
   }
   if (trunc)
      sprintf(&ArgusBuf[strlen(ArgusBuf)],"[|rerr]");
#else
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," rerr %u", length);
#endif
}

static void
#ifdef INET6
aodv_v6_draft_01_rreq(const union aodv *ap, const u_char *dat, u_int length)
#else
aodv_v6_draft_01_rreq(const union aodv *ap, const u_char *dat, u_int length)
#endif
{
#ifdef INET6
   u_int i;

   if (snapend < dat) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|aodv]");
      return;
   }
   i = min(length, (u_int)(snapend - dat));
   if (i < sizeof(ap->rreq6_draft_01)) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|rreq6]");
      return;
   }
   i -= sizeof(ap->rreq6_draft_01);
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," rreq %u %s%s%s%s%shops %u id 0x%08lx\n"
       "\tdst %s seq %lu src %s seq %lu", length,
       ap->rreq6_draft_01.rreq_type & RREQ_JOIN ? "[J]" : "",
       ap->rreq6_draft_01.rreq_type & RREQ_REPAIR ? "[R]" : "",
       ap->rreq6_draft_01.rreq_type & RREQ_GRAT ? "[G]" : "",
       ap->rreq6_draft_01.rreq_type & RREQ_DEST ? "[D]" : "",
       ap->rreq6_draft_01.rreq_type & RREQ_UNKNOWN ? "[U] " : " ",
       ap->rreq6_draft_01.rreq_hops,
       (unsigned long)EXTRACT_32BITS(&ap->rreq6_draft_01.rreq_id),
       ip6addr_string(&ap->rreq6_draft_01.rreq_da),
       (unsigned long)EXTRACT_32BITS(&ap->rreq6_draft_01.rreq_ds),
       ip6addr_string(&ap->rreq6_draft_01.rreq_oa),
       (unsigned long)EXTRACT_32BITS(&ap->rreq6_draft_01.rreq_os));
   if (i >= sizeof(struct aodv_ext))
      aodv_extension((void *)(&ap->rreq6_draft_01 + 1), i);
#else
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," rreq %u", length);
#endif
}

static void
#ifdef INET6
aodv_v6_draft_01_rrep(const union aodv *ap, const u_char *dat, u_int length)
#else
aodv_v6_draft_01_rrep(const union aodv *ap, const u_char *dat, u_int length)
#endif
{
#ifdef INET6
   u_int i;

   if (snapend < dat) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|aodv]");
      return;
   }
   i = min(length, (u_int)(snapend - dat));
   if (i < sizeof(ap->rrep6_draft_01)) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|rrep6]");
      return;
   }
   i -= sizeof(ap->rrep6_draft_01);
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," rrep %u %s%sprefix %u hops %u\n"
      "\tdst %s dseq %lu src %s %lu ms", length,
       ap->rrep6_draft_01.rrep_type & RREP_REPAIR ? "[R]" : "",
       ap->rrep6_draft_01.rrep_type & RREP_ACK ? "[A] " : " ",
       ap->rrep6_draft_01.rrep_ps & RREP_PREFIX_MASK,
       ap->rrep6_draft_01.rrep_hops,
       ip6addr_string(&ap->rrep6_draft_01.rrep_da),
       (unsigned long)EXTRACT_32BITS(&ap->rrep6_draft_01.rrep_ds),
       ip6addr_string(&ap->rrep6_draft_01.rrep_oa),
       (unsigned long)EXTRACT_32BITS(&ap->rrep6_draft_01.rrep_life));
   if (i >= sizeof(struct aodv_ext))
      aodv_extension((void *)(&ap->rrep6_draft_01 + 1), i);
#else
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," rrep %u", length);
#endif
}

static void
#ifdef INET6
aodv_v6_draft_01_rerr(const union aodv *ap, u_int length)
#else
aodv_v6_draft_01_rerr(const union aodv *ap, u_int length)
#endif
{
#ifdef INET6
   const struct rerr_unreach6_draft_01 *dp6 = NULL;
   int i, j, n, trunc;

   i = length - offsetof(struct aodv_rerr, r);
   j = sizeof(ap->rerr.r.dest6_draft_01[0]);
   dp6 = &ap->rerr.r.dest6_draft_01[0];
   n = ap->rerr.rerr_dc * j;
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," rerr %s [items %u] [%u]:",
       ap->rerr.rerr_flags & RERR_NODELETE ? "[D]" : "",
       ap->rerr.rerr_dc, length);
   trunc = n - (i/j);
   for (; i -= j >= 0; ++dp6) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," {%s}(%ld)", ip6addr_string(&dp6->u_da),
          (unsigned long)EXTRACT_32BITS(&dp6->u_ds));
   }
   if (trunc)
      sprintf(&ArgusBuf[strlen(ArgusBuf)],"[|rerr]");
#else
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," rerr %u", length);
#endif
}

char *
aodv_print(const u_char *dat, u_int length, int is_ip6)
{
   const union aodv *ap;

   ap = (union aodv *)dat;
   if (snapend < dat) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|aodv]");
      return ArgusBuf;
   }
   if (min(length, (u_int)(snapend - dat)) < sizeof(ap->rrep_ack)) {
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," [|aodv]");
      return ArgusBuf;
   }
   sprintf(&ArgusBuf[strlen(ArgusBuf)]," aodv");

   switch (ap->rerr.rerr_type) {

   case AODV_RREQ:
      if (is_ip6)
         aodv_v6_rreq(ap, dat, length);
      else
         aodv_rreq(ap, dat, length);
      break;

   case AODV_RREP:
      if (is_ip6)
         aodv_v6_rrep(ap, dat, length);
      else
         aodv_rrep(ap, dat, length);
      break;

   case AODV_RERR:
      if (is_ip6)
         aodv_v6_rerr(ap, length);
      else
         aodv_rerr(ap, dat, length);
      break;

   case AODV_RREP_ACK:
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," rrep-ack %u", length);
      break;

   case AODV_V6_DRAFT_01_RREQ:
      aodv_v6_draft_01_rreq(ap, dat, length);
      break;

   case AODV_V6_DRAFT_01_RREP:
      aodv_v6_draft_01_rrep(ap, dat, length);
      break;

   case AODV_V6_DRAFT_01_RERR:
      aodv_v6_draft_01_rerr(ap, length);
      break;

   case AODV_V6_DRAFT_01_RREP_ACK:
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," rrep-ack %u", length);
      break;

   default:
      sprintf(&ArgusBuf[strlen(ArgusBuf)]," %u %u", ap->rreq.rreq_type, length);
   }

   return ArgusBuf;
}
