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