lustre/utils/liblustreapi.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.sun.com/software/products/lustre/docs/GPLv2.pdf
  19  *
  20  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  21  * CA 95054 USA or visit www.sun.com if you need additional information or
  22  * have any questions.
  23  *
  24  * GPL HEADER END
  25  */
  26 /*
  27  * Copyright (c) 2003, 2010, Oracle and/or its affiliates. All rights reserved.
  28  * Use is subject to license terms.
  29  *
  30  * Copyright (c) 2011, 2014, Intel Corporation.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  *
  36  * lustre/utils/liblustreapi.c
  37  *
  38  * Author: Peter J. Braam <braam@clusterfs.com>
  39  * Author: Phil Schwan <phil@clusterfs.com>
  40  * Author: Robert Read <rread@clusterfs.com>
  41  */
  42
  43 /* for O_DIRECTORY */
  44 #ifndef _GNU_SOURCE
  45 #define _GNU_SOURCE
  46 #endif
  47
  48 #include <mntent.h>
  49 #include <stdlib.h>
  50 #include <stdio.h>
  51 #include <string.h>
  52 #include <stddef.h>
  53 #include <sys/ioctl.h>
  54 #include <unistd.h>
  55 #include <fcntl.h>
  56 #include <errno.h>
  57 #include <dirent.h>
  58 #include <stdarg.h>
  59 #include <sys/stat.h>
  60 #include <sys/statfs.h>
  61 #include <sys/types.h>
  62 #include <sys/syscall.h>
  63 #include <sys/xattr.h>
  64 #include <fnmatch.h>
  65 #include <glob.h>
  66 #include <libgen.h> /* for dirname() */
  67 #ifdef HAVE_LINUX_UNISTD_H
  68 #include <linux/unistd.h>
  69 #else
  70 #include <unistd.h>
  71 #endif
  72 #include <poll.h>
  73
  74 #include <libcfs/libcfs.h>
  75 #include <lnet/lnetctl.h>
  76 #include <lustre/lustreapi.h>
  77 #include <lustre_ioctl.h>
  78 #include "lustreapi_internal.h"
  79
  80 static int llapi_msg_level = LLAPI_MSG_MAX;
  81
  82 void llapi_msg_set_level(int level)
  83 {
  84         /* ensure level is in the good range */
  85         if (level < LLAPI_MSG_OFF)
  86                 llapi_msg_level = LLAPI_MSG_OFF;
  87         else if (level > LLAPI_MSG_MAX)
  88                 llapi_msg_level = LLAPI_MSG_MAX;
  89         else
  90                 llapi_msg_level = level;
  91 }
  92
  93 int llapi_msg_get_level(void)
  94 {
  95         return llapi_msg_level;
  96 }
  97
  98 static void error_callback_default(enum llapi_message_level level, int err,
  99                                    const char *fmt, va_list ap)
 100 {
 101         vfprintf(stderr, fmt, ap);
 102         if (level & LLAPI_MSG_NO_ERRNO)
 103                 fprintf(stderr, "\n");
 104         else
 105                 fprintf(stderr, ": %s (%d)\n", strerror(err), err);
 106 }
 107
 108 static void info_callback_default(enum llapi_message_level level, int err,
 109                                   const char *fmt, va_list ap)
 110 {
 111         vfprintf(stdout, fmt, ap);
 112 }
 113
 114 static llapi_log_callback_t llapi_error_callback = error_callback_default;
 115 static llapi_log_callback_t llapi_info_callback = info_callback_default;
 116
 117
 118 /* llapi_error will preserve errno */
 119 void llapi_error(enum llapi_message_level level, int err, const char *fmt, ...)
 120 {
 121         va_list  args;
 122         int      tmp_errno = errno;
 123
 124         if ((level & LLAPI_MSG_MASK) > llapi_msg_level)
 125                 return;
 126
 127         va_start(args, fmt);
 128         llapi_error_callback(level, abs(err), fmt, args);
 129         va_end(args);
 130         errno = tmp_errno;
 131 }
 132
 133 /* llapi_printf will preserve errno */
 134 void llapi_printf(enum llapi_message_level level, const char *fmt, ...)
 135 {
 136         va_list  args;
 137         int      tmp_errno = errno;
 138
 139         if ((level & LLAPI_MSG_MASK) > llapi_msg_level)
 140                 return;
 141
 142         va_start(args, fmt);
 143         llapi_info_callback(level, 0, fmt, args);
 144         va_end(args);
 145         errno = tmp_errno;
 146 }
 147
 148 /**
 149  * Set a custom error logging function. Passing in NULL will reset the logging
 150  * callback to its default value.
 151  *
 152  * This function returns the value of the old callback.
 153  */
 154 llapi_log_callback_t llapi_error_callback_set(llapi_log_callback_t cb)
 155 {
 156         llapi_log_callback_t    old = llapi_error_callback;
 157
 158         if (cb != NULL)
 159                 llapi_error_callback = cb;
 160         else
 161                 llapi_error_callback = error_callback_default;
 162
 163         return old;
 164 }
 165
 166 /**
 167  * Set a custom info logging function. Passing in NULL will reset the logging
 168  * callback to its default value.
 169  *
 170  * This function returns the value of the old callback.
 171  */
 172 llapi_log_callback_t llapi_info_callback_set(llapi_log_callback_t cb)
 173 {
 174         llapi_log_callback_t    old = llapi_info_callback;
 175
 176         if (cb != NULL)
 177                 llapi_info_callback = cb;
 178         else
 179                 llapi_info_callback = info_callback_default;
 180
 181         return old;
 182 }
 183
 184 /**
 185  * size_units is to be initialized (or zeroed) by caller.
 186  */
 187 int llapi_parse_size(const char *optarg, unsigned long long *size,
 188                      unsigned long long *size_units, int bytes_spec)
 189 {
 190         char *end;
 191
 192         if (strncmp(optarg, "-", 1) == 0)
 193                 return -1;
 194
 195         if (*size_units == 0)
 196                 *size_units = 1;
 197
 198         *size = strtoull(optarg, &end, 0);
 199
 200         if (*end != '\0') {
 201                 if ((*end == 'b') && *(end + 1) == '\0' &&
 202                     (*size & (~0ULL << (64 - 9))) == 0 &&
 203                     !bytes_spec) {
 204                         *size_units = 1 << 9;
 205                 } else if ((*end == 'b') &&
 206                            *(end + 1) == '\0' &&
 207                            bytes_spec) {
 208                         *size_units = 1;
 209                 } else if ((*end == 'k' || *end == 'K') &&
 210                            *(end + 1) == '\0' &&
 211                            (*size & (~0ULL << (64 - 10))) == 0) {
 212                         *size_units = 1 << 10;
 213                 } else if ((*end == 'm' || *end == 'M') &&
 214                            *(end + 1) == '\0' &&
 215                            (*size & (~0ULL << (64 - 20))) == 0) {
 216                         *size_units = 1 << 20;
 217                 } else if ((*end == 'g' || *end == 'G') &&
 218                            *(end + 1) == '\0' &&
 219                            (*size & (~0ULL << (64 - 30))) == 0) {
 220                         *size_units = 1 << 30;
 221                 } else if ((*end == 't' || *end == 'T') &&
 222                            *(end + 1) == '\0' &&
 223                            (*size & (~0ULL << (64 - 40))) == 0) {
 224                         *size_units = 1ULL << 40;
 225                 } else if ((*end == 'p' || *end == 'P') &&
 226                            *(end + 1) == '\0' &&
 227                            (*size & (~0ULL << (64 - 50))) == 0) {
 228                         *size_units = 1ULL << 50;
 229                 } else if ((*end == 'e' || *end == 'E') &&
 230                            *(end + 1) == '\0' &&
 231                            (*size & (~0ULL << (64 - 60))) == 0) {
 232                         *size_units = 1ULL << 60;
 233                 } else {
 234                         return -1;
 235                 }
 236         }
 237         *size *= *size_units;
 238         return 0;
 239 }
 240
 241 /* XXX: llapi_xxx() functions return negative values upon failure */
 242
 243 int llapi_stripe_limit_check(unsigned long long stripe_size, int stripe_offset,
 244                                 int stripe_count, int stripe_pattern)
 245 {
 246         int page_size, rc;
 247
 248         /* 64 KB is the largest common page size I'm aware of (on ia64), but
 249          * check the local page size just in case. */
 250         page_size = LOV_MIN_STRIPE_SIZE;
 251         if (getpagesize() > page_size) {
 252                 page_size = getpagesize();
 253                 llapi_err_noerrno(LLAPI_MSG_WARN,
 254                                 "warning: your page size (%u) is "
 255                                 "larger than expected (%u)", page_size,
 256                                 LOV_MIN_STRIPE_SIZE);
 257         }
 258         if (!llapi_stripe_size_is_aligned(stripe_size)) {
 259                 rc = -EINVAL;
 260                 llapi_error(LLAPI_MSG_ERROR, rc, "error: bad stripe_size %llu, "
 261                                 "must be an even multiple of %d bytes",
 262                                 stripe_size, page_size);
 263                 return rc;
 264         }
 265         if (!llapi_stripe_offset_is_valid(stripe_offset)) {
 266                 rc = -EINVAL;
 267                 llapi_error(LLAPI_MSG_ERROR, rc, "error: bad stripe offset %d",
 268                                 stripe_offset);
 269                 return rc;
 270         }
 271         if (!llapi_stripe_count_is_valid(stripe_count)) {
 272                 rc = -EINVAL;
 273                 llapi_error(LLAPI_MSG_ERROR, rc, "error: bad stripe count %d",
 274                                 stripe_count);
 275                 return rc;
 276         }
 277         if (llapi_stripe_size_is_too_big(stripe_size)) {
 278                 rc = -EINVAL;
 279                 llapi_error(LLAPI_MSG_ERROR, rc,
 280                                 "warning: stripe size 4G or larger "
 281                                 "is not currently supported and would wrap");
 282                 return rc;
 283         }
 284         return 0;
 285 }
 286
 287 /* return the first file matching this pattern */
 288 static int first_match(char *pattern, char *buffer)
 289 {
 290         glob_t glob_info;
 291
 292         if (glob(pattern, GLOB_BRACE, NULL, &glob_info))
 293                 return -ENOENT;
 294
 295         if (glob_info.gl_pathc < 1) {
 296                 globfree(&glob_info);
 297                 return -ENOENT;
 298         }
 299
 300         strcpy(buffer, glob_info.gl_pathv[0]);
 301
 302         globfree(&glob_info);
 303         return 0;
 304 }
 305
 306 static int find_target_obdpath(char *fsname, char *path)
 307 {
 308         glob_t glob_info;
 309         char pattern[PATH_MAX + 1];
 310         int rc;
 311
 312         snprintf(pattern, PATH_MAX,
 313                  "/proc/fs/lustre/lov/%s-*/target_obd",
 314                  fsname);
 315         rc = glob(pattern, GLOB_BRACE, NULL, &glob_info);
 316         if (rc == GLOB_NOMATCH)
 317                 return -ENODEV;
 318         else if (rc)
 319                 return -EINVAL;
 320
 321         strcpy(path, glob_info.gl_pathv[0]);
 322         globfree(&glob_info);
 323         return 0;
 324 }
 325
 326 static int find_poolpath(char *fsname, char *poolname, char *poolpath)
 327 {
 328         glob_t glob_info;
 329         char pattern[PATH_MAX + 1];
 330         int rc;
 331
 332         snprintf(pattern, PATH_MAX,
 333                  "/proc/fs/lustre/lov/%s-*/pools/%s",
 334                  fsname, poolname);
 335         rc = glob(pattern, GLOB_BRACE, NULL, &glob_info);
 336         /* If no pools, make sure the lov is available */
 337         if ((rc == GLOB_NOMATCH) &&
 338             (find_target_obdpath(fsname, poolpath) == -ENODEV))
 339                 return -ENODEV;
 340         if (rc)
 341                 return -EINVAL;
 342
 343         strcpy(poolpath, glob_info.gl_pathv[0]);
 344         globfree(&glob_info);
 345         return 0;
 346 }
 347
 348 /*
 349  * Trim a trailing newline from a string, if it exists.
 350  */
 351 int llapi_chomp_string(char *buf)
 352 {
 353         if (!buf || !*buf)
 354                 return 0;
 355
 356         while (buf[1])
 357                 buf++;
 358
 359         if (*buf != '\n')
 360                 return 0;
 361
 362         *buf = '\0';
 363         return '\n';
 364 }
 365
 366 /**
 367   * return a parameter string for a specific device type or mountpoint
 368   *
 369   * \param param_path the path to the file containing parameter data
 370   * \param result buffer for parameter value string
 371   * \param result_size size of buffer for return value
 372   *
 373   * The \param param_path is appended to /proc/{fs,sys}/{lnet,lustre} to
 374   * complete the absolute path to the file containing the parameter data
 375   * the user is requesting. If that file exist then the data is read from
 376   * the file and placed into the \param result buffer that is passed by
 377   * the user. Data is only copied up to the \param result_size to prevent
 378   * overflow of the array.
 379   *
 380   * Return 0 for success, with a NUL-terminated string in \param result.
 381   * Return -ve value for error.
 382   */
 383 int get_param(const char *param_path, char *result,
 384                      unsigned int result_size)
 385 {
 386         char file[PATH_MAX + 1], pattern[PATH_MAX + 1], buf[result_size];
 387         FILE *fp = NULL;
 388         int rc = 0;
 389
 390         snprintf(pattern, PATH_MAX, "/proc/{fs,sys}/{lnet,lustre}/%s",
 391                  param_path);
 392         rc = first_match(pattern, file);
 393         if (rc != 0 || result == NULL)
 394                 return rc;
 395
 396         fp = fopen(file, "r");
 397         if (fp != NULL) {
 398                 while (fgets(buf, result_size, fp) != NULL)
 399                         strcpy(result, buf);
 400                 fclose(fp);
 401         } else {
 402                 rc = -errno;
 403         }
 404         return rc;
 405 }
 406
 407 #define DEVICES_LIST "/proc/fs/lustre/devices"
 408
 409 /**
 410   * return a parameter string for a specific device type or mountpoint
 411   *
 412   * \param fsname Lustre filesystem name (optional)
 413   * \param file_path path to file in filesystem (optional, if fsname unset)
 414   * \param obd_type Lustre OBD device type
 415   * \param param_name parameter name to fetch
 416   * \param value return buffer for parameter value string
 417   * \param val_len size of buffer for return value
 418   *
 419   * If fsname is specified then the parameter will be from that filesystem
 420   * (if it exists). If file_path is given and it is in a mounted Lustre
 421   * filesystem, then the parameter will be otherwise the value may be
 422   * from any mounted filesystem (if there is more than one).
 423   *
 424   * If "obd_type" matches a Lustre device then the first matching device
 425   * (as with "lctl dl", constrained by \param fsname or \param mount_path)
 426   * will be used to provide the return value, otherwise the first such
 427   * device found will be used.
 428   *
 429   * Return 0 for success, with a NUL-terminated string in \param buffer.
 430   * Return -ve value for error.
 431   */
 432 static int get_param_obdvar(const char *fsname, const char *file_path,
 433                             const char *obd_type, const char *param_name,
 434                             char *value, unsigned int val_len)
 435 {
 436         char devices[PATH_MAX];
 437         char dev[PATH_MAX] = "*";
 438         char fs[PATH_MAX];
 439         FILE *fp = NULL;
 440         int rc = 0;
 441
 442         fp = fopen(DEVICES_LIST, "r");
 443         if (fp == NULL) {
 444                 rc = -errno;
 445                 llapi_error(LLAPI_MSG_ERROR, rc, "error: opening "DEVICES_LIST);
 446                 goto out;
 447         }
 448
 449         if (fsname == NULL && file_path != NULL) {
 450                 rc = llapi_search_fsname(file_path, fs);
 451                 if (rc) {
 452                         llapi_error(LLAPI_MSG_ERROR, rc,
 453                                     "'%s' is not on a Lustre filesystem",
 454                                     file_path);
 455                         goto out;
 456                 }
 457         } else if (fsname != NULL) {
 458                 rc = strlcpy(fs, fsname, sizeof(fs));
 459                 if (rc >= sizeof(fs)) {
 460                         rc = -E2BIG;
 461                         goto out;
 462                 }
 463         }
 464
 465         while (fgets(devices, sizeof(devices) - 1, fp) != NULL) {
 466                 char *bufp = devices, *tmp;
 467
 468                 devices[sizeof(devices) - 1] = '\0';
 469                 while (bufp[0] == ' ')
 470                         ++bufp;
 471
 472                 tmp = strstr(bufp, obd_type);
 473                 if (tmp != NULL) {
 474                         tmp += strlen(obd_type) + 1;
 475                         if (strcmp(tmp, fs))
 476                                 continue;
 477                         rc = strlcpy(dev, tmp, sizeof(dev));
 478                         if (rc >= sizeof(dev)) {
 479                                 rc = -E2BIG;
 480                                 goto out;
 481                         }
 482
 483                         tmp = strchr(dev, ' ');
 484                         if (tmp != NULL)
 485                                 *tmp = '\0';
 486                         break;
 487                 }
 488         }
 489
 490         if (dev[0] == '*' && strlen(fs)) {
 491                 rc = snprintf(dev, sizeof(dev), "%s-*", fs);
 492                 if (rc >= sizeof(dev)) {
 493                         rc = -E2BIG;
 494                         goto out;
 495                 }
 496         }
 497         rc = snprintf(devices, sizeof(devices), "%s/%s/%s", obd_type, dev,
 498                       param_name);
 499         if (rc >= sizeof(devices)) {
 500                 rc = -E2BIG;
 501                 goto out;
 502         }
 503
 504         fclose(fp);
 505         return get_param(devices, value, val_len);
 506 out:
 507         if (fp != NULL)
 508                 fclose(fp);
 509         return rc;
 510 }
 511
 512 /*
 513  * TYPE one of llite, lmv, lov.
 514  * /proc/fs/lustre/TYPE/INST the directory of interest.
 515  */
 516 static int get_param_cli(const char *type, const char *inst,
 517                          const char *param, char *buf, size_t buf_size)
 518 {
 519         char param_path[PATH_MAX + 1];
 520         FILE *param_file = NULL;
 521         int rc;
 522
 523         snprintf(param_path, sizeof(param_path),
 524                  "/proc/fs/lustre/%s/%s/%s", type, inst, param);
 525
 526         param_file = fopen(param_path, "r");
 527         if (param_file == NULL) {
 528                 rc = -errno;
 529                 goto out;
 530         }
 531
 532         if (fgets(buf, buf_size, param_file) == NULL) {
 533                 rc = -errno;
 534                 goto out;
 535         }
 536
 537         rc = 0;
 538 out:
 539         if (param_file != NULL)
 540                 fclose(param_file);
 541
 542         return rc;
 543 }
 544
 545 static int get_param_llite(const char *path,
 546                            const char *param, char *buf, size_t buf_size)
 547 {
 548         char inst[80];
 549         int rc;
 550
 551         rc = llapi_getname(path, inst, sizeof(inst));
 552         if (rc != 0)
 553                 return rc;
 554
 555         return get_param_cli("llite", inst, param, buf, buf_size);
 556 }
 557
 558 static int get_param_lov(const char *path,
 559                          const char *param, char *buf, size_t buf_size)
 560 {
 561         struct obd_uuid uuid;
 562         int rc;
 563
 564         rc = llapi_file_get_lov_uuid(path, &uuid);
 565         if (rc != 0)
 566                 return rc;
 567
 568         return get_param_cli("lov", uuid.uuid, param, buf, buf_size);
 569 }
 570
 571 static int get_param_lmv(const char *path,
 572                          const char *param, char *buf, size_t buf_size)
 573 {
 574         struct obd_uuid uuid;
 575         int rc;
 576
 577         rc = llapi_file_get_lmv_uuid(path, &uuid);
 578         if (rc != 0)
 579                 return rc;
 580
 581         return get_param_cli("lmv", uuid.uuid, param, buf, buf_size);
 582 }
 583
 584 static int get_mds_md_size(const char *path)
 585 {
 586         int md_size = lov_user_md_size(LOV_MAX_STRIPE_COUNT, LOV_USER_MAGIC_V3);
 587         char buf[80];
 588         int rc;
 589
 590         /* Get the max ea size from llite proc. */
 591         rc = get_param_llite(path, "max_easize", buf, sizeof(buf));
 592         if (rc != 0)
 593                 goto out;
 594
 595         rc = atoi(buf);
 596         if (rc > 0)
 597                 md_size = rc;
 598
 599 out:
 600         return md_size;
 601 }
 602
 603 int llapi_get_agent_uuid(char *path, char *buf, size_t bufsize)
 604 {
 605         return get_param_lmv(path, "uuid", buf, bufsize);
 606 }
 607
 608 /*
 609  * if pool is NULL, search ostname in target_obd
 610  * if pool is not NULL:
 611  *  if pool not found returns errno < 0
 612  *  if ostname is NULL, returns 1 if pool is not empty and 0 if pool empty
 613  *  if ostname is not NULL, returns 1 if OST is in pool and 0 if not
 614  */
 615 int llapi_search_ost(char *fsname, char *poolname, char *ostname)
 616 {
 617         FILE *fd;
 618         char buffer[PATH_MAX + 1];
 619         int len = 0, rc;
 620
 621         if (ostname != NULL)
 622                 len = strlen(ostname);
 623
 624         if (poolname == NULL) {
 625                 if (len == 0)
 626                         rc = -EINVAL;
 627                 else
 628                         rc = find_target_obdpath(fsname, buffer);
 629         } else {
 630                 rc = find_poolpath(fsname, poolname, buffer);
 631         }
 632         if (rc)
 633                 return rc;
 634
 635         fd = fopen(buffer, "r");
 636         if (fd == NULL)
 637                 return -errno;
 638
 639         while (fgets(buffer, sizeof(buffer), fd) != NULL) {
 640                 if (poolname == NULL) {
 641                         char *ptr;
 642                         /* Search for an ostname in the list of OSTs
 643                          Line format is IDX: fsname-OSTxxxx_UUID STATUS */
 644                         ptr = strchr(buffer, ' ');
 645                         if ((ptr != NULL) &&
 646                             (strncmp(ptr + 1, ostname, len) == 0)) {
 647                                 fclose(fd);
 648                                 return 1;
 649                         }
 650                 } else {
 651                         /* Search for an ostname in a pool,
 652                          (or an existing non-empty pool if no ostname) */
 653                         if ((ostname == NULL) ||
 654                             (strncmp(buffer, ostname, len) == 0)) {
 655                                 fclose(fd);
 656                                 return 1;
 657                         }
 658                 }
 659         }
 660         fclose(fd);
 661         return 0;
 662 }
 663
 664 /**
 665  * Open a Lustre file.
 666  *
 667  * \param name     the name of the file to be opened
 668  * \param flags    access mode, see flags in open(2)
 669  * \param mode     permission of the file if it is created, see mode in open(2)
 670  * \param param    stripe pattern of the newly created file
 671  *
 672  * \retval         file descriptor of opened file
 673  * \retval         negative errno on failure
 674  */
 675 int llapi_file_open_param(const char *name, int flags, mode_t mode,
 676                           const struct llapi_stripe_param *param)
 677 {
 678         char fsname[MAX_OBD_NAME + 1] = { 0 };
 679         char *pool_name = param->lsp_pool;
 680         struct lov_user_md *lum = NULL;
 681         size_t lum_size = sizeof(*lum);
 682         int fd, rc;
 683
 684         /* Make sure we are on a Lustre file system */
 685         rc = llapi_search_fsname(name, fsname);
 686         if (rc) {
 687                 llapi_error(LLAPI_MSG_ERROR, rc,
 688                             "'%s' is not on a Lustre filesystem",
 689                             name);
 690                 return rc;
 691         }
 692
 693         /* Check if the stripe pattern is sane. */
 694         rc = llapi_stripe_limit_check(param->lsp_stripe_size,
 695                                       param->lsp_stripe_offset,
 696                                       param->lsp_stripe_count,
 697                                       param->lsp_stripe_pattern);
 698         if (rc != 0)
 699                 return rc;
 700
 701         /* Make sure we have a good pool */
 702         if (pool_name != NULL) {
 703                 /* in case user gives the full pool name <fsname>.<poolname>,
 704                  * strip the fsname */
 705                 char *ptr = strchr(pool_name, '.');
 706                 if (ptr != NULL) {
 707                         *ptr = '\0';
 708                         if (strcmp(pool_name, fsname) != 0) {
 709                                 *ptr = '.';
 710                                 llapi_err_noerrno(LLAPI_MSG_ERROR,
 711                                         "Pool '%s' is not on filesystem '%s'",
 712                                         pool_name, fsname);
 713                                 return -EINVAL;
 714                         }
 715                         pool_name = ptr + 1;
 716                 }
 717
 718                 /* Make sure the pool exists and is non-empty */
 719                 rc = llapi_search_ost(fsname, pool_name, NULL);
 720                 if (rc < 1) {
 721                         char *err = rc == 0 ? "has no OSTs" : "does not exist";
 722
 723                         llapi_err_noerrno(LLAPI_MSG_ERROR, "pool '%s.%s' %s",
 724                                           fsname, pool_name, err);
 725                         return -EINVAL;
 726                 }
 727
 728                 lum_size = sizeof(struct lov_user_md_v3);
 729         }
 730
 731         /* sanity check of target list */
 732         if (param->lsp_is_specific) {
 733                 char ostname[MAX_OBD_NAME + 1];
 734                 bool found = false;
 735                 int i;
 736
 737                 for (i = 0; i < param->lsp_stripe_count; i++) {
 738                         snprintf(ostname, sizeof(ostname), "%s-OST%04x_UUID",
 739                                  fsname, param->lsp_osts[i]);
 740                         rc = llapi_search_ost(fsname, pool_name, ostname);
 741                         if (rc <= 0) {
 742                                 if (rc == 0)
 743                                         rc = -ENODEV;
 744
 745                                 llapi_error(LLAPI_MSG_ERROR, rc,
 746                                             "%s: cannot find OST %s in %s",
 747                                             __func__, ostname,
 748                                             pool_name != NULL ?
 749                                             "pool" : "system");
 750                                 return rc;
 751                         }
 752
 753                         /* Make sure stripe offset is in OST list. */
 754                         if (param->lsp_osts[i] == param->lsp_stripe_offset)
 755                                 found = true;
 756                 }
 757                 if (!found) {
 758                         llapi_error(LLAPI_MSG_ERROR, -EINVAL,
 759                                     "%s: stripe offset '%d' is not in the "
 760                                     "target list",
 761                                     __func__, param->lsp_stripe_offset);
 762                         return -EINVAL;
 763                 }
 764
 765                 lum_size = lov_user_md_size(param->lsp_stripe_count,
 766                                             LOV_USER_MAGIC_SPECIFIC);
 767         }
 768
 769         lum = calloc(1, lum_size);
 770         if (lum == NULL)
 771                 return -ENOMEM;
 772
 773 retry_open:
 774         fd = open(name, flags | O_LOV_DELAY_CREATE, mode);
 775         if (fd < 0) {
 776                 if (errno == EISDIR && !(flags & O_DIRECTORY)) {
 777                         flags = O_DIRECTORY | O_RDONLY;
 778                         goto retry_open;
 779                 }
 780         }
 781
 782         if (fd < 0) {
 783                 rc = -errno;
 784                 llapi_error(LLAPI_MSG_ERROR, rc, "unable to open '%s'", name);
 785                 free(lum);
 786                 return rc;
 787         }
 788
 789         /*  Initialize IOCTL striping pattern structure */
 790         lum->lmm_magic = LOV_USER_MAGIC_V1;
 791         lum->lmm_pattern = param->lsp_stripe_pattern;
 792         lum->lmm_stripe_size = param->lsp_stripe_size;
 793         lum->lmm_stripe_count = param->lsp_stripe_count;
 794         lum->lmm_stripe_offset = param->lsp_stripe_offset;
 795         if (pool_name != NULL) {
 796                 struct lov_user_md_v3 *lumv3 = (void *)lum;
 797
 798                 lumv3->lmm_magic = LOV_USER_MAGIC_V3;
 799                 strncpy(lumv3->lmm_pool_name, pool_name, LOV_MAXPOOLNAME);
 800         }
 801         if (param->lsp_is_specific) {
 802                 struct lov_user_md_v3 *lumv3 = (void *)lum;
 803                 int i;
 804
 805                 lumv3->lmm_magic = LOV_USER_MAGIC_SPECIFIC;
 806                 if (pool_name == NULL) {
 807                         /* LOV_USER_MAGIC_SPECIFIC uses v3 format plus specified
 808                          * OST list, therefore if pool is not specified we have
 809                          * to pack a null pool name for placeholder. */
 810                         memset(lumv3->lmm_pool_name, 0, LOV_MAXPOOLNAME);
 811                 }
 812
 813                 for (i = 0; i < param->lsp_stripe_count; i++)
 814                         lumv3->lmm_objects[i].l_ost_idx = param->lsp_osts[i];
 815         }
 816
 817         if (ioctl(fd, LL_IOC_LOV_SETSTRIPE, lum) != 0) {
 818                 char *errmsg = "stripe already set";
 819
 820                 rc = -errno;
 821                 if (errno != EEXIST && errno != EALREADY)
 822                         errmsg = strerror(errno);
 823
 824                 llapi_err_noerrno(LLAPI_MSG_ERROR,
 825                                   "error on ioctl "LPX64" for '%s' (%d): %s",
 826                                   (__u64)LL_IOC_LOV_SETSTRIPE, name, fd,
 827                                   errmsg);
 828
 829                 close(fd);
 830                 fd = rc;
 831         }
 832
 833         free(lum);
 834
 835         return fd;
 836 }
 837
 838 int llapi_file_open_pool(const char *name, int flags, int mode,
 839                          unsigned long long stripe_size, int stripe_offset,
 840                          int stripe_count, int stripe_pattern, char *pool_name)
 841 {
 842         const struct llapi_stripe_param param = {
 843                 .lsp_stripe_size = stripe_size,
 844                 .lsp_stripe_count = stripe_count,
 845                 .lsp_stripe_pattern = stripe_pattern,
 846                 .lsp_stripe_offset = stripe_offset,
 847                 .lsp_pool = pool_name
 848         };
 849         return llapi_file_open_param(name, flags, mode, &param);
 850 }
 851
 852 int llapi_file_open(const char *name, int flags, int mode,
 853                     unsigned long long stripe_size, int stripe_offset,
 854                     int stripe_count, int stripe_pattern)
 855 {
 856         return llapi_file_open_pool(name, flags, mode, stripe_size,
 857                                     stripe_offset, stripe_count,
 858                                     stripe_pattern, NULL);
 859 }
 860
 861 int llapi_file_create(const char *name, unsigned long long stripe_size,
 862                       int stripe_offset, int stripe_count, int stripe_pattern)
 863 {
 864         int fd;
 865
 866         fd = llapi_file_open_pool(name, O_CREAT | O_WRONLY, 0644, stripe_size,
 867                                   stripe_offset, stripe_count, stripe_pattern,
 868                                   NULL);
 869         if (fd < 0)
 870                 return fd;
 871
 872         close(fd);
 873         return 0;
 874 }
 875
 876 int llapi_file_create_pool(const char *name, unsigned long long stripe_size,
 877                            int stripe_offset, int stripe_count,
 878                            int stripe_pattern, char *pool_name)
 879 {
 880         int fd;
 881
 882         fd = llapi_file_open_pool(name, O_CREAT | O_WRONLY, 0644, stripe_size,
 883                                   stripe_offset, stripe_count, stripe_pattern,
 884                                   pool_name);
 885         if (fd < 0)
 886                 return fd;
 887
 888         close(fd);
 889         return 0;
 890 }
 891
 892 int llapi_dir_set_default_lmv_stripe(const char *name, int stripe_offset,
 893                                      int stripe_count, int stripe_pattern,
 894                                      const char *pool_name)
 895 {
 896         struct lmv_user_md      lum = { 0 };
 897         int                     fd;
 898         int                     rc = 0;
 899
 900         lum.lum_magic = LMV_USER_MAGIC;
 901         lum.lum_stripe_offset = stripe_offset;
 902         lum.lum_stripe_count = stripe_count;
 903         lum.lum_hash_type = stripe_pattern;
 904         if (pool_name != NULL) {
 905                 if (strlen(pool_name) >= sizeof(lum.lum_pool_name)) {
 906                         llapi_err_noerrno(LLAPI_MSG_ERROR,
 907                                   "error LL_IOC_LMV_SET_DEFAULT_STRIPE '%s'"
 908                                   ": too large pool name: %s", name, pool_name);
 909                         return -E2BIG;
 910                 }
 911                 strncpy(lum.lum_pool_name, pool_name,
 912                         sizeof(lum.lum_pool_name));
 913         }
 914
 915         fd = open(name, O_DIRECTORY | O_RDONLY);
 916         if (fd < 0) {
 917                 rc = -errno;
 918                 llapi_error(LLAPI_MSG_ERROR, rc, "unable to open '%s'", name);
 919                 return rc;
 920         }
 921
 922         rc = ioctl(fd, LL_IOC_LMV_SET_DEFAULT_STRIPE, &lum);
 923         if (rc < 0) {
 924                 char *errmsg = "stripe already set";
 925                 rc = -errno;
 926                 if (errno != EEXIST && errno != EALREADY)
 927                         errmsg = strerror(errno);
 928
 929                 llapi_err_noerrno(LLAPI_MSG_ERROR,
 930                                   "error on LL_IOC_LMV_SETSTRIPE '%s' (%d): %s",
 931                                   name, fd, errmsg);
 932         }
 933         close(fd);
 934         return rc;
 935 }
 936
 937 int llapi_dir_create_pool(const char *name, int mode, int stripe_offset,
 938                           int stripe_count, int stripe_pattern,
 939                           const char *pool_name)
 940 {
 941         struct lmv_user_md lmu = { 0 };
 942         struct obd_ioctl_data data = { 0 };
 943         char rawbuf[8192];
 944         char *buf = rawbuf;
 945         char *dirpath = NULL;
 946         char *namepath = NULL;
 947         char *dir;
 948         char *filename;
 949         int fd = -1;
 950         int rc;
 951
 952         dirpath = strdup(name);
 953         namepath = strdup(name);
 954         if (!dirpath || !namepath)
 955                 return -ENOMEM;
 956
 957         lmu.lum_magic = LMV_USER_MAGIC;
 958         lmu.lum_stripe_offset = stripe_offset;
 959         lmu.lum_stripe_count = stripe_count;
 960         lmu.lum_hash_type = stripe_pattern;
 961         if (pool_name != NULL) {
 962                 if (strlen(pool_name) > LOV_MAXPOOLNAME) {
 963                         llapi_err_noerrno(LLAPI_MSG_ERROR,
 964                                   "error LL_IOC_LMV_SETSTRIPE '%s' : too large"
 965                                   "pool name: %s", name, pool_name);
 966                         rc = -E2BIG;
 967                         goto out;
 968                 }
 969                 memcpy(lmu.lum_pool_name, pool_name, strlen(pool_name));
 970         }
 971
 972         filename = basename(namepath);
 973         dir = dirname(dirpath);
 974
 975         data.ioc_inlbuf1 = (char *)filename;
 976         data.ioc_inllen1 = strlen(filename) + 1;
 977         data.ioc_inlbuf2 = (char *)&lmu;
 978         data.ioc_inllen2 = sizeof(struct lmv_user_md);
 979         data.ioc_type = mode;
 980         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
 981         if (rc) {
 982                 llapi_error(LLAPI_MSG_ERROR, rc,
 983                             "error: LL_IOC_LMV_SETSTRIPE pack failed '%s'.",
 984                             name);
 985                 goto out;
 986         }
 987
 988         fd = open(dir, O_DIRECTORY | O_RDONLY);
 989         if (fd < 0) {
 990                 rc = -errno;
 991                 llapi_error(LLAPI_MSG_ERROR, rc, "unable to open '%s'", name);
 992                 goto out;
 993         }
 994
 995         if (ioctl(fd, LL_IOC_LMV_SETSTRIPE, buf)) {
 996                 char *errmsg = "stripe already set";
 997                 rc = -errno;
 998                 if (errno != EEXIST && errno != EALREADY)
 999                         errmsg = strerror(errno);
1000
1001                 llapi_err_noerrno(LLAPI_MSG_ERROR,
1002                                   "error on LL_IOC_LMV_SETSTRIPE '%s' (%d): %s",
1003                                   name, fd, errmsg);
1004         }
1005         close(fd);
1006 out:
1007         free(dirpath);
1008         free(namepath);
1009         return rc;
1010 }
1011
1012 int llapi_direntry_remove(char *dname)
1013 {
1014         char *dirpath = NULL;
1015         char *namepath = NULL;
1016         char *dir;
1017         char *filename;
1018         int fd = -1;
1019         int rc = 0;
1020
1021         dirpath = strdup(dname);
1022         namepath = strdup(dname);
1023         if (!dirpath || !namepath)
1024                 return -ENOMEM;
1025
1026         filename = basename(namepath);
1027
1028         dir = dirname(dirpath);
1029
1030         fd = open(dir, O_DIRECTORY | O_RDONLY);
1031         if (fd < 0) {
1032                 rc = -errno;
1033                 llapi_error(LLAPI_MSG_ERROR, rc, "unable to open '%s'",
1034                             filename);
1035                 goto out;
1036         }
1037
1038         if (ioctl(fd, LL_IOC_REMOVE_ENTRY, filename)) {
1039                 char *errmsg = strerror(errno);
1040                 llapi_err_noerrno(LLAPI_MSG_ERROR,
1041                                   "error on ioctl "LPX64" for '%s' (%d): %s",
1042                                   (__u64)LL_IOC_LMV_SETSTRIPE, filename,
1043                                   fd, errmsg);
1044         }
1045 out:
1046         free(dirpath);
1047         free(namepath);
1048         if (fd != -1)
1049                 close(fd);
1050         return rc;
1051 }
1052
1053 /*
1054  * Find the fsname, the full path, and/or an open fd.
1055  * Either the fsname or path must not be NULL
1056  */
1057 int get_root_path(int want, char *fsname, int *outfd, char *path, int index)
1058 {
1059         struct mntent mnt;
1060         char buf[PATH_MAX], mntdir[PATH_MAX];
1061         char *ptr;
1062         FILE *fp;
1063         int idx = 0, len = 0, mntlen, fd;
1064         int rc = -ENODEV;
1065
1066         /* get the mount point */
1067         fp = setmntent(MOUNTED, "r");
1068         if (fp == NULL) {
1069                 rc = -EIO;
1070                 llapi_error(LLAPI_MSG_ERROR, rc,
1071                             "setmntent(%s) failed", MOUNTED);
1072                 return rc;
1073         }
1074         while (1) {
1075                 if (getmntent_r(fp, &mnt, buf, sizeof(buf)) == NULL)
1076                         break;
1077
1078                 if (!llapi_is_lustre_mnt(&mnt))
1079                         continue;
1080
1081                 if ((want & WANT_INDEX) && (idx++ != index))
1082                         continue;
1083
1084                 mntlen = strlen(mnt.mnt_dir);
1085                 ptr = strrchr(mnt.mnt_fsname, '/');
1086                 /* thanks to the call to llapi_is_lustre_mnt() above,
1087                  * we are sure that mnt.mnt_fsname contains ":/",
1088                  * so ptr should never be NULL */
1089                 if (ptr == NULL)
1090                         continue;
1091                 ptr++;
1092
1093                 /* Check the fsname for a match, if given */
1094                 if (!(want & WANT_FSNAME) && fsname != NULL &&
1095                     (strlen(fsname) > 0) && (strcmp(ptr, fsname) != 0))
1096                         continue;
1097
1098                 /* If the path isn't set return the first one we find */
1099                 if (path == NULL || strlen(path) == 0) {
1100                         strcpy(mntdir, mnt.mnt_dir);
1101                         if ((want & WANT_FSNAME) && fsname != NULL)
1102                                 strcpy(fsname, ptr);
1103                         rc = 0;
1104                         break;
1105                 /* Otherwise find the longest matching path */
1106                 } else if ((strlen(path) >= mntlen) && (mntlen >= len) &&
1107                            (strncmp(mnt.mnt_dir, path, mntlen) == 0)) {
1108                         strcpy(mntdir, mnt.mnt_dir);
1109                         len = mntlen;
1110                         if ((want & WANT_FSNAME) && fsname != NULL)
1111                                 strcpy(fsname, ptr);
1112                         rc = 0;
1113                 }
1114         }
1115         endmntent(fp);
1116
1117         /* Found it */
1118         if (rc == 0) {
1119                 if ((want & WANT_PATH) && path != NULL)
1120                         strcpy(path, mntdir);
1121                 if (want & WANT_FD) {
1122                         fd = open(mntdir, O_RDONLY | O_DIRECTORY | O_NONBLOCK);
1123                         if (fd < 0) {
1124                                 rc = -errno;
1125                                 llapi_error(LLAPI_MSG_ERROR, rc,
1126                                             "error opening '%s'", mntdir);
1127
1128                         } else {
1129                                 *outfd = fd;
1130                         }
1131                 }
1132         } else if (want & WANT_ERROR)
1133                 llapi_err_noerrno(LLAPI_MSG_ERROR,
1134                                   "can't find fs root for '%s': %d",
1135                                   (want & WANT_PATH) ? fsname : path, rc);
1136         return rc;
1137 }
1138
1139 /*
1140  * search lustre mounts
1141  *
1142  * Calling this function will return to the user the mount point, mntdir, and
1143  * the file system name, fsname, if the user passed a buffer to this routine.
1144  *
1145  * The user inputs are pathname and index. If the pathname is supplied then
1146  * the value of the index will be ignored. The pathname will return data if
1147  * the pathname is located on a lustre mount. Index is used to pick which
1148  * mount point you want in the case of multiple mounted lustre file systems.
1149  * See function lfs_osts in lfs.c for a example of the index use.
1150  */
1151 int llapi_search_mounts(const char *pathname, int index, char *mntdir,
1152                         char *fsname)
1153 {
1154         int want = WANT_PATH, idx = -1;
1155
1156         if (!pathname || pathname[0] == '\0') {
1157                 want |= WANT_INDEX;
1158                 idx = index;
1159         } else
1160                 strcpy(mntdir, pathname);
1161
1162         if (fsname)
1163                 want |= WANT_FSNAME;
1164         return get_root_path(want, fsname, NULL, mntdir, idx);
1165 }
1166
1167 /* Given a path, find the corresponding Lustre fsname */
1168 int llapi_search_fsname(const char *pathname, char *fsname)
1169 {
1170         char *path;
1171         int rc;
1172
1173         path = realpath(pathname, NULL);
1174         if (path == NULL) {
1175                 char buf[PATH_MAX], *ptr;
1176
1177                 buf[0] = '\0';
1178                 if (pathname[0] != '/') {
1179                         /* Need an absolute path, but realpath() only works for
1180                          * pathnames that actually exist.  We go through the
1181                          * extra hurdle of dirname(getcwd() + pathname) in
1182                          * case the relative pathname contains ".." in it. */
1183                         if (getcwd(buf, sizeof(buf) - 2) == NULL)
1184                                 return -errno;
1185                         rc = strlcat(buf, "/", sizeof(buf));
1186                         if (rc >= sizeof(buf))
1187                                 return -E2BIG;
1188                 }
1189                 rc = strlcat(buf, pathname, sizeof(buf));
1190                 if (rc >= sizeof(buf))
1191                         return -E2BIG;
1192                 path = realpath(buf, NULL);
1193                 if (path == NULL) {
1194                         ptr = strrchr(buf, '/');
1195                         if (ptr == NULL)
1196                                 return -ENOENT;
1197                         *ptr = '\0';
1198                         path = realpath(buf, NULL);
1199                         if (path == NULL) {
1200                                 rc = -errno;
1201                                 llapi_error(LLAPI_MSG_ERROR, rc,
1202                                             "pathname '%s' cannot expand",
1203                                             pathname);
1204                                 return rc;
1205                         }
1206                 }
1207         }
1208         rc = get_root_path(WANT_FSNAME | WANT_ERROR, fsname, NULL, path, -1);
1209         free(path);
1210         return rc;
1211 }
1212
1213 int llapi_search_rootpath(char *pathname, const char *fsname)
1214 {
1215         return get_root_path(WANT_PATH, (char *)fsname, NULL, pathname, -1);
1216 }
1217
1218 int llapi_getname(const char *path, char *buf, size_t size)
1219 {
1220         struct obd_uuid uuid_buf;
1221         char *uuid = uuid_buf.uuid;
1222         int rc, nr;
1223
1224         memset(&uuid_buf, 0, sizeof(uuid_buf));
1225         rc = llapi_file_get_lov_uuid(path, &uuid_buf);
1226         if (rc)
1227                 return rc;
1228
1229         /* We want to turn lustre-clilov-ffff88002738bc00 into
1230          * lustre-ffff88002738bc00. */
1231
1232         nr = snprintf(buf, size, "%.*s-%s",
1233                       (int) (strlen(uuid) - 24), uuid,
1234                       uuid + strlen(uuid) - 16);
1235
1236         if (nr >= size)
1237                 rc = -ENAMETOOLONG;
1238
1239         return rc;
1240 }
1241
1242
1243 /*
1244  * find the pool directory path under /proc
1245  * (can be also used to test if a fsname is known)
1246  */
1247 static int poolpath(char *fsname, char *pathname, char *pool_pathname)
1248 {
1249         int rc = 0;
1250         char pattern[PATH_MAX + 1];
1251         char buffer[PATH_MAX];
1252
1253         if (fsname == NULL) {
1254                 rc = llapi_search_fsname(pathname, buffer);
1255                 if (rc != 0)
1256                         return rc;
1257                 fsname = buffer;
1258                 strcpy(pathname, fsname);
1259         }
1260
1261         snprintf(pattern, PATH_MAX, "/proc/fs/lustre/lov/%s-*/pools", fsname);
1262         rc = first_match(pattern, buffer);
1263         if (rc)
1264                 return rc;
1265
1266         /* in fsname test mode, pool_pathname is NULL */
1267         if (pool_pathname != NULL)
1268                 strcpy(pool_pathname, buffer);
1269
1270         return 0;
1271 }
1272
1273 /**
1274  * Get the list of pool members.
1275  * \param poolname    string of format \<fsname\>.\<poolname\>
1276  * \param members     caller-allocated array of char*
1277  * \param list_size   size of the members array
1278  * \param buffer      caller-allocated buffer for storing OST names
1279  * \param buffer_size size of the buffer
1280  *
1281  * \return number of members retrieved for this pool
1282  * \retval -error failure
1283  */
1284 int llapi_get_poolmembers(const char *poolname, char **members,
1285                           int list_size, char *buffer, int buffer_size)
1286 {
1287         char fsname[PATH_MAX];
1288         char *pool, *tmp;
1289         char pathname[PATH_MAX];
1290         char path[PATH_MAX];
1291         char buf[1024];
1292         FILE *fd;
1293         int rc = 0;
1294         int nb_entries = 0;
1295         int used = 0;
1296
1297         /* name is FSNAME.POOLNAME */
1298         if (strlen(poolname) >= sizeof(fsname))
1299                 return -EOVERFLOW;
1300         strlcpy(fsname, poolname, sizeof(fsname));
1301         pool = strchr(fsname, '.');
1302         if (pool == NULL)
1303                 return -EINVAL;
1304
1305         *pool = '\0';
1306         pool++;
1307
1308         rc = poolpath(fsname, NULL, pathname);
1309         if (rc != 0) {
1310                 llapi_error(LLAPI_MSG_ERROR, rc,
1311                             "Lustre filesystem '%s' not found",
1312                             fsname);
1313                 return rc;
1314         }
1315
1316         llapi_printf(LLAPI_MSG_NORMAL, "Pool: %s.%s\n", fsname, pool);
1317         rc = snprintf(path, sizeof(path), "%s/%s", pathname, pool);
1318         if (rc >= sizeof(path))
1319                 return -EOVERFLOW;
1320         fd = fopen(path, "r");
1321         if (fd == NULL) {
1322                 rc = -errno;
1323                 llapi_error(LLAPI_MSG_ERROR, rc, "Cannot open %s", path);
1324                 return rc;
1325         }
1326
1327         rc = 0;
1328         while (fgets(buf, sizeof(buf), fd) != NULL) {
1329                 if (nb_entries >= list_size) {
1330                         rc = -EOVERFLOW;
1331                         break;
1332                 }
1333                 buf[sizeof(buf) - 1] = '\0';
1334                 /* remove '\n' */
1335                 tmp = strchr(buf, '\n');
1336                 if (tmp != NULL)
1337                         *tmp='\0';
1338                 if (used + strlen(buf) + 1 > buffer_size) {
1339                         rc = -EOVERFLOW;
1340                         break;
1341                 }
1342
1343                 strcpy(buffer + used, buf);
1344                 members[nb_entries] = buffer + used;
1345                 used += strlen(buf) + 1;
1346                 nb_entries++;
1347                 rc = nb_entries;
1348         }
1349
1350         fclose(fd);
1351         return rc;
1352 }
1353
1354 /**
1355  * Get the list of pools in a filesystem.
1356  * \param name        filesystem name or path
1357  * \param poollist    caller-allocated array of char*
1358  * \param list_size   size of the poollist array
1359  * \param buffer      caller-allocated buffer for storing pool names
1360  * \param buffer_size size of the buffer
1361  *
1362  * \return number of pools retrieved for this filesystem
1363  * \retval -error failure
1364  */
1365 int llapi_get_poollist(const char *name, char **poollist, int list_size,
1366                        char *buffer, int buffer_size)
1367 {
1368         char fsname[PATH_MAX + 1], rname[PATH_MAX + 1], pathname[PATH_MAX + 1];
1369         char *ptr;
1370         DIR *dir;
1371         struct dirent pool;
1372         struct dirent *cookie = NULL;
1373         int rc = 0;
1374         unsigned int nb_entries = 0;
1375         unsigned int used = 0;
1376         unsigned int i;
1377
1378         /* initilize output array */
1379         for (i = 0; i < list_size; i++)
1380                 poollist[i] = NULL;
1381
1382         /* is name a pathname ? */
1383         ptr = strchr(name, '/');
1384         if (ptr != NULL) {
1385                 /* only absolute pathname is supported */
1386                 if (*name != '/')
1387                         return -EINVAL;
1388
1389                 if (!realpath(name, rname)) {
1390                         rc = -errno;
1391                         llapi_error(LLAPI_MSG_ERROR, rc, "invalid path '%s'",
1392                                     name);
1393                         return rc;
1394                 }
1395
1396                 rc = poolpath(NULL, rname, pathname);
1397                 if (rc != 0) {
1398                         llapi_error(LLAPI_MSG_ERROR, rc, "'%s' is not"
1399                                     " a Lustre filesystem", name);
1400                         return rc;
1401                 }
1402                 if (strlen(rname) > sizeof(fsname)-1)
1403                         return -E2BIG;
1404                 strncpy(fsname, rname, sizeof(fsname));
1405         } else {
1406                 /* name is FSNAME */
1407                 if (strlen(name) > sizeof(fsname)-1)
1408                         return -E2BIG;
1409                 strncpy(fsname, name, sizeof(fsname));
1410                 rc = poolpath(fsname, NULL, pathname);
1411         }
1412         if (rc != 0) {
1413                 llapi_error(LLAPI_MSG_ERROR, rc,
1414                             "Lustre filesystem '%s' not found", name);
1415                 return rc;
1416         }
1417
1418         llapi_printf(LLAPI_MSG_NORMAL, "Pools from %s:\n", fsname);
1419         dir = opendir(pathname);
1420         if (dir == NULL) {
1421                 rc = -errno;
1422                 llapi_error(LLAPI_MSG_ERROR, rc,
1423                             "Could not open pool list for '%s'",
1424                             name);
1425                 return rc;
1426         }
1427
1428         while(1) {
1429                 rc = readdir_r(dir, &pool, &cookie);
1430
1431                 if (rc != 0) {
1432                         rc = -errno;
1433                         llapi_error(LLAPI_MSG_ERROR, rc,
1434                                     "Error reading pool list for '%s'", name);
1435                         goto out;
1436                 } else if ((rc == 0) && (cookie == NULL)) {
1437                         /* end of directory */
1438                         break;
1439                 }
1440
1441                 /* ignore . and .. */
1442                 if (!strcmp(pool.d_name, ".") || !strcmp(pool.d_name, ".."))
1443                         continue;
1444
1445                 /* check output bounds */
1446                 if (nb_entries >= list_size) {
1447                         rc = -EOVERFLOW;
1448                         goto out;
1449                 }
1450
1451                 /* +2 for '.' and final '\0' */
1452                 if (used + strlen(pool.d_name) + strlen(fsname) + 2
1453                     > buffer_size) {
1454                         rc = -EOVERFLOW;
1455                         goto out;
1456                 }
1457
1458                 sprintf(buffer + used, "%s.%s", fsname, pool.d_name);
1459                 poollist[nb_entries] = buffer + used;
1460                 used += strlen(pool.d_name) + strlen(fsname) + 2;
1461                 nb_entries++;
1462         }
1463
1464 out:
1465         closedir(dir);
1466         return ((rc != 0) ? rc : nb_entries);
1467 }
1468
1469 /* wrapper for lfs.c and obd.c */
1470 int llapi_poollist(const char *name)
1471 {
1472         /* list of pool names (assume that pool count is smaller
1473            than OST count) */
1474         char **list, *buffer = NULL, *path = NULL, *fsname = NULL;
1475         int obdcount, bufsize, rc, nb, i;
1476         char *poolname = NULL, *tmp = NULL, data[16];
1477
1478         if (name == NULL)
1479                 return -EINVAL;
1480
1481         if (name[0] != '/') {
1482                 fsname = strdup(name);
1483                 if (fsname == NULL)
1484                         return -ENOMEM;
1485
1486                 poolname = strchr(fsname, '.');
1487                 if (poolname)
1488                         *poolname = '\0';
1489         } else {
1490                 path = (char *) name;
1491         }
1492
1493         rc = get_param_obdvar(fsname, path, "lov", "numobd",
1494                               data, sizeof(data));
1495         if (rc < 0)
1496                 goto err;
1497         obdcount = atoi(data);
1498
1499         /* Allocate space for each fsname-OST0000_UUID, 1 per OST,
1500          * and also an array to store the pointers for all that
1501          * allocated space. */
1502 retry_get_pools:
1503         bufsize = sizeof(struct obd_uuid) * obdcount;
1504         buffer = realloc(tmp, bufsize + sizeof(*list) * obdcount);
1505         if (buffer == NULL) {
1506                 rc = -ENOMEM;
1507                 goto err;
1508         }
1509         list = (char **) (buffer + bufsize);
1510
1511         if (!poolname) {
1512                 /* name is a path or fsname */
1513                 nb = llapi_get_poollist(name, list, obdcount,
1514                                         buffer, bufsize);
1515         } else {
1516                 /* name is a pool name (<fsname>.<poolname>) */
1517                 nb = llapi_get_poolmembers(name, list, obdcount,
1518                                            buffer, bufsize);
1519         }
1520
1521         if (nb == -EOVERFLOW) {
1522                 obdcount *= 2;
1523                 tmp = buffer;
1524                 goto retry_get_pools;
1525         }
1526
1527         for (i = 0; i < nb; i++)
1528                 llapi_printf(LLAPI_MSG_NORMAL, "%s\n", list[i]);
1529         rc = (nb < 0 ? nb : 0);
1530 err:
1531         if (buffer)
1532                 free(buffer);
1533         if (fsname)
1534                 free(fsname);
1535         return rc;
1536 }
1537
1538 typedef int (semantic_func_t)(char *path, DIR *parent, DIR **d,
1539                               void *data, struct dirent64 *de);
1540
1541 #define OBD_NOT_FOUND           (-1)
1542
1543 static int common_param_init(struct find_param *param, char *path)
1544 {
1545         int lum_size = get_mds_md_size(path);
1546
1547         if (lum_size < PATH_MAX + 1)
1548                 lum_size = PATH_MAX + 1;
1549
1550         param->fp_lum_size = lum_size;
1551         param->fp_lmd = malloc(sizeof(lstat_t) + param->fp_lum_size);
1552         if (param->fp_lmd == NULL) {
1553                 llapi_error(LLAPI_MSG_ERROR, -ENOMEM,
1554                             "error: allocation of %zu bytes for ioctl",
1555                             sizeof(lstat_t) + param->fp_lum_size);
1556                 return -ENOMEM;
1557         }
1558
1559         param->fp_lmv_stripe_count = 256;
1560         param->fp_lmv_md = malloc(lmv_user_md_size(param->fp_lmv_stripe_count,
1561                                                    LMV_MAGIC_V1));
1562         if (param->fp_lmv_md == NULL) {
1563                 llapi_error(LLAPI_MSG_ERROR, -ENOMEM,
1564                             "error: allocation of %d bytes for ioctl",
1565                             lmv_user_md_size(param->fp_lmv_stripe_count,
1566                                              LMV_MAGIC_V1));
1567                 return -ENOMEM;
1568         }
1569
1570         param->fp_got_uuids = 0;
1571         param->fp_obd_indexes = NULL;
1572         param->fp_obd_index = OBD_NOT_FOUND;
1573         if (!param->fp_migrate)
1574                 param->fp_mdt_index = OBD_NOT_FOUND;
1575         return 0;
1576 }
1577
1578 static void find_param_fini(struct find_param *param)
1579 {
1580         if (param->fp_obd_indexes)
1581                 free(param->fp_obd_indexes);
1582
1583         if (param->fp_lmd)
1584                 free(param->fp_lmd);
1585
1586         if (param->fp_lmv_md)
1587                 free(param->fp_lmv_md);
1588 }
1589
1590 static int cb_common_fini(char *path, DIR *parent, DIR **dirp, void *data,
1591                           struct dirent64 *de)
1592 {
1593         struct find_param *param = data;
1594         param->fp_depth--;
1595
1596         return 0;
1597 }
1598
1599 /* set errno upon failure */
1600 static DIR *opendir_parent(char *path)
1601 {
1602         DIR *parent;
1603         char *fname;
1604         char c;
1605
1606         fname = strrchr(path, '/');
1607         if (fname == NULL)
1608                 return opendir(".");
1609
1610         c = fname[1];
1611         fname[1] = '\0';
1612         parent = opendir(path);
1613         fname[1] = c;
1614         return parent;
1615 }
1616
1617 static int cb_get_dirstripe(char *path, DIR *d, struct find_param *param)
1618 {
1619         param->fp_lmv_md->lum_stripe_count = param->fp_lmv_stripe_count;
1620         if (param->fp_get_default_lmv)
1621                 param->fp_lmv_md->lum_magic = LMV_USER_MAGIC;
1622         else
1623                 param->fp_lmv_md->lum_magic = LMV_MAGIC_V1;
1624
1625         return ioctl(dirfd(d), LL_IOC_LMV_GETSTRIPE, param->fp_lmv_md);
1626 }
1627
1628 static int get_lmd_info(char *path, DIR *parent, DIR *dir,
1629                  struct lov_user_mds_data *lmd, int lumlen)
1630 {
1631         lstat_t *st = &lmd->lmd_st;
1632         int ret = 0;
1633
1634         if (parent == NULL && dir == NULL)
1635                 return -EINVAL;
1636
1637         if (dir) {
1638                 ret = ioctl(dirfd(dir), LL_IOC_MDC_GETINFO, (void *)lmd);
1639         } else if (parent) {
1640                 char *fname = strrchr(path, '/');
1641
1642                 /* To avoid opening, locking, and closing each file on the
1643                  * client if that is not needed. The GETFILEINFO ioctl can
1644                  * be done on the patent dir with a single open for all
1645                  * files in that directory, and it also doesn't pollute the
1646                  * client dcache with millions of dentries when traversing
1647                  * a large filesystem.  */
1648                 fname = (fname == NULL ? path : fname + 1);
1649                 /* retrieve needed file info */
1650                 strlcpy((char *)lmd, fname, lumlen);
1651                 ret = ioctl(dirfd(parent), IOC_MDC_GETFILEINFO, (void *)lmd);
1652         }
1653
1654         if (ret) {
1655                 if (errno == ENOTTY) {
1656                         /* ioctl is not supported, it is not a lustre fs.
1657                          * Do the regular lstat(2) instead. */
1658                         ret = lstat_f(path, st);
1659                         if (ret) {
1660                                 ret = -errno;
1661                                 llapi_error(LLAPI_MSG_ERROR, ret,
1662                                             "error: %s: lstat failed for %s",
1663                                             __func__, path);
1664                         }
1665                 } else if (errno == ENOENT) {
1666                         ret = -errno;
1667                         llapi_error(LLAPI_MSG_WARN, ret,
1668                                     "warning: %s: %s does not exist",
1669                                     __func__, path);
1670                 } else if (errno != EISDIR) {
1671                         ret = -errno;
1672                         llapi_error(LLAPI_MSG_ERROR, ret,
1673                                     "%s ioctl failed for %s.",
1674                                     dir ? "LL_IOC_MDC_GETINFO" :
1675                                     "IOC_MDC_GETFILEINFO", path);
1676                 } else {
1677                         ret = -errno;
1678                         llapi_error(LLAPI_MSG_ERROR, ret,
1679                                  "error: %s: IOC_MDC_GETFILEINFO failed for %s",
1680                                    __func__, path);
1681                 }
1682         }
1683         return ret;
1684 }
1685
1686 static int llapi_semantic_traverse(char *path, int size, DIR *parent,
1687                                    semantic_func_t sem_init,
1688                                    semantic_func_t sem_fini, void *data,
1689                                    struct dirent64 *de)
1690 {
1691         struct find_param *param = (struct find_param *)data;
1692         struct dirent64 *dent;
1693         int len, ret;
1694         DIR *d, *p = NULL;
1695
1696         ret = 0;
1697         len = strlen(path);
1698
1699         d = opendir(path);
1700         if (!d && errno != ENOTDIR) {
1701                 ret = -errno;
1702                 llapi_error(LLAPI_MSG_ERROR, ret, "%s: Failed to open '%s'",
1703                             __func__, path);
1704                 return ret;
1705         } else if (!d && !parent) {
1706                 /* ENOTDIR. Open the parent dir. */
1707                 p = opendir_parent(path);
1708                 if (!p) {
1709                         ret = -errno;
1710                         goto out;
1711                 }
1712         }
1713
1714         if (sem_init && (ret = sem_init(path, parent ?: p, &d, data, de)))
1715                 goto err;
1716
1717         if (d == NULL)
1718                 goto out;
1719
1720         while ((dent = readdir64(d)) != NULL) {
1721                 int rc;
1722
1723                 if (!strcmp(dent->d_name, ".") || !strcmp(dent->d_name, ".."))
1724                         continue;
1725
1726                 /* Don't traverse .lustre directory */
1727                 if (!(strcmp(dent->d_name, dot_lustre_name)))
1728                         continue;
1729
1730                 path[len] = 0;
1731                 if ((len + dent->d_reclen + 2) > size) {
1732                         llapi_err_noerrno(LLAPI_MSG_ERROR,
1733                                           "error: %s: string buffer is too small",
1734                                           __func__);
1735                         break;
1736                 }
1737                 strcat(path, "/");
1738                 strcat(path, dent->d_name);
1739
1740                 if (dent->d_type == DT_UNKNOWN) {
1741                         lstat_t *st = &param->fp_lmd->lmd_st;
1742
1743                         rc = get_lmd_info(path, d, NULL, param->fp_lmd,
1744                                            param->fp_lum_size);
1745                         if (rc == 0)
1746                                 dent->d_type = IFTODT(st->st_mode);
1747                         else if (ret == 0)
1748                                 ret = rc;
1749
1750                         if (rc == -ENOENT)
1751                                 continue;
1752                 }
1753                 switch (dent->d_type) {
1754                 case DT_UNKNOWN:
1755                         llapi_err_noerrno(LLAPI_MSG_ERROR,
1756                                           "error: %s: '%s' is UNKNOWN type %d",
1757                                           __func__, dent->d_name, dent->d_type);
1758                         break;
1759                 case DT_DIR:
1760                         rc = llapi_semantic_traverse(path, size, d, sem_init,
1761                                                       sem_fini, data, dent);
1762                         if (rc != 0 && ret == 0)
1763                                 ret = rc;
1764                         break;
1765                 default:
1766                         rc = 0;
1767                         if (sem_init) {
1768                                 rc = sem_init(path, d, NULL, data, dent);
1769                                 if (rc < 0 && ret == 0)
1770                                         ret = rc;
1771                         }
1772                         if (sem_fini && rc == 0)
1773                                 sem_fini(path, d, NULL, data, dent);
1774                 }
1775         }
1776
1777 out:
1778         path[len] = 0;
1779
1780         if (sem_fini)
1781                 sem_fini(path, parent, &d, data, de);
1782 err:
1783         if (d)
1784                 closedir(d);
1785         if (p)
1786                 closedir(p);
1787         return ret;
1788 }
1789
1790 static int param_callback(char *path, semantic_func_t sem_init,
1791                           semantic_func_t sem_fini, struct find_param *param)
1792 {
1793         int ret, len = strlen(path);
1794         char *buf;
1795
1796         if (len > PATH_MAX) {
1797                 ret = -EINVAL;
1798                 llapi_error(LLAPI_MSG_ERROR, ret,
1799                             "Path name '%s' is too long", path);
1800                 return ret;
1801         }
1802
1803         buf = (char *)malloc(PATH_MAX + 1);
1804         if (!buf)
1805                 return -ENOMEM;
1806
1807         strlcpy(buf, path, PATH_MAX + 1);
1808         ret = common_param_init(param, buf);
1809         if (ret)
1810                 goto out;
1811
1812         param->fp_depth = 0;
1813
1814         ret = llapi_semantic_traverse(buf, PATH_MAX + 1, NULL, sem_init,
1815                                       sem_fini, param, NULL);
1816 out:
1817         find_param_fini(param);
1818         free(buf);
1819         return ret < 0 ? ret : 0;
1820 }
1821
1822 int llapi_file_fget_lov_uuid(int fd, struct obd_uuid *lov_name)
1823 {
1824         int rc = ioctl(fd, OBD_IOC_GETNAME, lov_name);
1825         if (rc) {
1826                 rc = -errno;
1827                 llapi_error(LLAPI_MSG_ERROR, rc, "error: can't get lov name.");
1828         }
1829         return rc;
1830 }
1831
1832 int llapi_file_fget_lmv_uuid(int fd, struct obd_uuid *lov_name)
1833 {
1834         int rc = ioctl(fd, OBD_IOC_GETMDNAME, lov_name);
1835         if (rc) {
1836                 rc = -errno;
1837                 llapi_error(LLAPI_MSG_ERROR, rc, "error: can't get lmv name.");
1838         }
1839         return rc;
1840 }
1841
1842 int llapi_file_get_lov_uuid(const char *path, struct obd_uuid *lov_uuid)
1843 {
1844         int fd, rc;
1845
1846         fd = open(path, O_RDONLY);
1847         if (fd < 0) {
1848                 rc = -errno;
1849                 llapi_error(LLAPI_MSG_ERROR, rc, "error opening %s", path);
1850                 return rc;
1851         }
1852
1853         rc = llapi_file_fget_lov_uuid(fd, lov_uuid);
1854
1855         close(fd);
1856         return rc;
1857 }
1858
1859 int llapi_file_get_lmv_uuid(const char *path, struct obd_uuid *lov_uuid)
1860 {
1861         int fd, rc;
1862
1863         fd = open(path, O_RDONLY);
1864         if (fd < 0) {
1865                 rc = -errno;
1866                 llapi_error(LLAPI_MSG_ERROR, rc, "error opening %s", path);
1867                 return rc;
1868         }
1869
1870         rc = llapi_file_fget_lmv_uuid(fd, lov_uuid);
1871
1872         close(fd);
1873         return rc;
1874 }
1875
1876 enum tgt_type {
1877         LOV_TYPE = 1,
1878         LMV_TYPE
1879 };
1880 /*
1881  * If uuidp is NULL, return the number of available obd uuids.
1882  * If uuidp is non-NULL, then it will return the uuids of the obds. If
1883  * there are more OSTs then allocated to uuidp, then an error is returned with
1884  * the ost_count set to number of available obd uuids.
1885  */
1886 static int llapi_get_target_uuids(int fd, struct obd_uuid *uuidp,
1887                                   int *ost_count, enum tgt_type type)
1888 {
1889         struct obd_uuid name;
1890         char buf[1024];
1891         char format[32];
1892         FILE *fp;
1893         int rc = 0, index = 0;
1894
1895         /* Get the lov name */
1896         if (type == LOV_TYPE) {
1897                 rc = llapi_file_fget_lov_uuid(fd, &name);
1898                 if (rc)
1899                         return rc;
1900         } else {
1901                 rc = llapi_file_fget_lmv_uuid(fd, &name);
1902                 if (rc)
1903                         return rc;
1904         }
1905
1906         /* Now get the ost uuids from /proc */
1907         snprintf(buf, sizeof(buf), "/proc/fs/lustre/%s/%s/target_obd",
1908                  type == LOV_TYPE ? "lov" : "lmv", name.uuid);
1909         fp = fopen(buf, "r");
1910         if (fp == NULL) {
1911                 rc = -errno;
1912                 llapi_error(LLAPI_MSG_ERROR, rc, "error: opening '%s'", buf);
1913                 return rc;
1914         }
1915
1916         snprintf(format, sizeof(format),
1917                  "%%d: %%%zus", sizeof(uuidp[0].uuid) - 1);
1918         while (fgets(buf, sizeof(buf), fp) != NULL) {
1919                 if (uuidp && (index < *ost_count)) {
1920                         if (sscanf(buf, format, &index, uuidp[index].uuid) < 2)
1921                                 break;
1922                 }
1923                 index++;
1924         }
1925
1926         fclose(fp);
1927
1928         if (uuidp && (index > *ost_count))
1929                 rc = -EOVERFLOW;
1930
1931         *ost_count = index;
1932         return rc;
1933 }
1934
1935 int llapi_lov_get_uuids(int fd, struct obd_uuid *uuidp, int *ost_count)
1936 {
1937         return llapi_get_target_uuids(fd, uuidp, ost_count, LOV_TYPE);
1938 }
1939
1940 int llapi_get_obd_count(char *mnt, int *count, int is_mdt)
1941 {
1942         DIR *root;
1943         int rc;
1944
1945         root = opendir(mnt);
1946         if (!root) {
1947                 rc = -errno;
1948                 llapi_error(LLAPI_MSG_ERROR, rc, "open %s failed", mnt);
1949                 return rc;
1950         }
1951
1952         *count = is_mdt;
1953         rc = ioctl(dirfd(root), LL_IOC_GETOBDCOUNT, count);
1954         if (rc < 0)
1955                 rc = -errno;
1956
1957         closedir(root);
1958         return rc;
1959 }
1960
1961 /* Check if user specified value matches a real uuid.  Ignore _UUID,
1962  * -osc-4ba41334, other trailing gunk in comparison.
1963  * @param real_uuid ends in "_UUID"
1964  * @param search_uuid may or may not end in "_UUID"
1965  */
1966 int llapi_uuid_match(char *real_uuid, char *search_uuid)
1967 {
1968         int cmplen = strlen(real_uuid);
1969         int searchlen = strlen(search_uuid);
1970
1971         if (cmplen > 5 && strcmp(real_uuid + cmplen - 5, "_UUID") == 0)
1972                 cmplen -= 5;
1973         if (searchlen > 5 && strcmp(search_uuid + searchlen - 5, "_UUID") == 0)
1974                 searchlen -= 5;
1975
1976         /* The UUIDs may legitimately be different lengths, if
1977          * the system was upgraded from an older version. */
1978         if (cmplen != searchlen)
1979                 return 0;
1980
1981         return (strncmp(search_uuid, real_uuid, cmplen) == 0);
1982 }
1983
1984 /* Here, param->fp_obd_uuid points to a single obduuid, the index of which is
1985  * returned in param->fp_obd_index */
1986 static int setup_obd_uuid(DIR *dir, char *dname, struct find_param *param)
1987 {
1988         struct obd_uuid obd_uuid;
1989         char buf[1024];
1990         char format[32];
1991         FILE *fp;
1992         int rc = 0;
1993
1994         if (param->fp_got_uuids)
1995                 return rc;
1996
1997         /* Get the lov/lmv name */
1998         if (param->fp_get_lmv)
1999                 rc = llapi_file_fget_lmv_uuid(dirfd(dir), &obd_uuid);
2000         else
2001                 rc = llapi_file_fget_lov_uuid(dirfd(dir), &obd_uuid);
2002         if (rc) {
2003                 if (rc != -ENOTTY) {
2004                         llapi_error(LLAPI_MSG_ERROR, rc,
2005                                     "error: can't get lov name: %s", dname);
2006                 } else {
2007                         rc = 0;
2008                 }
2009                 return rc;
2010         }
2011
2012         param->fp_got_uuids = 1;
2013
2014         /* Now get the ost uuids from /proc */
2015         snprintf(buf, sizeof(buf), "/proc/fs/lustre/%s/%s/target_obd",
2016                  param->fp_get_lmv ? "lmv" : "lov", obd_uuid.uuid);
2017         fp = fopen(buf, "r");
2018         if (fp == NULL) {
2019                 rc = -errno;
2020                 llapi_error(LLAPI_MSG_ERROR, rc, "error: opening '%s'", buf);
2021                 return rc;
2022         }
2023
2024         if (!param->fp_obd_uuid && !param->fp_quiet && !param->fp_obds_printed)
2025                 llapi_printf(LLAPI_MSG_NORMAL, "%s:\n",
2026                              param->fp_get_lmv ? "MDTS" : "OBDS:");
2027
2028         snprintf(format, sizeof(format),
2029                  "%%d: %%%zus", sizeof(obd_uuid.uuid) - 1);
2030         while (fgets(buf, sizeof(buf), fp) != NULL) {
2031                 int index;
2032
2033                 if (sscanf(buf, format, &index, obd_uuid.uuid) < 2)
2034                         break;
2035
2036                 if (param->fp_obd_uuid) {
2037                         if (llapi_uuid_match(obd_uuid.uuid,
2038                                              param->fp_obd_uuid->uuid)) {
2039                                 param->fp_obd_index = index;
2040                                 break;
2041                         }
2042                 } else if (!param->fp_quiet && !param->fp_obds_printed) {
2043                         /* Print everything */
2044                         llapi_printf(LLAPI_MSG_NORMAL, "%s", buf);
2045                 }
2046         }
2047         param->fp_obds_printed = 1;
2048
2049         fclose(fp);
2050
2051         if (param->fp_obd_uuid && (param->fp_obd_index == OBD_NOT_FOUND)) {
2052                 llapi_err_noerrno(LLAPI_MSG_ERROR,
2053                                   "error: %s: unknown obduuid: %s",
2054                                   __func__, param->fp_obd_uuid->uuid);
2055                 rc = -EINVAL;
2056         }
2057
2058         return (rc);
2059 }
2060
2061 /* In this case, param->fp_obd_uuid will be an array of obduuids and
2062  * obd index for all these obduuids will be returned in
2063  * param->fp_obd_indexes */
2064 static int setup_indexes(DIR *dir, char *path, struct obd_uuid *obduuids,
2065                          int num_obds, int **obdindexes, int *obdindex,
2066                          enum tgt_type type)
2067 {
2068         int ret, obdcount, obd_valid = 0, obdnum;
2069         long i;
2070         struct obd_uuid *uuids = NULL;
2071         char buf[16];
2072         int *indexes;
2073
2074         if (type == LOV_TYPE)
2075                 ret = get_param_lov(path, "numobd", buf, sizeof(buf));
2076         else
2077                 ret = get_param_lmv(path, "numobd", buf, sizeof(buf));
2078         if (ret != 0)
2079                 return ret;
2080
2081         obdcount = atoi(buf);
2082         uuids = (struct obd_uuid *)malloc(obdcount *
2083                                           sizeof(struct obd_uuid));
2084         if (uuids == NULL)
2085                 return -ENOMEM;
2086
2087 retry_get_uuids:
2088         ret = llapi_get_target_uuids(dirfd(dir), uuids, &obdcount, type);
2089         if (ret) {
2090                 struct obd_uuid *uuids_temp;
2091
2092                 if (ret == -EOVERFLOW) {
2093                         uuids_temp = realloc(uuids, obdcount *
2094                                              sizeof(struct obd_uuid));
2095                         if (uuids_temp != NULL) {
2096                                 uuids = uuids_temp;
2097                                 goto retry_get_uuids;
2098                         }
2099                         else
2100                                 ret = -ENOMEM;
2101                 }
2102
2103                 llapi_error(LLAPI_MSG_ERROR, ret, "get ost uuid failed");
2104                 goto out_free;
2105         }
2106
2107         indexes = malloc(num_obds * sizeof(*obdindex));
2108         if (indexes == NULL) {
2109                 ret = -ENOMEM;
2110                 goto out_free;
2111         }
2112
2113         for (obdnum = 0; obdnum < num_obds; obdnum++) {
2114                 char *end = NULL;
2115
2116                 /* The user may have specified a simple index */
2117                 i = strtol(obduuids[obdnum].uuid, &end, 0);
2118                 if (end && *end == '\0' && i < obdcount) {
2119                         indexes[obdnum] = i;
2120                         obd_valid++;
2121                 } else {
2122                         for (i = 0; i < obdcount; i++) {
2123                                 if (llapi_uuid_match(uuids[i].uuid,
2124                                                      obduuids[obdnum].uuid)) {
2125                                         indexes[obdnum] = i;
2126                                         obd_valid++;
2127                                         break;
2128                                 }
2129                         }
2130                 }
2131                 if (i >= obdcount) {
2132                         indexes[obdnum] = OBD_NOT_FOUND;
2133                         llapi_err_noerrno(LLAPI_MSG_ERROR,
2134                                           "error: %s: unknown obduuid: %s",
2135                                           __func__, obduuids[obdnum].uuid);
2136                         ret = -EINVAL;
2137                 }
2138         }
2139
2140         if (obd_valid == 0)
2141                 *obdindex = OBD_NOT_FOUND;
2142         else
2143                 *obdindex = obd_valid;
2144
2145         *obdindexes = indexes;
2146 out_free:
2147         if (uuids)
2148                 free(uuids);
2149
2150         return ret;
2151 }
2152
2153 static int setup_target_indexes(DIR *dir, char *path, struct find_param *param)
2154 {
2155         int ret = 0;
2156
2157         if (param->fp_mdt_uuid) {
2158                 ret = setup_indexes(dir, path, param->fp_mdt_uuid,
2159                                     param->fp_num_mdts,
2160                                     &param->fp_mdt_indexes,
2161                                     &param->fp_mdt_index, LMV_TYPE);
2162                 if (ret)
2163                         return ret;
2164         }
2165
2166         if (param->fp_obd_uuid) {
2167                 ret = setup_indexes(dir, path, param->fp_obd_uuid,
2168                                     param->fp_num_obds,
2169                                     &param->fp_obd_indexes,
2170                                     &param->fp_obd_index, LOV_TYPE);
2171                 if (ret)
2172                         return ret;
2173         }
2174
2175         param->fp_got_uuids = 1;
2176
2177         return ret;
2178 }
2179
2180 int llapi_ostlist(char *path, struct find_param *param)
2181 {
2182         DIR *dir;
2183         int ret;
2184
2185         dir = opendir(path);
2186         if (dir == NULL)
2187                 return -errno;
2188
2189         ret = setup_obd_uuid(dir, path, param);
2190         closedir(dir);
2191
2192         return ret;
2193 }
2194
2195 /*
2196  * Given a filesystem name, or a pathname of a file on a lustre filesystem,
2197  * tries to determine the path to the filesystem's clilov directory under /proc
2198  *
2199  * fsname is limited to MTI_NAME_MAXLEN in lustre_idl.h
2200  * The NUL terminator is compensated by the additional "%s" bytes. */
2201 #define LOV_LEN (sizeof("/proc/fs/lustre/lov/%s-clilov-*") + MTI_NAME_MAXLEN)
2202 static int clilovpath(const char *fsname, const char *const pathname,
2203                       char *clilovpath)
2204 {
2205         int rc;
2206         char pattern[LOV_LEN];
2207         char buffer[PATH_MAX + 1];
2208
2209         if (fsname == NULL) {
2210                 rc = llapi_search_fsname(pathname, buffer);
2211                 if (rc != 0)
2212                         return rc;
2213                 fsname = buffer;
2214         }
2215
2216         snprintf(pattern, sizeof(pattern), "/proc/fs/lustre/lov/%s-clilov-*",
2217                  fsname);
2218
2219         rc = first_match(pattern, buffer);
2220         if (rc != 0)
2221                 return rc;
2222
2223         strlcpy(clilovpath, buffer, sizeof(buffer));
2224
2225         return 0;
2226 }
2227
2228 /*
2229  * Given the path to a stripe attribute proc file, tries to open and
2230  * read the attribute and return the value using the attr parameter
2231  */
2232 static int sattr_read_attr(const char *const fpath,
2233                            unsigned int *attr)
2234 {
2235
2236         FILE *f;
2237         char line[PATH_MAX + 1];
2238         int rc = 0;
2239
2240         f = fopen(fpath, "r");
2241         if (f == NULL) {
2242                 rc = -errno;
2243                 llapi_error(LLAPI_MSG_ERROR, rc, "Cannot open '%s'", fpath);
2244                 return rc;
2245         }
2246
2247         if (fgets(line, sizeof(line), f) != NULL) {
2248                 *attr = atoi(line);
2249         } else {
2250                 llapi_error(LLAPI_MSG_ERROR, errno, "Cannot read from '%s'", fpath);
2251                 rc = 1;
2252         }
2253
2254         fclose(f);
2255         return rc;
2256 }
2257
2258 /*
2259  * Tries to determine the default stripe attributes for a given filesystem. The
2260  * filesystem to check should be specified by fsname, or will be determined
2261  * using pathname.
2262  */
2263 static int sattr_get_defaults(const char *const fsname,
2264                               const char *const pathname,
2265                               unsigned int *scount,
2266                               unsigned int *ssize,
2267                               unsigned int *soffset)
2268 {
2269         int rc;
2270         char dpath[PATH_MAX + 1];
2271         char fpath[PATH_MAX + 1];
2272
2273         rc = clilovpath(fsname, pathname, dpath);
2274         if (rc != 0)
2275                 return rc;
2276
2277         if (scount) {
2278                 snprintf(fpath, PATH_MAX, "%s/stripecount", dpath);
2279                 rc = sattr_read_attr(fpath, scount);
2280                 if (rc != 0)
2281                         return rc;
2282         }
2283
2284         if (ssize) {
2285                 snprintf(fpath, PATH_MAX, "%s/stripesize", dpath);
2286                 rc = sattr_read_attr(fpath, ssize);
2287                 if (rc != 0)
2288                         return rc;
2289         }
2290
2291         if (soffset) {
2292                 snprintf(fpath, PATH_MAX, "%s/stripeoffset", dpath);
2293                 rc = sattr_read_attr(fpath, soffset);
2294                 if (rc != 0)
2295                         return rc;
2296         }
2297
2298         return 0;
2299 }
2300
2301 /*
2302  * Tries to gather the default stripe attributes for a given filesystem. If
2303  * the attributes can be determined, they are cached for easy retreival the
2304  * next time they are needed. Only a single filesystem's attributes are
2305  * cached at a time.
2306  */
2307 static int sattr_cache_get_defaults(const char *const fsname,
2308                                     const char *const pathname,
2309                                     unsigned int *scount,
2310                                     unsigned int *ssize,
2311                                     unsigned int *soffset)
2312 {
2313         static struct {
2314                 char fsname[PATH_MAX + 1];
2315                 unsigned int stripecount;
2316                 unsigned int stripesize;
2317                 unsigned int stripeoffset;
2318         } cache = {
2319                 .fsname = {'\0'}
2320         };
2321
2322         int rc;
2323         char fsname_buf[PATH_MAX + 1];
2324         unsigned int tmp[3];
2325
2326         if (fsname == NULL) {
2327                 rc = llapi_search_fsname(pathname, fsname_buf);
2328                 if (rc)
2329                         return rc;
2330         } else {
2331                 strlcpy(fsname_buf, fsname, sizeof(fsname_buf));
2332         }
2333
2334         if (strncmp(fsname_buf, cache.fsname, sizeof(fsname_buf) - 1) != 0) {
2335                 /*
2336                  * Ensure all 3 sattrs (count, size, and offset) are
2337                  * successfully retrieved and stored in tmp before writing to
2338                  * cache.
2339                  */
2340                 rc = sattr_get_defaults(fsname_buf, NULL, &tmp[0], &tmp[1],
2341                                         &tmp[2]);
2342                 if (rc != 0)
2343                         return rc;
2344
2345                 cache.stripecount = tmp[0];
2346                 cache.stripesize = tmp[1];
2347                 cache.stripeoffset = tmp[2];
2348                 strlcpy(cache.fsname, fsname_buf, sizeof(cache.fsname));
2349         }
2350
2351         if (scount)
2352                 *scount = cache.stripecount;
2353         if (ssize)
2354                 *ssize = cache.stripesize;
2355         if (soffset)
2356                 *soffset = cache.stripeoffset;
2357
2358         return 0;
2359 }
2360
2361 static void lov_dump_user_lmm_header(struct lov_user_md *lum, char *path,
2362                                      struct lov_user_ost_data_v1 *objects,
2363                                      int is_dir, int verbose, int depth,
2364                                      int raw, char *pool_name)
2365 {
2366         char *prefix = is_dir ? "" : "lmm_";
2367         char *seperator = "";
2368         int rc;
2369
2370         if (is_dir && lmm_oi_seq(&lum->lmm_oi) == FID_SEQ_LOV_DEFAULT) {
2371                 lmm_oi_set_seq(&lum->lmm_oi, 0);
2372                 if (verbose & VERBOSE_DETAIL)
2373                         llapi_printf(LLAPI_MSG_NORMAL, "(Default) ");
2374         }
2375
2376         if (depth && path && ((verbose != VERBOSE_OBJID) || !is_dir))
2377                 llapi_printf(LLAPI_MSG_NORMAL, "%s\n", path);
2378
2379         if ((verbose & VERBOSE_DETAIL) && !is_dir) {
2380                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_magic:          0x%08X\n",
2381                              lum->lmm_magic);
2382                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_seq:            "LPX64"\n",
2383                              lmm_oi_seq(&lum->lmm_oi));
2384                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_object_id:      "LPX64"\n",
2385                              lmm_oi_id(&lum->lmm_oi));
2386         }
2387
2388         if (verbose & VERBOSE_COUNT) {
2389                 if (verbose & ~VERBOSE_COUNT)
2390                         llapi_printf(LLAPI_MSG_NORMAL, "%sstripe_count:   ",
2391                                      prefix);
2392                 if (is_dir) {
2393                         if (!raw && lum->lmm_stripe_count == 0) {
2394                                 unsigned int scount;
2395                                 rc = sattr_cache_get_defaults(NULL, path,
2396                                                               &scount, NULL,
2397                                                               NULL);
2398                                 if (rc == 0)
2399                                         llapi_printf(LLAPI_MSG_NORMAL, "%d",
2400                                                      scount);
2401                                 else
2402                                         llapi_error(LLAPI_MSG_ERROR, rc,
2403                                                     "Cannot determine default"
2404                                                     " stripe count.");
2405                         } else {
2406                                 llapi_printf(LLAPI_MSG_NORMAL, "%d",
2407                                              lum->lmm_stripe_count ==
2408                                              (typeof(lum->lmm_stripe_count))(-1)
2409                                              ? -1 : lum->lmm_stripe_count);
2410                         }
2411                 } else {
2412                         llapi_printf(LLAPI_MSG_NORMAL, "%hd",
2413                                      (__s16)lum->lmm_stripe_count);
2414                 }
2415                 seperator = is_dir ? " " : "\n";
2416         }
2417
2418         if (verbose & VERBOSE_SIZE) {
2419                 llapi_printf(LLAPI_MSG_NORMAL, "%s", seperator);
2420                 if (verbose & ~VERBOSE_SIZE)
2421                         llapi_printf(LLAPI_MSG_NORMAL, "%sstripe_size:    ",
2422                                      prefix);
2423                 if (is_dir && !raw && lum->lmm_stripe_size == 0) {
2424                         unsigned int ssize;
2425                         rc = sattr_cache_get_defaults(NULL, path, NULL, &ssize,
2426                                                       NULL);
2427                         if (rc == 0)
2428                                 llapi_printf(LLAPI_MSG_NORMAL, "%u", ssize);
2429                         else
2430                                 llapi_error(LLAPI_MSG_ERROR, rc,
2431                                             "Cannot determine default"
2432                                             " stripe size.");
2433                 } else {
2434                         llapi_printf(LLAPI_MSG_NORMAL, "%u",
2435                                      lum->lmm_stripe_size);
2436                 }
2437                 seperator = is_dir ? " " : "\n";
2438         }
2439
2440         if ((verbose & VERBOSE_LAYOUT) && !is_dir) {
2441                 llapi_printf(LLAPI_MSG_NORMAL, "%s", seperator);
2442                 if (verbose & ~VERBOSE_LAYOUT)
2443                         llapi_printf(LLAPI_MSG_NORMAL, "%spattern:        ",
2444                                      prefix);
2445                 llapi_printf(LLAPI_MSG_NORMAL, "%.x", lum->lmm_pattern);
2446                 seperator = "\n";
2447         }
2448
2449         if ((verbose & VERBOSE_GENERATION) && !is_dir) {
2450                 llapi_printf(LLAPI_MSG_NORMAL, "%s", seperator);
2451                 if (verbose & ~VERBOSE_GENERATION)
2452                         llapi_printf(LLAPI_MSG_NORMAL, "%slayout_gen:     ",
2453                                      prefix);
2454                 llapi_printf(LLAPI_MSG_NORMAL, "%u",
2455                              (int)lum->lmm_layout_gen);
2456                 seperator = "\n";
2457         }
2458
2459         if (verbose & VERBOSE_OFFSET) {
2460                 llapi_printf(LLAPI_MSG_NORMAL, "%s", seperator);
2461                 if (verbose & ~VERBOSE_OFFSET)
2462                         llapi_printf(LLAPI_MSG_NORMAL, "%sstripe_offset:  ",
2463                                      prefix);
2464                 if (is_dir)
2465                         llapi_printf(LLAPI_MSG_NORMAL, "%d",
2466                                      lum->lmm_stripe_offset ==
2467                                      (typeof(lum->lmm_stripe_offset))(-1) ? -1 :
2468                                      lum->lmm_stripe_offset);
2469                 else
2470                         llapi_printf(LLAPI_MSG_NORMAL, "%u",
2471                                      objects[0].l_ost_idx);
2472                 seperator = is_dir ? " " : "\n";
2473         }
2474
2475         if ((verbose & VERBOSE_POOL) && (pool_name != NULL)) {
2476                 llapi_printf(LLAPI_MSG_NORMAL, "%s", seperator);
2477                 if (verbose & ~VERBOSE_POOL)
2478                         llapi_printf(LLAPI_MSG_NORMAL, "%spool:           ",
2479                                      prefix);
2480                 llapi_printf(LLAPI_MSG_NORMAL, "%s", pool_name);
2481         }
2482
2483         if (!is_dir || (is_dir && (verbose != VERBOSE_OBJID)))
2484                 llapi_printf(LLAPI_MSG_NORMAL, "\n");
2485 }
2486
2487 void lov_dump_user_lmm_v1v3(struct lov_user_md *lum, char *pool_name,
2488                             struct lov_user_ost_data_v1 *objects,
2489                             char *path, int is_dir, int obdindex,
2490                             int depth, int header, int raw)
2491 {
2492         int i, obdstripe = (obdindex != OBD_NOT_FOUND) ? 0 : 1;
2493
2494         if (!obdstripe) {
2495                 for (i = 0; !is_dir && i < lum->lmm_stripe_count; i++) {
2496                         if (obdindex == objects[i].l_ost_idx) {
2497                                 obdstripe = 1;
2498                                 break;
2499                         }
2500                 }
2501         }
2502
2503         if (obdstripe == 1)
2504                 lov_dump_user_lmm_header(lum, path, objects, is_dir, header,
2505                                          depth, raw, pool_name);
2506
2507         if (!is_dir && (header & VERBOSE_OBJID) &&
2508             !(lum->lmm_pattern & LOV_PATTERN_F_RELEASED)) {
2509                 if (obdstripe == 1)
2510                         llapi_printf(LLAPI_MSG_NORMAL,
2511                                    "\tobdidx\t\t objid\t\t objid\t\t group\n");
2512
2513                 for (i = 0; i < lum->lmm_stripe_count; i++) {
2514                         int idx = objects[i].l_ost_idx;
2515                         long long oid = ostid_id(&objects[i].l_ost_oi);
2516                         long long gr = ostid_seq(&objects[i].l_ost_oi);
2517                         if ((obdindex == OBD_NOT_FOUND) || (obdindex == idx)) {
2518                                 char fmt[48];
2519                                 sprintf(fmt, "%s%s%s\n",
2520                                         "\t%6u\t%14llu\t%#13llx\t",
2521                                         (fid_seq_is_rsvd(gr) ||
2522                                          fid_seq_is_mdt0(gr)) ?
2523                                          "%14llu" : "%#14llx", "%s");
2524                                 llapi_printf(LLAPI_MSG_NORMAL, fmt, idx, oid,
2525                                              oid, gr,
2526                                              obdindex == idx ? " *" : "");
2527                         }
2528
2529                 }
2530                 llapi_printf(LLAPI_MSG_NORMAL, "\n");
2531         }
2532 }
2533
2534 void lmv_dump_user_lmm(struct lmv_user_md *lum, char *pool_name,
2535                        char *path, int obdindex, int depth, int verbose)
2536 {
2537         struct lmv_user_mds_data *objects = lum->lum_objects;
2538         char *prefix = lum->lum_magic == LMV_USER_MAGIC ? "(Default)" : "";
2539         int i, obdstripe = 0;
2540         char *separator = "";
2541
2542         if (obdindex != OBD_NOT_FOUND) {
2543                 if (lum->lum_stripe_count == 0) {
2544                         if (obdindex == lum->lum_stripe_offset)
2545                                 obdstripe = 1;
2546                 } else {
2547                         for (i = 0; i < lum->lum_stripe_count; i++) {
2548                                 if (obdindex == objects[i].lum_mds) {
2549                                         llapi_printf(LLAPI_MSG_NORMAL,
2550                                                      "%s%s\n", prefix,
2551                                                      path);
2552                                         obdstripe = 1;
2553                                         break;
2554                                 }
2555                         }
2556                 }
2557         } else {
2558                 obdstripe = 1;
2559         }
2560
2561         if (!obdstripe)
2562                 return;
2563
2564         /* show all information default */
2565         if (!verbose) {
2566                 if (lum->lum_magic == LMV_USER_MAGIC)
2567                         verbose = VERBOSE_POOL | VERBOSE_COUNT | VERBOSE_OFFSET;
2568                 else
2569                         verbose = VERBOSE_OBJID;
2570         }
2571
2572         if (depth && path && ((verbose != VERBOSE_OBJID)))
2573                 llapi_printf(LLAPI_MSG_NORMAL, "%s%s\n", prefix, path);
2574
2575         if (verbose & VERBOSE_COUNT) {
2576                 llapi_printf(LLAPI_MSG_NORMAL, "%s", separator);
2577                 if (verbose & ~VERBOSE_COUNT)
2578                         llapi_printf(LLAPI_MSG_NORMAL, "lmv_stripe_count: ");
2579                 llapi_printf(LLAPI_MSG_NORMAL, "%u",
2580                              (int)lum->lum_stripe_count);
2581                 if (verbose & VERBOSE_OFFSET)
2582                         separator = " ";
2583                 else
2584                         separator = "\n";
2585         }
2586
2587         if (verbose & VERBOSE_OFFSET) {
2588                 llapi_printf(LLAPI_MSG_NORMAL, "%s", separator);
2589                 if (verbose & ~VERBOSE_OFFSET)
2590                         llapi_printf(LLAPI_MSG_NORMAL, "lmv_stripe_offset: ");
2591                 llapi_printf(LLAPI_MSG_NORMAL, "%d",
2592                              (int)lum->lum_stripe_offset);
2593                 separator = "\n";
2594         }
2595
2596         if (verbose & VERBOSE_OBJID && lum->lum_magic != LMV_USER_MAGIC) {
2597                 llapi_printf(LLAPI_MSG_NORMAL, "%s", separator);
2598                 if (obdstripe == 1 && lum->lum_stripe_count > 0)
2599                         llapi_printf(LLAPI_MSG_NORMAL,
2600                                      "mdtidx\t\t FID[seq:oid:ver]\n");
2601                 for (i = 0; i < lum->lum_stripe_count; i++) {
2602                         int idx = objects[i].lum_mds;
2603                         struct lu_fid *fid = &objects[i].lum_fid;
2604                         if ((obdindex == OBD_NOT_FOUND) || (obdindex == idx))
2605                                 llapi_printf(LLAPI_MSG_NORMAL,
2606                                              "%6u\t\t "DFID"\t\t%s\n",
2607                                             idx, PFID(fid),
2608                                             obdindex == idx ? " *" : "");
2609                 }
2610
2611         }
2612
2613         if ((verbose & VERBOSE_POOL) && (pool_name[0] != '\0')) {
2614                 llapi_printf(LLAPI_MSG_NORMAL, "%s", separator);
2615                 if (verbose & ~VERBOSE_POOL)
2616                         llapi_printf(LLAPI_MSG_NORMAL, "%slmv_pool:           ",
2617                                      prefix);
2618                 llapi_printf(LLAPI_MSG_NORMAL, "%s%c ", pool_name, ' ');
2619                 separator = "\n";
2620         }
2621
2622         if (!(verbose & VERBOSE_OBJID) || lum->lum_magic == LMV_USER_MAGIC)
2623                 llapi_printf(LLAPI_MSG_NORMAL, "\n");
2624 }
2625
2626 void llapi_lov_dump_user_lmm(struct find_param *param, char *path, int is_dir)
2627 {
2628         __u32 magic;
2629
2630         if (param->fp_get_lmv || param->fp_get_default_lmv)
2631                 magic = (__u32)param->fp_lmv_md->lum_magic;
2632         else
2633                 magic = *(__u32 *)&param->fp_lmd->lmd_lmm; /* lum->lmm_magic */
2634
2635         switch (magic) {
2636         case LOV_USER_MAGIC_V1:
2637                 lov_dump_user_lmm_v1v3(&param->fp_lmd->lmd_lmm, NULL,
2638                                        param->fp_lmd->lmd_lmm.lmm_objects,
2639                                        path, is_dir,
2640                                        param->fp_obd_index, param->fp_max_depth,
2641                                        param->fp_verbose, param->fp_raw);
2642                 break;
2643         case LOV_USER_MAGIC_V3: {
2644                 char pool_name[LOV_MAXPOOLNAME + 1];
2645                 struct lov_user_ost_data_v1 *objects;
2646                 struct lov_user_md_v3 *lmmv3 = (void *)&param->fp_lmd->lmd_lmm;
2647
2648                 strlcpy(pool_name, lmmv3->lmm_pool_name, sizeof(pool_name));
2649                 objects = lmmv3->lmm_objects;
2650                 lov_dump_user_lmm_v1v3(&param->fp_lmd->lmd_lmm, pool_name,
2651                                        objects, path, is_dir,
2652                                        param->fp_obd_index, param->fp_max_depth,
2653                                        param->fp_verbose, param->fp_raw);
2654                 break;
2655         }
2656         case LMV_MAGIC_V1:
2657         case LMV_USER_MAGIC: {
2658                 char pool_name[LOV_MAXPOOLNAME + 1];
2659                 struct lmv_user_md *lum;
2660
2661                 lum = (struct lmv_user_md *)param->fp_lmv_md;
2662                 strlcpy(pool_name, lum->lum_pool_name, sizeof(pool_name));
2663                 lmv_dump_user_lmm(lum, pool_name, path,
2664                                   param->fp_obd_index, param->fp_max_depth,
2665                                   param->fp_verbose);
2666                 break;
2667         }
2668         default:
2669                 llapi_printf(LLAPI_MSG_NORMAL, "unknown lmm_magic:  %#x "
2670                              "(expecting one of %#x %#x %#x %#x)\n",
2671                              *(__u32 *)&param->fp_lmd->lmd_lmm,
2672                              LOV_USER_MAGIC_V1, LOV_USER_MAGIC_V3,
2673                              LMV_USER_MAGIC, LMV_MAGIC_V1);
2674                 return;
2675         }
2676 }
2677
2678 int llapi_file_get_stripe(const char *path, struct lov_user_md *lum)
2679 {
2680         const char *fname;
2681         char *dname;
2682         int fd, rc = 0;
2683
2684         fname = strrchr(path, '/');
2685
2686         /* It should be a file (or other non-directory) */
2687         if (fname == NULL) {
2688                 dname = (char *)malloc(2);
2689                 if (dname == NULL)
2690                         return -ENOMEM;
2691                 strcpy(dname, ".");
2692                 fname = (char *)path;
2693         } else {
2694                 dname = (char *)malloc(fname - path + 1);
2695                 if (dname == NULL)
2696                         return -ENOMEM;
2697                 strncpy(dname, path, fname - path);
2698                 dname[fname - path] = '\0';
2699                 fname++;
2700         }
2701
2702         fd = open(dname, O_RDONLY);
2703         if (fd == -1) {
2704                 rc = -errno;
2705                 free(dname);
2706                 return rc;
2707         }
2708
2709         strcpy((char *)lum, fname);
2710         if (ioctl(fd, IOC_MDC_GETFILESTRIPE, (void *)lum) == -1)
2711                 rc = -errno;
2712
2713         if (close(fd) == -1 && rc == 0)
2714                 rc = -errno;
2715
2716         free(dname);
2717         return rc;
2718 }
2719
2720 int llapi_file_lookup(int dirfd, const char *name)
2721 {
2722         struct obd_ioctl_data data = { 0 };
2723         char rawbuf[8192];
2724         char *buf = rawbuf;
2725         int rc;
2726
2727         if (dirfd < 0 || name == NULL)
2728                 return -EINVAL;
2729
2730         data.ioc_version = OBD_IOCTL_VERSION;
2731         data.ioc_len = sizeof(data);
2732         data.ioc_inlbuf1 = (char *)name;
2733         data.ioc_inllen1 = strlen(name) + 1;
2734
2735         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2736         if (rc) {
2737                 llapi_error(LLAPI_MSG_ERROR, rc,
2738                             "error: IOC_MDC_LOOKUP pack failed for '%s': rc %d",
2739                             name, rc);
2740                 return rc;
2741         }
2742
2743         rc = ioctl(dirfd, IOC_MDC_LOOKUP, buf);
2744         if (rc < 0)
2745                 rc = -errno;
2746         return rc;
2747 }
2748
2749 /* Check if the value matches 1 of the given criteria (e.g. --atime +/-N).
2750  * @mds indicates if this is MDS timestamps and there are attributes on OSTs.
2751  *
2752  * The result is -1 if it does not match, 0 if not yet clear, 1 if matches.
2753  * The table below gives the answers for the specified parameters (value and
2754  * sign), 1st column is the answer for the MDS value, the 2nd is for the OST:
2755  * --------------------------------------
2756  * 1 | file > limit; sign > 0 | -1 / -1 |
2757  * 2 | file = limit; sign > 0 | -1 / -1 |
2758  * 3 | file < limit; sign > 0 |  ? /  1 |
2759  * 4 | file > limit; sign = 0 | -1 / -1 |
2760  * 5 | file = limit; sign = 0 |  ? /  1 |  <- (see the Note below)
2761  * 6 | file < limit; sign = 0 |  ? / -1 |
2762  * 7 | file > limit; sign < 0 |  1 /  1 |
2763  * 8 | file = limit; sign < 0 |  ? / -1 |
2764  * 9 | file < limit; sign < 0 |  ? / -1 |
2765  * --------------------------------------
2766  * Note: 5th actually means that the value is within the interval
2767  * (limit - margin, limit]. */
2768 static int find_value_cmp(unsigned long long file, unsigned long long limit,
2769                           int sign, int negopt, unsigned long long margin,
2770                           int mds)
2771 {
2772         int ret = -1;
2773
2774         if (sign > 0) {
2775                 /* Drop the fraction of margin (of days). */
2776                 if (file + margin <= limit)
2777                         ret = mds ? 0 : 1;
2778         } else if (sign == 0) {
2779                 if (file <= limit && file + margin > limit)
2780                         ret = mds ? 0 : 1;
2781                 else if (file + margin <= limit)
2782                         ret = mds ? 0 : -1;
2783         } else if (sign < 0) {
2784                 if (file > limit)
2785                         ret = 1;
2786                 else if (mds)
2787                         ret = 0;
2788         }
2789
2790         return negopt ? ~ret + 1 : ret;
2791 }
2792
2793 /* Check if the file time matches all the given criteria (e.g. --atime +/-N).
2794  * Return -1 or 1 if file timestamp does not or does match the given criteria
2795  * correspondingly. Return 0 if the MDS time is being checked and there are
2796  * attributes on OSTs and it is not yet clear if the timespamp matches.
2797  *
2798  * If 0 is returned, we need to do another RPC to the OSTs to obtain the
2799  * updated timestamps. */
2800 static int find_time_check(lstat_t *st, struct find_param *param, int mds)
2801 {
2802         int rc = 1;
2803         int rc2;
2804
2805         /* Check if file is accepted. */
2806         if (param->fp_atime) {
2807                 rc2 = find_value_cmp(st->st_atime, param->fp_atime,
2808                                      param->fp_asign, param->fp_exclude_atime,
2809                                      24 * 60 * 60, mds);
2810                 if (rc2 < 0)
2811                         return rc2;
2812                 rc = rc2;
2813         }
2814
2815         if (param->fp_mtime) {
2816                 rc2 = find_value_cmp(st->st_mtime, param->fp_mtime,
2817                                      param->fp_msign, param->fp_exclude_mtime,
2818                                      24 * 60 * 60, mds);
2819                 if (rc2 < 0)
2820                         return rc2;
2821
2822                 /* If the previous check matches, but this one is not yet clear,
2823                  * we should return 0 to do an RPC on OSTs. */
2824                 if (rc == 1)
2825                         rc = rc2;
2826         }
2827
2828         if (param->fp_ctime) {
2829                 rc2 = find_value_cmp(st->st_ctime, param->fp_ctime,
2830                                      param->fp_csign, param->fp_exclude_ctime,
2831                                      24 * 60 * 60, mds);
2832                 if (rc2 < 0)
2833                         return rc2;
2834
2835                 /* If the previous check matches, but this one is not yet clear,
2836                  * we should return 0 to do an RPC on OSTs. */
2837                 if (rc == 1)
2838                         rc = rc2;
2839         }
2840
2841         return rc;
2842 }
2843
2844 /**
2845  * Check whether the stripes matches the indexes user provided
2846  *       1   : matched
2847  *       0   : Unmatched
2848  */
2849 static int check_obd_match(struct find_param *param)
2850 {
2851         lstat_t *st = &param->fp_lmd->lmd_st;
2852         struct lov_user_ost_data_v1 *lmm_objects;
2853         int i, j;
2854
2855         if (param->fp_obd_uuid && param->fp_obd_index == OBD_NOT_FOUND)
2856                 return 0;
2857
2858         if (!S_ISREG(st->st_mode))
2859                 return 0;
2860
2861         /* Only those files should be accepted, which have a
2862          * stripe on the specified OST. */
2863         if (!param->fp_lmd->lmd_lmm.lmm_stripe_count)
2864                 return 0;
2865
2866         if (param->fp_lmd->lmd_lmm.lmm_magic ==
2867             LOV_USER_MAGIC_V3) {
2868                 struct lov_user_md_v3 *lmmv3 = (void *)&param->fp_lmd->lmd_lmm;
2869
2870                 lmm_objects = lmmv3->lmm_objects;
2871         } else if (param->fp_lmd->lmd_lmm.lmm_magic ==  LOV_USER_MAGIC_V1) {
2872                 lmm_objects = param->fp_lmd->lmd_lmm.lmm_objects;
2873         } else {
2874                 llapi_err_noerrno(LLAPI_MSG_ERROR, "%s:Unknown magic: 0x%08X\n",
2875                                   __func__, param->fp_lmd->lmd_lmm.lmm_magic);
2876                 return -EINVAL;
2877         }
2878
2879         for (i = 0; i < param->fp_lmd->lmd_lmm.lmm_stripe_count; i++) {
2880                 for (j = 0; j < param->fp_num_obds; j++) {
2881                         if (param->fp_obd_indexes[j] ==
2882                             lmm_objects[i].l_ost_idx) {
2883                                 if (param->fp_exclude_obd)
2884                                         return 0;
2885                                 return 1;
2886                         }
2887                 }
2888         }
2889
2890         if (param->fp_exclude_obd)
2891                 return 1;
2892
2893         return 0;
2894 }
2895
2896 static int check_mdt_match(struct find_param *param)
2897 {
2898         int i;
2899
2900         if (param->fp_mdt_uuid && param->fp_mdt_index == OBD_NOT_FOUND)
2901                 return 0;
2902
2903         /* FIXME: For striped dir, we should get stripe information and check */
2904         for (i = 0; i < param->fp_num_mdts; i++) {
2905                 if (param->fp_mdt_indexes[i] == param->fp_file_mdt_index)
2906                         return !param->fp_exclude_mdt;
2907         }
2908
2909         if (param->fp_exclude_mdt)
2910                 return 1;
2911
2912         return 0;
2913 }
2914
2915 /**
2916  * Check whether the obd is active or not, if it is
2917  * not active, just print the object affected by this
2918  * failed target
2919  **/
2920 static int print_failed_tgt(struct find_param *param, char *path, int type)
2921 {
2922         struct obd_statfs stat_buf;
2923         struct obd_uuid uuid_buf;
2924         int ret;
2925
2926         LASSERT(type == LL_STATFS_LOV || type == LL_STATFS_LMV);
2927
2928         memset(&stat_buf, 0, sizeof(struct obd_statfs));
2929         memset(&uuid_buf, 0, sizeof(struct obd_uuid));
2930         ret = llapi_obd_statfs(path, type,
2931                                param->fp_obd_index, &stat_buf,
2932                                &uuid_buf);
2933         if (ret) {
2934                 llapi_printf(LLAPI_MSG_NORMAL,
2935                              "obd_uuid: %s failed %s ",
2936                              param->fp_obd_uuid->uuid,
2937                              strerror(errno));
2938         }
2939
2940         return ret;
2941 }
2942
2943 static int cb_find_init(char *path, DIR *parent, DIR **dirp,
2944                         void *data, struct dirent64 *de)
2945 {
2946         struct find_param *param = (struct find_param *)data;
2947         DIR *dir = dirp == NULL ? NULL : *dirp;
2948         int decision = 1; /* 1 is accepted; -1 is rejected. */
2949         lstat_t *st = &param->fp_lmd->lmd_st;
2950         int lustre_fs = 1;
2951         int checked_type = 0;
2952         int ret = 0;
2953
2954         LASSERT(parent != NULL || dir != NULL);
2955
2956         param->fp_lmd->lmd_lmm.lmm_stripe_count = 0;
2957
2958         /* If a regular expression is presented, make the initial decision */
2959         if (param->fp_pattern != NULL) {
2960                 char *fname = strrchr(path, '/');
2961                 fname = (fname == NULL ? path : fname + 1);
2962                 ret = fnmatch(param->fp_pattern, fname, 0);
2963                 if ((ret == FNM_NOMATCH && !param->fp_exclude_pattern) ||
2964                     (ret == 0 && param->fp_exclude_pattern))
2965                         goto decided;
2966         }
2967
2968         /* See if we can check the file type from the dirent. */
2969         if (param->fp_type != 0 && de != NULL && de->d_type != DT_UNKNOWN) {
2970                 checked_type = 1;
2971
2972                 if (DTTOIF(de->d_type) == param->fp_type) {
2973                         if (param->fp_exclude_type)
2974                                 goto decided;
2975                 } else {
2976                         if (!param->fp_exclude_type)
2977                                 goto decided;
2978                 }
2979         }
2980
2981         ret = 0;
2982
2983         /* Request MDS for the stat info if some of these parameters need
2984          * to be compared. */
2985         if (param->fp_obd_uuid || param->fp_mdt_uuid ||
2986             param->fp_check_uid || param->fp_check_gid ||
2987             param->fp_atime || param->fp_mtime || param->fp_ctime ||
2988             param->fp_check_pool || param->fp_check_size ||
2989             param->fp_check_stripe_count || param->fp_check_stripe_size ||
2990             param->fp_check_layout)
2991                 decision = 0;
2992
2993         if (param->fp_type != 0 && checked_type == 0)
2994                 decision = 0;
2995
2996         if (decision == 0) {
2997                 ret = get_lmd_info(path, parent, dir, param->fp_lmd,
2998                                    param->fp_lum_size);
2999                 if (ret == 0 && param->fp_mdt_uuid != NULL) {
3000                         if (dir != NULL) {
3001                                 ret = llapi_file_fget_mdtidx(dirfd(dir),
3002                                                      &param->fp_file_mdt_index);
3003                         } else if (S_ISREG(st->st_mode)) {
3004                                 int fd;
3005
3006                                 /* FIXME: we could get the MDT index from the
3007                                  * file's FID in lmd->lmd_lmm.lmm_oi without
3008                                  * opening the file, once we are sure that
3009                                  * LFSCK2 (2.6) has fixed up pre-2.0 LOV EAs.
3010                                  * That would still be an ioctl() to map the
3011                                  * FID to the MDT, but not an open RPC. */
3012                                 fd = open(path, O_RDONLY);
3013                                 if (fd > 0) {
3014                                         ret = llapi_file_fget_mdtidx(fd,
3015                                                      &param->fp_file_mdt_index);
3016                                         close(fd);
3017                                 } else {
3018                                         ret = -errno;
3019                                 }
3020                         } else {
3021                                 /* For a special file, we assume it resides on
3022                                  * the same MDT as the parent directory. */
3023                                 ret = llapi_file_fget_mdtidx(dirfd(parent),
3024                                                      &param->fp_file_mdt_index);
3025                         }
3026                 }
3027                 if (ret != 0) {
3028                         if (ret == -ENOTTY)
3029                                 lustre_fs = 0;
3030                         if (ret == -ENOENT)
3031                                 goto decided;
3032
3033                         return ret;
3034                 }
3035         }
3036
3037         if (param->fp_type && !checked_type) {
3038                 if ((st->st_mode & S_IFMT) == param->fp_type) {
3039                         if (param->fp_exclude_type)
3040                                 goto decided;
3041                 } else {
3042                         if (!param->fp_exclude_type)
3043                                 goto decided;
3044                 }
3045         }
3046
3047         /* Prepare odb. */
3048         if (param->fp_obd_uuid || param->fp_mdt_uuid) {
3049                 if (lustre_fs && param->fp_got_uuids &&
3050                     param->fp_dev != st->st_dev) {
3051                         /* A lustre/lustre mount point is crossed. */
3052                         param->fp_got_uuids = 0;
3053                         param->fp_obds_printed = 0;
3054                         param->fp_mdt_index = OBD_NOT_FOUND;
3055                         param->fp_obd_index = OBD_NOT_FOUND;
3056                 }
3057
3058                 if (lustre_fs && !param->fp_got_uuids) {
3059                         ret = setup_target_indexes(dir ? dir : parent, path,
3060                                                    param);
3061                         if (ret)
3062                                 return ret;
3063
3064                         param->fp_dev = st->st_dev;
3065                 } else if (!lustre_fs && param->fp_got_uuids) {
3066                         /* A lustre/non-lustre mount point is crossed. */
3067                         param->fp_got_uuids = 0;
3068                         param->fp_mdt_index = OBD_NOT_FOUND;
3069                         param->fp_obd_index = OBD_NOT_FOUND;
3070                 }
3071         }
3072
3073         if (param->fp_check_stripe_size) {
3074                 decision = find_value_cmp(
3075                                 param->fp_lmd->lmd_lmm.lmm_stripe_size,
3076                                 param->fp_stripe_size,
3077                                 param->fp_stripe_size_sign,
3078                                 param->fp_exclude_stripe_size,
3079                                 param->fp_stripe_size_units, 0);
3080                 if (decision == -1)
3081                         goto decided;
3082         }
3083
3084         if (param->fp_check_stripe_count) {
3085                 decision = find_value_cmp(
3086                                 param->fp_lmd->lmd_lmm.lmm_stripe_count,
3087                                 param->fp_stripe_count,
3088                                 param->fp_stripe_count_sign,
3089                                 param->fp_exclude_stripe_count, 1, 0);
3090                 if (decision == -1)
3091                         goto decided;
3092         }
3093
3094         if (param->fp_check_layout) {
3095                 __u32 found;
3096
3097                 found = (param->fp_lmd->lmd_lmm.lmm_pattern & param->fp_layout);
3098                 if ((param->fp_lmd->lmd_lmm.lmm_pattern == 0xFFFFFFFF) ||
3099                     (found && param->fp_exclude_layout) ||
3100                     (!found && !param->fp_exclude_layout)) {
3101                         decision = -1;
3102                         goto decided;
3103                 }
3104         }
3105
3106         /* If an OBD UUID is specified but none matches, skip this file. */
3107         if ((param->fp_obd_uuid && param->fp_obd_index == OBD_NOT_FOUND) ||
3108             (param->fp_mdt_uuid && param->fp_mdt_index == OBD_NOT_FOUND))
3109                 goto decided;
3110
3111         /* If a OST or MDT UUID is given, and some OST matches,
3112          * check it here. */
3113         if (param->fp_obd_index != OBD_NOT_FOUND ||
3114             param->fp_mdt_index != OBD_NOT_FOUND) {
3115                 if (param->fp_obd_uuid) {
3116                         if (check_obd_match(param)) {
3117                                 /* If no mdtuuid is given, we are done.
3118                                  * Otherwise, fall through to the mdtuuid
3119                                  * check below. */
3120                                 if (!param->fp_mdt_uuid)
3121                                         goto obd_matches;
3122                         } else {
3123                                 goto decided;
3124                         }
3125                 }
3126
3127                 if (param->fp_mdt_uuid) {
3128                         if (check_mdt_match(param))
3129                                 goto obd_matches;
3130                         goto decided;
3131                 }
3132         }
3133
3134 obd_matches:
3135         if (param->fp_check_uid) {
3136                 if (st->st_uid == param->fp_uid) {
3137                         if (param->fp_exclude_uid)
3138                                 goto decided;
3139                 } else {
3140                         if (!param->fp_exclude_uid)
3141                                 goto decided;
3142                 }
3143         }
3144
3145         if (param->fp_check_gid) {
3146                 if (st->st_gid == param->fp_gid) {
3147                         if (param->fp_exclude_gid)
3148                                 goto decided;
3149                 } else {
3150                         if (!param->fp_exclude_gid)
3151                                 goto decided;
3152                 }
3153         }
3154
3155         if (param->fp_check_pool) {
3156                 struct lov_user_md_v3 *lmmv3 = (void *)&param->fp_lmd->lmd_lmm;
3157
3158                 /* empty requested pool is taken as no pool search => V1 */
3159                 if (((param->fp_lmd->lmd_lmm.lmm_magic == LOV_USER_MAGIC_V1) &&
3160                      (param->fp_poolname[0] == '\0')) ||
3161                     ((param->fp_lmd->lmd_lmm.lmm_magic == LOV_USER_MAGIC_V3) &&
3162                      (strncmp(lmmv3->lmm_pool_name,
3163                               param->fp_poolname, LOV_MAXPOOLNAME) == 0)) ||
3164                     ((param->fp_lmd->lmd_lmm.lmm_magic == LOV_USER_MAGIC_V3) &&
3165                      (strcmp(param->fp_poolname, "*") == 0))) {
3166                         if (param->fp_exclude_pool)
3167                                 goto decided;
3168                 } else {
3169                         if (!param->fp_exclude_pool)
3170                                 goto decided;
3171                 }
3172         }
3173
3174         /* Check the time on mds. */
3175         decision = 1;
3176         if (param->fp_atime || param->fp_mtime || param->fp_ctime) {
3177                 int for_mds;
3178
3179                 for_mds = lustre_fs ? (S_ISREG(st->st_mode) &&
3180                                        param->fp_lmd->lmd_lmm.lmm_stripe_count)
3181                         : 0;
3182                 decision = find_time_check(st, param, for_mds);
3183                 if (decision == -1)
3184                         goto decided;
3185         }
3186
3187         /* If file still fits the request, ask ost for updated info.
3188            The regular stat is almost of the same speed as some new
3189            'glimpse-size-ioctl'. */
3190
3191         if (param->fp_check_size && S_ISREG(st->st_mode) &&
3192             param->fp_lmd->lmd_lmm.lmm_stripe_count)
3193                 decision = 0;
3194
3195         if (param->fp_check_size && S_ISDIR(st->st_mode))
3196                 decision = 0;
3197
3198         if (!decision) {
3199                 /* For regular files with the stripe the decision may have not
3200                  * been taken yet if *time or size is to be checked. */
3201                 if (param->fp_obd_index != OBD_NOT_FOUND)
3202                         print_failed_tgt(param, path, LL_STATFS_LOV);
3203
3204                 if (param->fp_mdt_index != OBD_NOT_FOUND)
3205                         print_failed_tgt(param, path, LL_STATFS_LMV);
3206
3207                 if (S_ISDIR(st->st_mode))
3208                         ret = fstat_f(dirfd(dir), st);
3209                 else if (dir != NULL)
3210                         ret = ioctl(dirfd(dir), IOC_LOV_GETINFO,
3211                                     (void *)param->fp_lmd);
3212                 else
3213                         ret = ioctl(dirfd(parent), IOC_LOV_GETINFO,
3214                                     (void *)param->fp_lmd);
3215
3216                 if (ret) {
3217                         if (errno == ENOENT) {
3218                                 llapi_error(LLAPI_MSG_ERROR, -ENOENT,
3219                                             "warning: %s: %s does not exist",
3220                                             __func__, path);
3221                                 goto decided;
3222                         } else {
3223                                 ret = -errno;
3224                                 llapi_error(LLAPI_MSG_ERROR, ret,
3225                                             "%s: IOC_LOV_GETINFO on %s failed",
3226                                             __func__, path);
3227                                 return ret;
3228                         }
3229                 }
3230
3231                 /* Check the time on osc. */
3232                 decision = find_time_check(st, param, 0);
3233                 if (decision == -1)
3234                         goto decided;
3235         }
3236
3237         if (param->fp_check_size)
3238                 decision = find_value_cmp(st->st_size, param->fp_size,
3239                                           param->fp_size_sign,
3240                                           param->fp_exclude_size,
3241                                           param->fp_size_units, 0);
3242
3243         if (decision != -1) {
3244                 llapi_printf(LLAPI_MSG_NORMAL, "%s", path);
3245                 if (param->fp_zero_end)
3246                         llapi_printf(LLAPI_MSG_NORMAL, "%c", '\0');
3247                 else
3248                         llapi_printf(LLAPI_MSG_NORMAL, "\n");
3249         }
3250
3251 decided:
3252         /* Do not get down anymore? */
3253         if (param->fp_depth == param->fp_max_depth)
3254                 return 1;
3255
3256         param->fp_depth++;
3257
3258         return 0;
3259 }
3260
3261 static int cb_mv_init(char *path, DIR *parent, DIR **dirp,
3262                       void *param_data, struct dirent64 *de)
3263 {
3264         struct find_param       *param = (struct find_param *)param_data;
3265         DIR                     *dir = parent;
3266         char                    raw[OBD_MAX_IOCTL_BUFFER] = {'\0'};
3267         char                    *rawbuf = raw;
3268         struct obd_ioctl_data   data = { 0 };
3269         int                     fd;
3270         int                     ret;
3271         char                    *filename;
3272
3273         LASSERT(parent != NULL || dirp != NULL);
3274         if (dirp != NULL)
3275                 closedir(*dirp);
3276
3277         if (parent == NULL) {
3278                 dir = opendir_parent(path);
3279                 if (dir == NULL) {
3280                         *dirp = NULL;
3281                         ret = -errno;
3282                         llapi_error(LLAPI_MSG_ERROR, ret,
3283                                     "can not open %s", path);
3284                         return ret;
3285                 }
3286         }
3287
3288         fd = dirfd(dir);
3289
3290         filename = basename(path);
3291         data.ioc_inlbuf1 = (char *)filename;
3292         data.ioc_inllen1 = strlen(filename) + 1;
3293         data.ioc_inlbuf2 = (char *)&param->fp_mdt_index;
3294         data.ioc_inllen2 = sizeof(param->fp_mdt_index);
3295         ret = obd_ioctl_pack(&data, &rawbuf, sizeof(raw));
3296         if (ret != 0) {
3297                 llapi_error(LLAPI_MSG_ERROR, ret,
3298                             "llapi_obd_statfs: error packing ioctl data");
3299                 goto out;
3300         }
3301
3302         ret = ioctl(fd, LL_IOC_MIGRATE, rawbuf);
3303         if (ret != 0) {
3304                 ret = -errno;
3305                 fprintf(stderr, "%s migrate failed %d\n", path, ret);
3306                 goto out;
3307         } else if (param->fp_verbose & VERBOSE_DETAIL) {
3308                 fprintf(stdout, "migrate %s to MDT%d\n",
3309                         path, param->fp_mdt_index);
3310         }
3311
3312 out:
3313         if (dirp != NULL) {
3314                 /* If the directory is being migration, we need
3315                  * close the directory after migration,
3316                  * so the old directory cache will be cleanup
3317                  * on the client side, and re-open to get the
3318                  * new directory handle */
3319                 *dirp = opendir(path);
3320                 if (dirp == NULL) {
3321                         ret = -errno;
3322                         llapi_error(LLAPI_MSG_ERROR, ret,
3323                                     "%s: Failed to open '%s'", __func__, path);
3324                         return ret;
3325                 }
3326         }
3327
3328         if (parent == NULL)
3329                 closedir(dir);
3330
3331         return ret;
3332 }
3333
3334 int llapi_mv(char *path, struct find_param *param)
3335 {
3336         return param_callback(path, cb_mv_init, cb_common_fini, param);
3337 }
3338
3339 int llapi_find(char *path, struct find_param *param)
3340 {
3341         return param_callback(path, cb_find_init, cb_common_fini, param);
3342 }
3343
3344 /*
3345  * Get MDT number that the file/directory inode referenced
3346  * by the open fd resides on.
3347  * Return 0 and mdtidx on success, or -ve errno.
3348  */
3349 int llapi_file_fget_mdtidx(int fd, int *mdtidx)
3350 {
3351         if (ioctl(fd, LL_IOC_GET_MDTIDX, mdtidx) < 0)
3352                 return -errno;
3353         return 0;
3354 }
3355
3356 static int cb_get_mdt_index(char *path, DIR *parent, DIR **dirp, void *data,
3357                             struct dirent64 *de)
3358 {
3359         struct find_param *param = (struct find_param *)data;
3360         DIR *d = dirp == NULL ? NULL : *dirp;
3361         int ret;
3362         int mdtidx;
3363
3364         LASSERT(parent != NULL || d != NULL);
3365
3366         if (d != NULL) {
3367                 ret = llapi_file_fget_mdtidx(dirfd(d), &mdtidx);
3368         } else /* if (parent) */ {
3369                 int fd;
3370
3371                 fd = open(path, O_RDONLY | O_NOCTTY);
3372                 if (fd > 0) {
3373                         ret = llapi_file_fget_mdtidx(fd, &mdtidx);
3374                         close(fd);
3375                 } else {
3376                         ret = -errno;
3377                 }
3378         }
3379
3380         if (ret != 0) {
3381                 if (ret == -ENODATA) {
3382                         if (!param->fp_obd_uuid)
3383                                 llapi_printf(LLAPI_MSG_NORMAL,
3384                                              "'%s' has no stripe info\n", path);
3385                         goto out;
3386                 } else if (ret == -ENOENT) {
3387                         llapi_error(LLAPI_MSG_WARN, ret,
3388                                     "warning: %s: '%s' does not exist",
3389                                     __func__, path);
3390                         goto out;
3391                 } else if (ret == -ENOTTY) {
3392                         llapi_error(LLAPI_MSG_ERROR, ret,
3393                                     "%s: '%s' not on a Lustre fs",
3394                                     __func__, path);
3395                 } else {
3396                         llapi_error(LLAPI_MSG_ERROR, ret,
3397                                     "error: %s: '%s' failed get_mdtidx",
3398                                     __func__, path);
3399                 }
3400                 return ret;
3401         }
3402
3403         if (param->fp_quiet || !(param->fp_verbose & VERBOSE_DETAIL))
3404                 llapi_printf(LLAPI_MSG_NORMAL, "%d\n", mdtidx);
3405         else
3406                 llapi_printf(LLAPI_MSG_NORMAL, "%s\nmdt_index:\t%d\n",
3407                              path, mdtidx);
3408
3409 out:
3410         /* Do not go down anymore? */
3411         if (param->fp_depth == param->fp_max_depth)
3412                 return 1;
3413
3414         param->fp_depth++;
3415
3416         return 0;
3417 }
3418
3419 static int cb_getstripe(char *path, DIR *parent, DIR **dirp, void *data,
3420                         struct dirent64 *de)
3421 {
3422         struct find_param *param = (struct find_param *)data;
3423         DIR *d = dirp == NULL ? NULL : *dirp;
3424         int ret = 0;
3425
3426         LASSERT(parent != NULL || d != NULL);
3427
3428         if (param->fp_obd_uuid) {
3429                 param->fp_quiet = 1;
3430                 ret = setup_obd_uuid(d ? d : parent, path, param);
3431                 if (ret)
3432                         return ret;
3433         }
3434
3435         if (d) {
3436                 if (param->fp_get_lmv || param->fp_get_default_lmv) {
3437                         ret = cb_get_dirstripe(path, d, param);
3438                 } else {
3439                         ret = ioctl(dirfd(d), LL_IOC_LOV_GETSTRIPE,
3440                                      (void *)&param->fp_lmd->lmd_lmm);
3441                 }
3442
3443         } else if (parent && !param->fp_get_lmv && !param->fp_get_default_lmv) {
3444                 char *fname = strrchr(path, '/');
3445                 fname = (fname == NULL ? path : fname + 1);
3446
3447                 strlcpy((char *)&param->fp_lmd->lmd_lmm, fname,
3448                         param->fp_lum_size);
3449
3450                 ret = ioctl(dirfd(parent), IOC_MDC_GETFILESTRIPE,
3451                             (void *)&param->fp_lmd->lmd_lmm);
3452         } else {
3453                 return 0;
3454         }
3455
3456         if (ret) {
3457                 if (errno == ENODATA && d != NULL) {
3458                         /* We need to "fake" the "use the default" values
3459                          * since the lmm struct is zeroed out at this point.
3460                          * The magic needs to be set in order to satisfy
3461                          * a check later on in the code path.
3462                          * The object_seq needs to be set for the "(Default)"
3463                          * prefix to be displayed. */
3464                         if (param->fp_get_default_lmv) {
3465                                 struct lmv_user_md *lum = param->fp_lmv_md;
3466
3467                                 lum->lum_magic = LMV_USER_MAGIC;
3468                                 lum->lum_stripe_count = 0;
3469                                 lum->lum_stripe_offset = -1;
3470                                 goto dump;
3471                         } else if (param->fp_get_lmv) {
3472                                 struct lmv_user_md *lum = param->fp_lmv_md;
3473                                 int mdtidx;
3474
3475                                 ret = llapi_file_fget_mdtidx(dirfd(d), &mdtidx);
3476                                 if (ret != 0)
3477                                         goto err_out;
3478                                 lum->lum_magic = LMV_MAGIC_V1;
3479                                 lum->lum_stripe_count = 0;
3480                                 lum->lum_stripe_offset = mdtidx;
3481                                 goto dump;
3482                         } else {
3483                                 struct lov_user_md *lmm =
3484                                         &param->fp_lmd->lmd_lmm;
3485
3486                                 lmm->lmm_magic = LOV_USER_MAGIC_V1;
3487                                 if (!param->fp_raw)
3488                                         ostid_set_seq(&lmm->lmm_oi,
3489                                                       FID_SEQ_LOV_DEFAULT);
3490                                 lmm->lmm_stripe_count = 0;
3491                                 lmm->lmm_stripe_size = 0;
3492                                 lmm->lmm_stripe_offset = -1;
3493                                 goto dump;
3494                         }
3495                 } else if (errno == ENODATA && parent != NULL) {
3496                         if (!param->fp_obd_uuid && !param->fp_mdt_uuid)
3497                                 llapi_printf(LLAPI_MSG_NORMAL,
3498                                              "%s has no stripe info\n", path);
3499                         goto out;
3500                 } else if (errno == ENOENT) {
3501                         llapi_error(LLAPI_MSG_WARN, -ENOENT,
3502                                     "warning: %s: %s does not exist",
3503                                     __func__, path);
3504                         goto out;
3505                 } else if (errno == ENOTTY) {
3506                         ret = -errno;
3507                         llapi_error(LLAPI_MSG_ERROR, ret,
3508                                     "%s: '%s' not on a Lustre fs?",
3509                                     __func__, path);
3510                 } else {
3511                         ret = -errno;
3512 err_out:
3513                         llapi_error(LLAPI_MSG_ERROR, ret,
3514                                     "error: %s: %s failed for %s",
3515                                      __func__, d ? "LL_IOC_LOV_GETSTRIPE" :
3516                                     "IOC_MDC_GETFILESTRIPE", path);
3517                 }
3518
3519                 return ret;
3520         }
3521
3522 dump:
3523         if (!(param->fp_verbose & VERBOSE_MDTINDEX))
3524                 llapi_lov_dump_user_lmm(param, path, d ? 1 : 0);
3525
3526 out:
3527         /* Do not get down anymore? */
3528         if (param->fp_depth == param->fp_max_depth)
3529                 return 1;
3530
3531         param->fp_depth++;
3532
3533         return 0;
3534 }
3535
3536 int llapi_getstripe(char *path, struct find_param *param)
3537 {
3538         return param_callback(path, (param->fp_verbose & VERBOSE_MDTINDEX) ?
3539                               cb_get_mdt_index : cb_getstripe,
3540                               cb_common_fini, param);
3541 }
3542
3543 int llapi_obd_statfs(char *path, __u32 type, __u32 index,
3544                      struct obd_statfs *stat_buf,
3545                      struct obd_uuid *uuid_buf)
3546 {
3547         int fd;
3548         char raw[OBD_MAX_IOCTL_BUFFER] = {'\0'};
3549         char *rawbuf = raw;
3550         struct obd_ioctl_data data = { 0 };
3551         int rc = 0;
3552
3553         data.ioc_inlbuf1 = (char *)&type;
3554         data.ioc_inllen1 = sizeof(__u32);
3555         data.ioc_inlbuf2 = (char *)&index;
3556         data.ioc_inllen2 = sizeof(__u32);
3557         data.ioc_pbuf1 = (char *)stat_buf;
3558         data.ioc_plen1 = sizeof(struct obd_statfs);
3559         data.ioc_pbuf2 = (char *)uuid_buf;
3560         data.ioc_plen2 = sizeof(struct obd_uuid);
3561
3562         rc = obd_ioctl_pack(&data, &rawbuf, sizeof(raw));
3563         if (rc != 0) {
3564                 llapi_error(LLAPI_MSG_ERROR, rc,
3565                             "llapi_obd_statfs: error packing ioctl data");
3566                 return rc;
3567         }
3568
3569         fd = open(path, O_RDONLY);
3570         if (errno == EISDIR)
3571                 fd = open(path, O_DIRECTORY | O_RDONLY);
3572
3573         if (fd < 0) {
3574                 rc = errno ? -errno : -EBADF;
3575                 llapi_error(LLAPI_MSG_ERROR, rc, "error: %s: opening '%s'",
3576                             __func__, path);
3577                 /* If we can't even open a file on the filesystem (e.g. with
3578                  * -ESHUTDOWN), force caller to exit or it will loop forever. */
3579                 return -ENODEV;
3580         }
3581         rc = ioctl(fd, IOC_OBD_STATFS, (void *)rawbuf);
3582         if (rc)
3583                 rc = errno ? -errno : -EINVAL;
3584
3585         close(fd);
3586         return rc;
3587 }
3588
3589 #define MAX_STRING_SIZE 128
3590
3591 int llapi_ping(char *obd_type, char *obd_name)
3592 {
3593         char path[MAX_STRING_SIZE];
3594         char buf[1];
3595         int rc, fd;
3596
3597         snprintf(path, MAX_STRING_SIZE, "/proc/fs/lustre/%s/%s/ping",
3598                  obd_type, obd_name);
3599
3600         fd = open(path, O_WRONLY);
3601         if (fd < 0) {
3602                 rc = -errno;
3603                 llapi_error(LLAPI_MSG_ERROR, rc, "error opening %s", path);
3604                 return rc;
3605         }
3606
3607         /* The purpose is to send a byte as a ping, whatever this byte is. */
3608         /* coverity[uninit_use_in_call] */
3609         rc = write(fd, buf, 1);
3610         if (rc < 0)
3611                 rc = -errno;
3612         close(fd);
3613
3614         if (rc == 1)
3615                 return 0;
3616         return rc;
3617 }
3618
3619 int llapi_target_iterate(int type_num, char **obd_type,
3620                          void *args, llapi_cb_t cb)
3621 {
3622         char buf[MAX_STRING_SIZE];
3623         FILE *fp = fopen(DEVICES_LIST, "r");
3624         int i, rc = 0;
3625
3626         if (fp == NULL) {
3627                 rc = -errno;
3628                 llapi_error(LLAPI_MSG_ERROR, rc, "error: opening "DEVICES_LIST);
3629                 return rc;
3630         }
3631
3632         while (fgets(buf, sizeof(buf), fp) != NULL) {
3633                 char *obd_type_name = NULL;
3634                 char *obd_name = NULL;
3635                 char *obd_uuid = NULL;
3636                 char *bufp = buf;
3637                 struct obd_statfs osfs_buffer;
3638
3639                 while(bufp[0] == ' ')
3640                         ++bufp;
3641
3642                 for(i = 0; i < 3; i++) {
3643                         obd_type_name = strsep(&bufp, " ");
3644                 }
3645                 obd_name = strsep(&bufp, " ");
3646                 obd_uuid = strsep(&bufp, " ");
3647
3648                 memset(&osfs_buffer, 0, sizeof (osfs_buffer));
3649
3650                 for (i = 0; i < type_num; i++) {
3651                         if (strcmp(obd_type_name, obd_type[i]) != 0)
3652                                 continue;
3653
3654                         cb(obd_type_name, obd_name, obd_uuid, args);
3655                 }
3656         }
3657         fclose(fp);
3658         return 0;
3659 }
3660
3661 static void do_target_check(char *obd_type_name, char *obd_name,
3662                             char *obd_uuid, void *args)
3663 {
3664         int rc;
3665
3666         rc = llapi_ping(obd_type_name, obd_name);
3667         if (rc == ENOTCONN) {
3668                 llapi_printf(LLAPI_MSG_NORMAL, "%s inactive.\n", obd_name);
3669         } else if (rc) {
3670                 llapi_error(LLAPI_MSG_ERROR, rc, "error: check '%s'", obd_name);
3671         } else {
3672                 llapi_printf(LLAPI_MSG_NORMAL, "%s active.\n", obd_name);
3673         }
3674 }
3675
3676 int llapi_target_check(int type_num, char **obd_type, char *dir)
3677 {
3678         return llapi_target_iterate(type_num, obd_type, NULL, do_target_check);
3679 }
3680
3681 #undef MAX_STRING_SIZE
3682
3683 /* Is this a lustre fs? */
3684 int llapi_is_lustre_mnttype(const char *type)
3685 {
3686         return (strcmp(type, "lustre") == 0 || strcmp(type,"lustre_lite") == 0);
3687 }
3688
3689 /* Is this a lustre client fs? */
3690 int llapi_is_lustre_mnt(struct mntent *mnt)
3691 {
3692         return (llapi_is_lustre_mnttype(mnt->mnt_type) &&
3693                 strstr(mnt->mnt_fsname, ":/") != NULL);
3694 }
3695
3696 int llapi_quotacheck(char *mnt, int check_type)
3697 {
3698         DIR *root;
3699         int rc;
3700
3701         root = opendir(mnt);
3702         if (!root) {
3703                 rc = -errno;
3704                 llapi_error(LLAPI_MSG_ERROR, rc, "open %s failed", mnt);
3705                 return rc;
3706         }
3707
3708         rc = ioctl(dirfd(root), OBD_IOC_QUOTACHECK, check_type);
3709         if (rc < 0)
3710                 rc = -errno;
3711
3712         closedir(root);
3713         return rc;
3714 }
3715
3716 int llapi_poll_quotacheck(char *mnt, struct if_quotacheck *qchk)
3717 {
3718         DIR *root;
3719         int poll_intvl = 2;
3720         int rc;
3721
3722         root = opendir(mnt);
3723         if (!root) {
3724                 rc = -errno;
3725                 llapi_error(LLAPI_MSG_ERROR, rc, "open %s failed", mnt);
3726                 return rc;
3727         }
3728
3729         while (1) {
3730                 rc = ioctl(dirfd(root), OBD_IOC_POLL_QUOTACHECK, qchk);
3731                 if (!rc)
3732                         break;
3733                 sleep(poll_intvl);
3734                 if (poll_intvl < 30)
3735                         poll_intvl *= 2;
3736         }
3737
3738         closedir(root);
3739         return 0;
3740 }
3741
3742 int llapi_quotactl(char *mnt, struct if_quotactl *qctl)
3743 {
3744         DIR *root;
3745         int rc;
3746
3747         root = opendir(mnt);
3748         if (!root) {
3749                 rc = -errno;
3750                 llapi_error(LLAPI_MSG_ERROR, rc, "open %s failed", mnt);
3751                 return rc;
3752         }
3753
3754         rc = ioctl(dirfd(root), OBD_IOC_QUOTACTL, qctl);
3755         if (rc < 0)
3756                 rc = -errno;
3757
3758         closedir(root);
3759         return rc;
3760 }
3761
3762 static int cb_quotachown(char *path, DIR *parent, DIR **dirp, void *data,
3763                          struct dirent64 *de)
3764 {
3765         struct find_param *param = (struct find_param *)data;
3766         DIR *d = dirp == NULL ? NULL : *dirp;
3767         lstat_t *st;
3768         int rc;
3769
3770         LASSERT(parent != NULL || d != NULL);
3771
3772         rc = get_lmd_info(path, parent, d, param->fp_lmd, param->fp_lum_size);
3773         if (rc) {
3774                 if (rc == -ENODATA) {
3775                         if (!param->fp_obd_uuid && !param->fp_quiet)
3776                                 llapi_error(LLAPI_MSG_ERROR, -ENODATA,
3777                                           "%s has no stripe info", path);
3778                         rc = 0;
3779                 } else if (rc == -ENOENT) {
3780                         rc = 0;
3781                 }
3782                 return rc;
3783         }
3784
3785         st = &param->fp_lmd->lmd_st;
3786
3787         /* libc chown() will do extra check, and if the real owner is
3788          * the same as the ones to set, it won't fall into kernel, so
3789          * invoke syscall directly. */
3790         rc = syscall(SYS_chown, path, -1, -1);
3791         if (rc)
3792                 llapi_error(LLAPI_MSG_ERROR, errno,
3793                             "error: chown %s", path);
3794
3795         rc = chmod(path, st->st_mode);
3796         if (rc) {
3797                 rc = -errno;
3798                 llapi_error(LLAPI_MSG_ERROR, rc, "error: chmod %s (%hu)",
3799                             path, st->st_mode);
3800         }
3801
3802         return rc;
3803 }
3804
3805 int llapi_quotachown(char *path, int flag)
3806 {
3807         struct find_param param;
3808
3809         memset(&param, 0, sizeof(param));
3810         param.fp_recursive = 1;
3811         param.fp_verbose = 0;
3812         param.fp_quiet = 1;
3813
3814         return param_callback(path, cb_quotachown, NULL, &param);
3815 }
3816
3817 #include <pwd.h>
3818 #include <grp.h>
3819 #include <mntent.h>
3820 #include <sys/wait.h>
3821 #include <errno.h>
3822 #include <ctype.h>
3823
3824 static int rmtacl_notify(int ops)
3825 {
3826         FILE *fp;
3827         struct mntent *mnt;
3828         int found = 0, fd = 0, rc = 0;
3829
3830         fp = setmntent(MOUNTED, "r");
3831         if (fp == NULL) {
3832                 rc = -errno;
3833                 llapi_error(LLAPI_MSG_ERROR, rc,
3834                             "error setmntent(%s)", MOUNTED);
3835                 return rc;
3836         }
3837
3838         while (1) {
3839                 mnt = getmntent(fp);
3840                 if (!mnt)
3841                         break;
3842
3843                 if (!llapi_is_lustre_mnt(mnt))
3844                         continue;
3845
3846                 fd = open(mnt->mnt_dir, O_RDONLY | O_DIRECTORY);
3847                 if (fd < 0) {
3848                         rc = -errno;
3849                         llapi_error(LLAPI_MSG_ERROR, rc,
3850                                     "Can't open '%s'", mnt->mnt_dir);
3851                         goto out;
3852                 }
3853
3854                 rc = ioctl(fd, LL_IOC_RMTACL, ops);
3855                 close(fd);
3856                 if (rc < 0) {
3857                         rc = -errno;
3858                         llapi_error(LLAPI_MSG_ERROR, rc,
3859                                     "ioctl RMTACL on '%s' err %d",
3860                                     mnt->mnt_dir, rc);
3861                         goto out;
3862                 }
3863
3864                 found++;
3865         }
3866
3867 out:
3868         endmntent(fp);
3869         return ((rc != 0) ? rc : found);
3870 }
3871
3872 static char *next_token(char *p, int div)
3873 {
3874         if (p == NULL)
3875                 return NULL;
3876
3877         if (div)
3878                 while (*p && *p != ':' && !isspace(*p))
3879                         p++;
3880         else
3881                 while (*p == ':' || isspace(*p))
3882                         p++;
3883
3884         return *p ? p : NULL;
3885 }
3886
3887 static int rmtacl_name2id(char *name, int is_user)
3888 {
3889         if (is_user) {
3890                 struct passwd *pw;
3891
3892                 pw = getpwnam(name);
3893                 if (pw == NULL)
3894                         return INVALID_ID;
3895                 else
3896                         return (int)(pw->pw_uid);
3897         } else {
3898                 struct group *gr;
3899
3900                 gr = getgrnam(name);
3901                 if (gr == NULL)
3902                         return INVALID_ID;
3903                 else
3904                         return (int)(gr->gr_gid);
3905         }
3906 }
3907
3908 static int isodigit(int c)
3909 {
3910         return (c >= '0' && c <= '7') ? 1 : 0;
3911 }
3912
3913 /*
3914  * Whether the name is just digits string (uid/gid) already or not.
3915  * Return value:
3916  * 1: str is id
3917  * 0: str is not id
3918  */
3919 static int str_is_id(char *str)
3920 {
3921         if (str == NULL)
3922                 return 0;
3923
3924         if (*str == '0') {
3925                 str++;
3926                 if (*str == 'x' || *str == 'X') { /* for Hex. */
3927                         if (!isxdigit(*(++str)))
3928                                 return 0;
3929
3930                         while (isxdigit(*(++str)));
3931                 } else if (isodigit(*str)) { /* for Oct. */
3932                         while (isodigit(*(++str)));
3933                 }
3934         } else if (isdigit(*str)) { /* for Dec. */
3935                 while (isdigit(*(++str)));
3936         }
3937
3938         return (*str == 0) ? 1 : 0;
3939 }
3940
3941 typedef struct {
3942         char *name;
3943         int   length;
3944         int   is_user;
3945         int   next_token;
3946 } rmtacl_name_t;
3947
3948 #define RMTACL_OPTNAME(name) name, sizeof(name) - 1
3949
3950 static rmtacl_name_t rmtacl_namelist[] = {
3951         { RMTACL_OPTNAME("user:"),            1,      0 },
3952         { RMTACL_OPTNAME("group:"),           0,      0 },
3953         { RMTACL_OPTNAME("default:user:"),    1,      0 },
3954         { RMTACL_OPTNAME("default:group:"),   0,      0 },
3955         /* for --tabular option */
3956         { RMTACL_OPTNAME("user"),             1,      1 },
3957         { RMTACL_OPTNAME("group"),            0,      1 },
3958         { 0 }
3959 };
3960
3961 static int rgetfacl_output(char *str)
3962 {
3963         char *start = NULL, *end = NULL;
3964         int is_user = 0, n, id;
3965         char c;
3966         rmtacl_name_t *rn;
3967
3968         if (str == NULL)
3969                 return -1;
3970
3971         for (rn = rmtacl_namelist; rn->name; rn++) {
3972                 if(strncmp(str, rn->name, rn->length) == 0) {
3973                         if (!rn->next_token)
3974                                 start = str + rn->length;
3975                         else
3976                                 start = next_token(str + rn->length, 0);
3977                         is_user = rn->is_user;
3978                         break;
3979                 }
3980         }
3981
3982         end = next_token(start, 1);
3983         if (end == NULL || start == end) {
3984                 n = printf("%s", str);
3985                 return n;
3986         }
3987
3988         c = *end;
3989         *end = 0;
3990         id = rmtacl_name2id(start, is_user);
3991         if (id == INVALID_ID) {
3992                 if (str_is_id(start)) {
3993                         *end = c;
3994                         n = printf("%s", str);
3995                 } else
3996                         return -1;
3997         } else if ((id == NOBODY_UID && is_user) ||
3998                    (id == NOBODY_GID && !is_user)) {
3999                 *end = c;
4000                 n = printf("%s", str);
4001         } else {
4002                 *end = c;
4003                 *start = 0;
4004                 n = printf("%s%d%s", str, id, end);
4005         }
4006         return n;
4007 }
4008
4009 static int child_status(int status)
4010 {
4011         return WIFEXITED(status) ? WEXITSTATUS(status) : -1;
4012 }
4013
4014 static int do_rmtacl(int argc, char *argv[], int ops, int (output_func)(char *))
4015 {
4016         pid_t pid = 0;
4017         int fd[2], status, rc;
4018         FILE *fp;
4019         char buf[PIPE_BUF];
4020
4021         if (output_func) {
4022                 if (pipe(fd) < 0) {
4023                         rc = -errno;
4024                         llapi_error(LLAPI_MSG_ERROR, rc, "Can't create pipe");
4025                         return rc;
4026                 }
4027
4028                 pid = fork();
4029                 if (pid < 0) {
4030                         rc = -errno;
4031                         llapi_error(LLAPI_MSG_ERROR, rc, "Can't fork");
4032                         close(fd[0]);
4033                         close(fd[1]);
4034                         return rc;
4035                 } else if (!pid) {
4036                         /* child process redirects its output. */
4037                         close(fd[0]);
4038                         close(1);
4039                         if (dup2(fd[1], 1) < 0) {
4040                                 rc = -errno;
4041                                 llapi_error(LLAPI_MSG_ERROR, rc,
4042                                             "Can't dup2 %d", fd[1]);
4043                                 close(fd[1]);
4044                                 return rc;
4045                         }
4046                 } else {
4047                         close(fd[1]);
4048                 }
4049         }
4050
4051         if (!pid) {
4052                 status = rmtacl_notify(ops);
4053                 if (status < 0)
4054                         return -errno;
4055
4056                 exit(execvp(argv[0], argv));
4057         }
4058
4059         /* the following is parent process */
4060         fp = fdopen(fd[0], "r");
4061         if (fp == NULL) {
4062                 rc = -errno;
4063                 llapi_error(LLAPI_MSG_ERROR, rc, "fdopen %d failed", fd[0]);
4064                 kill(pid, SIGKILL);
4065                 close(fd[0]);
4066                 return rc;
4067         }
4068
4069         while (fgets(buf, PIPE_BUF, fp) != NULL) {
4070                 if (output_func(buf) < 0)
4071                         fprintf(stderr, "WARNING: unexpected error!\n[%s]\n",
4072                                 buf);
4073         }
4074         fclose(fp);
4075         close(fd[0]);
4076
4077         if (waitpid(pid, &status, 0) < 0) {
4078                 rc = -errno;
4079                 llapi_error(LLAPI_MSG_ERROR, rc, "waitpid %d failed", pid);
4080                 return rc;
4081         }
4082
4083         return child_status(status);
4084 }
4085
4086 int llapi_lsetfacl(int argc, char *argv[])
4087 {
4088         return do_rmtacl(argc, argv, RMT_LSETFACL, NULL);
4089 }
4090
4091 int llapi_lgetfacl(int argc, char *argv[])
4092 {
4093         return do_rmtacl(argc, argv, RMT_LGETFACL, NULL);
4094 }
4095
4096 int llapi_rsetfacl(int argc, char *argv[])
4097 {
4098         return do_rmtacl(argc, argv, RMT_RSETFACL, NULL);
4099 }
4100
4101 int llapi_rgetfacl(int argc, char *argv[])
4102 {
4103         return do_rmtacl(argc, argv, RMT_RGETFACL, rgetfacl_output);
4104 }
4105
4106 int llapi_cp(int argc, char *argv[])
4107 {
4108         int rc;
4109
4110         rc = rmtacl_notify(RMT_RSETFACL);
4111         if (rc < 0)
4112                 return rc;
4113
4114         exit(execvp(argv[0], argv));
4115 }
4116
4117 int llapi_ls(int argc, char *argv[])
4118 {
4119         int rc;
4120
4121         rc = rmtacl_notify(RMT_LGETFACL);
4122         if (rc < 0)
4123                 return rc;
4124
4125         exit(execvp(argv[0], argv));
4126 }
4127
4128 /* Print mdtname 'name' into 'buf' using 'format'.  Add -MDT0000 if needed.
4129  * format must have %s%s, buf must be > 16
4130  * Eg: if name = "lustre-MDT0000", "lustre", or "lustre-MDT0000_UUID"
4131  *     then buf = "lustre-MDT0000"
4132  */
4133 static int get_mdtname(char *name, char *format, char *buf)
4134 {
4135         char suffix[]="-MDT0000";
4136         int len = strlen(name);
4137
4138         if ((len > 5) && (strncmp(name + len - 5, "_UUID", 5) == 0)) {
4139                 name[len - 5] = '\0';
4140                 len -= 5;
4141         }
4142
4143         if (len > 8) {
4144                 if ((len <= 16) && strncmp(name + len - 8, "-MDT", 4) == 0) {
4145                         suffix[0] = '\0';
4146                 } else {
4147                         /* Not enough room to add suffix */
4148                         llapi_err_noerrno(LLAPI_MSG_ERROR,
4149                                           "MDT name too long |%s|", name);
4150                         return -EINVAL;
4151                 }
4152         }
4153
4154         return sprintf(buf, format, name, suffix);
4155 }
4156
4157 /** ioctl on filsystem root, with mdtindex sent as data
4158  * \param mdtname path, fsname, or mdtname (lutre-MDT0004)
4159  * \param mdtidxp pointer to integer within data to be filled in with the
4160  *    mdt index (0 if no mdt is specified).  NULL won't be filled.
4161  */
4162 int root_ioctl(const char *mdtname, int opc, void *data, int *mdtidxp,
4163                int want_error)
4164 {
4165         char fsname[20];
4166         char *ptr;
4167         int fd, rc;
4168         long index;
4169
4170         /* Take path, fsname, or MDTname.  Assume MDT0000 in the former cases.
4171          Open root and parse mdt index. */
4172         if (mdtname[0] == '/') {
4173                 index = 0;
4174                 rc = get_root_path(WANT_FD | want_error, NULL, &fd,
4175                                    (char *)mdtname, -1);
4176         } else {
4177                 if (get_mdtname((char *)mdtname, "%s%s", fsname) < 0)
4178                         return -EINVAL;
4179                 ptr = fsname + strlen(fsname) - 8;
4180                 *ptr = '\0';
4181                 index = strtol(ptr + 4, NULL, 10);
4182                 rc = get_root_path(WANT_FD | want_error, fsname, &fd, NULL, -1);
4183         }
4184         if (rc < 0) {
4185                 if (want_error)
4186                         llapi_err_noerrno(LLAPI_MSG_ERROR,
4187                                           "Can't open %s: %d\n", mdtname, rc);
4188                 return rc;
4189         }
4190
4191         if (mdtidxp)
4192                 *mdtidxp = index;
4193
4194         rc = ioctl(fd, opc, data);
4195         if (rc == -1)
4196                 rc = -errno;
4197         else
4198                 rc = 0;
4199         if (rc && want_error)
4200                 llapi_error(LLAPI_MSG_ERROR, rc, "ioctl %d err %d", opc, rc);
4201
4202         close(fd);
4203         return rc;
4204 }
4205
4206 /****** Changelog API ********/
4207
4208 static int changelog_ioctl(const char *mdtname, int opc, int id,
4209                            long long recno, int flags)
4210 {
4211         struct ioc_changelog data;
4212         int *idx;
4213
4214         data.icc_id = id;
4215         data.icc_recno = recno;
4216         data.icc_flags = flags;
4217         idx = (int *)(&data.icc_mdtindex);
4218
4219         return root_ioctl(mdtname, opc, &data, idx, WANT_ERROR);
4220 }
4221
4222 #define CHANGELOG_PRIV_MAGIC 0xCA8E1080
4223 struct changelog_private {
4224         int                             magic;
4225         enum changelog_send_flag        flags;
4226         lustre_kernelcomm               kuc;
4227 };
4228
4229 /** Start reading from a changelog
4230  * @param priv Opaque private control structure
4231  * @param flags Start flags (e.g. CHANGELOG_FLAG_BLOCK)
4232  * @param device Report changes recorded on this MDT
4233  * @param startrec Report changes beginning with this record number
4234  * (just call llapi_changelog_fini when done; don't need an endrec)
4235  */
4236 int llapi_changelog_start(void **priv, enum changelog_send_flag flags,
4237                           const char *device, long long startrec)
4238 {
4239         struct changelog_private        *cp;
4240         static bool                      warned;
4241         int                              rc;
4242
4243         /* Set up the receiver control struct */
4244         cp = calloc(1, sizeof(*cp));
4245         if (cp == NULL)
4246                 return -ENOMEM;
4247
4248         cp->magic = CHANGELOG_PRIV_MAGIC;
4249         cp->flags = flags;
4250
4251         /* Set up the receiver */
4252         rc = libcfs_ukuc_start(&cp->kuc, 0 /* no group registration */, 0);
4253         if (rc < 0)
4254                 goto out_free;
4255
4256         *priv = cp;
4257
4258         /* CHANGELOG_FLAG_JOBID will eventually become mandatory. Display a
4259          * warning if it's missing. */
4260         if (!(flags & CHANGELOG_FLAG_JOBID) && !warned) {
4261                 llapi_err_noerrno(LLAPI_MSG_WARN, "warning: %s() called "
4262                                   "w/o CHANGELOG_FLAG_JOBID", __func__);
4263                 warned = true;
4264         }
4265
4266         /* Tell the kernel to start sending */
4267         rc = changelog_ioctl(device, OBD_IOC_CHANGELOG_SEND, cp->kuc.lk_wfd,
4268                              startrec, flags);
4269         /* Only the kernel reference keeps the write side open */
4270         close(cp->kuc.lk_wfd);
4271         cp->kuc.lk_wfd = LK_NOFD;
4272         if (rc < 0) {
4273                 /* frees and clears priv */
4274                 llapi_changelog_fini(priv);
4275                 return rc;
4276         }
4277
4278         return 0;
4279
4280 out_free:
4281         free(cp);
4282         return rc;
4283 }
4284
4285 /** Finish reading from a changelog */
4286 int llapi_changelog_fini(void **priv)
4287 {
4288         struct changelog_private *cp = (struct changelog_private *)*priv;
4289
4290         if (!cp || (cp->magic != CHANGELOG_PRIV_MAGIC))
4291                 return -EINVAL;
4292
4293         libcfs_ukuc_stop(&cp->kuc);
4294         free(cp);
4295         *priv = NULL;
4296         return 0;
4297 }
4298
4299 /**
4300  * Convert all records to a same format according to the caller's wishes.
4301  * Default is CLF_VERSION | CLF_RENAME.
4302  * Add CLF_JOBID if explicitely requested.
4303  *
4304  * \param rec  The record to remap. It is expected to be big enough to
4305  *             properly handle the final format.
4306  * \return 1 if anything changed. 0 otherwise.
4307  */
4308 /** Read the next changelog entry
4309  * @param priv Opaque private control structure
4310  * @param rech Changelog record handle; record will be allocated here
4311  * @return 0 valid message received; rec is set
4312  *         <0 error code
4313  *         1 EOF
4314  */
4315 #define DEFAULT_RECORD_FMT      (CLF_VERSION | CLF_RENAME)
4316 int llapi_changelog_recv(void *priv, struct changelog_rec **rech)
4317 {
4318         struct changelog_private        *cp = (struct changelog_private *)priv;
4319         struct kuc_hdr                  *kuch;
4320         enum changelog_rec_flags         rec_fmt = DEFAULT_RECORD_FMT;
4321         int                              rc = 0;
4322
4323         if (!cp || (cp->magic != CHANGELOG_PRIV_MAGIC))
4324                 return -EINVAL;
4325         if (rech == NULL)
4326                 return -EINVAL;
4327         kuch = malloc(KUC_CHANGELOG_MSG_MAXSIZE);
4328         if (kuch == NULL)
4329                 return -ENOMEM;
4330
4331         if (cp->flags & CHANGELOG_FLAG_JOBID)
4332                 rec_fmt |= CLF_JOBID;
4333
4334 repeat:
4335         rc = libcfs_ukuc_msg_get(&cp->kuc, (char *)kuch,
4336                                  KUC_CHANGELOG_MSG_MAXSIZE,
4337                                  KUC_TRANSPORT_CHANGELOG);
4338         if (rc < 0)
4339                 goto out_free;
4340
4341         if ((kuch->kuc_transport != KUC_TRANSPORT_CHANGELOG) ||
4342             ((kuch->kuc_msgtype != CL_RECORD) &&
4343              (kuch->kuc_msgtype != CL_EOF))) {
4344                 llapi_err_noerrno(LLAPI_MSG_ERROR,
4345                                   "Unknown changelog message type %d:%d\n",
4346                                   kuch->kuc_transport, kuch->kuc_msgtype);
4347                 rc = -EPROTO;
4348                 goto out_free;
4349         }
4350
4351         if (kuch->kuc_msgtype == CL_EOF) {
4352                 if (cp->flags & CHANGELOG_FLAG_FOLLOW) {
4353                         /* Ignore EOFs */
4354                         goto repeat;
4355                 } else {
4356                         rc = 1;
4357                         goto out_free;
4358                 }
4359         }
4360
4361         /* Our message is a changelog_rec.  Use pointer math to skip
4362          * kuch_hdr and point directly to the message payload. */
4363         *rech = (struct changelog_rec *)(kuch + 1);
4364         changelog_remap_rec(*rech, rec_fmt);
4365
4366         return 0;
4367
4368 out_free:
4369         *rech = NULL;
4370         free(kuch);
4371         return rc;
4372 }
4373
4374 /** Release the changelog record when done with it. */
4375 int llapi_changelog_free(struct changelog_rec **rech)
4376 {
4377         if (*rech) {
4378                 /* We allocated memory starting at the kuc_hdr, but passed
4379                  * the consumer a pointer to the payload.
4380                  * Use pointer math to get back to the header.
4381                  */
4382                 struct kuc_hdr *kuch = (struct kuc_hdr *)*rech - 1;
4383                 free(kuch);
4384         }
4385         *rech = NULL;
4386         return 0;
4387 }
4388
4389 int llapi_changelog_clear(const char *mdtname, const char *idstr,
4390                           long long endrec)
4391 {
4392         long id;
4393
4394         if (endrec < 0) {
4395                 llapi_err_noerrno(LLAPI_MSG_ERROR,
4396                                   "can't purge negative records\n");
4397                 return -EINVAL;
4398         }
4399
4400         id = strtol(idstr + strlen(CHANGELOG_USER_PREFIX), NULL, 10);
4401         if ((id == 0) || (strncmp(idstr, CHANGELOG_USER_PREFIX,
4402                                   strlen(CHANGELOG_USER_PREFIX)) != 0)) {
4403                 llapi_err_noerrno(LLAPI_MSG_ERROR,
4404                                   "expecting id of the form '"
4405                                   CHANGELOG_USER_PREFIX
4406                                   "<num>'; got '%s'\n", idstr);
4407                 return -EINVAL;
4408         }
4409
4410         return changelog_ioctl(mdtname, OBD_IOC_CHANGELOG_CLEAR, id, endrec, 0);
4411 }
4412
4413 int llapi_fid2path(const char *device, const char *fidstr, char *buf,
4414                    int buflen, long long *recno, int *linkno)
4415 {
4416         struct lu_fid fid;
4417         struct getinfo_fid2path *gf;
4418         int rc;
4419
4420         while (*fidstr == '[')
4421                 fidstr++;
4422
4423         sscanf(fidstr, SFID, RFID(&fid));
4424         if (!fid_is_sane(&fid)) {
4425                 llapi_err_noerrno(LLAPI_MSG_ERROR,
4426                                   "bad FID format [%s], should be "DFID"\n",
4427                                   fidstr, (__u64)1, 2, 0);
4428                 return -EINVAL;
4429         }
4430
4431         gf = malloc(sizeof(*gf) + buflen);
4432         if (gf == NULL)
4433                 return -ENOMEM;
4434         gf->gf_fid = fid;
4435         gf->gf_recno = *recno;
4436         gf->gf_linkno = *linkno;
4437         gf->gf_pathlen = buflen;
4438
4439         /* Take path or fsname */
4440         rc = root_ioctl(device, OBD_IOC_FID2PATH, gf, NULL, 0);
4441         if (rc) {
4442                 if (rc != -ENOENT)
4443                         llapi_error(LLAPI_MSG_ERROR, rc, "ioctl err %d", rc);
4444         } else {
4445                 memcpy(buf, gf->gf_path, gf->gf_pathlen);
4446                 if (buf[0] == '\0') { /* ROOT path */
4447                         buf[0] = '/';
4448                         buf[1] = '\0';
4449                 }
4450                 *recno = gf->gf_recno;
4451                 *linkno = gf->gf_linkno;
4452         }
4453
4454         free(gf);
4455         return rc;
4456 }
4457
4458 static int fid_from_lma(const char *path, const int fd, lustre_fid *fid)
4459 {
4460         char                     buf[512];
4461         struct lustre_mdt_attrs *lma;
4462         int                      rc;
4463
4464         if (path == NULL)
4465                 rc = fgetxattr(fd, XATTR_NAME_LMA, buf, sizeof(buf));
4466         else
4467                 rc = lgetxattr(path, XATTR_NAME_LMA, buf, sizeof(buf));
4468         if (rc < 0)
4469                 return -errno;
4470         lma = (struct lustre_mdt_attrs *)buf;
4471         fid_le_to_cpu(fid, &lma->lma_self_fid);
4472         return 0;
4473 }
4474
4475 int llapi_get_mdt_index_by_fid(int fd, const lustre_fid *fid,
4476                                int *mdt_index)
4477 {
4478         int     rc;
4479
4480         rc = ioctl(fd, LL_IOC_FID2MDTIDX, fid);
4481         if (rc < 0)
4482                 return -errno;
4483
4484         *mdt_index = rc;
4485
4486         return rc;
4487 }
4488
4489 int llapi_fd2fid(const int fd, lustre_fid *fid)
4490 {
4491         int rc;
4492
4493         memset(fid, 0, sizeof(*fid));
4494
4495         rc = ioctl(fd, LL_IOC_PATH2FID, fid) < 0 ? -errno : 0;
4496         if (rc == -EINVAL || rc == -ENOTTY)
4497                 rc = fid_from_lma(NULL, fd, fid);
4498
4499         return rc;
4500 }
4501
4502 int llapi_path2fid(const char *path, lustre_fid *fid)
4503 {
4504         int fd, rc;
4505
4506         memset(fid, 0, sizeof(*fid));
4507         fd = open(path, O_RDONLY | O_NONBLOCK | O_NOFOLLOW);
4508         if (fd < 0) {
4509                 if (errno == ELOOP || errno == ENXIO)
4510                         return fid_from_lma(path, -1, fid);
4511                 return -errno;
4512         }
4513
4514         rc = llapi_fd2fid(fd, fid);
4515         if (rc == -EINVAL || rc == -ENOTTY)
4516                 rc = fid_from_lma(path, -1, fid);
4517
4518         close(fd);
4519         return rc;
4520 }
4521
4522 int llapi_fd2parent(int fd, unsigned int linkno, lustre_fid *parent_fid,
4523                     char *name, size_t name_size)
4524 {
4525         struct getparent        *gp;
4526         int                      rc;
4527
4528         gp = malloc(sizeof(*gp) + name_size);
4529         if (gp == NULL)
4530                 return -ENOMEM;
4531
4532         gp->gp_linkno = linkno;
4533         gp->gp_name_size = name_size;
4534
4535         rc = ioctl(fd, LL_IOC_GETPARENT, gp);
4536         if (rc < 0) {
4537                 rc = -errno;
4538                 goto err_free;
4539         }
4540
4541         *parent_fid = gp->gp_fid;
4542
4543         strncpy(name, gp->gp_name, name_size);
4544         name[name_size - 1] = '\0';
4545
4546 err_free:
4547         free(gp);
4548         return rc;
4549 }
4550
4551 int llapi_path2parent(const char *path, unsigned int linkno,
4552                       lustre_fid *parent_fid, char *name, size_t name_size)
4553 {
4554         int     fd;
4555         int     rc;
4556
4557         fd = open(path, O_RDONLY | O_NONBLOCK | O_NOFOLLOW);
4558         if (fd < 0)
4559                 return -errno;
4560
4561         rc = llapi_fd2parent(fd, linkno, parent_fid, name, name_size);
4562         close(fd);
4563         return rc;
4564 }
4565
4566 int llapi_get_connect_flags(const char *mnt, __u64 *flags)
4567 {
4568         DIR *root;
4569         int rc;
4570
4571         root = opendir(mnt);
4572         if (!root) {
4573                 rc = -errno;
4574                 llapi_error(LLAPI_MSG_ERROR, rc, "open %s failed", mnt);
4575                 return rc;
4576         }
4577
4578         rc = ioctl(dirfd(root), LL_IOC_GET_CONNECT_FLAGS, flags);
4579         if (rc < 0) {
4580                 rc = -errno;
4581                 llapi_error(LLAPI_MSG_ERROR, rc,
4582                             "ioctl on %s for getting connect flags failed", mnt);
4583         }
4584         closedir(root);
4585         return rc;
4586 }
4587
4588 int llapi_get_version(char *buffer, int buffer_size,
4589                       char **version)
4590 {
4591         int rc;
4592         int fd;
4593         struct obd_ioctl_data *data = (struct obd_ioctl_data *)buffer;
4594
4595         fd = open(OBD_DEV_PATH, O_RDONLY);
4596         if (fd == -1)
4597                 return -errno;
4598
4599         memset(buffer, 0, buffer_size);
4600         data->ioc_version = OBD_IOCTL_VERSION;
4601         data->ioc_inllen1 = buffer_size - cfs_size_round(sizeof(*data));
4602         data->ioc_inlbuf1 = buffer + cfs_size_round(sizeof(*data));
4603         data->ioc_len = obd_ioctl_packlen(data);
4604
4605         rc = ioctl(fd, OBD_GET_VERSION, buffer);
4606         if (rc == -1) {
4607                 rc = -errno;
4608                 close(fd);
4609                 return rc;
4610         }
4611         close(fd);
4612         *version = data->ioc_bulk;
4613         return 0;
4614 }
4615
4616 /**
4617  * Get a 64-bit value representing the version of file data pointed by fd.
4618  *
4619  * Each write or truncate, flushed on OST, will change this value. You can use
4620  * this value to verify if file data was modified. This only checks the file
4621  * data, not metadata.
4622  *
4623  * \param  flags  0: no flush pages, usually used it the process has already
4624  *                  taken locks;
4625  *                LL_DV_RD_FLUSH: OSTs will take LCK_PR to flush dirty pages
4626  *                  from clients;
4627  *                LL_DV_WR_FLUSH: OSTs will take LCK_PW to flush all caching
4628  *                  pages from clients.
4629  *
4630  * \retval 0 on success.
4631  * \retval -errno on error.
4632  */
4633 int llapi_get_data_version(int fd, __u64 *data_version, __u64 flags)
4634 {
4635         int rc;
4636         struct ioc_data_version idv;
4637
4638         idv.idv_flags = flags;
4639
4640         rc = ioctl(fd, LL_IOC_DATA_VERSION, &idv);
4641         if (rc)
4642                 rc = -errno;
4643         else
4644                 *data_version = idv.idv_version;
4645
4646         return rc;
4647 }
4648
4649 /*
4650  * Create a file without any name open it for read/write
4651  *
4652  * - file is created as if it were a standard file in the given \a directory
4653  * - file does not appear in \a directory and mtime does not change because
4654  *   the filename is handled specially by the Lustre MDS.
4655  * - file is removed at final close
4656  * - file modes are rw------- since it doesn't make sense to have a read-only
4657  *   or write-only file that cannot be opened again.
4658  * - if user wants another mode it must use fchmod() on the open file, no
4659  *   security problems arise because it cannot be opened by another process.
4660  *
4661  * \param[in]   directory       directory from which to inherit layout/MDT idx
4662  * \param[in]   idx             MDT index on which the file is created,
4663  *                              \a idx == -1 means no specific MDT is requested
4664  * \param[in]   open_flags      standard open(2) flags
4665  *
4666  * \retval      0 on success.
4667  * \retval      -errno on error.
4668  */
4669 int llapi_create_volatile_idx(char *directory, int idx, int open_flags)
4670 {
4671         char    file_path[PATH_MAX];
4672         char    filename[PATH_MAX];
4673         int     fd;
4674         int     rnumber;
4675         int     rc;
4676
4677         rnumber = random();
4678         if (idx == -1)
4679                 snprintf(filename, sizeof(filename),
4680                          LUSTRE_VOLATILE_HDR"::%.4X", rnumber);
4681         else
4682                 snprintf(filename, sizeof(filename),
4683                          LUSTRE_VOLATILE_HDR":%.4X:%.4X", idx, rnumber);
4684
4685         rc = snprintf(file_path, sizeof(file_path),
4686                       "%s/%s", directory, filename);
4687         if (rc >= sizeof(file_path))
4688                 return -E2BIG;
4689
4690         fd = open(file_path, O_RDWR | O_CREAT | open_flags, S_IRUSR | S_IWUSR);
4691         if (fd < 0) {
4692                 llapi_error(LLAPI_MSG_ERROR, errno,
4693                             "Cannot create volatile file '%s' in '%s'",
4694                             filename + LUSTRE_VOLATILE_HDR_LEN,
4695                             directory);
4696                 return -errno;
4697         }
4698         /* unlink file in case this wasn't a Lustre filesystem, and the
4699          * magic volatile filename wasn't handled as intended.  The effect
4700          * is the same. */
4701         unlink(file_path);
4702
4703         return fd;
4704 }
4705
4706 /**
4707  * Swap the layouts between 2 file descriptors
4708  * the 2 files must be open in write
4709  * first fd received the ioctl, second fd is passed as arg
4710  * this is assymetric but avoid use of root path for ioctl
4711  */
4712 int llapi_fswap_layouts(int fd1, int fd2, __u64 dv1, __u64 dv2, __u64 flags)
4713 {
4714         struct lustre_swap_layouts      lsl;
4715         int                             rc;
4716
4717         lsl.sl_fd = fd2;
4718         lsl.sl_flags = flags;
4719
4720         do
4721                 lsl.sl_gid = random();
4722         while (lsl.sl_gid == 0);
4723
4724         lsl.sl_dv1 = dv1;
4725         lsl.sl_dv2 = dv2;
4726         rc = ioctl(fd1, LL_IOC_LOV_SWAP_LAYOUTS, &lsl);
4727         if (rc)
4728                 rc = -errno;
4729         return rc;
4730 }
4731
4732 /**
4733  * Swap the layouts between 2 files
4734  * the 2 files are open in write
4735  */
4736 int llapi_swap_layouts(const char *path1, const char *path2,
4737                        __u64 dv1, __u64 dv2, __u64 flags)
4738 {
4739         int     fd1, fd2, rc;
4740
4741         fd1 = open(path1, O_WRONLY | O_LOV_DELAY_CREATE);
4742         if (fd1 < 0) {
4743                 rc = -errno;
4744                 llapi_error(LLAPI_MSG_ERROR, rc,
4745                             "error: cannot open '%s' for write", path1);
4746                 goto out;
4747         }
4748
4749         fd2 = open(path2, O_WRONLY | O_LOV_DELAY_CREATE);
4750         if (fd2 < 0) {
4751                 rc = -errno;
4752                 llapi_error(LLAPI_MSG_ERROR, rc,
4753                             "error: cannot open '%s' for write", path2);
4754                 goto out_close;
4755         }
4756
4757         rc = llapi_fswap_layouts(fd1, fd2, dv1, dv2, flags);
4758         if (rc < 0)
4759                 llapi_error(LLAPI_MSG_ERROR, rc,
4760                             "error: cannot swap layout between '%s' and '%s'",
4761                             path1, path2);
4762
4763         close(fd2);
4764 out_close:
4765         close(fd1);
4766 out:
4767         return rc;
4768 }
4769
4770 /**
4771  * Attempt to open a file with Lustre file identifier \a fid
4772  * and return an open file descriptor.
4773  *
4774  * \param[in] lustre_dir        path within Lustre filesystem containing \a fid
4775  * \param[in] fid               Lustre file identifier of file to open
4776  * \param[in] flags             open() flags
4777  *
4778  * \retval                      non-negative file descriptor on successful open
4779  * \retval                      -1 if an error occurred
4780  */
4781 int llapi_open_by_fid(const char *lustre_dir, const lustre_fid *fid, int flags)
4782 {
4783         char mntdir[PATH_MAX];
4784         char path[PATH_MAX];
4785         int rc;
4786
4787         rc = llapi_search_mounts(lustre_dir, 0, mntdir, NULL);
4788         if (rc != 0)
4789                 return -1;
4790
4791         snprintf(path, sizeof(path), "%s/.lustre/fid/"DFID, mntdir, PFID(fid));
4792         return open(path, flags);
4793 }
4794
4795 /**
4796  * Take group lock.
4797  *
4798  * \param fd   File to lock.
4799  * \param gid  Group Identifier.
4800  *
4801  * \retval 0 on success.
4802  * \retval -errno on failure.
4803  */
4804 int llapi_group_lock(int fd, int gid)
4805 {
4806         int rc;
4807
4808         rc = ioctl(fd, LL_IOC_GROUP_LOCK, gid);
4809         if (rc < 0) {
4810                 rc = -errno;
4811                 llapi_error(LLAPI_MSG_ERROR, rc, "cannot get group lock");
4812         }
4813         return rc;
4814 }
4815
4816 /**
4817  * Put group lock.
4818  *
4819  * \param fd   File to unlock.
4820  * \param gid  Group Identifier.
4821  *
4822  * \retval 0 on success.
4823  * \retval -errno on failure.
4824  */
4825 int llapi_group_unlock(int fd, int gid)
4826 {
4827         int rc;
4828
4829         rc = ioctl(fd, LL_IOC_GROUP_UNLOCK, gid);
4830         if (rc < 0) {
4831                 rc = -errno;
4832                 llapi_error(LLAPI_MSG_ERROR, rc, "cannot put group lock");
4833         }
4834         return rc;
4835 }