Whamcloud - gitweb
4ab76a3bb7511624517af081d101e8e70e9e15cf
[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 indicated\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, "cannot migrate '%s' to MDT%04x: %s\n",
1831                         argv[optind], param.fp_mdt_index, strerror(-rc));
1832         return rc;
1833 }
1834
1835 static int lfs_osts(int argc, char **argv)
1836 {
1837         return lfs_tgts(argc, argv);
1838 }
1839
1840 static int lfs_mdts(int argc, char **argv)
1841 {
1842         return lfs_tgts(argc, argv);
1843 }
1844
1845 #define COOK(value)                                                     \
1846 ({                                                                      \
1847         int radix = 0;                                                  \
1848         while (value > 1024) {                                          \
1849                 value /= 1024;                                          \
1850                 radix++;                                                \
1851         }                                                               \
1852         radix;                                                          \
1853 })
1854 #define UUF     "%-20s"
1855 #define CSF     "%11s"
1856 #define CDF     "%11llu"
1857 #define HDF     "%8.1f%c"
1858 #define RSF     "%4s"
1859 #define RDF     "%3d%%"
1860
1861 static int showdf(char *mntdir, struct obd_statfs *stat,
1862                   char *uuid, int ishow, int cooked,
1863                   char *type, int index, int rc)
1864 {
1865         long long avail, used, total;
1866         double ratio = 0;
1867         char *suffix = "KMGTPEZY";
1868         /* Note if we have >2^64 bytes/fs these buffers will need to be grown */
1869         char tbuf[3 * sizeof(__u64)];
1870         char ubuf[3 * sizeof(__u64)];
1871         char abuf[3 * sizeof(__u64)];
1872         char rbuf[3 * sizeof(__u64)];
1873
1874         if (!uuid || !stat)
1875                 return -EINVAL;
1876
1877         switch (rc) {
1878         case 0:
1879                 if (ishow) {
1880                         avail = stat->os_ffree;
1881                         used = stat->os_files - stat->os_ffree;
1882                         total = stat->os_files;
1883                 } else {
1884                         int shift = cooked ? 0 : 10;
1885
1886                         avail = (stat->os_bavail * stat->os_bsize) >> shift;
1887                         used  = ((stat->os_blocks - stat->os_bfree) *
1888                                  stat->os_bsize) >> shift;
1889                         total = (stat->os_blocks * stat->os_bsize) >> shift;
1890                 }
1891
1892                 if ((used + avail) > 0)
1893                         ratio = (double)used / (double)(used + avail);
1894
1895                 if (cooked) {
1896                         int i;
1897                         double cook_val;
1898
1899                         cook_val = (double)total;
1900                         i = COOK(cook_val);
1901                         if (i > 0)
1902                                 sprintf(tbuf, HDF, cook_val, suffix[i - 1]);
1903                         else
1904                                 sprintf(tbuf, CDF, total);
1905
1906                         cook_val = (double)used;
1907                         i = COOK(cook_val);
1908                         if (i > 0)
1909                                 sprintf(ubuf, HDF, cook_val, suffix[i - 1]);
1910                         else
1911                                 sprintf(ubuf, CDF, used);
1912
1913                         cook_val = (double)avail;
1914                         i = COOK(cook_val);
1915                         if (i > 0)
1916                                 sprintf(abuf, HDF, cook_val, suffix[i - 1]);
1917                         else
1918                                 sprintf(abuf, CDF, avail);
1919                 } else {
1920                         sprintf(tbuf, CDF, total);
1921                         sprintf(ubuf, CDF, used);
1922                         sprintf(abuf, CDF, avail);
1923                 }
1924
1925                 sprintf(rbuf, RDF, (int)(ratio * 100 + 0.5));
1926                 printf(UUF" "CSF" "CSF" "CSF" "RSF" %-s",
1927                        uuid, tbuf, ubuf, abuf, rbuf, mntdir);
1928                 if (type)
1929                         printf("[%s:%d]\n", type, index);
1930                 else
1931                         printf("\n");
1932
1933                 break;
1934         case -ENODATA:
1935                 printf(UUF": inactive device\n", uuid);
1936                 break;
1937         default:
1938                 printf(UUF": %s\n", uuid, strerror(-rc));
1939                 break;
1940         }
1941
1942         return 0;
1943 }
1944
1945 struct ll_stat_type {
1946         int   st_op;
1947         char *st_name;
1948 };
1949
1950 static int mntdf(char *mntdir, char *fsname, char *pool, int ishow,
1951                 int cooked, int lazy)
1952 {
1953         struct obd_statfs stat_buf, sum = { .os_bsize = 1 };
1954         struct obd_uuid uuid_buf;
1955         char *poolname = NULL;
1956         struct ll_stat_type types[] = { { LL_STATFS_LMV, "MDT" },
1957                                         { LL_STATFS_LOV, "OST" },
1958                                         { 0, NULL } };
1959         struct ll_stat_type *tp;
1960         __u64 ost_ffree = 0;
1961         __u32 index;
1962         __u32 type;
1963         int rc;
1964
1965         if (pool) {
1966                 poolname = strchr(pool, '.');
1967                 if (poolname != NULL) {
1968                         if (strncmp(fsname, pool, strlen(fsname))) {
1969                                 fprintf(stderr, "filesystem name incorrect\n");
1970                                 return -ENODEV;
1971                         }
1972                         poolname++;
1973                 } else
1974                         poolname = pool;
1975         }
1976
1977         if (ishow)
1978                 printf(UUF" "CSF" "CSF" "CSF" "RSF" %-s\n",
1979                        "UUID", "Inodes", "IUsed", "IFree",
1980                        "IUse%", "Mounted on");
1981         else
1982                 printf(UUF" "CSF" "CSF" "CSF" "RSF" %-s\n",
1983                        "UUID", cooked ? "bytes" : "1K-blocks",
1984                        "Used", "Available", "Use%", "Mounted on");
1985
1986         for (tp = types; tp->st_name != NULL; tp++) {
1987                 for (index = 0; ; index++) {
1988                         memset(&stat_buf, 0, sizeof(struct obd_statfs));
1989                         memset(&uuid_buf, 0, sizeof(struct obd_uuid));
1990                         type = lazy ? tp->st_op | LL_STATFS_NODELAY : tp->st_op;
1991                         rc = llapi_obd_statfs(mntdir, type, index,
1992                                               &stat_buf, &uuid_buf);
1993                         if (rc == -ENODEV)
1994                                 break;
1995
1996                         if (rc == -EAGAIN)
1997                                 continue;
1998
1999                         if (poolname && tp->st_op == LL_STATFS_LOV &&
2000                             llapi_search_ost(fsname, poolname,
2001                                              obd_uuid2str(&uuid_buf)) != 1)
2002                                 continue;
2003
2004                         /* the llapi_obd_statfs() call may have returned with
2005                          * an error, but if it filled in uuid_buf we will at
2006                          * lease use that to print out a message for that OBD.
2007                          * If we didn't get anything in the uuid_buf, then fill
2008                          * it in so that we can print an error message. */
2009                         if (uuid_buf.uuid[0] == '\0')
2010                                 sprintf(uuid_buf.uuid, "%s%04x",
2011                                         tp->st_name, index);
2012                         showdf(mntdir, &stat_buf, obd_uuid2str(&uuid_buf),
2013                                ishow, cooked, tp->st_name, index, rc);
2014
2015                         if (rc == 0) {
2016                                 if (tp->st_op == LL_STATFS_LMV) {
2017                                         sum.os_ffree += stat_buf.os_ffree;
2018                                         sum.os_files += stat_buf.os_files;
2019                                 } else /* if (tp->st_op == LL_STATFS_LOV) */ {
2020                                         sum.os_blocks += stat_buf.os_blocks *
2021                                                 stat_buf.os_bsize;
2022                                         sum.os_bfree  += stat_buf.os_bfree *
2023                                                 stat_buf.os_bsize;
2024                                         sum.os_bavail += stat_buf.os_bavail *
2025                                                 stat_buf.os_bsize;
2026                                         ost_ffree += stat_buf.os_ffree;
2027                                 }
2028                         } else if (rc == -EINVAL || rc == -EFAULT) {
2029                                 break;
2030                         }
2031                 }
2032         }
2033
2034         /* If we don't have as many objects free on the OST as inodes
2035          * on the MDS, we reduce the total number of inodes to
2036          * compensate, so that the "inodes in use" number is correct.
2037          * Matches ll_statfs_internal() so the results are consistent. */
2038         if (ost_ffree < sum.os_ffree) {
2039                 sum.os_files = (sum.os_files - sum.os_ffree) + ost_ffree;
2040                 sum.os_ffree = ost_ffree;
2041         }
2042         printf("\n");
2043         showdf(mntdir, &sum, "filesystem summary:", ishow, cooked, NULL, 0, 0);
2044         printf("\n");
2045         return 0;
2046 }
2047
2048 static int lfs_df(int argc, char **argv)
2049 {
2050         char mntdir[PATH_MAX] = {'\0'}, path[PATH_MAX] = {'\0'};
2051         int ishow = 0, cooked = 0;
2052         int lazy = 0;
2053         int c, rc = 0, index = 0;
2054         char fsname[PATH_MAX] = "", *pool_name = NULL;
2055         struct option long_opts[] = {
2056                 {"pool", required_argument, 0, 'p'},
2057                 {"lazy", 0, 0, 'l'},
2058                 {0, 0, 0, 0}
2059         };
2060
2061         while ((c = getopt_long(argc, argv, "hilp:", long_opts, NULL)) != -1) {
2062                 switch (c) {
2063                 case 'i':
2064                         ishow = 1;
2065                         break;
2066                 case 'h':
2067                         cooked = 1;
2068                         break;
2069                 case 'l':
2070                         lazy = 1;
2071                         break;
2072                 case 'p':
2073                         pool_name = optarg;
2074                         break;
2075                 default:
2076                         return CMD_HELP;
2077                 }
2078         }
2079         if (optind < argc && !realpath(argv[optind], path)) {
2080                 rc = -errno;
2081                 fprintf(stderr, "error: invalid path '%s': %s\n",
2082                         argv[optind], strerror(-rc));
2083                 return rc;
2084         }
2085
2086         while (!llapi_search_mounts(path, index++, mntdir, fsname)) {
2087                 /* Check if we have a mount point */
2088                 if (mntdir[0] == '\0')
2089                         continue;
2090
2091                 rc = mntdf(mntdir, fsname, pool_name, ishow, cooked, lazy);
2092                 if (rc || path[0] != '\0')
2093                         break;
2094                 fsname[0] = '\0'; /* avoid matching in next loop */
2095                 mntdir[0] = '\0'; /* avoid matching in next loop */
2096         }
2097
2098         return rc;
2099 }
2100
2101 static int lfs_getname(int argc, char **argv)
2102 {
2103         char mntdir[PATH_MAX] = "", path[PATH_MAX] = "", fsname[PATH_MAX] = "";
2104         int rc = 0, index = 0, c;
2105         char buf[sizeof(struct obd_uuid)];
2106
2107         while ((c = getopt(argc, argv, "h")) != -1)
2108                 return CMD_HELP;
2109
2110         if (optind == argc) { /* no paths specified, get all paths. */
2111                 while (!llapi_search_mounts(path, index++, mntdir, fsname)) {
2112                         rc = llapi_getname(mntdir, buf, sizeof(buf));
2113                         if (rc < 0) {
2114                                 fprintf(stderr,
2115                                         "cannot get name for `%s': %s\n",
2116                                         mntdir, strerror(-rc));
2117                                 break;
2118                         }
2119
2120                         printf("%s %s\n", buf, mntdir);
2121
2122                         path[0] = fsname[0] = mntdir[0] = 0;
2123                 }
2124         } else { /* paths specified, only attempt to search these. */
2125                 for (; optind < argc; optind++) {
2126                         rc = llapi_getname(argv[optind], buf, sizeof(buf));
2127                         if (rc < 0) {
2128                                 fprintf(stderr,
2129                                         "cannot get name for `%s': %s\n",
2130                                         argv[optind], strerror(-rc));
2131                                 break;
2132                         }
2133
2134                         printf("%s %s\n", buf, argv[optind]);
2135                 }
2136         }
2137         return rc;
2138 }
2139
2140 static int lfs_check(int argc, char **argv)
2141 {
2142         int rc;
2143         char mntdir[PATH_MAX] = {'\0'};
2144         int num_types = 1;
2145         char *obd_types[2];
2146         char obd_type1[4];
2147         char obd_type2[4];
2148
2149         if (argc != 2)
2150                 return CMD_HELP;
2151
2152         obd_types[0] = obd_type1;
2153         obd_types[1] = obd_type2;
2154
2155         if (strcmp(argv[1], "osts") == 0) {
2156                 strcpy(obd_types[0], "osc");
2157         } else if (strcmp(argv[1], "mds") == 0) {
2158                 strcpy(obd_types[0], "mdc");
2159         } else if (strcmp(argv[1], "servers") == 0) {
2160                 num_types = 2;
2161                 strcpy(obd_types[0], "osc");
2162                 strcpy(obd_types[1], "mdc");
2163         } else {
2164                 fprintf(stderr, "error: %s: option '%s' unrecognized\n",
2165                                 argv[0], argv[1]);
2166                         return CMD_HELP;
2167         }
2168
2169         rc = llapi_search_mounts(NULL, 0, mntdir, NULL);
2170         if (rc < 0 || mntdir[0] == '\0') {
2171                 fprintf(stderr, "No suitable Lustre mount found\n");
2172                 return rc;
2173         }
2174
2175         rc = llapi_target_check(num_types, obd_types, mntdir);
2176         if (rc)
2177                 fprintf(stderr, "error: %s: %s status failed\n",
2178                                 argv[0],argv[1]);
2179
2180         return rc;
2181
2182 }
2183
2184 static int lfs_join(int argc, char **argv)
2185 {
2186         fprintf(stderr, "join two lustre files into one.\n"
2187                         "obsolete, HEAD does not support it anymore.\n");
2188         return 0;
2189 }
2190
2191 #ifdef HAVE_SYS_QUOTA_H
2192 static int lfs_quotacheck(int argc, char **argv)
2193 {
2194         int c, check_type = 0;
2195         char *mnt;
2196         struct if_quotacheck qchk;
2197         struct if_quotactl qctl;
2198         char *obd_type = (char *)qchk.obd_type;
2199         int rc;
2200
2201         memset(&qchk, 0, sizeof(qchk));
2202
2203         while ((c = getopt(argc, argv, "gu")) != -1) {
2204                 switch (c) {
2205                 case 'u':
2206                         check_type |= 0x01;
2207                         break;
2208                 case 'g':
2209                         check_type |= 0x02;
2210                         break;
2211                 default:
2212                         fprintf(stderr, "error: %s: option '-%c' "
2213                                         "unrecognized\n", argv[0], c);
2214                         return CMD_HELP;
2215                 }
2216         }
2217
2218         if (check_type)
2219                 check_type--;
2220         else    /* do quotacheck for both user & group quota by default */
2221                 check_type = 0x02;
2222
2223         if (argc == optind)
2224                 return CMD_HELP;
2225
2226         mnt = argv[optind];
2227
2228         rc = llapi_quotacheck(mnt, check_type);
2229         if (rc == -EOPNOTSUPP) {
2230                 fprintf(stderr, "error: quotacheck not supported by the quota "
2231                         "master.\nPlease note that quotacheck is deprecated as "
2232                         "of lustre 2.4.0 since space accounting is always "
2233                         "enabled.\nFilesystems not formatted with 2.4 utils or "
2234                         "beyond can be upgraded with tunefs.lustre --quota.\n");
2235                 return rc;
2236         } else if (rc) {
2237                 fprintf(stderr, "quotacheck failed: %s\n", strerror(-rc));
2238                 return rc;
2239         }
2240
2241         rc = llapi_poll_quotacheck(mnt, &qchk);
2242         if (rc) {
2243                 if (*obd_type)
2244                         fprintf(stderr, "%s %s ", obd_type,
2245                                 obd_uuid2str(&qchk.obd_uuid));
2246                 fprintf(stderr, "quota check failed: %s\n", strerror(-rc));
2247                 return rc;
2248         }
2249
2250         memset(&qctl, 0, sizeof(qctl));
2251         qctl.qc_cmd = LUSTRE_Q_QUOTAON;
2252         qctl.qc_type = check_type;
2253         rc = llapi_quotactl(mnt, &qctl);
2254         if (rc && rc != -EALREADY) {
2255                 if (*obd_type)
2256                         fprintf(stderr, "%s %s ", (char *)qctl.obd_type,
2257                                 obd_uuid2str(&qctl.obd_uuid));
2258                 fprintf(stderr, "%s turn on quota failed: %s\n",
2259                         argv[0], strerror(-rc));
2260                 return rc;
2261         }
2262
2263         return 0;
2264 }
2265
2266 static int lfs_quotaon(int argc, char **argv)
2267 {
2268         int c;
2269         char *mnt;
2270         struct if_quotactl qctl;
2271         char *obd_type = (char *)qctl.obd_type;
2272         int rc;
2273
2274         memset(&qctl, 0, sizeof(qctl));
2275         qctl.qc_cmd = LUSTRE_Q_QUOTAON;
2276
2277         while ((c = getopt(argc, argv, "fgu")) != -1) {
2278                 switch (c) {
2279                 case 'u':
2280                         qctl.qc_type |= 0x01;
2281                         break;
2282                 case 'g':
2283                         qctl.qc_type |= 0x02;
2284                         break;
2285                 case 'f':
2286                         qctl.qc_cmd = LUSTRE_Q_QUOTAOFF;
2287                         break;
2288                 default:
2289                         fprintf(stderr, "error: %s: option '-%c' "
2290                                         "unrecognized\n", argv[0], c);
2291                         return CMD_HELP;
2292                 }
2293         }
2294
2295         if (qctl.qc_type)
2296                 qctl.qc_type--;
2297         else /* by default, enable quota for both user & group */
2298                 qctl.qc_type = 0x02;
2299
2300         if (argc == optind)
2301                 return CMD_HELP;
2302
2303         mnt = argv[optind];
2304
2305         rc = llapi_quotactl(mnt, &qctl);
2306         if (rc) {
2307                 if (rc == -EOPNOTSUPP) {
2308                         fprintf(stderr, "error: quotaon not supported by the "
2309                                 "quota master.\nPlease note that quotaon/off is"
2310                                 " deprecated as of lustre 2.4.0.\nQuota "
2311                                 "enforcement should now be enabled on the MGS "
2312                                 "via:\nmgs# lctl conf_param ${FSNAME}.quota."
2313                                 "<ost|mdt>=<u|g|ug>\n(ost for block quota, mdt "
2314                                 "for inode quota, u for user and g for group"
2315                                 "\n");
2316                 } else if (rc == -EALREADY) {
2317                         rc = 0;
2318                 } else if (rc == -ENOENT) {
2319                         fprintf(stderr, "error: cannot find quota database, "
2320                                         "make sure you have run quotacheck\n");
2321                 } else {
2322                         if (*obd_type)
2323                                 fprintf(stderr, "%s %s ", obd_type,
2324                                         obd_uuid2str(&qctl.obd_uuid));
2325                         fprintf(stderr, "%s failed: %s\n", argv[0],
2326                                 strerror(-rc));
2327                 }
2328         }
2329
2330         return rc;
2331 }
2332
2333 static int lfs_quotaoff(int argc, char **argv)
2334 {
2335         int c;
2336         char *mnt;
2337         struct if_quotactl qctl;
2338         char *obd_type = (char *)qctl.obd_type;
2339         int rc;
2340
2341         memset(&qctl, 0, sizeof(qctl));
2342         qctl.qc_cmd = LUSTRE_Q_QUOTAOFF;
2343
2344         while ((c = getopt(argc, argv, "gu")) != -1) {
2345                 switch (c) {
2346                 case 'u':
2347                         qctl.qc_type |= 0x01;
2348                         break;
2349                 case 'g':
2350                         qctl.qc_type |= 0x02;
2351                         break;
2352                 default:
2353                         fprintf(stderr, "error: %s: option '-%c' "
2354                                         "unrecognized\n", argv[0], c);
2355                         return CMD_HELP;
2356                 }
2357         }
2358
2359         if (qctl.qc_type)
2360                 qctl.qc_type--;
2361         else /* by default, disable quota for both user & group */
2362                 qctl.qc_type = 0x02;
2363
2364         if (argc == optind)
2365                 return CMD_HELP;
2366
2367         mnt = argv[optind];
2368
2369         rc = llapi_quotactl(mnt, &qctl);
2370         if (rc) {
2371                 if (rc == -EOPNOTSUPP) {
2372                         fprintf(stderr, "error: quotaoff not supported by the "
2373                                 "quota master.\nPlease note that quotaon/off is"
2374                                 " deprecated as of lustre 2.4.0.\nQuota "
2375                                 "enforcement can be disabled on the MGS via:\n"
2376                                 "mgs# lctl conf_param ${FSNAME}.quota.<ost|mdt>"
2377                                 "=\"\"\n");
2378                 } else if (rc == -EALREADY) {
2379                         rc = 0;
2380                 } else {
2381                         if (*obd_type)
2382                                 fprintf(stderr, "%s %s ", obd_type,
2383                                         obd_uuid2str(&qctl.obd_uuid));
2384                         fprintf(stderr, "quotaoff failed: %s\n",
2385                                 strerror(-rc));
2386                 }
2387         }
2388
2389         return rc;
2390 }
2391
2392 #define ARG2INT(nr, str, msg)                                           \
2393 do {                                                                    \
2394         char *endp;                                                     \
2395         nr = strtol(str, &endp, 0);                                     \
2396         if (*endp) {                                                    \
2397                 fprintf(stderr, "error: bad %s: %s\n", msg, str);       \
2398                 return CMD_HELP;                                        \
2399         }                                                               \
2400 } while (0)
2401
2402 #define ADD_OVERFLOW(a,b) ((a + b) < a) ? (a = ULONG_MAX) : (a = a + b)
2403
2404 /* Convert format time string "XXwXXdXXhXXmXXs" into seconds value
2405  * returns the value or ULONG_MAX on integer overflow or incorrect format
2406  * Notes:
2407  *        1. the order of specifiers is arbitrary (may be: 5w3s or 3s5w)
2408  *        2. specifiers may be encountered multiple times (2s3s is 5 seconds)
2409  *        3. empty integer value is interpreted as 0
2410  */
2411 static unsigned long str2sec(const char* timestr)
2412 {
2413         const char spec[] = "smhdw";
2414         const unsigned long mult[] = {1, 60, 60*60, 24*60*60, 7*24*60*60};
2415         unsigned long val = 0;
2416         char *tail;
2417
2418         if (strpbrk(timestr, spec) == NULL) {
2419                 /* no specifiers inside the time string,
2420                    should treat it as an integer value */
2421                 val = strtoul(timestr, &tail, 10);
2422                 return *tail ? ULONG_MAX : val;
2423         }
2424
2425         /* format string is XXwXXdXXhXXmXXs */
2426         while (*timestr) {
2427                 unsigned long v;
2428                 int ind;
2429                 char* ptr;
2430
2431                 v = strtoul(timestr, &tail, 10);
2432                 if (v == ULONG_MAX || *tail == '\0')
2433                         /* value too large (ULONG_MAX or more)
2434                            or missing specifier */
2435                         goto error;
2436
2437                 ptr = strchr(spec, *tail);
2438                 if (ptr == NULL)
2439                         /* unknown specifier */
2440                         goto error;
2441
2442                 ind = ptr - spec;
2443
2444                 /* check if product will overflow the type */
2445                 if (!(v < ULONG_MAX / mult[ind]))
2446                         goto error;
2447
2448                 ADD_OVERFLOW(val, mult[ind] * v);
2449                 if (val == ULONG_MAX)
2450                         goto error;
2451
2452                 timestr = tail + 1;
2453         }
2454
2455         return val;
2456
2457 error:
2458         return ULONG_MAX;
2459 }
2460
2461 #define ARG2ULL(nr, str, def_units)                                     \
2462 do {                                                                    \
2463         unsigned long long limit, units = def_units;                    \
2464         int rc;                                                         \
2465                                                                         \
2466         rc = llapi_parse_size(str, &limit, &units, 1);                  \
2467         if (rc < 0) {                                                   \
2468                 fprintf(stderr, "error: bad limit value %s\n", str);    \
2469                 return CMD_HELP;                                        \
2470         }                                                               \
2471         nr = limit;                                                     \
2472 } while (0)
2473
2474 static inline int has_times_option(int argc, char **argv)
2475 {
2476         int i;
2477
2478         for (i = 1; i < argc; i++)
2479                 if (!strcmp(argv[i], "-t"))
2480                         return 1;
2481
2482         return 0;
2483 }
2484
2485 int lfs_setquota_times(int argc, char **argv)
2486 {
2487         int c, rc;
2488         struct if_quotactl qctl;
2489         char *mnt, *obd_type = (char *)qctl.obd_type;
2490         struct obd_dqblk *dqb = &qctl.qc_dqblk;
2491         struct obd_dqinfo *dqi = &qctl.qc_dqinfo;
2492         struct option long_opts[] = {
2493                 {"block-grace",     required_argument, 0, 'b'},
2494                 {"group",           no_argument,       0, 'g'},
2495                 {"inode-grace",     required_argument, 0, 'i'},
2496                 {"times",           no_argument,       0, 't'},
2497                 {"user",            no_argument,       0, 'u'},
2498                 {0, 0, 0, 0}
2499         };
2500
2501         memset(&qctl, 0, sizeof(qctl));
2502         qctl.qc_cmd  = LUSTRE_Q_SETINFO;
2503         qctl.qc_type = UGQUOTA;
2504
2505         while ((c = getopt_long(argc, argv, "b:gi:tu", long_opts, NULL)) != -1) {
2506                 switch (c) {
2507                 case 'u':
2508                 case 'g':
2509                         if (qctl.qc_type != UGQUOTA) {
2510                                 fprintf(stderr, "error: -u and -g can't be used "
2511                                                 "more than once\n");
2512                                 return CMD_HELP;
2513                         }
2514                         qctl.qc_type = (c == 'u') ? USRQUOTA : GRPQUOTA;
2515                         break;
2516                 case 'b':
2517                         if ((dqi->dqi_bgrace = str2sec(optarg)) == ULONG_MAX) {
2518                                 fprintf(stderr, "error: bad block-grace: %s\n",
2519                                         optarg);
2520                                 return CMD_HELP;
2521                         }
2522                         dqb->dqb_valid |= QIF_BTIME;
2523                         break;
2524                 case 'i':
2525                         if ((dqi->dqi_igrace = str2sec(optarg)) == ULONG_MAX) {
2526                                 fprintf(stderr, "error: bad inode-grace: %s\n",
2527                                         optarg);
2528                                 return CMD_HELP;
2529                         }
2530                         dqb->dqb_valid |= QIF_ITIME;
2531                         break;
2532                 case 't': /* Yes, of course! */
2533                         break;
2534                 default: /* getopt prints error message for us when opterr != 0 */
2535                         return CMD_HELP;
2536                 }
2537         }
2538
2539         if (qctl.qc_type == UGQUOTA) {
2540                 fprintf(stderr, "error: neither -u nor -g specified\n");
2541                 return CMD_HELP;
2542         }
2543
2544         if (optind != argc - 1) {
2545                 fprintf(stderr, "error: unexpected parameters encountered\n");
2546                 return CMD_HELP;
2547         }
2548
2549         mnt = argv[optind];
2550         rc = llapi_quotactl(mnt, &qctl);
2551         if (rc) {
2552                 if (*obd_type)
2553                         fprintf(stderr, "%s %s ", obd_type,
2554                                 obd_uuid2str(&qctl.obd_uuid));
2555                 fprintf(stderr, "setquota failed: %s\n", strerror(-rc));
2556                 return rc;
2557         }
2558
2559         return 0;
2560 }
2561
2562 #define BSLIMIT (1 << 0)
2563 #define BHLIMIT (1 << 1)
2564 #define ISLIMIT (1 << 2)
2565 #define IHLIMIT (1 << 3)
2566
2567 int lfs_setquota(int argc, char **argv)
2568 {
2569         int c, rc;
2570         struct if_quotactl qctl;
2571         char *mnt, *obd_type = (char *)qctl.obd_type;
2572         struct obd_dqblk *dqb = &qctl.qc_dqblk;
2573         struct option long_opts[] = {
2574                 {"block-softlimit", required_argument, 0, 'b'},
2575                 {"block-hardlimit", required_argument, 0, 'B'},
2576                 {"group",           required_argument, 0, 'g'},
2577                 {"inode-softlimit", required_argument, 0, 'i'},
2578                 {"inode-hardlimit", required_argument, 0, 'I'},
2579                 {"user",            required_argument, 0, 'u'},
2580                 {0, 0, 0, 0}
2581         };
2582         unsigned limit_mask = 0;
2583         char *endptr;
2584
2585         if (has_times_option(argc, argv))
2586                 return lfs_setquota_times(argc, argv);
2587
2588         memset(&qctl, 0, sizeof(qctl));
2589         qctl.qc_cmd  = LUSTRE_Q_SETQUOTA;
2590         qctl.qc_type = UGQUOTA; /* UGQUOTA makes no sense for setquota,
2591                                  * so it can be used as a marker that qc_type
2592                                  * isn't reinitialized from command line */
2593
2594         while ((c = getopt_long(argc, argv, "b:B:g:i:I:u:", long_opts, NULL)) != -1) {
2595                 switch (c) {
2596                 case 'u':
2597                 case 'g':
2598                         if (qctl.qc_type != UGQUOTA) {
2599                                 fprintf(stderr, "error: -u and -g can't be used"
2600                                                 " more than once\n");
2601                                 return CMD_HELP;
2602                         }
2603                         qctl.qc_type = (c == 'u') ? USRQUOTA : GRPQUOTA;
2604                         rc = name2id(&qctl.qc_id, optarg,
2605                                      (qctl.qc_type == USRQUOTA) ? USER : GROUP);
2606                         if (rc) {
2607                                 qctl.qc_id = strtoul(optarg, &endptr, 10);
2608                                 if (*endptr != '\0') {
2609                                         fprintf(stderr, "error: can't find id "
2610                                                 "for name %s\n", optarg);
2611                                         return CMD_HELP;
2612                                 }
2613                         }
2614                         break;
2615                 case 'b':
2616                         ARG2ULL(dqb->dqb_bsoftlimit, optarg, 1024);
2617                         dqb->dqb_bsoftlimit >>= 10;
2618                         limit_mask |= BSLIMIT;
2619                         if (dqb->dqb_bsoftlimit &&
2620                             dqb->dqb_bsoftlimit <= 1024) /* <= 1M? */
2621                                 fprintf(stderr, "warning: block softlimit is "
2622                                         "smaller than the miminal qunit size, "
2623                                         "please see the help of setquota or "
2624                                         "Lustre manual for details.\n");
2625                         break;
2626                 case 'B':
2627                         ARG2ULL(dqb->dqb_bhardlimit, optarg, 1024);
2628                         dqb->dqb_bhardlimit >>= 10;
2629                         limit_mask |= BHLIMIT;
2630                         if (dqb->dqb_bhardlimit &&
2631                             dqb->dqb_bhardlimit <= 1024) /* <= 1M? */
2632                                 fprintf(stderr, "warning: block hardlimit is "
2633                                         "smaller than the miminal qunit size, "
2634                                         "please see the help of setquota or "
2635                                         "Lustre manual for details.\n");
2636                         break;
2637                 case 'i':
2638                         ARG2ULL(dqb->dqb_isoftlimit, optarg, 1);
2639                         limit_mask |= ISLIMIT;
2640                         if (dqb->dqb_isoftlimit &&
2641                             dqb->dqb_isoftlimit <= 1024) /* <= 1K inodes? */
2642                                 fprintf(stderr, "warning: inode softlimit is "
2643                                         "smaller than the miminal qunit size, "
2644                                         "please see the help of setquota or "
2645                                         "Lustre manual for details.\n");
2646                         break;
2647                 case 'I':
2648                         ARG2ULL(dqb->dqb_ihardlimit, optarg, 1);
2649                         limit_mask |= IHLIMIT;
2650                         if (dqb->dqb_ihardlimit &&
2651                             dqb->dqb_ihardlimit <= 1024) /* <= 1K inodes? */
2652                                 fprintf(stderr, "warning: inode hardlimit is "
2653                                         "smaller than the miminal qunit size, "
2654                                         "please see the help of setquota or "
2655                                         "Lustre manual for details.\n");
2656                         break;
2657                 default: /* getopt prints error message for us when opterr != 0 */
2658                         return CMD_HELP;
2659                 }
2660         }
2661
2662         if (qctl.qc_type == UGQUOTA) {
2663                 fprintf(stderr, "error: neither -u nor -g was specified\n");
2664                 return CMD_HELP;
2665         }
2666
2667         if (limit_mask == 0) {
2668                 fprintf(stderr, "error: at least one limit must be specified\n");
2669                 return CMD_HELP;
2670         }
2671
2672         if (optind != argc - 1) {
2673                 fprintf(stderr, "error: unexpected parameters encountered\n");
2674                 return CMD_HELP;
2675         }
2676
2677         mnt = argv[optind];
2678
2679         if ((!(limit_mask & BHLIMIT) ^ !(limit_mask & BSLIMIT)) ||
2680             (!(limit_mask & IHLIMIT) ^ !(limit_mask & ISLIMIT))) {
2681                 /* sigh, we can't just set blimits/ilimits */
2682                 struct if_quotactl tmp_qctl = {.qc_cmd  = LUSTRE_Q_GETQUOTA,
2683                                                .qc_type = qctl.qc_type,
2684                                                .qc_id   = qctl.qc_id};
2685
2686                 rc = llapi_quotactl(mnt, &tmp_qctl);
2687                 if (rc < 0) {
2688                         fprintf(stderr, "error: setquota failed while retrieving"
2689                                         " current quota settings (%s)\n",
2690                                         strerror(-rc));
2691                         return rc;
2692                 }
2693
2694                 if (!(limit_mask & BHLIMIT))
2695                         dqb->dqb_bhardlimit = tmp_qctl.qc_dqblk.dqb_bhardlimit;
2696                 if (!(limit_mask & BSLIMIT))
2697                         dqb->dqb_bsoftlimit = tmp_qctl.qc_dqblk.dqb_bsoftlimit;
2698                 if (!(limit_mask & IHLIMIT))
2699                         dqb->dqb_ihardlimit = tmp_qctl.qc_dqblk.dqb_ihardlimit;
2700                 if (!(limit_mask & ISLIMIT))
2701                         dqb->dqb_isoftlimit = tmp_qctl.qc_dqblk.dqb_isoftlimit;
2702
2703                 /* Keep grace times if we have got no softlimit arguments */
2704                 if ((limit_mask & BHLIMIT) && !(limit_mask & BSLIMIT)) {
2705                         dqb->dqb_valid |= QIF_BTIME;
2706                         dqb->dqb_btime = tmp_qctl.qc_dqblk.dqb_btime;
2707                 }
2708
2709                 if ((limit_mask & IHLIMIT) && !(limit_mask & ISLIMIT)) {
2710                         dqb->dqb_valid |= QIF_ITIME;
2711                         dqb->dqb_itime = tmp_qctl.qc_dqblk.dqb_itime;
2712                 }
2713         }
2714
2715         dqb->dqb_valid |= (limit_mask & (BHLIMIT | BSLIMIT)) ? QIF_BLIMITS : 0;
2716         dqb->dqb_valid |= (limit_mask & (IHLIMIT | ISLIMIT)) ? QIF_ILIMITS : 0;
2717
2718         rc = llapi_quotactl(mnt, &qctl);
2719         if (rc) {
2720                 if (*obd_type)
2721                         fprintf(stderr, "%s %s ", obd_type,
2722                                 obd_uuid2str(&qctl.obd_uuid));
2723                 fprintf(stderr, "setquota failed: %s\n", strerror(-rc));
2724                 return rc;
2725         }
2726
2727         return 0;
2728 }
2729
2730 static inline char *type2name(int check_type)
2731 {
2732         if (check_type == USRQUOTA)
2733                 return "user";
2734         else if (check_type == GRPQUOTA)
2735                 return "group";
2736         else
2737                 return "unknown";
2738 }
2739
2740 /* Converts seconds value into format string
2741  * result is returned in buf
2742  * Notes:
2743  *        1. result is in descenting order: 1w2d3h4m5s
2744  *        2. zero fields are not filled (except for p. 3): 5d1s
2745  *        3. zero seconds value is presented as "0s"
2746  */
2747 static char * __sec2str(time_t seconds, char *buf)
2748 {
2749         const char spec[] = "smhdw";
2750         const unsigned long mult[] = {1, 60, 60*60, 24*60*60, 7*24*60*60};
2751         unsigned long c;
2752         char *tail = buf;
2753         int i;
2754
2755         for (i = sizeof(mult) / sizeof(mult[0]) - 1 ; i >= 0; i--) {
2756                 c = seconds / mult[i];
2757
2758                 if (c > 0 || (i == 0 && buf == tail))
2759                         tail += snprintf(tail, 40-(tail-buf), "%lu%c", c, spec[i]);
2760
2761                 seconds %= mult[i];
2762         }
2763
2764         return tail;
2765 }
2766
2767 static void sec2str(time_t seconds, char *buf, int rc)
2768 {
2769         char *tail = buf;
2770
2771         if (rc)
2772                 *tail++ = '[';
2773
2774         tail = __sec2str(seconds, tail);
2775
2776         if (rc && tail - buf < 39) {
2777                 *tail++ = ']';
2778                 *tail++ = 0;
2779         }
2780 }
2781
2782 static void diff2str(time_t seconds, char *buf, time_t now)
2783 {
2784
2785         buf[0] = 0;
2786         if (!seconds)
2787                 return;
2788         if (seconds <= now) {
2789                 strcpy(buf, "none");
2790                 return;
2791         }
2792         __sec2str(seconds - now, buf);
2793 }
2794
2795 static void print_quota_title(char *name, struct if_quotactl *qctl,
2796                               bool human_readable)
2797 {
2798         printf("Disk quotas for %s %s (%cid %u):\n",
2799                type2name(qctl->qc_type), name,
2800                *type2name(qctl->qc_type), qctl->qc_id);
2801         printf("%15s%8s %7s%8s%8s%8s %7s%8s%8s\n",
2802                "Filesystem", human_readable ? "used" : "kbytes",
2803                "quota", "limit", "grace",
2804                "files", "quota", "limit", "grace");
2805 }
2806
2807 static void kbytes2str(__u64 num, char *buf, bool h)
2808 {
2809         if (!h) {
2810                 sprintf(buf, LPU64, num);
2811         } else {
2812                 if (num >> 30)
2813                         sprintf(buf, "%5.4gT", (double)num / (1 << 30));
2814                 else if (num >> 20)
2815                         sprintf(buf, "%5.4gG", (double)num / (1 << 20));
2816                 else if (num >> 10)
2817                         sprintf(buf, "%5.4gM", (double)num / (1 << 10));
2818                 else
2819                         sprintf(buf, LPU64"%s", num, "k");
2820         }
2821 }
2822
2823 static void print_quota(char *mnt, struct if_quotactl *qctl, int type,
2824                         int rc, bool h)
2825 {
2826         time_t now;
2827
2828         time(&now);
2829
2830         if (qctl->qc_cmd == LUSTRE_Q_GETQUOTA || qctl->qc_cmd == Q_GETOQUOTA) {
2831                 int bover = 0, iover = 0;
2832                 struct obd_dqblk *dqb = &qctl->qc_dqblk;
2833                 char numbuf[3][32];
2834                 char timebuf[40];
2835                 char strbuf[32];
2836
2837                 if (dqb->dqb_bhardlimit &&
2838                     lustre_stoqb(dqb->dqb_curspace) >= dqb->dqb_bhardlimit) {
2839                         bover = 1;
2840                 } else if (dqb->dqb_bsoftlimit && dqb->dqb_btime) {
2841                         if (dqb->dqb_btime > now) {
2842                                 bover = 2;
2843                         } else {
2844                                 bover = 3;
2845                         }
2846                 }
2847
2848                 if (dqb->dqb_ihardlimit &&
2849                     dqb->dqb_curinodes >= dqb->dqb_ihardlimit) {
2850                         iover = 1;
2851                 } else if (dqb->dqb_isoftlimit && dqb->dqb_itime) {
2852                         if (dqb->dqb_itime > now) {
2853                                 iover = 2;
2854                         } else {
2855                                 iover = 3;
2856                         }
2857                 }
2858
2859
2860                 if (strlen(mnt) > 15)
2861                         printf("%s\n%15s", mnt, "");
2862                 else
2863                         printf("%15s", mnt);
2864
2865                 if (bover)
2866                         diff2str(dqb->dqb_btime, timebuf, now);
2867
2868                 kbytes2str(lustre_stoqb(dqb->dqb_curspace), strbuf, h);
2869                 if (rc == -EREMOTEIO)
2870                         sprintf(numbuf[0], "%s*", strbuf);
2871                 else
2872                         sprintf(numbuf[0], (dqb->dqb_valid & QIF_SPACE) ?
2873                                 "%s" : "[%s]", strbuf);
2874
2875                 kbytes2str(dqb->dqb_bsoftlimit, strbuf, h);
2876                 if (type == QC_GENERAL)
2877                         sprintf(numbuf[1], (dqb->dqb_valid & QIF_BLIMITS) ?
2878                                 "%s" : "[%s]", strbuf);
2879                 else
2880                         sprintf(numbuf[1], "%s", "-");
2881
2882                 kbytes2str(dqb->dqb_bhardlimit, strbuf, h);
2883                 sprintf(numbuf[2], (dqb->dqb_valid & QIF_BLIMITS) ?
2884                         "%s" : "[%s]", strbuf);
2885
2886                 printf(" %7s%c %6s %7s %7s",
2887                        numbuf[0], bover ? '*' : ' ', numbuf[1],
2888                        numbuf[2], bover > 1 ? timebuf : "-");
2889
2890                 if (iover)
2891                         diff2str(dqb->dqb_itime, timebuf, now);
2892
2893                 sprintf(numbuf[0], (dqb->dqb_valid & QIF_INODES) ?
2894                         LPU64 : "["LPU64"]", dqb->dqb_curinodes);
2895
2896                 if (type == QC_GENERAL)
2897                         sprintf(numbuf[1], (dqb->dqb_valid & QIF_ILIMITS) ?
2898                                 LPU64 : "["LPU64"]", dqb->dqb_isoftlimit);
2899                 else
2900                         sprintf(numbuf[1], "%s", "-");
2901
2902                 sprintf(numbuf[2], (dqb->dqb_valid & QIF_ILIMITS) ?
2903                         LPU64 : "["LPU64"]", dqb->dqb_ihardlimit);
2904
2905                 if (type != QC_OSTIDX)
2906                         printf(" %7s%c %6s %7s %7s",
2907                                numbuf[0], iover ? '*' : ' ', numbuf[1],
2908                                numbuf[2], iover > 1 ? timebuf : "-");
2909                 else
2910                         printf(" %7s %7s %7s %7s", "-", "-", "-", "-");
2911                 printf("\n");
2912
2913         } else if (qctl->qc_cmd == LUSTRE_Q_GETINFO ||
2914                    qctl->qc_cmd == Q_GETOINFO) {
2915                 char bgtimebuf[40];
2916                 char igtimebuf[40];
2917
2918                 sec2str(qctl->qc_dqinfo.dqi_bgrace, bgtimebuf, rc);
2919                 sec2str(qctl->qc_dqinfo.dqi_igrace, igtimebuf, rc);
2920                 printf("Block grace time: %s; Inode grace time: %s\n",
2921                        bgtimebuf, igtimebuf);
2922         }
2923 }
2924
2925 static int print_obd_quota(char *mnt, struct if_quotactl *qctl, int is_mdt,
2926                            bool h, __u64 *total)
2927 {
2928         int rc = 0, rc1 = 0, count = 0;
2929         __u32 valid = qctl->qc_valid;
2930
2931         rc = llapi_get_obd_count(mnt, &count, is_mdt);
2932         if (rc) {
2933                 fprintf(stderr, "can not get %s count: %s\n",
2934                         is_mdt ? "mdt": "ost", strerror(-rc));
2935                 return rc;
2936         }
2937
2938         for (qctl->qc_idx = 0; qctl->qc_idx < count; qctl->qc_idx++) {
2939                 qctl->qc_valid = is_mdt ? QC_MDTIDX : QC_OSTIDX;
2940                 rc = llapi_quotactl(mnt, qctl);
2941                 if (rc) {
2942                         /* It is remote client case. */
2943                         if (-rc == EOPNOTSUPP) {
2944                                 rc = 0;
2945                                 goto out;
2946                         }
2947
2948                         if (!rc1)
2949                                 rc1 = rc;
2950                         fprintf(stderr, "quotactl %s%d failed.\n",
2951                                 is_mdt ? "mdt": "ost", qctl->qc_idx);
2952                         continue;
2953                 }
2954
2955                 print_quota(obd_uuid2str(&qctl->obd_uuid), qctl,
2956                             qctl->qc_valid, 0, h);
2957                 *total += is_mdt ? qctl->qc_dqblk.dqb_ihardlimit :
2958                                    qctl->qc_dqblk.dqb_bhardlimit;
2959         }
2960 out:
2961         qctl->qc_valid = valid;
2962         return rc ? : rc1;
2963 }
2964
2965 static int lfs_quota(int argc, char **argv)
2966 {
2967         int c;
2968         char *mnt, *name = NULL;
2969         struct if_quotactl qctl = { .qc_cmd = LUSTRE_Q_GETQUOTA,
2970                                     .qc_type = UGQUOTA };
2971         char *obd_type = (char *)qctl.obd_type;
2972         char *obd_uuid = (char *)qctl.obd_uuid.uuid;
2973         int rc, rc1 = 0, rc2 = 0, rc3 = 0,
2974             verbose = 0, pass = 0, quiet = 0, inacc;
2975         char *endptr;
2976         __u32 valid = QC_GENERAL, idx = 0;
2977         __u64 total_ialloc = 0, total_balloc = 0;
2978         bool human_readable = false;
2979
2980         while ((c = getopt(argc, argv, "gi:I:o:qtuvh")) != -1) {
2981                 switch (c) {
2982                 case 'u':
2983                         if (qctl.qc_type != UGQUOTA) {
2984                                 fprintf(stderr, "error: use either -u or -g\n");
2985                                 return CMD_HELP;
2986                         }
2987                         qctl.qc_type = USRQUOTA;
2988                         break;
2989                 case 'g':
2990                         if (qctl.qc_type != UGQUOTA) {
2991                                 fprintf(stderr, "error: use either -u or -g\n");
2992                                 return CMD_HELP;
2993                         }
2994                         qctl.qc_type = GRPQUOTA;
2995                         break;
2996                 case 't':
2997                         qctl.qc_cmd = LUSTRE_Q_GETINFO;
2998                         break;
2999                 case 'o':
3000                         valid = qctl.qc_valid = QC_UUID;
3001                         strlcpy(obd_uuid, optarg, sizeof(qctl.obd_uuid));
3002                         break;
3003                 case 'i':
3004                         valid = qctl.qc_valid = QC_MDTIDX;
3005                         idx = qctl.qc_idx = atoi(optarg);
3006                         break;
3007                 case 'I':
3008                         valid = qctl.qc_valid = QC_OSTIDX;
3009                         idx = qctl.qc_idx = atoi(optarg);
3010                         break;
3011                 case 'v':
3012                         verbose = 1;
3013                         break;
3014                 case 'q':
3015                         quiet = 1;
3016                         break;
3017                 case 'h':
3018                         human_readable = true;
3019                         break;
3020                 default:
3021                         fprintf(stderr, "error: %s: option '-%c' "
3022                                         "unrecognized\n", argv[0], c);
3023                         return CMD_HELP;
3024                 }
3025         }
3026
3027         /* current uid/gid info for "lfs quota /path/to/lustre/mount" */
3028         if (qctl.qc_cmd == LUSTRE_Q_GETQUOTA && qctl.qc_type == UGQUOTA &&
3029             optind == argc - 1) {
3030 ug_output:
3031                 memset(&qctl, 0, sizeof(qctl)); /* spoiled by print_*_quota */
3032                 qctl.qc_cmd = LUSTRE_Q_GETQUOTA;
3033                 qctl.qc_valid = valid;
3034                 qctl.qc_idx = idx;
3035                 if (pass++ == 0) {
3036                         qctl.qc_type = USRQUOTA;
3037                         qctl.qc_id = geteuid();
3038                 } else {
3039                         qctl.qc_type = GRPQUOTA;
3040                         qctl.qc_id = getegid();
3041                 }
3042                 rc = id2name(&name, qctl.qc_id,
3043                              (qctl.qc_type == USRQUOTA) ? USER : GROUP);
3044                 if (rc)
3045                         name = "<unknown>";
3046         /* lfs quota -u username /path/to/lustre/mount */
3047         } else if (qctl.qc_cmd == LUSTRE_Q_GETQUOTA) {
3048                 /* options should be followed by u/g-name and mntpoint */
3049                 if (optind + 2 != argc || qctl.qc_type == UGQUOTA) {
3050                         fprintf(stderr, "error: missing quota argument(s)\n");
3051                         return CMD_HELP;
3052                 }
3053
3054                 name = argv[optind++];
3055                 rc = name2id(&qctl.qc_id, name,
3056                              (qctl.qc_type == USRQUOTA) ? USER : GROUP);
3057                 if (rc) {
3058                         qctl.qc_id = strtoul(name, &endptr, 10);
3059                         if (*endptr != '\0') {
3060                                 fprintf(stderr, "error: can't find id for name "
3061                                         "%s\n", name);
3062                                 return CMD_HELP;
3063                         }
3064                 }
3065         } else if (optind + 1 != argc || qctl.qc_type == UGQUOTA) {
3066                 fprintf(stderr, "error: missing quota info argument(s)\n");
3067                 return CMD_HELP;
3068         }
3069
3070         mnt = argv[optind];
3071
3072         rc1 = llapi_quotactl(mnt, &qctl);
3073         if (rc1 < 0) {
3074                 switch (rc1) {
3075                 case -ESRCH:
3076                         fprintf(stderr, "%s quotas are not enabled.\n",
3077                                 qctl.qc_type == USRQUOTA ? "user" : "group");
3078                         goto out;
3079                 case -EPERM:
3080                         fprintf(stderr, "Permission denied.\n");
3081                 case -ENOENT:
3082                         /* We already got a "No such file..." message. */
3083                         goto out;
3084                 default:
3085                         fprintf(stderr, "Unexpected quotactl error: %s\n",
3086                                 strerror(-rc1));
3087                 }
3088         }
3089
3090         if (qctl.qc_cmd == LUSTRE_Q_GETQUOTA && !quiet)
3091                 print_quota_title(name, &qctl, human_readable);
3092
3093         if (rc1 && *obd_type)
3094                 fprintf(stderr, "%s %s ", obd_type, obd_uuid);
3095
3096         if (qctl.qc_valid != QC_GENERAL)
3097                 mnt = "";
3098
3099         inacc = (qctl.qc_cmd == LUSTRE_Q_GETQUOTA) &&
3100                 ((qctl.qc_dqblk.dqb_valid & (QIF_LIMITS|QIF_USAGE)) !=
3101                  (QIF_LIMITS|QIF_USAGE));
3102
3103         print_quota(mnt, &qctl, QC_GENERAL, rc1, human_readable);
3104
3105         if (qctl.qc_valid == QC_GENERAL && qctl.qc_cmd != LUSTRE_Q_GETINFO &&
3106             verbose) {
3107                 char strbuf[32];
3108
3109                 rc2 = print_obd_quota(mnt, &qctl, 1, human_readable,
3110                                       &total_ialloc);
3111                 rc3 = print_obd_quota(mnt, &qctl, 0, human_readable,
3112                                       &total_balloc);
3113                 kbytes2str(total_balloc, strbuf, human_readable);
3114                 printf("Total allocated inode limit: "LPU64", total "
3115                        "allocated block limit: %s\n", total_ialloc, strbuf);
3116         }
3117
3118         if (rc1 || rc2 || rc3 || inacc)
3119                 printf("Some errors happened when getting quota info. "
3120                        "Some devices may be not working or deactivated. "
3121                        "The data in \"[]\" is inaccurate.\n");
3122
3123 out:
3124         if (pass == 1)
3125                 goto ug_output;
3126
3127         return rc1;
3128 }
3129 #endif /* HAVE_SYS_QUOTA_H! */
3130
3131 static int flushctx_ioctl(char *mp)
3132 {
3133         int fd, rc;
3134
3135         fd = open(mp, O_RDONLY);
3136         if (fd == -1) {
3137                 fprintf(stderr, "flushctx: error open %s: %s\n",
3138                         mp, strerror(errno));
3139                 return -1;
3140         }
3141
3142         rc = ioctl(fd, LL_IOC_FLUSHCTX);
3143         if (rc == -1)
3144                 fprintf(stderr, "flushctx: error ioctl %s: %s\n",
3145                         mp, strerror(errno));
3146
3147         close(fd);
3148         return rc;
3149 }
3150
3151 static int lfs_flushctx(int argc, char **argv)
3152 {
3153         int     kdestroy = 0, c;
3154         char    mntdir[PATH_MAX] = {'\0'};
3155         int     index = 0;
3156         int     rc = 0;
3157
3158         while ((c = getopt(argc, argv, "k")) != -1) {
3159                 switch (c) {
3160                 case 'k':
3161                         kdestroy = 1;
3162                         break;
3163                 default:
3164                         fprintf(stderr, "error: %s: option '-%c' "
3165                                         "unrecognized\n", argv[0], c);
3166                         return CMD_HELP;
3167                 }
3168         }
3169
3170         if (kdestroy) {
3171             if ((rc = system("kdestroy > /dev/null")) != 0) {
3172                 rc = WEXITSTATUS(rc);
3173                 fprintf(stderr, "error destroying tickets: %d, continuing\n", rc);
3174             }
3175         }
3176
3177         if (optind >= argc) {
3178                 /* flush for all mounted lustre fs. */
3179                 while (!llapi_search_mounts(NULL, index++, mntdir, NULL)) {
3180                         /* Check if we have a mount point */
3181                         if (mntdir[0] == '\0')
3182                                 continue;
3183
3184                         if (flushctx_ioctl(mntdir))
3185                                 rc = -1;
3186
3187                         mntdir[0] = '\0'; /* avoid matching in next loop */
3188                 }
3189         } else {
3190                 /* flush fs as specified */
3191                 while (optind < argc) {
3192                         if (flushctx_ioctl(argv[optind++]))
3193                                 rc = -1;
3194                 }
3195         }
3196         return rc;
3197 }
3198
3199 static int lfs_lsetfacl(int argc, char **argv)
3200 {
3201         argv[0]++;
3202         return(llapi_lsetfacl(argc, argv));
3203 }
3204
3205 static int lfs_lgetfacl(int argc, char **argv)
3206 {
3207         argv[0]++;
3208         return(llapi_lgetfacl(argc, argv));
3209 }
3210
3211 static int lfs_rsetfacl(int argc, char **argv)
3212 {
3213         argv[0]++;
3214         return(llapi_rsetfacl(argc, argv));
3215 }
3216
3217 static int lfs_rgetfacl(int argc, char **argv)
3218 {
3219         argv[0]++;
3220         return(llapi_rgetfacl(argc, argv));
3221 }
3222
3223 static int lfs_cp(int argc, char **argv)
3224 {
3225         return(llapi_cp(argc, argv));
3226 }
3227
3228 static int lfs_ls(int argc, char **argv)
3229 {
3230         return(llapi_ls(argc, argv));
3231 }
3232
3233 static int lfs_changelog(int argc, char **argv)
3234 {
3235         void *changelog_priv;
3236         struct changelog_rec *rec;
3237         long long startrec = 0, endrec = 0;
3238         char *mdd;
3239         struct option long_opts[] = {
3240                 {"follow", no_argument, 0, 'f'},
3241                 {0, 0, 0, 0}
3242         };
3243         char short_opts[] = "f";
3244         int rc, follow = 0;
3245
3246         while ((rc = getopt_long(argc, argv, short_opts,
3247                                 long_opts, NULL)) != -1) {
3248                 switch (rc) {
3249                 case 'f':
3250                         follow++;
3251                         break;
3252                 case '?':
3253                         return CMD_HELP;
3254                 default:
3255                         fprintf(stderr, "error: %s: option '%s' unrecognized\n",
3256                                 argv[0], argv[optind - 1]);
3257                         return CMD_HELP;
3258                 }
3259         }
3260         if (optind >= argc)
3261                 return CMD_HELP;
3262
3263         mdd = argv[optind++];
3264         if (argc > optind)
3265                 startrec = strtoll(argv[optind++], NULL, 10);
3266         if (argc > optind)
3267                 endrec = strtoll(argv[optind++], NULL, 10);
3268
3269         rc = llapi_changelog_start(&changelog_priv,
3270                                    CHANGELOG_FLAG_BLOCK |
3271                                    CHANGELOG_FLAG_JOBID |
3272                                    (follow ? CHANGELOG_FLAG_FOLLOW : 0),
3273                                    mdd, startrec);
3274         if (rc < 0) {
3275                 fprintf(stderr, "Can't start changelog: %s\n",
3276                         strerror(errno = -rc));
3277                 return rc;
3278         }
3279
3280         while ((rc = llapi_changelog_recv(changelog_priv, &rec)) == 0) {
3281                 time_t secs;
3282                 struct tm ts;
3283
3284                 if (endrec && rec->cr_index > endrec) {
3285                         llapi_changelog_free(&rec);
3286                         break;
3287                 }
3288                 if (rec->cr_index < startrec) {
3289                         llapi_changelog_free(&rec);
3290                         continue;
3291                 }
3292
3293                 secs = rec->cr_time >> 30;
3294                 gmtime_r(&secs, &ts);
3295                 printf(LPU64" %02d%-5s %02d:%02d:%02d.%06d %04d.%02d.%02d "
3296                        "0x%x t="DFID, rec->cr_index, rec->cr_type,
3297                        changelog_type2str(rec->cr_type),
3298                        ts.tm_hour, ts.tm_min, ts.tm_sec,
3299                        (int)(rec->cr_time & ((1<<30) - 1)),
3300                        ts.tm_year + 1900, ts.tm_mon + 1, ts.tm_mday,
3301                        rec->cr_flags & CLF_FLAGMASK, PFID(&rec->cr_tfid));
3302
3303                 if (rec->cr_flags & CLF_JOBID) {
3304                         struct changelog_ext_jobid *jid =
3305                                 changelog_rec_jobid(rec);
3306
3307                         if (jid->cr_jobid[0] != '\0')
3308                                 printf(" j=%s", jid->cr_jobid);
3309                 }
3310
3311                 if (rec->cr_namelen)
3312                         printf(" p="DFID" %.*s", PFID(&rec->cr_pfid),
3313                                rec->cr_namelen, changelog_rec_name(rec));
3314
3315                 if (rec->cr_flags & CLF_RENAME) {
3316                         struct changelog_ext_rename *rnm =
3317                                 changelog_rec_rename(rec);
3318
3319                         if (!fid_is_zero(&rnm->cr_sfid))
3320                                 printf(" s="DFID" sp="DFID" %.*s",
3321                                        PFID(&rnm->cr_sfid),
3322                                        PFID(&rnm->cr_spfid),
3323                                        (int)changelog_rec_snamelen(rec),
3324                                        changelog_rec_sname(rec));
3325                 }
3326                 printf("\n");
3327
3328                 llapi_changelog_free(&rec);
3329         }
3330
3331         llapi_changelog_fini(&changelog_priv);
3332
3333         if (rc < 0)
3334                 fprintf(stderr, "Changelog: %s\n", strerror(errno = -rc));
3335
3336         return (rc == 1 ? 0 : rc);
3337 }
3338
3339 static int lfs_changelog_clear(int argc, char **argv)
3340 {
3341         long long endrec;
3342         int rc;
3343
3344         if (argc != 4)
3345                 return CMD_HELP;
3346
3347         endrec = strtoll(argv[3], NULL, 10);
3348
3349         rc = llapi_changelog_clear(argv[1], argv[2], endrec);
3350         if (rc)
3351                 fprintf(stderr, "%s error: %s\n", argv[0],
3352                         strerror(errno = -rc));
3353         return rc;
3354 }
3355
3356 static int lfs_fid2path(int argc, char **argv)
3357 {
3358         struct option long_opts[] = {
3359                 {"cur", no_argument, 0, 'c'},
3360                 {"link", required_argument, 0, 'l'},
3361                 {"rec", required_argument, 0, 'r'},
3362                 {0, 0, 0, 0}
3363         };
3364         char  short_opts[] = "cl:r:";
3365         char *device, *fid, *path;
3366         long long recno = -1;
3367         int linkno = -1;
3368         int lnktmp;
3369         int printcur = 0;
3370         int rc = 0;
3371
3372         while ((rc = getopt_long(argc, argv, short_opts,
3373                                 long_opts, NULL)) != -1) {
3374                 switch (rc) {
3375                 case 'c':
3376                         printcur++;
3377                         break;
3378                 case 'l':
3379                         linkno = strtol(optarg, NULL, 10);
3380                         break;
3381                 case 'r':
3382                         recno = strtoll(optarg, NULL, 10);
3383                         break;
3384                 case '?':
3385                         return CMD_HELP;
3386                 default:
3387                         fprintf(stderr, "error: %s: option '%s' unrecognized\n",
3388                                 argv[0], argv[optind - 1]);
3389                         return CMD_HELP;
3390                 }
3391         }
3392
3393         if (argc < 3)
3394                 return CMD_HELP;
3395
3396         device = argv[optind++];
3397         path = calloc(1, PATH_MAX);
3398         if (path == NULL) {
3399                 fprintf(stderr, "error: Not enough memory\n");
3400                 return -errno;
3401         }
3402
3403         rc = 0;
3404         while (optind < argc) {
3405                 fid = argv[optind++];
3406
3407                 lnktmp = (linkno >= 0) ? linkno : 0;
3408                 while (1) {
3409                         int oldtmp = lnktmp;
3410                         long long rectmp = recno;
3411                         int rc2;
3412                         rc2 = llapi_fid2path(device, fid, path, PATH_MAX,
3413                                              &rectmp, &lnktmp);
3414                         if (rc2 < 0) {
3415                                 fprintf(stderr, "%s: error on FID %s: %s\n",
3416                                         argv[0], fid, strerror(errno = -rc2));
3417                                 if (rc == 0)
3418                                         rc = rc2;
3419                                 break;
3420                         }
3421
3422                         if (printcur)
3423                                 fprintf(stdout, "%lld ", rectmp);
3424                         if (device[0] == '/') {
3425                                 fprintf(stdout, "%s", device);
3426                                 if (device[strlen(device) - 1] != '/')
3427                                         fprintf(stdout, "/");
3428                         } else if (path[0] == '\0') {
3429                                 fprintf(stdout, "/");
3430                         }
3431                         fprintf(stdout, "%s\n", path);
3432
3433                         if (linkno >= 0)
3434                                 /* specified linkno */
3435                                 break;
3436                         if (oldtmp == lnktmp)
3437                                 /* no more links */
3438                                 break;
3439                 }
3440         }
3441
3442         free(path);
3443         return rc;
3444 }
3445
3446 static int lfs_path2fid(int argc, char **argv)
3447 {
3448         struct option     long_opts[] = {
3449                 {"parents", no_argument, 0, 'p'},
3450                 {0, 0, 0, 0}
3451         };
3452         char            **path;
3453         const char        short_opts[] = "p";
3454         const char       *sep = "";
3455         lustre_fid        fid;
3456         int               rc = 0;
3457         bool              show_parents = false;
3458
3459         while ((rc = getopt_long(argc, argv, short_opts,
3460                                  long_opts, NULL)) != -1) {
3461                 switch (rc) {
3462                 case 'p':
3463                         show_parents = true;
3464                         break;
3465                 default:
3466                         fprintf(stderr, "error: %s: option '%s' unrecognized\n",
3467                                 argv[0], argv[optind - 1]);
3468                         return CMD_HELP;
3469                 }
3470         }
3471
3472         if (optind > argc - 1)
3473                 return CMD_HELP;
3474         else if (optind < argc - 1)
3475                 sep = ": ";
3476
3477         rc = 0;
3478         for (path = argv + optind; *path != NULL; path++) {
3479                 int err = 0;
3480                 if (!show_parents) {
3481                         err = llapi_path2fid(*path, &fid);
3482                         if (!err)
3483                                 printf("%s%s"DFID"\n",
3484                                        *sep != '\0' ? *path : "", sep,
3485                                        PFID(&fid));
3486                 } else {
3487                         char            name[NAME_MAX + 1];
3488                         unsigned int    linkno = 0;
3489
3490                         while ((err = llapi_path2parent(*path, linkno, &fid,
3491                                                 name, sizeof(name))) == 0) {
3492                                 if (*sep != '\0' && linkno == 0)
3493                                         printf("%s%s", *path, sep);
3494
3495                                 printf("%s"DFID"/%s", linkno != 0 ? "\t" : "",
3496                                        PFID(&fid), name);
3497                                 linkno++;
3498                         }
3499
3500                         /* err == -ENODATA is end-of-loop */
3501                         if (linkno > 0 && err == -ENODATA) {
3502                                 printf("\n");
3503                                 err = 0;
3504                         }
3505                 }
3506
3507                 if (err) {
3508                         fprintf(stderr, "%s: can't get %sfid for %s: %s\n",
3509                                 argv[0], show_parents ? "parent " : "", *path,
3510                                 strerror(-err));
3511                         if (rc == 0) {
3512                                 rc = err;
3513                                 errno = -err;
3514                         }
3515                 }
3516         }
3517
3518         return rc;
3519 }
3520
3521 static int lfs_data_version(int argc, char **argv)
3522 {
3523         char *path;
3524         __u64 data_version;
3525         int fd;
3526         int rc;
3527         int c;
3528         int data_version_flags = LL_DV_RD_FLUSH; /* Read by default */
3529
3530         if (argc < 2)
3531                 return CMD_HELP;
3532
3533         while ((c = getopt(argc, argv, "nrw")) != -1) {
3534                 switch (c) {
3535                 case 'n':
3536                         data_version_flags = 0;
3537                         break;
3538                 case 'r':
3539                         data_version_flags |= LL_DV_RD_FLUSH;
3540                         break;
3541                 case 'w':
3542                         data_version_flags |= LL_DV_WR_FLUSH;
3543                         break;
3544                 default:
3545                         return CMD_HELP;
3546                 }
3547         }
3548         if (optind == argc)
3549                 return CMD_HELP;
3550
3551         path = argv[optind];
3552         fd = open(path, O_RDONLY);
3553         if (fd < 0)
3554                 err(errno, "cannot open file %s", path);
3555
3556         rc = llapi_get_data_version(fd, &data_version, data_version_flags);
3557         if (rc < 0)
3558                 err(errno, "cannot get version for %s", path);
3559         else
3560                 printf(LPU64 "\n", data_version);
3561
3562         close(fd);
3563         return rc;
3564 }
3565
3566 static int lfs_hsm_state(int argc, char **argv)
3567 {
3568         int rc;
3569         int i = 1;
3570         char *path;
3571         struct hsm_user_state hus;
3572
3573         if (argc < 2)
3574                 return CMD_HELP;
3575
3576         do {
3577                 path = argv[i];
3578
3579                 rc = llapi_hsm_state_get(path, &hus);
3580                 if (rc) {
3581                         fprintf(stderr, "can't get hsm state for %s: %s\n",
3582                                 path, strerror(errno = -rc));
3583                         return rc;
3584                 }
3585
3586                 /* Display path name and status flags */
3587                 printf("%s: (0x%08x)", path, hus.hus_states);
3588
3589                 if (hus.hus_states & HS_RELEASED)
3590                         printf(" released");
3591                 if (hus.hus_states & HS_EXISTS)
3592                         printf(" exists");
3593                 if (hus.hus_states & HS_DIRTY)
3594                         printf(" dirty");
3595                 if (hus.hus_states & HS_ARCHIVED)
3596                         printf(" archived");
3597                 /* Display user-settable flags */
3598                 if (hus.hus_states & HS_NORELEASE)
3599                         printf(" never_release");
3600                 if (hus.hus_states & HS_NOARCHIVE)
3601                         printf(" never_archive");
3602                 if (hus.hus_states & HS_LOST)
3603                         printf(" lost_from_hsm");
3604
3605                 if (hus.hus_archive_id != 0)
3606                         printf(", archive_id:%d", hus.hus_archive_id);
3607                 printf("\n");
3608
3609         } while (++i < argc);
3610
3611         return 0;
3612 }
3613
3614 #define LFS_HSM_SET   0
3615 #define LFS_HSM_CLEAR 1
3616
3617 /**
3618  * Generic function to set or clear HSM flags.
3619  * Used by hsm_set and hsm_clear.
3620  *
3621  * @mode  if LFS_HSM_SET, set the flags, if LFS_HSM_CLEAR, clear the flags.
3622  */
3623 static int lfs_hsm_change_flags(int argc, char **argv, int mode)
3624 {
3625         struct option long_opts[] = {
3626                 {"lost", 0, 0, 'l'},
3627                 {"norelease", 0, 0, 'r'},
3628                 {"noarchive", 0, 0, 'a'},
3629                 {"archived", 0, 0, 'A'},
3630                 {"dirty", 0, 0, 'd'},
3631                 {"exists", 0, 0, 'e'},
3632                 {0, 0, 0, 0}
3633         };
3634         char short_opts[] = "lraAde";
3635         __u64 mask = 0;
3636         int c, rc;
3637         char *path;
3638
3639         if (argc < 3)
3640                 return CMD_HELP;
3641
3642         while ((c = getopt_long(argc, argv, short_opts,
3643                                 long_opts, NULL)) != -1) {
3644                 switch (c) {
3645                 case 'l':
3646                         mask |= HS_LOST;
3647                         break;
3648                 case 'a':
3649                         mask |= HS_NOARCHIVE;
3650                         break;
3651                 case 'A':
3652                         mask |= HS_ARCHIVED;
3653                         break;
3654                 case 'r':
3655                         mask |= HS_NORELEASE;
3656                         break;
3657                 case 'd':
3658                         mask |= HS_DIRTY;
3659                         break;
3660                 case 'e':
3661                         mask |= HS_EXISTS;
3662                         break;
3663                 case '?':
3664                         return CMD_HELP;
3665                 default:
3666                         fprintf(stderr, "error: %s: option '%s' unrecognized\n",
3667                                 argv[0], argv[optind - 1]);
3668                         return CMD_HELP;
3669                 }
3670         }
3671
3672         /* User should have specified a flag */
3673         if (mask == 0)
3674                 return CMD_HELP;
3675
3676         while (optind < argc) {
3677
3678                 path = argv[optind];
3679
3680                 /* If mode == 0, this means we apply the mask. */
3681                 if (mode == LFS_HSM_SET)
3682                         rc = llapi_hsm_state_set(path, mask, 0, 0);
3683                 else
3684                         rc = llapi_hsm_state_set(path, 0, mask, 0);
3685
3686                 if (rc != 0) {
3687                         fprintf(stderr, "Can't change hsm flags for %s: %s\n",
3688                                 path, strerror(errno = -rc));
3689                         return rc;
3690                 }
3691                 optind++;
3692         }
3693
3694         return 0;
3695 }
3696
3697 static int lfs_hsm_action(int argc, char **argv)
3698 {
3699         int                              rc;
3700         int                              i = 1;
3701         char                            *path;
3702         struct hsm_current_action        hca;
3703         struct hsm_extent                he;
3704         enum hsm_user_action             hua;
3705         enum hsm_progress_states         hps;
3706
3707         if (argc < 2)
3708                 return CMD_HELP;
3709
3710         do {
3711                 path = argv[i];
3712
3713                 rc = llapi_hsm_current_action(path, &hca);
3714                 if (rc) {
3715                         fprintf(stderr, "can't get hsm action for %s: %s\n",
3716                                 path, strerror(errno = -rc));
3717                         return rc;
3718                 }
3719                 he = hca.hca_location;
3720                 hua = hca.hca_action;
3721                 hps = hca.hca_state;
3722
3723                 printf("%s: %s", path, hsm_user_action2name(hua));
3724
3725                 /* Skip file without action */
3726                 if (hca.hca_action == HUA_NONE) {
3727                         printf("\n");
3728                         continue;
3729                 }
3730
3731                 printf(" %s ", hsm_progress_state2name(hps));
3732
3733                 if ((hps == HPS_RUNNING) &&
3734                     (hua == HUA_ARCHIVE || hua == HUA_RESTORE))
3735                         printf("("LPX64 " bytes moved)\n", he.length);
3736                 else if ((he.offset + he.length) == LUSTRE_EOF)
3737                         printf("(from "LPX64 " to EOF)\n", he.offset);
3738                 else
3739                         printf("(from "LPX64 " to "LPX64")\n",
3740                                he.offset, he.offset + he.length);
3741
3742         } while (++i < argc);
3743
3744         return 0;
3745 }
3746
3747 static int lfs_hsm_set(int argc, char **argv)
3748 {
3749         return lfs_hsm_change_flags(argc, argv, LFS_HSM_SET);
3750 }
3751
3752 static int lfs_hsm_clear(int argc, char **argv)
3753 {
3754         return lfs_hsm_change_flags(argc, argv, LFS_HSM_CLEAR);
3755 }
3756
3757 /**
3758  * Check file state and return its fid, to be used by lfs_hsm_request().
3759  *
3760  * \param[in]     file      Path to file to check
3761  * \param[in,out] fid       Pointer to allocated lu_fid struct.
3762  * \param[in,out] last_dev  Pointer to last device id used.
3763  *
3764  * \return 0 on success.
3765  */
3766 static int lfs_hsm_prepare_file(char *file, struct lu_fid *fid,
3767                                 dev_t *last_dev)
3768 {
3769         struct stat     st;
3770         int             rc;
3771
3772         rc = lstat(file, &st);
3773         if (rc) {
3774                 fprintf(stderr, "Cannot stat %s: %s\n", file, strerror(errno));
3775                 return -errno;
3776         }
3777         /* Checking for regular file as archiving as posix copytool
3778          * rejects archiving files other than regular files
3779          */
3780         if (!S_ISREG(st.st_mode)) {
3781                 fprintf(stderr, "error: \"%s\" is not a regular file\n", file);
3782                 return CMD_HELP;
3783         }
3784         /* A request should be ... */
3785         if (*last_dev != st.st_dev && *last_dev != 0) {
3786                 fprintf(stderr, "All files should be "
3787                         "on the same filesystem: %s\n", file);
3788                 return -EINVAL;
3789         }
3790         *last_dev = st.st_dev;
3791
3792         rc = llapi_path2fid(file, fid);
3793         if (rc) {
3794                 fprintf(stderr, "Cannot read FID of %s: %s\n",
3795                         file, strerror(-rc));
3796                 return rc;
3797         }
3798         return 0;
3799 }
3800
3801 static int lfs_hsm_request(int argc, char **argv, int action)
3802 {
3803         struct option            long_opts[] = {
3804                 {"filelist", 1, 0, 'l'},
3805                 {"data", 1, 0, 'D'},
3806                 {"archive", 1, 0, 'a'},
3807                 {0, 0, 0, 0}
3808         };
3809         dev_t                    last_dev = 0;
3810         char                     short_opts[] = "l:D:a:";
3811         struct hsm_user_request *hur, *oldhur;
3812         int                      c, i;
3813         size_t                   len;
3814         int                      nbfile;
3815         char                    *line = NULL;
3816         char                    *filelist = NULL;
3817         char                     fullpath[PATH_MAX];
3818         char                    *opaque = NULL;
3819         int                      opaque_len = 0;
3820         int                      archive_id = 0;
3821         FILE                    *fp;
3822         int                      nbfile_alloc = 0;
3823         char                     some_file[PATH_MAX+1] = "";
3824         int                      rc;
3825
3826         if (argc < 2)
3827                 return CMD_HELP;
3828
3829         while ((c = getopt_long(argc, argv, short_opts,
3830                                 long_opts, NULL)) != -1) {
3831                 switch (c) {
3832                 case 'l':
3833                         filelist = optarg;
3834                         break;
3835                 case 'D':
3836                         opaque = optarg;
3837                         break;
3838                 case 'a':
3839                         if (action != HUA_ARCHIVE) {
3840                                 fprintf(stderr,
3841                                         "error: -a is supported only "
3842                                         "when archiving\n");
3843                                 return CMD_HELP;
3844                         }
3845                         archive_id = atoi(optarg);
3846                         break;
3847                 case '?':
3848                         return CMD_HELP;
3849                 default:
3850                         fprintf(stderr, "error: %s: option '%s' unrecognized\n",
3851                                 argv[0], argv[optind - 1]);
3852                         return CMD_HELP;
3853                 }
3854         }
3855
3856         /* All remaining args are files, so we have at least nbfile */
3857         nbfile = argc - optind;
3858
3859         if ((nbfile == 0) && (filelist == NULL))
3860                 return CMD_HELP;
3861
3862         if (opaque != NULL)
3863                 opaque_len = strlen(opaque);
3864
3865         /* Alloc the request structure with enough place to store all files
3866          * from command line. */
3867         hur = llapi_hsm_user_request_alloc(nbfile, opaque_len);
3868         if (hur == NULL) {
3869                 fprintf(stderr, "Cannot create the request: %s\n",
3870                         strerror(errno));
3871                 return errno;
3872         }
3873         nbfile_alloc = nbfile;
3874
3875         hur->hur_request.hr_action = action;
3876         hur->hur_request.hr_archive_id = archive_id;
3877         hur->hur_request.hr_flags = 0;
3878
3879         /* All remaining args are files, add them */
3880         if (nbfile != 0) {
3881                 if (strlen(argv[optind]) > sizeof(some_file)-1) {
3882                         free(hur);
3883                         return -E2BIG;
3884                 }
3885                 strncpy(some_file, argv[optind], sizeof(some_file));
3886         }
3887
3888         for (i = 0; i < nbfile; i++) {
3889                 hur->hur_user_item[i].hui_extent.length = -1;
3890                 rc = lfs_hsm_prepare_file(argv[optind + i],
3891                                           &hur->hur_user_item[i].hui_fid,
3892                                           &last_dev);
3893                 hur->hur_request.hr_itemcount++;
3894                 if (rc)
3895                         goto out_free;
3896         }
3897
3898         /* from here stop using nb_file, use hur->hur_request.hr_itemcount */
3899
3900         /* If a filelist was specified, read the filelist from it. */
3901         if (filelist != NULL) {
3902                 fp = fopen(filelist, "r");
3903                 if (fp == NULL) {
3904                         fprintf(stderr, "Cannot read the file list %s: %s\n",
3905                                 filelist, strerror(errno));
3906                         rc = -errno;
3907                         goto out_free;
3908                 }
3909
3910                 while ((rc = getline(&line, &len, fp)) != -1) {
3911                         struct hsm_user_item *hui;
3912
3913                         /* If allocated buffer was too small, gets something
3914                          * bigger */
3915                         if (nbfile_alloc <= hur->hur_request.hr_itemcount) {
3916                                 ssize_t size;
3917                                 nbfile_alloc = nbfile_alloc * 2 + 1;
3918                                 oldhur = hur;
3919                                 hur = llapi_hsm_user_request_alloc(nbfile_alloc,
3920                                                                    opaque_len);
3921                                 if (hur == NULL) {
3922                                         fprintf(stderr, "hsm: cannot allocate "
3923                                                 "the request: %s\n",
3924                                                 strerror(errno));
3925                                         hur = oldhur;
3926                                         rc = -errno;
3927                                         fclose(fp);
3928                                         goto out_free;
3929                                 }
3930                                 size = hur_len(oldhur);
3931                                 if (size < 0) {
3932                                         fprintf(stderr, "hsm: cannot allocate "
3933                                                 "%u files + %u bytes data\n",
3934                                             oldhur->hur_request.hr_itemcount,
3935                                             oldhur->hur_request.hr_data_len);
3936                                         free(hur);
3937                                         hur = oldhur;
3938                                         rc = -E2BIG;
3939                                         fclose(fp);
3940                                         goto out_free;
3941                                 }
3942                                 memcpy(hur, oldhur, size);
3943                                 free(oldhur);
3944                         }
3945
3946                         /* Chop CR */
3947                         if (line[strlen(line) - 1] == '\n')
3948                                 line[strlen(line) - 1] = '\0';
3949
3950                         hui =
3951                              &hur->hur_user_item[hur->hur_request.hr_itemcount];
3952                         hui->hui_extent.length = -1;
3953                         rc = lfs_hsm_prepare_file(line, &hui->hui_fid,
3954                                                   &last_dev);
3955                         hur->hur_request.hr_itemcount++;
3956                         if (rc) {
3957                                 fclose(fp);
3958                                 goto out_free;
3959                         }
3960
3961                         if ((some_file[0] == '\0') &&
3962                             (strlen(line) < sizeof(some_file)))
3963                                 strcpy(some_file, line);
3964                 }
3965
3966                 rc = fclose(fp);
3967                 if (line)
3968                         free(line);
3969         }
3970
3971         /* If a --data was used, add it to the request */
3972         hur->hur_request.hr_data_len = opaque_len;
3973         if (opaque != NULL)
3974                 memcpy(hur_data(hur), opaque, opaque_len);
3975
3976         /* Send the HSM request */
3977         if (realpath(some_file, fullpath) == NULL) {
3978                 fprintf(stderr, "Could not find path '%s': %s\n",
3979                         some_file, strerror(errno));
3980         }
3981         rc = llapi_hsm_request(fullpath, hur);
3982         if (rc) {
3983                 fprintf(stderr, "Cannot send HSM request (use of %s): %s\n",
3984                         some_file, strerror(-rc));
3985                 goto out_free;
3986         }
3987
3988 out_free:
3989         free(hur);
3990         return rc;
3991 }
3992
3993 static int lfs_hsm_archive(int argc, char **argv)
3994 {
3995         return lfs_hsm_request(argc, argv, HUA_ARCHIVE);
3996 }
3997
3998 static int lfs_hsm_restore(int argc, char **argv)
3999 {
4000         return lfs_hsm_request(argc, argv, HUA_RESTORE);
4001 }
4002
4003 static int lfs_hsm_release(int argc, char **argv)
4004 {
4005         return lfs_hsm_request(argc, argv, HUA_RELEASE);
4006 }
4007
4008 static int lfs_hsm_remove(int argc, char **argv)
4009 {
4010         return lfs_hsm_request(argc, argv, HUA_REMOVE);
4011 }
4012
4013 static int lfs_hsm_cancel(int argc, char **argv)
4014 {
4015         return lfs_hsm_request(argc, argv, HUA_CANCEL);
4016 }
4017
4018 static int lfs_swap_layouts(int argc, char **argv)
4019 {
4020         if (argc != 3)
4021                 return CMD_HELP;
4022
4023         return llapi_swap_layouts(argv[1], argv[2], 0, 0,
4024                                   SWAP_LAYOUTS_KEEP_MTIME |
4025                                   SWAP_LAYOUTS_KEEP_ATIME);
4026 }
4027
4028 int main(int argc, char **argv)
4029 {
4030         int rc;
4031
4032         /* Ensure that liblustreapi constructor has run */
4033         if (!liblustreapi_initialized)
4034                 fprintf(stderr, "liblustreapi was not properly initialized\n");
4035
4036         setlinebuf(stdout);
4037
4038         Parser_init("lfs > ", cmdlist);
4039
4040         if (argc > 1) {
4041                 rc = Parser_execarg(argc - 1, argv + 1, cmdlist);
4042         } else {
4043                 rc = Parser_commands();
4044         }
4045
4046         return rc < 0 ? -rc : rc;
4047 }
4048
4049 #ifdef _LUSTRE_IDL_H_
4050 /* Everything we need here should be included by lustreapi.h. */
4051 # error "lfs should not depend on lustre_idl.h"
4052 #endif /* _LUSTRE_IDL_H_ */