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