/*      $NetBSD: atalk.c,v 1.21 2022/09/02 06:25:43 msaitoh Exp $       */

/*
* Copyright (c) 1983, 1988, 1993
*      The Regents of the University of California.  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. Neither the name of the University nor the names of its contributors
*    may be used to endorse or promote products derived from this software
*    without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS 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 REGENTS 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
#if 0
static char sccsid[] = "from @(#)atalk.c        1.1 (Whistle) 6/6/96";
#else
__RCSID("$NetBSD: atalk.c,v 1.21 2022/09/02 06:25:43 msaitoh Exp $");
#endif
#endif /* not lint */

#include <sys/param.h>
#include <sys/queue.h>
#include <sys/socket.h>
#include <sys/socketvar.h>
#include <sys/mbuf.h>
#include <sys/protosw.h>
#include <sys/sysctl.h>

#include <net/route.h>
#include <net/if.h>

#include <netinet/tcp_fsm.h>

#include <netatalk/at.h>
#include <netatalk/ddp_var.h>

#include <err.h>
#include <nlist.h>
#include <kvm.h>
#include <errno.h>
#include <stdio.h>
#include <string.h>
#include "netstat.h"
#include "prog_ops.h"

static int first = 1;

/*
* Print a summary of connections related to a Network Systems
* protocol.  For XXX, also give state of connection.
* Listening processes (aflag) are suppressed unless the
* -a (all) flag is specified.
*/

static const char *
at_pr_net(const struct sockaddr_at *sat, int numeric)
{
       static char mybuf[50];

       if (!numeric) {
               switch (sat->sat_addr.s_net) {
               case 0xffff:
                       return "????";
               case ATADDR_ANYNET:
                       return "*";
               }
       }
       (void)snprintf(mybuf, sizeof(mybuf), "%hu",
           ntohs(sat->sat_addr.s_net));
       return mybuf;
}

static const char *
at_pr_host(const struct sockaddr_at *sat, int numeric)
{
       static char mybuf[50];

       if (!numeric) {
               switch (sat->sat_addr.s_node) {
               case ATADDR_BCAST:
                       return "bcast";
               case ATADDR_ANYNODE:
                       return "*";
               }
       }
       (void)snprintf(mybuf, sizeof(mybuf), "%d",
           (unsigned int)sat->sat_addr.s_node);
       return mybuf;
}

static const char *
at_pr_port(const struct sockaddr_at *sat)
{
       static char mybuf[50];

       switch (sat->sat_port) {
       case ATADDR_ANYPORT:
               return "*";
       case 0xff:
               return "????";
       default:
               (void)snprintf(mybuf, sizeof(mybuf), "%d",
                   (unsigned int)sat->sat_port);
               return mybuf;
       }
}

static const char *
at_pr_range(const struct sockaddr_at *sat)
{
       static char mybuf[50];

       if (sat->sat_range.r_netrange.nr_firstnet
           != sat->sat_range.r_netrange.nr_lastnet) {
               (void)snprintf(mybuf, sizeof(mybuf), "%d-%d",
                       ntohs(sat->sat_range.r_netrange.nr_firstnet),
                       ntohs(sat->sat_range.r_netrange.nr_lastnet));
       } else {
               (void)snprintf(mybuf, sizeof(mybuf), "%d",
                       ntohs(sat->sat_range.r_netrange.nr_firstnet));
       }
       return mybuf;
}


/* what == 0 for addr only == 3
*      1 for net
*      2 for host
*      4 for port
*      8 for numeric only
*/
const char *
atalk_print(const struct sockaddr *sa, int what)
{
       const struct sockaddr_at *sat = (const struct sockaddr_at *) sa;
       static char mybuf[50];
       int numeric = (what & 0x08);

       mybuf[0] = 0;
       switch (what & 0x13) {
       case 0:
               mybuf[0] = 0;
               break;
       case 1:
               (void)snprintf(mybuf, sizeof(mybuf), "%s",
                   at_pr_net(sat, numeric));
               break;
       case 2:
               (void)snprintf(mybuf, sizeof(mybuf), "%s",
                   at_pr_host(sat, numeric));
               break;
       case 3:
               (void)snprintf(mybuf, sizeof(mybuf), "%s.%s",
                       at_pr_net(sat, numeric),
                       at_pr_host(sat, numeric));
               break;
       case 0x10:
               (void)snprintf(mybuf, sizeof(mybuf), "%s", at_pr_range(sat));
       }
       if (what & 4) {
               (void)snprintf(mybuf + strlen(mybuf),
                   sizeof(mybuf) - strlen(mybuf), ".%s", at_pr_port(sat));
       }
       return mybuf;
}

const char *
atalk_print2(const struct sockaddr *sa, const struct sockaddr *mask, int what)
{
       int             n, l;
       static char     buf[100];
       const struct sockaddr_at *sat1, *sat2;
       struct sockaddr_at thesockaddr;
       struct sockaddr *sa2;

       sat1 = (const struct sockaddr_at *) sa;
       sat2 = (const struct sockaddr_at *) mask;
       sa2 = (struct sockaddr *) & thesockaddr;

       thesockaddr.sat_addr.s_net = sat1->sat_addr.s_net &
           sat2->sat_addr.s_net;
       n = snprintf(buf, sizeof(buf), "%s", atalk_print(sa2, 1 | (what & 8)));
       if (n >= (int)sizeof(buf))
               n = sizeof(buf) - 1;
       else if (n == -1)
               n = 0;  /* What else can be done ? */
       if (sat2->sat_addr.s_net != 0xFFFF) {
               thesockaddr.sat_addr.s_net = sat1->sat_addr.s_net |
                   ~sat2->sat_addr.s_net;
               l = snprintf(buf + n, sizeof(buf) - n,
                   "-%s", atalk_print(sa2, 1 | (what & 8)));
               if (l >= (int)(sizeof(buf) - n))
                       l = sizeof(buf) - n - 1;
               if (l > 0)
                       n += l;
       }
       if (what & 2) {
               l = snprintf(buf + n, sizeof(buf) - n, ".%s",
                   atalk_print(sa, what & (~1)));
               if (l >= (int)(sizeof(buf) - n))
                       l = sizeof(buf) - n - 1;
               if (l > 0)
                       n += l;
       }
       return buf;
}

void
atalkprotopr(u_long off, const char *name)
{
       struct ddpcb ddpcb;
       struct socket sockb;
       struct ddpcb *next;
       struct ddpcb *initial;
       int width = 22;
       if (off == 0)
               return;
       if (kread(off, (char *)&initial, sizeof(struct ddpcb *)) < 0)
               return;
       for (next = initial; next != NULL;) {
               u_long  ppcb = (u_long)next;

               if (kread((u_long)next, (char *)&ddpcb, sizeof(ddpcb)) < 0)
                       return;
               next = ddpcb.ddp_next;
#if 0
               if (!aflag && atalk_nullhost(ddpcb.ddp_lsat))
                       continue;
#endif
               if (kread((u_long)ddpcb.ddp_socket,
                         (char *)&sockb, sizeof(sockb)) < 0)
                       return;
               if (first) {
                       printf("Active ATALK connections");
                       if (aflag)
                               printf(" (including servers)");
                       putchar('\n');
                       if (Aflag) {
                               width = 18;
                               printf("%-8.8s ", "PCB");
                       }
                       printf("%-5.5s %-6.6s %-6.6s  %*.*s %*.*s %s\n",
                              "Proto", "Recv-Q", "Send-Q",
                              -width, width, "Local Address",
                              -width, width, "Foreign Address", "(state)");
                       first = 0;
               }
               if (Aflag)
                       printf("%8lx ", ppcb);
               printf("%-5.5s %6ld %6ld ", name, sockb.so_rcv.sb_cc,
                      sockb.so_snd.sb_cc);
               printf(" %*.*s", -width, width,
                      atalk_print((struct sockaddr *)&ddpcb.ddp_lsat, 7));
               printf(" %*.*s", -width, width,
                      atalk_print((struct sockaddr *)&ddpcb.ddp_fsat, 7));
               putchar('\n');
       }
}
#define ANY(x,y,z)                                                            \
       ((sflag==1 || (x)) ?                                                  \
           printf("\t%llu %s%s%s\n",(unsigned long long)x,y,plural(x),z) : 0)

/*
* Dump DDP statistics structure.
*/
void
ddp_stats(u_long off, const char *name)
{
       uint64_t ddpstat[DDP_NSTATS];

       if (use_sysctl) {
               size_t size = sizeof(ddpstat);

               if (prog_sysctlbyname("net.atalk.ddp.stats", ddpstat, &size,
                                NULL, 0) == -1 && errno != ENOMEM)
                       return;
       } else {
               warnx("%s stats not available via KVM.", name);
               return;
       }

       printf("%s:\n", name);

       ANY(ddpstat[DDP_STAT_SHORT], "packet", " with short headers ");
       ANY(ddpstat[DDP_STAT_LONG], "packet", " with long headers ");
       ANY(ddpstat[DDP_STAT_NOSUM], "packet", " with no checksum ");
       ANY(ddpstat[DDP_STAT_TOOSHORT], "packet", " too short ");
       ANY(ddpstat[DDP_STAT_BADSUM], "packet", " with bad checksum ");
       ANY(ddpstat[DDP_STAT_TOOSMALL], "packet", " with not enough data ");
       ANY(ddpstat[DDP_STAT_FORWARD], "packet", " forwarded ");
       ANY(ddpstat[DDP_STAT_ENCAP], "packet", " encapsulated ");
       ANY(ddpstat[DDP_STAT_CANTFORWARD], "packet",
           " rcvd for unreachable dest ");
       ANY(ddpstat[DDP_STAT_NOSOCKSPACE], "packet",
           " dropped due to no socket space ");
}
#undef ANY