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