/*      $NetBSD: nfs_bootstatic.c,v 1.8 2009/10/23 02:32:34 snj Exp $   */

/*
 *
 * Copyright (c) 2004 Christian Limpach.
 * 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 ``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 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>
__KERNEL_RCSID(0, "$NetBSD: nfs_bootstatic.c,v 1.8 2009/10/23 02:32:34 snj Exp $");

#ifdef _KERNEL_OPT
#include "opt_nfs_boot.h"
#endif

#include <sys/param.h>
#include <sys/systm.h>
#include <sys/kernel.h>
#include <sys/ioctl.h>
#include <sys/proc.h>
#include <sys/mount.h>
#include <sys/mbuf.h>
#include <sys/socket.h>
#include <sys/socketvar.h>

#include <net/if.h>
#include <net/if_types.h>
#include <net/if_media.h>
#include <net/route.h>
#include <net/if_ether.h>

#include <netinet/in.h>
#include <netinet/if_inarp.h>

#include <nfs/rpcv2.h>

#include <nfs/nfsproto.h>
#include <nfs/nfs.h>
#include <nfs/nfsmount.h>
#include <nfs/nfsdiskless.h>

int (*nfs_bootstatic_callback)(struct nfs_diskless *) = NULL;

int
nfs_bootstatic(struct nfs_diskless *nd, struct lwp *lwp, int *_flags)
{
        struct ifnet *ifp = nd->nd_ifp;
        struct sockaddr_in *sin;
        int error, flags;

        if (nfs_bootstatic_callback)
                flags = (*nfs_bootstatic_callback)(nd);
        else
                flags = 0;

        if (flags & NFS_BOOT_NOSTATIC)
                return EOPNOTSUPP;

        if (flags == 0) {
#ifdef NFS_BOOTSTATIC_MYIP
                if (!(*_flags & NFS_BOOT_HAS_MYIP)) {
                        nd->nd_myip.s_addr = inet_addr(NFS_BOOTSTATIC_MYIP);
                        flags |= NFS_BOOT_HAS_MYIP;
                }
#endif
#ifdef NFS_BOOTSTATIC_GWIP
                if (!(*_flags & NFS_BOOT_HAS_GWIP)) {
                        nd->nd_gwip.s_addr = inet_addr(NFS_BOOTSTATIC_GWIP);
                        flags |= NFS_BOOT_HAS_GWIP;
                }
#endif
#ifdef NFS_BOOTSTATIC_MASK
                if (!(*_flags & NFS_BOOT_HAS_MASK)) {
                        nd->nd_mask.s_addr = inet_addr(NFS_BOOTSTATIC_MASK);
                        flags |= NFS_BOOT_HAS_MASK;
                }
#endif
#ifdef NFS_BOOTSTATIC_SERVADDR
                if (!(*_flags & NFS_BOOT_HAS_SERVADDR)) {
                        sin = (struct sockaddr_in *) &nd->nd_root.ndm_saddr;
                        memset((void *)sin, 0, sizeof(*sin));
                        sin->sin_len = sizeof(*sin);
                        sin->sin_family = AF_INET;
                        sin->sin_addr.s_addr = inet_addr(NFS_BOOTSTATIC_SERVADDR);
                        flags |= NFS_BOOT_HAS_SERVADDR;
                }
#endif
#ifdef NFS_BOOTSTATIC_SERVER
                if (!(*_flags & NFS_BOOT_HAS_SERVER)) {
                        strncpy(nd->nd_root.ndm_host, NFS_BOOTSTATIC_SERVER,
                                MNAMELEN);
                        flags |= NFS_BOOT_HAS_SERVER;
                        if (strchr(nd->nd_root.ndm_host, ':') != NULL)
                                flags |= NFS_BOOT_HAS_ROOTPATH;
                }
#endif
        }

        *_flags |= flags;

        if (!(*_flags & NFS_BOOT_HAS_SERVADDR) && (*_flags & NFS_BOOT_HAS_SERVER)) {
                char rootserver[MNAMELEN];
                char *sep;

                sep = strchr(nd->nd_root.ndm_host, ':');
                strlcpy(rootserver, nd->nd_root.ndm_host,
                        sep - nd->nd_root.ndm_host+1);

                sin = (struct sockaddr_in *) &nd->nd_root.ndm_saddr;
                memset((void *)sin, 0, sizeof(*sin));
                sin->sin_len = sizeof(*sin);
                sin->sin_family = AF_INET;
                sin->sin_addr.s_addr = inet_addr(rootserver);
                flags |= NFS_BOOT_HAS_SERVADDR;
                *_flags |= NFS_BOOT_HAS_SERVADDR;
        }

        if (flags & NFS_BOOT_HAS_MYIP)
                aprint_normal("nfs_boot: client_addr=%s\n",
                    inet_ntoa(nd->nd_myip));

        if (flags & NFS_BOOT_HAS_GWIP)
                aprint_normal("nfs_boot: gateway=%s\n",
                    inet_ntoa(nd->nd_gwip));

        if (flags & NFS_BOOT_HAS_MASK)
                aprint_normal("nfs_boot: netmask=%s\n",
                    inet_ntoa(nd->nd_mask));

        if (flags & NFS_BOOT_HAS_SERVADDR) {
                sin = (struct sockaddr_in *) &nd->nd_root.ndm_saddr;
                aprint_normal("nfs_boot: server=%s\n",
                    inet_ntoa(sin->sin_addr));
        }

        if (flags & NFS_BOOT_HAS_SERVER)
                aprint_normal("nfs_boot: root=%.*s\n", MNAMELEN,
                    nd->nd_root.ndm_host);

        /*
         * Do enough of ifconfig(8) so that the chosen interface
         * can talk to the servers.
         */
        error = nfs_boot_setaddress(ifp, lwp,
            *_flags & NFS_BOOT_HAS_MYIP ? nd->nd_myip.s_addr : INADDR_ANY,
            *_flags & NFS_BOOT_HAS_MASK ? nd->nd_mask.s_addr : INADDR_ANY,
            INADDR_ANY);
        if (error) {
                aprint_error("nfs_boot: set ifaddr, error=%d\n", error);
                goto out;
        }

        if ((*_flags & NFS_BOOT_ALLINFO) != NFS_BOOT_ALLINFO)
                return EADDRNOTAVAIL;

        error = 0;

 out:

        return error;
}