lustre/ptlrpc/nodemap_storage.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) 2015, Trustees of Indiana University
  24  *
  25  * Copyright (c) 2014, Intel Corporation.
  26  *
  27  * Author: Joshua Walgenbach <jjw@iu.edu>
  28  * Author: Kit Westneat <cwestnea@iu.edu>
  29  *
  30  * Implements the storage functionality for the nodemap configuration. Functions
  31  * in this file prepare, store, and load nodemap configuration data. Targets
  32  * using nodemap services should register a configuration file object. Nodemap
  33  * configuration changes that need to persist should call the appropriate
  34  * storage function for the data being modified.
  35  *
  36  * There are several index types as defined in enum nodemap_idx_type:
  37  *      NODEMAP_CLUSTER_IDX     stores the data found on the lu_nodemap struct,
  38  *                              like root squash and config flags, as well as
  39  *                              the name.
  40  *      NODEMAP_RANGE_IDX       stores NID range information for a nodemap
  41  *      NODEMAP_UIDMAP_IDX      stores a fs/client UID mapping pair
  42  *      NODEMAP_GIDMAP_IDX      stores a fs/client GID mapping pair
  43  *      NODEMAP_GLOBAL_IDX      stores whether or not nodemaps are active
  44  */
  45
  46 #include <libcfs/libcfs.h>
  47 #include <linux/err.h>
  48 #include <linux/kernel.h>
  49 #include <linux/list.h>
  50 #include <linux/mutex.h>
  51 #include <linux/string.h>
  52 #include <linux/types.h>
  53 #include <lnet/types.h>
  54 #include <lustre/lustre_idl.h>
  55 #include <dt_object.h>
  56 #include <lu_object.h>
  57 #include <lustre_net.h>
  58 #include <lustre_nodemap.h>
  59 #include <obd_class.h>
  60 #include <obd_support.h>
  61 #include "nodemap_internal.h"
  62
  63 /* list of registered nodemap index files, except MGS */
  64 static LIST_HEAD(ncf_list_head);
  65 static DEFINE_MUTEX(ncf_list_lock);
  66
  67 /* MGS index is different than others, others are listeners to MGS idx */
  68 static struct nm_config_file *nodemap_mgs_ncf;
  69
  70 /* lu_nodemap flags */
  71 enum nm_flag_shifts {
  72         NM_FL_ALLOW_ROOT_ACCESS = 0x1,
  73         NM_FL_TRUST_CLIENT_IDS = 0x2,
  74         NM_FL_DENY_UNKNOWN = 0x4,
  75 };
  76
  77 static void nodemap_cluster_key_init(struct nodemap_key *nk, unsigned int nm_id)
  78 {
  79         nk->nk_nodemap_id = cpu_to_le32(nm_idx_set_type(nm_id,
  80                                                         NODEMAP_CLUSTER_IDX));
  81         nk->nk_unused = 0;
  82 }
  83
  84 static void nodemap_cluster_rec_init(union nodemap_rec *nr,
  85                                      const struct lu_nodemap *nodemap)
  86 {
  87         CLASSERT(sizeof(nr->ncr.ncr_name) == sizeof(nodemap->nm_name));
  88
  89         strncpy(nr->ncr.ncr_name, nodemap->nm_name, sizeof(nodemap->nm_name));
  90         nr->ncr.ncr_squash_uid = cpu_to_le32(nodemap->nm_squash_uid);
  91         nr->ncr.ncr_squash_gid = cpu_to_le32(nodemap->nm_squash_gid);
  92         nr->ncr.ncr_flags = cpu_to_le32(
  93                 (nodemap->nmf_trust_client_ids ?
  94                         NM_FL_TRUST_CLIENT_IDS : 0) |
  95                 (nodemap->nmf_allow_root_access ?
  96                         NM_FL_ALLOW_ROOT_ACCESS : 0) |
  97                 (nodemap->nmf_deny_unknown ?
  98                         NM_FL_DENY_UNKNOWN : 0));
  99 }
 100
 101 static void nodemap_idmap_key_init(struct nodemap_key *nk, unsigned int nm_id,
 102                                    enum nodemap_id_type id_type,
 103                                    u32 id_client)
 104 {
 105         enum nodemap_idx_type idx_type;
 106
 107         if (id_type == NODEMAP_UID)
 108                 idx_type = NODEMAP_UIDMAP_IDX;
 109         else
 110                 idx_type = NODEMAP_GIDMAP_IDX;
 111
 112         nk->nk_nodemap_id = cpu_to_le32(nm_idx_set_type(nm_id, idx_type));
 113         nk->nk_id_client = cpu_to_le32(id_client);
 114 }
 115
 116 static void nodemap_idmap_rec_init(union nodemap_rec *nr, u32 id_fs)
 117 {
 118         nr->nir.nir_id_fs = cpu_to_le32(id_fs);
 119 }
 120
 121 static void nodemap_range_key_init(struct nodemap_key *nk, unsigned int nm_id,
 122                                    unsigned int rn_id)
 123 {
 124         nk->nk_nodemap_id = cpu_to_le32(nm_idx_set_type(nm_id,
 125                                                         NODEMAP_RANGE_IDX));
 126         nk->nk_range_id = cpu_to_le32(rn_id);
 127 }
 128
 129 static void nodemap_range_rec_init(union nodemap_rec *nr,
 130                                    const lnet_nid_t nid[2])
 131 {
 132         nr->nrr.nrr_start_nid = cpu_to_le64(nid[0]);
 133         nr->nrr.nrr_end_nid = cpu_to_le64(nid[1]);
 134 }
 135
 136 static void nodemap_global_key_init(struct nodemap_key *nk)
 137 {
 138         nk->nk_nodemap_id = cpu_to_le32(nm_idx_set_type(0, NODEMAP_GLOBAL_IDX));
 139         nk->nk_unused = 0;
 140 }
 141
 142 static void nodemap_global_rec_init(union nodemap_rec *nr, bool active)
 143 {
 144         nr->ngr.ngr_is_active = active;
 145 }
 146
 147 /* should be called with dt_write lock */
 148 static void nodemap_inc_version(const struct lu_env *env,
 149                                 struct dt_object *nodemap_idx,
 150                                 struct thandle *th)
 151 {
 152         u64 ver = dt_version_get(env, nodemap_idx);
 153         dt_version_set(env, nodemap_idx, ver + 1, th);
 154 }
 155
 156 static struct dt_object *nodemap_cache_find_create(const struct lu_env *env,
 157                                                    struct dt_device *dev,
 158                                                    struct local_oid_storage *los,
 159                                                    bool force_create)
 160 {
 161         struct lu_fid root_fid;
 162         struct dt_object *root_obj;
 163         struct dt_object *nm_obj;
 164         int rc = 0;
 165
 166         rc = dt_root_get(env, dev, &root_fid);
 167         if (rc < 0)
 168                 GOTO(out, nm_obj = ERR_PTR(rc));
 169
 170         root_obj = dt_locate(env, dev, &root_fid);
 171         if (unlikely(IS_ERR(root_obj)))
 172                 GOTO(out, nm_obj = root_obj);
 173
 174 again:
 175         /* if loading index fails the first time, try again with force_create */
 176         if (force_create) {
 177                 CDEBUG(D_INFO, "removing old index, creating new one\n");
 178                 rc = local_object_unlink(env, dev, root_obj,
 179                                          LUSTRE_NODEMAP_NAME);
 180                 if (rc < 0) {
 181                         /* XXX not sure the best way to get obd name. */
 182                         CERROR("cannot destroy nodemap index: rc = %d\n",
 183                                rc);
 184                         GOTO(out_root, nm_obj = ERR_PTR(rc));
 185                 }
 186         }
 187
 188         nm_obj = local_index_find_or_create(env, los, root_obj,
 189                                                 LUSTRE_NODEMAP_NAME,
 190                                                 S_IFREG | S_IRUGO | S_IWUSR,
 191                                                 &dt_nodemap_features);
 192         if (IS_ERR(nm_obj))
 193                 GOTO(out_root, nm_obj);
 194
 195         if (nm_obj->do_index_ops == NULL) {
 196                 rc = nm_obj->do_ops->do_index_try(env, nm_obj,
 197                                                       &dt_nodemap_features);
 198                 /* even if loading from tgt fails, connecting to MGS will
 199                  * rewrite the config
 200                  */
 201                 if (rc < 0 && !force_create) {
 202                         CERROR("cannot load nodemap index from disk, creating "
 203                                "new index: rc = %d\n", rc);
 204                         lu_object_put(env, &nm_obj->do_lu);
 205                         force_create = true;
 206                         goto again;
 207                 }
 208         }
 209
 210         if (rc < 0)
 211                 nm_obj = ERR_PTR(rc);
 212
 213 out_root:
 214         lu_object_put(env, &root_obj->do_lu);
 215 out:
 216         return nm_obj;
 217 }
 218
 219 static int nodemap_idx_insert(const struct lu_env *env,
 220                               struct dt_object *idx,
 221                               const struct nodemap_key *nk,
 222                               const union nodemap_rec *nr)
 223 {
 224         struct thandle          *th;
 225         struct dt_device        *dev = lu2dt_dev(idx->do_lu.lo_dev);
 226         int                      rc;
 227
 228         CLASSERT(sizeof(union nodemap_rec) == 32);
 229
 230         th = dt_trans_create(env, dev);
 231
 232         if (IS_ERR(th))
 233                 GOTO(out, rc = PTR_ERR(th));
 234
 235         rc = dt_declare_insert(env, idx,
 236                                (const struct dt_rec *)nr,
 237                                (const struct dt_key *)nk, th);
 238         if (rc != 0)
 239                 GOTO(out, rc);
 240
 241         rc = dt_declare_version_set(env, idx, th);
 242         if (rc != 0)
 243                 GOTO(out, rc);
 244
 245         rc = dt_trans_start_local(env, dev, th);
 246         if (rc != 0)
 247                 GOTO(out, rc);
 248
 249         dt_write_lock(env, idx, 0);
 250
 251         rc = dt_insert(env, idx, (const struct dt_rec *)nr,
 252                        (const struct dt_key *)nk, th, 1);
 253
 254         nodemap_inc_version(env, idx, th);
 255         dt_write_unlock(env, idx);
 256 out:
 257         dt_trans_stop(env, dev, th);
 258
 259         return rc;
 260 }
 261
 262 static int nodemap_idx_update(const struct lu_env *env,
 263                               struct dt_object *idx,
 264                               const struct nodemap_key *nk,
 265                               const union nodemap_rec *nr)
 266 {
 267         struct thandle          *th;
 268         struct dt_device        *dev = lu2dt_dev(idx->do_lu.lo_dev);
 269         int                      rc = 0;
 270
 271         th = dt_trans_create(env, dev);
 272
 273         if (IS_ERR(th))
 274                 GOTO(out, rc = PTR_ERR(th));
 275
 276         rc = dt_declare_delete(env, idx, (const struct dt_key *)nk, th);
 277         if (rc != 0)
 278                 GOTO(out, rc);
 279
 280         rc = dt_declare_insert(env, idx, (const struct dt_rec *)nr,
 281                                (const struct dt_key *)nk, th);
 282         if (rc != 0)
 283                 GOTO(out, rc);
 284
 285         rc = dt_declare_version_set(env, idx, th);
 286         if (rc != 0)
 287                 GOTO(out, rc);
 288
 289         rc = dt_trans_start_local(env, dev, th);
 290         if (rc != 0)
 291                 GOTO(out, rc);
 292
 293         dt_write_lock(env, idx, 0);
 294
 295         rc = dt_delete(env, idx, (const struct dt_key *)nk, th);
 296         if (rc != 0)
 297                 GOTO(out_lock, rc);
 298
 299         rc = dt_insert(env, idx, (const struct dt_rec *)nr,
 300                        (const struct dt_key *)nk, th, 1);
 301         if (rc != 0)
 302                 GOTO(out_lock, rc);
 303
 304         nodemap_inc_version(env, idx, th);
 305 out_lock:
 306         dt_write_unlock(env, idx);
 307 out:
 308         dt_trans_stop(env, dev, th);
 309
 310         return rc;
 311 }
 312
 313 static int nodemap_idx_delete(const struct lu_env *env,
 314                               struct dt_object *idx,
 315                               const struct nodemap_key *nk,
 316                               const union nodemap_rec *unused)
 317 {
 318         struct thandle          *th;
 319         struct dt_device        *dev = lu2dt_dev(idx->do_lu.lo_dev);
 320         int                      rc = 0;
 321
 322         th = dt_trans_create(env, dev);
 323
 324         if (IS_ERR(th))
 325                 GOTO(out, rc = PTR_ERR(th));
 326
 327         rc = dt_declare_delete(env, idx, (const struct dt_key *)nk, th);
 328         if (rc != 0)
 329                 GOTO(out, rc);
 330
 331         rc = dt_declare_version_set(env, idx, th);
 332         if (rc != 0)
 333                 GOTO(out, rc);
 334
 335         rc = dt_trans_start_local(env, dev, th);
 336         if (rc != 0)
 337                 GOTO(out, rc);
 338
 339         dt_write_lock(env, idx, 0);
 340
 341         rc = dt_delete(env, idx, (const struct dt_key *)nk, th);
 342
 343         nodemap_inc_version(env, idx, th);
 344
 345         dt_write_unlock(env, idx);
 346 out:
 347         dt_trans_stop(env, dev, th);
 348
 349         return rc;
 350 }
 351
 352 enum nm_add_update {
 353         NM_ADD = 0,
 354         NM_UPDATE = 1,
 355 };
 356
 357 static int nodemap_idx_nodemap_add_update(const struct lu_nodemap *nodemap,
 358                                           enum nm_add_update update)
 359 {
 360         struct nodemap_key nk;
 361         union nodemap_rec nr;
 362         struct lu_env env;
 363         int rc = 0;
 364
 365         ENTRY;
 366
 367         if (nodemap_mgs_ncf == NULL) {
 368                 CERROR("cannot add nodemap config to non-existing MGS.\n");
 369                 return -EINVAL;
 370         }
 371
 372         rc = lu_env_init(&env, LCT_LOCAL);
 373         if (rc)
 374                 RETURN(rc);
 375
 376         nodemap_cluster_key_init(&nk, nodemap->nm_id);
 377         nodemap_cluster_rec_init(&nr, nodemap);
 378
 379         if (update == NM_UPDATE)
 380                 rc = nodemap_idx_update(&env, nodemap_mgs_ncf->ncf_obj,
 381                                         &nk, &nr);
 382         else
 383                 rc = nodemap_idx_insert(&env, nodemap_mgs_ncf->ncf_obj,
 384                                         &nk, &nr);
 385
 386         lu_env_fini(&env);
 387
 388         RETURN(rc);
 389 }
 390
 391 int nodemap_idx_nodemap_add(const struct lu_nodemap *nodemap)
 392 {
 393         return nodemap_idx_nodemap_add_update(nodemap, NM_ADD);
 394 }
 395
 396 int nodemap_idx_nodemap_update(const struct lu_nodemap *nodemap)
 397 {
 398         return nodemap_idx_nodemap_add_update(nodemap, NM_UPDATE);
 399 }
 400
 401 int nodemap_idx_nodemap_del(const struct lu_nodemap *nodemap)
 402 {
 403         struct rb_root           root;
 404         struct lu_idmap         *idmap;
 405         struct lu_idmap         *temp;
 406         struct lu_nid_range     *range;
 407         struct lu_nid_range     *range_temp;
 408         struct nodemap_key       nk;
 409         struct lu_env            env;
 410         int                      rc = 0;
 411         int                      rc2 = 0;
 412
 413         ENTRY;
 414
 415         if (nodemap_mgs_ncf == NULL) {
 416                 CERROR("cannot add nodemap config to non-existing MGS.\n");
 417                 return -EINVAL;
 418         }
 419
 420         rc = lu_env_init(&env, LCT_LOCAL);
 421         if (rc != 0)
 422                 RETURN(rc);
 423
 424         root = nodemap->nm_fs_to_client_uidmap;
 425         nm_rbtree_postorder_for_each_entry_safe(idmap, temp, &root,
 426                                                 id_fs_to_client) {
 427                 nodemap_idmap_key_init(&nk, nodemap->nm_id, NODEMAP_UID,
 428                                        idmap->id_client);
 429                 rc2 = nodemap_idx_delete(&env, nodemap_mgs_ncf->ncf_obj,
 430                                          &nk, NULL);
 431                 if (rc2 < 0)
 432                         rc = rc2;
 433         }
 434
 435         root = nodemap->nm_client_to_fs_gidmap;
 436         nm_rbtree_postorder_for_each_entry_safe(idmap, temp, &root,
 437                                                 id_client_to_fs) {
 438                 nodemap_idmap_key_init(&nk, nodemap->nm_id, NODEMAP_GID,
 439                                        idmap->id_client);
 440                 rc2 = nodemap_idx_delete(&env, nodemap_mgs_ncf->ncf_obj,
 441                                          &nk, NULL);
 442                 if (rc2 < 0)
 443                         rc = rc2;
 444         }
 445
 446         list_for_each_entry_safe(range, range_temp, &nodemap->nm_ranges,
 447                                  rn_list) {
 448                 nodemap_range_key_init(&nk, nodemap->nm_id, range->rn_id);
 449                 rc2 = nodemap_idx_delete(&env, nodemap_mgs_ncf->ncf_obj,
 450                                          &nk, NULL);
 451                 if (rc2 < 0)
 452                         rc = rc2;
 453         }
 454
 455         nodemap_cluster_key_init(&nk, nodemap->nm_id);
 456         rc2 = nodemap_idx_delete(&env, nodemap_mgs_ncf->ncf_obj, &nk, NULL);
 457         if (rc2 < 0)
 458                 rc = rc2;
 459
 460         lu_env_fini(&env);
 461
 462         RETURN(rc);
 463 }
 464
 465 int nodemap_idx_range_add(const struct lu_nid_range *range,
 466                           const lnet_nid_t nid[2])
 467 {
 468         struct nodemap_key       nk;
 469         union nodemap_rec        nr;
 470         struct lu_env            env;
 471         int                      rc = 0;
 472         ENTRY;
 473
 474         if (nodemap_mgs_ncf == NULL) {
 475                 CERROR("cannot add nodemap config to non-existing MGS.\n");
 476                 return -EINVAL;
 477         }
 478
 479         rc = lu_env_init(&env, LCT_LOCAL);
 480         if (rc != 0)
 481                 RETURN(rc);
 482
 483         nodemap_range_key_init(&nk, range->rn_nodemap->nm_id, range->rn_id);
 484         nodemap_range_rec_init(&nr, nid);
 485
 486         rc = nodemap_idx_insert(&env, nodemap_mgs_ncf->ncf_obj, &nk, &nr);
 487         lu_env_fini(&env);
 488
 489         RETURN(rc);
 490 }
 491
 492 int nodemap_idx_range_del(const struct lu_nid_range *range)
 493 {
 494         struct nodemap_key       nk;
 495         struct lu_env            env;
 496         int                      rc = 0;
 497         ENTRY;
 498
 499         if (nodemap_mgs_ncf == NULL) {
 500                 CERROR("cannot add nodemap config to non-existing MGS.\n");
 501                 return -EINVAL;
 502         }
 503
 504         rc = lu_env_init(&env, LCT_LOCAL);
 505         if (rc != 0)
 506                 RETURN(rc);
 507
 508         nodemap_range_key_init(&nk, range->rn_nodemap->nm_id, range->rn_id);
 509
 510         rc = nodemap_idx_delete(&env, nodemap_mgs_ncf->ncf_obj, &nk, NULL);
 511         lu_env_fini(&env);
 512
 513         RETURN(rc);
 514 }
 515
 516 int nodemap_idx_idmap_add(const struct lu_nodemap *nodemap,
 517                           enum nodemap_id_type id_type,
 518                           const u32 map[2])
 519 {
 520         struct nodemap_key       nk;
 521         union nodemap_rec        nr;
 522         struct lu_env            env;
 523         int                      rc = 0;
 524         ENTRY;
 525
 526         if (nodemap_mgs_ncf == NULL) {
 527                 CERROR("cannot add nodemap config to non-existing MGS.\n");
 528                 return -EINVAL;
 529         }
 530
 531         rc = lu_env_init(&env, LCT_LOCAL);
 532         if (rc != 0)
 533                 RETURN(rc);
 534
 535         nodemap_idmap_key_init(&nk, nodemap->nm_id, id_type, map[0]);
 536         nodemap_idmap_rec_init(&nr, map[1]);
 537
 538         rc = nodemap_idx_insert(&env, nodemap_mgs_ncf->ncf_obj, &nk, &nr);
 539         lu_env_fini(&env);
 540
 541         RETURN(rc);
 542 }
 543
 544 int nodemap_idx_idmap_del(const struct lu_nodemap *nodemap,
 545                           enum nodemap_id_type id_type,
 546                           const u32 map[2])
 547 {
 548         struct nodemap_key       nk;
 549         struct lu_env            env;
 550         int                      rc = 0;
 551         ENTRY;
 552
 553         if (nodemap_mgs_ncf == NULL) {
 554                 CERROR("cannot add nodemap config to non-existing MGS.\n");
 555                 return -EINVAL;
 556         }
 557
 558         rc = lu_env_init(&env, LCT_LOCAL);
 559         if (rc != 0)
 560                 RETURN(rc);
 561
 562         nodemap_idmap_key_init(&nk, nodemap->nm_id, id_type, map[0]);
 563
 564         rc = nodemap_idx_delete(&env, nodemap_mgs_ncf->ncf_obj, &nk, NULL);
 565         lu_env_fini(&env);
 566
 567         RETURN(rc);
 568 }
 569
 570 static int nodemap_idx_global_add_update(bool value, enum nm_add_update update)
 571 {
 572         struct nodemap_key       nk;
 573         union nodemap_rec        nr;
 574         struct lu_env            env;
 575         int                      rc = 0;
 576         ENTRY;
 577
 578         if (nodemap_mgs_ncf == NULL) {
 579                 CERROR("cannot add nodemap config to non-existing MGS.\n");
 580                 return -EINVAL;
 581         }
 582
 583         rc = lu_env_init(&env, LCT_LOCAL);
 584         if (rc != 0)
 585                 RETURN(rc);
 586
 587         nodemap_global_key_init(&nk);
 588         nodemap_global_rec_init(&nr, value);
 589
 590         if (update == NM_UPDATE)
 591                 rc = nodemap_idx_update(&env, nodemap_mgs_ncf->ncf_obj,
 592                                         &nk, &nr);
 593         else
 594                 rc = nodemap_idx_insert(&env, nodemap_mgs_ncf->ncf_obj,
 595                                         &nk, &nr);
 596
 597         lu_env_fini(&env);
 598
 599         RETURN(rc);
 600 }
 601
 602 int nodemap_idx_nodemap_activate(bool value)
 603 {
 604         return nodemap_idx_global_add_update(value, NM_UPDATE);
 605 }
 606
 607 /**
 608  * Process a key/rec pair and modify the new configuration.
 609  *
 610  * \param       config          configuration to update with this key/rec data
 611  * \param       key             key of the record that was loaded
 612  * \param       rec             record that was loaded
 613  * \param       recent_nodemap  last referenced nodemap
 614  * \retval      type of record processed, see enum #nodemap_idx_type
 615  * \retval      -ENOENT         range or map loaded before nodemap record
 616  * \retval      -EINVAL         duplicate nodemap cluster records found with
 617  *                              different IDs, or nodemap has invalid name
 618  * \retval      -ENOMEM
 619  */
 620 static int nodemap_process_keyrec(struct nodemap_config *config,
 621                                   const struct nodemap_key *key,
 622                                   const union nodemap_rec *rec,
 623                                   struct lu_nodemap **recent_nodemap)
 624 {
 625         struct lu_nodemap       *nodemap = NULL;
 626         enum nodemap_idx_type    type;
 627         enum nodemap_id_type     id_type;
 628         u8                       flags;
 629         u32                      nodemap_id;
 630         lnet_nid_t               nid[2];
 631         u32                      map[2];
 632         int                      rc;
 633
 634         CLASSERT(sizeof(union nodemap_rec) == 32);
 635
 636         nodemap_id = le32_to_cpu(key->nk_nodemap_id);
 637         type = nm_idx_get_type(nodemap_id);
 638         nodemap_id = nm_idx_set_type(nodemap_id, 0);
 639
 640         CDEBUG(D_INFO, "found config entry, nm_id %d type %d\n",
 641                nodemap_id, type);
 642
 643         /* find the correct nodemap in the load list */
 644         if (type == NODEMAP_RANGE_IDX || type == NODEMAP_UIDMAP_IDX ||
 645             type == NODEMAP_GIDMAP_IDX) {
 646                 struct lu_nodemap *tmp = NULL;
 647
 648                 nodemap = *recent_nodemap;
 649
 650                 if (nodemap == NULL)
 651                         GOTO(out, rc = -ENOENT);
 652
 653                 if (nodemap->nm_id != nodemap_id) {
 654                         list_for_each_entry(tmp, &nodemap->nm_list, nm_list)
 655                                 if (tmp->nm_id == nodemap_id) {
 656                                         nodemap = tmp;
 657                                         break;
 658                                 }
 659
 660                         if (nodemap->nm_id != nodemap_id)
 661                                 GOTO(out, rc = -ENOENT);
 662                 }
 663
 664                 /* update most recently used nodemap if necessay */
 665                 if (nodemap != *recent_nodemap)
 666                         *recent_nodemap = nodemap;
 667         }
 668
 669         switch (type) {
 670         case NODEMAP_EMPTY_IDX:
 671                 if (nodemap_id != 0)
 672                         CWARN("Found nodemap config record without type field, "
 673                               " nodemap_id=%d. nodemap config file corrupt?\n",
 674                               nodemap_id);
 675                 break;
 676         case NODEMAP_CLUSTER_IDX:
 677                 nodemap = cfs_hash_lookup(config->nmc_nodemap_hash,
 678                                           rec->ncr.ncr_name);
 679                 if (nodemap == NULL) {
 680                         if (nodemap_id == LUSTRE_NODEMAP_DEFAULT_ID) {
 681                                 nodemap = nodemap_create(rec->ncr.ncr_name,
 682                                                          config, 1);
 683                                 config->nmc_default_nodemap = nodemap;
 684                         } else {
 685                                 nodemap = nodemap_create(rec->ncr.ncr_name,
 686                                                          config, 0);
 687                         }
 688                         if (IS_ERR(nodemap))
 689                                 GOTO(out, rc = PTR_ERR(nodemap));
 690
 691                         /* we need to override the local ID with the saved ID */
 692                         nodemap->nm_id = nodemap_id;
 693                         if (nodemap_id > config->nmc_nodemap_highest_id)
 694                                 config->nmc_nodemap_highest_id = nodemap_id;
 695
 696                 } else if (nodemap->nm_id != nodemap_id) {
 697                         nodemap_putref(nodemap);
 698                         GOTO(out, rc = -EINVAL);
 699                 }
 700
 701                 nodemap->nm_squash_uid =
 702                                 le32_to_cpu(rec->ncr.ncr_squash_uid);
 703                 nodemap->nm_squash_gid =
 704                                 le32_to_cpu(rec->ncr.ncr_squash_gid);
 705
 706                 flags = le32_to_cpu(rec->ncr.ncr_flags);
 707                 nodemap->nmf_allow_root_access =
 708                                         flags & NM_FL_ALLOW_ROOT_ACCESS;
 709                 nodemap->nmf_trust_client_ids =
 710                                         flags & NM_FL_TRUST_CLIENT_IDS;
 711                 nodemap->nmf_deny_unknown =
 712                                         flags & NM_FL_DENY_UNKNOWN;
 713
 714                 if (*recent_nodemap == NULL) {
 715                         *recent_nodemap = nodemap;
 716                         INIT_LIST_HEAD(&nodemap->nm_list);
 717                 } else {
 718                         list_add(&nodemap->nm_list,
 719                                  &(*recent_nodemap)->nm_list);
 720                 }
 721                 nodemap_putref(nodemap);
 722                 break;
 723         case NODEMAP_RANGE_IDX:
 724                 nid[0] = le64_to_cpu(rec->nrr.nrr_start_nid);
 725                 nid[1] = le64_to_cpu(rec->nrr.nrr_end_nid);
 726
 727                 rc = nodemap_add_range_helper(config, nodemap, nid,
 728                                         le32_to_cpu(key->nk_range_id));
 729                 if (rc != 0)
 730                         GOTO(out, rc);
 731                 break;
 732         case NODEMAP_UIDMAP_IDX:
 733         case NODEMAP_GIDMAP_IDX:
 734                 map[0] = le32_to_cpu(key->nk_id_client);
 735                 map[1] = le32_to_cpu(rec->nir.nir_id_fs);
 736
 737                 if (type == NODEMAP_UIDMAP_IDX)
 738                         id_type = NODEMAP_UID;
 739                 else
 740                         id_type = NODEMAP_GID;
 741
 742                 rc = nodemap_add_idmap_helper(nodemap, id_type, map);
 743                 if (rc != 0)
 744                         GOTO(out, rc);
 745                 break;
 746         case NODEMAP_GLOBAL_IDX:
 747                 config->nmc_nodemap_is_active = rec->ngr.ngr_is_active;
 748                 break;
 749         default:
 750                 CERROR("got keyrec pair for unknown type %d\n", type);
 751                 break;
 752         }
 753
 754         rc = type;
 755
 756 out:
 757         return rc;
 758 }
 759
 760 static int nodemap_load_entries(const struct lu_env *env,
 761                                 struct dt_object *nodemap_idx)
 762 {
 763         const struct dt_it_ops  *iops;
 764         struct dt_it            *it;
 765         struct lu_nodemap       *recent_nodemap = NULL;
 766         struct nodemap_config   *new_config = NULL;
 767         u64                      hash = 0;
 768         bool                     activate_nodemap = false;
 769         bool                     loaded_global_idx = false;
 770         int                      rc = 0;
 771
 772         ENTRY;
 773
 774         iops = &nodemap_idx->do_index_ops->dio_it;
 775
 776         dt_read_lock(env, nodemap_idx, 0);
 777         it = iops->init(env, nodemap_idx, 0);
 778         if (IS_ERR(it))
 779                 GOTO(out, rc = PTR_ERR(it));
 780
 781         rc = iops->load(env, it, hash);
 782         if (rc == 0) {
 783                 rc = iops->next(env, it);
 784                 if (rc != 0)
 785                         GOTO(out_iops, rc = 0);
 786         }
 787
 788         new_config = nodemap_config_alloc();
 789         if (IS_ERR(new_config)) {
 790                 rc = PTR_ERR(new_config);
 791                 new_config = NULL;
 792                 GOTO(out_lock, rc);
 793         }
 794
 795         do {
 796                 struct nodemap_key *key;
 797                 union nodemap_rec rec;
 798
 799                 key = (struct nodemap_key *)iops->key(env, it);
 800                 rc = iops->rec(env, it, (struct dt_rec *)&rec, 0);
 801                 if (rc != -ESTALE) {
 802                         if (rc != 0)
 803                                 GOTO(out_lock, rc);
 804                         rc = nodemap_process_keyrec(new_config, key, &rec,
 805                                                     &recent_nodemap);
 806                         if (rc < 0)
 807                                 GOTO(out_lock, rc);
 808                         if (rc == NODEMAP_GLOBAL_IDX)
 809                                 loaded_global_idx = true;
 810                 }
 811
 812                 do
 813                         rc = iops->next(env, it);
 814                 while (rc == -ESTALE);
 815         } while (rc == 0);
 816
 817         if (rc > 0)
 818                 rc = 0;
 819
 820 out_lock:
 821         if (rc != 0)
 822                 nodemap_config_dealloc(new_config);
 823         else
 824                 /* creating new default needs to be done outside dt read lock */
 825                 activate_nodemap = true;
 826 out_iops:
 827         iops->put(env, it);
 828         iops->fini(env, it);
 829 out:
 830         dt_read_unlock(env, nodemap_idx);
 831
 832         if (rc != 0)
 833                 CWARN("%s: failed to load nodemap configuration: rc = %d\n",
 834                       nodemap_idx->do_lu.lo_dev->ld_obd->obd_name, rc);
 835
 836         if (!activate_nodemap)
 837                 RETURN(rc);
 838
 839         if (new_config->nmc_default_nodemap == NULL) {
 840                 /* new MGS won't have a default nm on disk, so create it here */
 841                 new_config->nmc_default_nodemap =
 842                         nodemap_create(DEFAULT_NODEMAP, new_config, 1);
 843                 if (IS_ERR(new_config->nmc_default_nodemap)) {
 844                         rc = PTR_ERR(new_config->nmc_default_nodemap);
 845                 } else {
 846                         rc = nodemap_idx_nodemap_add_update(
 847                                         new_config->nmc_default_nodemap,
 848                                         NM_ADD);
 849                         nodemap_putref(new_config->nmc_default_nodemap);
 850                 }
 851         }
 852
 853         /* new nodemap config won't have an active/inactive record */
 854         if (rc == 0 && loaded_global_idx == false) {
 855                 struct nodemap_key       nk;
 856                 union nodemap_rec        nr;
 857
 858                 nodemap_global_key_init(&nk);
 859                 nodemap_global_rec_init(&nr, false);
 860                 rc = nodemap_idx_insert(env, nodemap_idx, &nk, &nr);
 861         }
 862
 863         if (rc == 0)
 864                 nodemap_config_set_active(new_config);
 865         else
 866                 nodemap_config_dealloc(new_config);
 867
 868         RETURN(rc);
 869 }
 870
 871 /**
 872  * Step through active config and write to disk.
 873  */
 874 int nodemap_save_config_cache(const struct lu_env *env,
 875                               struct nm_config_file *ncf)
 876 {
 877         struct dt_device *dev;
 878         struct dt_object *o;
 879         struct lu_nodemap *nodemap;
 880         struct lu_nodemap *nm_tmp;
 881         struct lu_nid_range *range;
 882         struct lu_nid_range *range_temp;
 883         struct lu_idmap *idmap;
 884         struct lu_idmap *id_tmp;
 885         struct rb_root root;
 886         struct nodemap_key nk;
 887         union nodemap_rec nr;
 888         LIST_HEAD(nodemap_list_head);
 889         int rc = 0, rc2;
 890
 891         ENTRY;
 892
 893         if (ncf->ncf_los == NULL || ncf->ncf_obj == NULL)
 894                 RETURN(-EIO);
 895
 896         dev = lu2dt_dev(ncf->ncf_obj->do_lu.lo_dev);
 897
 898         /* nodemap_cache_find_create will delete old conf file, so put here */
 899         lu_object_put_nocache(env, &ncf->ncf_obj->do_lu);
 900         ncf->ncf_obj = NULL;
 901
 902         /* force create a new index file to fill with active config */
 903         o = nodemap_cache_find_create(env, dev, ncf->ncf_los, true);
 904         if (IS_ERR(o))
 905                 GOTO(out, rc = PTR_ERR(o));
 906
 907         ncf->ncf_obj = o;
 908
 909         mutex_lock(&active_config_lock);
 910
 911         /* convert hash to list so we don't spin */
 912         cfs_hash_for_each_safe(active_config->nmc_nodemap_hash,
 913                                nm_hash_list_cb, &nodemap_list_head);
 914
 915         list_for_each_entry_safe(nodemap, nm_tmp, &nodemap_list_head, nm_list) {
 916                 nodemap_cluster_key_init(&nk, nodemap->nm_id);
 917                 nodemap_cluster_rec_init(&nr, nodemap);
 918
 919                 rc2 = nodemap_idx_insert(env, o, &nk, &nr);
 920                 if (rc2 < 0) {
 921                         rc = rc2;
 922                         continue;
 923                 }
 924
 925                 down_read(&active_config->nmc_range_tree_lock);
 926                 list_for_each_entry_safe(range, range_temp, &nodemap->nm_ranges,
 927                                          rn_list) {
 928                         lnet_nid_t nid[2] = {
 929                                 range->rn_node.in_extent.start,
 930                                 range->rn_node.in_extent.end
 931                         };
 932                         nodemap_range_key_init(&nk, nodemap->nm_id,
 933                                                range->rn_id);
 934                         nodemap_range_rec_init(&nr, nid);
 935                         rc2 = nodemap_idx_insert(env, o, &nk, &nr);
 936                         if (rc2 < 0)
 937                                 rc = rc2;
 938                 }
 939                 up_read(&active_config->nmc_range_tree_lock);
 940
 941                 /* we don't need to take nm_idmap_lock because active config
 942                  * lock prevents changes from happening to nodemaps
 943                  */
 944                 root = nodemap->nm_client_to_fs_uidmap;
 945                 nm_rbtree_postorder_for_each_entry_safe(idmap, id_tmp, &root,
 946                                                         id_client_to_fs) {
 947                         nodemap_idmap_key_init(&nk, nodemap->nm_id, NODEMAP_UID,
 948                                                idmap->id_client);
 949                         nodemap_idmap_rec_init(&nr, idmap->id_fs);
 950                         rc2 = nodemap_idx_insert(env, o, &nk, &nr);
 951                         if (rc2 < 0)
 952                                 rc = rc2;
 953                 }
 954
 955                 root = nodemap->nm_client_to_fs_gidmap;
 956                 nm_rbtree_postorder_for_each_entry_safe(idmap, id_tmp, &root,
 957                                                         id_client_to_fs) {
 958                         nodemap_idmap_key_init(&nk, nodemap->nm_id, NODEMAP_GID,
 959                                                idmap->id_client);
 960                         nodemap_idmap_rec_init(&nr, idmap->id_fs);
 961                         rc2 = nodemap_idx_insert(env, o, &nk, &nr);
 962                         if (rc2 < 0)
 963                                 rc = rc2;
 964                 }
 965         }
 966         nodemap_global_key_init(&nk);
 967         nodemap_global_rec_init(&nr, active_config->nmc_nodemap_is_active);
 968         rc2 = nodemap_idx_insert(env, o, &nk, &nr);
 969         if (rc2 < 0)
 970                 rc = rc2;
 971
 972 out:
 973         mutex_unlock(&active_config_lock);
 974         RETURN(rc);
 975 }
 976
 977 static void nodemap_save_all_caches(void)
 978 {
 979         struct nm_config_file   *ncf;
 980         struct lu_env            env;
 981         int                      rc = 0;
 982
 983         /* recreating nodemap cache requires fld_thread_key be in env */
 984         rc = lu_env_init(&env, LCT_MD_THREAD | LCT_DT_THREAD | LCT_MG_THREAD);
 985         if (rc != 0) {
 986                 CWARN("cannot init env for nodemap config: rc = %d\n", rc);
 987                 return;
 988         }
 989
 990         mutex_lock(&ncf_list_lock);
 991         list_for_each_entry(ncf, &ncf_list_head, ncf_list) {
 992                 rc = nodemap_save_config_cache(&env, ncf);
 993                 if (rc < 0 && ncf->ncf_obj != NULL)
 994                         CWARN("%s: error writing to nodemap config: rc = %d\n",
 995                               ncf->ncf_obj->do_lu.lo_dev->ld_obd->obd_name, rc);
 996         }
 997         mutex_unlock(&ncf_list_lock);
 998
 999         lu_env_fini(&env);
1000 }
1001
1002 /* tracks if config still needs to be loaded, either from disk or network */
1003 static bool nodemap_config_loaded;
1004 static DEFINE_MUTEX(nodemap_config_loaded_lock);
1005
1006 /**
1007  * Ensures that configs loaded over the wire are prioritized over those loaded
1008  * from disk.
1009  *
1010  * \param config        config to set as the active config
1011  */
1012 void nodemap_config_set_active_mgc(struct nodemap_config *config)
1013 {
1014         mutex_lock(&nodemap_config_loaded_lock);
1015         nodemap_config_set_active(config);
1016         nodemap_config_loaded = true;
1017         nodemap_save_all_caches();
1018         mutex_unlock(&nodemap_config_loaded_lock);
1019 }
1020 EXPORT_SYMBOL(nodemap_config_set_active_mgc);
1021
1022 /**
1023  * Register a dt_object representing the config index file. This should be
1024  * called by targets in order to load the nodemap configuration from disk. The
1025  * dt_object should be created with local_index_find_or_create and the index
1026  * features should be enabled with do_index_try.
1027  *
1028  * \param obj   dt_object returned by local_index_find_or_create
1029  *
1030  * \retval      on success: nm_config_file handle for later deregistration
1031  * \retval      -ENOMEM         memory allocation failure
1032  * \retval      -ENOENT         error loading nodemap config
1033  * \retval      -EINVAL         error loading nodemap config
1034  */
1035 struct nm_config_file *nm_config_file_register(const struct lu_env *env,
1036                                                struct dt_object *obj,
1037                                                struct local_oid_storage *los,
1038                                                enum nm_config_file_type ncf_type)
1039 {
1040         struct nm_config_file *ncf;
1041         bool save_config = false;
1042         int rc = 0;
1043         ENTRY;
1044
1045         OBD_ALLOC_PTR(ncf);
1046         if (ncf == NULL)
1047                 RETURN(ERR_PTR(-ENOMEM));
1048
1049         ncf->ncf_obj = obj;
1050         ncf->ncf_los = los;
1051
1052         if (ncf_type == NCFT_MGS) {
1053                 nodemap_mgs_ncf = ncf;
1054         } else {
1055                 mutex_lock(&ncf_list_lock);
1056                 list_add(&ncf->ncf_list, &ncf_list_head);
1057                 mutex_unlock(&ncf_list_lock);
1058         }
1059
1060         /* prevent activation of config loaded from MGS until disk is loaded
1061          * so disk config is overwritten by MGS config.
1062          */
1063         mutex_lock(&nodemap_config_loaded_lock);
1064         if (ncf_type == NCFT_MGS || !nodemap_config_loaded)
1065                 rc = nodemap_load_entries(env, obj);
1066         else
1067                 save_config = true;
1068         nodemap_config_loaded = true;
1069         mutex_unlock(&nodemap_config_loaded_lock);
1070
1071         /* sync on disk caches with loaded config in memory */
1072         if (save_config)
1073                 rc = nodemap_save_config_cache(env, ncf);
1074
1075         if (rc < 0) {
1076                 if (ncf_type == NCFT_MGS) {
1077                         nodemap_mgs_ncf = NULL;
1078                 } else {
1079                         mutex_lock(&ncf_list_lock);
1080                         list_del(&ncf->ncf_list);
1081                         mutex_unlock(&ncf_list_lock);
1082                 }
1083
1084                 OBD_FREE_PTR(ncf);
1085                 RETURN(ERR_PTR(rc));
1086         }
1087
1088         RETURN(ncf);
1089 }
1090 EXPORT_SYMBOL(nm_config_file_register);
1091
1092 /**
1093  * Deregister a nm_config_file. Should be called by targets during cleanup.
1094  *
1095  * \param ncf   config file to deregister
1096  */
1097 void nm_config_file_deregister(const struct lu_env *env,
1098                                struct nm_config_file *ncf,
1099                                enum nm_config_file_type ncf_type)
1100 {
1101         ENTRY;
1102
1103         if (ncf->ncf_obj)
1104                 lu_object_put(env, &ncf->ncf_obj->do_lu);
1105
1106         if (ncf_type == NCFT_TGT) {
1107                 mutex_lock(&ncf_list_lock);
1108                 list_del(&ncf->ncf_list);
1109                 mutex_unlock(&ncf_list_lock);
1110         } else {
1111                 nodemap_mgs_ncf = NULL;
1112         }
1113         OBD_FREE_PTR(ncf);
1114
1115         EXIT;
1116 }
1117 EXPORT_SYMBOL(nm_config_file_deregister);
1118
1119 int nodemap_process_idx_pages(struct nodemap_config *config, union lu_page *lip,
1120                               struct lu_nodemap **recent_nodemap)
1121 {
1122         struct nodemap_key *key;
1123         union nodemap_rec *rec;
1124         char *entry;
1125         int j;
1126         int k;
1127         int rc = 0;
1128         int size = dt_nodemap_features.dif_keysize_max +
1129                    dt_nodemap_features.dif_recsize_max;
1130         ENTRY;
1131
1132         for (j = 0; j < LU_PAGE_COUNT; j++) {
1133                 if (lip->lp_idx.lip_magic != LIP_MAGIC)
1134                         return -EINVAL;
1135
1136                 /* get and process keys and records from page */
1137                 for (k = 0; k < lip->lp_idx.lip_nr; k++) {
1138                         entry = lip->lp_idx.lip_entries + k * size;
1139                         key = (struct nodemap_key *)entry;
1140
1141                         entry += dt_nodemap_features.dif_keysize_max;
1142                         rec = (union nodemap_rec *)entry;
1143
1144                         rc = nodemap_process_keyrec(config, key, rec,
1145                                                     recent_nodemap);
1146                         if (rc < 0)
1147                                 return rc;
1148                 }
1149                 lip++;
1150         }
1151
1152         EXIT;
1153         return 0;
1154 }
1155 EXPORT_SYMBOL(nodemap_process_idx_pages);
1156
1157 int nodemap_index_read(struct lu_env *env,
1158                        struct nm_config_file *ncf,
1159                        struct idx_info *ii,
1160                        const struct lu_rdpg *rdpg)
1161 {
1162         struct dt_object        *nodemap_idx = ncf->ncf_obj;
1163         __u64                    version;
1164         int                      rc = 0;
1165
1166         ii->ii_keysize = dt_nodemap_features.dif_keysize_max;
1167         ii->ii_recsize = dt_nodemap_features.dif_recsize_max;
1168
1169         dt_read_lock(env, nodemap_idx, 0);
1170         version = dt_version_get(env, nodemap_idx);
1171         if (rdpg->rp_hash != 0 && ii->ii_version != version) {
1172                 CDEBUG(D_INFO, "nodemap config changed while sending, "
1173                                "old "LPU64", new "LPU64"\n",
1174                        ii->ii_version,
1175                        version);
1176                 ii->ii_hash_end = 0;
1177         } else {
1178                 rc = dt_index_walk(env, nodemap_idx, rdpg, NULL, ii);
1179                 CDEBUG(D_INFO, "walked index, hashend %llx\n", ii->ii_hash_end);
1180         }
1181
1182         if (rc >= 0)
1183                 ii->ii_version = version;
1184
1185         dt_read_unlock(env, nodemap_idx);
1186         return rc;
1187 }
1188 EXPORT_SYMBOL(nodemap_index_read);
1189
1190 /**
1191  * Returns the current nodemap configuration to MGC by walking the nodemap
1192  * config index and storing it in the response buffer.
1193  *
1194  * \param       req             incoming MGS_CONFIG_READ request
1195  * \retval      0               success
1196  * \retval      -EINVAL         malformed request
1197  * \retval      -ENOTCONN       client evicted/reconnected already
1198  * \retval      -ETIMEDOUT      client timeout or network error
1199  * \retval      -ENOMEM
1200  */
1201 int nodemap_get_config_req(struct obd_device *mgs_obd,
1202                            struct ptlrpc_request *req)
1203 {
1204         struct mgs_config_body *body;
1205         struct mgs_config_res *res;
1206         struct lu_rdpg rdpg;
1207         struct idx_info nodemap_ii;
1208         struct ptlrpc_bulk_desc *desc;
1209         struct l_wait_info lwi;
1210         struct tg_export_data *rqexp_ted = &req->rq_export->exp_target_data;
1211         int i;
1212         int page_count;
1213         int bytes = 0;
1214         int rc = 0;
1215
1216         body = req_capsule_client_get(&req->rq_pill, &RMF_MGS_CONFIG_BODY);
1217         if (!body)
1218                 RETURN(-EINVAL);
1219
1220         if (body->mcb_type != CONFIG_T_NODEMAP)
1221                 RETURN(-EINVAL);
1222
1223         rdpg.rp_count = (body->mcb_units << body->mcb_bits);
1224         rdpg.rp_npages = (rdpg.rp_count + PAGE_CACHE_SIZE - 1) >>
1225                 PAGE_CACHE_SHIFT;
1226         if (rdpg.rp_npages > PTLRPC_MAX_BRW_PAGES)
1227                 RETURN(-EINVAL);
1228
1229         CDEBUG(D_INFO, "reading nodemap log, name '%s', size = %u\n",
1230                body->mcb_name, rdpg.rp_count);
1231
1232         /* allocate pages to store the containers */
1233         OBD_ALLOC(rdpg.rp_pages, sizeof(*rdpg.rp_pages) * rdpg.rp_npages);
1234         if (rdpg.rp_pages == NULL)
1235                 RETURN(-ENOMEM);
1236         for (i = 0; i < rdpg.rp_npages; i++) {
1237                 rdpg.rp_pages[i] = alloc_page(GFP_IOFS);
1238                 if (rdpg.rp_pages[i] == NULL)
1239                         GOTO(out, rc = -ENOMEM);
1240         }
1241
1242         rdpg.rp_hash = body->mcb_offset;
1243         nodemap_ii.ii_magic = IDX_INFO_MAGIC;
1244         nodemap_ii.ii_flags = II_FL_NOHASH;
1245         nodemap_ii.ii_version = rqexp_ted->ted_nodemap_version;
1246
1247         bytes = nodemap_index_read(req->rq_svc_thread->t_env,
1248                                    mgs_obd->u.obt.obt_nodemap_config_file,
1249                                    &nodemap_ii, &rdpg);
1250         if (bytes < 0)
1251                 GOTO(out, rc = bytes);
1252
1253         rqexp_ted->ted_nodemap_version = nodemap_ii.ii_version;
1254
1255         res = req_capsule_server_get(&req->rq_pill, &RMF_MGS_CONFIG_RES);
1256         if (res == NULL)
1257                 GOTO(out, rc = -EINVAL);
1258         res->mcr_offset = nodemap_ii.ii_hash_end;
1259         res->mcr_size = bytes;
1260
1261         page_count = (bytes + PAGE_CACHE_SIZE - 1) >> PAGE_CACHE_SHIFT;
1262         LASSERT(page_count <= rdpg.rp_count);
1263         desc = ptlrpc_prep_bulk_exp(req, page_count, 1,
1264                                     PTLRPC_BULK_PUT_SOURCE |
1265                                         PTLRPC_BULK_BUF_KIOV,
1266                                     MGS_BULK_PORTAL,
1267                                     &ptlrpc_bulk_kiov_pin_ops);
1268         if (desc == NULL)
1269                 GOTO(out, rc = -ENOMEM);
1270
1271         for (i = 0; i < page_count && bytes > 0; i++) {
1272                 ptlrpc_prep_bulk_page_pin(desc, rdpg.rp_pages[i], 0,
1273                                           min_t(int, bytes, PAGE_CACHE_SIZE));
1274                 bytes -= PAGE_CACHE_SIZE;
1275         }
1276
1277         rc = target_bulk_io(req->rq_export, desc, &lwi);
1278         ptlrpc_free_bulk(desc);
1279
1280 out:
1281         if (rdpg.rp_pages != NULL) {
1282                 for (i = 0; i < rdpg.rp_npages; i++)
1283                         if (rdpg.rp_pages[i] != NULL)
1284                                 __free_page(rdpg.rp_pages[i]);
1285                 OBD_FREE(rdpg.rp_pages,
1286                          rdpg.rp_npages * sizeof(rdpg.rp_pages[0]));
1287         }
1288         return rc;
1289 }
1290 EXPORT_SYMBOL(nodemap_get_config_req);
1291
1292 int nodemap_fs_init(const struct lu_env *env, struct dt_device *dev,
1293                     struct obd_device *obd, struct local_oid_storage *los)
1294 {
1295         struct dt_object        *config_obj;
1296         struct nm_config_file   *nm_config_file;
1297         int                      rc = 0;
1298         ENTRY;
1299
1300         CDEBUG(D_INFO, "%s: finding nodemap index\n", obd->obd_name);
1301         /* load or create the index file from disk (don't force create) */
1302         config_obj = nodemap_cache_find_create(env, dev, los, false);
1303         if (IS_ERR(config_obj))
1304                 GOTO(out, rc = PTR_ERR(config_obj));
1305
1306         CDEBUG(D_INFO, "%s: registering nodemap index\n", obd->obd_name);
1307
1308         nm_config_file = nm_config_file_register(env, config_obj, los,
1309                                                  NCFT_TGT);
1310         if (IS_ERR(nm_config_file)) {
1311                 CERROR("%s: error loading nodemap config file, file must be "
1312                        "removed via ldiskfs: rc = %ld\n",
1313                        obd->obd_name, PTR_ERR(nm_config_file));
1314                 GOTO(out, rc = PTR_ERR(nm_config_file));
1315         }
1316
1317         obd->u.obt.obt_nodemap_config_file = nm_config_file;
1318
1319         /* save los in case object needs to be re-created */
1320         nm_config_file->ncf_los = los;
1321
1322         EXIT;
1323
1324 out:
1325         return rc;
1326 }
1327 EXPORT_SYMBOL(nodemap_fs_init);
1328
1329 void nodemap_fs_fini(const struct lu_env *env, struct obd_device *obd)
1330 {
1331         if (obd->u.obt.obt_nodemap_config_file == NULL)
1332                 return;
1333
1334         nm_config_file_deregister(env, obd->u.obt.obt_nodemap_config_file,
1335                                   NCFT_TGT);
1336         obd->u.obt.obt_nodemap_config_file = NULL;
1337 }
1338 EXPORT_SYMBOL(nodemap_fs_fini);