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) 2014, Intel Corporation.
  26  *
  27  * Author: Joshua Walgenbach <jjw@iu.edu>
  28  */
  29 #include <linux/module.h>
  30 #include <linux/sort.h>
  31 #include <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         write_lock(&nodemap->nm_idmap_lock);
  75         idmap_delete_tree(nodemap);
  76         write_unlock(&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 (LNET_NETTYP(LNET_NIDNET(nid)) == LOLND) {
 268                 lnet_process_id_t id;
 269                 int i = 0;
 270
 271                 do {
 272                         rc = LNetGetId(i++, &id);
 273                         if (rc < 0)
 274                                 RETURN(ERR_PTR(-EINVAL));
 275                 } while (LNET_NETTYP(LNET_NIDNET(id.nid)) == LOLND);
 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       name            name of nodemap
 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  */
 459 int nodemap_add_idmap_helper(struct lu_nodemap *nodemap,
 460                              enum nodemap_id_type id_type,
 461                              const __u32 map[2])
 462 {
 463         struct lu_idmap         *idmap;
 464         int                     rc = 0;
 465
 466         idmap = idmap_create(map[0], map[1]);
 467         if (idmap == NULL)
 468                 GOTO(out, rc = -ENOMEM);
 469
 470         write_lock(&nodemap->nm_idmap_lock);
 471         idmap_insert(id_type, idmap, nodemap);
 472         write_unlock(&nodemap->nm_idmap_lock);
 473         nm_member_revoke_locks(nodemap);
 474
 475 out:
 476         return rc;
 477 }
 478
 479 int nodemap_add_idmap(const char *name, enum nodemap_id_type id_type,
 480                       const __u32 map[2])
 481 {
 482         struct lu_nodemap       *nodemap = NULL;
 483         int                      rc;
 484
 485         mutex_lock(&active_config_lock);
 486         nodemap = nodemap_lookup(name);
 487         if (IS_ERR(nodemap)) {
 488                 mutex_unlock(&active_config_lock);
 489                 GOTO(out, rc = PTR_ERR(nodemap));
 490         }
 491
 492         if (is_default_nodemap(nodemap)) {
 493                 rc = -EINVAL;
 494         } else {
 495                 rc = nodemap_add_idmap_helper(nodemap, id_type, map);
 496                 if (rc == 0)
 497                         rc = nodemap_idx_idmap_add(nodemap, id_type, map);
 498         }
 499         mutex_unlock(&active_config_lock);
 500         nodemap_putref(nodemap);
 501
 502 out:
 503         return rc;
 504 }
 505 EXPORT_SYMBOL(nodemap_add_idmap);
 506
 507 /**
 508  * delete idmap from proper nodemap tree
 509  *
 510  * \param       name            name of nodemap
 511  * \param       id_type         NODEMAP_UID or NODEMAP_GID
 512  * \param       map             array[2] __u32 containing the mapA values
 513  *                              map[0] is client id
 514  *                              map[1] is the filesystem id
 515  *
 516  * \retval      0 on success
 517  */
 518 int nodemap_del_idmap(const char *name, enum nodemap_id_type id_type,
 519                       const __u32 map[2])
 520 {
 521         struct lu_nodemap       *nodemap = NULL;
 522         struct lu_idmap         *idmap = NULL;
 523         int                     rc = 0;
 524
 525         mutex_lock(&active_config_lock);
 526         nodemap = nodemap_lookup(name);
 527         if (IS_ERR(nodemap)) {
 528                 mutex_unlock(&active_config_lock);
 529                 GOTO(out, rc = PTR_ERR(nodemap));
 530         }
 531
 532         if (is_default_nodemap(nodemap))
 533                 GOTO(out_putref, rc = -EINVAL);
 534
 535         write_lock(&nodemap->nm_idmap_lock);
 536         idmap = idmap_search(nodemap, NODEMAP_CLIENT_TO_FS, id_type,
 537                              map[0]);
 538         if (idmap == NULL) {
 539                 rc = -EINVAL;
 540         } else {
 541                 idmap_delete(id_type, idmap, nodemap);
 542                 rc = nodemap_idx_idmap_del(nodemap, id_type, map);
 543         }
 544         write_unlock(&nodemap->nm_idmap_lock);
 545
 546 out_putref:
 547         mutex_unlock(&active_config_lock);
 548         if (rc == 0)
 549                 nm_member_revoke_locks(nodemap);
 550         nodemap_putref(nodemap);
 551
 552 out:
 553         return rc;
 554 }
 555 EXPORT_SYMBOL(nodemap_del_idmap);
 556
 557 /**
 558  * Get nodemap assigned to given export. Takes a reference on the nodemap.
 559  *
 560  * Note that this function may return either NULL, or an ERR_PTR()
 561  * or a valid nodemap pointer.  All of the functions accessing the
 562  * returned nodemap can check IS_ERR(nodemap) to see if an error is
 563  * returned.  NULL is not considered an error, which is OK since this
 564  * is a valid case if nodemap are not in use.  All nodemap handling
 565  * functions must check for nodemap == NULL and do nothing, and the
 566  * nodemap returned from this function should not be dereferenced.
 567  *
 568  * \param       export          export to get nodemap for
 569  *
 570  * \retval      pointer to nodemap on success
 571  * \retval      NULL    nodemap subsystem disabled
 572  * \retval      -EACCES export does not have nodemap assigned
 573  */
 574 struct lu_nodemap *nodemap_get_from_exp(struct obd_export *exp)
 575 {
 576         struct lu_nodemap *nodemap;
 577
 578         ENTRY;
 579
 580         if (!nodemap_active)
 581                 RETURN(NULL);
 582
 583         spin_lock(&exp->exp_target_data.ted_nodemap_lock);
 584         nodemap = exp->exp_target_data.ted_nodemap;
 585         if (nodemap)
 586                 nodemap_getref(nodemap);
 587         spin_unlock(&exp->exp_target_data.ted_nodemap_lock);
 588
 589         if (!nodemap) {
 590                 CDEBUG(D_INFO, "%s: nodemap null on export %s (at %s)\n",
 591                        exp->exp_obd->obd_name,
 592                        obd_uuid2str(&exp->exp_client_uuid),
 593                        obd_export_nid2str(exp));
 594                 RETURN(ERR_PTR(-EACCES));
 595         }
 596
 597         RETURN(nodemap);
 598 }
 599 EXPORT_SYMBOL(nodemap_get_from_exp);
 600
 601 /**
 602  * mapping function for nodemap idmaps
 603  *
 604  * \param       nodemap         lu_nodemap structure defining nodemap
 605  * \param       node_type       NODEMAP_UID or NODEMAP_GID
 606  * \param       tree_type       NODEMAP_CLIENT_TO_FS or
 607  *                              NODEMAP_FS_TO_CLIENT
 608  * \param       id              id to map
 609  *
 610  * \retval      mapped id according to the rules below.
 611  *
 612  * if the nodemap_active is false, just return the passed id without mapping
 613  *
 614  * if the id to be looked up is 0, check that root access is allowed and if it
 615  * is, return 0. Otherwise, return the squash uid or gid.
 616  *
 617  * if the nodemap is configured to trusted the ids from the client system, just
 618  * return the passwd id without mapping.
 619  *
 620  * if by this point, we haven't returned and the nodemap in question is the
 621  * default nodemap, return the squash uid or gid.
 622  *
 623  * after these checks, search the proper tree for the mapping, and if found
 624  * return the mapped value, otherwise return the squash uid or gid.
 625  */
 626 __u32 nodemap_map_id(struct lu_nodemap *nodemap,
 627                      enum nodemap_id_type id_type,
 628                      enum nodemap_tree_type tree_type, __u32 id)
 629 {
 630         struct lu_idmap         *idmap = NULL;
 631         __u32                    found_id;
 632
 633         ENTRY;
 634
 635         if (!nodemap_active)
 636                 goto out;
 637
 638         if (unlikely(nodemap == NULL))
 639                 goto out;
 640
 641         if (id == 0) {
 642                 if (nodemap->nmf_allow_root_access)
 643                         goto out;
 644                 else
 645                         goto squash;
 646         }
 647
 648         if (nodemap->nmf_trust_client_ids)
 649                 goto out;
 650
 651         if (is_default_nodemap(nodemap))
 652                 goto squash;
 653
 654         read_lock(&nodemap->nm_idmap_lock);
 655         idmap = idmap_search(nodemap, tree_type, id_type, id);
 656         if (idmap == NULL) {
 657                 read_unlock(&nodemap->nm_idmap_lock);
 658                 goto squash;
 659         }
 660
 661         if (tree_type == NODEMAP_FS_TO_CLIENT)
 662                 found_id = idmap->id_client;
 663         else
 664                 found_id = idmap->id_fs;
 665         read_unlock(&nodemap->nm_idmap_lock);
 666         RETURN(found_id);
 667
 668 squash:
 669         if (id_type == NODEMAP_UID)
 670                 RETURN(nodemap->nm_squash_uid);
 671         else
 672                 RETURN(nodemap->nm_squash_gid);
 673 out:
 674         RETURN(id);
 675 }
 676 EXPORT_SYMBOL(nodemap_map_id);
 677
 678 /**
 679  * Map posix ACL entries according to the nodemap membership. Removes any
 680  * squashed ACLs.
 681  *
 682  * \param       lu_nodemap      nodemap
 683  * \param       buf             buffer containing xattr encoded ACLs
 684  * \param       size            size of ACLs in bytes
 685  * \param       tree_type       direction of mapping
 686  * \retval      size            new size of ACLs in bytes
 687  * \retval      -EINVAL         bad \a size param, see posix_acl_xattr_count()
 688  */
 689 ssize_t nodemap_map_acl(struct lu_nodemap *nodemap, void *buf, size_t size,
 690                         enum nodemap_tree_type tree_type)
 691 {
 692         posix_acl_xattr_header  *header = buf;
 693         posix_acl_xattr_entry   *entry = &header->a_entries[0];
 694         posix_acl_xattr_entry   *new_entry = entry;
 695         posix_acl_xattr_entry   *end;
 696         int                      count;
 697
 698         if (!nodemap_active)
 699                 return size;
 700
 701         if (unlikely(nodemap == NULL))
 702                 return size;
 703
 704         count = posix_acl_xattr_count(size);
 705         if (count < 0)
 706                 return -EINVAL;
 707         if (count == 0)
 708                 return 0;
 709
 710         for (end = entry + count; entry != end; entry++) {
 711                 __u16 tag = le16_to_cpu(entry->e_tag);
 712                 __u32 id = le32_to_cpu(entry->e_id);
 713
 714                 switch (tag) {
 715                 case ACL_USER:
 716                         id = nodemap_map_id(nodemap, NODEMAP_UID,
 717                                             tree_type, id);
 718                         if (id == nodemap->nm_squash_uid)
 719                                 continue;
 720                         entry->e_id = cpu_to_le32(id);
 721                         break;
 722                 case ACL_GROUP:
 723                         id = nodemap_map_id(nodemap, NODEMAP_GID,
 724                                             tree_type, id);
 725                         if (id == nodemap->nm_squash_gid)
 726                                 continue;
 727                         entry->e_id = cpu_to_le32(id);
 728                         break;
 729                 }
 730
 731                 /* if we skip an ACL, copy the following ones over it */
 732                 if (new_entry != entry)
 733                         *new_entry = *entry;
 734
 735                 new_entry++;
 736         }
 737
 738         return (void *)new_entry - (void *)header;
 739 }
 740 EXPORT_SYMBOL(nodemap_map_acl);
 741
 742 /*
 743  * Add nid range to given nodemap
 744  *
 745  * \param       config          nodemap config to work on
 746  * \param       nodemap         nodemap to add range to
 747  * \param       nid             nid range to add
 748  * \param       range_id        should be 0 unless loading from disk
 749  * \retval      0               success
 750  * \retval      -ENOMEM
 751  *
 752  */
 753 int nodemap_add_range_helper(struct nodemap_config *config,
 754                              struct lu_nodemap *nodemap,
 755                              const lnet_nid_t nid[2],
 756                              unsigned int range_id)
 757 {
 758         struct lu_nid_range     *range;
 759         int rc;
 760
 761         down_write(&config->nmc_range_tree_lock);
 762         range = range_create(&config->nmc_range_tree, nid[0], nid[1],
 763                              nodemap, range_id);
 764         if (range == NULL) {
 765                 up_write(&config->nmc_range_tree_lock);
 766                 GOTO(out, rc = -ENOMEM);
 767         }
 768
 769         rc = range_insert(&config->nmc_range_tree, range);
 770         if (rc != 0) {
 771                 CERROR("cannot insert nodemap range into '%s': rc = %d\n",
 772                       nodemap->nm_name, rc);
 773                 up_write(&config->nmc_range_tree_lock);
 774                 list_del(&range->rn_list);
 775                 range_destroy(range);
 776                 GOTO(out, rc = -ENOMEM);
 777         }
 778
 779         list_add(&range->rn_list, &nodemap->nm_ranges);
 780         nm_member_reclassify_nodemap(config->nmc_default_nodemap);
 781         up_write(&config->nmc_range_tree_lock);
 782
 783         /* if range_id is non-zero, we are loading from disk */
 784         if (range_id == 0)
 785                 rc = nodemap_idx_range_add(range, nid);
 786
 787         nm_member_revoke_locks(config->nmc_default_nodemap);
 788         nm_member_revoke_locks(nodemap);
 789
 790 out:
 791         return rc;
 792 }
 793 int nodemap_add_range(const char *name, const lnet_nid_t nid[2])
 794 {
 795         struct lu_nodemap       *nodemap = NULL;
 796         int                      rc;
 797
 798         mutex_lock(&active_config_lock);
 799         nodemap = nodemap_lookup(name);
 800         if (IS_ERR(nodemap)) {
 801                 mutex_unlock(&active_config_lock);
 802                 GOTO(out, rc = PTR_ERR(nodemap));
 803         }
 804
 805         if (is_default_nodemap(nodemap))
 806                 rc = -EINVAL;
 807         else
 808                 rc = nodemap_add_range_helper(active_config, nodemap, nid, 0);
 809         mutex_unlock(&active_config_lock);
 810         nodemap_putref(nodemap);
 811 out:
 812         return rc;
 813 }
 814 EXPORT_SYMBOL(nodemap_add_range);
 815
 816 /**
 817  * delete a range
 818  * \param       name            nodemap name
 819  * \param       range_str       string containing range
 820  * \retval      0 on success
 821  *
 822  * Delete range from global range tree, and remove it
 823  * from the list in the associated nodemap.
 824  */
 825 int nodemap_del_range(const char *name, const lnet_nid_t nid[2])
 826 {
 827         struct lu_nodemap       *nodemap;
 828         struct lu_nid_range     *range;
 829         int                     rc = 0;
 830
 831         mutex_lock(&active_config_lock);
 832         nodemap = nodemap_lookup(name);
 833         if (IS_ERR(nodemap)) {
 834                 mutex_unlock(&active_config_lock);
 835                 GOTO(out, rc = PTR_ERR(nodemap));
 836         }
 837
 838         if (is_default_nodemap(nodemap))
 839                 GOTO(out_putref, rc = -EINVAL);
 840
 841         down_write(&active_config->nmc_range_tree_lock);
 842         range = range_find(&active_config->nmc_range_tree, nid[0], nid[1]);
 843         if (range == NULL) {
 844                 up_write(&active_config->nmc_range_tree_lock);
 845                 GOTO(out_putref, rc = -EINVAL);
 846         }
 847         rc = nodemap_idx_range_del(range);
 848         range_delete(&active_config->nmc_range_tree, range);
 849         nm_member_reclassify_nodemap(nodemap);
 850         up_write(&active_config->nmc_range_tree_lock);
 851
 852         nm_member_revoke_locks(active_config->nmc_default_nodemap);
 853         nm_member_revoke_locks(nodemap);
 854
 855 out_putref:
 856         mutex_unlock(&active_config_lock);
 857         nodemap_putref(nodemap);
 858 out:
 859         return rc;
 860 }
 861 EXPORT_SYMBOL(nodemap_del_range);
 862
 863 /**
 864  * set fileset on nodemap
 865  * \param       name            nodemap to set fileset on
 866  * \param       fileset         string containing fileset
 867  * \retval      0 on success
 868  *
 869  * set a fileset on the named nodemap
 870  */
 871 static int nodemap_set_fileset_helper(struct nodemap_config *config,
 872                                       struct lu_nodemap *nodemap,
 873                                       const char *fileset)
 874 {
 875         int rc = 0;
 876
 877         /* we allow fileset = "" which means clear fileset info */
 878         if (fileset == NULL || (fileset[0] != 0 && fileset[0] != '/'))
 879                 rc = -EINVAL;
 880         else if (strlcpy(nodemap->nm_fileset, fileset,
 881                          sizeof(nodemap->nm_fileset)) >=
 882                  sizeof(nodemap->nm_fileset))
 883                 rc = -ENAMETOOLONG;
 884
 885         return rc;
 886 }
 887
 888 int nodemap_set_fileset(const char *name, const char *fileset)
 889 {
 890         struct lu_nodemap       *nodemap = NULL;
 891         int                      rc = 0;
 892
 893         mutex_lock(&active_config_lock);
 894         nodemap = nodemap_lookup(name);
 895         if (IS_ERR(nodemap)) {
 896                 mutex_unlock(&active_config_lock);
 897                 GOTO(out, rc = PTR_ERR(nodemap));
 898         }
 899
 900         if (is_default_nodemap(nodemap))
 901                 rc = -EINVAL;
 902         else
 903                 rc = nodemap_set_fileset_helper(active_config, nodemap,
 904                                                 fileset);
 905         mutex_unlock(&active_config_lock);
 906
 907         nodemap_putref(nodemap);
 908 out:
 909         return rc;
 910 }
 911 EXPORT_SYMBOL(nodemap_set_fileset);
 912
 913 /**
 914  * get fileset defined on nodemap
 915  * \param       nodemap         nodemap to get fileset from
 916  * \retval      fileset name, or NULL if not defined or not activated
 917  *
 918  * get the fileset defined on the nodemap
 919  */
 920 char *nodemap_get_fileset(const struct lu_nodemap *nodemap)
 921 {
 922         if (!nodemap_active || is_default_nodemap(nodemap))
 923                 return NULL;
 924         else
 925                 return (char *)nodemap->nm_fileset;
 926 }
 927 EXPORT_SYMBOL(nodemap_get_fileset);
 928
 929 /**
 930  * Nodemap constructor
 931  *
 932  * Creates an lu_nodemap structure and assigns sane default
 933  * member values. If this is the default nodemap, the defaults
 934  * are the most restictive in xterms of mapping behavior. Otherwise
 935  * the default flags should be inherited from the default nodemap.
 936  * The adds nodemap to nodemap_hash.
 937  *
 938  * Requires that the caller take the active_config_lock
 939  *
 940  * \param       name            name of nodemap
 941  * \param       is_default      true if default nodemap
 942  * \retval      nodemap         success
 943  * \retval      -EINVAL         invalid nodemap name
 944  * \retval      -EEXIST         nodemap already exists
 945  * \retval      -ENOMEM         cannot allocate memory for nodemap
 946  */
 947 struct lu_nodemap *nodemap_create(const char *name,
 948                                   struct nodemap_config *config,
 949                                   bool is_default)
 950 {
 951         struct lu_nodemap       *nodemap = NULL;
 952         struct lu_nodemap       *default_nodemap;
 953         struct cfs_hash         *hash = config->nmc_nodemap_hash;
 954         int                      rc = 0;
 955
 956         default_nodemap = config->nmc_default_nodemap;
 957
 958         if (!nodemap_name_is_valid(name))
 959                 GOTO(out, rc = -EINVAL);
 960
 961         if (hash == NULL) {
 962                 CERROR("Config nodemap hash is NULL, unable to add %s\n", name);
 963                 GOTO(out, rc = -EINVAL);
 964         }
 965
 966         OBD_ALLOC_PTR(nodemap);
 967         if (nodemap == NULL) {
 968                 CERROR("cannot allocate memory (%zu bytes)"
 969                        "for nodemap '%s'\n", sizeof(*nodemap),
 970                        name);
 971                 GOTO(out, rc = -ENOMEM);
 972         }
 973
 974         /*
 975          * take an extra reference to prevent nodemap from being destroyed
 976          * while it's being created.
 977          */
 978         atomic_set(&nodemap->nm_refcount, 2);
 979         snprintf(nodemap->nm_name, sizeof(nodemap->nm_name), "%s", name);
 980         rc = cfs_hash_add_unique(hash, name, &nodemap->nm_hash);
 981         if (rc != 0) {
 982                 OBD_FREE_PTR(nodemap);
 983                 GOTO(out, rc = -EEXIST);
 984         }
 985
 986         INIT_LIST_HEAD(&nodemap->nm_ranges);
 987         INIT_LIST_HEAD(&nodemap->nm_list);
 988         INIT_LIST_HEAD(&nodemap->nm_member_list);
 989
 990         mutex_init(&nodemap->nm_member_list_lock);
 991         rwlock_init(&nodemap->nm_idmap_lock);
 992         nodemap->nm_fs_to_client_uidmap = RB_ROOT;
 993         nodemap->nm_client_to_fs_uidmap = RB_ROOT;
 994         nodemap->nm_fs_to_client_gidmap = RB_ROOT;
 995         nodemap->nm_client_to_fs_gidmap = RB_ROOT;
 996
 997         if (is_default) {
 998                 nodemap->nm_id = LUSTRE_NODEMAP_DEFAULT_ID;
 999                 config->nmc_default_nodemap = nodemap;
1000         } else {
1001                 config->nmc_nodemap_highest_id++;
1002                 nodemap->nm_id = config->nmc_nodemap_highest_id;
1003         }
1004
1005         if (is_default || default_nodemap == NULL) {
1006                 nodemap->nmf_trust_client_ids = 0;
1007                 nodemap->nmf_allow_root_access = 0;
1008
1009                 nodemap->nm_squash_uid = NODEMAP_NOBODY_UID;
1010                 nodemap->nm_squash_gid = NODEMAP_NOBODY_GID;
1011                 if (!is_default)
1012                         CWARN("adding nodemap '%s' to config without"
1013                               " default nodemap\n", nodemap->nm_name);
1014         } else {
1015                 nodemap->nmf_trust_client_ids =
1016                                 default_nodemap->nmf_trust_client_ids;
1017                 nodemap->nmf_allow_root_access =
1018                                 default_nodemap->nmf_allow_root_access;
1019
1020                 nodemap->nm_squash_uid = default_nodemap->nm_squash_uid;
1021                 nodemap->nm_squash_gid = default_nodemap->nm_squash_gid;
1022                 nodemap->nm_fileset[0] = 0;
1023         }
1024
1025         return nodemap;
1026
1027 out:
1028         CERROR("cannot add nodemap: '%s': rc = %d\n", name, rc);
1029         return ERR_PTR(rc);
1030 }
1031
1032 /**
1033  * update flag to turn on or off nodemap functions
1034  * \param       name            nodemap name
1035  * \param       admin_string    string containing updated value
1036  * \retval      0 on success
1037  *
1038  * Update admin flag to turn on or off nodemap functions.
1039  */
1040 int nodemap_set_allow_root(const char *name, bool allow_root)
1041 {
1042         struct lu_nodemap       *nodemap = NULL;
1043         int                     rc = 0;
1044
1045         mutex_lock(&active_config_lock);
1046         nodemap = nodemap_lookup(name);
1047         mutex_unlock(&active_config_lock);
1048         if (IS_ERR(nodemap))
1049                 GOTO(out, rc = PTR_ERR(nodemap));
1050
1051         nodemap->nmf_allow_root_access = allow_root;
1052         rc = nodemap_idx_nodemap_update(nodemap);
1053
1054         nm_member_revoke_locks(nodemap);
1055         nodemap_putref(nodemap);
1056 out:
1057         return rc;
1058 }
1059 EXPORT_SYMBOL(nodemap_set_allow_root);
1060
1061 /**
1062  * updated trust_client_ids flag for nodemap
1063  *
1064  * \param       name            nodemap name
1065  * \param       trust_string    new value for trust flag
1066  * \retval      0 on success
1067  *
1068  * Update the trust_client_ids flag for a nodemap.
1069  */
1070 int nodemap_set_trust_client_ids(const char *name, bool trust_client_ids)
1071 {
1072         struct lu_nodemap       *nodemap = NULL;
1073         int                     rc = 0;
1074
1075         mutex_lock(&active_config_lock);
1076         nodemap = nodemap_lookup(name);
1077         mutex_unlock(&active_config_lock);
1078         if (IS_ERR(nodemap))
1079                 GOTO(out, rc = PTR_ERR(nodemap));
1080
1081         nodemap->nmf_trust_client_ids = trust_client_ids;
1082         rc = nodemap_idx_nodemap_update(nodemap);
1083
1084         nm_member_revoke_locks(nodemap);
1085         nodemap_putref(nodemap);
1086 out:
1087         return rc;
1088 }
1089 EXPORT_SYMBOL(nodemap_set_trust_client_ids);
1090
1091 /**
1092  * update the squash_uid for a nodemap
1093  *
1094  * \param       name            nodemap name
1095  * \param       uid_string      string containing new squash_uid value
1096  * \retval      0 on success
1097  *
1098  * Update the squash_uid for a nodemap. The squash_uid is the uid
1099  * that the all client uids are mapped to if nodemap is active,
1100  * the trust_client_ids flag is not set, and the uid is not in
1101  * the idmap tree.
1102  */
1103 int nodemap_set_squash_uid(const char *name, uid_t uid)
1104 {
1105         struct lu_nodemap       *nodemap = NULL;
1106         int                     rc = 0;
1107
1108         mutex_lock(&active_config_lock);
1109         nodemap = nodemap_lookup(name);
1110         mutex_unlock(&active_config_lock);
1111         if (IS_ERR(nodemap))
1112                 GOTO(out, rc = PTR_ERR(nodemap));
1113
1114         nodemap->nm_squash_uid = uid;
1115         rc = nodemap_idx_nodemap_update(nodemap);
1116
1117         nm_member_revoke_locks(nodemap);
1118         nodemap_putref(nodemap);
1119 out:
1120         return rc;
1121 }
1122 EXPORT_SYMBOL(nodemap_set_squash_uid);
1123
1124 /**
1125  * Update the squash_gid for a nodemap.
1126  *
1127  * \param       name            nodemap name
1128  * \param       gid_string      string containing new squash_gid value
1129  * \retval      0 on success
1130  *
1131  * Update the squash_gid for a nodemap. The squash_uid is the gid
1132  * that the all client gids are mapped to if nodemap is active,
1133  * the trust_client_ids flag is not set, and the gid is not in
1134  * the idmap tree.
1135  */
1136 int nodemap_set_squash_gid(const char *name, gid_t gid)
1137 {
1138         struct lu_nodemap       *nodemap = NULL;
1139         int                     rc = 0;
1140
1141         mutex_lock(&active_config_lock);
1142         nodemap = nodemap_lookup(name);
1143         mutex_unlock(&active_config_lock);
1144         if (IS_ERR(nodemap))
1145                 GOTO(out, rc = PTR_ERR(nodemap));
1146
1147         nodemap->nm_squash_gid = gid;
1148         rc = nodemap_idx_nodemap_update(nodemap);
1149
1150         nm_member_revoke_locks(nodemap);
1151         nodemap_putref(nodemap);
1152 out:
1153         return rc;
1154 }
1155 EXPORT_SYMBOL(nodemap_set_squash_gid);
1156
1157 /**
1158  * Returns true if this nodemap has root user access. Always returns true if
1159  * nodemaps are not active.
1160  *
1161  * \param       nodemap         nodemap to check access for
1162  */
1163 bool nodemap_can_setquota(const struct lu_nodemap *nodemap)
1164 {
1165         return !nodemap_active || (nodemap && nodemap->nmf_allow_root_access);
1166 }
1167 EXPORT_SYMBOL(nodemap_can_setquota);
1168
1169 /**
1170  * Add a nodemap
1171  *
1172  * \param       name            name of nodemap
1173  * \retval      0               success
1174  * \retval      -EINVAL         invalid nodemap name
1175  * \retval      -EEXIST         nodemap already exists
1176  * \retval      -ENOMEM         cannot allocate memory for nodemap
1177  */
1178 int nodemap_add(const char *nodemap_name)
1179 {
1180         struct lu_nodemap *nodemap;
1181         int rc;
1182
1183         mutex_lock(&active_config_lock);
1184         nodemap = nodemap_create(nodemap_name, active_config, 0);
1185         if (IS_ERR(nodemap)) {
1186                 mutex_unlock(&active_config_lock);
1187                 return PTR_ERR(nodemap);
1188         }
1189
1190         rc = nodemap_idx_nodemap_add(nodemap);
1191         if (rc == 0)
1192                 rc = lprocfs_nodemap_register(nodemap, 0);
1193
1194         mutex_unlock(&active_config_lock);
1195         nodemap_putref(nodemap);
1196
1197         return rc;
1198 }
1199 EXPORT_SYMBOL(nodemap_add);
1200
1201 /**
1202  * Delete a nodemap
1203  *
1204  * \param       name            name of nodemmap
1205  * \retval      0               success
1206  * \retval      -EINVAL         invalid input
1207  * \retval      -ENOENT         no existing nodemap
1208  */
1209 int nodemap_del(const char *nodemap_name)
1210 {
1211         struct lu_nodemap       *nodemap;
1212         struct lu_nid_range     *range;
1213         struct lu_nid_range     *range_temp;
1214         int                      rc = 0;
1215         int                      rc2 = 0;
1216
1217         if (strcmp(nodemap_name, DEFAULT_NODEMAP) == 0)
1218                 RETURN(-EINVAL);
1219
1220         mutex_lock(&active_config_lock);
1221         nodemap = cfs_hash_del_key(active_config->nmc_nodemap_hash,
1222                                    nodemap_name);
1223         if (nodemap == NULL) {
1224                 mutex_unlock(&active_config_lock);
1225                 GOTO(out, rc = -ENOENT);
1226         }
1227
1228         /* erase nodemap from active ranges to prevent client assignment */
1229         down_write(&active_config->nmc_range_tree_lock);
1230         list_for_each_entry_safe(range, range_temp, &nodemap->nm_ranges,
1231                                  rn_list) {
1232                 rc2 = nodemap_idx_range_del(range);
1233                 if (rc2 < 0)
1234                         rc = rc2;
1235
1236                 range_delete(&active_config->nmc_range_tree, range);
1237         }
1238         up_write(&active_config->nmc_range_tree_lock);
1239
1240         rc2 = nodemap_idx_nodemap_del(nodemap);
1241         if (rc2 < 0)
1242                 rc = rc2;
1243
1244         /*
1245          * remove procfs here in case nodemap_create called with same name
1246          * before nodemap_destroy is run.
1247          */
1248         lprocfs_nodemap_remove(nodemap->nm_pde_data);
1249         nodemap->nm_pde_data = NULL;
1250
1251         /* reclassify all member exports from nodemap, so they put their refs */
1252         down_read(&active_config->nmc_range_tree_lock);
1253         nm_member_reclassify_nodemap(nodemap);
1254         up_read(&active_config->nmc_range_tree_lock);
1255
1256         if (!list_empty(&nodemap->nm_member_list))
1257                 CWARN("nodemap_del failed to reclassify all members\n");
1258
1259         mutex_unlock(&active_config_lock);
1260
1261         nodemap_putref(nodemap);
1262
1263 out:
1264         return rc;
1265 }
1266 EXPORT_SYMBOL(nodemap_del);
1267
1268 /**
1269  * activate nodemap functions
1270  *
1271  * \param       value           1 for on, 0 for off
1272  */
1273 void nodemap_activate(const bool value)
1274 {
1275         mutex_lock(&active_config_lock);
1276         active_config->nmc_nodemap_is_active = value;
1277
1278         /* copy active value to global to avoid locking in map functions */
1279         nodemap_active = value;
1280         nodemap_idx_nodemap_activate(value);
1281         mutex_unlock(&active_config_lock);
1282         nm_member_revoke_all();
1283 }
1284 EXPORT_SYMBOL(nodemap_activate);
1285
1286 /**
1287  * Helper iterator to convert nodemap hash to list.
1288  *
1289  * \param       hs                      hash structure
1290  * \param       bd                      bucket descriptor
1291  * \param       hnode                   hash node
1292  * \param       nodemap_list_head       list head for list of nodemaps in hash
1293  */
1294 static int nodemap_cleanup_iter_cb(struct cfs_hash *hs, struct cfs_hash_bd *bd,
1295                                    struct hlist_node *hnode,
1296                                    void *nodemap_list_head)
1297 {
1298         struct lu_nodemap *nodemap;
1299
1300         nodemap = hlist_entry(hnode, struct lu_nodemap, nm_hash);
1301         list_add(&nodemap->nm_list, nodemap_list_head);
1302
1303         cfs_hash_bd_del_locked(hs, bd, hnode);
1304
1305         return 0;
1306 }
1307
1308 struct nodemap_config *nodemap_config_alloc(void)
1309 {
1310         struct nodemap_config *config;
1311         int rc = 0;
1312
1313         OBD_ALLOC_PTR(config);
1314         if (config == NULL)
1315                 return ERR_PTR(-ENOMEM);
1316
1317         rc = nodemap_init_hash(config);
1318         if (rc != 0) {
1319                 OBD_FREE_PTR(config);
1320                 return ERR_PTR(rc);
1321         }
1322
1323         init_rwsem(&config->nmc_range_tree_lock);
1324
1325         return config;
1326 }
1327 EXPORT_SYMBOL(nodemap_config_alloc);
1328
1329 /**
1330  * Walk the nodemap_hash and remove all nodemaps.
1331  */
1332 void nodemap_config_dealloc(struct nodemap_config *config)
1333 {
1334         struct lu_nodemap       *nodemap = NULL;
1335         struct lu_nodemap       *nodemap_temp;
1336         struct lu_nid_range     *range;
1337         struct lu_nid_range     *range_temp;
1338         LIST_HEAD(nodemap_list_head);
1339
1340         cfs_hash_for_each_safe(config->nmc_nodemap_hash,
1341                                nodemap_cleanup_iter_cb, &nodemap_list_head);
1342         cfs_hash_putref(config->nmc_nodemap_hash);
1343
1344         /* Because nodemap_destroy might sleep, we can't destroy them
1345          * in cfs_hash_for_each, so we build a list there and destroy here
1346          */
1347         list_for_each_entry_safe(nodemap, nodemap_temp, &nodemap_list_head,
1348                                  nm_list) {
1349                 down_write(&config->nmc_range_tree_lock);
1350
1351                 /* move members to new config */
1352                 nm_member_reclassify_nodemap(nodemap);
1353                 list_for_each_entry_safe(range, range_temp, &nodemap->nm_ranges,
1354                                          rn_list)
1355                         range_delete(&config->nmc_range_tree, range);
1356                 up_write(&config->nmc_range_tree_lock);
1357
1358                 nodemap_putref(nodemap);
1359         }
1360         OBD_FREE_PTR(config);
1361 }
1362 EXPORT_SYMBOL(nodemap_config_dealloc);
1363
1364 static int nm_hash_list_cb(struct cfs_hash *hs, struct cfs_hash_bd *bd,
1365                            struct hlist_node *hnode,
1366                            void *nodemap_list_head)
1367 {
1368         struct lu_nodemap *nodemap;
1369
1370         nodemap = hlist_entry(hnode, struct lu_nodemap, nm_hash);
1371         list_add(&nodemap->nm_list, nodemap_list_head);
1372         return 0;
1373 }
1374
1375 void nodemap_config_set_active(struct nodemap_config *config)
1376 {
1377         struct nodemap_config   *old_config = active_config;
1378         struct lu_nodemap       *nodemap;
1379         struct lu_nodemap       *tmp;
1380         LIST_HEAD(nodemap_list_head);
1381
1382         ENTRY;
1383
1384         LASSERT(active_config != config);
1385         LASSERT(config->nmc_default_nodemap);
1386
1387         mutex_lock(&active_config_lock);
1388
1389         /* move proc entries from already existing nms, create for new nms */
1390         cfs_hash_for_each_safe(config->nmc_nodemap_hash,
1391                                nm_hash_list_cb, &nodemap_list_head);
1392         list_for_each_entry_safe(nodemap, tmp, &nodemap_list_head, nm_list) {
1393                 struct lu_nodemap *old_nm = NULL;
1394
1395                 if (active_config != NULL)
1396                         old_nm = cfs_hash_lookup(
1397                                         active_config->nmc_nodemap_hash,
1398                                         nodemap->nm_name);
1399                 if (old_nm != NULL) {
1400                         nodemap->nm_pde_data = old_nm->nm_pde_data;
1401                         old_nm->nm_pde_data = NULL;
1402                         nodemap_putref(old_nm);
1403                 } else {
1404                         bool is_def = (nodemap == config->nmc_default_nodemap);
1405
1406                         lprocfs_nodemap_register(nodemap, is_def);
1407                 }
1408         }
1409
1410         /* if new config is inactive, deactivate live config before switching */
1411         if (!config->nmc_nodemap_is_active)
1412                 nodemap_active = false;
1413         active_config = config;
1414         if (config->nmc_nodemap_is_active)
1415                 nodemap_active = true;
1416
1417         mutex_unlock(&active_config_lock);
1418
1419         if (old_config != NULL)
1420                 nodemap_config_dealloc(old_config);
1421
1422         nm_member_revoke_all();
1423
1424         EXIT;
1425 }
1426
1427 /**
1428  * Cleanup nodemap module on exit
1429  */
1430 void nodemap_mod_exit(void)
1431 {
1432         nodemap_config_dealloc(active_config);
1433         nodemap_procfs_exit();
1434 }
1435
1436 /**
1437  * Initialize the nodemap module
1438  */
1439 int nodemap_mod_init(void)
1440 {
1441         struct nodemap_config   *new_config;
1442         struct lu_nodemap       *nodemap;
1443         int                      rc = 0;
1444
1445         rc = nodemap_procfs_init();
1446         if (rc != 0)
1447                 return rc;
1448
1449         new_config = nodemap_config_alloc();
1450         if (IS_ERR(new_config)) {
1451                 nodemap_procfs_exit();
1452                 GOTO(out, rc = PTR_ERR(new_config));
1453         }
1454
1455         nodemap = nodemap_create(DEFAULT_NODEMAP, new_config, 1);
1456         if (IS_ERR(nodemap)) {
1457                 nodemap_config_dealloc(new_config);
1458                 nodemap_procfs_exit();
1459                 GOTO(out, rc = PTR_ERR(nodemap));
1460         }
1461
1462         nodemap_config_set_active(new_config);
1463         nodemap_putref(nodemap);
1464
1465 out:
1466         return rc;
1467 }
1468
1469 /**
1470  * Revoke locks for all nodemaps.
1471  */
1472 void nm_member_revoke_all(void)
1473 {
1474         struct lu_nodemap *nodemap;
1475         struct lu_nodemap *tmp;
1476         LIST_HEAD(nodemap_list_head);
1477
1478         mutex_lock(&active_config_lock);
1479         cfs_hash_for_each_safe(active_config->nmc_nodemap_hash,
1480                                nm_hash_list_cb, &nodemap_list_head);
1481
1482         /* revoke_locks sleeps, so can't call in cfs hash cb */
1483         list_for_each_entry_safe(nodemap, tmp, &nodemap_list_head, nm_list)
1484                 nm_member_revoke_locks(nodemap);
1485         mutex_unlock(&active_config_lock);
1486 }
1487
1488 /**
1489  * Returns the nodemap classification for a given nid into an ioctl buffer.
1490  * Useful for testing the nodemap configuration to make sure it is working as
1491  * expected.
1492  *
1493  * \param       nid             nid to classify
1494  * \param[out]  name_buf        buffer to write the nodemap name to
1495  * \param       name_len        length of buffer
1496  */
1497 void nodemap_test_nid(lnet_nid_t nid, char *name_buf, size_t name_len)
1498 {
1499         struct lu_nodemap       *nodemap;
1500
1501         mutex_lock(&active_config_lock);
1502         down_read(&active_config->nmc_range_tree_lock);
1503         nodemap = nodemap_classify_nid(nid);
1504         up_read(&active_config->nmc_range_tree_lock);
1505         mutex_unlock(&active_config_lock);
1506
1507         if (IS_ERR(nodemap))
1508                 return;
1509
1510         strncpy(name_buf, nodemap->nm_name, name_len);
1511         if (name_len > 0)
1512                 name_buf[name_len - 1] = '\0';
1513
1514         nodemap_putref(nodemap);
1515 }
1516 EXPORT_SYMBOL(nodemap_test_nid);
1517
1518 /**
1519  * Passes back the id mapping for a given nid/id pair. Useful for testing the
1520  * nodemap configuration to make sure it is working as expected.
1521  *
1522  * \param       nid             nid to classify
1523  * \param       idtype          uid or gid
1524  * \param       client_id       id to map to fs
1525  * \param       fs_id_buf       pointer to save mapped fs_id to
1526  *
1527  * \retval      0       success
1528  * \retval      -EINVAL invalid NID
1529  */
1530 int nodemap_test_id(lnet_nid_t nid, enum nodemap_id_type idtype,
1531                     __u32 client_id, __u32 *fs_id)
1532 {
1533         struct lu_nodemap       *nodemap;
1534
1535         mutex_lock(&active_config_lock);
1536         down_read(&active_config->nmc_range_tree_lock);
1537         nodemap = nodemap_classify_nid(nid);
1538         up_read(&active_config->nmc_range_tree_lock);
1539         mutex_unlock(&active_config_lock);
1540
1541         if (IS_ERR(nodemap))
1542                 return PTR_ERR(nodemap);
1543
1544         *fs_id = nodemap_map_id(nodemap, idtype, NODEMAP_CLIENT_TO_FS,
1545                                client_id);
1546         nodemap_putref(nodemap);
1547
1548         return 0;
1549 }
1550 EXPORT_SYMBOL(nodemap_test_id);