Whamcloud - gitweb
3f71f03db131204438c5f6595ff4852126edcd5c
[fs/lustre-release.git] / lustre / utils / obd.c
1 /*
2  * GPL HEADER START
3  *
4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License version 2 only,
8  * as published by the Free Software Foundation.
9  *
10  * This program is distributed in the hope that it will be useful, but
11  * WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * General Public License version 2 for more details (a copy is included
14  * in the LICENSE file that accompanied this code).
15  *
16  * You should have received a copy of the GNU General Public License
17  * version 2 along with this program; If not, see
18  * http://www.gnu.org/licenses/gpl-2.0.html
19  *
20  * GPL HEADER END
21  */
22 /*
23  * Copyright (c) 2002, 2010, Oracle and/or its affiliates. All rights reserved.
24  * Use is subject to license terms.
25  *
26  * Copyright (c) 2011, 2017, Intel Corporation.
27  */
28 /*
29  * This file is part of Lustre, http://www.lustre.org/
30  *
31  * lustre/utils/obd.c
32  *
33  * Author: Peter J. Braam <braam@clusterfs.com>
34  * Author: Phil Schwan <phil@clusterfs.com>
35  * Author: Andreas Dilger <adilger@clusterfs.com>
36  * Author: Robert Read <rread@clusterfs.com>
37  */
38
39 #include <sys/ioctl.h>
40 #include <sys/socket.h>
41 #include <sys/stat.h>
42 #include <sys/time.h>
43 #include <sys/types.h>
44 #include <sys/un.h>
45 #include <sys/wait.h>
46
47 #include <ctype.h>
48 #include <errno.h>
49 #include <fcntl.h>
50 #include <getopt.h>
51 #include <signal.h>
52 #include <stdarg.h>
53 #include <stdbool.h>
54 #include <stdio.h>
55 #include <stdlib.h>
56 #include <string.h>
57 #include <time.h>
58 #include <unistd.h>
59 #include <limits.h>
60 #include "obdctl.h"
61 #include "lustreapi_internal.h"
62 #include <libcfs/util/list.h>
63 #include <libcfs/util/ioctl.h>
64 #include <libcfs/util/param.h>
65 #include <libcfs/util/parser.h>
66 #include <libcfs/util/string.h>
67
68 #include <linux/lnet/nidstr.h>
69 #include <linux/lnet/lnetctl.h>
70 #ifdef HAVE_SERVER_SUPPPORT
71 #include <linux/lustre/lustre_barrier_user.h>
72 #include <linux/lustre/lustre_disk.h>
73 #endif
74 #include <linux/lustre/lustre_cfg.h>
75 #include <linux/lustre/lustre_ioctl.h>
76 #include <linux/lustre/lustre_ostid.h>
77 #include <linux/lustre/lustre_param.h>
78 #include <linux/lustre/lustre_ver.h>
79
80 #include <lustre/lustreapi.h>
81
82 #define MAX_STRING_SIZE 128
83
84 #if HAVE_LIBPTHREAD
85 #include <sys/ipc.h>
86 #include <sys/shm.h>
87 #include <pthread.h>
88
89 #define MAX_THREADS 4096
90 #define MAX_BASE_ID 0xffffffff
91 #define NIDSTRING_LENGTH 64
92 struct shared_data {
93         pthread_mutex_t mutex;
94         pthread_cond_t  cond;
95         int       stopping;
96         struct {
97                 __u64 counters[MAX_THREADS];
98                 __u64 offsets[MAX_THREADS];
99                 int   thr_running;
100                 int   start_barrier;
101                 int   stop_barrier;
102                 struct timeval start_time;
103                 struct timeval end_time;
104         } body;
105 };
106
107 static struct shared_data *shared_data;
108 static __u64 counter_snapshot[2][MAX_THREADS];
109 static int prev_valid;
110 static struct timeval prev_time;
111 static int thread;
112 static int nthreads;
113 #else
114 const int thread;
115 const int nthreads = 1;
116 #endif
117
118 static int cur_device = -1;
119
120 int lcfg_ioctl(char *func, int dev_id, struct lustre_cfg *lcfg)
121 {
122         struct obd_ioctl_data data;
123         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
124         int rc;
125
126         memset(&data, 0, sizeof(data));
127         data.ioc_dev = cur_device;
128         data.ioc_type = LUSTRE_CFG_TYPE;
129         data.ioc_plen1 = lustre_cfg_len(lcfg->lcfg_bufcount,
130                                         lcfg->lcfg_buflens);
131         data.ioc_pbuf1 = (void *)lcfg;
132         memset(buf, 0, sizeof(rawbuf));
133         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
134         if (rc) {
135                 fprintf(stderr, "error: %s: invalid ioctl\n", jt_cmdname(func));
136                 return rc;
137         }
138
139         rc =  l_ioctl(dev_id, OBD_IOC_PROCESS_CFG, buf);
140
141         return rc;
142 }
143
144 static int do_device(char *func, char *devname);
145
146 static int get_mgs_device(void)
147 {
148         char mgs[] = "$MGS";
149         static int mgs_device = -1;
150
151         if (mgs_device == -1) {
152                 int rc;
153
154                 do_disconnect(NULL, 1);
155                 rc = do_device("mgsioc", mgs);
156                 if (rc) {
157                         fprintf(stderr,
158                                 "This command must be run on the MGS.\n");
159                         errno = ENODEV;
160                         return -1;
161                 }
162                 mgs_device = cur_device;
163         }
164         return mgs_device;
165 }
166
167 /* Returns -1 on error with errno set */
168 int lcfg_mgs_ioctl(char *func, int dev_id, struct lustre_cfg *lcfg)
169 {
170         struct obd_ioctl_data data;
171         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
172         int rc;
173
174         memset(&data, 0, sizeof(data));
175         rc = data.ioc_dev = get_mgs_device();
176         if (rc < 0)
177                 goto out;
178         data.ioc_type = LUSTRE_CFG_TYPE;
179         data.ioc_plen1 = lustre_cfg_len(lcfg->lcfg_bufcount,
180                                         lcfg->lcfg_buflens);
181         data.ioc_pbuf1 = (void *)lcfg;
182         memset(buf, 0, sizeof(rawbuf));
183         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
184         if (rc) {
185                 fprintf(stderr, "error: %s: invalid ioctl\n", jt_cmdname(func));
186                 return rc;
187         }
188
189         rc = l_ioctl(dev_id, OBD_IOC_PARAM, buf);
190 out:
191         if (rc && errno == ENOSYS)
192                 fprintf(stderr, "Make sure cfg_device is set first.\n");
193
194         return rc;
195 }
196
197 char *obdo_print(struct obdo *obd)
198 {
199         char buf[1024];
200
201         snprintf(buf, sizeof(buf), "id: %#jx\ngrp: %#jx\natime: %ju\n"
202                  "mtime: %ju\nctime: %ju\nsize: %ju\nblocks: %ju"
203                  "\nblksize: %u\nmode: %o\nuid: %d\ngid: %d\nflags: %x\n"
204                  "misc: %x\nnlink: %d,\nvalid %#jx\n",
205                  (uintmax_t)ostid_id(&obd->o_oi),
206                  (uintmax_t)ostid_seq(&obd->o_oi),
207                  (uintmax_t)obd->o_atime, (uintmax_t)obd->o_mtime,
208                  (uintmax_t)obd->o_ctime, (uintmax_t)obd->o_size,
209                  (uintmax_t)obd->o_blocks, obd->o_blksize, obd->o_mode,
210                  obd->o_uid, obd->o_gid, obd->o_flags, obd->o_misc,
211                  obd->o_nlink, (uintmax_t)obd->o_valid);
212         return strdup(buf);
213 }
214
215 #define BAD_VERBOSE (-999999999)
216
217 #define N2D_OFF 0x100      /* So we can tell between error codes and devices */
218
219 static int do_name2dev(char *func, char *name)
220 {
221         struct obd_ioctl_data data;
222         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
223         int rc;
224
225         memset(&data, 0, sizeof(data));
226         data.ioc_dev = cur_device;
227         data.ioc_inllen1 = strlen(name) + 1;
228         data.ioc_inlbuf1 = name;
229
230         memset(buf, 0, sizeof(rawbuf));
231         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
232         if (rc < 0) {
233                 fprintf(stderr, "error: %s: invalid ioctl\n", jt_cmdname(func));
234                 return -rc;
235         }
236         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_NAME2DEV, buf);
237         if (rc < 0)
238                 return errno;
239         rc = llapi_ioctl_unpack(&data, buf, sizeof(rawbuf));
240         if (rc) {
241                 fprintf(stderr, "error: %s: invalid reply\n", jt_cmdname(func));
242                 return rc;
243         }
244
245         return data.ioc_dev + N2D_OFF;
246 }
247
248 /*
249  * resolve a device name to a device number.
250  * supports a number, $name or %uuid.
251  */
252 int parse_devname(char *func, char *name)
253 {
254         int rc;
255         int ret = -1;
256         int try_digit;
257
258         if (!name)
259                 return ret;
260
261         try_digit = isdigit(name[0]);
262
263         if (name[0] == '$' || name[0] == '%')
264                 name++;
265
266         rc = do_name2dev(func, name);
267         if (rc >= N2D_OFF)
268                 return rc - N2D_OFF;
269
270         if (try_digit)
271                 ret = strtoul(name, NULL, 0);
272         else
273                 fprintf(stderr, "No device found for name %s: %s\n",
274                         name, strerror(rc));
275         return ret;
276 }
277
278 char *jt_cmdname(char *func)
279 {
280         static char buf[512];
281
282         if (thread) {
283                 sprintf(buf, "%s-%d", func, thread);
284                 return buf;
285         }
286
287         return func;
288 }
289
290 #define difftime(a, b)                                  \
291         ((a)->tv_sec - (b)->tv_sec +                    \
292          ((a)->tv_usec - (b)->tv_usec) / 1000000.0)
293
294 static int be_verbose(int verbose, struct timeval *next_time,
295                       __u64 num, __u64 *next_num, int num_total)
296 {
297         struct timeval now;
298
299         if (!verbose)
300                 return 0;
301
302         if (next_time)
303                 gettimeofday(&now, NULL);
304
305         /* A positive verbosity means to print every X iterations */
306         if (verbose > 0 && (num >= *next_num || num >= num_total)) {
307                 *next_num += verbose;
308                 if (next_time) {
309                         next_time->tv_sec = now.tv_sec - verbose;
310                         next_time->tv_usec = now.tv_usec;
311                 }
312                 return 1;
313         }
314
315         /* A negative verbosity means to print at most each X seconds */
316         if (verbose < 0 && next_time && difftime(&now, next_time) >= 0.0) {
317                 next_time->tv_sec = now.tv_sec - verbose;
318                 next_time->tv_usec = now.tv_usec;
319                 *next_num = num;
320                 return 1;
321         }
322
323         return 0;
324 }
325
326 static int get_verbose(char *func, const char *arg)
327 {
328         int verbose;
329         char *end;
330
331         if (!arg || arg[0] == 'v') {
332                 verbose = 1;
333         } else if (arg[0] == 's' || arg[0] == 'q') {
334                 verbose = 0;
335         } else {
336                 verbose = (int)strtoul(arg, &end, 0);
337                 if (*end) {
338                         fprintf(stderr, "error: %s: bad verbose option '%s'\n",
339                                 jt_cmdname(func), arg);
340                         return BAD_VERBOSE;
341                 }
342         }
343
344         if (verbose < 0)
345                 printf("Print status every %d seconds\n", -verbose);
346         else if (verbose == 1)
347                 printf("Print status every operation\n");
348         else if (verbose > 1)
349                 printf("Print status every %d operations\n", verbose);
350
351         return verbose;
352 }
353
354 int do_disconnect(char *func, int verbose)
355 {
356         lcfg_set_devname(NULL);
357         cur_device = -1;
358         return 0;
359 }
360
361 #ifdef MAX_THREADS
362 static int shmem_setup(void)
363 {
364         pthread_mutexattr_t mattr;
365         pthread_condattr_t  cattr;
366         int rc;
367         int shmid;
368
369         /* Create new segment */
370         shmid = shmget(IPC_PRIVATE, sizeof(*shared_data), 0600);
371         if (shmid == -1) {
372                 fprintf(stderr, "Can't create shared data: %s\n",
373                         strerror(errno));
374                 return errno;
375         }
376
377         /* Attatch to new segment */
378         shared_data = (struct shared_data *)shmat(shmid, NULL, 0);
379
380         if (shared_data == (struct shared_data *)(-1)) {
381                 fprintf(stderr, "Can't attach shared data: %s\n",
382                         strerror(errno));
383                 shared_data = NULL;
384                 return errno;
385         }
386
387         /*
388          * Mark segment as destroyed, so it will disappear when we exit.
389          * Forks will inherit attached segments, so we should be OK.
390          */
391         if (shmctl(shmid, IPC_RMID, NULL) == -1) {
392                 fprintf(stderr, "Can't destroy shared data: %s\n",
393                         strerror(errno));
394                 return errno;
395         }
396
397         pthread_mutexattr_init(&mattr);
398         pthread_condattr_init(&cattr);
399
400         rc = pthread_mutexattr_setpshared(&mattr, PTHREAD_PROCESS_SHARED);
401         if (rc != 0) {
402                 fprintf(stderr, "Can't set shared mutex attr\n");
403                 goto out;
404         }
405
406         rc = pthread_condattr_setpshared(&cattr, PTHREAD_PROCESS_SHARED);
407         if (rc != 0) {
408                 fprintf(stderr, "Can't set shared cond attr\n");
409                 goto out;
410         }
411
412         pthread_mutex_init(&shared_data->mutex, &mattr);
413         pthread_cond_init(&shared_data->cond, &cattr);
414 out:
415         pthread_mutexattr_destroy(&mattr);
416         pthread_condattr_destroy(&cattr);
417
418         return rc;
419 }
420
421 static inline void shmem_lock(void)
422 {
423         pthread_mutex_lock(&shared_data->mutex);
424 }
425
426 static inline void shmem_unlock(void)
427 {
428         pthread_mutex_unlock(&shared_data->mutex);
429 }
430
431 static inline void shmem_wait(void)
432 {
433         pthread_cond_wait(&shared_data->cond, &shared_data->mutex);
434 }
435
436 static inline void shmem_wakeup_all(void)
437 {
438         pthread_cond_broadcast(&shared_data->cond);
439 }
440
441 static inline void shmem_reset(int total_threads)
442 {
443         if (!shared_data)
444                 return;
445
446         memset(&shared_data->body, 0, sizeof(shared_data->body));
447         memset(counter_snapshot, 0, sizeof(counter_snapshot));
448         prev_valid = 0;
449         shared_data->stopping = 0;
450         shared_data->body.start_barrier = total_threads;
451         shared_data->body.stop_barrier = total_threads;
452 }
453
454 static inline void shmem_bump(__u32 counter)
455 {
456         static bool running_not_bumped = true;
457
458         if (!shared_data || thread <= 0 || thread > MAX_THREADS)
459                 return;
460
461         shmem_lock();
462         shared_data->body.counters[thread - 1] += counter;
463         if (running_not_bumped) {
464                 shared_data->body.thr_running++;
465                 running_not_bumped = false;
466         }
467         shmem_unlock();
468 }
469
470 static void shmem_total(int total_threads)
471 {
472         __u64 total = 0;
473         double secs;
474         int i;
475
476         if (!shared_data || total_threads > MAX_THREADS)
477                 return;
478
479         shmem_lock();
480         for (i = 0; i < total_threads; i++)
481                 total += shared_data->body.counters[i];
482
483         secs = difftime(&shared_data->body.end_time,
484                         &shared_data->body.start_time);
485         shmem_unlock();
486
487         printf("Total: total %ju threads %d sec %f %f/second\n",
488                (uintmax_t)total, total_threads, secs, total / secs);
489 }
490
491 static void shmem_snap(int total_threads, int live_threads)
492 {
493         struct timeval this_time;
494         int non_zero = 0;
495         __u64 total = 0;
496         double secs;
497         int running;
498         int i;
499
500         if (!shared_data || total_threads > MAX_THREADS)
501                 return;
502
503         shmem_lock();
504         memcpy(counter_snapshot[0], shared_data->body.counters,
505                total_threads * sizeof(counter_snapshot[0][0]));
506         running = shared_data->body.thr_running;
507         shmem_unlock();
508
509         gettimeofday(&this_time, NULL);
510
511         for (i = 0; i < total_threads; i++) {
512                 long long this_count =
513                         counter_snapshot[0][i] - counter_snapshot[1][i];
514
515                 if (this_count != 0) {
516                         non_zero++;
517                         total += this_count;
518                 }
519         }
520
521         secs = difftime(&this_time, &prev_time);
522         if (prev_valid && secs > 1.0) {   /* someone screwed with the time? */
523                 printf("%d/%d Total: %f/second\n", non_zero, total_threads,
524                        total / secs);
525
526                 memcpy(counter_snapshot[1], counter_snapshot[0],
527                        total_threads * sizeof(counter_snapshot[0][0]));
528                 prev_time = this_time;
529         }
530         if (!prev_valid && running == total_threads) {
531                 prev_valid = 1;
532                 /* drop counters when all threads were started */
533                 memcpy(counter_snapshot[1], counter_snapshot[0],
534                        total_threads * sizeof(counter_snapshot[0][0]));
535                 prev_time = this_time;
536         }
537 }
538
539 static void shmem_stop(void)
540 {
541         if (!shared_data)
542                 return;
543
544         shared_data->stopping = 1;
545 }
546
547 static void shmem_cleanup(void)
548 {
549         if (!shared_data)
550                 return;
551
552         shmem_stop();
553
554         pthread_mutex_destroy(&shared_data->mutex);
555         pthread_cond_destroy(&shared_data->cond);
556 }
557
558 static int shmem_running(void)
559 {
560         return (!shared_data || !shared_data->stopping);
561 }
562
563 static void shmem_end_time_locked(void)
564 {
565         shared_data->body.stop_barrier--;
566         if (shared_data->body.stop_barrier == 0)
567                 gettimeofday(&shared_data->body.end_time, NULL);
568 }
569
570 static void shmem_start_time_locked(void)
571 {
572         shared_data->body.start_barrier--;
573         if (shared_data->body.start_barrier == 0) {
574                 shmem_wakeup_all();
575                 gettimeofday(&shared_data->body.start_time, NULL);
576         } else {
577                 shmem_wait();
578         }
579 }
580
581 #else
582 static int shmem_setup(void)
583 {
584         return 0;
585 }
586
587 static inline void shmem_reset(int total_threads)
588 {
589 }
590
591 static inline void shmem_bump(__u32 counters)
592 {
593 }
594
595 static void shmem_lock(void)
596 {
597 }
598
599 static void shmem_unlock(void)
600 {
601 }
602
603 static void shmem_cleanup(void)
604 {
605 }
606
607 static int shmem_running(void)
608 {
609         return 1;
610 }
611 #endif
612
613 extern command_t cmdlist[];
614
615 static int do_device(char *func, char *devname)
616 {
617         int dev;
618
619         dev = parse_devname(func, devname);
620         if (dev < 0)
621                 return -1;
622
623         lcfg_set_devname(devname);
624         cur_device = dev;
625         return 0;
626 }
627
628 int jt_obd_get_device(void)
629 {
630         return cur_device;
631 }
632
633 int jt_obd_device(int argc, char **argv)
634 {
635         int rc;
636
637         if (argc > 2)
638                 return CMD_HELP;
639
640         if (argc == 1) {
641                 printf("current device is %d - %s\n",
642                        cur_device, lcfg_get_devname() ? : "not set");
643                 return 0;
644         }
645         rc = do_device("device", argv[1]);
646         return rc;
647 }
648
649 int jt_opt_device(int argc, char **argv)
650 {
651         int ret;
652         int rc;
653
654         if (argc < 3)
655                 return CMD_HELP;
656
657         rc = do_device("device", argv[1]);
658
659         if (!rc)
660                 rc = Parser_execarg(argc - 2, argv + 2, cmdlist);
661
662         ret = do_disconnect(argv[0], 0);
663         if (!rc)
664                 rc = ret;
665
666         return rc;
667 }
668
669 #ifdef MAX_THREADS
670 static void parent_sighandler(int sig)
671 {
672 }
673
674 int jt_opt_threads(int argc, char **argv)
675 {
676         static char      cmdstr[129];
677         sigset_t         saveset;
678         sigset_t         sigset;
679         struct sigaction sigact;
680         struct sigaction saveact1;
681         struct sigaction saveact2;
682         unsigned long    threads;
683         __u64            next_thread;
684         int verbose;
685         int rc = 0;
686         int report_count = -1;
687         char *end;
688         int i;
689
690         if (argc < 5)
691                 return CMD_HELP;
692
693         threads = strtoul(argv[1], &end, 0);
694
695         if (*end == '.')
696                 report_count = strtoul(end + 1, &end, 0);
697
698         if (*end || threads > MAX_THREADS) {
699                 fprintf(stderr, "error: %s: invalid thread count '%s'\n",
700                         jt_cmdname(argv[0]), argv[1]);
701                 return CMD_HELP;
702         }
703
704         verbose = get_verbose(argv[0], argv[2]);
705         if (verbose == BAD_VERBOSE)
706                 return CMD_HELP;
707
708         if (verbose != 0) {
709                 snprintf(cmdstr, sizeof(cmdstr), "%s", argv[4]);
710                 for (i = 5; i < argc; i++)
711                         snprintf(cmdstr + strlen(cmdstr),
712                                  sizeof(cmdstr) - strlen(cmdstr),
713                                  " %s", argv[i]);
714
715                 printf("%s: starting %ld threads on device %s running %s\n",
716                        argv[0], threads, argv[3], cmdstr);
717         }
718
719         shmem_reset(threads);
720
721         sigemptyset(&sigset);
722         sigaddset(&sigset, SIGALRM);
723         sigaddset(&sigset, SIGCHLD);
724         sigprocmask(SIG_BLOCK, &sigset, &saveset);
725
726         nthreads = threads;
727
728         for (i = 1, next_thread = verbose; i <= threads; i++) {
729                 rc = fork();
730                 if (rc < 0) {
731                         fprintf(stderr, "error: %s: #%d - %s\n", argv[0], i,
732                                 strerror(rc = errno));
733                         break;
734                 } else if (rc == 0) {
735                         sigprocmask(SIG_SETMASK, &saveset, NULL);
736
737                         thread = i;
738                         argv[2] = "--device";
739                         exit(jt_opt_device(argc - 2, argv + 2));
740                 } else if (be_verbose(verbose, NULL, i, &next_thread, threads))
741                         printf("%s: thread #%d (PID %d) started\n",
742                                argv[0], i, rc);
743                 rc = 0;
744         }
745
746         if (!thread) {          /* parent process */
747                 int live_threads = threads;
748
749                 sigemptyset(&sigset);
750                 sigemptyset(&sigact.sa_mask);
751                 sigact.sa_handler = parent_sighandler;
752                 sigact.sa_flags = 0;
753
754                 sigaction(SIGALRM, &sigact, &saveact1);
755                 sigaction(SIGCHLD, &sigact, &saveact2);
756
757                 while (live_threads > 0) {
758                         int status;
759                         pid_t ret;
760
761                         if (verbose < 0)        /* periodic stats */
762                                 alarm(-verbose);
763
764                         sigsuspend(&sigset);
765                         alarm(0);
766
767                         while (live_threads > 0) {
768                                 ret = waitpid(0, &status, WNOHANG);
769                                 if (ret == 0)
770                                         break;
771
772                                 if (ret < 0) {
773                                         fprintf(stderr,
774                                                 "error: %s: wait - %s\n",
775                                                 argv[0], strerror(errno));
776                                         if (!rc)
777                                                 rc = errno;
778                                         continue;
779                                 } else {
780                                         /*
781                                          * This is a hack.  We _should_ be able
782                                          * to use WIFEXITED(status) to see if
783                                          * there was an error, but it appears
784                                          * to be broken and it always returns 1
785                                          * (OK).  See wait(2).
786                                          */
787                                         int err = WEXITSTATUS(status);
788
789                                         if (err || WIFSIGNALED(status))
790                                                 fprintf(stderr,
791                                                         "%s: PID %d had rc=%d\n",
792                                                         argv[0], ret, err);
793                                         if (!rc)
794                                                 rc = err;
795
796                                         live_threads--;
797                                 }
798                         }
799
800                         /* Show stats while all threads running */
801                         if (verbose < 0) {
802                                 shmem_snap(threads, live_threads);
803                                 if (report_count > 0 && --report_count == 0)
804                                         shmem_stop();
805                         }
806                 }
807                 sigaction(SIGCHLD, &saveact2, NULL);
808                 sigaction(SIGALRM, &saveact1, NULL);
809         }
810
811         shmem_total(threads);
812         sigprocmask(SIG_SETMASK, &saveset, NULL);
813
814         return rc;
815 }
816 #else
817 int jt_opt_threads(int argc, char **argv)
818 {
819         fprintf(stderr, "%s not-supported in a single-threaded runtime\n",
820                 jt_cmdname(argv[0]));
821         return CMD_HELP;
822 }
823 #endif
824
825 int jt_opt_net(int argc, char **argv)
826 {
827         char *arg2[3];
828         int rc;
829
830         if (argc < 3)
831                 return CMD_HELP;
832
833         arg2[0] = argv[0];
834         arg2[1] = argv[1];
835         arg2[2] = NULL;
836         rc = jt_ptl_network(2, arg2);
837
838         if (!rc)
839                 rc = Parser_execarg(argc - 2, argv + 2, cmdlist);
840
841         return rc;
842 }
843
844 #ifdef HAVE_SERVER_SUPPORT
845 /*
846  * Place this here so we can build tools that work with
847  * older Lustre versions
848  */
849 #ifndef OBD_IOC_NO_TRANSNO
850 #define OBD_IOC_NO_TRANSNO      _IOW('f', 140, OBD_IOC_DATA_TYPE)
851 #endif
852
853 int jt_obd_no_transno(int argc, char **argv)
854 {
855         struct obd_ioctl_data data;
856         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
857         glob_t path;
858         int count;
859         int fd;
860         int rc;
861
862         if (argc != 1)
863                 return CMD_HELP;
864
865         memset(buf, 0, sizeof(rawbuf));
866
867         rc = cfs_get_param_paths(&path, "no_transno");
868         if (rc != 0)
869                 goto old_ioctl;
870
871         fd = open(path.gl_pathv[0], O_WRONLY);
872         if (fd < 0) {
873                 cfs_free_param_data(&path);
874                 goto old_ioctl;
875         }
876
877         snprintf(rawbuf, sizeof(rawbuf), "%d", cur_device);
878
879         count = write(fd, rawbuf, strlen(rawbuf));
880         if (count < 0)
881                 rc = errno;
882
883         cfs_free_param_data(&path);
884         close(fd);
885         if (rc)
886                 goto old_ioctl;
887
888         return 0;
889
890 old_ioctl:
891 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(3, 0, 53, 0)
892         memset(&data, 0, sizeof(data));
893         data.ioc_dev = cur_device;
894
895         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
896         if (rc) {
897                 fprintf(stderr, "error: %s: invalid ioctl\n",
898                         jt_cmdname(argv[0]));
899                 return rc;
900         }
901         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_NO_TRANSNO, buf);
902         if (rc < 0)
903                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
904                         strerror(rc = errno));
905 #endif
906         return rc;
907 }
908
909 int jt_obd_set_readonly(int argc, char **argv)
910 {
911         struct obd_ioctl_data data;
912         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
913         int rc;
914
915         memset(&data, 0, sizeof(data));
916         data.ioc_dev = cur_device;
917
918         if (argc != 1)
919                 return CMD_HELP;
920
921         memset(buf, 0, sizeof(rawbuf));
922         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
923         if (rc) {
924                 fprintf(stderr, "error: %s: invalid ioctl\n",
925                         jt_cmdname(argv[0]));
926                 return rc;
927         }
928         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_SET_READONLY, buf);
929         if (rc < 0)
930                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
931                         strerror(rc = errno));
932
933         return rc;
934 }
935
936 static int obd_abort_recovery(char *cmd, enum obd_abort_recovery_flags flags)
937 {
938         struct obd_ioctl_data data = {
939                 .ioc_dev = cur_device,
940                 .ioc_type = flags,
941         };
942         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
943         int rc;
944
945         memset(buf, 0, sizeof(rawbuf));
946         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
947         if (rc) {
948                 fprintf(stderr, "error: %s: invalid ioctl\n",
949                         jt_cmdname(cmd));
950                 return rc;
951         }
952         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_ABORT_RECOVERY, buf);
953         if (rc < 0)
954                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(cmd),
955                         strerror(rc = errno));
956
957         return rc;
958 }
959
960 int jt_obd_abort_recovery(int argc, char **argv)
961 {
962         if (argc != 1)
963                 return CMD_HELP;
964
965         return obd_abort_recovery(argv[0], OBD_FLG_ABORT_RECOV_OST);
966 }
967
968 int jt_obd_abort_recovery_mdt(int argc, char **argv)
969 {
970         if (argc != 1)
971                 return CMD_HELP;
972
973         return obd_abort_recovery(argv[0], OBD_FLG_ABORT_RECOV_MDT);
974 }
975 #else /* ! HAVE_SERVER_SUPPROT */
976 int jt_obd_no_transno(int argc, char **argv)
977 {
978         if (argc != 1)
979                 return CMD_HELP;
980
981         fprintf(stderr, "error: %s: invalid ioctl\n",
982                 jt_cmdname(argv[0]));
983         return -EOPNOTSUPP;
984 }
985
986 int jt_obd_set_readonly(int argc, char **argv)
987 {
988         if (argc != 1)
989                 return CMD_HELP;
990
991         fprintf(stderr, "error: %s: invalid ioctl\n",
992                 jt_cmdname(argv[0]));
993         return -EOPNOTSUPP;
994 }
995
996 int jt_obd_abort_recovery(int argc, char **argv)
997 {
998         if (argc != 1)
999                 return CMD_HELP;
1000
1001         fprintf(stderr, "error: %s: invalid ioctl\n",
1002                 jt_cmdname(argv[0]));
1003         return -EOPNOTSUPP;
1004 }
1005
1006 int jt_obd_abort_recovery_mdt(int argc, char **argv)
1007 {
1008         if (argc != 1)
1009                 return CMD_HELP;
1010
1011         fprintf(stderr, "error: %s: invalid ioctl\n",
1012                 jt_cmdname(argv[0]));
1013         return -EOPNOTSUPP;
1014 }
1015 #endif /* HAVE_SERVER_SUPPORT */
1016
1017 int jt_get_version(int argc, char **argv)
1018 {
1019         char version[128];
1020         int rc;
1021
1022         if (argc != 1)
1023                 return CMD_HELP;
1024
1025         rc = llapi_get_version_string(version, sizeof(version));
1026         if (rc)
1027                 printf("Lustre version: %s\n", LUSTRE_VERSION_STRING);
1028         else
1029                 printf("Lustre version: %s\n", version);
1030
1031         return 0;
1032 }
1033
1034 static void print_obd_line(char *s)
1035 {
1036         const char *param = "osc/%s/ost_conn_uuid";
1037         char buf[MAX_STRING_SIZE];
1038         char obd_name[MAX_OBD_NAME];
1039         FILE *fp = NULL;
1040         glob_t path;
1041         char *ptr;
1042 retry:
1043         /* obd device type is the first 3 characters of param name */
1044         snprintf(buf, sizeof(buf), " %%*d %%*s %.3s %%%zus %%*s %%*d ",
1045                  param, sizeof(obd_name) - 1);
1046         if (sscanf(s, buf, obd_name) == 0)
1047                 goto try_mdc;
1048         if (cfs_get_param_paths(&path, param, obd_name) != 0)
1049                 goto try_mdc;
1050         fp = fopen(path.gl_pathv[0], "r");
1051         if (!fp) {
1052                 /* need to free path data before retry */
1053                 cfs_free_param_data(&path);
1054 try_mdc:
1055                 if (param[0] == 'o') { /* failed with osc, try mdc */
1056                         param = "mdc/%s/mds_conn_uuid";
1057                         goto retry;
1058                 }
1059                 buf[0] = '\0';
1060                 goto fail_print;
1061         }
1062
1063         /* should not ignore fgets(3)'s return value */
1064         if (!fgets(buf, sizeof(buf), fp)) {
1065                 fprintf(stderr, "reading from %s: %s", buf, strerror(errno));
1066                 goto fail_close;
1067         }
1068
1069 fail_close:
1070         fclose(fp);
1071         cfs_free_param_data(&path);
1072
1073         /* trim trailing newlines */
1074         ptr = strrchr(buf, '\n');
1075         if (ptr)
1076                 *ptr = '\0';
1077 fail_print:
1078         ptr = strrchr(s, '\n');
1079         if (ptr)
1080                 *ptr = '\0';
1081         printf("%s%s%s\n", s, buf[0] ? " " : "", buf);
1082 }
1083
1084 /* get device list by ioctl */
1085 int jt_obd_list_ioctl(int argc, char **argv)
1086 {
1087         int rc, index;
1088         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1089         struct obd_ioctl_data *data = (struct obd_ioctl_data *)buf;
1090
1091         if (argc > 2)
1092                 return CMD_HELP;
1093         /* Just ignore a -t option.  Only supported with /proc. */
1094         else if (argc == 2 && strcmp(argv[1], "-t") != 0)
1095                 return CMD_HELP;
1096
1097         for (index = 0;; index++) {
1098                 memset(buf, 0, sizeof(rawbuf));
1099                 data->ioc_version = OBD_IOCTL_VERSION;
1100                 data->ioc_inllen1 =
1101                         sizeof(rawbuf) - __ALIGN_KERNEL(sizeof(*data), 8);
1102                 data->ioc_inlbuf1 = buf + __ALIGN_KERNEL(sizeof(*data), 8);
1103                 data->ioc_len = obd_ioctl_packlen(data);
1104                 data->ioc_count = index;
1105
1106                 rc = l_ioctl(OBD_DEV_ID, OBD_IOC_GETDEVICE, buf);
1107                 if (rc != 0)
1108                         break;
1109                 printf("%s\n", (char *)data->ioc_bulk);
1110         }
1111         if (rc != 0) {
1112                 if (errno == ENOENT)
1113                         /* no device or the last device */
1114                         rc = 0;
1115                 else
1116                         fprintf(stderr,
1117                                 "Error getting device list: %s: check dmesg\n",
1118                                 strerror(errno));
1119         }
1120         return rc;
1121 }
1122
1123 int jt_obd_list(int argc, char **argv)
1124 {
1125         char buf[MAX_STRING_SIZE];
1126         int print_obd = 0;
1127         glob_t path;
1128         FILE *fp;
1129
1130         if (argc > 2)
1131                 return CMD_HELP;
1132
1133         if (argc == 2) {
1134                 if (strcmp(argv[1], "-t") == 0)
1135                         print_obd = 1;
1136                 else
1137                         return CMD_HELP;
1138         }
1139
1140         if (cfs_get_param_paths(&path, "devices") ||
1141             !(fp = fopen(path.gl_pathv[0], "r"))) {
1142                 cfs_free_param_data(&path);
1143
1144                 return jt_obd_list_ioctl(argc, argv);
1145         }
1146
1147         while (fgets(buf, sizeof(buf), fp) != NULL)
1148                 if (print_obd)
1149                         print_obd_line(buf);
1150                 else
1151                         printf("%s", buf);
1152
1153         cfs_free_param_data(&path);
1154         fclose(fp);
1155         return 0;
1156 }
1157
1158 struct jt_fid_space {
1159         __u64   jt_seq;
1160         __u64   jt_id;
1161         int     jt_width;
1162 };
1163
1164 int jt_obd_alloc_fids(struct jt_fid_space *space, struct lu_fid *fid,
1165                       __u64 *count)
1166 {
1167         int rc;
1168
1169         if (space->jt_seq == 0 || space->jt_id == space->jt_width) {
1170                 struct obd_ioctl_data  data;
1171                 char rawbuf[MAX_IOC_BUFLEN];
1172                 char *buf = rawbuf;
1173                 __u64 seqnr;
1174                 int max_count;
1175
1176                 memset(&data, 0, sizeof(data));
1177                 data.ioc_dev = cur_device;
1178
1179                 data.ioc_pbuf1 = (char *)&seqnr;
1180                 data.ioc_plen1 = sizeof(seqnr);
1181
1182                 data.ioc_pbuf2 = (char *)&max_count;
1183                 data.ioc_plen2 = sizeof(max_count);
1184
1185                 memset(buf, 0, sizeof(rawbuf));
1186                 rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
1187                 if (rc) {
1188                         fprintf(stderr, "error: invalid ioctl rc = %d\n", rc);
1189                         return rc;
1190                 }
1191
1192                 rc = l_ioctl(OBD_DEV_ID, OBD_IOC_ECHO_ALLOC_SEQ, buf);
1193                 if (rc) {
1194                         fprintf(stderr, "ioctl error: rc = %d\n", rc);
1195                         return rc;
1196                 }
1197
1198                 space->jt_seq = *(__u64 *)data.ioc_pbuf1;
1199                 space->jt_width = *(int *)data.ioc_pbuf2;
1200                 space->jt_id = 1;
1201         }
1202         fid->f_seq = space->jt_seq;
1203         fid->f_oid = space->jt_id;
1204         fid->f_ver = 0;
1205
1206         space->jt_id = space->jt_id + *count;
1207         if (space->jt_id > space->jt_width)
1208                 space->jt_id = space->jt_width;
1209
1210         *count = space->jt_id - fid->f_oid;
1211         return 0;
1212 }
1213
1214 #define MD_STEP_COUNT 1000
1215 int jt_obd_md_common(int argc, char **argv, int cmd)
1216 {
1217         struct obd_ioctl_data  data;
1218         struct timeval         start;
1219         struct timeval         end_time;
1220         char rawbuf[MAX_IOC_BUFLEN];
1221         char *buf = rawbuf;
1222         int mode = 0000644;
1223         int create_mode;
1224         int rc = 0;
1225         char *parent_basedir = NULL;
1226         char dirname[4096];
1227         int parent_base_id = 0;
1228         int parent_count = 1;
1229         __u64 child_base_id = -1;
1230         int stripe_count = 0;
1231         int stripe_index = -1;
1232         int count = 0;
1233         char *end;
1234         __u64 seconds = 0;
1235         double diff;
1236         int c;
1237         __u64 total_count = 0;
1238         char *name = NULL;
1239         struct jt_fid_space fid_space = {0};
1240         int version = 0;
1241         struct option long_opts[] = {
1242         { .val = 'b',   .name = "child_base_id",
1243                                                 .has_arg = required_argument },
1244         { .val = 'c',   .name = "stripe_count",
1245                                                 .has_arg = required_argument },
1246         { .val = 'd',   .name = "parent_basedir",
1247                                                 .has_arg = required_argument },
1248         { .val = 'D',   .name = "parent_dircount",
1249                                                 .has_arg = required_argument },
1250         { .val = 'i',   .name = "stripe_index", .has_arg = required_argument },
1251         { .val = 'm',   .name = "mode",         .has_arg = required_argument },
1252         { .val = 'n',   .name = "count",        .has_arg = required_argument },
1253         { .val = 't',   .name = "time",         .has_arg = required_argument },
1254         { .val = 'v',   .name = "version",      .has_arg = no_argument },
1255         { .name = NULL } };
1256
1257         while ((c = getopt_long(argc, argv, "b:c:d:D:m:n:t:v",
1258                                 long_opts, NULL)) >= 0) {
1259                 switch (c) {
1260                 case 'b':
1261                         child_base_id = strtoull(optarg, &end, 0);
1262                         if (*end) {
1263                                 fprintf(stderr,
1264                                         "error: %s: bad child_base_id '%s'\n",
1265                                         jt_cmdname(argv[0]), optarg);
1266                                 return CMD_HELP;
1267                         }
1268                         break;
1269                 case 'c':
1270                         stripe_count = strtoul(optarg, &end, 0);
1271                         if (*end) {
1272                                 fprintf(stderr,
1273                                         "error: %s: bad stripe count '%s'\n",
1274                                         jt_cmdname(argv[0]), optarg);
1275                                 return CMD_HELP;
1276                         }
1277                         break;
1278                 case 'd':
1279                         parent_basedir = optarg;
1280                         break;
1281                 case 'D':
1282                         parent_count = strtoul(optarg, &end, 0);
1283                         if (*end) {
1284                                 fprintf(stderr,
1285                                         "error: %s: bad parent count '%s'\n",
1286                                         jt_cmdname(argv[0]), optarg);
1287                                 return CMD_HELP;
1288                         }
1289                         break;
1290                 case 'i':
1291                         stripe_index = strtoul(optarg, &end, 0);
1292                         if (*end) {
1293                                 fprintf(stderr,
1294                                         "error: %s: bad stripe index '%s'\n",
1295                                         jt_cmdname(argv[0]), optarg);
1296                                 return CMD_HELP;
1297                         }
1298                         break;
1299                 case 'm':
1300                         mode = strtoul(optarg, &end, 0);
1301                         if (*end) {
1302                                 fprintf(stderr, "error: %s: bad mode '%s'\n",
1303                                         jt_cmdname(argv[0]), optarg);
1304                                 return CMD_HELP;
1305                         }
1306                         break;
1307                 case 'n':
1308                         total_count = strtoul(optarg, &end, 0);
1309                         if (*end || total_count == 0) {
1310                                 fprintf(stderr, "%s: bad child count '%s'\n",
1311                                         jt_cmdname(argv[0]), optarg);
1312                                 return CMD_HELP;
1313                         }
1314                         break;
1315                 case 't':
1316                         seconds = strtoull(optarg, &end, 0);
1317                         if (*end) {
1318                                 fprintf(stderr, "error: %s: seconds '%s'\n",
1319                                         jt_cmdname(argv[0]), optarg);
1320                                 return CMD_HELP;
1321                         }
1322                         break;
1323                 case 'v':
1324                         version = 1;
1325                         break;
1326                 default:
1327                         fprintf(stderr,
1328                                 "error: %s: option '%s' unrecognized\n",
1329                                 argv[0], argv[optind - 1]);
1330                         return CMD_HELP;
1331                 }
1332         }
1333
1334         memset(&data, 0, sizeof(data));
1335         data.ioc_dev = cur_device;
1336         if (child_base_id == -1) {
1337                 if (optind >= argc)
1338                         return CMD_HELP;
1339                 name = argv[optind];
1340                 total_count = 1;
1341         } else {
1342                 if (optind < argc) {
1343                         fprintf(stderr,
1344                                 "child_base_id and name can not specified at the same time\n");
1345                         return CMD_HELP;
1346                 }
1347         }
1348
1349         if (stripe_count == 0 && stripe_index != -1) {
1350                 fprintf(stderr,
1351                         "If stripe_count is 0, stripe_index can not be specified\n");
1352                 return CMD_HELP;
1353         }
1354
1355         if (total_count == 0 && seconds == 0) {
1356                 fprintf(stderr, "count or seconds needs to be indicated\n");
1357                 return CMD_HELP;
1358         }
1359
1360         if (parent_count <= 0) {
1361                 fprintf(stderr, "parent count must < 0\n");
1362                 return CMD_HELP;
1363         }
1364
1365 #ifdef MAX_THREADS
1366         if (thread) {
1367                 shmem_lock();
1368                 /* threads interleave */
1369                 if (parent_base_id != -1)
1370                         parent_base_id += (thread - 1) % parent_count;
1371
1372                 if (child_base_id != -1)
1373                         child_base_id +=  (thread - 1) *
1374                                 (MAX_BASE_ID / nthreads);
1375
1376                 shmem_start_time_locked();
1377                 shmem_unlock();
1378         }
1379 #endif
1380         /*
1381          * If parent directory is not specified, try to get the directory
1382          * from name
1383          */
1384         if (!parent_basedir) {
1385                 char *last_lash;
1386
1387                 if (!name) {
1388                         fprintf(stderr,
1389                                 "parent_basedir or name must be indicated!\n");
1390                         return CMD_HELP;
1391                 }
1392                 /*Get directory and name from name*/
1393                 last_lash = strrchr(name, '/');
1394                 if (!last_lash || name[0] != '/') {
1395                         fprintf(stderr, "Can not locate %s\n", name);
1396                         return CMD_HELP;
1397                 }
1398
1399                 if (last_lash == name) {
1400                         sprintf(dirname, "%s", "/");
1401                         name++;
1402                 } else {
1403                         int namelen = (unsigned long)last_lash -
1404                                       (unsigned long)name + 1;
1405                         snprintf(dirname, namelen, "%s", name);
1406                         name = last_lash + 1;
1407                 }
1408
1409                 data.ioc_pbuf1 = dirname;
1410                 data.ioc_plen1 = strlen(dirname);
1411
1412                 data.ioc_pbuf2 = name;
1413                 data.ioc_plen2 = strlen(name);
1414         } else {
1415                 if (name) {
1416                         data.ioc_pbuf2 = name;
1417                         data.ioc_plen2 = strlen(name);
1418                 }
1419                 if (parent_base_id > 0)
1420                         sprintf(dirname, "%s%d", parent_basedir,
1421                                 parent_base_id);
1422                 else
1423                         sprintf(dirname, "%s", parent_basedir);
1424                 data.ioc_pbuf1 = dirname;
1425                 data.ioc_plen1 = strlen(dirname);
1426         }
1427
1428         if (cmd == ECHO_MD_MKDIR || cmd == ECHO_MD_RMDIR)
1429                 create_mode = S_IFDIR;
1430         else
1431                 create_mode = S_IFREG;
1432
1433         data.ioc_obdo1.o_mode = mode | S_IFDIR;
1434         data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLTYPE | OBD_MD_FLMODE |
1435                                  OBD_MD_FLFLAGS | OBD_MD_FLGROUP;
1436         data.ioc_command = cmd;
1437
1438         gettimeofday(&start, NULL);
1439         while (shmem_running()) {
1440                 struct lu_fid fid = { 0 };
1441
1442                 if (child_base_id != -1)
1443                         data.ioc_obdo2.o_oi.oi.oi_id = child_base_id;
1444                 data.ioc_obdo2.o_mode = mode | create_mode;
1445                 data.ioc_obdo2.o_valid = OBD_MD_FLID | OBD_MD_FLTYPE |
1446                                          OBD_MD_FLMODE | OBD_MD_FLFLAGS |
1447                                          OBD_MD_FLGROUP;
1448                 data.ioc_obdo2.o_misc = stripe_count;
1449                 data.ioc_obdo2.o_stripe_idx = stripe_index;
1450
1451                 if (total_count > 0) {
1452                         if ((total_count - count) > MD_STEP_COUNT)
1453                                 data.ioc_count = MD_STEP_COUNT;
1454                         else
1455                                 data.ioc_count = total_count - count;
1456                 } else {
1457                         data.ioc_count = MD_STEP_COUNT;
1458                 }
1459
1460                 if (cmd == ECHO_MD_CREATE || cmd == ECHO_MD_MKDIR) {
1461                         /*Allocate fids for the create */
1462                         rc = jt_obd_alloc_fids(&fid_space, &fid,
1463                                                &data.ioc_count);
1464                         if (rc) {
1465                                 fprintf(stderr, "Allocate fids error %d.\n",
1466                                         rc);
1467                                 return rc;
1468                         }
1469                         data.ioc_obdo1.o_oi.oi_fid = fid;
1470                 }
1471
1472                 child_base_id += data.ioc_count;
1473                 count += data.ioc_count;
1474
1475                 memset(buf, 0, sizeof(rawbuf));
1476                 rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
1477                 if (rc) {
1478                         fprintf(stderr, "error: %s: invalid ioctl %d\n",
1479                                 jt_cmdname(argv[0]), rc);
1480                         return rc;
1481                 }
1482
1483                 rc = l_ioctl(OBD_DEV_ID, OBD_IOC_ECHO_MD, buf);
1484                 if (rc) {
1485                         fprintf(stderr, "error: %s: %s\n",
1486                                 jt_cmdname(argv[0]), strerror(rc = errno));
1487                         return rc;
1488                 }
1489                 shmem_bump(data.ioc_count);
1490
1491                 gettimeofday(&end_time, NULL);
1492                 diff = difftime(&end_time, &start);
1493                 if (seconds > 0 && (__u64)diff > seconds)
1494                         break;
1495
1496                 if (count >= total_count && total_count > 0)
1497                         break;
1498         }
1499
1500         if (count > 0 && version) {
1501                 gettimeofday(&end_time, NULL);
1502                 diff = difftime(&end_time, &start);
1503                 printf("%s: %d in %.3fs (%.3f /s): %s",
1504                        jt_cmdname(argv[0]), count, diff,
1505                        (double)count / diff, ctime(&end_time.tv_sec));
1506         }
1507
1508 #ifdef MAX_THREADS
1509         if (thread) {
1510                 shmem_lock();
1511                 shmem_end_time_locked();
1512                 shmem_unlock();
1513         }
1514 #endif
1515         return rc;
1516 }
1517
1518 int jt_obd_test_create(int argc, char **argv)
1519 {
1520         return jt_obd_md_common(argc, argv, ECHO_MD_CREATE);
1521 }
1522
1523 int jt_obd_test_mkdir(int argc, char **argv)
1524 {
1525         return jt_obd_md_common(argc, argv, ECHO_MD_MKDIR);
1526 }
1527
1528 int jt_obd_test_destroy(int argc, char **argv)
1529 {
1530         return jt_obd_md_common(argc, argv, ECHO_MD_DESTROY);
1531 }
1532
1533 int jt_obd_test_rmdir(int argc, char **argv)
1534 {
1535         return jt_obd_md_common(argc, argv, ECHO_MD_RMDIR);
1536 }
1537
1538 int jt_obd_test_lookup(int argc, char **argv)
1539 {
1540         return jt_obd_md_common(argc, argv, ECHO_MD_LOOKUP);
1541 }
1542
1543 int jt_obd_test_setxattr(int argc, char **argv)
1544 {
1545         return jt_obd_md_common(argc, argv, ECHO_MD_SETATTR);
1546 }
1547
1548 int jt_obd_test_md_getattr(int argc, char **argv)
1549 {
1550         return jt_obd_md_common(argc, argv, ECHO_MD_GETATTR);
1551 }
1552
1553 int jt_obd_create(int argc, char **argv)
1554 {
1555         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1556         struct obd_ioctl_data data;
1557         struct timeval next_time;
1558         __u64 count = 1, next_count, base_id = 1;
1559         int verbose = 1, mode = 0100644, rc = 0, i;
1560         char *end;
1561
1562         memset(&data, 0, sizeof(data));
1563         data.ioc_dev = cur_device;
1564         if (argc < 2 || argc > 4)
1565                 return CMD_HELP;
1566
1567         count = strtoull(argv[1], &end, 0);
1568         if (*end) {
1569                 fprintf(stderr, "error: %s: invalid iteration count '%s'\n",
1570                         jt_cmdname(argv[0]), argv[1]);
1571                 return CMD_HELP;
1572         }
1573
1574         if (argc > 2) {
1575                 mode = strtoul(argv[2], &end, 0);
1576                 if (*end) {
1577                         fprintf(stderr, "error: %s: invalid mode '%s'\n",
1578                                 jt_cmdname(argv[0]), argv[2]);
1579                         return CMD_HELP;
1580                 }
1581                 if (!(mode & S_IFMT))
1582                         mode |= S_IFREG;
1583         }
1584
1585         if (argc > 3) {
1586                 verbose = get_verbose(argv[0], argv[3]);
1587                 if (verbose == BAD_VERBOSE)
1588                         return CMD_HELP;
1589         }
1590
1591         printf("%s: %jd objects\n", jt_cmdname(argv[0]), (uintmax_t)count);
1592         gettimeofday(&next_time, NULL);
1593         next_time.tv_sec -= verbose;
1594
1595         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1596         for (i = 1, next_count = verbose; i <= count && shmem_running(); i++) {
1597                 /*
1598                  * base_id is 1 so we don't need to worry about it being
1599                  * greater than OBIF_MAX_OID
1600                  */
1601                 data.ioc_obdo1.o_oi.oi_fid.f_oid = base_id;
1602                 data.ioc_obdo1.o_mode = mode;
1603                 data.ioc_obdo1.o_uid = 0;
1604                 data.ioc_obdo1.o_gid = 0;
1605                 data.ioc_obdo1.o_projid = 0;
1606                 data.ioc_obdo1.o_valid = OBD_MD_FLTYPE | OBD_MD_FLMODE |
1607                                          OBD_MD_FLID | OBD_MD_FLUID |
1608                                          OBD_MD_FLGID | OBD_MD_FLGROUP |
1609                                          OBD_MD_FLPROJID;
1610
1611                 memset(buf, 0, sizeof(rawbuf));
1612                 rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
1613                 if (rc) {
1614                         fprintf(stderr, "error: %s: invalid ioctl\n",
1615                                 jt_cmdname(argv[0]));
1616                         return rc;
1617                 }
1618                 rc = l_ioctl(OBD_DEV_ID, OBD_IOC_CREATE, buf);
1619                 llapi_ioctl_unpack(&data, buf, sizeof(rawbuf));
1620                 shmem_bump(1);
1621                 if (rc < 0) {
1622                         fprintf(stderr, "error: %s: #%d - %s\n",
1623                                 jt_cmdname(argv[0]), i, strerror(rc = errno));
1624                         break;
1625                 }
1626                 if (!(data.ioc_obdo1.o_valid & OBD_MD_FLID)) {
1627                         fprintf(stderr, "error: %s: oid not valid #%d:%#jx\n",
1628                                 jt_cmdname(argv[0]), i,
1629                                 (uintmax_t)data.ioc_obdo1.o_valid);
1630                         rc = EINVAL;
1631                         break;
1632                 }
1633
1634                 if (be_verbose(verbose, &next_time, i, &next_count, count))
1635                         printf("%s: #%d is object id %#jx\n",
1636                                jt_cmdname(argv[0]), i,
1637                                (uintmax_t)ostid_id(&data.ioc_obdo1.o_oi));
1638         }
1639
1640         return rc;
1641 }
1642
1643 int jt_obd_setattr(int argc, char **argv)
1644 {
1645         struct obd_ioctl_data data;
1646         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1647         __u64 objid;
1648         char *end;
1649         int mode;
1650         int rc;
1651
1652         memset(&data, 0, sizeof(data));
1653         data.ioc_dev = cur_device;
1654         if (argc != 2)
1655                 return CMD_HELP;
1656
1657         objid = strtoull(argv[1], &end, 0);
1658         if (*end) {
1659                 fprintf(stderr, "error: %s: objid '%s' is not a number\n",
1660                         jt_cmdname(argv[0]), argv[1]);
1661                 return CMD_HELP;
1662         }
1663
1664         if (objid >= OBIF_MAX_OID) {
1665                 fprintf(stderr, "error: %s: invalid objid '%s'\n",
1666                         jt_cmdname(argv[0]), argv[1]);
1667                 return CMD_HELP;
1668         }
1669
1670         mode = strtoul(argv[2], &end, 0);
1671         if (*end) {
1672                 fprintf(stderr, "error: %s: invalid mode '%s'\n",
1673                         jt_cmdname(argv[0]), argv[2]);
1674                 return CMD_HELP;
1675         }
1676
1677         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1678         data.ioc_obdo1.o_mode = S_IFREG | mode;
1679         data.ioc_obdo1.o_oi.oi_fid.f_oid = objid;
1680         data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLTYPE | OBD_MD_FLMODE;
1681
1682         memset(buf, 0, sizeof(rawbuf));
1683         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
1684         if (rc) {
1685                 fprintf(stderr, "error: %s: invalid ioctl\n",
1686                         jt_cmdname(argv[0]));
1687                 return rc;
1688         }
1689         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_SETATTR, buf);
1690         if (rc < 0)
1691                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
1692                         strerror(rc = errno));
1693
1694         return rc;
1695 }
1696
1697 int jt_obd_test_setattr(int argc, char **argv)
1698 {
1699         struct obd_ioctl_data data;
1700         struct timeval start, next_time;
1701         __u64 i, count, next_count;
1702         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1703         int verbose = 1;
1704         __u64 objid = 3;
1705         char *end;
1706         int rc = 0;
1707
1708         if (argc < 2 || argc > 4)
1709                 return CMD_HELP;
1710
1711         memset(&data, 0, sizeof(data));
1712         data.ioc_dev = cur_device;
1713         count = strtoull(argv[1], &end, 0);
1714         if (*end) {
1715                 fprintf(stderr, "error: %s: invalid iteration count '%s'\n",
1716                         jt_cmdname(argv[0]), argv[1]);
1717                 return CMD_HELP;
1718         }
1719
1720         if (argc >= 3) {
1721                 verbose = get_verbose(argv[0], argv[2]);
1722                 if (verbose == BAD_VERBOSE)
1723                         return CMD_HELP;
1724         }
1725
1726         if (argc >= 4) {
1727                 if (argv[3][0] == 't') {
1728                         objid = strtoull(argv[3] + 1, &end, 0);
1729                         if (thread)
1730                                 objid += thread - 1;
1731                 } else {
1732                         objid = strtoull(argv[3], &end, 0);
1733                 }
1734                 if (*end) {
1735                         fprintf(stderr, "error: %s: invalid objid '%s'\n",
1736                                 jt_cmdname(argv[0]), argv[3]);
1737                         return CMD_HELP;
1738                 }
1739         }
1740
1741         gettimeofday(&start, NULL);
1742         next_time.tv_sec = start.tv_sec - verbose;
1743         next_time.tv_usec = start.tv_usec;
1744         if (verbose != 0)
1745                 printf("%s: setting %jd attrs (objid %#jx): %s",
1746                        jt_cmdname(argv[0]), (uintmax_t)count,
1747                        (uintmax_t)objid, ctime(&start.tv_sec));
1748
1749         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1750         for (i = 1, next_count = verbose; i <= count && shmem_running(); i++) {
1751                 if (objid >= OBIF_MAX_OID) {
1752                         fprintf(stderr, "errr: %s: invalid objid '%llu'\n",
1753                                 jt_cmdname(argv[0]), (unsigned long long)objid);
1754                         return -E2BIG;
1755                 }
1756
1757                 data.ioc_obdo1.o_oi.oi_fid.f_oid = objid;
1758                 data.ioc_obdo1.o_mode = S_IFREG;
1759                 data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLTYPE |
1760                                          OBD_MD_FLMODE;
1761                 memset(buf, 0, sizeof(rawbuf));
1762                 rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
1763                 if (rc) {
1764                         fprintf(stderr, "error: %s: invalid ioctl\n",
1765                                 jt_cmdname(argv[0]));
1766                         return rc;
1767                 }
1768                 rc = l_ioctl(OBD_DEV_ID, OBD_IOC_SETATTR, &data);
1769                 shmem_bump(1);
1770                 if (rc < 0) {
1771                         fprintf(stderr, "error: %s: #%jd - %d:%s\n",
1772                                 jt_cmdname(argv[0]), (uintmax_t)i,
1773                                 errno, strerror(rc = errno));
1774                         break;
1775                 }
1776                 if (be_verbose(verbose, &next_time, i, &next_count, count))
1777                         printf("%s: set attr #%jd\n",
1778                                jt_cmdname(argv[0]), (uintmax_t)i);
1779         }
1780
1781         if (!rc) {
1782                 struct timeval end;
1783                 double diff;
1784
1785                 gettimeofday(&end, NULL);
1786
1787                 diff = difftime(&end, &start);
1788
1789                 --i;
1790                 if (verbose != 0)
1791                         printf("%s: %jd attrs in %.3fs (%.3f attr/s): %s",
1792                                jt_cmdname(argv[0]), (uintmax_t)i, diff,
1793                                i / diff, ctime(&end.tv_sec));
1794         }
1795         return rc;
1796 }
1797
1798 int jt_obd_destroy(int argc, char **argv)
1799 {
1800         struct obd_ioctl_data data;
1801         struct timeval next_time;
1802         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1803         __u64 count = 1, next_count;
1804         int verbose = 1;
1805         __u64 id;
1806         char *end;
1807         int rc = 0, i;
1808
1809         memset(&data, 0, sizeof(data));
1810         data.ioc_dev = cur_device;
1811         if (argc < 2 || argc > 4)
1812                 return CMD_HELP;
1813
1814         errno = 0;
1815         id = strtoull(argv[1], &end, 0);
1816         if (*end || id == 0 || errno != 0) {
1817                 fprintf(stderr, "error: %s: invalid objid '%s'\n",
1818                         jt_cmdname(argv[0]), argv[1]);
1819                 return CMD_HELP;
1820         }
1821         if (argc > 2) {
1822                 count = strtoull(argv[2], &end, 0);
1823                 if (*end) {
1824                         fprintf(stderr,
1825                                 "error: %s: invalid iteration count '%s'\n",
1826                                 jt_cmdname(argv[0]), argv[2]);
1827                         return CMD_HELP;
1828                 }
1829         }
1830
1831         if (argc > 3) {
1832                 verbose = get_verbose(argv[0], argv[3]);
1833                 if (verbose == BAD_VERBOSE)
1834                         return CMD_HELP;
1835         }
1836
1837         printf("%s: %jd objects\n", jt_cmdname(argv[0]), (uintmax_t)count);
1838         gettimeofday(&next_time, NULL);
1839         next_time.tv_sec -= verbose;
1840
1841         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1842         for (i = 1, next_count = verbose; i <= count && shmem_running();
1843              i++, id++) {
1844                 if (id >= OBIF_MAX_OID) {
1845                         fprintf(stderr, "errr: %s: invalid objid '%llu'\n",
1846                                 jt_cmdname(argv[0]), (unsigned long long)id);
1847                         return -E2BIG;
1848                 }
1849
1850                 data.ioc_obdo1.o_oi.oi_fid.f_oid = id;
1851                 data.ioc_obdo1.o_mode = S_IFREG | 0644;
1852                 data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLMODE;
1853
1854                 memset(buf, 0, sizeof(rawbuf));
1855                 rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
1856                 if (rc) {
1857                         fprintf(stderr, "error: %s: invalid ioctl\n",
1858                                 jt_cmdname(argv[0]));
1859                         return rc;
1860                 }
1861                 rc = l_ioctl(OBD_DEV_ID, OBD_IOC_DESTROY, buf);
1862                 llapi_ioctl_unpack(&data, buf, sizeof(rawbuf));
1863                 shmem_bump(1);
1864                 if (rc < 0) {
1865                         fprintf(stderr, "error: %s: objid %#jx: %s\n",
1866                                 jt_cmdname(argv[0]), (uintmax_t)id,
1867                                 strerror(rc = errno));
1868                         break;
1869                 }
1870
1871                 if (be_verbose(verbose, &next_time, i, &next_count, count))
1872                         printf("%s: #%d is object id %#jx\n",
1873                                jt_cmdname(argv[0]), i, (uintmax_t)id);
1874         }
1875
1876         return rc;
1877 }
1878
1879 static int jt_str_to_ost_id(const char *str, struct ost_id *oi)
1880 {
1881         __u64 oid;
1882         char *end;
1883
1884         oid = strtoull(str, &end, 0);
1885         if (*end == '\0') {
1886                 /* If str is a single number then assume old echo
1887                  * client usage. */
1888                 if (oid >= OBIF_MAX_OID)
1889                         return -EINVAL;
1890
1891                 ostid_set_seq_echo(oi);
1892                 oi->oi_fid.f_oid = oid;
1893                 return 0;
1894         }
1895
1896         return llapi_fid_parse(str, &oi->oi_fid, NULL);
1897 }
1898
1899 int jt_obd_getattr(int argc, char **argv)
1900 {
1901         struct obd_ioctl_data data;
1902         struct obdo *oa;
1903         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1904         int rc;
1905
1906         if (argc != 2)
1907                 return CMD_HELP;
1908
1909         memset(&data, 0, sizeof(data));
1910         data.ioc_dev = cur_device;
1911         oa = &data.ioc_obdo1;
1912
1913         rc = jt_str_to_ost_id(argv[1], &oa->o_oi);
1914         if (rc < 0) {
1915                 fprintf(stderr, "error: %s: invalid objid of FID '%s'\n",
1916                         jt_cmdname(argv[0]), argv[1]);
1917                 return CMD_HELP;
1918         }
1919
1920         oa->o_valid = OBD_MD_FLID | OBD_MD_FLGROUP;
1921
1922         memset(buf, 0, sizeof(rawbuf));
1923         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
1924         if (rc) {
1925                 fprintf(stderr, "error: %s: invalid ioctl\n",
1926                         jt_cmdname(argv[0]));
1927                 return rc;
1928         }
1929         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_GETATTR, buf);
1930         llapi_ioctl_unpack(&data, buf, sizeof(rawbuf));
1931         if (rc) {
1932                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
1933                         strerror(rc = errno));
1934                 return rc;
1935         }
1936
1937 #define OP4(bits, name, format, value)                                  \
1938         do {                                                            \
1939                 if ((oa->o_valid & (bits)) == (bits))                   \
1940                         printf("%s: "format"\n", (name), value);        \
1941         } while (0)
1942
1943 #define OPM(bits, member, format)                                       \
1944         OP4(bits, #member, format, (uintmax_t)(oa->o_ ## member))
1945
1946 #define OPO(bits, member) OPM(bits, member, "%#jo")
1947 #define OPU(bits, member) OPM(bits, member, "%ju")
1948 #define OPX(bits, member) OPM(bits, member, "%#jx")
1949
1950         OPX(0, valid);
1951         OPX(OBD_MD_FLID | OBD_MD_FLGROUP, oi.oi.oi_id);
1952         OPX(OBD_MD_FLID | OBD_MD_FLGROUP, oi.oi.oi_seq);
1953         OP4(OBD_MD_FLID | OBD_MD_FLGROUP, "oi.oi_fid", DFID, PFID(&oa->o_oi.oi_fid));
1954         OPU(OBD_MD_FLATIME, atime);
1955         OPU(OBD_MD_FLMTIME, mtime);
1956         OPU(OBD_MD_FLCTIME, ctime);
1957         OPU(OBD_MD_FLSIZE, size);
1958         OPU(OBD_MD_FLBLOCKS, blocks);
1959         OPU(OBD_MD_FLBLKSZ, blksize);
1960         OPO(OBD_MD_FLMODE | OBD_MD_FLTYPE, mode);
1961         OPU(OBD_MD_FLUID, uid);
1962         OPU(OBD_MD_FLGID, gid);
1963         OPU(OBD_MD_FLFLAGS, flags);
1964         OPU(OBD_MD_FLNLINK, nlink);
1965         OPX(OBD_MD_FLPARENT | OBD_MD_FLFID, parent_seq);
1966         OPX(OBD_MD_FLPARENT | OBD_MD_FLFID, parent_oid);
1967         OPX(OBD_MD_FLPARENT | OBD_MD_FLFID, parent_ver);
1968         OPU(OBD_MD_LAYOUT_VERSION, layout_version);
1969         OPU(OBD_MD_FLGRANT, grant);
1970         OPU(OBD_MD_FLPROJID, projid);
1971         OPU(OBD_MD_FLDATAVERSION, data_version);
1972 #undef OP4
1973 #undef OPM
1974 #undef OPO
1975 #undef OPU
1976 #undef OPX
1977
1978         return 0;
1979 }
1980
1981 int jt_obd_test_getattr(int argc, char **argv)
1982 {
1983         struct obd_ioctl_data data;
1984         struct timeval start, next_time;
1985         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1986         __u64 i, count, next_count;
1987         int verbose = 1;
1988         __u64 objid = 3;
1989         char *end;
1990         int rc = 0;
1991
1992         if (argc < 2 || argc > 4)
1993                 return CMD_HELP;
1994
1995         memset(&data, 0, sizeof(data));
1996         data.ioc_dev = cur_device;
1997         count = strtoull(argv[1], &end, 0);
1998         if (*end) {
1999                 fprintf(stderr, "error: %s: invalid iteration count '%s'\n",
2000                         jt_cmdname(argv[0]), argv[1]);
2001                 return CMD_HELP;
2002         }
2003
2004         if (argc >= 3) {
2005                 verbose = get_verbose(argv[0], argv[2]);
2006                 if (verbose == BAD_VERBOSE)
2007                         return CMD_HELP;
2008         }
2009
2010         if (argc >= 4) {
2011                 if (argv[3][0] == 't') {
2012                         objid = strtoull(argv[3] + 1, &end, 0);
2013                         if (thread)
2014                                 objid += thread - 1;
2015                 } else {
2016                         objid = strtoull(argv[3], &end, 0);
2017                 }
2018                 if (*end) {
2019                         fprintf(stderr, "error: %s: invalid objid '%s'\n",
2020                                 jt_cmdname(argv[0]), argv[3]);
2021                         return CMD_HELP;
2022                 }
2023         }
2024
2025         gettimeofday(&start, NULL);
2026         next_time.tv_sec = start.tv_sec - verbose;
2027         next_time.tv_usec = start.tv_usec;
2028         if (verbose != 0)
2029                 printf("%s: getting %jd attrs (objid %#jx): %s",
2030                        jt_cmdname(argv[0]), (uintmax_t)count,
2031                        (uintmax_t)objid, ctime(&start.tv_sec));
2032
2033         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
2034         for (i = 1, next_count = verbose; i <= count && shmem_running(); i++) {
2035                 if (objid >= OBIF_MAX_OID) {
2036                         fprintf(stderr, "errr: %s: invalid objid '%llu'\n",
2037                                 jt_cmdname(argv[0]), (unsigned long long)objid);
2038                         return -E2BIG;
2039                 }
2040
2041                 data.ioc_obdo1.o_oi.oi_fid.f_oid = objid;
2042                 data.ioc_obdo1.o_mode = S_IFREG;
2043                 data.ioc_obdo1.o_valid = 0xffffffff;
2044                 memset(buf, 0, sizeof(rawbuf));
2045                 rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
2046                 if (rc) {
2047                         fprintf(stderr, "error: %s: invalid ioctl\n",
2048                                 jt_cmdname(argv[0]));
2049                         return rc;
2050                 }
2051                 rc = l_ioctl(OBD_DEV_ID, OBD_IOC_GETATTR, &data);
2052                 shmem_bump(1);
2053                 if (rc < 0) {
2054                         fprintf(stderr, "error: %s: #%jd - %d:%s\n",
2055                                 jt_cmdname(argv[0]), (uintmax_t)i,
2056                                 errno, strerror(rc = errno));
2057                         break;
2058                 }
2059                 if (be_verbose(verbose, &next_time, i, &next_count, count))
2060                         printf("%s: got attr #%jd\n",
2061                                jt_cmdname(argv[0]), (uintmax_t)i);
2062         }
2063
2064         if (!rc) {
2065                 struct timeval end;
2066                 double diff;
2067
2068                 gettimeofday(&end, NULL);
2069
2070                 diff = difftime(&end, &start);
2071
2072                 --i;
2073                 if (verbose != 0)
2074                         printf("%s: %jd attrs in %.3fs (%.3f attr/s): %s",
2075                                jt_cmdname(argv[0]), (uintmax_t)i, diff,
2076                                i / diff, ctime(&end.tv_sec));
2077         }
2078
2079         return rc;
2080 }
2081
2082 /*
2083  * test_brw <cnt>                                               count
2084  *      <r|w[r(repeat)x(noverify)]>                             mode
2085  *      <q|v|#(print interval)>                                 verbosity
2086  *      <npages[+offset]>                                       blocksize
2087  *      <[[<interleave_threads>]t(inc obj by thread#)]obj>      object
2088  *      [p|g<args>]                                             batch
2089  */
2090 int jt_obd_test_brw(int argc, char **argv)
2091 {
2092         struct obd_ioctl_data data;
2093         struct timeval start, next_time;
2094         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2095         __u64 count, next_count, len, stride, thr_offset = 0, objid = 3;
2096         int write = 0, verbose = 1, cmd, i, rc = 0, pages = 1;
2097         int offset_pages = 0;
2098         long n;
2099         int repeat_offset = 0;
2100         unsigned long long ull;
2101         int  nthr_per_obj = 0;
2102         int  verify = 1;
2103         int  obj_idx = 0;
2104         char *end;
2105
2106         if (argc < 2 || argc > 7) {
2107                 fprintf(stderr, "error: %s: bad number of arguments: %d\n",
2108                         jt_cmdname(argv[0]), argc);
2109                 return CMD_HELP;
2110         }
2111
2112         count = strtoull(argv[1], &end, 0);
2113         if (*end) {
2114                 fprintf(stderr, "error: %s: bad iteration count '%s'\n",
2115                         jt_cmdname(argv[0]), argv[1]);
2116                 return CMD_HELP;
2117         }
2118
2119         if (argc >= 3) {
2120                 if (argv[2][0] == 'w' || argv[2][0] == '1')
2121                         write = 1;
2122                 /* else it's a read */
2123
2124                 if (argv[2][0] != 0)
2125                         for (i = 1; argv[2][i] != 0; i++)
2126                                 switch (argv[2][i]) {
2127                                 case 'r':
2128                                         repeat_offset = 1;
2129                                         break;
2130
2131                                 case 'x':
2132                                         verify = 0;
2133                                         break;
2134
2135                                 default:
2136                                         fprintf(stderr,
2137                                                 "Can't parse cmd '%s'\n",
2138                                                 argv[2]);
2139                                         return CMD_HELP;
2140                                 }
2141         }
2142
2143         if (argc >= 4) {
2144                 verbose = get_verbose(argv[0], argv[3]);
2145                 if (verbose == BAD_VERBOSE)
2146                         return CMD_HELP;
2147         }
2148
2149         if (argc >= 5) {
2150                 pages = strtoul(argv[4], &end, 0);
2151
2152                 if (*end == '+')
2153                         offset_pages = strtoul(end + 1, &end, 0);
2154
2155                 if (*end != 0 || offset_pages < 0 || offset_pages >= pages) {
2156                         fprintf(stderr, "error: %s: bad npages[+offset] parameter '%s'\n",
2157                                 jt_cmdname(argv[0]), argv[4]);
2158                         return CMD_HELP;
2159                 }
2160         }
2161
2162         if (argc >= 6) {
2163                 if (thread && (n = strtol(argv[5], &end, 0)) > 0 &&
2164                     *end == 't' && (ull = strtoull(end + 1, &end, 0)) > 0 &&
2165                     *end == 0) {
2166                         nthr_per_obj = n;
2167                         objid = ull;
2168                 } else if (thread && argv[5][0] == 't') {
2169                         nthr_per_obj = 1;
2170                         objid = strtoull(argv[5] + 1, &end, 0);
2171                 } else {
2172                         nthr_per_obj = 0;
2173                         objid = strtoull(argv[5], &end, 0);
2174                 }
2175                 if (*end) {
2176                         fprintf(stderr, "error: %s: bad objid '%s'\n",
2177                                 jt_cmdname(argv[0]), argv[5]);
2178                         return CMD_HELP;
2179                 }
2180         }
2181
2182         memset(&data, 0, sizeof(data));
2183         data.ioc_dev = cur_device;
2184
2185         /*
2186          * communicate the 'type' of brw test and batching to echo_client.
2187          * don't start.  we'd love to refactor this lctl->echo_client
2188          * interface
2189          */
2190         data.ioc_pbuf1 = (void *)1;
2191         data.ioc_plen1 = 1;
2192
2193         if (argc >= 7) {
2194                 switch (argv[6][0]) {
2195                 case 'g': /* plug and unplug */
2196                         data.ioc_pbuf1 = (void *)2;
2197                         data.ioc_plen1 = strtoull(argv[6] + 1, &end, 0);
2198                         break;
2199                 case 'p': /* prep and commit */
2200                         data.ioc_pbuf1 = (void *)3;
2201                         data.ioc_plen1 = strtoull(argv[6] + 1, &end, 0);
2202                         break;
2203                 default:
2204                         fprintf(stderr,
2205                                 "error: %s: batching '%s' needs to specify 'p' or 'g'\n",
2206                                 jt_cmdname(argv[0]), argv[6]);
2207                         return CMD_HELP;
2208                 }
2209
2210                 if (*end) {
2211                         fprintf(stderr, "error: %s: bad batching '%s'\n",
2212                                 jt_cmdname(argv[0]), argv[6]);
2213                         return CMD_HELP;
2214                 }
2215                 data.ioc_plen1 *= getpagesize();
2216         }
2217
2218         len = pages * getpagesize();
2219         thr_offset = offset_pages * getpagesize();
2220         stride = len;
2221
2222 #ifdef MAX_THREADS
2223         if (thread) {
2224                 shmem_lock();
2225                 if (nthr_per_obj != 0) {
2226                         /* threads interleave */
2227                         obj_idx = (thread - 1) / nthr_per_obj;
2228                         objid += obj_idx;
2229                         stride *= nthr_per_obj;
2230                         if ((thread - 1) % nthr_per_obj == 0) {
2231                                 shared_data->body.offsets[obj_idx] =
2232                                         stride + thr_offset;
2233                         }
2234                         thr_offset += ((thread - 1) % nthr_per_obj) * len;
2235                 } else {
2236                         /* threads disjoint */
2237                         thr_offset += (thread - 1) * len;
2238                 }
2239
2240                 shmem_start_time_locked();
2241                 shmem_unlock();
2242         }
2243 #endif
2244
2245         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
2246         if (objid >= OBIF_MAX_OID) {
2247                 fprintf(stderr, "errr: %s: invalid objid '%llu'\n",
2248                         jt_cmdname(argv[0]), (unsigned long long)objid);
2249                 return -E2BIG;
2250         }
2251
2252         data.ioc_obdo1.o_oi.oi_fid.f_oid = objid;
2253         data.ioc_obdo1.o_mode = S_IFREG;
2254         data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLTYPE | OBD_MD_FLMODE |
2255                                  OBD_MD_FLFLAGS | OBD_MD_FLGROUP;
2256         data.ioc_obdo1.o_flags = (verify ? OBD_FL_DEBUG_CHECK : 0);
2257         data.ioc_count = len;
2258         data.ioc_offset = (repeat_offset ? 0 : thr_offset);
2259
2260         gettimeofday(&start, NULL);
2261         next_time.tv_sec = start.tv_sec - verbose;
2262         next_time.tv_usec = start.tv_usec;
2263
2264         if (verbose != 0)
2265                 printf("%s: %s %jux%d pages (obj %#jx, off %ju): %s",
2266                        jt_cmdname(argv[0]), write ? "writing" : "reading",
2267                        (uintmax_t)count, pages, (uintmax_t)objid,
2268                        (uintmax_t)data.ioc_offset, ctime(&start.tv_sec));
2269
2270         cmd = write ? OBD_IOC_BRW_WRITE : OBD_IOC_BRW_READ;
2271         for (i = 1, next_count = verbose; i <= count && shmem_running(); i++) {
2272                 data.ioc_obdo1.o_valid &= ~(OBD_MD_FLBLOCKS | OBD_MD_FLGRANT);
2273                 memset(buf, 0, sizeof(rawbuf));
2274                 rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
2275                 if (rc) {
2276                         fprintf(stderr, "error: %s: invalid ioctl\n",
2277                                 jt_cmdname(argv[0]));
2278                         return rc;
2279                 }
2280                 rc = l_ioctl(OBD_DEV_ID, cmd, buf);
2281                 shmem_bump(1);
2282                 if (rc) {
2283                         fprintf(stderr, "error: %s: #%d - %s on %s\n",
2284                                 jt_cmdname(argv[0]), i, strerror(rc = errno),
2285                                 write ? "write" : "read");
2286                         break;
2287                 } else if (be_verbose(verbose, &next_time, i,
2288                                       &next_count, count)) {
2289                         shmem_lock();
2290                         printf("%s: %s number %d @ %jd:%ju for %d\n",
2291                                jt_cmdname(argv[0]), write ? "write" : "read", i,
2292                                (uintmax_t)ostid_id(&data.ioc_obdo1.o_oi),
2293                                (uintmax_t)data.ioc_offset,
2294                                (int)(pages * getpagesize()));
2295                         shmem_unlock();
2296                 }
2297
2298                 if (!repeat_offset) {
2299 #ifdef MAX_THREADS
2300                         if (stride == len) {
2301                                 data.ioc_offset += stride;
2302                         } else if (i < count) {
2303                                 shmem_lock();
2304                                 data.ioc_offset =
2305                                         shared_data->body.offsets[obj_idx];
2306                                 shared_data->body.offsets[obj_idx] += len;
2307                                 shmem_unlock();
2308                         }
2309 #else
2310                         data.ioc_offset += len;
2311                         obj_idx = 0; /* avoids an unused var warning */
2312 #endif
2313                 }
2314         }
2315
2316         if (!rc) {
2317                 struct timeval end;
2318                 double diff;
2319
2320                 gettimeofday(&end, NULL);
2321
2322                 diff = difftime(&end, &start);
2323
2324                 --i;
2325                 if (verbose != 0)
2326                         printf("%s: %s %dx%d pages in %.3fs (%.3f MB/s): %s",
2327                                jt_cmdname(argv[0]), write ? "wrote" : "read",
2328                                i, pages, diff,
2329                                ((double)i * pages * getpagesize()) /
2330                                (diff * 1048576.0), ctime(&end.tv_sec));
2331         }
2332
2333 #ifdef MAX_THREADS
2334         if (thread) {
2335                 shmem_lock();
2336                 shmem_end_time_locked();
2337                 shmem_unlock();
2338         }
2339 #endif
2340         return rc;
2341 }
2342
2343 static int do_activate(int argc, char **argv, int flag)
2344 {
2345         struct obd_ioctl_data data;
2346         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2347         int rc;
2348
2349         memset(&data, 0, sizeof(data));
2350         data.ioc_dev = cur_device;
2351         if (argc != 1)
2352                 return CMD_HELP;
2353
2354         /* reuse offset for 'active' */
2355         data.ioc_offset = flag;
2356
2357         memset(buf, 0, sizeof(rawbuf));
2358         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
2359         if (rc) {
2360                 fprintf(stderr, "error: %s: invalid ioctl\n",
2361                         jt_cmdname(argv[0]));
2362                 return rc;
2363         }
2364         rc = l_ioctl(OBD_DEV_ID, IOC_OSC_SET_ACTIVE, buf);
2365         if (rc)
2366                 fprintf(stderr, "error: %s: failed: %s\n",
2367                         jt_cmdname(argv[0]), strerror(rc = errno));
2368
2369         return rc;
2370 }
2371
2372 /**
2373  * Replace nids for given device.
2374  * lctl replace_nids <devicename> <nid1>[,nid2,nid3]
2375  * Command should be started on MGS server.
2376  * Only MGS server should be started (command execution
2377  * returns error in another cases). Command mount
2378  * -t lustre <MDT partition> -o nosvc <mount point>
2379  * can be used for that.
2380  *
2381  * llogs for MDTs and clients are processed. All
2382  * records copied as is except add_uuid and setup. This records
2383  * are skipped and recorded with new nids and uuid.
2384  *
2385  * \see mgs_replace_nids
2386  * \see mgs_replace_log
2387  * \see mgs_replace_nids_handler
2388  */
2389 int jt_replace_nids(int argc, char **argv)
2390 {
2391         int rc;
2392         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2393         struct obd_ioctl_data data;
2394
2395         memset(&data, 0, sizeof(data));
2396         data.ioc_dev = get_mgs_device();
2397         if (argc != 3)
2398                 return CMD_HELP;
2399
2400         data.ioc_inllen1 = strlen(argv[1]) + 1;
2401         data.ioc_inlbuf1 = argv[1];
2402
2403         data.ioc_inllen2 = strlen(argv[2]) + 1;
2404         data.ioc_inlbuf2 = argv[2];
2405         memset(buf, 0, sizeof(rawbuf));
2406         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
2407         if (rc) {
2408                 fprintf(stderr, "error: %s: invalid ioctl\n",
2409                         jt_cmdname(argv[0]));
2410                 return rc;
2411         }
2412
2413         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_REPLACE_NIDS, buf);
2414         if (rc < 0) {
2415                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
2416                         strerror(rc = errno));
2417         }
2418
2419         return rc;
2420 }
2421
2422 int jt_obd_deactivate(int argc, char **argv)
2423 {
2424         return do_activate(argc, argv, 0);
2425 }
2426
2427 int jt_obd_activate(int argc, char **argv)
2428 {
2429         return do_activate(argc, argv, 1);
2430 }
2431
2432 int jt_obd_recover(int argc, char **argv)
2433 {
2434         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2435         struct obd_ioctl_data data;
2436         int rc;
2437
2438         memset(&data, 0, sizeof(data));
2439         data.ioc_dev = cur_device;
2440         if (argc > 2)
2441                 return CMD_HELP;
2442
2443         if (argc == 2) {
2444                 data.ioc_inllen1 = strlen(argv[1]) + 1;
2445                 data.ioc_inlbuf1 = argv[1];
2446         }
2447
2448         memset(buf, 0, sizeof(rawbuf));
2449         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
2450         if (rc) {
2451                 fprintf(stderr, "error: %s: invalid ioctl\n",
2452                         jt_cmdname(argv[0]));
2453                 return rc;
2454         }
2455         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_CLIENT_RECOVER, buf);
2456         if (rc < 0) {
2457                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
2458                         strerror(rc = errno));
2459         }
2460
2461         return rc;
2462 }
2463
2464 int jt_obd_mdc_lookup(int argc, char **argv)
2465 {
2466         struct obd_ioctl_data data;
2467         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2468         char *parent, *child;
2469         int rc, fd, verbose = 1;
2470
2471         if (argc < 3 || argc > 4)
2472                 return CMD_HELP;
2473
2474         parent = argv[1];
2475         child = argv[2];
2476         if (argc == 4)
2477                 verbose = get_verbose(argv[0], argv[3]);
2478
2479         memset(&data, 0, sizeof(data));
2480         data.ioc_dev = cur_device;
2481
2482         data.ioc_inllen1 = strlen(child) + 1;
2483         data.ioc_inlbuf1 = child;
2484
2485         memset(buf, 0, sizeof(rawbuf));
2486         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
2487         if (rc) {
2488                 fprintf(stderr, "error: %s: invalid ioctl\n",
2489                         jt_cmdname(argv[0]));
2490                 return rc;
2491         }
2492
2493         fd = open(parent, O_RDONLY);
2494         if (fd < 0) {
2495                 fprintf(stderr, "open \"%s\" failed: %s\n", parent,
2496                         strerror(errno));
2497                 return -1;
2498         }
2499
2500         rc = ioctl(fd, IOC_MDC_LOOKUP, buf);
2501         if (rc < 0) {
2502                 fprintf(stderr, "error: %s: ioctl error: %s\n",
2503                         jt_cmdname(argv[0]), strerror(rc = errno));
2504         }
2505         close(fd);
2506
2507         if (verbose) {
2508                 rc = llapi_ioctl_unpack(&data, buf, sizeof(rawbuf));
2509                 if (rc) {
2510                         fprintf(stderr, "error: %s: invalid reply\n",
2511                                 jt_cmdname(argv[0]));
2512                         return rc;
2513                 }
2514                 printf("%s: mode %o uid %d gid %d\n",
2515                        child, data.ioc_obdo1.o_mode, data.ioc_obdo1.o_uid,
2516                        data.ioc_obdo1.o_gid);
2517         }
2518
2519         return rc;
2520 }
2521
2522 #ifdef HAVE_SERVER_SUPPORT
2523 /**
2524  * Clear config logs for given device or filesystem.
2525  * lctl clear_conf <devicename|fsname>
2526  * Command has to be run on MGS node having MGS device mounted with -o
2527  * nosvc.
2528  *
2529  * Configuration logs for filesystem or one particular log is
2530  * processed. New log is created, original log is read, its records
2531  * marked SKIP do not get copied to new log. Others are copied as-is.
2532  * Original file is renamed to log.${time}.bak.
2533  *
2534  * \see mgs_clear_configs
2535  * \see mgs_replace_log
2536  * \see mgs_clear_config_handler
2537  **/
2538 int jt_lcfg_clear(int argc, char **argv)
2539 {
2540         int rc;
2541         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2542         struct obd_ioctl_data data;
2543
2544         memset(&data, 0, sizeof(data));
2545         data.ioc_dev = get_mgs_device();
2546         if (argc != 2)
2547                 return CMD_HELP;
2548
2549         data.ioc_inllen1 = strlen(argv[1]) + 1;
2550         data.ioc_inlbuf1 = argv[1];
2551
2552         memset(buf, 0, sizeof(rawbuf));
2553         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
2554         if (rc) {
2555                 fprintf(stderr, "error: %s: invalid ioctl\n",
2556                         jt_cmdname(argv[0]));
2557                 return rc;
2558         }
2559
2560         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_CLEAR_CONFIGS, buf);
2561         if (rc < 0) {
2562                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
2563                         strerror(rc = errno));
2564         }
2565
2566         return rc;
2567 }
2568
2569 int jt_lcfg_fork(int argc, char **argv)
2570 {
2571         struct obd_ioctl_data data;
2572         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2573         int rc;
2574
2575         if (argc != 3)
2576                 return CMD_HELP;
2577
2578         memset(&data, 0, sizeof(data));
2579         data.ioc_dev = get_mgs_device();
2580         data.ioc_inllen1 = strlen(argv[1]) + 1;
2581         data.ioc_inlbuf1 = argv[1];
2582         data.ioc_inllen2 = strlen(argv[2]) + 1;
2583         data.ioc_inlbuf2 = argv[2];
2584
2585         memset(buf, 0, sizeof(rawbuf));
2586         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
2587         if (rc) {
2588                 fprintf(stderr, "error: %s: invalid ioctl\n",
2589                         jt_cmdname(argv[0]));
2590                 return rc;
2591         }
2592
2593         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LCFG_FORK, buf);
2594         if (rc < 0)
2595                 fprintf(stderr, "error: %s: OBD_IOC_LCFG_FORK failed: %s\n",
2596                         jt_cmdname(argv[0]), strerror(errno));
2597
2598         return rc;
2599 }
2600
2601 int jt_lcfg_erase(int argc, char **argv)
2602 {
2603         struct obd_ioctl_data data;
2604         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2605         int rc;
2606
2607         if (argc == 3) {
2608                 if (strncmp(argv[2], "-q", strlen("-q")) != 0 &&
2609                     strncmp(argv[2], "--quiet", strlen("--quiet")) != 0)
2610                         return CMD_HELP;
2611         } else if (argc != 2) {
2612                 return CMD_HELP;
2613         }
2614
2615         memset(&data, 0, sizeof(data));
2616         data.ioc_dev = get_mgs_device();
2617         data.ioc_inllen1 = strlen(argv[1]) + 1;
2618         data.ioc_inlbuf1 = argv[1];
2619
2620         memset(buf, 0, sizeof(rawbuf));
2621         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
2622         if (rc) {
2623                 fprintf(stderr, "error: %s: invalid ioctl\n",
2624                         jt_cmdname(argv[0]));
2625                 return rc;
2626         }
2627
2628         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LCFG_ERASE, buf);
2629         if (rc < 0)
2630                 fprintf(stderr, "error: %s: OBD_IOC_LCFG_ERASE failed: %s\n",
2631                         jt_cmdname(argv[0]), strerror(errno));
2632
2633         return rc;
2634 }
2635 #else /* !HAVE_SERVER_SUPPORT */
2636 int jt_lcfg_clear(int argc, char **argv)
2637 {
2638         if (argc != 2)
2639                 return CMD_HELP;
2640
2641         fprintf(stderr, "error: %s: invalid ioctl\n",
2642                 jt_cmdname(argv[0]));
2643         return -EOPNOTSUPP;
2644 }
2645
2646 int jt_lcfg_fork(int argc, char **argv)
2647 {
2648         if (argc != 3)
2649                 return CMD_HELP;
2650
2651         fprintf(stderr, "error: %s: invalid ioctl\n",
2652                 jt_cmdname(argv[0]));
2653         return -EOPNOTSUPP;
2654 }
2655
2656 int jt_lcfg_erase(int argc, char **argv)
2657 {
2658         if (argc != 3)
2659                 return CMD_HELP;
2660
2661         fprintf(stderr, "error: %s: invalid ioctl\n",
2662                 jt_cmdname(argv[0]));
2663         return -EOPNOTSUPP;
2664 }
2665 #endif /* HAVE_SERVER_SUPPORT */
2666
2667 enum llog_default_dev_op {
2668         LLOG_DFLT_MGS_SET = 0,
2669         LLOG_DFLT_DEV_RESET
2670 };
2671
2672 static int llog_default_device(enum llog_default_dev_op op)
2673 {
2674         int rc = 0;
2675         static int dflt_dev = -1;
2676
2677         if (op == LLOG_DFLT_MGS_SET && (cur_device == -1)) {
2678                 char mgs[] = "$MGS";
2679
2680                 rc = do_device("llog_default_device", mgs);
2681                 dflt_dev = cur_device;
2682
2683         } else if (op == LLOG_DFLT_DEV_RESET && (dflt_dev != -1)) {
2684                 do_disconnect(NULL, 1);
2685                 dflt_dev = -1;
2686         }
2687
2688         return rc;
2689 }
2690
2691 int jt_llog_catlist(int argc, char **argv)
2692 {
2693         struct obd_ioctl_data data;
2694         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2695         char *tmp = NULL;
2696         int start = 0;
2697         int rc;
2698
2699         if (argc != 1)
2700                 return CMD_HELP;
2701
2702         if (llog_default_device(LLOG_DFLT_MGS_SET))
2703                 return CMD_INCOMPLETE;
2704
2705         do {
2706                 memset(&data, 0, sizeof(data));
2707                 data.ioc_dev = cur_device;
2708                 data.ioc_inllen1 = sizeof(rawbuf) -
2709                                    __ALIGN_KERNEL(sizeof(data), 8);
2710                 data.ioc_count = start;
2711                 memset(buf, 0, sizeof(rawbuf));
2712                 rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
2713                 if (rc) {
2714                         fprintf(stderr, "error: %s: invalid ioctl\n",
2715                                 jt_cmdname(argv[0]));
2716                         goto err;
2717                 }
2718                 rc = l_ioctl(OBD_DEV_ID, OBD_IOC_CATLOGLIST, buf);
2719                 if (rc < 0)
2720                         break;
2721                 tmp = ((struct obd_ioctl_data *)buf)->ioc_bulk;
2722                 if (strlen(tmp) > 0)
2723                         fprintf(stdout, "%s", tmp);
2724                 else
2725                         break;
2726                 start = ((struct obd_ioctl_data *)buf)->ioc_count;
2727         } while (start);
2728
2729         if (rc < 0)
2730                 fprintf(stderr, "OBD_IOC_CATLOGLIST failed: %s\n",
2731                         strerror(errno));
2732
2733 err:
2734         llog_default_device(LLOG_DFLT_DEV_RESET);
2735
2736         return rc;
2737 }
2738
2739 int jt_llog_info(int argc, char **argv)
2740 {
2741         const struct option long_opts[] = {
2742         /* Allow optional "--catalog" for compatibility with llog commands. */
2743         { .val = 'c',   .name = "catalog",      .has_arg = required_argument },
2744         { .val = 'h',   .name = "help",         .has_arg = no_argument },
2745         { .name = NULL } };
2746         struct obd_ioctl_data data = { 0 };
2747         char rawbuf[MAX_IOC_BUFLEN] = "", *buf = rawbuf;
2748         char *cmd = argv[0];
2749         char *catalog = NULL;
2750         int rc, c;
2751
2752         while ((c = getopt_long(argc, argv, "c:h", long_opts, NULL)) != -1) {
2753                 switch (c) {
2754                 case 'c':
2755                         catalog = optarg;
2756                         break;
2757                 case 'h':
2758                 default:
2759                         return CMD_HELP;
2760                 }
2761         }
2762         argc -= optind;
2763         argv += optind;
2764         /* support "logname" positional parameter */
2765         if (argc == 1) {
2766                 if (catalog) {
2767                         fprintf(stderr,
2768                                 "%s: catalog is set, unknown argument '%s'\n",
2769                                 cmd, optarg);
2770                         return CMD_HELP;
2771                 }
2772                 catalog = argv[0];
2773         } else if (!catalog || argc > 1) {
2774                 return CMD_HELP;
2775         }
2776
2777         /* Manage default device */
2778         if (llog_default_device(LLOG_DFLT_MGS_SET))
2779                 return CMD_INCOMPLETE;
2780
2781         data.ioc_dev = cur_device;
2782         data.ioc_inllen1 = strlen(catalog) + 1;
2783         data.ioc_inlbuf1 = catalog;
2784         data.ioc_inllen2 = sizeof(rawbuf) - __ALIGN_KERNEL(sizeof(data), 8) -
2785                            __ALIGN_KERNEL(data.ioc_inllen1, 8);
2786         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
2787         if (rc) {
2788                 fprintf(stderr, "%s: ioctl_pack failed for catalog '%s': %s\n",
2789                         jt_cmdname(cmd), catalog, strerror(-rc));
2790                 goto err;
2791         }
2792
2793         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_INFO, buf);
2794         if (rc == 0)
2795                 fprintf(stdout, "%s", ((struct obd_ioctl_data *)buf)->ioc_bulk);
2796         else
2797                 fprintf(stderr, "%s: OBD_IOC_LLOG_INFO failed: %s\n",
2798                         jt_cmdname(cmd), strerror(errno));
2799
2800 err:
2801         llog_default_device(LLOG_DFLT_DEV_RESET);
2802
2803         return rc;
2804 }
2805
2806 int jt_llog_print_cb(const char *record, void *private)
2807 {
2808         printf("%s\n", record);
2809
2810         return 0;
2811 }
2812
2813 static int
2814 llog_process_records(int (record_cb)(const char *record, void *private),
2815                      const char *record, void *private, bool reverse)
2816 {
2817         char *ptr = NULL;
2818         char *tmp = NULL;
2819         int rc = 0;
2820
2821         if (!reverse) {
2822                 do {
2823                         ptr = strchr(record, '\n');
2824                         if (ptr)
2825                                 *ptr = '\0';
2826                         rc = record_cb(record, private);
2827                         if (rc)
2828                                 goto out;
2829                         if (ptr)
2830                                 record = ptr + 1;
2831                 } while (ptr && *(ptr + 1));
2832         } else {
2833                 tmp = (char *)record;
2834
2835                 ptr = strrchr(record, '\n');
2836                 if (ptr)
2837                         *ptr = '\0';
2838                 else
2839                         goto out;
2840                 while ((ptr = strrchr(record, '\n'))) {
2841                         tmp = ptr + 1;
2842                         *ptr = '\0';
2843                         rc = record_cb(tmp, private);
2844                         if (rc)
2845                                 goto out;
2846                 };
2847                 rc = record_cb(record, private);
2848                 if (rc)
2849                         goto out;
2850         }
2851 out:
2852         return rc;
2853 }
2854
2855 /**
2856  * Iterate over llog records, typically YAML-formatted configuration logs
2857  *
2858  * \param logname[in]   name of llog file or FID
2859  * \param start[in]     first record to process
2860  * \param end[in]       last record to process (inclusive)
2861  * \param cb[in]        callback for records. Return -ve error, or +ve abort.
2862  * \param private[in,out] private data passed to the \a record_cb function
2863  * \param reverse[in]   print the llog records from the beginning or the end
2864  *
2865  * \retval              0 on success
2866  *                      others handled by the caller
2867  */
2868 int jt_llog_print_iter(char *logname, long start, long end,
2869                        int (record_cb)(const char *record, void *private),
2870                        void *private, bool reverse, bool raw)
2871 {
2872         struct obd_ioctl_data data = { 0 };
2873         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2874         char startbuf[16], endbuf[16];
2875         static long inc = sizeof(rawbuf) / 128;
2876         long rec;
2877         int rc = 0;
2878
2879         if (end == -1)
2880                 end = 0x7fffffff;
2881
2882         data.ioc_dev = cur_device;
2883         data.ioc_inlbuf1 = logname;
2884         data.ioc_inllen1 = strlen(logname) + 1;
2885
2886         /*
2887          * Estimate about 128 characters per configuration record.  Not all
2888          * records will be printed in any case, so they should easily fit.  If
2889          * not, the kernel will return -EOVERFLOW and ask for fewer records.
2890          *
2891          * We don't want to request records from the kernel one-at-a-time, as
2892          * it restarts the config llog iteration from the beginning, so we
2893          * fetch multiple records from the kernel per call and split locally.
2894          */
2895         for (rec = start; rec < end; rec += inc) {
2896                 char *record = ((struct obd_ioctl_data *)buf)->ioc_bulk;
2897
2898 retry:
2899                 snprintf(startbuf, sizeof(startbuf), "%lu", rec);
2900                 snprintf(endbuf, sizeof(endbuf), "%lu",
2901                          end < rec + inc - 1 ? end : rec + inc - 1);
2902
2903                 data.ioc_u32_1 = raw ? 1 : 0;
2904                 /* start and end record numbers are passed as ASCII digits */
2905                 data.ioc_inlbuf2 = startbuf;
2906                 data.ioc_inllen2 = strlen(startbuf) + 1;
2907                 data.ioc_inlbuf3 = endbuf;
2908                 data.ioc_inllen3 = strlen(endbuf) + 1;
2909
2910                 data.ioc_inllen4 = sizeof(rawbuf) -
2911                         __ALIGN_KERNEL(sizeof(data), 8) -
2912                         __ALIGN_KERNEL(data.ioc_inllen1, 8) -
2913                         __ALIGN_KERNEL(data.ioc_inllen2, 8) -
2914                         __ALIGN_KERNEL(data.ioc_inllen3, 8);
2915                 memset(buf, 0, sizeof(rawbuf));
2916                 rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
2917                 if (rc) {
2918                         fprintf(stderr, "%s: invalid ioctl data\n", logname);
2919                         goto out;
2920                 }
2921
2922                 rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_PRINT, buf);
2923                 if (rc == -EOVERFLOW && inc > 2) {
2924                         inc /= 2;
2925                         goto retry;
2926                 }
2927                 if (rc) {
2928                         fprintf(stderr, "%s: OBD_IOC_LLOG_PRINT failed: %s\n",
2929                                 logname, strerror(errno));
2930                         rc = -errno;
2931                         goto out;
2932                 }
2933
2934                 /* There is no "end of list" marker, record was not modified */
2935                 if (strcmp(record, logname) == 0)
2936                         break;
2937
2938                 rc = llog_process_records(record_cb, record, private, reverse);
2939                 if (rc)
2940                         goto out;
2941         }
2942
2943 out:
2944         return rc;
2945 }
2946
2947 static int llog_parse_catalog_options(int *argc, char ***argv, char **catalog,
2948                                       long *start, long *end, int *raw)
2949 {
2950         const struct option long_opts[] = {
2951         /* the --catalog option is not required, just for consistency */
2952         { .val = 'c',   .name = "catalog",      .has_arg = required_argument },
2953         { .val = 'e',   .name = "end",          .has_arg = required_argument },
2954         { .val = 'h',   .name = "help",         .has_arg = no_argument },
2955         { .val = 'r',   .name = "raw",          .has_arg = no_argument },
2956         { .val = 's',   .name = "start",        .has_arg = required_argument },
2957         { .name = NULL } };
2958         char *cmd = (*argv)[0];
2959         char *endp;
2960         int c;
2961
2962         if (!catalog || !start || !end)
2963                 return -EINVAL;
2964
2965         /* now process command line arguments*/
2966         while ((c = getopt_long(*argc, *argv, "c:e:hrs:",
2967                                 long_opts, NULL)) != -1) {
2968                 switch (c) {
2969                 case 'c':
2970                         *catalog = optarg;
2971                         break;
2972                 case 'e':
2973                         *end = strtol(optarg, &endp, 0);
2974                         if (*endp != '\0') {
2975                                 fprintf(stderr, "%s: bad end value '%s'\n",
2976                                         cmd, optarg);
2977                                 return CMD_HELP;
2978                         }
2979                         break;
2980                 case 'r':
2981                         if (!raw)
2982                                 return CMD_HELP;
2983                         *raw = 1;
2984                         break;
2985                 case 's':
2986                         *start = strtol(optarg, &endp, 0);
2987                         if (*endp != '\0') {
2988                                 fprintf(stderr, "%s: bad start value '%s'\n",
2989                                         cmd, optarg);
2990                                 return CMD_HELP;
2991                         }
2992                         break;
2993                 case 'h':
2994                 default:
2995                         return CMD_HELP;
2996                 }
2997         }
2998         *argc -= optind;
2999         *argv += optind;
3000
3001         /*
3002          * support old optional positional parameters only if they were
3003          * not already specified with named arguments: logname [start [end]]
3004          */
3005         if (*argc >= 1) {
3006                 if (*catalog) {
3007                         fprintf(stderr,
3008                                 "%s: logname is set, unknown argument '%s'\n",
3009                                 cmd, (*argv)[0]);
3010                         return CMD_HELP;
3011                 }
3012                 *catalog = (*argv)[0];
3013                 (*argc)--;
3014                 (*argv)++;
3015         }
3016
3017         if (*catalog == NULL) {
3018                 fprintf(stderr, "%s: no logname specified\n", cmd);
3019                 return CMD_HELP;
3020         }
3021
3022         if (*argc >= 1) {
3023                 if (*start != 1) {
3024                         fprintf(stderr,
3025                                 "%s: --start is set, unknown argument '%s'\n",
3026                                 cmd, (*argv)[0]);
3027                         return CMD_HELP;
3028                 }
3029
3030                 *start = strtol((*argv)[0], &endp, 0);
3031                 if (*endp != '\0') {
3032                         fprintf(stderr, "%s: bad start value '%s'\n",
3033                                 cmd, (*argv)[0]);
3034                         return CMD_HELP;
3035                 }
3036                 (*argc)--;
3037                 (*argv)++;
3038         }
3039         if (*argc >= 1) {
3040                 if (*end != -1) {
3041                         fprintf(stderr,
3042                                 "%s: --end is set, unknown argument '%s'\n",
3043                                 cmd, (*argv)[0]);
3044                         return CMD_HELP;
3045                 }
3046
3047                 *end = strtol((*argv)[0], &endp, 0);
3048                 if (*endp != '\0') {
3049                         fprintf(stderr, "%s: bad end value '%s'\n",
3050                                 cmd, (*argv)[0]);
3051                         return CMD_HELP;
3052                 }
3053                 (*argc)--;
3054                 (*argv)++;
3055         }
3056         if (*argc > 1) {
3057                 fprintf(stderr, "%s: unknown argument '%s'\n", cmd, (*argv)[0]);
3058                 return CMD_HELP;
3059         }
3060
3061         if (*end != -1 && *end < *start) {
3062                 fprintf(stderr, "%s: end '%lu' less than than start '%lu'\n",
3063                         cmd, *end, *start);
3064                 return CMD_HELP;
3065         }
3066
3067         return 0;
3068 }
3069
3070 int jt_llog_print(int argc, char **argv)
3071 {
3072         char *catalog = NULL;
3073         long start = 1, end = -1;
3074         int raw = 0;
3075         int rc;
3076
3077         rc = llog_parse_catalog_options(&argc, &argv, &catalog, &start, &end,
3078                                         &raw);
3079         if (rc)
3080                 return rc;
3081
3082         if (llog_default_device(LLOG_DFLT_MGS_SET))
3083                 return CMD_INCOMPLETE;
3084
3085         rc = jt_llog_print_iter(catalog, start, end, jt_llog_print_cb,
3086                                 NULL, false, !!raw);
3087
3088         llog_default_device(LLOG_DFLT_DEV_RESET);
3089
3090         return rc;
3091 }
3092
3093 /*
3094  * Parse catalog, log ID, and optionally a log index with either optional
3095  * arguments or positional arguments.  Only the initial catalog argument
3096  * may be positional with other optional arguments.
3097  *
3098  * The positional arguments option should eventually be phased out.
3099  */
3100 static int llog_parse_catalog_log_idx(int *argc, char ***argv, const char *opts,
3101                                       struct obd_ioctl_data *data)
3102 {
3103         const struct option long_opts[] = {
3104         /* the --catalog option is not required, just for consistency */
3105         { .val = 'c',   .name = "catalog",      .has_arg = required_argument },
3106         { .val = 'h',   .name = "help",         .has_arg = no_argument },
3107         { .val = 'i',   .name = "log_idx",      .has_arg = required_argument },
3108         { .val = 'l',   .name = "log_id",       .has_arg = required_argument },
3109         { .name = NULL } };
3110         int c;
3111
3112         /* sanity check */
3113         if (!data || *argc <= 1)
3114                 return -1;
3115
3116         data->ioc_dev = cur_device;
3117
3118         /* now process command line arguments*/
3119         while ((c = getopt_long(*argc, *argv, opts, long_opts, NULL)) != -1) {
3120                 switch (c) {
3121                 case 'c':
3122                         data->ioc_inllen1 = strlen(optarg) + 1;
3123                         data->ioc_inlbuf1 = optarg;
3124                         break;
3125                 case 'i':
3126                         data->ioc_inllen3 = strlen(optarg) + 1;
3127                         data->ioc_inlbuf3 = optarg;
3128                         break;
3129                 case 'l': /* The log_id option isn't currently needed for
3130                            * cancel as mdt_iocontrol() handles IOC_LLOG_CANCEL,
3131                            * but we may as well keep it for now.
3132                            */
3133                         data->ioc_inllen2 = strlen(optarg) + 1;
3134                         data->ioc_inlbuf2 = optarg;
3135                         break;
3136                 case 'h':
3137                 default:
3138                         return CMD_HELP;
3139                 }
3140         }
3141
3142         *argc -= optind;
3143         *argv += optind;
3144
3145         /* Allow catalog to be specified as first option without --catalog */
3146         if (!data->ioc_inlbuf1 && *argc > 0) {
3147                 data->ioc_inlbuf1 = (*argv)[0];
3148                 data->ioc_inllen1 = strlen((*argv)[0]) + 1;
3149                 (*argc)--;
3150                 (*argv)++;
3151         }
3152
3153         return 0;
3154 }
3155
3156 int jt_llog_cancel(int argc, char **argv)
3157 {
3158         struct obd_ioctl_data data = { 0 };
3159         char rawbuf[MAX_IOC_BUFLEN] = "", *buf = rawbuf;
3160         char *cmd = argv[0];
3161         int rc;
3162
3163         /* Manage default device */
3164         if (llog_default_device(LLOG_DFLT_MGS_SET))
3165                 return CMD_INCOMPLETE;
3166
3167         /* Parse catalog file (in inlbuf1) and named parameters */
3168         rc = llog_parse_catalog_log_idx(&argc, &argv, "c:hi:l:", &data);
3169
3170         /*
3171          * Handle old positional parameters if not using named parameters,
3172          * either "<catalog> <log_idx>" or "<catalog> <log_id> <log_idx>".
3173          * It was "inlbuf3 = log_idx", and "inlbuf2 = log_id" (ignored by
3174          * config log cancel), and shows why I hate positional parameters.
3175          */
3176         if (argc == 1) {
3177                 data.ioc_inllen3 = strlen(argv[0]) + 1;
3178                 data.ioc_inlbuf3 = argv[0];
3179         } else if (argc == 2) {
3180                 data.ioc_inllen2 = strlen(argv[0]) + 1;
3181                 data.ioc_inlbuf2 = argv[0];
3182                 data.ioc_inllen3 = strlen(argv[1]) + 1;
3183                 data.ioc_inlbuf3 = argv[1];
3184         }
3185
3186         if (!data.ioc_inlbuf1 || !data.ioc_inlbuf3) {
3187                 /* missing mandatory parameters */
3188                 rc = CMD_HELP;
3189                 goto err;
3190         }
3191
3192         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
3193         if (rc) {
3194                 fprintf(stderr, "%s: ioctl_pack for catalog '%s' failed: %s\n",
3195                         jt_cmdname(cmd), data.ioc_inlbuf1, strerror(-rc));
3196                 goto err;
3197         }
3198
3199         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_CANCEL, buf);
3200         if (rc)
3201                 fprintf(stderr, "%s: cancel catalog '%s:%s' failed: %s\n",
3202                         jt_cmdname(cmd), data.ioc_inlbuf1, data.ioc_inlbuf3,
3203                         strerror(errno));
3204
3205 err:
3206         llog_default_device(LLOG_DFLT_DEV_RESET);
3207         return rc;
3208 }
3209
3210 int jt_llog_check(int argc, char **argv)
3211 {
3212         struct obd_ioctl_data data = { 0 };
3213         char rawbuf[MAX_IOC_BUFLEN] = "", *buf = rawbuf;
3214         char *catalog = NULL;
3215         char startbuf[16], endbuf[16];
3216         long start = 1, end = -1;
3217         char *cmd = argv[0];
3218         int rc;
3219
3220         rc = llog_parse_catalog_options(&argc, &argv, &catalog, &start,
3221                                         &end, NULL);
3222         if (rc)
3223                 return rc;
3224
3225         if (llog_default_device(LLOG_DFLT_MGS_SET))
3226                 return CMD_INCOMPLETE;
3227
3228         if (end == -1)
3229                 end = 0x7fffffff;
3230
3231         data.ioc_dev = cur_device;
3232         data.ioc_inllen1 = strlen(catalog) + 1;
3233         data.ioc_inlbuf1 = catalog;
3234
3235         snprintf(startbuf, sizeof(startbuf), "%lu", start);
3236         snprintf(endbuf, sizeof(endbuf), "%lu", end);
3237         /* start and end record numbers are passed as ASCII digits */
3238         data.ioc_inllen2 = strlen(startbuf) + 1;
3239         data.ioc_inlbuf2 = startbuf;
3240         data.ioc_inllen3 = strlen(endbuf) + 1;
3241         data.ioc_inlbuf3 = endbuf;
3242
3243         data.ioc_inllen4 = sizeof(rawbuf) - __ALIGN_KERNEL(sizeof(data), 8) -
3244                            __ALIGN_KERNEL(data.ioc_inllen1, 8) -
3245                            __ALIGN_KERNEL(data.ioc_inllen2, 8) -
3246                            __ALIGN_KERNEL(data.ioc_inllen3, 8);
3247         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
3248         if (rc) {
3249                 fprintf(stderr, "%s: ioctl_pack failed for catalog '%s': %s\n",
3250                         jt_cmdname(cmd), data.ioc_inlbuf1, strerror(-rc));
3251                 goto err;
3252         }
3253
3254         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_CHECK, buf);
3255         if (rc == 0)
3256                 fprintf(stdout, "%s", ((struct obd_ioctl_data *)buf)->ioc_bulk);
3257         else
3258                 fprintf(stderr, "%s: OBD_IOC_LLOG_CHECK failed: %s\n",
3259                         jt_cmdname(cmd), strerror(errno));
3260 err:
3261         llog_default_device(LLOG_DFLT_DEV_RESET);
3262         return rc;
3263 }
3264
3265 int jt_llog_remove(int argc, char **argv)
3266 {
3267         struct obd_ioctl_data data = { 0 };
3268         char rawbuf[MAX_IOC_BUFLEN] = "", *buf = rawbuf;
3269         char *cmd = argv[0];
3270         int rc;
3271
3272         if (llog_default_device(LLOG_DFLT_MGS_SET))
3273                 return CMD_INCOMPLETE;
3274
3275         rc = llog_parse_catalog_log_idx(&argc, &argv, "c:hl:", &data);
3276         if (rc)
3277                 goto err;
3278
3279         if (argc == 1) {
3280                 if (data.ioc_inlbuf2) {
3281                         fprintf(stderr,
3282                                 "%s: --log_id is set, unknown argument '%s'\n",
3283                                 jt_cmdname(cmd), argv[0]);
3284                         rc = CMD_HELP;
3285                         goto err;
3286                 }
3287
3288                 data.ioc_inllen2 = strlen(argv[0]) + 1;
3289                 data.ioc_inlbuf2 = argv[0];
3290         }
3291
3292         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
3293         if (rc) {
3294                 fprintf(stderr, "%s: ioctl_pack for catalog '%s' failed: %s\n",
3295                         jt_cmdname(cmd), data.ioc_inlbuf1, strerror(-rc));
3296                 goto err;
3297         }
3298
3299         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_REMOVE, buf);
3300         if (rc)
3301                 fprintf(stderr, "%s: cancel catalog '%s:%s' failed: %s\n",
3302                         jt_cmdname(cmd), data.ioc_inlbuf1, data.ioc_inlbuf2,
3303                         strerror(-rc));
3304
3305 err:
3306         llog_default_device(LLOG_DFLT_DEV_RESET);
3307         return rc;
3308 }
3309
3310 static void signal_server(int sig)
3311 {
3312         if (sig == SIGINT) {
3313                 do_disconnect("sigint", 1);
3314                 exit(1);
3315         } else {
3316                 fprintf(stderr, "%s: got signal %d\n", jt_cmdname("sigint"),
3317                         sig);
3318         }
3319 }
3320
3321 int obd_initialize(int argc, char **argv)
3322 {
3323         if (shmem_setup() != 0)
3324                 return -1;
3325
3326         register_ioc_dev(OBD_DEV_ID, OBD_DEV_PATH);
3327
3328         return 0;
3329 }
3330
3331 void obd_finalize(int argc, char **argv)
3332 {
3333         struct sigaction sigact;
3334
3335         /* sigact initialization */
3336         sigact.sa_handler = signal_server;
3337         sigfillset(&sigact.sa_mask);
3338         sigact.sa_flags = SA_RESTART;
3339         /* coverity[uninit_use_in_call] */
3340         sigaction(SIGINT, &sigact, NULL);
3341
3342         shmem_cleanup();
3343         do_disconnect(argv[0], 1);
3344 }
3345
3346 /**
3347  * Get the index of the last llog record
3348  *
3349  * logid:            [0x3:0xa:0x0]:0
3350  * flags:            4 (plain)
3351  * records_count:    57
3352  * last_index:       57
3353  *
3354  * \param logname[in]   pointer to config log name
3355  *
3356  * \retval              > 0 on success
3357  *                      <= 0 on error
3358  */
3359 static long llog_last_index(char *logname)
3360 {
3361         struct obd_ioctl_data data = { 0 };
3362         char rawbuf[MAX_IOC_BUFLEN] = "", *buf = rawbuf;
3363         char *last_index;
3364         long rc;
3365
3366         data.ioc_dev = cur_device;
3367         data.ioc_inllen1 = strlen(logname) + 1;
3368         data.ioc_inlbuf1 = logname;
3369         data.ioc_inllen2 = sizeof(rawbuf) - __ALIGN_KERNEL(sizeof(data), 8) -
3370                            __ALIGN_KERNEL(data.ioc_inllen1, 8);
3371         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
3372         if (rc) {
3373                 fprintf(stderr, "%s: ioctl_pack failed for catalog '%s': %s\n",
3374                         __func__, logname, strerror(-rc));
3375                 return rc;
3376         }
3377
3378         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_INFO, buf);
3379         if (rc == 0) {
3380                 last_index = strstr(((struct obd_ioctl_data *)buf)->ioc_bulk,
3381                                     "last_index:");
3382                 return strtol(last_index + 11, NULL, 10);
3383         }
3384
3385         rc = -errno;
3386
3387         return rc;
3388 }
3389
3390 static char *get_llog_event_name(__u32 cmd)
3391 {
3392 #ifdef HAVE_SERVER_SUPPORT
3393         struct lcfg_type_data *data;
3394
3395         data = lcfg_cmd2data(cmd);
3396         if (data)
3397                 return data->ltd_name;
3398 #endif
3399         return NULL;
3400 }
3401
3402 static char *get_event_filter(__u32 cmd)
3403 {
3404         char *event_name;
3405         char *filter = NULL;
3406         int len;
3407
3408         event_name = get_llog_event_name(cmd);
3409         if (event_name) {
3410                 /* 9 bytes for "event: , " */
3411                 len = 9 + strlen(event_name);
3412                 filter = malloc(len + 1);
3413                 if (!filter)
3414                         return NULL;
3415                 memset(filter, 0, len + 1);
3416                 snprintf(filter, len, "event: %s, ", event_name);
3417                 return filter;
3418         }
3419
3420         return NULL;
3421 }
3422
3423 /**
3424  * Callback to search ostname in llog
3425  * - { index: 23, event: attach, device: lustre-OST0000-osc, type: osc,
3426  *     UUID: lustre-clilov_UUID }
3427  * - { index: 24, event: setup, device: lustre-OST0000-osc,
3428  *     UUID: lustre-OST0000_UUID, node: 192.168.0.120@tcp }
3429  * - { index: 25, event: add_osc, device: lustre-clilov,
3430  *     ost: lustre-OST0000_UUID, index: 0, gen: 1 }
3431  *
3432  * \param record[in]    pointer to llog record
3433  * \param data[in]      pointer to ostname
3434  *
3435  * \retval              1 if ostname is found
3436  *                      0 if ostname is not found
3437  *                      -ENOENT if ostname is deleted
3438  */
3439 static int llog_search_ost_cb(const char *record, void *data)
3440 {
3441         char *ostname = data;
3442         char ost_filter[MAX_STRING_SIZE] = {'\0'};
3443         char *add_osc, *del_osc, *setup, *cleanup;
3444
3445         add_osc = get_event_filter(LCFG_LOV_ADD_OBD);
3446         del_osc = get_event_filter(LCFG_LOV_DEL_OBD);
3447         setup = get_event_filter(LCFG_SETUP);
3448         cleanup = get_event_filter(LCFG_CLEANUP);
3449         if (!add_osc || !del_osc || !setup || !cleanup)
3450                 return -ENOMEM;
3451
3452         if (ostname && ostname[0])
3453                 snprintf(ost_filter, sizeof(ost_filter), " %s,", ostname);
3454
3455         if (strstr(record, ost_filter)) {
3456                 if (strstr(record, add_osc) || strstr(record, setup))
3457                         return 1;
3458                 if (strstr(record, del_osc) || strstr(record, cleanup))
3459                         return -ENOENT;
3460         }
3461
3462         free(add_osc);
3463         free(del_osc);
3464         free(setup);
3465         free(cleanup);
3466
3467         return 0;
3468 }
3469
3470 /**
3471  * Search ost in llog
3472  *
3473  * \param logname[in]           pointer to config log name
3474  * \param last_index[in]        the index of the last llog record
3475  * \param ostname[in]           pointer to ost name
3476  *
3477  * \retval                      1 if ostname is found
3478  *                              0 if ostname is not found
3479  */
3480 static int llog_search_ost(char *logname, long last_index, char *ostname)
3481 {
3482         long start, end, inc = MAX_IOC_BUFLEN / 128;
3483         int rc = 0;
3484
3485         for (end = last_index; end > 1; end -= inc) {
3486                 start = end - inc > 0 ? end - inc : 1;
3487                 rc = jt_llog_print_iter(logname, start, end, llog_search_ost_cb,
3488                                         ostname, true, false);
3489                 if (rc)
3490                         break;
3491         }
3492
3493         return (rc == 1 ? 1 : 0);
3494 }
3495
3496 struct llog_pool_data {
3497         char lpd_fsname[LUSTRE_MAXFSNAME + 1];
3498         char lpd_poolname[LOV_MAXPOOLNAME + 1];
3499         char lpd_ostname[MAX_OBD_NAME + 1];
3500         enum lcfg_command_type lpd_cmd_type;
3501         bool lpd_pool_exists;
3502         int lpd_ost_num;
3503 };
3504
3505 /**
3506  * Called for each formatted line in the config log (within range).
3507  *
3508  * - { index: 74, event: new_pool, device: tfs-clilov, fsname: tfs, pool: tmp }
3509  * - { index: 77, event: add_pool, device: tfs-clilov, fsname: tfs, pool: tmp,
3510  *     ost: tfs-OST0000_UUID }
3511  * - { index: 224, event: remove_pool, device: tfs-clilov, fsname: tfs,
3512  *     pool: tmp, ost: tfs-OST0003_UUID }
3513  * - { index: 227, event: del_pool, device: tfs-clilov, fsname: tfs, pool: tmp }
3514  *
3515  * \param record[in]    pointer to llog record
3516  * \param data[in]      pointer to llog_pool_data
3517  *
3518  * \retval              1 if pool or OST is found
3519  *                      0 if pool or OST is not found
3520  *                      -ENOENT if pool or OST is removed
3521  */
3522 static int llog_search_pool_cb(const char *record, void *data)
3523 {
3524         struct llog_pool_data *lpd = data;
3525         char pool_filter[MAX_STRING_SIZE] = "";
3526         char *new_pool, *del_pool, *add_pool, *rem_pool;
3527         char *found = NULL;
3528         int fs_pool_len = 0, rc = 0;
3529
3530         new_pool = get_event_filter(LCFG_POOL_NEW);
3531         del_pool = get_event_filter(LCFG_POOL_DEL);
3532         add_pool = get_event_filter(LCFG_POOL_ADD);
3533         rem_pool = get_event_filter(LCFG_POOL_REM);
3534         if (!new_pool || !del_pool || !add_pool || !rem_pool) {
3535                 rc = -ENOMEM;
3536                 goto out;
3537         }
3538
3539         fs_pool_len = 16 + strlen(lpd->lpd_fsname) + strlen(lpd->lpd_poolname);
3540         snprintf(pool_filter, fs_pool_len + 1, "fsname: %s, pool: %s",
3541                  lpd->lpd_fsname, lpd->lpd_poolname);
3542
3543         /* search poolname */
3544         found = strstr(record, pool_filter);
3545         if (found &&
3546             (found[fs_pool_len] == ' ' || found[fs_pool_len] == ',')) {
3547                 if (strstr(record, new_pool)) {
3548                         lpd->lpd_pool_exists = true;
3549                         rc = 1;
3550                         goto out;
3551                 }
3552                 if (strstr(record, del_pool)) {
3553                         lpd->lpd_pool_exists = false;
3554                         rc = -ENOENT;
3555                         goto out;
3556                 }
3557
3558                 if (lpd->lpd_cmd_type == LCFG_POOL_NEW ||
3559                     lpd->lpd_cmd_type == LCFG_POOL_DEL) {
3560                         /* In function mgs_pool_cmd(), a pool's pool_new/add
3561                          * record will be marked as "SKIP" if its pool_destroy/
3562                          * remove record is logged later. That means the "SKIP"
3563                          * record won't be printed here and thus lpd_ost_num
3564                          * doesn't need to be decreased as well.
3565                          */
3566                         if (strstr(record, add_pool))
3567                                 lpd->lpd_ost_num++;
3568                 } else if (lpd->lpd_ostname && lpd->lpd_ostname[0]) {
3569                         if (strstr(record, lpd->lpd_ostname)) {
3570                                 lpd->lpd_pool_exists = true;
3571                                 if (strstr(record, add_pool)) {
3572                                         lpd->lpd_ost_num = 1;
3573                                         rc = 1;
3574                                         goto out;
3575                                 }
3576                                 if (strstr(record, rem_pool)) {
3577                                         lpd->lpd_ost_num = 0;
3578                                         rc = -ENOENT;
3579                                         goto out;
3580                                 }
3581                         }
3582                 }
3583         }
3584 out:
3585         if (new_pool)
3586                 free(new_pool);
3587         if (del_pool)
3588                 free(del_pool);
3589         if (add_pool)
3590                 free(add_pool);
3591         if (rem_pool)
3592                 free(rem_pool);
3593
3594         return rc;
3595 }
3596
3597 /* Search pool and its ost in llog
3598  *
3599  * \param logname[in]           pointer to config log name
3600  * \param last_index[in]        the index of the last llog record
3601  * \param fsname[in]            pointer to filesystem name
3602  * \param poolname[in]          pointer pool name
3603  * \param ostname[in]           pointer to OST name(OSTnnnn-UUID)
3604  * \param cmd[in]               pool command type
3605  *
3606  * \retval                      < 0 on error
3607  *                              0 if pool is empty or OST is not found
3608  *                              1 if pool is not empty or OST is found
3609  */
3610 static int llog_search_pool(char *logname, long last_index, char *fsname,
3611                             char *poolname, char *ostname,
3612                             enum lcfg_command_type cmd)
3613 {
3614         struct llog_pool_data lpd;
3615         long start, end, inc = MAX_IOC_BUFLEN / 128;
3616         int rc = 0;
3617
3618         memset(&lpd, 0, sizeof(lpd));
3619         lpd.lpd_cmd_type = cmd;
3620         lpd.lpd_pool_exists = false;
3621         lpd.lpd_ost_num = 0;
3622         strncpy(lpd.lpd_fsname, fsname, sizeof(lpd.lpd_fsname) - 1);
3623         if (poolname && poolname[0])
3624                 strncpy(lpd.lpd_poolname, poolname,
3625                         sizeof(lpd.lpd_poolname) - 1);
3626         if (ostname && ostname[0])
3627                 strncpy(lpd.lpd_ostname, ostname, sizeof(lpd.lpd_ostname) - 1);
3628
3629         for (end = last_index; end > 1; end -= inc) {
3630                 start = end - inc > 0 ? end - inc : 1;
3631                 rc = jt_llog_print_iter(logname, start, end,
3632                                         llog_search_pool_cb, &lpd, true, false);
3633                 if (rc) {
3634                         if (rc == 1 && lpd.lpd_pool_exists)
3635                                 rc = lpd.lpd_ost_num ? 1 : 0;
3636                         else if (rc == -ENOENT && lpd.lpd_pool_exists &&
3637                                  !lpd.lpd_ost_num)
3638                                 rc = 0;
3639                         goto out;
3640                 }
3641         }
3642
3643         rc = -ENOENT;
3644 out:
3645         return rc;
3646 }
3647
3648 static bool combined_mgs_mds(char *fsname)
3649 {
3650         glob_t path;
3651         int rc;
3652
3653         rc = cfs_get_param_paths(&path, "mdt/%s-MDT0000", fsname);
3654         if (!rc)
3655                 cfs_free_param_data(&path);
3656
3657         if (get_mgs_device() > 0 && !rc)
3658                 return true;
3659
3660         return false;
3661 }
3662
3663 /*
3664  * if pool is NULL, search ostname in target_obd
3665  * if pool is not NULL:
3666  *  - if pool not found returns errno < 0
3667  *  - if ostname is NULL, returns 1 if pool is not empty and 0 if pool empty
3668  *  - if ostname is not NULL, returns 1 if OST is in pool and 0 if not
3669  */
3670 int lctl_search_ost(char *fsname, char *poolname, char *ostname,
3671                     enum lcfg_command_type cmd)
3672 {
3673         char logname[MAX_OBD_NAME] = {'\0'};
3674         long last_index;
3675
3676         if (fsname && fsname[0] == '\0')
3677                 fsname = NULL;
3678         if (!fsname)
3679                 return -EINVAL;
3680
3681         if (combined_mgs_mds(fsname))
3682                 return llapi_search_ost(fsname, poolname, ostname);
3683
3684         /* fetch the last_index of llog record */
3685         snprintf(logname, sizeof(logname), "%s-client", fsname);
3686         last_index = llog_last_index(logname);
3687         if (last_index < 0)
3688                 return last_index;
3689
3690         /* if pool is NULL, search ostname in target_obd */
3691         if (!poolname && ostname)
3692                 return llog_search_ost(logname, last_index, ostname);
3693
3694         return llog_search_pool(logname, last_index, fsname, poolname,
3695                                 ostname, cmd);
3696 }
3697
3698 static int check_pool_cmd(enum lcfg_command_type cmd, char *fsname,
3699                           char *poolname, char *ostname)
3700 {
3701         int rc;
3702
3703         rc = lctl_search_ost(fsname, poolname, ostname, cmd);
3704         if (rc < 0 && (cmd != LCFG_POOL_NEW)) {
3705                 fprintf(stderr, "Pool %s.%s not found\n",
3706                         fsname, poolname);
3707                 return rc;
3708         }
3709
3710         switch (cmd) {
3711         case LCFG_POOL_NEW: {
3712                 if (ostname)
3713                         return -EINVAL;
3714
3715                 if (rc >= 0) {
3716                         fprintf(stderr, "Pool %s.%s already exists\n",
3717                                 fsname, poolname);
3718                         return -EEXIST;
3719                 }
3720                 return 0;
3721         }
3722         case LCFG_POOL_DEL: {
3723                 if (ostname)
3724                         return -EINVAL;
3725
3726                 if (rc == 1) {
3727                         fprintf(stderr,
3728                                 "Pool %s.%s not empty, please remove all members\n",
3729                                 fsname, poolname);
3730                         return -ENOTEMPTY;
3731                 }
3732                 return 0;
3733         }
3734         case LCFG_POOL_ADD: {
3735                 if (rc == 1) {
3736                         fprintf(stderr, "OST %s is already in pool %s.%s\n",
3737                                 ostname, fsname, poolname);
3738                         return -EEXIST;
3739                 }
3740                 rc = lctl_search_ost(fsname, NULL, ostname, cmd);
3741                 if (rc == 0) {
3742                         fprintf(stderr, "OST %s is not part of the '%s' fs.\n",
3743                                 ostname, fsname);
3744                         return -ENOENT;
3745                 }
3746                 return 0;
3747         }
3748         case LCFG_POOL_REM: {
3749                 if (rc == 0) {
3750                         fprintf(stderr, "OST %s not found in pool %s.%s\n",
3751                                 ostname, fsname, poolname);
3752                         return -ENOENT;
3753                 }
3754                 return 0;
3755         }
3756         default:
3757                 break;
3758         } /* switch */
3759         return -EINVAL;
3760 }
3761
3762 /*
3763  * This check only verifies that the changes have been "pushed out" to
3764  * the client successfully.  This involves waiting for a config update,
3765  * and so may fail because of problems in that code or post-command
3766  * network loss. So reporting a warning is appropriate, but not a failure.
3767  */
3768 static int check_pool_cmd_result(enum lcfg_command_type cmd, char *fsname,
3769                                  char *poolname, char *ostname)
3770 {
3771         int cpt = 10;
3772         int rc = 0;
3773
3774         switch (cmd) {
3775         case LCFG_POOL_NEW: {
3776                 do {
3777                         rc = lctl_search_ost(fsname, poolname, NULL, cmd);
3778                         if (rc == -ENODEV)
3779                                 return rc;
3780                         if (rc < 0)
3781                                 sleep(1);
3782                         cpt--;
3783                 } while ((rc < 0) && (cpt > 0));
3784                 if (rc >= 0) {
3785                         fprintf(stderr, "Pool %s.%s created\n",
3786                                 fsname, poolname);
3787                         return 0;
3788                 }
3789
3790                 fprintf(stderr, "Warning, pool %s.%s not found\n", fsname,
3791                         poolname);
3792                 return -ENOENT;
3793         }
3794         case LCFG_POOL_DEL: {
3795                 do {
3796                         rc = lctl_search_ost(fsname, poolname, NULL, cmd);
3797                         if (rc == -ENODEV)
3798                                 return rc;
3799                         if (rc >= 0)
3800                                 sleep(1);
3801                         cpt--;
3802                 } while ((rc >= 0) && (cpt > 0));
3803                 if (rc < 0) {
3804                         fprintf(stderr, "Pool %s.%s destroyed\n",
3805                                 fsname, poolname);
3806                         return 0;
3807                 }
3808
3809                 fprintf(stderr, "Warning, pool %s.%s still found\n", fsname,
3810                         poolname);
3811                 return -EEXIST;
3812         }
3813         case LCFG_POOL_ADD: {
3814                 do {
3815                         rc = lctl_search_ost(fsname, poolname, ostname, cmd);
3816                         if (rc == -ENODEV)
3817                                 return rc;
3818                         if (rc != 1)
3819                                 sleep(1);
3820                         cpt--;
3821                 } while ((rc != 1) && (cpt > 0));
3822                 if (rc == 1) {
3823                         fprintf(stderr, "OST %s added to pool %s.%s\n",
3824                                 ostname, fsname, poolname);
3825                         return 0;
3826                 }
3827                 fprintf(stderr, "Warning, OST %s not found in pool %s.%s\n",
3828                         ostname, fsname, poolname);
3829                 return -ENOENT;
3830         }
3831         case LCFG_POOL_REM: {
3832                 do {
3833                         rc = lctl_search_ost(fsname, poolname, ostname, cmd);
3834                         if (rc == -ENODEV)
3835                                 return rc;
3836                         if (rc == 1)
3837                                 sleep(1);
3838                         cpt--;
3839                 } while ((rc == 1) && (cpt > 0));
3840                 if (rc != 1) {
3841                         fprintf(stderr, "OST %s removed from pool %s.%s\n",
3842                                 ostname, fsname, poolname);
3843                         return 0;
3844                 }
3845                 fprintf(stderr, "Warning, OST %s still found in pool %s.%s\n",
3846                         ostname, fsname, poolname);
3847                 return -EEXIST;
3848         }
3849         default:
3850                 break;
3851         }
3852         return -EINVAL;
3853 }
3854
3855 static int check_and_complete_ostname(char *fsname, char *ostname)
3856 {
3857         char *ptr;
3858         char real_ostname[MAX_OBD_NAME + 1];
3859         char i;
3860
3861         /* if OST name does not start with fsname, we add it */
3862         /* if not check if the fsname is the right one */
3863         ptr = strchr(ostname, '-');
3864         if (!ptr) {
3865                 sprintf(real_ostname, "%s-%s", fsname, ostname);
3866         } else if (strncmp(ostname, fsname, strlen(fsname)) != 0) {
3867                 fprintf(stderr, "%s does not start with fsname %s\n",
3868                         ostname, fsname);
3869                 return -EINVAL;
3870         } else {
3871                 if (strlen(ostname) > sizeof(real_ostname) - 1)
3872                         return -E2BIG;
3873
3874                 strncpy(real_ostname, ostname, sizeof(real_ostname));
3875         }
3876
3877         /* real_ostname is fsname-????? */
3878         ptr = real_ostname + strlen(fsname) + 1;
3879         if (strncmp(ptr, "OST", 3) != 0) {
3880                 fprintf(stderr, "%s does not start by %s-OST nor OST\n",
3881                         ostname, fsname);
3882                 return -EINVAL;
3883         }
3884         /* real_ostname is fsname-OST????? */
3885         ptr += 3;
3886         for (i = 0; i < 4; i++) {
3887                 if (!isxdigit(*ptr)) {
3888                         fprintf(stderr,
3889                                 "ost's index in %s is not an hexa number\n",
3890                                 ostname);
3891                         return -EINVAL;
3892                 }
3893                 ptr++;
3894         }
3895         /* real_ostname is fsname-OSTXXXX????? */
3896         /* if OST name does not end with _UUID, we add it */
3897         if (*ptr == '\0') {
3898                 strcat(real_ostname, "_UUID");
3899         } else if (strcmp(ptr, "_UUID") != 0) {
3900                 fprintf(stderr,
3901                         "ostname %s does not end with _UUID\n", ostname);
3902                 return -EINVAL;
3903         }
3904         /* real_ostname is fsname-OSTXXXX_UUID */
3905         strcpy(ostname, real_ostname);
3906         return 0;
3907 }
3908
3909 /* returns 0 or -errno */
3910 static int pool_cmd(enum lcfg_command_type cmd, char *cmdname,
3911                     char *fullpoolname, char *fsname, char *poolname,
3912                     char *ostname)
3913 {
3914         int rc = 0;
3915         struct obd_ioctl_data data;
3916         struct lustre_cfg_bufs bufs;
3917         struct lustre_cfg *lcfg;
3918         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
3919
3920         rc = check_pool_cmd(cmd, fsname, poolname, ostname);
3921         if (rc == -ENODEV)
3922                 fprintf(stderr,
3923                         "Can't verify pool command since there is no local MDT or client, proceeding anyhow...\n");
3924         else if (rc)
3925                 return rc;
3926
3927         lustre_cfg_bufs_reset(&bufs, NULL);
3928         lustre_cfg_bufs_set_string(&bufs, 0, cmdname);
3929         lustre_cfg_bufs_set_string(&bufs, 1, fullpoolname);
3930         if (ostname)
3931                 lustre_cfg_bufs_set_string(&bufs, 2, ostname);
3932
3933         lcfg = malloc(lustre_cfg_len(bufs.lcfg_bufcount, bufs.lcfg_buflen));
3934         if (!lcfg)
3935                 return -ENOMEM;
3936         lustre_cfg_init(lcfg, cmd, &bufs);
3937
3938         memset(&data, 0, sizeof(data));
3939         rc = data.ioc_dev = get_mgs_device();
3940         if (rc < 0)
3941                 goto out;
3942
3943         data.ioc_type = LUSTRE_CFG_TYPE;
3944         data.ioc_plen1 = lustre_cfg_len(lcfg->lcfg_bufcount,
3945                                         lcfg->lcfg_buflens);
3946         data.ioc_pbuf1 = (void *)lcfg;
3947
3948         memset(buf, 0, sizeof(rawbuf));
3949         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
3950         if (rc) {
3951                 fprintf(stderr, "error: %s: invalid ioctl\n",
3952                         jt_cmdname(cmdname));
3953                 free(lcfg);
3954                 return rc;
3955         }
3956         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_POOL, buf);
3957 out:
3958         if (rc)
3959                 rc = -errno;
3960         switch (rc) {
3961         case -ENAMETOOLONG:
3962                 fprintf(stderr,
3963                         "error: %s: either the pool or file system name is too long (max pool name len is %d and file system name is %d)\n",
3964                         jt_cmdname(cmdname), LOV_MAXPOOLNAME, LUSTRE_MAXFSNAME);
3965                 break;
3966         case -EINVAL:
3967                 fprintf(stderr,
3968                         "error: %s can contain only alphanumeric characters, underscores, and dashes besides the required '.'\n",
3969                         jt_cmdname(cmdname));
3970         default:
3971                 break;
3972         }
3973         free(lcfg);
3974         return rc;
3975 }
3976
3977 #ifdef HAVE_SERVER_SUPPORT
3978 /**
3979  * Format and send the ioctl to the MGS.
3980  *
3981  * \param       cmd             IOCTL to send
3982  * \param       ret_data        void pointer to return anything from
3983  *                              ioctl
3984  * \param       num_args        number of arguments to pack into the
3985  *                              ioctl buffer
3986  * \param       argv[]          variable number of string arguments
3987  *
3988  * \retval                      0 on success
3989  */
3990 static int nodemap_cmd(enum lcfg_command_type cmd, void *ret_data,
3991                        unsigned int ret_size, ...)
3992 {
3993         va_list                 ap;
3994         char                    *arg;
3995         int                     i = 0;
3996         struct lustre_cfg_bufs  bufs;
3997         struct obd_ioctl_data   data;
3998         struct lustre_cfg       *lcfg;
3999         char                    rawbuf[MAX_IOC_BUFLEN];
4000         char                    *buf = rawbuf;
4001         int                     rc = 0;
4002
4003         lustre_cfg_bufs_reset(&bufs, NULL);
4004
4005         va_start(ap, ret_size);
4006         arg = va_arg(ap, char *);
4007         while (arg) {
4008                 lustre_cfg_bufs_set_string(&bufs, i, arg);
4009                 i++;
4010                 arg = va_arg(ap, char *);
4011         }
4012         va_end(ap);
4013
4014         lcfg = malloc(lustre_cfg_len(bufs.lcfg_bufcount, bufs.lcfg_buflen));
4015         if (!lcfg)
4016                 return -ENOMEM;
4017         lustre_cfg_init(lcfg, cmd, &bufs);
4018
4019         memset(&data, 0, sizeof(data));
4020         rc = data.ioc_dev = get_mgs_device();
4021         if (rc < 0)
4022                 goto out;
4023
4024         data.ioc_type = LUSTRE_CFG_TYPE;
4025         data.ioc_plen1 = lustre_cfg_len(lcfg->lcfg_bufcount,
4026                                         lcfg->lcfg_buflens);
4027         data.ioc_pbuf1 = (void *)lcfg;
4028
4029         memset(buf, 0, sizeof(rawbuf));
4030         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
4031         if (rc) {
4032                 fprintf(stderr,
4033                         "error: invalid ioctl request: %08x errno: %d: %s\n",
4034                         cmd, errno, strerror(-rc));
4035                 goto out;
4036         }
4037
4038         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_NODEMAP, buf);
4039         if (rc) {
4040                 fprintf(stderr,
4041                         "error: invalid ioctl: %08x errno: %d: %s\n",
4042                         cmd, errno, strerror(errno));
4043                 goto out;
4044         }
4045
4046         if (ret_data) {
4047                 rc = llapi_ioctl_unpack(&data, buf, sizeof(rawbuf));
4048                 if (rc)
4049                         goto out;
4050
4051                 if (ret_size > data.ioc_plen1)
4052                         ret_size = data.ioc_plen1;
4053
4054                 memcpy(ret_data, data.ioc_pbuf1, ret_size);
4055         }
4056 out:
4057         free(lcfg);
4058
4059         return rc;
4060 }
4061
4062 /**
4063  * activate nodemap functions
4064  *
4065  * \param       argc            number of args
4066  * \param       argv[]          variable string arguments
4067  *
4068  * argv[0]                      1 for activate or 0 for deactivate
4069  *
4070  * \retval                      0 on success
4071  */
4072 int jt_nodemap_activate(int argc, char **argv)
4073 {
4074         int rc;
4075
4076         rc = nodemap_cmd(LCFG_NODEMAP_ACTIVATE, NULL, 0, argv[0], argv[1],
4077                          NULL);
4078
4079         if (rc != 0) {
4080                 errno = -rc;
4081                 perror(argv[0]);
4082         }
4083
4084         return rc;
4085 }
4086
4087 /**
4088  * add a nodemap
4089  *
4090  * \param       argc            number of args
4091  * \param       argv[]          variable string arguments
4092  *
4093  * argv[0]                      nodemap name
4094  *
4095  * \retval                      0 on success
4096  */
4097 int jt_nodemap_add(int argc, char **argv)
4098 {
4099         int rc;
4100
4101         rc = llapi_nodemap_exists(argv[1]);
4102         if (rc == 0) {
4103                 fprintf(stderr, "error: %s existing nodemap name\n", argv[1]);
4104                 return 1;
4105         }
4106
4107         rc = nodemap_cmd(LCFG_NODEMAP_ADD, NULL, 0, argv[0], argv[1], NULL);
4108
4109         if (rc != 0) {
4110                 errno = -rc;
4111                 perror(argv[0]);
4112         }
4113
4114         return rc;
4115 }
4116
4117 /**
4118  * delete a nodemap
4119  *
4120  * \param       argc            number of args
4121  * \param       argv[]          variable string arguments
4122  *
4123  * argv[0]                      nodemap name
4124  *
4125  * \retval                      0 on success
4126  */
4127 int jt_nodemap_del(int argc, char **argv)
4128 {
4129         int rc;
4130
4131         rc = llapi_nodemap_exists(argv[1]);
4132         if (rc != 0) {
4133                 fprintf(stderr, "error: %s not existing nodemap name\n",
4134                         argv[1]);
4135                 return rc;
4136         }
4137         rc = nodemap_cmd(LCFG_NODEMAP_DEL, NULL, 0, argv[0], argv[1], NULL);
4138
4139         if (rc != 0) {
4140                 errno = -rc;
4141                 perror(argv[0]);
4142         }
4143
4144         return rc;
4145 }
4146
4147 /**
4148  * test a nid for nodemap membership
4149  *
4150  * \param       argc            number of args
4151  * \param       argv[]          variable string arguments
4152  *
4153  * argv[0]                      properly formatted nid
4154  *
4155  * \retval                      0 on success
4156  */
4157 int jt_nodemap_test_nid(int argc, char **argv)
4158 {
4159         char    rawbuf[MAX_IOC_BUFLEN];
4160         int     rc;
4161
4162         rc = nodemap_cmd(LCFG_NODEMAP_TEST_NID, &rawbuf, sizeof(rawbuf),
4163                          argv[0], argv[1], NULL);
4164         if (rc == 0)
4165                 printf("%s\n", (char *)rawbuf);
4166
4167         return rc;
4168 }
4169
4170 /**
4171  * test a nodemap id pair for mapping
4172  *
4173  * \param       argc            number of args
4174  * \param       argv[[]         variable string arguments
4175  *
4176  * \retval                      0 on success
4177  *
4178  * The argv array should contain the nodemap name, the id
4179  * to checking the mapping on, and the id type (UID or GID)
4180  *
4181  */
4182 int jt_nodemap_test_id(int argc, char **argv)
4183 {
4184         char    rawbuf[MAX_IOC_BUFLEN];
4185         char    *nidstr = NULL;
4186         char    *idstr = NULL;
4187         char    *typestr = NULL;
4188         int     rc = 0;
4189         int     c;
4190
4191         static struct option long_opts[] = {
4192                 { .val = 'i',   .name = "id",   .has_arg = required_argument },
4193                 { .val = 'n',   .name = "nid",  .has_arg = required_argument },
4194                 { .val = 't',   .name = "idtype",
4195                                                 .has_arg = required_argument },
4196                 { .name = NULL } };
4197
4198         while ((c = getopt_long(argc, argv, "n:t:i:",
4199                                 long_opts, NULL)) != -1) {
4200                 switch (c) {
4201                 case 'n':
4202                         nidstr = optarg;
4203                         break;
4204                 case 't':
4205                         typestr = optarg;
4206                         break;
4207                 case 'i':
4208                         idstr = optarg;
4209                         break;
4210                 }
4211         }
4212
4213         if (!nidstr || !typestr || !idstr) {
4214                 fprintf(stderr,
4215                         "usage: nodemap_test_id --nid <nid> --idtype [uid|gid] --id <id>\n");
4216                 return -1;
4217         }
4218
4219         rc = nodemap_cmd(LCFG_NODEMAP_TEST_ID, &rawbuf, sizeof(rawbuf),
4220                          argv[0], nidstr, typestr, idstr, NULL);
4221         if (rc == 0)
4222                 printf("%s\n", (char *)rawbuf);
4223
4224         return rc;
4225 }
4226
4227 /**
4228  * parse nid range
4229  *
4230  * \param       nodemap_range   --range string
4231  * \param       nid_range       nid range string, min_nid:max_nid
4232  *
4233  * \retval                      0 on success
4234  */
4235 static int parse_nid_range(char *nodemap_range, char *nid_range, int range_len)
4236 {
4237         char                    min_nid[LNET_NIDSTR_SIZE + 1];
4238         char                    max_nid[LNET_NIDSTR_SIZE + 1];
4239         struct list_head        nidlist;
4240         int                     rc = 0;
4241
4242         INIT_LIST_HEAD(&nidlist);
4243
4244         if (cfs_parse_nidlist(nodemap_range, strlen(nodemap_range),
4245                               &nidlist) <= 0) {
4246                 fprintf(stderr,
4247                         "error: nodemap_xxx_range: can't parse nid range: %s\n",
4248                         nodemap_range);
4249                 return -EINVAL;
4250         }
4251
4252         rc = cfs_nidrange_find_min_max(&nidlist, &min_nid[0], &max_nid[0],
4253                                        LNET_NIDSTR_SIZE);
4254         if (rc < 0) {
4255                 if (rc == -EINVAL)
4256                         fprintf(stderr,
4257                                 "error: nodemap_xxx_range: nid range uses currently unsupported features\n");
4258                 else if (rc == -ERANGE)
4259                         fprintf(stderr,
4260                                 "error: nodemap_xxx_range: nodemap ranges must be contiguous\n");
4261
4262                 return rc;
4263         }
4264
4265         snprintf(nid_range, range_len, "%s:%s", min_nid, max_nid);
4266
4267         return rc;
4268 }
4269
4270 /**
4271  * add an nid range to a nodemap
4272  *
4273  * \param       argc            number of args
4274  * \param       argv[]          variable string arguments
4275  *
4276  * --name                       nodemap name
4277  * --range                      properly formatted nid range
4278  *
4279  * \retval                      0 on success, -errno on error
4280  */
4281 int jt_nodemap_add_range(int argc, char **argv)
4282 {
4283         char                    *nodemap_name = NULL;
4284         char                    *nodemap_range = NULL;
4285         char                    nid_range[2 * LNET_NIDSTR_SIZE + 2];
4286         int                     rc = 0;
4287         int                     c;
4288
4289         static struct option long_opts[] = {
4290         { .val = 'n',   .name = "name",         .has_arg = required_argument },
4291         { .val = 'r',   .name = "range",        .has_arg = required_argument },
4292         { .name = NULL } };
4293
4294         while ((c = getopt_long(argc, argv, "n:r:",
4295                                 long_opts, NULL)) != -1) {
4296                 switch (c) {
4297                 case 'n':
4298                         nodemap_name = optarg;
4299                         break;
4300                 case 'r':
4301                         nodemap_range = optarg;
4302                         break;
4303                 }
4304         }
4305
4306         if (!nodemap_name || !nodemap_range) {
4307                 fprintf(stderr,
4308                         "usage: nodemap_add_range --name <name> --range <range>\n");
4309                 return -EINVAL;
4310         }
4311
4312         rc = parse_nid_range(nodemap_range, nid_range, sizeof(nid_range));
4313         if (rc) {
4314                 return rc;
4315         }
4316         rc = nodemap_cmd(LCFG_NODEMAP_ADD_RANGE, NULL, 0, argv[0],
4317                          nodemap_name, nid_range, NULL);
4318         if (rc) {
4319                 fprintf(stderr,
4320                         "error: %s: cannot add range '%s' to nodemap '%s': %s\n",
4321                         jt_cmdname(argv[0]), nodemap_range, nodemap_name,
4322                         strerror(-rc));
4323         }
4324
4325         return rc;
4326 }
4327
4328 /**
4329  * delete an nid range to a nodemap
4330  *
4331  * \param       argc            number of args
4332  * \param       argv[]          variable string arguments
4333  *
4334  * --name                       nodemap name
4335  * --range                      properly formatted nid range
4336  *
4337  * \retval                      0 on success
4338  */
4339 int jt_nodemap_del_range(int argc, char **argv)
4340 {
4341         char                    *nodemap_name = NULL;
4342         char                    *nodemap_range = NULL;
4343         char                    nid_range[2 * LNET_NIDSTR_SIZE + 2];
4344         int                     rc = 0;
4345         int                     c;
4346
4347         static struct option long_opts[] = {
4348         { .val = 'n',   .name = "name",         .has_arg = required_argument },
4349         { .val = 'r',   .name = "range",        .has_arg = required_argument },
4350         { .name = NULL } };
4351
4352         while ((c = getopt_long(argc, argv, "n:r:",
4353                                 long_opts, NULL)) != -1) {
4354                 switch (c) {
4355                 case 'n':
4356                         nodemap_name = optarg;
4357                         break;
4358                 case 'r':
4359                         nodemap_range = optarg;
4360                         break;
4361                 }
4362         }
4363
4364         if (!nodemap_name || !nodemap_range) {
4365                 fprintf(stderr,
4366                         "usage: nodemap_del_range --name <name> --range <range>\n");
4367                 return -1;
4368         }
4369
4370         rc = parse_nid_range(nodemap_range, nid_range, sizeof(nid_range));
4371         if (rc) {
4372                 errno = -rc;
4373                 return rc;
4374         }
4375         rc = nodemap_cmd(LCFG_NODEMAP_DEL_RANGE, NULL, 0, argv[0],
4376                          nodemap_name, nid_range, NULL);
4377         if (rc != 0) {
4378                 errno = -rc;
4379                 fprintf(stderr,
4380                         "error: %s: cannot delete range '%s' to nodemap '%s': rc = %d\n",
4381                         jt_cmdname(argv[0]), nodemap_range, nodemap_name, rc);
4382         }
4383
4384         return rc;
4385 }
4386
4387 /**
4388  * set a fileset on a nodemap
4389  *
4390  * \param       argc            number of args
4391  * \param       argv[]          variable string arguments
4392  *
4393  * --name                       nodemap name
4394  * --fileset                    fileset name
4395  *
4396  * \retval                      0 on success
4397  */
4398 int jt_nodemap_set_fileset(int argc, char **argv)
4399 {
4400         char *nodemap_name = NULL;
4401         char *fileset_name = NULL;
4402         int   rc = 0;
4403         int   c;
4404
4405         static struct option long_opts[] = {
4406         { .val = 'f',   .name = "fileset",      .has_arg = required_argument },
4407         { .val = 'n',   .name = "name",         .has_arg = required_argument },
4408         { .name = NULL } };
4409
4410         while ((c = getopt_long(argc, argv, "n:f:",
4411                                 long_opts, NULL)) != -1) {
4412                 switch (c) {
4413                 case 'n':
4414                         nodemap_name = optarg;
4415                         break;
4416                 case 'f':
4417                         fileset_name = optarg;
4418                         break;
4419                 }
4420         }
4421
4422         if (!nodemap_name || !fileset_name) {
4423                 fprintf(stderr,
4424                         "usage: nodemap_set_fileset --name <name> --fileset <fileset>\n");
4425                 return -1;
4426         }
4427
4428         rc = nodemap_cmd(LCFG_NODEMAP_SET_FILESET, NULL, 0, argv[0],
4429                          nodemap_name, fileset_name, NULL);
4430         if (rc != 0) {
4431                 errno = -rc;
4432                 fprintf(stderr,
4433                         "error: %s: cannot set fileset '%s' on nodemap '%s': rc = %d\n",
4434                         jt_cmdname(argv[0]), fileset_name, nodemap_name, rc);
4435         }
4436
4437         return rc;
4438 }
4439
4440 /**
4441  * set SELinux policy info on a nodemap
4442  *
4443  * \param       argc            number of args
4444  * \param       argv[]          variable string arguments
4445  *
4446  * --name                       nodemap name
4447  * --sepol                      SELinux policy info
4448  *
4449  * \retval                      0 on success
4450  */
4451 int jt_nodemap_set_sepol(int argc, char **argv)
4452 {
4453         char *nodemap_name = NULL;
4454         char *sepol = NULL;
4455         int   rc = 0;
4456         int   c;
4457
4458         static struct option long_options[] = {
4459                 {
4460                         .name           = "name",
4461                         .has_arg        = required_argument,
4462                         .val            = 'n',
4463                 },
4464                 {
4465                         .name           = "sepol",
4466                         .has_arg        = required_argument,
4467                         .val            = 's',
4468                 },
4469                 {
4470                         .name = NULL,
4471                 }
4472         };
4473
4474         while ((c = getopt_long(argc, argv, "n:s:",
4475                                 long_options, NULL)) != -1) {
4476                 switch (c) {
4477                 case 'n':
4478                         nodemap_name = optarg;
4479                         break;
4480                 case 's':
4481                         sepol = optarg;
4482                         break;
4483                 }
4484         }
4485
4486         if (!nodemap_name || !sepol) {
4487                 fprintf(stderr,
4488                         "usage: nodemap_set_sepol --name <name> --sepol <sepol>\n");
4489                 return -1;
4490         }
4491
4492         rc = nodemap_cmd(LCFG_NODEMAP_SET_SEPOL, NULL, 0, argv[0],
4493                          nodemap_name, sepol, NULL);
4494         if (rc != 0) {
4495                 errno = -rc;
4496                 fprintf(stderr,
4497                         "error: %s: cannot set sepol '%s' on nodemap '%s': rc = %d\n",
4498                         jt_cmdname(argv[0]), sepol, nodemap_name, rc);
4499         }
4500
4501         return rc;
4502 }
4503
4504 /**
4505  * modify a nodemap's behavior
4506  *
4507  * \param       argc            number of args
4508  * \param       argv[]          variable string arguments
4509  *
4510  * --name                       nodemap name
4511  * --property                   nodemap property to change
4512  *                              admin, trusted, squash_uid, squash_gid)
4513  * --value                      value to set property
4514  *
4515  * \retval                      0 on success
4516  */
4517 int jt_nodemap_modify(int argc, char **argv)
4518 {
4519         int                     c;
4520         int                     rc = 0;
4521         enum lcfg_command_type  cmd = 0;
4522         char                    *nodemap_name = NULL;
4523         char                    *param = NULL;
4524         char                    *value = NULL;
4525
4526         static struct option long_opts[] = {
4527         { .val = 'n',   .name = "name",         .has_arg = required_argument },
4528         { .val = 'p',   .name = "property",     .has_arg = required_argument },
4529         { .val = 'v',   .name = "value",        .has_arg = required_argument },
4530         { .name = NULL } };
4531
4532         while ((c = getopt_long(argc, argv, "n:p:v:",
4533                                 long_opts, NULL)) != -1) {
4534                 switch (c) {
4535                 case 'n':
4536                         nodemap_name = optarg;
4537                         break;
4538                 case 'p':
4539                         param = optarg;
4540                         break;
4541                 case 'v':
4542                         value = optarg;
4543                         break;
4544                 }
4545         }
4546
4547         if (!nodemap_name || !param || !value) {
4548                 fprintf(stderr,
4549                         "usage: nodemap_modify --name <nodemap_name> --property <property_name> --value <value>\n");
4550                 fprintf(stderr,
4551                         "valid properties: admin trusted map_mode squash_uid squash_gid squash_projid deny_unknown audit_mode forbid_encryption\n");
4552                 return -1;
4553         }
4554
4555         if (strcmp("admin", param) == 0) {
4556                 cmd = LCFG_NODEMAP_ADMIN;
4557         } else if (strcmp("trusted", param) == 0) {
4558                 cmd = LCFG_NODEMAP_TRUSTED;
4559         } else if (strcmp("deny_unknown", param) == 0) {
4560                 cmd = LCFG_NODEMAP_DENY_UNKNOWN;
4561         } else if (strcmp("squash_uid", param) == 0) {
4562                 cmd = LCFG_NODEMAP_SQUASH_UID;
4563         } else if (strcmp("squash_gid", param) == 0) {
4564                 cmd = LCFG_NODEMAP_SQUASH_GID;
4565         } else if (strcmp("squash_projid", param) == 0) {
4566                 cmd = LCFG_NODEMAP_SQUASH_PROJID;
4567         } else if (strcmp("map_mode", param) == 0) {
4568                 cmd = LCFG_NODEMAP_MAP_MODE;
4569         } else if (strcmp("audit_mode", param) == 0) {
4570                 cmd = LCFG_NODEMAP_AUDIT_MODE;
4571         } else if (strcmp("forbid_encryption", param) == 0) {
4572                 cmd = LCFG_NODEMAP_FORBID_ENCRYPT;
4573         } else {
4574                 fprintf(stderr,
4575                         "error: %s: nodemap_modify invalid subcommand: %s\n",
4576                         jt_cmdname(argv[0]), param);
4577                 return -1;
4578         }
4579
4580         rc = nodemap_cmd(cmd, NULL, 0, argv[0], nodemap_name, param,
4581                          value, NULL);
4582         if (rc != 0) {
4583                 errno = -rc;
4584                 fprintf(stderr,
4585                         "error: %s: cannot modify nodemap '%s' to param '%s': value '%s': rc = %d\n",
4586                         jt_cmdname(argv[0]), nodemap_name, param, value, rc);
4587         }
4588
4589         return rc;
4590 }
4591
4592 int jt_nodemap_add_idmap(int argc, char **argv)
4593 {
4594         int                     c;
4595         enum                    lcfg_command_type cmd = 0;
4596         char                    *nodemap_name = NULL;
4597         char                    *idmap = NULL;
4598         char                    *idtype = NULL;
4599         int                     rc = 0;
4600
4601         static struct option long_opts[] = {
4602         { .val = 'i',   .name = "idtype",       .has_arg = required_argument },
4603         { .val = 'm',   .name = "idmap",        .has_arg = required_argument },
4604         { .val = 'n',   .name = "name",         .has_arg = required_argument },
4605         { .name = NULL } };
4606
4607         while ((c = getopt_long(argc, argv, "n:m:i:",
4608                                 long_opts, NULL)) != -1) {
4609                 switch (c) {
4610                 case 'n':
4611                         nodemap_name = optarg;
4612                         break;
4613                 case 'm':
4614                         idmap = optarg;
4615                         break;
4616                 case 'i':
4617                         idtype = optarg;
4618                         break;
4619                 }
4620         }
4621
4622         if (!nodemap_name || !idmap || !idtype) {
4623                 fprintf(stderr,
4624                         "usage: %s --name <name> --idtype [uid | gid | projid] --idmap <client id>:<filesystem id>\n",
4625                         argv[0]);
4626                 return -1;
4627         }
4628
4629         if (strcmp("uid", idtype) == 0) {
4630                 cmd = LCFG_NODEMAP_ADD_UIDMAP;
4631         } else if (strcmp("gid", idtype) == 0) {
4632                 cmd = LCFG_NODEMAP_ADD_GIDMAP;
4633         } else if (strcmp("projid", idtype) == 0) {
4634                 cmd = LCFG_NODEMAP_ADD_PROJIDMAP;
4635         } else {
4636                 fprintf(stderr,
4637                         "usage: %s --name <name> --idtype [uid | gid | projid] --idmap <client id>:<filesystem id>\n",
4638                         argv[0]);
4639                 return -1;
4640         }
4641
4642         rc = nodemap_cmd(cmd, NULL, 0, argv[0], nodemap_name, idmap, NULL);
4643         if (rc != 0) {
4644                 errno = -rc;
4645                 fprintf(stderr,
4646                         "cannot add %smap '%s' to nodemap '%s': rc = %d\n",
4647                         idtype, idmap, nodemap_name, rc);
4648         }
4649
4650         return rc;
4651 }
4652
4653 int jt_nodemap_del_idmap(int argc, char **argv)
4654 {
4655         int                     c;
4656         enum                    lcfg_command_type cmd = 0;
4657         char                    *nodemap_name = NULL;
4658         char                    *idmap = NULL;
4659         char                    *idtype = NULL;
4660         int                     rc = 0;
4661
4662         static struct option long_opts[] = {
4663         { .val = 'i',   .name = "idtype",       .has_arg = required_argument },
4664         { .val = 'm',   .name = "idmap",        .has_arg = required_argument },
4665         { .val = 'n',   .name = "name",         .has_arg = required_argument },
4666         { .name = NULL } };
4667
4668         while ((c = getopt_long(argc, argv, "n:m:i:",
4669                                 long_opts, NULL)) != -1) {
4670                 switch (c) {
4671                 case 'n':
4672                         nodemap_name = optarg;
4673                         break;
4674                 case 'm':
4675                         idmap = optarg;
4676                         break;
4677                 case 'i':
4678                         idtype = optarg;
4679                         break;
4680                 }
4681         }
4682
4683         if (!nodemap_name || !idmap || !idtype) {
4684                 fprintf(stderr,
4685                         "usage: %s --name <name> --idtype [uid | gid | projid] --idmap <client id>:<filesystem id>\n",
4686                         argv[0]);
4687                 return -1;
4688         }
4689
4690         if (strcmp("uid", idtype) == 0) {
4691                 cmd = LCFG_NODEMAP_DEL_UIDMAP;
4692         } else if (strcmp("gid", idtype) == 0) {
4693                 cmd = LCFG_NODEMAP_DEL_GIDMAP;
4694         } else if (strcmp("projid", idtype) == 0) {
4695                 cmd = LCFG_NODEMAP_DEL_PROJIDMAP;
4696         } else {
4697                 fprintf(stderr,
4698                         "usage: %s --name <name> --idtype [uid | gid | projid] --idmap <client id>:<filesystem id>\n",
4699                         argv[0]);
4700                 return -1;
4701         }
4702
4703         rc = nodemap_cmd(cmd, NULL, 0, argv[0], nodemap_name, idmap, NULL);
4704         if (rc != 0) {
4705                 errno = -rc;
4706                 fprintf(stderr,
4707                         "cannot delete %smap '%s' from nodemap '%s': rc = %d\n",
4708                         idtype, idmap, nodemap_name, rc);
4709         }
4710
4711         return rc;
4712 }
4713 #else /* !HAVE_SERVER_SUPPORT */
4714 int jt_nodemap_activate(int argc, char **argv)
4715 {
4716         fprintf(stderr, "error: %s: invalid ioctl\n",
4717                 jt_cmdname(argv[0]));
4718         return -EOPNOTSUPP;
4719 }
4720
4721 int jt_nodemap_add(int argc, char **argv)
4722 {
4723         fprintf(stderr, "error: %s: invalid ioctl\n",
4724                 jt_cmdname(argv[0]));
4725         return -EOPNOTSUPP;
4726 }
4727
4728 int jt_nodemap_del(int argc, char **argv)
4729 {
4730         fprintf(stderr, "error: %s: invalid ioctl\n",
4731                 jt_cmdname(argv[0]));
4732         return -EOPNOTSUPP;
4733 }
4734
4735 int jt_nodemap_modify(int argc, char **argv)
4736 {
4737         fprintf(stderr, "error: %s: invalid ioctl\n",
4738                 jt_cmdname(argv[0]));
4739         return -EOPNOTSUPP;
4740 }
4741
4742 int jt_nodemap_add_range(int argc, char **argv)
4743 {
4744         fprintf(stderr, "error: %s: invalid ioctl\n",
4745                 jt_cmdname(argv[0]));
4746         return -EOPNOTSUPP;
4747 }
4748
4749 int jt_nodemap_test_nid(int argc, char **argv)
4750 {
4751         fprintf(stderr, "error: %s: invalid ioctl\n",
4752                 jt_cmdname(argv[0]));
4753         return -EOPNOTSUPP;
4754 }
4755
4756 int jt_nodemap_del_range(int argc, char **argv)
4757 {
4758         fprintf(stderr, "error: %s: invalid ioctl\n",
4759                 jt_cmdname(argv[0]));
4760         return -EOPNOTSUPP;
4761 }
4762
4763 int jt_nodemap_add_idmap(int argc, char **argv)
4764 {
4765         fprintf(stderr, "error: %s: invalid ioctl\n",
4766                 jt_cmdname(argv[0]));
4767         return -EOPNOTSUPP;
4768 }
4769
4770 int jt_nodemap_del_idmap(int argc, char **argv)
4771 {
4772         fprintf(stderr, "error: %s: invalid ioctl\n",
4773                 jt_cmdname(argv[0]));
4774         return -EOPNOTSUPP;
4775 }
4776
4777 int jt_nodemap_test_id(int argc, char **argv)
4778 {
4779         fprintf(stderr, "error: %s: invalid ioctl\n",
4780                 jt_cmdname(argv[0]));
4781         return -EOPNOTSUPP;
4782 }
4783
4784 int jt_nodemap_set_fileset(int argc, char **argv)
4785 {
4786         fprintf(stderr, "error: %s: invalid ioctl\n",
4787                 jt_cmdname(argv[0]));
4788         return -EOPNOTSUPP;
4789 }
4790
4791 int jt_nodemap_set_sepol(int argc, char **argv)
4792 {
4793         fprintf(stderr, "error: %s: invalid ioctl\n",
4794                 jt_cmdname(argv[0]));
4795         return -EOPNOTSUPP;
4796 }
4797
4798 int jt_nodemap_info(int argc, char **argv)
4799 {
4800         fprintf(stderr, "error: %s: invalid ioctl\n",
4801                 jt_cmdname(argv[0]));
4802         return -EOPNOTSUPP;
4803 }
4804 #endif /* HAVE_SERVER_SUPPORT */
4805
4806 /*
4807  * this function tranforms a rule [start-end/step] into an array
4808  * of matching numbers
4809  * supported forms are:
4810  * [start]                : just this number
4811  * [start-end]            : all numbers from start to end
4812  * [start-end/step]       : numbers from start to end with increment of step
4813  * on return, format contains a printf format string which can be used
4814  * to generate all the strings
4815  */
4816 static int get_array_idx(char *rule, char *format, int **array)
4817 {
4818         char *start, *end, *ptr;
4819         unsigned int lo, hi, step;
4820         int array_sz = 0;
4821         int i, array_idx;
4822         int rc;
4823
4824         start = strchr(rule, '[');
4825         end = strchr(rule, ']');
4826         if ((!start) || (!end)) {
4827                 *array = malloc(sizeof(int));
4828                 if (!*array)
4829                         return 0;
4830                 strcpy(format, rule);
4831                 array_sz = 1;
4832                 return array_sz;
4833         }
4834         *start = '\0';
4835         *end = '\0';
4836         end++;
4837         start++;
4838         /* put in format the printf format (the rule without the range) */
4839         sprintf(format, "%s%%.4x%s", rule, end);
4840
4841         array_idx = 0;
4842         array_sz = 0;
4843         *array = NULL;
4844         /* loop on , separator */
4845         do {
4846                 /* extract the 3 fields */
4847                 rc = sscanf(start, "%x-%x/%u", &lo, &hi, &step);
4848                 switch (rc) {
4849                 case 0:
4850                         goto err;
4851                 case 1: {
4852                         void *tmp;
4853
4854                         array_sz++;
4855                         tmp = realloc(*array, array_sz * sizeof(int));
4856                         if (!tmp)
4857                                 goto err;
4858                         *array = tmp;
4859                         (*array)[array_idx] = lo;
4860                         array_idx++;
4861                         break;
4862                 }
4863                 case 2: {
4864                         step = 1;
4865                         /* do not break to share code with case 3: */
4866                 }
4867                 case 3: {
4868                         void *tmp;
4869
4870                         if ((hi < lo) || (step == 0))
4871                                 goto err;
4872                         array_sz += (hi - lo) / step + 1;
4873                         tmp = realloc(*array, array_sz * sizeof(int));
4874                         if (!tmp)
4875                                 goto err;
4876                         *array = tmp;
4877                         for (i = lo; i <= hi; i += step, array_idx++)
4878                                 (*array)[array_idx] = i;
4879                         break;
4880                 }
4881                 }
4882                 ptr = strchr(start, ',');
4883                 if (ptr)
4884                         start = ptr + 1;
4885
4886         } while (ptr);
4887         return array_sz;
4888 err:
4889         if (*array) {
4890                 free(*array);
4891                 *array = NULL;
4892         }
4893         return 0;
4894 }
4895
4896 struct llog_pool_name {
4897         char lpn_name[UUID_MAX];
4898         struct list_head lpn_list;
4899 };
4900
4901 struct llog_pool_list_data {
4902         char lpld_fsname[LUSTRE_MAXFSNAME + 1];
4903         char lpld_poolname[LOV_MAXPOOLNAME + 1];
4904         bool lpld_exists;
4905         struct list_head lpld_list_head;
4906 };
4907
4908 /**
4909  * Callback to list pool information in llog
4910  * - { index: 74, event: new_pool, device: tfs-clilov, fsname: tfs, pool: tmp }
4911  * - { index: 77, event: add_pool, device: tfs-clilov, fsname: tfs, pool: tmp,
4912  *     ost: tfs-OST0000_UUID }
4913  * - { index: 224, event: remove_pool, device: tfs-clilov, fsname: tfs,
4914  *     pool: tmp, ost: tfs-OST0003_UUID }
4915  * - { index: 227, event: del_pool, device: tfs-clilov, fsname: tfs, pool: tmp }
4916  *
4917  * \param record[in]    pointer to llog record
4918  * \param data[in]      pointer to struct llog_pool_list_data
4919  *
4920  * \retval              0 on success
4921  *                      <0 on error
4922  */
4923 static int llog_poollist_cb(const char *record, void *data)
4924 {
4925         struct llog_pool_list_data *lpld = data;
4926         char pool_filter[MAX_STRING_SIZE] = "";
4927         char *new_record, *del_record, *del_pool, *found;
4928         char type[10] = "";
4929         int filter_len, rc = 0;
4930
4931         filter_len = snprintf(pool_filter, sizeof(pool_filter), " fsname: %s,",
4932                               lpld->lpld_fsname);
4933         if (lpld->lpld_poolname[0] == '\0') {
4934                 new_record = get_event_filter(LCFG_POOL_NEW);
4935                 del_record = get_event_filter(LCFG_POOL_DEL);
4936                 strncpy(type, " pool: ", sizeof(type));
4937         } else {
4938                 filter_len += snprintf(pool_filter + filter_len,
4939                                        sizeof(pool_filter) - filter_len,
4940                                        " pool: %s", lpld->lpld_poolname);
4941                 new_record = get_event_filter(LCFG_POOL_ADD);
4942                 del_record = get_event_filter(LCFG_POOL_REM);
4943                 strncpy(type, " ost: ", sizeof(type));
4944         }
4945         del_pool = get_event_filter(LCFG_POOL_DEL);
4946
4947         if (!new_record || !del_record || !del_pool) {
4948                 rc = -ENOMEM;
4949                 goto out;
4950         }
4951
4952         found = strstr(record, pool_filter);
4953         if (found &&
4954             (found[filter_len] == ' ' || found[filter_len] == ',')) {
4955                 struct llog_pool_name *tmp = NULL;
4956                 struct list_head *head = &lpld->lpld_list_head;
4957                 char *name;
4958                 int name_len, type_len = strlen(type);
4959
4960                 lpld->lpld_exists = true;
4961                 if (strstr(record, new_record)) {
4962                         name = strstr(record, type);
4963                         /* 2 bytes for " }" */
4964                         name_len = strlen(name) - type_len - 2;
4965                         if (name_len <= 0 || name_len > sizeof(tmp->lpn_name))
4966                                 return -EINVAL;
4967                         tmp = malloc(sizeof(struct llog_pool_name));
4968                         if (!tmp) {
4969                                 rc = -ENOMEM;
4970                                 goto out;
4971                         }
4972                         memset(tmp, 0, sizeof(struct llog_pool_name));
4973                         strncpy(tmp->lpn_name, name + type_len, name_len);
4974                         list_add_tail(&tmp->lpn_list, &lpld->lpld_list_head);
4975                 } else if (strstr(record, del_record)) {
4976                         name = strstr(record, type);
4977                         name_len = strlen(name) - type_len - 2;
4978                         list_for_each_entry(tmp, head, lpn_list) {
4979                                 if (strncmp(tmp->lpn_name, name + type_len,
4980                                             name_len) == 0 &&
4981                                             tmp->lpn_name[name_len] == '\0') {
4982                                         list_del(&tmp->lpn_list);
4983                                         free(tmp);
4984                                         break;
4985                                 }
4986                         }
4987                 }
4988                 /* verify if the specified pool still exists */
4989                 if (lpld->lpld_poolname[0] && strstr(record, del_pool))
4990                         lpld->lpld_exists = false;
4991         }
4992 out:
4993         if (new_record)
4994                 free(new_record);
4995         if (del_record)
4996                 free(del_record);
4997         if (del_pool)
4998                 free(del_pool);
4999
5000         return rc;
5001 }
5002
5003 /**
5004  * List pool information by config log
5005  *
5006  * \param fsname[in]    pointer to filesystem name
5007  * \param poolname[in]  pointer to pool name
5008  *
5009  * \retval              0 on success
5010  *                      < 0 on error
5011  */
5012 int llog_poollist(char *fsname, char *poolname)
5013 {
5014         char logname[MAX_OBD_NAME] = {'\0'};
5015         struct llog_pool_list_data lpld;
5016         struct llog_pool_name *tmp;
5017         struct list_head *head;
5018         int rc = 0;
5019
5020         if (fsname && fsname[0] == '\0')
5021                 fsname = NULL;
5022         if (!fsname)
5023                 return -EINVAL;
5024
5025         memset(&lpld, 0, sizeof(lpld));
5026         INIT_LIST_HEAD(&lpld.lpld_list_head);
5027         lpld.lpld_exists = false;
5028         strncpy(lpld.lpld_fsname, fsname, sizeof(lpld.lpld_fsname) - 1);
5029         if (poolname && poolname[0])
5030                 strncpy(lpld.lpld_poolname, poolname,
5031                         sizeof(lpld.lpld_poolname) - 1);
5032         snprintf(logname, sizeof(logname), "%s-client", fsname);
5033         rc = jt_llog_print_iter(logname, 0, -1, llog_poollist_cb, &lpld, false,
5034                                 false);
5035
5036         if (poolname && poolname[0])
5037                 printf("Pool: %s.%s\n", fsname, poolname);
5038         else
5039                 printf("Pools from %s:\n", fsname);
5040
5041         head = &lpld.lpld_list_head;
5042         if (poolname && poolname[0] && !lpld.lpld_exists && list_empty(head))
5043                 return -ENOENT;
5044
5045         list_for_each_entry(tmp, head, lpn_list) {
5046                 if (poolname && poolname[0])
5047                         printf("%s\n", tmp->lpn_name);
5048                 else
5049                         printf("%s.%s\n", fsname, tmp->lpn_name);
5050                 list_del(&tmp->lpn_list);
5051                 free(tmp);
5052         }
5053
5054         return rc;
5055 }
5056
5057 static bool get_pools_path(char *fsname)
5058 {
5059         glob_t path;
5060         int rc;
5061
5062         rc = cfs_get_param_paths(&path, "lov/%s-*/pools", fsname);
5063         if (!rc)
5064                 cfs_free_param_data(&path);
5065
5066         return (rc == 0);
5067 }
5068
5069 static int extract_fsname_poolname(char **argv, char *fsname,
5070                                    char *poolname)
5071 {
5072         char *cmd = argv[0], *param = argv[1];
5073         char *ptr;
5074         int rc;
5075
5076         snprintf(fsname, PATH_MAX + 1, "%s", param);
5077         ptr = strchr(fsname, '.');
5078         if (!ptr) {
5079                 if (strcmp(cmd, "pool_list") == 0) {
5080                         poolname = NULL;
5081                         goto out;
5082                 }
5083                 fprintf(stderr, ". is missing in %s\n", fsname);
5084                 rc = -EINVAL;
5085                 goto err;
5086         }
5087
5088         if ((ptr - fsname) == 0) {
5089                 fprintf(stderr, "fsname is empty\n");
5090                 rc = -EINVAL;
5091                 goto err;
5092         }
5093
5094         *ptr = '\0';
5095         ++ptr;
5096
5097         if (strlen(ptr) == 0) {
5098                 fprintf(stderr, "poolname is empty\n");
5099                 rc = -EINVAL;
5100                 goto err;
5101         }
5102
5103         strncpy(poolname, ptr, LOV_MAXPOOLNAME);
5104         poolname[LOV_MAXPOOLNAME] = '\0';
5105
5106         if (strncmp(poolname, "none", LOV_MAXPOOLNAME) == 0) {
5107                 fprintf(stderr, "poolname cannot be 'none'\n");
5108                 return -EINVAL;
5109         }
5110 out:
5111         return 0;
5112
5113 err:
5114         fprintf(stderr, "argument %s must be <fsname>.<poolname>\n", param);
5115         return rc;
5116 }
5117
5118 int jt_pool_cmd(int argc, char **argv)
5119 {
5120         enum lcfg_command_type cmd;
5121         char fsname[PATH_MAX + 1];
5122         char poolname[LOV_MAXPOOLNAME + 1];
5123         char *ostnames_buf = NULL;
5124         int i, rc;
5125         int *array = NULL, array_sz;
5126         struct {
5127                 int     rc;
5128                 char   *ostname;
5129         } *cmds = NULL;
5130
5131         switch (argc) {
5132         case 0:
5133         case 1: return CMD_HELP;
5134         case 2: {
5135                 rc = extract_fsname_poolname(argv, fsname, poolname);
5136                 if (rc)
5137                         break;
5138
5139                 if (strcmp("pool_new", argv[0]) == 0) {
5140                         cmd = LCFG_POOL_NEW;
5141                 } else if (strcmp("pool_destroy", argv[0]) == 0) {
5142                         cmd = LCFG_POOL_DEL;
5143                 } else if (strcmp("pool_list", argv[0]) == 0) {
5144                         if (get_pools_path(fsname))
5145                                 return llapi_poollist(argv[1]);
5146                         if (get_mgs_device() > 0)
5147                                 return llog_poollist(fsname, poolname);
5148                         fprintf(stderr,
5149                                 "Cannot run pool_list command since there is no local MGS/MDT or client\n");
5150                         return CMD_HELP;
5151                 } else {
5152                         return CMD_HELP;
5153                 }
5154
5155                 rc = pool_cmd(cmd, argv[0], argv[1], fsname, poolname, NULL);
5156                 if (rc)
5157                         break;
5158
5159                 check_pool_cmd_result(cmd, fsname, poolname, NULL);
5160                 break;
5161         }
5162         default: {
5163                 char format[2 * MAX_OBD_NAME];
5164
5165                 if (strcmp("pool_remove", argv[0]) == 0)
5166                         cmd = LCFG_POOL_REM;
5167                 else if (strcmp("pool_add", argv[0]) == 0)
5168                         cmd = LCFG_POOL_ADD;
5169                 else
5170                         return CMD_HELP;
5171
5172                 rc = extract_fsname_poolname(argv, fsname, poolname);
5173                 if (rc)
5174                         break;
5175
5176                 for (i = 2; i < argc; i++) {
5177                         int j;
5178
5179                         array_sz = get_array_idx(argv[i], format, &array);
5180                         if (array_sz == 0)
5181                                 return CMD_HELP;
5182
5183                         cmds = malloc(array_sz * sizeof(cmds[0]));
5184                         if (cmds) {
5185                                 ostnames_buf = malloc(array_sz *
5186                                                       (MAX_OBD_NAME + 1));
5187                         } else {
5188                                 free(array);
5189                                 rc = -ENOMEM;
5190                                 goto out;
5191                         }
5192
5193                         for (j = 0; j < array_sz; j++) {
5194                                 char ostname[MAX_OBD_NAME + 1];
5195                                 int rc2;
5196
5197                                 snprintf(ostname, MAX_OBD_NAME, format,
5198                                          array[j]);
5199                                 ostname[MAX_OBD_NAME] = '\0';
5200
5201                                 rc2 = check_and_complete_ostname(fsname,
5202                                                                 ostname);
5203                                 if (rc2) {
5204                                         free(array);
5205                                         free(cmds);
5206                                         if (ostnames_buf)
5207                                                 free(ostnames_buf);
5208                                         rc = rc ? rc : rc2;
5209                                         goto out;
5210                                 }
5211                                 if (ostnames_buf) {
5212                                         cmds[j].ostname =
5213                                         &ostnames_buf[(MAX_OBD_NAME + 1) * j];
5214                                         strcpy(cmds[j].ostname, ostname);
5215                                 } else {
5216                                         cmds[j].ostname = NULL;
5217                                 }
5218                                 cmds[j].rc = pool_cmd(cmd, argv[0], argv[1],
5219                                                       fsname, poolname,
5220                                                       ostname);
5221                                 /* Return an err if any of the add/dels fail */
5222                                 if (!rc)
5223                                         rc = cmds[j].rc;
5224                         }
5225                         for (j = 0; j < array_sz; j++) {
5226                                 if (!cmds[j].rc) {
5227                                         char ostname[MAX_OBD_NAME + 1];
5228
5229                                         if (!cmds[j].ostname) {
5230                                                 snprintf(ostname, MAX_OBD_NAME,
5231                                                          format, array[j]);
5232                                                 ostname[MAX_OBD_NAME] = '\0';
5233                                                 check_and_complete_ostname(
5234                                                         fsname, ostname);
5235                                         } else {
5236                                                 strcpy(ostname,
5237                                                        cmds[j].ostname);
5238                                         }
5239                                         check_pool_cmd_result(cmd, fsname,
5240                                                               poolname,
5241                                                               ostname);
5242                                 }
5243                         }
5244                         if (array_sz > 0)
5245                                 free(array);
5246                         if (cmds)
5247                                 free(cmds);
5248                         if (ostnames_buf)
5249                                 free(ostnames_buf);
5250                 }
5251                 fallthrough;
5252         }
5253         } /* switch */
5254
5255 out:
5256         if (rc != 0) {
5257                 errno = -rc;
5258                 perror(argv[0]);
5259         }
5260
5261         return rc;
5262 }
5263
5264 #ifdef HAVE_SERVER_SUPPORT
5265 static const char *barrier_status2name(enum barrier_status status)
5266 {
5267         switch (status) {
5268         case BS_INIT:
5269                 return "init";
5270         case BS_FREEZING_P1:
5271                 return "freezing_p1";
5272         case BS_FREEZING_P2:
5273                 return "freezing_p2";
5274         case BS_FROZEN:
5275                 return "frozen";
5276         case BS_THAWING:
5277                 return "thawing";
5278         case BS_THAWED:
5279                 return "thawed";
5280         case BS_FAILED:
5281                 return "failed";
5282         case BS_EXPIRED:
5283                 return "expired";
5284         case BS_RESCAN:
5285                 return "rescan";
5286         default:
5287                 return "unknown";
5288         }
5289 }
5290
5291 int jt_barrier_freeze(int argc, char **argv)
5292 {
5293         struct obd_ioctl_data data;
5294         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
5295         struct barrier_ctl bc;
5296         int rc;
5297
5298         if (argc < 2 || argc > 3)
5299                 return CMD_HELP;
5300
5301         memset(&data, 0, sizeof(data));
5302         rc = data.ioc_dev = get_mgs_device();
5303         if (rc < 0)
5304                 return rc;
5305
5306         memset(&bc, 0, sizeof(bc));
5307         bc.bc_version = BARRIER_VERSION_V1;
5308         bc.bc_cmd = BC_FREEZE;
5309         if (argc == 3)
5310                 bc.bc_timeout = atoi(argv[2]);
5311         if (bc.bc_timeout == 0)
5312                 bc.bc_timeout = BARRIER_TIMEOUT_DEFAULT;
5313
5314         if (strlen(argv[1]) > 8) {
5315                 fprintf(stderr,
5316                         "%s: fsname name %s is too long. It should not exceed 8.\n",
5317                         argv[0], argv[1]);
5318                 return -EINVAL;
5319         }
5320
5321         strncpy(bc.bc_name, argv[1], sizeof(bc.bc_name));
5322         data.ioc_inlbuf1 = (char *)&bc;
5323         data.ioc_inllen1 = sizeof(bc);
5324         memset(buf, 0, sizeof(rawbuf));
5325         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
5326         if (rc) {
5327                 fprintf(stderr, "Fail to pack ioctl data: rc = %d.\n", rc);
5328                 return rc;
5329         }
5330
5331         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_BARRIER, buf);
5332         if (rc < 0)
5333                 fprintf(stderr, "Fail to freeze barrier for %s: %s\n",
5334                         argv[1], strerror(errno));
5335
5336         return rc;
5337 }
5338
5339 int jt_barrier_thaw(int argc, char **argv)
5340 {
5341         struct obd_ioctl_data data;
5342         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
5343         struct barrier_ctl bc;
5344         int rc;
5345
5346         if (argc != 2)
5347                 return CMD_HELP;
5348
5349         memset(&data, 0, sizeof(data));
5350         rc = data.ioc_dev = get_mgs_device();
5351         if (rc < 0)
5352                 return rc;
5353
5354         memset(&bc, 0, sizeof(bc));
5355         bc.bc_version = BARRIER_VERSION_V1;
5356         bc.bc_cmd = BC_THAW;
5357
5358         if (strlen(argv[1]) > 8) {
5359                 fprintf(stderr,
5360                         "fsname name %s is too long. It should not exceed 8.\n",
5361                         argv[1]);
5362                 return -EINVAL;
5363         }
5364
5365         strncpy(bc.bc_name, argv[1], sizeof(bc.bc_name));
5366         data.ioc_inlbuf1 = (char *)&bc;
5367         data.ioc_inllen1 = sizeof(bc);
5368         memset(buf, 0, sizeof(rawbuf));
5369         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
5370         if (rc) {
5371                 fprintf(stderr, "Fail to pack ioctl data: rc = %d.\n", rc);
5372                 return rc;
5373         }
5374
5375         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_BARRIER, buf);
5376         if (rc < 0)
5377                 fprintf(stderr, "Fail to thaw barrier for %s: %s\n",
5378                         argv[1], strerror(errno));
5379
5380         return rc;
5381 }
5382
5383 int __jt_barrier_stat(const char *fsname, struct barrier_ctl *bc)
5384 {
5385         struct obd_ioctl_data data;
5386         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
5387         int rc;
5388
5389         memset(&data, 0, sizeof(data));
5390         rc = data.ioc_dev = get_mgs_device();
5391         if (rc < 0)
5392                 return rc;
5393
5394         memset(bc, 0, sizeof(*bc));
5395         bc->bc_version = BARRIER_VERSION_V1;
5396         bc->bc_cmd = BC_STAT;
5397         strncpy(bc->bc_name, fsname, sizeof(bc->bc_name) - 1);
5398         data.ioc_inlbuf1 = (char *)bc;
5399         data.ioc_inllen1 = sizeof(*bc);
5400         memset(buf, 0, sizeof(rawbuf));
5401         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
5402         if (rc) {
5403                 fprintf(stderr, "Fail to pack ioctl data: rc = %d.\n", rc);
5404                 return rc;
5405         }
5406
5407         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_BARRIER, buf);
5408         if (rc < 0)
5409                 fprintf(stderr, "Fail to query barrier for %s: %s\n",
5410                         fsname, strerror(errno));
5411         else
5412                 llapi_ioctl_unpack(&data, buf, sizeof(rawbuf));
5413
5414         return rc;
5415 }
5416
5417 int jt_barrier_stat(int argc, char **argv)
5418 {
5419         struct barrier_ctl bc;
5420         static struct option long_opt_barrier_stat[] = {
5421                 {
5422                         .val            = 's',
5423                         .name           = "state",
5424                         .has_arg        = no_argument,
5425                 },
5426                 {       .val            = 't',
5427                         .name           = "timeout",
5428                         .has_arg        = no_argument,
5429                 },
5430                 {
5431                         NULL
5432                 }
5433         };
5434         const char *name;
5435         int index;
5436         int opt;
5437         int rc;
5438         bool state = false;
5439         bool timeout = false;
5440
5441         while ((opt = getopt_long(argc, argv, "st", long_opt_barrier_stat,
5442                                   &index)) != EOF) {
5443                 switch (opt) {
5444                 case 's':
5445                         state = true;
5446                         break;
5447                 case 't':
5448                         timeout = true;
5449                         break;
5450                 default:
5451                         return CMD_HELP;
5452                 }
5453         }
5454
5455         if (optind >= argc)
5456                 return CMD_HELP;
5457
5458         name = argv[optind];
5459         if (strlen(name) > 8) {
5460                 fprintf(stderr,
5461                         "fsname name %s is too long. It should not exceed 8.\n",
5462                         name);
5463                 return -EINVAL;
5464         }
5465
5466         rc = __jt_barrier_stat(name, &bc);
5467         if (!rc) {
5468                 if (state && !timeout)
5469                         printf("%s\n", barrier_status2name(bc.bc_status));
5470                 else if (timeout && !state)
5471                         printf("%d\n",
5472                                (bc.bc_status == BS_FREEZING_P1 ||
5473                                 bc.bc_status == BS_FREEZING_P2 ||
5474                                 bc.bc_status == BS_FROZEN) ?
5475                                bc.bc_timeout : 0);
5476                 else
5477                         printf("state: %s\ntimeout: %d seconds\n",
5478                                barrier_status2name(bc.bc_status),
5479                                (bc.bc_status == BS_FREEZING_P1 ||
5480                                 bc.bc_status == BS_FREEZING_P2 ||
5481                                 bc.bc_status == BS_FROZEN) ?
5482                                bc.bc_timeout : 0);
5483         }
5484
5485         return rc;
5486 }
5487
5488 int jt_barrier_rescan(int argc, char **argv)
5489 {
5490         struct obd_ioctl_data data;
5491         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
5492         struct barrier_ctl bc;
5493         int rc;
5494
5495         if (argc < 2 || argc > 3)
5496                 return CMD_HELP;
5497
5498         memset(&data, 0, sizeof(data));
5499         rc = data.ioc_dev = get_mgs_device();
5500         if (rc < 0)
5501                 return rc;
5502
5503         memset(&bc, 0, sizeof(bc));
5504         bc.bc_version = BARRIER_VERSION_V1;
5505         bc.bc_cmd = BC_RESCAN;
5506         if (argc == 3)
5507                 bc.bc_timeout = atoi(argv[2]);
5508         if (bc.bc_timeout == 0)
5509                 bc.bc_timeout = BARRIER_TIMEOUT_DEFAULT;
5510
5511         if (strlen(argv[1]) > 8) {
5512                 fprintf(stderr,
5513                         "fsname name %s is too long. It should not exceed 8.\n",
5514                         argv[1]);
5515                 return -EINVAL;
5516         }
5517
5518         strncpy(bc.bc_name, argv[1], sizeof(bc.bc_name));
5519         data.ioc_inlbuf1 = (char *)&bc;
5520         data.ioc_inllen1 = sizeof(bc);
5521         memset(buf, 0, sizeof(rawbuf));
5522         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
5523         if (rc) {
5524                 fprintf(stderr, "Fail to pack ioctl data: rc = %d.\n", rc);
5525                 return rc;
5526         }
5527
5528         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_BARRIER, buf);
5529         if (rc < 0) {
5530                 fprintf(stderr, "Fail to rescan barrier bitmap for %s: %s\n",
5531                         argv[1], strerror(errno));
5532         } else {
5533                 llapi_ioctl_unpack(&data, buf, sizeof(rawbuf));
5534                 printf("%u of %u MDT(s) in the filesystem %s are inactive\n",
5535                        bc.bc_absence, bc.bc_total, argv[1]);
5536         }
5537
5538         return rc;
5539 }
5540 #endif /* HAVE_SERVER_SUPPORT */
5541
5542 int jt_get_obj_version(int argc, char **argv)
5543 {
5544         struct lu_fid fid;
5545         struct obd_ioctl_data data;
5546         __u64 version, id = ULLONG_MAX, group = ULLONG_MAX;
5547         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf, *fidstr;
5548         int rc, c;
5549
5550         while ((c = getopt(argc, argv, "i:g:")) != -1) {
5551                 switch (c) {
5552                 case 'i':
5553                         id = strtoull(optarg, NULL, 0);
5554                         break;
5555                 case 'g':
5556                         group = strtoull(optarg, NULL, 0);
5557                         break;
5558                 default:
5559                         return CMD_HELP;
5560                 }
5561         }
5562
5563         argc -= optind;
5564         fidstr = *(argv + optind);
5565
5566         if (!(id != ULLONG_MAX && group != ULLONG_MAX && argc == 0) &&
5567             !(id == ULLONG_MAX && group == ULLONG_MAX && argc == 1))
5568                 return CMD_HELP;
5569
5570         memset(&data, 0, sizeof(data));
5571         data.ioc_dev = cur_device;
5572         if (argc == 1) {
5573                 rc = llapi_fid_parse(fidstr, &fid, NULL);
5574                 if (rc) {
5575                         fprintf(stderr, "%s: error parsing FID '%s': %s\n",
5576                                 jt_cmdname(argv[0]), fidstr, strerror(-rc));
5577                         return rc;
5578                 }
5579
5580                 data.ioc_inlbuf1 = (char *)&fid;
5581                 data.ioc_inllen1 = sizeof(fid);
5582         } else {
5583                 data.ioc_inlbuf3 = (char *)&id;
5584                 data.ioc_inllen3 = sizeof(id);
5585                 data.ioc_inlbuf4 = (char *)&group;
5586                 data.ioc_inllen4 = sizeof(group);
5587         }
5588         data.ioc_inlbuf2 = (char *)&version;
5589         data.ioc_inllen2 = sizeof(version);
5590
5591         memset(buf, 0, sizeof(*buf));
5592         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
5593         if (rc) {
5594                 fprintf(stderr, "error: %s: packing ioctl arguments: %s\n",
5595                         jt_cmdname(argv[0]), strerror(-rc));
5596                 return rc;
5597         }
5598
5599         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_GET_OBJ_VERSION, buf);
5600         if (rc) {
5601                 fprintf(stderr, "error: %s: ioctl: %s\n",
5602                         jt_cmdname(argv[0]), strerror(errno));
5603                 return -errno;
5604         }
5605
5606         llapi_ioctl_unpack(&data, buf, sizeof(rawbuf));
5607         printf("%#jx\n", (uintmax_t)version);
5608
5609         return 0;
5610 }
5611
5612 #ifdef HAVE_SERVER_SUPPORT
5613 int jt_changelog_register(int argc, char **argv)
5614 {
5615         struct option long_opts[] = {
5616         { .val = 'h', .name = "help", .has_arg = no_argument },
5617         { .val = 'm', .name = "mask", .has_arg = required_argument },
5618         { .val = 'n', .name = "nameonly", .has_arg = no_argument },
5619         { .val = 'u', .name = "user", .has_arg = required_argument },
5620         { .name = NULL } };
5621         struct obd_ioctl_data data = { 0 };
5622         char rawbuf[MAX_IOC_BUFLEN] = "";
5623         char *buf = rawbuf;
5624         char *device = lcfg_get_devname();
5625         char *username = NULL, *usermask = NULL;
5626         bool print_name_only = false;
5627         int c;
5628         int rc;
5629
5630         if (cur_device < 0 || !device)
5631                 return CMD_HELP;
5632
5633         while ((c = getopt_long(argc, argv, "hm:nu:", long_opts, NULL)) != -1) {
5634                 switch (c) {
5635                 case 'm':
5636                         usermask = strdup(optarg);
5637                         if (!usermask) {
5638                                 fprintf(stderr,
5639                                         "error: %s: %s: cannot copy '%s'\n",
5640                                         jt_cmdname(argv[0]), strerror(errno),
5641                                         optarg);
5642                                 return -errno;
5643                         }
5644                         break;
5645                 case 'n':
5646                         print_name_only = true;
5647                         break;
5648                 case 'u':
5649                         username = strdup(optarg);
5650                         if (!username) {
5651                                 fprintf(stderr,
5652                                         "error: %s: %s: cannot copy '%s'\n",
5653                                         jt_cmdname(argv[0]), strerror(errno),
5654                                         optarg);
5655                                 return -errno;
5656                         }
5657                         break;
5658                 case 'h':
5659                 default:
5660                         free(username);
5661                         free(usermask);
5662                         return CMD_HELP;
5663                 }
5664         }
5665
5666         data.ioc_dev = cur_device;
5667         if (username) {
5668                 data.ioc_inlbuf1 = username;
5669                 data.ioc_inllen1 = strlen(username) + 1;
5670         }
5671
5672         if (usermask) {
5673                 data.ioc_inlbuf2 = usermask;
5674                 data.ioc_inllen2 = strlen(usermask) + 1;
5675         }
5676
5677         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
5678         if (rc < 0) {
5679                 fprintf(stderr, "error: %s: cannot pack ioctl: %s\n",
5680                         jt_cmdname(argv[0]), strerror(-rc));
5681                 goto out;
5682         }
5683         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_CHANGELOG_REG, buf);
5684         if (rc < 0) {
5685                 rc = -errno;
5686                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
5687                         rc == -EEXIST ? "User exists" : strerror(-rc));
5688                 goto out;
5689         }
5690
5691         llapi_ioctl_unpack(&data, buf, sizeof(rawbuf));
5692
5693         if (data.ioc_u32_1 == 0) {
5694                 fprintf(stderr, "received invalid userid!\n");
5695                 rc = -EPROTO;
5696                 goto out;
5697         }
5698
5699         if (print_name_only)
5700                 printf("%s%u%s%s\n", CHANGELOG_USER_PREFIX, data.ioc_u32_1,
5701                        username ? "-" : "", username ? : "");
5702         else
5703                 printf("%s: Registered changelog userid '%s%u%s%s'\n",
5704                        device, CHANGELOG_USER_PREFIX, data.ioc_u32_1,
5705                        username ? "-" : "", username ? : "");
5706 out:
5707         free(usermask);
5708         free(username);
5709         return rc;
5710 }
5711
5712 int jt_changelog_deregister(int argc, char **argv)
5713 {
5714         struct option long_opts[] = {
5715         { .val = 'h', .name = "help", .has_arg = no_argument },
5716         { .val = 'u', .name = "user", .has_arg = required_argument },
5717         { .name = NULL } };
5718         struct obd_ioctl_data data = { 0 };
5719         char rawbuf[MAX_IOC_BUFLEN] = "";
5720         char *buf = rawbuf;
5721         char *device = lcfg_get_devname();
5722         char *username = NULL;
5723         int id = 0;
5724         int c, rc;
5725
5726         if (cur_device < 0 || !device)
5727                 return CMD_HELP;
5728
5729         while ((c = getopt_long(argc, argv, "hu:", long_opts, NULL)) != -1) {
5730                 switch (c) {
5731                 case 'u':
5732                         username = strdup(optarg);
5733                         if (!username) {
5734                                 fprintf(stderr,
5735                                         "error: %s: %s: cannot copy '%s'\n",
5736                                         jt_cmdname(argv[0]), strerror(errno),
5737                                         optarg);
5738                                 return -errno;
5739                         }
5740                         break;
5741                 case 'h':
5742                 default:
5743                         free(username);
5744                         return CMD_HELP;
5745                 }
5746         }
5747
5748         if (1 == optind && argc > 1) {
5749                 /* first check if pure ID was passed */
5750                 id = atoi(argv[optind]);
5751                 /* nameless cl<ID> format or cl<ID>-... format, only ID matters */
5752                 if (id == 0)
5753                         sscanf(argv[optind], CHANGELOG_USER_PREFIX"%d", &id);
5754
5755                 /* no valid ID was parsed */
5756                 if (id <= 0) {
5757                         rc = -EINVAL;
5758                         fprintf(stderr,
5759                                 "error: %s: expect <ID> or cl<ID>[-name] got '%s'\n",
5760                                 strerror(-rc), argv[optind]);
5761                         return CMD_HELP;
5762                 }
5763                 optind++;
5764         }
5765
5766         if (optind < argc || argc == 1)
5767                 return CMD_HELP;
5768
5769         data.ioc_dev = cur_device;
5770         data.ioc_u32_1 = id;
5771         if (username) {
5772                 data.ioc_inlbuf1 = username;
5773                 data.ioc_inllen1 = strlen(username) + 1;
5774         }
5775
5776         rc = llapi_ioctl_pack(&data, &buf, sizeof(rawbuf));
5777         if (rc < 0) {
5778                 fprintf(stderr, "error: %s: invalid ioctl\n",
5779                         jt_cmdname(argv[0]));
5780                 return rc;
5781         }
5782
5783         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_CHANGELOG_DEREG, buf);
5784         if (rc < 0) {
5785                 rc = -errno;
5786                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
5787                         rc == -ENOENT ? "User not found" : strerror(-rc));
5788                 return rc;
5789         }
5790
5791         llapi_ioctl_unpack(&data, buf, sizeof(rawbuf));
5792         printf("%s: Deregistered changelog user #%u\n", device, data.ioc_u32_1);
5793
5794         return 0;
5795 }
5796 #else /* !HAVE_SERVER_SUPPORT */
5797 int jt_changelog_register(int argc, char **argv)
5798 {
5799         fprintf(stderr, "error: %s: invalid ioctl\n",
5800                 jt_cmdname(argv[0]));
5801         return -EOPNOTSUPP;
5802 }
5803
5804 int jt_changelog_deregister(int argc, char **argv)
5805 {
5806         fprintf(stderr, "error: %s: invalid ioctl\n",
5807                 jt_cmdname(argv[0]));
5808         return -EOPNOTSUPP;
5809 }
5810 #endif /* HAVE_SERVER_SUPPORT */
5811
5812 int jt_pcc_add(int argc, char **argv)
5813 {
5814         struct option long_opts[] = {
5815                 { .val = 'p', .name = "param", .has_arg = required_argument },
5816                 { .name = NULL } };
5817         const char *mntpath;
5818         const char *pccpath;
5819         char *param = NULL;
5820         char cmd[PATH_MAX];
5821         int rc;
5822
5823         optind = 1;
5824         while ((rc = getopt_long(argc, argv, "p:", long_opts, NULL)) != -1) {
5825                 switch (rc) {
5826                 case 'p':
5827                         param = optarg;
5828                         break;
5829                 default:
5830                         return CMD_HELP;
5831                 }
5832         }
5833
5834         if (!param) {
5835                 fprintf(stderr, "%s: must specify the config param for PCC\n",
5836                         jt_cmdname(argv[0]));
5837                 return CMD_HELP;
5838         }
5839
5840         if (optind + 2 != argc) {
5841                 fprintf(stderr,
5842                         "%s: must specify mount path and PCC path %d:%d\n",
5843                         jt_cmdname(argv[0]), optind, argc);
5844                 return CMD_HELP;
5845         }
5846
5847         mntpath = argv[optind++];
5848         pccpath = argv[optind];
5849
5850         snprintf(cmd, PATH_MAX, "add %s %s", pccpath, param);
5851         rc = llapi_pccdev_set(mntpath, cmd);
5852         if (rc < 0)
5853                 fprintf(stderr, "%s: failed to run '%s' on %s\n",
5854                         jt_cmdname(argv[0]), cmd, mntpath);
5855
5856         return rc;
5857 }
5858
5859 int jt_pcc_del(int argc, char **argv)
5860 {
5861         const char *mntpath;
5862         const char *pccpath;
5863         char cmd[PATH_MAX];
5864         int rc;
5865
5866         optind = 1;
5867         if (argc != 3) {
5868                 fprintf(stderr, "%s: require 3 arguments\n",
5869                         jt_cmdname(argv[0]));
5870                 return CMD_HELP;
5871         }
5872
5873         mntpath = argv[optind++];
5874         pccpath = argv[optind++];
5875
5876         snprintf(cmd, PATH_MAX, "del %s", pccpath);
5877         rc = llapi_pccdev_set(mntpath, cmd);
5878         if (rc < 0)
5879                 fprintf(stderr, "%s: failed to run '%s' on %s\n",
5880                         jt_cmdname(argv[0]), cmd, mntpath);
5881
5882         return rc;
5883 }
5884
5885 int jt_pcc_clear(int argc, char **argv)
5886 {
5887         const char *mntpath;
5888         int rc;
5889
5890         optind = 1;
5891         if (argc != 2) {
5892                 fprintf(stderr, "%s: require 2 arguments\n",
5893                         jt_cmdname(argv[0]));
5894                 return CMD_HELP;
5895         }
5896
5897         mntpath = argv[optind];
5898         rc = llapi_pccdev_set(mntpath, "clear");
5899         if (rc < 0)
5900                 fprintf(stderr, "%s: failed to run 'clear' on %s\n",
5901                         jt_cmdname(argv[0]), mntpath);
5902
5903         return rc;
5904 }
5905
5906 int jt_pcc_list(int argc, char **argv)
5907 {
5908         const char *mntpath;
5909         int rc;
5910
5911         optind = 1;
5912         if (argc != 2) {
5913                 fprintf(stderr, "%s: require 2 arguments\n",
5914                         jt_cmdname(argv[0]));
5915                 return CMD_HELP;
5916         }
5917
5918         mntpath = argv[optind];
5919         rc = llapi_pccdev_get(mntpath);
5920         if (rc < 0)
5921                 fprintf(stderr, "%s: failed to run 'pcc list' on %s\n",
5922                         jt_cmdname(argv[0]), mntpath);
5923
5924         return rc;
5925 }