lustre/ptlrpc/nodemap_lproc.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, 2017, Intel Corporation.
  26  *
  27  * Author: Joshua Walgenbach <jjw@iu.edu>
  28  */
  29
  30 #define NODEMAP_LPROC_ID_LEN 16
  31 #define NODEMAP_LPROC_FLAG_LEN 2
  32
  33 #include <lprocfs_status.h>
  34 #include <lustre_net.h>
  35 #include <lustre_export.h>
  36 #include <obd_class.h>
  37 #include <interval_tree.h>
  38 #include "nodemap_internal.h"
  39
  40 static LIST_HEAD(nodemap_pde_list);
  41
  42 /**
  43  * Reads and prints the idmap for the given nodemap.
  44  *
  45  * \param       m               seq file in proc fs
  46  * \param       data            unused
  47  * \retval      0               success
  48  */
  49 static int nodemap_idmap_show(struct seq_file *m, void *data)
  50 {
  51         struct lu_nodemap       *nodemap;
  52         struct lu_idmap         *idmap;
  53         struct rb_node          *node;
  54         bool                    cont = 0;
  55         int rc;
  56
  57         mutex_lock(&active_config_lock);
  58         nodemap = nodemap_lookup(m->private);
  59         mutex_unlock(&active_config_lock);
  60         if (IS_ERR(nodemap)) {
  61                 rc = PTR_ERR(nodemap);
  62                 CERROR("cannot find nodemap '%s': rc = %d\n",
  63                         (char *)m->private, rc);
  64                 return rc;
  65         }
  66
  67         seq_printf(m, "[\n");
  68         down_read(&nodemap->nm_idmap_lock);
  69         for (node = rb_first(&nodemap->nm_client_to_fs_uidmap); node;
  70                                 node = rb_next(node)) {
  71                 if (cont)
  72                         seq_printf(m, ",\n");
  73                 cont = 1;
  74                 idmap = rb_entry(node, struct lu_idmap, id_client_to_fs);
  75                 if (idmap != NULL)
  76                         seq_printf(m, " { idtype: uid, client_id: %u, "
  77                                    "fs_id: %u }", idmap->id_client,
  78                                    idmap->id_fs);
  79         }
  80         for (node = rb_first(&nodemap->nm_client_to_fs_gidmap);
  81                                 node; node = rb_next(node)) {
  82                 if (cont)
  83                         seq_printf(m, ",\n");
  84                 idmap = rb_entry(node, struct lu_idmap, id_client_to_fs);
  85                 if (idmap != NULL)
  86                         seq_printf(m, " { idtype: gid, client_id: %u, "
  87                                    "fs_id: %u }", idmap->id_client,
  88                                    idmap->id_fs);
  89         }
  90         up_read(&nodemap->nm_idmap_lock);
  91         seq_printf(m, "\n");
  92         seq_printf(m, "]\n");
  93
  94         nodemap_putref(nodemap);
  95         return 0;
  96 }
  97
  98 /**
  99  * Attaches nodemap_idmap_show to proc file.
 100  *
 101  * \param       inode           inode of seq file in proc fs
 102  * \param       file            seq file
 103  * \retval      0               success
 104  */
 105 static int nodemap_idmap_open(struct inode *inode, struct file *file)
 106 {
 107         return single_open(file, nodemap_idmap_show, PDE_DATA(inode));
 108 }
 109
 110 /**
 111  * Reads and prints the NID ranges for the given nodemap.
 112  *
 113  * \param       m               seq file in proc fs
 114  * \param       data            unused
 115  * \retval      0               success
 116  */
 117 static int nodemap_ranges_show(struct seq_file *m, void *data)
 118 {
 119         struct lu_nodemap               *nodemap;
 120         struct lu_nid_range             *range;
 121         struct interval_node_extent     ext;
 122         char                            start_nidstr[LNET_NIDSTR_SIZE];
 123         char                            end_nidstr[LNET_NIDSTR_SIZE];
 124         bool                            cont = false;
 125         int rc;
 126
 127         mutex_lock(&active_config_lock);
 128         nodemap = nodemap_lookup(m->private);
 129         if (IS_ERR(nodemap)) {
 130                 mutex_unlock(&active_config_lock);
 131                 rc = PTR_ERR(nodemap);
 132                 CERROR("cannot find nodemap '%s': rc = %d\n",
 133                         (char *)m->private, rc);
 134                 return rc;
 135         }
 136
 137         seq_printf(m, "[\n");
 138         down_read(&active_config->nmc_range_tree_lock);
 139         list_for_each_entry(range, &nodemap->nm_ranges, rn_list) {
 140                 if (cont)
 141                         seq_printf(m, ",\n");
 142                 cont = 1;
 143                 ext = range->rn_node.in_extent;
 144                 libcfs_nid2str_r(ext.start, start_nidstr, sizeof(start_nidstr));
 145                 libcfs_nid2str_r(ext.end, end_nidstr, sizeof(end_nidstr));
 146                 seq_printf(m, " { id: %u, start_nid: %s, end_nid: %s }",
 147                            range->rn_id, start_nidstr, end_nidstr);
 148         }
 149         up_read(&active_config->nmc_range_tree_lock);
 150         mutex_unlock(&active_config_lock);
 151         seq_printf(m, "\n");
 152         seq_printf(m, "]\n");
 153
 154         nodemap_putref(nodemap);
 155         return 0;
 156 }
 157
 158 /**
 159  * Connects nodemap_idmap_show to proc file.
 160  *
 161  * \param       inode           inode of seq file in proc fs
 162  * \param       file            seq file
 163  * \retval      0               success
 164  */
 165 static int nodemap_ranges_open(struct inode *inode, struct file *file)
 166 {
 167         return single_open(file, nodemap_ranges_show, PDE_DATA(inode));
 168 }
 169
 170 /**
 171  * Reads and prints the fileset for the given nodemap.
 172  *
 173  * \param       m               seq file in proc fs
 174  * \param       data            unused
 175  * \retval      0               success
 176  */
 177 static int nodemap_fileset_seq_show(struct seq_file *m, void *data)
 178 {
 179         struct lu_nodemap *nodemap;
 180         int rc = 0;
 181
 182         mutex_lock(&active_config_lock);
 183         nodemap = nodemap_lookup(m->private);
 184         mutex_unlock(&active_config_lock);
 185         if (IS_ERR(nodemap)) {
 186                 rc = PTR_ERR(nodemap);
 187                 CERROR("cannot find nodemap '%s': rc = %d\n",
 188                         (char *)m->private, rc);
 189                 return rc;
 190         }
 191
 192         seq_printf(m, "%s\n", nodemap->nm_fileset);
 193         nodemap_putref(nodemap);
 194         return rc;
 195 }
 196
 197 /**
 198  * Set a fileset on a nodemap.
 199  *
 200  * \param[in] file      proc file
 201  * \param[in] buffer    string, "<fileset>"
 202  * \param[in] count     \a buffer length
 203  * \param[in] off       unused
 204  * \retval              \a count on success
 205  * \retval              negative number on error
 206  */
 207 static ssize_t
 208 nodemap_fileset_seq_write(struct file *file,
 209                                       const char __user *buffer,
 210                                       size_t count, loff_t *off)
 211 {
 212         struct seq_file *m = file->private_data;
 213         char *nm_fileset;
 214         int rc = 0;
 215         ENTRY;
 216
 217         if (count == 0)
 218                 RETURN(0);
 219
 220         if (count > PATH_MAX)
 221                 RETURN(-EINVAL);
 222
 223         OBD_ALLOC(nm_fileset, count + 1);
 224         /* OBD_ALLOC zero-fills the buffer */
 225         if (nm_fileset == NULL)
 226                 RETURN(-ENOMEM);
 227
 228         if (copy_from_user(nm_fileset, buffer, count))
 229                 GOTO(out, rc = -EFAULT);
 230
 231         rc = nodemap_set_fileset(m->private, nm_fileset);
 232         if (rc != 0)
 233                 GOTO(out, rc = -EINVAL);
 234
 235         rc = count;
 236 out:
 237         OBD_FREE(nm_fileset, count + 1);
 238
 239         return rc;
 240 }
 241 LPROC_SEQ_FOPS(nodemap_fileset);
 242
 243 /**
 244  * Reads and prints the exports attached to the given nodemap.
 245  *
 246  * \param       m               seq file in proc fs, stores nodemap
 247  * \param       data            unused
 248  * \retval      0               success
 249  */
 250 static int nodemap_exports_show(struct seq_file *m, void *data)
 251 {
 252         struct lu_nodemap *nodemap;
 253         struct obd_export *exp;
 254         char nidstr[LNET_NIDSTR_SIZE] = "<unknown>";
 255         int rc;
 256
 257         mutex_lock(&active_config_lock);
 258         nodemap = nodemap_lookup(m->private);
 259         mutex_unlock(&active_config_lock);
 260         if (IS_ERR(nodemap)) {
 261                 rc = PTR_ERR(nodemap);
 262                 CERROR("cannot find nodemap '%s': rc = %d\n",
 263                         (char *)m->private, rc);
 264                 return rc;
 265         }
 266
 267         seq_printf(m, "[\n");
 268
 269         mutex_lock(&nodemap->nm_member_list_lock);
 270         list_for_each_entry(exp, &nodemap->nm_member_list,
 271                             exp_target_data.ted_nodemap_member) {
 272                 if (exp->exp_connection != NULL)
 273                         libcfs_nid2str_r(exp->exp_connection->c_peer.nid,
 274                                          nidstr, sizeof(nidstr));
 275
 276                 seq_printf(m, " { nid: %s, uuid: %s },",
 277                            nidstr, exp->exp_client_uuid.uuid);
 278         }
 279         mutex_unlock(&nodemap->nm_member_list_lock);
 280
 281         seq_printf(m, "\n");
 282         seq_printf(m, "]\n");
 283
 284         nodemap_putref(nodemap);
 285         return 0;
 286 }
 287
 288 /**
 289  * Attaches nodemap_idmap_show to proc file.
 290  *
 291  * \param       inode           inode of seq file in proc fs
 292  * \param       file            seq file
 293  * \retval      0               success
 294  */
 295 static int nodemap_exports_open(struct inode *inode, struct file *file)
 296 {
 297         return single_open(file, nodemap_exports_show, PDE_DATA(inode));
 298 }
 299
 300 /**
 301  * Reads and prints the active flag for the given nodemap.
 302  *
 303  * \param       m               seq file in proc fs
 304  * \param       data            unused
 305  * \retval      0               success
 306  */
 307 static int nodemap_active_seq_show(struct seq_file *m, void *data)
 308 {
 309         seq_printf(m, "%u\n", (unsigned int)nodemap_active);
 310         return 0;
 311 }
 312
 313 /**
 314  * Activate/deactivate nodemap.
 315  *
 316  * \param[in] file      proc file
 317  * \param[in] buffer    string, "1" or "0" to activate/deactivate nodemap
 318  * \param[in] count     \a buffer length
 319  * \param[in] off       unused
 320  * \retval              \a count on success
 321  * \retval              negative number on error
 322  */
 323 static ssize_t
 324 nodemap_active_seq_write(struct file *file, const char __user *buffer,
 325                          size_t count, loff_t *off)
 326 {
 327         char                    active_string[NODEMAP_LPROC_FLAG_LEN + 1];
 328         long unsigned int       active;
 329         int                     rc;
 330
 331         if (count == 0)
 332                 return 0;
 333
 334         if (count >= sizeof(active_string))
 335                 return -EINVAL;
 336
 337         if (copy_from_user(active_string, buffer, count))
 338                 return -EFAULT;
 339
 340         active_string[count] = '\0';
 341         rc = kstrtoul(active_string, 10, &active);
 342         if (rc != 0)
 343                 return -EINVAL;
 344
 345         nodemap_activate(active);
 346
 347         return count;
 348 }
 349 LPROC_SEQ_FOPS(nodemap_active);
 350
 351 /**
 352  * Reads and prints the nodemap ID for the given nodemap.
 353  *
 354  * \param       m               seq file in proc fs
 355  * \param       data            unused
 356  * \retval      0               success
 357  */
 358 static int nodemap_id_seq_show(struct seq_file *m, void *data)
 359 {
 360         struct lu_nodemap *nodemap;
 361
 362         mutex_lock(&active_config_lock);
 363         nodemap = nodemap_lookup(m->private);
 364         mutex_unlock(&active_config_lock);
 365         if (IS_ERR(nodemap)) {
 366                 int rc = PTR_ERR(nodemap);
 367                 CERROR("cannot find nodemap '%s': rc = %d\n",
 368                         (char *)m->private, rc);
 369                 return rc;
 370         }
 371
 372         seq_printf(m, "%u\n", nodemap->nm_id);
 373         nodemap_putref(nodemap);
 374         return 0;
 375 }
 376 LPROC_SEQ_FOPS_RO(nodemap_id);
 377
 378 /**
 379  * Reads and prints the root squash UID for the given nodemap.
 380  *
 381  * \param       m               seq file in proc fs
 382  * \param       data            unused
 383  * \retval      0               success
 384  */
 385 static int nodemap_squash_uid_seq_show(struct seq_file *m, void *data)
 386 {
 387         struct lu_nodemap *nodemap;
 388
 389         mutex_lock(&active_config_lock);
 390         nodemap = nodemap_lookup(m->private);
 391         mutex_unlock(&active_config_lock);
 392         if (IS_ERR(nodemap)) {
 393                 int rc = PTR_ERR(nodemap);
 394                 CERROR("cannot find nodemap '%s': rc = %d\n",
 395                         (char *)m->private, rc);
 396                 return rc;
 397         }
 398
 399         seq_printf(m, "%u\n", nodemap->nm_squash_uid);
 400         nodemap_putref(nodemap);
 401         return 0;
 402 }
 403
 404 /**
 405  * Reads and prints the root squash GID for the given nodemap.
 406  *
 407  * \param       m               seq file in proc fs
 408  * \param       data            unused
 409  * \retval      0               success
 410  */
 411 static int nodemap_squash_gid_seq_show(struct seq_file *m, void *data)
 412 {
 413         struct lu_nodemap *nodemap;
 414
 415         mutex_lock(&active_config_lock);
 416         nodemap = nodemap_lookup(m->private);
 417         mutex_unlock(&active_config_lock);
 418         if (IS_ERR(nodemap)) {
 419                 int rc = PTR_ERR(nodemap);
 420                 CERROR("cannot find nodemap '%s': rc = %d\n",
 421                         (char *)m->private, rc);
 422                 return rc;
 423         }
 424
 425         seq_printf(m, "%u\n", nodemap->nm_squash_gid);
 426         nodemap_putref(nodemap);
 427         return 0;
 428 }
 429
 430 /**
 431  * Reads and prints the trusted flag for the given nodemap.
 432  *
 433  * \param       m               seq file in proc fs
 434  * \param       data            unused
 435  * \retval      0               success
 436  */
 437 static int nodemap_trusted_seq_show(struct seq_file *m, void *data)
 438 {
 439         struct lu_nodemap *nodemap;
 440
 441         mutex_lock(&active_config_lock);
 442         nodemap = nodemap_lookup(m->private);
 443         mutex_unlock(&active_config_lock);
 444         if (IS_ERR(nodemap)) {
 445                 int rc = PTR_ERR(nodemap);
 446
 447                 CERROR("cannot find nodemap '%s': rc = %d\n",
 448                         (char *)m->private, rc);
 449                 return rc;
 450         }
 451
 452         seq_printf(m, "%d\n", (int)nodemap->nmf_trust_client_ids);
 453         nodemap_putref(nodemap);
 454         return 0;
 455 }
 456
 457 /**
 458  * Reads and prints the admin flag for the given nodemap.
 459  *
 460  * \param       m               seq file in proc fs
 461  * \param       data            unused
 462  * \retval      0               success
 463  */
 464 static int nodemap_admin_seq_show(struct seq_file *m, void *data)
 465 {
 466         struct lu_nodemap *nodemap;
 467         int rc;
 468
 469         mutex_lock(&active_config_lock);
 470         nodemap = nodemap_lookup(m->private);
 471         mutex_unlock(&active_config_lock);
 472         if (IS_ERR(nodemap)) {
 473                 rc = PTR_ERR(nodemap);
 474                 CERROR("cannot find nodemap '%s': rc = %d\n",
 475                         (char *)m->private, rc);
 476                 return rc;
 477         }
 478
 479         seq_printf(m, "%d\n", (int)nodemap->nmf_allow_root_access);
 480         nodemap_putref(nodemap);
 481         return 0;
 482 }
 483
 484 /**
 485  * Reads and prints the mapping mode for the given nodemap.
 486  *
 487  * \param       m               seq file in proc fs
 488  * \param       data            unused
 489  * \retval      0               success
 490  */
 491 static int nodemap_map_mode_seq_show(struct seq_file *m, void *data)
 492 {
 493         struct lu_nodemap *nodemap;
 494         int rc;
 495
 496         mutex_lock(&active_config_lock);
 497         nodemap = nodemap_lookup(m->private);
 498         mutex_unlock(&active_config_lock);
 499         if (IS_ERR(nodemap)) {
 500                 rc = PTR_ERR(nodemap);
 501                 CERROR("cannot find nodemap '%s': rc = %d\n",
 502                         (char *)m->private, rc);
 503                 return rc;
 504         }
 505
 506         if (nodemap->nmf_map_uid_only)
 507                 seq_printf(m, "uid_only\n");
 508         else if (nodemap->nmf_map_gid_only)
 509                 seq_printf(m, "gid_only\n");
 510         else
 511                 seq_printf(m, "both\n");
 512
 513         nodemap_putref(nodemap);
 514         return 0;
 515 }
 516
 517 /**
 518  * Reads and prints the deny_unknown flag for the given nodemap.
 519  *
 520  * \param       m               seq file in proc fs
 521  * \param       data            unused
 522  * \retval      0               success
 523  */
 524 static int nodemap_deny_unknown_seq_show(struct seq_file *m, void *data)
 525 {
 526         struct lu_nodemap *nodemap;
 527         int rc;
 528
 529         mutex_lock(&active_config_lock);
 530         nodemap = nodemap_lookup(m->private);
 531         mutex_unlock(&active_config_lock);
 532         if (IS_ERR(nodemap)) {
 533                 rc = PTR_ERR(nodemap);
 534                 CERROR("cannot find nodemap '%s': rc = %d\n",
 535                         (char *)m->private, rc);
 536                 return rc;
 537         }
 538
 539         seq_printf(m, "%d\n", (int)nodemap->nmf_deny_unknown);
 540         nodemap_putref(nodemap);
 541         return 0;
 542 }
 543
 544 /**
 545  * Reads and prints the audit_mode flag for the given nodemap.
 546  *
 547  * \param       m               seq file in proc fs
 548  * \param       data            unused
 549  * \retval      0               success
 550  */
 551 static int nodemap_audit_mode_seq_show(struct seq_file *m, void *data)
 552 {
 553         struct lu_nodemap *nodemap;
 554         int rc;
 555
 556         mutex_lock(&active_config_lock);
 557         nodemap = nodemap_lookup(m->private);
 558         mutex_unlock(&active_config_lock);
 559         if (IS_ERR(nodemap)) {
 560                 rc = PTR_ERR(nodemap);
 561                 CERROR("cannot find nodemap '%s': rc = %d\n",
 562                        (char *)m->private, rc);
 563                 return rc;
 564         }
 565
 566         seq_printf(m, "%d\n", (int)nodemap->nmf_enable_audit);
 567         nodemap_putref(nodemap);
 568         return 0;
 569 }
 570
 571 static struct lprocfs_vars lprocfs_nm_module_vars[] = {
 572         {
 573                 .name           = "active",
 574                 .fops           = &nodemap_active_fops,
 575         },
 576         {
 577                 NULL
 578         }
 579 };
 580
 581 LPROC_SEQ_FOPS_RO(nodemap_trusted);
 582 LPROC_SEQ_FOPS_RO(nodemap_admin);
 583 LPROC_SEQ_FOPS_RO(nodemap_squash_uid);
 584 LPROC_SEQ_FOPS_RO(nodemap_squash_gid);
 585
 586 LPROC_SEQ_FOPS_RO(nodemap_deny_unknown);
 587 LPROC_SEQ_FOPS_RO(nodemap_map_mode);
 588 LPROC_SEQ_FOPS_RO(nodemap_audit_mode);
 589
 590 const struct file_operations nodemap_ranges_fops = {
 591         .open                   = nodemap_ranges_open,
 592         .read                   = seq_read,
 593         .llseek                 = seq_lseek,
 594         .release                = single_release
 595 };
 596
 597 const struct file_operations nodemap_idmap_fops = {
 598         .open                   = nodemap_idmap_open,
 599         .read                   = seq_read,
 600         .llseek                 = seq_lseek,
 601         .release                = single_release
 602 };
 603
 604 const struct file_operations nodemap_exports_fops = {
 605         .open                   = nodemap_exports_open,
 606         .read                   = seq_read,
 607         .llseek                 = seq_lseek,
 608         .release                = single_release
 609 };
 610
 611 static struct lprocfs_vars lprocfs_nodemap_vars[] = {
 612         {
 613                 .name           = "id",
 614                 .fops           = &nodemap_id_fops,
 615         },
 616         {
 617                 .name           = "trusted_nodemap",
 618                 .fops           = &nodemap_trusted_fops,
 619         },
 620         {
 621                 .name           = "admin_nodemap",
 622                 .fops           = &nodemap_admin_fops,
 623         },
 624         {
 625                 .name           = "deny_unknown",
 626                 .fops           = &nodemap_deny_unknown_fops,
 627         },
 628         {
 629                 .name           = "map_mode",
 630                 .fops           = &nodemap_map_mode_fops,
 631         },
 632         {
 633                 .name           = "audit_mode",
 634                 .fops           = &nodemap_audit_mode_fops,
 635         },
 636         {
 637                 .name           = "squash_uid",
 638                 .fops           = &nodemap_squash_uid_fops,
 639         },
 640         {
 641                 .name           = "squash_gid",
 642                 .fops           = &nodemap_squash_gid_fops,
 643         },
 644         {
 645                 .name           = "ranges",
 646                 .fops           = &nodemap_ranges_fops,
 647         },
 648         {
 649                 .name           = "fileset",
 650                 .fops           = &nodemap_fileset_fops,
 651         },
 652         {
 653                 .name           = "exports",
 654                 .fops           = &nodemap_exports_fops,
 655         },
 656         {
 657                 .name           = "idmap",
 658                 .fops           = &nodemap_idmap_fops,
 659         },
 660         {
 661                 NULL
 662         }
 663 };
 664
 665 static struct lprocfs_vars lprocfs_default_nodemap_vars[] = {
 666         {
 667                 .name           = "id",
 668                 .fops           = &nodemap_id_fops,
 669         },
 670         {
 671                 .name           = "trusted_nodemap",
 672                 .fops           = &nodemap_trusted_fops,
 673         },
 674         {
 675                 .name           = "admin_nodemap",
 676                 .fops           = &nodemap_admin_fops,
 677         },
 678         {
 679                 .name           = "squash_uid",
 680                 .fops           = &nodemap_squash_uid_fops,
 681         },
 682         {
 683                 .name           = "squash_gid",
 684                 .fops           = &nodemap_squash_gid_fops,
 685         },
 686         {
 687                 .name           = "fileset",
 688                 .fops           = &nodemap_fileset_fops,
 689         },
 690         {
 691                 .name           = "exports",
 692                 .fops           = &nodemap_exports_fops,
 693         },
 694         {
 695                 .name           = "audit_mode",
 696                 .fops           = &nodemap_audit_mode_fops,
 697         },
 698         {
 699                 NULL
 700         }
 701 };
 702
 703 /**
 704  * Initialize the nodemap procfs directory.
 705  *
 706  * \retval      0               success
 707  */
 708 int nodemap_procfs_init(void)
 709 {
 710         int rc = 0;
 711
 712         proc_lustre_nodemap_root = lprocfs_register(LUSTRE_NODEMAP_NAME,
 713                                                     proc_lustre_root,
 714                                                     lprocfs_nm_module_vars,
 715                                                     NULL);
 716         if (IS_ERR(proc_lustre_nodemap_root)) {
 717                 rc = PTR_ERR(proc_lustre_nodemap_root);
 718                 CERROR("cannot create 'nodemap' directory: rc = %d\n",
 719                        rc);
 720                 proc_lustre_nodemap_root = NULL;
 721         }
 722         return rc;
 723 }
 724
 725 /**
 726  * Cleanup nodemap proc entry data structures.
 727  */
 728 void nodemap_procfs_exit(void)
 729 {
 730         struct nodemap_pde *nm_pde;
 731         struct nodemap_pde *tmp;
 732
 733         lprocfs_remove(&proc_lustre_nodemap_root);
 734         list_for_each_entry_safe(nm_pde, tmp, &nodemap_pde_list,
 735                                  npe_list_member) {
 736                 list_del(&nm_pde->npe_list_member);
 737                 OBD_FREE_PTR(nm_pde);
 738         }
 739 }
 740
 741 /**
 742  * Remove a nodemap's procfs entry and related data.
 743  */
 744 void lprocfs_nodemap_remove(struct nodemap_pde *nm_pde)
 745 {
 746         lprocfs_remove(&nm_pde->npe_proc_entry);
 747         list_del(&nm_pde->npe_list_member);
 748         OBD_FREE_PTR(nm_pde);
 749 }
 750
 751 /**
 752  * Register the proc directory for a nodemap
 753  *
 754  * \param       nodemap         nodemap to make the proc dir for
 755  * \param       is_default:     1 if default nodemap
 756  * \retval      0               success
 757  */
 758 int lprocfs_nodemap_register(struct lu_nodemap *nodemap, bool is_default)
 759 {
 760         struct nodemap_pde      *nm_entry;
 761         int                      rc = 0;
 762
 763         OBD_ALLOC_PTR(nm_entry);
 764         if (nm_entry == NULL)
 765                 GOTO(out, rc = -ENOMEM);
 766
 767         nm_entry->npe_proc_entry = proc_mkdir(nodemap->nm_name,
 768                                               proc_lustre_nodemap_root);
 769         if (IS_ERR(nm_entry->npe_proc_entry))
 770                 GOTO(out, rc = PTR_ERR(nm_entry->npe_proc_entry));
 771
 772         snprintf(nm_entry->npe_name, sizeof(nm_entry->npe_name), "%s",
 773                  nodemap->nm_name);
 774
 775         /* Use the nodemap name as stored on the PDE as the private data. This
 776          * is so a nodemap struct can be replaced without updating the proc
 777          * entries.
 778          */
 779         rc = lprocfs_add_vars(nm_entry->npe_proc_entry,
 780                               (is_default ? lprocfs_default_nodemap_vars :
 781                                             lprocfs_nodemap_vars),
 782                               nm_entry->npe_name);
 783         if (rc != 0)
 784                 lprocfs_remove(&nm_entry->npe_proc_entry);
 785         else
 786                 list_add(&nm_entry->npe_list_member, &nodemap_pde_list);
 787
 788 out:
 789         if (rc != 0) {
 790                 CERROR("cannot create 'nodemap/%s': rc = %d\n",
 791                        nodemap->nm_name, rc);
 792                 if (nm_entry != NULL) {
 793                         OBD_FREE_PTR(nm_entry);
 794                         nm_entry = NULL;
 795                 }
 796         }
 797
 798         nodemap->nm_pde_data = nm_entry;
 799
 800         return rc;
 801 }