lustre/utils/obd.c

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