/*      $NetBSD: dev_net.c,v 1.23 2008/04/28 20:24:06 martin Exp $      */

/*-
 * Copyright (c) 1997 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Gordon W. Ross.
 *
 * 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 NETBSD FOUNDATION, INC. 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 FOUNDATION 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.
 */

/*-
 * This module implements a "raw device" interface suitable for
 * use by the stand-alone I/O library NFS code.  This interface
 * does not support any "block" access, and exists only for the
 * purpose of initializing the network interface, getting boot
 * parameters, and performing the NFS mount.
 *
 * At open time, this does:
 *
 * find interface      - netif_open()
 * RARP for IP address - rarp_getipaddress()
 * RPC/bootparams      - callrpc(d, RPC_BOOTPARAMS, ...)
 * RPC/mountd          - nfs_mount(sock, ip, path)
 *
 * the root file handle from mountd is saved in a global
 * for use by the NFS open code (NFS/lookup).
 */

#include <sys/param.h>
#include <sys/socket.h>
#include <sys/stdarg.h>

#include <net/if.h>
#include <netinet/in.h>
#include <netinet/in_systm.h>

#include <stand.h>
#include <stddef.h>
#include <string.h>
#include <net.h>
#include <netif.h>
#include <bootp.h>
#include <bootparam.h>

#include "dev_net.h"
#include "bootstrap.h"

#ifndef NETPROTO_DEFAULT
# define NETPROTO_DEFAULT NET_NFS
#endif

static char *netdev_name;
static int netdev_sock = -1;
static int netdev_opens;

static int      net_init(void);
static int      net_open(struct open_file *, ...);
static int      net_close(struct open_file *);
static void     net_cleanup(void);
static int      net_strategy(void *, int, daddr_t, size_t, char *, size_t *);
static int      net_print(int);

static int net_getparams(int sock);

struct devsw netdev = {
        .dv_name = "net",
        .dv_type = DEVT_NET,
        .dv_init = net_init,
        .dv_strategy = net_strategy,
        .dv_open = net_open,
        .dv_close = net_close,
        .dv_ioctl = noioctl,
        .dv_print = net_print,
        .dv_cleanup = net_cleanup,
};

static struct uri_scheme {
        const char *scheme;
        int proto;
} uri_schemes[] = {
        { "tftp:/", NET_TFTP },
        { "nfs:/", NET_NFS },
};

static int
net_init(void)
{

        return (0);
}

/*
 * Called by devopen after it sets f->f_dev to our devsw entry.
 * This opens the low-level device and sets dev->d_opendata.
 * This is declared with variable arguments...
 */
static int
net_open(struct open_file *f, ...)
{
        struct iodesc *d;
        va_list args;
        struct devdesc *dev;
        const char *devname;    /* Device part of file name (or NULL). */
        int error = 0;

        va_start(args, f);
        dev = va_arg(args, struct devdesc *);
        va_end(args);

        devname = dev->d_dev->dv_name;
        /* Before opening another interface, close the previous one first. */
        if (netdev_sock >= 0 && strcmp(devname, netdev_name) != 0)
                net_cleanup();

        /* On first open, do netif open, mount, etc. */
        if (netdev_opens == 0) {
                /* Find network interface. */
                if (netdev_sock < 0) {
                        netdev_sock = netif_open(dev);
                        if (netdev_sock < 0) {
                                printf("%s: netif_open() failed\n", __func__);
                                return (ENXIO);
                        }
                        netdev_name = strdup(devname);
                        DEBUG_PRINTF(1,("%s: netif_open() succeeded %#x\n",
                                __func__, rootip.s_addr));
                }
                /*
                 * If network params were not set by netif_open(), try to get
                 * them via bootp, rarp, etc.
                 */
                if (rootip.s_addr == 0) {
                        /* Get root IP address, and path, etc. */
                        error = net_getparams(netdev_sock);
                        if (error) {
                                /* getparams makes its own noise */
                                free(netdev_name);
                                netif_close(netdev_sock);
                                netdev_sock = -1;
                                return (error);
                        }
                }
                /*
                 * Set the variables required by the kernel's nfs_diskless
                 * mechanism.  This is the minimum set of variables required to
                 * mount a root filesystem without needing to obtain additional
                 * info from bootp or other sources.
                 */
                d = socktodesc(netdev_sock);
                setenv("boot.netif.hwaddr", ether_sprintf(d->myea), 1);
                setenv("boot.netif.ip", inet_ntoa(myip), 1);
                setenv("boot.netif.netmask", intoa(netmask), 1);
                setenv("boot.netif.gateway", inet_ntoa(gateip), 1);
                setenv("boot.netif.server", inet_ntoa(rootip), 1);
                if (netproto == NET_TFTP) {
                        setenv("boot.tftproot.server", inet_ntoa(rootip), 1);
                        setenv("boot.tftproot.path", rootpath, 1);
                } else if (netproto == NET_NFS) {
                        setenv("boot.nfsroot.server", inet_ntoa(rootip), 1);
                        setenv("boot.nfsroot.path", rootpath, 1);
                }
                if (intf_mtu != 0) {
                        char mtu[16];
                        snprintf(mtu, sizeof(mtu), "%u", intf_mtu);
                        setenv("boot.netif.mtu", mtu, 1);
                }

                DEBUG_PRINTF(1,("%s: netproto=%d\n", __func__, netproto));
        }
        netdev_opens++;
        dev->d_opendata = &netdev_sock;
        return (error);
}

static int
net_close(struct open_file *f)
{
        struct devdesc *dev;

        DEBUG_PRINTF(2,("%s: opens=%d\n", __func__, netdev_opens));

        dev = f->f_devdata;
        dev->d_opendata = NULL;

        return (0);
}

static void
net_cleanup(void)
{

        if (netdev_sock >= 0) {
                DEBUG_PRINTF(1,("%s: calling netif_close()\n", __func__));
                rootip.s_addr = 0;
                free(netdev_name);
                netif_close(netdev_sock);
                netdev_sock = -1;
        }
}

static int
net_strategy(void *devdata, int rw, daddr_t blk, size_t size, char *buf,
    size_t *rsize)
{

        return (EIO);
}

#define SUPPORT_BOOTP

/*
 * Get info for NFS boot: our IP address, our hostname,
 * server IP address, and our root path on the server.
 * There are two ways to do this:  The old, Sun way,
 * and the more modern, BOOTP way. (RFC951, RFC1048)
 *
 * The default is to use the Sun bootparams RPC
 * (because that is what the kernel will do).
 * MD code can make try_bootp initialied data,
 * which will override this common definition.
 */
#ifdef  SUPPORT_BOOTP
int try_bootp = 1;
#endif

extern n_long ip_convertaddr(char *p);

static int
net_getparams(int sock)
{
        char buf[MAXHOSTNAMELEN];
        n_long rootaddr, smask;

#ifdef  SUPPORT_BOOTP
        /*
         * Try to get boot info using BOOTP.  If we succeed, then
         * the server IP address, gateway, and root path will all
         * be initialized.  If any remain uninitialized, we will
         * use RARP and RPC/bootparam (the Sun way) to get them.
         */
        if (try_bootp)
                bootp(sock);
        if (myip.s_addr != 0)
                goto exit;
        DEBUG_PRINTF(1,("%s: BOOTP failed, trying RARP/RPC...\n", __func__));
#endif

        /*
         * Use RARP to get our IP address.  This also sets our
         * netmask to the "natural" default for our address.
         */
        if (rarp_getipaddress(sock)) {
                printf("%s: RARP failed\n", __func__);
                return (EIO);
        }
        printf("%s: client addr: %s\n", __func__, inet_ntoa(myip));

        /* Get our hostname, server IP address, gateway. */
        if (bp_whoami(sock)) {
                printf("%s: bootparam/whoami RPC failed\n", __func__);
                return (EIO);
        }
        DEBUG_PRINTF(1,("%s: client name: %s\n", __func__, hostname));

        /*
         * Ignore the gateway from whoami (unreliable).
         * Use the "gateway" parameter instead.
         */
        smask = 0;
        gateip.s_addr = 0;
        if (bp_getfile(sock, "gateway", &gateip, buf) == 0) {
                /* Got it!  Parse the netmask. */
                smask = ip_convertaddr(buf);
        }
        if (smask) {
                netmask = smask;
                DEBUG_PRINTF(1,("%s: subnet mask: %s\n", __func__,
                        intoa(netmask)));
        }
        if (gateip.s_addr)
                DEBUG_PRINTF(1,("%s: net gateway: %s\n", __func__,
                        inet_ntoa(gateip)));

        /* Get the root server and pathname. */
        if (bp_getfile(sock, "root", &rootip, rootpath)) {
                printf("%s: bootparam/getfile RPC failed\n", __func__);
                return (EIO);
        }
exit:
        if ((rootaddr = net_parse_rootpath()) != htonl(INADDR_NONE))
                rootip.s_addr = rootaddr;

        DEBUG_PRINTF(1,("%s: proto: %d\n", __func__, netproto));
        DEBUG_PRINTF(1,("%s: server addr: %s\n", __func__, inet_ntoa(rootip)));
        DEBUG_PRINTF(1,("%s: server port: %d\n", __func__, rootport));
        DEBUG_PRINTF(1,("%s: server path: %s\n", __func__, rootpath));

        return (0);
}

static int
net_print(int verbose)
{
        struct netif_driver *drv;
        int i, d, cnt;
        int ret = 0;

        if (netif_drivers[0] == NULL)
                return (ret);

        printf("%s devices:", netdev.dv_name);
        if ((ret = pager_output("\n")) != 0)
                return (ret);

        cnt = 0;
        for (d = 0; netif_drivers[d]; d++) {
                drv = netif_drivers[d];
                for (i = 0; i < drv->netif_nifs; i++) {
                        printf("\t%s%d:", netdev.dv_name, cnt++);
                        if (verbose) {
                                printf(" (%s%d)", drv->netif_bname,
                                    drv->netif_ifs[i].dif_unit);
                        }
                        if ((ret = pager_output("\n")) != 0)
                                return (ret);
                }
        }
        return (ret);
}

bool
is_tftp(void)
{
    return (netproto == NET_TFTP);
}

/*
 * Parses the rootpath if present
 *
 * The rootpath format can be in the form
 * <scheme>://ip[:port]/path
 * <scheme>:/path
 *
 * For compatibility with previous behaviour it also accepts as an NFS scheme
 * ip:/path
 * /path
 *
 * If an ip is set it returns it in network byte order.
 * The default scheme defined in the global netproto, if not set it defaults to
 * NFS.
 * It leaves just the pathname in the global rootpath.
 */
uint32_t
net_parse_rootpath(void)
{
        n_long addr = 0;
        size_t i;
        char ip[FNAME_SIZE];
        char *ptr, *portp, *val;

        netproto = NET_NONE;

        for (i = 0; i < nitems(uri_schemes); i++) {
                if (strncmp(rootpath, uri_schemes[i].scheme,
                    strlen(uri_schemes[i].scheme)) != 0)
                        continue;

                netproto = uri_schemes[i].proto;
                break;
        }
        ptr = rootpath;
        /* Fallback for compatibility mode */
        if (netproto == NET_NONE) {
                netproto = NETPROTO_DEFAULT;
                (void)strsep(&ptr, ":");
                if (ptr != NULL) {
                        addr = inet_addr(rootpath);
                        DEBUG_PRINTF(1,("rootpath=%s addr=%#x\n",
                                rootpath, addr));
                        bcopy(ptr, rootpath, strlen(ptr) + 1);
                }
        } else {
                ptr += strlen(uri_schemes[i].scheme);
                if (*ptr == '/') {
                        /* we are in the form <scheme>://, we do expect an ip */
                        ptr++;
                        portp = val = strchr(ptr, ':');
                        if (val != NULL) {
                                val++;
                                rootport = strtol(val, NULL, 10);
                        }
                        val = strchr(ptr, '/');
                        if (val != NULL) {
                                if (portp == NULL)
                                        portp = val;
                                snprintf(ip, sizeof(ip), "%.*s",
                                    (int)(portp - ptr),
                                    ptr);
                                addr = inet_addr(ip);
                                DEBUG_PRINTF(1,("ip=%s addr=%#x\n",
                                        ip, addr));
                                bcopy(val, rootpath, strlen(val) + 1);
                        }
                } else {
                        ptr--;
                        bcopy(ptr, rootpath, strlen(ptr) + 1);
                }
        }
        if (addr == 0)
                addr = htonl(INADDR_NONE);
        return (addr);
}