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