/*      $NetBSD: nfs_fha_new.c,v 1.2 2016/12/13 21:50:32 pgoyette Exp $ */
/*-
* Copyright (c) 2008 Isilon Inc http://www.isilon.com/
* Copyright (c) 2013 Spectra Logic Corporation
*
* 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>
/* __FBSDID("FreeBSD: head/sys/fs/nfsserver/nfs_fha_new.c 259765 2013-12-23 08:43:16Z mav "); */
__RCSID("$NetBSD: nfs_fha_new.c,v 1.2 2016/12/13 21:50:32 pgoyette Exp $");

#include <fs/nfs/common/nfsport.h>

#include <rpc/rpc.h>
#include <fs/nfs/common/nfs_fha.h>
#include <fs/nfs/common/xdr_subs.h>
#include <fs/nfs/common/nfs.h>
#include <fs/nfs/common/nfsproto.h>
#include <fs/nfs/common/nfsm_subs.h>
#include <fs/nfs/server/nfs_fha_new.h>

static void fhanew_init(void *foo);
static void fhanew_uninit(void *foo);
rpcproc_t fhanew_get_procnum(rpcproc_t procnum);
int fhanew_realign(struct mbuf **mb, int malloc_flags);
int fhanew_get_fh(uint64_t *fh, int v3, struct mbuf **md, caddr_t *dpos);
int fhanew_is_read(rpcproc_t procnum);
int fhanew_is_write(rpcproc_t procnum);
int fhanew_get_offset(struct mbuf **md, caddr_t *dpos, int v3,
                     struct fha_info *info);
int fhanew_no_offset(rpcproc_t procnum);
void fhanew_set_locktype(rpcproc_t procnum, struct fha_info *info);
static int fhenew_stats_sysctl(SYSCTL_HANDLER_ARGS);

static struct fha_params fhanew_softc;

SYSCTL_DECL(_vfs_nfsd);

extern int newnfs_nfsv3_procid[];
extern SVCPOOL  *nfsrvd_pool;

SYSINIT(nfs_fhanew, SI_SUB_ROOT_CONF, SI_ORDER_ANY, fhanew_init, NULL);
SYSUNINIT(nfs_fhanew, SI_SUB_ROOT_CONF, SI_ORDER_ANY, fhanew_uninit, NULL);

static void
fhanew_init(void *foo)
{
       struct fha_params *softc;

       softc = &fhanew_softc;

       bzero(softc, sizeof(*softc));

       /*
        * Setup the callbacks for this FHA personality.
        */
       softc->callbacks.get_procnum = fhanew_get_procnum;
       softc->callbacks.realign = fhanew_realign;
       softc->callbacks.get_fh = fhanew_get_fh;
       softc->callbacks.is_read = fhanew_is_read;
       softc->callbacks.is_write = fhanew_is_write;
       softc->callbacks.get_offset = fhanew_get_offset;
       softc->callbacks.no_offset = fhanew_no_offset;
       softc->callbacks.set_locktype = fhanew_set_locktype;
       softc->callbacks.fhe_stats_sysctl = fhenew_stats_sysctl;

       snprintf(softc->server_name, sizeof(softc->server_name),
           FHANEW_SERVER_NAME);

       softc->pool = &nfsrvd_pool;

       /*
        * Initialize the sysctl context list for the fha module.
        */
       sysctl_ctx_init(&softc->sysctl_ctx);
       softc->sysctl_tree = SYSCTL_ADD_NODE(&softc->sysctl_ctx,
           SYSCTL_STATIC_CHILDREN(_vfs_nfsd), OID_AUTO, "fha", CTLFLAG_RD,
           0, "fha node");
       if (softc->sysctl_tree == NULL) {
               printf("%s: unable to allocate sysctl tree\n", __func__);
               return;
       }

       fha_init(softc);
}

static void
fhanew_uninit(void *foo)
{
       struct fha_params *softc;

       softc = &fhanew_softc;

       fha_uninit(softc);
}

rpcproc_t
fhanew_get_procnum(rpcproc_t procnum)
{
       if (procnum > NFSV2PROC_STATFS)
               return (-1);

       return (newnfs_nfsv3_procid[procnum]);
}

int
fhanew_realign(struct mbuf **mb, int malloc_flags)
{
       return (newnfs_realign(mb, malloc_flags));
}

int
fhanew_get_fh(uint64_t *fh, int v3, struct mbuf **md, caddr_t *dpos)
{
       struct nfsrv_descript lnd, *nd;
       uint32_t *tl;
       uint8_t *buf;
       uint64_t t;
       int error, len, i;

       error = 0;
       len = 0;
       nd = &lnd;

       nd->nd_md = *md;
       nd->nd_dpos = *dpos;

       if (v3) {
               NFSM_DISSECT_NONBLOCK(tl, uint32_t *, NFSX_UNSIGNED);
               if ((len = fxdr_unsigned(int, *tl)) <= 0 || len > NFSX_FHMAX) {
                       error = EBADRPC;
                       goto nfsmout;
               }
       } else {
               len = NFSX_V2FH;
       }

       t = 0;
       if (len != 0) {
               NFSM_DISSECT_NONBLOCK(buf, uint8_t *, len);
               for (i = 0; i < len; i++)
                       t ^= ((uint64_t)buf[i] << (i & 7) * 8);
       }
       *fh = t;

nfsmout:
       *md = nd->nd_md;
       *dpos = nd->nd_dpos;

       return (error);
}

int
fhanew_is_read(rpcproc_t procnum)
{
       if (procnum == NFSPROC_READ)
               return (1);
       else
               return (0);
}

int
fhanew_is_write(rpcproc_t procnum)
{
       if (procnum == NFSPROC_WRITE)
               return (1);
       else
               return (0);
}

int
fhanew_get_offset(struct mbuf **md, caddr_t *dpos, int v3,
                 struct fha_info *info)
{
       struct nfsrv_descript lnd, *nd;
       uint32_t *tl;
       int error;

       error = 0;

       nd = &lnd;
       nd->nd_md = *md;
       nd->nd_dpos = *dpos;

       if (v3) {
               NFSM_DISSECT_NONBLOCK(tl, uint32_t *, 2 * NFSX_UNSIGNED);
               info->offset = fxdr_hyper(tl);
       } else {
               NFSM_DISSECT_NONBLOCK(tl, uint32_t *, NFSX_UNSIGNED);
               info->offset = fxdr_unsigned(uint32_t, *tl);
       }

nfsmout:
       *md = nd->nd_md;
       *dpos = nd->nd_dpos;

       return (error);
}

int
fhanew_no_offset(rpcproc_t procnum)
{
       if (procnum == NFSPROC_FSSTAT ||
           procnum == NFSPROC_FSINFO ||
           procnum == NFSPROC_PATHCONF ||
           procnum == NFSPROC_NOOP ||
           procnum == NFSPROC_NULL)
               return (1);
       else
               return (0);
}

void
fhanew_set_locktype(rpcproc_t procnum, struct fha_info *info)
{
       switch (procnum) {
       case NFSPROC_NULL:
       case NFSPROC_GETATTR:
       case NFSPROC_LOOKUP:
       case NFSPROC_ACCESS:
       case NFSPROC_READLINK:
       case NFSPROC_READ:
       case NFSPROC_READDIR:
       case NFSPROC_READDIRPLUS:
       case NFSPROC_WRITE:
               info->locktype = LK_SHARED;
               break;
       case NFSPROC_SETATTR:
       case NFSPROC_CREATE:
       case NFSPROC_MKDIR:
       case NFSPROC_SYMLINK:
       case NFSPROC_MKNOD:
       case NFSPROC_REMOVE:
       case NFSPROC_RMDIR:
       case NFSPROC_RENAME:
       case NFSPROC_LINK:
       case NFSPROC_FSSTAT:
       case NFSPROC_FSINFO:
       case NFSPROC_PATHCONF:
       case NFSPROC_COMMIT:
       case NFSPROC_NOOP:
               info->locktype = LK_EXCLUSIVE;
               break;
       }
}

static int
fhenew_stats_sysctl(SYSCTL_HANDLER_ARGS)
{
       return (fhe_stats_sysctl(oidp, arg1, arg2, req, &fhanew_softc));
}


SVCTHREAD *
fhanew_assign(SVCTHREAD *this_thread, struct svc_req *req)
{
       return (fha_assign(this_thread, req, &fhanew_softc));
}