Whamcloud - gitweb
LU-14489 utils: fix 'lfs find --mdt-count'
[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, 2017, Intel Corporation.
27  */
28 /*
29  * This file is part of Lustre, http://www.lustre.org/
30  *
31  * lustre/utils/liblustreapi.c
32  *
33  * Author: Peter J. Braam <braam@clusterfs.com>
34  * Author: Phil Schwan <phil@clusterfs.com>
35  * Author: Robert Read <rread@clusterfs.com>
36  */
37
38 /* for O_DIRECTORY */
39 #ifndef _GNU_SOURCE
40 #define _GNU_SOURCE
41 #endif
42
43 #include <ctype.h>
44 #include <mntent.h>
45 #include <stdlib.h>
46 #include <stdio.h>
47 #include <string.h>
48 #include <stddef.h>
49 #include <sys/ioctl.h>
50 #include <unistd.h>
51 #include <fcntl.h>
52 #include <errno.h>
53 #include <dirent.h>
54 #include <stdarg.h>
55 #include <sys/stat.h>
56 #include <sys/statfs.h>
57 #include <sys/syscall.h>
58 #include <sys/time.h>
59 #include <sys/types.h>
60 #include <sys/xattr.h>
61 #include <sys/sysmacros.h>
62 #include <time.h>
63 #include <fnmatch.h>
64 #include <libgen.h> /* for dirname() */
65 #include <linux/limits.h>
66 #ifdef HAVE_LINUX_UNISTD_H
67 #include <linux/unistd.h>
68 #else
69 #include <unistd.h>
70 #endif
71 #include <poll.h>
72 #include <time.h>
73 #include <inttypes.h>
74
75 #include <libcfs/util/ioctl.h>
76 #include <libcfs/util/param.h>
77 #include <libcfs/util/string.h>
78 #include <linux/lnet/lnetctl.h>
79 #include <lustre/lustreapi.h>
80 #include <linux/lustre/lustre_ostid.h>
81 #include <linux/lustre/lustre_ioctl.h>
82 #include "lustreapi_internal.h"
83 #include "lstddef.h"
84
85 static int llapi_msg_level = LLAPI_MSG_MAX;
86 const char *liblustreapi_cmd;
87
88 struct lustre_foreign_type lu_foreign_types[] = {
89         {.lft_type = LU_FOREIGN_TYPE_NONE, .lft_name = "none"},
90         {.lft_type = LU_FOREIGN_TYPE_SYMLINK, .lft_name = "symlink"},
91         /* must be the last element */
92         {.lft_type = LU_FOREIGN_TYPE_UNKNOWN, .lft_name = NULL}
93         /* array max dimension must be <= UINT32_MAX */
94 };
95
96 void llapi_msg_set_level(int level)
97 {
98         /* ensure level is in the good range */
99         if (level < LLAPI_MSG_OFF)
100                 llapi_msg_level = LLAPI_MSG_OFF;
101         else if (level > LLAPI_MSG_MAX)
102                 llapi_msg_level = LLAPI_MSG_MAX;
103         else
104                 llapi_msg_level = level;
105 }
106
107 int llapi_msg_get_level(void)
108 {
109         return llapi_msg_level;
110 }
111
112 void llapi_set_command_name(const char *cmd)
113 {
114         liblustreapi_cmd = cmd;
115 }
116
117 void llapi_clear_command_name(void)
118 {
119         liblustreapi_cmd = NULL;
120 }
121
122 static void error_callback_default(enum llapi_message_level level, int err,
123                                    const char *fmt, va_list ap)
124 {
125         bool has_nl = strchr(fmt, '\n') != NULL;
126
127         if (liblustreapi_cmd != NULL)
128                 fprintf(stderr, "%s %s: ", program_invocation_short_name,
129                         liblustreapi_cmd);
130         else
131                 fprintf(stderr, "%s: ", program_invocation_short_name);
132
133
134         if (level & LLAPI_MSG_NO_ERRNO) {
135                 vfprintf(stderr, fmt, ap);
136                 if (!has_nl)
137                         fprintf(stderr, "\n");
138         } else {
139                 char *newfmt;
140
141                 /*
142                  * Remove trailing linefeed so error string can be appended.
143                  * @fmt is a const string, so we can't modify it directly.
144                  */
145                 if (has_nl && (newfmt = strdup(fmt)))
146                         *strrchr(newfmt, '\n') = '\0';
147                 else
148                         newfmt = (char *)fmt;
149
150                 vfprintf(stderr, newfmt, ap);
151                 if (newfmt != fmt)
152                         free(newfmt);
153                 fprintf(stderr, ": %s (%d)\n", strerror(err), err);
154         }
155 }
156
157 static void info_callback_default(enum llapi_message_level level, int err,
158                                   const char *fmt, va_list ap)
159 {
160         if (err != 0) {
161                 if (liblustreapi_cmd != NULL) {
162                         fprintf(stdout, "%s %s: ",
163                                 program_invocation_short_name,
164                                 liblustreapi_cmd);
165                 } else {
166                         fprintf(stdout, "%s: ", program_invocation_short_name);
167                 }
168         }
169         vfprintf(stdout, fmt, ap);
170 }
171
172 static llapi_log_callback_t llapi_error_callback = error_callback_default;
173 static llapi_log_callback_t llapi_info_callback = info_callback_default;
174
175
176 /* llapi_error will preserve errno */
177 void llapi_error(enum llapi_message_level level, int err, const char *fmt, ...)
178 {
179         va_list  args;
180         int      tmp_errno = errno;
181
182         if ((level & LLAPI_MSG_MASK) > llapi_msg_level)
183                 return;
184
185         va_start(args, fmt);
186         llapi_error_callback(level, abs(err), fmt, args);
187         va_end(args);
188         errno = tmp_errno;
189 }
190
191 /* llapi_printf will preserve errno */
192 void llapi_printf(enum llapi_message_level level, const char *fmt, ...)
193 {
194         va_list  args;
195         int      tmp_errno = errno;
196
197         if ((level & LLAPI_MSG_MASK) > llapi_msg_level)
198                 return;
199
200         va_start(args, fmt);
201         llapi_info_callback(level, 0, fmt, args);
202         va_end(args);
203         errno = tmp_errno;
204 }
205
206 /**
207  * Set a custom error logging function. Passing in NULL will reset the logging
208  * callback to its default value.
209  *
210  * This function returns the value of the old callback.
211  */
212 llapi_log_callback_t llapi_error_callback_set(llapi_log_callback_t cb)
213 {
214         llapi_log_callback_t    old = llapi_error_callback;
215
216         if (cb != NULL)
217                 llapi_error_callback = cb;
218         else
219                 llapi_error_callback = error_callback_default;
220
221         return old;
222 }
223
224 /**
225  * Set a custom info logging function. Passing in NULL will reset the logging
226  * callback to its default value.
227  *
228  * This function returns the value of the old callback.
229  */
230 llapi_log_callback_t llapi_info_callback_set(llapi_log_callback_t cb)
231 {
232         llapi_log_callback_t    old = llapi_info_callback;
233
234         if (cb != NULL)
235                 llapi_info_callback = cb;
236         else
237                 llapi_info_callback = info_callback_default;
238
239         return old;
240 }
241
242 /**
243  * size_units is to be initialized (or zeroed) by caller.
244  */
245 int llapi_parse_size(const char *optarg, unsigned long long *size,
246                      unsigned long long *size_units, int bytes_spec)
247 {
248         char *end;
249         char *argbuf = (char *)optarg;
250         unsigned long long frac = 0, frac_d = 1;
251
252         if (strncmp(optarg, "-", 1) == 0)
253                 return -1;
254
255         if (*size_units == 0)
256                 *size_units = 1;
257
258         *size = strtoull(argbuf, &end, 0);
259         if (end != NULL && *end == '.') {
260                 int i;
261
262                 argbuf = end + 1;
263                 frac = strtoull(argbuf, &end, 10);
264                 /* count decimal places */
265                 for (i = 0; i < (end - argbuf); i++)
266                         frac_d *= 10;
267         }
268
269         if (*end != '\0') {
270                 if ((*end == 'b') && *(end + 1) == '\0' &&
271                     (*size & (~0ULL << (64 - 9))) == 0 &&
272                     !bytes_spec) {
273                         *size_units = 1 << 9;
274                 } else if ((*end == 'b') &&
275                            *(end + 1) == '\0' &&
276                            bytes_spec) {
277                         *size_units = 1;
278                 } else if ((*end == 'k' || *end == 'K') &&
279                            *(end + 1) == '\0' &&
280                            (*size & (~0ULL << (64 - 10))) == 0) {
281                         *size_units = 1 << 10;
282                 } else if ((*end == 'm' || *end == 'M') &&
283                            *(end + 1) == '\0' &&
284                            (*size & (~0ULL << (64 - 20))) == 0) {
285                         *size_units = 1 << 20;
286                 } else if ((*end == 'g' || *end == 'G') &&
287                            *(end + 1) == '\0' &&
288                            (*size & (~0ULL << (64 - 30))) == 0) {
289                         *size_units = 1 << 30;
290                 } else if ((*end == 't' || *end == 'T') &&
291                            *(end + 1) == '\0' &&
292                            (*size & (~0ULL << (64 - 40))) == 0) {
293                         *size_units = 1ULL << 40;
294                 } else if ((*end == 'p' || *end == 'P') &&
295                            *(end + 1) == '\0' &&
296                            (*size & (~0ULL << (64 - 50))) == 0) {
297                         *size_units = 1ULL << 50;
298                 } else if ((*end == 'e' || *end == 'E') &&
299                            *(end + 1) == '\0' &&
300                            (*size & (~0ULL << (64 - 60))) == 0) {
301                         *size_units = 1ULL << 60;
302                 } else {
303                         return -1;
304                 }
305         }
306         *size = *size * *size_units + frac * *size_units / frac_d;
307
308         return 0;
309 }
310
311 int llapi_ioctl_pack(struct obd_ioctl_data *data, char **pbuf, int max_len)
312 {
313         struct obd_ioctl_data *overlay;
314         char *ptr;
315
316         data->ioc_len = obd_ioctl_packlen(data);
317         data->ioc_version = OBD_IOCTL_VERSION;
318
319         if (*pbuf != NULL && data->ioc_len > max_len) {
320                 llapi_error(LLAPI_MSG_ERROR, -EINVAL,
321                             "pbuf = %p, ioc_len = %u, max_len = %d",
322                             *pbuf, data->ioc_len, max_len);
323                 return -EINVAL;
324         }
325
326         if (*pbuf == NULL)
327                 *pbuf = malloc(data->ioc_len);
328
329         if (*pbuf == NULL)
330                 return -ENOMEM;
331
332         overlay = (struct obd_ioctl_data *)*pbuf;
333         memcpy(*pbuf, data, sizeof(*data));
334
335         ptr = overlay->ioc_bulk;
336         if (data->ioc_inlbuf1) {
337                 memcpy(ptr, data->ioc_inlbuf1, data->ioc_inllen1);
338                 ptr += __ALIGN_KERNEL(data->ioc_inllen1, 8);
339         }
340
341         if (data->ioc_inlbuf2) {
342                 memcpy(ptr, data->ioc_inlbuf2, data->ioc_inllen2);
343                 ptr += __ALIGN_KERNEL(data->ioc_inllen2, 8);
344         }
345
346         if (data->ioc_inlbuf3) {
347                 memcpy(ptr, data->ioc_inlbuf3, data->ioc_inllen3);
348                 ptr += __ALIGN_KERNEL(data->ioc_inllen3, 8);
349         }
350
351         if (data->ioc_inlbuf4) {
352                 memcpy(ptr, data->ioc_inlbuf4, data->ioc_inllen4);
353                 ptr += __ALIGN_KERNEL(data->ioc_inllen4, 8);
354         }
355
356         return 0;
357 }
358
359 int llapi_ioctl_unpack(struct obd_ioctl_data *data, char *pbuf, int max_len)
360 {
361         struct obd_ioctl_data *overlay;
362         char *ptr;
363
364         if (pbuf == NULL)
365                 return 1;
366
367         overlay = (struct obd_ioctl_data *)pbuf;
368
369         /* Preserve the caller's buffer pointers */
370         overlay->ioc_inlbuf1 = data->ioc_inlbuf1;
371         overlay->ioc_inlbuf2 = data->ioc_inlbuf2;
372         overlay->ioc_inlbuf3 = data->ioc_inlbuf3;
373         overlay->ioc_inlbuf4 = data->ioc_inlbuf4;
374
375         memcpy(data, pbuf, sizeof(*data));
376
377         ptr = overlay->ioc_bulk;
378         if (data->ioc_inlbuf1) {
379                 memcpy(data->ioc_inlbuf1, ptr, data->ioc_inllen1);
380                 ptr += __ALIGN_KERNEL(data->ioc_inllen1, 8);
381         }
382
383         if (data->ioc_inlbuf2) {
384                 memcpy(data->ioc_inlbuf2, ptr, data->ioc_inllen2);
385                 ptr += __ALIGN_KERNEL(data->ioc_inllen2, 8);
386         }
387
388         if (data->ioc_inlbuf3) {
389                 memcpy(data->ioc_inlbuf3, ptr, data->ioc_inllen3);
390                 ptr += __ALIGN_KERNEL(data->ioc_inllen3, 8);
391         }
392
393         if (data->ioc_inlbuf4) {
394                 memcpy(data->ioc_inlbuf4, ptr, data->ioc_inllen4);
395                 ptr += __ALIGN_KERNEL(data->ioc_inllen4, 8);
396         }
397
398         return 0;
399 }
400
401 /* XXX: llapi_xxx() functions return negative values upon failure */
402
403 int llapi_layout_search_ost(__u32 ost, char *pname, char *fsname)
404 {
405         char ostname[MAX_OBD_NAME + 64];
406         char *pool_name = pname;
407         int rc = 0;
408
409         /**
410          * The current policy is that the pool does not have to exist at the
411          * setstripe time, see sanity-pfl/-flr tests.
412          * If this logic will change, re-enable it.
413          *
414          * if (pname && strlen(pname) == 0)
415          */
416                 pool_name = NULL;
417
418         snprintf(ostname, sizeof(ostname), "%s-OST%04x_UUID",
419                  fsname, ost);
420         rc = llapi_search_ost(fsname, pool_name, ostname);
421         if (rc <= 0) {
422                 if (rc == 0)
423                         rc = -ENODEV;
424
425                 llapi_error(LLAPI_MSG_ERROR, rc,
426                             "%s: cannot find OST %s in %s", __func__, ostname,
427                             pool_name != NULL ? "pool" : "system");
428                 return rc;
429         }
430
431         return 0;
432 }
433
434 /**
435  * Verify the setstripe parameters before using.
436  * This is a pair method for comp_args_to_layout()/llapi_layout_sanity_cb()
437  * when just 1 component or a non-PFL layout is given.
438  *
439  * \param[in] param             stripe parameters
440  * \param[in] pool_name         pool name
441  * \param[in] fsname            lustre FS name
442  *
443  * \retval                      0, success
444  *                              < 0, error code on failre
445  */
446 static int llapi_stripe_param_verify(const struct llapi_stripe_param *param,
447                                      char **pool_name,
448                                      char *fsname)
449 {
450         int count;
451         static int page_size;
452         int rc = 0;
453
454         if (page_size == 0) {
455                 /*
456                  * 64 KB is the largest common page size (on ia64/PPC/ARM),
457                  * but check the local page size just in case. The page_size
458                  * will not change for the lifetime of this process at least.
459                  */
460                 page_size = LOV_MIN_STRIPE_SIZE;
461                 if (getpagesize() > page_size) {
462                         page_size = getpagesize();
463                         llapi_err_noerrno(LLAPI_MSG_WARN,
464                                           "warning: page size (%u) larger than expected (%u)",
465                                           page_size, LOV_MIN_STRIPE_SIZE);
466                 }
467         }
468         if (!llapi_stripe_size_is_aligned(param->lsp_stripe_size)) {
469                 rc = -EINVAL;
470                 llapi_error(LLAPI_MSG_ERROR, rc,
471                             "error: bad stripe_size %llu, must be an even multiple of %d bytes",
472                             param->lsp_stripe_size, page_size);
473                 goto out;
474         }
475         if (!llapi_stripe_index_is_valid(param->lsp_stripe_offset)) {
476                 rc = -EINVAL;
477                 llapi_error(LLAPI_MSG_ERROR, rc, "error: bad stripe offset %d",
478                             param->lsp_stripe_offset);
479                 goto out;
480         }
481         if (llapi_stripe_size_is_too_big(param->lsp_stripe_size)) {
482                 rc = -EINVAL;
483                 llapi_error(LLAPI_MSG_ERROR, rc,
484                             "error: stripe size '%llu' over 4GB limit",
485                             param->lsp_stripe_size);
486                 goto out;
487         }
488
489         count = param->lsp_stripe_count;
490         if (param->lsp_stripe_pattern == LOV_PATTERN_MDT) {
491                 rc = -EINVAL;
492                 llapi_error(LLAPI_MSG_ERROR, rc,
493                             "Invalid pattern: %d, must be specified with -E\n",
494                             param->lsp_stripe_pattern);
495                 goto out;
496         } else {
497                 if (!llapi_stripe_count_is_valid(count)) {
498                         rc = -EINVAL;
499                         llapi_error(LLAPI_MSG_ERROR, rc,
500                                     "Invalid stripe count %d\n", count);
501                         goto out;
502                 }
503         }
504
505         /* Make sure we have a good pool */
506         if (*pool_name != NULL) {
507                 if (!llapi_pool_name_is_valid(pool_name, fsname)) {
508                         rc = -EINVAL;
509                         llapi_error(LLAPI_MSG_ERROR, rc,
510                                     "Pool '%s' is not on filesystem '%s'",
511                                     *pool_name, fsname);
512                         goto out;
513                 }
514
515                 /* Make sure the pool exists and is non-empty */
516                 rc = llapi_search_ost(fsname, *pool_name, NULL);
517                 if (rc < 1) {
518                         char *err = rc == 0 ? "has no OSTs" : "does not exist";
519
520                         rc = -EINVAL;
521                         llapi_error(LLAPI_MSG_ERROR, rc, "pool '%s.%s' %s",
522                                     fsname, *pool_name, err);
523                         goto out;
524                 }
525                 rc = 0;
526         }
527
528         /* sanity check of target list */
529         if (param->lsp_is_specific) {
530                 bool found = false;
531                 int i;
532
533                 for (i = 0; i < count; i++) {
534                         rc = llapi_layout_search_ost(param->lsp_osts[i],
535                                                      *pool_name, fsname);
536                         if (rc)
537                                 goto out;
538
539                         /* Make sure stripe offset is in OST list. */
540                         if (param->lsp_osts[i] == param->lsp_stripe_offset)
541                                 found = true;
542                 }
543                 if (!found) {
544                         rc = -EINVAL;
545                         llapi_error(LLAPI_MSG_ERROR, rc,
546                                     "%s: stripe offset '%d' is not in the target list",
547                                     __func__, param->lsp_stripe_offset);
548                         goto out;
549                 }
550         } else if (param->lsp_stripe_offset != -1) {
551                 rc = llapi_layout_search_ost(param->lsp_stripe_offset,
552                                              *pool_name, fsname);
553                 if (rc)
554                         goto out;
555         }
556 out:
557         errno = -rc;
558         return rc;
559 }
560
561 int llapi_dir_stripe_limit_check(int stripe_offset, int stripe_count,
562                                  int hash_type)
563 {
564         int rc;
565
566         if (!llapi_dir_stripe_index_is_valid(stripe_offset)) {
567                 rc = -EINVAL;
568                 llapi_error(LLAPI_MSG_ERROR, rc, "error: bad stripe offset %d",
569                                 stripe_offset);
570                 return rc;
571         }
572         if (!llapi_dir_stripe_count_is_valid(stripe_count)) {
573                 rc = -EINVAL;
574                 llapi_error(LLAPI_MSG_ERROR, rc, "error: bad stripe count %d",
575                                 stripe_count);
576                 return rc;
577         }
578
579         if (!llapi_dir_hash_type_is_valid(hash_type)) {
580                 rc = -EINVAL;
581                 llapi_error(LLAPI_MSG_ERROR, rc, "error: bad hash type %d",
582                                 hash_type);
583                 return rc;
584         }
585         return 0;
586 }
587
588 /*
589  * Trim a trailing newline from a string, if it exists.
590  */
591 int llapi_chomp_string(char *buf)
592 {
593         if (!buf || !*buf)
594                 return 0;
595
596         while (buf[1])
597                 buf++;
598
599         if (*buf != '\n')
600                 return 0;
601
602         *buf = '\0';
603         return '\n';
604 }
605
606 /*
607  * Wrapper to grab parameter settings for lov.*-clilov-*.* values
608  */
609 static int get_param_lov(const char *path, const char *param,
610                          char *buf, size_t buf_size)
611 {
612         struct obd_uuid uuid;
613         int rc;
614
615         rc = llapi_file_get_lov_uuid(path, &uuid);
616         if (rc != 0)
617                 return rc;
618
619         return get_lustre_param_value("lov", uuid.uuid, FILTER_BY_EXACT, param,
620                                       buf, buf_size);
621 }
622
623 /*
624  * Wrapper to grab parameter settings for lmv.*-clilov-*.* values
625  */
626 static int get_param_lmv(const char *path, const char *param,
627                          char *buf, size_t buf_size)
628 {
629         struct obd_uuid uuid;
630         int rc;
631
632         rc = llapi_file_get_lmv_uuid(path, &uuid);
633         if (rc != 0)
634                 return rc;
635
636         return get_lustre_param_value("lmv", uuid.uuid, FILTER_BY_EXACT, param,
637                                buf, buf_size);
638 }
639
640 static int get_mds_md_size(const char *path)
641 {
642         int md_size = lov_user_md_size(LOV_MAX_STRIPE_COUNT, LOV_USER_MAGIC_V3);
643
644         /*
645          * Rather than open the file and do the ioctl to get the
646          * instance name and close the file and search for the param
647          * file and open the param file and read the param file and
648          * parse the value and close the param file, let's just return
649          * a large enough value. It's 2020, RAM is cheap and this is
650          * much faster.
651          */
652
653         if (md_size < XATTR_SIZE_MAX)
654                 md_size = XATTR_SIZE_MAX;
655
656         return md_size;
657 }
658
659 int llapi_get_agent_uuid(char *path, char *buf, size_t bufsize)
660 {
661         return get_param_lmv(path, "uuid", buf, bufsize);
662 }
663
664 /**
665  * Open a Lustre file.
666  *
667  * \param name     the name of the file to be opened
668  * \param flags    access mode, see flags in open(2)
669  * \param mode     permission of the file if it is created, see mode in open(2)
670  * \param param    stripe pattern of the newly created file
671  *
672  * \retval         file descriptor of opened file
673  * \retval         negative errno on failure
674  */
675 int llapi_file_open_param(const char *name, int flags, mode_t mode,
676                           const struct llapi_stripe_param *param)
677 {
678         char fsname[MAX_OBD_NAME + 1] = { 0 };
679         struct lov_user_md *lum = NULL;
680         char *pool_name = param->lsp_pool;
681         size_t lum_size;
682         int fd, rc;
683
684         /* Make sure we are on a Lustre file system */
685         rc = llapi_search_fsname(name, fsname);
686         if (rc) {
687                 llapi_error(LLAPI_MSG_ERROR, rc,
688                             "'%s' is not on a Lustre filesystem",
689                             name);
690                 return rc;
691         }
692
693         /* Check if the stripe pattern is sane. */
694         rc = llapi_stripe_param_verify(param, &pool_name, fsname);
695         if (rc != 0)
696                 return rc;
697
698         if (param->lsp_is_specific)
699                 lum_size = lov_user_md_size(param->lsp_stripe_count,
700                                             LOV_USER_MAGIC_SPECIFIC);
701         else if (pool_name)
702                 lum_size = sizeof(struct lov_user_md_v3);
703         else
704                 lum_size = sizeof(*lum);
705
706         lum = calloc(1, lum_size);
707         if (lum == NULL)
708                 return -ENOMEM;
709
710 retry_open:
711         fd = open(name, flags | O_LOV_DELAY_CREATE, mode);
712         if (fd < 0) {
713                 if (errno == EISDIR && !(flags & O_DIRECTORY)) {
714                         flags = O_DIRECTORY | O_RDONLY;
715                         goto retry_open;
716                 }
717         }
718
719         if (fd < 0) {
720                 rc = -errno;
721                 llapi_error(LLAPI_MSG_ERROR, rc, "unable to open '%s'", name);
722                 free(lum);
723                 return rc;
724         }
725
726         /*  Initialize IOCTL striping pattern structure */
727         lum->lmm_magic = LOV_USER_MAGIC_V1;
728         lum->lmm_pattern = param->lsp_stripe_pattern;
729         lum->lmm_stripe_size = param->lsp_stripe_size;
730         lum->lmm_stripe_count = param->lsp_stripe_count;
731         lum->lmm_stripe_offset = param->lsp_stripe_offset;
732         if (pool_name != NULL) {
733                 struct lov_user_md_v3 *lumv3 = (void *)lum;
734
735                 lumv3->lmm_magic = LOV_USER_MAGIC_V3;
736                 strncpy(lumv3->lmm_pool_name, pool_name, LOV_MAXPOOLNAME);
737         }
738         if (param->lsp_is_specific) {
739                 struct lov_user_md_v3 *lumv3 = (void *)lum;
740                 int i;
741
742                 lumv3->lmm_magic = LOV_USER_MAGIC_SPECIFIC;
743                 if (pool_name == NULL) {
744                         /*
745                          * LOV_USER_MAGIC_SPECIFIC uses v3 format plus specified
746                          * OST list, therefore if pool is not specified we have
747                          * to pack a null pool name for placeholder.
748                          */
749                         memset(lumv3->lmm_pool_name, 0, LOV_MAXPOOLNAME);
750                 }
751
752                 for (i = 0; i < param->lsp_stripe_count; i++)
753                         lumv3->lmm_objects[i].l_ost_idx = param->lsp_osts[i];
754         }
755
756         if (ioctl(fd, LL_IOC_LOV_SETSTRIPE, lum) != 0) {
757                 char errmsg[512] = "stripe already set";
758
759                 rc = -errno;
760                 if (errno != EEXIST && errno != EALREADY)
761                         strncpy(errmsg, strerror(errno), sizeof(errmsg) - 1);
762                 if (rc == -EREMOTEIO)
763                         snprintf(errmsg, sizeof(errmsg),
764                                  "inactive OST among your specified %d OST(s)",
765                                  param->lsp_stripe_count);
766
767                 llapi_err_noerrno(LLAPI_MSG_ERROR,
768                                   "setstripe error for '%s': %s", name, errmsg);
769
770                 close(fd);
771                 fd = rc;
772         }
773
774         free(lum);
775
776         return fd;
777 }
778
779 int llapi_file_open_pool(const char *name, int flags, int mode,
780                          unsigned long long stripe_size, int stripe_offset,
781                          int stripe_count, int stripe_pattern, char *pool_name)
782 {
783         const struct llapi_stripe_param param = {
784                 .lsp_stripe_size = stripe_size,
785                 .lsp_stripe_count = stripe_count,
786                 .lsp_stripe_pattern = stripe_pattern,
787                 .lsp_stripe_offset = stripe_offset,
788                 .lsp_pool = pool_name
789         };
790         return llapi_file_open_param(name, flags, mode, &param);
791 }
792
793 int llapi_file_open(const char *name, int flags, int mode,
794                     unsigned long long stripe_size, int stripe_offset,
795                     int stripe_count, int stripe_pattern)
796 {
797         return llapi_file_open_pool(name, flags, mode, stripe_size,
798                                     stripe_offset, stripe_count,
799                                     stripe_pattern, NULL);
800 }
801
802 int llapi_file_create_foreign(const char *name, mode_t mode, __u32 type,
803                               __u32 flags, char *foreign_lov)
804 {
805         size_t len;
806         struct lov_foreign_md *lfm;
807         int fd, rc;
808
809         if (foreign_lov == NULL) {
810                 rc = -EINVAL;
811                 llapi_error(LLAPI_MSG_ERROR, rc,
812                             "foreign LOV EA content must be provided");
813                 goto out_err;
814         }
815
816         len = strlen(foreign_lov);
817         if (len > XATTR_SIZE_MAX - offsetof(struct lov_foreign_md, lfm_value) ||
818             len <= 0) {
819                 rc = -EINVAL;
820                 llapi_error(LLAPI_MSG_ERROR, rc,
821                             "foreign LOV EA size %zu (must be 0 < len < %zu)",
822                             len, XATTR_SIZE_MAX -
823                             offsetof(struct lov_foreign_md, lfm_value));
824                 goto out_err;
825         }
826
827         lfm = malloc(len + offsetof(struct lov_foreign_md, lfm_value));
828         if (lfm == NULL) {
829                 rc = -ENOMEM;
830                 llapi_error(LLAPI_MSG_ERROR, rc,
831                             "failed to allocate lov_foreign_md");
832                 goto out_err;
833         }
834
835         fd = open(name, O_WRONLY|O_CREAT|O_LOV_DELAY_CREATE, mode);
836         if (fd == -1) {
837                 fd = -errno;
838                 llapi_error(LLAPI_MSG_ERROR, fd, "open '%s' failed", name);
839                 goto out_free;
840         }
841
842         lfm->lfm_magic = LOV_USER_MAGIC_FOREIGN;
843         lfm->lfm_length = len;
844         lfm->lfm_type = type;
845         lfm->lfm_flags = flags;
846         memcpy(lfm->lfm_value, foreign_lov, len);
847
848         if (ioctl(fd, LL_IOC_LOV_SETSTRIPE, lfm) != 0) {
849                 char *errmsg = "stripe already set";
850
851                 rc = -errno;
852                 if (errno == ENOTTY)
853                         errmsg = "not on a Lustre filesystem";
854                 else if (errno == EEXIST || errno == EALREADY)
855                         errmsg = "stripe already set";
856                 else
857                         errmsg = strerror(errno);
858
859                 llapi_err_noerrno(LLAPI_MSG_ERROR,
860                                   "setstripe error for '%s': %s", name, errmsg);
861
862                 close(fd);
863                 fd = rc;
864         }
865
866 out_free:
867         free(lfm);
868
869         return fd;
870
871 out_err:
872         errno = -rc;
873         return rc;
874 }
875
876 int llapi_file_create(const char *name, unsigned long long stripe_size,
877                       int stripe_offset, int stripe_count, int stripe_pattern)
878 {
879         int fd;
880
881         fd = llapi_file_open_pool(name, O_CREAT | O_WRONLY, 0644, stripe_size,
882                                   stripe_offset, stripe_count, stripe_pattern,
883                                   NULL);
884         if (fd < 0)
885                 return fd;
886
887         close(fd);
888         return 0;
889 }
890
891 int llapi_file_create_pool(const char *name, unsigned long long stripe_size,
892                            int stripe_offset, int stripe_count,
893                            int stripe_pattern, char *pool_name)
894 {
895         int fd;
896
897         fd = llapi_file_open_pool(name, O_CREAT | O_WRONLY, 0644, stripe_size,
898                                   stripe_offset, stripe_count, stripe_pattern,
899                                   pool_name);
900         if (fd < 0)
901                 return fd;
902
903         close(fd);
904         return 0;
905 }
906
907 static int verify_dir_param(const char *name,
908                             const struct llapi_stripe_param *param)
909 {
910         char fsname[MAX_OBD_NAME + 1] = { 0 };
911         char *pool_name = param->lsp_pool;
912         int rc;
913
914         /* Make sure we are on a Lustre file system */
915         rc = llapi_search_fsname(name, fsname);
916         if (rc) {
917                 llapi_error(LLAPI_MSG_ERROR, rc,
918                             "'%s' is not on a Lustre filesystem",
919                             name);
920                 return rc;
921         }
922
923         /* Check if the stripe pattern is sane. */
924         rc = llapi_dir_stripe_limit_check(param->lsp_stripe_offset,
925                                           param->lsp_stripe_count,
926                                           param->lsp_stripe_pattern);
927         if (rc != 0)
928                 return rc;
929
930         /* Make sure we have a good pool */
931         if (pool_name != NULL) {
932                 /*
933                  * in case user gives the full pool name <fsname>.<poolname>,
934                  * strip the fsname
935                  */
936                 char *ptr = strchr(pool_name, '.');
937
938                 if (ptr != NULL) {
939                         *ptr = '\0';
940                         if (strcmp(pool_name, fsname) != 0) {
941                                 *ptr = '.';
942                                 llapi_err_noerrno(LLAPI_MSG_ERROR,
943                                         "Pool '%s' is not on filesystem '%s'",
944                                         pool_name, fsname);
945                                 return -EINVAL;
946                         }
947                         pool_name = ptr + 1;
948                 }
949
950                 /* Make sure the pool exists and is non-empty */
951                 rc = llapi_search_tgt(fsname, pool_name, NULL, true);
952                 if (rc < 1) {
953                         char *err = rc == 0 ? "has no OSTs" : "does not exist";
954
955                         llapi_err_noerrno(LLAPI_MSG_ERROR, "pool '%s.%s' %s",
956                                           fsname, pool_name, err);
957                         return -EINVAL;
958                 }
959         }
960
961         /* sanity check of target list */
962         if (param->lsp_is_specific) {
963                 char mdtname[MAX_OBD_NAME + 64];
964                 bool found = false;
965                 int i;
966
967                 for (i = 0; i < param->lsp_stripe_count; i++) {
968                         snprintf(mdtname, sizeof(mdtname), "%s-MDT%04x_UUID",
969                                  fsname, param->lsp_tgts[i]);
970                         rc = llapi_search_tgt(fsname, pool_name, mdtname, true);
971                         if (rc <= 0) {
972                                 if (rc == 0)
973                                         rc = -ENODEV;
974
975                                 llapi_error(LLAPI_MSG_ERROR, rc,
976                                             "%s: cannot find MDT %s in %s",
977                                             __func__, mdtname,
978                                             pool_name != NULL ?
979                                             "pool" : "system");
980                                 return rc;
981                         }
982
983                         /* Make sure stripe offset is in MDT list. */
984                         if (param->lsp_tgts[i] == param->lsp_stripe_offset)
985                                 found = true;
986                 }
987                 if (!found) {
988                         llapi_error(LLAPI_MSG_ERROR, -EINVAL,
989                                     "%s: stripe offset '%d' is not in the target list",
990                                     __func__, param->lsp_stripe_offset);
991                         return -EINVAL;
992                 }
993         }
994
995         return 0;
996 }
997
998 static inline void param2lmu(struct lmv_user_md *lmu,
999                              const struct llapi_stripe_param *param)
1000 {
1001         lmu->lum_magic = param->lsp_is_specific ? LMV_USER_MAGIC_SPECIFIC :
1002                                                   LMV_USER_MAGIC;
1003         lmu->lum_stripe_count = param->lsp_stripe_count;
1004         lmu->lum_stripe_offset = param->lsp_stripe_offset;
1005         lmu->lum_hash_type = param->lsp_stripe_pattern;
1006         lmu->lum_max_inherit = param->lsp_max_inherit;
1007         lmu->lum_max_inherit_rr = param->lsp_max_inherit_rr;
1008         if (param->lsp_pool != NULL)
1009                 strncpy(lmu->lum_pool_name, param->lsp_pool, LOV_MAXPOOLNAME);
1010         if (param->lsp_is_specific) {
1011                 int i;
1012
1013                 for (i = 0; i < param->lsp_stripe_count; i++)
1014                         lmu->lum_objects[i].lum_mds = param->lsp_tgts[i];
1015         }
1016 }
1017
1018 int llapi_dir_set_default_lmv(const char *name,
1019                               const struct llapi_stripe_param *param)
1020 {
1021         struct lmv_user_md lmu = { 0 };
1022         int fd;
1023         int rc = 0;
1024
1025         rc = verify_dir_param(name, param);
1026         if (rc)
1027                 return rc;
1028
1029         /* TODO: default lmv doesn't support specific targets yet */
1030         if (param->lsp_is_specific)
1031                 return -EINVAL;
1032
1033         param2lmu(&lmu, param);
1034
1035         fd = open(name, O_DIRECTORY | O_RDONLY);
1036         if (fd < 0) {
1037                 rc = -errno;
1038                 llapi_error(LLAPI_MSG_ERROR, rc, "unable to open '%s'", name);
1039                 return rc;
1040         }
1041
1042         rc = ioctl(fd, LL_IOC_LMV_SET_DEFAULT_STRIPE, &lmu);
1043         if (rc < 0) {
1044                 char *errmsg = "stripe already set";
1045
1046                 rc = -errno;
1047                 if (errno != EEXIST && errno != EALREADY)
1048                         errmsg = strerror(errno);
1049
1050                 llapi_err_noerrno(LLAPI_MSG_ERROR,
1051                                   "default dirstripe error on '%s': %s",
1052                                   name, errmsg);
1053         }
1054         close(fd);
1055         return rc;
1056 }
1057
1058 int llapi_dir_set_default_lmv_stripe(const char *name, int stripe_offset,
1059                                      int stripe_count, int stripe_pattern,
1060                                      const char *pool_name)
1061 {
1062         const struct llapi_stripe_param param = {
1063                 .lsp_stripe_count = stripe_count,
1064                 .lsp_stripe_offset = stripe_offset,
1065                 .lsp_stripe_pattern = stripe_pattern,
1066                 .lsp_pool = (char *)pool_name
1067         };
1068
1069         return llapi_dir_set_default_lmv(name, &param);
1070 }
1071
1072 /**
1073  * Create a Lustre directory.
1074  *
1075  * \param name     the name of the directory to be created
1076  * \param mode     permission of the file if it is created, see mode in open(2)
1077  * \param param    stripe pattern of the newly created directory
1078  *
1079  * \retval         0 on success
1080  * \retval         negative errno on failure
1081  */
1082 int llapi_dir_create(const char *name, mode_t mode,
1083                      const struct llapi_stripe_param *param)
1084 {
1085         struct lmv_user_md *lmu = NULL;
1086         size_t lmu_size;
1087         struct obd_ioctl_data data = { 0 };
1088         char rawbuf[8192];
1089         char *buf = rawbuf;
1090         char *dirpath = NULL;
1091         char *namepath = NULL;
1092         char *dir;
1093         char *filename;
1094         int fd, rc;
1095
1096         rc = verify_dir_param(name, param);
1097         if (rc)
1098                 return rc;
1099
1100         lmu_size = lmv_user_md_size(param->lsp_stripe_count,
1101                                     param->lsp_is_specific ?
1102                                          LMV_USER_MAGIC_SPECIFIC :
1103                                          LMV_USER_MAGIC);
1104
1105         lmu = calloc(1, lmu_size);
1106         if (lmu == NULL)
1107                 return -ENOMEM;
1108
1109         dirpath = strdup(name);
1110         if (!dirpath) {
1111                 free(lmu);
1112                 return -ENOMEM;
1113         }
1114
1115         namepath = strdup(name);
1116         if (!namepath) {
1117                 free(dirpath);
1118                 free(lmu);
1119                 return -ENOMEM;
1120         }
1121
1122         param2lmu(lmu, param);
1123
1124         filename = basename(namepath);
1125         dir = dirname(dirpath);
1126
1127         data.ioc_inlbuf1 = (char *)filename;
1128         data.ioc_inllen1 = strlen(filename) + 1;
1129         data.ioc_inlbuf2 = (char *)lmu;
1130         data.ioc_inllen2 = lmu_size;
1131         data.ioc_type = mode;
1132         if (param->lsp_is_create)
1133                 /* borrow obdo1.o_flags to store this flag */
1134                 data.ioc_obdo1.o_flags = OBD_FL_OBDMDEXISTS;
1135         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
1136         if (rc) {
1137                 llapi_error(LLAPI_MSG_ERROR, rc,
1138                             "error: LL_IOC_LMV_SETSTRIPE pack failed '%s'.",
1139                             name);
1140                 goto out;
1141         }
1142
1143         fd = open(dir, O_DIRECTORY | O_RDONLY);
1144         if (fd < 0) {
1145                 rc = -errno;
1146                 llapi_error(LLAPI_MSG_ERROR, rc, "unable to open '%s'", name);
1147                 goto out;
1148         }
1149
1150         if (ioctl(fd, LL_IOC_LMV_SETSTRIPE, buf)) {
1151                 char *errmsg = "stripe already set";
1152
1153                 rc = -errno;
1154                 if (errno != EEXIST && errno != EALREADY)
1155                         errmsg = strerror(errno);
1156
1157                 llapi_err_noerrno(LLAPI_MSG_ERROR,
1158                                   "dirstripe error on '%s': %s", name, errmsg);
1159         }
1160         close(fd);
1161 out:
1162         free(namepath);
1163         free(dirpath);
1164         free(lmu);
1165         return rc;
1166 }
1167
1168 /**
1169  * Create a foreign directory.
1170  *
1171  * \param name     the name of the directory to be created
1172  * \param mode     permission of the file if it is created, see mode in open(2)
1173  * \param type     foreign type to be set in LMV EA
1174  * \param flags    foreign flags to be set in LMV EA
1175  * \param value    foreign pattern to be set in LMV EA
1176  *
1177  * \retval         0 on success
1178  * \retval         negative errno on failure
1179  */
1180 int llapi_dir_create_foreign(const char *name, mode_t mode, __u32 type,
1181                              __u32 flags, const char *value)
1182 {
1183         struct lmv_foreign_md *lfm = NULL;
1184         size_t lfm_size, len;
1185         struct obd_ioctl_data data = { 0 };
1186         char rawbuf[8192];
1187         char *buf = rawbuf;
1188         char *dirpath = NULL;
1189         char *namepath = NULL;
1190         char *dir;
1191         char *filename;
1192         int fd, rc;
1193
1194         len = strlen(value);
1195         if (len > XATTR_SIZE_MAX - offsetof(struct lmv_foreign_md, lfm_value) ||
1196             len <= 0) {
1197                 rc = -EINVAL;
1198                 llapi_error(LLAPI_MSG_ERROR, rc,
1199                             "invalid LOV EA length %zu (must be 0 < len < %zu)",
1200                             len, XATTR_SIZE_MAX -
1201                             offsetof(struct lmv_foreign_md, lfm_value));
1202                 return rc;
1203         }
1204         lfm_size = len + offsetof(struct lmv_foreign_md, lfm_value);
1205         lfm = calloc(1, lfm_size);
1206         if (lfm == NULL)
1207                 return -ENOMEM;
1208
1209         dirpath = strdup(name);
1210         if (!dirpath) {
1211                 free(lfm);
1212                 return -ENOMEM;
1213         }
1214
1215         namepath = strdup(name);
1216         if (!namepath) {
1217                 free(dirpath);
1218                 free(lfm);
1219                 return -ENOMEM;
1220         }
1221
1222         lfm->lfm_magic = LMV_MAGIC_FOREIGN;
1223         lfm->lfm_length = len;
1224         lfm->lfm_type = type;
1225         lfm->lfm_flags = flags;
1226         memcpy(lfm->lfm_value, value, len);
1227
1228         filename = basename(namepath);
1229         dir = dirname(dirpath);
1230
1231         data.ioc_inlbuf1 = (char *)filename;
1232         data.ioc_inllen1 = strlen(filename) + 1;
1233         data.ioc_inlbuf2 = (char *)lfm;
1234         data.ioc_inllen2 = lfm_size;
1235         data.ioc_type = mode;
1236         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
1237         if (rc) {
1238                 llapi_error(LLAPI_MSG_ERROR, rc,
1239                             "error: LL_IOC_LMV_SETSTRIPE pack failed '%s'.",
1240                             name);
1241                 goto out;
1242         }
1243
1244         fd = open(dir, O_DIRECTORY | O_RDONLY);
1245         if (fd < 0) {
1246                 rc = -errno;
1247                 llapi_error(LLAPI_MSG_ERROR, rc, "unable to open '%s'", name);
1248                 goto out;
1249         }
1250
1251         if (ioctl(fd, LL_IOC_LMV_SETSTRIPE, buf)) {
1252                 char *errmsg = "stripe already set";
1253
1254                 rc = -errno;
1255                 if (errno != EEXIST && errno != EALREADY)
1256                         errmsg = strerror(errno);
1257
1258                 llapi_err_noerrno(LLAPI_MSG_ERROR,
1259                                   "dirstripe error on '%s': %s", name, errmsg);
1260         }
1261         close(fd);
1262 out:
1263         free(namepath);
1264         free(dirpath);
1265         free(lfm);
1266         return rc;
1267 }
1268
1269 int llapi_dir_create_pool(const char *name, int mode, int stripe_offset,
1270                           int stripe_count, int stripe_pattern,
1271                           const char *pool_name)
1272 {
1273         const struct llapi_stripe_param param = {
1274                 .lsp_stripe_count = stripe_count,
1275                 .lsp_stripe_offset = stripe_offset,
1276                 .lsp_stripe_pattern = stripe_pattern,
1277                 .lsp_pool = (char *)pool_name
1278         };
1279
1280         return llapi_dir_create(name, mode, &param);
1281 }
1282
1283 /*
1284  * Find the fsname, the full path, and/or an open fd.
1285  * Either the fsname or path must not be NULL
1286  */
1287 int get_root_path(int want, char *fsname, int *outfd, char *path, int index)
1288 {
1289         struct mntent mnt;
1290         char buf[PATH_MAX], mntdir[PATH_MAX];
1291         char *ptr, *ptr_end;
1292         FILE *fp;
1293         int idx = 0, mntlen = 0, fd;
1294         int rc = -ENODEV;
1295         int fsnamelen, mountlen;
1296
1297         /* get the mount point */
1298         fp = setmntent(PROC_MOUNTS, "r");
1299         if (fp == NULL) {
1300                 rc = -EIO;
1301                 llapi_error(LLAPI_MSG_ERROR, rc,
1302                             "cannot retrieve filesystem mount point");
1303                 return rc;
1304         }
1305         while (1) {
1306                 if (getmntent_r(fp, &mnt, buf, sizeof(buf)) == NULL)
1307                         break;
1308
1309                 if (!llapi_is_lustre_mnt(&mnt))
1310                         continue;
1311
1312                 if ((want & WANT_INDEX) && (idx++ != index))
1313                         continue;
1314
1315                 mntlen = strlen(mnt.mnt_dir);
1316                 ptr = strchr(mnt.mnt_fsname, '/');
1317                 while (ptr && *ptr == '/')
1318                         ptr++;
1319                 /*
1320                  * thanks to the call to llapi_is_lustre_mnt() above,
1321                  * we are sure that mnt.mnt_fsname contains ":/",
1322                  * so ptr should never be NULL
1323                  */
1324                 if (ptr == NULL)
1325                         continue;
1326                 ptr_end = ptr;
1327                 while (*ptr_end != '/' && *ptr_end != '\0')
1328                         ptr_end++;
1329
1330                 /* Check the fsname for a match, if given */
1331                 mountlen = ptr_end - ptr;
1332                 if (!(want & WANT_FSNAME) && fsname != NULL &&
1333                     (fsnamelen = strlen(fsname)) > 0 &&
1334                     (fsnamelen != mountlen ||
1335                     (strncmp(ptr, fsname, mountlen) != 0)))
1336                         continue;
1337
1338                 /* If the path isn't set return the first one we find */
1339                 if (path == NULL || strlen(path) == 0) {
1340                         strncpy(mntdir, mnt.mnt_dir, sizeof(mntdir) - 1);
1341                         mntdir[sizeof(mntdir) - 1] = '\0';
1342                         if ((want & WANT_FSNAME) && fsname != NULL) {
1343                                 strncpy(fsname, ptr, mountlen);
1344                                 fsname[mountlen] = '\0';
1345                         }
1346                         rc = 0;
1347                         break;
1348                 /* Otherwise find the longest matching path */
1349                 } else if ((strlen(path) >= mntlen) &&
1350                            (strncmp(mnt.mnt_dir, path, mntlen) == 0)) {
1351                         /* check the path format */
1352                         if (strlen(path) > mntlen && path[mntlen] != '/')
1353                                 continue;
1354                         strncpy(mntdir, mnt.mnt_dir, sizeof(mntdir) - 1);
1355                         mntdir[sizeof(mntdir) - 1] = '\0';
1356                         if ((want & WANT_FSNAME) && fsname != NULL) {
1357                                 strncpy(fsname, ptr, mountlen);
1358                                 fsname[mountlen] = '\0';
1359                         }
1360                         rc = 0;
1361                         break;
1362                 }
1363         }
1364         endmntent(fp);
1365
1366         /* Found it */
1367         if (rc == 0) {
1368                 if ((want & WANT_PATH) && path != NULL) {
1369                         strncpy(path, mntdir, mntlen);
1370                         path[mntlen] = '\0';
1371                 }
1372                 if (want & WANT_FD) {
1373                         fd = open(mntdir, O_RDONLY | O_DIRECTORY | O_NONBLOCK);
1374                         if (fd < 0) {
1375                                 rc = -errno;
1376                                 llapi_error(LLAPI_MSG_ERROR, rc,
1377                                             "cannot open '%s'", mntdir);
1378
1379                         } else {
1380                                 *outfd = fd;
1381                         }
1382                 }
1383         } else if (want & WANT_ERROR)
1384                 llapi_err_noerrno(LLAPI_MSG_ERROR,
1385                                   "'%s' not on a mounted Lustre filesystem",
1386                                   (want & WANT_PATH) ? fsname : path);
1387         return rc;
1388 }
1389
1390 /*
1391  * search lustre mounts
1392  *
1393  * Calling this function will return to the user the mount point, mntdir, and
1394  * the file system name, fsname, if the user passed a buffer to this routine.
1395  *
1396  * The user inputs are pathname and index. If the pathname is supplied then
1397  * the value of the index will be ignored. The pathname will return data if
1398  * the pathname is located on a lustre mount. Index is used to pick which
1399  * mount point you want in the case of multiple mounted lustre file systems.
1400  * See function lfs_osts in lfs.c for an example of the index use.
1401  */
1402 int llapi_search_mounts(const char *pathname, int index, char *mntdir,
1403                         char *fsname)
1404 {
1405         int want = WANT_PATH, idx = -1;
1406
1407         if (!pathname || pathname[0] == '\0') {
1408                 want |= WANT_INDEX;
1409                 idx = index;
1410         } else {
1411                 strcpy(mntdir, pathname);
1412         }
1413
1414         if (fsname)
1415                 want |= WANT_FSNAME;
1416         return get_root_path(want, fsname, NULL, mntdir, idx);
1417 }
1418
1419 /* Given a path, find the corresponding Lustre fsname */
1420 int llapi_search_fsname(const char *pathname, char *fsname)
1421 {
1422         char *path;
1423         int rc;
1424
1425         path = realpath(pathname, NULL);
1426         if (path == NULL) {
1427                 char tmp[PATH_MAX - 1];
1428                 char buf[PATH_MAX];
1429                 char *ptr;
1430
1431                 tmp[0] = '\0';
1432                 buf[0] = '\0';
1433                 if (pathname[0] != '/') {
1434                         /*
1435                          * Need an absolute path, but realpath() only works for
1436                          * pathnames that actually exist.  We go through the
1437                          * extra hurdle of dirname(getcwd() + pathname) in
1438                          * case the relative pathname contains ".." in it.
1439                          */
1440                         char realpath[PATH_MAX - 1];
1441
1442                         if (getcwd(realpath, sizeof(realpath) - 2) == NULL) {
1443                                 rc = -errno;
1444                                 llapi_error(LLAPI_MSG_ERROR, rc,
1445                                             "cannot get current working directory");
1446                                 return rc;
1447                         }
1448
1449                         rc = snprintf(tmp, sizeof(tmp), "%s/", realpath);
1450                         if (rc >= sizeof(tmp)) {
1451                                 rc = -E2BIG;
1452                                 llapi_error(LLAPI_MSG_ERROR, rc,
1453                                             "invalid parent path '%s'",
1454                                             tmp);
1455                                 return rc;
1456                         }
1457                 }
1458
1459                 rc = snprintf(buf, sizeof(buf), "%s%s", tmp, pathname);
1460                 if (rc >= sizeof(buf)) {
1461                         rc = -E2BIG;
1462                         llapi_error(LLAPI_MSG_ERROR, rc,
1463                                     "invalid path '%s'", pathname);
1464                         return rc;
1465                 }
1466                 path = realpath(buf, NULL);
1467                 if (path == NULL) {
1468                         ptr = strrchr(buf, '/');
1469                         if (ptr == NULL) {
1470                                 llapi_error(LLAPI_MSG_ERROR |
1471                                             LLAPI_MSG_NO_ERRNO, 0,
1472                                             "cannot resolve path '%s'",
1473                                             buf);
1474                                 return -ENOENT;
1475                         }
1476                         *ptr = '\0';
1477                         path = realpath(buf, NULL);
1478                         if (path == NULL) {
1479                                 rc = -errno;
1480                                 llapi_error(LLAPI_MSG_ERROR, rc,
1481                                             "cannot resolve path '%s'",
1482                                              pathname);
1483                                 return rc;
1484                         }
1485                 }
1486         }
1487         rc = get_root_path(WANT_FSNAME | WANT_ERROR, fsname, NULL, path, -1);
1488         free(path);
1489         return rc;
1490 }
1491
1492 int llapi_search_rootpath(char *pathname, const char *fsname)
1493 {
1494         /*
1495          * pathname can be used as an argument by get_root_path(),
1496          * clear it for safety
1497          */
1498         pathname[0] = 0;
1499         return get_root_path(WANT_PATH, (char *)fsname, NULL, pathname, -1);
1500 }
1501
1502 /**
1503  * Get the list of pool members.
1504  * \param poolname    string of format \<fsname\>.\<poolname\>
1505  * \param members     caller-allocated array of char*
1506  * \param list_size   size of the members array
1507  * \param buffer      caller-allocated buffer for storing OST names
1508  * \param buffer_size size of the buffer
1509  *
1510  * \return number of members retrieved for this pool
1511  * \retval -error failure
1512  */
1513 int llapi_get_poolmembers(const char *poolname, char **members,
1514                           int list_size, char *buffer, int buffer_size)
1515 {
1516         char fsname[PATH_MAX];
1517         char *pool, *tmp;
1518         glob_t pathname;
1519         char buf[PATH_MAX];
1520         FILE *fd;
1521         int rc = 0;
1522         int nb_entries = 0;
1523         int used = 0;
1524
1525         /* name is FSNAME.POOLNAME */
1526         if (strlen(poolname) >= sizeof(fsname))
1527                 return -EOVERFLOW;
1528
1529         snprintf(fsname, sizeof(fsname), "%s", poolname);
1530         pool = strchr(fsname, '.');
1531         if (pool == NULL)
1532                 return -EINVAL;
1533
1534         *pool = '\0';
1535         pool++;
1536
1537         rc = poolpath(&pathname, fsname, NULL);
1538         if (rc != 0) {
1539                 llapi_error(LLAPI_MSG_ERROR, rc,
1540                             "Lustre filesystem '%s' not found",
1541                             fsname);
1542                 return rc;
1543         }
1544
1545         llapi_printf(LLAPI_MSG_NORMAL, "Pool: %s.%s\n", fsname, pool);
1546         rc = snprintf(buf, sizeof(buf), "%s/%s", pathname.gl_pathv[0], pool);
1547         cfs_free_param_data(&pathname);
1548         if (rc >= sizeof(buf))
1549                 return -EOVERFLOW;
1550         fd = fopen(buf, "r");
1551         if (fd == NULL) {
1552                 rc = -errno;
1553                 llapi_error(LLAPI_MSG_ERROR, rc, "cannot open %s", buf);
1554                 return rc;
1555         }
1556
1557         rc = 0;
1558         while (fgets(buf, sizeof(buf), fd) != NULL) {
1559                 if (nb_entries >= list_size) {
1560                         rc = -EOVERFLOW;
1561                         break;
1562                 }
1563                 buf[sizeof(buf) - 1] = '\0';
1564                 /* remove '\n' */
1565                 tmp = strchr(buf, '\n');
1566                 if (tmp != NULL)
1567                         *tmp = '\0';
1568                 if (used + strlen(buf) + 1 > buffer_size) {
1569                         rc = -EOVERFLOW;
1570                         break;
1571                 }
1572
1573                 strcpy(buffer + used, buf);
1574                 members[nb_entries] = buffer + used;
1575                 used += strlen(buf) + 1;
1576                 nb_entries++;
1577                 rc = nb_entries;
1578         }
1579
1580         fclose(fd);
1581         return rc;
1582 }
1583
1584 /**
1585  * Get the list of pools in a filesystem.
1586  * \param name        filesystem name or path
1587  * \param poollist    caller-allocated array of char*
1588  * \param list_size   size of the poollist array
1589  * \param buffer      caller-allocated buffer for storing pool names
1590  * \param buffer_size size of the buffer
1591  *
1592  * \return number of pools retrieved for this filesystem
1593  * \retval -error failure
1594  */
1595 int llapi_get_poollist(const char *name, char **poollist, int list_size,
1596                        char *buffer, int buffer_size)
1597 {
1598         glob_t pathname;
1599         char *fsname;
1600         char *ptr;
1601         DIR *dir;
1602         struct dirent *pool;
1603         int rc = 0;
1604         unsigned int nb_entries = 0;
1605         unsigned int used = 0;
1606         unsigned int i;
1607
1608         /* initialize output array */
1609         for (i = 0; i < list_size; i++)
1610                 poollist[i] = NULL;
1611
1612         /* is name a pathname ? */
1613         ptr = strchr(name, '/');
1614         if (ptr != NULL) {
1615                 char fsname_buf[MAXNAMLEN];
1616
1617                 /* We will need fsname for printing later */
1618                 rc = llapi_getname(name, fsname_buf, sizeof(fsname_buf));
1619                 if (rc)
1620                         return rc;
1621
1622                 ptr = strrchr(fsname_buf, '-');
1623                 if (ptr)
1624                         *ptr = '\0';
1625
1626                 fsname = strdup(fsname_buf);
1627                 if (!fsname)
1628                         return -ENOMEM;
1629         } else {
1630                 /* name is FSNAME */
1631                 fsname = strdup(name);
1632                 if (!fsname)
1633                         return -ENOMEM;
1634         }
1635
1636         rc = poolpath(&pathname, fsname, NULL);
1637         if (rc != 0) {
1638                 llapi_error(LLAPI_MSG_ERROR, rc,
1639                             "Lustre filesystem '%s' not found", name);
1640                 goto free_path;
1641         }
1642
1643         dir = opendir(pathname.gl_pathv[0]);
1644         if (dir == NULL) {
1645                 rc = -errno;
1646                 llapi_error(LLAPI_MSG_ERROR, rc,
1647                             "Could not open pool list for '%s'",
1648                             name);
1649                 goto free_path;
1650         }
1651
1652         do {
1653                 errno = 0;
1654                 pool = readdir(dir);
1655                 if (pool == NULL) {
1656                         rc = -errno;
1657                         goto free_dir;
1658                 }
1659
1660                 /* ignore . and .. */
1661                 if (!strcmp(pool->d_name, ".") || !strcmp(pool->d_name, ".."))
1662                         continue;
1663
1664                 /* check output bounds */
1665                 if (nb_entries >= list_size) {
1666                         rc = -EOVERFLOW;
1667                         goto free_dir_no_msg;
1668                 }
1669
1670                 /* +2 for '.' and final '\0' */
1671                 if (used + strlen(pool->d_name) + strlen(fsname) + 2
1672                     > buffer_size) {
1673                         rc = -EOVERFLOW;
1674                         goto free_dir_no_msg;
1675                 }
1676
1677                 sprintf(buffer + used, "%s.%s", fsname, pool->d_name);
1678                 poollist[nb_entries] = buffer + used;
1679                 used += strlen(pool->d_name) + strlen(fsname) + 2;
1680                 nb_entries++;
1681         } while (1);
1682
1683 free_dir:
1684         if (rc)
1685                 llapi_error(LLAPI_MSG_ERROR, rc,
1686                             "Error reading pool list for '%s'", name);
1687         else
1688                 llapi_printf(LLAPI_MSG_NORMAL, "Pools from %s:\n", fsname);
1689
1690 free_dir_no_msg:
1691         closedir(dir);
1692 free_path:
1693         cfs_free_param_data(&pathname);
1694         if (fsname)
1695                 free(fsname);
1696         return rc != 0 ? rc : nb_entries;
1697 }
1698
1699 /* wrapper for lfs.c and obd.c */
1700 int llapi_poollist(const char *name)
1701 {
1702         int poolcount, rc, i;
1703         char *buf, **pools;
1704
1705         rc = llapi_get_poolbuf(name, &buf, &pools, &poolcount);
1706         if (rc)
1707                 return rc;
1708
1709         for (i = 0; i < poolcount; i++)
1710                 llapi_printf(LLAPI_MSG_NORMAL, "%s\n", pools[i]);
1711         free(buf);
1712
1713         return 0;
1714 }
1715
1716 /**
1717  * Get buffer that holds uuids and the list of pools in a filesystem.
1718  *
1719  * \param name          filesystem name or path
1720  * \param buf           bufffer that has to be freed if function returns 0
1721  * \param pools         pointer to the list of pools in buffer
1722  * \param poolcount     number of pools
1723  *
1724  * \return 0 when found at least 1 pool, i.e. poolcount  > 0
1725  * \retval -error failure
1726  */
1727 int llapi_get_poolbuf(const char *name, char **buf,
1728                       char ***pools, int *poolcount)
1729 {
1730         /*
1731          * list of pool names (assume that pool count is smaller
1732          * than OST count)
1733          */
1734         char **list, *buffer = NULL, *fsname = (char *)name;
1735         char *poolname = NULL, *tmp = NULL, data[16];
1736         enum param_filter type = FILTER_BY_PATH;
1737         int obdcount, bufsize, rc, nb;
1738
1739         if (name == NULL)
1740                 return -EINVAL;
1741
1742         if (name[0] != '/') {
1743                 fsname = strdup(name);
1744                 if (fsname == NULL)
1745                         return -ENOMEM;
1746
1747                 poolname = strchr(fsname, '.');
1748                 if (poolname)
1749                         *poolname = '\0';
1750                 type = FILTER_BY_FS_NAME;
1751         }
1752
1753         rc = get_lustre_param_value("lov", fsname, type, "numobd",
1754                                     data, sizeof(data));
1755         if (rc < 0)
1756                 goto err;
1757         obdcount = atoi(data);
1758
1759         /*
1760          * Allocate space for each fsname-OST0000_UUID, 1 per OST,
1761          * and also an array to store the pointers for all that
1762          * allocated space.
1763          */
1764 retry_get_pools:
1765         bufsize = sizeof(struct obd_uuid) * obdcount;
1766         buffer = realloc(tmp, bufsize + sizeof(*list) * obdcount);
1767         if (buffer == NULL) {
1768                 rc = -ENOMEM;
1769                 goto err;
1770         }
1771         list = (char **) (buffer + bufsize);
1772
1773         if (!poolname) {
1774                 /* name is a path or fsname */
1775                 nb = llapi_get_poollist(name, list, obdcount,
1776                                         buffer, bufsize);
1777         } else {
1778                 /* name is a pool name (<fsname>.<poolname>) */
1779                 nb = llapi_get_poolmembers(name, list, obdcount,
1780                                            buffer, bufsize);
1781         }
1782
1783         if (nb == -EOVERFLOW) {
1784                 obdcount *= 2;
1785                 tmp = buffer;
1786                 goto retry_get_pools;
1787         }
1788
1789         rc = (nb < 0 ? nb : 0);
1790         if (!rc) {
1791                 *buf = buffer;
1792                 *pools = list;
1793                 *poolcount = nb;
1794         }
1795 err:
1796         /* Don't free buffer, it will be used later */
1797         if (rc && buffer)
1798                 free(buffer);
1799         if (fsname != NULL && type == FILTER_BY_FS_NAME)
1800                 free(fsname);
1801         return rc;
1802 }
1803
1804 typedef int (semantic_func_t)(char *path, int p, int *d,
1805                               void *data, struct dirent64 *de);
1806
1807 #define OBD_NOT_FOUND           (-1)
1808
1809 static bool lmv_is_foreign(__u32 magic)
1810 {
1811         return magic == LMV_MAGIC_FOREIGN;
1812 }
1813
1814 static void find_param_fini(struct find_param *param)
1815 {
1816         if (param->fp_migrate)
1817                 return;
1818
1819         if (param->fp_obd_indexes) {
1820                 free(param->fp_obd_indexes);
1821                 param->fp_obd_indexes = NULL;
1822         }
1823
1824         if (param->fp_lmd) {
1825                 free(param->fp_lmd);
1826                 param->fp_lmd = NULL;
1827         }
1828
1829         if (param->fp_lmv_md) {
1830                 free(param->fp_lmv_md);
1831                 param->fp_lmv_md = NULL;
1832         }
1833 }
1834
1835 static int common_param_init(struct find_param *param, char *path)
1836 {
1837         int lum_size = get_mds_md_size(path);
1838
1839         if (lum_size < 0)
1840                 return lum_size;
1841
1842         /* migrate has fp_lmv_md initialized outside */
1843         if (param->fp_migrate)
1844                 return 0;
1845
1846         if (lum_size < PATH_MAX + 1)
1847                 lum_size = PATH_MAX + 1;
1848
1849         param->fp_lum_size = lum_size;
1850         param->fp_lmd = calloc(1, offsetof(typeof(*param->fp_lmd), lmd_lmm) +
1851                                lum_size);
1852         if (param->fp_lmd == NULL) {
1853                 llapi_error(LLAPI_MSG_ERROR, -ENOMEM,
1854                             "error: allocate %zu bytes for layout failed",
1855                             sizeof(lstat_t) + param->fp_lum_size);
1856                 return -ENOMEM;
1857         }
1858
1859         param->fp_lmv_stripe_count = 256;
1860         param->fp_lmv_md = calloc(1,
1861                                   lmv_user_md_size(param->fp_lmv_stripe_count,
1862                                                    LMV_USER_MAGIC_SPECIFIC));
1863         if (param->fp_lmv_md == NULL) {
1864                 llapi_error(LLAPI_MSG_ERROR, -ENOMEM,
1865                             "error: allocation of %d bytes for ioctl",
1866                             lmv_user_md_size(param->fp_lmv_stripe_count,
1867                                              LMV_USER_MAGIC_SPECIFIC));
1868                 find_param_fini(param);
1869                 return -ENOMEM;
1870         }
1871
1872         param->fp_got_uuids = 0;
1873         param->fp_obd_indexes = NULL;
1874         param->fp_obd_index = OBD_NOT_FOUND;
1875         param->fp_mdt_index = OBD_NOT_FOUND;
1876         return 0;
1877 }
1878
1879 static int cb_common_fini(char *path, int p, int *dp, void *data,
1880                           struct dirent64 *de)
1881 {
1882         struct find_param *param = data;
1883
1884         param->fp_depth--;
1885         return 0;
1886 }
1887
1888 /* set errno upon failure */
1889 static int open_parent(const char *path)
1890 {
1891         char *path_copy;
1892         char *parent_path;
1893         int parent;
1894
1895         path_copy = strdup(path);
1896         if (path_copy == NULL)
1897                 return -1;
1898
1899         parent_path = dirname(path_copy);
1900         parent = open(parent_path, O_RDONLY|O_NDELAY|O_DIRECTORY);
1901         free(path_copy);
1902
1903         return parent;
1904 }
1905
1906 static int cb_get_dirstripe(char *path, int *d, struct find_param *param)
1907 {
1908         int ret;
1909         bool did_nofollow = false;
1910
1911         if (!d || *d < 0)
1912                 return -ENOTDIR;
1913 again:
1914         param->fp_lmv_md->lum_stripe_count = param->fp_lmv_stripe_count;
1915         if (param->fp_get_default_lmv)
1916                 param->fp_lmv_md->lum_magic = LMV_USER_MAGIC;
1917         else
1918                 param->fp_lmv_md->lum_magic = LMV_MAGIC_V1;
1919
1920         ret = ioctl(*d, LL_IOC_LMV_GETSTRIPE, param->fp_lmv_md);
1921
1922         /* if ENOTTY likely to be a fake symlink, so try again after
1923          * new open() with O_NOFOLLOW, but only once to prevent any
1924          * loop like for the path of a file/dir not on Lustre !!
1925          */
1926         if (ret < 0 && errno == ENOTTY && !did_nofollow) {
1927                 int fd, ret2;
1928
1929                 did_nofollow = true;
1930                 fd = open(path, O_RDONLY | O_NOFOLLOW);
1931                 if (fd < 0) {
1932                         /* restore original errno */
1933                         errno = ENOTTY;
1934                         return ret;
1935                 }
1936
1937                 /* close original fd and set new */
1938                 close(*d);
1939                 *d = fd;
1940                 ret2 = ioctl(fd, LL_IOC_LMV_GETSTRIPE, param->fp_lmv_md);
1941                 if (ret2 < 0 && errno != E2BIG) {
1942                         /* restore original errno */
1943                         errno = ENOTTY;
1944                         return ret;
1945                 }
1946                 /* LMV is ok or need to handle E2BIG case now */
1947                 ret = ret2;
1948         }
1949
1950         if (errno == E2BIG && ret != 0) {
1951                 int stripe_count;
1952                 int lmv_size;
1953
1954                 /* if foreign LMV case, fake stripes number */
1955                 if (lmv_is_foreign(param->fp_lmv_md->lum_magic)) {
1956                         struct lmv_foreign_md *lfm;
1957
1958                         lfm = (struct lmv_foreign_md *)param->fp_lmv_md;
1959                         if (lfm->lfm_length < XATTR_SIZE_MAX -
1960                             offsetof(typeof(*lfm), lfm_value)) {
1961                                 uint32_t size = lfm->lfm_length +
1962                                              offsetof(typeof(*lfm), lfm_value);
1963
1964                                 stripe_count = lmv_foreign_to_md_stripes(size);
1965                         } else {
1966                                 llapi_error(LLAPI_MSG_ERROR, -EINVAL,
1967                                             "error: invalid %d foreign size returned from ioctl",
1968                                             lfm->lfm_length);
1969                                 return -EINVAL;
1970                         }
1971                 } else {
1972                         stripe_count = param->fp_lmv_md->lum_stripe_count;
1973                 }
1974                 if (stripe_count <= param->fp_lmv_stripe_count)
1975                         return ret;
1976
1977                 free(param->fp_lmv_md);
1978                 param->fp_lmv_stripe_count = stripe_count;
1979                 lmv_size = lmv_user_md_size(stripe_count,
1980                                             LMV_USER_MAGIC_SPECIFIC);
1981                 param->fp_lmv_md = malloc(lmv_size);
1982                 if (param->fp_lmv_md == NULL) {
1983                         llapi_error(LLAPI_MSG_ERROR, -ENOMEM,
1984                                     "error: allocation of %d bytes for ioctl",
1985                                     lmv_user_md_size(param->fp_lmv_stripe_count,
1986                                                      LMV_USER_MAGIC_SPECIFIC));
1987                         return -ENOMEM;
1988                 }
1989                 goto again;
1990         }
1991
1992         return ret;
1993 }
1994
1995 static void convert_lmd_statx(struct lov_user_mds_data *lmd_v2, lstat_t *st,
1996                               bool strict)
1997 {
1998         memset(&lmd_v2->lmd_stx, 0, sizeof(lmd_v2->lmd_stx));
1999         lmd_v2->lmd_stx.stx_blksize = st->st_blksize;
2000         lmd_v2->lmd_stx.stx_nlink = st->st_nlink;
2001         lmd_v2->lmd_stx.stx_uid = st->st_uid;
2002         lmd_v2->lmd_stx.stx_gid = st->st_gid;
2003         lmd_v2->lmd_stx.stx_mode = st->st_mode;
2004         lmd_v2->lmd_stx.stx_ino = st->st_ino;
2005         lmd_v2->lmd_stx.stx_size = st->st_size;
2006         lmd_v2->lmd_stx.stx_blocks = st->st_blocks;
2007         lmd_v2->lmd_stx.stx_atime.tv_sec = st->st_atime;
2008         lmd_v2->lmd_stx.stx_ctime.tv_sec = st->st_ctime;
2009         lmd_v2->lmd_stx.stx_mtime.tv_sec = st->st_mtime;
2010         lmd_v2->lmd_stx.stx_rdev_major = major(st->st_rdev);
2011         lmd_v2->lmd_stx.stx_rdev_minor = minor(st->st_rdev);
2012         lmd_v2->lmd_stx.stx_dev_major = major(st->st_dev);
2013         lmd_v2->lmd_stx.stx_dev_minor = minor(st->st_dev);
2014         lmd_v2->lmd_stx.stx_mask |= STATX_BASIC_STATS;
2015
2016         lmd_v2->lmd_flags = 0;
2017         if (strict) {
2018                 lmd_v2->lmd_flags |= OBD_MD_FLSIZE | OBD_MD_FLBLOCKS;
2019         } else {
2020                 lmd_v2->lmd_stx.stx_mask &= ~(STATX_SIZE | STATX_BLOCKS);
2021                 if (lmd_v2->lmd_stx.stx_size)
2022                         lmd_v2->lmd_flags |= OBD_MD_FLLAZYSIZE;
2023                 if (lmd_v2->lmd_stx.stx_blocks)
2024                         lmd_v2->lmd_flags |= OBD_MD_FLLAZYBLOCKS;
2025         }
2026         lmd_v2->lmd_flags |= OBD_MD_FLATIME | OBD_MD_FLMTIME | OBD_MD_FLCTIME |
2027                              OBD_MD_FLBLKSZ | OBD_MD_FLMODE | OBD_MD_FLTYPE |
2028                              OBD_MD_FLUID | OBD_MD_FLGID | OBD_MD_FLNLINK |
2029                              OBD_MD_FLRDEV;
2030
2031 }
2032
2033 static int convert_lmdbuf_v1v2(void *lmdbuf, int lmdlen)
2034 {
2035         struct lov_user_mds_data_v1 *lmd_v1 = lmdbuf;
2036         struct lov_user_mds_data *lmd_v2 = lmdbuf;
2037         lstat_t st;
2038         int size;
2039
2040         size = lov_comp_md_size((struct lov_comp_md_v1 *)&lmd_v1->lmd_lmm);
2041         if (size < 0)
2042                 return size;
2043
2044         if (lmdlen < sizeof(lmd_v1->lmd_st) + size)
2045                 return -EOVERFLOW;
2046
2047         st = lmd_v1->lmd_st;
2048         memmove(&lmd_v2->lmd_lmm, &lmd_v1->lmd_lmm,
2049                 lmdlen - (&lmd_v2->lmd_lmm - &lmd_v1->lmd_lmm));
2050         convert_lmd_statx(lmd_v2, &st, false);
2051         lmd_v2->lmd_lmmsize = 0;
2052         lmd_v2->lmd_padding = 0;
2053
2054         return 0;
2055 }
2056
2057 int get_lmd_info_fd(const char *path, int parent_fd, int dir_fd,
2058                     void *lmdbuf, int lmdlen, enum get_lmd_info_type type)
2059 {
2060         struct lov_user_mds_data *lmd = lmdbuf;
2061         static bool use_old_ioctl;
2062         unsigned long cmd;
2063         int ret = 0;
2064
2065         if (parent_fd < 0 && dir_fd < 0)
2066                 return -EINVAL;
2067         if (type != GET_LMD_INFO && type != GET_LMD_STRIPE)
2068                 return -EINVAL;
2069
2070         if (dir_fd >= 0) {
2071                 /*
2072                  * LL_IOC_MDC_GETINFO operates on the current directory inode
2073                  * and returns struct lov_user_mds_data, while
2074                  * LL_IOC_LOV_GETSTRIPE returns only struct lov_user_md.
2075                  */
2076                 if (type == GET_LMD_INFO)
2077                         cmd = use_old_ioctl ? LL_IOC_MDC_GETINFO_V1 :
2078                                               LL_IOC_MDC_GETINFO_V2;
2079                 else
2080                         cmd = LL_IOC_LOV_GETSTRIPE;
2081
2082 retry_getinfo:
2083                 ret = ioctl(dir_fd, cmd, lmdbuf);
2084                 if (ret < 0 && errno == ENOTTY &&
2085                     cmd == LL_IOC_MDC_GETINFO_V2) {
2086                         cmd = LL_IOC_MDC_GETINFO_V1;
2087                         use_old_ioctl = true;
2088                         goto retry_getinfo;
2089                 }
2090
2091                 if (cmd == LL_IOC_MDC_GETINFO_V1 && !ret)
2092                         ret = convert_lmdbuf_v1v2(lmdbuf, lmdlen);
2093
2094                 if (ret < 0 && errno == ENOTTY && type == GET_LMD_STRIPE) {
2095                         int dir_fd2;
2096
2097                         /* retry ioctl() after new open() with O_NOFOLLOW
2098                          * just in case it could be a fake symlink
2099                          * need using a new open() as dir_fd is being closed
2100                          * by caller
2101                          */
2102
2103                         dir_fd2 = open(path, O_RDONLY | O_NDELAY | O_NOFOLLOW);
2104                         if (dir_fd2 < 0) {
2105                                 /* return original error */
2106                                 errno = ENOTTY;
2107                         } else {
2108                                 ret = ioctl(dir_fd2, cmd, lmdbuf);
2109                                 /* pass new errno or success back to caller */
2110
2111                                 close(dir_fd2);
2112                         }
2113                 }
2114
2115         } else if (parent_fd >= 0) {
2116                 const char *fname = strrchr(path, '/');
2117
2118                 /*
2119                  * IOC_MDC_GETFILEINFO takes as input the filename (relative to
2120                  * the parent directory) and returns struct lov_user_mds_data,
2121                  * while IOC_MDC_GETFILESTRIPE returns only struct lov_user_md.
2122                  *
2123                  * This avoids opening, locking, and closing each file on the
2124                  * client if that is not needed. Multiple of these ioctl() can
2125                  * be done on the parent dir with a single open for all
2126                  * files in that directory, and it also doesn't pollute the
2127                  * client dcache with millions of dentries when traversing
2128                  * a large filesystem.
2129                  */
2130                 fname = (fname == NULL ? path : fname + 1);
2131
2132                 ret = snprintf(lmdbuf, lmdlen, "%s", fname);
2133                 if (ret < 0)
2134                         errno = -ret;
2135                 else if (ret >= lmdlen || ret++ == 0)
2136                         errno = EINVAL;
2137                 else {
2138                         if (type == GET_LMD_INFO)
2139                                 cmd = use_old_ioctl ? IOC_MDC_GETFILEINFO_V1 :
2140                                                       IOC_MDC_GETFILEINFO_V2;
2141                         else
2142                                 cmd = IOC_MDC_GETFILESTRIPE;
2143
2144 retry_getfileinfo:
2145                         ret = ioctl(parent_fd, cmd, lmdbuf);
2146                         if (ret < 0 && errno == ENOTTY &&
2147                             cmd == IOC_MDC_GETFILEINFO_V2) {
2148                                 cmd = IOC_MDC_GETFILEINFO_V1;
2149                                 use_old_ioctl = true;
2150                                 goto retry_getfileinfo;
2151                         }
2152
2153                         if (cmd == IOC_MDC_GETFILEINFO_V1 && !ret)
2154                                 ret = convert_lmdbuf_v1v2(lmdbuf, lmdlen);
2155                 }
2156         }
2157
2158         if (ret && type == GET_LMD_INFO) {
2159                 if (errno == ENOTTY) {
2160                         lstat_t st;
2161
2162                         /*
2163                          * ioctl is not supported, it is not a lustre fs.
2164                          * Do the regular lstat(2) instead.
2165                          */
2166                         ret = lstat_f(path, &st);
2167                         if (ret) {
2168                                 ret = -errno;
2169                                 llapi_error(LLAPI_MSG_ERROR, ret,
2170                                             "error: %s: lstat failed for %s",
2171                                             __func__, path);
2172                         }
2173
2174                         convert_lmd_statx(lmd, &st, true);
2175                         /*
2176                          * It may be wrong to set use_old_ioctl with true as
2177                          * the file is not a lustre fs. So reset it with false
2178                          * directly here.
2179                          */
2180                         use_old_ioctl = false;
2181                 } else if (errno == ENOENT) {
2182                         ret = -errno;
2183                         llapi_error(LLAPI_MSG_WARN, ret,
2184                                     "warning: %s does not exist", path);
2185                 } else if (errno != EISDIR && errno != ENODATA) {
2186                         ret = -errno;
2187                         llapi_error(LLAPI_MSG_ERROR, ret,
2188                                     "%s ioctl failed for %s.",
2189                                     dir_fd >= 0 ? "LL_IOC_MDC_GETINFO" :
2190                                     "IOC_MDC_GETFILEINFO", path);
2191                 }
2192         }
2193
2194         return ret;
2195 }
2196
2197 static int llapi_semantic_traverse(char *path, int size, int parent,
2198                                    semantic_func_t sem_init,
2199                                    semantic_func_t sem_fini, void *data,
2200                                    struct dirent64 *de)
2201 {
2202         struct find_param *param = (struct find_param *)data;
2203         struct dirent64 *dent;
2204         int len, ret, d, p = -1;
2205         DIR *dir = NULL;
2206
2207         ret = 0;
2208         len = strlen(path);
2209
2210         d = open(path, O_RDONLY|O_NDELAY|O_DIRECTORY);
2211         /* if an invalid fake dir symlink, opendir() will return EINVAL
2212          * instead of ENOTDIR. If a valid but dangling faked or real file/dir
2213          * symlink ENOENT will be returned. For a valid/resolved fake or real
2214          * file symlink ENOTDIR will be returned as for a regular file.
2215          * opendir() will be successful for a  valid and resolved fake or real
2216          * dir simlink or a regular dir.
2217          */
2218         if (d == -1 && errno != ENOTDIR && errno != EINVAL && errno != ENOENT) {
2219                 ret = -errno;
2220                 llapi_error(LLAPI_MSG_ERROR, ret, "%s: Failed to open '%s'",
2221                             __func__, path);
2222                 return ret;
2223         } else if (d == -1) {
2224                 if (errno == ENOENT || errno == EINVAL) {
2225                         int old_errno = errno;
2226
2227                         /* try to open with O_NOFOLLOW this will help
2228                          * differentiate fake vs real symlinks
2229                          * it is ok to not use O_DIRECTORY with O_RDONLY
2230                          * and it will prevent the need to deal with ENOTDIR
2231                          * error, instead of ELOOP, being returned by recent
2232                          * kernels for real symlinks
2233                          */
2234                         d = open(path, O_RDONLY|O_NDELAY|O_NOFOLLOW);
2235                         /* if a dangling real symlink should return ELOOP, or
2236                          * again ENOENT if really non-existing path, or E...??
2237                          * So return original error. If success or ENOTDIR, path
2238                          * is likely to be a fake dir/file symlink, so continue
2239                          */
2240                         if (d == -1) {
2241                                 ret =  -old_errno;
2242                                 goto out;
2243                         }
2244
2245                 }
2246
2247                 /* ENOTDIR */
2248                 if (parent == -1 && d == -1) {
2249                         /* Open the parent dir. */
2250                         p = open_parent(path);
2251                         if (p == -1) {
2252                                 ret = -errno;
2253                                 goto out;
2254                         }
2255                 }
2256         } else { /* d != -1 */
2257                 int d2;
2258
2259                 /* try to reopen dir with O_NOFOLLOW just in case of a foreign
2260                  * symlink dir
2261                  */
2262                 d2 = open(path, O_RDONLY|O_NDELAY|O_NOFOLLOW);
2263                 if (d2 != -1) {
2264                         close(d);
2265                         d = d2;
2266                 } else {
2267                         /* continue with d */
2268                         errno = 0;
2269                 }
2270         }
2271
2272         if (sem_init) {
2273                 ret = sem_init(path, (parent != -1) ? parent : p, &d, data, de);
2274                 if (ret)
2275                         goto err;
2276         }
2277
2278         if (d == -1)
2279                 goto out;
2280
2281         dir = fdopendir(d);
2282         if (dir == NULL) {
2283                 /* ENOTDIR if fake symlink, do not consider it as an error */
2284                 if (errno != ENOTDIR)
2285                         llapi_error(LLAPI_MSG_ERROR, errno,
2286                                     "fdopendir() failed");
2287                 else
2288                         errno = 0;
2289
2290                 goto out;
2291         }
2292
2293         while ((dent = readdir64(dir)) != NULL) {
2294                 int rc;
2295
2296                 if (!strcmp(dent->d_name, ".") || !strcmp(dent->d_name, ".."))
2297                         continue;
2298
2299                 path[len] = 0;
2300                 if ((len + dent->d_reclen + 2) > size) {
2301                         llapi_err_noerrno(LLAPI_MSG_ERROR,
2302                                           "error: %s: string buffer too small for %s",
2303                                           __func__, path);
2304                         break;
2305                 }
2306                 strcat(path, "/");
2307                 strcat(path, dent->d_name);
2308
2309                 if (dent->d_type == DT_UNKNOWN) {
2310                         struct lov_user_mds_data *lmd = param->fp_lmd;
2311
2312                         rc = get_lmd_info_fd(path, d, -1, param->fp_lmd,
2313                                              param->fp_lum_size, GET_LMD_INFO);
2314                         if (rc == 0)
2315                                 dent->d_type = IFTODT(lmd->lmd_stx.stx_mode);
2316                         else if (ret == 0)
2317                                 ret = rc;
2318
2319                         if (rc == -ENOENT)
2320                                 continue;
2321                 }
2322                 switch (dent->d_type) {
2323                 case DT_UNKNOWN:
2324                         llapi_err_noerrno(LLAPI_MSG_ERROR,
2325                                           "error: %s: '%s' is UNKNOWN type %d",
2326                                           __func__, dent->d_name, dent->d_type);
2327                         break;
2328                 case DT_DIR:
2329                         rc = llapi_semantic_traverse(path, size, d, sem_init,
2330                                                       sem_fini, data, dent);
2331                         if (rc != 0 && ret == 0)
2332                                 ret = rc;
2333                         break;
2334                 default:
2335                         rc = 0;
2336                         if (sem_init) {
2337                                 rc = sem_init(path, d, NULL, data, dent);
2338                                 if (rc < 0 && ret == 0) {
2339                                         ret = rc;
2340                                         break;
2341                                 }
2342                         }
2343                         if (sem_fini && rc == 0)
2344                                 sem_fini(path, d, NULL, data, dent);
2345                 }
2346         }
2347
2348 out:
2349         path[len] = 0;
2350
2351         if (sem_fini)
2352                 sem_fini(path, parent, &d, data, de);
2353 err:
2354         if (d != -1) {
2355                 if (dir)
2356                         closedir(dir);
2357                 else
2358                         close(d);
2359         }
2360         if (p != -1)
2361                 close(p);
2362         return ret;
2363 }
2364
2365 static int param_callback(char *path, semantic_func_t sem_init,
2366                           semantic_func_t sem_fini, struct find_param *param)
2367 {
2368         int ret, len = strlen(path);
2369         char *buf;
2370
2371         if (len > PATH_MAX) {
2372                 ret = -EINVAL;
2373                 llapi_error(LLAPI_MSG_ERROR, ret,
2374                             "Path name '%s' is too long", path);
2375                 return ret;
2376         }
2377
2378         buf = (char *)malloc(2 * PATH_MAX);
2379         if (!buf)
2380                 return -ENOMEM;
2381
2382         snprintf(buf, PATH_MAX + 1, "%s", path);
2383         ret = common_param_init(param, buf);
2384         if (ret)
2385                 goto out;
2386
2387         param->fp_depth = 0;
2388
2389         ret = llapi_semantic_traverse(buf, 2 * PATH_MAX, -1, sem_init,
2390                                       sem_fini, param, NULL);
2391 out:
2392         find_param_fini(param);
2393         free(buf);
2394         return ret < 0 ? ret : 0;
2395 }
2396
2397 int llapi_file_fget_lov_uuid(int fd, struct obd_uuid *lov_name)
2398 {
2399         int rc;
2400
2401         rc = ioctl(fd, OBD_IOC_GETDTNAME, lov_name);
2402         if (rc && errno == ENOTTY)
2403                 rc = ioctl(fd, OBD_IOC_GETNAME_OLD, lov_name);
2404         if (rc) {
2405                 rc = -errno;
2406                 llapi_error(LLAPI_MSG_ERROR, rc, "cannot get lov name");
2407         }
2408
2409         return rc;
2410 }
2411
2412 int llapi_file_fget_lmv_uuid(int fd, struct obd_uuid *lov_name)
2413 {
2414         int rc;
2415
2416         rc = ioctl(fd, OBD_IOC_GETMDNAME, lov_name);
2417         if (rc) {
2418                 rc = -errno;
2419                 llapi_error(LLAPI_MSG_ERROR, rc, "error: can't get lmv name.");
2420         }
2421
2422         return rc;
2423 }
2424
2425 int llapi_file_get_lov_uuid(const char *path, struct obd_uuid *lov_uuid)
2426 {
2427         int fd, rc;
2428
2429         /* do not follow faked symlinks */
2430         fd = open(path, O_RDONLY | O_NONBLOCK | O_NOFOLLOW);
2431         if (fd < 0) {
2432                 /* real symlink should have failed with ELOOP so retry without
2433                  * O_NOFOLLOW just in case
2434                  */
2435                 fd = open(path, O_RDONLY | O_NONBLOCK);
2436                 if (fd < 0) {
2437                         rc = -errno;
2438                         llapi_error(LLAPI_MSG_ERROR, rc, "cannot open '%s'",
2439                                     path);
2440                         return rc;
2441                 }
2442         }
2443
2444         rc = llapi_file_fget_lov_uuid(fd, lov_uuid);
2445
2446         close(fd);
2447         return rc;
2448 }
2449
2450 int llapi_file_get_lmv_uuid(const char *path, struct obd_uuid *lov_uuid)
2451 {
2452         int fd, rc;
2453
2454         fd = open(path, O_RDONLY | O_NONBLOCK);
2455         if (fd < 0) {
2456                 rc = -errno;
2457                 llapi_error(LLAPI_MSG_ERROR, rc, "error opening %s", path);
2458                 return rc;
2459         }
2460
2461         rc = llapi_file_fget_lmv_uuid(fd, lov_uuid);
2462
2463         close(fd);
2464         return rc;
2465 }
2466
2467 enum tgt_type {
2468         LOV_TYPE = 1,
2469         LMV_TYPE
2470 };
2471
2472 /*
2473  * If uuidp is NULL, return the number of available obd uuids.
2474  * If uuidp is non-NULL, then it will return the uuids of the obds. If
2475  * there are more OSTs than allocated to uuidp, then an error is returned with
2476  * the ost_count set to number of available obd uuids.
2477  */
2478 static int llapi_get_target_uuids(int fd, struct obd_uuid *uuidp,
2479                                   int *ost_count, enum tgt_type type)
2480 {
2481         char buf[PATH_MAX], format[32];
2482         int rc = 0, index = 0;
2483         struct obd_uuid name;
2484         glob_t param;
2485         FILE *fp;
2486
2487         /* Get the lov name */
2488         if (type == LOV_TYPE)
2489                 rc = llapi_file_fget_lov_uuid(fd, &name);
2490         else
2491                 rc = llapi_file_fget_lmv_uuid(fd, &name);
2492         if (rc != 0)
2493                 return rc;
2494
2495         /* Now get the ost uuids */
2496         rc = get_lustre_param_path(type == LOV_TYPE ? "lov" : "lmv", name.uuid,
2497                                    FILTER_BY_EXACT, "target_obd", &param);
2498         if (rc != 0)
2499                 return -ENOENT;
2500
2501         fp = fopen(param.gl_pathv[0], "r");
2502         if (fp == NULL) {
2503                 rc = -errno;
2504                 llapi_error(LLAPI_MSG_ERROR, rc, "error: opening '%s'",
2505                             param.gl_pathv[0]);
2506                 goto free_param;
2507         }
2508
2509         snprintf(format, sizeof(format),
2510                  "%%d: %%%zus", sizeof(uuidp[0].uuid) - 1);
2511         while (fgets(buf, sizeof(buf), fp) != NULL) {
2512                 if (uuidp && (index < *ost_count)) {
2513                         if (sscanf(buf, format, &index, uuidp[index].uuid) < 2)
2514                                 break;
2515                 }
2516                 index++;
2517         }
2518
2519         fclose(fp);
2520
2521         if (uuidp && (index > *ost_count))
2522                 rc = -EOVERFLOW;
2523
2524         *ost_count = index;
2525 free_param:
2526         cfs_free_param_data(&param);
2527         return rc;
2528 }
2529
2530 int llapi_lov_get_uuids(int fd, struct obd_uuid *uuidp, int *ost_count)
2531 {
2532         return llapi_get_target_uuids(fd, uuidp, ost_count, LOV_TYPE);
2533 }
2534
2535 int llapi_get_obd_count(char *mnt, int *count, int is_mdt)
2536 {
2537         int root;
2538         int rc;
2539
2540         root = open(mnt, O_RDONLY | O_DIRECTORY);
2541         if (root < 0) {
2542                 rc = -errno;
2543                 llapi_error(LLAPI_MSG_ERROR, rc, "open %s failed", mnt);
2544                 return rc;
2545         }
2546
2547         *count = is_mdt;
2548         rc = ioctl(root, LL_IOC_GETOBDCOUNT, count);
2549         if (rc < 0)
2550                 rc = -errno;
2551
2552         close(root);
2553         return rc;
2554 }
2555
2556 /*
2557  * Check if user specified value matches a real uuid.  Ignore _UUID,
2558  * -osc-4ba41334, other trailing gunk in comparison.
2559  * @param real_uuid ends in "_UUID"
2560  * @param search_uuid may or may not end in "_UUID"
2561  */
2562 int llapi_uuid_match(char *real_uuid, char *search_uuid)
2563 {
2564         int cmplen = strlen(real_uuid);
2565         int searchlen = strlen(search_uuid);
2566
2567         if (cmplen > 5 && strcmp(real_uuid + cmplen - 5, "_UUID") == 0)
2568                 cmplen -= 5;
2569         if (searchlen > 5 && strcmp(search_uuid + searchlen - 5, "_UUID") == 0)
2570                 searchlen -= 5;
2571
2572         /*
2573          * The UUIDs may legitimately be different lengths, if
2574          * the system was upgraded from an older version.
2575          */
2576         if (cmplen != searchlen)
2577                 return 0;
2578
2579         return (strncmp(search_uuid, real_uuid, cmplen) == 0);
2580 }
2581
2582 /*
2583  * Here, param->fp_obd_uuid points to a single obduuid, the index of which is
2584  * returned in param->fp_obd_index
2585  */
2586 static int setup_obd_uuid(int fd, char *dname, struct find_param *param)
2587 {
2588         struct obd_uuid obd_uuid;
2589         char buf[PATH_MAX];
2590         glob_t param_data;
2591         char format[32];
2592         int rc = 0;
2593         FILE *fp;
2594
2595         if (param->fp_got_uuids)
2596                 return rc;
2597
2598         /* Get the lov/lmv name */
2599         if (param->fp_get_lmv)
2600                 rc = llapi_file_fget_lmv_uuid(fd, &obd_uuid);
2601         else
2602                 rc = llapi_file_fget_lov_uuid(fd, &obd_uuid);
2603         if (rc) {
2604                 if (rc != -ENOTTY) {
2605                         llapi_error(LLAPI_MSG_ERROR, rc,
2606                                     "error: can't get %s name: %s",
2607                                     param->fp_get_lmv ? "lmv" : "lov",
2608                                     dname);
2609                 } else {
2610                         rc = 0;
2611                 }
2612                 return rc;
2613         }
2614
2615         param->fp_got_uuids = 1;
2616
2617         /* Now get the ost uuids */
2618         rc = get_lustre_param_path(param->fp_get_lmv ? "lmv" : "lov",
2619                                    obd_uuid.uuid, FILTER_BY_EXACT,
2620                                    "target_obd", &param_data);
2621         if (rc != 0)
2622                 return -ENOENT;
2623
2624         fp = fopen(param_data.gl_pathv[0], "r");
2625         if (fp == NULL) {
2626                 rc = -errno;
2627                 llapi_error(LLAPI_MSG_ERROR, rc, "error: opening '%s'",
2628                             param_data.gl_pathv[0]);
2629                 goto free_param;
2630         }
2631
2632         if (!param->fp_obd_uuid && !param->fp_quiet && !param->fp_obds_printed)
2633                 llapi_printf(LLAPI_MSG_NORMAL, "%s:\n",
2634                              param->fp_get_lmv ? "MDTS" : "OBDS");
2635
2636         snprintf(format, sizeof(format),
2637                  "%%d: %%%zus", sizeof(obd_uuid.uuid) - 1);
2638         while (fgets(buf, sizeof(buf), fp) != NULL) {
2639                 int index;
2640
2641                 if (sscanf(buf, format, &index, obd_uuid.uuid) < 2)
2642                         break;
2643
2644                 if (param->fp_obd_uuid) {
2645                         if (llapi_uuid_match(obd_uuid.uuid,
2646                                              param->fp_obd_uuid->uuid)) {
2647                                 param->fp_obd_index = index;
2648                                 break;
2649                         }
2650                 } else if (!param->fp_quiet && !param->fp_obds_printed) {
2651                         /* Print everything */
2652                         llapi_printf(LLAPI_MSG_NORMAL, "%s", buf);
2653                 }
2654         }
2655         param->fp_obds_printed = 1;
2656
2657         fclose(fp);
2658
2659         if (param->fp_obd_uuid && (param->fp_obd_index == OBD_NOT_FOUND)) {
2660                 llapi_err_noerrno(LLAPI_MSG_ERROR,
2661                                   "error: %s: unknown obduuid: %s",
2662                                   __func__, param->fp_obd_uuid->uuid);
2663                 rc = -EINVAL;
2664         }
2665 free_param:
2666         cfs_free_param_data(&param_data);
2667         return rc;
2668 }
2669
2670 /*
2671  * In this case, param->fp_obd_uuid will be an array of obduuids and
2672  * obd index for all these obduuids will be returned in
2673  * param->fp_obd_indexes
2674  */
2675 static int setup_indexes(int d, char *path, struct obd_uuid *obduuids,
2676                          int num_obds, int **obdindexes, int *obdindex,
2677                          enum tgt_type type)
2678 {
2679         int ret, obdcount, obd_valid = 0, obdnum;
2680         long i;
2681         struct obd_uuid *uuids = NULL;
2682         char buf[16];
2683         int *indexes;
2684
2685         if (type == LOV_TYPE)
2686                 ret = get_param_lov(path, "numobd", buf, sizeof(buf));
2687         else
2688                 ret = get_param_lmv(path, "numobd", buf, sizeof(buf));
2689         if (ret != 0)
2690                 return ret;
2691
2692         obdcount = atoi(buf);
2693         uuids = malloc(obdcount * sizeof(struct obd_uuid));
2694         if (uuids == NULL)
2695                 return -ENOMEM;
2696
2697 retry_get_uuids:
2698         ret = llapi_get_target_uuids(d, uuids, &obdcount, type);
2699         if (ret) {
2700                 if (ret == -EOVERFLOW) {
2701                         struct obd_uuid *uuids_temp;
2702
2703                         uuids_temp = realloc(uuids, obdcount *
2704                                              sizeof(struct obd_uuid));
2705                         if (uuids_temp != NULL) {
2706                                 uuids = uuids_temp;
2707                                 goto retry_get_uuids;
2708                         }
2709                         ret = -ENOMEM;
2710                 }
2711
2712                 llapi_error(LLAPI_MSG_ERROR, ret, "cannot get ost uuid");
2713                 goto out_free;
2714         }
2715
2716         indexes = malloc(num_obds * sizeof(*obdindex));
2717         if (indexes == NULL) {
2718                 ret = -ENOMEM;
2719                 goto out_free;
2720         }
2721
2722         for (obdnum = 0; obdnum < num_obds; obdnum++) {
2723                 char *end = NULL;
2724
2725                 /* The user may have specified a simple index */
2726                 i = strtol(obduuids[obdnum].uuid, &end, 0);
2727                 if (end && *end == '\0' && i < obdcount) {
2728                         indexes[obdnum] = i;
2729                         obd_valid++;
2730                 } else {
2731                         for (i = 0; i < obdcount; i++) {
2732                                 if (llapi_uuid_match(uuids[i].uuid,
2733                                                      obduuids[obdnum].uuid)) {
2734                                         indexes[obdnum] = i;
2735                                         obd_valid++;
2736                                         break;
2737                                 }
2738                         }
2739                 }
2740                 if (i >= obdcount) {
2741                         indexes[obdnum] = OBD_NOT_FOUND;
2742                         llapi_err_noerrno(LLAPI_MSG_ERROR,
2743                                           "invalid obduuid '%s'",
2744                                           obduuids[obdnum].uuid);
2745                         ret = -EINVAL;
2746                 }
2747         }
2748
2749         if (obd_valid == 0)
2750                 *obdindex = OBD_NOT_FOUND;
2751         else
2752                 *obdindex = obd_valid;
2753
2754         *obdindexes = indexes;
2755 out_free:
2756         if (uuids)
2757                 free(uuids);
2758
2759         return ret;
2760 }
2761
2762 static int setup_target_indexes(int d, char *path, struct find_param *param)
2763 {
2764         int ret = 0;
2765
2766         if (param->fp_mdt_uuid) {
2767                 ret = setup_indexes(d, path, param->fp_mdt_uuid,
2768                                     param->fp_num_mdts,
2769                                     &param->fp_mdt_indexes,
2770                                     &param->fp_mdt_index, LMV_TYPE);
2771                 if (ret)
2772                         return ret;
2773         }
2774
2775         if (param->fp_obd_uuid) {
2776                 ret = setup_indexes(d, path, param->fp_obd_uuid,
2777                                     param->fp_num_obds,
2778                                     &param->fp_obd_indexes,
2779                                     &param->fp_obd_index, LOV_TYPE);
2780                 if (ret)
2781                         return ret;
2782         }
2783
2784         param->fp_got_uuids = 1;
2785
2786         return ret;
2787 }
2788
2789 int llapi_ostlist(char *path, struct find_param *param)
2790 {
2791         int fd;
2792         int ret;
2793
2794         fd = open(path, O_RDONLY | O_DIRECTORY);
2795         if (fd < 0)
2796                 return -errno;
2797
2798         ret = setup_obd_uuid(fd, path, param);
2799         close(fd);
2800
2801         return ret;
2802 }
2803
2804 /*
2805  * Tries to determine the default stripe attributes for a given filesystem. The
2806  * filesystem to check should be specified by fsname, or will be determined
2807  * using pathname.
2808  */
2809 static int sattr_get_defaults(const char *const fsname,
2810                               unsigned int *scount,
2811                               unsigned int *ssize,
2812                               unsigned int *soffset)
2813 {
2814         char val[PATH_MAX];
2815         int rc;
2816
2817         if (scount) {
2818                 rc = get_lustre_param_value("lov", fsname, FILTER_BY_FS_NAME,
2819                                             "stripecount", val, sizeof(val));
2820                 if (rc != 0)
2821                         return rc;
2822                 *scount = atoi(val);
2823         }
2824
2825         if (ssize) {
2826                 rc = get_lustre_param_value("lov", fsname, FILTER_BY_FS_NAME,
2827                                             "stripesize", val, sizeof(val));
2828                 if (rc != 0)
2829                         return rc;
2830                 *ssize = atoi(val);
2831         }
2832
2833         if (soffset) {
2834                 rc = get_lustre_param_value("lov", fsname, FILTER_BY_FS_NAME,
2835                                             "stripeoffset", val, sizeof(val));
2836                 if (rc != 0)
2837                         return rc;
2838                 *soffset = atoi(val);
2839         }
2840
2841         return 0;
2842 }
2843
2844 /*
2845  * Tries to gather the default stripe attributes for a given filesystem. If
2846  * the attributes can be determined, they are cached for easy retreival the
2847  * next time they are needed. Only a single filesystem's attributes are
2848  * cached at a time.
2849  */
2850 int sattr_cache_get_defaults(const char *const fsname,
2851                              const char *const pathname, unsigned int *scount,
2852                              unsigned int *ssize, unsigned int *soffset)
2853 {
2854         static struct {
2855                 char fsname[PATH_MAX + 1];
2856                 unsigned int stripecount;
2857                 unsigned int stripesize;
2858                 unsigned int stripeoffset;
2859         } cache = {
2860                 .fsname = {'\0'}
2861         };
2862
2863         int rc;
2864         char fsname_buf[PATH_MAX + 1];
2865         unsigned int tmp[3];
2866
2867         if (fsname == NULL) {
2868                 rc = llapi_search_fsname(pathname, fsname_buf);
2869                 if (rc)
2870                         return rc;
2871         } else {
2872                 snprintf(fsname_buf, sizeof(fsname_buf), "%s", fsname);
2873         }
2874
2875         if (strncmp(fsname_buf, cache.fsname, sizeof(fsname_buf) - 1) != 0) {
2876                 /*
2877                  * Ensure all 3 sattrs (count, size, and offset) are
2878                  * successfully retrieved and stored in tmp before writing to
2879                  * cache.
2880                  */
2881                 rc = sattr_get_defaults(fsname_buf, &tmp[0], &tmp[1], &tmp[2]);
2882                 if (rc != 0)
2883                         return rc;
2884
2885                 cache.stripecount = tmp[0];
2886                 cache.stripesize = tmp[1];
2887                 cache.stripeoffset = tmp[2];
2888                 snprintf(cache.fsname, sizeof(cache.fsname), "%s", fsname_buf);
2889         }
2890
2891         if (scount)
2892                 *scount = cache.stripecount;
2893         if (ssize)
2894                 *ssize = cache.stripesize;
2895         if (soffset)
2896                 *soffset = cache.stripeoffset;
2897
2898         return 0;
2899 }
2900
2901 static char *layout2name(__u32 layout_pattern)
2902 {
2903         if (layout_pattern & LOV_PATTERN_F_RELEASED)
2904                 return "released";
2905         else if (layout_pattern == LOV_PATTERN_MDT)
2906                 return "mdt";
2907         else if (layout_pattern == LOV_PATTERN_RAID0)
2908                 return "raid0";
2909         else if (layout_pattern ==
2910                         (LOV_PATTERN_RAID0 | LOV_PATTERN_OVERSTRIPING))
2911                 return "raid0,overstriped";
2912         else
2913                 return "unknown";
2914 }
2915
2916 enum lov_dump_flags {
2917         LDF_IS_DIR      = 0x0001,
2918         LDF_IS_RAW      = 0x0002,
2919         LDF_INDENT      = 0x0004,
2920         LDF_SKIP_OBJS   = 0x0008,
2921         LDF_YAML        = 0x0010,
2922         LDF_EXTENSION   = 0x0020,
2923 };
2924
2925 static void lov_dump_user_lmm_header(struct lov_user_md *lum, char *path,
2926                                      struct lov_user_ost_data_v1 *objects,
2927                                      enum llapi_layout_verbose verbose,
2928                                      int depth, char *pool_name,
2929                                      enum lov_dump_flags flags)
2930 {
2931         bool is_dir = flags & LDF_IS_DIR;
2932         bool is_raw = flags & LDF_IS_RAW;
2933         bool indent = flags & LDF_INDENT;
2934         bool yaml = flags & LDF_YAML;
2935         bool skip_objs = flags & LDF_SKIP_OBJS;
2936         bool extension = flags & LDF_EXTENSION;
2937         char *prefix = is_dir ? "" : "lmm_";
2938         char *separator = "";
2939         char *space = indent ? "      " : "";
2940         int rc;
2941
2942         if (is_dir && lmm_oi_seq(&lum->lmm_oi) == FID_SEQ_LOV_DEFAULT) {
2943                 lmm_oi_set_seq(&lum->lmm_oi, 0);
2944                 if (!indent && (verbose & VERBOSE_DETAIL))
2945                         llapi_printf(LLAPI_MSG_NORMAL, "%s(Default) ", space);
2946         }
2947
2948         if (!yaml && !indent && depth && path &&
2949             ((verbose != VERBOSE_OBJID) || !is_dir))
2950                 llapi_printf(LLAPI_MSG_NORMAL, "%s\n", path);
2951
2952         if ((verbose & VERBOSE_DETAIL) && !is_dir) {
2953                 llapi_printf(LLAPI_MSG_NORMAL, "%s%smagic:         0x%08X\n",
2954                              space, prefix, lum->lmm_magic);
2955                 llapi_printf(LLAPI_MSG_NORMAL, "%s%sseq:           %#jx\n",
2956                              space, prefix,
2957                              (uintmax_t)lmm_oi_seq(&lum->lmm_oi));
2958                 llapi_printf(LLAPI_MSG_NORMAL, "%s%sobject_id:     %#jx\n",
2959                              space, prefix,
2960                              (uintmax_t)lmm_oi_id(&lum->lmm_oi));
2961         }
2962
2963         if (verbose & (VERBOSE_DETAIL | VERBOSE_DFID)) {
2964                 __u64 seq;
2965                 __u32 oid;
2966                 __u32 ver;
2967
2968                 if (verbose & ~VERBOSE_DFID)
2969                         llapi_printf(LLAPI_MSG_NORMAL, "%slmm_fid:           ",
2970                                      space);
2971
2972                 if (is_dir) {
2973                         struct lu_fid dir_fid;
2974
2975                         rc = llapi_path2fid(path, &dir_fid);
2976                         if (rc)
2977                                 llapi_error(LLAPI_MSG_ERROR, rc,
2978                                             "Cannot determine directory fid.");
2979
2980                         seq = dir_fid.f_seq;
2981                         oid = dir_fid.f_oid;
2982                         ver = dir_fid.f_ver;
2983                 } else {
2984                         /*
2985                          * This needs a bit of hand-holding since old 1.x
2986                          * lmm_oi have { oi.oi_id = mds_inum, oi.oi_seq = 0 }
2987                          * and 2.x lmm_oi have { oi.oi_id = mds_oid,
2988                          * oi.oi_seq = mds_seq } instead of a real FID.
2989                          * Ideally the 2.x code would have stored this like a
2990                          * FID with { oi_id = mds_seq, oi_seq = mds_oid } so
2991                          * the ostid union lu_fid { f_seq = mds_seq,
2992                          * f_oid = mds_oid } worked properly (especially since
2993                          * IGIF FIDs use mds_inum as the FID SEQ), but
2994                          * unfortunately that didn't happen.
2995                          *
2996                          * Print it to look like an IGIF FID, even though the
2997                          * fields are reversed on disk, so that it makes sense
2998                          * to userspace.
2999                          *
3000                          * Don't use ostid_id() and ostid_seq(), since they
3001                          * assume the oi_fid fields are in the right order.
3002                          * This is why there are separate lmm_oi_seq() and
3003                          * lmm_oi_id() routines for this.
3004                          *
3005                          * For newer layout types hopefully this will be a
3006                          * real FID.
3007                          */
3008                         seq = lmm_oi_seq(&lum->lmm_oi) == 0 ?
3009                                 lmm_oi_id(&lum->lmm_oi) :
3010                                 lmm_oi_seq(&lum->lmm_oi);
3011                         oid = lmm_oi_seq(&lum->lmm_oi) == 0 ?
3012                             0 : (__u32)lmm_oi_id(&lum->lmm_oi);
3013                         ver = (__u32)(lmm_oi_id(&lum->lmm_oi) >> 32);
3014                 }
3015
3016                 if (yaml)
3017                         llapi_printf(LLAPI_MSG_NORMAL, DFID_NOBRACE"\n",
3018                                      (unsigned long long)seq, oid, ver);
3019                 else
3020                         llapi_printf(LLAPI_MSG_NORMAL, DFID"\n",
3021                                      (unsigned long long)seq, oid, ver);
3022         }
3023
3024         if (verbose & VERBOSE_STRIPE_COUNT) {
3025                 if (verbose & ~VERBOSE_STRIPE_COUNT)
3026                         llapi_printf(LLAPI_MSG_NORMAL, "%s%sstripe_count:  ",
3027                                      space, prefix);
3028                 if (is_dir) {
3029                         if (!is_raw && lum->lmm_stripe_count == 0 &&
3030                             lov_pattern(lum->lmm_pattern) != LOV_PATTERN_MDT) {
3031                                 unsigned int scount;
3032
3033                                 rc = sattr_cache_get_defaults(NULL, path,
3034                                                               &scount, NULL,
3035                                                               NULL);
3036                                 if (rc == 0)
3037                                         llapi_printf(LLAPI_MSG_NORMAL, "%d",
3038                                                      scount);
3039                                 else
3040                                         llapi_error(LLAPI_MSG_ERROR, rc,
3041                                                     "Cannot determine default stripe count.");
3042                         } else {
3043                                 llapi_printf(LLAPI_MSG_NORMAL, "%d",
3044                                              extension ? 0 :
3045                                              (__s16)lum->lmm_stripe_count);
3046                         }
3047                 } else {
3048                         llapi_printf(LLAPI_MSG_NORMAL, "%hd",
3049                                      extension ? 0 :
3050                                      (__s16)lum->lmm_stripe_count);
3051                 }
3052                 if (!yaml && is_dir)
3053                         separator = " ";
3054                 else
3055                         separator = "\n";
3056         }
3057
3058         if (((verbose & VERBOSE_STRIPE_SIZE) && !extension) ||
3059             ((verbose & VERBOSE_EXT_SIZE) && extension)) {
3060                 llapi_printf(LLAPI_MSG_NORMAL, "%s", separator);
3061                 if (verbose & ~VERBOSE_EXT_SIZE && extension)
3062                         llapi_printf(LLAPI_MSG_NORMAL, "%s%sextension_size: ",
3063                                      space, prefix);
3064                 if (verbose & ~VERBOSE_STRIPE_SIZE && !extension)
3065                         llapi_printf(LLAPI_MSG_NORMAL, "%s%sstripe_size:   ",
3066                                      space, prefix);
3067                 if (is_dir && !is_raw && lum->lmm_stripe_size == 0) {
3068                         unsigned int ssize;
3069
3070                         rc = sattr_cache_get_defaults(NULL, path, NULL, &ssize,
3071                                                       NULL);
3072                         if (rc == 0)
3073                                 llapi_printf(LLAPI_MSG_NORMAL, "%u", ssize);
3074                         else
3075                                 llapi_error(LLAPI_MSG_ERROR, rc,
3076                                             "Cannot determine default stripe size.");
3077                 } else {
3078                         /* Extension size is in KiB */
3079                         llapi_printf(LLAPI_MSG_NORMAL, "%llu",
3080                                      extension ?
3081                                      (unsigned long long)(lum->lmm_stripe_size * SEL_UNIT_SIZE) :
3082                                      (unsigned long long)lum->lmm_stripe_size);
3083                 }
3084                 if (!yaml && is_dir)
3085                         separator = " ";
3086                 else
3087                         separator = "\n";
3088         }
3089
3090         if ((verbose & VERBOSE_PATTERN)) {
3091                 llapi_printf(LLAPI_MSG_NORMAL, "%s", separator);
3092                 if (verbose & ~VERBOSE_PATTERN)
3093                         llapi_printf(LLAPI_MSG_NORMAL, "%s%spattern:       ",
3094                                      space, prefix);
3095                 if (lov_pattern_supported(lum->lmm_pattern))
3096                         llapi_printf(LLAPI_MSG_NORMAL, "%s",
3097                                      layout2name(lum->lmm_pattern));
3098                 else
3099                         llapi_printf(LLAPI_MSG_NORMAL, "%x", lum->lmm_pattern);
3100                 separator = (!yaml && is_dir) ? " " : "\n";
3101         }
3102
3103         if ((verbose & VERBOSE_GENERATION) && !is_dir) {
3104                 llapi_printf(LLAPI_MSG_NORMAL, "%s", separator);
3105                 if (verbose & ~VERBOSE_GENERATION)
3106                         llapi_printf(LLAPI_MSG_NORMAL, "%s%slayout_gen:    ",
3107                                      space, prefix);
3108                 llapi_printf(LLAPI_MSG_NORMAL, "%u",
3109                              skip_objs ? 0 : (int)lum->lmm_layout_gen);
3110                 separator = "\n";
3111         }
3112
3113         if (verbose & VERBOSE_STRIPE_OFFSET) {
3114                 llapi_printf(LLAPI_MSG_NORMAL, "%s", separator);
3115                 if (verbose & ~VERBOSE_STRIPE_OFFSET)
3116                         llapi_printf(LLAPI_MSG_NORMAL, "%s%sstripe_offset: ",
3117                                      space, prefix);
3118                 if (is_dir || skip_objs)
3119                         llapi_printf(LLAPI_MSG_NORMAL, "%d",
3120                                      lum->lmm_stripe_offset ==
3121                                      (typeof(lum->lmm_stripe_offset))(-1) ? -1 :
3122                                      lum->lmm_stripe_offset);
3123                 else if (lov_pattern(lum->lmm_pattern) == LOV_PATTERN_MDT)
3124                         llapi_printf(LLAPI_MSG_NORMAL, "0");
3125                 else
3126                         llapi_printf(LLAPI_MSG_NORMAL, "%u",
3127                                      objects[0].l_ost_idx);
3128                 if (!yaml && is_dir)
3129                         separator = " ";
3130                 else
3131                         separator = "\n";