1d96b98a3SKenneth D. Merry /*- 2d63027b6SPedro F. Giffuni * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 3d63027b6SPedro F. Giffuni * 4d96b98a3SKenneth D. Merry * Copyright (c) 2008 Isilon Inc http://www.isilon.com/ 5d96b98a3SKenneth D. Merry * Copyright (c) 2013 Spectra Logic Corporation 6d96b98a3SKenneth D. Merry * 7d96b98a3SKenneth D. Merry * Redistribution and use in source and binary forms, with or without 8d96b98a3SKenneth D. Merry * modification, are permitted provided that the following conditions 9d96b98a3SKenneth D. Merry * are met: 10d96b98a3SKenneth D. Merry * 1. Redistributions of source code must retain the above copyright 11d96b98a3SKenneth D. Merry * notice, this list of conditions and the following disclaimer. 12d96b98a3SKenneth D. Merry * 2. Redistributions in binary form must reproduce the above copyright 13d96b98a3SKenneth D. Merry * notice, this list of conditions and the following disclaimer in the 14d96b98a3SKenneth D. Merry * documentation and/or other materials provided with the distribution. 15d96b98a3SKenneth D. Merry * 16d96b98a3SKenneth D. Merry * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 17d96b98a3SKenneth D. Merry * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 18d96b98a3SKenneth D. Merry * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 19d96b98a3SKenneth D. Merry * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 20d96b98a3SKenneth D. Merry * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21d96b98a3SKenneth D. Merry * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 22d96b98a3SKenneth D. Merry * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23d96b98a3SKenneth D. Merry * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 24d96b98a3SKenneth D. Merry * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 25d96b98a3SKenneth D. Merry * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26d96b98a3SKenneth D. Merry * SUCH DAMAGE. 27d96b98a3SKenneth D. Merry */ 28d96b98a3SKenneth D. Merry 29d96b98a3SKenneth D. Merry #include <sys/cdefs.h> 30d96b98a3SKenneth D. Merry __FBSDID("$FreeBSD$"); 31d96b98a3SKenneth D. Merry 329897e357SRick Macklem #include <sys/types.h> 339897e357SRick Macklem #include <sys/mbuf.h> 349897e357SRick Macklem #include <sys/sbuf.h> 359897e357SRick Macklem 36d96b98a3SKenneth D. Merry #include <fs/nfs/nfsport.h> 379897e357SRick Macklem #include <fs/nfsserver/nfs_fha_new.h> 38d96b98a3SKenneth D. Merry 39d96b98a3SKenneth D. Merry #include <rpc/rpc.h> 409897e357SRick Macklem 419897e357SRick Macklem static MALLOC_DEFINE(M_NFS_FHA, "NFS FHA", "NFS FHA"); 42d96b98a3SKenneth D. Merry 43d96b98a3SKenneth D. Merry static void fhanew_init(void *foo); 44d96b98a3SKenneth D. Merry static void fhanew_uninit(void *foo); 459897e357SRick Macklem static rpcproc_t fhanew_get_procnum(rpcproc_t procnum); 469897e357SRick Macklem static int fhanew_get_fh(uint64_t *fh, int v3, struct mbuf **md, 479897e357SRick Macklem caddr_t *dpos); 489897e357SRick Macklem static int fhanew_is_read(rpcproc_t procnum); 499897e357SRick Macklem static int fhanew_is_write(rpcproc_t procnum); 509897e357SRick Macklem static int fhanew_get_offset(struct mbuf **md, caddr_t *dpos, 519897e357SRick Macklem int v3, struct fha_info *info); 529897e357SRick Macklem static int fhanew_no_offset(rpcproc_t procnum); 539897e357SRick Macklem static void fhanew_set_locktype(rpcproc_t procnum, 54d96b98a3SKenneth D. Merry struct fha_info *info); 55d96b98a3SKenneth D. Merry static int fhenew_stats_sysctl(SYSCTL_HANDLER_ARGS); 569897e357SRick Macklem static void fha_extract_info(struct svc_req *req, 579897e357SRick Macklem struct fha_info *i); 58d96b98a3SKenneth D. Merry 59cd406ac9SRick Macklem NFSD_VNET_DEFINE_STATIC(struct fha_params *, fhanew_softc); 60d96b98a3SKenneth D. Merry 61d96b98a3SKenneth D. Merry SYSCTL_DECL(_vfs_nfsd); 62d96b98a3SKenneth D. Merry 63d96b98a3SKenneth D. Merry extern int newnfs_nfsv3_procid[]; 647e44856eSRick Macklem 65*ed03776cSRick Macklem VNET_SYSINIT(nfs_fhanew, SI_SUB_VNET_DONE, SI_ORDER_ANY, fhanew_init, NULL); 66*ed03776cSRick Macklem VNET_SYSUNINIT(nfs_fhanew, SI_SUB_VNET_DONE, SI_ORDER_ANY, fhanew_uninit, NULL); 67d96b98a3SKenneth D. Merry 68d96b98a3SKenneth D. Merry static void 69d96b98a3SKenneth D. Merry fhanew_init(void *foo) 70d96b98a3SKenneth D. Merry { 71d96b98a3SKenneth D. Merry struct fha_params *softc; 729897e357SRick Macklem int i; 73d96b98a3SKenneth D. Merry 74cd406ac9SRick Macklem NFSD_VNET(fhanew_softc) = malloc(sizeof(struct fha_params), M_TEMP, 75cd406ac9SRick Macklem M_WAITOK | M_ZERO); 76cd406ac9SRick Macklem softc = NFSD_VNET(fhanew_softc); 77d96b98a3SKenneth D. Merry 78d96b98a3SKenneth D. Merry snprintf(softc->server_name, sizeof(softc->server_name), 79d96b98a3SKenneth D. Merry FHANEW_SERVER_NAME); 80d96b98a3SKenneth D. Merry 81d96b98a3SKenneth D. Merry /* 82d96b98a3SKenneth D. Merry * Initialize the sysctl context list for the fha module. 83d96b98a3SKenneth D. Merry */ 84d96b98a3SKenneth D. Merry sysctl_ctx_init(&softc->sysctl_ctx); 85cd406ac9SRick Macklem if (IS_DEFAULT_VNET(curvnet)) { 86d96b98a3SKenneth D. Merry softc->sysctl_tree = SYSCTL_ADD_NODE(&softc->sysctl_ctx, 877029da5cSPawel Biernacki SYSCTL_STATIC_CHILDREN(_vfs_nfsd), OID_AUTO, "fha", 887029da5cSPawel Biernacki CTLFLAG_RD | CTLFLAG_MPSAFE, 0, "NFS File Handle Affinity (FHA)"); 89d96b98a3SKenneth D. Merry if (softc->sysctl_tree == NULL) { 90d96b98a3SKenneth D. Merry printf("%s: unable to allocate sysctl tree\n", __func__); 91d96b98a3SKenneth D. Merry return; 92d96b98a3SKenneth D. Merry } 93cd406ac9SRick Macklem } 94d96b98a3SKenneth D. Merry 959897e357SRick Macklem for (i = 0; i < FHA_HASH_SIZE; i++) 969897e357SRick Macklem mtx_init(&softc->fha_hash[i].mtx, "fhalock", NULL, MTX_DEF); 979897e357SRick Macklem 989897e357SRick Macklem /* 999897e357SRick Macklem * Set the default tuning parameters. 1009897e357SRick Macklem */ 1019897e357SRick Macklem softc->ctls.enable = FHA_DEF_ENABLE; 1029897e357SRick Macklem softc->ctls.read = FHA_DEF_READ; 1039897e357SRick Macklem softc->ctls.write = FHA_DEF_WRITE; 1049897e357SRick Macklem softc->ctls.bin_shift = FHA_DEF_BIN_SHIFT; 1059897e357SRick Macklem softc->ctls.max_nfsds_per_fh = FHA_DEF_MAX_NFSDS_PER_FH; 1069897e357SRick Macklem softc->ctls.max_reqs_per_nfsd = FHA_DEF_MAX_REQS_PER_NFSD; 1079897e357SRick Macklem 1089897e357SRick Macklem /* 1099897e357SRick Macklem * Add sysctls so the user can change the tuning parameters. 1109897e357SRick Macklem */ 111cd406ac9SRick Macklem if (IS_DEFAULT_VNET(curvnet)) { 1129897e357SRick Macklem SYSCTL_ADD_UINT(&softc->sysctl_ctx, SYSCTL_CHILDREN(softc->sysctl_tree), 1139897e357SRick Macklem OID_AUTO, "enable", CTLFLAG_RWTUN, 1149897e357SRick Macklem &softc->ctls.enable, 0, "Enable NFS File Handle Affinity (FHA)"); 1159897e357SRick Macklem 1169897e357SRick Macklem SYSCTL_ADD_UINT(&softc->sysctl_ctx, SYSCTL_CHILDREN(softc->sysctl_tree), 1179897e357SRick Macklem OID_AUTO, "read", CTLFLAG_RWTUN, 1189897e357SRick Macklem &softc->ctls.read, 0, "Enable NFS FHA read locality"); 1199897e357SRick Macklem 1209897e357SRick Macklem SYSCTL_ADD_UINT(&softc->sysctl_ctx, SYSCTL_CHILDREN(softc->sysctl_tree), 1219897e357SRick Macklem OID_AUTO, "write", CTLFLAG_RWTUN, 1229897e357SRick Macklem &softc->ctls.write, 0, "Enable NFS FHA write locality"); 1239897e357SRick Macklem 1249897e357SRick Macklem SYSCTL_ADD_UINT(&softc->sysctl_ctx, SYSCTL_CHILDREN(softc->sysctl_tree), 1259897e357SRick Macklem OID_AUTO, "bin_shift", CTLFLAG_RWTUN, 1269897e357SRick Macklem &softc->ctls.bin_shift, 0, 1279897e357SRick Macklem "Maximum locality distance 2^(bin_shift) bytes"); 1289897e357SRick Macklem 1299897e357SRick Macklem SYSCTL_ADD_UINT(&softc->sysctl_ctx, SYSCTL_CHILDREN(softc->sysctl_tree), 1309897e357SRick Macklem OID_AUTO, "max_nfsds_per_fh", CTLFLAG_RWTUN, 1319897e357SRick Macklem &softc->ctls.max_nfsds_per_fh, 0, "Maximum nfsd threads that " 1329897e357SRick Macklem "should be working on requests for the same file handle"); 1339897e357SRick Macklem 1349897e357SRick Macklem SYSCTL_ADD_UINT(&softc->sysctl_ctx, SYSCTL_CHILDREN(softc->sysctl_tree), 1359897e357SRick Macklem OID_AUTO, "max_reqs_per_nfsd", CTLFLAG_RWTUN, 1369897e357SRick Macklem &softc->ctls.max_reqs_per_nfsd, 0, "Maximum requests that " 1379897e357SRick Macklem "single nfsd thread should be working on at any time"); 1389897e357SRick Macklem 1399897e357SRick Macklem SYSCTL_ADD_OID(&softc->sysctl_ctx, SYSCTL_CHILDREN(softc->sysctl_tree), 1409897e357SRick Macklem OID_AUTO, "fhe_stats", CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_MPSAFE, 1419897e357SRick Macklem 0, 0, fhenew_stats_sysctl, "A", ""); 142d96b98a3SKenneth D. Merry } 143cd406ac9SRick Macklem } 144d96b98a3SKenneth D. Merry 145d96b98a3SKenneth D. Merry static void 146d96b98a3SKenneth D. Merry fhanew_uninit(void *foo) 147d96b98a3SKenneth D. Merry { 148d96b98a3SKenneth D. Merry struct fha_params *softc; 1499897e357SRick Macklem int i; 150d96b98a3SKenneth D. Merry 151cd406ac9SRick Macklem softc = NFSD_VNET(fhanew_softc); 152d96b98a3SKenneth D. Merry 1539897e357SRick Macklem sysctl_ctx_free(&softc->sysctl_ctx); 1549897e357SRick Macklem for (i = 0; i < FHA_HASH_SIZE; i++) 1559897e357SRick Macklem mtx_destroy(&softc->fha_hash[i].mtx); 156cd406ac9SRick Macklem free(softc, M_TEMP); 157d96b98a3SKenneth D. Merry } 158d96b98a3SKenneth D. Merry 1599897e357SRick Macklem static rpcproc_t 160d96b98a3SKenneth D. Merry fhanew_get_procnum(rpcproc_t procnum) 161d96b98a3SKenneth D. Merry { 162d96b98a3SKenneth D. Merry if (procnum > NFSV2PROC_STATFS) 163d96b98a3SKenneth D. Merry return (-1); 164d96b98a3SKenneth D. Merry 165d96b98a3SKenneth D. Merry return (newnfs_nfsv3_procid[procnum]); 166d96b98a3SKenneth D. Merry } 167d96b98a3SKenneth D. Merry 1689897e357SRick Macklem static int 16910f8f58dSAlexander Motin fhanew_get_fh(uint64_t *fh, int v3, struct mbuf **md, caddr_t *dpos) 170d96b98a3SKenneth D. Merry { 171d96b98a3SKenneth D. Merry struct nfsrv_descript lnd, *nd; 172d96b98a3SKenneth D. Merry uint32_t *tl; 17310f8f58dSAlexander Motin uint8_t *buf; 17410f8f58dSAlexander Motin uint64_t t; 17510f8f58dSAlexander Motin int error, len, i; 176d96b98a3SKenneth D. Merry 177d96b98a3SKenneth D. Merry error = 0; 178d96b98a3SKenneth D. Merry len = 0; 179d96b98a3SKenneth D. Merry nd = &lnd; 180d96b98a3SKenneth D. Merry 181d96b98a3SKenneth D. Merry nd->nd_md = *md; 182d96b98a3SKenneth D. Merry nd->nd_dpos = *dpos; 183d96b98a3SKenneth D. Merry 184d96b98a3SKenneth D. Merry if (v3) { 185d96b98a3SKenneth D. Merry NFSM_DISSECT_NONBLOCK(tl, uint32_t *, NFSX_UNSIGNED); 186d96b98a3SKenneth D. Merry if ((len = fxdr_unsigned(int, *tl)) <= 0 || len > NFSX_FHMAX) { 187d96b98a3SKenneth D. Merry error = EBADRPC; 188d96b98a3SKenneth D. Merry goto nfsmout; 189d96b98a3SKenneth D. Merry } 190d96b98a3SKenneth D. Merry } else { 191d96b98a3SKenneth D. Merry len = NFSX_V2FH; 192d96b98a3SKenneth D. Merry } 193d96b98a3SKenneth D. Merry 19410f8f58dSAlexander Motin t = 0; 195d96b98a3SKenneth D. Merry if (len != 0) { 19610f8f58dSAlexander Motin NFSM_DISSECT_NONBLOCK(buf, uint8_t *, len); 19710f8f58dSAlexander Motin for (i = 0; i < len; i++) 19810f8f58dSAlexander Motin t ^= ((uint64_t)buf[i] << (i & 7) * 8); 19910f8f58dSAlexander Motin } 20010f8f58dSAlexander Motin *fh = t; 201d96b98a3SKenneth D. Merry 202d96b98a3SKenneth D. Merry nfsmout: 203d96b98a3SKenneth D. Merry *md = nd->nd_md; 204d96b98a3SKenneth D. Merry *dpos = nd->nd_dpos; 205d96b98a3SKenneth D. Merry 206d96b98a3SKenneth D. Merry return (error); 207d96b98a3SKenneth D. Merry } 208d96b98a3SKenneth D. Merry 2099897e357SRick Macklem static int 210d96b98a3SKenneth D. Merry fhanew_is_read(rpcproc_t procnum) 211d96b98a3SKenneth D. Merry { 212d96b98a3SKenneth D. Merry if (procnum == NFSPROC_READ) 213d96b98a3SKenneth D. Merry return (1); 214d96b98a3SKenneth D. Merry else 215d96b98a3SKenneth D. Merry return (0); 216d96b98a3SKenneth D. Merry } 217d96b98a3SKenneth D. Merry 2189897e357SRick Macklem static int 219d96b98a3SKenneth D. Merry fhanew_is_write(rpcproc_t procnum) 220d96b98a3SKenneth D. Merry { 221d96b98a3SKenneth D. Merry if (procnum == NFSPROC_WRITE) 222d96b98a3SKenneth D. Merry return (1); 223d96b98a3SKenneth D. Merry else 224d96b98a3SKenneth D. Merry return (0); 225d96b98a3SKenneth D. Merry } 226d96b98a3SKenneth D. Merry 2279897e357SRick Macklem static int 228d96b98a3SKenneth D. Merry fhanew_get_offset(struct mbuf **md, caddr_t *dpos, int v3, 229d96b98a3SKenneth D. Merry struct fha_info *info) 230d96b98a3SKenneth D. Merry { 231d96b98a3SKenneth D. Merry struct nfsrv_descript lnd, *nd; 232d96b98a3SKenneth D. Merry uint32_t *tl; 233d96b98a3SKenneth D. Merry int error; 234d96b98a3SKenneth D. Merry 235d96b98a3SKenneth D. Merry error = 0; 236d96b98a3SKenneth D. Merry 237d96b98a3SKenneth D. Merry nd = &lnd; 238d96b98a3SKenneth D. Merry nd->nd_md = *md; 239d96b98a3SKenneth D. Merry nd->nd_dpos = *dpos; 240d96b98a3SKenneth D. Merry 241d96b98a3SKenneth D. Merry if (v3) { 242d96b98a3SKenneth D. Merry NFSM_DISSECT_NONBLOCK(tl, uint32_t *, 2 * NFSX_UNSIGNED); 243d96b98a3SKenneth D. Merry info->offset = fxdr_hyper(tl); 244d96b98a3SKenneth D. Merry } else { 245d96b98a3SKenneth D. Merry NFSM_DISSECT_NONBLOCK(tl, uint32_t *, NFSX_UNSIGNED); 246d96b98a3SKenneth D. Merry info->offset = fxdr_unsigned(uint32_t, *tl); 247d96b98a3SKenneth D. Merry } 248d96b98a3SKenneth D. Merry 249d96b98a3SKenneth D. Merry nfsmout: 250d96b98a3SKenneth D. Merry *md = nd->nd_md; 251d96b98a3SKenneth D. Merry *dpos = nd->nd_dpos; 252d96b98a3SKenneth D. Merry 253d96b98a3SKenneth D. Merry return (error); 254d96b98a3SKenneth D. Merry } 255d96b98a3SKenneth D. Merry 2569897e357SRick Macklem static int 257d96b98a3SKenneth D. Merry fhanew_no_offset(rpcproc_t procnum) 258d96b98a3SKenneth D. Merry { 259d96b98a3SKenneth D. Merry if (procnum == NFSPROC_FSSTAT || 260d96b98a3SKenneth D. Merry procnum == NFSPROC_FSINFO || 261d96b98a3SKenneth D. Merry procnum == NFSPROC_PATHCONF || 262d96b98a3SKenneth D. Merry procnum == NFSPROC_NOOP || 263d96b98a3SKenneth D. Merry procnum == NFSPROC_NULL) 264d96b98a3SKenneth D. Merry return (1); 265d96b98a3SKenneth D. Merry else 266d96b98a3SKenneth D. Merry return (0); 267d96b98a3SKenneth D. Merry } 268d96b98a3SKenneth D. Merry 2699897e357SRick Macklem static void 270d96b98a3SKenneth D. Merry fhanew_set_locktype(rpcproc_t procnum, struct fha_info *info) 271d96b98a3SKenneth D. Merry { 272d96b98a3SKenneth D. Merry switch (procnum) { 273d96b98a3SKenneth D. Merry case NFSPROC_NULL: 274d96b98a3SKenneth D. Merry case NFSPROC_GETATTR: 275d96b98a3SKenneth D. Merry case NFSPROC_LOOKUP: 276d96b98a3SKenneth D. Merry case NFSPROC_ACCESS: 277d96b98a3SKenneth D. Merry case NFSPROC_READLINK: 278d96b98a3SKenneth D. Merry case NFSPROC_READ: 279d96b98a3SKenneth D. Merry case NFSPROC_READDIR: 280d96b98a3SKenneth D. Merry case NFSPROC_READDIRPLUS: 281d96b98a3SKenneth D. Merry case NFSPROC_WRITE: 282d96b98a3SKenneth D. Merry info->locktype = LK_SHARED; 283d96b98a3SKenneth D. Merry break; 284d96b98a3SKenneth D. Merry case NFSPROC_SETATTR: 285d96b98a3SKenneth D. Merry case NFSPROC_CREATE: 286d96b98a3SKenneth D. Merry case NFSPROC_MKDIR: 287d96b98a3SKenneth D. Merry case NFSPROC_SYMLINK: 288d96b98a3SKenneth D. Merry case NFSPROC_MKNOD: 289d96b98a3SKenneth D. Merry case NFSPROC_REMOVE: 290d96b98a3SKenneth D. Merry case NFSPROC_RMDIR: 291d96b98a3SKenneth D. Merry case NFSPROC_RENAME: 292d96b98a3SKenneth D. Merry case NFSPROC_LINK: 293d96b98a3SKenneth D. Merry case NFSPROC_FSSTAT: 294d96b98a3SKenneth D. Merry case NFSPROC_FSINFO: 295d96b98a3SKenneth D. Merry case NFSPROC_PATHCONF: 296d96b98a3SKenneth D. Merry case NFSPROC_COMMIT: 297d96b98a3SKenneth D. Merry case NFSPROC_NOOP: 298d96b98a3SKenneth D. Merry info->locktype = LK_EXCLUSIVE; 299d96b98a3SKenneth D. Merry break; 300d96b98a3SKenneth D. Merry } 301d96b98a3SKenneth D. Merry } 302d96b98a3SKenneth D. Merry 3039897e357SRick Macklem /* 3049897e357SRick Macklem * This just specifies that offsets should obey affinity when within 3059897e357SRick Macklem * the same 1Mbyte (1<<20) chunk for the file (reads only for now). 3069897e357SRick Macklem */ 3079897e357SRick Macklem static void 3089897e357SRick Macklem fha_extract_info(struct svc_req *req, struct fha_info *i) 309d96b98a3SKenneth D. Merry { 3109897e357SRick Macklem struct mbuf *md; 3119897e357SRick Macklem caddr_t dpos; 3129897e357SRick Macklem static u_int64_t random_fh = 0; 3139897e357SRick Macklem int error; 3149897e357SRick Macklem int v3 = (req->rq_vers == 3); 3159897e357SRick Macklem rpcproc_t procnum; 3169897e357SRick Macklem 3179897e357SRick Macklem /* 3189897e357SRick Macklem * We start off with a random fh. If we get a reasonable 3199897e357SRick Macklem * procnum, we set the fh. If there's a concept of offset 3209897e357SRick Macklem * that we're interested in, we set that. 3219897e357SRick Macklem */ 3229897e357SRick Macklem i->fh = ++random_fh; 3239897e357SRick Macklem i->offset = 0; 3249897e357SRick Macklem i->locktype = LK_EXCLUSIVE; 3259897e357SRick Macklem i->read = i->write = 0; 3269897e357SRick Macklem 3279897e357SRick Macklem /* 3289897e357SRick Macklem * Extract the procnum and convert to v3 form if necessary, 3299897e357SRick Macklem * taking care to deal with out-of-range procnums. Caller will 3309897e357SRick Macklem * ensure that rq_vers is either 2 or 3. 3319897e357SRick Macklem */ 3329897e357SRick Macklem procnum = req->rq_proc; 3339897e357SRick Macklem if (!v3) { 3349897e357SRick Macklem rpcproc_t tmp_procnum; 3359897e357SRick Macklem 3369897e357SRick Macklem tmp_procnum = fhanew_get_procnum(procnum); 3379897e357SRick Macklem if (tmp_procnum == -1) 3389897e357SRick Macklem goto out; 3399897e357SRick Macklem procnum = tmp_procnum; 340d96b98a3SKenneth D. Merry } 341d96b98a3SKenneth D. Merry 3429897e357SRick Macklem /* 3439897e357SRick Macklem * We do affinity for most. However, we divide a realm of affinity 3449897e357SRick Macklem * by file offset so as to allow for concurrent random access. We 3459897e357SRick Macklem * only do this for reads today, but this may change when IFS supports 3469897e357SRick Macklem * efficient concurrent writes. 3479897e357SRick Macklem */ 3489897e357SRick Macklem if (fhanew_no_offset(procnum)) 3499897e357SRick Macklem goto out; 350d96b98a3SKenneth D. Merry 3519897e357SRick Macklem i->read = fhanew_is_read(procnum); 3529897e357SRick Macklem i->write = fhanew_is_write(procnum); 3539897e357SRick Macklem 3549897e357SRick Macklem error = newnfs_realign(&req->rq_args, M_NOWAIT); 3559897e357SRick Macklem if (error) 3569897e357SRick Macklem goto out; 3579897e357SRick Macklem md = req->rq_args; 3589897e357SRick Macklem dpos = mtod(md, caddr_t); 3599897e357SRick Macklem 3609897e357SRick Macklem /* Grab the filehandle. */ 3619897e357SRick Macklem error = fhanew_get_fh(&i->fh, v3, &md, &dpos); 3629897e357SRick Macklem if (error) 3639897e357SRick Macklem goto out; 3649897e357SRick Macklem 3659897e357SRick Macklem /* Content ourselves with zero offset for all but reads. */ 3669897e357SRick Macklem if (i->read || i->write) 3679897e357SRick Macklem fhanew_get_offset(&md, &dpos, v3, i); 3689897e357SRick Macklem 3699897e357SRick Macklem out: 3709897e357SRick Macklem fhanew_set_locktype(procnum, i); 3719897e357SRick Macklem } 3729897e357SRick Macklem 3739897e357SRick Macklem static struct fha_hash_entry * 3749897e357SRick Macklem fha_hash_entry_new(u_int64_t fh) 3759897e357SRick Macklem { 3769897e357SRick Macklem struct fha_hash_entry *e; 3779897e357SRick Macklem 3789897e357SRick Macklem e = malloc(sizeof(*e), M_NFS_FHA, M_WAITOK); 3799897e357SRick Macklem e->fh = fh; 3809897e357SRick Macklem e->num_rw = 0; 3819897e357SRick Macklem e->num_exclusive = 0; 3829897e357SRick Macklem e->num_threads = 0; 3839897e357SRick Macklem LIST_INIT(&e->threads); 3849897e357SRick Macklem 3859897e357SRick Macklem return (e); 3869897e357SRick Macklem } 3879897e357SRick Macklem 3889897e357SRick Macklem static void 3899897e357SRick Macklem fha_hash_entry_destroy(struct fha_hash_entry *e) 3909897e357SRick Macklem { 3919897e357SRick Macklem 3929897e357SRick Macklem mtx_assert(e->mtx, MA_OWNED); 3939897e357SRick Macklem KASSERT(e->num_rw == 0, 3949897e357SRick Macklem ("%d reqs on destroyed fhe %p", e->num_rw, e)); 3959897e357SRick Macklem KASSERT(e->num_exclusive == 0, 3969897e357SRick Macklem ("%d exclusive reqs on destroyed fhe %p", e->num_exclusive, e)); 3979897e357SRick Macklem KASSERT(e->num_threads == 0, 3989897e357SRick Macklem ("%d threads on destroyed fhe %p", e->num_threads, e)); 3999897e357SRick Macklem free(e, M_NFS_FHA); 4009897e357SRick Macklem } 4019897e357SRick Macklem 4029897e357SRick Macklem static void 4039897e357SRick Macklem fha_hash_entry_remove(struct fha_hash_entry *e) 4049897e357SRick Macklem { 4059897e357SRick Macklem 4069897e357SRick Macklem mtx_assert(e->mtx, MA_OWNED); 4079897e357SRick Macklem LIST_REMOVE(e, link); 4089897e357SRick Macklem fha_hash_entry_destroy(e); 4099897e357SRick Macklem } 4109897e357SRick Macklem 4119897e357SRick Macklem static struct fha_hash_entry * 4129897e357SRick Macklem fha_hash_entry_lookup(struct fha_params *softc, u_int64_t fh) 4139897e357SRick Macklem { 4149897e357SRick Macklem struct fha_hash_slot *fhs; 4159897e357SRick Macklem struct fha_hash_entry *fhe, *new_fhe; 4169897e357SRick Macklem 4179897e357SRick Macklem fhs = &softc->fha_hash[fh % FHA_HASH_SIZE]; 4189897e357SRick Macklem new_fhe = fha_hash_entry_new(fh); 4199897e357SRick Macklem new_fhe->mtx = &fhs->mtx; 4209897e357SRick Macklem mtx_lock(&fhs->mtx); 4219897e357SRick Macklem LIST_FOREACH(fhe, &fhs->list, link) 4229897e357SRick Macklem if (fhe->fh == fh) 4239897e357SRick Macklem break; 4249897e357SRick Macklem if (!fhe) { 4259897e357SRick Macklem fhe = new_fhe; 4269897e357SRick Macklem LIST_INSERT_HEAD(&fhs->list, fhe, link); 4279897e357SRick Macklem } else 4289897e357SRick Macklem fha_hash_entry_destroy(new_fhe); 4299897e357SRick Macklem return (fhe); 4309897e357SRick Macklem } 4319897e357SRick Macklem 4329897e357SRick Macklem static void 4339897e357SRick Macklem fha_hash_entry_add_thread(struct fha_hash_entry *fhe, SVCTHREAD *thread) 4349897e357SRick Macklem { 4359897e357SRick Macklem 4369897e357SRick Macklem mtx_assert(fhe->mtx, MA_OWNED); 4379897e357SRick Macklem thread->st_p2 = 0; 4389897e357SRick Macklem LIST_INSERT_HEAD(&fhe->threads, thread, st_alink); 4399897e357SRick Macklem fhe->num_threads++; 4409897e357SRick Macklem } 4419897e357SRick Macklem 4429897e357SRick Macklem static void 4439897e357SRick Macklem fha_hash_entry_remove_thread(struct fha_hash_entry *fhe, SVCTHREAD *thread) 4449897e357SRick Macklem { 4459897e357SRick Macklem 4469897e357SRick Macklem mtx_assert(fhe->mtx, MA_OWNED); 4479897e357SRick Macklem KASSERT(thread->st_p2 == 0, 4489897e357SRick Macklem ("%d reqs on removed thread %p", thread->st_p2, thread)); 4499897e357SRick Macklem LIST_REMOVE(thread, st_alink); 4509897e357SRick Macklem fhe->num_threads--; 4519897e357SRick Macklem } 4529897e357SRick Macklem 4539897e357SRick Macklem /* 4549897e357SRick Macklem * Account for an ongoing operation associated with this file. 4559897e357SRick Macklem */ 4569897e357SRick Macklem static void 4579897e357SRick Macklem fha_hash_entry_add_op(struct fha_hash_entry *fhe, int locktype, int count) 4589897e357SRick Macklem { 4599897e357SRick Macklem 4609897e357SRick Macklem mtx_assert(fhe->mtx, MA_OWNED); 4619897e357SRick Macklem if (LK_EXCLUSIVE == locktype) 4629897e357SRick Macklem fhe->num_exclusive += count; 4639897e357SRick Macklem else 4649897e357SRick Macklem fhe->num_rw += count; 4659897e357SRick Macklem } 4669897e357SRick Macklem 4679897e357SRick Macklem /* 4689897e357SRick Macklem * Get the service thread currently associated with the fhe that is 4699897e357SRick Macklem * appropriate to handle this operation. 4709897e357SRick Macklem */ 4719897e357SRick Macklem static SVCTHREAD * 4729897e357SRick Macklem fha_hash_entry_choose_thread(struct fha_params *softc, 4739897e357SRick Macklem struct fha_hash_entry *fhe, struct fha_info *i, SVCTHREAD *this_thread) 4749897e357SRick Macklem { 4759897e357SRick Macklem SVCTHREAD *thread, *min_thread = NULL; 4769897e357SRick Macklem int req_count, min_count = 0; 4779897e357SRick Macklem off_t offset1, offset2; 4789897e357SRick Macklem 4799897e357SRick Macklem LIST_FOREACH(thread, &fhe->threads, st_alink) { 4809897e357SRick Macklem req_count = thread->st_p2; 4819897e357SRick Macklem 4829897e357SRick Macklem /* If there are any writes in progress, use the first thread. */ 4839897e357SRick Macklem if (fhe->num_exclusive) { 4849897e357SRick Macklem #if 0 4859897e357SRick Macklem ITRACE_CURPROC(ITRACE_NFS, ITRACE_INFO, 4869897e357SRick Macklem "fha: %p(%d)w", thread, req_count); 4879897e357SRick Macklem #endif 4889897e357SRick Macklem return (thread); 4899897e357SRick Macklem } 4909897e357SRick Macklem 4919897e357SRick Macklem /* Check whether we should consider locality. */ 4929897e357SRick Macklem if ((i->read && !softc->ctls.read) || 4939897e357SRick Macklem (i->write && !softc->ctls.write)) 4949897e357SRick Macklem goto noloc; 4959897e357SRick Macklem 4969897e357SRick Macklem /* 4979897e357SRick Macklem * Check for locality, making sure that we won't 4989897e357SRick Macklem * exceed our per-thread load limit in the process. 4999897e357SRick Macklem */ 5009897e357SRick Macklem offset1 = i->offset; 5019897e357SRick Macklem offset2 = thread->st_p3; 5029897e357SRick Macklem 5039897e357SRick Macklem if (((offset1 >= offset2) 5049897e357SRick Macklem && ((offset1 - offset2) < (1 << softc->ctls.bin_shift))) 5059897e357SRick Macklem || ((offset2 > offset1) 5069897e357SRick Macklem && ((offset2 - offset1) < (1 << softc->ctls.bin_shift)))) { 5079897e357SRick Macklem if ((softc->ctls.max_reqs_per_nfsd == 0) || 5089897e357SRick Macklem (req_count < softc->ctls.max_reqs_per_nfsd)) { 5099897e357SRick Macklem #if 0 5109897e357SRick Macklem ITRACE_CURPROC(ITRACE_NFS, ITRACE_INFO, 5119897e357SRick Macklem "fha: %p(%d)r", thread, req_count); 5129897e357SRick Macklem #endif 5139897e357SRick Macklem return (thread); 5149897e357SRick Macklem } 5159897e357SRick Macklem } 5169897e357SRick Macklem 5179897e357SRick Macklem noloc: 5189897e357SRick Macklem /* 5199897e357SRick Macklem * We don't have a locality match, so skip this thread, 5209897e357SRick Macklem * but keep track of the most attractive thread in case 5219897e357SRick Macklem * we need to come back to it later. 5229897e357SRick Macklem */ 5239897e357SRick Macklem #if 0 5249897e357SRick Macklem ITRACE_CURPROC(ITRACE_NFS, ITRACE_INFO, 5259897e357SRick Macklem "fha: %p(%d)s off1 %llu off2 %llu", thread, 5269897e357SRick Macklem req_count, offset1, offset2); 5279897e357SRick Macklem #endif 5289897e357SRick Macklem if ((min_thread == NULL) || (req_count < min_count)) { 5299897e357SRick Macklem min_count = req_count; 5309897e357SRick Macklem min_thread = thread; 5319897e357SRick Macklem } 5329897e357SRick Macklem } 5339897e357SRick Macklem 5349897e357SRick Macklem /* 5359897e357SRick Macklem * We didn't find a good match yet. See if we can add 5369897e357SRick Macklem * a new thread to this file handle entry's thread list. 5379897e357SRick Macklem */ 5389897e357SRick Macklem if ((softc->ctls.max_nfsds_per_fh == 0) || 5399897e357SRick Macklem (fhe->num_threads < softc->ctls.max_nfsds_per_fh)) { 5409897e357SRick Macklem thread = this_thread; 5419897e357SRick Macklem #if 0 5429897e357SRick Macklem ITRACE_CURPROC(ITRACE_NFS, ITRACE_INFO, 5439897e357SRick Macklem "fha: %p(%d)t", thread, thread->st_p2); 5449897e357SRick Macklem #endif 5459897e357SRick Macklem fha_hash_entry_add_thread(fhe, thread); 5469897e357SRick Macklem } else { 5479897e357SRick Macklem /* 5489897e357SRick Macklem * We don't want to use any more threads for this file, so 5499897e357SRick Macklem * go back to the most attractive nfsd we're already using. 5509897e357SRick Macklem */ 5519897e357SRick Macklem thread = min_thread; 5529897e357SRick Macklem } 5539897e357SRick Macklem 5549897e357SRick Macklem return (thread); 5559897e357SRick Macklem } 5569897e357SRick Macklem 5579897e357SRick Macklem /* 5589897e357SRick Macklem * After getting a request, try to assign it to some thread. Usually we 5599897e357SRick Macklem * handle it ourselves. 5609897e357SRick Macklem */ 561d96b98a3SKenneth D. Merry SVCTHREAD * 562d96b98a3SKenneth D. Merry fhanew_assign(SVCTHREAD *this_thread, struct svc_req *req) 563d96b98a3SKenneth D. Merry { 564cd406ac9SRick Macklem struct fha_params *softc; 5659897e357SRick Macklem SVCTHREAD *thread; 5669897e357SRick Macklem struct fha_info i; 5679897e357SRick Macklem struct fha_hash_entry *fhe; 5689897e357SRick Macklem 569cd406ac9SRick Macklem NFSD_CURVNET_SET(NFSD_TD_TO_VNET(curthread)); 570cd406ac9SRick Macklem softc = NFSD_VNET(fhanew_softc); 5719897e357SRick Macklem /* Check to see whether we're enabled. */ 5729897e357SRick Macklem if (softc->ctls.enable == 0) 5739897e357SRick Macklem goto thist; 5749897e357SRick Macklem 5759897e357SRick Macklem /* 5769897e357SRick Macklem * Only do placement if this is an NFS request. 5779897e357SRick Macklem */ 5789897e357SRick Macklem if (req->rq_prog != NFS_PROG) 5799897e357SRick Macklem goto thist; 5809897e357SRick Macklem 5819897e357SRick Macklem if (req->rq_vers != 2 && req->rq_vers != 3) 5829897e357SRick Macklem goto thist; 5839897e357SRick Macklem 5849897e357SRick Macklem fha_extract_info(req, &i); 5859897e357SRick Macklem 5869897e357SRick Macklem /* 5879897e357SRick Macklem * We save the offset associated with this request for later 5889897e357SRick Macklem * nfsd matching. 5899897e357SRick Macklem */ 5909897e357SRick Macklem fhe = fha_hash_entry_lookup(softc, i.fh); 5919897e357SRick Macklem req->rq_p1 = fhe; 5929897e357SRick Macklem req->rq_p2 = i.locktype; 5939897e357SRick Macklem req->rq_p3 = i.offset; 5949897e357SRick Macklem 5959897e357SRick Macklem /* 5969897e357SRick Macklem * Choose a thread, taking into consideration locality, thread load, 5979897e357SRick Macklem * and the number of threads already working on this file. 5989897e357SRick Macklem */ 5999897e357SRick Macklem thread = fha_hash_entry_choose_thread(softc, fhe, &i, this_thread); 6009897e357SRick Macklem KASSERT(thread, ("fha_assign: NULL thread!")); 6019897e357SRick Macklem fha_hash_entry_add_op(fhe, i.locktype, 1); 6029897e357SRick Macklem thread->st_p2++; 6039897e357SRick Macklem thread->st_p3 = i.offset; 6049897e357SRick Macklem 6059897e357SRick Macklem /* 6069897e357SRick Macklem * Grab the pool lock here to not let chosen thread go away before 6079897e357SRick Macklem * the new request inserted to its queue while we drop fhe lock. 6089897e357SRick Macklem */ 6099897e357SRick Macklem mtx_lock(&thread->st_lock); 6109897e357SRick Macklem mtx_unlock(fhe->mtx); 6119897e357SRick Macklem 612cd406ac9SRick Macklem NFSD_CURVNET_RESTORE(); 6139897e357SRick Macklem return (thread); 6149897e357SRick Macklem thist: 6159897e357SRick Macklem req->rq_p1 = NULL; 616cd406ac9SRick Macklem NFSD_CURVNET_RESTORE(); 6179897e357SRick Macklem mtx_lock(&this_thread->st_lock); 6189897e357SRick Macklem return (this_thread); 6199897e357SRick Macklem } 6209897e357SRick Macklem 6219897e357SRick Macklem /* 6229897e357SRick Macklem * Called when we're done with an operation. The request has already 6239897e357SRick Macklem * been de-queued. 6249897e357SRick Macklem */ 6259897e357SRick Macklem void 6269897e357SRick Macklem fhanew_nd_complete(SVCTHREAD *thread, struct svc_req *req) 6279897e357SRick Macklem { 6289897e357SRick Macklem struct fha_hash_entry *fhe = req->rq_p1; 6299897e357SRick Macklem struct mtx *mtx; 6309897e357SRick Macklem 631cd406ac9SRick Macklem NFSD_CURVNET_SET(NFSD_TD_TO_VNET(curthread)); 6329897e357SRick Macklem /* 6339897e357SRick Macklem * This may be called for reqs that didn't go through 6349897e357SRick Macklem * fha_assign (e.g. extra NULL ops used for RPCSEC_GSS. 6359897e357SRick Macklem */ 636cd406ac9SRick Macklem if (!fhe) { 637cd406ac9SRick Macklem NFSD_CURVNET_RESTORE(); 6389897e357SRick Macklem return; 639cd406ac9SRick Macklem } 6409897e357SRick Macklem 6419897e357SRick Macklem mtx = fhe->mtx; 6429897e357SRick Macklem mtx_lock(mtx); 6439897e357SRick Macklem fha_hash_entry_add_op(fhe, req->rq_p2, -1); 6449897e357SRick Macklem thread->st_p2--; 6459897e357SRick Macklem KASSERT(thread->st_p2 >= 0, ("Negative request count %d on %p", 6469897e357SRick Macklem thread->st_p2, thread)); 6479897e357SRick Macklem if (thread->st_p2 == 0) { 6489897e357SRick Macklem fha_hash_entry_remove_thread(fhe, thread); 6499897e357SRick Macklem if (0 == fhe->num_rw + fhe->num_exclusive) 6509897e357SRick Macklem fha_hash_entry_remove(fhe); 6519897e357SRick Macklem } 6529897e357SRick Macklem mtx_unlock(mtx); 653cd406ac9SRick Macklem NFSD_CURVNET_RESTORE(); 6549897e357SRick Macklem } 6559897e357SRick Macklem 6569897e357SRick Macklem static int 6579897e357SRick Macklem fhenew_stats_sysctl(SYSCTL_HANDLER_ARGS) 6589897e357SRick Macklem { 659cd406ac9SRick Macklem struct fha_params *softc; 6609897e357SRick Macklem int error, i; 6619897e357SRick Macklem struct sbuf sb; 6629897e357SRick Macklem struct fha_hash_entry *fhe; 6639897e357SRick Macklem bool_t first, hfirst; 6649897e357SRick Macklem SVCTHREAD *thread; 6659897e357SRick Macklem 6669897e357SRick Macklem sbuf_new(&sb, NULL, 65536, SBUF_FIXEDLEN); 6679897e357SRick Macklem 668cd406ac9SRick Macklem NFSD_CURVNET_SET(NFSD_TD_TO_VNET(curthread)); 669cd406ac9SRick Macklem softc = NFSD_VNET(fhanew_softc); 6709897e357SRick Macklem for (i = 0; i < FHA_HASH_SIZE; i++) 6719897e357SRick Macklem if (!LIST_EMPTY(&softc->fha_hash[i].list)) 6729897e357SRick Macklem break; 6739897e357SRick Macklem 6749897e357SRick Macklem if (i == FHA_HASH_SIZE) { 6759897e357SRick Macklem sbuf_printf(&sb, "No file handle entries.\n"); 6769897e357SRick Macklem goto out; 6779897e357SRick Macklem } 6789897e357SRick Macklem 6799897e357SRick Macklem hfirst = TRUE; 6809897e357SRick Macklem for (; i < FHA_HASH_SIZE; i++) { 6819897e357SRick Macklem mtx_lock(&softc->fha_hash[i].mtx); 6829897e357SRick Macklem if (LIST_EMPTY(&softc->fha_hash[i].list)) { 6839897e357SRick Macklem mtx_unlock(&softc->fha_hash[i].mtx); 6849897e357SRick Macklem continue; 6859897e357SRick Macklem } 6869897e357SRick Macklem sbuf_printf(&sb, "%shash %d: {\n", hfirst ? "" : ", ", i); 6879897e357SRick Macklem first = TRUE; 6889897e357SRick Macklem LIST_FOREACH(fhe, &softc->fha_hash[i].list, link) { 6899897e357SRick Macklem sbuf_printf(&sb, "%sfhe %p: {\n", first ? " " : ", ", 6909897e357SRick Macklem fhe); 6919897e357SRick Macklem sbuf_printf(&sb, " fh: %ju\n", (uintmax_t) fhe->fh); 6929897e357SRick Macklem sbuf_printf(&sb, " num_rw/exclusive: %d/%d\n", 6939897e357SRick Macklem fhe->num_rw, fhe->num_exclusive); 6949897e357SRick Macklem sbuf_printf(&sb, " num_threads: %d\n", 6959897e357SRick Macklem fhe->num_threads); 6969897e357SRick Macklem 6979897e357SRick Macklem LIST_FOREACH(thread, &fhe->threads, st_alink) { 6989897e357SRick Macklem sbuf_printf(&sb, " thread %p offset %ju " 6999897e357SRick Macklem "reqs %d\n", thread, 7009897e357SRick Macklem thread->st_p3, thread->st_p2); 7019897e357SRick Macklem } 7029897e357SRick Macklem 7039897e357SRick Macklem sbuf_printf(&sb, " }"); 7049897e357SRick Macklem first = FALSE; 7059897e357SRick Macklem } 7069897e357SRick Macklem sbuf_printf(&sb, "\n}"); 7079897e357SRick Macklem mtx_unlock(&softc->fha_hash[i].mtx); 7089897e357SRick Macklem hfirst = FALSE; 7099897e357SRick Macklem } 7109897e357SRick Macklem 7119897e357SRick Macklem out: 712cd406ac9SRick Macklem NFSD_CURVNET_RESTORE(); 7139897e357SRick Macklem sbuf_trim(&sb); 7149897e357SRick Macklem sbuf_finish(&sb); 7159897e357SRick Macklem error = sysctl_handle_string(oidp, sbuf_data(&sb), sbuf_len(&sb), req); 7169897e357SRick Macklem sbuf_delete(&sb); 7179897e357SRick Macklem return (error); 718d96b98a3SKenneth D. Merry } 719