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