lustre/ptlrpc/nodemap_internal.h

   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) 2013, 2014, Intel Corporation.
  26  *
  27  * Author: Joshua Walgenbach <jjw@iu.edu>
  28  */
  29
  30 #ifndef _NODEMAP_INTERNAL_H
  31 #define _NODEMAP_INTERNAL_H
  32
  33 #include <lustre_nodemap.h>
  34 #include <interval_tree.h>
  35
  36 #define DEFAULT_NODEMAP "default"
  37
  38 /* Turn on proc debug interface to allow OSS and
  39  * MDS nodes to configure nodemap independently of
  40  * MGS (since the nodemap distribution is not written
  41  * yet */
  42 #define NODEMAP_PROC_DEBUG 1
  43
  44 /* Default nobody uid and gid values */
  45
  46 #define NODEMAP_NOBODY_UID 99
  47 #define NODEMAP_NOBODY_GID 99
  48
  49 struct lprocfs_static_vars;
  50
  51 /* nodemap root proc directory under fs/lustre */
  52 extern struct proc_dir_entry *proc_lustre_nodemap_root;
  53 /* flag if nodemap is active */
  54 extern bool nodemap_active;
  55
  56 extern struct mutex active_config_lock;
  57 extern struct nodemap_config *active_config;
  58
  59 struct lu_nid_range {
  60         /* unique id set by mgs */
  61         unsigned int             rn_id;
  62         /* lu_nodemap containing this range */
  63         struct lu_nodemap       *rn_nodemap;
  64         /* list for nodemap */
  65         struct list_head         rn_list;
  66         /* nid interval tree */
  67         struct interval_node     rn_node;
  68 };
  69
  70 struct lu_idmap {
  71         /* uid/gid of client */
  72         __u32           id_client;
  73         /* uid/gid on filesystem */
  74         __u32           id_fs;
  75         /* tree mapping client ids to filesystem ids */
  76         struct rb_node  id_client_to_fs;
  77         /* tree mappung filesystem to client */
  78         struct rb_node  id_fs_to_client;
  79 };
  80
  81 /* first 4 bits of the nodemap_id is the index type */
  82 struct nodemap_key {
  83         __u32 nk_nodemap_id;
  84         union {
  85                 __u32 nk_range_id;
  86                 __u32 nk_id_client;
  87                 __u32 nk_unused;
  88         };
  89 };
  90
  91 enum nodemap_idx_type {
  92         NODEMAP_EMPTY_IDX = 0,          /* index created with blank record */
  93         NODEMAP_CLUSTER_IDX = 1,        /* a nodemap cluster of nodes */
  94         NODEMAP_RANGE_IDX = 2,          /* nid range assigned to a nm cluster */
  95         NODEMAP_UIDMAP_IDX = 3,         /* uid map assigned to a nm cluster */
  96         NODEMAP_GIDMAP_IDX = 4,         /* gid map assigned to a nm cluster */
  97         NODEMAP_GLOBAL_IDX = 15,        /* stores nodemap activation status */
  98 };
  99
 100 #define NM_TYPE_MASK 0x0FFFFFFF
 101 #define NM_TYPE_SHIFT 28
 102
 103 static inline enum nodemap_idx_type nm_idx_get_type(unsigned int id)
 104 {
 105         return id >> NM_TYPE_SHIFT;
 106 }
 107
 108 static inline __u32 nm_idx_set_type(unsigned int id, enum nodemap_idx_type t)
 109 {
 110         return (id & NM_TYPE_MASK) | (t << NM_TYPE_SHIFT);
 111 }
 112
 113 struct lu_nodemap *nodemap_create(const char *name,
 114                                   struct nodemap_config *config,
 115                                   bool is_default);
 116 void nodemap_putref(struct lu_nodemap *nodemap);
 117 struct lu_nodemap *nodemap_lookup(const char *name);
 118
 119 int nodemap_procfs_init(void);
 120 void nodemap_procfs_exit(void);
 121 int lprocfs_nodemap_register(struct lu_nodemap *nodemap,
 122                              bool is_default_nodemap);
 123 void lprocfs_nodemap_remove(struct nodemap_pde *nodemap_pde);
 124 struct lu_nid_range *nodemap_range_find(lnet_nid_t start_nid,
 125                                         lnet_nid_t end_nid);
 126 struct lu_nid_range *range_create(struct nodemap_range_tree *nm_range_tree,
 127                                   lnet_nid_t start_nid, lnet_nid_t end_nid,
 128                                   struct lu_nodemap *nodemap,
 129                                   unsigned int range_id);
 130 void range_destroy(struct lu_nid_range *range);
 131 int range_insert(struct nodemap_range_tree *nm_range_tree,
 132                  struct lu_nid_range *data);
 133 void range_delete(struct nodemap_range_tree *nm_range_tree,
 134                   struct lu_nid_range *data);
 135 struct lu_nid_range *range_search(struct nodemap_range_tree *nm_range_tree,
 136                                   lnet_nid_t nid);
 137 struct lu_nid_range *range_find(struct nodemap_range_tree *nm_range_tree,
 138                                 lnet_nid_t start_nid, lnet_nid_t end_nid);
 139 int range_parse_nidstring(char *range_string, lnet_nid_t *start_nid,
 140                           lnet_nid_t *end_nid);
 141 void range_init_tree(void);
 142 struct lu_idmap *idmap_create(__u32 client_id, __u32 fs_id);
 143 void idmap_insert(enum nodemap_id_type id_type, struct lu_idmap *idmap,
 144                  struct lu_nodemap *nodemap);
 145 void idmap_delete(enum nodemap_id_type id_type,  struct lu_idmap *idmap,
 146                   struct lu_nodemap *nodemap);
 147 void idmap_delete_tree(struct lu_nodemap *nodemap);
 148 struct lu_idmap *idmap_search(struct lu_nodemap *nodemap,
 149                               enum nodemap_tree_type,
 150                               enum nodemap_id_type id_type,
 151                               __u32 id);
 152 int nm_member_add(struct lu_nodemap *nodemap, struct obd_export *exp);
 153 void nm_member_del(struct lu_nodemap *nodemap, struct obd_export *exp);
 154 void nm_member_delete_list(struct lu_nodemap *nodemap);
 155 struct lu_nodemap *nodemap_classify_nid(lnet_nid_t nid);
 156 void nm_member_reclassify_nodemap(struct lu_nodemap *nodemap);
 157 void nm_member_revoke_locks(struct lu_nodemap *nodemap);
 158 void nm_member_revoke_all(void);
 159
 160 int nodemap_add_idmap_helper(struct lu_nodemap *nodemap,
 161                              enum nodemap_id_type id_type,
 162                              const __u32 map[2]);
 163 int nodemap_add_range_helper(struct nodemap_config *config,
 164                              struct lu_nodemap *nodemap,
 165                              const lnet_nid_t nid[2],
 166                              unsigned int range_id);
 167
 168 struct rb_node *nm_rb_next_postorder(const struct rb_node *node);
 169 struct rb_node *nm_rb_first_postorder(const struct rb_root *root);
 170 void nodemap_getref(struct lu_nodemap *nodemap);
 171 void nodemap_putref(struct lu_nodemap *nodemap);
 172
 173 #define nm_rbtree_postorder_for_each_entry_safe(pos, n,                 \
 174                                                 root, field)            \
 175         for (pos = nm_rb_first_postorder(root) ?                        \
 176                 rb_entry(nm_rb_first_postorder(root), typeof(*pos),     \
 177                 field) : NULL,                                          \
 178                 n = (pos && nm_rb_next_postorder(&pos->field)) ?        \
 179                 rb_entry(nm_rb_next_postorder(&pos->field),             \
 180                 typeof(*pos), field) : NULL;                            \
 181                 pos != NULL;                                            \
 182                 pos = n,                                                \
 183                 n = (pos && nm_rb_next_postorder(&pos->field)) ?        \
 184                 rb_entry(nm_rb_next_postorder(&pos->field),             \
 185                 typeof(*pos), field) : NULL)
 186
 187 int nodemap_idx_nodemap_add(const struct lu_nodemap *nodemap);
 188 int nodemap_idx_nodemap_update(const struct lu_nodemap *nodemap);
 189 int nodemap_idx_nodemap_del(const struct lu_nodemap *nodemap);
 190 int nodemap_idx_idmap_add(const struct lu_nodemap *nodemap,
 191                           enum nodemap_id_type id_type,
 192                           const __u32 map[2]);
 193 int nodemap_idx_idmap_del(const struct lu_nodemap *nodemap,
 194                           enum nodemap_id_type id_type,
 195                           const __u32 map[2]);
 196 int nodemap_idx_range_add(const struct lu_nid_range *range,
 197                           const lnet_nid_t nid[2]);
 198 int nodemap_idx_range_del(const struct lu_nid_range *range);
 199 int nodemap_idx_nodemap_activate(bool value);
 200 #endif  /* _NODEMAP_INTERNAL_H */