lustre/ptlrpc/nodemap_handler.c

   1 /*
   2  * GPL HEADER START
   3  *
   4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License version 2 only,
   8  * as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License version 2 for more details (a copy is included
  14  * in the LICENSE file that accompanied this code).
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * version 2 along with this program; If not, see
  18  * http://www.gnu.org/licenses/gpl-2.0.html
  19  *
  20  * GPL HEADER END
  21  */
  22 /*
  23  * Copyright (C) 2013, Trustees of Indiana University
  24  *
  25  * Copyright (c) 2017, Intel Corporation.
  26  *
  27  * Author: Joshua Walgenbach <jjw@iu.edu>
  28  */
  29 #include <linux/module.h>
  30 #include <linux/sort.h>
  31 #include <uapi/linux/lnet/nidstr.h>
  32 #include <lustre_net.h>
  33 #include <lustre_acl.h>
  34 #include <lustre_eacl.h>
  35 #include <obd_class.h>
  36 #include "nodemap_internal.h"
  37
  38 #define HASH_NODEMAP_BKT_BITS 3
  39 #define HASH_NODEMAP_CUR_BITS 3
  40 #define HASH_NODEMAP_MAX_BITS 7
  41
  42 #define DEFAULT_NODEMAP "default"
  43
  44 /* nodemap proc root proc directory under fs/lustre */
  45 struct proc_dir_entry *proc_lustre_nodemap_root;
  46
  47 /* Copy of config active flag to avoid locking in mapping functions */
  48 bool nodemap_active;
  49
  50 /* Lock protecting the active config, useful primarily when proc and
  51  * nodemap_hash might be replaced when loading a new config
  52  * Any time the active config is referenced, the lock should be held.
  53  */
  54 DEFINE_MUTEX(active_config_lock);
  55 struct nodemap_config *active_config;
  56
  57 /**
  58  * Nodemap destructor
  59  *
  60  * \param       nodemap         nodemap to destroy
  61  */
  62 static void nodemap_destroy(struct lu_nodemap *nodemap)
  63 {
  64         ENTRY;
  65
  66         if (nodemap->nm_pde_data != NULL)
  67                 lprocfs_nodemap_remove(nodemap->nm_pde_data);
  68
  69         mutex_lock(&active_config_lock);
  70         down_read(&active_config->nmc_range_tree_lock);
  71         nm_member_reclassify_nodemap(nodemap);
  72         up_read(&active_config->nmc_range_tree_lock);
  73
  74         down_write(&nodemap->nm_idmap_lock);
  75         idmap_delete_tree(nodemap);
  76         up_write(&nodemap->nm_idmap_lock);
  77
  78         mutex_unlock(&active_config_lock);
  79
  80         if (!list_empty(&nodemap->nm_member_list))
  81                 CWARN("nodemap_destroy failed to reclassify all members\n");
  82
  83         nm_member_delete_list(nodemap);
  84
  85         OBD_FREE_PTR(nodemap);
  86
  87         EXIT;
  88 }
  89
  90 /**
  91  * Functions used for the cfs_hash
  92  */
  93 void nodemap_getref(struct lu_nodemap *nodemap)
  94 {
  95         atomic_inc(&nodemap->nm_refcount);
  96         CDEBUG(D_INFO, "GETting nodemap %s(p=%p) : new refcount %d\n",
  97                nodemap->nm_name, nodemap, atomic_read(&nodemap->nm_refcount));
  98 }
  99
 100 /**
 101  * Destroy nodemap if last reference is put. Should be called outside
 102  * active_config_lock
 103  */
 104 void nodemap_putref(struct lu_nodemap *nodemap)
 105 {
 106         if (!nodemap)
 107                 return;
 108
 109         LASSERT(atomic_read(&nodemap->nm_refcount) > 0);
 110
 111         CDEBUG(D_INFO, "PUTting nodemap %s(p=%p) : new refcount %d\n",
 112                nodemap->nm_name, nodemap,
 113                atomic_read(&nodemap->nm_refcount) - 1);
 114
 115         if (atomic_dec_and_test(&nodemap->nm_refcount))
 116                 nodemap_destroy(nodemap);
 117 }
 118 EXPORT_SYMBOL(nodemap_putref);
 119
 120 static __u32 nodemap_hashfn(struct cfs_hash *hash_body,
 121                             const void *key, unsigned mask)
 122 {
 123         return cfs_hash_djb2_hash(key, strlen(key), mask);
 124 }
 125
 126 static void *nodemap_hs_key(struct hlist_node *hnode)
 127 {
 128         struct lu_nodemap *nodemap;
 129
 130         nodemap = hlist_entry(hnode, struct lu_nodemap, nm_hash);
 131
 132         return nodemap->nm_name;
 133 }
 134
 135 static int nodemap_hs_keycmp(const void *key,
 136                              struct hlist_node *compared_hnode)
 137 {
 138         char *nodemap_name;
 139
 140         nodemap_name = nodemap_hs_key(compared_hnode);
 141
 142         return !strcmp(key, nodemap_name);
 143 }
 144
 145 static void *nodemap_hs_hashobject(struct hlist_node *hnode)
 146 {
 147         return hlist_entry(hnode, struct lu_nodemap, nm_hash);
 148 }
 149
 150 static void nodemap_hs_get(struct cfs_hash *hs, struct hlist_node *hnode)
 151 {
 152         struct lu_nodemap *nodemap;
 153
 154         nodemap = hlist_entry(hnode, struct lu_nodemap, nm_hash);
 155         nodemap_getref(nodemap);
 156 }
 157
 158 static void nodemap_hs_put_locked(struct cfs_hash *hs,
 159                                   struct hlist_node *hnode)
 160 {
 161         struct lu_nodemap *nodemap;
 162
 163         nodemap = hlist_entry(hnode, struct lu_nodemap, nm_hash);
 164         nodemap_putref(nodemap);
 165 }
 166
 167 static struct cfs_hash_ops nodemap_hash_operations = {
 168         .hs_hash        = nodemap_hashfn,
 169         .hs_key         = nodemap_hs_key,
 170         .hs_keycmp      = nodemap_hs_keycmp,
 171         .hs_object      = nodemap_hs_hashobject,
 172         .hs_get         = nodemap_hs_get,
 173         .hs_put_locked  = nodemap_hs_put_locked,
 174 };
 175
 176 /* end of cfs_hash functions */
 177
 178 /**
 179  * Initialize nodemap_hash
 180  *
 181  * \retval      0               success
 182  * \retval      -ENOMEM         cannot create hash
 183  */
 184 static int nodemap_init_hash(struct nodemap_config *nmc)
 185 {
 186         nmc->nmc_nodemap_hash = cfs_hash_create("NODEMAP",
 187                                                 HASH_NODEMAP_CUR_BITS,
 188                                                 HASH_NODEMAP_MAX_BITS,
 189                                                 HASH_NODEMAP_BKT_BITS, 0,
 190                                                 CFS_HASH_MIN_THETA,
 191                                                 CFS_HASH_MAX_THETA,
 192                                                 &nodemap_hash_operations,
 193                                                 CFS_HASH_DEFAULT);
 194
 195         if (nmc->nmc_nodemap_hash == NULL) {
 196                 CERROR("cannot create nodemap_hash table\n");
 197                 return -ENOMEM;
 198         }
 199
 200         return 0;
 201 }
 202
 203 /**
 204  * Check for valid nodemap name
 205  *
 206  * \param       name            nodemap name
 207  * \retval      true            valid
 208  * \retval      false           invalid
 209  */
 210 static bool nodemap_name_is_valid(const char *name)
 211 {
 212         if (strlen(name) > LUSTRE_NODEMAP_NAME_LENGTH ||
 213             strlen(name) == 0)
 214                 return false;
 215
 216         for (; *name != '\0'; name++) {
 217                 if (!isalnum(*name) && *name != '_')
 218                         return false;
 219         }
 220
 221         return true;
 222 }
 223
 224 /**
 225  * Nodemap lookup
 226  *
 227  * Look nodemap up in the active_config nodemap hash. Caller should hold the
 228  * active_config_lock.
 229  *
 230  * \param       name            name of nodemap
 231  * \retval      nodemap         pointer set to found nodemap
 232  * \retval      -EINVAL         name is not valid
 233  * \retval      -ENOENT         nodemap not found
 234  */
 235 struct lu_nodemap *nodemap_lookup(const char *name)
 236 {
 237         struct lu_nodemap *nodemap = NULL;
 238
 239         if (!nodemap_name_is_valid(name))
 240                 return ERR_PTR(-EINVAL);
 241
 242         nodemap = cfs_hash_lookup(active_config->nmc_nodemap_hash, name);
 243         if (nodemap == NULL)
 244                 return ERR_PTR(-ENOENT);
 245
 246         return nodemap;
 247 }
 248
 249 /**
 250  * Classify the nid into the proper nodemap. Caller must hold active config and
 251  * nm_range_tree_lock, and call nodemap_putref when done with nodemap.
 252  *
 253  * \param       nid                     nid to classify
 254  * \retval      nodemap                 nodemap containing the nid
 255  * \retval      default_nodemap         default nodemap
 256  * \retval      -EINVAL                 LO nid given without other local nid
 257  */
 258 struct lu_nodemap *nodemap_classify_nid(lnet_nid_t nid)
 259 {
 260         struct lu_nid_range *range;
 261         struct lu_nodemap *nodemap;
 262         int rc;
 263
 264         ENTRY;
 265
 266         /* don't use 0@lo, use the first non-lo local NID instead */
 267         if (nid == LNET_NID_LO_0) {
 268                 struct lnet_process_id id;
 269                 int i = 0;
 270
 271                 do {
 272                         rc = LNetGetId(i++, &id);
 273                         if (rc < 0)
 274                                 RETURN(ERR_PTR(-EINVAL));
 275                 } while (id.nid == LNET_NID_LO_0);
 276
 277                 nid = id.nid;
 278                 CDEBUG(D_INFO, "found nid %s\n", libcfs_nid2str(nid));
 279         }
 280
 281         range = range_search(&active_config->nmc_range_tree, nid);
 282         if (range != NULL)
 283                 nodemap = range->rn_nodemap;
 284         else
 285                 nodemap = active_config->nmc_default_nodemap;
 286
 287         LASSERT(nodemap != NULL);
 288         nodemap_getref(nodemap);
 289
 290         RETURN(nodemap);
 291 }
 292
 293 /**
 294  * simple check for default nodemap
 295  */
 296 static bool is_default_nodemap(const struct lu_nodemap *nodemap)
 297 {
 298         return nodemap->nm_id == 0;
 299 }
 300
 301 /**
 302  * parse a nodemap range string into two nids
 303  *
 304  * \param       range_str               string to parse
 305  * \param       range[2]                array of two nids
 306  * \reyval      0 on success
 307  */
 308 int nodemap_parse_range(const char *range_str, lnet_nid_t range[2])
 309 {
 310         char    buf[LNET_NIDSTR_SIZE * 2 + 2];
 311         char    *ptr = NULL;
 312         char    *start_nidstr;
 313         char    *end_nidstr;
 314         int     rc = 0;
 315
 316         snprintf(buf, sizeof(buf), "%s", range_str);
 317         ptr = buf;
 318         start_nidstr = strsep(&ptr, ":");
 319         end_nidstr = strsep(&ptr, ":");
 320
 321         if (start_nidstr == NULL || end_nidstr == NULL)
 322                 GOTO(out, rc = -EINVAL);
 323
 324         range[0] = libcfs_str2nid(start_nidstr);
 325         range[1] = libcfs_str2nid(end_nidstr);
 326
 327 out:
 328         return rc;
 329
 330 }
 331 EXPORT_SYMBOL(nodemap_parse_range);
 332
 333 /**
 334  * parse a string containing an id map of form "client_id:filesystem_id"
 335  * into an array of __u32 * for use in mapping functions
 336  *
 337  * \param       idmap_str               map string
 338  * \param       idmap                   array[2] of __u32
 339  *
 340  * \retval      0 on success
 341  * \retval      -EINVAL if idmap cannot be parsed
 342  */
 343 int nodemap_parse_idmap(char *idmap_str, __u32 idmap[2])
 344 {
 345         char                    *sep;
 346         long unsigned int        idmap_buf;
 347         int                      rc;
 348
 349         if (idmap_str == NULL)
 350                 return -EINVAL;
 351
 352         sep = strchr(idmap_str, ':');
 353         if (sep == NULL)
 354                 return -EINVAL;
 355         *sep = '\0';
 356         sep++;
 357
 358         rc = kstrtoul(idmap_str, 10, &idmap_buf);
 359         if (rc != 0)
 360                 return -EINVAL;
 361         idmap[0] = idmap_buf;
 362
 363         rc = kstrtoul(sep, 10, &idmap_buf);
 364         if (rc != 0)
 365                 return -EINVAL;
 366         idmap[1] = idmap_buf;
 367
 368         return 0;
 369 }
 370 EXPORT_SYMBOL(nodemap_parse_idmap);
 371
 372 /**
 373  * add a member to a nodemap
 374  *
 375  * \param       nid             nid to add to the members
 376  * \param       exp             obd_export structure for the connection
 377  *                              that is being added
 378  * \retval      -EINVAL         export is NULL, or has invalid NID
 379  * \retval      -EEXIST         export is already member of a nodemap
 380  */
 381 int nodemap_add_member(lnet_nid_t nid, struct obd_export *exp)
 382 {
 383         struct lu_nodemap *nodemap;
 384         int rc = 0;
 385         ENTRY;
 386
 387         mutex_lock(&active_config_lock);
 388         down_read(&active_config->nmc_range_tree_lock);
 389
 390         nodemap = nodemap_classify_nid(nid);
 391
 392         if (IS_ERR(nodemap)) {
 393                 CWARN("%s: error adding to nodemap, no valid NIDs found\n",
 394                           exp->exp_obd->obd_name);
 395                 rc = -EINVAL;
 396         } else {
 397                 rc = nm_member_add(nodemap, exp);
 398         }
 399
 400         up_read(&active_config->nmc_range_tree_lock);
 401         mutex_unlock(&active_config_lock);
 402
 403         if (!IS_ERR(nodemap))
 404                 nodemap_putref(nodemap);
 405
 406         RETURN(rc);
 407 }
 408 EXPORT_SYMBOL(nodemap_add_member);
 409
 410 /**
 411  * delete a member from a nodemap
 412  *
 413  * \param       exp             export to remove from a nodemap
 414  */
 415 void nodemap_del_member(struct obd_export *exp)
 416 {
 417         struct lu_nodemap *nodemap;
 418
 419         ENTRY;
 420
 421         /* using ac lock to prevent nodemap reclassification while deleting */
 422         mutex_lock(&active_config_lock);
 423
 424         /* use of ted_nodemap is protected by active_config_lock. we take an
 425          * extra reference to make sure nodemap isn't destroyed under
 426          * active_config_lock
 427          */
 428         nodemap = exp->exp_target_data.ted_nodemap;
 429         if (nodemap == NULL)
 430                 goto out;
 431         else
 432                 nodemap_getref(nodemap);
 433
 434         mutex_lock(&nodemap->nm_member_list_lock);
 435         nm_member_del(nodemap, exp);
 436         mutex_unlock(&nodemap->nm_member_list_lock);
 437
 438 out:
 439         mutex_unlock(&active_config_lock);
 440
 441         if (nodemap)
 442                 nodemap_putref(nodemap);
 443
 444         EXIT;
 445 }
 446 EXPORT_SYMBOL(nodemap_del_member);
 447
 448 /**
 449  * add an idmap to the proper nodemap trees
 450  *
 451  * \param       nodemap         nodemap to add idmap to
 452  * \param       id_type         NODEMAP_UID or NODEMAP_GID
 453  * \param       map             array[2] __u32 containing the map values
 454  *                              map[0] is client id
 455  *                              map[1] is the filesystem id
 456  *
 457  * \retval      0       on success
 458  * \retval      < 0     if error occurs
 459  */
 460 int nodemap_add_idmap_helper(struct lu_nodemap *nodemap,
 461                              enum nodemap_id_type id_type,
 462                              const __u32 map[2])
 463 {
 464         struct lu_idmap         *idmap;
 465         struct lu_idmap         *temp;
 466         int                     rc = 0;
 467
 468         idmap = idmap_create(map[0], map[1]);
 469         if (idmap == NULL)
 470                 GOTO(out, rc = -ENOMEM);
 471
 472         down_write(&nodemap->nm_idmap_lock);
 473         temp = idmap_insert(id_type, idmap, nodemap);
 474         /* If the new id_client or id_fs is matched, the old idmap and its
 475          * index should be deleted according to its id_client before the new
 476          * idmap is added again.
 477          */
 478         if (IS_ERR(temp))
 479                 GOTO(out_insert, rc = PTR_ERR(temp));
 480         if (temp) {
 481                 __u32 del_map[2];
 482
 483                 del_map[0] = temp->id_client;
 484                 idmap_delete(id_type, temp, nodemap);
 485                 rc = nodemap_idx_idmap_del(nodemap, id_type, del_map);
 486                 /* In case there is any corrupted idmap */
 487                 if (!rc || unlikely(rc == -ENOENT)) {
 488                         temp = idmap_insert(id_type, idmap, nodemap);
 489                         if (IS_ERR(temp))
 490                                 rc = PTR_ERR(temp);
 491                         else if (!temp)
 492                                 rc = 0;
 493                         else
 494                                 rc = -EPERM;
 495                 }
 496         }
 497 out_insert:
 498         if (rc)
 499                 OBD_FREE_PTR(idmap);
 500         up_write(&nodemap->nm_idmap_lock);
 501         nm_member_revoke_locks(nodemap);
 502
 503 out:
 504         return rc;
 505 }
 506
 507 int nodemap_add_idmap(const char *name, enum nodemap_id_type id_type,
 508                       const __u32 map[2])
 509 {
 510         struct lu_nodemap       *nodemap = NULL;
 511         int                      rc;
 512
 513         ENTRY;
 514
 515         mutex_lock(&active_config_lock);
 516         nodemap = nodemap_lookup(name);
 517         if (IS_ERR(nodemap)) {
 518                 mutex_unlock(&active_config_lock);
 519                 GOTO(out, rc = PTR_ERR(nodemap));
 520         }
 521
 522         if (is_default_nodemap(nodemap)) {
 523                 rc = -EINVAL;
 524         } else {
 525                 rc = nodemap_add_idmap_helper(nodemap, id_type, map);
 526                 if (rc == 0)
 527                         rc = nodemap_idx_idmap_add(nodemap, id_type, map);
 528         }
 529         mutex_unlock(&active_config_lock);
 530         nodemap_putref(nodemap);
 531
 532 out:
 533         RETURN(rc);
 534 }
 535 EXPORT_SYMBOL(nodemap_add_idmap);
 536
 537 /**
 538  * delete idmap from proper nodemap tree
 539  *
 540  * \param       name            name of nodemap
 541  * \param       id_type         NODEMAP_UID or NODEMAP_GID
 542  * \param       map             array[2] __u32 containing the mapA values
 543  *                              map[0] is client id
 544  *                              map[1] is the filesystem id
 545  *
 546  * \retval      0 on success
 547  */
 548 int nodemap_del_idmap(const char *name, enum nodemap_id_type id_type,
 549                       const __u32 map[2])
 550 {
 551         struct lu_nodemap       *nodemap = NULL;
 552         struct lu_idmap         *idmap = NULL;
 553         int                     rc = 0;
 554
 555         ENTRY;
 556
 557         mutex_lock(&active_config_lock);
 558         nodemap = nodemap_lookup(name);
 559         if (IS_ERR(nodemap)) {
 560                 mutex_unlock(&active_config_lock);
 561                 GOTO(out, rc = PTR_ERR(nodemap));
 562         }
 563
 564         if (is_default_nodemap(nodemap))
 565                 GOTO(out_putref, rc = -EINVAL);
 566
 567         down_write(&nodemap->nm_idmap_lock);
 568         idmap = idmap_search(nodemap, NODEMAP_CLIENT_TO_FS, id_type,
 569                              map[0]);
 570         if (idmap == NULL) {
 571                 rc = -EINVAL;
 572         } else {
 573                 idmap_delete(id_type, idmap, nodemap);
 574                 rc = nodemap_idx_idmap_del(nodemap, id_type, map);
 575         }
 576         up_write(&nodemap->nm_idmap_lock);
 577
 578 out_putref:
 579         mutex_unlock(&active_config_lock);
 580         if (rc == 0)
 581                 nm_member_revoke_locks(nodemap);
 582         nodemap_putref(nodemap);
 583
 584 out:
 585         RETURN(rc);
 586 }
 587 EXPORT_SYMBOL(nodemap_del_idmap);
 588
 589 /**
 590  * Get nodemap assigned to given export. Takes a reference on the nodemap.
 591  *
 592  * Note that this function may return either NULL, or an ERR_PTR()
 593  * or a valid nodemap pointer.  All of the functions accessing the
 594  * returned nodemap can check IS_ERR(nodemap) to see if an error is
 595  * returned.  NULL is not considered an error, which is OK since this
 596  * is a valid case if nodemap are not in use.  All nodemap handling
 597  * functions must check for nodemap == NULL and do nothing, and the
 598  * nodemap returned from this function should not be dereferenced.
 599  *
 600  * \param       export          export to get nodemap for
 601  *
 602  * \retval      pointer to nodemap on success
 603  * \retval      NULL    nodemap subsystem disabled
 604  * \retval      -EACCES export does not have nodemap assigned
 605  */
 606 struct lu_nodemap *nodemap_get_from_exp(struct obd_export *exp)
 607 {
 608         struct lu_nodemap *nodemap;
 609
 610         ENTRY;
 611
 612         if (!nodemap_active)
 613                 RETURN(NULL);
 614
 615         spin_lock(&exp->exp_target_data.ted_nodemap_lock);
 616         nodemap = exp->exp_target_data.ted_nodemap;
 617         if (nodemap)
 618                 nodemap_getref(nodemap);
 619         spin_unlock(&exp->exp_target_data.ted_nodemap_lock);
 620
 621         if (!nodemap) {
 622                 CDEBUG(D_INFO, "%s: nodemap null on export %s (at %s)\n",
 623                        exp->exp_obd->obd_name,
 624                        obd_uuid2str(&exp->exp_client_uuid),
 625                        obd_export_nid2str(exp));
 626                 RETURN(ERR_PTR(-EACCES));
 627         }
 628
 629         RETURN(nodemap);
 630 }
 631 EXPORT_SYMBOL(nodemap_get_from_exp);
 632
 633 /**
 634  * mapping function for nodemap idmaps
 635  *
 636  * \param       nodemap         lu_nodemap structure defining nodemap
 637  * \param       node_type       NODEMAP_UID or NODEMAP_GID
 638  * \param       tree_type       NODEMAP_CLIENT_TO_FS or
 639  *                              NODEMAP_FS_TO_CLIENT
 640  * \param       id              id to map
 641  *
 642  * \retval      mapped id according to the rules below.
 643  *
 644  * if the nodemap_active is false, just return the passed id without mapping
 645  *
 646  * if the id to be looked up is 0, check that root access is allowed and if it
 647  * is, return 0. Otherwise, return the squash uid or gid.
 648  *
 649  * if the nodemap is configured to trusted the ids from the client system, just
 650  * return the passed id without mapping.
 651  *
 652  * if by this point, we haven't returned and the nodemap in question is the
 653  * default nodemap, return the squash uid or gid.
 654  *
 655  * after these checks, search the proper tree for the mapping, and if found
 656  * return the mapped value, otherwise return the squash uid or gid.
 657  */
 658 __u32 nodemap_map_id(struct lu_nodemap *nodemap,
 659                      enum nodemap_id_type id_type,
 660                      enum nodemap_tree_type tree_type, __u32 id)
 661 {
 662         struct lu_idmap         *idmap = NULL;
 663         __u32                    found_id;
 664
 665         ENTRY;
 666
 667         if (!nodemap_active)
 668                 goto out;
 669
 670         if (unlikely(nodemap == NULL))
 671                 goto out;
 672
 673         if (nodemap->nmf_map_uid_only && id_type == NODEMAP_GID)
 674                 goto out;
 675
 676         if (nodemap->nmf_map_gid_only && id_type == NODEMAP_UID)
 677                 goto out;
 678
 679         if (id == 0) {
 680                 if (nodemap->nmf_allow_root_access)
 681                         goto out;
 682                 else
 683                         goto squash;
 684         }
 685
 686         if (nodemap->nmf_trust_client_ids)
 687                 goto out;
 688
 689         if (is_default_nodemap(nodemap))
 690                 goto squash;
 691
 692         down_read(&nodemap->nm_idmap_lock);
 693         idmap = idmap_search(nodemap, tree_type, id_type, id);
 694         if (idmap == NULL) {
 695                 up_read(&nodemap->nm_idmap_lock);
 696                 goto squash;
 697         }
 698
 699         if (tree_type == NODEMAP_FS_TO_CLIENT)
 700                 found_id = idmap->id_client;
 701         else
 702                 found_id = idmap->id_fs;
 703         up_read(&nodemap->nm_idmap_lock);
 704         RETURN(found_id);
 705
 706 squash:
 707         if (id_type == NODEMAP_UID)
 708                 RETURN(nodemap->nm_squash_uid);
 709         else
 710                 RETURN(nodemap->nm_squash_gid);
 711 out:
 712         RETURN(id);
 713 }
 714 EXPORT_SYMBOL(nodemap_map_id);
 715
 716 /**
 717  * Map posix ACL entries according to the nodemap membership. Removes any
 718  * squashed ACLs.
 719  *
 720  * \param       lu_nodemap      nodemap
 721  * \param       buf             buffer containing xattr encoded ACLs
 722  * \param       size            size of ACLs in bytes
 723  * \param       tree_type       direction of mapping
 724  * \retval      size            new size of ACLs in bytes
 725  * \retval      -EINVAL         bad \a size param, see posix_acl_xattr_count()
 726  */
 727 ssize_t nodemap_map_acl(struct lu_nodemap *nodemap, void *buf, size_t size,
 728                         enum nodemap_tree_type tree_type)
 729 {
 730         posix_acl_xattr_header  *header = buf;
 731         posix_acl_xattr_entry   *entry = GET_POSIX_ACL_XATTR_ENTRY(header);
 732         posix_acl_xattr_entry   *new_entry = entry;
 733         posix_acl_xattr_entry   *end;
 734         int                      count;
 735
 736         ENTRY;
 737
 738         if (!nodemap_active)
 739                 RETURN(size);
 740
 741         if (unlikely(nodemap == NULL))
 742                 RETURN(size);
 743
 744         count = posix_acl_xattr_count(size);
 745         if (count < 0)
 746                 RETURN(-EINVAL);
 747         if (count == 0)
 748                 RETURN(0);
 749
 750         for (end = entry + count; entry != end; entry++) {
 751                 __u16 tag = le16_to_cpu(entry->e_tag);
 752                 __u32 id = le32_to_cpu(entry->e_id);
 753
 754                 switch (tag) {
 755                 case ACL_USER:
 756                         id = nodemap_map_id(nodemap, NODEMAP_UID,
 757                                             tree_type, id);
 758                         if (id == nodemap->nm_squash_uid)
 759                                 continue;
 760                         entry->e_id = cpu_to_le32(id);
 761                         break;
 762                 case ACL_GROUP:
 763                         id = nodemap_map_id(nodemap, NODEMAP_GID,
 764                                             tree_type, id);
 765                         if (id == nodemap->nm_squash_gid)
 766                                 continue;
 767                         entry->e_id = cpu_to_le32(id);
 768                         break;
 769                 }
 770
 771                 /* if we skip an ACL, copy the following ones over it */
 772                 if (new_entry != entry)
 773                         *new_entry = *entry;
 774
 775                 new_entry++;
 776         }
 777
 778         RETURN((void *)new_entry - (void *)header);
 779 }
 780 EXPORT_SYMBOL(nodemap_map_acl);
 781
 782 /*
 783  * Add nid range to given nodemap
 784  *
 785  * \param       config          nodemap config to work on
 786  * \param       nodemap         nodemap to add range to
 787  * \param       nid             nid range to add
 788  * \param       range_id        should be 0 unless loading from disk
 789  * \retval      0               success
 790  * \retval      -ENOMEM
 791  *
 792  */
 793 int nodemap_add_range_helper(struct nodemap_config *config,
 794                              struct lu_nodemap *nodemap,
 795                              const lnet_nid_t nid[2],
 796                              unsigned int range_id)
 797 {
 798         struct lu_nid_range     *range;
 799         int rc;
 800
 801         down_write(&config->nmc_range_tree_lock);
 802         range = range_create(&config->nmc_range_tree, nid[0], nid[1],
 803                              nodemap, range_id);
 804         if (range == NULL) {
 805                 up_write(&config->nmc_range_tree_lock);
 806                 GOTO(out, rc = -ENOMEM);
 807         }
 808
 809         rc = range_insert(&config->nmc_range_tree, range);
 810         if (rc != 0) {
 811                 CERROR("cannot insert nodemap range into '%s': rc = %d\n",
 812                       nodemap->nm_name, rc);
 813                 up_write(&config->nmc_range_tree_lock);
 814                 list_del(&range->rn_list);
 815                 range_destroy(range);
 816                 GOTO(out, rc = -ENOMEM);
 817         }
 818
 819         list_add(&range->rn_list, &nodemap->nm_ranges);
 820
 821         /* nodemaps have no members if they aren't on the active config */
 822         if (config == active_config)
 823                 nm_member_reclassify_nodemap(config->nmc_default_nodemap);
 824
 825         up_write(&config->nmc_range_tree_lock);
 826
 827         /* if range_id is non-zero, we are loading from disk */
 828         if (range_id == 0)
 829                 rc = nodemap_idx_range_add(range, nid);
 830
 831         if (config == active_config) {
 832                 nm_member_revoke_locks(config->nmc_default_nodemap);
 833                 nm_member_revoke_locks(nodemap);
 834         }
 835
 836 out:
 837         return rc;
 838 }
 839 int nodemap_add_range(const char *name, const lnet_nid_t nid[2])
 840 {
 841         struct lu_nodemap       *nodemap = NULL;
 842         int                      rc;
 843
 844         mutex_lock(&active_config_lock);
 845         nodemap = nodemap_lookup(name);
 846         if (IS_ERR(nodemap)) {
 847                 mutex_unlock(&active_config_lock);
 848                 GOTO(out, rc = PTR_ERR(nodemap));
 849         }
 850
 851         if (is_default_nodemap(nodemap))
 852                 rc = -EINVAL;
 853         else
 854                 rc = nodemap_add_range_helper(active_config, nodemap, nid, 0);
 855         mutex_unlock(&active_config_lock);
 856         nodemap_putref(nodemap);
 857 out:
 858         return rc;
 859 }
 860 EXPORT_SYMBOL(nodemap_add_range);
 861
 862 /**
 863  * delete a range
 864  * \param       name            nodemap name
 865  * \param       nid             nid range
 866  * \retval      0 on success
 867  *
 868  * Delete range from global range tree, and remove it
 869  * from the list in the associated nodemap.
 870  */
 871 int nodemap_del_range(const char *name, const lnet_nid_t nid[2])
 872 {
 873         struct lu_nodemap       *nodemap;
 874         struct lu_nid_range     *range;
 875         int                     rc = 0;
 876
 877         mutex_lock(&active_config_lock);
 878         nodemap = nodemap_lookup(name);
 879         if (IS_ERR(nodemap)) {
 880                 mutex_unlock(&active_config_lock);
 881                 GOTO(out, rc = PTR_ERR(nodemap));
 882         }
 883
 884         if (is_default_nodemap(nodemap))
 885                 GOTO(out_putref, rc = -EINVAL);
 886
 887         down_write(&active_config->nmc_range_tree_lock);
 888         range = range_find(&active_config->nmc_range_tree, nid[0], nid[1]);
 889         if (range == NULL) {
 890                 up_write(&active_config->nmc_range_tree_lock);
 891                 GOTO(out_putref, rc = -EINVAL);
 892         }
 893         if (range->rn_nodemap != nodemap) {
 894                 up_write(&active_config->nmc_range_tree_lock);
 895                 GOTO(out_putref, rc = -EINVAL);
 896         }
 897         rc = nodemap_idx_range_del(range);
 898         range_delete(&active_config->nmc_range_tree, range);
 899         nm_member_reclassify_nodemap(nodemap);
 900         up_write(&active_config->nmc_range_tree_lock);
 901
 902         nm_member_revoke_locks(active_config->nmc_default_nodemap);
 903         nm_member_revoke_locks(nodemap);
 904
 905 out_putref:
 906         mutex_unlock(&active_config_lock);
 907         nodemap_putref(nodemap);
 908 out:
 909         return rc;
 910 }
 911 EXPORT_SYMBOL(nodemap_del_range);
 912
 913 /**
 914  * set fileset on nodemap
 915  * \param       name            nodemap to set fileset on
 916  * \param       fileset         string containing fileset
 917  * \retval      0 on success
 918  *
 919  * set a fileset on the named nodemap
 920  */
 921 static int nodemap_set_fileset_helper(struct nodemap_config *config,
 922                                       struct lu_nodemap *nodemap,
 923                                       const char *fileset)
 924 {
 925         int rc = 0;
 926
 927         /* Allow 'fileset=clear' in addition to 'fileset=""' to clear fileset
 928          * because either command 'lctl set_param -P *.*.fileset=""' or
 929          * 'lctl nodemap_set_fileset --fileset ""' can only work correctly
 930          * on MGS, while on other servers, both commands will invoke upcall
 931          * "/usr/sbin/lctl set_param nodemap.default.fileset=" by function
 932          * process_param2_config(), which will cause "no value" error and
 933          * won't clear fileset.
 934          * 'fileset=""' is still kept for compatibility reason.
 935          */
 936         if (fileset == NULL)
 937                 rc = -EINVAL;
 938         else if (fileset[0] == '\0' || strcmp(fileset, "clear") == 0)
 939                 nodemap->nm_fileset[0] = '\0';
 940         else if (fileset[0] != '/')
 941                 rc = -EINVAL;
 942         else if (strlcpy(nodemap->nm_fileset, fileset,
 943                          sizeof(nodemap->nm_fileset)) >=
 944                  sizeof(nodemap->nm_fileset))
 945                 rc = -ENAMETOOLONG;
 946
 947         return rc;
 948 }
 949
 950 int nodemap_set_fileset(const char *name, const char *fileset)
 951 {
 952         struct lu_nodemap       *nodemap = NULL;
 953         int                      rc = 0;
 954
 955         mutex_lock(&active_config_lock);
 956         nodemap = nodemap_lookup(name);
 957         if (IS_ERR(nodemap)) {
 958                 mutex_unlock(&active_config_lock);
 959                 GOTO(out, rc = PTR_ERR(nodemap));
 960         }
 961
 962         rc = nodemap_set_fileset_helper(active_config, nodemap, fileset);
 963         mutex_unlock(&active_config_lock);
 964
 965         nodemap_putref(nodemap);
 966 out:
 967         return rc;
 968 }
 969 EXPORT_SYMBOL(nodemap_set_fileset);
 970
 971 /**
 972  * get fileset defined on nodemap
 973  * \param       nodemap         nodemap to get fileset from
 974  * \retval      fileset name, or NULL if not defined or not activated
 975  *
 976  * get the fileset defined on the nodemap
 977  */
 978 char *nodemap_get_fileset(const struct lu_nodemap *nodemap)
 979 {
 980         if (!nodemap_active)
 981                 return NULL;
 982
 983         return (char *)nodemap->nm_fileset;
 984 }
 985 EXPORT_SYMBOL(nodemap_get_fileset);
 986
 987 static int nodemap_validate_sepol(const char *sepol)
 988 {
 989         char buf[LUSTRE_NODEMAP_SEPOL_LENGTH + 1];
 990         char *p = (char *)sepol;
 991         char *q = buf;
 992         char polname[NAME_MAX + 1] = "";
 993         char hash[SELINUX_POLICY_HASH_LEN + 1] = "";
 994         unsigned char mode;
 995         unsigned short ver;
 996
 997         BUILD_BUG_ON(sizeof(buf) != sizeof(((struct lu_nodemap *)0)->nm_sepol));
 998
 999         if (sepol == NULL)
1000                 return -EINVAL;
1001
1002         /* we allow sepol = "" which means clear SELinux policy info */
1003         if (sepol[0] == '\0')
1004                 return 0;
1005
1006         /* make a copy of sepol, by replacing ':' with space
1007          * so that we can use sscanf over the string
1008          */
1009         while (p-sepol < sizeof(buf)) {
1010                 if (*p == ':')
1011                         *q = ' ';
1012                 else
1013                         *q = *p;
1014                 if (*p == '\0')
1015                         break;
1016                 p++;
1017                 q++;
1018         }
1019         if (p-sepol == sizeof(buf))
1020                 return -ENAMETOOLONG;
1021
1022         if (sscanf(buf, "%1hhu %s %hu %s", &mode, polname, &ver, hash) != 4)
1023                 return -EINVAL;
1024
1025         if (mode != 0 && mode != 1)
1026                 return -EINVAL;
1027
1028         return 0;
1029 }
1030
1031 /**
1032  * set SELinux policy on nodemap
1033  * \param       name            nodemap to set SELinux policy info on
1034  * \param       sepol           string containing SELinux policy info
1035  * \retval      0 on success
1036  *
1037  * set SELinux policy info on the named nodemap
1038  */
1039 int nodemap_set_sepol(const char *name, const char *sepol)
1040 {
1041         struct lu_nodemap       *nodemap = NULL;
1042         int                      rc;
1043
1044         rc = nodemap_validate_sepol(sepol);
1045         if (rc < 0)
1046                 GOTO(out, rc);
1047
1048         mutex_lock(&active_config_lock);
1049         nodemap = nodemap_lookup(name);
1050         if (IS_ERR(nodemap)) {
1051                 mutex_unlock(&active_config_lock);
1052                 GOTO(out, rc = PTR_ERR(nodemap));
1053         }
1054
1055         if (is_default_nodemap(nodemap)) {
1056                 /* We do not want nodes in the default nodemap to have
1057                  * SELinux restrictions. Sec admin should create dedicated
1058                  * nodemap entries for this.
1059                  */
1060                 GOTO(out_putref, rc = -EINVAL);
1061         }
1062
1063         /* truncation cannot happen, as string length was checked in
1064          * nodemap_validate_sepol()
1065          */
1066         strlcpy(nodemap->nm_sepol, sepol, sizeof(nodemap->nm_sepol));
1067
1068 out_putref:
1069         mutex_unlock(&active_config_lock);
1070         nodemap_putref(nodemap);
1071 out:
1072         return rc;
1073 }
1074 EXPORT_SYMBOL(nodemap_set_sepol);
1075
1076 /**
1077  * get SELinux policy info defined on nodemap
1078  * \param       nodemap         nodemap to get SELinux policy info from
1079  * \retval      SELinux policy info, or NULL if not defined or not activated
1080  *
1081  * get the SELinux policy info defined on the nodemap
1082  */
1083 const char *nodemap_get_sepol(const struct lu_nodemap *nodemap)
1084 {
1085         if (is_default_nodemap(nodemap))
1086                 return NULL;
1087         else
1088                 return (char *)nodemap->nm_sepol;
1089 }
1090 EXPORT_SYMBOL(nodemap_get_sepol);
1091
1092 /**
1093  * Nodemap constructor
1094  *
1095  * Creates an lu_nodemap structure and assigns sane default
1096  * member values. If this is the default nodemap, the defaults
1097  * are the most restrictive in terms of mapping behavior. Otherwise
1098  * the default flags should be inherited from the default nodemap.
1099  * The adds nodemap to nodemap_hash.
1100  *
1101  * Requires that the caller take the active_config_lock
1102  *
1103  * \param       name            name of nodemap
1104  * \param       is_default      true if default nodemap
1105  * \retval      nodemap         success
1106  * \retval      -EINVAL         invalid nodemap name
1107  * \retval      -EEXIST         nodemap already exists
1108  * \retval      -ENOMEM         cannot allocate memory for nodemap
1109  */
1110 struct lu_nodemap *nodemap_create(const char *name,
1111                                   struct nodemap_config *config,
1112                                   bool is_default)
1113 {
1114         struct lu_nodemap       *nodemap = NULL;
1115         struct lu_nodemap       *default_nodemap;
1116         struct cfs_hash         *hash = config->nmc_nodemap_hash;
1117         int                      rc = 0;
1118         ENTRY;
1119
1120         default_nodemap = config->nmc_default_nodemap;
1121
1122         if (!nodemap_name_is_valid(name))
1123                 GOTO(out, rc = -EINVAL);
1124
1125         if (hash == NULL) {
1126                 CERROR("Config nodemap hash is NULL, unable to add %s\n", name);
1127                 GOTO(out, rc = -EINVAL);
1128         }
1129
1130         OBD_ALLOC_PTR(nodemap);
1131         if (nodemap == NULL) {
1132                 CERROR("cannot allocate memory (%zu bytes) for nodemap '%s'\n",
1133                        sizeof(*nodemap), name);
1134                 GOTO(out, rc = -ENOMEM);
1135         }
1136
1137         /*
1138          * take an extra reference to prevent nodemap from being destroyed
1139          * while it's being created.
1140          */
1141         atomic_set(&nodemap->nm_refcount, 2);
1142         snprintf(nodemap->nm_name, sizeof(nodemap->nm_name), "%s", name);
1143         rc = cfs_hash_add_unique(hash, name, &nodemap->nm_hash);
1144         if (rc != 0) {
1145                 OBD_FREE_PTR(nodemap);
1146                 GOTO(out, rc = -EEXIST);
1147         }
1148
1149         INIT_LIST_HEAD(&nodemap->nm_ranges);
1150         INIT_LIST_HEAD(&nodemap->nm_list);
1151         INIT_LIST_HEAD(&nodemap->nm_member_list);
1152
1153         mutex_init(&nodemap->nm_member_list_lock);
1154         init_rwsem(&nodemap->nm_idmap_lock);
1155         nodemap->nm_fs_to_client_uidmap = RB_ROOT;
1156         nodemap->nm_client_to_fs_uidmap = RB_ROOT;
1157         nodemap->nm_fs_to_client_gidmap = RB_ROOT;
1158         nodemap->nm_client_to_fs_gidmap = RB_ROOT;
1159
1160         if (is_default) {
1161                 nodemap->nm_id = LUSTRE_NODEMAP_DEFAULT_ID;
1162                 config->nmc_default_nodemap = nodemap;
1163         } else {
1164                 config->nmc_nodemap_highest_id++;
1165                 nodemap->nm_id = config->nmc_nodemap_highest_id;
1166         }
1167
1168         if (is_default || default_nodemap == NULL) {
1169                 nodemap->nmf_trust_client_ids = 0;
1170                 nodemap->nmf_allow_root_access = 0;
1171                 nodemap->nmf_deny_unknown = 0;
1172                 nodemap->nmf_map_uid_only = 0;
1173                 nodemap->nmf_map_gid_only = 0;
1174                 nodemap->nmf_enable_audit = 1;
1175                 nodemap->nmf_forbid_encryption = 0;
1176
1177                 nodemap->nm_squash_uid = NODEMAP_NOBODY_UID;
1178                 nodemap->nm_squash_gid = NODEMAP_NOBODY_GID;
1179                 nodemap->nm_fileset[0] = '\0';
1180                 nodemap->nm_sepol[0] = '\0';
1181                 if (!is_default)
1182                         CWARN("adding nodemap '%s' to config without"
1183                               " default nodemap\n", nodemap->nm_name);
1184         } else {
1185                 nodemap->nmf_trust_client_ids =
1186                                 default_nodemap->nmf_trust_client_ids;
1187                 nodemap->nmf_allow_root_access =
1188                                 default_nodemap->nmf_allow_root_access;
1189                 nodemap->nmf_deny_unknown =
1190                                 default_nodemap->nmf_deny_unknown;
1191                 nodemap->nmf_map_uid_only =
1192                                 default_nodemap->nmf_map_uid_only;
1193                 nodemap->nmf_map_gid_only =
1194                                 default_nodemap->nmf_map_gid_only;
1195                 nodemap->nmf_enable_audit =
1196                         default_nodemap->nmf_enable_audit;
1197                 nodemap->nmf_forbid_encryption =
1198                         default_nodemap->nmf_forbid_encryption;
1199
1200                 nodemap->nm_squash_uid = default_nodemap->nm_squash_uid;
1201                 nodemap->nm_squash_gid = default_nodemap->nm_squash_gid;
1202                 nodemap->nm_fileset[0] = '\0';
1203                 nodemap->nm_sepol[0] = '\0';
1204         }
1205
1206         RETURN(nodemap);
1207
1208 out:
1209         CERROR("cannot add nodemap: '%s': rc = %d\n", name, rc);
1210         RETURN(ERR_PTR(rc));
1211 }
1212
1213 /**
1214  * Set the nmf_deny_unknown flag to true or false.
1215  * \param       name            nodemap name
1216  * \param       deny_unknown    if true, squashed users will get EACCES
1217  * \retval      0 on success
1218  *
1219  */
1220 int nodemap_set_deny_unknown(const char *name, bool deny_unknown)
1221 {
1222         struct lu_nodemap       *nodemap = NULL;
1223         int                     rc = 0;
1224
1225         mutex_lock(&active_config_lock);
1226         nodemap = nodemap_lookup(name);
1227         mutex_unlock(&active_config_lock);
1228         if (IS_ERR(nodemap))
1229                 GOTO(out, rc = PTR_ERR(nodemap));
1230
1231         nodemap->nmf_deny_unknown = deny_unknown;
1232         rc = nodemap_idx_nodemap_update(nodemap);
1233
1234         nm_member_revoke_locks(nodemap);
1235         nodemap_putref(nodemap);
1236 out:
1237         return rc;
1238 }
1239 EXPORT_SYMBOL(nodemap_set_deny_unknown);
1240
1241 /**
1242  * Set the nmf_allow_root_access flag to true or false.
1243  * \param       name            nodemap name
1244  * \param       allow_root      if true, nodemap will not squash the root user
1245  * \retval      0 on success
1246  *
1247  */
1248 int nodemap_set_allow_root(const char *name, bool allow_root)
1249 {
1250         struct lu_nodemap       *nodemap = NULL;
1251         int                     rc = 0;
1252
1253         mutex_lock(&active_config_lock);
1254         nodemap = nodemap_lookup(name);
1255         mutex_unlock(&active_config_lock);
1256         if (IS_ERR(nodemap))
1257                 GOTO(out, rc = PTR_ERR(nodemap));
1258
1259         nodemap->nmf_allow_root_access = allow_root;
1260         rc = nodemap_idx_nodemap_update(nodemap);
1261
1262         nm_member_revoke_locks(nodemap);
1263         nodemap_putref(nodemap);
1264 out:
1265         return rc;
1266 }
1267 EXPORT_SYMBOL(nodemap_set_allow_root);
1268
1269 /**
1270  * Set the nmf_trust_client_ids flag to true or false.
1271  *
1272  * \param       name                    nodemap name
1273  * \param       trust_client_ids        if true, nodemap will not map its IDs
1274  * \retval      0 on success
1275  *
1276  */
1277 int nodemap_set_trust_client_ids(const char *name, bool trust_client_ids)
1278 {
1279         struct lu_nodemap       *nodemap = NULL;
1280         int                     rc = 0;
1281
1282         mutex_lock(&active_config_lock);
1283         nodemap = nodemap_lookup(name);
1284         mutex_unlock(&active_config_lock);
1285         if (IS_ERR(nodemap))
1286                 GOTO(out, rc = PTR_ERR(nodemap));
1287
1288         nodemap->nmf_trust_client_ids = trust_client_ids;
1289         rc = nodemap_idx_nodemap_update(nodemap);
1290
1291         nm_member_revoke_locks(nodemap);
1292         nodemap_putref(nodemap);
1293 out:
1294         return rc;
1295 }
1296 EXPORT_SYMBOL(nodemap_set_trust_client_ids);
1297
1298 int nodemap_set_mapping_mode(const char *name, enum nodemap_mapping_modes mode)
1299 {
1300         struct lu_nodemap       *nodemap = NULL;
1301         int                     rc = 0;
1302
1303         mutex_lock(&active_config_lock);
1304         nodemap = nodemap_lookup(name);
1305         mutex_unlock(&active_config_lock);
1306         if (IS_ERR(nodemap))
1307                 GOTO(out, rc = PTR_ERR(nodemap));
1308
1309         switch (mode) {
1310         case NODEMAP_MAP_BOTH:
1311                 nodemap->nmf_map_uid_only = 0;
1312                 nodemap->nmf_map_gid_only = 0;
1313                 break;
1314         case NODEMAP_MAP_UID_ONLY:
1315                 nodemap->nmf_map_uid_only = 1;
1316                 nodemap->nmf_map_gid_only = 0;
1317                 break;
1318         case NODEMAP_MAP_GID_ONLY:
1319                 nodemap->nmf_map_uid_only = 0;
1320                 nodemap->nmf_map_gid_only = 1;
1321                 break;
1322         default:
1323                 CWARN("cannot set unknown mapping mode, mode = %d\n", mode);
1324         }
1325         rc = nodemap_idx_nodemap_update(nodemap);
1326
1327         nm_member_revoke_locks(nodemap);
1328         nodemap_putref(nodemap);
1329 out:
1330         return rc;
1331 }
1332 EXPORT_SYMBOL(nodemap_set_mapping_mode);
1333
1334 /**
1335  * Update the squash_uid for a nodemap.
1336  *
1337  * \param       name            nodemap name
1338  * \param       uid             the new uid to squash unknown users to
1339  * \retval      0 on success
1340  *
1341  * Update the squash_uid for a nodemap. The squash_uid is the uid
1342  * that the all client uids are mapped to if nodemap is active,
1343  * the trust_client_ids flag is not set, and the uid is not in
1344  * the idmap tree.
1345  */
1346 int nodemap_set_squash_uid(const char *name, uid_t uid)
1347 {
1348         struct lu_nodemap       *nodemap = NULL;
1349         int                     rc = 0;
1350
1351         mutex_lock(&active_config_lock);
1352         nodemap = nodemap_lookup(name);
1353         mutex_unlock(&active_config_lock);
1354         if (IS_ERR(nodemap))
1355                 GOTO(out, rc = PTR_ERR(nodemap));
1356
1357         nodemap->nm_squash_uid = uid;
1358         rc = nodemap_idx_nodemap_update(nodemap);
1359
1360         nm_member_revoke_locks(nodemap);
1361         nodemap_putref(nodemap);
1362 out:
1363         return rc;
1364 }
1365 EXPORT_SYMBOL(nodemap_set_squash_uid);
1366
1367 /**
1368  * Update the squash_gid for a nodemap.
1369  *
1370  * \param       name            nodemap name
1371  * \param       gid             the new gid to squash unknown gids to
1372  * \retval      0 on success
1373  *
1374  * Update the squash_gid for a nodemap. The squash_uid is the gid
1375  * that the all client gids are mapped to if nodemap is active,
1376  * the trust_client_ids flag is not set, and the gid is not in
1377  * the idmap tree.
1378  */
1379 int nodemap_set_squash_gid(const char *name, gid_t gid)
1380 {
1381         struct lu_nodemap       *nodemap = NULL;
1382         int                     rc = 0;
1383
1384         mutex_lock(&active_config_lock);
1385         nodemap = nodemap_lookup(name);
1386         mutex_unlock(&active_config_lock);
1387         if (IS_ERR(nodemap))
1388                 GOTO(out, rc = PTR_ERR(nodemap));
1389
1390         nodemap->nm_squash_gid = gid;
1391         rc = nodemap_idx_nodemap_update(nodemap);
1392
1393         nm_member_revoke_locks(nodemap);
1394         nodemap_putref(nodemap);
1395 out:
1396         return rc;
1397 }
1398 EXPORT_SYMBOL(nodemap_set_squash_gid);
1399
1400 /**
1401  * Returns true if this nodemap has root user access. Always returns true if
1402  * nodemaps are not active.
1403  *
1404  * \param       nodemap         nodemap to check access for
1405  */
1406 bool nodemap_can_setquota(const struct lu_nodemap *nodemap)
1407 {
1408         return !nodemap_active || (nodemap && nodemap->nmf_allow_root_access);
1409 }
1410 EXPORT_SYMBOL(nodemap_can_setquota);
1411
1412 /**
1413  * Set the nmf_enable_audit flag to true or false.
1414  * \param       name            nodemap name
1415  * \param       audit_mode      if true, allow audit
1416  * \retval      0 on success
1417  *
1418  */
1419 int nodemap_set_audit_mode(const char *name, bool enable_audit)
1420 {
1421         struct lu_nodemap       *nodemap = NULL;
1422         int                     rc = 0;
1423
1424         mutex_lock(&active_config_lock);
1425         nodemap = nodemap_lookup(name);
1426         mutex_unlock(&active_config_lock);
1427         if (IS_ERR(nodemap))
1428                 GOTO(out, rc = PTR_ERR(nodemap));
1429
1430         nodemap->nmf_enable_audit = enable_audit;
1431         rc = nodemap_idx_nodemap_update(nodemap);
1432
1433         nm_member_revoke_locks(nodemap);
1434         nodemap_putref(nodemap);
1435 out:
1436         return rc;
1437 }
1438 EXPORT_SYMBOL(nodemap_set_audit_mode);
1439
1440 /**
1441  * Set the nmf_forbid_encryption flag to true or false.
1442  * \param       name                    nodemap name
1443  * \param       forbid_encryption       if true, forbid encryption
1444  * \retval      0 on success
1445  *
1446  */
1447 int nodemap_set_forbid_encryption(const char *name, bool forbid_encryption)
1448 {
1449         struct lu_nodemap       *nodemap = NULL;
1450         int                     rc = 0;
1451
1452         mutex_lock(&active_config_lock);
1453         nodemap = nodemap_lookup(name);
1454         mutex_unlock(&active_config_lock);
1455         if (IS_ERR(nodemap))
1456                 GOTO(out, rc = PTR_ERR(nodemap));
1457
1458         nodemap->nmf_forbid_encryption = forbid_encryption;
1459         rc = nodemap_idx_nodemap_update(nodemap);
1460
1461         nm_member_revoke_locks(nodemap);
1462         nodemap_putref(nodemap);
1463 out:
1464         return rc;
1465 }
1466 EXPORT_SYMBOL(nodemap_set_forbid_encryption);
1467
1468
1469 /**
1470  * Add a nodemap
1471  *
1472  * \param       name            name of nodemap
1473  * \retval      0               success
1474  * \retval      -EINVAL         invalid nodemap name
1475  * \retval      -EEXIST         nodemap already exists
1476  * \retval      -ENOMEM         cannot allocate memory for nodemap
1477  */
1478 int nodemap_add(const char *nodemap_name)
1479 {
1480         struct lu_nodemap *nodemap;
1481         int rc;
1482
1483         mutex_lock(&active_config_lock);
1484         nodemap = nodemap_create(nodemap_name, active_config, 0);
1485         if (IS_ERR(nodemap)) {
1486                 mutex_unlock(&active_config_lock);
1487                 return PTR_ERR(nodemap);
1488         }
1489
1490         rc = nodemap_idx_nodemap_add(nodemap);
1491         if (rc == 0)
1492                 rc = lprocfs_nodemap_register(nodemap, 0);
1493
1494         mutex_unlock(&active_config_lock);
1495         nodemap_putref(nodemap);
1496
1497         return rc;
1498 }
1499 EXPORT_SYMBOL(nodemap_add);
1500
1501 /**
1502  * Delete a nodemap
1503  *
1504  * \param       name            name of nodemmap
1505  * \retval      0               success
1506  * \retval      -EINVAL         invalid input
1507  * \retval      -ENOENT         no existing nodemap
1508  */
1509 int nodemap_del(const char *nodemap_name)
1510 {
1511         struct lu_nodemap       *nodemap;
1512         struct lu_nid_range     *range;
1513         struct lu_nid_range     *range_temp;
1514         int                      rc = 0;
1515         int                      rc2 = 0;
1516
1517         if (strcmp(nodemap_name, DEFAULT_NODEMAP) == 0)
1518                 RETURN(-EINVAL);
1519
1520         mutex_lock(&active_config_lock);
1521         nodemap = cfs_hash_del_key(active_config->nmc_nodemap_hash,
1522                                    nodemap_name);
1523         if (nodemap == NULL) {
1524                 mutex_unlock(&active_config_lock);
1525                 GOTO(out, rc = -ENOENT);
1526         }
1527
1528         /* erase nodemap from active ranges to prevent client assignment */
1529         down_write(&active_config->nmc_range_tree_lock);
1530         list_for_each_entry_safe(range, range_temp, &nodemap->nm_ranges,
1531                                  rn_list) {
1532                 rc2 = nodemap_idx_range_del(range);
1533                 if (rc2 < 0)
1534                         rc = rc2;
1535
1536                 range_delete(&active_config->nmc_range_tree, range);
1537         }
1538         up_write(&active_config->nmc_range_tree_lock);
1539
1540         rc2 = nodemap_idx_nodemap_del(nodemap);
1541         if (rc2 < 0)
1542                 rc = rc2;
1543
1544         /*
1545          * remove procfs here in case nodemap_create called with same name
1546          * before nodemap_destroy is run.
1547          */
1548         lprocfs_nodemap_remove(nodemap->nm_pde_data);
1549         nodemap->nm_pde_data = NULL;
1550
1551         /* reclassify all member exports from nodemap, so they put their refs */
1552         down_read(&active_config->nmc_range_tree_lock);
1553         nm_member_reclassify_nodemap(nodemap);
1554         up_read(&active_config->nmc_range_tree_lock);
1555
1556         if (!list_empty(&nodemap->nm_member_list))
1557                 CWARN("nodemap_del failed to reclassify all members\n");
1558
1559         mutex_unlock(&active_config_lock);
1560
1561         nodemap_putref(nodemap);
1562
1563 out:
1564         return rc;
1565 }
1566 EXPORT_SYMBOL(nodemap_del);
1567
1568 /**
1569  * activate nodemap functions
1570  *
1571  * \param       value           1 for on, 0 for off
1572  */
1573 void nodemap_activate(const bool value)
1574 {
1575         mutex_lock(&active_config_lock);
1576         active_config->nmc_nodemap_is_active = value;
1577
1578         /* copy active value to global to avoid locking in map functions */
1579         nodemap_active = value;
1580         nodemap_idx_nodemap_activate(value);
1581         mutex_unlock(&active_config_lock);
1582         nm_member_revoke_all();
1583 }
1584 EXPORT_SYMBOL(nodemap_activate);
1585
1586 /**
1587  * Helper iterator to convert nodemap hash to list.
1588  *
1589  * \param       hs                      hash structure
1590  * \param       bd                      bucket descriptor
1591  * \param       hnode                   hash node
1592  * \param       nodemap_list_head       list head for list of nodemaps in hash
1593  */
1594 static int nodemap_cleanup_iter_cb(struct cfs_hash *hs, struct cfs_hash_bd *bd,
1595                                    struct hlist_node *hnode,
1596                                    void *nodemap_list_head)
1597 {
1598         struct lu_nodemap *nodemap;
1599
1600         nodemap = hlist_entry(hnode, struct lu_nodemap, nm_hash);
1601         list_add(&nodemap->nm_list, nodemap_list_head);
1602
1603         cfs_hash_bd_del_locked(hs, bd, hnode);
1604
1605         return 0;
1606 }
1607
1608 struct nodemap_config *nodemap_config_alloc(void)
1609 {
1610         struct nodemap_config *config;
1611         int rc = 0;
1612
1613         OBD_ALLOC_PTR(config);
1614         if (config == NULL)
1615                 return ERR_PTR(-ENOMEM);
1616
1617         rc = nodemap_init_hash(config);
1618         if (rc != 0) {
1619                 OBD_FREE_PTR(config);
1620                 return ERR_PTR(rc);
1621         }
1622
1623         init_rwsem(&config->nmc_range_tree_lock);
1624
1625         return config;
1626 }
1627 EXPORT_SYMBOL(nodemap_config_alloc);
1628
1629 /**
1630  * Walk the nodemap_hash and remove all nodemaps.
1631  */
1632 void nodemap_config_dealloc(struct nodemap_config *config)
1633 {
1634         struct lu_nodemap       *nodemap = NULL;
1635         struct lu_nodemap       *nodemap_temp;
1636         struct lu_nid_range     *range;
1637         struct lu_nid_range     *range_temp;
1638         LIST_HEAD(nodemap_list_head);
1639
1640         cfs_hash_for_each_safe(config->nmc_nodemap_hash,
1641                                nodemap_cleanup_iter_cb, &nodemap_list_head);
1642         cfs_hash_putref(config->nmc_nodemap_hash);
1643
1644         /* Because nodemap_destroy might sleep, we can't destroy them
1645          * in cfs_hash_for_each, so we build a list there and destroy here
1646          */
1647         list_for_each_entry_safe(nodemap, nodemap_temp, &nodemap_list_head,
1648                                  nm_list) {
1649                 mutex_lock(&active_config_lock);
1650                 down_write(&config->nmc_range_tree_lock);
1651
1652                 /* move members to new config, requires ac lock */
1653                 nm_member_reclassify_nodemap(nodemap);
1654                 list_for_each_entry_safe(range, range_temp, &nodemap->nm_ranges,
1655                                          rn_list)
1656                         range_delete(&config->nmc_range_tree, range);
1657                 up_write(&config->nmc_range_tree_lock);
1658                 mutex_unlock(&active_config_lock);
1659
1660                 /* putref must be outside of ac lock if nm could be destroyed */
1661                 nodemap_putref(nodemap);
1662         }
1663         OBD_FREE_PTR(config);
1664 }
1665 EXPORT_SYMBOL(nodemap_config_dealloc);
1666
1667 /*
1668  * callback for cfs_hash_for_each_safe used to convert a nodemap hash to a
1669  * nodemap list, generally for locking purposes as a hash cb can't sleep.
1670  */
1671 int nm_hash_list_cb(struct cfs_hash *hs, struct cfs_hash_bd *bd,
1672                     struct hlist_node *hnode,
1673                     void *nodemap_list_head)
1674 {
1675         struct lu_nodemap *nodemap;
1676
1677         nodemap = hlist_entry(hnode, struct lu_nodemap, nm_hash);
1678         list_add(&nodemap->nm_list, nodemap_list_head);
1679         return 0;
1680 }
1681
1682 void nodemap_config_set_active(struct nodemap_config *config)
1683 {
1684         struct nodemap_config   *old_config = active_config;
1685         struct lu_nodemap       *nodemap;
1686         struct lu_nodemap       *tmp;
1687         bool revoke_locks;
1688         LIST_HEAD(nodemap_list_head);
1689
1690         ENTRY;
1691
1692         LASSERT(active_config != config);
1693         LASSERT(config->nmc_default_nodemap);
1694
1695         mutex_lock(&active_config_lock);
1696
1697         /* move proc entries from already existing nms, create for new nms */
1698         cfs_hash_for_each_safe(config->nmc_nodemap_hash,
1699                                nm_hash_list_cb, &nodemap_list_head);
1700         list_for_each_entry_safe(nodemap, tmp, &nodemap_list_head, nm_list) {
1701                 struct lu_nodemap *old_nm = NULL;
1702
1703                 if (active_config != NULL)
1704                         old_nm = cfs_hash_lookup(
1705                                         active_config->nmc_nodemap_hash,
1706                                         nodemap->nm_name);
1707                 if (old_nm != NULL) {
1708                         nodemap->nm_pde_data = old_nm->nm_pde_data;
1709                         old_nm->nm_pde_data = NULL;
1710                         nodemap_putref(old_nm);
1711                 } else {
1712                         bool is_def = (nodemap == config->nmc_default_nodemap);
1713
1714                         lprocfs_nodemap_register(nodemap, is_def);
1715                 }
1716         }
1717
1718         /*
1719          * We only need to revoke locks if old nodemap was active, and new
1720          * config is now nodemap inactive. nodemap_config_dealloc will
1721          * reclassify exports, triggering a lock revoke if and only if new
1722          * nodemap is active.
1723          */
1724         revoke_locks = !config->nmc_nodemap_is_active && nodemap_active;
1725
1726         /* if new config is inactive, deactivate live config before switching */
1727         if (!config->nmc_nodemap_is_active)
1728                 nodemap_active = false;
1729         active_config = config;
1730         if (config->nmc_nodemap_is_active)
1731                 nodemap_active = true;
1732
1733         mutex_unlock(&active_config_lock);
1734
1735         if (old_config != NULL)
1736                 nodemap_config_dealloc(old_config);
1737
1738         if (revoke_locks)
1739                 nm_member_revoke_all();
1740
1741         EXIT;
1742 }
1743
1744 /**
1745  * Cleanup nodemap module on exit
1746  */
1747 void nodemap_mod_exit(void)
1748 {
1749         nodemap_config_dealloc(active_config);
1750         nodemap_procfs_exit();
1751 }
1752
1753 /**
1754  * Initialize the nodemap module
1755  */
1756 int nodemap_mod_init(void)
1757 {
1758         struct nodemap_config   *new_config;
1759         struct lu_nodemap       *nodemap;
1760         int                      rc = 0;
1761
1762         rc = nodemap_procfs_init();
1763         if (rc != 0)
1764                 return rc;
1765
1766         new_config = nodemap_config_alloc();
1767         if (IS_ERR(new_config)) {
1768                 nodemap_procfs_exit();
1769                 GOTO(out, rc = PTR_ERR(new_config));
1770         }
1771
1772         nodemap = nodemap_create(DEFAULT_NODEMAP, new_config, 1);
1773         if (IS_ERR(nodemap)) {
1774                 nodemap_config_dealloc(new_config);
1775                 nodemap_procfs_exit();
1776                 GOTO(out, rc = PTR_ERR(nodemap));
1777         }
1778
1779         nodemap_config_set_active(new_config);
1780         nodemap_putref(nodemap);
1781
1782 out:
1783         return rc;
1784 }
1785
1786 /**
1787  * Revoke locks for all nodemaps.
1788  */
1789 void nm_member_revoke_all(void)
1790 {
1791         struct lu_nodemap *nodemap;
1792         struct lu_nodemap *tmp;
1793         LIST_HEAD(nodemap_list_head);
1794
1795         mutex_lock(&active_config_lock);
1796         cfs_hash_for_each_safe(active_config->nmc_nodemap_hash,
1797                                nm_hash_list_cb, &nodemap_list_head);
1798
1799         /* revoke_locks sleeps, so can't call in cfs hash cb */
1800         list_for_each_entry_safe(nodemap, tmp, &nodemap_list_head, nm_list)
1801                 nm_member_revoke_locks_always(nodemap);
1802         mutex_unlock(&active_config_lock);
1803 }
1804
1805 /**
1806  * Returns the nodemap classification for a given nid into an ioctl buffer.
1807  * Useful for testing the nodemap configuration to make sure it is working as
1808  * expected.
1809  *
1810  * \param       nid             nid to classify
1811  * \param[out]  name_buf        buffer to write the nodemap name to
1812  * \param       name_len        length of buffer
1813  */
1814 void nodemap_test_nid(lnet_nid_t nid, char *name_buf, size_t name_len)
1815 {
1816         struct lu_nodemap       *nodemap;
1817
1818         mutex_lock(&active_config_lock);
1819         down_read(&active_config->nmc_range_tree_lock);
1820         nodemap = nodemap_classify_nid(nid);
1821         up_read(&active_config->nmc_range_tree_lock);
1822         mutex_unlock(&active_config_lock);
1823
1824         if (IS_ERR(nodemap))
1825                 return;
1826
1827         strncpy(name_buf, nodemap->nm_name, name_len);
1828         if (name_len > 0)
1829                 name_buf[name_len - 1] = '\0';
1830
1831         nodemap_putref(nodemap);
1832 }
1833 EXPORT_SYMBOL(nodemap_test_nid);
1834
1835 /**
1836  * Passes back the id mapping for a given nid/id pair. Useful for testing the
1837  * nodemap configuration to make sure it is working as expected.
1838  *
1839  * \param       nid             nid to classify
1840  * \param       idtype          uid or gid
1841  * \param       client_id       id to map to fs
1842  * \param       fs_id_buf       pointer to save mapped fs_id to
1843  *
1844  * \retval      0       success
1845  * \retval      -EINVAL invalid NID
1846  */
1847 int nodemap_test_id(lnet_nid_t nid, enum nodemap_id_type idtype,
1848                     __u32 client_id, __u32 *fs_id)
1849 {
1850         struct lu_nodemap       *nodemap;
1851
1852         mutex_lock(&active_config_lock);
1853         down_read(&active_config->nmc_range_tree_lock);
1854         nodemap = nodemap_classify_nid(nid);
1855         up_read(&active_config->nmc_range_tree_lock);
1856         mutex_unlock(&active_config_lock);
1857
1858         if (IS_ERR(nodemap))
1859                 return PTR_ERR(nodemap);
1860
1861         *fs_id = nodemap_map_id(nodemap, idtype, NODEMAP_CLIENT_TO_FS,
1862                                client_id);
1863         nodemap_putref(nodemap);
1864
1865         return 0;
1866 }
1867 EXPORT_SYMBOL(nodemap_test_id);