Whamcloud - gitweb
b=14836
[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  2008 Sun Microsystems, Inc. All rights reserved
30  * Use is subject to license terms.
31  */
32 /*
33  * This file is part of Lustre, http://www.lustre.org/
34  * Lustre is a trademark of Sun Microsystems, Inc.
35  *
36  * lustre/utils/liblustreapi.c
37  *
38  * Author: Peter J. Braam <braam@clusterfs.com>
39  * Author: Phil Schwan <phil@clusterfs.com>
40  * Author: Robert Read <rread@clusterfs.com>
41  */
42
43 /* for O_DIRECTORY */
44 #ifndef _GNU_SOURCE
45 #define _GNU_SOURCE
46 #endif
47
48 #include <stdlib.h>
49 #include <stdio.h>
50 #include <string.h>
51 #include <stddef.h>
52 #include <sys/ioctl.h>
53 #include <unistd.h>
54 #include <fcntl.h>
55 #include <errno.h>
56 #include <dirent.h>
57 #include <stdarg.h>
58 #include <sys/stat.h>
59 #include <sys/types.h>
60 #include <sys/syscall.h>
61 #include <fnmatch.h>
62 #include <glob.h>
63 #ifdef HAVE_LINUX_UNISTD_H
64 #include <linux/unistd.h>
65 #else
66 #include <unistd.h>
67 #endif
68
69 #include <liblustre.h>
70 #include <lnet/lnetctl.h>
71 #include <obd.h>
72 #include <lustre_lib.h>
73 #include <obd_lov.h>
74 #include <lustre/liblustreapi.h>
75
76 static unsigned llapi_dir_filetype_table[] = {
77         [DT_UNKNOWN]= 0,
78         [DT_FIFO]= S_IFIFO,
79         [DT_CHR] = S_IFCHR,
80         [DT_DIR] = S_IFDIR,
81         [DT_BLK] = S_IFBLK,
82         [DT_REG] = S_IFREG,
83         [DT_LNK] = S_IFLNK,
84         [DT_SOCK]= S_IFSOCK,
85 #if defined(DT_DOOR) && defined(S_IFDOOR)
86         [DT_DOOR]= S_IFDOOR,
87 #endif
88 };
89
90 #if defined(DT_DOOR) && defined(S_IFDOOR)
91 static const int DT_MAX = DT_DOOR;
92 #else
93 static const int DT_MAX = DT_SOCK;
94 #endif
95
96 static unsigned llapi_filetype_dir_table[] = {
97         [0]= DT_UNKNOWN,
98         [S_IFIFO]= DT_FIFO,
99         [S_IFCHR] = DT_CHR,
100         [S_IFDIR] = DT_DIR,
101         [S_IFBLK] = DT_BLK,
102         [S_IFREG] = DT_REG,
103         [S_IFLNK] = DT_LNK,
104         [S_IFSOCK]= DT_SOCK,
105 #if defined(DT_DOOR) && defined(S_IFDOOR)
106         [S_IFDOOR]= DT_DOOR,
107 #endif
108 };
109
110 #if defined(DT_DOOR) && defined(S_IFDOOR)
111 static const int S_IFMAX = DT_DOOR;
112 #else
113 static const int S_IFMAX = DT_SOCK;
114 #endif
115
116 /* liblustreapi message level */
117 static int llapi_msg_level = LLAPI_MSG_MAX;
118
119 void llapi_msg_set_level(int level)
120 {
121         /* ensure level is in the good range */
122         if (level < LLAPI_MSG_OFF)
123                 llapi_msg_level = LLAPI_MSG_OFF;
124         else if (level > LLAPI_MSG_MAX)
125                 llapi_msg_level = LLAPI_MSG_MAX;
126         else
127                 llapi_msg_level = level;
128 }
129
130 void llapi_err(int level, char *fmt, ...)
131 {
132         va_list args;
133         int tmp_errno = abs(errno);
134
135         if ((level & LLAPI_MSG_MASK) > llapi_msg_level)
136                 return;
137
138         va_start(args, fmt);
139         vfprintf(stderr, fmt, args);
140         va_end(args);
141
142         if (level & LLAPI_MSG_NO_ERRNO)
143                 fprintf(stderr, "\n");
144         else
145                 fprintf(stderr, ": %s (%d)\n", strerror(tmp_errno), tmp_errno);
146 }
147
148 #define llapi_err_noerrno(level, fmt, a...)                             \
149         llapi_err((level) | LLAPI_MSG_NO_ERRNO, fmt, ## a)
150
151 void llapi_printf(int level, char *fmt, ...)
152 {
153         va_list args;
154
155         if ((level & LLAPI_MSG_MASK) > llapi_msg_level)
156                 return;
157
158         va_start(args, fmt);
159         vfprintf(stdout, fmt, args);
160         va_end(args);
161 }
162
163 int parse_size(char *optarg, unsigned long long *size,
164                unsigned long long *size_units)
165 {
166         char *end;
167
168         *size = strtoul(optarg, &end, 0);
169
170         if (*end != '\0') {
171                 if ((*end == 'b') && *(end+1) == '\0' &&
172                     (*size & (~0ULL << (64 - 9))) == 0) {
173                         *size <<= 9;
174                         *size_units = 1 << 9;
175                 } else if ((*end == 'k' || *end == 'K') &&
176                            *(end+1) == '\0' && (*size &
177                            (~0ULL << (64 - 10))) == 0) {
178                         *size <<= 10;
179                         *size_units = 1 << 10;
180                 } else if ((*end == 'm' || *end == 'M') &&
181                            *(end+1) == '\0' && (*size &
182                            (~0ULL << (64 - 20))) == 0) {
183                         *size <<= 20;
184                         *size_units = 1 << 20;
185                 } else if ((*end == 'g' || *end == 'G') &&
186                            *(end+1) == '\0' && (*size &
187                            (~0ULL << (64 - 30))) == 0) {
188                         *size <<= 30;
189                         *size_units = 1 << 30;
190                 } else if ((*end == 't' || *end == 'T') &&
191                            *(end+1) == '\0' && (*size &
192                            (~0ULL << (64 - 40))) == 0) {
193                         *size <<= 40;
194                         *size_units = 1ULL << 40;
195                 } else if ((*end == 'p' || *end == 'P') &&
196                            *(end+1) == '\0' && (*size &
197                            (~0ULL << (64 - 50))) == 0) {
198                         *size <<= 50;
199                         *size_units = 1ULL << 50;
200                 } else if ((*end == 'e' || *end == 'E') &&
201                            *(end+1) == '\0' && (*size &
202                            (~0ULL << (64 - 60))) == 0) {
203                         *size <<= 60;
204                         *size_units = 1ULL << 60;
205                 } else {
206                         return -1;
207                 }
208         }
209
210         return 0;
211 }
212
213 int llapi_stripe_limit_check(unsigned long stripe_size, int stripe_offset,
214                              int stripe_count, int stripe_pattern)
215 {
216         int page_size;
217
218         /* 64 KB is the largest common page size I'm aware of (on ia64), but
219          * check the local page size just in case. */
220         page_size = LOV_MIN_STRIPE_SIZE;
221         if (getpagesize() > page_size) {
222                 page_size = getpagesize();
223                 llapi_err_noerrno(LLAPI_MSG_WARN,
224                                   "warning: your page size (%u) is "
225                                   "larger than expected (%u)", page_size,
226                                   LOV_MIN_STRIPE_SIZE);
227         }
228         if (stripe_size < 0 || (stripe_size & (LOV_MIN_STRIPE_SIZE - 1))) {
229                 llapi_err(LLAPI_MSG_ERROR, "error: bad stripe_size %lu, "
230                           "must be an even multiple of %d bytes",
231                           stripe_size, page_size);
232                 return -EINVAL;
233         }
234         if (stripe_offset < -1 || stripe_offset > MAX_OBD_DEVICES) {
235                 llapi_err(LLAPI_MSG_ERROR, "error: bad stripe offset %d",
236                           stripe_offset);
237                 return -EINVAL;
238         }
239         if (stripe_count < -1 || stripe_count > LOV_MAX_STRIPE_COUNT) {
240                 llapi_err(LLAPI_MSG_ERROR, "error: bad stripe count %d",
241                           stripe_count);
242                 return -EINVAL;
243         }
244         if (stripe_count > 0 && (__u64)stripe_size * stripe_count > 0xffffffff){
245                 llapi_err(LLAPI_MSG_ERROR, "error: stripe_size %lu * "
246                           "stripe_count %u exceeds 4GB", stripe_size,
247                           stripe_count);
248                 return -EINVAL;
249         }
250         return 0;
251 }
252
253 int llapi_file_open(const char *name, int flags, int mode,
254                     unsigned long stripe_size, int stripe_offset,
255                     int stripe_count, int stripe_pattern)
256 {
257         struct lov_user_md lum = { 0 };
258         int fd, rc = 0;
259         int isdir = 0;
260
261         fd = open(name, flags | O_LOV_DELAY_CREATE, mode);
262         if (fd < 0 && errno == EISDIR) {
263                 fd = open(name, O_DIRECTORY | O_RDONLY);
264                 isdir++;
265         }
266
267         if (fd < 0) {
268                 rc = -errno;
269                 llapi_err(LLAPI_MSG_ERROR, "unable to open '%s'", name);
270                 return rc;
271         }
272
273         if ((rc = llapi_stripe_limit_check(stripe_size, stripe_offset,
274                                            stripe_count, stripe_pattern)) != 0) {
275                 errno = rc;
276                 goto out;
277         }
278
279         /*  Initialize IOCTL striping pattern structure */
280         lum.lmm_magic = LOV_USER_MAGIC;
281         lum.lmm_pattern = stripe_pattern;
282         lum.lmm_stripe_size = stripe_size;
283         lum.lmm_stripe_count = stripe_count;
284         lum.lmm_stripe_offset = stripe_offset;
285
286         if (ioctl(fd, LL_IOC_LOV_SETSTRIPE, &lum)) {
287                 char *errmsg = "stripe already set";
288                 rc = -errno;
289                 if (errno != EEXIST && errno != EALREADY)
290                         errmsg = strerror(errno);
291
292                 llapi_err_noerrno(LLAPI_MSG_ERROR,
293                                   "error on ioctl "LPX64" for '%s' (%d): %s",
294                                   (__u64)LL_IOC_LOV_SETSTRIPE, name, fd, errmsg);
295         }
296 out:
297         if (rc) {
298                 close(fd);
299                 fd = rc;
300         }
301
302         return fd;
303 }
304
305 static int poolpath(char *fsname, char *pathname, char *pool_pathname);
306
307 int llapi_file_open_pool(const char *name, int flags, int mode,
308                          unsigned long stripe_size, int stripe_offset,
309                          int stripe_count, int stripe_pattern, char *pool_name)
310 {
311         struct lov_user_md_v3 lum = { 0 };
312         int fd, rc = 0;
313         int isdir = 0;
314         char fsname[MAX_OBD_NAME + 1], *ptr;
315
316         fd = open(name, flags | O_LOV_DELAY_CREATE, mode);
317         if (fd < 0 && errno == EISDIR) {
318                 fd = open(name, O_DIRECTORY | O_RDONLY);
319                 isdir++;
320         }
321
322         if (fd < 0) {
323                 rc = -errno;
324                 llapi_err(LLAPI_MSG_ERROR, "unable to open '%s'", name);
325                 return rc;
326         }
327
328         if ((rc = llapi_stripe_limit_check(stripe_size, stripe_offset,
329                                            stripe_count, stripe_pattern)) != 0) {
330                 errno = rc;
331                 goto out;
332         }
333
334         /* in case user give the full pool name <fsname>.<poolname>, skip
335          * the fsname */
336         ptr = strchr(pool_name, '.');
337         if (ptr != NULL) {
338                 strncpy(fsname, pool_name, ptr - pool_name);
339                 fsname[ptr - pool_name] = '\0';
340                 /* if fsname matches a fs skip it
341                  * if not keep the poolname as is */
342                 if (poolpath(fsname, NULL, NULL) == 0)
343                         pool_name = ptr + 1;
344         }
345
346         /*  Initialize IOCTL striping pattern structure */
347         lum.lmm_magic = LOV_USER_MAGIC_V3;
348         lum.lmm_pattern = stripe_pattern;
349         lum.lmm_stripe_size = stripe_size;
350         lum.lmm_stripe_count = stripe_count;
351         lum.lmm_stripe_offset = stripe_offset;
352         strncpy(lum.lmm_pool_name, pool_name, MAXPOOLNAME);
353
354         if (ioctl(fd, LL_IOC_LOV_SETSTRIPE, &lum)) {
355                 char *errmsg = "stripe already set";
356                 rc = -errno;
357                 if (errno != EEXIST && errno != EALREADY)
358                         errmsg = strerror(errno);
359
360                 llapi_err_noerrno(LLAPI_MSG_ERROR,
361                                   "error on ioctl "LPX64" for '%s' (%d): %s",
362                                   (__u64)LL_IOC_LOV_SETSTRIPE, name, fd, errmsg);
363         }
364 out:
365         if (rc) {
366                 close(fd);
367                 fd = rc;
368         }
369
370         return fd;
371 }
372
373 int llapi_file_create(const char *name, unsigned long stripe_size,
374                       int stripe_offset, int stripe_count, int stripe_pattern)
375 {
376         int fd;
377
378         fd = llapi_file_open(name, O_CREAT | O_WRONLY, 0644, stripe_size,
379                              stripe_offset, stripe_count, stripe_pattern);
380         if (fd < 0)
381                 return fd;
382
383         close(fd);
384         return 0;
385 }
386
387 int llapi_file_create_pool(const char *name, unsigned long stripe_size,
388                            int stripe_offset, int stripe_count,
389                            int stripe_pattern, char *pool_name)
390 {
391         int fd;
392
393         fd = llapi_file_open_pool(name, O_CREAT | O_WRONLY, 0644, stripe_size,
394                                   stripe_offset, stripe_count, stripe_pattern,
395                                   pool_name);
396         if (fd < 0)
397                 return fd;
398
399         close(fd);
400         return 0;
401 }
402
403
404 static int print_pool_members(char *fs, char *pool_dir, char *pool_file)
405 {
406         char path[PATH_MAX + 1];
407         char buf[1024];
408         FILE *fd;
409
410         llapi_printf(LLAPI_MSG_NORMAL, "Pool: %s.%s\n", fs, pool_file);
411         sprintf(path, "%s/%s", pool_dir, pool_file);
412         if ((fd = fopen(path, "r")) == NULL) {
413                 llapi_err(LLAPI_MSG_ERROR, "Cannot open %s\n", path);
414                 return -EINVAL;
415         }
416         while (fgets(buf, sizeof(buf), fd) != NULL)
417                llapi_printf(LLAPI_MSG_NORMAL, buf);
418
419         fclose(fd);
420         return 0;
421 }
422
423 /*
424  * search lustre fsname from pathname
425  *
426  */
427 static int search_fsname(char *pathname, char *fsname)
428 {
429         char *ptr;
430         FILE *fp;
431         struct mntent *mnt = NULL;
432
433         /* get the mount point */
434         fp = setmntent(MOUNTED, "r");
435         if (fp == NULL) {
436                  llapi_err(LLAPI_MSG_ERROR,
437                            "setmntent(%s) failed: %s:", MOUNTED,
438                            strerror (errno));
439                  return -EIO;
440         }
441         mnt = getmntent(fp);
442         while ((feof(fp) == 0) && ferror(fp) == 0) {
443                 if (llapi_is_lustre_mnt(mnt)) {
444                         /* search by pathname */
445                         if (strncmp(mnt->mnt_dir, pathname,
446                                     strlen(mnt->mnt_dir)) == 0) {
447                                 ptr = strchr(mnt->mnt_fsname, '/');
448                                 if (ptr == NULL)
449                                         return -EINVAL;
450                                 ptr++;
451                                 strcpy(fsname, ptr);
452                                 return 0;
453                         }
454                 }
455                 mnt = getmntent(fp);
456         }
457         endmntent(fp);
458         return -ENOENT;
459
460 }
461
462 /*
463  * find the pool directory path under /proc
464  * (can be also used to test if a fsname is known)
465  */
466 static int poolpath(char *fsname, char *pathname, char *pool_pathname)
467 {
468         int rc = 0;
469         glob_t glob_info;
470         char pattern[PATH_MAX + 1];
471         char buffer[PATH_MAX];
472
473         if (fsname == NULL) {
474                 rc = search_fsname(pathname, buffer);
475                 if (rc != 0)
476                         return rc;
477                 fsname = buffer;
478                 strcpy(pathname, fsname);
479         }
480
481         snprintf(pattern, PATH_MAX,
482                  "/proc/fs/lustre/lov/%s-*/pools",
483                  fsname);
484         rc = glob(pattern, GLOB_BRACE, NULL, &glob_info);
485         if (rc)
486                 return -ENOENT;
487
488         if (glob_info.gl_pathc == 0) {
489                 globfree(&glob_info);
490                 return -ENOENT;
491         }
492
493         /* in fsname test mode, pool_pathname is NULL */
494         if (pool_pathname != NULL)
495                 strcpy(pool_pathname, glob_info.gl_pathv[0]);
496
497         return 0;
498 }
499
500 int llapi_poollist(char *name)
501 {
502         char *poolname;
503         char *fsname;
504         char rname[PATH_MAX + 1], pathname[PATH_MAX + 1];
505         char *ptr;
506         int rc = 0;
507
508         /* is name a pathname ? */
509         ptr = strchr(name, '/');
510         if (ptr != NULL) {
511                 /* only absolute pathname is supported */
512                 if (*name != '/')
513                         return -EINVAL;
514                 if (!realpath(name, rname)) {
515                         rc = -errno;
516                         llapi_err(LLAPI_MSG_ERROR,
517                                   "llapi_poollist: invalid path '%s'",
518                                   name);
519                         return rc;
520                 }
521
522                 rc = poolpath(NULL, rname, pathname);
523                 if (rc != 0) {
524                         errno = -rc;
525                         llapi_err(LLAPI_MSG_ERROR,
526                                   "llapi_poollist: '%s' is not"
527                                   " a Lustre filesystem",
528                                   name);
529                         return rc;
530                 }
531                 fsname = rname;
532                 poolname = NULL;
533         } else {
534                 /* name is FSNAME[.POOLNAME] */
535                 fsname = name;
536                 poolname = strchr(name, '.');
537                 if (poolname != NULL) {
538                         *poolname = '\0';
539                         poolname++;
540                 }
541                 rc = poolpath(fsname, NULL, pathname);
542                 if (rc != 0) {
543                         errno = -rc;
544                         llapi_err(LLAPI_MSG_ERROR,
545                                   "llapi_poollist: Lustre filesystem '%s'"
546                                   " not found", name);
547                         return rc;
548                 }
549         }
550         if (rc != 0) {
551                 errno = -rc;
552                 llapi_err(LLAPI_MSG_ERROR,
553                           "llapi_poollist: Lustre filesystem '%s' not found",
554                           name);
555                 return rc;
556         }
557
558         if (poolname != NULL) {
559                 rc = print_pool_members(fsname, pathname, poolname);
560                 poolname--;
561                 *poolname = '.';
562         } else {
563                 DIR *dir;
564                 struct dirent *pool;
565
566                 llapi_printf(LLAPI_MSG_NORMAL, "Pools from %s:\n", fsname);
567                 if ((dir = opendir(pathname)) == NULL) {
568                         return -EINVAL;
569                 }
570                 while ((pool = readdir(dir)) != NULL) {
571                         if (!((pool->d_name[0] == '.') &&
572                               (pool->d_name[1] == '\0')) &&
573                             !((pool->d_name[0] == '.') &&
574                               (pool->d_name[1] == '.') &&
575                               (pool->d_name[2] == '\0')))
576                         llapi_printf(LLAPI_MSG_NORMAL, " %s.%s\n", fsname, pool->d_name);
577                 }
578                 closedir(dir);
579         }
580         return rc;
581 }
582
583 typedef int (semantic_func_t)(char *path, DIR *parent, DIR *d,
584                               void *data, cfs_dirent_t *de);
585
586 #define MAX_LOV_UUID_COUNT      max(LOV_MAX_STRIPE_COUNT, 1000)
587 #define OBD_NOT_FOUND           (-1)
588
589 static int common_param_init(struct find_param *param)
590 {
591         param->lumlen = lov_mds_md_size(MAX_LOV_UUID_COUNT, LOV_MAGIC_V3);
592         if ((param->lmd = malloc(sizeof(lstat_t) + param->lumlen)) == NULL) {
593                 llapi_err(LLAPI_MSG_ERROR,
594                           "error: allocation of %d bytes for ioctl",
595                           sizeof(lstat_t) + param->lumlen);
596                 return -ENOMEM;
597         }
598
599         param->got_uuids = 0;
600         param->obdindexes = NULL;
601         param->obdindex = OBD_NOT_FOUND;
602         return 0;
603 }
604
605 static void find_param_fini(struct find_param *param)
606 {
607         if (param->obdindexes)
608                 free(param->obdindexes);
609
610         if (param->lmd)
611                 free(param->lmd);
612 }
613
614 int llapi_file_fget_lov_uuid(int fd, struct obd_uuid *lov_name)
615 {
616         int rc = ioctl(fd, OBD_IOC_GETNAME, lov_name);
617         if (rc) {
618                 rc = errno;
619                 llapi_err(LLAPI_MSG_ERROR, "error: can't get lov name.");
620         }
621         return rc;
622 }
623
624 int llapi_file_get_lov_uuid(const char *path, struct obd_uuid *lov_uuid)
625 {
626         int fd, rc;
627
628         fd = open(path, O_RDONLY);
629         if (fd < 0) {
630                 rc = errno;
631                 llapi_err(LLAPI_MSG_ERROR, "error opening %s", path);
632                 return rc;
633         }
634
635         rc = llapi_file_fget_lov_uuid(fd, lov_uuid);
636
637         close(fd);
638
639         return rc;
640 }
641
642 /*
643  * If uuidp is NULL, return the number of available obd uuids.
644  * If uuidp is non-NULL, then it will return the uuids of the obds. If
645  * there are more OSTs then allocated to uuidp, then an error is returned with
646  * the ost_count set to number of available obd uuids.
647  */
648 int llapi_lov_get_uuids(int fd, struct obd_uuid *uuidp, int *ost_count)
649 {
650         struct obd_uuid lov_name;
651         char buf[1024];
652         FILE *fp;
653         int rc = 0, index = 0;
654
655         /* Get the lov name */
656         rc = llapi_file_fget_lov_uuid(fd, &lov_name);
657         if (rc)
658                 return rc;
659
660         /* Now get the ost uuids from /proc */
661         snprintf(buf, sizeof(buf), "/proc/fs/lustre/lov/%s/target_obd",
662                  lov_name.uuid);
663         fp = fopen(buf, "r");
664         if (fp == NULL) {
665                 rc = errno;
666                 llapi_err(LLAPI_MSG_ERROR, "error: opening '%s'", buf);
667                 return rc;
668         }
669
670         while (fgets(buf, sizeof(buf), fp) != NULL) {
671                 if (uuidp && (index < *ost_count)) {
672                         if (sscanf(buf, "%d: %s", &index, uuidp[index].uuid) <2)
673                                 break;
674                 }
675                 index++;
676         }
677
678         fclose(fp);
679
680         if (uuidp && (index >= *ost_count))
681                 return -EOVERFLOW;
682
683         *ost_count = index;
684         return rc;
685 }
686
687 /* Here, param->obduuid points to a single obduuid, the index of which is
688  * returned in param->obdindex */
689 static int setup_obd_uuid(DIR *dir, char *dname, struct find_param *param)
690 {
691         struct obd_uuid lov_uuid;
692         char uuid[sizeof(struct obd_uuid)];
693         char buf[1024];
694         FILE *fp;
695         int rc = 0, index;
696
697         /* Get the lov name */
698         rc = llapi_file_fget_lov_uuid(dirfd(dir), &lov_uuid);
699         if (rc) {
700                 if (errno != ENOTTY) {
701                         rc = errno;
702                         llapi_err(LLAPI_MSG_ERROR,
703                                   "error: can't get lov name: %s", dname);
704                 } else {
705                         rc = 0;
706                 }
707                 return rc;
708         }
709
710         param->got_uuids = 1;
711
712         /* Now get the ost uuids from /proc */
713         snprintf(buf, sizeof(buf), "/proc/fs/lustre/lov/%s/target_obd",
714                  lov_uuid.uuid);
715         fp = fopen(buf, "r");
716         if (fp == NULL) {
717                 rc = errno;
718                 llapi_err(LLAPI_MSG_ERROR, "error: opening '%s'", buf);
719                 return rc;
720         }
721
722         if (!param->obduuid && !param->quiet && !param->obds_printed)
723                 llapi_printf(LLAPI_MSG_NORMAL, "OBDS:\n");
724
725         while (fgets(buf, sizeof(buf), fp) != NULL) {
726                 if (sscanf(buf, "%d: %s", &index, uuid) < 2)
727                         break;
728
729                 if (param->obduuid) {
730                         if (strncmp(param->obduuid->uuid, uuid,
731                                     sizeof(uuid)) == 0) {
732                                 param->obdindex = index;
733                                 break;
734                         }
735                 } else if (!param->quiet && !param->obds_printed) {
736                         /* Print everything */
737                         llapi_printf(LLAPI_MSG_NORMAL, "%s", buf);
738                 }
739         }
740         param->obds_printed = 1;
741
742         fclose(fp);
743
744         if (!param->quiet && param->obduuid &&
745             (param->obdindex == OBD_NOT_FOUND)) {
746                 llapi_err_noerrno(LLAPI_MSG_ERROR,
747                                   "error: %s: unknown obduuid: %s",
748                                   __FUNCTION__, param->obduuid->uuid);
749                 //rc = EINVAL;
750         }
751
752         return (rc);
753 }
754
755 /* In this case, param->obduuid will be an array of obduuids and
756  * obd index for all these obduuids will be returned in
757  * param->obdindexes */
758 static int setup_obd_indexes(DIR *dir, struct find_param *param)
759 {
760         struct obd_uuid *uuids = NULL;
761         int obdcount = INIT_ALLOC_NUM_OSTS;
762         int ret, obd_valid = 0, obdnum, i;
763
764         uuids = (struct obd_uuid *)malloc(INIT_ALLOC_NUM_OSTS *
765                                           sizeof(struct obd_uuid));
766         if (uuids == NULL)
767                 return -ENOMEM;
768
769 retry_get_uuids:
770         ret = llapi_lov_get_uuids(dirfd(dir), uuids,
771                                   &obdcount);
772         if (ret) {
773                 struct obd_uuid *uuids_temp;
774
775                 if (ret == -EOVERFLOW) {
776                         uuids_temp = realloc(uuids, obdcount *
777                                              sizeof(struct obd_uuid));
778                         if (uuids_temp != NULL)
779                                 goto retry_get_uuids;
780                         else
781                                 ret = -ENOMEM;
782                 }
783
784                 llapi_err(LLAPI_MSG_ERROR, "get ost uuid failed");
785                 return ret;
786         }
787
788         param->obdindexes = malloc(param->num_obds * sizeof(param->obdindex));
789         if (param->obdindexes == NULL)
790                 return -ENOMEM;
791
792         for (obdnum = 0; obdnum < param->num_obds; obdnum++) {
793                 for (i = 0; i <= obdcount; i++) {
794                         if (strcmp((char *)&param->obduuid[obdnum].uuid,
795                                    (char *)&uuids[i]) == 0) {
796                                 param->obdindexes[obdnum] = i;
797                                 obd_valid++;
798                                 break;
799                         }
800                 }
801                 if (i == obdcount)
802                         param->obdindexes[obdnum] = OBD_NOT_FOUND;
803         }
804
805         if (obd_valid == 0)
806                 param->obdindex = OBD_NOT_FOUND;
807         else
808                 param->obdindex = obd_valid;
809
810         param->got_uuids = 1;
811
812         return 0;
813 }
814
815 void lov_dump_user_lmm_v1v3(struct lov_user_md *lum, char *pool_name,
816                             struct lov_user_ost_data_v1 *objects,
817                             char *path, int is_dir,
818                             int obdindex, int quiet, int header, int body)
819 {
820         int i, obdstripe = 0;
821
822         if (obdindex != OBD_NOT_FOUND) {
823                 for (i = 0; !is_dir && i < lum->lmm_stripe_count; i++) {
824                         if (obdindex == objects[i].l_ost_idx) {
825                                 llapi_printf(LLAPI_MSG_NORMAL, "%s\n", path);
826                                 obdstripe = 1;
827                                 break;
828                         }
829                 }
830         } else if (!quiet) {
831                 llapi_printf(LLAPI_MSG_NORMAL, "%s\n", path);
832                 obdstripe = 1;
833         }
834
835         /* if it's a directory */
836         if (is_dir) {
837                 if (obdstripe == 1) {
838                         if (lum->lmm_object_gr == LOV_OBJECT_GROUP_DEFAULT) {
839                                 llapi_printf(LLAPI_MSG_NORMAL, "(Default) ");
840                                 lum->lmm_object_gr = LOV_OBJECT_GROUP_CLEAR;
841                         }
842                         llapi_printf(LLAPI_MSG_NORMAL,
843                                      "stripe_count: %d stripe_size: %u "
844                                      "stripe_offset: %d%s%s\n",
845                                      lum->lmm_stripe_count == (__u16)-1 ? -1 :
846                                         lum->lmm_stripe_count,
847                                      lum->lmm_stripe_size,
848                                      lum->lmm_stripe_offset == (__u16)-1 ? -1 :
849                                         lum->lmm_stripe_offset,
850                                      pool_name != NULL ? " pool: " : "",
851                                      pool_name != NULL ? pool_name : "");
852                 }
853                 return;
854         }
855
856         if (header && (obdstripe == 1)) {
857                 llapi_printf(LLAPI_MSG_NORMAL,
858                              "lmm_magic:          0x%08X\n",  lum->lmm_magic);
859                 llapi_printf(LLAPI_MSG_NORMAL,
860                              "lmm_object_gr:      "LPX64"\n", lum->lmm_object_gr);
861                 llapi_printf(LLAPI_MSG_NORMAL,
862                              "lmm_object_id:      "LPX64"\n", lum->lmm_object_id);
863                 llapi_printf(LLAPI_MSG_NORMAL,
864                              "lmm_stripe_count:   %u\n", (int)lum->lmm_stripe_count);
865                 llapi_printf(LLAPI_MSG_NORMAL,
866                              "lmm_stripe_size:    %u\n",      lum->lmm_stripe_size);
867                 llapi_printf(LLAPI_MSG_NORMAL,
868                              "lmm_stripe_pattern: %x\n",      lum->lmm_pattern);
869                 if (pool_name != NULL)
870                         llapi_printf(LLAPI_MSG_NORMAL,
871                                      "lmm_pool_name:      %s\n",      pool_name);
872         }
873
874         if (body) {
875                 if ((!quiet) && (obdstripe == 1))
876                         llapi_printf(LLAPI_MSG_NORMAL,
877                                      "\tobdidx\t\t objid\t\tobjid\t\t group\n");
878
879                 for (i = 0; i < lum->lmm_stripe_count; i++) {
880                         int idx = objects[i].l_ost_idx;
881                         long long oid = objects[i].l_object_id;
882                         long long gr = objects[i].l_object_gr;
883                         if ((obdindex == OBD_NOT_FOUND) || (obdindex == idx))
884                                 llapi_printf(LLAPI_MSG_NORMAL,
885                                              "\t%6u\t%14llu\t%#13llx\t%14llu%s\n",
886                                              idx, oid, oid, gr,
887                                              obdindex == idx ? " *" : "");
888                 }
889                 llapi_printf(LLAPI_MSG_NORMAL, "\n");
890         }
891 }
892
893 void lov_dump_user_lmm_join(struct lov_user_md_v1 *lum, char *path,
894                             int is_dir, int obdindex, int quiet,
895                             int header, int body)
896 {
897         struct lov_user_md_join *lumj = (struct lov_user_md_join *)lum;
898         int i, obdstripe = 0;
899
900         if (obdindex != OBD_NOT_FOUND) {
901                 for (i = 0; i < lumj->lmm_stripe_count; i++) {
902                         if (obdindex == lumj->lmm_objects[i].l_ost_idx) {
903                                 llapi_printf(LLAPI_MSG_NORMAL, "%s\n", path);
904                                 obdstripe = 1;
905                                 break;
906                         }
907                 }
908         } else if (!quiet) {
909                 llapi_printf(LLAPI_MSG_NORMAL, "%s\n", path);
910                 obdstripe = 1;
911         }
912
913         if (header && obdstripe == 1) {
914                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_magic:          0x%08X\n",
915                              lumj->lmm_magic);
916                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_object_gr:      "LPX64"\n",
917                              lumj->lmm_object_gr);
918                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_object_id:      "LPX64"\n",
919                              lumj->lmm_object_id);
920                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_stripe_count:   %u\n",
921                              (int)lumj->lmm_stripe_count);
922                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_stripe_size:    %u\n",
923                              lumj->lmm_stripe_size);
924                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_stripe_pattern: %x\n",
925                              lumj->lmm_pattern);
926                 llapi_printf(LLAPI_MSG_NORMAL, "lmm_extent_count:   %x\n",
927                              lumj->lmm_extent_count);
928         }
929
930         if (body) {
931                 unsigned long long start = -1, end = 0;
932                 if (!quiet && obdstripe == 1)
933                         llapi_printf(LLAPI_MSG_NORMAL,
934                                      "joined\tobdidx\t\t objid\t\tobjid\t\t group"
935                                      "\t\tstart\t\tend\n");
936                 for (i = 0; i < lumj->lmm_stripe_count; i++) {
937                         int idx = lumj->lmm_objects[i].l_ost_idx;
938                         long long oid = lumj->lmm_objects[i].l_object_id;
939                         long long gr = lumj->lmm_objects[i].l_object_gr;
940                         if (obdindex == OBD_NOT_FOUND || obdindex == idx)
941                                 llapi_printf(LLAPI_MSG_NORMAL,
942                                              "\t%6u\t%14llu\t%#13llx\t%14llu%s",
943                                              idx, oid, oid, gr,
944                                              obdindex == idx ? " *" : "");
945                         if (start != lumj->lmm_objects[i].l_extent_start ||
946                             end != lumj->lmm_objects[i].l_extent_end) {
947                                 start = lumj->lmm_objects[i].l_extent_start;
948                                 llapi_printf(LLAPI_MSG_NORMAL, "\t%14llu", start);
949                                 end = lumj->lmm_objects[i].l_extent_end;
950                                 if (end == (unsigned long long)-1)
951                                         llapi_printf(LLAPI_MSG_NORMAL,
952                                                      "\t\tEOF\n");
953                                 else
954                                         llapi_printf(LLAPI_MSG_NORMAL,
955                                                      "\t\t%llu\n", end);
956                         } else {
957                                 llapi_printf(LLAPI_MSG_NORMAL, "\t\t\t\t\n");
958                         }
959                 }
960                 llapi_printf(LLAPI_MSG_NORMAL, "\n");
961         }
962 }
963
964 void llapi_lov_dump_user_lmm(struct find_param *param,
965                              char *path, int is_dir)
966 {
967         switch(*(__u32 *)&param->lmd->lmd_lmm) { /* lum->lmm_magic */
968         case LOV_USER_MAGIC_V1:
969                 lov_dump_user_lmm_v1v3(&param->lmd->lmd_lmm, NULL,
970                                        param->lmd->lmd_lmm.lmm_objects,
971                                        path, is_dir,
972                                        param->obdindex, param->quiet,
973                                        param->verbose,
974                                        (param->verbose || !param->obduuid));
975                 break;
976         case LOV_USER_MAGIC_JOIN:
977                 lov_dump_user_lmm_join(&param->lmd->lmd_lmm, path, is_dir,
978                                        param->obdindex, param->quiet,
979                                        param->verbose,
980                                        (param->verbose || !param->obduuid));
981                 break;
982         case LOV_USER_MAGIC_V3: {
983                 char pool_name[MAXPOOLNAME + 1];
984                 struct lov_user_ost_data_v1 *objects;
985
986                 strncpy(pool_name,
987                         ((struct lov_user_md_v3 *)(&param->lmd->lmd_lmm))->lmm_pool_name,
988                         MAXPOOLNAME);
989                 pool_name[MAXPOOLNAME] = '\0';
990                 objects = ((struct lov_user_md_v3 *)(&param->lmd->lmd_lmm))->lmm_objects;
991                 lov_dump_user_lmm_v1v3(&param->lmd->lmd_lmm, pool_name,
992                                       objects, path, is_dir,
993                                       param->obdindex, param->quiet,
994                                       param->verbose,
995                                       (param->verbose || !param->obduuid));
996                 break;
997         }
998         default:
999                 llapi_printf(LLAPI_MSG_NORMAL, "unknown lmm_magic:  %#x "
1000                              "(expecting one of %#x %#x %#x)\n",
1001                              *(__u32 *)&param->lmd->lmd_lmm,
1002                              LOV_USER_MAGIC_V1, LOV_USER_MAGIC_JOIN,
1003                              LOV_USER_MAGIC_V3);
1004                 return;
1005         }
1006 }
1007
1008 int llapi_file_get_stripe(const char *path, struct lov_user_md *lum)
1009 {
1010         const char *fname;
1011         char *dname;
1012         int fd, rc = 0;
1013
1014         fname = strrchr(path, '/');
1015
1016         /* It should be a file (or other non-directory) */
1017         if (fname == NULL) {
1018                 dname = (char *)malloc(2);
1019                 if (dname == NULL)
1020                         return ENOMEM;
1021                 strcpy(dname, ".");
1022                 fname = (char *)path;
1023         } else {
1024                 dname = (char *)malloc(fname - path + 1);
1025                 if (dname == NULL)
1026                         return ENOMEM;
1027                 strncpy(dname, path, fname - path);
1028                 dname[fname - path] = '\0';
1029                 fname++;
1030         }
1031
1032         if ((fd = open(dname, O_RDONLY)) == -1) {
1033                 rc = errno;
1034                 free(dname);
1035                 return rc;
1036         }
1037
1038         strcpy((char *)lum, fname);
1039         if (ioctl(fd, IOC_MDC_GETFILESTRIPE, (void *)lum) == -1)
1040                 rc = errno;
1041
1042         if (close(fd) == -1 && rc == 0)
1043                 rc = errno;
1044
1045         free(dname);
1046
1047         return rc;
1048 }
1049
1050 int llapi_file_lookup(int dirfd, const char *name)
1051 {
1052         struct obd_ioctl_data data = { 0 };
1053         char rawbuf[8192];
1054         char *buf = rawbuf;
1055         int rc;
1056
1057         if (dirfd < 0 || name == NULL)
1058                 return -EINVAL;
1059
1060         data.ioc_version = OBD_IOCTL_VERSION;
1061         data.ioc_len = sizeof(data);
1062         data.ioc_inlbuf1 = (char *)name;
1063         data.ioc_inllen1 = strlen(name) + 1;
1064
1065         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1066         if (rc) {
1067                 llapi_err(LLAPI_MSG_ERROR,
1068                           "error: IOC_MDC_LOOKUP pack failed for '%s': rc %d",
1069                           name, rc);
1070                 return rc;
1071         }
1072
1073         return ioctl(dirfd, IOC_MDC_LOOKUP, buf);
1074 }
1075
1076 int llapi_mds_getfileinfo(char *path, DIR *parent,
1077                           struct lov_user_mds_data *lmd)
1078 {
1079         lstat_t *st = &lmd->lmd_st;
1080         char *fname = strrchr(path, '/');
1081         int ret = 0;
1082
1083         if (parent == NULL)
1084                 return -EINVAL;
1085
1086         fname = (fname == NULL ? path : fname + 1);
1087         /* retrieve needed file info */
1088         strncpy((char *)lmd, fname,
1089                 lov_mds_md_size(MAX_LOV_UUID_COUNT, LOV_MAGIC));
1090         ret = ioctl(dirfd(parent), IOC_MDC_GETFILEINFO, (void *)lmd);
1091
1092         if (ret) {
1093                 if (errno == ENOTTY) {
1094                         /* ioctl is not supported, it is not a lustre fs.
1095                          * Do the regular lstat(2) instead. */
1096                         ret = lstat_f(path, st);
1097                         if (ret) {
1098                                 llapi_err(LLAPI_MSG_ERROR,
1099                                           "error: %s: lstat failed for %s",
1100                                           __FUNCTION__, path);
1101                                 return ret;
1102                         }
1103                 } else if (errno == ENOENT) {
1104                         llapi_err(LLAPI_MSG_WARN,
1105                                   "warning: %s: %s does not exist",
1106                                   __FUNCTION__, path);
1107                         return -ENOENT;
1108                 } else {
1109                         llapi_err(LLAPI_MSG_ERROR,
1110                                   "error: %s: IOC_MDC_GETFILEINFO failed for %s",
1111                                   __FUNCTION__, path);
1112                         return ret;
1113                 }
1114         }
1115
1116         return 0;
1117 }
1118
1119 static DIR *opendir_parent(char *path)
1120 {
1121         DIR *parent;
1122         char *fname;
1123         char c;
1124
1125         fname = strrchr(path, '/');
1126         if (fname == NULL)
1127                 return opendir(".");
1128
1129         c = fname[1];
1130         fname[1] = '\0';
1131         parent = opendir(path);
1132         fname[1] = c;
1133         return parent;
1134 }
1135
1136 static int llapi_semantic_traverse(char *path, int size, DIR *parent,
1137                                    semantic_func_t sem_init,
1138                                    semantic_func_t sem_fini, void *data,
1139                                    cfs_dirent_t *de)
1140 {
1141         cfs_dirent_t *dent;
1142         int len, ret;
1143         DIR *d, *p = NULL;
1144
1145         ret = 0;
1146         len = strlen(path);
1147
1148         d = opendir(path);
1149         if (!d && errno != ENOTDIR) {
1150                 llapi_err(LLAPI_MSG_ERROR, "%s: Failed to open '%s'",
1151                           __FUNCTION__, path);
1152                 return -EINVAL;
1153         } else if (!d && !parent) {
1154                 /* ENOTDIR. Open the parent dir. */
1155                 p = opendir_parent(path);
1156                 if (!p)
1157                         GOTO(out, ret = -EINVAL);
1158         }
1159
1160         if (sem_init && (ret = sem_init(path, parent ?: p, d, data, de)))
1161                 goto err;
1162
1163         if (!d)
1164                 GOTO(out, ret = 0);
1165
1166         while ((dent = readdir64(d)) != NULL) {
1167                 ((struct find_param *)data)->have_fileinfo = 0;
1168
1169                 if (!strcmp(dent->d_name, ".") || !strcmp(dent->d_name, ".."))
1170                         continue;
1171
1172                 path[len] = 0;
1173                 if ((len + dent->d_reclen + 2) > size) {
1174                         llapi_err(LLAPI_MSG_ERROR,
1175                                   "error: %s: string buffer is too small",
1176                                   __FUNCTION__);
1177                         break;
1178                 }
1179                 strcat(path, "/");
1180                 strcat(path, dent->d_name);
1181
1182                 if (dent->d_type == DT_UNKNOWN) {
1183                         lstat_t *st = &((struct find_param *)data)->lmd->lmd_st;
1184
1185                         ret = llapi_mds_getfileinfo(path, d,
1186                                              ((struct find_param *)data)->lmd);
1187                         if (ret == 0) {
1188                                 ((struct find_param *)data)->have_fileinfo = 1;
1189                                 dent->d_type = llapi_filetype_dir_table[st->st_mode &
1190                                                                         S_IFMT];
1191                         }
1192                         if (ret == -ENOENT)
1193                                 continue;
1194                 }
1195
1196                 switch (dent->d_type) {
1197                 case DT_UNKNOWN:
1198                         llapi_err(LLAPI_MSG_ERROR,
1199                                   "error: %s: '%s' is UNKNOWN type %d",
1200                                   __FUNCTION__, dent->d_name, dent->d_type);
1201                         break;
1202                 case DT_DIR:
1203                         ret = llapi_semantic_traverse(path, size, d, sem_init,
1204                                                       sem_fini, data, dent);
1205                         if (ret < 0)
1206                                 goto out;
1207                         break;
1208                 default:
1209                         ret = 0;
1210                         if (sem_init) {
1211                                 ret = sem_init(path, d, NULL, data, dent);
1212                                 if (ret < 0)
1213                                         goto out;
1214                         }
1215                         if (sem_fini && ret == 0)
1216                                 sem_fini(path, d, NULL, data, dent);
1217                 }
1218         }
1219
1220 out:
1221         path[len] = 0;
1222
1223         if (sem_fini)
1224                 sem_fini(path, parent, d, data, de);
1225 err:
1226         if (d)
1227                 closedir(d);
1228         if (p)
1229                 closedir(p);
1230         return ret;
1231 }
1232
1233 /* Check if the value matches 1 of the given criteria (e.g. --atime +/-N).
1234  * @mds indicates if this is MDS timestamps and there are attributes on OSTs.
1235  *
1236  * The result is -1 if it does not match, 0 if not yet clear, 1 if matches.
1237  * The table bolow gives the answers for the specified parameters (value and
1238  * sign), 1st column is the answer for the MDS value, the 2nd is for the OST:
1239  * --------------------------------------
1240  * 1 | file > limit; sign > 0 | -1 / -1 |
1241  * 2 | file = limit; sign > 0 |  ? /  1 |
1242  * 3 | file < limit; sign > 0 |  ? /  1 |
1243  * 4 | file > limit; sign = 0 | -1 / -1 |
1244  * 5 | file = limit; sign = 0 |  ? /  1 |  <- (see the Note below)
1245  * 6 | file < limit; sign = 0 |  ? / -1 |
1246  * 7 | file > limit; sign < 0 |  1 /  1 |
1247  * 8 | file = limit; sign < 0 |  ? / -1 |
1248  * 9 | file < limit; sign < 0 |  ? / -1 |
1249  * --------------------------------------
1250  * Note: 5th actually means that the value is within the interval
1251  * (limit - margin, limit]. */
1252 static int find_value_cmp(unsigned int file, unsigned int limit, int sign,
1253                           unsigned long long margin, int mds)
1254 {
1255         if (sign > 0) {
1256                 if (file < limit)
1257                         return mds ? 0 : 1;
1258         }
1259
1260         if (sign == 0) {
1261                 if (file <= limit && file + margin > limit)
1262                         return mds ? 0 : 1;
1263                 if (file + margin <= limit)
1264                         return mds ? 0 : -1;
1265         }
1266
1267         if (sign < 0) {
1268                 if (file > limit)
1269                         return 1;
1270                 if (mds)
1271                         return 0;
1272         }
1273
1274         return -1;
1275 }
1276
1277 /* Check if the file time matches all the given criteria (e.g. --atime +/-N).
1278  * Return -1 or 1 if file timestamp does not or does match the given criteria
1279  * correspondingly. Return 0 if the MDS time is being checked and there are
1280  * attributes on OSTs and it is not yet clear if the timespamp matches.
1281  *
1282  * If 0 is returned, we need to do another RPC to the OSTs to obtain the
1283  * updated timestamps. */
1284 static int find_time_check(lstat_t *st, struct find_param *param, int mds)
1285 {
1286         int ret;
1287         int rc = 0;
1288
1289         /* Check if file is accepted. */
1290         if (param->atime) {
1291                 ret = find_value_cmp(st->st_atime, param->atime,
1292                                      param->asign, 24 * 60 * 60, mds);
1293                 if (ret < 0)
1294                         return ret;
1295                 rc = ret;
1296         }
1297
1298         if (param->mtime) {
1299                 ret = find_value_cmp(st->st_mtime, param->mtime,
1300                                      param->msign, 24 * 60 * 60, mds);
1301                 if (ret < 0)
1302                         return ret;
1303
1304                 /* If the previous check matches, but this one is not yet clear,
1305                  * we should return 0 to do an RPC on OSTs. */
1306                 if (rc == 1)
1307                         rc = ret;
1308         }
1309
1310         if (param->ctime) {
1311                 ret = find_value_cmp(st->st_ctime, param->ctime,
1312                                      param->csign, 24 * 60 * 60, mds);
1313                 if (ret < 0)
1314                         return ret;
1315
1316                 /* If the previous check matches, but this one is not yet clear,
1317                  * we should return 0 to do an RPC on OSTs. */
1318                 if (rc == 1)
1319                         rc = ret;
1320         }
1321
1322         return rc;
1323 }
1324
1325 static int cb_find_init(char *path, DIR *parent, DIR *dir,
1326                         void *data, cfs_dirent_t *de)
1327 {
1328         struct find_param *param = (struct find_param *)data;
1329         int decision = 1; /* 1 is accepted; -1 is rejected. */
1330         lstat_t *st = &param->lmd->lmd_st;
1331         int lustre_fs = 1;
1332         int checked_type = 0;
1333         int ret = 0;
1334
1335         LASSERT(parent != NULL || dir != NULL);
1336
1337         param->lmd->lmd_lmm.lmm_stripe_count = 0;
1338
1339         /* If a regular expression is presented, make the initial decision */
1340         if (param->pattern != NULL) {
1341                 char *fname = strrchr(path, '/');
1342                 fname = (fname == NULL ? path : fname + 1);
1343                 ret = fnmatch(param->pattern, fname, 0);
1344                 if ((ret == FNM_NOMATCH && !param->exclude_pattern) ||
1345                     (ret == 0 && param->exclude_pattern))
1346                         goto decided;
1347         }
1348
1349         /* See if we can check the file type from the dirent. */
1350         if (param->type && de != NULL && de->d_type != DT_UNKNOWN &&
1351             de->d_type < DT_MAX) {
1352                 checked_type = 1;
1353                 if (llapi_dir_filetype_table[de->d_type] == param->type) {
1354                         if (param->exclude_type)
1355                                 goto decided;
1356                 } else {
1357                         if (!param->exclude_type)
1358                                 goto decided;
1359                 }
1360         }
1361
1362
1363         /* If a time or OST should be checked, the decision is not taken yet. */
1364         if (param->atime || param->ctime || param->mtime || param->obduuid ||
1365             param->size)
1366                 decision = 0;
1367
1368         ret = 0;
1369         /* Request MDS for the stat info. */
1370         if (param->have_fileinfo == 0) {
1371                 if (dir) {
1372                         /* retrieve needed file info */
1373                         ret = ioctl(dirfd(dir), LL_IOC_MDC_GETINFO,
1374                                     (void *)param->lmd);
1375                 } else {
1376                         char *fname = strrchr(path, '/');
1377                         fname = (fname == NULL ? path : fname + 1);
1378
1379                         /* retrieve needed file info */
1380                         strncpy((char *)param->lmd, fname, param->lumlen);
1381                         ret = ioctl(dirfd(parent), IOC_MDC_GETFILEINFO,
1382                                    (void *)param->lmd);
1383                 }
1384         }
1385
1386         if (ret) {
1387                 if (errno == ENOTTY) {
1388                         /* ioctl is not supported, it is not a lustre fs.
1389                          * Do the regular lstat(2) instead. */
1390                         lustre_fs = 0;
1391                         ret = lstat_f(path, st);
1392                         if (ret) {
1393                                 llapi_err(LLAPI_MSG_ERROR,
1394                                           "error: %s: lstat failed for %s",
1395                                           __FUNCTION__, path);
1396                                 return ret;
1397                         }
1398                 } else if (errno == ENOENT) {
1399                         llapi_err(LLAPI_MSG_WARN,
1400                                   "warning: %s: %s does not exist",
1401                                   __FUNCTION__, path);
1402                         goto decided;
1403                 } else {
1404                         llapi_err(LLAPI_MSG_ERROR, "error: %s: %s failed for %s",
1405                                   __FUNCTION__, dir ? "LL_IOC_MDC_GETINFO" :
1406                                   "IOC_MDC_GETFILEINFO", path);
1407                         return ret;
1408                 }
1409         }
1410
1411         if (param->type && !checked_type) {
1412                 if ((st->st_mode & S_IFMT) == param->type) {
1413                         if (param->exclude_type)
1414                                 goto decided;
1415                 } else {
1416                         if (!param->exclude_type)
1417                                 goto decided;
1418                 }
1419         }
1420
1421         /* Prepare odb. */
1422         if (param->obduuid) {
1423                 if (lustre_fs && param->got_uuids &&
1424                     param->st_dev != st->st_dev) {
1425                         /* A lustre/lustre mount point is crossed. */
1426                         param->got_uuids = 0;
1427                         param->obds_printed = 0;
1428                         param->obdindex = OBD_NOT_FOUND;
1429                 }
1430
1431                 if (lustre_fs && !param->got_uuids) {
1432                         ret = setup_obd_indexes(dir ? dir : parent, param);
1433                         if (ret)
1434                                 return ret;
1435
1436                         param->st_dev = st->st_dev;
1437                 } else if (!lustre_fs && param->got_uuids) {
1438                         /* A lustre/non-lustre mount point is crossed. */
1439                         param->got_uuids = 0;
1440                         param->obdindex = OBD_NOT_FOUND;
1441                 }
1442         }
1443
1444         /* If an OBD UUID is specified but no one matches, skip this file. */
1445         if (param->obduuid && param->obdindex == OBD_NOT_FOUND)
1446                 goto decided;
1447
1448         /* If a OST UUID is given, and some OST matches, check it here. */
1449         if (param->obdindex != OBD_NOT_FOUND) {
1450                 if (!S_ISREG(st->st_mode))
1451                         goto decided;
1452
1453                 /* Only those files should be accepted, which have a
1454                  * stripe on the specified OST. */
1455                 if (!param->lmd->lmd_lmm.lmm_stripe_count) {
1456                         goto decided;
1457                 } else {
1458                         int i, j;
1459                         struct lov_user_ost_data_v1 *lmm_objects;
1460
1461                         if (param->lmd->lmd_lmm.lmm_magic ==
1462                             LOV_USER_MAGIC_V3) {
1463                                 lmm_objects =
1464                                  ((struct lov_user_md_v3 *)(&(param->lmd->lmd_lmm)))->lmm_objects;
1465                         } else {
1466                                 lmm_objects = param->lmd->lmd_lmm.lmm_objects;
1467                         }
1468
1469                         for (i = 0;
1470                              i < param->lmd->lmd_lmm.lmm_stripe_count; i++) {
1471                                 for (j = 0; j < param->num_obds; j++) {
1472                                         if (param->obdindexes[j] ==
1473                                             lmm_objects[i].l_ost_idx)
1474                                                 goto obd_matches;
1475                                 }
1476                         }
1477
1478                         if (i == param->lmd->lmd_lmm.lmm_stripe_count)
1479                                 goto decided;
1480                 }
1481         }
1482
1483         if (param->check_uid) {
1484                 if (st->st_uid == param->uid) {
1485                         if (param->exclude_uid)
1486                                 goto decided;
1487                 } else {
1488                         if (!param->exclude_uid)
1489                                 goto decided;
1490                 }
1491         }
1492
1493         if (param->check_gid) {
1494                 if (st->st_gid == param->gid) {
1495                         if (param->exclude_gid)
1496                                 goto decided;
1497                 } else {
1498                         if (!param->exclude_gid)
1499                                 goto decided;
1500                 }
1501         }
1502
1503         if (param->check_pool) {
1504                 /* empty requested pool is taken as no pool search => V1 */
1505                 if (((param->lmd->lmd_lmm.lmm_magic == LOV_USER_MAGIC_V1) &&
1506                      (param->poolname[0] == '\0')) ||
1507                     ((param->lmd->lmd_lmm.lmm_magic == LOV_USER_MAGIC_V3) &&
1508                      (strncmp(((struct lov_user_md_v3 *)(&(param->lmd->lmd_lmm)))->lmm_pool_name,
1509                               param->poolname, MAXPOOLNAME) == 0)) ||
1510                     ((param->lmd->lmd_lmm.lmm_magic == LOV_USER_MAGIC_V3) &&
1511                      (strcmp(param->poolname, "*") == 0))) {
1512                         if (param->exclude_pool)
1513                                 goto decided;
1514                 } else {
1515                         if (!param->exclude_pool)
1516                                 goto decided;
1517                 }
1518         }
1519
1520         /* Check the time on mds. */
1521         if (!decision) {
1522                 int for_mds;
1523
1524                 for_mds = lustre_fs ? (S_ISREG(st->st_mode) &&
1525                                        param->lmd->lmd_lmm.lmm_stripe_count)
1526                                     : 0;
1527                 decision = find_time_check(st, param, for_mds);
1528         }
1529
1530 obd_matches:
1531         /* If file still fits the request, ask osd for updated info.
1532            The regulat stat is almost of the same speed as some new
1533            'glimpse-size-ioctl'. */
1534         if (!decision && S_ISREG(st->st_mode) &&
1535             (param->lmd->lmd_lmm.lmm_stripe_count || param->size)) {
1536                 if (param->obdindex != OBD_NOT_FOUND) {
1537                         /* Check whether the obd is active or not, if it is
1538                          * not active, just print the object affected by this
1539                          * failed ost
1540                          * */
1541                         struct obd_statfs stat_buf;
1542                         struct obd_uuid uuid_buf;
1543
1544                         memset(&stat_buf, 0, sizeof(struct obd_statfs));
1545                         memset(&uuid_buf, 0, sizeof(struct obd_uuid));
1546                         ret = llapi_obd_statfs(path, LL_STATFS_LOV,
1547                                                param->obdindex, &stat_buf,
1548                                                &uuid_buf);
1549                         if (ret) {
1550                                 if (ret == -ENODATA || ret == -ENODEV
1551                                     || ret == -EIO)
1552                                         errno = EIO;
1553                                 llapi_printf(LLAPI_MSG_NORMAL,
1554                                              "obd_uuid: %s failed %s ",
1555                                              param->obduuid->uuid,
1556                                              strerror(errno));
1557                                 goto print_path;
1558                         }
1559                 }
1560                 if (dir) {
1561                         ret = ioctl(dirfd(dir), IOC_LOV_GETINFO,
1562                                     (void *)param->lmd);
1563                 } else if (parent) {
1564                         ret = ioctl(dirfd(parent), IOC_LOV_GETINFO,
1565                                     (void *)param->lmd);
1566                 }
1567
1568                 if (ret) {
1569                         if (errno == ENOENT) {
1570                                 llapi_err(LLAPI_MSG_ERROR,
1571                                           "warning: %s: %s does not exist",
1572                                           __FUNCTION__, path);
1573                                 goto decided;
1574                         } else {
1575                                 llapi_err(LLAPI_MSG_ERROR,
1576                                           "%s: IOC_LOV_GETINFO on %s failed",
1577                                           __FUNCTION__, path);
1578                                 return ret;
1579                         }
1580                 }
1581
1582                 /* Check the time on osc. */
1583                 decision = find_time_check(st, param, 0);
1584                 if (decision == -1)
1585                         goto decided;
1586         }
1587
1588         if (param->size)
1589                 decision = find_value_cmp(st->st_size, param->size,
1590                                           param->size_sign, param->size_units,
1591                                           0);
1592
1593 print_path:
1594         if (decision != -1) {
1595                 llapi_printf(LLAPI_MSG_NORMAL, "%s", path);
1596                 if (param->zeroend)
1597                         llapi_printf(LLAPI_MSG_NORMAL, "%c", '\0');
1598                 else
1599                         llapi_printf(LLAPI_MSG_NORMAL, "\n");
1600         }
1601
1602 decided:
1603         /* Do not get down anymore? */
1604         if (param->depth == param->maxdepth)
1605                 return 1;
1606
1607         param->depth++;
1608         return 0;
1609 }
1610
1611 static int cb_common_fini(char *path, DIR *parent, DIR *d, void *data,
1612                           cfs_dirent_t *de)
1613 {
1614         struct find_param *param = (struct find_param *)data;
1615         param->depth--;
1616         return 0;
1617 }
1618
1619 int llapi_find(char *path, struct find_param *param)
1620 {
1621         char *buf;
1622         int ret, len = strlen(path);
1623
1624         if (len > PATH_MAX) {
1625                 llapi_err(LLAPI_MSG_ERROR, "%s: Path name '%s' is too long",
1626                           __FUNCTION__, path);
1627                 return -EINVAL;
1628         }
1629
1630         buf = (char *)malloc(PATH_MAX + 1);
1631         if (!buf)
1632                 return -ENOMEM;
1633
1634         ret = common_param_init(param);
1635         if (ret) {
1636                 free(buf);
1637                 return ret;
1638         }
1639
1640         param->depth = 0;
1641
1642         strncpy(buf, path, PATH_MAX + 1);
1643         ret = llapi_semantic_traverse(buf, PATH_MAX + 1, NULL, cb_find_init,
1644                                       cb_common_fini, param, NULL);
1645
1646         find_param_fini(param);
1647         free(buf);
1648         return ret < 0 ? ret : 0;
1649 }
1650
1651 static int cb_getstripe(char *path, DIR *parent, DIR *d, void *data,
1652                         cfs_dirent_t *de)
1653 {
1654         struct find_param *param = (struct find_param *)data;
1655         int ret = 0;
1656
1657         LASSERT(parent != NULL || d != NULL);
1658
1659         /* Prepare odb. */
1660         if (!param->got_uuids) {
1661                 ret = setup_obd_uuid(d ? d : parent, path, param);
1662                 if (ret)
1663                         return ret;
1664         }
1665
1666         if (d) {
1667                 ret = ioctl(dirfd(d), LL_IOC_LOV_GETSTRIPE,
1668                             (void *)&param->lmd->lmd_lmm);
1669         } else if (parent) {
1670                 char *fname = strrchr(path, '/');
1671                 fname = (fname == NULL ? path : fname + 1);
1672
1673                 strncpy((char *)&param->lmd->lmd_lmm, fname, param->lumlen);
1674                 ret = ioctl(dirfd(parent), IOC_MDC_GETFILESTRIPE,
1675                             (void *)&param->lmd->lmd_lmm);
1676         }
1677
1678         if (ret) {
1679                 if (errno == ENODATA) {
1680                         if (!param->obduuid && !param->quiet)
1681                                 llapi_printf(LLAPI_MSG_NORMAL,
1682                                              "%s has no stripe info\n", path);
1683                         goto out;
1684                 } else if (errno == ENOTTY) {
1685                         llapi_err(LLAPI_MSG_ERROR,
1686                                   "%s: '%s' not on a Lustre fs?",
1687                                   __FUNCTION__, path);
1688                 } else if (errno == ENOENT) {
1689                         llapi_err(LLAPI_MSG_WARN,
1690                                   "warning: %s: %s does not exist",
1691                                   __FUNCTION__, path);
1692                         goto out;
1693                 } else {
1694                         llapi_err(LLAPI_MSG_ERROR,
1695                                   "error: %s: %s failed for %s",
1696                                    __FUNCTION__, d ? "LL_IOC_LOV_GETSTRIPE" :
1697                                   "IOC_MDC_GETFILESTRIPE", path);
1698                 }
1699
1700                 return ret;
1701         }
1702
1703         llapi_lov_dump_user_lmm(param, path, d ? 1 : 0);
1704 out:
1705         /* Do not get down anymore? */
1706         if (param->depth == param->maxdepth)
1707                 return 1;
1708
1709         param->depth++;
1710         return 0;
1711 }
1712
1713 int llapi_getstripe(char *path, struct find_param *param)
1714 {
1715         char *buf;
1716         int ret = 0, len = strlen(path);
1717
1718         if (len > PATH_MAX) {
1719                 llapi_err(LLAPI_MSG_ERROR,
1720                           "%s: Path name '%s' is too long",
1721                           __FUNCTION__, path);
1722                 return -EINVAL;
1723         }
1724
1725         buf = (char *)malloc(PATH_MAX + 1);
1726         if (!buf)
1727                 return -ENOMEM;
1728
1729         ret = common_param_init(param);
1730         if (ret) {
1731                 free(buf);
1732                 return ret;
1733         }
1734
1735         param->depth = 0;
1736
1737         strncpy(buf, path, PATH_MAX + 1);
1738         ret = llapi_semantic_traverse(buf, PATH_MAX + 1, NULL, cb_getstripe,
1739                                       cb_common_fini, param, NULL);
1740         find_param_fini(param);
1741         free(buf);
1742         return ret < 0 ? ret : 0;
1743 }
1744
1745 int llapi_obd_statfs(char *path, __u32 type, __u32 index,
1746                      struct obd_statfs *stat_buf,
1747                      struct obd_uuid *uuid_buf)
1748 {
1749         int fd;
1750         char raw[OBD_MAX_IOCTL_BUFFER] = {'\0'};
1751         char *rawbuf = raw;
1752         struct obd_ioctl_data data = { 0 };
1753         int rc = 0;
1754
1755         data.ioc_inlbuf1 = (char *)&type;
1756         data.ioc_inllen1 = sizeof(__u32);
1757         data.ioc_inlbuf2 = (char *)&index;
1758         data.ioc_inllen2 = sizeof(__u32);
1759         data.ioc_pbuf1 = (char *)stat_buf;
1760         data.ioc_plen1 = sizeof(struct obd_statfs);
1761         data.ioc_pbuf2 = (char *)uuid_buf;
1762         data.ioc_plen2 = sizeof(struct obd_uuid);
1763
1764         if ((rc = obd_ioctl_pack(&data, &rawbuf, sizeof(raw))) != 0) {
1765                 llapi_err(LLAPI_MSG_ERROR,
1766                           "llapi_obd_statfs: error packing ioctl data");
1767                 return rc;
1768         }
1769
1770         fd = open(path, O_RDONLY);
1771         if (errno == EISDIR)
1772                 fd = open(path, O_DIRECTORY | O_RDONLY);
1773
1774         if (fd < 0) {
1775                 rc = errno ? -errno : -EBADF;
1776                 llapi_err(LLAPI_MSG_ERROR, "error: %s: opening '%s'",
1777                           __FUNCTION__, path);
1778                 return rc;
1779         }
1780         rc = ioctl(fd, IOC_OBD_STATFS, (void *)rawbuf);
1781         if (rc)
1782                 rc = errno ? -errno : -EINVAL;
1783
1784         close(fd);
1785         return rc;
1786 }
1787
1788 #define MAX_STRING_SIZE 128
1789 #define DEVICES_LIST "/proc/fs/lustre/devices"
1790
1791 int llapi_ping(char *obd_type, char *obd_name)
1792 {
1793         char path[MAX_STRING_SIZE];
1794         char buf[1];
1795         int rc, fd;
1796
1797         snprintf(path, MAX_STRING_SIZE, "/proc/fs/lustre/%s/%s/ping",
1798                  obd_type, obd_name);
1799
1800         fd = open(path, O_WRONLY);
1801         if (fd < 0) {
1802                 rc = errno;
1803                 llapi_err(LLAPI_MSG_ERROR, "error opening %s", path);
1804                 return rc;
1805         }
1806
1807         rc = write(fd, buf, 1);
1808         close(fd);
1809
1810         if (rc == 1)
1811                 return 0;
1812         return rc;
1813 }
1814
1815 int llapi_target_iterate(int type_num, char **obd_type, void *args, llapi_cb_t cb)
1816 {
1817         char buf[MAX_STRING_SIZE];
1818         FILE *fp = fopen(DEVICES_LIST, "r");
1819         int i, rc = 0;
1820
1821         if (fp == NULL) {
1822                 rc = errno;
1823                 llapi_err(LLAPI_MSG_ERROR, "error: opening "DEVICES_LIST);
1824                 return rc;
1825         }
1826
1827         while (fgets(buf, sizeof(buf), fp) != NULL) {
1828                 char *obd_type_name = NULL;
1829                 char *obd_name = NULL;
1830                 char *obd_uuid = NULL;
1831                 char rawbuf[OBD_MAX_IOCTL_BUFFER];
1832                 char *bufl = rawbuf;
1833                 char *bufp = buf;
1834                 struct obd_ioctl_data datal = { 0, };
1835                 struct obd_statfs osfs_buffer;
1836
1837                 while(bufp[0] == ' ')
1838                         ++bufp;
1839
1840                 for(i = 0; i < 3; i++) {
1841                         obd_type_name = strsep(&bufp, " ");
1842                 }
1843                 obd_name = strsep(&bufp, " ");
1844                 obd_uuid = strsep(&bufp, " ");
1845
1846                 memset(&osfs_buffer, 0, sizeof (osfs_buffer));
1847
1848                 memset(bufl, 0, sizeof(rawbuf));
1849                 datal.ioc_pbuf1 = (char *)&osfs_buffer;
1850                 datal.ioc_plen1 = sizeof(osfs_buffer);
1851
1852                 for (i = 0; i < type_num; i++) {
1853                         if (strcmp(obd_type_name, obd_type[i]) != 0)
1854                                 continue;
1855
1856                         cb(obd_type_name, obd_name, obd_uuid, args);
1857                 }
1858         }
1859         fclose(fp);
1860         return rc;
1861 }
1862
1863 static void do_target_check(char *obd_type_name, char *obd_name,
1864                             char *obd_uuid, void *args)
1865 {
1866         int rc;
1867
1868         rc = llapi_ping(obd_type_name, obd_name);
1869         if (rc == ENOTCONN) {
1870                 llapi_printf(LLAPI_MSG_NORMAL, "%s inactive.\n", obd_name);
1871         } else if (rc) {
1872                 llapi_err(LLAPI_MSG_ERROR, "error: check '%s'", obd_name);
1873         } else {
1874                 llapi_printf(LLAPI_MSG_NORMAL, "%s active.\n", obd_name);
1875         }
1876 }
1877
1878 int llapi_target_check(int type_num, char **obd_type, char *dir)
1879 {
1880         return llapi_target_iterate(type_num, obd_type, NULL, do_target_check);
1881 }
1882
1883 #undef MAX_STRING_SIZE
1884
1885 int llapi_catinfo(char *dir, char *keyword, char *node_name)
1886 {
1887         char raw[OBD_MAX_IOCTL_BUFFER];
1888         char out[LLOG_CHUNK_SIZE];
1889         char *buf = raw;
1890         struct obd_ioctl_data data = { 0 };
1891         char key[30];
1892         DIR *root;
1893         int rc;
1894
1895         sprintf(key, "%s", keyword);
1896         memset(raw, 0, sizeof(raw));
1897         memset(out, 0, sizeof(out));
1898         data.ioc_inlbuf1 = key;
1899         data.ioc_inllen1 = strlen(key) + 1;
1900         if (node_name) {
1901                 data.ioc_inlbuf2 = node_name;
1902                 data.ioc_inllen2 = strlen(node_name) + 1;
1903         }
1904         data.ioc_pbuf1 = out;
1905         data.ioc_plen1 = sizeof(out);
1906         rc = obd_ioctl_pack(&data, &buf, sizeof(raw));
1907         if (rc)
1908                 return rc;
1909
1910         root = opendir(dir);
1911         if (root == NULL) {
1912                 rc = errno;
1913                 llapi_err(LLAPI_MSG_ERROR, "open %s failed", dir);
1914                 return rc;
1915         }
1916
1917         rc = ioctl(dirfd(root), OBD_IOC_LLOG_CATINFO, buf);
1918         if (rc)
1919                 llapi_err(LLAPI_MSG_ERROR, "ioctl OBD_IOC_CATINFO failed");
1920         else
1921                 llapi_printf(LLAPI_MSG_NORMAL, "%s", data.ioc_pbuf1);
1922
1923         closedir(root);
1924         return rc;
1925 }
1926
1927 /* Is this a lustre fs? */
1928 int llapi_is_lustre_mnttype(const char *type)
1929 {
1930         return (strcmp(type, "lustre") == 0 || strcmp(type,"lustre_lite") == 0);
1931 }
1932
1933 /* Is this a lustre client fs? */
1934 int llapi_is_lustre_mnt(struct mntent *mnt)
1935 {
1936         return (llapi_is_lustre_mnttype(mnt->mnt_type) &&
1937                 strstr(mnt->mnt_fsname, ":/") != NULL);
1938 }
1939
1940 int llapi_quotacheck(char *mnt, int check_type)
1941 {
1942         DIR *root;
1943         int rc;
1944
1945         root = opendir(mnt);
1946         if (!root) {
1947                 llapi_err(LLAPI_MSG_ERROR, "open %s failed", mnt);
1948                 return -1;
1949         }
1950
1951         rc = ioctl(dirfd(root), LL_IOC_QUOTACHECK, check_type);
1952
1953         closedir(root);
1954         return rc;
1955 }
1956
1957 int llapi_poll_quotacheck(char *mnt, struct if_quotacheck *qchk)
1958 {
1959         DIR *root;
1960         int poll_intvl = 2;
1961         int rc;
1962
1963         root = opendir(mnt);
1964         if (!root) {
1965                 llapi_err(LLAPI_MSG_ERROR, "open %s failed", mnt);
1966                 return -1;
1967         }
1968
1969         while (1) {
1970                 rc = ioctl(dirfd(root), LL_IOC_POLL_QUOTACHECK, qchk);
1971                 if (!rc)
1972                         break;
1973                 sleep(poll_intvl);
1974                 if (poll_intvl < 30)
1975                         poll_intvl *= 2;
1976         }
1977
1978         closedir(root);
1979         return rc;
1980 }
1981
1982 int llapi_quotactl(char *mnt, struct if_quotactl *qctl)
1983 {
1984         DIR *root;
1985         int rc;
1986
1987         root = opendir(mnt);
1988         if (!root) {
1989                 llapi_err(LLAPI_MSG_ERROR, "open %s failed", mnt);
1990                 return -1;
1991         }
1992
1993         rc = ioctl(dirfd(root), LL_IOC_QUOTACTL, qctl);
1994
1995         closedir(root);
1996         return rc;
1997 }
1998
1999 static int cb_quotachown(char *path, DIR *parent, DIR *d, void *data,
2000                          cfs_dirent_t *de)
2001 {
2002         struct find_param *param = (struct find_param *)data;
2003         lstat_t *st;
2004         int rc;
2005
2006         LASSERT(parent != NULL || d != NULL);
2007
2008         if (d) {
2009                 rc = ioctl(dirfd(d), LL_IOC_MDC_GETINFO,
2010                            (void *)param->lmd);
2011         } else if (parent) {
2012                 char *fname = strrchr(path, '/');
2013                 fname = (fname == NULL ? path : fname + 1);
2014
2015                 strncpy((char *)param->lmd, fname, param->lumlen);
2016                 rc = ioctl(dirfd(parent), IOC_MDC_GETFILEINFO,
2017                            (void *)param->lmd);
2018         } else {
2019                 return 0;
2020         }
2021
2022         if (rc) {
2023                 if (errno == ENODATA) {
2024                         if (!param->obduuid && !param->quiet)
2025                                 llapi_err(LLAPI_MSG_ERROR,
2026                                           "%s has no stripe info", path);
2027                         rc = 0;
2028                 } else if (errno == ENOENT) {
2029                         llapi_err(LLAPI_MSG_ERROR,
2030                                   "warning: %s: %s does not exist",
2031                                   __FUNCTION__, path);
2032                         rc = 0;
2033                 } else if (errno != EISDIR) {
2034                         rc = errno;
2035                         llapi_err(LLAPI_MSG_ERROR, "%s ioctl failed for %s.",
2036                                   d ? "LL_IOC_MDC_GETINFO" :
2037                                   "IOC_MDC_GETFILEINFO", path);
2038                 }
2039                 return rc;
2040         }
2041
2042         st = &param->lmd->lmd_st;
2043
2044         /* libc chown() will do extra check, and if the real owner is
2045          * the same as the ones to set, it won't fall into kernel, so
2046          * invoke syscall directly. */
2047         rc = syscall(SYS_chown, path, -1, -1);
2048         if (rc)
2049                 llapi_err(LLAPI_MSG_ERROR,"error: chown %s (%u,%u)", path);
2050
2051         rc = chmod(path, st->st_mode);
2052         if (rc)
2053                 llapi_err(LLAPI_MSG_ERROR,"error: chmod %s (%hu)", path, st->st_mode);
2054
2055         return rc;
2056 }
2057
2058 int llapi_quotachown(char *path, int flag)
2059 {
2060         struct find_param param;
2061         char *buf;
2062         int ret = 0, len = strlen(path);
2063
2064         if (len > PATH_MAX) {
2065                 llapi_err(LLAPI_MSG_ERROR, "%s: Path name '%s' is too long",
2066                           __FUNCTION__, path);
2067                 return -EINVAL;
2068         }
2069
2070         buf = (char *)malloc(PATH_MAX + 1);
2071         if (!buf)
2072                 return -ENOMEM;
2073
2074         memset(&param, 0, sizeof(param));
2075         param.recursive = 1;
2076         param.verbose = 0;
2077         param.quiet = 1;
2078
2079         ret = common_param_init(&param);
2080         if (ret)
2081                 goto out;
2082
2083         strncpy(buf, path, PATH_MAX + 1);
2084         ret = llapi_semantic_traverse(buf, PATH_MAX + 1, NULL, cb_quotachown,
2085                                       NULL, &param, NULL);
2086 out:
2087         find_param_fini(&param);
2088         free(buf);
2089         return ret;
2090 }
2091
2092 #include <pwd.h>
2093 #include <grp.h>
2094 #include <mntent.h>
2095 #include <sys/wait.h>
2096 #include <errno.h>
2097 #include <ctype.h>
2098
2099 static int rmtacl_notify(int ops)
2100 {
2101         FILE *fp;
2102         struct mntent *mnt;
2103         int found = 0, fd, rc;
2104
2105         fp = setmntent(MOUNTED, "r");
2106         if (fp == NULL) {
2107                 perror("setmntent");
2108                 return -1;
2109         }
2110
2111         while (1) {
2112                 mnt = getmntent(fp);
2113                 if (!mnt)
2114                         break;
2115
2116                 if (!llapi_is_lustre_mnt(mnt))
2117                         continue;
2118
2119                 fd = open(mnt->mnt_dir, O_RDONLY | O_DIRECTORY);
2120                 if (fd < 0) {
2121                         perror("open");
2122                         return -1;
2123                 }
2124
2125                 rc = ioctl(fd, LL_IOC_RMTACL, ops);
2126                 if (rc < 0) {
2127                         perror("ioctl");
2128                 return -1;
2129         }
2130
2131                 found++;
2132         }
2133         endmntent(fp);
2134         return found;
2135 }
2136
2137 static char *next_token(char *p, int div)
2138 {
2139         if (p == NULL)
2140                 return NULL;
2141
2142         if (div)
2143                 while (*p && *p != ':' && !isspace(*p))
2144                         p++;
2145         else
2146                 while (*p == ':' || isspace(*p))
2147                         p++;
2148
2149         return *p ? p : NULL;
2150 }
2151
2152 static int rmtacl_name2id(char *name, int is_user)
2153 {
2154         if (is_user) {
2155                 struct passwd *pw;
2156
2157                 if ((pw = getpwnam(name)) == NULL)
2158                         return INVALID_ID;
2159                 else
2160                         return (int)(pw->pw_uid);
2161         } else {
2162                 struct group *gr;
2163
2164                 if ((gr = getgrnam(name)) == NULL)
2165                         return INVALID_ID;
2166                 else
2167                         return (int)(gr->gr_gid);
2168         }
2169 }
2170
2171 static int isodigit(int c)
2172 {
2173         return (c >= '0' && c <= '7') ? 1 : 0;
2174 }
2175
2176 /*
2177  * Whether the name is just digits string (uid/gid) already or not.
2178  * Return value:
2179  * 1: str is id
2180  * 0: str is not id
2181  */
2182 static int str_is_id(char *str)
2183 {
2184         if (str == NULL)
2185                 return 0;
2186
2187         if (*str == '0') {
2188                 str++;
2189                 if (*str == 'x' || *str == 'X') { /* for Hex. */
2190                         if (!isxdigit(*(++str)))
2191                                 return 0;
2192
2193                         while (isxdigit(*(++str)));
2194                 } else if (isodigit(*str)) { /* for Oct. */
2195                         while (isodigit(*(++str)));
2196                 }
2197         } else if (isdigit(*str)) { /* for Dec. */
2198                 while (isdigit(*(++str)));
2199         }
2200
2201         return (*str == 0) ? 1 : 0;
2202 }
2203
2204 typedef struct {
2205         char *name;
2206         int   length;
2207         int   is_user;
2208         int   next_token;
2209 } rmtacl_name_t;
2210
2211 #define RMTACL_OPTNAME(name) name, sizeof(name) - 1
2212
2213 static rmtacl_name_t rmtacl_namelist[] = {
2214         { RMTACL_OPTNAME("user:"),            1,      0 },
2215         { RMTACL_OPTNAME("group:"),           0,      0 },
2216         { RMTACL_OPTNAME("default:user:"),    1,      0 },
2217         { RMTACL_OPTNAME("default:group:"),   0,      0 },
2218         /* for --tabular option */
2219         { RMTACL_OPTNAME("user"),             1,      1 },
2220         { RMTACL_OPTNAME("group"),            0,      1 },
2221         { 0 }
2222 };
2223
2224 static int rgetfacl_output(char *str)
2225 {
2226         char *start = NULL, *end = NULL;
2227         int is_user = 0, n, id;
2228         char c;
2229         rmtacl_name_t *rn;
2230
2231         if (str == NULL)
2232                 return -1;
2233
2234         for (rn = rmtacl_namelist; rn->name; rn++) {
2235                 if(strncmp(str, rn->name, rn->length) == 0) {
2236                         if (!rn->next_token)
2237                                 start = str + rn->length;
2238                         else
2239                                 start = next_token(str + rn->length, 0);
2240                         is_user = rn->is_user;
2241                         break;
2242                 }
2243         }
2244
2245         end = next_token(start, 1);
2246         if (end == NULL || start == end) {
2247                 n = printf("%s", str);
2248                 return n;
2249         }
2250
2251         c = *end;
2252         *end = 0;
2253         id = rmtacl_name2id(start, is_user);
2254         if (id == INVALID_ID) {
2255                 if (str_is_id(start)) {
2256                         *end = c;
2257                         n = printf("%s", str);
2258                 } else
2259                         return -1;
2260         } else if ((id == NOBODY_UID && is_user) ||
2261                    (id == NOBODY_GID && !is_user)) {
2262                 *end = c;
2263                 n = printf("%s", str);
2264         } else {
2265                 *end = c;
2266                 *start = 0;
2267                 n = printf("%s%d%s", str, id, end);
2268         }
2269         return n;
2270 }
2271
2272 static int child_status(int status)
2273 {
2274         return WIFEXITED(status) ? WEXITSTATUS(status) : -1;
2275 }
2276
2277 static int do_rmtacl(int argc, char *argv[], int ops, int (output_func)(char *))
2278 {
2279         pid_t pid = 0;
2280         int fd[2], status;
2281         FILE *fp;
2282         char buf[PIPE_BUF];
2283
2284         if (output_func) {
2285                 if (pipe(fd) < 0) {
2286                         perror("pipe");
2287                         return -1;
2288                 }
2289
2290                 if ((pid = fork()) < 0) {
2291                         perror("fork");
2292                         close(fd[0]);
2293                         close(fd[1]);
2294                         return -1;
2295                 } else if (!pid) {
2296                         /* child process redirects its output. */
2297                         close(fd[0]);
2298                         close(1);
2299                         if (dup2(fd[1], 1) < 0) {
2300                                 perror("dup2");
2301                                 close(fd[1]);
2302                                 return -1;
2303                         }
2304                 } else {
2305                         close(fd[1]);
2306                 }
2307         }
2308
2309         if (!pid) {
2310                 status = rmtacl_notify(ops);
2311                 if (status < 0)
2312                         return -1;
2313
2314                 exit(execvp(argv[0], argv));
2315         }
2316
2317         /* the following is parent process */
2318         if ((fp = fdopen(fd[0], "r")) == NULL) {
2319                 perror("fdopen");
2320                 kill(pid, SIGKILL);
2321                 close(fd[0]);
2322                 return -1;
2323         }
2324
2325         while (fgets(buf, PIPE_BUF, fp) != NULL) {
2326                 if (output_func(buf) < 0)
2327                         fprintf(stderr, "WARNING: unexpected error!\n[%s]\n",
2328                                 buf);
2329         }
2330         fclose(fp);
2331         close(fd[0]);
2332
2333         if (waitpid(pid, &status, 0) < 0) {
2334                 perror("waitpid");
2335                 return -1;
2336         }
2337
2338         return child_status(status);
2339 }
2340
2341 int llapi_lsetfacl(int argc, char *argv[])
2342 {
2343         return do_rmtacl(argc, argv, RMT_LSETFACL, NULL);
2344 }
2345
2346 int llapi_lgetfacl(int argc, char *argv[])
2347 {
2348         return do_rmtacl(argc, argv, RMT_LGETFACL, NULL);
2349 }
2350
2351 int llapi_rsetfacl(int argc, char *argv[])
2352 {
2353         return do_rmtacl(argc, argv, RMT_RSETFACL, NULL);
2354 }
2355
2356 int llapi_rgetfacl(int argc, char *argv[])
2357 {
2358         return do_rmtacl(argc, argv, RMT_RGETFACL, rgetfacl_output);
2359 }
2360
2361 int llapi_cp(int argc, char *argv[])
2362 {
2363         int rc;
2364
2365         rc = rmtacl_notify(RMT_RSETFACL);
2366         if (rc < 0)
2367                 return -1;
2368
2369         exit(execvp(argv[0], argv));
2370 }
2371
2372 int llapi_ls(int argc, char *argv[])
2373 {
2374         int rc;
2375
2376         rc = rmtacl_notify(RMT_LGETFACL);
2377         if (rc < 0)
2378                 return -1;
2379
2380         exit(execvp(argv[0], argv));
2381 }