lustre/utils/liblustreapi.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  * GPL HEADER START
   5  *
   6  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License version 2 only,
  10  * as published by the Free Software Foundation.
  11  *
  12  * This program is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * General Public License version 2 for more details (a copy is included
  16  * in the LICENSE file that accompanied this code).
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * version 2 along with this program; If not, see
  20  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  21  *
  22  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  23  * CA 95054 USA or visit www.sun.com if you need additional information or
  24  * have any questions.
  25  *
  26  * GPL HEADER END
  27  */
  28 /*
  29  * Copyright (c) 2003, 2010, Oracle and/or its affiliates. All rights reserved.
  30  * Use is subject to license terms.
  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 #ifdef HAVE_LINUX_UNISTD_H
  65 #include <linux/unistd.h>
  66 #else
  67 #include <unistd.h>
  68 #endif
  69 #include <poll.h>
  70
  71 #include <liblustre.h>
  72 #include <lnet/lnetctl.h>
  73 #include <obd.h>
  74 #include <lustre_lib.h>
  75 #include <obd_lov.h>
  76 #include <lustre/liblustreapi.h>
  77
  78 static unsigned llapi_dir_filetype_table[] = {
  79         [DT_UNKNOWN]= 0,
  80         [DT_FIFO]= S_IFIFO,
  81         [DT_CHR] = S_IFCHR,
  82         [DT_DIR] = S_IFDIR,
  83         [DT_BLK] = S_IFBLK,
  84         [DT_REG] = S_IFREG,
  85         [DT_LNK] = S_IFLNK,
  86         [DT_SOCK]= S_IFSOCK,
  87 #if defined(DT_DOOR) && defined(S_IFDOOR)
  88         [DT_DOOR]= S_IFDOOR,
  89 #endif
  90 };
  91
  92 #if defined(DT_DOOR) && defined(S_IFDOOR)
  93 static const int DT_MAX = DT_DOOR;
  94 #else
  95 static const int DT_MAX = DT_SOCK;
  96 #endif
  97
  98 static unsigned llapi_filetype_dir_table[] = {
  99         [0]= DT_UNKNOWN,
 100         [S_IFIFO]= DT_FIFO,
 101         [S_IFCHR] = DT_CHR,
 102         [S_IFDIR] = DT_DIR,
 103         [S_IFBLK] = DT_BLK,
 104         [S_IFREG] = DT_REG,
 105         [S_IFLNK] = DT_LNK,
 106         [S_IFSOCK]= DT_SOCK,
 107 #if defined(DT_DOOR) && defined(S_IFDOOR)
 108         [S_IFDOOR]= DT_DOOR,
 109 #endif
 110 };
 111
 112 #if defined(DT_DOOR) && defined(S_IFDOOR)
 113 static const int S_IFMAX = DT_DOOR;
 114 #else
 115 static const int S_IFMAX = DT_SOCK;
 116 #endif
 117
 118 /* liblustreapi message level */
 119 static int llapi_msg_level = LLAPI_MSG_MAX;
 120
 121 void llapi_msg_set_level(int level)
 122 {
 123         /* ensure level is in the good range */
 124         if (level < LLAPI_MSG_OFF)
 125                 llapi_msg_level = LLAPI_MSG_OFF;
 126         else if (level > LLAPI_MSG_MAX)
 127                 llapi_msg_level = LLAPI_MSG_MAX;
 128         else
 129                 llapi_msg_level = level;
 130 }
 131
 132 void llapi_err(int level, char *fmt, ...)
 133 {
 134         va_list args;
 135         int tmp_errno = abs(errno);
 136
 137         if ((level & LLAPI_MSG_MASK) > llapi_msg_level)
 138                 return;
 139
 140         va_start(args, fmt);
 141         vfprintf(stderr, fmt, args);
 142         va_end(args);
 143
 144         if (level & LLAPI_MSG_NO_ERRNO)
 145                 fprintf(stderr, "\n");
 146         else
 147                 fprintf(stderr, ": %s (%d)\n", strerror(tmp_errno), tmp_errno);
 148 }
 149
 150 #define llapi_err_noerrno(level, fmt, a...)                             \
 151         llapi_err((level) | LLAPI_MSG_NO_ERRNO, fmt, ## a)
 152
 153 void llapi_printf(int level, char *fmt, ...)
 154 {
 155         va_list args;
 156
 157         if ((level & LLAPI_MSG_MASK) > llapi_msg_level)
 158                 return;
 159
 160         va_start(args, fmt);
 161         vfprintf(stdout, fmt, args);
 162         va_end(args);
 163 }
 164
 165 /**
 166  * size_units is unchanged if no specifier used
 167  */
 168 int parse_size(char *optarg, unsigned long long *size,
 169                unsigned long long *size_units, int bytes_spec)
 170 {
 171         char *end;
 172
 173         *size = strtoull(optarg, &end, 0);
 174
 175         if (*end != '\0') {
 176                 if ((*end == 'b') && *(end+1) == '\0' &&
 177                     (*size & (~0ULL << (64 - 9))) == 0 &&
 178                     !bytes_spec) {
 179                         *size <<= 9;
 180                         *size_units = 1 << 9;
 181                 } else if ((*end == 'b') && *(end+1) == '\0' &&
 182                            bytes_spec) {
 183                         *size_units = 1;
 184                 } else if ((*end == 'k' || *end == 'K') &&
 185                            *(end+1) == '\0' && (*size &
 186                            (~0ULL << (64 - 10))) == 0) {
 187                         *size <<= 10;
 188                         *size_units = 1 << 10;
 189                 } else if ((*end == 'm' || *end == 'M') &&
 190                            *(end+1) == '\0' && (*size &
 191                            (~0ULL << (64 - 20))) == 0) {
 192                         *size <<= 20;
 193                         *size_units = 1 << 20;
 194                 } else if ((*end == 'g' || *end == 'G') &&
 195                            *(end+1) == '\0' && (*size &
 196                            (~0ULL << (64 - 30))) == 0) {
 197                         *size <<= 30;
 198                         *size_units = 1 << 30;
 199                 } else if ((*end == 't' || *end == 'T') &&
 200                            *(end+1) == '\0' && (*size &
 201                            (~0ULL << (64 - 40))) == 0) {
 202                         *size <<= 40;
 203                         *size_units = 1ULL << 40;
 204                 } else if ((*end == 'p' || *end == 'P') &&
 205                            *(end+1) == '\0' && (*size &
 206                            (~0ULL << (64 - 50))) == 0) {
 207                         *size <<= 50;
 208                         *size_units = 1ULL << 50;
 209                 } else if ((*end == 'e' || *end == 'E') &&
 210                            *(end+1) == '\0' && (*size &
 211                            (~0ULL << (64 - 60))) == 0) {
 212                         *size <<= 60;
 213                         *size_units = 1ULL << 60;
 214                 } else {
 215                         return -1;
 216                 }
 217         }
 218
 219         return 0;
 220 }
 221
 222 int llapi_stripe_limit_check(unsigned long long stripe_size, int stripe_offset,
 223                              int stripe_count, int stripe_pattern)
 224 {
 225         int page_size;
 226
 227         /* 64 KB is the largest common page size I'm aware of (on ia64), but
 228          * check the local page size just in case. */
 229         page_size = LOV_MIN_STRIPE_SIZE;
 230         if (getpagesize() > page_size) {
 231                 page_size = getpagesize();
 232                 llapi_err_noerrno(LLAPI_MSG_WARN,
 233                                   "warning: your page size (%u) is "
 234                                   "larger than expected (%u)", page_size,
 235                                   LOV_MIN_STRIPE_SIZE);
 236         }
 237         if (stripe_size < 0 || (stripe_size & (LOV_MIN_STRIPE_SIZE - 1))) {
 238                 llapi_err(LLAPI_MSG_ERROR, "error: bad stripe_size %lu, "
 239                           "must be an even multiple of %d bytes",
 240                           stripe_size, page_size);
 241                 return -EINVAL;
 242         }
 243         if (stripe_offset < -1 || stripe_offset > MAX_OBD_DEVICES) {
 244                 errno = -EINVAL;
 245                 llapi_err(LLAPI_MSG_ERROR, "error: bad stripe offset %d",
 246                           stripe_offset);
 247                 return -EINVAL;
 248         }
 249         if (stripe_count < -1 || stripe_count > LOV_MAX_STRIPE_COUNT) {
 250                 errno = -EINVAL;
 251                 llapi_err(LLAPI_MSG_ERROR, "error: bad stripe count %d",
 252                           stripe_count);
 253                 return -EINVAL;
 254         }
 255         if (stripe_size >= (1ULL << 32)){
 256                 errno = -EINVAL;
 257                 llapi_err(LLAPI_MSG_ERROR, "warning: stripe size larger than 4G"
 258                           " is not currently supported and would wrap");
 259                 return -EINVAL;
 260         }
 261         return 0;
 262 }
 263
 264 static int find_target_obdpath(char *fsname, char *path)
 265 {
 266         glob_t glob_info;
 267         char pattern[PATH_MAX + 1];
 268         int rc;
 269
 270         snprintf(pattern, PATH_MAX,
 271                  "/proc/fs/lustre/lov/%s-*/target_obd",
 272                  fsname);
 273         rc = glob(pattern, GLOB_BRACE, NULL, &glob_info);
 274         if (rc == GLOB_NOMATCH)
 275                 return -ENODEV;
 276         else if (rc)
 277                 return -EINVAL;
 278
 279         strcpy(path, glob_info.gl_pathv[0]);
 280         globfree(&glob_info);
 281         return 0;
 282 }
 283
 284 static int find_poolpath(char *fsname, char *poolname, char *poolpath)
 285 {
 286         glob_t glob_info;
 287         char pattern[PATH_MAX + 1];
 288         int rc;
 289
 290         snprintf(pattern, PATH_MAX,
 291                  "/proc/fs/lustre/lov/%s-*/pools/%s",
 292                  fsname, poolname);
 293         rc = glob(pattern, GLOB_BRACE, NULL, &glob_info);
 294         /* If no pools, make sure the lov is available */
 295         if ((rc == GLOB_NOMATCH) &&
 296             (find_target_obdpath(fsname, poolpath) == -ENODEV))
 297                 return -ENODEV;
 298         if (rc)
 299                 return -EINVAL;
 300
 301         strcpy(poolpath, glob_info.gl_pathv[0]);
 302         globfree(&glob_info);
 303         return 0;
 304 }
 305
 306 /*
 307  * if pool is NULL, search ostname in target_obd
 308  * if pool is not NULL:
 309  *  if pool not found returns errno < 0
 310  *  if ostname is NULL, returns 1 if pool is not empty and 0 if pool empty
 311  *  if ostname is not NULL, returns 1 if OST is in pool and 0 if not
 312  */
 313 int llapi_search_ost(char *fsname, char *poolname, char *ostname)
 314 {
 315         FILE *fd;
 316         char buffer[PATH_MAX + 1];
 317         int len = 0, rc;
 318
 319         if (ostname != NULL)
 320                 len = strlen(ostname);
 321
 322         if (poolname == NULL)
 323                 rc = find_target_obdpath(fsname, buffer);
 324         else
 325                 rc = find_poolpath(fsname, poolname, buffer);
 326         if (rc)
 327                 return rc;
 328
 329         if ((fd = fopen(buffer, "r")) == NULL)
 330                 return -EINVAL;
 331
 332         while (fgets(buffer, sizeof(buffer), fd) != NULL) {
 333                 if (poolname == NULL) {
 334                         char *ptr;
 335                         /* Search for an ostname in the list of OSTs
 336                          Line format is IDX: fsname-OSTxxxx_UUID STATUS */
 337                         ptr = strchr(buffer, ' ');
 338                         if ((ptr != NULL) &&
 339                             (strncmp(ptr + 1, ostname, len) == 0)) {
 340                                 fclose(fd);
 341                                 return 1;
 342                         }
 343                 } else {
 344                         /* Search for an ostname in a pool,
 345                          (or an existing non-empty pool if no ostname) */
 346                         if ((ostname == NULL) ||
 347                             (strncmp(buffer, ostname, len) == 0)) {
 348                                 fclose(fd);
 349                                 return 1;
 350                         }
 351                 }
 352         }
 353         fclose(fd);
 354         return 0;
 355 }
 356
 357 int llapi_file_open_pool(const char *name, int flags, int mode,
 358                          unsigned long long stripe_size, int stripe_offset,
 359                          int stripe_count, int stripe_pattern, char *pool_name)
 360 {
 361         struct lov_user_md_v3 lum = { 0 };
 362         int fd, rc = 0;
 363         int isdir = 0;
 364
 365         /* Make sure we have a good pool */
 366         if (pool_name != NULL) {
 367                 char fsname[MAX_OBD_NAME + 1], *ptr;
 368
 369                 if (llapi_search_fsname(name, fsname)) {
 370                     llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO,
 371                               "'%s' is not on a Lustre filesystem", name);
 372                     return -EINVAL;
 373                 }
 374
 375                 /* in case user gives the full pool name <fsname>.<poolname>,
 376                  * strip the fsname */
 377                 ptr = strchr(pool_name, '.');
 378                 if (ptr != NULL) {
 379                         *ptr = '\0';
 380                         if (strcmp(pool_name, fsname) != 0) {
 381                                 *ptr = '.';
 382                                 llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO,
 383                                    "Pool '%s' is not on filesystem '%s'",
 384                                    pool_name, fsname);
 385                                 return -EINVAL;
 386                         }
 387                         pool_name = ptr + 1;
 388                 }
 389
 390                 /* Make sure the pool exists and is non-empty */
 391                 if ((rc = llapi_search_ost(fsname, pool_name, NULL)) < 1) {
 392                         llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO,
 393                                   "pool '%s.%s' %s", fsname, pool_name,
 394                                   rc == 0 ? "has no OSTs" : "does not exist");
 395                         return -EINVAL;
 396                 }
 397         }
 398
 399         fd = open(name, flags | O_LOV_DELAY_CREATE, mode);
 400         if (fd < 0 && errno == EISDIR) {
 401                 fd = open(name, O_DIRECTORY | O_RDONLY);
 402                 isdir++;
 403         }
 404
 405         if (fd < 0) {
 406                 rc = -errno;
 407                 llapi_err(LLAPI_MSG_ERROR, "unable to open '%s'", name);
 408                 return rc;
 409         }
 410
 411         if ((rc = llapi_stripe_limit_check(stripe_size, stripe_offset,
 412                                            stripe_count, stripe_pattern)) != 0){
 413                 errno = rc;
 414                 goto out;
 415         }
 416
 417         /*  Initialize IOCTL striping pattern structure */
 418         lum.lmm_magic = LOV_USER_MAGIC_V3;
 419         lum.lmm_pattern = stripe_pattern;
 420         lum.lmm_stripe_size = stripe_size;
 421         lum.lmm_stripe_count = stripe_count;
 422         lum.lmm_stripe_offset = stripe_offset;
 423         if (pool_name != NULL) {
 424                 strncpy(lum.lmm_pool_name, pool_name, LOV_MAXPOOLNAME);
 425         } else {
 426                 /* If no pool is specified at all, use V1 request */
 427                 lum.lmm_magic = LOV_USER_MAGIC_V1;
 428         }
 429
 430         if (ioctl(fd, LL_IOC_LOV_SETSTRIPE, &lum)) {
 431                 char *errmsg = "stripe already set";
 432                 rc = -errno;
 433                 if (errno != EEXIST && errno != EALREADY)
 434                         errmsg = strerror(errno);
 435
 436                 llapi_err_noerrno(LLAPI_MSG_ERROR,
 437                                   "error on ioctl "LPX64" for '%s' (%d): %s",
 438                                   (__u64)LL_IOC_LOV_SETSTRIPE, name, fd,errmsg);
 439         }
 440 out:
 441         if (rc) {
 442                 close(fd);
 443                 fd = rc;
 444         }
 445
 446         return fd;
 447 }
 448
 449 int llapi_file_open(const char *name, int flags, int mode,
 450                     unsigned long long stripe_size, int stripe_offset,
 451                     int stripe_count, int stripe_pattern)
 452 {
 453         return llapi_file_open_pool(name, flags, mode, stripe_size,
 454                                     stripe_offset, stripe_count,
 455                                     stripe_pattern, NULL);
 456 }
 457
 458 int llapi_file_create(const char *name, unsigned long long stripe_size,
 459                       int stripe_offset, int stripe_count, int stripe_pattern)
 460 {
 461         int fd;
 462
 463         fd = llapi_file_open_pool(name, O_CREAT | O_WRONLY, 0644, stripe_size,
 464                                   stripe_offset, stripe_count, stripe_pattern,
 465                                   NULL);
 466         if (fd < 0)
 467                 return fd;
 468
 469         close(fd);
 470         return 0;
 471 }
 472
 473 int llapi_file_create_pool(const char *name, unsigned long long stripe_size,
 474                            int stripe_offset, int stripe_count,
 475                            int stripe_pattern, char *pool_name)
 476 {
 477         int fd;
 478
 479         fd = llapi_file_open_pool(name, O_CREAT | O_WRONLY, 0644, stripe_size,
 480                                   stripe_offset, stripe_count, stripe_pattern,
 481                                   pool_name);
 482         if (fd < 0)
 483                 return fd;
 484
 485         close(fd);
 486         return 0;
 487 }
 488
 489 /*
 490  * Find the fsname, the full path, and/or an open fd.
 491  * Either the fsname or path must not be NULL
 492  */
 493 #define WANT_PATH   0x1
 494 #define WANT_FSNAME 0x2
 495 #define WANT_FD     0x4
 496 #define WANT_INDEX  0x8
 497 #define WANT_ERROR  0x10
 498 static int get_root_path(int want, char *fsname, int *outfd, char *path,
 499                          int index)
 500 {
 501         struct mntent mnt;
 502         char buf[PATH_MAX], mntdir[PATH_MAX];
 503         char *ptr;
 504         FILE *fp;
 505         int idx = 0, len = 0, mntlen, fd;
 506         int rc = -ENODEV;
 507
 508         /* get the mount point */
 509         fp = setmntent(MOUNTED, "r");
 510         if (fp == NULL) {
 511                  llapi_err(LLAPI_MSG_ERROR,
 512                            "setmntent(%s) failed: %s:", MOUNTED,
 513                            strerror (errno));
 514                  return -EIO;
 515         }
 516         while (1) {
 517                 if (getmntent_r(fp, &mnt, buf, sizeof(buf)) == NULL)
 518                         break;
 519
 520                 if (!llapi_is_lustre_mnt(&mnt))
 521                         continue;
 522
 523                 if ((want & WANT_INDEX) && (idx++ != index))
 524                         continue;
 525
 526                 mntlen = strlen(mnt.mnt_dir);
 527                 ptr = strrchr(mnt.mnt_fsname, '/');
 528                 if (!ptr && !len) {
 529                         rc = -EINVAL;
 530                         break;
 531                 }
 532                 ptr++;
 533
 534                 /* Check the fsname for a match, if given */
 535                 if (!(want & WANT_FSNAME) && fsname != NULL &&
 536                     (strlen(fsname) > 0) && (strcmp(ptr, fsname) != 0))
 537                         continue;
 538
 539                 /* If the path isn't set return the first one we find */
 540                 if (path == NULL || strlen(path) == 0) {
 541                         strcpy(mntdir, mnt.mnt_dir);
 542                         if ((want & WANT_FSNAME) && fsname != NULL)
 543                                 strcpy(fsname, ptr);
 544                         rc = 0;
 545                         break;
 546                 /* Otherwise find the longest matching path */
 547                 } else if ((strlen(path) >= mntlen) && (mntlen >= len) &&
 548                            (strncmp(mnt.mnt_dir, path, mntlen) == 0)) {
 549                         strcpy(mntdir, mnt.mnt_dir);
 550                         len = mntlen;
 551                         if ((want & WANT_FSNAME) && fsname != NULL)
 552                                 strcpy(fsname, ptr);
 553                         rc = 0;
 554                 }
 555         }
 556         endmntent(fp);
 557
 558         /* Found it */
 559         if (rc == 0) {
 560                 if ((want & WANT_PATH) && path != NULL)
 561                         strcpy(path, mntdir);
 562                 if (want & WANT_FD) {
 563                         fd = open(mntdir, O_RDONLY | O_DIRECTORY | O_NONBLOCK);
 564                         if (fd < 0) {
 565                                 perror("open");
 566                                 rc = -errno;
 567                         } else {
 568                                 *outfd = fd;
 569                         }
 570                 }
 571         } else if (want & WANT_ERROR)
 572                 llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO,
 573                           "can't find fs root for '%s': %d",
 574                           (want & WANT_PATH) ? fsname : path, rc);
 575         return rc;
 576 }
 577
 578 /*
 579  * search lustre mounts
 580  *
 581  * Calling this function will return to the user the mount point, mntdir, and
 582  * the file system name, fsname, if the user passed a buffer to this routine.
 583  *
 584  * The user inputs are pathname and index. If the pathname is supplied then
 585  * the value of the index will be ignored. The pathname will return data if
 586  * the pathname is located on a lustre mount. Index is used to pick which
 587  * mount point you want in the case of multiple mounted lustre file systems.
 588  * See function lfs_osts in lfs.c for a example of the index use.
 589  */
 590 int llapi_search_mounts(const char *pathname, int index, char *mntdir,
 591                         char *fsname)
 592 {
 593         int want = WANT_PATH, idx = -1;
 594
 595         if (!pathname || pathname[0] == '\0') {
 596                 want |= WANT_INDEX;
 597                 idx = index;
 598         } else
 599                 strcpy(mntdir, pathname);
 600
 601         if (fsname)
 602                 want |= WANT_FSNAME;
 603         return get_root_path(want, fsname, NULL, mntdir, idx);
 604 }
 605
 606 /* Given a path, find the corresponding Lustre fsname */
 607 int llapi_search_fsname(const char *pathname, char *fsname)
 608 {
 609         char *path = (char*)pathname, buf[PATH_MAX + 1];
 610
 611         if (pathname[0] != '/') { /* Need a absolute path */
 612                 memset(buf, '\0', sizeof(buf));
 613                 if (realpath(pathname, buf) == NULL) {
 614                         llapi_err(LLAPI_MSG_ERROR, "pathname '%s' cannot expand",
 615                                   pathname);
 616                         return -EINVAL;
 617                 }
 618                 path = buf;
 619         }
 620         return get_root_path(WANT_FSNAME | WANT_ERROR, fsname, NULL,
 621                              path, -1);
 622 }
 623
 624 /* return the first file matching this pattern */
 625 static int first_match(char *pattern, char *buffer)
 626 {
 627         glob_t glob_info;
 628
 629         if (glob(pattern, GLOB_BRACE, NULL, &glob_info))
 630                 return -ENOENT;
 631
 632         if (glob_info.gl_pathc < 1) {
 633                 globfree(&glob_info);
 634                 return -ENOENT;
 635         }
 636
 637         strcpy(buffer, glob_info.gl_pathv[0]);
 638
 639         globfree(&glob_info);
 640         return 0;
 641 }
 642
 643 /*
 644  * find the pool directory path under /proc
 645  * (can be also used to test if a fsname is known)
 646  */
 647 static int poolpath(char *fsname, char *pathname, char *pool_pathname)
 648 {
 649         int rc = 0;
 650         char pattern[PATH_MAX + 1];
 651         char buffer[PATH_MAX];
 652
 653         if (fsname == NULL) {
 654                 rc = llapi_search_fsname(pathname, buffer);
 655                 if (rc != 0)
 656                         return rc;
 657                 fsname = buffer;
 658                 strcpy(pathname, fsname);
 659         }
 660
 661         snprintf(pattern, PATH_MAX, "/proc/fs/lustre/lov/%s-*/pools", fsname);
 662         rc = first_match(pattern, buffer);
 663         if (rc)
 664                 return rc;
 665
 666         /* in fsname test mode, pool_pathname is NULL */
 667         if (pool_pathname != NULL)
 668                 strcpy(pool_pathname, buffer);
 669
 670         return 0;
 671 }
 672
 673 /**
 674  * Get the list of pool members.
 675  * \param poolname    string of format \<fsname\>.\<poolname\>
 676  * \param members     caller-allocated array of char*
 677  * \param list_size   size of the members array
 678  * \param buffer      caller-allocated buffer for storing OST names
 679  * \param buffer_size size of the buffer
 680  *
 681  * \return number of members retrieved for this pool
 682  * \retval -error failure
 683  */
 684 int llapi_get_poolmembers(const char *poolname, char **members,
 685                           int list_size, char *buffer, int buffer_size)
 686 {
 687         char fsname[PATH_MAX + 1];
 688         char *pool, *tmp;
 689         char pathname[PATH_MAX + 1];
 690         char path[PATH_MAX + 1];
 691         char buf[1024];
 692         FILE *fd;
 693         int rc = 0;
 694         int nb_entries = 0;
 695         int used = 0;
 696
 697         /* name is FSNAME.POOLNAME */
 698         if (strlen(poolname) > PATH_MAX)
 699                 return -EOVERFLOW;
 700         strcpy(fsname, poolname);
 701         pool = strchr(fsname, '.');
 702         if (pool == NULL)
 703                 return -EINVAL;
 704
 705         *pool = '\0';
 706         pool++;
 707
 708         rc = poolpath(fsname, NULL, pathname);
 709         if (rc != 0) {
 710                 errno = -rc;
 711                 llapi_err(LLAPI_MSG_ERROR, "Lustre filesystem '%s' not found",
 712                           fsname);
 713                 return rc;
 714         }
 715
 716         llapi_printf(LLAPI_MSG_NORMAL, "Pool: %s.%s\n", fsname, pool);
 717         sprintf(path, "%s/%s", pathname, pool);
 718         if ((fd = fopen(path, "r")) == NULL) {
 719                 llapi_err(LLAPI_MSG_ERROR, "Cannot open %s", path);
 720                 return -EINVAL;
 721         }
 722
 723         rc = 0;
 724         while (fgets(buf, sizeof(buf), fd) != NULL) {
 725                 if (nb_entries >= list_size) {
 726                         rc = -EOVERFLOW;
 727                         break;
 728                 }
 729                 /* remove '\n' */
 730                 if ((tmp = strchr(buf, '\n')) != NULL)
 731                         *tmp='\0';
 732                 if (used + strlen(buf) + 1 > buffer_size) {
 733                         rc = -EOVERFLOW;
 734                         break;
 735                 }
 736
 737                 strcpy(buffer + used, buf);
 738                 members[nb_entries] = buffer + used;
 739                 used += strlen(buf) + 1;
 740                 nb_entries++;
 741                 rc = nb_entries;
 742         }
 743
 744         fclose(fd);
 745         return rc;
 746 }
 747
 748 /**
 749  * Get the list of pools in a filesystem.
 750  * \param name        filesystem name or path
 751  * \param poollist    caller-allocated array of char*
 752  * \param list_size   size of the poollist array
 753  * \param buffer      caller-allocated buffer for storing pool names
 754  * \param buffer_size size of the buffer
 755  *
 756  * \return number of pools retrieved for this filesystem
 757  * \retval -error failure
 758  */
 759 int llapi_get_poollist(const char *name, char **poollist, int list_size,
 760                        char *buffer, int buffer_size)
 761 {
 762         char fsname[PATH_MAX + 1], rname[PATH_MAX + 1], pathname[PATH_MAX + 1];
 763         char *ptr;
 764         DIR *dir;
 765         struct dirent pool;
 766         struct dirent *cookie = NULL;
 767         int rc = 0;
 768         unsigned int nb_entries = 0;
 769         unsigned int used = 0;
 770         unsigned int i;
 771
 772         /* initilize output array */
 773         for (i = 0; i < list_size; i++)
 774                 poollist[i] = NULL;
 775
 776         /* is name a pathname ? */
 777         ptr = strchr(name, '/');
 778         if (ptr != NULL) {
 779                 /* only absolute pathname is supported */
 780                 if (*name != '/')
 781                         return -EINVAL;
 782                 if (!realpath(name, rname)) {
 783                         rc = -errno;
 784                         llapi_err(LLAPI_MSG_ERROR, "invalid path '%s'", name);
 785                         return rc;
 786                 }
 787
 788                 rc = poolpath(NULL, rname, pathname);
 789                 if (rc != 0) {
 790                         errno = -rc;
 791                         llapi_err(LLAPI_MSG_ERROR, "'%s' is not"
 792                                   " a Lustre filesystem", name);
 793                         return rc;
 794                 }
 795                 strcpy(fsname, rname);
 796         } else {
 797                 /* name is FSNAME */
 798                 strcpy(fsname, name);
 799                 rc = poolpath(fsname, NULL, pathname);
 800         }
 801         if (rc != 0) {
 802                 errno = -rc;
 803                 llapi_err(LLAPI_MSG_ERROR, "Lustre filesystem '%s' not found",
 804                           name);
 805                 return rc;
 806         }
 807
 808         llapi_printf(LLAPI_MSG_NORMAL, "Pools from %s:\n", fsname);
 809         if ((dir = opendir(pathname)) == NULL) {
 810                 llapi_err(LLAPI_MSG_ERROR, "Could not open pool list for '%s'",
 811                           name);
 812                 return -errno;
 813         }
 814
 815         while(1) {
 816                 rc = readdir_r(dir, &pool, &cookie);
 817
 818                 if (rc != 0) {
 819                         llapi_err(LLAPI_MSG_ERROR,
 820                                   "Error reading pool list for '%s'", name);
 821                         return -errno;
 822                 } else if ((rc == 0) && (cookie == NULL))
 823                         /* end of directory */
 824                         break;
 825
 826                 /* ignore . and .. */
 827                 if (!strcmp(pool.d_name, ".") || !strcmp(pool.d_name, ".."))
 828                         continue;
 829
 830                 /* check output bounds */
 831                 if (nb_entries >= list_size)
 832                         return -EOVERFLOW;
 833
 834                 /* +2 for '.' and final '\0' */
 835                 if (used + strlen(pool.d_name) + strlen(fsname) + 2
 836                     > buffer_size)
 837                         return -EOVERFLOW;
 838
 839                 sprintf(buffer + used, "%s.%s", fsname, pool.d_name);
 840                 poollist[nb_entries] = buffer + used;
 841                 used += strlen(pool.d_name) + strlen(fsname) + 2;
 842                 nb_entries++;
 843         }
 844
 845         closedir(dir);
 846         return nb_entries;
 847 }
 848
 849 /* wrapper for lfs.c and obd.c */
 850 int llapi_poollist(const char *name)
 851 {
 852         /* list of pool names (assume that pool count is smaller
 853            than OST count) */
 854         char *list[FIND_MAX_OSTS];
 855         char *buffer;
 856         /* fsname-OST0000_UUID < 32 char, 1 per OST */
 857         int bufsize = FIND_MAX_OSTS * 32;
 858         int i, nb;
 859
 860         buffer = malloc(bufsize);
 861         if (buffer == NULL)
 862                 return -ENOMEM;
 863
 864         if ((name[0] == '/') || (strchr(name, '.') == NULL))
 865                 /* name is a path or fsname */
 866                 nb = llapi_get_poollist(name, list, FIND_MAX_OSTS, buffer,
 867                                         bufsize);
 868         else
 869                 /* name is a pool name (<fsname>.<poolname>) */
 870                 nb = llapi_get_poolmembers(name, list, FIND_MAX_OSTS, buffer,
 871                                            bufsize);
 872
 873         for (i = 0; i < nb; i++)
 874                 llapi_printf(LLAPI_MSG_NORMAL, "%s\n", list[i]);
 875
 876         free(buffer);
 877         return (nb < 0 ? nb : 0);
 878 }
 879
 880
 881 typedef int (semantic_func_t)(char *path, DIR *parent, DIR *d,
 882                               void *data, cfs_dirent_t *de);
 883
 884 #define MAX_LOV_UUID_COUNT      max(LOV_MAX_STRIPE_COUNT, 1000)
 885 #define OBD_NOT_FOUND           (-1)
 886
 887 static int common_param_init(struct find_param *param)
 888 {
 889         param->lumlen = lov_mds_md_size(MAX_LOV_UUID_COUNT, LOV_MAGIC_V3);
 890         if ((param->lmd = malloc(sizeof(lstat_t) + param->lumlen)) == NULL) {
 891                 llapi_err(LLAPI_MSG_ERROR,
 892                           "error: allocation of %d bytes for ioctl",
 893                           sizeof(lstat_t) + param->lumlen);
 894                 return -ENOMEM;
 895         }
 896
 897         param->got_uuids = 0;
 898         param->obdindexes = NULL;
 899         param->obdindex = OBD_NOT_FOUND;
 900         return 0;
 901 }
 902
 903 static void find_param_fini(struct find_param *param)
 904 {
 905         if (param->obdindexes)
 906                 free(param->obdindexes);
 907
 908         if (param->lmd)
 909                 free(param->lmd);
 910 }
 911
 912 static int cb_common_fini(char *path, DIR *parent, DIR *d, void *data,
 913                           cfs_dirent_t *de)
 914 {
 915         struct find_param *param = (struct find_param *)data;
 916         param->depth--;
 917         return 0;
 918 }
 919
 920 static DIR *opendir_parent(char *path)
 921 {
 922         DIR *parent;
 923         char *fname;
 924         char c;
 925
 926         fname = strrchr(path, '/');
 927         if (fname == NULL)
 928                 return opendir(".");
 929
 930         c = fname[1];
 931         fname[1] = '\0';
 932         parent = opendir(path);
 933         fname[1] = c;
 934         return parent;
 935 }
 936
 937 int llapi_mds_getfileinfo(char *path, DIR *parent,
 938                           struct lov_user_mds_data *lmd)
 939 {
 940         lstat_t *st = &lmd->lmd_st;
 941         char *fname = strrchr(path, '/');
 942         int ret = 0;
 943
 944         if (parent == NULL)
 945                 return -EINVAL;
 946
 947         fname = (fname == NULL ? path : fname + 1);
 948         /* retrieve needed file info */
 949         strncpy((char *)lmd, fname,
 950                 lov_mds_md_size(MAX_LOV_UUID_COUNT, LOV_MAGIC));
 951         ret = ioctl(dirfd(parent), IOC_MDC_GETFILEINFO, (void *)lmd);
 952
 953         if (ret) {
 954                 if (errno == ENOTTY) {
 955                         /* ioctl is not supported, it is not a lustre fs.
 956                          * Do the regular lstat(2) instead. */
 957                         ret = lstat_f(path, st);
 958                         if (ret) {
 959                                 llapi_err(LLAPI_MSG_ERROR,
 960                                           "error: %s: lstat failed for %s",
 961                                           __func__, path);
 962                                 return ret;
 963                         }
 964                 } else if (errno == ENOENT) {
 965                         llapi_err(LLAPI_MSG_WARN,
 966                                   "warning: %s: %s does not exist",
 967                                   __func__, path);
 968                         return -ENOENT;
 969                 } else {
 970                         llapi_err(LLAPI_MSG_ERROR,
 971                                  "error: %s: IOC_MDC_GETFILEINFO failed for %s",
 972                                  __func__, path);
 973                         return ret;
 974                 }
 975         }
 976
 977         return 0;
 978 }
 979
 980 static int llapi_semantic_traverse(char *path, int size, DIR *parent,
 981                                    semantic_func_t sem_init,
 982                                    semantic_func_t sem_fini, void *data,
 983                                    cfs_dirent_t *de)
 984 {
 985         cfs_dirent_t *dent;
 986         int len, ret;
 987         DIR *d, *p = NULL;
 988
 989         ret = 0;
 990         len = strlen(path);
 991
 992         d = opendir(path);
 993         if (!d && errno != ENOTDIR) {
 994                 llapi_err(LLAPI_MSG_ERROR, "%s: Failed to open '%s'",
 995                           __func__, path);
 996                 return -EINVAL;
 997         } else if (!d && !parent) {
 998                 /* ENOTDIR. Open the parent dir. */
 999                 p = opendir_parent(path);
1000                 if (!p)
1001                         GOTO(out, ret = -EINVAL);
1002         }
1003
1004         if (sem_init && (ret = sem_init(path, parent ?: p, d, data, de)))
1005                 goto err;
1006
1007         if (!d)
1008                 GOTO(out, ret = 0);
1009
1010         while ((dent = readdir64(d)) != NULL) {
1011                 ((struct find_param *)data)->have_fileinfo = 0;
1012
1013                 if (!strcmp(dent->d_name, ".") || !strcmp(dent->d_name, ".."))
1014                         continue;
1015
1016                 /* Don't traverse .lustre directory */
1017                 if (!(strcmp(dent->d_name, dot_lustre_name)))
1018                         continue;
1019
1020                 path[len] = 0;
1021                 if ((len + dent->d_reclen + 2) > size) {
1022                         llapi_err(LLAPI_MSG_ERROR,
1023                                   "error: %s: string buffer is too small",
1024                                   __func__);
1025                         break;
1026                 }
1027                 strcat(path, "/");
1028                 strcat(path, dent->d_name);
1029
1030                 if (dent->d_type == DT_UNKNOWN) {
1031                         lstat_t *st = &((struct find_param *)data)->lmd->lmd_st;
1032
1033                         ret = llapi_mds_getfileinfo(path, d,
1034                                              ((struct find_param *)data)->lmd);
1035                         if (ret == 0) {
1036                                 ((struct find_param *)data)->have_fileinfo = 1;
1037                                 dent->d_type =
1038                                         llapi_filetype_dir_table[st->st_mode &
1039                                                                  S_IFMT];
1040                         }
1041                         if (ret == -ENOENT)
1042                                 continue;
1043                 }
1044
1045                 switch (dent->d_type) {
1046                 case DT_UNKNOWN:
1047                         llapi_err(LLAPI_MSG_ERROR,
1048                                   "error: %s: '%s' is UNKNOWN type %d",
1049                                   __func__, dent->d_name, dent->d_type);
1050                         break;
1051                 case DT_DIR:
1052                         ret = llapi_semantic_traverse(path, size, d, sem_init,
1053                                                       sem_fini, data, dent);
1054                         if (ret < 0)
1055                                 goto out;
1056                         break;
1057                 default:
1058                         ret = 0;
1059                         if (sem_init) {
1060                                 ret = sem_init(path, d, NULL, data, dent);
1061                                 if (ret < 0)
1062                                         goto out;
1063                         }
1064                         if (sem_fini && ret == 0)
1065                                 sem_fini(path, d, NULL, data, dent);
1066                 }
1067         }
1068
1069 out:
1070         path[len] = 0;
1071
1072         if (sem_fini)
1073                 sem_fini(path, parent, d, data, de);
1074 err:
1075         if (d)
1076                 closedir(d);
1077         if (p)
1078                 closedir(p);
1079         return ret;
1080 }
1081
1082 static int param_callback(char *path, semantic_func_t sem_init,
1083                           semantic_func_t sem_fini, struct find_param *param)
1084 {
1085         int ret, len = strlen(path);
1086         char *buf;
1087
1088         if (len > PATH_MAX) {
1089                 llapi_err(LLAPI_MSG_ERROR, "Path name '%s' is too long", path);
1090                 return -EINVAL;
1091         }
1092
1093         buf = (char *)malloc(PATH_MAX + 1);
1094         if (!buf)
1095                 return -ENOMEM;
1096
1097         ret = common_param_init(param);
1098         if (ret)
1099                 goto out;
1100         param->depth = 0;
1101
1102         strncpy(buf, path, PATH_MAX + 1);
1103         ret = llapi_semantic_traverse(buf, PATH_MAX + 1, NULL, sem_init,
1104                                       sem_fini, param, NULL);
1105 out:
1106         find_param_fini(param);
1107         free(buf);
1108         return ret < 0 ? ret : 0;
1109 }
1110
1111 int llapi_file_fget_lov_uuid(int fd, struct obd_uuid *lov_name)
1112 {
1113         int rc = ioctl(fd, OBD_IOC_GETNAME, lov_name);
1114         if (rc) {
1115                 rc = errno;
1116                 llapi_err(LLAPI_MSG_ERROR, "error: can't get lov name.");
1117         }
1118         return rc;
1119 }
1120
1121 int llapi_file_get_lov_uuid(const char *path, struct obd_uuid *lov_uuid)
1122 {
1123         int fd, rc;
1124
1125         fd = open(path, O_RDONLY);
1126         if (fd < 0) {
1127                 rc = errno;
1128                 llapi_err(LLAPI_MSG_ERROR, "error opening %s", path);
1129                 return rc;
1130         }
1131
1132         rc = llapi_file_fget_lov_uuid(fd, lov_uuid);
1133
1134         close(fd);
1135
1136         return rc;
1137 }
1138
1139 /*
1140  * If uuidp is NULL, return the number of available obd uuids.
1141  * If uuidp is non-NULL, then it will return the uuids of the obds. If
1142  * there are more OSTs then allocated to uuidp, then an error is returned with
1143  * the ost_count set to number of available obd uuids.
1144  */
1145 int llapi_lov_get_uuids(int fd, struct obd_uuid *uuidp, int *ost_count)
1146 {
1147         struct obd_uuid lov_name;
1148         char buf[1024];
1149         FILE *fp;
1150         int rc = 0, index = 0;
1151
1152         /* Get the lov name */
1153         rc = llapi_file_fget_lov_uuid(fd, &lov_name);
1154         if (rc)
1155                 return rc;
1156
1157         /* Now get the ost uuids from /proc */
1158         snprintf(buf, sizeof(buf), "/proc/fs/lustre/lov/%s/target_obd",
1159                  lov_name.uuid);
1160         fp = fopen(buf, "r");
1161         if (fp == NULL) {
1162                 rc = errno;
1163                 llapi_err(LLAPI_MSG_ERROR, "error: opening '%s'", buf);
1164                 return rc;
1165         }
1166
1167         while (fgets(buf, sizeof(buf), fp) != NULL) {
1168                 if (uuidp && (index < *ost_count)) {
1169                         if (sscanf(buf, "%d: %s", &index, uuidp[index].uuid) <2)
1170                                 break;
1171                 }
1172                 index++;
1173         }
1174
1175         fclose(fp);
1176
1177         if (uuidp && (index >= *ost_count))
1178                 return -EOVERFLOW;
1179
1180         *ost_count = index;
1181         return rc;
1182 }
1183
1184 int llapi_get_obd_count(char *mnt, int *count, int is_mdt)
1185 {
1186         DIR *root;
1187         int rc;
1188
1189         root = opendir(mnt);
1190         if (!root) {
1191                 llapi_err(LLAPI_MSG_ERROR, "open %s failed", mnt);
1192                 return -1;
1193         }
1194
1195         *count = is_mdt;
1196         rc = ioctl(dirfd(root), LL_IOC_GETOBDCOUNT, count);
1197
1198         closedir(root);
1199         return rc;
1200 }
1201
1202 /* Check if user specified value matches a real uuid.  Ignore _UUID,
1203  * -osc-4ba41334, other trailing gunk in comparison.
1204  * @param real_uuid ends in "_UUID"
1205  * @param search_uuid may or may not end in "_UUID"
1206  */
1207 int llapi_uuid_match(char *real_uuid, char *search_uuid)
1208 {
1209         int cmplen = strlen(real_uuid);
1210         int searchlen = strlen(search_uuid);
1211
1212         if (cmplen > 5 && strcmp(real_uuid + cmplen - 5, "_UUID") == 0)
1213                 cmplen -= 5;
1214         if (searchlen > 5 && strcmp(search_uuid + searchlen - 5, "_UUID") == 0)
1215                 searchlen -= 5;
1216
1217         /* The UUIDs may legitimately be different lengths, if
1218          * the system was upgraded from an older version. */
1219         if (cmplen != searchlen)
1220                 return 0;
1221
1222         return (strncmp(search_uuid, real_uuid, cmplen) == 0);
1223 }
1224
1225 /* Here, param->obduuid points to a single obduuid, the index of which is
1226  * returned in param->obdindex */
1227 static int setup_obd_uuid(DIR *dir, char *dname, struct find_param *param)
1228 {
1229         struct obd_uuid lov_uuid;
1230         char uuid[sizeof(struct obd_uuid)];
1231         char buf[1024];
1232         FILE *fp;
1233         int rc = 0, index;
1234
1235         if (param->got_uuids)
1236                 return rc;
1237
1238         /* Get the lov name */
1239         rc = llapi_file_fget_lov_uuid(dirfd(dir), &lov_uuid);
1240         if (rc) {
1241                 if (errno != ENOTTY) {
1242                         rc = errno;
1243                         llapi_err(LLAPI_MSG_ERROR,
1244                                   "error: can't get lov name: %s", dname);
1245                 } else {
1246                         rc = 0;
1247                 }
1248                 return rc;
1249         }
1250
1251         param->got_uuids = 1;
1252
1253         /* Now get the ost uuids from /proc */
1254         snprintf(buf, sizeof(buf), "/proc/fs/lustre/lov/%s/target_obd",
1255                  lov_uuid.uuid);
1256         fp = fopen(buf, "r");
1257         if (fp == NULL) {
1258                 rc = errno;
1259                 llapi_err(LLAPI_MSG_ERROR, "error: opening '%s'", buf);
1260                 return rc;
1261         }
1262
1263         if (!param->obduuid && !param->quiet && !param->obds_printed)
1264                 llapi_printf(LLAPI_MSG_NORMAL, "OBDS:\n");
1265
1266         while (fgets(buf, sizeof(buf), fp) != NULL) {
1267                 if (sscanf(buf, "%d: %s", &index, uuid) < 2)
1268                         break;
1269
1270                 if (param->obduuid) {
1271                         if (llapi_uuid_match(uuid, param->obduuid->uuid)) {
1272                                 param->obdindex = index;
1273                                 break;
1274                         }
1275                 } else if (!param->quiet && !param->obds_printed) {
1276                         /* Print everything */
1277                         llapi_printf(LLAPI_MSG_NORMAL, "%s", buf);
1278                 }
1279         }
1280         param->obds_printed = 1;
1281
1282         fclose(fp);
1283
1284         if (param->obduuid && (param->obdindex == OBD_NOT_FOUND)) {
1285                 llapi_err_noerrno(LLAPI_MSG_ERROR,
1286                                   "error: %s: unknown obduuid: %s",
1287                                   __func__, param->obduuid->uuid);
1288                 rc = -EINVAL;
1289         }
1290
1291         return (rc);
1292 }
1293
1294 /* In this case, param->obduuid will be an array of obduuids and
1295  * obd index for all these obduuids will be returned in
1296  * param->obdindexes */
1297 static int setup_obd_indexes(DIR *dir, struct find_param *param)
1298 {
1299         struct obd_uuid *uuids = NULL;
1300         int obdcount = INIT_ALLOC_NUM_OSTS;
1301         int ret, obd_valid = 0, obdnum, i;
1302
1303         uuids = (struct obd_uuid *)malloc(INIT_ALLOC_NUM_OSTS *
1304                                           sizeof(struct obd_uuid));
1305         if (uuids == NULL)
1306                 return -ENOMEM;
1307
1308 retry_get_uuids:
1309         ret = llapi_lov_get_uuids(dirfd(dir), uuids,
1310                                   &obdcount);
1311         if (ret) {
1312                 struct obd_uuid *uuids_temp;
1313
1314                 if (ret == -EOVERFLOW) {
1315                         uuids_temp = realloc(uuids, obdcount *
1316                                              sizeof(struct obd_uuid));
1317                         if (uuids_temp != NULL)
1318                                 goto retry_get_uuids;
1319                         else
1320                                 ret = -ENOMEM;
1321                 }
1322
1323                 llapi_err(LLAPI_MSG_ERROR, "get ost uuid failed");
1324                 return ret;
1325         }
1326
1327         param->obdindexes = malloc(param->num_obds * sizeof(param->obdindex));
1328         if (param->obdindexes == NULL)
1329                 return -ENOMEM;
1330
1331         for (obdnum = 0; obdnum < param->num_obds; obdnum++) {
1332                 for (i = 0; i < obdcount; i++) {
1333                         if (llapi_uuid_match(uuids[i].uuid,
1334                                              param->obduuid[obdnum].uuid)) {
1335                                 param->obdindexes[obdnum] = i;
1336                                 obd_valid++;
1337                                 break;
1338                         }
1339                 }
1340                 if (i >= obdcount) {
1341                         param->obdindexes[obdnum] = OBD_NOT_FOUND;
1342                         llapi_err_noerrno(LLAPI_MSG_ERROR,
1343                                           "error: %s: unknown obduuid: %s",
1344                                           __func__,
1345                                           param->obduuid[obdnum].uuid);
1346                         ret = -EINVAL;
1347                 }
1348         }
1349
1350         if (obd_valid == 0)
1351                 param->obdindex = OBD_NOT_FOUND;
1352         else
1353                 param->obdindex = obd_valid;
1354
1355         param->got_uuids = 1;
1356
1357         return ret;
1358 }
1359
1360 int llapi_ostlist(char *path, struct find_param *param)
1361 {
1362         DIR *dir;
1363         int ret;
1364
1365         dir = opendir(path);
1366         if (dir == NULL)
1367                 return -errno;
1368
1369         ret = setup_obd_uuid(dir, path, param);
1370         closedir(dir);
1371
1372         return ret;
1373 }
1374
1375 static void lov_dump_user_lmm_header(struct lov_user_md *lum, char *path,
1376                                      struct lov_user_ost_data_v1 *objects,
1377                                      int is_dir, int verbose, int depth,
1378                                      char *pool_name)
1379 {
1380         char *prefix = is_dir ? "" : "lmm_";
1381         char nl = is_dir ? ' ' : '\n';
1382
1383         if (is_dir && lum->lmm_object_seq == LOV_OBJECT_GROUP_DEFAULT) {
1384                 lum->lmm_object_seq = LOV_OBJECT_GROUP_CLEAR;
1385                 if (verbose & VERBOSE_DETAIL)
1386                         llapi_printf(LLAPI_MSG_NORMAL, "(Default) ");
1387         }
1388
1389         if (depth && path && ((verbose != VERBOSE_OBJID) || !is_dir))
1390                 llapi_printf(LLAPI_MSG_NORMAL, "%s\n", path);
1391
1392         if ((verbose & VERBOSE_DETAIL) && !is_dir) {
1393                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_magic:          0x%08X\n",
1394                              lum->lmm_magic);
1395                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_seq:            "LPX64"\n",
1396                              lum->lmm_object_seq);
1397                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_object_id:      "LPX64"\n",
1398                              lum->lmm_object_id);
1399         }
1400
1401         if (verbose & VERBOSE_COUNT) {
1402                 if (verbose & ~VERBOSE_COUNT)
1403                         llapi_printf(LLAPI_MSG_NORMAL, "%sstripe_count:   ",
1404                                      prefix);
1405                 llapi_printf(LLAPI_MSG_NORMAL, "%hd%c",
1406                              (__s16)lum->lmm_stripe_count, nl);
1407         }
1408
1409         if (verbose & VERBOSE_SIZE) {
1410                 if (verbose & ~VERBOSE_SIZE)
1411                         llapi_printf(LLAPI_MSG_NORMAL, "%sstripe_size:    ",
1412                                      prefix);
1413                 llapi_printf(LLAPI_MSG_NORMAL, "%u%c", lum->lmm_stripe_size,
1414                              nl);
1415         }
1416
1417         if ((verbose & VERBOSE_DETAIL) && !is_dir) {
1418                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_stripe_pattern: %x%c",
1419                              lum->lmm_pattern, nl);
1420         }
1421
1422         if (verbose & VERBOSE_OFFSET) {
1423                 if (verbose & ~VERBOSE_OFFSET)
1424                         llapi_printf(LLAPI_MSG_NORMAL, "%sstripe_offset:  ",
1425                                      prefix);
1426                 if (is_dir)
1427                         llapi_printf(LLAPI_MSG_NORMAL, "%d%c",
1428                                      lum->lmm_stripe_offset ==
1429                                      (typeof(lum->lmm_stripe_offset))(-1) ? -1 :
1430                                      lum->lmm_stripe_offset, nl);
1431                 else
1432                         llapi_printf(LLAPI_MSG_NORMAL, "%u%c",
1433                                      objects[0].l_ost_idx, nl);
1434         }
1435
1436         if ((verbose & VERBOSE_POOL) && (pool_name != NULL)) {
1437                 if (verbose & ~VERBOSE_POOL)
1438                         llapi_printf(LLAPI_MSG_NORMAL, "%spool:           ",
1439                                      prefix);
1440                 llapi_printf(LLAPI_MSG_NORMAL, "%s%c", pool_name, nl);
1441         }
1442
1443         if (is_dir && (verbose != VERBOSE_OBJID))
1444                 llapi_printf(LLAPI_MSG_NORMAL, "\n");
1445 }
1446
1447 void lov_dump_user_lmm_v1v3(struct lov_user_md *lum, char *pool_name,
1448                             struct lov_user_ost_data_v1 *objects,
1449                             char *path, int is_dir,
1450                             int obdindex, int depth, int header)
1451 {
1452         int i, obdstripe = (obdindex != OBD_NOT_FOUND) ? 0 : 1;
1453
1454         if (!obdstripe) {
1455                 for (i = 0; !is_dir && i < lum->lmm_stripe_count; i++) {
1456                         if (obdindex == objects[i].l_ost_idx) {
1457                                 obdstripe = 1;
1458                                 break;
1459                         }
1460                 }
1461         }
1462
1463         if (obdstripe == 1)
1464                 lov_dump_user_lmm_header(lum, path, objects, is_dir, header, depth,
1465                                          pool_name);
1466
1467         if (!is_dir && (header & VERBOSE_OBJID)) {
1468                 if (obdstripe == 1)
1469                         llapi_printf(LLAPI_MSG_NORMAL,
1470                                      "\tobdidx\t\t objid\t\tobjid\t\t group\n");
1471
1472                 for (i = 0; i < lum->lmm_stripe_count; i++) {
1473                         int idx = objects[i].l_ost_idx;
1474                         long long oid = objects[i].l_object_id;
1475                         long long gr = objects[i].l_object_seq;
1476                         if ((obdindex == OBD_NOT_FOUND) || (obdindex == idx))
1477                                 llapi_printf(LLAPI_MSG_NORMAL,
1478                                            "\t%6u\t%14llu\t%#13llx\t%14llu%s\n",
1479                                            idx, oid, oid, gr,
1480                                            obdindex == idx ? " *" : "");
1481                 }
1482                 llapi_printf(LLAPI_MSG_NORMAL, "\n");
1483         }
1484 }
1485
1486 void llapi_lov_dump_user_lmm(struct find_param *param,
1487                              char *path, int is_dir)
1488 {
1489         switch(*(__u32 *)&param->lmd->lmd_lmm) { /* lum->lmm_magic */
1490         case LOV_USER_MAGIC_V1:
1491                 lov_dump_user_lmm_v1v3(&param->lmd->lmd_lmm, NULL,
1492                                        param->lmd->lmd_lmm.lmm_objects,
1493                                        path, is_dir,
1494                                        param->obdindex, param->maxdepth,
1495                                        param->verbose);
1496                 break;
1497         case LOV_USER_MAGIC_V3: {
1498                 char pool_name[LOV_MAXPOOLNAME + 1];
1499                 struct lov_user_ost_data_v1 *objects;
1500                 struct lov_user_md_v3 *lmmv3 = (void *)&param->lmd->lmd_lmm;
1501
1502                 strncpy(pool_name, lmmv3->lmm_pool_name, LOV_MAXPOOLNAME);
1503                 pool_name[LOV_MAXPOOLNAME] = '\0';
1504                 objects = lmmv3->lmm_objects;
1505                 lov_dump_user_lmm_v1v3(&param->lmd->lmd_lmm, pool_name,
1506                                        objects, path, is_dir,
1507                                        param->obdindex, param->maxdepth,
1508                                        param->verbose);
1509                 break;
1510         }
1511         default:
1512                 llapi_printf(LLAPI_MSG_NORMAL, "unknown lmm_magic:  %#x "
1513                              "(expecting one of %#x %#x %#x)\n",
1514                              *(__u32 *)&param->lmd->lmd_lmm,
1515                              LOV_USER_MAGIC_V1, LOV_USER_MAGIC_V3);
1516                 return;
1517         }
1518 }
1519
1520 int llapi_file_get_stripe(const char *path, struct lov_user_md *lum)
1521 {
1522         const char *fname;
1523         char *dname;
1524         int fd, rc = 0;
1525
1526         fname = strrchr(path, '/');
1527
1528         /* It should be a file (or other non-directory) */
1529         if (fname == NULL) {
1530                 dname = (char *)malloc(2);
1531                 if (dname == NULL)
1532                         return ENOMEM;
1533                 strcpy(dname, ".");
1534                 fname = (char *)path;
1535         } else {
1536                 dname = (char *)malloc(fname - path + 1);
1537                 if (dname == NULL)
1538                         return ENOMEM;
1539                 strncpy(dname, path, fname - path);
1540                 dname[fname - path] = '\0';
1541                 fname++;
1542         }
1543
1544         if ((fd = open(dname, O_RDONLY)) == -1) {
1545                 rc = errno;
1546                 free(dname);
1547                 return rc;
1548         }
1549
1550         strcpy((char *)lum, fname);
1551         if (ioctl(fd, IOC_MDC_GETFILESTRIPE, (void *)lum) == -1)
1552                 rc = errno;
1553
1554         if (close(fd) == -1 && rc == 0)
1555                 rc = errno;
1556
1557         free(dname);
1558
1559         return rc;
1560 }
1561
1562 int llapi_file_lookup(int dirfd, const char *name)
1563 {
1564         struct obd_ioctl_data data = { 0 };
1565         char rawbuf[8192];
1566         char *buf = rawbuf;
1567         int rc;
1568
1569         if (dirfd < 0 || name == NULL)
1570                 return -EINVAL;
1571
1572         data.ioc_version = OBD_IOCTL_VERSION;
1573         data.ioc_len = sizeof(data);
1574         data.ioc_inlbuf1 = (char *)name;
1575         data.ioc_inllen1 = strlen(name) + 1;
1576
1577         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1578         if (rc) {
1579                 llapi_err(LLAPI_MSG_ERROR,
1580                           "error: IOC_MDC_LOOKUP pack failed for '%s': rc %d",
1581                           name, rc);
1582                 return rc;
1583         }
1584
1585         return ioctl(dirfd, IOC_MDC_LOOKUP, buf);
1586 }
1587
1588 /* Check if the value matches 1 of the given criteria (e.g. --atime +/-N).
1589  * @mds indicates if this is MDS timestamps and there are attributes on OSTs.
1590  *
1591  * The result is -1 if it does not match, 0 if not yet clear, 1 if matches.
1592  * The table below gives the answers for the specified parameters (value and
1593  * sign), 1st column is the answer for the MDS value, the 2nd is for the OST:
1594  * --------------------------------------
1595  * 1 | file > limit; sign > 0 | -1 / -1 |
1596  * 2 | file = limit; sign > 0 |  ? /  1 |
1597  * 3 | file < limit; sign > 0 |  ? /  1 |
1598  * 4 | file > limit; sign = 0 | -1 / -1 |
1599  * 5 | file = limit; sign = 0 |  ? /  1 |  <- (see the Note below)
1600  * 6 | file < limit; sign = 0 |  ? / -1 |
1601  * 7 | file > limit; sign < 0 |  1 /  1 |
1602  * 8 | file = limit; sign < 0 |  ? / -1 |
1603  * 9 | file < limit; sign < 0 |  ? / -1 |
1604  * --------------------------------------
1605  * Note: 5th actually means that the value is within the interval
1606  * (limit - margin, limit]. */
1607 static int find_value_cmp(unsigned int file, unsigned int limit, int sign,
1608                           int negopt, unsigned long long margin, int mds)
1609 {
1610         int ret = -1;
1611
1612         if (sign > 0) {
1613                 if (file <= limit)
1614                         ret = mds ? 0 : 1;
1615         } else if (sign == 0) {
1616                 if (file <= limit && file + margin >= limit)
1617                         ret = mds ? 0 : 1;
1618                 else if (file + margin <= limit)
1619                         ret = mds ? 0 : -1;
1620         } else if (sign < 0) {
1621                 if (file >= limit)
1622                         ret = 1;
1623                 else if (mds)
1624                         ret = 0;
1625         }
1626
1627         return negopt ? ~ret + 1 : ret;
1628 }
1629
1630 /* Check if the file time matches all the given criteria (e.g. --atime +/-N).
1631  * Return -1 or 1 if file timestamp does not or does match the given criteria
1632  * correspondingly. Return 0 if the MDS time is being checked and there are
1633  * attributes on OSTs and it is not yet clear if the timespamp matches.
1634  *
1635  * If 0 is returned, we need to do another RPC to the OSTs to obtain the
1636  * updated timestamps. */
1637 static int find_time_check(lstat_t *st, struct find_param *param, int mds)
1638 {
1639         int ret;
1640         int rc = 0;
1641
1642         /* Check if file is accepted. */
1643         if (param->atime) {
1644                 ret = find_value_cmp(st->st_atime, param->atime,
1645                                      param->asign, param->exclude_atime,
1646                                      24 * 60 * 60, mds);
1647                 if (ret < 0)
1648                         return ret;
1649                 rc = ret;
1650         }
1651
1652         if (param->mtime) {
1653                 ret = find_value_cmp(st->st_mtime, param->mtime,
1654                                      param->msign, param->exclude_mtime,
1655                                      24 * 60 * 60, mds);
1656                 if (ret < 0)
1657                         return ret;
1658
1659                 /* If the previous check matches, but this one is not yet clear,
1660                  * we should return 0 to do an RPC on OSTs. */
1661                 if (rc == 1)
1662                         rc = ret;
1663         }
1664
1665         if (param->ctime) {
1666                 ret = find_value_cmp(st->st_ctime, param->ctime,
1667                                      param->csign, param->exclude_ctime,
1668                                      24 * 60 * 60, mds);
1669                 if (ret < 0)
1670                         return ret;
1671
1672                 /* If the previous check matches, but this one is not yet clear,
1673                  * we should return 0 to do an RPC on OSTs. */
1674                 if (rc == 1)
1675                         rc = ret;
1676         }
1677
1678         return rc;
1679 }
1680
1681 static int cb_find_init(char *path, DIR *parent, DIR *dir,
1682                         void *data, cfs_dirent_t *de)
1683 {
1684         struct find_param *param = (struct find_param *)data;
1685         int decision = 1; /* 1 is accepted; -1 is rejected. */
1686         lstat_t *st = &param->lmd->lmd_st;
1687         int lustre_fs = 1;
1688         int checked_type = 0;
1689         int ret = 0;
1690
1691         LASSERT(parent != NULL || dir != NULL);
1692
1693         param->lmd->lmd_lmm.lmm_stripe_count = 0;
1694
1695         /* If a regular expression is presented, make the initial decision */
1696         if (param->pattern != NULL) {
1697                 char *fname = strrchr(path, '/');
1698                 fname = (fname == NULL ? path : fname + 1);
1699                 ret = fnmatch(param->pattern, fname, 0);
1700                 if ((ret == FNM_NOMATCH && !param->exclude_pattern) ||
1701                     (ret == 0 && param->exclude_pattern))
1702                         goto decided;
1703         }
1704
1705         /* See if we can check the file type from the dirent. */
1706         if (param->type && de != NULL && de->d_type != DT_UNKNOWN &&
1707             de->d_type < DT_MAX) {
1708                 checked_type = 1;
1709                 if (llapi_dir_filetype_table[de->d_type] == param->type) {
1710                         if (param->exclude_type)
1711                                 goto decided;
1712                 } else {
1713                         if (!param->exclude_type)
1714                                 goto decided;
1715                 }
1716         }
1717
1718
1719         /* If a time or OST should be checked, the decision is not taken yet. */
1720         if (param->atime || param->ctime || param->mtime || param->obduuid ||
1721             param->check_size)
1722                 decision = 0;
1723
1724         ret = 0;
1725         /* Request MDS for the stat info. */
1726         if (param->have_fileinfo == 0) {
1727                 if (dir) {
1728                         /* retrieve needed file info */
1729                         ret = ioctl(dirfd(dir), LL_IOC_MDC_GETINFO,
1730                                     (void *)param->lmd);
1731                 } else {
1732                         char *fname = strrchr(path, '/');
1733                         fname = (fname == NULL ? path : fname + 1);
1734
1735                         /* retrieve needed file info */
1736                         strncpy((char *)param->lmd, fname, param->lumlen);
1737                         ret = ioctl(dirfd(parent), IOC_MDC_GETFILEINFO,
1738                                    (void *)param->lmd);
1739                 }
1740         }
1741
1742         if (ret) {
1743                 if (errno == ENOTTY) {
1744                         /* ioctl is not supported, it is not a lustre fs.
1745                          * Do the regular lstat(2) instead. */
1746                         lustre_fs = 0;
1747                         ret = lstat_f(path, st);
1748                         if (ret) {
1749                                 llapi_err(LLAPI_MSG_ERROR,
1750                                           "error: %s: lstat failed for %s",
1751                                           __func__, path);
1752                                 return ret;
1753                         }
1754                 } else if (errno == ENOENT) {
1755                         llapi_err(LLAPI_MSG_WARN,
1756                                   "warning: %s: %s does not exist",
1757                                   __func__, path);
1758                         goto decided;
1759                 } else {
1760                         llapi_err(LLAPI_MSG_ERROR,"error: %s: %s failed for %s",
1761                                   __func__, dir ? "LL_IOC_MDC_GETINFO" :
1762                                   "IOC_MDC_GETFILEINFO", path);
1763                         return ret;
1764                 }
1765         }
1766
1767         if (param->type && !checked_type) {
1768                 if ((st->st_mode & S_IFMT) == param->type) {
1769                         if (param->exclude_type)
1770                                 goto decided;
1771                 } else {
1772                         if (!param->exclude_type)
1773                                 goto decided;
1774                 }
1775         }
1776
1777         /* Prepare odb. */
1778         if (param->obduuid) {
1779                 if (lustre_fs && param->got_uuids &&
1780                     param->st_dev != st->st_dev) {
1781                         /* A lustre/lustre mount point is crossed. */
1782                         param->got_uuids = 0;
1783                         param->obds_printed = 0;
1784                         param->obdindex = OBD_NOT_FOUND;
1785                 }
1786
1787                 if (lustre_fs && !param->got_uuids) {
1788                         ret = setup_obd_indexes(dir ? dir : parent, param);
1789                         if (ret)
1790                                 return ret;
1791
1792                         param->st_dev = st->st_dev;
1793                 } else if (!lustre_fs && param->got_uuids) {
1794                         /* A lustre/non-lustre mount point is crossed. */
1795                         param->got_uuids = 0;
1796                         param->obdindex = OBD_NOT_FOUND;
1797                 }
1798         }
1799
1800         /* If an OBD UUID is specified but no one matches, skip this file. */
1801         if (param->obduuid && param->obdindex == OBD_NOT_FOUND)
1802                 goto decided;
1803
1804         /* If a OST UUID is given, and some OST matches, check it here. */
1805         if (param->obdindex != OBD_NOT_FOUND) {
1806                 if (!S_ISREG(st->st_mode))
1807                         goto decided;
1808
1809                 /* Only those files should be accepted, which have a
1810                  * stripe on the specified OST. */
1811                 if (!param->lmd->lmd_lmm.lmm_stripe_count) {
1812                         goto decided;
1813                 } else {
1814                         int i, j;
1815                         struct lov_user_ost_data_v1 *lmm_objects;
1816
1817                         if (param->lmd->lmd_lmm.lmm_magic ==
1818                             LOV_USER_MAGIC_V3) {
1819                                 struct lov_user_md_v3 *lmmv3 =
1820                                         (void *)&param->lmd->lmd_lmm;
1821
1822                                 lmm_objects = lmmv3->lmm_objects;
1823                         } else {
1824                                 lmm_objects = param->lmd->lmd_lmm.lmm_objects;
1825                         }
1826
1827                         for (i = 0;
1828                              i < param->lmd->lmd_lmm.lmm_stripe_count; i++) {
1829                                 for (j = 0; j < param->num_obds; j++) {
1830                                         if (param->obdindexes[j] ==
1831                                             lmm_objects[i].l_ost_idx) {
1832                                                 if (param->exclude_obd)
1833                                                         goto decided;
1834                                                 goto obd_matches;
1835                                         }
1836                                 }
1837                         }
1838
1839                         if (i == param->lmd->lmd_lmm.lmm_stripe_count) {
1840                                 if (param->exclude_obd)
1841                                         goto obd_matches;
1842                                 goto decided;
1843                         }
1844                 }
1845         }
1846
1847         if (param->check_uid) {
1848                 if (st->st_uid == param->uid) {
1849                         if (param->exclude_uid)
1850                                 goto decided;
1851                 } else {
1852                         if (!param->exclude_uid)
1853                                 goto decided;
1854                 }
1855         }
1856
1857         if (param->check_gid) {
1858                 if (st->st_gid == param->gid) {
1859                         if (param->exclude_gid)
1860                                 goto decided;
1861                 } else {
1862                         if (!param->exclude_gid)
1863                                 goto decided;
1864                 }
1865         }
1866
1867         if (param->check_pool) {
1868                 struct lov_user_md_v3 *lmmv3 = (void *)&param->lmd->lmd_lmm;
1869
1870                 /* empty requested pool is taken as no pool search => V1 */
1871                 if (((param->lmd->lmd_lmm.lmm_magic == LOV_USER_MAGIC_V1) &&
1872                      (param->poolname[0] == '\0')) ||
1873                     ((param->lmd->lmd_lmm.lmm_magic == LOV_USER_MAGIC_V3) &&
1874                      (strncmp(lmmv3->lmm_pool_name,
1875                               param->poolname, LOV_MAXPOOLNAME) == 0)) ||
1876                     ((param->lmd->lmd_lmm.lmm_magic == LOV_USER_MAGIC_V3) &&
1877                      (strcmp(param->poolname, "*") == 0))) {
1878                         if (param->exclude_pool)
1879                                 goto decided;
1880                 } else {
1881                         if (!param->exclude_pool)
1882                                 goto decided;
1883                 }
1884         }
1885
1886         /* Check the time on mds. */
1887         if (!decision) {
1888                 int for_mds;
1889
1890                 for_mds = lustre_fs ? (S_ISREG(st->st_mode) &&
1891                                        param->lmd->lmd_lmm.lmm_stripe_count)
1892                                     : 0;
1893                 decision = find_time_check(st, param, for_mds);
1894         }
1895
1896 obd_matches:
1897         /* If file still fits the request, ask ost for updated info.
1898            The regular stat is almost of the same speed as some new
1899            'glimpse-size-ioctl'. */
1900         if (!decision && S_ISREG(st->st_mode) &&
1901             param->lmd->lmd_lmm.lmm_stripe_count &&
1902             (param->check_size ||param->atime || param->mtime || param->ctime)) {
1903                 if (param->obdindex != OBD_NOT_FOUND) {
1904                         /* Check whether the obd is active or not, if it is
1905                          * not active, just print the object affected by this
1906                          * failed ost
1907                          * */
1908                         struct obd_statfs stat_buf;
1909                         struct obd_uuid uuid_buf;
1910
1911                         memset(&stat_buf, 0, sizeof(struct obd_statfs));
1912                         memset(&uuid_buf, 0, sizeof(struct obd_uuid));
1913                         ret = llapi_obd_statfs(path, LL_STATFS_LOV,
1914                                                param->obdindex, &stat_buf,
1915                                                &uuid_buf);
1916                         if (ret) {
1917                                 if (ret == -ENODATA || ret == -ENODEV
1918                                     || ret == -EIO)
1919                                         errno = EIO;
1920                                 llapi_printf(LLAPI_MSG_NORMAL,
1921                                              "obd_uuid: %s failed %s ",
1922                                              param->obduuid->uuid,
1923                                              strerror(errno));
1924                                 goto print_path;
1925                         }
1926                 }
1927                 if (dir) {
1928                         ret = ioctl(dirfd(dir), IOC_LOV_GETINFO,
1929                                     (void *)param->lmd);
1930                 } else if (parent) {
1931                         ret = ioctl(dirfd(parent), IOC_LOV_GETINFO,
1932                                     (void *)param->lmd);
1933                 }
1934
1935                 if (ret) {
1936                         if (errno == ENOENT) {
1937                                 llapi_err(LLAPI_MSG_ERROR,
1938                                           "warning: %s: %s does not exist",
1939                                           __func__, path);
1940                                 goto decided;
1941                         } else {
1942                                 llapi_err(LLAPI_MSG_ERROR,
1943                                           "%s: IOC_LOV_GETINFO on %s failed",
1944                                           __func__, path);
1945                                 return ret;
1946                         }
1947                 }
1948
1949                 /* Check the time on osc. */
1950                 decision = find_time_check(st, param, 0);
1951                 if (decision == -1)
1952                         goto decided;
1953         }
1954
1955         if (param->check_size)
1956                 decision = find_value_cmp(st->st_size, param->size,
1957                                           param->size_sign, param->exclude_size,
1958                                           param->size_units, 0);
1959
1960 print_path:
1961         if (decision != -1) {
1962                 llapi_printf(LLAPI_MSG_NORMAL, "%s", path);
1963                 if (param->zeroend)
1964                         llapi_printf(LLAPI_MSG_NORMAL, "%c", '\0');
1965                 else
1966                         llapi_printf(LLAPI_MSG_NORMAL, "\n");
1967         }
1968
1969 decided:
1970         /* Do not get down anymore? */
1971         if (param->depth == param->maxdepth)
1972                 return 1;
1973
1974         param->depth++;
1975         return 0;
1976 }
1977
1978 int llapi_find(char *path, struct find_param *param)
1979 {
1980         return param_callback(path, cb_find_init, cb_common_fini, param);
1981 }
1982
1983 /*
1984  * Get MDT number that the file/directory inode referenced
1985  * by the open fd resides on.
1986  * Return 0 and mdtidx on success, or -ve errno.
1987  */
1988 int llapi_file_fget_mdtidx(int fd, int *mdtidx)
1989 {
1990         if (ioctl(fd, LL_IOC_GET_MDTIDX, &mdtidx) < 0)
1991                 return -errno;
1992         return 0;
1993 }
1994
1995 static int cb_get_mdt_index(char *path, DIR *parent, DIR *d, void *data,
1996                             cfs_dirent_t *de)
1997 {
1998         struct find_param *param = (struct find_param *)data;
1999         int ret = 0;
2000         int mdtidx;
2001
2002         LASSERT(parent != NULL || d != NULL);
2003
2004         if (d) {
2005                 ret = llapi_file_fget_mdtidx(dirfd(d), &mdtidx);
2006         } else if (parent) {
2007                 int fd;
2008
2009                 fd = open(path, O_RDONLY);
2010                 if (fd > 0) {
2011                         ret = llapi_file_fget_mdtidx(fd, &mdtidx);
2012                         close(fd);
2013                 } else {
2014                         ret = fd;
2015                 }
2016         }
2017
2018         if (ret) {
2019                 if (errno == ENODATA) {
2020                         if (!param->obduuid)
2021                                 llapi_printf(LLAPI_MSG_NORMAL,
2022                                              "%s has no stripe info\n", path);
2023                         goto out;
2024                 } else if (errno == ENOTTY) {
2025                         llapi_err(LLAPI_MSG_ERROR,
2026                                   "%s: '%s' not on a Lustre fs?",
2027                                   __func__, path);
2028                 } else if (errno == ENOENT) {
2029                         llapi_err(LLAPI_MSG_WARN,
2030                                   "warning: %s: %s does not exist",
2031                                   __func__, path);
2032                         goto out;
2033                 } else {
2034                         llapi_err(LLAPI_MSG_ERROR,
2035                                   "error: %s: LL_IOC_GET_MDTIDX failed for %s",
2036                                    __func__, path);
2037                 }
2038                 return ret;
2039         }
2040
2041         if (param->quiet)
2042                 llapi_printf(LLAPI_MSG_NORMAL, "%d\n", mdtidx);
2043         else
2044                 llapi_printf(LLAPI_MSG_NORMAL, "%s MDT index: %d\n",
2045                              path, mdtidx);
2046
2047 out:
2048         /* Do not get down anymore? */
2049         if (param->depth == param->maxdepth)
2050                 return 1;
2051
2052         param->depth++;
2053         return 0;
2054 }
2055
2056 static int cb_getstripe(char *path, DIR *parent, DIR *d, void *data,
2057                         cfs_dirent_t *de)
2058 {
2059         struct find_param *param = (struct find_param *)data;
2060         int ret = 0;
2061
2062         LASSERT(parent != NULL || d != NULL);
2063
2064         if (param->obduuid) {
2065                 param->quiet = 1;
2066                 ret = setup_obd_uuid(d ? d : parent, path, param);
2067                 if (ret)
2068                         return ret;
2069         }
2070
2071         if (d) {
2072                 ret = ioctl(dirfd(d), LL_IOC_LOV_GETSTRIPE,
2073                             (void *)&param->lmd->lmd_lmm);
2074         } else if (parent) {
2075                 char *fname = strrchr(path, '/');
2076                 fname = (fname == NULL ? path : fname + 1);
2077
2078                 strncpy((char *)&param->lmd->lmd_lmm, fname, param->lumlen);
2079
2080                 ret = ioctl(dirfd(parent), IOC_MDC_GETFILESTRIPE,
2081                             (void *)&param->lmd->lmd_lmm);
2082         }
2083
2084         if (ret) {
2085                 if (errno == ENODATA) {
2086                         if (!param->obduuid)
2087                                 llapi_printf(LLAPI_MSG_NORMAL,
2088                                              "%s has no stripe info\n", path);
2089                         goto out;
2090                 } else if (errno == ENOTTY) {
2091                         llapi_err(LLAPI_MSG_ERROR,
2092                                   "%s: '%s' not on a Lustre fs?",
2093                                   __func__, path);
2094                 } else if (errno == ENOENT) {
2095                         llapi_err(LLAPI_MSG_WARN,
2096                                   "warning: %s: %s does not exist",
2097                                   __func__, path);
2098                         goto out;
2099                 } else {
2100                         llapi_err(LLAPI_MSG_ERROR,
2101                                   "error: %s: %s failed for %s",
2102                                    __func__, d ? "LL_IOC_LOV_GETSTRIPE" :
2103                                   "IOC_MDC_GETFILESTRIPE", path);
2104                 }
2105
2106                 return ret;
2107         }
2108
2109         if (!param->get_mdt_index)
2110                 llapi_lov_dump_user_lmm(param, path, d ? 1 : 0);
2111
2112 out:
2113         /* Do not get down anymore? */
2114         if (param->depth == param->maxdepth)
2115                 return 1;
2116
2117         param->depth++;
2118         return 0;
2119 }
2120
2121 int llapi_getstripe(char *path, struct find_param *param)
2122 {
2123         return param_callback(path, param->get_mdt_index ?
2124                               cb_get_mdt_index : cb_getstripe,
2125                               cb_common_fini, param);
2126 }
2127
2128 int llapi_obd_statfs(char *path, __u32 type, __u32 index,
2129                      struct obd_statfs *stat_buf,
2130                      struct obd_uuid *uuid_buf)
2131 {
2132         int fd;
2133         char raw[OBD_MAX_IOCTL_BUFFER] = {'\0'};
2134         char *rawbuf = raw;
2135         struct obd_ioctl_data data = { 0 };
2136         int rc = 0;
2137
2138         data.ioc_inlbuf1 = (char *)&type;
2139         data.ioc_inllen1 = sizeof(__u32);
2140         data.ioc_inlbuf2 = (char *)&index;
2141         data.ioc_inllen2 = sizeof(__u32);
2142         data.ioc_pbuf1 = (char *)stat_buf;
2143         data.ioc_plen1 = sizeof(struct obd_statfs);
2144         data.ioc_pbuf2 = (char *)uuid_buf;
2145         data.ioc_plen2 = sizeof(struct obd_uuid);
2146
2147         if ((rc = obd_ioctl_pack(&data, &rawbuf, sizeof(raw))) != 0) {
2148                 llapi_err(LLAPI_MSG_ERROR,
2149                           "llapi_obd_statfs: error packing ioctl data");
2150                 return rc;
2151         }
2152
2153         fd = open(path, O_RDONLY);
2154         if (errno == EISDIR)
2155                 fd = open(path, O_DIRECTORY | O_RDONLY);
2156
2157         if (fd < 0) {
2158                 rc = errno ? -errno : -EBADF;
2159                 llapi_err(LLAPI_MSG_ERROR, "error: %s: opening '%s'",
2160                           __func__, path);
2161                 return rc;
2162         }
2163         rc = ioctl(fd, IOC_OBD_STATFS, (void *)rawbuf);
2164         if (rc)
2165                 rc = errno ? -errno : -EINVAL;
2166
2167         close(fd);
2168         return rc;
2169 }
2170
2171 #define MAX_STRING_SIZE 128
2172 #define DEVICES_LIST "/proc/fs/lustre/devices"
2173
2174 int llapi_ping(char *obd_type, char *obd_name)
2175 {
2176         char path[MAX_STRING_SIZE];
2177         char buf[1];
2178         int rc, fd;
2179
2180         snprintf(path, MAX_STRING_SIZE, "/proc/fs/lustre/%s/%s/ping",
2181                  obd_type, obd_name);
2182
2183         fd = open(path, O_WRONLY);
2184         if (fd < 0) {
2185                 rc = errno;
2186                 llapi_err(LLAPI_MSG_ERROR, "error opening %s", path);
2187                 return rc;
2188         }
2189
2190         rc = write(fd, buf, 1);
2191         close(fd);
2192
2193         if (rc == 1)
2194                 return 0;
2195         return rc;
2196 }
2197
2198 int llapi_target_iterate(int type_num, char **obd_type,void *args,llapi_cb_t cb)
2199 {
2200         char buf[MAX_STRING_SIZE];
2201         FILE *fp = fopen(DEVICES_LIST, "r");
2202         int i, rc = 0;
2203
2204         if (fp == NULL) {
2205                 rc = errno;
2206                 llapi_err(LLAPI_MSG_ERROR, "error: opening "DEVICES_LIST);
2207                 return rc;
2208         }
2209
2210         while (fgets(buf, sizeof(buf), fp) != NULL) {
2211                 char *obd_type_name = NULL;
2212                 char *obd_name = NULL;
2213                 char *obd_uuid = NULL;
2214                 char *bufp = buf;
2215                 struct obd_ioctl_data datal = { 0, };
2216                 struct obd_statfs osfs_buffer;
2217
2218                 while(bufp[0] == ' ')
2219                         ++bufp;
2220
2221                 for(i = 0; i < 3; i++) {
2222                         obd_type_name = strsep(&bufp, " ");
2223                 }
2224                 obd_name = strsep(&bufp, " ");
2225                 obd_uuid = strsep(&bufp, " ");
2226
2227                 memset(&osfs_buffer, 0, sizeof (osfs_buffer));
2228
2229                 datal.ioc_pbuf1 = (char *)&osfs_buffer;
2230                 datal.ioc_plen1 = sizeof(osfs_buffer);
2231
2232                 for (i = 0; i < type_num; i++) {
2233                         if (strcmp(obd_type_name, obd_type[i]) != 0)
2234                                 continue;
2235
2236                         cb(obd_type_name, obd_name, obd_uuid, args);
2237                 }
2238         }
2239         fclose(fp);
2240         return rc;
2241 }
2242
2243 static void do_target_check(char *obd_type_name, char *obd_name,
2244                             char *obd_uuid, void *args)
2245 {
2246         int rc;
2247
2248         rc = llapi_ping(obd_type_name, obd_name);
2249         if (rc == ENOTCONN) {
2250                 llapi_printf(LLAPI_MSG_NORMAL, "%s inactive.\n", obd_name);
2251         } else if (rc) {
2252                 llapi_err(LLAPI_MSG_ERROR, "error: check '%s'", obd_name);
2253         } else {
2254                 llapi_printf(LLAPI_MSG_NORMAL, "%s active.\n", obd_name);
2255         }
2256 }
2257
2258 int llapi_target_check(int type_num, char **obd_type, char *dir)
2259 {
2260         return llapi_target_iterate(type_num, obd_type, NULL, do_target_check);
2261 }
2262
2263 #undef MAX_STRING_SIZE
2264
2265 int llapi_catinfo(char *dir, char *keyword, char *node_name)
2266 {
2267         char raw[OBD_MAX_IOCTL_BUFFER];
2268         char out[LLOG_CHUNK_SIZE];
2269         char *buf = raw;
2270         struct obd_ioctl_data data = { 0 };
2271         char key[30];
2272         DIR *root;
2273         int rc;
2274
2275         sprintf(key, "%s", keyword);
2276         memset(raw, 0, sizeof(raw));
2277         memset(out, 0, sizeof(out));
2278         data.ioc_inlbuf1 = key;
2279         data.ioc_inllen1 = strlen(key) + 1;
2280         if (node_name) {
2281                 data.ioc_inlbuf2 = node_name;
2282                 data.ioc_inllen2 = strlen(node_name) + 1;
2283         }
2284         data.ioc_pbuf1 = out;
2285         data.ioc_plen1 = sizeof(out);
2286         rc = obd_ioctl_pack(&data, &buf, sizeof(raw));
2287         if (rc)
2288                 return rc;
2289
2290         root = opendir(dir);
2291         if (root == NULL) {
2292                 rc = errno;
2293                 llapi_err(LLAPI_MSG_ERROR, "open %s failed", dir);
2294                 return rc;
2295         }
2296
2297         rc = ioctl(dirfd(root), OBD_IOC_LLOG_CATINFO, buf);
2298         if (rc)
2299                 llapi_err(LLAPI_MSG_ERROR, "ioctl OBD_IOC_CATINFO failed");
2300         else
2301                 llapi_printf(LLAPI_MSG_NORMAL, "%s", data.ioc_pbuf1);
2302
2303         closedir(root);
2304         return rc;
2305 }
2306
2307 /* Is this a lustre fs? */
2308 int llapi_is_lustre_mnttype(const char *type)
2309 {
2310         return (strcmp(type, "lustre") == 0 || strcmp(type,"lustre_lite") == 0);
2311 }
2312
2313 /* Is this a lustre client fs? */
2314 int llapi_is_lustre_mnt(struct mntent *mnt)
2315 {
2316         return (llapi_is_lustre_mnttype(mnt->mnt_type) &&
2317                 strstr(mnt->mnt_fsname, ":/") != NULL);
2318 }
2319
2320 int llapi_quotacheck(char *mnt, int check_type)
2321 {
2322         DIR *root;
2323         int rc;
2324
2325         root = opendir(mnt);
2326         if (!root) {
2327                 llapi_err(LLAPI_MSG_ERROR, "open %s failed", mnt);
2328                 return -1;
2329         }
2330
2331         rc = ioctl(dirfd(root), LL_IOC_QUOTACHECK, check_type);
2332
2333         closedir(root);
2334         return rc;
2335 }
2336
2337 int llapi_poll_quotacheck(char *mnt, struct if_quotacheck *qchk)
2338 {
2339         DIR *root;
2340         int poll_intvl = 2;
2341         int rc;
2342
2343         root = opendir(mnt);
2344         if (!root) {
2345                 llapi_err(LLAPI_MSG_ERROR, "open %s failed", mnt);
2346                 return -1;
2347         }
2348
2349         while (1) {
2350                 rc = ioctl(dirfd(root), LL_IOC_POLL_QUOTACHECK, qchk);
2351                 if (!rc)
2352                         break;
2353                 sleep(poll_intvl);
2354                 if (poll_intvl < 30)
2355                         poll_intvl *= 2;
2356         }
2357
2358         closedir(root);
2359         return rc;
2360 }
2361
2362 int llapi_quotactl(char *mnt, struct if_quotactl *qctl)
2363 {
2364         DIR *root;
2365         int rc;
2366
2367         root = opendir(mnt);
2368         if (!root) {
2369                 llapi_err(LLAPI_MSG_ERROR, "open %s failed", mnt);
2370                 return -1;
2371         }
2372
2373         rc = ioctl(dirfd(root), LL_IOC_QUOTACTL, qctl);
2374
2375         closedir(root);
2376         return rc;
2377 }
2378
2379 static int cb_quotachown(char *path, DIR *parent, DIR *d, void *data,
2380                          cfs_dirent_t *de)
2381 {
2382         struct find_param *param = (struct find_param *)data;
2383         lstat_t *st;
2384         int rc;
2385
2386         LASSERT(parent != NULL || d != NULL);
2387
2388         if (d) {
2389                 rc = ioctl(dirfd(d), LL_IOC_MDC_GETINFO,
2390                            (void *)param->lmd);
2391         } else if (parent) {
2392                 char *fname = strrchr(path, '/');
2393                 fname = (fname == NULL ? path : fname + 1);
2394
2395                 strncpy((char *)param->lmd, fname, param->lumlen);
2396                 rc = ioctl(dirfd(parent), IOC_MDC_GETFILEINFO,
2397                            (void *)param->lmd);
2398         } else {
2399                 return 0;
2400         }
2401
2402         if (rc) {
2403                 if (errno == ENODATA) {
2404                         if (!param->obduuid && !param->quiet)
2405                                 llapi_err(LLAPI_MSG_ERROR,
2406                                           "%s has no stripe info", path);
2407                         rc = 0;
2408                 } else if (errno == ENOENT) {
2409                         llapi_err(LLAPI_MSG_ERROR,
2410                                   "warning: %s: %s does not exist",
2411                                   __func__, path);
2412                         rc = 0;
2413                 } else if (errno != EISDIR) {
2414                         rc = errno;
2415                         llapi_err(LLAPI_MSG_ERROR, "%s ioctl failed for %s.",
2416                                   d ? "LL_IOC_MDC_GETINFO" :
2417                                   "IOC_MDC_GETFILEINFO", path);
2418                 }
2419                 return rc;
2420         }
2421
2422         st = &param->lmd->lmd_st;
2423
2424         /* libc chown() will do extra check, and if the real owner is
2425          * the same as the ones to set, it won't fall into kernel, so
2426          * invoke syscall directly. */
2427         rc = syscall(SYS_chown, path, -1, -1);
2428         if (rc)
2429                 llapi_err(LLAPI_MSG_ERROR,"error: chown %s (%u,%u)", path);
2430
2431         rc = chmod(path, st->st_mode);
2432         if (rc)
2433                 llapi_err(LLAPI_MSG_ERROR, "error: chmod %s (%hu)",
2434                           path, st->st_mode);
2435
2436         return rc;
2437 }
2438
2439 int llapi_quotachown(char *path, int flag)
2440 {
2441         struct find_param param;
2442
2443         memset(&param, 0, sizeof(param));
2444         param.recursive = 1;
2445         param.verbose = 0;
2446         param.quiet = 1;
2447
2448         return param_callback(path, cb_quotachown, NULL, &param);
2449 }
2450
2451 #include <pwd.h>
2452 #include <grp.h>
2453 #include <mntent.h>
2454 #include <sys/wait.h>
2455 #include <errno.h>
2456 #include <ctype.h>
2457
2458 static int rmtacl_notify(int ops)
2459 {
2460         FILE *fp;
2461         struct mntent *mnt;
2462         int found = 0, fd, rc;
2463
2464         fp = setmntent(MOUNTED, "r");
2465         if (fp == NULL) {
2466                 perror("setmntent");
2467                 return -1;
2468         }
2469
2470         while (1) {
2471                 mnt = getmntent(fp);
2472                 if (!mnt)
2473                         break;
2474
2475                 if (!llapi_is_lustre_mnt(mnt))
2476                         continue;
2477
2478                 fd = open(mnt->mnt_dir, O_RDONLY | O_DIRECTORY);
2479                 if (fd < 0) {
2480                         perror("open");
2481                         return -1;
2482                 }
2483
2484                 rc = ioctl(fd, LL_IOC_RMTACL, ops);
2485                 if (rc < 0) {
2486                         perror("ioctl");
2487                 return -1;
2488                 }
2489
2490                 found++;
2491         }
2492         endmntent(fp);
2493         return found;
2494 }
2495
2496 static char *next_token(char *p, int div)
2497 {
2498         if (p == NULL)
2499                 return NULL;
2500
2501         if (div)
2502                 while (*p && *p != ':' && !isspace(*p))
2503                         p++;
2504         else
2505                 while (*p == ':' || isspace(*p))
2506                         p++;
2507
2508         return *p ? p : NULL;
2509 }
2510
2511 static int rmtacl_name2id(char *name, int is_user)
2512 {
2513         if (is_user) {
2514                 struct passwd *pw;
2515
2516                 if ((pw = getpwnam(name)) == NULL)
2517                         return INVALID_ID;
2518                 else
2519                         return (int)(pw->pw_uid);
2520         } else {
2521                 struct group *gr;
2522
2523                 if ((gr = getgrnam(name)) == NULL)
2524                         return INVALID_ID;
2525                 else
2526                         return (int)(gr->gr_gid);
2527         }
2528 }
2529
2530 static int isodigit(int c)
2531 {
2532         return (c >= '0' && c <= '7') ? 1 : 0;
2533 }
2534
2535 /*
2536  * Whether the name is just digits string (uid/gid) already or not.
2537  * Return value:
2538  * 1: str is id
2539  * 0: str is not id
2540  */
2541 static int str_is_id(char *str)
2542 {
2543         if (str == NULL)
2544                 return 0;
2545
2546         if (*str == '0') {
2547                 str++;
2548                 if (*str == 'x' || *str == 'X') { /* for Hex. */
2549                         if (!isxdigit(*(++str)))
2550                                 return 0;
2551
2552                         while (isxdigit(*(++str)));
2553                 } else if (isodigit(*str)) { /* for Oct. */
2554                         while (isodigit(*(++str)));
2555                 }
2556         } else if (isdigit(*str)) { /* for Dec. */
2557                 while (isdigit(*(++str)));
2558         }
2559
2560         return (*str == 0) ? 1 : 0;
2561 }
2562
2563 typedef struct {
2564         char *name;
2565         int   length;
2566         int   is_user;
2567         int   next_token;
2568 } rmtacl_name_t;
2569
2570 #define RMTACL_OPTNAME(name) name, sizeof(name) - 1
2571
2572 static rmtacl_name_t rmtacl_namelist[] = {
2573         { RMTACL_OPTNAME("user:"),            1,      0 },
2574         { RMTACL_OPTNAME("group:"),           0,      0 },
2575         { RMTACL_OPTNAME("default:user:"),    1,      0 },
2576         { RMTACL_OPTNAME("default:group:"),   0,      0 },
2577         /* for --tabular option */
2578         { RMTACL_OPTNAME("user"),             1,      1 },
2579         { RMTACL_OPTNAME("group"),            0,      1 },
2580         { 0 }
2581 };
2582
2583 static int rgetfacl_output(char *str)
2584 {
2585         char *start = NULL, *end = NULL;
2586         int is_user = 0, n, id;
2587         char c;
2588         rmtacl_name_t *rn;
2589
2590         if (str == NULL)
2591                 return -1;
2592
2593         for (rn = rmtacl_namelist; rn->name; rn++) {
2594                 if(strncmp(str, rn->name, rn->length) == 0) {
2595                         if (!rn->next_token)
2596                                 start = str + rn->length;
2597                         else
2598                                 start = next_token(str + rn->length, 0);
2599                         is_user = rn->is_user;
2600                         break;
2601                 }
2602         }
2603
2604         end = next_token(start, 1);
2605         if (end == NULL || start == end) {
2606                 n = printf("%s", str);
2607                 return n;
2608         }
2609
2610         c = *end;
2611         *end = 0;
2612         id = rmtacl_name2id(start, is_user);
2613         if (id == INVALID_ID) {
2614                 if (str_is_id(start)) {
2615                         *end = c;
2616                         n = printf("%s", str);
2617                 } else
2618                         return -1;
2619         } else if ((id == NOBODY_UID && is_user) ||
2620                    (id == NOBODY_GID && !is_user)) {
2621                 *end = c;
2622                 n = printf("%s", str);
2623         } else {
2624                 *end = c;
2625                 *start = 0;
2626                 n = printf("%s%d%s", str, id, end);
2627         }
2628         return n;
2629 }
2630
2631 static int child_status(int status)
2632 {
2633         return WIFEXITED(status) ? WEXITSTATUS(status) : -1;
2634 }
2635
2636 static int do_rmtacl(int argc, char *argv[], int ops, int (output_func)(char *))
2637 {
2638         pid_t pid = 0;
2639         int fd[2], status;
2640         FILE *fp;
2641         char buf[PIPE_BUF];
2642
2643         if (output_func) {
2644                 if (pipe(fd) < 0) {
2645                         perror("pipe");
2646                         return -1;
2647                 }
2648
2649                 if ((pid = fork()) < 0) {
2650                         perror("fork");
2651                         close(fd[0]);
2652                         close(fd[1]);
2653                         return -1;
2654                 } else if (!pid) {
2655                         /* child process redirects its output. */
2656                         close(fd[0]);
2657                         close(1);
2658                         if (dup2(fd[1], 1) < 0) {
2659                                 perror("dup2");
2660                                 close(fd[1]);
2661                                 return -1;
2662                         }
2663                 } else {
2664                         close(fd[1]);
2665                 }
2666         }
2667
2668         if (!pid) {
2669                 status = rmtacl_notify(ops);
2670                 if (status < 0)
2671                         return -1;
2672
2673                 exit(execvp(argv[0], argv));
2674         }
2675
2676         /* the following is parent process */
2677         if ((fp = fdopen(fd[0], "r")) == NULL) {
2678                 perror("fdopen");
2679                 kill(pid, SIGKILL);
2680                 close(fd[0]);
2681                 return -1;
2682         }
2683
2684         while (fgets(buf, PIPE_BUF, fp) != NULL) {
2685                 if (output_func(buf) < 0)
2686                         fprintf(stderr, "WARNING: unexpected error!\n[%s]\n",
2687                                 buf);
2688         }
2689         fclose(fp);
2690         close(fd[0]);
2691
2692         if (waitpid(pid, &status, 0) < 0) {
2693                 perror("waitpid");
2694                 return -1;
2695         }
2696
2697         return child_status(status);
2698 }
2699
2700 int llapi_lsetfacl(int argc, char *argv[])
2701 {
2702         return do_rmtacl(argc, argv, RMT_LSETFACL, NULL);
2703 }
2704
2705 int llapi_lgetfacl(int argc, char *argv[])
2706 {
2707         return do_rmtacl(argc, argv, RMT_LGETFACL, NULL);
2708 }
2709
2710 int llapi_rsetfacl(int argc, char *argv[])
2711 {
2712         return do_rmtacl(argc, argv, RMT_RSETFACL, NULL);
2713 }
2714
2715 int llapi_rgetfacl(int argc, char *argv[])
2716 {
2717         return do_rmtacl(argc, argv, RMT_RGETFACL, rgetfacl_output);
2718 }
2719
2720 int llapi_cp(int argc, char *argv[])
2721 {
2722         int rc;
2723
2724         rc = rmtacl_notify(RMT_RSETFACL);
2725         if (rc < 0)
2726                 return -1;
2727
2728         exit(execvp(argv[0], argv));
2729 }
2730
2731 int llapi_ls(int argc, char *argv[])
2732 {
2733         int rc;
2734
2735         rc = rmtacl_notify(RMT_LGETFACL);
2736         if (rc < 0)
2737                 return -1;
2738
2739         exit(execvp(argv[0], argv));
2740 }
2741
2742 /* Print mdtname 'name' into 'buf' using 'format'.  Add -MDT0000 if needed.
2743  * format must have %s%s, buf must be > 16
2744  * Eg: if name = "lustre-MDT0000", "lustre", or "lustre-MDT0000_UUID"
2745  *     then buf = "lustre-MDT0000"
2746  */
2747 static int get_mdtname(char *name, char *format, char *buf)
2748 {
2749         char suffix[]="-MDT0000";
2750         int len = strlen(name);
2751
2752         if ((len > 5) && (strncmp(name + len - 5, "_UUID", 5) == 0)) {
2753                 name[len - 5] = '\0';
2754                 len -= 5;
2755         }
2756
2757         if (len > 8) {
2758                 if ((len <= 16) && strncmp(name + len - 8, "-MDT", 4) == 0) {
2759                         suffix[0] = '\0';
2760                 } else {
2761                         /* Not enough room to add suffix */
2762                         llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO,
2763                                   "MDT name too long |%s|", name);
2764                         return -EINVAL;
2765                 }
2766         }
2767
2768         return sprintf(buf, format, name, suffix);
2769 }
2770
2771 /** ioctl on filsystem root, with mdtindex sent as data
2772  * \param mdtname path, fsname, or mdtname (lutre-MDT0004)
2773  * \param mdtidxp pointer to integer within data to be filled in with the
2774  *    mdt index (0 if no mdt is specified).  NULL won't be filled.
2775  */
2776 static int root_ioctl(const char *mdtname, int opc, void *data, int *mdtidxp,
2777                       int want_error)
2778 {
2779         char fsname[20];
2780         char *ptr;
2781         int fd, index, rc;
2782
2783         /* Take path, fsname, or MDTname.  Assume MDT0000 in the former cases.
2784          Open root and parse mdt index. */
2785         if (mdtname[0] == '/') {
2786                 index = 0;
2787                 rc = get_root_path(WANT_FD | want_error, NULL, &fd,
2788                                    (char *)mdtname, -1);
2789         } else {
2790                 if (get_mdtname((char *)mdtname, "%s%s", fsname) < 0)
2791                         return -EINVAL;
2792                 ptr = fsname + strlen(fsname) - 8;
2793                 *ptr = '\0';
2794                 index = strtol(ptr + 4, NULL, 10);
2795                 rc = get_root_path(WANT_FD | want_error, fsname, &fd, NULL, -1);
2796         }
2797         if (rc < 0) {
2798                 if (want_error)
2799                         llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO,
2800                                   "Can't open %s: %d\n", mdtname, rc);
2801                 return rc;
2802         }
2803
2804         if (mdtidxp)
2805                 *mdtidxp = index;
2806
2807         rc = ioctl(fd, opc, data);
2808         if (rc && want_error)
2809                 llapi_err(LLAPI_MSG_ERROR, "ioctl %d err %d", opc, rc);
2810
2811         close(fd);
2812         return rc;
2813 }
2814
2815 /****** Changelog API ********/
2816
2817 static int changelog_ioctl(const char *mdtname, int opc, int id,
2818                            long long recno, int flags)
2819 {
2820         struct ioc_changelog data;
2821         int *idx;
2822
2823         data.icc_id = id;
2824         data.icc_recno = recno;
2825         data.icc_flags = flags;
2826         idx = (int *)(&data.icc_mdtindex);
2827
2828         return root_ioctl(mdtname, opc, &data, idx, WANT_ERROR);
2829 }
2830
2831 #define CHANGELOG_PRIV_MAGIC 0xCA8E1080
2832 struct changelog_private {
2833         int magic;
2834         int flags;
2835         lustre_kernelcomm kuc;
2836 };
2837
2838 /** Start reading from a changelog
2839  * @param priv Opaque private control structure
2840  * @param flags Start flags (e.g. CHANGELOG_FLAG_BLOCK)
2841  * @param device Report changes recorded on this MDT
2842  * @param startrec Report changes beginning with this record number
2843  * (just call llapi_changelog_fini when done; don't need an endrec)
2844  */
2845 int llapi_changelog_start(void **priv, int flags, const char *device,
2846                           long long startrec)
2847 {
2848         struct changelog_private *cp;
2849         int rc;
2850
2851         /* Set up the receiver control struct */
2852         cp = calloc(1, sizeof(*cp));
2853         if (cp == NULL)
2854                 return -ENOMEM;
2855
2856         cp->magic = CHANGELOG_PRIV_MAGIC;
2857         cp->flags = flags;
2858
2859         /* Set up the receiver */
2860         rc = libcfs_ukuc_start(&cp->kuc, 0 /* no group registration */);
2861         if (rc < 0)
2862                 goto out_free;
2863
2864         *priv = cp;
2865
2866         /* Tell the kernel to start sending */
2867         rc = changelog_ioctl(device, OBD_IOC_CHANGELOG_SEND, cp->kuc.lk_wfd,
2868                              startrec, flags);
2869         /* Only the kernel reference keeps the write side open */
2870         close(cp->kuc.lk_wfd);
2871         cp->kuc.lk_wfd = 0;
2872         if (rc < 0) {
2873                 /* frees and clears priv */
2874                 llapi_changelog_fini(priv);
2875                 return rc;
2876         }
2877
2878         return 0;
2879
2880 out_free:
2881         free(cp);
2882         return rc;
2883 }
2884
2885 /** Finish reading from a changelog */
2886 int llapi_changelog_fini(void **priv)
2887 {
2888         struct changelog_private *cp = (struct changelog_private *)*priv;
2889
2890         if (!cp || (cp->magic != CHANGELOG_PRIV_MAGIC))
2891                 return -EINVAL;
2892
2893         libcfs_ukuc_stop(&cp->kuc);
2894         free(cp);
2895         *priv = NULL;
2896         return 0;
2897 }
2898
2899 /** Read the next changelog entry
2900  * @param priv Opaque private control structure
2901  * @param rech Changelog record handle; record will be allocated here
2902  * @return 0 valid message received; rec is set
2903  *         <0 error code
2904  *         1 EOF
2905  */
2906 int llapi_changelog_recv(void *priv, struct changelog_rec **rech)
2907 {
2908         struct changelog_private *cp = (struct changelog_private *)priv;
2909         struct kuc_hdr *kuch;
2910         int rc = 0;
2911
2912         if (!cp || (cp->magic != CHANGELOG_PRIV_MAGIC))
2913                 return -EINVAL;
2914         if (rech == NULL)
2915                 return -EINVAL;
2916         kuch = malloc(CR_MAXSIZE + sizeof(*kuch));
2917         if (kuch == NULL)
2918                 return -ENOMEM;
2919
2920 repeat:
2921         rc = libcfs_ukuc_msg_get(&cp->kuc, (char *)kuch,
2922                                  CR_MAXSIZE + sizeof(*kuch),
2923                                  KUC_TRANSPORT_CHANGELOG);
2924         if (rc < 0)
2925                 goto out_free;
2926
2927         if ((kuch->kuc_transport != KUC_TRANSPORT_CHANGELOG) ||
2928             ((kuch->kuc_msgtype != CL_RECORD) &&
2929              (kuch->kuc_msgtype != CL_EOF))) {
2930                 llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO,
2931                           "Unknown changelog message type %d:%d\n",
2932                           kuch->kuc_transport, kuch->kuc_msgtype);
2933                 rc = -EPROTO;
2934                 goto out_free;
2935         }
2936
2937         if (kuch->kuc_msgtype == CL_EOF) {
2938                 if (cp->flags & CHANGELOG_FLAG_FOLLOW) {
2939                         /* Ignore EOFs */
2940                         goto repeat;
2941                 } else {
2942                         rc = 1;
2943                         goto out_free;
2944                 }
2945         }
2946
2947         /* Our message is a changelog_rec.  Use pointer math to skip
2948          * kuch_hdr and point directly to the message payload.
2949          */
2950         *rech = (struct changelog_rec *)(kuch + 1);
2951
2952         return 0;
2953
2954 out_free:
2955         *rech = NULL;
2956         free(kuch);
2957         return rc;
2958 }
2959
2960 /** Release the changelog record when done with it. */
2961 int llapi_changelog_free(struct changelog_rec **rech)
2962 {
2963         if (*rech) {
2964                 /* We allocated memory starting at the kuc_hdr, but passed
2965                  * the consumer a pointer to the payload.
2966                  * Use pointer math to get back to the header.
2967                  */
2968                 struct kuc_hdr *kuch = (struct kuc_hdr *)*rech - 1;
2969                 free(kuch);
2970         }
2971         *rech = NULL;
2972         return 0;
2973 }
2974
2975 int llapi_changelog_clear(const char *mdtname, const char *idstr,
2976                           long long endrec)
2977 {
2978         int id;
2979
2980         if (endrec < 0) {
2981                 llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO,
2982                           "can't purge negative records\n");
2983                 return -EINVAL;
2984         }
2985
2986         id = strtol(idstr + strlen(CHANGELOG_USER_PREFIX), NULL, 10);
2987         if ((id == 0) || (strncmp(idstr, CHANGELOG_USER_PREFIX,
2988                                   strlen(CHANGELOG_USER_PREFIX)) != 0)) {
2989                 llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO,
2990                           "expecting id of the form '"CHANGELOG_USER_PREFIX
2991                           "<num>'; got '%s'\n", idstr);
2992                 return -EINVAL;
2993         }
2994
2995         return changelog_ioctl(mdtname, OBD_IOC_CHANGELOG_CLEAR, id, endrec, 0);
2996 }
2997
2998 int llapi_fid2path(const char *device, const char *fidstr, char *buf,
2999                    int buflen, long long *recno, int *linkno)
3000 {
3001         struct lu_fid fid;
3002         struct getinfo_fid2path *gf;
3003         int rc;
3004
3005         while (*fidstr == '[')
3006                 fidstr++;
3007
3008         sscanf(fidstr, SFID, RFID(&fid));
3009         if (!fid_is_sane(&fid)) {
3010                 llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO,
3011                           "bad FID format [%s], should be "DFID"\n",
3012                           fidstr, (__u64)1, 2, 0);
3013                 return -EINVAL;
3014         }
3015
3016         gf = malloc(sizeof(*gf) + buflen);
3017         if (gf == NULL)
3018                 return -ENOMEM;
3019         gf->gf_fid = fid;
3020         gf->gf_recno = *recno;
3021         gf->gf_linkno = *linkno;
3022         gf->gf_pathlen = buflen;
3023
3024         /* Take path or fsname */
3025         rc = root_ioctl(device, OBD_IOC_FID2PATH, gf, NULL, 0);
3026         if (rc) {
3027                 llapi_err(LLAPI_MSG_ERROR, "ioctl err %d", rc);
3028         } else {
3029                 memcpy(buf, gf->gf_path, gf->gf_pathlen);
3030                 *recno = gf->gf_recno;
3031                 *linkno = gf->gf_linkno;
3032         }
3033
3034         free(gf);
3035         return rc;
3036 }
3037
3038 static int path2fid_from_lma(const char *path, lustre_fid *fid)
3039 {
3040         char buf[512];
3041         struct lustre_mdt_attrs *lma;
3042         int rc;
3043
3044         rc = lgetxattr(path, XATTR_NAME_LMA, buf, sizeof(buf));
3045         if (rc < 0)
3046                 return -errno;
3047         lma = (struct lustre_mdt_attrs *)buf;
3048         fid_le_to_cpu(fid, &lma->lma_self_fid);
3049         return 0;
3050 }
3051
3052 int llapi_path2fid(const char *path, lustre_fid *fid)
3053 {
3054         int fd, rc;
3055
3056         memset(fid, 0, sizeof(*fid));
3057         fd = open(path, O_RDONLY | O_NONBLOCK | O_NOFOLLOW);
3058         if (fd < 0) {
3059                 if (errno == ELOOP) /* symbolic link */
3060                         return path2fid_from_lma(path, fid);
3061                 return -errno;
3062         }
3063
3064         rc = ioctl(fd, LL_IOC_PATH2FID, fid) < 0 ? -errno : 0;
3065         if (rc == -EINVAL) /* char special device */
3066                 rc = path2fid_from_lma(path, fid);
3067
3068         close(fd);
3069         return rc;
3070 }
3071
3072 /****** HSM Copytool API ********/
3073 #define CT_PRIV_MAGIC 0xC0BE2001
3074 struct copytool_private {
3075         int magic;
3076         char *fsname;
3077         lustre_kernelcomm kuc;
3078         __u32 archives;
3079 };
3080
3081 #include <libcfs/libcfs.h>
3082
3083 /** Register a copytool
3084  * @param[out] priv Opaque private control structure
3085  * @param fsname Lustre filesystem
3086  * @param flags Open flags, currently unused (e.g. O_NONBLOCK)
3087  * @param archive_count
3088  * @param archives Which archive numbers this copytool is responsible for
3089  */
3090 int llapi_copytool_start(void **priv, char *fsname, int flags,
3091                          int archive_count, int *archives)
3092 {
3093         struct copytool_private *ct;
3094         int rc;
3095
3096         if (archive_count > 0 && archives == NULL) {
3097                 llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO,
3098                           "NULL archive numbers");
3099                 return -EINVAL;
3100         }
3101
3102         ct = calloc(1, sizeof(*ct));
3103         if (ct == NULL)
3104                 return -ENOMEM;
3105
3106         ct->fsname = malloc(strlen(fsname) + 1);
3107         if (ct->fsname == NULL) {
3108                 rc = -ENOMEM;
3109                 goto out_err;
3110         }
3111         strcpy(ct->fsname, fsname);
3112         ct->magic = CT_PRIV_MAGIC;
3113         ct->archives = 0;
3114         for (rc = 0; rc < archive_count; rc++) {
3115                 if (archives[rc] > sizeof(ct->archives)) {
3116                         llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO,
3117                                   "Maximum of %d archives supported",
3118                                   sizeof(ct->archives));
3119                         goto out_err;
3120                 }
3121                 ct->archives |= 1 << archives[rc];
3122         }
3123         /* special case: if no archives specified, default to archive #0. */
3124         if (ct->archives == 0)
3125                 ct->archives = 1;
3126
3127         rc = libcfs_ukuc_start(&ct->kuc, KUC_GRP_HSM);
3128         if (rc < 0)
3129                 goto out_err;
3130
3131         /* Storing archive(s) in lk_data; see mdc_ioc_hsm_ct_start */
3132         ct->kuc.lk_data = ct->archives;
3133         rc = root_ioctl(ct->fsname, LL_IOC_HSM_CT_START, &(ct->kuc), NULL,
3134                         WANT_ERROR);
3135         /* Only the kernel reference keeps the write side open */
3136         close(ct->kuc.lk_wfd);
3137         ct->kuc.lk_wfd = 0;
3138         if (rc < 0)
3139                 goto out_err;
3140
3141         *priv = ct;
3142         return 0;
3143
3144 out_err:
3145         if (ct->fsname)
3146                 free(ct->fsname);
3147         free(ct);
3148         return rc;
3149 }
3150
3151 /** Deregister a copytool */
3152 int llapi_copytool_fini(void **priv)
3153 {
3154         struct copytool_private *ct = (struct copytool_private *)*priv;
3155
3156         if (!ct || (ct->magic != CT_PRIV_MAGIC))
3157                 return -EINVAL;
3158
3159         /* Tell the kernel to stop sending us messages */
3160         ct->kuc.lk_flags = LK_FLG_STOP;
3161         root_ioctl(ct->fsname, LL_IOC_HSM_CT_START, &(ct->kuc), NULL, 0);
3162
3163         /* Shut down the kernelcomms */
3164         libcfs_ukuc_stop(&ct->kuc);
3165
3166         free(ct->fsname);
3167         free(ct);
3168         *priv = NULL;
3169         return 0;
3170 }
3171
3172 /** Wait for the next hsm_action_list
3173  * @param priv Opaque private control structure
3174  * @param halh Action list handle, will be allocated here
3175  * @param msgsize Number of bytes in the message, will be set here
3176  * @return 0 valid message received; halh and msgsize are set
3177  *         <0 error code
3178  */
3179 int llapi_copytool_recv(void *priv, struct hsm_action_list **halh, int *msgsize)
3180 {
3181         struct copytool_private *ct = (struct copytool_private *)priv;
3182         struct kuc_hdr *kuch;
3183         struct hsm_action_list *hal;
3184         int rc = 0;
3185
3186         if (!ct || (ct->magic != CT_PRIV_MAGIC))
3187                 return -EINVAL;
3188         if (halh == NULL || msgsize == NULL)
3189                 return -EINVAL;
3190
3191         kuch = malloc(HAL_MAXSIZE + sizeof(*kuch));
3192         if (kuch == NULL)
3193                 return -ENOMEM;
3194
3195         rc = libcfs_ukuc_msg_get(&ct->kuc, (char *)kuch,
3196                                  HAL_MAXSIZE + sizeof(*kuch),
3197                                  KUC_TRANSPORT_HSM);
3198         if (rc < 0)
3199                 goto out_free;
3200
3201         /* Handle generic messages */
3202         if (kuch->kuc_transport == KUC_TRANSPORT_GENERIC &&
3203             kuch->kuc_msgtype == KUC_MSG_SHUTDOWN) {
3204                 rc = -ESHUTDOWN;
3205                 goto out_free;
3206         }
3207
3208         if (kuch->kuc_transport != KUC_TRANSPORT_HSM ||
3209             kuch->kuc_msgtype != HMT_ACTION_LIST) {
3210                 llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO,
3211                           "Unknown HSM message type %d:%d\n",
3212                           kuch->kuc_transport, kuch->kuc_msgtype);
3213                 rc = -EPROTO;
3214                 goto out_free;
3215         }
3216
3217         /* Our message is a hsm_action_list.  Use pointer math to skip
3218          * kuch_hdr and point directly to the message payload.
3219          */
3220         hal = (struct hsm_action_list *)(kuch + 1);
3221
3222         /* Check that we have registered for this archive # */
3223         if (((1 << hal->hal_archive_num) & ct->archives) == 0) {
3224                     llapi_err(LLAPI_MSG_INFO | LLAPI_MSG_NO_ERRNO,
3225                           "Ignoring request for archive #%d (bitmask %#x)\n",
3226                           hal->hal_archive_num, ct->archives);
3227                 rc = 0;
3228                 goto out_free;
3229         }
3230
3231         *halh = hal;
3232         *msgsize = kuch->kuc_msglen - sizeof(*kuch);
3233         return 0;
3234
3235 out_free:
3236         *halh = NULL;
3237         *msgsize = 0;
3238         free(kuch);
3239         return rc;
3240 }
3241
3242 /** Release the action list when done with it. */
3243 int llapi_copytool_free(struct hsm_action_list **hal)
3244 {
3245         /* Reuse the llapi_changelog_free function */
3246         return llapi_changelog_free((struct changelog_rec **)hal);
3247 }
3248
3249 int llapi_get_connect_flags(const char *mnt, __u64 *flags)
3250 {
3251         DIR *root;
3252         int rc;
3253
3254         root = opendir(mnt);
3255         if (!root) {
3256                 llapi_err(LLAPI_MSG_ERROR, "open %s failed", mnt);
3257                 return -1;
3258         }
3259
3260         rc = ioctl(dirfd(root), LL_IOC_GET_CONNECT_FLAGS, flags);
3261         closedir(root);
3262         if (rc < 0)
3263                 llapi_err(LLAPI_MSG_ERROR,
3264                           "ioctl on %s for getting connect flags failed", mnt);
3265         return rc;
3266 }
3267
3268 int llapi_get_version(char *buffer, int buffer_size,
3269                       char **version)
3270 {
3271         int rc;
3272         int fd;
3273         struct obd_ioctl_data *data = (struct obd_ioctl_data *)buffer;
3274
3275         fd = open(OBD_DEV_PATH, O_RDONLY);
3276         if (fd == -1)
3277                 return -errno;
3278
3279         memset(buffer, 0, buffer_size);
3280         data->ioc_version = OBD_IOCTL_VERSION;
3281         data->ioc_inllen1 = buffer_size - cfs_size_round(sizeof(*data));
3282         data->ioc_inlbuf1 = buffer + cfs_size_round(sizeof(*data));
3283         data->ioc_len = obd_ioctl_packlen(data);
3284
3285         rc = ioctl(fd, OBD_GET_VERSION, buffer);
3286         if (rc == -1) {
3287                 rc = errno;
3288                 close(fd);
3289                 return -rc;
3290         }
3291         close(fd);
3292         *version = data->ioc_bulk;
3293         return 0;
3294 }