+/**
+ * Format and send the ioctl to the MGS.
+ *
+ * \param cmd IOCTL to send
+ * \param ret_data void pointer to return anything from
+ * ioctl
+ * \param num_args number of arguments to pack into the
+ * ioctl buffer
+ * \param argv[] variable number of string arguments
+ *
+ * \retval 0 on success
+ */
+static int nodemap_cmd(enum lcfg_command_type cmd, void *ret_data,
+ unsigned int ret_size, ...)
+{
+ va_list ap;
+ char *arg;
+ int i = 0;
+ struct lustre_cfg_bufs bufs;
+ struct obd_ioctl_data data;
+ struct lustre_cfg *lcfg;
+ char rawbuf[MAX_IOC_BUFLEN];
+ char *buf = rawbuf;
+ int rc = 0;
+
+ lustre_cfg_bufs_reset(&bufs, NULL);
+
+ va_start(ap, ret_size);
+ arg = va_arg(ap, char *);
+ while (arg != NULL) {
+ lustre_cfg_bufs_set_string(&bufs, i, arg);
+ i++;
+ arg = va_arg(ap, char *);
+ }
+ va_end(ap);
+
+ lcfg = lustre_cfg_new(cmd, &bufs);
+ if (lcfg == NULL)
+ return -ENOMEM;
+
+ memset(&data, 0, sizeof(data));
+ rc = data.ioc_dev = get_mgs_device();
+ if (rc < 0)
+ goto out;
+
+ data.ioc_type = LUSTRE_CFG_TYPE;
+ data.ioc_plen1 = lustre_cfg_len(lcfg->lcfg_bufcount,
+ lcfg->lcfg_buflens);
+ data.ioc_pbuf1 = (void *)lcfg;
+
+ memset(buf, 0, sizeof(rawbuf));
+ rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
+ if (rc != 0) {
+ fprintf(stderr, "error: invalid ioctl: %08x errno: %d with "
+ "rc=%d\n", cmd, errno, rc);
+ goto out;
+ }
+
+ rc = l_ioctl(OBD_DEV_ID, OBD_IOC_NODEMAP, buf);
+ if (rc != 0) {
+ fprintf(stderr, "error: invalid ioctl: %08x errno: %d with "
+ "rc=%d\n", cmd, errno, rc);
+ goto out;
+ }
+
+ if (ret_data != NULL) {
+ rc = obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
+ if (rc != 0)
+ goto out;
+
+ memcpy(ret_data, data.ioc_pbuf1, data.ioc_plen1);
+ if (ret_data == NULL || sizeof(ret_data) != ret_size)
+ rc = -errno;
+ }
+out:
+ lustre_cfg_free(lcfg);
+
+ return rc;
+}
+
+/**
+ * activate nodemap functions
+ *
+ * \param argc number of args
+ * \param argv[] variable string arguments
+ *
+ * argv[0] 1 for activate or 0 for deactivate
+ *
+ * \retval 0 on success
+ */
+int jt_nodemap_activate(int argc, char **argv)
+{
+ int rc;
+
+ rc = nodemap_cmd(LCFG_NODEMAP_ACTIVATE, NULL, 0, argv[0], argv[1],
+ NULL);
+
+ if (rc != 0) {
+ errno = -rc;
+ perror(argv[0]);
+ }
+
+ return rc;
+}
+
+/**
+ * add a nodemap
+ *
+ * \param argc number of args
+ * \param argv[] variable string arguments
+ *
+ * argv[0] nodemap name
+ *
+ * \retval 0 on success
+ */
+int jt_nodemap_add(int argc, char **argv)
+{
+ int rc;
+
+ rc = llapi_nodemap_exists(argv[1]);
+ if (rc == 0) {
+ fprintf(stderr, "error: %s existing nodemap name\n", argv[1]);
+ return 1;
+ }
+
+ rc = nodemap_cmd(LCFG_NODEMAP_ADD, NULL, 0, argv[0], argv[1], NULL);
+
+ if (rc != 0) {
+ errno = -rc;
+ perror(argv[0]);
+ }
+
+ return rc;
+}
+
+/**
+ * delete a nodemap
+ *
+ * \param argc number of args
+ * \param argv[] variable string arguments
+ *
+ * argv[0] nodemap name
+ *
+ * \retval 0 on success
+ */
+int jt_nodemap_del(int argc, char **argv)
+{
+ int rc;
+
+ rc = llapi_nodemap_exists(argv[1]);
+ if (rc != 0) {
+ fprintf(stderr, "error: %s not existing nodemap name\n",
+ argv[1]);
+ return rc;
+ }
+ rc = nodemap_cmd(LCFG_NODEMAP_DEL, NULL, 0, argv[0], argv[1], NULL);
+
+ if (rc != 0) {
+ errno = -rc;
+ perror(argv[0]);
+ }
+
+ return rc;
+}
+
+/**
+ * test a nid for nodemap membership
+ *
+ * \param argc number of args
+ * \param argv[] variable string arguments
+ *
+ * argv[0] properly formatted nid
+ *
+ * \retval 0 on success
+ */
+int jt_nodemap_test_nid(int argc, char **argv)
+{
+
+ char rawbuf[MAX_IOC_BUFLEN];
+ int rc;
+
+ rc = nodemap_cmd(LCFG_NODEMAP_TEST_NID, &rawbuf, sizeof(rawbuf),
+ argv[0], argv[1], NULL);
+ if (rc == 0)
+ printf("%s\n", (char *)rawbuf);
+
+ return rc;
+}
+
+/**
+ * test a nodemap id pair for mapping
+ *
+ * \param argc number of args
+ * \param argv[[] variable string arguments
+ *
+ * \retval 0 on success
+ *
+ * The argv array should contain the nodemap name, the id
+ * to checking the mapping on, and the id type (UID or GID)
+ *
+ */
+int jt_nodemap_test_id(int argc, char **argv)
+{
+ char rawbuf[MAX_IOC_BUFLEN];
+ char *nidstr = NULL;
+ char *idstr = NULL;
+ char *typestr = NULL;
+ int rc = 0;
+ int c;
+
+ static struct option long_options[] = {
+ {
+ .name = "nid",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'n',
+ },
+ {
+ .name = "idtype",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 't',
+ },
+ {
+ .name = "id",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'i',
+ },
+ {
+ NULL
+ }
+ };
+
+ while ((c = getopt_long(argc, argv, "n:t:i:",
+ long_options, NULL)) != -1) {
+ switch (c) {
+ case 'n':
+ nidstr = optarg;
+ break;
+ case 't':
+ typestr = optarg;
+ break;
+ case 'i':
+ idstr = optarg;
+ break;
+ }
+ }
+
+ if (nidstr == NULL || typestr == NULL || idstr == NULL) {
+ fprintf(stderr, "usage: nodemap_test_id --nid <nid> "
+ "--idtype [uid|gid] --id <id>\n");
+ return -1;
+ }
+
+ rc = nodemap_cmd(LCFG_NODEMAP_TEST_ID, &rawbuf, sizeof(rawbuf),
+ argv[0], nidstr, typestr, idstr);
+ if (rc == 0)
+ printf("%s\n", (char *)rawbuf);
+
+ return rc;
+}
+
+/**
+ * add an nid range to a nodemap
+ *
+ * \param argc number of args
+ * \param argv[] variable string arguments
+ *
+ * --name nodemap name
+ * --range properly formatted nid range
+ *
+ * \retval 0 on success
+ */
+int jt_nodemap_add_range(int argc, char **argv)
+{
+ char *nodemap_name = NULL;
+ char *nodemap_range = NULL;
+ struct list_head nidlist;
+ char min_nid[LNET_NIDSTR_SIZE + 1];
+ char max_nid[LNET_NIDSTR_SIZE + 1];
+ char nid_range[2 * LNET_NIDSTR_SIZE + 2];
+ int rc = 0;
+ int c;
+
+ static struct option long_options[] = {
+ {
+ .name = "name",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'n',
+ },
+ {
+ .name = "range",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'r',
+ },
+ {
+ NULL
+ }
+ };
+
+ INIT_LIST_HEAD(&nidlist);
+
+ while ((c = getopt_long(argc, argv, "n:r:",
+ long_options, NULL)) != -1) {
+ switch (c) {
+ case 'n':
+ nodemap_name = optarg;
+ break;
+ case 'r':
+ nodemap_range = optarg;
+ break;
+ }
+ }
+
+ if (nodemap_name == NULL || nodemap_range == NULL) {
+ fprintf(stderr, "usage: nodemap_add_range --name <name> "
+ "--range <range>\n");
+ return -1;
+ }
+
+ if (cfs_parse_nidlist(nodemap_range, strlen(nodemap_range),
+ &nidlist) <= 0) {
+ fprintf(stderr, "error: %s: can't parse nid range: %s\n",
+ jt_cmdname(argv[0]), nodemap_range);
+ return -1;
+ }
+
+ if (!cfs_nidrange_is_contiguous(&nidlist)) {
+ fprintf(stderr, "error: %s: nodemap ranges must be "
+ "contiguous\n", jt_cmdname(argv[0]));
+ return -1;
+ }
+
+ cfs_nidrange_find_min_max(&nidlist, &min_nid[0], &max_nid[0],
+ LNET_NIDSTR_SIZE);
+ snprintf(nid_range, sizeof(nid_range), "%s:%s", min_nid, max_nid);
+
+ rc = nodemap_cmd(LCFG_NODEMAP_ADD_RANGE, NULL, 0, argv[0],
+ nodemap_name, nid_range, NULL);
+ if (rc != 0) {
+ errno = -rc;
+ fprintf(stderr, "error: %s: cannot add range '%s' to nodemap "
+ "'%s': rc = %d\n",
+ jt_cmdname(argv[0]), nodemap_range, nodemap_name, rc);
+ }
+
+ return rc;
+}
+
+/**
+ * delete an nid range to a nodemap
+ *
+ * \param argc number of args
+ * \param argv[] variable string arguments
+ *
+ * --name nodemap name
+ * --range properly formatted nid range
+ *
+ * \retval 0 on success
+ */
+int jt_nodemap_del_range(int argc, char **argv)
+{
+ char *nodemap_name = NULL;
+ char *nodemap_range = NULL;
+ struct list_head nidlist;
+ char min_nid[LNET_NIDSTR_SIZE + 1];
+ char max_nid[LNET_NIDSTR_SIZE + 1];
+ char nid_range[2 * LNET_NIDSTR_SIZE + 2];
+ int rc = 0;
+ int c;
+
+ static struct option long_options[] = {
+ {
+ .name = "name",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'n',
+ },
+ {
+ .name = "range",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'r',
+ },
+ {
+ NULL
+ }
+ };
+
+ INIT_LIST_HEAD(&nidlist);
+
+ while ((c = getopt_long(argc, argv, "n:r:",
+ long_options, NULL)) != -1) {
+ switch (c) {
+ case 'n':
+ nodemap_name = optarg;
+ break;
+ case 'r':
+ nodemap_range = optarg;
+ break;
+ }
+ }
+
+ if (nodemap_name == NULL || nodemap_range == NULL) {
+ fprintf(stderr, "usage: nodemap_del_range --name <name> "
+ "--range <range>\n");
+ return -1;
+ }
+
+ if (cfs_parse_nidlist(nodemap_range, strlen(nodemap_range),
+ &nidlist) <= 0) {
+ fprintf(stderr, "error: %s: can't parse nid range: %s\n",
+ jt_cmdname(argv[0]), nodemap_range);
+ return -1;
+ }
+
+ if (!cfs_nidrange_is_contiguous(&nidlist)) {
+ fprintf(stderr, "error: %s: nodemap ranges must be "
+ "contiguous\n", jt_cmdname(argv[0]));
+ return -1;
+ }
+
+ cfs_nidrange_find_min_max(&nidlist, &min_nid[0], &max_nid[0],
+ LNET_NIDSTR_SIZE);
+ snprintf(nid_range, sizeof(nid_range), "%s:%s", min_nid, max_nid);
+
+ rc = nodemap_cmd(LCFG_NODEMAP_DEL_RANGE, NULL, 0, argv[0],
+ nodemap_name, nid_range, NULL);
+ if (rc != 0) {
+ errno = -rc;
+ fprintf(stderr, "error: %s: cannot delete range '%s' to "
+ "nodemap '%s': rc = %d\n",
+ jt_cmdname(argv[0]), nodemap_range, nodemap_name, rc);
+ }
+
+ return rc;
+}
+
+/**
+ * modify a nodemap's behavior
+ *
+ * \param argc number of args
+ * \param argv[] variable string arguments
+ *
+ * --name nodemap name
+ * --property nodemap property to change
+ * admin, trusted, squash_uid, squash_gid)
+ * --value value to set property
+ *
+ * \retval 0 on success
+ */
+int jt_nodemap_modify(int argc, char **argv)
+{
+ int c;
+ int rc = 0;
+ enum lcfg_command_type cmd = 0;
+ char *nodemap_name = NULL;
+ char *param = NULL;
+ char *value = NULL;
+
+ static struct option long_options[] = {
+ {
+ .name = "name",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'n',
+ },
+ {
+ .name = "property",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'p',
+ },
+ {
+ .name = "value",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'v',
+ },
+ {
+ NULL
+ }
+ };
+
+ while ((c = getopt_long(argc, argv, "n:p:v:",
+ long_options, NULL)) != -1) {
+ switch (c) {
+ case 'n':
+ nodemap_name = optarg;
+ break;
+ case 'p':
+ param = optarg;
+ break;
+ case 'v':
+ value = optarg;
+ break;
+ }
+ }
+
+ if (nodemap_name == NULL || param == NULL || value == NULL) {
+ fprintf(stderr, "usage: nodemap_modify --name <name> "
+ "--property <range> --value <value>\n");
+ return -1;
+ }
+
+ if (strcmp("admin", param) == 0) {
+ cmd = LCFG_NODEMAP_ADMIN;
+ } else if (strcmp("trusted", param) == 0) {
+ cmd = LCFG_NODEMAP_TRUSTED;
+ } else if (strcmp("squash_uid", param) == 0) {
+ cmd = LCFG_NODEMAP_SQUASH_UID;
+ } else if (strcmp("squash_gid", param) == 0) {
+ cmd = LCFG_NODEMAP_SQUASH_GID;
+ } else {
+ fprintf(stderr, "error: %s: nodemap_modify invalid "
+ "subcommand: %s\n",
+ jt_cmdname(argv[0]), param);
+ return -1;
+ }
+
+ rc = nodemap_cmd(cmd, NULL, 0, argv[0], nodemap_name, param,
+ value, NULL);
+ if (rc != 0) {
+ errno = -rc;
+ fprintf(stderr, "error: %s: cannot modify nodemap '%s' "
+ "to param '%s': value '%s': rc = %d\n",
+ jt_cmdname(argv[0]), nodemap_name, param, value, rc);
+ }
+
+ return rc;
+}
+
+int jt_nodemap_add_idmap(int argc, char **argv)
+{
+ int c;
+ enum lcfg_command_type cmd = 0;
+ char *nodemap_name = NULL;
+ char *idmap = NULL;
+ char *idtype = NULL;
+ int rc = 0;
+
+ static struct option long_options[] = {
+ {
+ .name = "name",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'n',
+ },
+ {
+ .name = "idmap",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'm',
+ },
+ {
+ .name = "idtype",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'i',
+ },
+ {
+ NULL
+ }
+ };
+
+ while ((c = getopt_long(argc, argv, "n:m:i:",
+ long_options, NULL)) != -1) {
+ switch (c) {
+ case 'n':
+ nodemap_name = optarg;
+ break;
+ case 'm':
+ idmap = optarg;
+ break;
+ case 'i':
+ idtype = optarg;
+ break;
+ }
+ }
+
+ if (nodemap_name == NULL || idmap == NULL || idtype == NULL) {
+ fprintf(stderr, "usage: %s --name <name> --idtype [uid | gid]"
+ " --idmap <client id>:<filesystem id>\n", argv[0]);
+ return -1;
+ }
+
+ if (strcmp("uid", idtype) == 0) {
+ cmd = LCFG_NODEMAP_ADD_UIDMAP;
+ } else if (strcmp("gid", idtype) == 0) {
+ cmd = LCFG_NODEMAP_ADD_GIDMAP;
+ } else {
+ fprintf(stderr, "usage: %s --name <name> --idtype [uid | gid]"
+ " --idmap <client id>:<filesystem id>\n", argv[0]);
+ return -1;
+ }
+
+ rc = nodemap_cmd(cmd, NULL, 0, argv[0], nodemap_name, idmap, NULL);
+ if (rc != 0) {
+ errno = -rc;
+ fprintf(stderr, "cannot add %smap '%s' to nodemap '%s'"
+ ": rc = %d\n", idtype, idmap, nodemap_name, rc);
+ }
+
+ return rc;
+}
+
+int jt_nodemap_del_idmap(int argc, char **argv)
+{
+ int c;
+ enum lcfg_command_type cmd = 0;
+ char *nodemap_name = NULL;
+ char *idmap = NULL;
+ char *idtype = NULL;
+ int rc = 0;
+
+ static struct option long_options[] = {
+ {
+ .name = "name",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'n',
+ },
+ {
+ .name = "idmap",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'm',
+ },
+ {
+ .name = "idtype",
+ .has_arg = required_argument,
+ .flag = 0,
+ .val = 'i',
+ },
+ {
+ NULL
+ }
+ };
+
+ while ((c = getopt_long(argc, argv, "n:m:i:",
+ long_options, NULL)) != -1) {
+ switch (c) {
+ case 'n':
+ nodemap_name = optarg;
+ break;
+ case 'm':
+ idmap = optarg;
+ break;
+ case 'i':
+ idtype = optarg;
+ break;
+ }
+ }
+
+ if (nodemap_name == NULL || idmap == NULL || idtype == NULL) {
+ fprintf(stderr, "usage: %s --name <name> --idtype [uid | gid]"
+ " --idmap <client id>:<filesystem id>\n", argv[0]);
+ return -1;
+ }
+
+ if (strcmp("uid", idtype) == 0)
+ cmd = LCFG_NODEMAP_DEL_UIDMAP;
+ else
+ cmd = LCFG_NODEMAP_DEL_GIDMAP;
+
+ rc = nodemap_cmd(cmd, NULL, 0, argv[0], nodemap_name, idmap, NULL);
+ if (rc != 0) {
+ errno = -rc;
+ fprintf(stderr, "cannot add %smap '%s' to nodemap '%s'"
+ ": rc = %d\n", idtype, idmap, nodemap_name, rc);
+ }
+
+ return rc;
+}
+