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