/*      $NetBSD: ip.c,v 1.18 2014/06/03 22:22:41 joerg Exp $    */

/*
* Copyright (c) 1999, 2000 Andrew Doran <[email protected]>
* 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.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR 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 THE AUTHOR 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 <sys/cdefs.h>
#ifndef lint
__RCSID("$NetBSD: ip.c,v 1.18 2014/06/03 22:22:41 joerg Exp $");
#endif /* not lint */

#include <sys/param.h>
#include <sys/sysctl.h>

#include <netinet/in.h>
#include <netinet/in_systm.h>
#include <netinet/ip.h>
#include <netinet/ip_var.h>
#include <netinet/udp.h>
#include <netinet/udp_var.h>

#include <string.h>

#include "systat.h"
#include "extern.h"

#define LHD(row, str)           mvwprintw(wnd, row, 10, str)
#define RHD(row, str)           mvwprintw(wnd, row, 45, str);
#define SHOW(stat, row, col) \
   mvwprintw(wnd, row, col, "%9llu", (unsigned long long)curstat.stat)

struct mystat {
       uint64_t i[IP_NSTATS];
       uint64_t u[UDP_NSTATS];
};

enum update {
       UPDATE_TIME,
       UPDATE_BOOT,
       UPDATE_RUN,
};

static enum update update = UPDATE_TIME;
static struct mystat curstat;
static struct mystat oldstat;
static struct mystat newstat;

WINDOW *
openip(void)
{

       return (subwin(stdscr, -1, 0, 5, 0));
}

void
closeip(WINDOW *w)
{

       if (w != NULL) {
               wclear(w);
               wrefresh(w);
               delwin(w);
       }
}

void
labelip(void)
{

       wmove(wnd, 0, 0); wclrtoeol(wnd);

       LHD(0,  "total packets received");
       LHD(1,  "  passed to upper layers");
       LHD(2,  "  with bad checksums");
       LHD(3,  "  too short for header");
       LHD(4,  "  too short for data");
       LHD(5,  "  with invalid hlen");
       LHD(6,  "  with invalid length");
       LHD(7,  "  with invalid version");
       LHD(8,  "  too large");
       LHD(9,  "  option errors");
       LHD(10, "  fragments received");
       LHD(11, "  fragments dropped");
       LHD(12, "  fragments timed out");
       LHD(13, "  packets reassembled ok");

       LHD(15, "packets forwarded");
       LHD(16, "  fast forwarded");
       LHD(17, "  unreachable dests");
       LHD(18, "  redirects generated");

       RHD(0,  "total packets sent");
       RHD(1,  "  generated locally");
       RHD(2,  "  output drops");
       RHD(3,  "  output fragments generated");
       RHD(4,  "  fragmentation failed");
       RHD(5,  "  destinations unreachable");
       RHD(6,  "  packets output via raw IP");
       RHD(7,  "  total UDP packets sent");

       RHD(9, "total UDP packets received");
       RHD(10, "  too short for header");
       RHD(11, "  invalid checksum");
       RHD(12, "  invalid length");
       RHD(13, "  no socket for dest port");
       RHD(14, "  no socket for broadcast");
       RHD(15, "  socket buffer full");
}

void
showip(void)
{
       u_quad_t totalout;

       totalout = curstat.i[IP_STAT_FORWARD] + curstat.i[IP_STAT_LOCALOUT];

       SHOW(i[IP_STAT_TOTAL], 0, 0);
       mvwprintw(wnd, 0, 35, "%9llu", (unsigned long long)totalout);
       SHOW(i[IP_STAT_DELIVERED], 1, 0);
       SHOW(i[IP_STAT_BADSUM], 2, 0);
       SHOW(i[IP_STAT_TOOSHORT], 3, 0);
       SHOW(i[IP_STAT_TOOSMALL], 4, 0);
       SHOW(i[IP_STAT_BADHLEN], 5, 0);
       SHOW(i[IP_STAT_BADLEN], 6, 0);
       SHOW(i[IP_STAT_BADVERS], 7, 0);
       SHOW(i[IP_STAT_TOOLONG], 8, 0);
       SHOW(i[IP_STAT_BADOPTIONS], 9, 0);

       SHOW(i[IP_STAT_LOCALOUT], 1, 35);
       SHOW(i[IP_STAT_ODROPPED], 2, 35);
       SHOW(i[IP_STAT_OFRAGMENTS], 3, 35);
       SHOW(i[IP_STAT_CANTFRAG], 4, 35);
       SHOW(i[IP_STAT_NOROUTE], 5, 35);
       SHOW(i[IP_STAT_RAWOUT], 6, 35);
       SHOW(u[UDP_STAT_OPACKETS], 7, 35);

       SHOW(i[IP_STAT_FRAGMENTS], 10, 0);
       SHOW(i[IP_STAT_FRAGDROPPED], 11, 0);
       SHOW(i[IP_STAT_FRAGTIMEOUT], 12, 0);
       SHOW(i[IP_STAT_REASSEMBLED], 13, 0);

       SHOW(i[IP_STAT_FORWARD], 15, 0);
       SHOW(i[IP_STAT_FASTFORWARD], 16, 0);
       SHOW(i[IP_STAT_CANTFORWARD], 17, 0);
       SHOW(i[IP_STAT_REDIRECTSENT], 18, 0);

       SHOW(u[UDP_STAT_IPACKETS], 9, 35);
       SHOW(u[UDP_STAT_HDROPS], 10, 35);
       SHOW(u[UDP_STAT_BADSUM], 11, 35);
       SHOW(u[UDP_STAT_BADLEN], 12, 35);
       SHOW(u[UDP_STAT_NOPORT], 13, 35);
       SHOW(u[UDP_STAT_NOPORTBCAST], 14, 35);
       SHOW(u[UDP_STAT_FULLSOCK], 15, 35);
}

int
initip(void)
{

       return 1;
}

void
fetchip(void)
{
       size_t size;

       size = sizeof(newstat.i);
       if (sysctlbyname("net.inet.ip.stats", newstat.i, &size, NULL, 0) == -1)
               return;
       size = sizeof(newstat.u);
       if (sysctlbyname("net.inet.udp.stats", newstat.u, &size, NULL, 0) == -1)
               return;

       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_TOTAL]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_DELIVERED]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_BADSUM]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_TOOSHORT]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_TOOSMALL]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_BADHLEN]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_BADLEN]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_BADVERS]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_TOOLONG]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_BADOPTIONS]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_LOCALOUT]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_ODROPPED]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_OFRAGMENTS]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_CANTFRAG]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_NOROUTE]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_RAWOUT]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_FRAGMENTS]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_FRAGDROPPED]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_FRAGTIMEOUT]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_REASSEMBLED]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_FORWARD]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_FASTFORWARD]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_CANTFORWARD]);
       ADJINETCTR(curstat, oldstat, newstat, i[IP_STAT_REDIRECTSENT]);
       ADJINETCTR(curstat, oldstat, newstat, u[UDP_STAT_OPACKETS]);
       ADJINETCTR(curstat, oldstat, newstat, u[UDP_STAT_IPACKETS]);
       ADJINETCTR(curstat, oldstat, newstat, u[UDP_STAT_HDROPS]);
       ADJINETCTR(curstat, oldstat, newstat, u[UDP_STAT_BADSUM]);
       ADJINETCTR(curstat, oldstat, newstat, u[UDP_STAT_BADLEN]);
       ADJINETCTR(curstat, oldstat, newstat, u[UDP_STAT_NOPORT]);
       ADJINETCTR(curstat, oldstat, newstat, u[UDP_STAT_NOPORTBCAST]);
       ADJINETCTR(curstat, oldstat, newstat, u[UDP_STAT_FULLSOCK]);

       if (update == UPDATE_TIME)
               memcpy(&oldstat, &newstat, sizeof(oldstat));
}

void
ip_boot(char *args)
{

       memset(&oldstat, 0, sizeof(oldstat));
       update = UPDATE_BOOT;
}

void
ip_run(char *args)
{

       if (update != UPDATE_RUN) {
               memcpy(&oldstat, &newstat, sizeof(oldstat));
               update = UPDATE_RUN;
       }
}

void
ip_time(char *args)
{

       if (update != UPDATE_TIME) {
               memcpy(&oldstat, &newstat, sizeof(oldstat));
               update = UPDATE_TIME;
       }
}

void
ip_zero(char *args)
{

       if (update == UPDATE_RUN)
               memcpy(&oldstat, &newstat, sizeof(oldstat));
}