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