Whamcloud - gitweb
LU-6401 uapi: turn lustre_ioctl.h into a proper UAPI header
[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, 2016, Intel Corporation.
27  */
28 /*
29  * This file is part of Lustre, http://www.lustre.org/
30  * Lustre is a trademark of Sun Microsystems, Inc.
31  *
32  * lustre/utils/obd.c
33  *
34  * Author: Peter J. Braam <braam@clusterfs.com>
35  * Author: Phil Schwan <phil@clusterfs.com>
36  * Author: Andreas Dilger <adilger@clusterfs.com>
37  * Author: Robert Read <rread@clusterfs.com>
38  */
39
40 #include <sys/ioctl.h>
41 #include <sys/socket.h>
42 #include <sys/stat.h>
43 #include <sys/time.h>
44 #include <sys/types.h>
45 #include <sys/un.h>
46 #include <sys/wait.h>
47
48 #include <ctype.h>
49 #include <errno.h>
50 #include <fcntl.h>
51 #include <getopt.h>
52 #include <signal.h>
53 #include <stdarg.h>
54 #include <stdbool.h>
55 #include <stdio.h>
56 #include <stdlib.h>
57 #include <string.h>
58 #include <time.h>
59 #include <unistd.h>
60 #include <limits.h>
61
62 #include "obdctl.h"
63 #include "lustreapi_internal.h"
64 #include <libcfs/util/list.h>
65 #include <libcfs/util/ioctl.h>
66 #include <libcfs/util/param.h>
67 #include <libcfs/util/parser.h>
68 #include <libcfs/util/string.h>
69
70 #include <lnet/nidstr.h>
71 #include <lustre/lustre_ostid.h>
72 #include <lustre_cfg.h>
73 #include <linux/lustre_ioctl.h>
74 #include <lustre_ver.h>
75
76 #include <lnet/lnetctl.h>
77 #include <lustre/lustreapi.h>
78 #include <lustre_param.h>
79 #include <lustre/lustre_barrier_user.h>
80
81 #define MAX_STRING_SIZE 128
82
83 #if HAVE_LIBPTHREAD
84 #include <sys/ipc.h>
85 #include <sys/shm.h>
86 #include <pthread.h>
87
88 #define MAX_THREADS 4096
89 #define MAX_BASE_ID 0xffffffff
90 #define NIDSTRING_LENGTH 64
91 struct shared_data {
92         pthread_mutex_t mutex;
93         pthread_cond_t  cond;
94         int       stopping;
95         struct {
96                 __u64 counters[MAX_THREADS];
97                 __u64 offsets[MAX_THREADS];
98                 int   thr_running;
99                 int   start_barrier;
100                 int   stop_barrier;
101                 struct timeval start_time;
102                 struct timeval end_time;
103         } body;
104 };
105
106 static struct shared_data *shared_data;
107 static __u64 counter_snapshot[2][MAX_THREADS];
108 static int prev_valid;
109 static struct timeval prev_time;
110 static int thread;
111 static int nthreads;
112 #else
113 const int thread = 0;
114 const int nthreads = 1;
115 #endif
116
117 static int cur_device = -1;
118
119 static int l2_ioctl(int dev_id, int opc, void *buf)
120 {
121         return l_ioctl(dev_id, opc, buf);
122 }
123
124 int lcfg_ioctl(char * func, int dev_id, struct lustre_cfg *lcfg)
125 {
126         struct obd_ioctl_data data;
127         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
128         int rc;
129
130         memset(&data, 0, sizeof(data));
131         data.ioc_dev = cur_device;
132         data.ioc_type = LUSTRE_CFG_TYPE;
133         data.ioc_plen1 = lustre_cfg_len(lcfg->lcfg_bufcount,
134                                         lcfg->lcfg_buflens);
135         data.ioc_pbuf1 = (void *)lcfg;
136         memset(buf, 0, sizeof(rawbuf));
137         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
138         if (rc) {
139                 fprintf(stderr, "error: %s: invalid ioctl\n",
140                         jt_cmdname(func));
141                 return rc;
142         }
143
144         rc =  l_ioctl(dev_id, OBD_IOC_PROCESS_CFG, buf);
145
146         return rc;
147 }
148
149 static int do_device(char *func, char *devname);
150
151 static int get_mgs_device()
152 {
153         char mgs[] = "$MGS";
154         static int mgs_device = -1;
155
156         if (mgs_device == -1) {
157                 int rc;
158                 do_disconnect(NULL, 1);
159                 rc = do_device("mgsioc", mgs);
160                 if (rc) {
161                         fprintf(stderr,
162                                 "This command must be run on the MGS.\n");
163                         errno = ENODEV;
164                         return -1;
165                 }
166                 mgs_device = cur_device;
167         }
168         return mgs_device;
169 }
170
171 /* Returns -1 on error with errno set */
172 int lcfg_mgs_ioctl(char *func, int dev_id, struct lustre_cfg *lcfg)
173 {
174         struct obd_ioctl_data data;
175         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
176         int rc;
177
178         memset(&data, 0, sizeof(data));
179         rc = data.ioc_dev = get_mgs_device();
180         if (rc < 0)
181                 goto out;
182         data.ioc_type = LUSTRE_CFG_TYPE;
183         data.ioc_plen1 = lustre_cfg_len(lcfg->lcfg_bufcount,
184                                         lcfg->lcfg_buflens);
185         data.ioc_pbuf1 = (void *)lcfg;
186         memset(buf, 0, sizeof(rawbuf));
187         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
188         if (rc) {
189                 fprintf(stderr, "error: %s: invalid ioctl\n",
190                         jt_cmdname(func));
191                 return rc;
192         }
193
194         rc = l_ioctl(dev_id, OBD_IOC_PARAM, buf);
195 out:
196         if (rc) {
197                 if (errno == ENOSYS)
198                         fprintf(stderr, "Make sure cfg_device is set first.\n");
199         }
200         return rc;
201 }
202
203 char *obdo_print(struct obdo *obd)
204 {
205         char buf[1024];
206
207         snprintf(buf, sizeof(buf), "id: %#jx\ngrp: %#jx\natime: %ju\n"
208                  "mtime: %ju\nctime: %ju\nsize: %ju\nblocks: %ju"
209                  "\nblksize: %u\nmode: %o\nuid: %d\ngid: %d\nflags: %x\n"
210                  "misc: %x\nnlink: %d,\nvalid %#jx\n",
211                  (uintmax_t)ostid_id(&obd->o_oi),
212                  (uintmax_t)ostid_seq(&obd->o_oi),
213                  (uintmax_t)obd->o_atime, (uintmax_t)obd->o_mtime,
214                  (uintmax_t)obd->o_ctime, (uintmax_t)obd->o_size,
215                  (uintmax_t)obd->o_blocks, obd->o_blksize, obd->o_mode,
216                  obd->o_uid, obd->o_gid, obd->o_flags, obd->o_misc,
217                  obd->o_nlink, (uintmax_t)obd->o_valid);
218         return strdup(buf);
219 }
220
221
222 #define BAD_VERBOSE (-999999999)
223
224 #define N2D_OFF 0x100      /* So we can tell between error codes and devices */
225
226 static int do_name2dev(char *func, char *name)
227 {
228         struct obd_ioctl_data data;
229         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
230         int rc;
231
232         memset(&data, 0, sizeof(data));
233         data.ioc_dev = cur_device;
234         data.ioc_inllen1 = strlen(name) + 1;
235         data.ioc_inlbuf1 = name;
236
237         memset(buf, 0, sizeof(rawbuf));
238         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
239         if (rc < 0) {
240                 fprintf(stderr, "error: %s: invalid ioctl\n",
241                         jt_cmdname(func));
242                 return -rc;
243         }
244         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_NAME2DEV, buf);
245         if (rc < 0)
246                 return errno;
247         rc = obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
248         if (rc) {
249                 fprintf(stderr, "error: %s: invalid reply\n",
250                         jt_cmdname(func));
251                 return rc;
252         }
253
254         return data.ioc_dev + N2D_OFF;
255 }
256
257 /*
258  * resolve a device name to a device number.
259  * supports a number, $name or %uuid.
260  */
261 int parse_devname(char *func, char *name)
262 {
263         int rc;
264         int ret = -1;
265
266         if (!name)
267                 return ret;
268         if (isdigit(name[0])) {
269                 ret = strtoul(name, NULL, 0);
270         } else {
271                 if (name[0] == '$' || name[0] == '%')
272                         name++;
273                 rc = do_name2dev(func, name);
274                 if (rc >= N2D_OFF) {
275                         ret = rc - N2D_OFF;
276                         // printf("Name %s is device %d\n", name, ret);
277                 } else {
278                         fprintf(stderr, "No device found for name %s: %s\n",
279                                 name, strerror(rc));
280                 }
281         }
282         return ret;
283 }
284
285 char *jt_cmdname(char *func)
286 {
287         static char buf[512];
288
289         if (thread) {
290                 sprintf(buf, "%s-%d", func, thread);
291                 return buf;
292         }
293
294         return func;
295 }
296
297 #define difftime(a, b)                                  \
298         ((a)->tv_sec - (b)->tv_sec +                    \
299          ((a)->tv_usec - (b)->tv_usec) / 1000000.0)
300
301 static int be_verbose(int verbose, struct timeval *next_time,
302                       __u64 num, __u64 *next_num, int num_total)
303 {
304         struct timeval now;
305
306         if (!verbose)
307                 return 0;
308
309         if (next_time != NULL)
310                 gettimeofday(&now, NULL);
311
312         /* A positive verbosity means to print every X iterations */
313         if (verbose > 0 && (num >= *next_num || num >= num_total)) {
314                 *next_num += verbose;
315                 if (next_time) {
316                         next_time->tv_sec = now.tv_sec - verbose;
317                         next_time->tv_usec = now.tv_usec;
318                 }
319                 return 1;
320         }
321
322         /* A negative verbosity means to print at most each X seconds */
323         if (verbose < 0 && next_time != NULL &&
324             difftime(&now, next_time) >= 0.0){
325                 next_time->tv_sec = now.tv_sec - verbose;
326                 next_time->tv_usec = now.tv_usec;
327                 *next_num = num;
328                 return 1;
329         }
330
331         return 0;
332 }
333
334 static int get_verbose(char *func, const char *arg)
335 {
336         int verbose;
337         char *end;
338
339         if (!arg || arg[0] == 'v')
340                 verbose = 1;
341         else if (arg[0] == 's' || arg[0] == 'q')
342                 verbose = 0;
343         else {
344                 verbose = (int)strtoul(arg, &end, 0);
345                 if (*end) {
346                         fprintf(stderr, "error: %s: bad verbose option '%s'\n",
347                                 jt_cmdname(func), arg);
348                         return BAD_VERBOSE;
349                 }
350         }
351
352         if (verbose < 0)
353                 printf("Print status every %d seconds\n", -verbose);
354         else if (verbose == 1)
355                 printf("Print status every operation\n");
356         else if (verbose > 1)
357                 printf("Print status every %d operations\n", verbose);
358
359         return verbose;
360 }
361
362 int do_disconnect(char *func, int verbose)
363 {
364         lcfg_set_devname(NULL);
365         cur_device = -1;
366         return 0;
367 }
368
369 #ifdef MAX_THREADS
370 static int shmem_setup(void)
371 {
372         pthread_mutexattr_t mattr;
373         pthread_condattr_t  cattr;
374         int                 rc;
375         int                 shmid;
376
377         /* Create new segment */
378         shmid = shmget(IPC_PRIVATE, sizeof(*shared_data), 0600);
379         if (shmid == -1) {
380                 fprintf(stderr, "Can't create shared data: %s\n",
381                         strerror(errno));
382                 return errno;
383         }
384
385         /* Attatch to new segment */
386         shared_data = (struct shared_data *)shmat(shmid, NULL, 0);
387
388         if (shared_data == (struct shared_data *)(-1)) {
389                 fprintf(stderr, "Can't attach shared data: %s\n",
390                         strerror(errno));
391                 shared_data = NULL;
392                 return errno;
393         }
394
395         /* Mark segment as destroyed, so it will disappear when we exit.
396          * Forks will inherit attached segments, so we should be OK.
397          */
398         if (shmctl(shmid, IPC_RMID, NULL) == -1) {
399                 fprintf(stderr, "Can't destroy shared data: %s\n",
400                         strerror(errno));
401                 return errno;
402         }
403
404         pthread_mutexattr_init(&mattr);
405         pthread_condattr_init(&cattr);
406
407         rc = pthread_mutexattr_setpshared(&mattr, PTHREAD_PROCESS_SHARED);
408         if (rc != 0) {
409                 fprintf(stderr, "Can't set shared mutex attr\n");
410                 goto out;
411         }
412
413         rc = pthread_condattr_setpshared(&cattr, PTHREAD_PROCESS_SHARED);
414         if (rc != 0) {
415                 fprintf(stderr, "Can't set shared cond attr\n");
416                 goto out;
417         }
418
419         pthread_mutex_init(&shared_data->mutex, &mattr);
420         pthread_cond_init(&shared_data->cond, &cattr);
421 out:
422         pthread_mutexattr_destroy(&mattr);
423         pthread_condattr_destroy(&cattr);
424
425         return rc;
426 }
427
428 static inline void shmem_lock(void)
429 {
430         pthread_mutex_lock(&shared_data->mutex);
431 }
432
433 static inline void shmem_unlock(void)
434 {
435         pthread_mutex_unlock(&shared_data->mutex);
436 }
437
438 static inline void shmem_wait(void)
439 {
440         pthread_cond_wait(&shared_data->cond, &shared_data->mutex);
441 }
442
443 static inline void shmem_wakeup_all(void)
444 {
445         pthread_cond_broadcast(&shared_data->cond);
446 }
447
448 static inline void shmem_reset(int total_threads)
449 {
450         if (shared_data == NULL)
451                 return;
452
453         memset(&shared_data->body, 0, sizeof(shared_data->body));
454         memset(counter_snapshot, 0, sizeof(counter_snapshot));
455         prev_valid = 0;
456         shared_data->stopping = 0;
457         shared_data->body.start_barrier = total_threads;
458         shared_data->body.stop_barrier = total_threads;
459 }
460
461 static inline void shmem_bump(__u32 counter)
462 {
463         static bool running_not_bumped = true;
464
465         if (shared_data == NULL || thread <= 0 || thread > MAX_THREADS)
466                 return;
467
468         shmem_lock();
469         shared_data->body.counters[thread - 1] += counter;
470         if (running_not_bumped) {
471                 shared_data->body.thr_running++;
472                 running_not_bumped = false;
473         }
474         shmem_unlock();
475 }
476
477 static void shmem_total(int total_threads)
478 {
479         __u64 total = 0;
480         double secs;
481         int i;
482
483         if (shared_data == NULL || total_threads > MAX_THREADS)
484                 return;
485
486         shmem_lock();
487         for (i = 0; i < total_threads; i++)
488                 total += shared_data->body.counters[i];
489
490         secs = difftime(&shared_data->body.end_time,
491                         &shared_data->body.start_time);
492         shmem_unlock();
493
494         printf("Total: total %ju threads %d sec %f %f/second\n",
495                (uintmax_t)total, total_threads, secs, total / secs);
496
497         return;
498 }
499
500 static void shmem_snap(int total_threads, int live_threads)
501 {
502         struct timeval this_time;
503         int non_zero = 0;
504         __u64 total = 0;
505         double secs;
506         int running;
507         int i;
508
509         if (shared_data == NULL || total_threads > MAX_THREADS)
510                 return;
511
512         shmem_lock();
513         memcpy(counter_snapshot[0], shared_data->body.counters,
514                total_threads * sizeof(counter_snapshot[0][0]));
515         running = shared_data->body.thr_running;
516         shmem_unlock();
517
518         gettimeofday(&this_time, NULL);
519
520         for (i = 0; i < total_threads; i++) {
521                 long long this_count =
522                         counter_snapshot[0][i] - counter_snapshot[1][i];
523
524                 if (this_count != 0) {
525                         non_zero++;
526                         total += this_count;
527                 }
528         }
529
530         secs = difftime(&this_time, &prev_time);
531         if (prev_valid && secs > 1.0)    /* someone screwed with the time? */
532                 printf("%d/%d Total: %f/second\n", non_zero, total_threads,
533                        total / secs);
534
535         memcpy(counter_snapshot[1], counter_snapshot[0],
536                total_threads * sizeof(counter_snapshot[0][0]));
537         prev_time = this_time;
538         if (!prev_valid &&
539             running == total_threads)
540                 prev_valid = 1;
541 }
542
543 static void shmem_stop(void)
544 {
545         if (shared_data == NULL)
546                 return;
547
548         shared_data->stopping = 1;
549 }
550
551 static void shmem_cleanup(void)
552 {
553         if (shared_data == NULL)
554                 return;
555
556         shmem_stop();
557
558         pthread_mutex_destroy(&shared_data->mutex);
559         pthread_cond_destroy(&shared_data->cond);
560 }
561
562 static int shmem_running(void)
563 {
564         return (shared_data == NULL || !shared_data->stopping);
565 }
566
567 static void shmem_end_time_locked(void)
568 {
569         shared_data->body.stop_barrier--;
570         if (shared_data->body.stop_barrier == 0)
571                 gettimeofday(&shared_data->body.end_time, NULL);
572 }
573
574 static void shmem_start_time_locked(void)
575 {
576         shared_data->body.start_barrier--;
577         if (shared_data->body.start_barrier == 0) {
578                 shmem_wakeup_all();
579                 gettimeofday(&shared_data->body.start_time, NULL);
580         } else {
581                 shmem_wait();
582         }
583 }
584
585 #else
586 static int shmem_setup(void)
587 {
588         return 0;
589 }
590
591 static inline void shmem_reset(int total_threads)
592 {
593 }
594
595 static inline void shmem_bump(__u32 counters)
596 {
597 }
598
599 static void shmem_lock()
600 {
601 }
602
603 static void shmem_unlock()
604 {
605 }
606
607 static void shmem_cleanup(void)
608 {
609 }
610
611 static int shmem_running(void)
612 {
613         return 1;
614 }
615 #endif
616
617 extern command_t cmdlist[];
618
619 static int do_device(char *func, char *devname)
620 {
621         int dev;
622
623         dev = parse_devname(func, devname);
624         if (dev < 0)
625                 return -1;
626
627         lcfg_set_devname(devname);
628         cur_device = dev;
629         return 0;
630 }
631
632 int jt_obd_get_device()
633 {
634         return cur_device;
635 }
636
637 int jt_obd_device(int argc, char **argv)
638 {
639         int rc;
640
641         if (argc > 2)
642                 return CMD_HELP;
643
644         if (argc == 1) {
645                 printf("current device is %d - %s\n",
646                        cur_device, lcfg_get_devname() ? : "not set");
647                 return 0;
648         }
649         rc = do_device("device", argv[1]);
650         return rc;
651 }
652
653 int jt_opt_device(int argc, char **argv)
654 {
655         int ret;
656         int rc;
657
658         if (argc < 3)
659                 return CMD_HELP;
660
661         rc = do_device("device", argv[1]);
662
663         if (!rc)
664                 rc = Parser_execarg(argc - 2, argv + 2, cmdlist);
665
666         ret = do_disconnect(argv[0], 0);
667         if (!rc)
668                 rc = ret;
669
670         return rc;
671 }
672
673 #ifdef MAX_THREADS
674 static void parent_sighandler (int sig)
675 {
676         return;
677 }
678
679 int jt_opt_threads(int argc, char **argv)
680 {
681         static char      cmdstr[128];
682         sigset_t         saveset;
683         sigset_t         sigset;
684         struct sigaction sigact;
685         struct sigaction saveact1;
686         struct sigaction saveact2;
687         unsigned long    threads;
688         __u64            next_thread;
689         int verbose;
690         int rc = 0;
691         int report_count = -1;
692         char *end;
693         int i;
694
695         if (argc < 5)
696                 return CMD_HELP;
697
698         threads = strtoul(argv[1], &end, 0);
699
700         if (*end == '.')
701                 report_count = strtoul(end + 1, &end, 0);
702
703         if (*end || threads > MAX_THREADS) {
704                 fprintf(stderr, "error: %s: invalid thread count '%s'\n",
705                         jt_cmdname(argv[0]), argv[1]);
706                 return CMD_HELP;
707         }
708
709         verbose = get_verbose(argv[0], argv[2]);
710         if (verbose == BAD_VERBOSE)
711                 return CMD_HELP;
712
713         if (verbose != 0) {
714                 snprintf(cmdstr, sizeof(cmdstr), "%s", argv[4]);
715                 for (i = 5; i < argc; i++)
716                         snprintf(cmdstr + strlen(cmdstr), sizeof(cmdstr),
717                                  " %s", argv[i]);
718
719                 printf("%s: starting %ld threads on device %s running %s\n",
720                        argv[0], threads, argv[3], cmdstr);
721         }
722
723         shmem_reset(threads);
724
725         sigemptyset(&sigset);
726         sigaddset(&sigset, SIGALRM);
727         sigaddset(&sigset, SIGCHLD);
728         sigprocmask(SIG_BLOCK, &sigset, &saveset);
729
730         nthreads = threads;
731
732         for (i = 1, next_thread = verbose; i <= threads; i++) {
733                 rc = fork();
734                 if (rc < 0) {
735                         fprintf(stderr, "error: %s: #%d - %s\n", argv[0], i,
736                                 strerror(rc = errno));
737                         break;
738                 } else if (rc == 0) {
739                         sigprocmask(SIG_SETMASK, &saveset, NULL);
740
741                         thread = i;
742                         argv[2] = "--device";
743                         exit(jt_opt_device(argc - 2, argv + 2));
744                 } else if (be_verbose(verbose, NULL, i, &next_thread, threads))
745                         printf("%s: thread #%d (PID %d) started\n",
746                                argv[0], i, rc);
747                 rc = 0;
748         }
749
750         if (!thread) {          /* parent process */
751                 int live_threads = threads;
752
753                 sigemptyset(&sigset);
754                 sigemptyset(&sigact.sa_mask);
755                 sigact.sa_handler = parent_sighandler;
756                 sigact.sa_flags = 0;
757
758                 sigaction(SIGALRM, &sigact, &saveact1);
759                 sigaction(SIGCHLD, &sigact, &saveact2);
760
761                 while (live_threads > 0) {
762                         int status;
763                         pid_t ret;
764
765                         if (verbose < 0)        /* periodic stats */
766                                 alarm(-verbose);
767
768                         sigsuspend(&sigset);
769                         alarm(0);
770
771                         while (live_threads > 0) {
772                                 ret = waitpid(0, &status, WNOHANG);
773                                 if (ret == 0)
774                                         break;
775
776                                 if (ret < 0) {
777                                         fprintf(stderr, "error: %s: wait - %s\n",
778                                                 argv[0], strerror(errno));
779                                         if (!rc)
780                                                 rc = errno;
781                                         continue;
782                                 } else {
783                                         /*
784                                          * This is a hack.  We _should_ be able
785                                          * to use WIFEXITED(status) to see if
786                                          * there was an error, but it appears
787                                          * to be broken and it always returns 1
788                                          * (OK).  See wait(2).
789                                          */
790                                         int err = WEXITSTATUS(status);
791                                         if (err || WIFSIGNALED(status))
792                                                 fprintf(stderr,
793                                                         "%s: PID %d had rc=%d\n",
794                                                         argv[0], ret, err);
795                                         if (!rc)
796                                                 rc = err;
797
798                                         live_threads--;
799                                 }
800                         }
801
802                         /* Show stats while all threads running */
803                         if (verbose < 0) {
804                                 shmem_snap(threads, live_threads);
805                                 if (report_count > 0 && --report_count == 0)
806                                         shmem_stop();
807                         }
808                 }
809                 sigaction(SIGCHLD, &saveact2, NULL);
810                 sigaction(SIGALRM, &saveact1, NULL);
811         }
812
813         shmem_total(threads);
814         sigprocmask(SIG_SETMASK, &saveset, NULL);
815
816         return rc;
817 }
818 #else
819 int jt_opt_threads(int argc, char **argv)
820 {
821         fprintf(stderr, "%s not-supported in a single-threaded runtime\n",
822                 jt_cmdname(argv[0]));
823         return CMD_HELP;
824 }
825 #endif
826
827 int jt_opt_net(int argc, char **argv)
828 {
829         char *arg2[3];
830         int rc;
831
832         if (argc < 3)
833                 return CMD_HELP;
834
835         arg2[0] = argv[0];
836         arg2[1] = argv[1];
837         arg2[2] = NULL;
838         rc = jt_ptl_network (2, arg2);
839
840         if (!rc)
841                 rc = Parser_execarg(argc - 2, argv + 2, cmdlist);
842
843         return rc;
844 }
845
846 int jt_obd_no_transno(int argc, char **argv)
847 {
848         struct obd_ioctl_data data;
849         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
850         int rc;
851
852         memset(&data, 0, sizeof(data));
853         data.ioc_dev = cur_device;
854
855         if (argc != 1)
856                 return CMD_HELP;
857
858         memset(buf, 0, sizeof(rawbuf));
859         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
860         if (rc) {
861                 fprintf(stderr, "error: %s: invalid ioctl\n",
862                         jt_cmdname(argv[0]));
863                 return rc;
864         }
865         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_NO_TRANSNO, buf);
866         if (rc < 0)
867                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
868                         strerror(rc = errno));
869
870         return rc;
871 }
872
873 int jt_obd_set_readonly(int argc, char **argv)
874 {
875         struct obd_ioctl_data data;
876         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
877         int rc;
878
879         memset(&data, 0, sizeof(data));
880         data.ioc_dev = cur_device;
881
882         if (argc != 1)
883                 return CMD_HELP;
884
885         memset(buf, 0, sizeof(rawbuf));
886         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
887         if (rc) {
888                 fprintf(stderr, "error: %s: invalid ioctl\n",
889                         jt_cmdname(argv[0]));
890                 return rc;
891         }
892         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_SET_READONLY, buf);
893         if (rc < 0)
894                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
895                         strerror(rc = errno));
896
897         return rc;
898 }
899
900 int jt_obd_abort_recovery(int argc, char **argv)
901 {
902         struct obd_ioctl_data data;
903         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
904         int rc;
905
906         memset(&data, 0, sizeof(data));
907         data.ioc_dev = cur_device;
908
909         if (argc != 1)
910                 return CMD_HELP;
911
912         memset(buf, 0, sizeof(rawbuf));
913         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
914         if (rc) {
915                 fprintf(stderr, "error: %s: invalid ioctl\n",
916                         jt_cmdname(argv[0]));
917                 return rc;
918         }
919         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_ABORT_RECOVERY, buf);
920         if (rc < 0)
921                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
922                         strerror(rc = errno));
923
924         return rc;
925 }
926
927 int jt_get_version(int argc, char **argv)
928 {
929         char version[128];
930         int rc;
931
932         if (argc != 1)
933                 return CMD_HELP;
934
935         rc = llapi_get_version_string(version, sizeof(version));
936         if (rc)
937                 printf("Lustre version: %s\n", LUSTRE_VERSION_STRING);
938         else
939                 printf("Lustre version: %s\n", version);
940
941         return 0;
942 }
943
944 static void print_obd_line(char *s)
945 {
946         const char *param = "osc/%s/ost_conn_uuid";
947         char buf[MAX_STRING_SIZE];
948         char obd_name[MAX_OBD_NAME];
949         FILE *fp = NULL;
950         glob_t path;
951         char *ptr;
952 retry:
953         /* obd device type is the first 3 characters of param name */
954         snprintf(buf, sizeof(buf), " %%*d %%*s %.3s %%%zus %%*s %%*d ",
955                  param, sizeof(obd_name) - 1);
956         if (sscanf(s, buf, obd_name) == 0)
957                 goto try_mdc;
958         if (cfs_get_param_paths(&path, param, obd_name) != 0)
959                 goto try_mdc;
960         fp = fopen(path.gl_pathv[0], "r");
961         if (fp == NULL) {
962                 /* need to free path data before retry */
963                 cfs_free_param_data(&path);
964 try_mdc:
965                 if (param[0] == 'o') { /* failed with osc, try mdc */
966                         param = "mdc/%s/mds_conn_uuid";
967                         goto retry;
968                 }
969                 buf[0] = '\0';
970                 goto fail_print;
971         }
972
973         /* should not ignore fgets(3)'s return value */
974         if (!fgets(buf, sizeof(buf), fp)) {
975                 fprintf(stderr, "reading from %s: %s", buf, strerror(errno));
976                 goto fail_close;
977         }
978
979 fail_close:
980         fclose(fp);
981         cfs_free_param_data(&path);
982
983         /* trim trailing newlines */
984         ptr = strrchr(buf, '\n');
985         if (ptr)
986                 *ptr = '\0';
987 fail_print:
988         ptr = strrchr(s, '\n');
989         if (ptr)
990                 *ptr = '\0';
991         printf("%s%s%s\n", s, buf[0] ? " " : "", buf);
992 }
993
994 /* get device list by ioctl */
995 int jt_obd_list_ioctl(int argc, char **argv)
996 {
997         int rc, index;
998         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
999         struct obd_ioctl_data *data = (struct obd_ioctl_data *)buf;
1000
1001         if (argc > 2)
1002                 return CMD_HELP;
1003         /* Just ignore a -t option.  Only supported with /proc. */
1004         else if (argc == 2 && strcmp(argv[1], "-t") != 0)
1005                 return CMD_HELP;
1006
1007         for (index = 0;; index++) {
1008                 memset(buf, 0, sizeof(rawbuf));
1009                 data->ioc_version = OBD_IOCTL_VERSION;
1010                 data->ioc_inllen1 =
1011                         sizeof(rawbuf) - cfs_size_round(sizeof(*data));
1012                 data->ioc_inlbuf1 = buf + cfs_size_round(sizeof(*data));
1013                 data->ioc_len = obd_ioctl_packlen(data);
1014                 data->ioc_count = index;
1015
1016                 rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_GETDEVICE, buf);
1017                 if (rc != 0)
1018                         break;
1019                 printf("%s\n", (char *)data->ioc_bulk);
1020         }
1021         if (rc != 0) {
1022                 if (errno == ENOENT)
1023                         /* no device or the last device */
1024                         rc = 0;
1025                 else
1026                         fprintf(stderr, "Error getting device list: %s: "
1027                                 "check dmesg.\n", strerror(errno));
1028         }
1029         return rc;
1030 }
1031
1032 int jt_obd_list(int argc, char **argv)
1033 {
1034         char buf[MAX_STRING_SIZE];
1035         int print_obd = 0;
1036         glob_t path;
1037         FILE *fp;
1038
1039         if (argc > 2)
1040                 return CMD_HELP;
1041         else if (argc == 2) {
1042                 if (strcmp(argv[1], "-t") == 0)
1043                         print_obd = 1;
1044                 else
1045                         return CMD_HELP;
1046         }
1047
1048         if (cfs_get_param_paths(&path, "devices") != 0)
1049                 return -errno;
1050
1051         fp = fopen(path.gl_pathv[0], "r");
1052         if (fp == NULL) {
1053                 fprintf(stderr, "error: %s: %s opening %s\n",
1054                         jt_cmdname(argv[0]), strerror(errno), path.gl_pathv[0]);
1055                 cfs_free_param_data(&path);
1056                 return jt_obd_list_ioctl(argc, argv);
1057         }
1058
1059         while (fgets(buf, sizeof(buf), fp) != NULL)
1060                 if (print_obd)
1061                         print_obd_line(buf);
1062                 else
1063                         printf("%s", buf);
1064
1065         cfs_free_param_data(&path);
1066         fclose(fp);
1067         return 0;
1068 }
1069
1070 struct jt_fid_space {
1071         __u64   jt_seq;
1072         __u64   jt_id;
1073         int     jt_width;
1074 };
1075
1076 int jt_obd_alloc_fids(struct jt_fid_space *space, struct lu_fid *fid,
1077                       __u64 *count)
1078 {
1079         int rc;
1080
1081         if (space->jt_seq == 0 || space->jt_id == space->jt_width) {
1082                 struct obd_ioctl_data  data;
1083                 char rawbuf[MAX_IOC_BUFLEN];
1084                 char *buf = rawbuf;
1085                 __u64 seqnr;
1086                 int max_count;
1087
1088                 memset(&data, 0, sizeof(data));
1089                 data.ioc_dev = cur_device;
1090
1091                 data.ioc_pbuf1 = (char *)&seqnr;
1092                 data.ioc_plen1 = sizeof(seqnr);
1093
1094                 data.ioc_pbuf2 = (char *)&max_count;
1095                 data.ioc_plen2 = sizeof(max_count);
1096
1097                 memset(buf, 0, sizeof(rawbuf));
1098                 rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1099                 if (rc) {
1100                         fprintf(stderr, "error: invalid ioctl rc = %d\n", rc);
1101                         return rc;
1102                 }
1103
1104                 rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_ECHO_ALLOC_SEQ, buf);
1105                 if (rc) {
1106                         fprintf(stderr, "ioctl error: rc = %d\n", rc);
1107                         return rc;
1108                 }
1109
1110                 space->jt_seq = *(__u64 *)data.ioc_pbuf1;
1111                 space->jt_width = *(int *)data.ioc_pbuf2;
1112                 space->jt_id = 1;
1113         }
1114         fid->f_seq = space->jt_seq;
1115         fid->f_oid = space->jt_id;
1116         fid->f_ver = 0;
1117
1118         space->jt_id = space->jt_id + *count;
1119         if (space->jt_id > space->jt_width)
1120                 space->jt_id = space->jt_width;
1121
1122         *count = space->jt_id - fid->f_oid;
1123         return 0;
1124 }
1125
1126 #define MD_STEP_COUNT 1000
1127 int jt_obd_md_common(int argc, char **argv, int cmd)
1128 {
1129         struct obd_ioctl_data  data;
1130         struct timeval         start;
1131         struct timeval         end_time;
1132         char                   rawbuf[MAX_IOC_BUFLEN];
1133         char                  *buf = rawbuf;
1134         int                    mode = 0000644;
1135         int                    create_mode;
1136         int                    rc = 0;
1137         char                  *parent_basedir = NULL;
1138         char                   dirname[4096];
1139         int                    parent_base_id = 0;
1140         int                    parent_count = 1;
1141         __u64                  child_base_id = -1;
1142         int                    stripe_count = 0;
1143         int                    stripe_index = -1;
1144         int                    count = 0;
1145         char                  *end;
1146         __u64                  seconds = 0;
1147         double                 diff;
1148         int                    c;
1149         __u64                  total_count = 0;
1150         char                  *name = NULL;
1151         struct jt_fid_space    fid_space = {0};
1152         int                    version = 0;
1153         struct option          long_opts[] = {
1154                 {"child_base_id",     required_argument, 0, 'b'},
1155                 {"stripe_count",      required_argument, 0, 'c'},
1156                 {"parent_basedir",    required_argument, 0, 'd'},
1157                 {"parent_dircount",   required_argument, 0, 'D'},
1158                 {"stripe_index",      required_argument, 0, 'i'},
1159                 {"mode",              required_argument, 0, 'm'},
1160                 {"count",             required_argument, 0, 'n'},
1161                 {"time",              required_argument, 0, 't'},
1162                 {"version",           no_argument,       0, 'v'},
1163                 {0, 0, 0, 0}
1164         };
1165
1166         while ((c = getopt_long(argc, argv, "b:c:d:D:m:n:t:v",
1167                                 long_opts, NULL)) >= 0) {
1168                 switch (c) {
1169                 case 'b':
1170                         child_base_id = strtoull(optarg, &end, 0);
1171                         if (*end) {
1172                                 fprintf(stderr, "error: %s: bad child_base_id"
1173                                         " '%s'\n", jt_cmdname(argv[0]), optarg);
1174                                 return CMD_HELP;
1175                         }
1176                         break;
1177                 case 'c':
1178                         stripe_count = strtoul(optarg, &end, 0);
1179                         if (*end) {
1180                                 fprintf(stderr, "error: %s: bad stripe count"
1181                                         " '%s'\n", jt_cmdname(argv[0]), optarg);
1182                                 return CMD_HELP;
1183                         }
1184                         break;
1185                 case 'd':
1186                         parent_basedir = optarg;
1187                         break;
1188                 case 'D':
1189                         parent_count = strtoul(optarg, &end, 0);
1190                         if (*end) {
1191                                 fprintf(stderr, "error: %s: bad parent count"
1192                                         " '%s'\n", jt_cmdname(argv[0]), optarg);
1193                                 return CMD_HELP;
1194                         }
1195                         break;
1196                 case 'i':
1197                         stripe_index = strtoul(optarg, &end, 0);
1198                         if (*end) {
1199                                 fprintf(stderr, "error: %s: bad stripe index"
1200                                         " '%s'\n", jt_cmdname(argv[0]), optarg);
1201                                 return CMD_HELP;
1202                         }
1203                         break;
1204                 case 'm':
1205                         mode = strtoul(optarg, &end, 0);
1206                         if (*end) {
1207                                 fprintf(stderr, "error: %s: bad mode '%s'\n",
1208                                         jt_cmdname(argv[0]), optarg);
1209                                 return CMD_HELP;
1210                         }
1211                         break;
1212                 case 'n':
1213                         total_count = strtoul(optarg, &end, 0);
1214                         if (*end || total_count == 0) {
1215                                 fprintf(stderr, "%s: bad child count '%s'\n",
1216                                         jt_cmdname(argv[0]), optarg);
1217                                 return CMD_HELP;
1218                         }
1219                         break;
1220                 case 't':
1221                         seconds = strtoull(optarg, &end, 0);
1222                         if (*end) {
1223                                 fprintf(stderr, "error: %s: seconds '%s'\n",
1224                                         jt_cmdname(argv[0]), optarg);
1225                                 return CMD_HELP;
1226                         }
1227                         break;
1228                 case 'v':
1229                         version = 1;
1230                         break;
1231                 default:
1232                         fprintf(stderr, "error: %s: option '%s' "
1233                                 "unrecognized\n", argv[0], argv[optind - 1]);
1234                         return CMD_HELP;
1235                 }
1236         }
1237
1238         memset(&data, 0, sizeof(data));
1239         data.ioc_dev = cur_device;
1240         if (child_base_id == -1) {
1241                 if (optind >= argc)
1242                         return CMD_HELP;
1243                 name = argv[optind];
1244                 total_count = 1;
1245         } else {
1246                 if (optind < argc) {
1247                         fprintf(stderr, "child_base_id and name can not"
1248                                         " specified at the same time\n");
1249                         return CMD_HELP;
1250                 }
1251         }
1252
1253         if (stripe_count == 0 && stripe_index != -1) {
1254                 fprintf(stderr, "If stripe_count is 0, stripe_index can not"
1255                                 "be specified\n");
1256                 return CMD_HELP;
1257         }
1258
1259         if (total_count == 0 && seconds == 0) {
1260                 fprintf(stderr, "count or seconds needs to be indicated\n");
1261                 return CMD_HELP;
1262         }
1263
1264         if (parent_count <= 0) {
1265                 fprintf(stderr, "parent count must < 0\n");
1266                 return CMD_HELP;
1267         }
1268
1269 #ifdef MAX_THREADS
1270         if (thread) {
1271                 shmem_lock();
1272                 /* threads interleave */
1273                 if (parent_base_id != -1)
1274                         parent_base_id += (thread - 1) % parent_count;
1275
1276                 if (child_base_id != -1)
1277                         child_base_id +=  (thread - 1) * \
1278                                           (MAX_BASE_ID / nthreads);
1279
1280                 shmem_start_time_locked();
1281                 shmem_unlock();
1282         }
1283 #endif
1284         /* If parent directory is not specified, try to get the directory
1285          * from name */
1286         if (parent_basedir == NULL) {
1287                 char *last_lash;
1288                 if (name == NULL) {
1289                         fprintf(stderr, "parent_basedir or name must be"
1290                                         "indicated!\n");
1291                         return CMD_HELP;
1292                 }
1293                 /*Get directory and name from name*/
1294                 last_lash = strrchr(name, '/');
1295                 if (last_lash == NULL || name[0] != '/') {
1296                         fprintf(stderr, "Can not locate %s\n", name);
1297                         return CMD_HELP;
1298                 }
1299
1300                 if (last_lash == name) {
1301                         sprintf(dirname, "%s", "/");
1302                         name++;
1303                 } else {
1304                         int namelen = (unsigned long)last_lash -
1305                                       (unsigned long)name + 1;
1306                         snprintf(dirname, namelen, "%s", name);
1307                         name = last_lash + 1;
1308                 }
1309
1310                 data.ioc_pbuf1 = dirname;
1311                 data.ioc_plen1 = strlen(dirname);
1312
1313                 data.ioc_pbuf2 = name;
1314                 data.ioc_plen2 = strlen(name);
1315         } else {
1316                 if (name != NULL) {
1317                         data.ioc_pbuf2 = name;
1318                         data.ioc_plen2 = strlen(name);
1319                 }
1320                 if (parent_base_id > 0)
1321                         sprintf(dirname, "%s%d", parent_basedir,
1322                                 parent_base_id);
1323                 else
1324                         sprintf(dirname, "%s", parent_basedir);
1325                 data.ioc_pbuf1 = dirname;
1326                 data.ioc_plen1 = strlen(dirname);
1327         }
1328
1329         if (cmd == ECHO_MD_MKDIR || cmd == ECHO_MD_RMDIR)
1330                 create_mode = S_IFDIR;
1331         else
1332                 create_mode = S_IFREG;
1333
1334         data.ioc_obdo1.o_mode = mode | S_IFDIR;
1335         data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLTYPE | OBD_MD_FLMODE |
1336                                  OBD_MD_FLFLAGS | OBD_MD_FLGROUP;
1337         data.ioc_command = cmd;
1338
1339         gettimeofday(&start, NULL);
1340         while (shmem_running()) {
1341                 struct lu_fid fid = { 0 };
1342
1343                 if (child_base_id != -1)
1344                         data.ioc_obdo2.o_oi.oi.oi_id = child_base_id;
1345                 data.ioc_obdo2.o_mode = mode | create_mode;
1346                 data.ioc_obdo2.o_valid = OBD_MD_FLID | OBD_MD_FLTYPE |
1347                                          OBD_MD_FLMODE | OBD_MD_FLFLAGS |
1348                                          OBD_MD_FLGROUP;
1349                 data.ioc_obdo2.o_misc = stripe_count;
1350                 data.ioc_obdo2.o_stripe_idx = stripe_index;
1351
1352                 if (total_count > 0) {
1353                         if ((total_count - count) > MD_STEP_COUNT)
1354                                 data.ioc_count = MD_STEP_COUNT;
1355                         else
1356                                 data.ioc_count = total_count - count;
1357                 } else {
1358                         data.ioc_count = MD_STEP_COUNT;
1359                 }
1360
1361                 if (cmd == ECHO_MD_CREATE || cmd == ECHO_MD_MKDIR) {
1362                         /*Allocate fids for the create */
1363                         rc = jt_obd_alloc_fids(&fid_space, &fid,
1364                                                &data.ioc_count);
1365                         if (rc) {
1366                                 fprintf(stderr, "Allocate fids error %d.\n",rc);
1367                                 return rc;
1368                         }
1369                         fid_to_ostid(&fid, &data.ioc_obdo1.o_oi);
1370                 }
1371
1372                 child_base_id += data.ioc_count;
1373                 count += data.ioc_count;
1374
1375                 memset(buf, 0, sizeof(rawbuf));
1376                 rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1377                 if (rc) {
1378                         fprintf(stderr, "error: %s: invalid ioctl %d\n",
1379                                 jt_cmdname(argv[0]), rc);
1380                         return rc;
1381                 }
1382
1383                 rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_ECHO_MD, buf);
1384                 if (rc) {
1385                         fprintf(stderr, "error: %s: %s\n",
1386                                 jt_cmdname(argv[0]), strerror(rc = errno));
1387                         return rc;
1388                 }
1389                 shmem_bump(data.ioc_count);
1390
1391                 gettimeofday(&end_time, NULL);
1392                 diff = difftime(&end_time, &start);
1393                 if (seconds > 0 && (__u64)diff > seconds)
1394                         break;
1395
1396                 if (count >= total_count && total_count > 0)
1397                         break;
1398         }
1399
1400         if (count > 0 && version) {
1401                 gettimeofday(&end_time, NULL);
1402                 diff = difftime(&end_time, &start);
1403                 printf("%s: %d in %.3fs (%.3f /s): %s",
1404                         jt_cmdname(argv[0]), count, diff,
1405                         (double)count/diff, ctime(&end_time.tv_sec));
1406         }
1407
1408 #ifdef MAX_THREADS
1409         if (thread) {
1410                 shmem_lock();
1411                 shmem_end_time_locked();
1412                 shmem_unlock();
1413         }
1414 #endif
1415         return rc;
1416 }
1417
1418 int jt_obd_test_create(int argc, char **argv)
1419 {
1420         return jt_obd_md_common(argc, argv, ECHO_MD_CREATE);
1421 }
1422
1423 int jt_obd_test_mkdir(int argc, char **argv)
1424 {
1425         return jt_obd_md_common(argc, argv, ECHO_MD_MKDIR);
1426 }
1427
1428 int jt_obd_test_destroy(int argc, char **argv)
1429 {
1430         return jt_obd_md_common(argc, argv, ECHO_MD_DESTROY);
1431 }
1432
1433 int jt_obd_test_rmdir(int argc, char **argv)
1434 {
1435         return jt_obd_md_common(argc, argv, ECHO_MD_RMDIR);
1436 }
1437
1438 int jt_obd_test_lookup(int argc, char **argv)
1439 {
1440         return jt_obd_md_common(argc, argv, ECHO_MD_LOOKUP);
1441 }
1442
1443 int jt_obd_test_setxattr(int argc, char **argv)
1444 {
1445         return jt_obd_md_common(argc, argv, ECHO_MD_SETATTR);
1446 }
1447
1448 int jt_obd_test_md_getattr(int argc, char **argv)
1449 {
1450         return jt_obd_md_common(argc, argv, ECHO_MD_GETATTR);
1451 }
1452
1453 int jt_obd_create(int argc, char **argv)
1454 {
1455         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1456         struct obd_ioctl_data data;
1457         struct timeval next_time;
1458         __u64 count = 1, next_count, base_id = 1;
1459         int verbose = 1, mode = 0100644, rc = 0, i;
1460         char *end;
1461
1462         memset(&data, 0, sizeof(data));
1463         data.ioc_dev = cur_device;
1464         if (argc < 2 || argc > 4)
1465                 return CMD_HELP;
1466
1467         count = strtoull(argv[1], &end, 0);
1468         if (*end) {
1469                 fprintf(stderr, "error: %s: invalid iteration count '%s'\n",
1470                         jt_cmdname(argv[0]), argv[1]);
1471                 return CMD_HELP;
1472         }
1473
1474         if (argc > 2) {
1475                 mode = strtoul(argv[2], &end, 0);
1476                 if (*end) {
1477                         fprintf(stderr, "error: %s: invalid mode '%s'\n",
1478                                 jt_cmdname(argv[0]), argv[2]);
1479                         return CMD_HELP;
1480                 }
1481                 if (!(mode & S_IFMT))
1482                         mode |= S_IFREG;
1483         }
1484
1485         if (argc > 3) {
1486                 verbose = get_verbose(argv[0], argv[3]);
1487                 if (verbose == BAD_VERBOSE)
1488                         return CMD_HELP;
1489         }
1490
1491         printf("%s: %jd objects\n", jt_cmdname(argv[0]), (uintmax_t)count);
1492         gettimeofday(&next_time, NULL);
1493         next_time.tv_sec -= verbose;
1494
1495         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1496         for (i = 1, next_count = verbose; i <= count && shmem_running(); i++) {
1497                 data.ioc_obdo1.o_mode = mode;
1498                 ostid_set_id(&data.ioc_obdo1.o_oi, base_id);
1499                 data.ioc_obdo1.o_uid = 0;
1500                 data.ioc_obdo1.o_gid = 0;
1501                 data.ioc_obdo1.o_projid = 0;
1502                 data.ioc_obdo1.o_valid = OBD_MD_FLTYPE | OBD_MD_FLMODE |
1503                                          OBD_MD_FLID | OBD_MD_FLUID |
1504                                          OBD_MD_FLGID | OBD_MD_FLGROUP |
1505                                          OBD_MD_FLPROJID;
1506
1507                 memset(buf, 0, sizeof(rawbuf));
1508                 rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1509                 if (rc) {
1510                         fprintf(stderr, "error: %s: invalid ioctl\n",
1511                                 jt_cmdname(argv[0]));
1512                         return rc;
1513                 }
1514                 rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_CREATE, buf);
1515                 obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
1516                 shmem_bump(1);
1517                 if (rc < 0) {
1518                         fprintf(stderr, "error: %s: #%d - %s\n",
1519                                 jt_cmdname(argv[0]), i, strerror(rc = errno));
1520                         break;
1521                 }
1522                 if (!(data.ioc_obdo1.o_valid & OBD_MD_FLID)) {
1523                         fprintf(stderr, "error: %s: oid not valid #%d:%#jx\n",
1524                                 jt_cmdname(argv[0]), i,
1525                                 (uintmax_t)data.ioc_obdo1.o_valid);
1526                         rc = EINVAL;
1527                         break;
1528                 }
1529
1530                 if (be_verbose(verbose, &next_time, i, &next_count, count))
1531                         printf("%s: #%d is object id %#jx\n",
1532                                jt_cmdname(argv[0]), i,
1533                                (uintmax_t) ostid_id(&data.ioc_obdo1.o_oi));
1534         }
1535         return rc;
1536 }
1537
1538 int jt_obd_setattr(int argc, char **argv)
1539 {
1540         struct obd_ioctl_data data;
1541         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1542         char *end;
1543         int rc;
1544
1545         memset(&data, 0, sizeof(data));
1546         data.ioc_dev = cur_device;
1547         if (argc != 2)
1548                 return CMD_HELP;
1549
1550         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1551         ostid_set_id(&data.ioc_obdo1.o_oi, strtoull(argv[1], &end, 0));
1552         if (*end) {
1553                 fprintf(stderr, "error: %s: invalid objid '%s'\n",
1554                         jt_cmdname(argv[0]), argv[1]);
1555                 return CMD_HELP;
1556         }
1557         data.ioc_obdo1.o_mode = S_IFREG | strtoul(argv[2], &end, 0);
1558         if (*end) {
1559                 fprintf(stderr, "error: %s: invalid mode '%s'\n",
1560                         jt_cmdname(argv[0]), argv[2]);
1561                 return CMD_HELP;
1562         }
1563         data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLTYPE | OBD_MD_FLMODE;
1564
1565         memset(buf, 0, sizeof(rawbuf));
1566         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1567         if (rc) {
1568                 fprintf(stderr, "error: %s: invalid ioctl\n",
1569                         jt_cmdname(argv[0]));
1570                 return rc;
1571         }
1572         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_SETATTR, buf);
1573         if (rc < 0)
1574                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
1575                         strerror(rc = errno));
1576
1577         return rc;
1578 }
1579
1580 int jt_obd_test_setattr(int argc, char **argv)
1581 {
1582         struct obd_ioctl_data data;
1583         struct timeval start, next_time;
1584         __u64 i, count, next_count;
1585         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1586         int verbose = 1;
1587         __u64 objid = 3;
1588         char *end;
1589         int rc = 0;
1590
1591         if (argc < 2 || argc > 4)
1592                 return CMD_HELP;
1593
1594         memset(&data, 0, sizeof(data));
1595         data.ioc_dev = cur_device;
1596         count = strtoull(argv[1], &end, 0);
1597         if (*end) {
1598                 fprintf(stderr, "error: %s: invalid iteration count '%s'\n",
1599                         jt_cmdname(argv[0]), argv[1]);
1600                 return CMD_HELP;
1601         }
1602
1603         if (argc >= 3) {
1604                 verbose = get_verbose(argv[0], argv[2]);
1605                 if (verbose == BAD_VERBOSE)
1606                         return CMD_HELP;
1607         }
1608
1609         if (argc >= 4) {
1610                 if (argv[3][0] == 't') {
1611                         objid = strtoull(argv[3] + 1, &end, 0);
1612                         if (thread)
1613                                 objid += thread - 1;
1614                 } else
1615                         objid = strtoull(argv[3], &end, 0);
1616                 if (*end) {
1617                         fprintf(stderr, "error: %s: invalid objid '%s'\n",
1618                                 jt_cmdname(argv[0]), argv[3]);
1619                         return CMD_HELP;
1620                 }
1621         }
1622
1623         gettimeofday(&start, NULL);
1624         next_time.tv_sec = start.tv_sec - verbose;
1625         next_time.tv_usec = start.tv_usec;
1626         if (verbose != 0)
1627                 printf("%s: setting %jd attrs (objid %#jx): %s",
1628                        jt_cmdname(argv[0]), (uintmax_t)count,
1629                        (uintmax_t)objid, ctime(&start.tv_sec));
1630
1631         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1632         for (i = 1, next_count = verbose; i <= count && shmem_running(); i++) {
1633                 ostid_set_id(&data.ioc_obdo1.o_oi, objid);
1634                 data.ioc_obdo1.o_mode = S_IFREG;
1635                 data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLTYPE | OBD_MD_FLMODE;
1636                 memset(buf, 0, sizeof(rawbuf));
1637                 rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1638                 if (rc) {
1639                         fprintf(stderr, "error: %s: invalid ioctl\n",
1640                                 jt_cmdname(argv[0]));
1641                         return rc;
1642                 }
1643                 rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_SETATTR, &data);
1644                 shmem_bump(1);
1645                 if (rc < 0) {
1646                         fprintf(stderr, "error: %s: #%jd - %d:%s\n",
1647                                 jt_cmdname(argv[0]), (uintmax_t)i,
1648                                 errno, strerror(rc = errno));
1649                         break;
1650                 } else {
1651                         if (be_verbose
1652                             (verbose, &next_time, i, &next_count, count))
1653                                 printf("%s: set attr #%jd\n",
1654                                        jt_cmdname(argv[0]), (uintmax_t)i);
1655                 }
1656         }
1657
1658         if (!rc) {
1659                 struct timeval end;
1660                 double diff;
1661
1662                 gettimeofday(&end, NULL);
1663
1664                 diff = difftime(&end, &start);
1665
1666                 --i;
1667                 if (verbose != 0)
1668                         printf("%s: %jd attrs in %.3fs (%.3f attr/s): %s",
1669                                jt_cmdname(argv[0]), (uintmax_t)i, diff,
1670                                i / diff, ctime(&end.tv_sec));
1671         }
1672         return rc;
1673 }
1674
1675 int jt_obd_destroy(int argc, char **argv)
1676 {
1677         struct obd_ioctl_data data;
1678         struct timeval next_time;
1679         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1680         __u64 count = 1, next_count;
1681         int verbose = 1;
1682         __u64 id;
1683         char *end;
1684         int rc = 0, i;
1685
1686         memset(&data, 0, sizeof(data));
1687         data.ioc_dev = cur_device;
1688         if (argc < 2 || argc > 4)
1689                 return CMD_HELP;
1690
1691         id = strtoull(argv[1], &end, 0);
1692         if (*end || id == 0 || errno != 0) {
1693                 fprintf(stderr, "error: %s: invalid objid '%s'\n",
1694                         jt_cmdname(argv[0]), argv[1]);
1695                 return CMD_HELP;
1696         }
1697         if (argc > 2) {
1698                 count = strtoull(argv[2], &end, 0);
1699                 if (*end) {
1700                         fprintf(stderr,
1701                                 "error: %s: invalid iteration count '%s'\n",
1702                                 jt_cmdname(argv[0]), argv[2]);
1703                         return CMD_HELP;
1704                 }
1705         }
1706
1707         if (argc > 3) {
1708                 verbose = get_verbose(argv[0], argv[3]);
1709                 if (verbose == BAD_VERBOSE)
1710                         return CMD_HELP;
1711         }
1712
1713         printf("%s: %jd objects\n", jt_cmdname(argv[0]), (uintmax_t)count);
1714         gettimeofday(&next_time, NULL);
1715         next_time.tv_sec -= verbose;
1716
1717         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1718         for (i = 1, next_count = verbose; i <= count && shmem_running(); i++, id++) {
1719                 ostid_set_id(&data.ioc_obdo1.o_oi, id);
1720                 data.ioc_obdo1.o_mode = S_IFREG | 0644;
1721                 data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLMODE;
1722
1723                 memset(buf, 0, sizeof(rawbuf));
1724                 rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1725                 if (rc) {
1726                         fprintf(stderr, "error: %s: invalid ioctl\n",
1727                                 jt_cmdname(argv[0]));
1728                         return rc;
1729                 }
1730                 rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_DESTROY, buf);
1731                 obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
1732                 shmem_bump(1);
1733                 if (rc < 0) {
1734                         fprintf(stderr, "error: %s: objid %#jx: %s\n",
1735                                 jt_cmdname(argv[0]), (uintmax_t)id,
1736                                 strerror(rc = errno));
1737                         break;
1738                 }
1739
1740                 if (be_verbose(verbose, &next_time, i, &next_count, count))
1741                         printf("%s: #%d is object id %#jx\n",
1742                                jt_cmdname(argv[0]), i, (uintmax_t)id);
1743         }
1744
1745         return rc;
1746 }
1747
1748 int jt_obd_getattr(int argc, char **argv)
1749 {
1750         struct obd_ioctl_data data;
1751         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1752         char *end;
1753         int rc;
1754
1755         if (argc != 2)
1756                 return CMD_HELP;
1757
1758         memset(&data, 0, sizeof(data));
1759         data.ioc_dev = cur_device;
1760         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1761         ostid_set_id(&data.ioc_obdo1.o_oi, strtoull(argv[1], &end, 0));
1762         if (*end) {
1763                 fprintf(stderr, "error: %s: invalid objid '%s'\n",
1764                         jt_cmdname(argv[0]), argv[1]);
1765                 return CMD_HELP;
1766         }
1767         /* to help obd filter */
1768         data.ioc_obdo1.o_mode = 0100644;
1769         data.ioc_obdo1.o_valid = 0xffffffff;
1770         printf("%s: object id %#jx\n", jt_cmdname(argv[0]),
1771                (uintmax_t)ostid_id(&data.ioc_obdo1.o_oi));
1772
1773         memset(buf, 0, sizeof(rawbuf));
1774         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1775         if (rc) {
1776                 fprintf(stderr, "error: %s: invalid ioctl\n",
1777                         jt_cmdname(argv[0]));
1778                 return rc;
1779         }
1780         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_GETATTR, buf);
1781         obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
1782         if (rc) {
1783                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
1784                         strerror(rc = errno));
1785         } else {
1786                 printf("%s: object id %ju, mode %o\n", jt_cmdname(argv[0]),
1787                        (uintmax_t)ostid_id(&data.ioc_obdo1.o_oi),
1788                        data.ioc_obdo1.o_mode);
1789         }
1790         return rc;
1791 }
1792
1793 int jt_obd_test_getattr(int argc, char **argv)
1794 {
1795         struct obd_ioctl_data data;
1796         struct timeval start, next_time;
1797         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1798         __u64 i, count, next_count;
1799         int verbose = 1;
1800         __u64 objid = 3;
1801         char *end;
1802         int rc = 0;
1803
1804         if (argc < 2 || argc > 4)
1805                 return CMD_HELP;
1806
1807         memset(&data, 0, sizeof(data));
1808         data.ioc_dev = cur_device;
1809         count = strtoull(argv[1], &end, 0);
1810         if (*end) {
1811                 fprintf(stderr, "error: %s: invalid iteration count '%s'\n",
1812                         jt_cmdname(argv[0]), argv[1]);
1813                 return CMD_HELP;
1814         }
1815
1816         if (argc >= 3) {
1817                 verbose = get_verbose(argv[0], argv[2]);
1818                 if (verbose == BAD_VERBOSE)
1819                         return CMD_HELP;
1820         }
1821
1822         if (argc >= 4) {
1823                 if (argv[3][0] == 't') {
1824                         objid = strtoull(argv[3] + 1, &end, 0);
1825                         if (thread)
1826                                 objid += thread - 1;
1827                 } else
1828                         objid = strtoull(argv[3], &end, 0);
1829                 if (*end) {
1830                         fprintf(stderr, "error: %s: invalid objid '%s'\n",
1831                                 jt_cmdname(argv[0]), argv[3]);
1832                         return CMD_HELP;
1833                 }
1834         }
1835
1836         gettimeofday(&start, NULL);
1837         next_time.tv_sec = start.tv_sec - verbose;
1838         next_time.tv_usec = start.tv_usec;
1839         if (verbose != 0)
1840                 printf("%s: getting %jd attrs (objid %#jx): %s",
1841                        jt_cmdname(argv[0]), (uintmax_t) count,
1842                        (uintmax_t)objid, ctime(&start.tv_sec));
1843
1844         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1845         for (i = 1, next_count = verbose; i <= count && shmem_running(); i++) {
1846                 ostid_set_id(&data.ioc_obdo1.o_oi, objid);
1847                 data.ioc_obdo1.o_mode = S_IFREG;
1848                 data.ioc_obdo1.o_valid = 0xffffffff;
1849                 memset(buf, 0, sizeof(rawbuf));
1850                 rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1851                 if (rc) {
1852                         fprintf(stderr, "error: %s: invalid ioctl\n",
1853                                 jt_cmdname(argv[0]));
1854                         return rc;
1855                 }
1856                 rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_GETATTR, &data);
1857                 shmem_bump(1);
1858                 if (rc < 0) {
1859                         fprintf(stderr, "error: %s: #%jd - %d:%s\n",
1860                                 jt_cmdname(argv[0]), (uintmax_t)i,
1861                                 errno, strerror(rc = errno));
1862                         break;
1863                 } else {
1864                         if (be_verbose
1865                             (verbose, &next_time, i, &next_count, count))
1866                                 printf("%s: got attr #%jd\n",
1867                                        jt_cmdname(argv[0]), (uintmax_t)i);
1868                 }
1869         }
1870
1871         if (!rc) {
1872                 struct timeval end;
1873                 double diff;
1874
1875                 gettimeofday(&end, NULL);
1876
1877                 diff = difftime(&end, &start);
1878
1879                 --i;
1880                 if (verbose != 0)
1881                         printf("%s: %jd attrs in %.3fs (%.3f attr/s): %s",
1882                                jt_cmdname(argv[0]), (uintmax_t) i, diff,
1883                                i / diff, ctime(&end.tv_sec));
1884         }
1885         return rc;
1886 }
1887
1888 /* test_brw <cnt>                                               count
1889         <r|w[r(repeat)x(noverify)]>                             mode
1890         <q|v|#(print interval)>                                 verbosity
1891         <npages[+offset]>                                       blocksize
1892         <[[<interleave_threads>]t(inc obj by thread#)]obj>      object
1893         [p|g<args>]                                             batch */
1894 int jt_obd_test_brw(int argc, char **argv)
1895 {
1896         struct obd_ioctl_data data;
1897         struct timeval start, next_time;
1898         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1899         __u64 count, next_count, len, stride, thr_offset = 0, objid = 3;
1900         int write = 0, verbose = 1, cmd, i, rc = 0, pages = 1;
1901         int offset_pages = 0;
1902         long n;
1903         int repeat_offset = 0;
1904         unsigned long long ull;
1905         int  nthr_per_obj = 0;
1906         int  verify = 1;
1907         int  obj_idx = 0;
1908         char *end;
1909
1910         if (argc < 2 || argc > 7) {
1911                 fprintf(stderr, "error: %s: bad number of arguments: %d\n",
1912                         jt_cmdname(argv[0]), argc);
1913                 return CMD_HELP;
1914         }
1915
1916         count = strtoull(argv[1], &end, 0);
1917         if (*end) {
1918                 fprintf(stderr, "error: %s: bad iteration count '%s'\n",
1919                         jt_cmdname(argv[0]), argv[1]);
1920                 return CMD_HELP;
1921         }
1922
1923         if (argc >= 3) {
1924                 if (argv[2][0] == 'w' || argv[2][0] == '1')
1925                         write = 1;
1926                 /* else it's a read */
1927
1928                 if (argv[2][0] != 0)
1929                         for (i = 1; argv[2][i] != 0; i++)
1930                                 switch (argv[2][i]) {
1931                                 case 'r':
1932                                         repeat_offset = 1;
1933                                         break;
1934
1935                                 case 'x':
1936                                         verify = 0;
1937                                         break;
1938
1939                                 default:
1940                                         fprintf (stderr, "Can't parse cmd '%s'\n",
1941                                                  argv[2]);
1942                                         return CMD_HELP;
1943                                 }
1944         }
1945
1946         if (argc >= 4) {
1947                 verbose = get_verbose(argv[0], argv[3]);
1948                 if (verbose == BAD_VERBOSE)
1949                         return CMD_HELP;
1950         }
1951
1952         if (argc >= 5) {
1953                 pages = strtoul(argv[4], &end, 0);
1954
1955                 if (*end == '+')
1956                         offset_pages = strtoul(end + 1, &end, 0);
1957
1958                 if (*end != 0 ||
1959                     offset_pages < 0 || offset_pages >= pages) {
1960                         fprintf(stderr, "error: %s: bad npages[+offset] parameter '%s'\n",
1961                                 jt_cmdname(argv[0]), argv[4]);
1962                         return CMD_HELP;
1963                 }
1964         }
1965
1966         if (argc >= 6) {
1967                 if (thread &&
1968                     (n = strtol(argv[5], &end, 0)) > 0 &&
1969                     *end == 't' &&
1970                     (ull = strtoull(end + 1, &end, 0)) > 0 &&
1971                     *end == 0) {
1972                         nthr_per_obj = n;
1973                         objid = ull;
1974                 } else if (thread &&
1975                            argv[5][0] == 't') {
1976                         nthr_per_obj = 1;
1977                         objid = strtoull(argv[5] + 1, &end, 0);
1978                 } else {
1979                         nthr_per_obj = 0;
1980                         objid = strtoull(argv[5], &end, 0);
1981                 }
1982                 if (*end) {
1983                         fprintf(stderr, "error: %s: bad objid '%s'\n",
1984                                 jt_cmdname(argv[0]), argv[5]);
1985                         return CMD_HELP;
1986                 }
1987         }
1988
1989         memset(&data, 0, sizeof(data));
1990         data.ioc_dev = cur_device;
1991
1992         /* communicate the 'type' of brw test and batching to echo_client.
1993          * don't start.  we'd love to refactor this lctl->echo_client
1994          * interface */
1995         data.ioc_pbuf1 = (void *)1;
1996         data.ioc_plen1 = 1;
1997
1998         if (argc >= 7) {
1999                 switch(argv[6][0]) {
2000                         case 'g': /* plug and unplug */
2001                                 data.ioc_pbuf1 = (void *)2;
2002                                 data.ioc_plen1 = strtoull(argv[6] + 1, &end,
2003                                                           0);
2004                                 break;
2005                         case 'p': /* prep and commit */
2006                                 data.ioc_pbuf1 = (void *)3;
2007                                 data.ioc_plen1 = strtoull(argv[6] + 1, &end,
2008                                                           0);
2009                                 break;
2010                         default:
2011                                 fprintf(stderr, "error: %s: batching '%s' "
2012                                         "needs to specify 'p' or 'g'\n",
2013                                         jt_cmdname(argv[0]), argv[6]);
2014                                 return CMD_HELP;
2015                 }
2016
2017                 if (*end) {
2018                         fprintf(stderr, "error: %s: bad batching '%s'\n",
2019                                 jt_cmdname(argv[0]), argv[6]);
2020                         return CMD_HELP;
2021                 }
2022                 data.ioc_plen1 *= getpagesize();
2023         }
2024
2025         len = pages * getpagesize();
2026         thr_offset = offset_pages * getpagesize();
2027         stride = len;
2028
2029 #ifdef MAX_THREADS
2030         if (thread) {
2031                 shmem_lock ();
2032                 if (nthr_per_obj != 0) {
2033                         /* threads interleave */
2034                         obj_idx = (thread - 1)/nthr_per_obj;
2035                         objid += obj_idx;
2036                         stride *= nthr_per_obj;
2037                         if ((thread - 1) % nthr_per_obj == 0) {
2038                                 shared_data->body.offsets[obj_idx] =
2039                                         stride + thr_offset;
2040                         }
2041                         thr_offset += ((thread - 1) % nthr_per_obj) * len;
2042                 } else {
2043                         /* threads disjoint */
2044                         thr_offset += (thread - 1) * len;
2045                 }
2046
2047                 shmem_start_time_locked();
2048                 shmem_unlock ();
2049         }
2050 #endif
2051
2052         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
2053         ostid_set_id(&data.ioc_obdo1.o_oi, objid);
2054         data.ioc_obdo1.o_mode = S_IFREG;
2055         data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLTYPE | OBD_MD_FLMODE |
2056                                  OBD_MD_FLFLAGS | OBD_MD_FLGROUP;
2057         data.ioc_obdo1.o_flags = (verify ? OBD_FL_DEBUG_CHECK : 0);
2058         data.ioc_count = len;
2059         data.ioc_offset = (repeat_offset ? 0 : thr_offset);
2060
2061         gettimeofday(&start, NULL);
2062         next_time.tv_sec = start.tv_sec - verbose;
2063         next_time.tv_usec = start.tv_usec;
2064
2065         if (verbose != 0)
2066                 printf("%s: %s %jux%d pages (obj %#jx, off %ju): %s",
2067                        jt_cmdname(argv[0]), write ? "writing" : "reading",
2068                        (uintmax_t)count, pages, (uintmax_t) objid,
2069                        (uintmax_t)data.ioc_offset, ctime(&start.tv_sec));
2070
2071         cmd = write ? OBD_IOC_BRW_WRITE : OBD_IOC_BRW_READ;
2072         for (i = 1, next_count = verbose; i <= count && shmem_running(); i++) {
2073                 data.ioc_obdo1.o_valid &= ~(OBD_MD_FLBLOCKS|OBD_MD_FLGRANT);
2074                 memset(buf, 0, sizeof(rawbuf));
2075                 rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2076                 if (rc) {
2077                         fprintf(stderr, "error: %s: invalid ioctl\n",
2078                                 jt_cmdname(argv[0]));
2079                         return rc;
2080                 }
2081                 rc = l2_ioctl(OBD_DEV_ID, cmd, buf);
2082                 shmem_bump(1);
2083                 if (rc) {
2084                         fprintf(stderr, "error: %s: #%d - %s on %s\n",
2085                                 jt_cmdname(argv[0]), i, strerror(rc = errno),
2086                                 write ? "write" : "read");
2087                         break;
2088                 } else if (be_verbose(verbose, &next_time,i, &next_count,count)) {
2089                         shmem_lock ();
2090                         printf("%s: %s number %d @ %jd:%ju for %d\n",
2091                                jt_cmdname(argv[0]), write ? "write" : "read", i,
2092                                (uintmax_t)ostid_id(&data.ioc_obdo1.o_oi),
2093                                (uintmax_t)data.ioc_offset,
2094                                (int)(pages * getpagesize()));
2095                         shmem_unlock ();
2096                 }
2097
2098                 if (!repeat_offset) {
2099 #ifdef MAX_THREADS
2100                         if (stride == len) {
2101                                 data.ioc_offset += stride;
2102                         } else if (i < count) {
2103                                 shmem_lock ();
2104                                 data.ioc_offset =
2105                                         shared_data->body.offsets[obj_idx];
2106                                 shared_data->body.offsets[obj_idx] += len;
2107                                 shmem_unlock ();
2108                         }
2109 #else
2110                         data.ioc_offset += len;
2111                         obj_idx = 0; /* avoids an unused var warning */
2112 #endif
2113                 }
2114         }
2115
2116         if (!rc) {
2117                 struct timeval end;
2118                 double diff;
2119
2120                 gettimeofday(&end, NULL);
2121
2122                 diff = difftime(&end, &start);
2123
2124                 --i;
2125                 if (verbose != 0)
2126                         printf("%s: %s %dx%d pages in %.3fs (%.3f MB/s): %s",
2127                                jt_cmdname(argv[0]), write ? "wrote" : "read",
2128                                i, pages, diff,
2129                                ((double)i * pages * getpagesize()) /
2130                                (diff * 1048576.0),
2131                                ctime(&end.tv_sec));
2132         }
2133
2134 #ifdef MAX_THREADS
2135         if (thread) {
2136                 shmem_lock();
2137                 shmem_end_time_locked();
2138                 shmem_unlock();
2139         }
2140 #endif
2141         return rc;
2142 }
2143
2144 int jt_obd_lov_getconfig(int argc, char **argv)
2145 {
2146         struct obd_ioctl_data data;
2147         struct lov_desc desc;
2148         struct obd_uuid *uuidarray;
2149         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2150         __u32 *obdgens;
2151         char *path;
2152         int rc, fd;
2153
2154         memset(&data, 0, sizeof(data));
2155         data.ioc_dev = cur_device;
2156
2157         if (argc != 2)
2158                 return CMD_HELP;
2159
2160         path = argv[1];
2161         fd = open(path, O_RDONLY);
2162         if (fd < 0) {
2163                 fprintf(stderr, "open \"%s\" failed: %s\n", path,
2164                         strerror(errno));
2165                 return -errno;
2166         }
2167
2168         memset(&desc, 0, sizeof(desc));
2169         obd_str2uuid(&desc.ld_uuid, argv[1]);
2170         desc.ld_tgt_count = ((OBD_MAX_IOCTL_BUFFER-sizeof(data)-sizeof(desc)) /
2171                              (sizeof(*uuidarray) + sizeof(*obdgens)));
2172
2173 repeat:
2174         uuidarray = calloc(desc.ld_tgt_count, sizeof(*uuidarray));
2175         if (!uuidarray) {
2176                 fprintf(stderr, "error: %s: no memory for %d uuid's\n",
2177                         jt_cmdname(argv[0]), desc.ld_tgt_count);
2178                 rc = -ENOMEM;
2179                 goto out;
2180         }
2181         obdgens = calloc(desc.ld_tgt_count, sizeof(*obdgens));
2182         if (!obdgens) {
2183                 fprintf(stderr, "error: %s: no memory for %d generation #'s\n",
2184                         jt_cmdname(argv[0]), desc.ld_tgt_count);
2185                 rc = -ENOMEM;
2186                 goto out_uuidarray;
2187         }
2188
2189         memset(buf, 0, sizeof(rawbuf));
2190         data.ioc_inllen1 = sizeof(desc);
2191         data.ioc_inlbuf1 = (char *)&desc;
2192         data.ioc_inllen2 = desc.ld_tgt_count * sizeof(*uuidarray);
2193         data.ioc_inlbuf2 = (char *)uuidarray;
2194         data.ioc_inllen3 = desc.ld_tgt_count * sizeof(*obdgens);
2195         data.ioc_inlbuf3 = (char *)obdgens;
2196
2197         if (obd_ioctl_pack(&data, &buf, sizeof(rawbuf))) {
2198                 fprintf(stderr, "error: %s: invalid ioctl\n",
2199                         jt_cmdname(argv[0]));
2200                 rc = -EINVAL;
2201                 goto out_obdgens;
2202         }
2203         rc = ioctl(fd, OBD_IOC_LOV_GET_CONFIG, buf);
2204         if (rc == -ENOSPC) {
2205                 free(uuidarray);
2206                 free(obdgens);
2207                 goto repeat;
2208         } else if (rc) {
2209                 fprintf(stderr, "error: %s: ioctl error: %s\n",
2210                         jt_cmdname(argv[0]), strerror(rc = errno));
2211         } else {
2212                 struct obd_uuid *uuidp;
2213                 __u32 *genp;
2214                 int i;
2215
2216                 if (obd_ioctl_unpack(&data, buf, sizeof(rawbuf))) {
2217                         fprintf(stderr, "error: %s: invalid reply\n",
2218                                 jt_cmdname(argv[0]));
2219                         rc = -EINVAL;
2220                         goto out;
2221                 }
2222                 if (desc.ld_default_stripe_count == (__u32)-1)
2223                         printf("default_stripe_count: %d\n", -1);
2224                 else
2225                         printf("default_stripe_count: %u\n",
2226                                desc.ld_default_stripe_count);
2227                 printf("default_stripe_size: %ju\n",
2228                        (uintmax_t)desc.ld_default_stripe_size);
2229                 printf("default_stripe_offset: %ju\n",
2230                        (uintmax_t)desc.ld_default_stripe_offset);
2231                 printf("default_stripe_pattern: %u\n", desc.ld_pattern);
2232                 printf("obd_count: %u\n", desc.ld_tgt_count);
2233                 printf("OBDS:\tobdidx\t\tobdgen\t\t obduuid\n");
2234                 uuidp = uuidarray;
2235                 genp = obdgens;
2236                 for (i = 0; i < desc.ld_tgt_count; i++, uuidp++, genp++)
2237                         printf("\t%6u\t%14u\t\t %s\n", i, *genp, (char *)uuidp);
2238         }
2239 out_obdgens:
2240         free(obdgens);
2241 out_uuidarray:
2242         free(uuidarray);
2243 out:
2244         close(fd);
2245         return rc;
2246 }
2247
2248 static int do_activate(int argc, char **argv, int flag)
2249 {
2250         struct obd_ioctl_data data;
2251         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2252         int rc;
2253
2254         memset(&data, 0, sizeof(data));
2255         data.ioc_dev = cur_device;
2256         if (argc != 1)
2257                 return CMD_HELP;
2258
2259         /* reuse offset for 'active' */
2260         data.ioc_offset = flag;
2261
2262         memset(buf, 0, sizeof(rawbuf));
2263         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2264         if (rc) {
2265                 fprintf(stderr, "error: %s: invalid ioctl\n",
2266                         jt_cmdname(argv[0]));
2267                 return rc;
2268         }
2269         rc = l2_ioctl(OBD_DEV_ID, IOC_OSC_SET_ACTIVE, buf);
2270         if (rc)
2271                 fprintf(stderr, "error: %s: failed: %s\n",
2272                         jt_cmdname(argv[0]), strerror(rc = errno));
2273
2274         return rc;
2275 }
2276
2277 /**
2278  * Replace nids for given device.
2279  * lctl replace_nids <devicename> <nid1>[,nid2,nid3]
2280  * Command should be started on MGS server.
2281  * Only MGS server should be started (command execution
2282  * returns error in another cases). Command mount
2283  * -t lustre <MDT partition> -o nosvc <mount point>
2284  * can be used for that.
2285  *
2286  * llogs for MDTs and clients are processed. All
2287  * records copied as is except add_uuid and setup. This records
2288  * are skipped and recorded with new nids and uuid.
2289  *
2290  * \see mgs_replace_nids
2291  * \see mgs_replace_nids_log
2292  * \see mgs_replace_handler
2293  */
2294 int jt_replace_nids(int argc, char **argv)
2295 {
2296         int rc;
2297         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2298         struct obd_ioctl_data data;
2299
2300         memset(&data, 0, sizeof(data));
2301         data.ioc_dev = get_mgs_device();
2302         if (argc != 3)
2303                 return CMD_HELP;
2304
2305         data.ioc_inllen1 = strlen(argv[1]) + 1;
2306         data.ioc_inlbuf1 = argv[1];
2307
2308         data.ioc_inllen2 = strlen(argv[2]) + 1;
2309         data.ioc_inlbuf2 = argv[2];
2310         memset(buf, 0, sizeof(rawbuf));
2311         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2312         if (rc) {
2313                 fprintf(stderr, "error: %s: invalid ioctl\n",
2314                         jt_cmdname(argv[0]));
2315                 return rc;
2316         }
2317
2318         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_REPLACE_NIDS, buf);
2319         if (rc < 0) {
2320                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
2321                         strerror(rc = errno));
2322         }
2323
2324         return rc;
2325 }
2326
2327 int jt_obd_deactivate(int argc, char **argv)
2328 {
2329         return do_activate(argc, argv, 0);
2330 }
2331
2332 int jt_obd_activate(int argc, char **argv)
2333 {
2334         return do_activate(argc, argv, 1);
2335 }
2336
2337 int jt_obd_recover(int argc, char **argv)
2338 {
2339         int rc;
2340         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2341         struct obd_ioctl_data data;
2342
2343         memset(&data, 0, sizeof(data));
2344         data.ioc_dev = cur_device;
2345         if (argc > 2)
2346                 return CMD_HELP;
2347
2348         if (argc == 2) {
2349                 data.ioc_inllen1 = strlen(argv[1]) + 1;
2350                 data.ioc_inlbuf1 = argv[1];
2351         }
2352
2353         memset(buf, 0, sizeof(rawbuf));
2354         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2355         if (rc) {
2356                 fprintf(stderr, "error: %s: invalid ioctl\n",
2357                         jt_cmdname(argv[0]));
2358                 return rc;
2359         }
2360         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_CLIENT_RECOVER, buf);
2361         if (rc < 0) {
2362                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
2363                         strerror(rc = errno));
2364         }
2365
2366         return rc;
2367 }
2368
2369 int jt_obd_mdc_lookup(int argc, char **argv)
2370 {
2371         struct obd_ioctl_data data;
2372         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2373         char *parent, *child;
2374         int rc, fd, verbose = 1;
2375
2376         if (argc < 3 || argc > 4)
2377                 return CMD_HELP;
2378
2379         parent = argv[1];
2380         child = argv[2];
2381         if (argc == 4)
2382                 verbose = get_verbose(argv[0], argv[3]);
2383
2384         memset(&data, 0, sizeof(data));
2385         data.ioc_dev = cur_device;
2386
2387         data.ioc_inllen1 = strlen(child) + 1;
2388         data.ioc_inlbuf1 = child;
2389
2390         memset(buf, 0, sizeof(rawbuf));
2391         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2392         if (rc) {
2393                 fprintf(stderr, "error: %s: invalid ioctl\n",
2394                         jt_cmdname(argv[0]));
2395                 return rc;
2396         }
2397
2398         fd = open(parent, O_RDONLY);
2399         if (fd < 0) {
2400                 fprintf(stderr, "open \"%s\" failed: %s\n", parent,
2401                         strerror(errno));
2402                 return -1;
2403         }
2404
2405         rc = ioctl(fd, IOC_MDC_LOOKUP, buf);
2406         if (rc < 0) {
2407                 fprintf(stderr, "error: %s: ioctl error: %s\n",
2408                         jt_cmdname(argv[0]), strerror(rc = errno));
2409         }
2410         close(fd);
2411
2412         if (verbose) {
2413                 rc = obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
2414                 if (rc) {
2415                         fprintf(stderr, "error: %s: invalid reply\n",
2416                                 jt_cmdname(argv[0]));
2417                         return rc;
2418                 }
2419                 printf("%s: mode %o uid %d gid %d\n", child,
2420                        data.ioc_obdo1.o_mode, data.ioc_obdo1.o_uid,
2421                        data.ioc_obdo1.o_gid);
2422         }
2423
2424         return rc;
2425 }
2426
2427 int jt_lcfg_fork(int argc, char **argv)
2428 {
2429         struct obd_ioctl_data data;
2430         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2431         int rc;
2432
2433         if (argc != 3)
2434                 return CMD_HELP;
2435
2436         memset(&data, 0, sizeof(data));
2437         data.ioc_dev = get_mgs_device();
2438         data.ioc_inllen1 = strlen(argv[1]) + 1;
2439         data.ioc_inlbuf1 = argv[1];
2440         data.ioc_inllen2 = strlen(argv[2]) + 1;
2441         data.ioc_inlbuf2 = argv[2];
2442
2443         memset(buf, 0, sizeof(rawbuf));
2444         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2445         if (rc) {
2446                 fprintf(stderr, "error: %s: invalid ioctl\n",
2447                         jt_cmdname(argv[0]));
2448                 return rc;
2449         }
2450
2451         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LCFG_FORK, buf);
2452         if (rc < 0)
2453                 fprintf(stderr, "error: %s: OBD_IOC_LCFG_FORK failed: %s\n",
2454                         jt_cmdname(argv[0]), strerror(errno));
2455
2456         return rc;
2457 }
2458
2459 int jt_lcfg_erase(int argc, char **argv)
2460 {
2461         struct obd_ioctl_data data;
2462         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2463         int rc;
2464
2465         if (argc == 3) {
2466                 if (strncmp(argv[2], "-q", strlen("-q")) != 0 &&
2467                     strncmp(argv[2], "--quiet", strlen("--quiet")) != 0)
2468                         return CMD_HELP;
2469         } else if (argc != 2) {
2470                 return CMD_HELP;
2471         }
2472
2473         memset(&data, 0, sizeof(data));
2474         data.ioc_dev = get_mgs_device();
2475         data.ioc_inllen1 = strlen(argv[1]) + 1;
2476         data.ioc_inlbuf1 = argv[1];
2477
2478         memset(buf, 0, sizeof(rawbuf));
2479         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2480         if (rc) {
2481                 fprintf(stderr, "error: %s: invalid ioctl\n",
2482                         jt_cmdname(argv[0]));
2483                 return rc;
2484         }
2485
2486         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LCFG_ERASE, buf);
2487         if (rc < 0)
2488                 fprintf(stderr, "error: %s: OBD_IOC_LCFG_ERASE failed: %s\n",
2489                         jt_cmdname(argv[0]), strerror(errno));
2490
2491         return rc;
2492 }
2493
2494 int jt_llog_catlist(int argc, char **argv)
2495 {
2496         struct obd_ioctl_data data;
2497         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2498         int rc;
2499
2500         if (argc != 1)
2501                 return CMD_HELP;
2502
2503         memset(&data, 0, sizeof(data));
2504         data.ioc_dev = cur_device;
2505         data.ioc_inllen1 = sizeof(rawbuf) - cfs_size_round(sizeof(data));
2506         memset(buf, 0, sizeof(rawbuf));
2507         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2508         if (rc) {
2509                 fprintf(stderr, "error: %s: invalid ioctl\n",
2510                         jt_cmdname(argv[0]));
2511                 return rc;
2512         }
2513         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_CATLOGLIST, buf);
2514         if (rc == 0)
2515                 fprintf(stdout, "%s", ((struct obd_ioctl_data*)buf)->ioc_bulk);
2516         else
2517                 fprintf(stderr, "OBD_IOC_CATLOGLIST failed: %s\n",
2518                         strerror(errno));
2519
2520         return rc;
2521 }
2522
2523 int jt_llog_info(int argc, char **argv)
2524 {
2525         struct obd_ioctl_data data;
2526         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2527         int rc;
2528
2529         if (argc != 2)
2530                 return CMD_HELP;
2531
2532         memset(&data, 0, sizeof(data));
2533         data.ioc_dev = cur_device;
2534         data.ioc_inllen1 = strlen(argv[1]) + 1;
2535         data.ioc_inlbuf1 = argv[1];
2536         data.ioc_inllen2 = sizeof(rawbuf) - cfs_size_round(sizeof(data)) -
2537                 cfs_size_round(data.ioc_inllen1);
2538         memset(buf, 0, sizeof(rawbuf));
2539         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2540         if (rc) {
2541                 fprintf(stderr, "error: %s: invalid ioctl\n",
2542                         jt_cmdname(argv[0]));
2543                 return rc;
2544         }
2545
2546         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_INFO, buf);
2547         if (rc == 0)
2548                 fprintf(stdout, "%s", ((struct obd_ioctl_data*)buf)->ioc_bulk);
2549         else
2550                 fprintf(stderr, "OBD_IOC_LLOG_INFO failed: %s\n",
2551                         strerror(errno));
2552
2553         return rc;
2554 }
2555
2556 int jt_llog_print(int argc, char **argv)
2557 {
2558         struct obd_ioctl_data data;
2559         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2560         int rc;
2561
2562         if (argc != 2 && argc != 4)
2563                 return CMD_HELP;
2564
2565         memset(&data, 0, sizeof(data));
2566         data.ioc_dev = cur_device;
2567         data.ioc_inllen1 = strlen(argv[1]) + 1;
2568         data.ioc_inlbuf1 = argv[1];
2569         if (argc == 4) {
2570                 data.ioc_inllen2 = strlen(argv[2]) + 1;
2571                 data.ioc_inlbuf2 = argv[2];
2572                 data.ioc_inllen3 = strlen(argv[3]) + 1;
2573                 data.ioc_inlbuf3 = argv[3];
2574         } else {
2575                 char from[2] = "1", to[3] = "-1";
2576                 data.ioc_inllen2 = strlen(from) + 1;
2577                 data.ioc_inlbuf2 = from;
2578                 data.ioc_inllen3 = strlen(to) + 1;
2579                 data.ioc_inlbuf3 = to;
2580         }
2581         data.ioc_inllen4 = sizeof(rawbuf) - cfs_size_round(sizeof(data)) -
2582                 cfs_size_round(data.ioc_inllen1) -
2583                 cfs_size_round(data.ioc_inllen2) -
2584                 cfs_size_round(data.ioc_inllen3);
2585         memset(buf, 0, sizeof(rawbuf));
2586         rc = obd_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_LLOG_PRINT, buf);
2594         if (rc == 0)
2595                 fprintf(stdout, "%s", ((struct obd_ioctl_data*)buf)->ioc_bulk);
2596         else
2597                 fprintf(stderr, "OBD_IOC_LLOG_PRINT failed: %s\n",
2598                         strerror(errno));
2599
2600         return rc;
2601 }
2602
2603 static int llog_cancel_parse_optional(int argc, char **argv,
2604                                       struct obd_ioctl_data *data)
2605 {
2606         int cOpt;
2607         const char *const short_options = "c:l:i:h";
2608         const struct option long_options[] = {
2609                 {"catalog", required_argument, NULL, 'c'},
2610                 {"log_id", required_argument, NULL, 'l'},
2611                 {"log_idx", required_argument, NULL, 'i'},
2612                 {"help", no_argument, NULL, 'h'},
2613                 {NULL, 0, NULL, 0}
2614         };
2615
2616         /* sanity check */
2617         if (!data || argc <= 1) {
2618                 return -1;
2619         }
2620
2621         /*now process command line arguments*/
2622         while ((cOpt = getopt_long(argc, argv, short_options,
2623                                         long_options, NULL)) != -1) {
2624                 switch (cOpt) {
2625                 case 'c':
2626                         data->ioc_inllen1 = strlen(optarg) + 1;
2627                         data->ioc_inlbuf1 = optarg;
2628                         break;
2629
2630                 case 'l':
2631                         data->ioc_inllen2 = strlen(optarg) + 1;
2632                         data->ioc_inlbuf2 = optarg;
2633                         break;
2634
2635                 case 'i':
2636                         data->ioc_inllen3 = strlen(optarg) + 1;
2637                         data->ioc_inlbuf3 = optarg;
2638                         break;
2639
2640                 case 'h':
2641                 default:
2642                         return -1;
2643                 }
2644         }
2645
2646         if ((data->ioc_inlbuf1 == NULL) || (data->ioc_inlbuf3 == NULL)) {
2647                 /* missing mandatory parameters */
2648                 return -1;
2649         }
2650
2651         return 0;
2652 }
2653
2654 int jt_llog_cancel(int argc, char **argv)
2655 {
2656         struct obd_ioctl_data data;
2657         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2658         int rc, i;
2659
2660         /* check that the arguments provided are either all
2661          * optional or all positional.  No mixing allowed
2662          *
2663          * if argc is 4 or 3 then check all arguments to ensure that none
2664          * of them start with a '-'.  If so then this is invalid.
2665          * Otherwise if arg is > 4 then assume that this is optional
2666          * arguments, and parse as such ignoring any thing that's not
2667          * optional.  The result is that user must use optional arguments
2668          * for all mandatory parameters.  Code will ignore all other args
2669          *
2670          * The positional arguments option should eventually be phased out.
2671          */
2672         memset(&data, 0, sizeof(data));
2673         data.ioc_dev = cur_device;
2674
2675         if (argc == 3 || argc == 4) {
2676                 for (i = 1; i < argc; i++) {
2677                         if (argv[i][0] == '-')
2678                                 return CMD_HELP;
2679                 }
2680                 data.ioc_inllen1 = strlen(argv[1]) + 1;
2681                 data.ioc_inlbuf1 = argv[1];
2682                 if (argc == 4) {
2683                         data.ioc_inllen2 = strlen(argv[2]) + 1;
2684                         data.ioc_inlbuf2 = argv[2];
2685                         data.ioc_inllen3 = strlen(argv[3]) + 1;
2686                         data.ioc_inlbuf3 = argv[3];
2687                 } else {
2688                         data.ioc_inllen3 = strlen(argv[2]) + 1;
2689                         data.ioc_inlbuf3 = argv[2];
2690                 }
2691         } else {
2692                 if (llog_cancel_parse_optional(argc, argv, &data) != 0)
2693                         return CMD_HELP;
2694         }
2695
2696         memset(buf, 0, sizeof(rawbuf));
2697         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2698         if (rc) {
2699                 fprintf(stderr, "error: %s: invalid ioctl\n",
2700                         jt_cmdname(argv[0]));
2701                 return rc;
2702         }
2703
2704         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_CANCEL, buf);
2705         if (rc == 0)
2706                 fprintf(stdout, "index %s was canceled.\n",
2707                         argc == 4 ? argv[3] : argv[2]);
2708         else
2709                 fprintf(stderr, "OBD_IOC_LLOG_CANCEL failed: %s\n",
2710                         strerror(errno));
2711
2712         return rc;
2713 }
2714
2715 int jt_llog_check(int argc, char **argv)
2716 {
2717         struct obd_ioctl_data data;
2718         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2719         int rc;
2720
2721         if (argc != 2 && argc != 4)
2722                 return CMD_HELP;
2723
2724         memset(&data, 0, sizeof(data));
2725         data.ioc_dev = cur_device;
2726         data.ioc_inllen1 = strlen(argv[1]) + 1;
2727         data.ioc_inlbuf1 = argv[1];
2728         if (argc == 4) {
2729                 data.ioc_inllen2 = strlen(argv[2]) + 1;
2730                 data.ioc_inlbuf2 = argv[2];
2731                 data.ioc_inllen3 = strlen(argv[3]) + 1;
2732                 data.ioc_inlbuf3 = argv[3];
2733         } else {
2734                 char from[2] = "1", to[3] = "-1";
2735                 data.ioc_inllen2 = strlen(from) + 1;
2736                 data.ioc_inlbuf2 = from;
2737                 data.ioc_inllen3 = strlen(to) + 1;
2738                 data.ioc_inlbuf3 = to;
2739         }
2740         data.ioc_inllen4 = sizeof(rawbuf) - cfs_size_round(sizeof(data)) -
2741                 cfs_size_round(data.ioc_inllen1) -
2742                 cfs_size_round(data.ioc_inllen2) -
2743                 cfs_size_round(data.ioc_inllen3);
2744         memset(buf, 0, sizeof(rawbuf));
2745         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2746         if (rc) {
2747                 fprintf(stderr, "error: %s: invalid ioctl\n",
2748                         jt_cmdname(argv[0]));
2749                 return rc;
2750         }
2751
2752         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_CHECK, buf);
2753         if (rc == 0)
2754                 fprintf(stdout, "%s", ((struct obd_ioctl_data*)buf)->ioc_bulk);
2755         else
2756                 fprintf(stderr, "OBD_IOC_LLOG_CHECK failed: %s\n",
2757                         strerror(errno));
2758         return rc;
2759 }
2760
2761 int jt_llog_remove(int argc, char **argv)
2762 {
2763         struct obd_ioctl_data data;
2764         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2765         int rc;
2766
2767         if (argc != 3 && argc != 2)
2768                 return CMD_HELP;
2769
2770         memset(&data, 0, sizeof(data));
2771         data.ioc_dev = cur_device;
2772         data.ioc_inllen1 = strlen(argv[1]) + 1;
2773         data.ioc_inlbuf1 = argv[1];
2774         if (argc == 3){
2775                 data.ioc_inllen2 = strlen(argv[2]) + 1;
2776                 data.ioc_inlbuf2 = argv[2];
2777         }
2778         memset(buf, 0, sizeof(rawbuf));
2779         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2780         if (rc) {
2781                 fprintf(stderr, "error: %s: invalid ioctl\n",
2782                         jt_cmdname(argv[0]));
2783                 return rc;
2784         }
2785
2786         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_REMOVE, buf);
2787         if (rc == 0) {
2788                 if (argc == 2)
2789                         fprintf(stdout, "log %s is removed.\n", argv[1]);
2790                 else
2791                         fprintf(stdout, "the log in catalog %s is removed. \n",
2792                                 argv[1]);
2793         } else
2794                 fprintf(stderr, "OBD_IOC_LLOG_REMOVE failed: %s\n",
2795                         strerror(errno));
2796
2797         return rc;
2798 }
2799
2800 static void signal_server(int sig)
2801 {
2802         if (sig == SIGINT) {
2803                 do_disconnect("sigint", 1);
2804                 exit(1);
2805         } else
2806                 fprintf(stderr, "%s: got signal %d\n", jt_cmdname("sigint"), sig);
2807 }
2808
2809 int obd_initialize(int argc, char **argv)
2810 {
2811         if (shmem_setup() != 0)
2812                 return -1;
2813
2814         register_ioc_dev(OBD_DEV_ID, OBD_DEV_PATH,
2815                          OBD_DEV_MAJOR, OBD_DEV_MINOR);
2816
2817         return 0;
2818 }
2819
2820 void obd_finalize(int argc, char **argv)
2821 {
2822         struct sigaction sigact;
2823
2824         /* sigact initialization */
2825         sigact.sa_handler = signal_server;
2826         sigfillset(&sigact.sa_mask);
2827         sigact.sa_flags = SA_RESTART;
2828         /* coverity[uninit_use_in_call] */
2829         sigaction(SIGINT, &sigact, NULL);
2830
2831         shmem_cleanup();
2832         do_disconnect(argv[0], 1);
2833 }
2834
2835 static int check_pool_cmd(enum lcfg_command_type cmd,
2836                           char *fsname, char *poolname,
2837                           char *ostname)
2838 {
2839         int rc;
2840
2841         rc = llapi_search_ost(fsname, poolname, ostname);
2842         if (rc < 0 && (cmd != LCFG_POOL_NEW)) {
2843                 fprintf(stderr, "Pool %s.%s not found\n",
2844                         fsname, poolname);
2845                 return rc;
2846         }
2847
2848         switch (cmd) {
2849         case LCFG_POOL_NEW: {
2850                 if (ostname != NULL)
2851                         return -EINVAL;
2852
2853                 if (rc >= 0) {
2854                         fprintf(stderr, "Pool %s.%s already exists\n",
2855                                 fsname, poolname);
2856                         return -EEXIST;
2857                 }
2858                 return 0;
2859         }
2860         case LCFG_POOL_DEL: {
2861                 if (ostname != NULL)
2862                         return -EINVAL;
2863
2864                 if (rc == 1) {
2865                         fprintf(stderr, "Pool %s.%s not empty, "
2866                                 "please remove all members\n",
2867                                 fsname, poolname);
2868                         return -ENOTEMPTY;
2869                 }
2870                 return 0;
2871         }
2872         case LCFG_POOL_ADD: {
2873                 if (rc == 1) {
2874                         fprintf(stderr, "OST %s is already in pool %s.%s\n",
2875                                 ostname, fsname, poolname);
2876                         return -EEXIST;
2877                 }
2878                 rc = llapi_search_ost(fsname, NULL, ostname);
2879                 if (rc == 0) {
2880                         fprintf(stderr, "OST %s is not part of the '%s' fs.\n",
2881                                 ostname, fsname);
2882                         return -ENOENT;
2883                 }
2884                 return 0;
2885         }
2886         case LCFG_POOL_REM: {
2887                 if (rc == 0) {
2888                         fprintf(stderr, "OST %s not found in pool %s.%s\n",
2889                                 ostname, fsname, poolname);
2890                         return -ENOENT;
2891                 }
2892                 return 0;
2893         }
2894         default:
2895                 break;
2896         } /* switch */
2897         return -EINVAL;
2898 }
2899
2900 /* This check only verifies that the changes have been "pushed out" to
2901    the client successfully.  This involves waiting for a config update,
2902    and so may fail because of problems in that code or post-command
2903    network loss. So reporting a warning is appropriate, but not a failure.
2904 */
2905 static int check_pool_cmd_result(enum lcfg_command_type cmd,
2906                                  char *fsname, char *poolname,
2907                                  char *ostname)
2908 {
2909         int cpt = 10;
2910         int rc = 0;
2911
2912         switch (cmd) {
2913         case LCFG_POOL_NEW: {
2914                 do {
2915                         rc = llapi_search_ost(fsname, poolname, NULL);
2916                         if (rc == -ENODEV)
2917                                 return rc;
2918                         if (rc < 0)
2919                                 sleep(2);
2920                         cpt--;
2921                 } while ((rc < 0) && (cpt > 0));
2922                 if (rc >= 0) {
2923                         fprintf(stderr, "Pool %s.%s created\n",
2924                                 fsname, poolname);
2925                         return 0;
2926                 } else {
2927                         fprintf(stderr, "Warning, pool %s.%s not found\n",
2928                                 fsname, poolname);
2929                         return -ENOENT;
2930                 }
2931         }
2932         case LCFG_POOL_DEL: {
2933                 do {
2934                         rc = llapi_search_ost(fsname, poolname, NULL);
2935                         if (rc == -ENODEV)
2936                                 return rc;
2937                         if (rc >= 0)
2938                                 sleep(2);
2939                         cpt--;
2940                 } while ((rc >= 0) && (cpt > 0));
2941                 if (rc < 0) {
2942                         fprintf(stderr, "Pool %s.%s destroyed\n",
2943                                 fsname, poolname);
2944                         return 0;
2945                 } else {
2946                         fprintf(stderr, "Warning, pool %s.%s still found\n",
2947                                 fsname, poolname);
2948                         return -EEXIST;
2949                 }
2950         }
2951         case LCFG_POOL_ADD: {
2952                 do {
2953                         rc = llapi_search_ost(fsname, poolname, ostname);
2954                         if (rc == -ENODEV)
2955                                 return rc;
2956                         if (rc != 1)
2957                                 sleep(2);
2958                         cpt--;
2959                 } while ((rc != 1) && (cpt > 0));
2960                 if (rc == 1) {
2961                         fprintf(stderr, "OST %s added to pool %s.%s\n",
2962                                 ostname, fsname, poolname);
2963                         return 0;
2964                 } else {
2965                         fprintf(stderr, "Warning, OST %s not found in pool %s.%s\n",
2966                                 ostname, fsname, poolname);
2967                         return -ENOENT;
2968                 }
2969         }
2970         case LCFG_POOL_REM: {
2971                 do {
2972                         rc = llapi_search_ost(fsname, poolname, ostname);
2973                         if (rc == -ENODEV)
2974                                 return rc;
2975                         if (rc == 1)
2976                                 sleep(2);
2977                         cpt--;
2978                 } while ((rc == 1) && (cpt > 0));
2979                 if (rc != 1) {
2980                         fprintf(stderr, "OST %s removed from pool %s.%s\n",
2981                                 ostname, fsname, poolname);
2982                         return 0;
2983                 } else {
2984                         fprintf(stderr, "Warning, OST %s still found in pool %s.%s\n",
2985                                 ostname, fsname, poolname);
2986                         return -EEXIST;
2987                 }
2988         }
2989         default:
2990                 break;
2991         }
2992         return -EINVAL;
2993 }
2994
2995 static int check_and_complete_ostname(char *fsname, char *ostname)
2996 {
2997         char *ptr;
2998         char real_ostname[MAX_OBD_NAME + 1];
2999         char i;
3000
3001         /* if OST name does not start with fsname, we add it */
3002         /* if not check if the fsname is the right one */
3003         ptr = strchr(ostname, '-');
3004         if (ptr == NULL) {
3005                 sprintf(real_ostname, "%s-%s", fsname, ostname);
3006         } else if (strncmp(ostname, fsname, strlen(fsname)) != 0) {
3007                 fprintf(stderr, "%s does not start with fsname %s\n",
3008                         ostname, fsname);
3009                 return -EINVAL;
3010         } else {
3011                 if (strlen(ostname) > sizeof(real_ostname)-1)
3012                         return -E2BIG;
3013                 strncpy(real_ostname, ostname, sizeof(real_ostname));
3014         }
3015         /* real_ostname is fsname-????? */
3016         ptr = real_ostname + strlen(fsname) + 1;
3017         if (strncmp(ptr, "OST", 3) != 0) {
3018                 fprintf(stderr, "%s does not start by %s-OST nor OST\n",
3019                         ostname, fsname);
3020                 return -EINVAL;
3021         }
3022         /* real_ostname is fsname-OST????? */
3023         ptr += 3;
3024         for (i = 0; i < 4; i++) {
3025                 if (!isxdigit(*ptr)) {
3026                         fprintf(stderr,
3027                                 "ost's index in %s is not an hexa number\n",
3028                                 ostname);
3029                         return -EINVAL;
3030                 }
3031                 ptr++;
3032         }
3033         /* real_ostname is fsname-OSTXXXX????? */
3034         /* if OST name does not end with _UUID, we add it */
3035         if (*ptr == '\0') {
3036                 strcat(real_ostname, "_UUID");
3037         } else if (strcmp(ptr, "_UUID") != 0) {
3038                 fprintf(stderr,
3039                         "ostname %s does not end with _UUID\n", ostname);
3040                 return -EINVAL;
3041         }
3042         /* real_ostname is fsname-OSTXXXX_UUID */
3043         strcpy(ostname, real_ostname);
3044         return 0;
3045 }
3046
3047 /* returns 0 or -errno */
3048 static int pool_cmd(enum lcfg_command_type cmd,
3049                     char *cmdname, char *fullpoolname,
3050                     char *fsname, char *poolname, char *ostname)
3051 {
3052         int rc = 0;
3053         struct obd_ioctl_data data;
3054         struct lustre_cfg_bufs bufs;
3055         struct lustre_cfg *lcfg;
3056         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
3057
3058         rc = check_pool_cmd(cmd, fsname, poolname, ostname);
3059         if (rc == -ENODEV)
3060                 fprintf(stderr, "Can't verify pool command since there "
3061                         "is no local MDT or client, proceeding anyhow...\n");
3062         else if (rc)
3063                 return rc;
3064
3065         lustre_cfg_bufs_reset(&bufs, NULL);
3066         lustre_cfg_bufs_set_string(&bufs, 0, cmdname);
3067         lustre_cfg_bufs_set_string(&bufs, 1, fullpoolname);
3068         if (ostname != NULL)
3069                 lustre_cfg_bufs_set_string(&bufs, 2, ostname);
3070
3071         lcfg = lustre_cfg_new(cmd, &bufs);
3072         if (lcfg == NULL)
3073                 return -ENOMEM;
3074
3075         memset(&data, 0, sizeof(data));
3076         rc = data.ioc_dev = get_mgs_device();
3077         if (rc < 0)
3078                 goto out;
3079
3080         data.ioc_type = LUSTRE_CFG_TYPE;
3081         data.ioc_plen1 = lustre_cfg_len(lcfg->lcfg_bufcount,
3082                                         lcfg->lcfg_buflens);
3083         data.ioc_pbuf1 = (void *)lcfg;
3084
3085         memset(buf, 0, sizeof(rawbuf));
3086         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
3087         if (rc) {
3088                 fprintf(stderr, "error: %s: invalid ioctl\n",
3089                         jt_cmdname(cmdname));
3090                 lustre_cfg_free(lcfg);
3091                 return rc;
3092         }
3093         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_POOL, buf);
3094 out:
3095         if (rc)
3096                 rc = -errno;
3097         lustre_cfg_free(lcfg);
3098         return rc;
3099 }
3100
3101 /**
3102  * Format and send the ioctl to the MGS.
3103  *
3104  * \param       cmd             IOCTL to send
3105  * \param       ret_data        void pointer to return anything from
3106  *                              ioctl
3107  * \param       num_args        number of arguments to pack into the
3108  *                              ioctl buffer
3109  * \param       argv[]          variable number of string arguments
3110  *
3111  * \retval                      0 on success
3112  */
3113 static int nodemap_cmd(enum lcfg_command_type cmd, void *ret_data,
3114                        unsigned int ret_size, ...)
3115 {
3116         va_list                 ap;
3117         char                    *arg;
3118         int                     i = 0;
3119         struct lustre_cfg_bufs  bufs;
3120         struct obd_ioctl_data   data;
3121         struct lustre_cfg       *lcfg;
3122         char                    rawbuf[MAX_IOC_BUFLEN];
3123         char                    *buf = rawbuf;
3124         int                     rc = 0;
3125
3126         lustre_cfg_bufs_reset(&bufs, NULL);
3127
3128         va_start(ap, ret_size);
3129         arg = va_arg(ap, char *);
3130         while (arg != NULL) {
3131                 lustre_cfg_bufs_set_string(&bufs, i, arg);
3132                 i++;
3133                 arg = va_arg(ap, char *);
3134         }
3135         va_end(ap);
3136
3137         lcfg = lustre_cfg_new(cmd, &bufs);
3138         if (lcfg == NULL)
3139                 return -ENOMEM;
3140
3141         memset(&data, 0, sizeof(data));
3142         rc = data.ioc_dev = get_mgs_device();
3143         if (rc < 0)
3144                 goto out;
3145
3146         data.ioc_type = LUSTRE_CFG_TYPE;
3147         data.ioc_plen1 = lustre_cfg_len(lcfg->lcfg_bufcount,
3148                          lcfg->lcfg_buflens);
3149         data.ioc_pbuf1 = (void *)lcfg;
3150
3151         memset(buf, 0, sizeof(rawbuf));
3152         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
3153         if (rc != 0) {
3154                 fprintf(stderr, "error: invalid ioctl: %08x errno: %d with "
3155                                "rc=%d\n", cmd, errno, rc);
3156                 goto out;
3157         }
3158
3159         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_NODEMAP, buf);
3160         if (rc != 0) {
3161                 fprintf(stderr, "error: invalid ioctl: %08x errno: %d with "
3162                                "rc=%d\n", cmd, errno, rc);
3163                 goto out;
3164         }
3165
3166         if (ret_data != NULL) {
3167                 rc = obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
3168                 if (rc != 0)
3169                         goto out;
3170
3171                 if (ret_size > data.ioc_plen1)
3172                         ret_size = data.ioc_plen1;
3173
3174                 memcpy(ret_data, data.ioc_pbuf1, ret_size);
3175         }
3176 out:
3177         lustre_cfg_free(lcfg);
3178
3179         return rc;
3180 }
3181
3182 /**
3183  * activate nodemap functions
3184  *
3185  * \param       argc            number of args
3186  * \param       argv[]          variable string arguments
3187  *
3188  * argv[0]                      1 for activate or 0 for deactivate
3189  *
3190  * \retval                      0 on success
3191  */
3192 int jt_nodemap_activate(int argc, char **argv)
3193 {
3194         int rc;
3195
3196         rc = nodemap_cmd(LCFG_NODEMAP_ACTIVATE, NULL, 0, argv[0], argv[1],
3197                          NULL);
3198
3199         if (rc != 0) {
3200                 errno = -rc;
3201                 perror(argv[0]);
3202         }
3203
3204         return rc;
3205 }
3206
3207 /**
3208  * add a nodemap
3209  *
3210  * \param       argc            number of args
3211  * \param       argv[]          variable string arguments
3212  *
3213  * argv[0]                      nodemap name
3214  *
3215  * \retval                      0 on success
3216  */
3217 int jt_nodemap_add(int argc, char **argv)
3218 {
3219         int rc;
3220
3221         rc = llapi_nodemap_exists(argv[1]);
3222         if (rc == 0) {
3223                 fprintf(stderr, "error: %s existing nodemap name\n", argv[1]);
3224                 return 1;
3225         }
3226
3227         rc = nodemap_cmd(LCFG_NODEMAP_ADD, NULL, 0, argv[0], argv[1], NULL);
3228
3229         if (rc != 0) {
3230                 errno = -rc;
3231                 perror(argv[0]);
3232         }
3233
3234         return rc;
3235 }
3236
3237 /**
3238  * delete a nodemap
3239  *
3240  * \param       argc            number of args
3241  * \param       argv[]          variable string arguments
3242  *
3243  * argv[0]                      nodemap name
3244  *
3245  * \retval                      0 on success
3246  */
3247 int jt_nodemap_del(int argc, char **argv)
3248 {
3249         int rc;
3250
3251         rc = llapi_nodemap_exists(argv[1]);
3252         if (rc != 0) {
3253                 fprintf(stderr, "error: %s not existing nodemap name\n",
3254                         argv[1]);
3255                 return rc;
3256         }
3257         rc = nodemap_cmd(LCFG_NODEMAP_DEL, NULL, 0, argv[0], argv[1], NULL);
3258
3259         if (rc != 0) {
3260                 errno = -rc;
3261                 perror(argv[0]);
3262         }
3263
3264         return rc;
3265 }
3266
3267 /**
3268  * test a nid for nodemap membership
3269  *
3270  * \param       argc            number of args
3271  * \param       argv[]          variable string arguments
3272  *
3273  * argv[0]                      properly formatted nid
3274  *
3275  * \retval                      0 on success
3276  */
3277 int jt_nodemap_test_nid(int argc, char **argv)
3278 {
3279
3280         char    rawbuf[MAX_IOC_BUFLEN];
3281         int     rc;
3282
3283         rc = nodemap_cmd(LCFG_NODEMAP_TEST_NID, &rawbuf, sizeof(rawbuf),
3284                          argv[0], argv[1], NULL);
3285         if (rc == 0)
3286                 printf("%s\n", (char *)rawbuf);
3287
3288         return rc;
3289 }
3290
3291 /**
3292  * test a nodemap id pair for mapping
3293  *
3294  * \param       argc            number of args
3295  * \param       argv[[]         variable string arguments
3296  *
3297  * \retval                      0 on success
3298  *
3299  * The argv array should contain the nodemap name, the id
3300  * to checking the mapping on, and the id type (UID or GID)
3301  *
3302  */
3303 int jt_nodemap_test_id(int argc, char **argv)
3304 {
3305         char    rawbuf[MAX_IOC_BUFLEN];
3306         char    *nidstr = NULL;
3307         char    *idstr = NULL;
3308         char    *typestr = NULL;
3309         int     rc = 0;
3310         int     c;
3311
3312         static struct option long_options[] = {
3313                 {
3314                         .name           = "nid",
3315                         .has_arg        = required_argument,
3316                         .flag           = 0,
3317                         .val            = 'n',
3318                 },
3319                 {
3320                         .name           = "idtype",
3321                         .has_arg        = required_argument,
3322                         .flag           = 0,
3323                         .val            = 't',
3324                 },
3325                 {
3326                         .name           = "id",
3327                         .has_arg        = required_argument,
3328                         .flag           = 0,
3329                         .val            = 'i',
3330                 },
3331                 {
3332                         NULL
3333                 }
3334         };
3335
3336         while ((c = getopt_long(argc, argv, "n:t:i:",
3337                                 long_options, NULL)) != -1) {
3338                 switch (c) {
3339                 case 'n':
3340                         nidstr = optarg;
3341                         break;
3342                 case 't':
3343                         typestr = optarg;
3344                         break;
3345                 case 'i':
3346                         idstr = optarg;
3347                         break;
3348                 }
3349         }
3350
3351         if (nidstr == NULL || typestr == NULL || idstr == NULL) {
3352                 fprintf(stderr, "usage: nodemap_test_id --nid <nid> "
3353                                 "--idtype [uid|gid] --id <id>\n");
3354                 return -1;
3355         }
3356
3357         rc = nodemap_cmd(LCFG_NODEMAP_TEST_ID, &rawbuf, sizeof(rawbuf),
3358                          argv[0], nidstr, typestr, idstr);
3359         if (rc == 0)
3360                 printf("%s\n", (char *)rawbuf);
3361
3362         return rc;
3363 }
3364
3365 /**
3366  * add an nid range to a nodemap
3367  *
3368  * \param       argc            number of args
3369  * \param       argv[]          variable string arguments
3370  *
3371  * --name                       nodemap name
3372  * --range                      properly formatted nid range
3373  *
3374  * \retval                      0 on success
3375  */
3376 int jt_nodemap_add_range(int argc, char **argv)
3377 {
3378         char                    *nodemap_name = NULL;
3379         char                    *nodemap_range = NULL;
3380         struct list_head        nidlist;
3381         char                    min_nid[LNET_NIDSTR_SIZE + 1];
3382         char                    max_nid[LNET_NIDSTR_SIZE + 1];
3383         char                    nid_range[2 * LNET_NIDSTR_SIZE + 2];
3384         int                     rc = 0;
3385         int                     c;
3386
3387         static struct option long_options[] = {
3388                 {
3389                         .name           = "name",
3390                         .has_arg        = required_argument,
3391                         .flag           = 0,
3392                         .val            = 'n',
3393                 },
3394                 {
3395                         .name           = "range",
3396                         .has_arg        = required_argument,
3397                         .flag           = 0,
3398                         .val            = 'r',
3399                 },
3400                 {
3401                         NULL
3402                 }
3403         };
3404
3405         INIT_LIST_HEAD(&nidlist);
3406
3407         while ((c = getopt_long(argc, argv, "n:r:",
3408                                 long_options, NULL)) != -1) {
3409                 switch (c) {
3410                 case 'n':
3411                         nodemap_name = optarg;
3412                         break;
3413                 case 'r':
3414                         nodemap_range = optarg;
3415                         break;
3416                 }
3417         }
3418
3419         if (nodemap_name == NULL || nodemap_range == NULL) {
3420                 fprintf(stderr, "usage: nodemap_add_range --name <name> "
3421                                 "--range <range>\n");
3422                 return -1;
3423         }
3424
3425         if (cfs_parse_nidlist(nodemap_range, strlen(nodemap_range),
3426                               &nidlist) <= 0) {
3427                 fprintf(stderr, "error: %s: can't parse nid range: %s\n",
3428                         jt_cmdname(argv[0]), nodemap_range);
3429                 return -1;
3430         }
3431
3432         if (!cfs_nidrange_is_contiguous(&nidlist)) {
3433                 fprintf(stderr, "error: %s: nodemap ranges must be "
3434                         "contiguous\n", jt_cmdname(argv[0]));
3435                 return -1;
3436         }
3437
3438         cfs_nidrange_find_min_max(&nidlist, &min_nid[0], &max_nid[0],
3439                                   LNET_NIDSTR_SIZE);
3440         snprintf(nid_range, sizeof(nid_range), "%s:%s", min_nid, max_nid);
3441
3442         rc = nodemap_cmd(LCFG_NODEMAP_ADD_RANGE, NULL, 0, argv[0],
3443                          nodemap_name, nid_range, NULL);
3444         if (rc != 0) {
3445                 errno = -rc;
3446                 fprintf(stderr, "error: %s: cannot add range '%s' to nodemap "
3447                                 "'%s': rc = %d\n",
3448                         jt_cmdname(argv[0]), nodemap_range, nodemap_name, rc);
3449         }
3450
3451         return rc;
3452 }
3453
3454 /**
3455  * delete an nid range to a nodemap
3456  *
3457  * \param       argc            number of args
3458  * \param       argv[]          variable string arguments
3459  *
3460  * --name                       nodemap name
3461  * --range                      properly formatted nid range
3462  *
3463  * \retval                      0 on success
3464  */
3465 int jt_nodemap_del_range(int argc, char **argv)
3466 {
3467         char                    *nodemap_name = NULL;
3468         char                    *nodemap_range = NULL;
3469         struct list_head        nidlist;
3470         char                    min_nid[LNET_NIDSTR_SIZE + 1];
3471         char                    max_nid[LNET_NIDSTR_SIZE + 1];
3472         char                    nid_range[2 * LNET_NIDSTR_SIZE + 2];
3473         int                     rc = 0;
3474         int                     c;
3475
3476         static struct option long_options[] = {
3477                 {
3478                         .name           = "name",
3479                         .has_arg        = required_argument,
3480                         .flag           = 0,
3481                         .val            = 'n',
3482                 },
3483                 {
3484                         .name           = "range",
3485                         .has_arg        = required_argument,
3486                         .flag           = 0,
3487                         .val            = 'r',
3488                 },
3489                 {
3490                         NULL
3491                 }
3492         };
3493
3494         INIT_LIST_HEAD(&nidlist);
3495
3496         while ((c = getopt_long(argc, argv, "n:r:",
3497                                 long_options, NULL)) != -1) {
3498                 switch (c) {
3499                 case 'n':
3500                         nodemap_name = optarg;
3501                         break;
3502                 case 'r':
3503                         nodemap_range = optarg;
3504                         break;
3505                 }
3506         }
3507
3508         if (nodemap_name == NULL || nodemap_range == NULL) {
3509                 fprintf(stderr, "usage: nodemap_del_range --name <name> "
3510                                 "--range <range>\n");
3511                 return -1;
3512         }
3513
3514         if (cfs_parse_nidlist(nodemap_range, strlen(nodemap_range),
3515                               &nidlist) <= 0) {
3516                 fprintf(stderr, "error: %s: can't parse nid range: %s\n",
3517                         jt_cmdname(argv[0]), nodemap_range);
3518                 return -1;
3519         }
3520
3521         if (!cfs_nidrange_is_contiguous(&nidlist)) {
3522                 fprintf(stderr, "error: %s: nodemap ranges must be "
3523                         "contiguous\n", jt_cmdname(argv[0]));
3524                 return -1;
3525         }
3526
3527         cfs_nidrange_find_min_max(&nidlist, &min_nid[0], &max_nid[0],
3528                                   LNET_NIDSTR_SIZE);
3529         snprintf(nid_range, sizeof(nid_range), "%s:%s", min_nid, max_nid);
3530
3531         rc = nodemap_cmd(LCFG_NODEMAP_DEL_RANGE, NULL, 0, argv[0],
3532                          nodemap_name, nid_range, NULL);
3533         if (rc != 0) {
3534                 errno = -rc;
3535                 fprintf(stderr, "error: %s: cannot delete range '%s' to "
3536                                "nodemap '%s': rc = %d\n",
3537                         jt_cmdname(argv[0]), nodemap_range, nodemap_name, rc);
3538         }
3539
3540         return rc;
3541 }
3542
3543 /**
3544  * set a fileset on a nodemap
3545  *
3546  * \param       argc            number of args
3547  * \param       argv[]          variable string arguments
3548  *
3549  * --name                       nodemap name
3550  * --fileset                    fileset name
3551  *
3552  * \retval                      0 on success
3553  */
3554 int jt_nodemap_set_fileset(int argc, char **argv)
3555 {
3556         char *nodemap_name = NULL;
3557         char *fileset_name = NULL;
3558         int   rc = 0;
3559         int   c;
3560
3561         static struct option long_options[] = {
3562                 {
3563                         .name           = "name",
3564                         .has_arg        = required_argument,
3565                         .flag           = 0,
3566                         .val            = 'n',
3567                 },
3568                 {
3569                         .name           = "fileset",
3570                         .has_arg        = required_argument,
3571                         .flag           = 0,
3572                         .val            = 'f',
3573                 },
3574                 {
3575                         NULL
3576                 }
3577         };
3578
3579         while ((c = getopt_long(argc, argv, "n:f:",
3580                                 long_options, NULL)) != -1) {
3581                 switch (c) {
3582                 case 'n':
3583                         nodemap_name = optarg;
3584                         break;
3585                 case 'f':
3586                         fileset_name = optarg;
3587                         break;
3588                 }
3589         }
3590
3591         if (nodemap_name == NULL || fileset_name == NULL) {
3592                 fprintf(stderr, "usage: nodemap_set_fileset --name <name> "
3593                                 "--fileset <fileset>\n");
3594                 return -1;
3595         }
3596
3597         rc = nodemap_cmd(LCFG_NODEMAP_SET_FILESET, NULL, 0, argv[0],
3598                          nodemap_name, fileset_name, NULL);
3599         if (rc != 0) {
3600                 errno = -rc;
3601                 fprintf(stderr, "error: %s: cannot set fileset '%s' on nodemap "
3602                                 "'%s': rc = %d\n",
3603                         jt_cmdname(argv[0]), fileset_name, nodemap_name, rc);
3604         }
3605
3606         return rc;
3607 }
3608
3609 /**
3610  * modify a nodemap's behavior
3611  *
3612  * \param       argc            number of args
3613  * \param       argv[]          variable string arguments
3614  *
3615  * --name                       nodemap name
3616  * --property                   nodemap property to change
3617  *                              admin, trusted, squash_uid, squash_gid)
3618  * --value                      value to set property
3619  *
3620  * \retval                      0 on success
3621  */
3622 int jt_nodemap_modify(int argc, char **argv)
3623 {
3624         int                     c;
3625         int                     rc = 0;
3626         enum lcfg_command_type  cmd = 0;
3627         char                    *nodemap_name = NULL;
3628         char                    *param = NULL;
3629         char                    *value = NULL;
3630
3631         static struct option long_options[] = {
3632                 {
3633                         .name           = "name",
3634                         .has_arg        = required_argument,
3635                         .flag           = 0,
3636                         .val            = 'n',
3637                 },
3638                 {
3639                         .name           = "property",
3640                         .has_arg        = required_argument,
3641                         .flag           = 0,
3642                         .val            = 'p',
3643                 },
3644                 {
3645                         .name           = "value",
3646                         .has_arg        = required_argument,
3647                         .flag           = 0,
3648                         .val            = 'v',
3649                 },
3650                 {
3651                         NULL
3652                 }
3653         };
3654
3655         while ((c = getopt_long(argc, argv, "n:p:v:",
3656                                 long_options, NULL)) != -1) {
3657                 switch (c) {
3658                 case 'n':
3659                         nodemap_name = optarg;
3660                         break;
3661                 case 'p':
3662                         param = optarg;
3663                         break;
3664                 case 'v':
3665                         value = optarg;
3666                         break;
3667                 }
3668         }
3669
3670         if (nodemap_name == NULL || param == NULL || value == NULL) {
3671                 fprintf(stderr, "usage: nodemap_modify --name <nodemap_name> "
3672                                 "--property <property_name> --value <value>\n");
3673                 fprintf(stderr, "valid properties: admin trusted map_mode "
3674                                 "squash_uid squash_gid deny_unknown\n");
3675                 return -1;
3676         }
3677
3678         if (strcmp("admin", param) == 0) {
3679                 cmd = LCFG_NODEMAP_ADMIN;
3680         } else if (strcmp("trusted", param) == 0) {
3681                 cmd = LCFG_NODEMAP_TRUSTED;
3682         } else if (strcmp("deny_unknown", param) == 0) {
3683                 cmd = LCFG_NODEMAP_DENY_UNKNOWN;
3684         } else if (strcmp("squash_uid", param) == 0) {
3685                 cmd = LCFG_NODEMAP_SQUASH_UID;
3686         } else if (strcmp("squash_gid", param) == 0) {
3687                 cmd = LCFG_NODEMAP_SQUASH_GID;
3688         } else if (strcmp("map_mode", param) == 0) {
3689                 cmd = LCFG_NODEMAP_MAP_MODE;
3690         } else {
3691                 fprintf(stderr, "error: %s: nodemap_modify invalid "
3692                                 "subcommand: %s\n",
3693                         jt_cmdname(argv[0]), param);
3694                 return -1;
3695         }
3696
3697         rc = nodemap_cmd(cmd, NULL, 0, argv[0], nodemap_name, param,
3698                          value, NULL);
3699         if (rc != 0) {
3700                 errno = -rc;
3701                 fprintf(stderr, "error: %s: cannot modify nodemap '%s' "
3702                                "to param '%s': value '%s': rc = %d\n",
3703                         jt_cmdname(argv[0]), nodemap_name, param, value, rc);
3704         }
3705
3706         return rc;
3707 }
3708
3709 int jt_nodemap_add_idmap(int argc, char **argv)
3710 {
3711         int                     c;
3712         enum                    lcfg_command_type cmd = 0;
3713         char                    *nodemap_name = NULL;
3714         char                    *idmap = NULL;
3715         char                    *idtype = NULL;
3716         int                     rc = 0;
3717
3718         static struct option long_options[] = {
3719                 {
3720                         .name           = "name",
3721                         .has_arg        = required_argument,
3722                         .flag           = 0,
3723                         .val            = 'n',
3724                 },
3725                 {
3726                         .name           = "idmap",
3727                         .has_arg        = required_argument,
3728                         .flag           = 0,
3729                         .val            = 'm',
3730                 },
3731                 {
3732                         .name           = "idtype",
3733                         .has_arg        = required_argument,
3734                         .flag           = 0,
3735                         .val            = 'i',
3736                 },
3737                 {
3738                         NULL
3739                 }
3740         };
3741
3742         while ((c = getopt_long(argc, argv, "n:m:i:",
3743                                 long_options, NULL)) != -1) {
3744                 switch (c) {
3745                 case 'n':
3746                         nodemap_name = optarg;
3747                         break;
3748                 case 'm':
3749                         idmap = optarg;
3750                         break;
3751                 case 'i':
3752                         idtype = optarg;
3753                         break;
3754                 }
3755         }
3756
3757         if (nodemap_name == NULL || idmap == NULL || idtype == NULL) {
3758                 fprintf(stderr, "usage: %s --name <name> --idtype [uid | gid]"
3759                         " --idmap <client id>:<filesystem id>\n", argv[0]);
3760                 return -1;
3761         }
3762
3763         if (strcmp("uid", idtype) == 0) {
3764                 cmd = LCFG_NODEMAP_ADD_UIDMAP;
3765         } else if (strcmp("gid", idtype) == 0) {
3766                 cmd = LCFG_NODEMAP_ADD_GIDMAP;
3767         } else {
3768                 fprintf(stderr, "usage: %s --name <name> --idtype [uid | gid]"
3769                         " --idmap <client id>:<filesystem id>\n", argv[0]);
3770                 return -1;
3771         }
3772
3773         rc = nodemap_cmd(cmd, NULL, 0, argv[0], nodemap_name, idmap, NULL);
3774         if (rc != 0) {
3775                 errno = -rc;
3776                 fprintf(stderr, "cannot add %smap '%s' to nodemap '%s'"
3777                         ": rc = %d\n", idtype, idmap, nodemap_name, rc);
3778         }
3779
3780         return rc;
3781 }
3782
3783 int jt_nodemap_del_idmap(int argc, char **argv)
3784 {
3785         int                     c;
3786         enum                    lcfg_command_type cmd = 0;
3787         char                    *nodemap_name = NULL;
3788         char                    *idmap = NULL;
3789         char                    *idtype = NULL;
3790         int                     rc = 0;
3791
3792         static struct option long_options[] = {
3793                 {
3794                         .name           = "name",
3795                         .has_arg        = required_argument,
3796                         .flag           = 0,
3797                         .val            = 'n',
3798                 },
3799                 {
3800                         .name           = "idmap",
3801                         .has_arg        = required_argument,
3802                         .flag           = 0,
3803                         .val            = 'm',
3804                 },
3805                 {
3806                         .name           = "idtype",
3807                         .has_arg        = required_argument,
3808                         .flag           = 0,
3809                         .val            = 'i',
3810                 },
3811                 {
3812                         NULL
3813                 }
3814         };
3815
3816         while ((c = getopt_long(argc, argv, "n:m:i:",
3817                                 long_options, NULL)) != -1) {
3818                 switch (c) {
3819                 case 'n':
3820                         nodemap_name = optarg;
3821                         break;
3822                 case 'm':
3823                         idmap = optarg;
3824                         break;
3825                 case 'i':
3826                         idtype = optarg;
3827                         break;
3828                 }
3829         }
3830
3831         if (nodemap_name == NULL || idmap == NULL || idtype == NULL) {
3832                 fprintf(stderr, "usage: %s --name <name> --idtype [uid | gid]"
3833                         " --idmap <client id>:<filesystem id>\n", argv[0]);
3834                 return -1;
3835         }
3836
3837         if (strcmp("uid", idtype) == 0)
3838                 cmd = LCFG_NODEMAP_DEL_UIDMAP;
3839         else
3840                 cmd = LCFG_NODEMAP_DEL_GIDMAP;
3841
3842         rc = nodemap_cmd(cmd, NULL, 0, argv[0], nodemap_name, idmap, NULL);
3843         if (rc != 0) {
3844                 errno = -rc;
3845                 fprintf(stderr, "cannot delete %smap '%s' from nodemap '%s'"
3846                         ": rc = %d\n", idtype, idmap, nodemap_name, rc);
3847         }
3848
3849         return rc;
3850 }
3851
3852 /*
3853  * this function tranforms a rule [start-end/step] into an array
3854  * of matching numbers
3855  * supported forms are:
3856  * [start]                : just this number
3857  * [start-end]            : all numbers from start to end
3858  * [start-end/step]       : numbers from start to end with increment of step
3859  * on return, format contains a printf format string which can be used
3860  * to generate all the strings
3861  */
3862 static int get_array_idx(char *rule, char *format, int **array)
3863 {
3864         char *start, *end, *ptr;
3865         unsigned int lo, hi, step;
3866         int array_sz = 0;
3867         int i, array_idx;
3868         int rc;
3869
3870         start = strchr(rule, '[');
3871         end = strchr(rule, ']');
3872         if ((start == NULL) || (end == NULL)) {
3873                 *array = malloc(sizeof(int));
3874                 if (*array == NULL)
3875                         return 0;
3876                 strcpy(format, rule);
3877                 array_sz = 1;
3878                 return array_sz;
3879         }
3880         *start = '\0';
3881         *end = '\0';
3882         end++;
3883         start++;
3884         /* put in format the printf format (the rule without the range) */
3885         sprintf(format, "%s%%.4x%s", rule, end);
3886
3887         array_idx = 0;
3888         array_sz = 0;
3889         *array = NULL;
3890         /* loop on , separator */
3891         do {
3892                 /* extract the 3 fields */
3893                 rc = sscanf(start, "%x-%x/%u", &lo, &hi, &step);
3894                 switch (rc) {
3895                 case 0:
3896                         goto err;
3897                 case 1: {
3898                         void *tmp;
3899
3900                         array_sz++;
3901                         tmp = realloc(*array, array_sz * sizeof(int));
3902                         if (tmp == NULL)
3903                                 goto err;
3904                         *array = tmp;
3905                         (*array)[array_idx] = lo;
3906                         array_idx++;
3907                         break;
3908                 }
3909                 case 2: {
3910                         step = 1;
3911                         /* do not break to share code with case 3: */
3912                 }
3913                 case 3: {
3914                         void *tmp;
3915
3916                         if ((hi < lo) || (step == 0))
3917                                 goto err;
3918                         array_sz += (hi - lo) / step + 1;
3919                         tmp = realloc(*array, array_sz * sizeof(int));
3920                         if (tmp == NULL)
3921                                 goto err;
3922                         *array = tmp;
3923                         for (i = lo; i <= hi; i+=step, array_idx++)
3924                                 (*array)[array_idx] = i;
3925                         break;
3926                 }
3927                 }
3928                 ptr = strchr(start, ',');
3929                 if (ptr != NULL)
3930                         start = ptr + 1;
3931
3932         } while (ptr != NULL);
3933         return array_sz;
3934 err:
3935         if (*array != NULL) {
3936                 free(*array);
3937                 *array = NULL;
3938         }
3939         return 0;
3940 }
3941
3942 static int extract_fsname_poolname(const char *arg, char *fsname,
3943                                    char *poolname)
3944 {
3945         char *ptr;
3946         int rc;
3947
3948         strlcpy(fsname, arg, PATH_MAX + 1);
3949         ptr = strchr(fsname, '.');
3950         if (ptr == NULL) {
3951                 fprintf(stderr, ". is missing in %s\n", fsname);
3952                 rc = -EINVAL;
3953                 goto err;
3954         }
3955
3956         if ((ptr - fsname) == 0) {
3957                 fprintf(stderr, "fsname is empty\n");
3958                 rc = -EINVAL;
3959                 goto err;
3960         }
3961
3962         *ptr = '\0';
3963         ++ptr;
3964
3965         rc = lustre_is_fsname_valid(fsname, 1, LUSTRE_MAXFSNAME);
3966         if (rc < 0) {
3967                 fprintf(stderr, "filesystem name %s must be 1-%d chars\n",
3968                         fsname, LUSTRE_MAXFSNAME);
3969                 rc = -EINVAL;
3970                 goto err;
3971         } else if (rc > 0) {
3972                 fprintf(stderr, "char '%c' not allowed in filesystem name\n",
3973                         rc);
3974                 rc = -EINVAL;
3975                 goto err;
3976         }
3977
3978         rc = lustre_is_poolname_valid(ptr, 1, LOV_MAXPOOLNAME);
3979         if (rc == -1) {
3980                 fprintf(stderr, "poolname is empty\n");
3981                 rc = -EINVAL;
3982                 goto err;
3983         } else if (rc == -2) {
3984                 fprintf(stderr,
3985                         "poolname %s is too long (max is %d)\n",
3986                         ptr, LOV_MAXPOOLNAME);
3987                 rc = -ENAMETOOLONG;
3988                 goto err;
3989         } else if (rc > 0) {
3990                 fprintf(stderr, "char '%c' not allowed in pool name '%s'\n",
3991                         rc, ptr);
3992                 rc = -EINVAL;
3993                 goto err;
3994         }
3995
3996         strncpy(poolname, ptr, LOV_MAXPOOLNAME);
3997         poolname[LOV_MAXPOOLNAME] = '\0';
3998         return 0;
3999
4000 err:
4001         fprintf(stderr, "argument %s must be <fsname>.<poolname>\n", arg);
4002         return rc;
4003 }
4004
4005 int jt_pool_cmd(int argc, char **argv)
4006 {
4007         enum lcfg_command_type cmd;
4008         char fsname[PATH_MAX + 1];
4009         char poolname[LOV_MAXPOOLNAME + 1];
4010         char *ostnames_buf = NULL;
4011         int i, rc;
4012         int *array = NULL, array_sz;
4013         struct {
4014                 int     rc;
4015                 char   *ostname;
4016         } *cmds = NULL;
4017
4018         switch (argc) {
4019         case 0:
4020         case 1: return CMD_HELP;
4021         case 2: {
4022                 if (strcmp("pool_new", argv[0]) == 0)
4023                         cmd = LCFG_POOL_NEW;
4024                 else if (strcmp("pool_destroy", argv[0]) == 0)
4025                         cmd = LCFG_POOL_DEL;
4026                 else if (strcmp("pool_list", argv[0]) == 0)
4027                          return llapi_poollist(argv[1]);
4028                 else return CMD_HELP;
4029
4030                 rc = extract_fsname_poolname(argv[1], fsname, poolname);
4031                 if (rc)
4032                         break;
4033
4034                 rc = pool_cmd(cmd, argv[0], argv[1], fsname, poolname, NULL);
4035                 if (rc)
4036                         break;
4037
4038                 check_pool_cmd_result(cmd, fsname, poolname, NULL);
4039                 break;
4040         }
4041         default: {
4042                 char format[2*MAX_OBD_NAME];
4043
4044                 if (strcmp("pool_remove", argv[0]) == 0) {
4045                         cmd = LCFG_POOL_REM;
4046                 } else if (strcmp("pool_add", argv[0]) == 0) {
4047                         cmd = LCFG_POOL_ADD;
4048                 } else {
4049                         return CMD_HELP;
4050                 }
4051
4052                 rc = extract_fsname_poolname(argv[1], fsname, poolname);
4053                 if (rc)
4054                         break;
4055
4056                 for (i = 2; i < argc; i++) {
4057                         int j;
4058
4059                         array_sz = get_array_idx(argv[i], format, &array);
4060                         if (array_sz == 0)
4061                                 return CMD_HELP;
4062
4063                         cmds = malloc(array_sz * sizeof(cmds[0]));
4064                         if (cmds != NULL) {
4065                                 ostnames_buf = malloc(array_sz *
4066                                                       (MAX_OBD_NAME + 1));
4067                         } else {
4068                                 free(array);
4069                                 rc = -ENOMEM;
4070                                 goto out;
4071                         }
4072
4073                         for (j = 0; j < array_sz; j++) {
4074                                 char ostname[MAX_OBD_NAME + 1];
4075
4076                                 snprintf(ostname, MAX_OBD_NAME, format,
4077                                          array[j]);
4078                                 ostname[MAX_OBD_NAME] = '\0';
4079
4080                                 rc = check_and_complete_ostname(fsname,ostname);
4081                                 if (rc) {
4082                                         free(array);
4083                                         free(cmds);
4084                                         if (ostnames_buf)
4085                                                 free(ostnames_buf);
4086                                         goto out;
4087                                 }
4088                                 if (ostnames_buf != NULL) {
4089                                         cmds[j].ostname =
4090                                           &ostnames_buf[(MAX_OBD_NAME + 1) * j];
4091                                         strcpy(cmds[j].ostname, ostname);
4092                                 } else {
4093                                         cmds[j].ostname = NULL;
4094                                 }
4095                                 cmds[j].rc = pool_cmd(cmd, argv[0], argv[1],
4096                                                       fsname, poolname,
4097                                                       ostname);
4098                                 /* Return an err if any of the add/dels fail */
4099                                 if (!rc)
4100                                         rc = cmds[j].rc;
4101                         }
4102                         for (j = 0; j < array_sz; j++) {
4103                                 if (!cmds[j].rc) {
4104                                         char ostname[MAX_OBD_NAME + 1];
4105
4106                                         if (!cmds[j].ostname) {
4107                                                 snprintf(ostname, MAX_OBD_NAME,
4108                                                          format, array[j]);
4109                                                 ostname[MAX_OBD_NAME] = '\0';
4110                                                 check_and_complete_ostname(
4111                                                         fsname, ostname);
4112                                         } else {
4113                                                 strcpy(ostname,
4114                                                        cmds[j].ostname);
4115                                         }
4116                                         check_pool_cmd_result(cmd, fsname,
4117                                                               poolname,ostname);
4118                                 }
4119                         }
4120                         if (array_sz > 0)
4121                                 free(array);
4122                         if (cmds)
4123                                 free(cmds);
4124                         if (ostnames_buf != NULL)
4125                                 free(ostnames_buf);
4126                 }
4127                 /* fall through */
4128         }
4129         } /* switch */
4130
4131 out:
4132         if (rc != 0) {
4133                 errno = -rc;
4134                 perror(argv[0]);
4135         }
4136
4137         return rc;
4138 }
4139
4140 static const char *barrier_status2name(enum barrier_status status)
4141 {
4142         switch (status) {
4143         case BS_INIT:
4144                 return "init";
4145         case BS_FREEZING_P1:
4146                 return "freezing_p1";
4147         case BS_FREEZING_P2:
4148                 return "freezing_p2";
4149         case BS_FROZEN:
4150                 return "frozen";
4151         case BS_THAWING:
4152                 return "thawing";
4153         case BS_THAWED:
4154                 return "thawed";
4155         case BS_FAILED:
4156                 return "failed";
4157         case BS_EXPIRED:
4158                 return "expired";
4159         case BS_RESCAN:
4160                 return "rescan";
4161         default:
4162                 return "unknown";
4163         }
4164 }
4165
4166 int jt_barrier_freeze(int argc, char **argv)
4167 {
4168         struct obd_ioctl_data data;
4169         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
4170         struct barrier_ctl bc;
4171         int rc;
4172
4173         if (argc < 2 || argc > 3)
4174                 return CMD_HELP;
4175
4176         memset(&data, 0, sizeof(data));
4177         rc = data.ioc_dev = get_mgs_device();
4178         if (rc < 0)
4179                 return rc;
4180
4181         memset(&bc, 0, sizeof(bc));
4182         bc.bc_version = BARRIER_VERSION_V1;
4183         bc.bc_cmd = BC_FREEZE;
4184         if (argc == 3)
4185                 bc.bc_timeout = atoi(argv[2]);
4186         if (bc.bc_timeout == 0)
4187                 bc.bc_timeout = BARRIER_TIMEOUT_DEFAULT;
4188
4189         if (strlen(argv[1]) > 8) {
4190                 fprintf(stderr, "%s: fsname name %s is too long. "
4191                         "It should not exceed 8.\n", argv[0], argv[1]);
4192                 return -EINVAL;
4193         }
4194
4195         strncpy(bc.bc_name, argv[1], sizeof(bc.bc_name));
4196         data.ioc_inlbuf1 = (char *)&bc;
4197         data.ioc_inllen1 = sizeof(bc);
4198         memset(buf, 0, sizeof(rawbuf));
4199         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
4200         if (rc) {
4201                 fprintf(stderr, "Fail to pack ioctl data: rc = %d.\n", rc);
4202                 return rc;
4203         }
4204
4205         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_BARRIER, buf);
4206         if (rc < 0)
4207                 fprintf(stderr, "Fail to freeze barrier for %s: %s\n",
4208                         argv[1], strerror(errno));
4209
4210         return rc;
4211 }
4212
4213 int jt_barrier_thaw(int argc, char **argv)
4214 {
4215         struct obd_ioctl_data data;
4216         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
4217         struct barrier_ctl bc;
4218         int rc;
4219
4220         if (argc != 2)
4221                 return CMD_HELP;
4222
4223         memset(&data, 0, sizeof(data));
4224         rc = data.ioc_dev = get_mgs_device();
4225         if (rc < 0)
4226                 return rc;
4227
4228         memset(&bc, 0, sizeof(bc));
4229         bc.bc_version = BARRIER_VERSION_V1;
4230         bc.bc_cmd = BC_THAW;
4231
4232         if (strlen(argv[1]) > 8) {
4233                 fprintf(stderr, "fsname name %s is too long. "
4234                         "It should not exceed 8.\n", argv[1]);
4235                 return -EINVAL;
4236         }
4237
4238         strncpy(bc.bc_name, argv[1], sizeof(bc.bc_name));
4239         data.ioc_inlbuf1 = (char *)&bc;
4240         data.ioc_inllen1 = sizeof(bc);
4241         memset(buf, 0, sizeof(rawbuf));
4242         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
4243         if (rc) {
4244                 fprintf(stderr, "Fail to pack ioctl data: rc = %d.\n", rc);
4245                 return rc;
4246         }
4247
4248         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_BARRIER, buf);
4249         if (rc < 0)
4250                 fprintf(stderr, "Fail to thaw barrier for %s: %s\n",
4251                         argv[1], strerror(errno));
4252
4253         return rc;
4254 }
4255
4256 int __jt_barrier_stat(int argc, char **argv, struct barrier_ctl *bc)
4257 {
4258         struct obd_ioctl_data data;
4259         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
4260         int rc;
4261
4262         memset(&data, 0, sizeof(data));
4263         rc = data.ioc_dev = get_mgs_device();
4264         if (rc < 0)
4265                 return rc;
4266
4267         memset(bc, 0, sizeof(*bc));
4268         bc->bc_version = BARRIER_VERSION_V1;
4269         bc->bc_cmd = BC_STAT;
4270         strncpy(bc->bc_name, argv[1], sizeof(bc->bc_name));
4271         data.ioc_inlbuf1 = (char *)bc;
4272         data.ioc_inllen1 = sizeof(*bc);
4273         memset(buf, 0, sizeof(rawbuf));
4274         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
4275         if (rc) {
4276                 fprintf(stderr, "Fail to pack ioctl data: rc = %d.\n", rc);
4277                 return rc;
4278         }
4279
4280         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_BARRIER, buf);
4281         if (rc < 0)
4282                 fprintf(stderr, "Fail to query barrier for %s: %s\n",
4283                         argv[1], strerror(errno));
4284         else
4285                 obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
4286
4287         return rc;
4288 }
4289
4290 int jt_barrier_stat(int argc, char **argv)
4291 {
4292         struct barrier_ctl bc;
4293         int rc;
4294
4295         if (argc != 2)
4296                 return CMD_HELP;
4297
4298         if (strlen(argv[1]) > 8) {
4299                 fprintf(stderr, "fsname name %s is too long. "
4300                         "It should not exceed 8.\n", argv[1]);
4301                 return -EINVAL;
4302         }
4303
4304         rc = __jt_barrier_stat(argc, argv, &bc);
4305         if (!rc) {
4306                 printf("The barrier for %s is in '%s'\n",
4307                        argv[1], barrier_status2name(bc.bc_status));
4308                 if (bc.bc_status == BS_FREEZING_P1 ||
4309                     bc.bc_status == BS_FREEZING_P2 ||
4310                     bc.bc_status == BS_FROZEN)
4311                         printf("The barrier will be expired after %d "
4312                                "seconds\n", bc.bc_timeout);
4313         }
4314
4315         return rc;
4316 }
4317
4318 int jt_barrier_rescan(int argc, char **argv)
4319 {
4320         struct obd_ioctl_data data;
4321         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
4322         struct barrier_ctl bc;
4323         int rc;
4324
4325         if (argc < 2 || argc > 3)
4326                 return CMD_HELP;
4327
4328         memset(&data, 0, sizeof(data));
4329         rc = data.ioc_dev = get_mgs_device();
4330         if (rc < 0)
4331                 return rc;
4332
4333         memset(&bc, 0, sizeof(bc));
4334         bc.bc_version = BARRIER_VERSION_V1;
4335         bc.bc_cmd = BC_RESCAN;
4336         if (argc == 3)
4337                 bc.bc_timeout = atoi(argv[2]);
4338         if (bc.bc_timeout == 0)
4339                 bc.bc_timeout = BARRIER_TIMEOUT_DEFAULT;
4340
4341         if (strlen(argv[1]) > 8) {
4342                 fprintf(stderr, "fsname name %s is too long. "
4343                         "It should not exceed 8.\n", argv[1]);
4344                 return -EINVAL;
4345         }
4346
4347         strncpy(bc.bc_name, argv[1], sizeof(bc.bc_name));
4348         data.ioc_inlbuf1 = (char *)&bc;
4349         data.ioc_inllen1 = sizeof(bc);
4350         memset(buf, 0, sizeof(rawbuf));
4351         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
4352         if (rc) {
4353                 fprintf(stderr, "Fail to pack ioctl data: rc = %d.\n", rc);
4354                 return rc;
4355         }
4356
4357         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_BARRIER, buf);
4358         if (rc < 0) {
4359                 fprintf(stderr, "Fail to rescan barrier bitmap for %s: %s\n",
4360                         argv[1], strerror(errno));
4361         } else {
4362                 obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
4363                 printf("%u of %u MDT(s) in the filesystem %s are inactive\n",
4364                        bc.bc_absence, bc.bc_total, argv[1]);
4365         }
4366
4367         return rc;
4368 }
4369
4370 int jt_get_obj_version(int argc, char **argv)
4371 {
4372         struct lu_fid fid;
4373         struct obd_ioctl_data data;
4374         __u64 version, id = ULLONG_MAX, group = ULLONG_MAX;
4375         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf, *fidstr;
4376         int rc, c;
4377
4378         while ((c = getopt(argc, argv, "i:g:")) != -1) {
4379                 switch (c) {
4380                 case 'i':
4381                         id = strtoull(optarg, NULL, 0);
4382                         break;
4383                 case 'g':
4384                         group = strtoull(optarg, NULL, 0);
4385                         break;
4386                 default:
4387                         return CMD_HELP;
4388                 }
4389         }
4390
4391         argc -= optind;
4392         argv += optind;
4393
4394         if (!(id != ULLONG_MAX && group != ULLONG_MAX && argc == 0) &&
4395             !(id == ULLONG_MAX && group == ULLONG_MAX && argc == 1))
4396                 return CMD_HELP;
4397
4398         memset(&data, 0, sizeof data);
4399         data.ioc_dev = cur_device;
4400         if (argc == 1) {
4401                 fidstr = *argv;
4402                 while (*fidstr == '[')
4403                         fidstr++;
4404                 sscanf(fidstr, SFID, RFID(&fid));
4405
4406                 data.ioc_inlbuf1 = (char *) &fid;
4407                 data.ioc_inllen1 = sizeof fid;
4408         } else {
4409                 data.ioc_inlbuf3 = (char *) &id;
4410                 data.ioc_inllen3 = sizeof id;
4411                 data.ioc_inlbuf4 = (char *) &group;
4412                 data.ioc_inllen4 = sizeof group;
4413         }
4414         data.ioc_inlbuf2 = (char *) &version;
4415         data.ioc_inllen2 = sizeof version;
4416
4417         memset(buf, 0, sizeof *buf);
4418         rc = obd_ioctl_pack(&data, &buf, sizeof rawbuf);
4419         if (rc) {
4420                 fprintf(stderr, "error: %s: packing ioctl arguments: %s\n",
4421                         jt_cmdname(argv[0]), strerror(-rc));
4422                 return rc;
4423         }
4424
4425         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_GET_OBJ_VERSION, buf);
4426         if (rc == -1) {
4427                 fprintf(stderr, "error: %s: ioctl: %s\n",
4428                         jt_cmdname(argv[0]), strerror(errno));
4429                 return -errno;
4430         }
4431
4432         obd_ioctl_unpack(&data, buf, sizeof rawbuf);
4433         printf("%#jx\n", (uintmax_t)version);
4434         return 0;
4435 }
4436
4437 int jt_changelog_register(int argc, char **argv)
4438 {
4439         struct obd_ioctl_data    data = { 0 };
4440         char                     rawbuf[MAX_IOC_BUFLEN] = "";
4441         char                    *buf = rawbuf;
4442         char                    *device = lcfg_get_devname();
4443         bool                     print_name_only = false;
4444         int                      c;
4445         int                      rc;
4446
4447         if (argc > 2)
4448                 return CMD_HELP;
4449
4450         while ((c = getopt(argc, argv, "hn")) >= 0) {
4451                 switch (c) {
4452                 case 'n':
4453                         print_name_only = true;
4454                         break;
4455                 case 'h':
4456                 default:
4457                         return CMD_HELP;
4458                 }
4459         }
4460
4461         if (cur_device < 0 || device == NULL)
4462                 return CMD_HELP;
4463
4464         data.ioc_dev = cur_device;
4465
4466         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
4467         if (rc < 0) {
4468                 fprintf(stderr, "error: %s: cannot pack ioctl: %s\n",
4469                         jt_cmdname(argv[0]), strerror(-rc));
4470                return rc;
4471         }
4472
4473         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_CHANGELOG_REG, buf);
4474         if (rc < 0) {
4475                 rc = -errno;
4476                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
4477                         strerror(-rc));
4478                 return rc;
4479         }
4480
4481         obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
4482
4483         if (data.ioc_u32_1 == 0) {
4484                 fprintf(stderr, "received invalid userid!\n");
4485                 return -EPROTO;
4486         }
4487
4488         if (print_name_only)
4489                 printf("%s%u\n", CHANGELOG_USER_PREFIX, data.ioc_u32_1);
4490         else
4491                 printf("%s: Registered changelog userid '%s%u'\n",
4492                        device, CHANGELOG_USER_PREFIX, data.ioc_u32_1);
4493
4494         return 0;
4495 }
4496
4497 int jt_changelog_deregister(int argc, char **argv)
4498 {
4499         struct obd_ioctl_data    data = { 0 };
4500         char                     rawbuf[MAX_IOC_BUFLEN] = "";
4501         char                    *buf = rawbuf;
4502         char                    *device = lcfg_get_devname();
4503         int                      id;
4504         int                      rc;
4505
4506         if (argc != 2 || cur_device < 0 || device == NULL)
4507                 return CMD_HELP;
4508
4509         rc = sscanf(argv[1], CHANGELOG_USER_PREFIX"%d", &id);
4510         if (rc != 1 || id <= 0) {
4511                 fprintf(stderr,
4512                         "error: %s: expected id of the form %s<num> got '%s'\n",
4513                         jt_cmdname(argv[0]), CHANGELOG_USER_PREFIX, argv[1]);
4514                 return CMD_HELP;
4515         }
4516
4517         data.ioc_dev = cur_device;
4518         data.ioc_u32_1 = id;
4519
4520         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
4521         if (rc < 0) {
4522                 fprintf(stderr, "error: %s: invalid ioctl\n",
4523                         jt_cmdname(argv[0]));
4524                 return rc;
4525         }
4526
4527         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_CHANGELOG_DEREG, buf);
4528         if (rc < 0) {
4529                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
4530                         strerror(rc = errno));
4531                 return rc;
4532         }
4533
4534         obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
4535         printf("%s: Deregistered changelog user '%s%u'\n",
4536                device, CHANGELOG_USER_PREFIX, data.ioc_u32_1);
4537
4538         return 0;
4539 }