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