lustre/utils/obd.c

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