/*      $NetBSD: route.c,v 1.88 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: @(#)route.c       8.3 (Berkeley) 3/9/94";
#else
__RCSID("$NetBSD: route.c,v 1.88 2022/09/02 06:25:43 msaitoh Exp $");
#endif
#endif /* not lint */

#include <stdbool.h>
#include <sys/param.h>
#include <sys/protosw.h>
#include <sys/socket.h>
#include <sys/mbuf.h>
#include <sys/un.h>

#include <net/if.h>
#include <net/if_dl.h>
#include <net/if_types.h>
#include <net/route.h>
#include <netinet/in.h>
#include <netatalk/at.h>
#include <netmpls/mpls.h>

#include <sys/sysctl.h>

#include <arpa/inet.h>

#include <err.h>
#include <kvm.h>
#include <netdb.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "netstat.h"
#include "rtutil.h"

#define kget(p, d) (kread((u_long)(p), (char *)&(d), sizeof (d)))

/*
 * XXX we put all of the sockaddr types in here to force the alignment
 * to be correct.
 */
static union sockaddr_union {
        struct  sockaddr u_sa;
        struct  sockaddr_in u_in;
        struct  sockaddr_un u_un;
        struct  sockaddr_at u_at;
        struct  sockaddr_dl u_dl;
        u_short u_data[128];
        int u_dummy;            /* force word-alignment */
} pt_u;

int     do_rtent = 0;
struct  rtentry rtentry;
struct  radix_node rnode;
struct  radix_mask rmask;

static struct sockaddr *kgetsa(const struct sockaddr *);
static void p_tree(struct radix_node *);
static void p_rtnode(void);
static void p_krtentry(struct rtentry *);

/*
 * Print routing tables.
 */
void
routepr(u_long rtree)
{
        struct radix_node_head *rnh, head;
        struct radix_node_head *rt_nodes[AF_MAX + 1];
        int i;

        printf("Routing tables\n");

        if (rtree == 0) {
                printf("rt_tables: symbol not in namelist\n");
                return;
        }

        kget(rtree, rt_nodes);
        for (i = 0; i <= AF_MAX; i++) {
                if ((rnh = rt_nodes[i]) == 0)
                        continue;
                kget(rnh, head);
                if (i == AF_UNSPEC) {
                        if (Aflag && (af == 0 || af == 0xff)) {
                                printf("Netmasks:\n");
                                p_tree(head.rnh_treetop);
                        }
                } else if (af == AF_UNSPEC || af == i) {
                        p_family(i);
                        do_rtent = 1;
                        p_rthdr(i, Aflag);
                        p_tree(head.rnh_treetop);
                }
        }
}

static struct sockaddr *
kgetsa(const struct sockaddr *dst)
{

        kget(dst, pt_u.u_sa);
        if (pt_u.u_sa.sa_len > sizeof (pt_u.u_sa))
                kread((u_long)dst, (char *)pt_u.u_data, pt_u.u_sa.sa_len);
        return (&pt_u.u_sa);
}

static void
p_tree(struct radix_node *rn)
{

again:
        kget(rn, rnode);
        if (rnode.rn_b < 0) {
                if (Aflag)
                        printf("%-8.8lx ", (u_long) rn);
                if (rnode.rn_flags & RNF_ROOT) {
                        if (Aflag)
                                printf("(root node)%s",
                                    rnode.rn_dupedkey ? " =>\n" : "\n");
                } else if (do_rtent) {
                        kget(rn, rtentry);
                        p_krtentry(&rtentry);
                        if (Aflag)
                                p_rtnode();
                } else {
                        p_sockaddr(
                                kgetsa((const struct sockaddr *)rnode.rn_key),
                                NULL, 0, 44, nflag);
                        putchar('\n');
                }
                if ((rn = rnode.rn_dupedkey) != NULL)
                        goto again;
        } else {
                if (Aflag && do_rtent) {
                        printf("%-8.8lx ", (u_long) rn);
                        p_rtnode();
                }
                rn = rnode.rn_r;
                p_tree(rnode.rn_l);
                p_tree(rn);
        }
}

static void
p_rtnode(void)
{
        struct radix_mask *rm = rnode.rn_mklist;
        char    nbuf[20];

        if (rnode.rn_b < 0) {
                if (rnode.rn_mask) {
                        printf("\t  mask ");
                        p_sockaddr(
                                kgetsa((const struct sockaddr *)rnode.rn_mask),
                                NULL, 0, -1, nflag);
                } else if (rm == 0)
                        return;
        } else {
                (void)snprintf(nbuf, sizeof nbuf, "(%d)", rnode.rn_b);
                printf("%6.6s %8.8lx : %8.8lx", nbuf, (u_long) rnode.rn_l,
                    (u_long) rnode.rn_r);
        }
        while (rm) {
                kget(rm, rmask);
                (void)snprintf(nbuf, sizeof nbuf, " %d refs, ", rmask.rm_refs);
                printf(" mk = %8.8lx {(%d),%s", (u_long) rm,
                    -1 - rmask.rm_b, rmask.rm_refs ? nbuf : " ");
                if (rmask.rm_flags & RNF_NORMAL) {
                        struct radix_node rnode_aux;
                        printf(" <normal>, ");
                        kget(rmask.rm_leaf, rnode_aux);
                        p_sockaddr(kgetsa((const struct sockaddr *)rnode_aux.rn_mask),
                            NULL, 0, -1, nflag);
                } else
                        p_sockaddr(
                                kgetsa((const struct sockaddr *)rmask.rm_mask),
                                NULL, 0, -1, nflag);
                putchar('}');
                if ((rm = rmask.rm_mklist) != NULL)
                        printf(" ->");
        }
        putchar('\n');
}

static struct sockaddr *sockcopy(struct sockaddr *, union sockaddr_union *);

/*
 * copy a sockaddr into an allocated region, allocate at least sockaddr
 * bytes and zero unused
 */
static struct sockaddr *
sockcopy(struct sockaddr *sp, union sockaddr_union *dp)
{
        int len;

        if (sp == 0 || sp->sa_len == 0)
                (void)memset(dp, 0, sizeof (*sp));
        else {
                len = (sp->sa_len >= sizeof (*sp)) ? sp->sa_len : sizeof (*sp);
                (void)memcpy(dp, sp, len);
        }
        return ((struct sockaddr *)dp);
}

static void
p_krtentry(struct rtentry *rt)
{
        static struct ifnet ifnet, *lastif;
        union sockaddr_union addr_un, mask_un;
        struct sockaddr *addr, *mask;

        memset(&addr_un, 0, sizeof(addr_un));
        memset(&mask_un, 0, sizeof(mask_un));
        addr = sockcopy(kgetsa(rt_getkey(rt)), &addr_un);
        if (rt_mask(rt))
                mask = sockcopy(kgetsa(rt_mask(rt)), &mask_un);
        else
                mask = sockcopy(NULL, &mask_un);
        p_addr(addr, mask, rt->rt_flags, nflag);
        p_gwaddr(kgetsa(rt->rt_gateway), kgetsa(rt->rt_gateway)->sa_family,
            nflag);
        p_flags(rt->rt_flags);
        printf("%6d %8"PRIu64" ", rt->rt_refcnt, rt->rt_use);
        if (rt->rt_rmx.rmx_mtu)
                printf("%6"PRIu64, rt->rt_rmx.rmx_mtu);
        else
                printf("%6s", "-");
        putchar((rt->rt_rmx.rmx_locks & RTV_MTU) ? 'L' : ' ');
        if (tagflag == 1) {
#ifndef SMALL
                if (rt->rt_tag != NULL) {
                        const struct sockaddr *tagsa = kgetsa(rt->rt_tag);
                        char *tagstr;

                        if (tagsa->sa_family == AF_MPLS) {
                                tagstr = mpls_ntoa(tagsa);
                                if (strlen(tagstr) < 7)
                                        printf("%7s", tagstr);
                                else
                                        printf("%s", tagstr);
                        } else
                                printf("%7s", "-");
                } else
#endif
                        printf("%7s", "-");
        }
        if (rt->rt_ifp) {
                if (rt->rt_ifp != lastif) {
                        kget(rt->rt_ifp, ifnet);
                        lastif = rt->rt_ifp;
                }
                printf(" %.16s%s", ifnet.if_xname,
                        rt->rt_nodes[0].rn_dupedkey ? " =>" : "");
        }
        putchar('\n');
#ifndef SMALL
        if (vflag)
                p_rtrmx(&rt->rt_rmx);
#endif
}

/*
 * Print routing statistics
 */
void
rt_stats(u_long off)
{
        struct rtstat rtstats;

        if (use_sysctl) {
                size_t rtsize = sizeof(rtstats);

                if (sysctlbyname("net.rtable.stats", &rtstats, &rtsize,
                    NULL, 0) == -1)
                        err(1, "rt_stats: sysctl");
        } else if (off == 0) {
                printf("rtstat: symbol not in namelist\n");
                return;
        } else
                kread(off, (char *)&rtstats, sizeof(rtstats));

        printf("routing:\n");
        printf("\t%llu bad routing redirect%s\n",
            (unsigned long long)rtstats.rts_badredirect,
            plural(rtstats.rts_badredirect));
        printf("\t%llu dynamically created route%s\n",
            (unsigned long long)rtstats.rts_dynamic,
            plural(rtstats.rts_dynamic));
        printf("\t%llu new gateway%s due to redirects\n",
            (unsigned long long)rtstats.rts_newgateway,
            plural(rtstats.rts_newgateway));
        printf("\t%llu destination%s found unreachable\n",
            (unsigned long long)rtstats.rts_unreach,
            plural(rtstats.rts_unreach));
        printf("\t%llu use%s of a wildcard route\n",
            (unsigned long long)rtstats.rts_wildcard,
            plural(rtstats.rts_wildcard));
}