Whamcloud - gitweb
LU-2430 utils: fix "lfs mv" command parsing
[fs/lustre-release.git] / lustre / utils / lfs.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, 2014, 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/lfs.c
37  *
38  * Author: Peter J. Braam <braam@clusterfs.com>
39  * Author: Phil Schwan <phil@clusterfs.com>
40  * Author: Robert Read <rread@clusterfs.com>
41  */
42
43 /* for O_DIRECTORY */
44 #ifndef _GNU_SOURCE
45 #define _GNU_SOURCE
46 #endif
47
48 #include <stdlib.h>
49 #include <stdio.h>
50 #include <getopt.h>
51 #include <string.h>
52 #include <mntent.h>
53 #include <errno.h>
54 #include <err.h>
55 #include <pwd.h>
56 #include <grp.h>
57 #include <sys/quota.h>
58 #include <sys/types.h>
59 #include <sys/stat.h>
60 #include <fcntl.h>
61 #include <dirent.h>
62 #include <time.h>
63 #include <ctype.h>
64 #ifdef HAVE_SYS_QUOTA_H
65 # include <sys/quota.h>
66 #endif
67
68 #include <libcfs/libcfs.h>
69 #include <libcfs/libcfsutil.h>
70 #include <lustre/lustreapi.h>
71 #include <lustre_ver.h>
72
73 /* all functions */
74 static int lfs_setstripe(int argc, char **argv);
75 static int lfs_find(int argc, char **argv);
76 static int lfs_getstripe(int argc, char **argv);
77 static int lfs_getdirstripe(int argc, char **argv);
78 static int lfs_setdirstripe(int argc, char **argv);
79 static int lfs_rmentry(int argc, char **argv);
80 static int lfs_osts(int argc, char **argv);
81 static int lfs_mdts(int argc, char **argv);
82 static int lfs_df(int argc, char **argv);
83 static int lfs_getname(int argc, char **argv);
84 static int lfs_check(int argc, char **argv);
85 #ifdef HAVE_SYS_QUOTA_H
86 static int lfs_quotacheck(int argc, char **argv);
87 static int lfs_quotaon(int argc, char **argv);
88 static int lfs_quotaoff(int argc, char **argv);
89 static int lfs_setquota(int argc, char **argv);
90 static int lfs_quota(int argc, char **argv);
91 #endif
92 static int lfs_flushctx(int argc, char **argv);
93 static int lfs_join(int argc, char **argv);
94 static int lfs_lsetfacl(int argc, char **argv);
95 static int lfs_lgetfacl(int argc, char **argv);
96 static int lfs_rsetfacl(int argc, char **argv);
97 static int lfs_rgetfacl(int argc, char **argv);
98 static int lfs_cp(int argc, char **argv);
99 static int lfs_ls(int argc, char **argv);
100 static int lfs_poollist(int argc, char **argv);
101 static int lfs_changelog(int argc, char **argv);
102 static int lfs_changelog_clear(int argc, char **argv);
103 static int lfs_fid2path(int argc, char **argv);
104 static int lfs_path2fid(int argc, char **argv);
105 static int lfs_data_version(int argc, char **argv);
106 static int lfs_hsm_state(int argc, char **argv);
107 static int lfs_hsm_set(int argc, char **argv);
108 static int lfs_hsm_clear(int argc, char **argv);
109 static int lfs_hsm_action(int argc, char **argv);
110 static int lfs_hsm_archive(int argc, char **argv);
111 static int lfs_hsm_restore(int argc, char **argv);
112 static int lfs_hsm_release(int argc, char **argv);
113 static int lfs_hsm_remove(int argc, char **argv);
114 static int lfs_hsm_cancel(int argc, char **argv);
115 static int lfs_swap_layouts(int argc, char **argv);
116 static int lfs_mv(int argc, char **argv);
117
118 #define SETSTRIPE_USAGE(_cmd, _tgt) \
119         "usage: "_cmd" [--stripe-count|-c <stripe_count>]\n"\
120         "                 [--stripe-index|-i <start_ost_idx>]\n"\
121         "                 [--stripe-size|-S <stripe_size>]\n"\
122         "                 [--pool|-p <pool_name>]\n"\
123         "                 [--block|-b] "_tgt"\n"\
124         "                 [--ost-list|-o <ost_indices>]\n"\
125         "\tstripe_size:  Number of bytes on each OST (0 filesystem default)\n"\
126         "\t              Can be specified with k, m or g (in KB, MB and GB\n"\
127         "\t              respectively)\n"\
128         "\tstart_ost_idx: OST index of first stripe (-1 default)\n"\
129         "\tstripe_count: Number of OSTs to stripe over (0 default, -1 all)\n"\
130         "\tpool_name:    Name of OST pool to use (default none)\n"\
131         "\tblock:        Block file access during data migration\n"\
132         "\tost_indices:  List of OST indices, can be repeated multiple times\n"\
133         "\t              Indices be specified in a format of:\n"\
134         "\t                -o <ost_1>,<ost_i>-<ost_j>,<ost_n>\n"\
135         "\t              Or:\n"\
136         "\t                -o <ost_1> -o <ost_i>-<ost_j> -o <ost_n>\n"\
137         "\t              If --pool is set with --ost-list, then the OSTs\n"\
138         "\t              must be the members of the pool."
139
140 /* all avaialable commands */
141 command_t cmdlist[] = {
142         {"setstripe", lfs_setstripe, 0,
143          "Create a new file with a specific striping pattern or\n"
144          "set the default striping pattern on an existing directory or\n"
145          "delete the default striping pattern from an existing directory\n"
146          "usage: setstripe -d <directory>   (to delete default striping)\n"\
147          " or\n"
148          SETSTRIPE_USAGE("setstripe", "<directory|filename>")},
149         {"getstripe", lfs_getstripe, 0,
150          "To list the striping info for a given file or files in a\n"
151          "directory or recursively for all files in a directory tree.\n"
152          "usage: getstripe [--ost|-O <uuid>] [--quiet | -q] [--verbose | -v]\n"
153          "                 [--stripe-count|-c] [--stripe-index|-i]\n"
154          "                 [--pool|-p] [--stripe-size|-S] [--directory|-d]\n"
155          "                 [--mdt-index|-M] [--recursive|-r] [--raw|-R]\n"
156          "                 [--layout|-L]\n"
157          "                 <directory|filename> ..."},
158         {"setdirstripe", lfs_setdirstripe, 0,
159          "To create a striped directory on a specified MDT. This can only\n"
160          "be done on MDT0 with the right of administrator.\n"
161          "usage: setdirstripe <--count|-c stripe_count>\n"
162          "              [--index|-i mdt_index] [--hash-type|-t hash_type]\n"
163          "              [--default_stripe|-D ] [--mode|-m mode] <dir>\n"
164          "\tstripe_count: stripe count of the striped directory\n"
165          "\tmdt_index:  MDT index of first stripe\n"
166          "\thash_type:  hash type of the striped directory. Hash types:\n"
167          "      fnv_1a_64 FNV-1a hash algorithm (default)\n"
168          "      all_char  sum of characters % MDT_COUNT (not recommended)\n"
169          "\tdefault_stripe: set default dirstripe of the directory\n"
170          "\tmode: the mode of the directory\n"},
171         {"getdirstripe", lfs_getdirstripe, 0,
172          "To list the striping info for a given directory\n"
173          "or recursively for all directories in a directory tree.\n"
174          "usage: getdirstripe [--obd|-O <uuid>] [--quiet|-q] [--verbose|-v]\n"
175          "               [--count|-c ] [--index|-i ] [--raw|-R]\n"
176          "               [--recursive | -r] [ --default_stripe | -D ] <dir> "},
177         {"mkdir", lfs_setdirstripe, 0,
178          "To create a striped directory on a specified MDT. This can only\n"
179          "be done on MDT0 with the right of administrator.\n"
180          "usage: mkdir <--count|-c stripe_count>\n"
181          "              [--index|-i mdt_index] [--hash-type|-t hash_type]\n"
182          "              [--default_stripe|-D ] [--mode|-m mode] <dir>\n"
183          "\tstripe_count: stripe count of the striped directory\n"
184          "\tmdt_index:  MDT index of first stripe\n"
185          "\thash_type:  hash type of the striped directory. Hash types:\n"
186          "      fnv_1a_64 FNV-1a hash algorithm (default)\n"
187          "      all_char  sum of characters % MDT_COUNT (not recommended)\n"
188          "\tdefault_stripe: set default dirstripe of the directory\n"
189          "\tmode: the mode of the directory\n"},
190         {"rm_entry", lfs_rmentry, 0,
191          "To remove the name entry of the remote directory. Note: This\n"
192          "command will only delete the name entry, i.e. the remote directory\n"
193          "will become inaccessable after this command. This can only be done\n"
194          "by the administrator\n"
195          "usage: rm_entry <dir>\n"},
196         {"pool_list", lfs_poollist, 0,
197          "List pools or pool OSTs\n"
198          "usage: pool_list <fsname>[.<pool>] | <pathname>\n"},
199         {"find", lfs_find, 0,
200          "find files matching given attributes recursively in directory tree.\n"
201          "usage: find <directory|filename> ...\n"
202          "     [[!] --atime|-A [+-]N] [[!] --ctime|-C [+-]N]\n"
203          "     [[!] --mtime|-M [+-]N] [[!] --mdt|-m <uuid|index,...>]\n"
204          "     [--maxdepth|-D N] [[!] --name|-n <pattern>]\n"
205          "     [[!] --ost|-O <uuid|index,...>] [--print|-p] [--print0|-P]\n"
206          "     [[!] --size|-s [+-]N[bkMGTPE]]\n"
207          "     [[!] --stripe-count|-c [+-]<stripes>]\n"
208          "     [[!] --stripe-index|-i <index,...>]\n"
209          "     [[!] --stripe-size|-S [+-]N[kMGT]] [[!] --type|-t <filetype>]\n"
210          "     [[!] --gid|-g|--group|-G <gid>|<gname>]\n"
211          "     [[!] --uid|-u|--user|-U <uid>|<uname>] [[!] --pool <pool>]\n"
212          "     [[!] --layout|-L released,raid0]\n"
213          "\t !: used before an option indicates 'NOT' requested attribute\n"
214          "\t -: used before a value indicates 'AT MOST' requested value\n"
215          "\t +: used before a value indicates 'AT LEAST' requested value\n"},
216         {"check", lfs_check, 0,
217          "Display the status of MDS or OSTs (as specified in the command)\n"
218          "or all the servers (MDS and OSTs).\n"
219          "usage: check <osts|mds|servers>"},
220         {"join", lfs_join, 0,
221          "join two lustre files into one.\n"
222          "obsolete, HEAD does not support it anymore.\n"},
223         {"osts", lfs_osts, 0, "list OSTs connected to client "
224          "[for specified path only]\n" "usage: osts [path]"},
225         {"mdts", lfs_mdts, 0, "list MDTs connected to client "
226          "[for specified path only]\n" "usage: mdts [path]"},
227         {"df", lfs_df, 0,
228          "report filesystem disk space usage or inodes usage"
229          "of each MDS and all OSDs or a batch belonging to a specific pool .\n"
230          "Usage: df [-i] [-h] [--lazy|-l] [--pool|-p <fsname>[.<pool>] [path]"},
231         {"getname", lfs_getname, 0, "list instances and specified mount points "
232          "[for specified path only]\n"
233          "Usage: getname [-h]|[path ...] "},
234 #ifdef HAVE_SYS_QUOTA_H
235         {"quotacheck", lfs_quotacheck, 0,
236          "Scan the specified filesystem for disk usage, and create,\n"
237          "or update quota files. Deprecated as of 2.4.0.\n"
238          "usage: quotacheck [ -ug ] <filesystem>"},
239         {"quotaon", lfs_quotaon, 0, "Turn filesystem"
240          " quotas on. Deprecated as of 2.4.0.\n"
241          "usage: quotaon [ -ugf ] <filesystem>"},
242         {"quotaoff", lfs_quotaoff, 0, "Turn filesystem"
243          " quotas off. Deprecated as of 2.4.0.\n"
244          "usage: quotaoff [ -ug ] <filesystem>"},
245         {"setquota", lfs_setquota, 0, "Set filesystem quotas.\n"
246          "usage: setquota <-u|-g> <uname>|<uid>|<gname>|<gid>\n"
247          "                -b <block-softlimit> -B <block-hardlimit>\n"
248          "                -i <inode-softlimit> -I <inode-hardlimit> <filesystem>\n"
249          "       setquota <-u|--user|-g|--group> <uname>|<uid>|<gname>|<gid>\n"
250          "                [--block-softlimit <block-softlimit>]\n"
251          "                [--block-hardlimit <block-hardlimit>]\n"
252          "                [--inode-softlimit <inode-softlimit>]\n"
253          "                [--inode-hardlimit <inode-hardlimit>] <filesystem>\n"
254          "       setquota [-t] <-u|--user|-g|--group>\n"
255          "                [--block-grace <block-grace>]\n"
256          "                [--inode-grace <inode-grace>] <filesystem>\n"
257          "       -b can be used instead of --block-softlimit/--block-grace\n"
258          "       -B can be used instead of --block-hardlimit\n"
259          "       -i can be used instead of --inode-softlimit/--inode-grace\n"
260          "       -I can be used instead of --inode-hardlimit\n\n"
261          "Note: The total quota space will be split into many qunits and\n"
262          "      balanced over all server targets, the minimal qunit size is\n"
263          "      1M bytes for block space and 1K inodes for inode space.\n\n"
264          "      Quota space rebalancing process will stop when this mininum\n"
265          "      value is reached. As a result, quota exceeded can be returned\n"
266          "      while many targets still have 1MB or 1K inodes of spare\n"
267          "      quota space."},
268         {"quota", lfs_quota, 0, "Display disk usage and limits.\n"
269          "usage: quota [-q] [-v] [-h] [-o <obd_uuid>|-i <mdt_idx>|-I "
270                        "<ost_idx>]\n"
271          "             [<-u|-g> <uname>|<uid>|<gname>|<gid>] <filesystem>\n"
272          "       quota [-o <obd_uuid>|-i <mdt_idx>|-I <ost_idx>] -t <-u|-g> <filesystem>"},
273 #endif
274         {"flushctx", lfs_flushctx, 0, "Flush security context for current user.\n"
275          "usage: flushctx [-k] [mountpoint...]"},
276         {"lsetfacl", lfs_lsetfacl, 0,
277          "Remote user setfacl for user/group on the same remote client.\n"
278          "usage: lsetfacl [-bkndRLPvh] [{-m|-x} acl_spec] [{-M|-X} acl_file] file ..."},
279         {"lgetfacl", lfs_lgetfacl, 0,
280          "Remote user getfacl for user/group on the same remote client.\n"
281          "usage: lgetfacl [-dRLPvh] file ..."},
282         {"rsetfacl", lfs_rsetfacl, 0,
283          "Remote user setfacl for user/group on other clients.\n"
284          "usage: rsetfacl [-bkndRLPvh] [{-m|-x} acl_spec] [{-M|-X} acl_file] file ..."},
285         {"rgetfacl", lfs_rgetfacl, 0,
286          "Remote user getfacl for user/group on other clients.\n"
287          "usage: rgetfacl [-dRLPvh] file ..."},
288         {"cp", lfs_cp, 0,
289          "Remote user copy files and directories.\n"
290          "usage: cp [OPTION]... [-T] SOURCE DEST\n\tcp [OPTION]... SOURCE... DIRECTORY\n\tcp [OPTION]... -t DIRECTORY SOURCE..."},
291         {"ls", lfs_ls, 0,
292          "Remote user list directory contents.\n"
293          "usage: ls [OPTION]... [FILE]..."},
294         {"changelog", lfs_changelog, 0,
295          "Show the metadata changes on an MDT."
296          "\nusage: changelog <mdtname> [startrec [endrec]]"},
297         {"changelog_clear", lfs_changelog_clear, 0,
298          "Indicate that old changelog records up to <endrec> are no longer of "
299          "interest to consumer <id>, allowing the system to free up space.\n"
300          "An <endrec> of 0 means all records.\n"
301          "usage: changelog_clear <mdtname> <id> <endrec>"},
302         {"fid2path", lfs_fid2path, 0,
303          "Resolve the full path(s) for given FID(s). For a specific hardlink "
304          "specify link number <linkno>.\n"
305         /* "For a historical link name, specify changelog record <recno>.\n" */
306          "usage: fid2path [--link <linkno>] <fsname|rootpath> <fid> ..."
307                 /* [ --rec <recno> ] */ },
308         {"path2fid", lfs_path2fid, 0, "Display the fid(s) for a given path(s).\n"
309          "usage: path2fid [--parents] <path> ..."},
310         {"data_version", lfs_data_version, 0, "Display file data version for "
311          "a given path.\n" "usage: data_version -[n|r|w] <path>"},
312         {"hsm_state", lfs_hsm_state, 0, "Display the HSM information (states, "
313          "undergoing actions) for given files.\n usage: hsm_state <file> ..."},
314         {"hsm_set", lfs_hsm_set, 0, "Set HSM user flag on specified files.\n"
315          "usage: hsm_set [--norelease] [--noarchive] [--dirty] [--exists] "
316          "[--archived] [--lost] <file> ..."},
317         {"hsm_clear", lfs_hsm_clear, 0, "Clear HSM user flag on specified "
318          "files.\n"
319          "usage: hsm_clear [--norelease] [--noarchive] [--dirty] [--exists] "
320          "[--archived] [--lost] <file> ..."},
321         {"hsm_action", lfs_hsm_action, 0, "Display current HSM request for "
322          "given files.\n" "usage: hsm_action <file> ..."},
323         {"hsm_archive", lfs_hsm_archive, 0,
324          "Archive file to external storage.\n"
325          "usage: hsm_archive [--filelist FILELIST] [--data DATA] [--archive NUM] "
326          "<file> ..."},
327         {"hsm_restore", lfs_hsm_restore, 0,
328          "Restore file from external storage.\n"
329          "usage: hsm_restore [--filelist FILELIST] [--data DATA] <file> ..."},
330         {"hsm_release", lfs_hsm_release, 0,
331          "Release files from Lustre.\n"
332          "usage: hsm_release [--filelist FILELIST] [--data DATA] <file> ..."},
333         {"hsm_remove", lfs_hsm_remove, 0,
334          "Remove file copy from external storage.\n"
335          "usage: hsm_remove [--filelist FILELIST] [--data DATA] <file> ..."},
336         {"hsm_cancel", lfs_hsm_cancel, 0,
337          "Cancel requests related to specified files.\n"
338          "usage: hsm_cancel [--filelist FILELIST] [--data DATA] <file> ..."},
339         {"swap_layouts", lfs_swap_layouts, 0, "Swap layouts between 2 files.\n"
340          "usage: swap_layouts <path1> <path2>"},
341         {"migrate", lfs_setstripe, 0, "migrate file from one OST layout to "
342          "another (may be not safe with concurent writes).\n"
343          SETSTRIPE_USAGE("migrate  ", "<filename>")},
344         {"mv", lfs_mv, 0,
345          "To move directories between MDTs.\n"
346          "usage: mv <directory|filename> [--mdt-index|-M] <mdt_index> "
347          "[--verbose|-v]\n"},
348         {"help", Parser_help, 0, "help"},
349         {"exit", Parser_quit, 0, "quit"},
350         {"quit", Parser_quit, 0, "quit"},
351         {"--version", Parser_version, 0,
352          "output build version of the utility and exit"},
353         { 0, 0, 0, NULL }
354 };
355
356 #define MIGRATION_BLOCKS 1
357
358 static int lfs_migrate(char *name, __u64 migration_flags,
359                        struct llapi_stripe_param *param)
360 {
361         int                      fd, fdv;
362         char                     volatile_file[PATH_MAX +
363                                                 LUSTRE_VOLATILE_HDR_LEN + 4];
364         char                     parent[PATH_MAX];
365         char                    *ptr;
366         int                      rc;
367         __u64                    dv1;
368         struct lov_user_md      *lum = NULL;
369         int                      lumsz;
370         int                      bufsz;
371         void                    *buf = NULL;
372         int                      rsize, wsize;
373         __u64                    rpos, wpos, bufoff;
374         int                      gid;
375         int                      have_gl = 0;
376         struct stat              st, stv;
377
378         /* find the right size for the IO and allocate the buffer */
379         lumsz = lov_user_md_size(LOV_MAX_STRIPE_COUNT, LOV_USER_MAGIC_V3);
380         lum = malloc(lumsz);
381         if (lum == NULL) {
382                 rc = -ENOMEM;
383                 goto free;
384         }
385
386         rc = llapi_file_get_stripe(name, lum);
387         /* failure can come from may case and some may be not real error
388          * (eg: no stripe)
389          * in case of a real error, a later call will failed with a better
390          * error management */
391         if (rc < 0)
392                 bufsz = 1024*1024;
393         else
394                 bufsz = lum->lmm_stripe_size;
395         rc = posix_memalign(&buf, getpagesize(), bufsz);
396         if (rc != 0) {
397                 rc = -rc;
398                 goto free;
399         }
400
401         /* search for file directory pathname */
402         if (strlen(name) > sizeof(parent)-1) {
403                 rc = -E2BIG;
404                 goto free;
405         }
406         strncpy(parent, name, sizeof(parent));
407         ptr = strrchr(parent, '/');
408         if (ptr == NULL) {
409                 if (getcwd(parent, sizeof(parent)) == NULL) {
410                         rc = -errno;
411                         goto free;
412                 }
413         } else {
414                 if (ptr == parent)
415                         strcpy(parent, "/");
416                 else
417                         *ptr = '\0';
418         }
419         rc = snprintf(volatile_file, sizeof(volatile_file), "%s/%s::", parent,
420                       LUSTRE_VOLATILE_HDR);
421         if (rc >= sizeof(volatile_file)) {
422                 rc = -E2BIG;
423                 goto free;
424         }
425
426         /* create, open a volatile file, use caching (ie no directio) */
427         /* exclusive create is not needed because volatile files cannot
428          * conflict on name by construction */
429         fdv = llapi_file_open_param(volatile_file, O_CREAT | O_WRONLY, 0644,
430                                     param);
431         if (fdv < 0) {
432                 rc = fdv;
433                 fprintf(stderr, "cannot create volatile file in %s (%s)\n",
434                         parent, strerror(-rc));
435                 goto free;
436         }
437
438         /* open file, direct io */
439         /* even if the file is only read, WR mode is nedeed to allow
440          * layout swap on fd */
441         fd = open(name, O_RDWR | O_DIRECT);
442         if (fd == -1) {
443                 rc = -errno;
444                 fprintf(stderr, "cannot open %s (%s)\n", name, strerror(-rc));
445                 close(fdv);
446                 goto free;
447         }
448
449         /* Not-owner (root?) special case.
450          * Need to set owner/group of volatile file like original.
451          * This will allow to pass related check during layout_swap.
452          */
453         rc = fstat(fd, &st);
454         if (rc != 0) {
455                 rc = -errno;
456                 fprintf(stderr, "cannot stat %s (%s)\n", name,
457                         strerror(errno));
458                 goto error;
459         }
460         rc = fstat(fdv, &stv);
461         if (rc != 0) {
462                 rc = -errno;
463                 fprintf(stderr, "cannot stat %s (%s)\n", volatile_file,
464                         strerror(errno));
465                 goto error;
466         }
467         if (st.st_uid != stv.st_uid || st.st_gid != stv.st_gid) {
468                 rc = fchown(fdv, st.st_uid, st.st_gid);
469                 if (rc != 0) {
470                         rc = -errno;
471                         fprintf(stderr, "cannot chown %s (%s)\n", name,
472                                 strerror(errno));
473                         goto error;
474                 }
475         }
476
477         /* get file data version */
478         rc = llapi_get_data_version(fd, &dv1, LL_DV_RD_FLUSH);
479         if (rc != 0) {
480                 fprintf(stderr, "cannot get dataversion on %s (%s)\n",
481                         name, strerror(-rc));
482                 goto error;
483         }
484
485         do
486                 gid = random();
487         while (gid == 0);
488         if (migration_flags & MIGRATION_BLOCKS) {
489                 /* take group lock to limit concurent access
490                  * this will be no more needed when exclusive access will
491                  * be implemented (see LU-2919) */
492                 /* group lock is taken after data version read because it
493                  * blocks data version call */
494                 rc = llapi_group_lock(fd, gid);
495                 if (rc < 0) {
496                         fprintf(stderr, "cannot get group lock on %s (%s)\n",
497                                 name, strerror(-rc));
498                         goto error;
499                 }
500                 have_gl = 1;
501         }
502
503         /* copy data */
504         rpos = 0;
505         wpos = 0;
506         bufoff = 0;
507         rsize = -1;
508         do {
509                 /* read new data only if we have written all
510                  * previously read data */
511                 if (wpos == rpos) {
512                         rsize = read(fd, buf, bufsz);
513                         if (rsize < 0) {
514                                 rc = -errno;
515                                 fprintf(stderr, "read failed on %s"
516                                         " (%s)\n", name,
517                                         strerror(-rc));
518                                 goto error;
519                         }
520                         rpos += rsize;
521                         bufoff = 0;
522                 }
523                 /* eof ? */
524                 if (rsize == 0)
525                         break;
526                 wsize = write(fdv, buf + bufoff, rpos - wpos);
527                 if (wsize < 0) {
528                         rc = -errno;
529                         fprintf(stderr, "write failed on volatile"
530                                 " for %s (%s)\n", name, strerror(-rc));
531                         goto error;
532                 }
533                 wpos += wsize;
534                 bufoff += wsize;
535         } while (1);
536
537         /* flush data */
538         fsync(fdv);
539
540         if (migration_flags & MIGRATION_BLOCKS) {
541                 /* give back group lock */
542                 rc = llapi_group_unlock(fd, gid);
543                 if (rc < 0)
544                         fprintf(stderr, "cannot put group lock on %s (%s)\n",
545                                 name, strerror(-rc));
546                 have_gl = 0;
547         }
548
549         /* swap layouts
550          * for a migration we need to:
551          * - check data version on file did not change
552          * - keep file mtime
553          * - keep file atime
554          */
555         rc = llapi_fswap_layouts(fd, fdv, dv1, 0,
556                                  SWAP_LAYOUTS_CHECK_DV1 |
557                                  SWAP_LAYOUTS_KEEP_MTIME |
558                                  SWAP_LAYOUTS_KEEP_ATIME);
559         if (rc == -EAGAIN) {
560                 fprintf(stderr, "%s: dataversion changed during copy, "
561                         "migration aborted\n", name);
562                 goto error;
563         }
564         if (rc != 0)
565                 fprintf(stderr, "%s: swap layout to new file failed: %s\n",
566                         name, strerror(-rc));
567
568 error:
569         /* give back group lock */
570         if ((migration_flags & MIGRATION_BLOCKS) && have_gl) {
571                 int rc2;
572
573                 /* we keep the original error in rc */
574                 rc2 = llapi_group_unlock(fd, gid);
575                 if (rc2 < 0)
576                         fprintf(stderr, "cannot put group lock on %s (%s)\n",
577                                 name, strerror(-rc2));
578         }
579
580         close(fdv);
581         close(fd);
582 free:
583         if (lum)
584                 free(lum);
585         if (buf)
586                 free(buf);
587         return rc;
588 }
589
590 /**
591  * Parse a string containing an OST index list into an array of integers.
592  *
593  * The input string contains a comma delimited list of individual
594  * indices and ranges, for example "1,2-4,7". Add the indices into the
595  * \a osts array and remove duplicates.
596  *
597  * \param[out] osts    array to store indices in
598  * \param[in] size     size of \a osts array
599  * \param[in] offset   starting index in \a osts
600  * \param[in] arg      string containing OST index list
601  *
602  * \retval positive    number of indices in \a osts
603  * \retval -EINVAL     unable to parse \a arg
604  */
605 static int parse_targets(__u32 *osts, int size, int offset, char *arg)
606 {
607         int rc;
608         int nr = offset;
609         int slots = size - offset;
610         char *ptr = NULL;
611         bool end_of_loop;
612
613         if (arg == NULL)
614                 return -EINVAL;
615
616         end_of_loop = false;
617         while (!end_of_loop) {
618                 int start_index;
619                 int end_index;
620                 int i;
621                 char *endptr = NULL;
622
623                 rc = -EINVAL;
624
625                 ptr = strchrnul(arg, ',');
626
627                 end_of_loop = *ptr == '\0';
628                 *ptr = '\0';
629
630                 start_index = strtol(arg, &endptr, 0);
631                 if (endptr == arg) /* no data at all */
632                         break;
633                 if (*endptr != '-' && *endptr != '\0') /* has invalid data */
634                         break;
635                 if (start_index < 0)
636                         break;
637
638                 end_index = start_index;
639                 if (*endptr == '-') {
640                         end_index = strtol(endptr + 1, &endptr, 0);
641                         if (*endptr != '\0')
642                                 break;
643                         if (end_index < start_index)
644                                 break;
645                 }
646
647                 for (i = start_index; i <= end_index && slots > 0; i++) {
648                         int j;
649
650                         /* remove duplicate */
651                         for (j = 0; j < offset; j++) {
652                                 if (osts[j] == i)
653                                         break;
654                         }
655                         if (j == offset) { /* no duplicate */
656                                 osts[nr++] = i;
657                                 --slots;
658                         }
659                 }
660                 if (slots == 0 && i < end_index)
661                         break;
662
663                 *ptr = ',';
664                 arg = ++ptr;
665                 offset = nr;
666                 rc = 0;
667         }
668         if (!end_of_loop && ptr != NULL)
669                 *ptr = ',';
670
671         return rc < 0 ? rc : nr;
672 }
673
674 /* functions */
675 static int lfs_setstripe(int argc, char **argv)
676 {
677         struct llapi_stripe_param       *param;
678         char                            *fname;
679         int                              result;
680         unsigned long long               st_size;
681         int                              st_offset, st_count;
682         char                            *end;
683         int                              c;
684         int                              delete = 0;
685         char                            *stripe_size_arg = NULL;
686         char                            *stripe_off_arg = NULL;
687         char                            *stripe_count_arg = NULL;
688         char                            *pool_name_arg = NULL;
689         unsigned long long               size_units = 1;
690         bool                             migrate_mode = false;
691         __u64                            migration_flags = 0;
692         __u32                            osts[LOV_MAX_STRIPE_COUNT] = { 0 };
693         int                              nr_osts = 0;
694
695         struct option            long_opts[] = {
696                 /* valid only in migrate mode */
697                 {"block",        no_argument,       0, 'b'},
698 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 9, 53, 0)
699                 /* This formerly implied "stripe-count", but was explicitly
700                  * made "stripe-count" for consistency with other options,
701                  * and to separate it from "mdt-count" when DNE arrives. */
702                 {"count",        required_argument, 0, 'c'},
703 #endif
704                 {"stripe-count", required_argument, 0, 'c'},
705                 {"stripe_count", required_argument, 0, 'c'},
706                 {"delete",       no_argument,       0, 'd'},
707 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 9, 53, 0)
708                 /* This formerly implied "stripe-index", but was explicitly
709                  * made "stripe-index" for consistency with other options,
710                  * and to separate it from "mdt-index" when DNE arrives. */
711                 {"index",        required_argument, 0, 'i'},
712 #endif
713                 {"stripe-index", required_argument, 0, 'i'},
714                 {"stripe_index", required_argument, 0, 'i'},
715                 {"ost-list",     required_argument, 0, 'o'},
716                 {"ost_list",     required_argument, 0, 'o'},
717                 {"pool",         required_argument, 0, 'p'},
718 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 9, 53, 0)
719                 /* This formerly implied "--stripe-size", but was confusing
720                  * with "lfs find --size|-s", which means "file size", so use
721                  * the consistent "--stripe-size|-S" for all commands. */
722                 {"size",         required_argument, 0, 's'},
723 #endif
724                 {"stripe-size",  required_argument, 0, 'S'},
725                 {"stripe_size",  required_argument, 0, 'S'},
726                 {0, 0, 0, 0}
727         };
728
729         st_size = 0;
730         st_offset = -1;
731         st_count = 0;
732
733         if (strcmp(argv[0], "migrate") == 0)
734                 migrate_mode = true;
735
736         while ((c = getopt_long(argc, argv, "bc:di:o:p:s:S:",
737                                 long_opts, NULL)) >= 0) {
738                 switch (c) {
739                 case 0:
740                         /* Long options. */
741                         break;
742                 case 'b':
743                         if (!migrate_mode) {
744                                 fprintf(stderr, "--block is valid only for"
745                                                 " migrate mode");
746                                 return CMD_HELP;
747                         }
748                         migration_flags |= MIGRATION_BLOCKS;
749                         break;
750                 case 'c':
751 #if LUSTRE_VERSION_CODE >= OBD_OCD_VERSION(2, 6, 53, 0)
752                         if (strcmp(argv[optind - 1], "--count") == 0)
753                                 fprintf(stderr, "warning: '--count' deprecated"
754                                         ", use '--stripe-count' instead\n");
755 #endif
756                         stripe_count_arg = optarg;
757                         break;
758                 case 'd':
759                         /* delete the default striping pattern */
760                         delete = 1;
761                         break;
762                 case 'o':
763                         nr_osts = parse_targets(osts, ARRAY_SIZE(osts), nr_osts,
764                                                 optarg);
765                         if (nr_osts < 0) {
766                                 fprintf(stderr,
767                                         "error: %s: bad OST indices '%s'\n",
768                                         argv[0], optarg);
769                                 return CMD_HELP;
770                         }
771
772                         if (st_offset == -1) /* first in the command line */
773                                 st_offset = osts[0];
774                         break;
775                 case 'i':
776 #if LUSTRE_VERSION_CODE >= OBD_OCD_VERSION(2, 6, 53, 0)
777                         if (strcmp(argv[optind - 1], "--index") == 0)
778                                 fprintf(stderr, "warning: '--index' deprecated"
779                                         ", use '--stripe-index' instead\n");
780 #endif
781                         stripe_off_arg = optarg;
782                         break;
783 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 9, 53, 0)
784                 case 's':
785 #if LUSTRE_VERSION_CODE >= OBD_OCD_VERSION(2, 6, 53, 0)
786                         fprintf(stderr, "warning: '--size|-s' deprecated, "
787                                 "use '--stripe-size|-S' instead\n");
788 #endif
789 #endif /* LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 9, 53, 0) */
790                 case 'S':
791                         stripe_size_arg = optarg;
792                         break;
793                 case 'p':
794                         pool_name_arg = optarg;
795                         break;
796                 default:
797                         return CMD_HELP;
798                 }
799         }
800
801         fname = argv[optind];
802
803         if (delete &&
804             (stripe_size_arg != NULL || stripe_off_arg != NULL ||
805              stripe_count_arg != NULL || pool_name_arg != NULL)) {
806                 fprintf(stderr, "error: %s: cannot specify -d with "
807                         "-s, -c, -o, or -p options\n",
808                         argv[0]);
809                 return CMD_HELP;
810         }
811
812         if (optind == argc) {
813                 fprintf(stderr, "error: %s: missing filename|dirname\n",
814                         argv[0]);
815                 return CMD_HELP;
816         }
817
818         if (pool_name_arg && strlen(pool_name_arg) > LOV_MAXPOOLNAME) {
819                 fprintf(stderr,
820                         "error: %s: pool name '%s' is too long (max is %d characters)\n",
821                         argv[0], pool_name_arg, LOV_MAXPOOLNAME);
822                 return CMD_HELP;
823         }
824
825         /* get the stripe size */
826         if (stripe_size_arg != NULL) {
827                 result = llapi_parse_size(stripe_size_arg, &st_size,
828                                           &size_units, 0);
829                 if (result) {
830                         fprintf(stderr, "error: %s: bad stripe size '%s'\n",
831                                 argv[0], stripe_size_arg);
832                         return CMD_HELP;
833                 }
834         }
835         /* get the stripe offset */
836         if (stripe_off_arg != NULL) {
837                 st_offset = strtol(stripe_off_arg, &end, 0);
838                 if (*end != '\0') {
839                         fprintf(stderr, "error: %s: bad stripe offset '%s'\n",
840                                 argv[0], stripe_off_arg);
841                         return CMD_HELP;
842                 }
843         }
844         /* get the stripe count */
845         if (stripe_count_arg != NULL) {
846                 st_count = strtoul(stripe_count_arg, &end, 0);
847                 if (*end != '\0') {
848                         fprintf(stderr, "error: %s: bad stripe count '%s'\n",
849                                 argv[0], stripe_count_arg);
850                         return CMD_HELP;
851                 }
852         }
853
854         /* initialize stripe parameters */
855         param = calloc(1, offsetof(typeof(*param), lsp_osts[nr_osts]));
856         if (param == NULL) {
857                 fprintf(stderr, "error: %s: run out of memory\n", argv[0]);
858                 return CMD_HELP;
859         }
860
861         param->lsp_stripe_size = st_size;
862         param->lsp_stripe_offset = st_offset;
863         param->lsp_stripe_count = st_count;
864         param->lsp_stripe_pattern = 0;
865         param->lsp_pool = pool_name_arg;
866         param->lsp_is_specific = false;
867         if (nr_osts > 0) {
868                 if (st_count > 0 && nr_osts != st_count) {
869                         fprintf(stderr, "error: %s: stripe count '%d' doesn't "
870                                 "match the number of OSTs: %d\n",
871                                 argv[0], st_count, nr_osts);
872                         return CMD_HELP;
873                 }
874
875                 param->lsp_is_specific = true;
876                 param->lsp_stripe_count = nr_osts;
877                 memcpy(param->lsp_osts, osts, sizeof(*osts) * nr_osts);
878         }
879
880         do {
881                 if (!migrate_mode) {
882                         result = llapi_file_open_param(fname,
883                                                        O_CREAT | O_WRONLY,
884                                                        0644, param);
885                         if (result >= 0) {
886                                 close(result);
887                                 result = 0;
888                         }
889                 } else {
890                         result = lfs_migrate(fname, migration_flags, param);
891                 }
892                 if (result) {
893                         fprintf(stderr,
894                                 "error: %s: %s stripe file '%s' failed\n",
895                                 argv[0], migrate_mode ? "migrate" : "create",
896                                 fname);
897                         break;
898                 }
899                 fname = argv[++optind];
900         } while (fname != NULL);
901
902         free(param);
903         return result;
904 }
905
906 static int lfs_poollist(int argc, char **argv)
907 {
908         if (argc != 2)
909                 return CMD_HELP;
910
911         return llapi_poollist(argv[1]);
912 }
913
914 static int set_time(time_t *time, time_t *set, char *str)
915 {
916         time_t t;
917         int res = 0;
918
919         if (str[0] == '+')
920                 res = 1;
921         else if (str[0] == '-')
922                 res = -1;
923
924         if (res)
925                 str++;
926
927         t = strtol(str, NULL, 0);
928         if (*time < t * 24 * 60 * 60) {
929                 if (res)
930                         str--;
931                 fprintf(stderr, "Wrong time '%s' is specified.\n", str);
932                 return INT_MAX;
933         }
934
935         *set = *time - t * 24 * 60 * 60;
936         return res;
937 }
938
939 #define USER 0
940 #define GROUP 1
941
942 static int name2id(unsigned int *id, char *name, int type)
943 {
944         if (type == USER) {
945                 struct passwd *entry;
946
947                 if (!(entry = getpwnam(name))) {
948                         if (!errno)
949                                 errno = ENOENT;
950                         return -1;
951                 }
952
953                 *id = entry->pw_uid;
954         } else {
955                 struct group *entry;
956
957                 if (!(entry = getgrnam(name))) {
958                         if (!errno)
959                                 errno = ENOENT;
960                         return -1;
961                 }
962
963                 *id = entry->gr_gid;
964         }
965
966         return 0;
967 }
968
969 static int id2name(char **name, unsigned int id, int type)
970 {
971         if (type == USER) {
972                 struct passwd *entry;
973
974                 if (!(entry = getpwuid(id))) {
975                         if (!errno)
976                                 errno = ENOENT;
977                         return -1;
978                 }
979
980                 *name = entry->pw_name;
981         } else {
982                 struct group *entry;
983
984                 if (!(entry = getgrgid(id))) {
985                         if (!errno)
986                                 errno = ENOENT;
987                         return -1;
988                 }
989
990                 *name = entry->gr_name;
991         }
992
993         return 0;
994 }
995
996 static int name2layout(__u32 *layout, char *name)
997 {
998         char *ptr, *lyt;
999
1000         *layout = 0;
1001         for (ptr = name; ; ptr = NULL) {
1002                 lyt = strtok(ptr, ",");
1003                 if (lyt == NULL)
1004                         break;
1005                 if (strcmp(lyt, "released") == 0)
1006                         *layout |= LOV_PATTERN_F_RELEASED;
1007                 else if (strcmp(lyt, "raid0") == 0)
1008                         *layout |= LOV_PATTERN_RAID0;
1009                 else
1010                         return -1;
1011         }
1012         return 0;
1013 }
1014
1015 #define FIND_POOL_OPT 3
1016 static int lfs_find(int argc, char **argv)
1017 {
1018         int c, rc;
1019         int ret = 0;
1020         time_t t;
1021         struct find_param param = {
1022                 .fp_max_depth = -1,
1023                 .fp_quiet = 1,
1024         };
1025         struct option long_opts[] = {
1026                 {"atime",        required_argument, 0, 'A'},
1027                 {"stripe-count", required_argument, 0, 'c'},
1028                 {"stripe_count", required_argument, 0, 'c'},
1029                 {"ctime",        required_argument, 0, 'C'},
1030                 {"maxdepth",     required_argument, 0, 'D'},
1031                 {"gid",          required_argument, 0, 'g'},
1032                 {"group",        required_argument, 0, 'G'},
1033                 {"stripe-index", required_argument, 0, 'i'},
1034                 {"stripe_index", required_argument, 0, 'i'},
1035                 {"layout",       required_argument, 0, 'L'},
1036                 {"mdt",          required_argument, 0, 'm'},
1037                 {"mtime",        required_argument, 0, 'M'},
1038                 {"name",         required_argument, 0, 'n'},
1039      /* reserve {"or",           no_argument,     , 0, 'o'}, to match find(1) */
1040                 {"obd",          required_argument, 0, 'O'},
1041                 {"ost",          required_argument, 0, 'O'},
1042                 /* no short option for pool, p/P already used */
1043                 {"pool",         required_argument, 0, FIND_POOL_OPT},
1044                 {"print0",       no_argument,       0, 'p'},
1045                 {"print",        no_argument,       0, 'P'},
1046                 {"size",         required_argument, 0, 's'},
1047                 {"stripe-size",  required_argument, 0, 'S'},
1048                 {"stripe_size",  required_argument, 0, 'S'},
1049                 {"type",         required_argument, 0, 't'},
1050                 {"uid",          required_argument, 0, 'u'},
1051                 {"user",         required_argument, 0, 'U'},
1052                 {0, 0, 0, 0}
1053         };
1054         int pathstart = -1;
1055         int pathend = -1;
1056         int neg_opt = 0;
1057         time_t *xtime;
1058         int *xsign;
1059         int isoption;
1060         char *endptr;
1061
1062         time(&t);
1063
1064         /* when getopt_long_only() hits '!' it returns 1, puts "!" in optarg */
1065         while ((c = getopt_long_only(argc, argv,
1066                                      "-A:c:C:D:g:G:i:L:m:M:n:O:Ppqrs:S:t:u:U:v",
1067                                      long_opts, NULL)) >= 0) {
1068                 xtime = NULL;
1069                 xsign = NULL;
1070                 if (neg_opt)
1071                         --neg_opt;
1072                 /* '!' is part of option */
1073                 /* when getopt_long_only() finds a string which is not
1074                  * an option nor a known option argument it returns 1
1075                  * in that case if we already have found pathstart and pathend
1076                  * (i.e. we have the list of pathnames),
1077                  * the only supported value is "!"
1078                  */
1079                 isoption = (c != 1) || (strcmp(optarg, "!") == 0);
1080                 if (!isoption && pathend != -1) {
1081                         fprintf(stderr, "err: %s: filename|dirname must either "
1082                                         "precede options or follow options\n",
1083                                         argv[0]);
1084                         ret = CMD_HELP;
1085                         goto err;
1086                 }
1087                 if (!isoption && pathstart == -1)
1088                         pathstart = optind - 1;
1089                 if (isoption && pathstart != -1 && pathend == -1)
1090                         pathend = optind - 2;
1091                 switch (c) {
1092                 case 0:
1093                         /* Long options. */
1094                         break;
1095                 case 1:
1096                         /* unknown; opt is "!" or path component,
1097                          * checking done above.
1098                          */
1099                         if (strcmp(optarg, "!") == 0)
1100                                 neg_opt = 2;
1101                         break;
1102                 case 'A':
1103                         xtime = &param.fp_atime;
1104                         xsign = &param.fp_asign;
1105                         param.fp_exclude_atime = !!neg_opt;
1106                         /* no break, this falls through to 'C' for ctime */
1107                 case 'C':
1108                         if (c == 'C') {
1109                                 xtime = &param.fp_ctime;
1110                                 xsign = &param.fp_csign;
1111                                 param.fp_exclude_ctime = !!neg_opt;
1112                         }
1113                         /* no break, this falls through to 'M' for mtime */
1114                 case 'M':
1115                         if (c == 'M') {
1116                                 xtime = &param.fp_mtime;
1117                                 xsign = &param.fp_msign;
1118                                 param.fp_exclude_mtime = !!neg_opt;
1119                         }
1120                         rc = set_time(&t, xtime, optarg);
1121                         if (rc == INT_MAX) {
1122                                 ret = -1;
1123                                 goto err;
1124                         }
1125                         if (rc)
1126                                 *xsign = rc;
1127                         break;
1128                 case 'c':
1129                         if (optarg[0] == '+') {
1130                                 param.fp_stripe_count_sign = -1;
1131                                 optarg++;
1132                         } else if (optarg[0] == '-') {
1133                                 param.fp_stripe_count_sign =  1;
1134                                 optarg++;
1135                         }
1136
1137                         param.fp_stripe_count = strtoul(optarg, &endptr, 0);
1138                         if (*endptr != '\0') {
1139                                 fprintf(stderr,"error: bad stripe_count '%s'\n",
1140                                         optarg);
1141                                 ret = -1;
1142                                 goto err;
1143                         }
1144                         param.fp_check_stripe_count = 1;
1145                         param.fp_exclude_stripe_count = !!neg_opt;
1146                         break;
1147                 case 'D':
1148                         param.fp_max_depth = strtol(optarg, 0, 0);
1149                         break;
1150                 case 'g':
1151                 case 'G':
1152                         rc = name2id(&param.fp_gid, optarg, GROUP);
1153                         if (rc) {
1154                                 param.fp_gid = strtoul(optarg, &endptr, 10);
1155                                 if (*endptr != '\0') {
1156                                         fprintf(stderr, "Group/GID: %s cannot "
1157                                                 "be found.\n", optarg);
1158                                         ret = -1;
1159                                         goto err;
1160                                 }
1161                         }
1162                         param.fp_exclude_gid = !!neg_opt;
1163                         param.fp_check_gid = 1;
1164                         break;
1165                 case 'L':
1166                         ret = name2layout(&param.fp_layout, optarg);
1167                         if (ret)
1168                                 goto err;
1169                         param.fp_exclude_layout = !!neg_opt;
1170                         param.fp_check_layout = 1;
1171                         break;
1172                 case 'u':
1173                 case 'U':
1174                         rc = name2id(&param.fp_uid, optarg, USER);
1175                         if (rc) {
1176                                 param.fp_uid = strtoul(optarg, &endptr, 10);
1177                                 if (*endptr != '\0') {
1178                                         fprintf(stderr, "User/UID: %s cannot "
1179                                                 "be found.\n", optarg);
1180                                         ret = -1;
1181                                         goto err;
1182                                 }
1183                         }
1184                         param.fp_exclude_uid = !!neg_opt;
1185                         param.fp_check_uid = 1;
1186                         break;
1187                 case FIND_POOL_OPT:
1188                         if (strlen(optarg) > LOV_MAXPOOLNAME) {
1189                                 fprintf(stderr,
1190                                         "Pool name %s is too long"
1191                                         " (max is %d)\n", optarg,
1192                                         LOV_MAXPOOLNAME);
1193                                 ret = -1;
1194                                 goto err;
1195                         }
1196                         /* we do check for empty pool because empty pool
1197                          * is used to find V1 lov attributes */
1198                         strncpy(param.fp_poolname, optarg, LOV_MAXPOOLNAME);
1199                         param.fp_poolname[LOV_MAXPOOLNAME] = '\0';
1200                         param.fp_exclude_pool = !!neg_opt;
1201                         param.fp_check_pool = 1;
1202                         break;
1203                 case 'n':
1204                         param.fp_pattern = (char *)optarg;
1205                         param.fp_exclude_pattern = !!neg_opt;
1206                         break;
1207                 case 'm':
1208                 case 'i':
1209                 case 'O': {
1210                         char *buf, *token, *next, *p;
1211                         int len = 1;
1212                         void *tmp;
1213
1214                         buf = strdup(optarg);
1215                         if (buf == NULL) {
1216                                 ret = -ENOMEM;
1217                                 goto err;
1218                         }
1219
1220                         param.fp_exclude_obd = !!neg_opt;
1221
1222                         token = buf;
1223                         while (token && *token) {
1224                                 token = strchr(token, ',');
1225                                 if (token) {
1226                                         len++;
1227                                         token++;
1228                                 }
1229                         }
1230                         if (c == 'm') {
1231                                 param.fp_exclude_mdt = !!neg_opt;
1232                                 param.fp_num_alloc_mdts += len;
1233                                 tmp = realloc(param.fp_mdt_uuid,
1234                                               param.fp_num_alloc_mdts *
1235                                               sizeof(*param.fp_mdt_uuid));
1236                                 if (tmp == NULL) {
1237                                         ret = -ENOMEM;
1238                                         goto err_free;
1239                                 }
1240
1241                                 param.fp_mdt_uuid = tmp;
1242                         } else {
1243                                 param.fp_exclude_obd = !!neg_opt;
1244                                 param.fp_num_alloc_obds += len;
1245                                 tmp = realloc(param.fp_obd_uuid,
1246                                               param.fp_num_alloc_obds *
1247                                               sizeof(*param.fp_obd_uuid));
1248                                 if (tmp == NULL) {
1249                                         ret = -ENOMEM;
1250                                         goto err_free;
1251                                 }
1252
1253                                 param.fp_obd_uuid = tmp;
1254                         }
1255                         for (token = buf; token && *token; token = next) {
1256                                 struct obd_uuid *puuid;
1257                                 if (c == 'm') {
1258                                         puuid =
1259                                         &param.fp_mdt_uuid[param.fp_num_mdts++];
1260                                 } else {
1261                                         puuid =
1262                                         &param.fp_obd_uuid[param.fp_num_obds++];
1263                                 }
1264                                 p = strchr(token, ',');
1265                                 next = 0;
1266                                 if (p) {
1267                                         *p = 0;
1268                                         next = p+1;
1269                                 }
1270
1271                                 if (strlen(token) > sizeof(puuid->uuid) - 1) {
1272                                         ret = -E2BIG;
1273                                         goto err_free;
1274                                 }
1275
1276                                 strncpy(puuid->uuid, token,
1277                                         sizeof(puuid->uuid));
1278                         }
1279 err_free:
1280                         if (buf)
1281                                 free(buf);
1282                         break;
1283                 }
1284                 case 'p':
1285                         param.fp_zero_end = 1;
1286                         break;
1287                 case 'P':
1288                         break;
1289                 case 's':
1290                         if (optarg[0] == '+') {
1291                                 param.fp_size_sign = -1;
1292                                 optarg++;
1293                         } else if (optarg[0] == '-') {
1294                                 param.fp_size_sign =  1;
1295                                 optarg++;
1296                         }
1297
1298                         ret = llapi_parse_size(optarg, &param.fp_size,
1299                                                &param.fp_size_units, 0);
1300                         if (ret) {
1301                                 fprintf(stderr, "error: bad file size '%s'\n",
1302                                         optarg);
1303                                 goto err;
1304                         }
1305                         param.fp_check_size = 1;
1306                         param.fp_exclude_size = !!neg_opt;
1307                         break;
1308                 case 'S':
1309                         if (optarg[0] == '+') {
1310                                 param.fp_stripe_size_sign = -1;
1311                                 optarg++;
1312                         } else if (optarg[0] == '-') {
1313                                 param.fp_stripe_size_sign =  1;
1314                                 optarg++;
1315                         }
1316
1317                         ret = llapi_parse_size(optarg, &param.fp_stripe_size,
1318                                                &param.fp_stripe_size_units, 0);
1319                         if (ret) {
1320                                 fprintf(stderr, "error: bad stripe_size '%s'\n",
1321                                         optarg);
1322                                 goto err;
1323                         }
1324                         param.fp_check_stripe_size = 1;
1325                         param.fp_exclude_stripe_size = !!neg_opt;
1326                         break;
1327                 case 't':
1328                         param.fp_exclude_type = !!neg_opt;
1329                         switch (optarg[0]) {
1330                         case 'b':
1331                                 param.fp_type = S_IFBLK;
1332                                 break;
1333                         case 'c':
1334                                 param.fp_type = S_IFCHR;
1335                                 break;
1336                         case 'd':
1337                                 param.fp_type = S_IFDIR;
1338                                 break;
1339                         case 'f':
1340                                 param.fp_type = S_IFREG;
1341                                 break;
1342                         case 'l':
1343                                 param.fp_type = S_IFLNK;
1344                                 break;
1345                         case 'p':
1346                                 param.fp_type = S_IFIFO;
1347                                 break;
1348                         case 's':
1349                                 param.fp_type = S_IFSOCK;
1350                                 break;
1351                         default:
1352                                 fprintf(stderr, "error: %s: bad type '%s'\n",
1353                                         argv[0], optarg);
1354                                 ret = CMD_HELP;
1355                                 goto err;
1356                         };
1357                         break;
1358                 default:
1359                         ret = CMD_HELP;
1360                         goto err;
1361                 };
1362         }
1363
1364         if (pathstart == -1) {
1365                 fprintf(stderr, "error: %s: no filename|pathname\n",
1366                         argv[0]);
1367                 ret = CMD_HELP;
1368                 goto err;
1369         } else if (pathend == -1) {
1370                 /* no options */
1371                 pathend = argc;
1372         }
1373
1374         do {
1375                 rc = llapi_find(argv[pathstart], &param);
1376                 if (rc != 0 && ret == 0)
1377                         ret = rc;
1378         } while (++pathstart < pathend);
1379
1380         if (ret)
1381                 fprintf(stderr, "error: %s failed for %s.\n",
1382                         argv[0], argv[optind - 1]);
1383 err:
1384         if (param.fp_obd_uuid && param.fp_num_alloc_obds)
1385                 free(param.fp_obd_uuid);
1386
1387         if (param.fp_mdt_uuid && param.fp_num_alloc_mdts)
1388                 free(param.fp_mdt_uuid);
1389
1390         return ret;
1391 }
1392
1393 static int lfs_getstripe_internal(int argc, char **argv,
1394                                   struct find_param *param)
1395 {
1396         struct option long_opts[] = {
1397 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 9, 53, 0)
1398                 /* This formerly implied "stripe-count", but was explicitly
1399                  * made "stripe-count" for consistency with other options,
1400                  * and to separate it from "mdt-count" when DNE arrives. */
1401                 {"count",               no_argument,            0, 'c'},
1402 #endif
1403                 {"stripe-count",        no_argument,            0, 'c'},
1404                 {"stripe_count",        no_argument,            0, 'c'},
1405                 {"directory",           no_argument,            0, 'd'},
1406                 {"default",             no_argument,            0, 'D'},
1407                 {"generation",          no_argument,            0, 'g'},
1408 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 9, 53, 0)
1409                 /* This formerly implied "stripe-index", but was explicitly
1410                  * made "stripe-index" for consistency with other options,
1411                  * and to separate it from "mdt-index" when DNE arrives. */
1412                 {"index",               no_argument,            0, 'i'},
1413 #endif
1414                 {"stripe-index",        no_argument,            0, 'i'},
1415                 {"stripe_index",        no_argument,            0, 'i'},
1416                 {"layout",              no_argument,            0, 'L'},
1417                 {"mdt-index",           no_argument,            0, 'M'},
1418                 {"mdt_index",           no_argument,            0, 'M'},
1419 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 9, 53, 0)
1420                 /* This formerly implied "stripe-index", but was confusing
1421                  * with "file offset" (which will eventually be needed for
1422                  * with different layouts by offset), so deprecate it. */
1423                 {"offset",              no_argument,            0, 'o'},
1424 #endif
1425                 {"obd",                 required_argument,      0, 'O'},
1426                 {"ost",                 required_argument,      0, 'O'},
1427                 {"pool",                no_argument,            0, 'p'},
1428                 {"quiet",               no_argument,            0, 'q'},
1429                 {"recursive",           no_argument,            0, 'r'},
1430                 {"raw",                 no_argument,            0, 'R'},
1431 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 9, 53, 0)
1432                 /* This formerly implied "--stripe-size", but was confusing
1433                  * with "lfs find --size|-s", which means "file size", so use
1434                  * the consistent "--stripe-size|-S" for all commands. */
1435                 {"size",                no_argument,            0, 's'},
1436 #endif
1437                 {"stripe-size",         no_argument,            0, 'S'},
1438                 {"stripe_size",         no_argument,            0, 'S'},
1439                 {"verbose",             no_argument,            0, 'v'},
1440                 {0, 0, 0, 0}
1441         };
1442         int c, rc;
1443
1444         param->fp_max_depth = 1;
1445         while ((c = getopt_long(argc, argv, "cdDghiLMoO:pqrRsSv",
1446                                 long_opts, NULL)) != -1) {
1447                 switch (c) {
1448                 case 'O':
1449                         if (param->fp_obd_uuid) {
1450                                 fprintf(stderr,
1451                                         "error: %s: only one obduuid allowed",
1452                                         argv[0]);
1453                                 return CMD_HELP;
1454                         }
1455                         param->fp_obd_uuid = (struct obd_uuid *)optarg;
1456                         break;
1457                 case 'q':
1458                         param->fp_quiet++;
1459                         break;
1460                 case 'd':
1461                         param->fp_max_depth = 0;
1462                         break;
1463                 case 'D':
1464                         param->fp_get_default_lmv = 1;
1465                         break;
1466                 case 'r':
1467                         param->fp_recursive = 1;
1468                         break;
1469                 case 'v':
1470                         param->fp_verbose = VERBOSE_ALL | VERBOSE_DETAIL;
1471                         break;
1472                 case 'c':
1473 #if LUSTRE_VERSION_CODE >= OBD_OCD_VERSION(2, 6, 53, 0)
1474                         if (strcmp(argv[optind - 1], "--count") == 0)
1475                                 fprintf(stderr, "warning: '--count' deprecated,"
1476                                         " use '--stripe-count' instead\n");
1477 #endif
1478                         if (!(param->fp_verbose & VERBOSE_DETAIL)) {
1479                                 param->fp_verbose |= VERBOSE_COUNT;
1480                                 param->fp_max_depth = 0;
1481                         }
1482                         break;
1483 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 9, 53, 0)
1484                 case 's':
1485 #if LUSTRE_VERSION_CODE >= OBD_OCD_VERSION(2, 6, 53, 0)
1486                         fprintf(stderr, "warning: '--size|-s' deprecated, "
1487                                 "use '--stripe-size|-S' instead\n");
1488 #endif
1489 #endif /* LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 9, 53, 0) */
1490                 case 'S':
1491                         if (!(param->fp_verbose & VERBOSE_DETAIL)) {
1492                                 param->fp_verbose |= VERBOSE_SIZE;
1493                                 param->fp_max_depth = 0;
1494                         }
1495                         break;
1496 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 9, 53, 0)
1497                 case 'o':
1498                         fprintf(stderr, "warning: '--offset|-o' deprecated, "
1499                                 "use '--stripe-index|-i' instead\n");
1500 #endif
1501                 case 'i':
1502 #if LUSTRE_VERSION_CODE >= OBD_OCD_VERSION(2, 6, 53, 0)
1503                         if (strcmp(argv[optind - 1], "--index") == 0)
1504                                 fprintf(stderr, "warning: '--index' deprecated"
1505                                         ", use '--stripe-index' instead\n");
1506 #endif
1507                         if (!(param->fp_verbose & VERBOSE_DETAIL)) {
1508                                 param->fp_verbose |= VERBOSE_OFFSET;
1509                                 param->fp_max_depth = 0;
1510                         }
1511                         break;
1512                 case 'p':
1513                         if (!(param->fp_verbose & VERBOSE_DETAIL)) {
1514                                 param->fp_verbose |= VERBOSE_POOL;
1515                                 param->fp_max_depth = 0;
1516                         }
1517                         break;
1518                 case 'g':
1519                         if (!(param->fp_verbose & VERBOSE_DETAIL)) {
1520                                 param->fp_verbose |= VERBOSE_GENERATION;
1521                                 param->fp_max_depth = 0;
1522                         }
1523                         break;
1524                 case 'L':
1525                         if (!(param->fp_verbose & VERBOSE_DETAIL)) {
1526                                 param->fp_verbose |= VERBOSE_LAYOUT;
1527                                 param->fp_max_depth = 0;
1528                         }
1529                         break;
1530                 case 'M':
1531                         if (!(param->fp_verbose & VERBOSE_DETAIL))
1532                                 param->fp_max_depth = 0;
1533                         param->fp_verbose |= VERBOSE_MDTINDEX;
1534                         break;
1535                 case 'R':
1536                         param->fp_raw = 1;
1537                         break;
1538                 default:
1539                         return CMD_HELP;
1540                 }
1541         }
1542
1543         if (optind >= argc)
1544                 return CMD_HELP;
1545
1546         if (param->fp_recursive)
1547                 param->fp_max_depth = -1;
1548
1549         if (!param->fp_verbose)
1550                 param->fp_verbose = VERBOSE_ALL;
1551         if (param->fp_quiet)
1552                 param->fp_verbose = VERBOSE_OBJID;
1553
1554         do {
1555                 rc = llapi_getstripe(argv[optind], param);
1556         } while (++optind < argc && !rc);
1557
1558         if (rc)
1559                 fprintf(stderr, "error: %s failed for %s.\n",
1560                         argv[0], argv[optind - 1]);
1561         return rc;
1562 }
1563
1564 static int lfs_tgts(int argc, char **argv)
1565 {
1566         char mntdir[PATH_MAX] = {'\0'}, path[PATH_MAX] = {'\0'};
1567         struct find_param param;
1568         int index = 0, rc=0;
1569
1570         if (argc > 2)
1571                 return CMD_HELP;
1572
1573         if (argc == 2 && !realpath(argv[1], path)) {
1574                 rc = -errno;
1575                 fprintf(stderr, "error: invalid path '%s': %s\n",
1576                         argv[1], strerror(-rc));
1577                 return rc;
1578         }
1579
1580         while (!llapi_search_mounts(path, index++, mntdir, NULL)) {
1581                 /* Check if we have a mount point */
1582                 if (mntdir[0] == '\0')
1583                         continue;
1584
1585                 memset(&param, 0, sizeof(param));
1586                 if (!strcmp(argv[0], "mdts"))
1587                         param.fp_get_lmv = 1;
1588
1589                 rc = llapi_ostlist(mntdir, &param);
1590                 if (rc) {
1591                         fprintf(stderr, "error: %s: failed on %s\n",
1592                                 argv[0], mntdir);
1593                 }
1594                 if (path[0] != '\0')
1595                         break;
1596                 memset(mntdir, 0, PATH_MAX);
1597         }
1598
1599         return rc;
1600 }
1601
1602 static int lfs_getstripe(int argc, char **argv)
1603 {
1604         struct find_param param = { 0 };
1605         return lfs_getstripe_internal(argc, argv, &param);
1606 }
1607
1608 /* functions */
1609 static int lfs_getdirstripe(int argc, char **argv)
1610 {
1611         struct find_param param = { 0 };
1612
1613         param.fp_get_lmv = 1;
1614         return lfs_getstripe_internal(argc, argv, &param);
1615 }
1616
1617 /* functions */
1618 static int lfs_setdirstripe(int argc, char **argv)
1619 {
1620         char                    *dname;
1621         int                     result;
1622         unsigned int            stripe_offset = -1;
1623         unsigned int            stripe_count = 1;
1624         enum lmv_hash_type      hash_type;
1625         char                    *end;
1626         int                     c;
1627         char                    *stripe_offset_opt = NULL;
1628         char                    *stripe_count_opt = NULL;
1629         char                    *stripe_hash_opt = NULL;
1630         char                    *mode_opt = NULL;
1631         int                     default_stripe = 0;
1632         mode_t                  mode = S_IRWXU | S_IRWXG | S_IRWXO;
1633         mode_t                  previous_mode = 0;
1634
1635         struct option long_opts[] = {
1636                 {"count",       required_argument, 0, 'c'},
1637                 {"index",       required_argument, 0, 'i'},
1638                 {"mode",        required_argument, 0, 'm'},
1639                 {"hash-type",   required_argument, 0, 't'},
1640                 {"default_stripe", no_argument, 0, 'D'},
1641                 {0, 0, 0, 0}
1642         };
1643
1644         while ((c = getopt_long(argc, argv, "c:Di:m:t:", long_opts,
1645                                 NULL)) >= 0) {
1646                 switch (c) {
1647                 case 0:
1648                         /* Long options. */
1649                         break;
1650                 case 'c':
1651                         stripe_count_opt = optarg;
1652                         break;
1653                 case 'D':
1654                         default_stripe = 1;
1655                         break;
1656                 case 'i':
1657                         stripe_offset_opt = optarg;
1658                         break;
1659                 case 'm':
1660                         mode_opt = optarg;
1661                         break;
1662                 case 't':
1663                         stripe_hash_opt = optarg;
1664                         break;
1665                 default:
1666                         fprintf(stderr, "error: %s: option '%s' "
1667                                         "unrecognized\n",
1668                                         argv[0], argv[optind - 1]);
1669                         return CMD_HELP;
1670                 }
1671         }
1672
1673         if (optind == argc) {
1674                 fprintf(stderr, "error: %s: missing dirname\n",
1675                         argv[0]);
1676                 return CMD_HELP;
1677         }
1678
1679         if (stripe_offset_opt == NULL && stripe_count_opt == NULL) {
1680                 fprintf(stderr, "error: %s: missing stripe offset and count.\n",
1681                         argv[0]);
1682                 return CMD_HELP;
1683         }
1684
1685         if (stripe_offset_opt != NULL) {
1686                 /* get the stripe offset */
1687                 stripe_offset = strtoul(stripe_offset_opt, &end, 0);
1688                 if (*end != '\0') {
1689                         fprintf(stderr, "error: %s: bad stripe offset '%s'\n",
1690                                 argv[0], stripe_offset_opt);
1691                         return CMD_HELP;
1692                 }
1693         }
1694
1695         if (mode_opt != NULL) {
1696                 mode = strtoul(mode_opt, &end, 8);
1697                 if (*end != '\0') {
1698                         fprintf(stderr, "error: %s: bad mode '%s'\n",
1699                                 argv[0], mode_opt);
1700                         return CMD_HELP;
1701                 }
1702                 previous_mode = umask(0);
1703         }
1704
1705         if (stripe_hash_opt == NULL ||
1706             strcmp(stripe_hash_opt, LMV_HASH_NAME_FNV_1A_64) == 0) {
1707                 hash_type = LMV_HASH_TYPE_FNV_1A_64;
1708         } else if (strcmp(stripe_hash_opt, LMV_HASH_NAME_ALL_CHARS) == 0) {
1709                 hash_type = LMV_HASH_TYPE_ALL_CHARS;
1710         } else {
1711                 fprintf(stderr, "error: %s: bad stripe hash type '%s'\n",
1712                         argv[0], stripe_hash_opt);
1713                 return CMD_HELP;
1714         }
1715
1716         /* get the stripe count */
1717         if (stripe_count_opt != NULL) {
1718                 stripe_count = strtoul(stripe_count_opt, &end, 0);
1719                 if (*end != '\0') {
1720                         fprintf(stderr, "error: %s: bad stripe count '%s'\n",
1721                                 argv[0], stripe_count_opt);
1722                         return CMD_HELP;
1723                 }
1724         }
1725
1726         dname = argv[optind];
1727         do {
1728                 if (default_stripe == 1) {
1729                         result = llapi_dir_set_default_lmv_stripe(dname,
1730                                                     stripe_offset, stripe_count,
1731                                                     hash_type, NULL);
1732                 } else {
1733                         result = llapi_dir_create_pool(dname, mode,
1734                                                        stripe_offset,
1735                                                        stripe_count, hash_type,
1736                                                        NULL);
1737                 }
1738
1739                 if (result) {
1740                         fprintf(stderr, "error: %s: create stripe dir '%s' "
1741                                 "failed\n", argv[0], dname);
1742                         break;
1743                 }
1744                 dname = argv[++optind];
1745         } while (dname != NULL);
1746
1747         if (mode_opt != NULL)
1748                 umask(previous_mode);
1749
1750         return result;
1751 }
1752
1753 /* functions */
1754 static int lfs_rmentry(int argc, char **argv)
1755 {
1756         char *dname;
1757         int   index;
1758         int   result = 0;
1759
1760         if (argc <= 1) {
1761                 fprintf(stderr, "error: %s: missing dirname\n",
1762                         argv[0]);
1763                 return CMD_HELP;
1764         }
1765
1766         index = 1;
1767         dname = argv[index];
1768         while (dname != NULL) {
1769                 result = llapi_direntry_remove(dname);
1770                 if (result) {
1771                         fprintf(stderr, "error: %s: remove dir entry '%s' "
1772                                 "failed\n", argv[0], dname);
1773                         break;
1774                 }
1775                 dname = argv[++index];
1776         }
1777         return result;
1778 }
1779
1780 static int lfs_mv(int argc, char **argv)
1781 {
1782         struct  find_param param = {
1783                 .fp_max_depth = -1,
1784                 .fp_mdt_index = -1,
1785         };
1786         char   *end;
1787         int     c;
1788         int     rc = 0;
1789         struct option long_opts[] = {
1790                 {"mdt-index", required_argument, 0, 'M'},
1791                 {"verbose",     no_argument,       0, 'v'},
1792                 {0, 0, 0, 0}
1793         };
1794
1795         while ((c = getopt_long(argc, argv, "M:v", long_opts, NULL)) != -1) {
1796                 switch (c) {
1797                 case 'M': {
1798                         param.fp_mdt_index = strtoul(optarg, &end, 0);
1799                         if (*end != '\0') {
1800                                 fprintf(stderr, "%s: invalid MDT index'%s'\n",
1801                                         argv[0], optarg);
1802                                 return CMD_HELP;
1803                         }
1804                         break;
1805                 }
1806                 case 'v': {
1807                         param.fp_verbose = VERBOSE_DETAIL;
1808                         break;
1809                 }
1810                 default:
1811                         fprintf(stderr, "error: %s: unrecognized option '%s'\n",
1812                                 argv[0], argv[optind - 1]);
1813                         return CMD_HELP;
1814                 }
1815         }
1816
1817         if (param.fp_mdt_index == -1) {
1818                 fprintf(stderr, "%s: MDT index must be specified\n", argv[0]);
1819                 return CMD_HELP;
1820         }
1821
1822         if (optind >= argc) {
1823                 fprintf(stderr, "%s: missing operand path\n", argv[0]);
1824                 return CMD_HELP;
1825         }
1826
1827         param.fp_migrate = 1;
1828         rc = llapi_mv(argv[optind], &param);
1829         if (rc != 0)
1830                 fprintf(stderr, "%s: cannot migrate '%s' to MDT%04x: %s\n",
1831                         argv[0], argv[optind], param.fp_mdt_index,
1832                         strerror(-rc));
1833         return rc;
1834 }
1835
1836 static int lfs_osts(int argc, char **argv)
1837 {
1838         return lfs_tgts(argc, argv);
1839 }
1840
1841 static int lfs_mdts(int argc, char **argv)
1842 {
1843         return lfs_tgts(argc, argv);
1844 }
1845
1846 #define COOK(value)                                                     \
1847 ({                                                                      \
1848         int radix = 0;                                                  \
1849         while (value > 1024) {                                          \
1850                 value /= 1024;                                          \
1851                 radix++;                                                \
1852         }                                                               \
1853         radix;                                                          \
1854 })
1855 #define UUF     "%-20s"
1856 #define CSF     "%11s"
1857 #define CDF     "%11llu"
1858 #define HDF     "%8.1f%c"
1859 #define RSF     "%4s"
1860 #define RDF     "%3d%%"
1861
1862 static int showdf(char *mntdir, struct obd_statfs *stat,
1863                   char *uuid, int ishow, int cooked,
1864                   char *type, int index, int rc)
1865 {
1866         long long avail, used, total;
1867         double ratio = 0;
1868         char *suffix = "KMGTPEZY";
1869         /* Note if we have >2^64 bytes/fs these buffers will need to be grown */
1870         char tbuf[3 * sizeof(__u64)];
1871         char ubuf[3 * sizeof(__u64)];
1872         char abuf[3 * sizeof(__u64)];
1873         char rbuf[3 * sizeof(__u64)];
1874
1875         if (!uuid || !stat)
1876                 return -EINVAL;
1877
1878         switch (rc) {
1879         case 0:
1880                 if (ishow) {
1881                         avail = stat->os_ffree;
1882                         used = stat->os_files - stat->os_ffree;
1883                         total = stat->os_files;
1884                 } else {
1885                         int shift = cooked ? 0 : 10;
1886
1887                         avail = (stat->os_bavail * stat->os_bsize) >> shift;
1888                         used  = ((stat->os_blocks - stat->os_bfree) *
1889                                  stat->os_bsize) >> shift;
1890                         total = (stat->os_blocks * stat->os_bsize) >> shift;
1891                 }
1892
1893                 if ((used + avail) > 0)
1894                         ratio = (double)used / (double)(used + avail);
1895
1896                 if (cooked) {
1897                         int i;
1898                         double cook_val;
1899
1900                         cook_val = (double)total;
1901                         i = COOK(cook_val);
1902                         if (i > 0)
1903                                 sprintf(tbuf, HDF, cook_val, suffix[i - 1]);
1904                         else
1905                                 sprintf(tbuf, CDF, total);
1906
1907                         cook_val = (double)used;
1908                         i = COOK(cook_val);
1909                         if (i > 0)
1910                                 sprintf(ubuf, HDF, cook_val, suffix[i - 1]);
1911                         else
1912                                 sprintf(ubuf, CDF, used);
1913
1914                         cook_val = (double)avail;
1915                         i = COOK(cook_val);
1916                         if (i > 0)
1917                                 sprintf(abuf, HDF, cook_val, suffix[i - 1]);
1918                         else
1919                                 sprintf(abuf, CDF, avail);
1920                 } else {
1921                         sprintf(tbuf, CDF, total);
1922                         sprintf(ubuf, CDF, used);
1923                         sprintf(abuf, CDF, avail);
1924                 }
1925
1926                 sprintf(rbuf, RDF, (int)(ratio * 100 + 0.5));
1927                 printf(UUF" "CSF" "CSF" "CSF" "RSF" %-s",
1928                        uuid, tbuf, ubuf, abuf, rbuf, mntdir);
1929                 if (type)
1930                         printf("[%s:%d]\n", type, index);
1931                 else
1932                         printf("\n");
1933
1934                 break;
1935         case -ENODATA:
1936                 printf(UUF": inactive device\n", uuid);
1937                 break;
1938         default:
1939                 printf(UUF": %s\n", uuid, strerror(-rc));
1940                 break;
1941         }
1942
1943         return 0;
1944 }
1945
1946 struct ll_stat_type {
1947         int   st_op;
1948         char *st_name;
1949 };
1950
1951 static int mntdf(char *mntdir, char *fsname, char *pool, int ishow,
1952                 int cooked, int lazy)
1953 {
1954         struct obd_statfs stat_buf, sum = { .os_bsize = 1 };
1955         struct obd_uuid uuid_buf;
1956         char *poolname = NULL;
1957         struct ll_stat_type types[] = { { LL_STATFS_LMV, "MDT" },
1958                                         { LL_STATFS_LOV, "OST" },
1959                                         { 0, NULL } };
1960         struct ll_stat_type *tp;
1961         __u64 ost_ffree = 0;
1962         __u32 index;
1963         __u32 type;
1964         int rc;
1965
1966         if (pool) {
1967                 poolname = strchr(pool, '.');
1968                 if (poolname != NULL) {
1969                         if (strncmp(fsname, pool, strlen(fsname))) {
1970                                 fprintf(stderr, "filesystem name incorrect\n");
1971                                 return -ENODEV;
1972                         }
1973                         poolname++;
1974                 } else
1975                         poolname = pool;
1976         }
1977
1978         if (ishow)
1979                 printf(UUF" "CSF" "CSF" "CSF" "RSF" %-s\n",
1980                        "UUID", "Inodes", "IUsed", "IFree",
1981                        "IUse%", "Mounted on");
1982         else
1983                 printf(UUF" "CSF" "CSF" "CSF" "RSF" %-s\n",
1984                        "UUID", cooked ? "bytes" : "1K-blocks",
1985                        "Used", "Available", "Use%", "Mounted on");
1986
1987         for (tp = types; tp->st_name != NULL; tp++) {
1988                 for (index = 0; ; index++) {
1989                         memset(&stat_buf, 0, sizeof(struct obd_statfs));
1990                         memset(&uuid_buf, 0, sizeof(struct obd_uuid));
1991                         type = lazy ? tp->st_op | LL_STATFS_NODELAY : tp->st_op;
1992                         rc = llapi_obd_statfs(mntdir, type, index,
1993                                               &stat_buf, &uuid_buf);
1994                         if (rc == -ENODEV)
1995                                 break;
1996
1997                         if (rc == -EAGAIN)
1998                                 continue;
1999
2000                         if (poolname && tp->st_op == LL_STATFS_LOV &&
2001                             llapi_search_ost(fsname, poolname,
2002                                              obd_uuid2str(&uuid_buf)) != 1)
2003                                 continue;
2004
2005                         /* the llapi_obd_statfs() call may have returned with
2006                          * an error, but if it filled in uuid_buf we will at
2007                          * lease use that to print out a message for that OBD.
2008                          * If we didn't get anything in the uuid_buf, then fill
2009                          * it in so that we can print an error message. */
2010                         if (uuid_buf.uuid[0] == '\0')
2011                                 sprintf(uuid_buf.uuid, "%s%04x",
2012                                         tp->st_name, index);
2013                         showdf(mntdir, &stat_buf, obd_uuid2str(&uuid_buf),
2014                                ishow, cooked, tp->st_name, index, rc);
2015
2016                         if (rc == 0) {
2017                                 if (tp->st_op == LL_STATFS_LMV) {
2018                                         sum.os_ffree += stat_buf.os_ffree;
2019                                         sum.os_files += stat_buf.os_files;
2020                                 } else /* if (tp->st_op == LL_STATFS_LOV) */ {
2021                                         sum.os_blocks += stat_buf.os_blocks *
2022                                                 stat_buf.os_bsize;
2023                                         sum.os_bfree  += stat_buf.os_bfree *
2024                                                 stat_buf.os_bsize;
2025                                         sum.os_bavail += stat_buf.os_bavail *
2026                                                 stat_buf.os_bsize;
2027                                         ost_ffree += stat_buf.os_ffree;
2028                                 }
2029                         } else if (rc == -EINVAL || rc == -EFAULT) {
2030                                 break;
2031                         }
2032                 }
2033         }
2034
2035         /* If we don't have as many objects free on the OST as inodes
2036          * on the MDS, we reduce the total number of inodes to
2037          * compensate, so that the "inodes in use" number is correct.
2038          * Matches ll_statfs_internal() so the results are consistent. */
2039         if (ost_ffree < sum.os_ffree) {
2040                 sum.os_files = (sum.os_files - sum.os_ffree) + ost_ffree;
2041                 sum.os_ffree = ost_ffree;
2042         }
2043         printf("\n");
2044         showdf(mntdir, &sum, "filesystem summary:", ishow, cooked, NULL, 0, 0);
2045         printf("\n");
2046         return 0;
2047 }
2048
2049 static int lfs_df(int argc, char **argv)
2050 {
2051         char mntdir[PATH_MAX] = {'\0'}, path[PATH_MAX] = {'\0'};
2052         int ishow = 0, cooked = 0;
2053         int lazy = 0;
2054         int c, rc = 0, index = 0;
2055         char fsname[PATH_MAX] = "", *pool_name = NULL;
2056         struct option long_opts[] = {
2057                 {"pool", required_argument, 0, 'p'},
2058                 {"lazy", 0, 0, 'l'},
2059                 {0, 0, 0, 0}
2060         };
2061
2062         while ((c = getopt_long(argc, argv, "hilp:", long_opts, NULL)) != -1) {
2063                 switch (c) {
2064                 case 'i':
2065                         ishow = 1;
2066                         break;
2067                 case 'h':
2068                         cooked = 1;
2069                         break;
2070                 case 'l':
2071                         lazy = 1;
2072                         break;
2073                 case 'p':
2074                         pool_name = optarg;
2075                         break;
2076                 default:
2077                         return CMD_HELP;
2078                 }
2079         }
2080         if (optind < argc && !realpath(argv[optind], path)) {
2081                 rc = -errno;
2082                 fprintf(stderr, "error: invalid path '%s': %s\n",
2083                         argv[optind], strerror(-rc));
2084                 return rc;
2085         }
2086
2087         while (!llapi_search_mounts(path, index++, mntdir, fsname)) {
2088                 /* Check if we have a mount point */
2089                 if (mntdir[0] == '\0')
2090                         continue;
2091
2092                 rc = mntdf(mntdir, fsname, pool_name, ishow, cooked, lazy);
2093                 if (rc || path[0] != '\0')
2094                         break;
2095                 fsname[0] = '\0'; /* avoid matching in next loop */
2096                 mntdir[0] = '\0'; /* avoid matching in next loop */
2097         }
2098
2099         return rc;
2100 }
2101
2102 static int lfs_getname(int argc, char **argv)
2103 {
2104         char mntdir[PATH_MAX] = "", path[PATH_MAX] = "", fsname[PATH_MAX] = "";
2105         int rc = 0, index = 0, c;
2106         char buf[sizeof(struct obd_uuid)];
2107
2108         while ((c = getopt(argc, argv, "h")) != -1)
2109                 return CMD_HELP;
2110
2111         if (optind == argc) { /* no paths specified, get all paths. */
2112                 while (!llapi_search_mounts(path, index++, mntdir, fsname)) {
2113                         rc = llapi_getname(mntdir, buf, sizeof(buf));
2114                         if (rc < 0) {
2115                                 fprintf(stderr,
2116                                         "cannot get name for `%s': %s\n",
2117                                         mntdir, strerror(-rc));
2118                                 break;
2119                         }
2120
2121                         printf("%s %s\n", buf, mntdir);
2122
2123                         path[0] = fsname[0] = mntdir[0] = 0;
2124                 }
2125         } else { /* paths specified, only attempt to search these. */
2126                 for (; optind < argc; optind++) {
2127                         rc = llapi_getname(argv[optind], buf, sizeof(buf));
2128                         if (rc < 0) {
2129                                 fprintf(stderr,
2130                                         "cannot get name for `%s': %s\n",
2131                                         argv[optind], strerror(-rc));
2132                                 break;
2133                         }
2134
2135                         printf("%s %s\n", buf, argv[optind]);
2136                 }
2137         }
2138         return rc;
2139 }
2140
2141 static int lfs_check(int argc, char **argv)
2142 {
2143         int rc;
2144         char mntdir[PATH_MAX] = {'\0'};
2145         int num_types = 1;
2146         char *obd_types[2];
2147         char obd_type1[4];
2148         char obd_type2[4];
2149
2150         if (argc != 2)
2151                 return CMD_HELP;
2152
2153         obd_types[0] = obd_type1;
2154         obd_types[1] = obd_type2;
2155
2156         if (strcmp(argv[1], "osts") == 0) {
2157                 strcpy(obd_types[0], "osc");
2158         } else if (strcmp(argv[1], "mds") == 0) {
2159                 strcpy(obd_types[0], "mdc");
2160         } else if (strcmp(argv[1], "servers") == 0) {
2161                 num_types = 2;
2162                 strcpy(obd_types[0], "osc");
2163                 strcpy(obd_types[1], "mdc");
2164         } else {
2165                 fprintf(stderr, "error: %s: option '%s' unrecognized\n",
2166                                 argv[0], argv[1]);
2167                         return CMD_HELP;
2168         }
2169
2170         rc = llapi_search_mounts(NULL, 0, mntdir, NULL);
2171         if (rc < 0 || mntdir[0] == '\0') {
2172                 fprintf(stderr, "No suitable Lustre mount found\n");
2173                 return rc;
2174         }
2175
2176         rc = llapi_target_check(num_types, obd_types, mntdir);
2177         if (rc)
2178                 fprintf(stderr, "error: %s: %s status failed\n",
2179                                 argv[0],argv[1]);
2180
2181         return rc;
2182
2183 }
2184
2185 static int lfs_join(int argc, char **argv)
2186 {
2187         fprintf(stderr, "join two lustre files into one.\n"
2188                         "obsolete, HEAD does not support it anymore.\n");
2189         return 0;
2190 }
2191
2192 #ifdef HAVE_SYS_QUOTA_H
2193 static int lfs_quotacheck(int argc, char **argv)
2194 {
2195         int c, check_type = 0;
2196         char *mnt;
2197         struct if_quotacheck qchk;
2198         struct if_quotactl qctl;
2199         char *obd_type = (char *)qchk.obd_type;
2200         int rc;
2201
2202         memset(&qchk, 0, sizeof(qchk));
2203
2204         while ((c = getopt(argc, argv, "gu")) != -1) {
2205                 switch (c) {
2206                 case 'u':
2207                         check_type |= 0x01;
2208                         break;
2209                 case 'g':
2210                         check_type |= 0x02;
2211                         break;
2212                 default:
2213                         fprintf(stderr, "error: %s: option '-%c' "
2214                                         "unrecognized\n", argv[0], c);
2215                         return CMD_HELP;
2216                 }
2217         }
2218
2219         if (check_type)
2220                 check_type--;
2221         else    /* do quotacheck for both user & group quota by default */
2222                 check_type = 0x02;
2223
2224         if (argc == optind)
2225                 return CMD_HELP;
2226
2227         mnt = argv[optind];
2228
2229         rc = llapi_quotacheck(mnt, check_type);
2230         if (rc == -EOPNOTSUPP) {
2231                 fprintf(stderr, "error: quotacheck not supported by the quota "
2232                         "master.\nPlease note that quotacheck is deprecated as "
2233                         "of lustre 2.4.0 since space accounting is always "
2234                         "enabled.\nFilesystems not formatted with 2.4 utils or "
2235                         "beyond can be upgraded with tunefs.lustre --quota.\n");
2236                 return rc;
2237         } else if (rc) {
2238                 fprintf(stderr, "quotacheck failed: %s\n", strerror(-rc));
2239                 return rc;
2240         }
2241
2242         rc = llapi_poll_quotacheck(mnt, &qchk);
2243         if (rc) {
2244                 if (*obd_type)
2245                         fprintf(stderr, "%s %s ", obd_type,
2246                                 obd_uuid2str(&qchk.obd_uuid));
2247                 fprintf(stderr, "quota check failed: %s\n", strerror(-rc));
2248                 return rc;
2249         }
2250
2251         memset(&qctl, 0, sizeof(qctl));
2252         qctl.qc_cmd = LUSTRE_Q_QUOTAON;
2253         qctl.qc_type = check_type;
2254         rc = llapi_quotactl(mnt, &qctl);
2255         if (rc && rc != -EALREADY) {
2256                 if (*obd_type)
2257                         fprintf(stderr, "%s %s ", (char *)qctl.obd_type,
2258                                 obd_uuid2str(&qctl.obd_uuid));
2259                 fprintf(stderr, "%s turn on quota failed: %s\n",
2260                         argv[0], strerror(-rc));
2261                 return rc;
2262         }
2263
2264         return 0;
2265 }
2266
2267 static int lfs_quotaon(int argc, char **argv)
2268 {
2269         int c;
2270         char *mnt;
2271         struct if_quotactl qctl;
2272         char *obd_type = (char *)qctl.obd_type;
2273         int rc;
2274
2275         memset(&qctl, 0, sizeof(qctl));
2276         qctl.qc_cmd = LUSTRE_Q_QUOTAON;
2277
2278         while ((c = getopt(argc, argv, "fgu")) != -1) {
2279                 switch (c) {
2280                 case 'u':
2281                         qctl.qc_type |= 0x01;
2282                         break;
2283                 case 'g':
2284                         qctl.qc_type |= 0x02;
2285                         break;
2286                 case 'f':
2287                         qctl.qc_cmd = LUSTRE_Q_QUOTAOFF;
2288                         break;
2289                 default:
2290                         fprintf(stderr, "error: %s: option '-%c' "
2291                                         "unrecognized\n", argv[0], c);
2292                         return CMD_HELP;
2293                 }
2294         }
2295
2296         if (qctl.qc_type)
2297                 qctl.qc_type--;
2298         else /* by default, enable quota for both user & group */
2299                 qctl.qc_type = 0x02;
2300
2301         if (argc == optind)
2302                 return CMD_HELP;
2303
2304         mnt = argv[optind];
2305
2306         rc = llapi_quotactl(mnt, &qctl);
2307         if (rc) {
2308                 if (rc == -EOPNOTSUPP) {
2309                         fprintf(stderr, "error: quotaon not supported by the "
2310                                 "quota master.\nPlease note that quotaon/off is"
2311                                 " deprecated as of lustre 2.4.0.\nQuota "
2312                                 "enforcement should now be enabled on the MGS "
2313                                 "via:\nmgs# lctl conf_param ${FSNAME}.quota."
2314                                 "<ost|mdt>=<u|g|ug>\n(ost for block quota, mdt "
2315                                 "for inode quota, u for user and g for group"
2316                                 "\n");
2317                 } else if (rc == -EALREADY) {
2318                         rc = 0;
2319                 } else if (rc == -ENOENT) {
2320                         fprintf(stderr, "error: cannot find quota database, "
2321                                         "make sure you have run quotacheck\n");
2322                 } else {
2323                         if (*obd_type)
2324                                 fprintf(stderr, "%s %s ", obd_type,
2325                                         obd_uuid2str(&qctl.obd_uuid));
2326                         fprintf(stderr, "%s failed: %s\n", argv[0],
2327                                 strerror(-rc));
2328                 }
2329         }
2330
2331         return rc;
2332 }
2333
2334 static int lfs_quotaoff(int argc, char **argv)
2335 {
2336         int c;
2337         char *mnt;
2338         struct if_quotactl qctl;
2339         char *obd_type = (char *)qctl.obd_type;
2340         int rc;
2341
2342         memset(&qctl, 0, sizeof(qctl));
2343         qctl.qc_cmd = LUSTRE_Q_QUOTAOFF;
2344
2345         while ((c = getopt(argc, argv, "gu")) != -1) {
2346                 switch (c) {
2347                 case 'u':
2348                         qctl.qc_type |= 0x01;
2349                         break;
2350                 case 'g':
2351                         qctl.qc_type |= 0x02;
2352                         break;
2353                 default:
2354                         fprintf(stderr, "error: %s: option '-%c' "
2355                                         "unrecognized\n", argv[0], c);
2356                         return CMD_HELP;
2357                 }
2358         }
2359
2360         if (qctl.qc_type)
2361                 qctl.qc_type--;
2362         else /* by default, disable quota for both user & group */
2363                 qctl.qc_type = 0x02;
2364
2365         if (argc == optind)
2366                 return CMD_HELP;
2367
2368         mnt = argv[optind];
2369
2370         rc = llapi_quotactl(mnt, &qctl);
2371         if (rc) {
2372                 if (rc == -EOPNOTSUPP) {
2373                         fprintf(stderr, "error: quotaoff not supported by the "
2374                                 "quota master.\nPlease note that quotaon/off is"
2375                                 " deprecated as of lustre 2.4.0.\nQuota "
2376                                 "enforcement can be disabled on the MGS via:\n"
2377                                 "mgs# lctl conf_param ${FSNAME}.quota.<ost|mdt>"
2378                                 "=\"\"\n");
2379                 } else if (rc == -EALREADY) {
2380                         rc = 0;
2381                 } else {
2382                         if (*obd_type)
2383                                 fprintf(stderr, "%s %s ", obd_type,
2384                                         obd_uuid2str(&qctl.obd_uuid));
2385                         fprintf(stderr, "quotaoff failed: %s\n",
2386                                 strerror(-rc));
2387                 }
2388         }
2389
2390         return rc;
2391 }
2392
2393 #define ARG2INT(nr, str, msg)                                           \
2394 do {                                                                    \
2395         char *endp;                                                     \
2396         nr = strtol(str, &endp, 0);                                     \
2397         if (*endp) {                                                    \
2398                 fprintf(stderr, "error: bad %s: %s\n", msg, str);       \
2399                 return CMD_HELP;                                        \
2400         }                                                               \
2401 } while (0)
2402
2403 #define ADD_OVERFLOW(a,b) ((a + b) < a) ? (a = ULONG_MAX) : (a = a + b)
2404
2405 /* Convert format time string "XXwXXdXXhXXmXXs" into seconds value
2406  * returns the value or ULONG_MAX on integer overflow or incorrect format
2407  * Notes:
2408  *        1. the order of specifiers is arbitrary (may be: 5w3s or 3s5w)
2409  *        2. specifiers may be encountered multiple times (2s3s is 5 seconds)
2410  *        3. empty integer value is interpreted as 0
2411  */
2412 static unsigned long str2sec(const char* timestr)
2413 {
2414         const char spec[] = "smhdw";
2415         const unsigned long mult[] = {1, 60, 60*60, 24*60*60, 7*24*60*60};
2416         unsigned long val = 0;
2417         char *tail;
2418
2419         if (strpbrk(timestr, spec) == NULL) {
2420                 /* no specifiers inside the time string,
2421                    should treat it as an integer value */
2422                 val = strtoul(timestr, &tail, 10);
2423                 return *tail ? ULONG_MAX : val;
2424         }
2425
2426         /* format string is XXwXXdXXhXXmXXs */
2427         while (*timestr) {
2428                 unsigned long v;
2429                 int ind;
2430                 char* ptr;
2431
2432                 v = strtoul(timestr, &tail, 10);
2433                 if (v == ULONG_MAX || *tail == '\0')
2434                         /* value too large (ULONG_MAX or more)
2435                            or missing specifier */
2436                         goto error;
2437
2438                 ptr = strchr(spec, *tail);
2439                 if (ptr == NULL)
2440                         /* unknown specifier */
2441                         goto error;
2442
2443                 ind = ptr - spec;
2444
2445                 /* check if product will overflow the type */
2446                 if (!(v < ULONG_MAX / mult[ind]))
2447                         goto error;
2448
2449                 ADD_OVERFLOW(val, mult[ind] * v);
2450                 if (val == ULONG_MAX)
2451                         goto error;
2452
2453                 timestr = tail + 1;
2454         }
2455
2456         return val;
2457
2458 error:
2459         return ULONG_MAX;
2460 }
2461
2462 #define ARG2ULL(nr, str, def_units)                                     \
2463 do {                                                                    \
2464         unsigned long long limit, units = def_units;                    \
2465         int rc;                                                         \
2466                                                                         \
2467         rc = llapi_parse_size(str, &limit, &units, 1);                  \
2468         if (rc < 0) {                                                   \
2469                 fprintf(stderr, "error: bad limit value %s\n", str);    \
2470                 return CMD_HELP;                                        \
2471         }                                                               \
2472         nr = limit;                                                     \
2473 } while (0)
2474
2475 static inline int has_times_option(int argc, char **argv)
2476 {
2477         int i;
2478
2479         for (i = 1; i < argc; i++)
2480                 if (!strcmp(argv[i], "-t"))
2481                         return 1;
2482
2483         return 0;
2484 }
2485
2486 int lfs_setquota_times(int argc, char **argv)
2487 {
2488         int c, rc;
2489         struct if_quotactl qctl;
2490         char *mnt, *obd_type = (char *)qctl.obd_type;
2491         struct obd_dqblk *dqb = &qctl.qc_dqblk;
2492         struct obd_dqinfo *dqi = &qctl.qc_dqinfo;
2493         struct option long_opts[] = {
2494                 {"block-grace",     required_argument, 0, 'b'},
2495                 {"group",           no_argument,       0, 'g'},
2496                 {"inode-grace",     required_argument, 0, 'i'},
2497                 {"times",           no_argument,       0, 't'},
2498                 {"user",            no_argument,       0, 'u'},
2499                 {0, 0, 0, 0}
2500         };
2501
2502         memset(&qctl, 0, sizeof(qctl));
2503         qctl.qc_cmd  = LUSTRE_Q_SETINFO;
2504         qctl.qc_type = UGQUOTA;
2505
2506         while ((c = getopt_long(argc, argv, "b:gi:tu", long_opts, NULL)) != -1) {
2507                 switch (c) {
2508                 case 'u':
2509                 case 'g':
2510                         if (qctl.qc_type != UGQUOTA) {
2511                                 fprintf(stderr, "error: -u and -g can't be used "
2512                                                 "more than once\n");
2513                                 return CMD_HELP;
2514                         }
2515                         qctl.qc_type = (c == 'u') ? USRQUOTA : GRPQUOTA;
2516                         break;
2517                 case 'b':
2518                         if ((dqi->dqi_bgrace = str2sec(optarg)) == ULONG_MAX) {
2519                                 fprintf(stderr, "error: bad block-grace: %s\n",
2520                                         optarg);
2521                                 return CMD_HELP;
2522                         }
2523                         dqb->dqb_valid |= QIF_BTIME;
2524                         break;
2525                 case 'i':
2526                         if ((dqi->dqi_igrace = str2sec(optarg)) == ULONG_MAX) {
2527                                 fprintf(stderr, "error: bad inode-grace: %s\n",
2528                                         optarg);
2529                                 return CMD_HELP;
2530                         }
2531                         dqb->dqb_valid |= QIF_ITIME;
2532                         break;
2533                 case 't': /* Yes, of course! */
2534                         break;
2535                 default: /* getopt prints error message for us when opterr != 0 */
2536                         return CMD_HELP;
2537                 }
2538         }
2539
2540         if (qctl.qc_type == UGQUOTA) {
2541                 fprintf(stderr, "error: neither -u nor -g specified\n");
2542                 return CMD_HELP;
2543         }
2544
2545         if (optind != argc - 1) {
2546                 fprintf(stderr, "error: unexpected parameters encountered\n");
2547                 return CMD_HELP;
2548         }
2549
2550         mnt = argv[optind];
2551         rc = llapi_quotactl(mnt, &qctl);
2552         if (rc) {
2553                 if (*obd_type)
2554                         fprintf(stderr, "%s %s ", obd_type,
2555                                 obd_uuid2str(&qctl.obd_uuid));
2556                 fprintf(stderr, "setquota failed: %s\n", strerror(-rc));
2557                 return rc;
2558         }
2559
2560         return 0;
2561 }
2562
2563 #define BSLIMIT (1 << 0)
2564 #define BHLIMIT (1 << 1)
2565 #define ISLIMIT (1 << 2)
2566 #define IHLIMIT (1 << 3)
2567
2568 int lfs_setquota(int argc, char **argv)
2569 {
2570         int c, rc;
2571         struct if_quotactl qctl;
2572         char *mnt, *obd_type = (char *)qctl.obd_type;
2573         struct obd_dqblk *dqb = &qctl.qc_dqblk;
2574         struct option long_opts[] = {
2575                 {"block-softlimit", required_argument, 0, 'b'},
2576                 {"block-hardlimit", required_argument, 0, 'B'},
2577                 {"group",           required_argument, 0, 'g'},
2578                 {"inode-softlimit", required_argument, 0, 'i'},
2579                 {"inode-hardlimit", required_argument, 0, 'I'},
2580                 {"user",            required_argument, 0, 'u'},
2581                 {0, 0, 0, 0}
2582         };
2583         unsigned limit_mask = 0;
2584         char *endptr;
2585
2586         if (has_times_option(argc, argv))
2587                 return lfs_setquota_times(argc, argv);
2588
2589         memset(&qctl, 0, sizeof(qctl));
2590         qctl.qc_cmd  = LUSTRE_Q_SETQUOTA;
2591         qctl.qc_type = UGQUOTA; /* UGQUOTA makes no sense for setquota,
2592                                  * so it can be used as a marker that qc_type
2593                                  * isn't reinitialized from command line */
2594
2595         while ((c = getopt_long(argc, argv, "b:B:g:i:I:u:", long_opts, NULL)) != -1) {
2596                 switch (c) {
2597                 case 'u':
2598                 case 'g':
2599                         if (qctl.qc_type != UGQUOTA) {
2600                                 fprintf(stderr, "error: -u and -g can't be used"
2601                                                 " more than once\n");
2602                                 return CMD_HELP;
2603                         }
2604                         qctl.qc_type = (c == 'u') ? USRQUOTA : GRPQUOTA;
2605                         rc = name2id(&qctl.qc_id, optarg,
2606                                      (qctl.qc_type == USRQUOTA) ? USER : GROUP);
2607                         if (rc) {
2608                                 qctl.qc_id = strtoul(optarg, &endptr, 10);
2609                                 if (*endptr != '\0') {
2610                                         fprintf(stderr, "error: can't find id "
2611                                                 "for name %s\n", optarg);
2612                                         return CMD_HELP;
2613                                 }
2614                         }
2615                         break;
2616                 case 'b':
2617                         ARG2ULL(dqb->dqb_bsoftlimit, optarg, 1024);
2618                         dqb->dqb_bsoftlimit >>= 10;
2619                         limit_mask |= BSLIMIT;
2620                         if (dqb->dqb_bsoftlimit &&
2621                             dqb->dqb_bsoftlimit <= 1024) /* <= 1M? */
2622                                 fprintf(stderr, "warning: block softlimit is "
2623                                         "smaller than the miminal qunit size, "
2624                                         "please see the help of setquota or "
2625                                         "Lustre manual for details.\n");
2626                         break;
2627                 case 'B':
2628                         ARG2ULL(dqb->dqb_bhardlimit, optarg, 1024);
2629                         dqb->dqb_bhardlimit >>= 10;
2630                         limit_mask |= BHLIMIT;
2631                         if (dqb->dqb_bhardlimit &&
2632                             dqb->dqb_bhardlimit <= 1024) /* <= 1M? */
2633                                 fprintf(stderr, "warning: block hardlimit is "
2634                                         "smaller than the miminal qunit size, "
2635                                         "please see the help of setquota or "
2636                                         "Lustre manual for details.\n");
2637                         break;
2638                 case 'i':
2639                         ARG2ULL(dqb->dqb_isoftlimit, optarg, 1);
2640                         limit_mask |= ISLIMIT;
2641                         if (dqb->dqb_isoftlimit &&
2642                             dqb->dqb_isoftlimit <= 1024) /* <= 1K inodes? */
2643                                 fprintf(stderr, "warning: inode softlimit is "
2644                                         "smaller than the miminal qunit size, "
2645                                         "please see the help of setquota or "
2646                                         "Lustre manual for details.\n");
2647                         break;
2648                 case 'I':
2649                         ARG2ULL(dqb->dqb_ihardlimit, optarg, 1);
2650                         limit_mask |= IHLIMIT;
2651                         if (dqb->dqb_ihardlimit &&
2652                             dqb->dqb_ihardlimit <= 1024) /* <= 1K inodes? */
2653                                 fprintf(stderr, "warning: inode hardlimit is "
2654                                         "smaller than the miminal qunit size, "
2655                                         "please see the help of setquota or "
2656                                         "Lustre manual for details.\n");
2657                         break;
2658                 default: /* getopt prints error message for us when opterr != 0 */
2659                         return CMD_HELP;
2660                 }
2661         }
2662
2663         if (qctl.qc_type == UGQUOTA) {
2664                 fprintf(stderr, "error: neither -u nor -g was specified\n");
2665                 return CMD_HELP;
2666         }
2667
2668         if (limit_mask == 0) {
2669                 fprintf(stderr, "error: at least one limit must be specified\n");
2670                 return CMD_HELP;
2671         }
2672
2673         if (optind != argc - 1) {
2674                 fprintf(stderr, "error: unexpected parameters encountered\n");
2675                 return CMD_HELP;
2676         }
2677
2678         mnt = argv[optind];
2679
2680         if ((!(limit_mask & BHLIMIT) ^ !(limit_mask & BSLIMIT)) ||
2681             (!(limit_mask & IHLIMIT) ^ !(limit_mask & ISLIMIT))) {
2682                 /* sigh, we can't just set blimits/ilimits */
2683                 struct if_quotactl tmp_qctl = {.qc_cmd  = LUSTRE_Q_GETQUOTA,
2684                                                .qc_type = qctl.qc_type,
2685                                                .qc_id   = qctl.qc_id};
2686
2687                 rc = llapi_quotactl(mnt, &tmp_qctl);
2688                 if (rc < 0) {
2689                         fprintf(stderr, "error: setquota failed while retrieving"
2690                                         " current quota settings (%s)\n",
2691                                         strerror(-rc));
2692                         return rc;
2693                 }
2694
2695                 if (!(limit_mask & BHLIMIT))
2696                         dqb->dqb_bhardlimit = tmp_qctl.qc_dqblk.dqb_bhardlimit;
2697                 if (!(limit_mask & BSLIMIT))
2698                         dqb->dqb_bsoftlimit = tmp_qctl.qc_dqblk.dqb_bsoftlimit;
2699                 if (!(limit_mask & IHLIMIT))
2700                         dqb->dqb_ihardlimit = tmp_qctl.qc_dqblk.dqb_ihardlimit;
2701                 if (!(limit_mask & ISLIMIT))
2702                         dqb->dqb_isoftlimit = tmp_qctl.qc_dqblk.dqb_isoftlimit;
2703
2704                 /* Keep grace times if we have got no softlimit arguments */
2705                 if ((limit_mask & BHLIMIT) && !(limit_mask & BSLIMIT)) {
2706                         dqb->dqb_valid |= QIF_BTIME;
2707                         dqb->dqb_btime = tmp_qctl.qc_dqblk.dqb_btime;
2708                 }
2709
2710                 if ((limit_mask & IHLIMIT) && !(limit_mask & ISLIMIT)) {
2711                         dqb->dqb_valid |= QIF_ITIME;
2712                         dqb->dqb_itime = tmp_qctl.qc_dqblk.dqb_itime;
2713                 }
2714         }
2715
2716         dqb->dqb_valid |= (limit_mask & (BHLIMIT | BSLIMIT)) ? QIF_BLIMITS : 0;
2717         dqb->dqb_valid |= (limit_mask & (IHLIMIT | ISLIMIT)) ? QIF_ILIMITS : 0;
2718
2719         rc = llapi_quotactl(mnt, &qctl);
2720         if (rc) {
2721                 if (*obd_type)
2722                         fprintf(stderr, "%s %s ", obd_type,
2723                                 obd_uuid2str(&qctl.obd_uuid));
2724                 fprintf(stderr, "setquota failed: %s\n", strerror(-rc));
2725                 return rc;
2726         }
2727
2728         return 0;
2729 }
2730
2731 static inline char *type2name(int check_type)
2732 {
2733         if (check_type == USRQUOTA)
2734                 return "user";
2735         else if (check_type == GRPQUOTA)
2736                 return "group";
2737         else
2738                 return "unknown";
2739 }
2740
2741 /* Converts seconds value into format string
2742  * result is returned in buf
2743  * Notes:
2744  *        1. result is in descenting order: 1w2d3h4m5s
2745  *        2. zero fields are not filled (except for p. 3): 5d1s
2746  *        3. zero seconds value is presented as "0s"
2747  */
2748 static char * __sec2str(time_t seconds, char *buf)
2749 {
2750         const char spec[] = "smhdw";
2751         const unsigned long mult[] = {1, 60, 60*60, 24*60*60, 7*24*60*60};
2752         unsigned long c;
2753         char *tail = buf;
2754         int i;
2755
2756         for (i = sizeof(mult) / sizeof(mult[0]) - 1 ; i >= 0; i--) {
2757                 c = seconds / mult[i];
2758
2759                 if (c > 0 || (i == 0 && buf == tail))
2760                         tail += snprintf(tail, 40-(tail-buf), "%lu%c", c, spec[i]);
2761
2762                 seconds %= mult[i];
2763         }
2764
2765         return tail;
2766 }
2767
2768 static void sec2str(time_t seconds, char *buf, int rc)
2769 {
2770         char *tail = buf;
2771
2772         if (rc)
2773                 *tail++ = '[';
2774
2775         tail = __sec2str(seconds, tail);
2776
2777         if (rc && tail - buf < 39) {
2778                 *tail++ = ']';
2779                 *tail++ = 0;
2780         }
2781 }
2782
2783 static void diff2str(time_t seconds, char *buf, time_t now)
2784 {
2785
2786         buf[0] = 0;
2787         if (!seconds)
2788                 return;
2789         if (seconds <= now) {
2790                 strcpy(buf, "none");
2791                 return;
2792         }
2793         __sec2str(seconds - now, buf);
2794 }
2795
2796 static void print_quota_title(char *name, struct if_quotactl *qctl,
2797                               bool human_readable)
2798 {
2799         printf("Disk quotas for %s %s (%cid %u):\n",
2800                type2name(qctl->qc_type), name,
2801                *type2name(qctl->qc_type), qctl->qc_id);
2802         printf("%15s%8s %7s%8s%8s%8s %7s%8s%8s\n",
2803                "Filesystem", human_readable ? "used" : "kbytes",
2804                "quota", "limit", "grace",
2805                "files", "quota", "limit", "grace");
2806 }
2807
2808 static void kbytes2str(__u64 num, char *buf, bool h)
2809 {
2810         if (!h) {
2811                 sprintf(buf, LPU64, num);
2812         } else {
2813                 if (num >> 30)
2814                         sprintf(buf, "%5.4gT", (double)num / (1 << 30));
2815                 else if (num >> 20)
2816                         sprintf(buf, "%5.4gG", (double)num / (1 << 20));
2817                 else if (num >> 10)
2818                         sprintf(buf, "%5.4gM", (double)num / (1 << 10));
2819                 else
2820                         sprintf(buf, LPU64"%s", num, "k");
2821         }
2822 }
2823
2824 static void print_quota(char *mnt, struct if_quotactl *qctl, int type,
2825                         int rc, bool h)
2826 {
2827         time_t now;
2828
2829         time(&now);
2830
2831         if (qctl->qc_cmd == LUSTRE_Q_GETQUOTA || qctl->qc_cmd == Q_GETOQUOTA) {
2832                 int bover = 0, iover = 0;
2833                 struct obd_dqblk *dqb = &qctl->qc_dqblk;
2834                 char numbuf[3][32];
2835                 char timebuf[40];
2836                 char strbuf[32];
2837
2838                 if (dqb->dqb_bhardlimit &&
2839                     lustre_stoqb(dqb->dqb_curspace) >= dqb->dqb_bhardlimit) {
2840                         bover = 1;
2841                 } else if (dqb->dqb_bsoftlimit && dqb->dqb_btime) {
2842                         if (dqb->dqb_btime > now) {
2843                                 bover = 2;
2844                         } else {
2845                                 bover = 3;
2846                         }
2847                 }
2848
2849                 if (dqb->dqb_ihardlimit &&
2850                     dqb->dqb_curinodes >= dqb->dqb_ihardlimit) {
2851                         iover = 1;
2852                 } else if (dqb->dqb_isoftlimit && dqb->dqb_itime) {
2853                         if (dqb->dqb_itime > now) {
2854                                 iover = 2;
2855                         } else {
2856                                 iover = 3;
2857                         }
2858                 }
2859
2860
2861                 if (strlen(mnt) > 15)
2862                         printf("%s\n%15s", mnt, "");
2863                 else
2864                         printf("%15s", mnt);
2865
2866                 if (bover)
2867                         diff2str(dqb->dqb_btime, timebuf, now);
2868
2869                 kbytes2str(lustre_stoqb(dqb->dqb_curspace), strbuf, h);
2870                 if (rc == -EREMOTEIO)
2871                         sprintf(numbuf[0], "%s*", strbuf);
2872                 else
2873                         sprintf(numbuf[0], (dqb->dqb_valid & QIF_SPACE) ?
2874                                 "%s" : "[%s]", strbuf);
2875
2876                 kbytes2str(dqb->dqb_bsoftlimit, strbuf, h);
2877                 if (type == QC_GENERAL)
2878                         sprintf(numbuf[1], (dqb->dqb_valid & QIF_BLIMITS) ?
2879                                 "%s" : "[%s]", strbuf);
2880                 else
2881                         sprintf(numbuf[1], "%s", "-");
2882
2883                 kbytes2str(dqb->dqb_bhardlimit, strbuf, h);
2884                 sprintf(numbuf[2], (dqb->dqb_valid & QIF_BLIMITS) ?
2885                         "%s" : "[%s]", strbuf);
2886
2887                 printf(" %7s%c %6s %7s %7s",
2888                        numbuf[0], bover ? '*' : ' ', numbuf[1],
2889                        numbuf[2], bover > 1 ? timebuf : "-");
2890
2891                 if (iover)
2892                         diff2str(dqb->dqb_itime, timebuf, now);
2893
2894                 sprintf(numbuf[0], (dqb->dqb_valid & QIF_INODES) ?
2895                         LPU64 : "["LPU64"]", dqb->dqb_curinodes);
2896
2897                 if (type == QC_GENERAL)
2898                         sprintf(numbuf[1], (dqb->dqb_valid & QIF_ILIMITS) ?
2899                                 LPU64 : "["LPU64"]", dqb->dqb_isoftlimit);
2900                 else
2901                         sprintf(numbuf[1], "%s", "-");
2902
2903                 sprintf(numbuf[2], (dqb->dqb_valid & QIF_ILIMITS) ?
2904                         LPU64 : "["LPU64"]", dqb->dqb_ihardlimit);
2905
2906                 if (type != QC_OSTIDX)
2907                         printf(" %7s%c %6s %7s %7s",
2908                                numbuf[0], iover ? '*' : ' ', numbuf[1],
2909                                numbuf[2], iover > 1 ? timebuf : "-");
2910                 else
2911                         printf(" %7s %7s %7s %7s", "-", "-", "-", "-");
2912                 printf("\n");
2913
2914         } else if (qctl->qc_cmd == LUSTRE_Q_GETINFO ||
2915                    qctl->qc_cmd == Q_GETOINFO) {
2916                 char bgtimebuf[40];
2917                 char igtimebuf[40];
2918
2919                 sec2str(qctl->qc_dqinfo.dqi_bgrace, bgtimebuf, rc);
2920                 sec2str(qctl->qc_dqinfo.dqi_igrace, igtimebuf, rc);
2921                 printf("Block grace time: %s; Inode grace time: %s\n",
2922                        bgtimebuf, igtimebuf);
2923         }
2924 }
2925
2926 static int print_obd_quota(char *mnt, struct if_quotactl *qctl, int is_mdt,
2927                            bool h, __u64 *total)
2928 {
2929         int rc = 0, rc1 = 0, count = 0;
2930         __u32 valid = qctl->qc_valid;
2931
2932         rc = llapi_get_obd_count(mnt, &count, is_mdt);
2933         if (rc) {
2934                 fprintf(stderr, "can not get %s count: %s\n",
2935                         is_mdt ? "mdt": "ost", strerror(-rc));
2936                 return rc;
2937         }
2938
2939         for (qctl->qc_idx = 0; qctl->qc_idx < count; qctl->qc_idx++) {
2940                 qctl->qc_valid = is_mdt ? QC_MDTIDX : QC_OSTIDX;
2941                 rc = llapi_quotactl(mnt, qctl);
2942                 if (rc) {
2943                         /* It is remote client case. */
2944                         if (-rc == EOPNOTSUPP) {
2945                                 rc = 0;
2946                                 goto out;
2947                         }
2948
2949                         if (!rc1)
2950                                 rc1 = rc;
2951                         fprintf(stderr, "quotactl %s%d failed.\n",
2952                                 is_mdt ? "mdt": "ost", qctl->qc_idx);
2953                         continue;
2954                 }
2955
2956                 print_quota(obd_uuid2str(&qctl->obd_uuid), qctl,
2957                             qctl->qc_valid, 0, h);
2958                 *total += is_mdt ? qctl->qc_dqblk.dqb_ihardlimit :
2959                                    qctl->qc_dqblk.dqb_bhardlimit;
2960         }
2961 out:
2962         qctl->qc_valid = valid;
2963         return rc ? : rc1;
2964 }
2965
2966 static int lfs_quota(int argc, char **argv)
2967 {
2968         int c;
2969         char *mnt, *name = NULL;
2970         struct if_quotactl qctl = { .qc_cmd = LUSTRE_Q_GETQUOTA,
2971                                     .qc_type = UGQUOTA };
2972         char *obd_type = (char *)qctl.obd_type;
2973         char *obd_uuid = (char *)qctl.obd_uuid.uuid;
2974         int rc, rc1 = 0, rc2 = 0, rc3 = 0,
2975             verbose = 0, pass = 0, quiet = 0, inacc;
2976         char *endptr;
2977         __u32 valid = QC_GENERAL, idx = 0;
2978         __u64 total_ialloc = 0, total_balloc = 0;
2979         bool human_readable = false;
2980
2981         while ((c = getopt(argc, argv, "gi:I:o:qtuvh")) != -1) {
2982                 switch (c) {
2983                 case 'u':
2984                         if (qctl.qc_type != UGQUOTA) {
2985                                 fprintf(stderr, "error: use either -u or -g\n");
2986                                 return CMD_HELP;
2987                         }
2988                         qctl.qc_type = USRQUOTA;
2989                         break;
2990                 case 'g':
2991                         if (qctl.qc_type != UGQUOTA) {
2992                                 fprintf(stderr, "error: use either -u or -g\n");
2993                                 return CMD_HELP;
2994                         }
2995                         qctl.qc_type = GRPQUOTA;
2996                         break;
2997                 case 't':
2998                         qctl.qc_cmd = LUSTRE_Q_GETINFO;
2999                         break;
3000                 case 'o':
3001                         valid = qctl.qc_valid = QC_UUID;
3002                         strlcpy(obd_uuid, optarg, sizeof(qctl.obd_uuid));
3003                         break;
3004                 case 'i':
3005                         valid = qctl.qc_valid = QC_MDTIDX;
3006                         idx = qctl.qc_idx = atoi(optarg);
3007                         break;
3008                 case 'I':
3009                         valid = qctl.qc_valid = QC_OSTIDX;
3010                         idx = qctl.qc_idx = atoi(optarg);
3011                         break;
3012                 case 'v':
3013                         verbose = 1;
3014                         break;
3015                 case 'q':
3016                         quiet = 1;
3017                         break;
3018                 case 'h':
3019                         human_readable = true;
3020                         break;
3021                 default:
3022                         fprintf(stderr, "error: %s: option '-%c' "
3023                                         "unrecognized\n", argv[0], c);
3024                         return CMD_HELP;
3025                 }
3026         }
3027
3028         /* current uid/gid info for "lfs quota /path/to/lustre/mount" */
3029         if (qctl.qc_cmd == LUSTRE_Q_GETQUOTA && qctl.qc_type == UGQUOTA &&
3030             optind == argc - 1) {
3031 ug_output:
3032                 memset(&qctl, 0, sizeof(qctl)); /* spoiled by print_*_quota */
3033                 qctl.qc_cmd = LUSTRE_Q_GETQUOTA;
3034                 qctl.qc_valid = valid;
3035                 qctl.qc_idx = idx;
3036                 if (pass++ == 0) {
3037                         qctl.qc_type = USRQUOTA;
3038                         qctl.qc_id = geteuid();
3039                 } else {
3040                         qctl.qc_type = GRPQUOTA;
3041                         qctl.qc_id = getegid();
3042                 }
3043                 rc = id2name(&name, qctl.qc_id,
3044                              (qctl.qc_type == USRQUOTA) ? USER : GROUP);
3045                 if (rc)
3046                         name = "<unknown>";
3047         /* lfs quota -u username /path/to/lustre/mount */
3048         } else if (qctl.qc_cmd == LUSTRE_Q_GETQUOTA) {
3049                 /* options should be followed by u/g-name and mntpoint */
3050                 if (optind + 2 != argc || qctl.qc_type == UGQUOTA) {
3051                         fprintf(stderr, "error: missing quota argument(s)\n");
3052                         return CMD_HELP;
3053                 }
3054
3055                 name = argv[optind++];
3056                 rc = name2id(&qctl.qc_id, name,
3057                              (qctl.qc_type == USRQUOTA) ? USER : GROUP);
3058                 if (rc) {
3059                         qctl.qc_id = strtoul(name, &endptr, 10);
3060                         if (*endptr != '\0') {
3061                                 fprintf(stderr, "error: can't find id for name "
3062                                         "%s\n", name);
3063                                 return CMD_HELP;
3064                         }
3065                 }
3066         } else if (optind + 1 != argc || qctl.qc_type == UGQUOTA) {
3067                 fprintf(stderr, "error: missing quota info argument(s)\n");
3068                 return CMD_HELP;
3069         }
3070
3071         mnt = argv[optind];
3072
3073         rc1 = llapi_quotactl(mnt, &qctl);
3074         if (rc1 < 0) {
3075                 switch (rc1) {
3076                 case -ESRCH:
3077                         fprintf(stderr, "%s quotas are not enabled.\n",
3078                                 qctl.qc_type == USRQUOTA ? "user" : "group");
3079                         goto out;
3080                 case -EPERM:
3081                         fprintf(stderr, "Permission denied.\n");
3082                 case -ENOENT:
3083                         /* We already got a "No such file..." message. */
3084                         goto out;
3085                 default:
3086                         fprintf(stderr, "Unexpected quotactl error: %s\n",
3087                                 strerror(-rc1));
3088                 }
3089         }
3090
3091         if (qctl.qc_cmd == LUSTRE_Q_GETQUOTA && !quiet)
3092                 print_quota_title(name, &qctl, human_readable);
3093
3094         if (rc1 && *obd_type)
3095                 fprintf(stderr, "%s %s ", obd_type, obd_uuid);
3096
3097         if (qctl.qc_valid != QC_GENERAL)
3098                 mnt = "";
3099
3100         inacc = (qctl.qc_cmd == LUSTRE_Q_GETQUOTA) &&
3101                 ((qctl.qc_dqblk.dqb_valid & (QIF_LIMITS|QIF_USAGE)) !=
3102                  (QIF_LIMITS|QIF_USAGE));
3103
3104         print_quota(mnt, &qctl, QC_GENERAL, rc1, human_readable);
3105
3106         if (qctl.qc_valid == QC_GENERAL && qctl.qc_cmd != LUSTRE_Q_GETINFO &&
3107             verbose) {
3108                 char strbuf[32];
3109
3110                 rc2 = print_obd_quota(mnt, &qctl, 1, human_readable,
3111                                       &total_ialloc);
3112                 rc3 = print_obd_quota(mnt, &qctl, 0, human_readable,
3113                                       &total_balloc);
3114                 kbytes2str(total_balloc, strbuf, human_readable);
3115                 printf("Total allocated inode limit: "LPU64", total "
3116                        "allocated block limit: %s\n", total_ialloc, strbuf);
3117         }
3118
3119         if (rc1 || rc2 || rc3 || inacc)
3120                 printf("Some errors happened when getting quota info. "
3121                        "Some devices may be not working or deactivated. "
3122                        "The data in \"[]\" is inaccurate.\n");
3123
3124 out:
3125         if (pass == 1)
3126                 goto ug_output;
3127
3128         return rc1;
3129 }
3130 #endif /* HAVE_SYS_QUOTA_H! */
3131
3132 static int flushctx_ioctl(char *mp)
3133 {
3134         int fd, rc;
3135
3136         fd = open(mp, O_RDONLY);
3137         if (fd == -1) {
3138                 fprintf(stderr, "flushctx: error open %s: %s\n",
3139                         mp, strerror(errno));
3140                 return -1;
3141         }
3142
3143         rc = ioctl(fd, LL_IOC_FLUSHCTX);
3144         if (rc == -1)
3145                 fprintf(stderr, "flushctx: error ioctl %s: %s\n",
3146                         mp, strerror(errno));
3147
3148         close(fd);
3149         return rc;
3150 }
3151
3152 static int lfs_flushctx(int argc, char **argv)
3153 {
3154         int     kdestroy = 0, c;
3155         char    mntdir[PATH_MAX] = {'\0'};
3156         int     index = 0;
3157         int     rc = 0;
3158
3159         while ((c = getopt(argc, argv, "k")) != -1) {
3160                 switch (c) {
3161                 case 'k':
3162                         kdestroy = 1;
3163                         break;
3164                 default:
3165                         fprintf(stderr, "error: %s: option '-%c' "
3166                                         "unrecognized\n", argv[0], c);
3167                         return CMD_HELP;
3168                 }
3169         }
3170
3171         if (kdestroy) {
3172             if ((rc = system("kdestroy > /dev/null")) != 0) {
3173                 rc = WEXITSTATUS(rc);
3174                 fprintf(stderr, "error destroying tickets: %d, continuing\n", rc);
3175             }
3176         }
3177
3178         if (optind >= argc) {
3179                 /* flush for all mounted lustre fs. */
3180                 while (!llapi_search_mounts(NULL, index++, mntdir, NULL)) {
3181                         /* Check if we have a mount point */
3182                         if (mntdir[0] == '\0')
3183                                 continue;
3184
3185                         if (flushctx_ioctl(mntdir))
3186                                 rc = -1;
3187
3188                         mntdir[0] = '\0'; /* avoid matching in next loop */
3189                 }
3190         } else {
3191                 /* flush fs as specified */
3192                 while (optind < argc) {
3193                         if (flushctx_ioctl(argv[optind++]))
3194                                 rc = -1;
3195                 }
3196         }
3197         return rc;
3198 }
3199
3200 static int lfs_lsetfacl(int argc, char **argv)
3201 {
3202         argv[0]++;
3203         return(llapi_lsetfacl(argc, argv));
3204 }
3205
3206 static int lfs_lgetfacl(int argc, char **argv)
3207 {
3208         argv[0]++;
3209         return(llapi_lgetfacl(argc, argv));
3210 }
3211
3212 static int lfs_rsetfacl(int argc, char **argv)
3213 {
3214         argv[0]++;
3215         return(llapi_rsetfacl(argc, argv));
3216 }
3217
3218 static int lfs_rgetfacl(int argc, char **argv)
3219 {
3220         argv[0]++;
3221         return(llapi_rgetfacl(argc, argv));
3222 }
3223
3224 static int lfs_cp(int argc, char **argv)
3225 {
3226         return(llapi_cp(argc, argv));
3227 }
3228
3229 static int lfs_ls(int argc, char **argv)
3230 {
3231         return(llapi_ls(argc, argv));
3232 }
3233
3234 static int lfs_changelog(int argc, char **argv)
3235 {
3236         void *changelog_priv;
3237         struct changelog_rec *rec;
3238         long long startrec = 0, endrec = 0;
3239         char *mdd;
3240         struct option long_opts[] = {
3241                 {"follow", no_argument, 0, 'f'},
3242                 {0, 0, 0, 0}
3243         };
3244         char short_opts[] = "f";
3245         int rc, follow = 0;
3246
3247         while ((rc = getopt_long(argc, argv, short_opts,
3248                                 long_opts, NULL)) != -1) {
3249                 switch (rc) {
3250                 case 'f':
3251                         follow++;
3252                         break;
3253                 case '?':
3254                         return CMD_HELP;
3255                 default:
3256                         fprintf(stderr, "error: %s: option '%s' unrecognized\n",
3257                                 argv[0], argv[optind - 1]);
3258                         return CMD_HELP;
3259                 }
3260         }
3261         if (optind >= argc)
3262                 return CMD_HELP;
3263
3264         mdd = argv[optind++];
3265         if (argc > optind)
3266                 startrec = strtoll(argv[optind++], NULL, 10);
3267         if (argc > optind)
3268                 endrec = strtoll(argv[optind++], NULL, 10);
3269
3270         rc = llapi_changelog_start(&changelog_priv,
3271                                    CHANGELOG_FLAG_BLOCK |
3272                                    CHANGELOG_FLAG_JOBID |
3273                                    (follow ? CHANGELOG_FLAG_FOLLOW : 0),
3274                                    mdd, startrec);
3275         if (rc < 0) {
3276                 fprintf(stderr, "Can't start changelog: %s\n",
3277                         strerror(errno = -rc));
3278                 return rc;
3279         }
3280
3281         while ((rc = llapi_changelog_recv(changelog_priv, &rec)) == 0) {
3282                 time_t secs;
3283                 struct tm ts;
3284
3285                 if (endrec && rec->cr_index > endrec) {
3286                         llapi_changelog_free(&rec);
3287                         break;
3288                 }
3289                 if (rec->cr_index < startrec) {
3290                         llapi_changelog_free(&rec);
3291                         continue;
3292                 }
3293
3294                 secs = rec->cr_time >> 30;
3295                 gmtime_r(&secs, &ts);
3296                 printf(LPU64" %02d%-5s %02d:%02d:%02d.%06d %04d.%02d.%02d "
3297                        "0x%x t="DFID, rec->cr_index, rec->cr_type,
3298                        changelog_type2str(rec->cr_type),
3299                        ts.tm_hour, ts.tm_min, ts.tm_sec,
3300                        (int)(rec->cr_time & ((1<<30) - 1)),
3301                        ts.tm_year + 1900, ts.tm_mon + 1, ts.tm_mday,
3302                        rec->cr_flags & CLF_FLAGMASK, PFID(&rec->cr_tfid));
3303
3304                 if (rec->cr_flags & CLF_JOBID) {
3305                         struct changelog_ext_jobid *jid =
3306                                 changelog_rec_jobid(rec);
3307
3308                         if (jid->cr_jobid[0] != '\0')
3309                                 printf(" j=%s", jid->cr_jobid);
3310                 }
3311
3312                 if (rec->cr_namelen)
3313                         printf(" p="DFID" %.*s", PFID(&rec->cr_pfid),
3314                                rec->cr_namelen, changelog_rec_name(rec));
3315
3316                 if (rec->cr_flags & CLF_RENAME) {
3317                         struct changelog_ext_rename *rnm =
3318                                 changelog_rec_rename(rec);
3319
3320                         if (!fid_is_zero(&rnm->cr_sfid))
3321                                 printf(" s="DFID" sp="DFID" %.*s",
3322                                        PFID(&rnm->cr_sfid),
3323                                        PFID(&rnm->cr_spfid),
3324                                        (int)changelog_rec_snamelen(rec),
3325                                        changelog_rec_sname(rec));
3326                 }
3327                 printf("\n");
3328
3329                 llapi_changelog_free(&rec);
3330         }
3331
3332         llapi_changelog_fini(&changelog_priv);
3333
3334         if (rc < 0)
3335                 fprintf(stderr, "Changelog: %s\n", strerror(errno = -rc));
3336
3337         return (rc == 1 ? 0 : rc);
3338 }
3339
3340 static int lfs_changelog_clear(int argc, char **argv)
3341 {
3342         long long endrec;
3343         int rc;
3344
3345         if (argc != 4)
3346                 return CMD_HELP;
3347
3348         endrec = strtoll(argv[3], NULL, 10);
3349
3350         rc = llapi_changelog_clear(argv[1], argv[2], endrec);
3351         if (rc)
3352                 fprintf(stderr, "%s error: %s\n", argv[0],
3353                         strerror(errno = -rc));
3354         return rc;
3355 }
3356
3357 static int lfs_fid2path(int argc, char **argv)
3358 {
3359         struct option long_opts[] = {
3360                 {"cur", no_argument, 0, 'c'},
3361                 {"link", required_argument, 0, 'l'},
3362                 {"rec", required_argument, 0, 'r'},
3363                 {0, 0, 0, 0}
3364         };
3365         char  short_opts[] = "cl:r:";
3366         char *device, *fid, *path;
3367         long long recno = -1;
3368         int linkno = -1;
3369         int lnktmp;
3370         int printcur = 0;
3371         int rc = 0;
3372
3373         while ((rc = getopt_long(argc, argv, short_opts,
3374                                 long_opts, NULL)) != -1) {
3375                 switch (rc) {
3376                 case 'c':
3377                         printcur++;
3378                         break;
3379                 case 'l':
3380                         linkno = strtol(optarg, NULL, 10);
3381                         break;
3382                 case 'r':
3383                         recno = strtoll(optarg, NULL, 10);
3384                         break;
3385                 case '?':
3386                         return CMD_HELP;
3387                 default:
3388                         fprintf(stderr, "error: %s: option '%s' unrecognized\n",
3389                                 argv[0], argv[optind - 1]);
3390                         return CMD_HELP;
3391                 }
3392         }
3393
3394         if (argc < 3)
3395                 return CMD_HELP;
3396
3397         device = argv[optind++];
3398         path = calloc(1, PATH_MAX);
3399         if (path == NULL) {
3400                 fprintf(stderr, "error: Not enough memory\n");
3401                 return -errno;
3402         }
3403
3404         rc = 0;
3405         while (optind < argc) {
3406                 fid = argv[optind++];
3407
3408                 lnktmp = (linkno >= 0) ? linkno : 0;
3409                 while (1) {
3410                         int oldtmp = lnktmp;
3411                         long long rectmp = recno;
3412                         int rc2;
3413                         rc2 = llapi_fid2path(device, fid, path, PATH_MAX,
3414                                              &rectmp, &lnktmp);
3415                         if (rc2 < 0) {
3416                                 fprintf(stderr, "%s: error on FID %s: %s\n",
3417                                         argv[0], fid, strerror(errno = -rc2));
3418                                 if (rc == 0)
3419                                         rc = rc2;
3420                                 break;
3421                         }
3422
3423                         if (printcur)
3424                                 fprintf(stdout, "%lld ", rectmp);
3425                         if (device[0] == '/') {
3426                                 fprintf(stdout, "%s", device);
3427                                 if (device[strlen(device) - 1] != '/')
3428                                         fprintf(stdout, "/");
3429                         } else if (path[0] == '\0') {
3430                                 fprintf(stdout, "/");
3431                         }
3432                         fprintf(stdout, "%s\n", path);
3433
3434                         if (linkno >= 0)
3435                                 /* specified linkno */
3436                                 break;
3437                         if (oldtmp == lnktmp)
3438                                 /* no more links */
3439                                 break;
3440                 }
3441         }
3442
3443         free(path);
3444         return rc;
3445 }
3446
3447 static int lfs_path2fid(int argc, char **argv)
3448 {
3449         struct option     long_opts[] = {
3450                 {"parents", no_argument, 0, 'p'},
3451                 {0, 0, 0, 0}
3452         };
3453         char            **path;
3454         const char        short_opts[] = "p";
3455         const char       *sep = "";
3456         lustre_fid        fid;
3457         int               rc = 0;
3458         bool              show_parents = false;
3459
3460         while ((rc = getopt_long(argc, argv, short_opts,
3461                                  long_opts, NULL)) != -1) {
3462                 switch (rc) {
3463                 case 'p':
3464                         show_parents = true;
3465                         break;
3466                 default:
3467                         fprintf(stderr, "error: %s: option '%s' unrecognized\n",
3468                                 argv[0], argv[optind - 1]);
3469                         return CMD_HELP;
3470                 }
3471         }
3472
3473         if (optind > argc - 1)
3474                 return CMD_HELP;
3475         else if (optind < argc - 1)
3476                 sep = ": ";
3477
3478         rc = 0;
3479         for (path = argv + optind; *path != NULL; path++) {
3480                 int err = 0;
3481                 if (!show_parents) {
3482                         err = llapi_path2fid(*path, &fid);
3483                         if (!err)
3484                                 printf("%s%s"DFID"\n",
3485                                        *sep != '\0' ? *path : "", sep,
3486                                        PFID(&fid));
3487                 } else {
3488                         char            name[NAME_MAX + 1];
3489                         unsigned int    linkno = 0;
3490
3491                         while ((err = llapi_path2parent(*path, linkno, &fid,
3492                                                 name, sizeof(name))) == 0) {
3493                                 if (*sep != '\0' && linkno == 0)
3494                                         printf("%s%s", *path, sep);
3495
3496                                 printf("%s"DFID"/%s", linkno != 0 ? "\t" : "",
3497                                        PFID(&fid), name);
3498                                 linkno++;
3499                         }
3500
3501                         /* err == -ENODATA is end-of-loop */
3502                         if (linkno > 0 && err == -ENODATA) {
3503                                 printf("\n");
3504                                 err = 0;
3505                         }
3506                 }
3507
3508                 if (err) {
3509                         fprintf(stderr, "%s: can't get %sfid for %s: %s\n",
3510                                 argv[0], show_parents ? "parent " : "", *path,
3511                                 strerror(-err));
3512                         if (rc == 0) {
3513                                 rc = err;
3514                                 errno = -err;
3515                         }
3516                 }
3517         }
3518
3519         return rc;
3520 }
3521
3522 static int lfs_data_version(int argc, char **argv)
3523 {
3524         char *path;
3525         __u64 data_version;
3526         int fd;
3527         int rc;
3528         int c;
3529         int data_version_flags = LL_DV_RD_FLUSH; /* Read by default */
3530
3531         if (argc < 2)
3532                 return CMD_HELP;
3533
3534         while ((c = getopt(argc, argv, "nrw")) != -1) {
3535                 switch (c) {
3536                 case 'n':
3537                         data_version_flags = 0;
3538                         break;
3539                 case 'r':
3540                         data_version_flags |= LL_DV_RD_FLUSH;
3541                         break;
3542                 case 'w':
3543                         data_version_flags |= LL_DV_WR_FLUSH;
3544                         break;
3545                 default:
3546                         return CMD_HELP;
3547                 }
3548         }
3549         if (optind == argc)
3550                 return CMD_HELP;
3551
3552         path = argv[optind];
3553         fd = open(path, O_RDONLY);
3554         if (fd < 0)
3555                 err(errno, "cannot open file %s", path);
3556
3557         rc = llapi_get_data_version(fd, &data_version, data_version_flags);
3558         if (rc < 0)
3559                 err(errno, "cannot get version for %s", path);
3560         else
3561                 printf(LPU64 "\n", data_version);
3562
3563         close(fd);
3564         return rc;
3565 }
3566
3567 static int lfs_hsm_state(int argc, char **argv)
3568 {
3569         int rc;
3570         int i = 1;
3571         char *path;
3572         struct hsm_user_state hus;
3573
3574         if (argc < 2)
3575                 return CMD_HELP;
3576
3577         do {
3578                 path = argv[i];
3579
3580                 rc = llapi_hsm_state_get(path, &hus);
3581                 if (rc) {
3582                         fprintf(stderr, "can't get hsm state for %s: %s\n",
3583                                 path, strerror(errno = -rc));
3584                         return rc;
3585                 }
3586
3587                 /* Display path name and status flags */
3588                 printf("%s: (0x%08x)", path, hus.hus_states);
3589
3590                 if (hus.hus_states & HS_RELEASED)
3591                         printf(" released");
3592                 if (hus.hus_states & HS_EXISTS)
3593                         printf(" exists");
3594                 if (hus.hus_states & HS_DIRTY)
3595                         printf(" dirty");
3596                 if (hus.hus_states & HS_ARCHIVED)
3597                         printf(" archived");
3598                 /* Display user-settable flags */
3599                 if (hus.hus_states & HS_NORELEASE)
3600                         printf(" never_release");
3601                 if (hus.hus_states & HS_NOARCHIVE)
3602                         printf(" never_archive");
3603                 if (hus.hus_states & HS_LOST)
3604                         printf(" lost_from_hsm");
3605
3606                 if (hus.hus_archive_id != 0)
3607                         printf(", archive_id:%d", hus.hus_archive_id);
3608                 printf("\n");
3609
3610         } while (++i < argc);
3611
3612         return 0;
3613 }
3614
3615 #define LFS_HSM_SET   0
3616 #define LFS_HSM_CLEAR 1
3617
3618 /**
3619  * Generic function to set or clear HSM flags.
3620  * Used by hsm_set and hsm_clear.
3621  *
3622  * @mode  if LFS_HSM_SET, set the flags, if LFS_HSM_CLEAR, clear the flags.
3623  */
3624 static int lfs_hsm_change_flags(int argc, char **argv, int mode)
3625 {
3626         struct option long_opts[] = {
3627                 {"lost", 0, 0, 'l'},
3628                 {"norelease", 0, 0, 'r'},
3629                 {"noarchive", 0, 0, 'a'},
3630                 {"archived", 0, 0, 'A'},
3631                 {"dirty", 0, 0, 'd'},
3632                 {"exists", 0, 0, 'e'},
3633                 {0, 0, 0, 0}
3634         };
3635         char short_opts[] = "lraAde";
3636         __u64 mask = 0;
3637         int c, rc;
3638         char *path;
3639
3640         if (argc < 3)
3641                 return CMD_HELP;
3642
3643         while ((c = getopt_long(argc, argv, short_opts,
3644                                 long_opts, NULL)) != -1) {
3645                 switch (c) {
3646                 case 'l':
3647                         mask |= HS_LOST;
3648                         break;
3649                 case 'a':
3650                         mask |= HS_NOARCHIVE;
3651                         break;
3652                 case 'A':
3653                         mask |= HS_ARCHIVED;
3654                         break;
3655                 case 'r':
3656                         mask |= HS_NORELEASE;
3657                         break;
3658                 case 'd':
3659                         mask |= HS_DIRTY;
3660                         break;
3661                 case 'e':
3662                         mask |= HS_EXISTS;
3663                         break;
3664                 case '?':
3665                         return CMD_HELP;
3666                 default:
3667                         fprintf(stderr, "error: %s: option '%s' unrecognized\n",
3668                                 argv[0], argv[optind - 1]);
3669                         return CMD_HELP;
3670                 }
3671         }
3672
3673         /* User should have specified a flag */
3674         if (mask == 0)
3675                 return CMD_HELP;
3676
3677         while (optind < argc) {
3678
3679                 path = argv[optind];
3680
3681                 /* If mode == 0, this means we apply the mask. */
3682                 if (mode == LFS_HSM_SET)
3683                         rc = llapi_hsm_state_set(path, mask, 0, 0);
3684                 else
3685                         rc = llapi_hsm_state_set(path, 0, mask, 0);
3686
3687                 if (rc != 0) {
3688                         fprintf(stderr, "Can't change hsm flags for %s: %s\n",
3689                                 path, strerror(errno = -rc));
3690                         return rc;
3691                 }
3692                 optind++;
3693         }
3694
3695         return 0;
3696 }
3697
3698 static int lfs_hsm_action(int argc, char **argv)
3699 {
3700         int                              rc;
3701         int                              i = 1;
3702         char                            *path;
3703         struct hsm_current_action        hca;
3704         struct hsm_extent                he;
3705         enum hsm_user_action             hua;
3706         enum hsm_progress_states         hps;
3707
3708         if (argc < 2)
3709                 return CMD_HELP;
3710
3711         do {
3712                 path = argv[i];
3713
3714                 rc = llapi_hsm_current_action(path, &hca);
3715                 if (rc) {
3716                         fprintf(stderr, "can't get hsm action for %s: %s\n",
3717                                 path, strerror(errno = -rc));
3718                         return rc;
3719                 }
3720                 he = hca.hca_location;
3721                 hua = hca.hca_action;
3722                 hps = hca.hca_state;
3723
3724                 printf("%s: %s", path, hsm_user_action2name(hua));
3725
3726                 /* Skip file without action */
3727                 if (hca.hca_action == HUA_NONE) {
3728                         printf("\n");
3729                         continue;
3730                 }
3731
3732                 printf(" %s ", hsm_progress_state2name(hps));
3733
3734                 if ((hps == HPS_RUNNING) &&
3735                     (hua == HUA_ARCHIVE || hua == HUA_RESTORE))
3736                         printf("("LPX64 " bytes moved)\n", he.length);
3737                 else if ((he.offset + he.length) == LUSTRE_EOF)
3738                         printf("(from "LPX64 " to EOF)\n", he.offset);
3739                 else
3740                         printf("(from "LPX64 " to "LPX64")\n",
3741                                he.offset, he.offset + he.length);
3742
3743         } while (++i < argc);
3744
3745         return 0;
3746 }
3747
3748 static int lfs_hsm_set(int argc, char **argv)
3749 {
3750         return lfs_hsm_change_flags(argc, argv, LFS_HSM_SET);
3751 }
3752
3753 static int lfs_hsm_clear(int argc, char **argv)
3754 {
3755         return lfs_hsm_change_flags(argc, argv, LFS_HSM_CLEAR);
3756 }
3757
3758 /**
3759  * Check file state and return its fid, to be used by lfs_hsm_request().
3760  *
3761  * \param[in]     file      Path to file to check
3762  * \param[in,out] fid       Pointer to allocated lu_fid struct.
3763  * \param[in,out] last_dev  Pointer to last device id used.
3764  *
3765  * \return 0 on success.
3766  */
3767 static int lfs_hsm_prepare_file(char *file, struct lu_fid *fid,
3768                                 dev_t *last_dev)
3769 {
3770         struct stat     st;
3771         int             rc;
3772
3773         rc = lstat(file, &st);
3774         if (rc) {
3775                 fprintf(stderr, "Cannot stat %s: %s\n", file, strerror(errno));
3776                 return -errno;
3777         }
3778         /* Checking for regular file as archiving as posix copytool
3779          * rejects archiving files other than regular files
3780          */
3781         if (!S_ISREG(st.st_mode)) {
3782                 fprintf(stderr, "error: \"%s\" is not a regular file\n", file);
3783                 return CMD_HELP;
3784         }
3785         /* A request should be ... */
3786         if (*last_dev != st.st_dev && *last_dev != 0) {
3787                 fprintf(stderr, "All files should be "
3788                         "on the same filesystem: %s\n", file);
3789                 return -EINVAL;
3790         }
3791         *last_dev = st.st_dev;
3792
3793         rc = llapi_path2fid(file, fid);
3794         if (rc) {
3795                 fprintf(stderr, "Cannot read FID of %s: %s\n",
3796                         file, strerror(-rc));
3797                 return rc;
3798         }
3799         return 0;
3800 }
3801
3802 static int lfs_hsm_request(int argc, char **argv, int action)
3803 {
3804         struct option            long_opts[] = {
3805                 {"filelist", 1, 0, 'l'},
3806                 {"data", 1, 0, 'D'},
3807                 {"archive", 1, 0, 'a'},
3808                 {0, 0, 0, 0}
3809         };
3810         dev_t                    last_dev = 0;
3811         char                     short_opts[] = "l:D:a:";
3812         struct hsm_user_request *hur, *oldhur;
3813         int                      c, i;
3814         size_t                   len;
3815         int                      nbfile;
3816         char                    *line = NULL;
3817         char                    *filelist = NULL;
3818         char                     fullpath[PATH_MAX];
3819         char                    *opaque = NULL;
3820         int                      opaque_len = 0;
3821         int                      archive_id = 0;
3822         FILE                    *fp;
3823         int                      nbfile_alloc = 0;
3824         char                     some_file[PATH_MAX+1] = "";
3825         int                      rc;
3826
3827         if (argc < 2)
3828                 return CMD_HELP;
3829
3830         while ((c = getopt_long(argc, argv, short_opts,
3831                                 long_opts, NULL)) != -1) {
3832                 switch (c) {
3833                 case 'l':
3834                         filelist = optarg;
3835                         break;
3836                 case 'D':
3837                         opaque = optarg;
3838                         break;
3839                 case 'a':
3840                         if (action != HUA_ARCHIVE) {
3841                                 fprintf(stderr,
3842                                         "error: -a is supported only "
3843                                         "when archiving\n");
3844                                 return CMD_HELP;
3845                         }
3846                         archive_id = atoi(optarg);
3847                         break;
3848                 case '?':
3849                         return CMD_HELP;
3850                 default:
3851                         fprintf(stderr, "error: %s: option '%s' unrecognized\n",
3852                                 argv[0], argv[optind - 1]);
3853                         return CMD_HELP;
3854                 }
3855         }
3856
3857         /* All remaining args are files, so we have at least nbfile */
3858         nbfile = argc - optind;
3859
3860         if ((nbfile == 0) && (filelist == NULL))
3861                 return CMD_HELP;
3862
3863         if (opaque != NULL)
3864                 opaque_len = strlen(opaque);
3865
3866         /* Alloc the request structure with enough place to store all files
3867          * from command line. */
3868         hur = llapi_hsm_user_request_alloc(nbfile, opaque_len);
3869         if (hur == NULL) {
3870                 fprintf(stderr, "Cannot create the request: %s\n",
3871                         strerror(errno));
3872                 return errno;
3873         }
3874         nbfile_alloc = nbfile;
3875
3876         hur->hur_request.hr_action = action;
3877         hur->hur_request.hr_archive_id = archive_id;
3878         hur->hur_request.hr_flags = 0;
3879
3880         /* All remaining args are files, add them */
3881         if (nbfile != 0) {
3882                 if (strlen(argv[optind]) > sizeof(some_file)-1) {
3883                         free(hur);
3884                         return -E2BIG;
3885                 }
3886                 strncpy(some_file, argv[optind], sizeof(some_file));
3887         }
3888
3889         for (i = 0; i < nbfile; i++) {
3890                 hur->hur_user_item[i].hui_extent.length = -1;
3891                 rc = lfs_hsm_prepare_file(argv[optind + i],
3892                                           &hur->hur_user_item[i].hui_fid,
3893                                           &last_dev);
3894                 hur->hur_request.hr_itemcount++;
3895                 if (rc)
3896                         goto out_free;
3897         }
3898
3899         /* from here stop using nb_file, use hur->hur_request.hr_itemcount */
3900
3901         /* If a filelist was specified, read the filelist from it. */
3902         if (filelist != NULL) {
3903                 fp = fopen(filelist, "r");
3904                 if (fp == NULL) {
3905                         fprintf(stderr, "Cannot read the file list %s: %s\n",
3906                                 filelist, strerror(errno));
3907                         rc = -errno;
3908                         goto out_free;
3909                 }
3910
3911                 while ((rc = getline(&line, &len, fp)) != -1) {
3912                         struct hsm_user_item *hui;
3913
3914                         /* If allocated buffer was too small, gets something
3915                          * bigger */
3916                         if (nbfile_alloc <= hur->hur_request.hr_itemcount) {
3917                                 ssize_t size;
3918                                 nbfile_alloc = nbfile_alloc * 2 + 1;
3919                                 oldhur = hur;
3920                                 hur = llapi_hsm_user_request_alloc(nbfile_alloc,
3921                                                                    opaque_len);
3922                                 if (hur == NULL) {
3923                                         fprintf(stderr, "hsm: cannot allocate "
3924                                                 "the request: %s\n",
3925                                                 strerror(errno));
3926                                         hur = oldhur;
3927                                         rc = -errno;
3928                                         fclose(fp);
3929                                         goto out_free;
3930                                 }
3931                                 size = hur_len(oldhur);
3932                                 if (size < 0) {
3933                                         fprintf(stderr, "hsm: cannot allocate "
3934                                                 "%u files + %u bytes data\n",
3935                                             oldhur->hur_request.hr_itemcount,
3936                                             oldhur->hur_request.hr_data_len);
3937                                         free(hur);
3938                                         hur = oldhur;
3939                                         rc = -E2BIG;
3940                                         fclose(fp);
3941                                         goto out_free;
3942                                 }
3943                                 memcpy(hur, oldhur, size);
3944                                 free(oldhur);
3945                         }
3946
3947                         /* Chop CR */
3948                         if (line[strlen(line) - 1] == '\n')
3949                                 line[strlen(line) - 1] = '\0';
3950
3951                         hui =
3952                              &hur->hur_user_item[hur->hur_request.hr_itemcount];
3953                         hui->hui_extent.length = -1;
3954                         rc = lfs_hsm_prepare_file(line, &hui->hui_fid,
3955                                                   &last_dev);
3956                         hur->hur_request.hr_itemcount++;
3957                         if (rc) {
3958                                 fclose(fp);
3959                                 goto out_free;
3960                         }
3961
3962                         if ((some_file[0] == '\0') &&
3963                             (strlen(line) < sizeof(some_file)))
3964                                 strcpy(some_file, line);
3965                 }
3966
3967                 rc = fclose(fp);
3968                 if (line)
3969                         free(line);
3970         }
3971
3972         /* If a --data was used, add it to the request */
3973         hur->hur_request.hr_data_len = opaque_len;
3974         if (opaque != NULL)
3975                 memcpy(hur_data(hur), opaque, opaque_len);
3976
3977         /* Send the HSM request */
3978         if (realpath(some_file, fullpath) == NULL) {
3979                 fprintf(stderr, "Could not find path '%s': %s\n",
3980                         some_file, strerror(errno));
3981         }
3982         rc = llapi_hsm_request(fullpath, hur);
3983         if (rc) {
3984                 fprintf(stderr, "Cannot send HSM request (use of %s): %s\n",
3985                         some_file, strerror(-rc));
3986                 goto out_free;
3987         }
3988
3989 out_free:
3990         free(hur);
3991         return rc;
3992 }
3993
3994 static int lfs_hsm_archive(int argc, char **argv)
3995 {
3996         return lfs_hsm_request(argc, argv, HUA_ARCHIVE);
3997 }
3998
3999 static int lfs_hsm_restore(int argc, char **argv)
4000 {
4001         return lfs_hsm_request(argc, argv, HUA_RESTORE);
4002 }
4003
4004 static int lfs_hsm_release(int argc, char **argv)
4005 {
4006         return lfs_hsm_request(argc, argv, HUA_RELEASE);
4007 }
4008
4009 static int lfs_hsm_remove(int argc, char **argv)
4010 {
4011         return lfs_hsm_request(argc, argv, HUA_REMOVE);
4012 }
4013
4014 static int lfs_hsm_cancel(int argc, char **argv)
4015 {
4016         return lfs_hsm_request(argc, argv, HUA_CANCEL);
4017 }
4018
4019 static int lfs_swap_layouts(int argc, char **argv)
4020 {
4021         if (argc != 3)
4022                 return CMD_HELP;
4023
4024         return llapi_swap_layouts(argv[1], argv[2], 0, 0,
4025                                   SWAP_LAYOUTS_KEEP_MTIME |
4026                                   SWAP_LAYOUTS_KEEP_ATIME);
4027 }
4028
4029 int main(int argc, char **argv)
4030 {
4031         int rc;
4032
4033         /* Ensure that liblustreapi constructor has run */
4034         if (!liblustreapi_initialized)
4035                 fprintf(stderr, "liblustreapi was not properly initialized\n");
4036
4037         setlinebuf(stdout);
4038
4039         Parser_init("lfs > ", cmdlist);
4040
4041         if (argc > 1) {
4042                 rc = Parser_execarg(argc - 1, argv + 1, cmdlist);
4043         } else {
4044                 rc = Parser_commands();
4045         }
4046
4047         return rc < 0 ? -rc : rc;
4048 }
4049
4050 #ifdef _LUSTRE_IDL_H_
4051 /* Everything we need here should be included by lustreapi.h. */
4052 # error "lfs should not depend on lustre_idl.h"
4053 #endif /* _LUSTRE_IDL_H_ */