lustre/utils/liblustreapi.c

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