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.sun.com/software/products/lustre/docs/GPLv2.pdf
  19  *
  20  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  21  * CA 95054 USA or visit www.sun.com if you need additional information or
  22  * have any questions.
  23  *
  24  * GPL HEADER END
  25  */
  26 /*
  27  * Copyright (c) 2002, 2010, Oracle and/or its affiliates. All rights reserved.
  28  * Use is subject to license terms.
  29  *
  30  * Copyright (c) 2011, 2015, Intel Corporation.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  *
  36  * lustre/utils/obd.c
  37  *
  38  * Author: Peter J. Braam <braam@clusterfs.com>
  39  * Author: Phil Schwan <phil@clusterfs.com>
  40  * Author: Andreas Dilger <adilger@clusterfs.com>
  41  * Author: Robert Read <rread@clusterfs.com>
  42  */
  43
  44 #include <sys/ioctl.h>
  45 #include <sys/socket.h>
  46 #include <sys/stat.h>
  47 #include <sys/time.h>
  48 #include <sys/types.h>
  49 #include <sys/un.h>
  50 #include <sys/wait.h>
  51
  52 #include <ctype.h>
  53 #include <errno.h>
  54 #include <fcntl.h>
  55 #include <getopt.h>
  56 #include <signal.h>
  57 #include <stdarg.h>
  58 #include <stdbool.h>
  59 #include <stdio.h>
  60 #include <stdlib.h>
  61 #include <string.h>
  62 #include <time.h>
  63 #include <unistd.h>
  64 #include <limits.h>
  65
  66 #include "obdctl.h"
  67 #include <libcfs/util/list.h>
  68 #include <libcfs/util/ioctl.h>
  69 #include <libcfs/util/param.h>
  70 #include <libcfs/util/parser.h>
  71 #include <libcfs/util/string.h>
  72
  73 #include <lnet/nidstr.h>
  74 #include <lustre/lustre_idl.h>
  75 #include <lustre_cfg.h>
  76 #include <lustre_ioctl.h>
  77 #include <lustre_ver.h>
  78
  79 #include <lnet/lnetctl.h>
  80 #include <lustre/lustreapi.h>
  81 #include <lustre_param.h>
  82
  83 #define MAX_STRING_SIZE 128
  84
  85 #if HAVE_LIBPTHREAD
  86 #include <sys/ipc.h>
  87 #include <sys/shm.h>
  88 #include <pthread.h>
  89
  90 #define MAX_THREADS 4096
  91 #define MAX_BASE_ID 0xffffffff
  92 #define NIDSTRING_LENGTH 64
  93 struct shared_data {
  94         pthread_mutex_t mutex;
  95         pthread_cond_t  cond;
  96         int       stopping;
  97         struct {
  98                 __u64 counters[MAX_THREADS];
  99                 __u64 offsets[MAX_THREADS];
 100                 int   thr_running;
 101                 int   start_barrier;
 102                 int   stop_barrier;
 103                 struct timeval start_time;
 104                 struct timeval end_time;
 105         } body;
 106 };
 107
 108 static struct shared_data *shared_data;
 109 static __u64 counter_snapshot[2][MAX_THREADS];
 110 static int prev_valid;
 111 static struct timeval prev_time;
 112 static int thread;
 113 static int nthreads;
 114 #else
 115 const int thread = 0;
 116 const int nthreads = 1;
 117 #endif
 118
 119 static int cur_device = -1;
 120
 121 static int l2_ioctl(int dev_id, int opc, void *buf)
 122 {
 123         return l_ioctl(dev_id, opc, buf);
 124 }
 125
 126 int lcfg_ioctl(char * func, int dev_id, struct lustre_cfg *lcfg)
 127 {
 128         struct obd_ioctl_data data;
 129         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
 130         int rc;
 131
 132         memset(&data, 0, sizeof(data));
 133         data.ioc_dev = cur_device;
 134         data.ioc_type = LUSTRE_CFG_TYPE;
 135         data.ioc_plen1 = lustre_cfg_len(lcfg->lcfg_bufcount,
 136                                         lcfg->lcfg_buflens);
 137         data.ioc_pbuf1 = (void *)lcfg;
 138         memset(buf, 0, sizeof(rawbuf));
 139         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
 140         if (rc) {
 141                 fprintf(stderr, "error: %s: invalid ioctl\n",
 142                         jt_cmdname(func));
 143                 return rc;
 144         }
 145
 146         rc =  l_ioctl(dev_id, OBD_IOC_PROCESS_CFG, buf);
 147
 148         return rc;
 149 }
 150
 151 static int do_device(char *func, char *devname);
 152
 153 static int get_mgs_device()
 154 {
 155         char mgs[] = "$MGS";
 156         static int mgs_device = -1;
 157
 158         if (mgs_device == -1) {
 159                 int rc;
 160                 do_disconnect(NULL, 1);
 161                 rc = do_device("mgsioc", mgs);
 162                 if (rc) {
 163                         fprintf(stderr,
 164                                 "This command must be run on the MGS.\n");
 165                         errno = ENODEV;
 166                         return -1;
 167                 }
 168                 mgs_device = cur_device;
 169         }
 170         return mgs_device;
 171 }
 172
 173 /* Returns -1 on error with errno set */
 174 int lcfg_mgs_ioctl(char *func, int dev_id, struct lustre_cfg *lcfg)
 175 {
 176         struct obd_ioctl_data data;
 177         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
 178         int rc;
 179
 180         memset(&data, 0, sizeof(data));
 181         rc = data.ioc_dev = get_mgs_device();
 182         if (rc < 0)
 183                 goto out;
 184         data.ioc_type = LUSTRE_CFG_TYPE;
 185         data.ioc_plen1 = lustre_cfg_len(lcfg->lcfg_bufcount,
 186                                         lcfg->lcfg_buflens);
 187         data.ioc_pbuf1 = (void *)lcfg;
 188         memset(buf, 0, sizeof(rawbuf));
 189         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
 190         if (rc) {
 191                 fprintf(stderr, "error: %s: invalid ioctl\n",
 192                         jt_cmdname(func));
 193                 return rc;
 194         }
 195
 196         rc = l_ioctl(dev_id, OBD_IOC_PARAM, buf);
 197 out:
 198         if (rc) {
 199                 if (errno == ENOSYS)
 200                         fprintf(stderr, "Make sure cfg_device is set first.\n");
 201         }
 202         return rc;
 203 }
 204
 205 char *obdo_print(struct obdo *obd)
 206 {
 207         char buf[1024];
 208
 209         snprintf(buf, sizeof(buf), "id: %#jx\ngrp: %#jx\natime: %ju\n"
 210                  "mtime: %ju\nctime: %ju\nsize: %ju\nblocks: %ju"
 211                  "\nblksize: %u\nmode: %o\nuid: %d\ngid: %d\nflags: %x\n"
 212                  "misc: %x\nnlink: %d,\nvalid %#jx\n",
 213                  (uintmax_t)ostid_id(&obd->o_oi),
 214                  (uintmax_t)ostid_seq(&obd->o_oi),
 215                  (uintmax_t)obd->o_atime, (uintmax_t)obd->o_mtime,
 216                  (uintmax_t)obd->o_ctime, (uintmax_t)obd->o_size,
 217                  (uintmax_t)obd->o_blocks, obd->o_blksize, obd->o_mode,
 218                  obd->o_uid, obd->o_gid, obd->o_flags, obd->o_misc,
 219                  obd->o_nlink, (uintmax_t)obd->o_valid);
 220         return strdup(buf);
 221 }
 222
 223
 224 #define BAD_VERBOSE (-999999999)
 225
 226 #define N2D_OFF 0x100      /* So we can tell between error codes and devices */
 227
 228 static int do_name2dev(char *func, char *name)
 229 {
 230         struct obd_ioctl_data data;
 231         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
 232         int rc;
 233
 234         memset(&data, 0, sizeof(data));
 235         data.ioc_dev = cur_device;
 236         data.ioc_inllen1 = strlen(name) + 1;
 237         data.ioc_inlbuf1 = name;
 238
 239         memset(buf, 0, sizeof(rawbuf));
 240         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
 241         if (rc < 0) {
 242                 fprintf(stderr, "error: %s: invalid ioctl\n",
 243                         jt_cmdname(func));
 244                 return -rc;
 245         }
 246         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_NAME2DEV, buf);
 247         if (rc < 0)
 248                 return errno;
 249         rc = obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
 250         if (rc) {
 251                 fprintf(stderr, "error: %s: invalid reply\n",
 252                         jt_cmdname(func));
 253                 return rc;
 254         }
 255
 256         return data.ioc_dev + N2D_OFF;
 257 }
 258
 259 /*
 260  * resolve a device name to a device number.
 261  * supports a number, $name or %uuid.
 262  */
 263 int parse_devname(char *func, char *name)
 264 {
 265         int rc;
 266         int ret = -1;
 267
 268         if (!name)
 269                 return ret;
 270         if (isdigit(name[0])) {
 271                 ret = strtoul(name, NULL, 0);
 272         } else {
 273                 if (name[0] == '$' || name[0] == '%')
 274                         name++;
 275                 rc = do_name2dev(func, name);
 276                 if (rc >= N2D_OFF) {
 277                         ret = rc - N2D_OFF;
 278                         // printf("Name %s is device %d\n", name, ret);
 279                 } else {
 280                         fprintf(stderr, "No device found for name %s: %s\n",
 281                                 name, strerror(rc));
 282                 }
 283         }
 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                         fid_to_ostid(&fid, &data.ioc_obdo1.o_oi);
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                 data.ioc_obdo1.o_mode = mode;
1500                 ostid_set_id(&data.ioc_obdo1.o_oi, base_id);
1501                 data.ioc_obdo1.o_uid = 0;
1502                 data.ioc_obdo1.o_gid = 0;
1503                 data.ioc_obdo1.o_valid = OBD_MD_FLTYPE | OBD_MD_FLMODE |
1504                                          OBD_MD_FLID | OBD_MD_FLUID |
1505                                          OBD_MD_FLGID | OBD_MD_FLGROUP;
1506
1507                 memset(buf, 0, sizeof(rawbuf));
1508                 rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1509                 if (rc) {
1510                         fprintf(stderr, "error: %s: invalid ioctl\n",
1511                                 jt_cmdname(argv[0]));
1512                         return rc;
1513                 }
1514                 rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_CREATE, buf);
1515                 obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
1516                 shmem_bump(1);
1517                 if (rc < 0) {
1518                         fprintf(stderr, "error: %s: #%d - %s\n",
1519                                 jt_cmdname(argv[0]), i, strerror(rc = errno));
1520                         break;
1521                 }
1522                 if (!(data.ioc_obdo1.o_valid & OBD_MD_FLID)) {
1523                         fprintf(stderr, "error: %s: oid not valid #%d:%#jx\n",
1524                                 jt_cmdname(argv[0]), i,
1525                                 (uintmax_t)data.ioc_obdo1.o_valid);
1526                         rc = EINVAL;
1527                         break;
1528                 }
1529
1530                 if (be_verbose(verbose, &next_time, i, &next_count, count))
1531                         printf("%s: #%d is object id %#jx\n",
1532                                jt_cmdname(argv[0]), i,
1533                                (uintmax_t) ostid_id(&data.ioc_obdo1.o_oi));
1534         }
1535         return rc;
1536 }
1537
1538 int jt_obd_setattr(int argc, char **argv)
1539 {
1540         struct obd_ioctl_data data;
1541         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1542         char *end;
1543         int rc;
1544
1545         memset(&data, 0, sizeof(data));
1546         data.ioc_dev = cur_device;
1547         if (argc != 2)
1548                 return CMD_HELP;
1549
1550         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1551         ostid_set_id(&data.ioc_obdo1.o_oi, strtoull(argv[1], &end, 0));
1552         if (*end) {
1553                 fprintf(stderr, "error: %s: invalid objid '%s'\n",
1554                         jt_cmdname(argv[0]), argv[1]);
1555                 return CMD_HELP;
1556         }
1557         data.ioc_obdo1.o_mode = S_IFREG | strtoul(argv[2], &end, 0);
1558         if (*end) {
1559                 fprintf(stderr, "error: %s: invalid mode '%s'\n",
1560                         jt_cmdname(argv[0]), argv[2]);
1561                 return CMD_HELP;
1562         }
1563         data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLTYPE | OBD_MD_FLMODE;
1564
1565         memset(buf, 0, sizeof(rawbuf));
1566         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1567         if (rc) {
1568                 fprintf(stderr, "error: %s: invalid ioctl\n",
1569                         jt_cmdname(argv[0]));
1570                 return rc;
1571         }
1572         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_SETATTR, buf);
1573         if (rc < 0)
1574                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
1575                         strerror(rc = errno));
1576
1577         return rc;
1578 }
1579
1580 int jt_obd_test_setattr(int argc, char **argv)
1581 {
1582         struct obd_ioctl_data data;
1583         struct timeval start, next_time;
1584         __u64 i, count, next_count;
1585         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1586         int verbose = 1;
1587         __u64 objid = 3;
1588         char *end;
1589         int rc = 0;
1590
1591         if (argc < 2 || argc > 4)
1592                 return CMD_HELP;
1593
1594         memset(&data, 0, sizeof(data));
1595         data.ioc_dev = cur_device;
1596         count = strtoull(argv[1], &end, 0);
1597         if (*end) {
1598                 fprintf(stderr, "error: %s: invalid iteration count '%s'\n",
1599                         jt_cmdname(argv[0]), argv[1]);
1600                 return CMD_HELP;
1601         }
1602
1603         if (argc >= 3) {
1604                 verbose = get_verbose(argv[0], argv[2]);
1605                 if (verbose == BAD_VERBOSE)
1606                         return CMD_HELP;
1607         }
1608
1609         if (argc >= 4) {
1610                 if (argv[3][0] == 't') {
1611                         objid = strtoull(argv[3] + 1, &end, 0);
1612                         if (thread)
1613                                 objid += thread - 1;
1614                 } else
1615                         objid = strtoull(argv[3], &end, 0);
1616                 if (*end) {
1617                         fprintf(stderr, "error: %s: invalid objid '%s'\n",
1618                                 jt_cmdname(argv[0]), argv[3]);
1619                         return CMD_HELP;
1620                 }
1621         }
1622
1623         gettimeofday(&start, NULL);
1624         next_time.tv_sec = start.tv_sec - verbose;
1625         next_time.tv_usec = start.tv_usec;
1626         if (verbose != 0)
1627                 printf("%s: setting %jd attrs (objid %#jx): %s",
1628                        jt_cmdname(argv[0]), (uintmax_t)count,
1629                        (uintmax_t)objid, ctime(&start.tv_sec));
1630
1631         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1632         for (i = 1, next_count = verbose; i <= count && shmem_running(); i++) {
1633                 ostid_set_id(&data.ioc_obdo1.o_oi, objid);
1634                 data.ioc_obdo1.o_mode = S_IFREG;
1635                 data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLTYPE | OBD_MD_FLMODE;
1636                 memset(buf, 0, sizeof(rawbuf));
1637                 rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1638                 if (rc) {
1639                         fprintf(stderr, "error: %s: invalid ioctl\n",
1640                                 jt_cmdname(argv[0]));
1641                         return rc;
1642                 }
1643                 rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_SETATTR, &data);
1644                 shmem_bump(1);
1645                 if (rc < 0) {
1646                         fprintf(stderr, "error: %s: #%jd - %d:%s\n",
1647                                 jt_cmdname(argv[0]), (uintmax_t)i,
1648                                 errno, strerror(rc = errno));
1649                         break;
1650                 } else {
1651                         if (be_verbose
1652                             (verbose, &next_time, i, &next_count, count))
1653                                 printf("%s: set attr #%jd\n",
1654                                        jt_cmdname(argv[0]), (uintmax_t)i);
1655                 }
1656         }
1657
1658         if (!rc) {
1659                 struct timeval end;
1660                 double diff;
1661
1662                 gettimeofday(&end, NULL);
1663
1664                 diff = difftime(&end, &start);
1665
1666                 --i;
1667                 if (verbose != 0)
1668                         printf("%s: %jd attrs in %.3fs (%.3f attr/s): %s",
1669                                jt_cmdname(argv[0]), (uintmax_t)i, diff,
1670                                i / diff, ctime(&end.tv_sec));
1671         }
1672         return rc;
1673 }
1674
1675 int jt_obd_destroy(int argc, char **argv)
1676 {
1677         struct obd_ioctl_data data;
1678         struct timeval next_time;
1679         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1680         __u64 count = 1, next_count;
1681         int verbose = 1;
1682         __u64 id;
1683         char *end;
1684         int rc = 0, i;
1685
1686         memset(&data, 0, sizeof(data));
1687         data.ioc_dev = cur_device;
1688         if (argc < 2 || argc > 4)
1689                 return CMD_HELP;
1690
1691         id = strtoull(argv[1], &end, 0);
1692         if (*end || id == 0 || errno != 0) {
1693                 fprintf(stderr, "error: %s: invalid objid '%s'\n",
1694                         jt_cmdname(argv[0]), argv[1]);
1695                 return CMD_HELP;
1696         }
1697         if (argc > 2) {
1698                 count = strtoull(argv[2], &end, 0);
1699                 if (*end) {
1700                         fprintf(stderr,
1701                                 "error: %s: invalid iteration count '%s'\n",
1702                                 jt_cmdname(argv[0]), argv[2]);
1703                         return CMD_HELP;
1704                 }
1705         }
1706
1707         if (argc > 3) {
1708                 verbose = get_verbose(argv[0], argv[3]);
1709                 if (verbose == BAD_VERBOSE)
1710                         return CMD_HELP;
1711         }
1712
1713         printf("%s: %jd objects\n", jt_cmdname(argv[0]), (uintmax_t)count);
1714         gettimeofday(&next_time, NULL);
1715         next_time.tv_sec -= verbose;
1716
1717         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1718         for (i = 1, next_count = verbose; i <= count && shmem_running(); i++, id++) {
1719                 ostid_set_id(&data.ioc_obdo1.o_oi, id);
1720                 data.ioc_obdo1.o_mode = S_IFREG | 0644;
1721                 data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLMODE;
1722
1723                 memset(buf, 0, sizeof(rawbuf));
1724                 rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1725                 if (rc) {
1726                         fprintf(stderr, "error: %s: invalid ioctl\n",
1727                                 jt_cmdname(argv[0]));
1728                         return rc;
1729                 }
1730                 rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_DESTROY, buf);
1731                 obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
1732                 shmem_bump(1);
1733                 if (rc < 0) {
1734                         fprintf(stderr, "error: %s: objid %#jx: %s\n",
1735                                 jt_cmdname(argv[0]), (uintmax_t)id,
1736                                 strerror(rc = errno));
1737                         break;
1738                 }
1739
1740                 if (be_verbose(verbose, &next_time, i, &next_count, count))
1741                         printf("%s: #%d is object id %#jx\n",
1742                                jt_cmdname(argv[0]), i, (uintmax_t)id);
1743         }
1744
1745         return rc;
1746 }
1747
1748 int jt_obd_getattr(int argc, char **argv)
1749 {
1750         struct obd_ioctl_data data;
1751         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1752         char *end;
1753         int rc;
1754
1755         if (argc != 2)
1756                 return CMD_HELP;
1757
1758         memset(&data, 0, sizeof(data));
1759         data.ioc_dev = cur_device;
1760         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1761         ostid_set_id(&data.ioc_obdo1.o_oi, strtoull(argv[1], &end, 0));
1762         if (*end) {
1763                 fprintf(stderr, "error: %s: invalid objid '%s'\n",
1764                         jt_cmdname(argv[0]), argv[1]);
1765                 return CMD_HELP;
1766         }
1767         /* to help obd filter */
1768         data.ioc_obdo1.o_mode = 0100644;
1769         data.ioc_obdo1.o_valid = 0xffffffff;
1770         printf("%s: object id %#jx\n", jt_cmdname(argv[0]),
1771                (uintmax_t)ostid_id(&data.ioc_obdo1.o_oi));
1772
1773         memset(buf, 0, sizeof(rawbuf));
1774         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1775         if (rc) {
1776                 fprintf(stderr, "error: %s: invalid ioctl\n",
1777                         jt_cmdname(argv[0]));
1778                 return rc;
1779         }
1780         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_GETATTR, buf);
1781         obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
1782         if (rc) {
1783                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
1784                         strerror(rc = errno));
1785         } else {
1786                 printf("%s: object id %ju, mode %o\n", jt_cmdname(argv[0]),
1787                        (uintmax_t)ostid_id(&data.ioc_obdo1.o_oi),
1788                        data.ioc_obdo1.o_mode);
1789         }
1790         return rc;
1791 }
1792
1793 int jt_obd_test_getattr(int argc, char **argv)
1794 {
1795         struct obd_ioctl_data data;
1796         struct timeval start, next_time;
1797         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1798         __u64 i, count, next_count;
1799         int verbose = 1;
1800         __u64 objid = 3;
1801         char *end;
1802         int rc = 0;
1803
1804         if (argc < 2 || argc > 4)
1805                 return CMD_HELP;
1806
1807         memset(&data, 0, sizeof(data));
1808         data.ioc_dev = cur_device;
1809         count = strtoull(argv[1], &end, 0);
1810         if (*end) {
1811                 fprintf(stderr, "error: %s: invalid iteration count '%s'\n",
1812                         jt_cmdname(argv[0]), argv[1]);
1813                 return CMD_HELP;
1814         }
1815
1816         if (argc >= 3) {
1817                 verbose = get_verbose(argv[0], argv[2]);
1818                 if (verbose == BAD_VERBOSE)
1819                         return CMD_HELP;
1820         }
1821
1822         if (argc >= 4) {
1823                 if (argv[3][0] == 't') {
1824                         objid = strtoull(argv[3] + 1, &end, 0);
1825                         if (thread)
1826                                 objid += thread - 1;
1827                 } else
1828                         objid = strtoull(argv[3], &end, 0);
1829                 if (*end) {
1830                         fprintf(stderr, "error: %s: invalid objid '%s'\n",
1831                                 jt_cmdname(argv[0]), argv[3]);
1832                         return CMD_HELP;
1833                 }
1834         }
1835
1836         gettimeofday(&start, NULL);
1837         next_time.tv_sec = start.tv_sec - verbose;
1838         next_time.tv_usec = start.tv_usec;
1839         if (verbose != 0)
1840                 printf("%s: getting %jd attrs (objid %#jx): %s",
1841                        jt_cmdname(argv[0]), (uintmax_t) count,
1842                        (uintmax_t)objid, ctime(&start.tv_sec));
1843
1844         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
1845         for (i = 1, next_count = verbose; i <= count && shmem_running(); i++) {
1846                 ostid_set_id(&data.ioc_obdo1.o_oi, objid);
1847                 data.ioc_obdo1.o_mode = S_IFREG;
1848                 data.ioc_obdo1.o_valid = 0xffffffff;
1849                 memset(buf, 0, sizeof(rawbuf));
1850                 rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
1851                 if (rc) {
1852                         fprintf(stderr, "error: %s: invalid ioctl\n",
1853                                 jt_cmdname(argv[0]));
1854                         return rc;
1855                 }
1856                 rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_GETATTR, &data);
1857                 shmem_bump(1);
1858                 if (rc < 0) {
1859                         fprintf(stderr, "error: %s: #%jd - %d:%s\n",
1860                                 jt_cmdname(argv[0]), (uintmax_t)i,
1861                                 errno, strerror(rc = errno));
1862                         break;
1863                 } else {
1864                         if (be_verbose
1865                             (verbose, &next_time, i, &next_count, count))
1866                                 printf("%s: got attr #%jd\n",
1867                                        jt_cmdname(argv[0]), (uintmax_t)i);
1868                 }
1869         }
1870
1871         if (!rc) {
1872                 struct timeval end;
1873                 double diff;
1874
1875                 gettimeofday(&end, NULL);
1876
1877                 diff = difftime(&end, &start);
1878
1879                 --i;
1880                 if (verbose != 0)
1881                         printf("%s: %jd attrs in %.3fs (%.3f attr/s): %s",
1882                                jt_cmdname(argv[0]), (uintmax_t) i, diff,
1883                                i / diff, ctime(&end.tv_sec));
1884         }
1885         return rc;
1886 }
1887
1888 /* test_brw <cnt>                                               count
1889         <r|w[r(repeat)x(noverify)]>                             mode
1890         <q|v|#(print interval)>                                 verbosity
1891         <npages[+offset]>                                       blocksize
1892         <[[<interleave_threads>]t(inc obj by thread#)]obj>      object
1893         [p|g<args>]                                             batch */
1894 int jt_obd_test_brw(int argc, char **argv)
1895 {
1896         struct obd_ioctl_data data;
1897         struct timeval start, next_time;
1898         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
1899         __u64 count, next_count, len, stride, thr_offset = 0, objid = 3;
1900         int write = 0, verbose = 1, cmd, i, rc = 0, pages = 1;
1901         int offset_pages = 0;
1902         long n;
1903         int repeat_offset = 0;
1904         unsigned long long ull;
1905         int  nthr_per_obj = 0;
1906         int  verify = 1;
1907         int  obj_idx = 0;
1908         char *end;
1909
1910         if (argc < 2 || argc > 7) {
1911                 fprintf(stderr, "error: %s: bad number of arguments: %d\n",
1912                         jt_cmdname(argv[0]), argc);
1913                 return CMD_HELP;
1914         }
1915
1916         count = strtoull(argv[1], &end, 0);
1917         if (*end) {
1918                 fprintf(stderr, "error: %s: bad iteration count '%s'\n",
1919                         jt_cmdname(argv[0]), argv[1]);
1920                 return CMD_HELP;
1921         }
1922
1923         if (argc >= 3) {
1924                 if (argv[2][0] == 'w' || argv[2][0] == '1')
1925                         write = 1;
1926                 /* else it's a read */
1927
1928                 if (argv[2][0] != 0)
1929                         for (i = 1; argv[2][i] != 0; i++)
1930                                 switch (argv[2][i]) {
1931                                 case 'r':
1932                                         repeat_offset = 1;
1933                                         break;
1934
1935                                 case 'x':
1936                                         verify = 0;
1937                                         break;
1938
1939                                 default:
1940                                         fprintf (stderr, "Can't parse cmd '%s'\n",
1941                                                  argv[2]);
1942                                         return CMD_HELP;
1943                                 }
1944         }
1945
1946         if (argc >= 4) {
1947                 verbose = get_verbose(argv[0], argv[3]);
1948                 if (verbose == BAD_VERBOSE)
1949                         return CMD_HELP;
1950         }
1951
1952         if (argc >= 5) {
1953                 pages = strtoul(argv[4], &end, 0);
1954
1955                 if (*end == '+')
1956                         offset_pages = strtoul(end + 1, &end, 0);
1957
1958                 if (*end != 0 ||
1959                     offset_pages < 0 || offset_pages >= pages) {
1960                         fprintf(stderr, "error: %s: bad npages[+offset] parameter '%s'\n",
1961                                 jt_cmdname(argv[0]), argv[4]);
1962                         return CMD_HELP;
1963                 }
1964         }
1965
1966         if (argc >= 6) {
1967                 if (thread &&
1968                     (n = strtol(argv[5], &end, 0)) > 0 &&
1969                     *end == 't' &&
1970                     (ull = strtoull(end + 1, &end, 0)) > 0 &&
1971                     *end == 0) {
1972                         nthr_per_obj = n;
1973                         objid = ull;
1974                 } else if (thread &&
1975                            argv[5][0] == 't') {
1976                         nthr_per_obj = 1;
1977                         objid = strtoull(argv[5] + 1, &end, 0);
1978                 } else {
1979                         nthr_per_obj = 0;
1980                         objid = strtoull(argv[5], &end, 0);
1981                 }
1982                 if (*end) {
1983                         fprintf(stderr, "error: %s: bad objid '%s'\n",
1984                                 jt_cmdname(argv[0]), argv[5]);
1985                         return CMD_HELP;
1986                 }
1987         }
1988
1989         memset(&data, 0, sizeof(data));
1990         data.ioc_dev = cur_device;
1991
1992         /* communicate the 'type' of brw test and batching to echo_client.
1993          * don't start.  we'd love to refactor this lctl->echo_client
1994          * interface */
1995         data.ioc_pbuf1 = (void *)1;
1996         data.ioc_plen1 = 1;
1997
1998         if (argc >= 7) {
1999                 switch(argv[6][0]) {
2000                         case 'g': /* plug and unplug */
2001                                 data.ioc_pbuf1 = (void *)2;
2002                                 data.ioc_plen1 = strtoull(argv[6] + 1, &end,
2003                                                           0);
2004                                 break;
2005                         case 'p': /* prep and commit */
2006                                 data.ioc_pbuf1 = (void *)3;
2007                                 data.ioc_plen1 = strtoull(argv[6] + 1, &end,
2008                                                           0);
2009                                 break;
2010                         default:
2011                                 fprintf(stderr, "error: %s: batching '%s' "
2012                                         "needs to specify 'p' or 'g'\n",
2013                                         jt_cmdname(argv[0]), argv[6]);
2014                                 return CMD_HELP;
2015                 }
2016
2017                 if (*end) {
2018                         fprintf(stderr, "error: %s: bad batching '%s'\n",
2019                                 jt_cmdname(argv[0]), argv[6]);
2020                         return CMD_HELP;
2021                 }
2022                 data.ioc_plen1 *= getpagesize();
2023         }
2024
2025         len = pages * getpagesize();
2026         thr_offset = offset_pages * getpagesize();
2027         stride = len;
2028
2029 #ifdef MAX_THREADS
2030         if (thread) {
2031                 shmem_lock ();
2032                 if (nthr_per_obj != 0) {
2033                         /* threads interleave */
2034                         obj_idx = (thread - 1)/nthr_per_obj;
2035                         objid += obj_idx;
2036                         stride *= nthr_per_obj;
2037                         if ((thread - 1) % nthr_per_obj == 0) {
2038                                 shared_data->body.offsets[obj_idx] =
2039                                         stride + thr_offset;
2040                         }
2041                         thr_offset += ((thread - 1) % nthr_per_obj) * len;
2042                 } else {
2043                         /* threads disjoint */
2044                         thr_offset += (thread - 1) * len;
2045                 }
2046
2047                 shmem_start_time_locked();
2048                 shmem_unlock ();
2049         }
2050 #endif
2051
2052         ostid_set_seq_echo(&data.ioc_obdo1.o_oi);
2053         ostid_set_id(&data.ioc_obdo1.o_oi, objid);
2054         data.ioc_obdo1.o_mode = S_IFREG;
2055         data.ioc_obdo1.o_valid = OBD_MD_FLID | OBD_MD_FLTYPE | OBD_MD_FLMODE |
2056                                  OBD_MD_FLFLAGS | OBD_MD_FLGROUP;
2057         data.ioc_obdo1.o_flags = (verify ? OBD_FL_DEBUG_CHECK : 0);
2058         data.ioc_count = len;
2059         data.ioc_offset = (repeat_offset ? 0 : thr_offset);
2060
2061         gettimeofday(&start, NULL);
2062         next_time.tv_sec = start.tv_sec - verbose;
2063         next_time.tv_usec = start.tv_usec;
2064
2065         if (verbose != 0)
2066                 printf("%s: %s %jux%d pages (obj %#jx, off %ju): %s",
2067                        jt_cmdname(argv[0]), write ? "writing" : "reading",
2068                        (uintmax_t)count, pages, (uintmax_t) objid,
2069                        (uintmax_t)data.ioc_offset, ctime(&start.tv_sec));
2070
2071         cmd = write ? OBD_IOC_BRW_WRITE : OBD_IOC_BRW_READ;
2072         for (i = 1, next_count = verbose; i <= count && shmem_running(); i++) {
2073                 data.ioc_obdo1.o_valid &= ~(OBD_MD_FLBLOCKS|OBD_MD_FLGRANT);
2074                 memset(buf, 0, sizeof(rawbuf));
2075                 rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2076                 if (rc) {
2077                         fprintf(stderr, "error: %s: invalid ioctl\n",
2078                                 jt_cmdname(argv[0]));
2079                         return rc;
2080                 }
2081                 rc = l2_ioctl(OBD_DEV_ID, cmd, buf);
2082                 shmem_bump(1);
2083                 if (rc) {
2084                         fprintf(stderr, "error: %s: #%d - %s on %s\n",
2085                                 jt_cmdname(argv[0]), i, strerror(rc = errno),
2086                                 write ? "write" : "read");
2087                         break;
2088                 } else if (be_verbose(verbose, &next_time,i, &next_count,count)) {
2089                         shmem_lock ();
2090                         printf("%s: %s number %d @ %jd:%ju for %d\n",
2091                                jt_cmdname(argv[0]), write ? "write" : "read", i,
2092                                (uintmax_t)ostid_id(&data.ioc_obdo1.o_oi),
2093                                (uintmax_t)data.ioc_offset,
2094                                (int)(pages * getpagesize()));
2095                         shmem_unlock ();
2096                 }
2097
2098                 if (!repeat_offset) {
2099 #ifdef MAX_THREADS
2100                         if (stride == len) {
2101                                 data.ioc_offset += stride;
2102                         } else if (i < count) {
2103                                 shmem_lock ();
2104                                 data.ioc_offset =
2105                                         shared_data->body.offsets[obj_idx];
2106                                 shared_data->body.offsets[obj_idx] += len;
2107                                 shmem_unlock ();
2108                         }
2109 #else
2110                         data.ioc_offset += len;
2111                         obj_idx = 0; /* avoids an unused var warning */
2112 #endif
2113                 }
2114         }
2115
2116         if (!rc) {
2117                 struct timeval end;
2118                 double diff;
2119
2120                 gettimeofday(&end, NULL);
2121
2122                 diff = difftime(&end, &start);
2123
2124                 --i;
2125                 if (verbose != 0)
2126                         printf("%s: %s %dx%d pages in %.3fs (%.3f MB/s): %s",
2127                                jt_cmdname(argv[0]), write ? "wrote" : "read",
2128                                i, pages, diff,
2129                                ((double)i * pages * getpagesize()) /
2130                                (diff * 1048576.0),
2131                                ctime(&end.tv_sec));
2132         }
2133
2134 #ifdef MAX_THREADS
2135         if (thread) {
2136                 shmem_lock();
2137                 shmem_end_time_locked();
2138                 shmem_unlock();
2139         }
2140 #endif
2141         return rc;
2142 }
2143
2144 int jt_obd_lov_getconfig(int argc, char **argv)
2145 {
2146         struct obd_ioctl_data data;
2147         struct lov_desc desc;
2148         struct obd_uuid *uuidarray;
2149         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2150         __u32 *obdgens;
2151         char *path;
2152         int rc, fd;
2153
2154         memset(&data, 0, sizeof(data));
2155         data.ioc_dev = cur_device;
2156
2157         if (argc != 2)
2158                 return CMD_HELP;
2159
2160         path = argv[1];
2161         fd = open(path, O_RDONLY);
2162         if (fd < 0) {
2163                 fprintf(stderr, "open \"%s\" failed: %s\n", path,
2164                         strerror(errno));
2165                 return -errno;
2166         }
2167
2168         memset(&desc, 0, sizeof(desc));
2169         obd_str2uuid(&desc.ld_uuid, argv[1]);
2170         desc.ld_tgt_count = ((OBD_MAX_IOCTL_BUFFER-sizeof(data)-sizeof(desc)) /
2171                              (sizeof(*uuidarray) + sizeof(*obdgens)));
2172
2173 repeat:
2174         uuidarray = calloc(desc.ld_tgt_count, sizeof(*uuidarray));
2175         if (!uuidarray) {
2176                 fprintf(stderr, "error: %s: no memory for %d uuid's\n",
2177                         jt_cmdname(argv[0]), desc.ld_tgt_count);
2178                 rc = -ENOMEM;
2179                 goto out;
2180         }
2181         obdgens = calloc(desc.ld_tgt_count, sizeof(*obdgens));
2182         if (!obdgens) {
2183                 fprintf(stderr, "error: %s: no memory for %d generation #'s\n",
2184                         jt_cmdname(argv[0]), desc.ld_tgt_count);
2185                 rc = -ENOMEM;
2186                 goto out_uuidarray;
2187         }
2188
2189         memset(buf, 0, sizeof(rawbuf));
2190         data.ioc_inllen1 = sizeof(desc);
2191         data.ioc_inlbuf1 = (char *)&desc;
2192         data.ioc_inllen2 = desc.ld_tgt_count * sizeof(*uuidarray);
2193         data.ioc_inlbuf2 = (char *)uuidarray;
2194         data.ioc_inllen3 = desc.ld_tgt_count * sizeof(*obdgens);
2195         data.ioc_inlbuf3 = (char *)obdgens;
2196
2197         if (obd_ioctl_pack(&data, &buf, sizeof(rawbuf))) {
2198                 fprintf(stderr, "error: %s: invalid ioctl\n",
2199                         jt_cmdname(argv[0]));
2200                 rc = -EINVAL;
2201                 goto out_obdgens;
2202         }
2203         rc = ioctl(fd, OBD_IOC_LOV_GET_CONFIG, buf);
2204         if (rc == -ENOSPC) {
2205                 free(uuidarray);
2206                 free(obdgens);
2207                 goto repeat;
2208         } else if (rc) {
2209                 fprintf(stderr, "error: %s: ioctl error: %s\n",
2210                         jt_cmdname(argv[0]), strerror(rc = errno));
2211         } else {
2212                 struct obd_uuid *uuidp;
2213                 __u32 *genp;
2214                 int i;
2215
2216                 if (obd_ioctl_unpack(&data, buf, sizeof(rawbuf))) {
2217                         fprintf(stderr, "error: %s: invalid reply\n",
2218                                 jt_cmdname(argv[0]));
2219                         rc = -EINVAL;
2220                         goto out;
2221                 }
2222                 if (desc.ld_default_stripe_count == (__u32)-1)
2223                         printf("default_stripe_count: %d\n", -1);
2224                 else
2225                         printf("default_stripe_count: %u\n",
2226                                desc.ld_default_stripe_count);
2227                 printf("default_stripe_size: %ju\n",
2228                        (uintmax_t)desc.ld_default_stripe_size);
2229                 printf("default_stripe_offset: %ju\n",
2230                        (uintmax_t)desc.ld_default_stripe_offset);
2231                 printf("default_stripe_pattern: %u\n", desc.ld_pattern);
2232                 printf("obd_count: %u\n", desc.ld_tgt_count);
2233                 printf("OBDS:\tobdidx\t\tobdgen\t\t obduuid\n");
2234                 uuidp = uuidarray;
2235                 genp = obdgens;
2236                 for (i = 0; i < desc.ld_tgt_count; i++, uuidp++, genp++)
2237                         printf("\t%6u\t%14u\t\t %s\n", i, *genp, (char *)uuidp);
2238         }
2239 out_obdgens:
2240         free(obdgens);
2241 out_uuidarray:
2242         free(uuidarray);
2243 out:
2244         close(fd);
2245         return rc;
2246 }
2247
2248 static int do_activate(int argc, char **argv, int flag)
2249 {
2250         struct obd_ioctl_data data;
2251         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2252         int rc;
2253
2254         memset(&data, 0, sizeof(data));
2255         data.ioc_dev = cur_device;
2256         if (argc != 1)
2257                 return CMD_HELP;
2258
2259         /* reuse offset for 'active' */
2260         data.ioc_offset = flag;
2261
2262         memset(buf, 0, sizeof(rawbuf));
2263         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2264         if (rc) {
2265                 fprintf(stderr, "error: %s: invalid ioctl\n",
2266                         jt_cmdname(argv[0]));
2267                 return rc;
2268         }
2269         rc = l2_ioctl(OBD_DEV_ID, IOC_OSC_SET_ACTIVE, buf);
2270         if (rc)
2271                 fprintf(stderr, "error: %s: failed: %s\n",
2272                         jt_cmdname(argv[0]), strerror(rc = errno));
2273
2274         return rc;
2275 }
2276
2277 /**
2278  * Replace nids for given device.
2279  * lctl replace_nids <devicename> <nid1>[,nid2,nid3]
2280  * Command should be started on MGS server.
2281  * Only MGS server should be started (command execution
2282  * returns error in another cases). Command mount
2283  * -t lustre <MDT partition> -o nosvc <mount point>
2284  * can be used for that.
2285  *
2286  * llogs for MDTs and clients are processed. All
2287  * records copied as is except add_uuid and setup. This records
2288  * are skipped and recorded with new nids and uuid.
2289  *
2290  * \see mgs_replace_nids
2291  * \see mgs_replace_nids_log
2292  * \see mgs_replace_handler
2293  */
2294 int jt_replace_nids(int argc, char **argv)
2295 {
2296         int rc;
2297         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2298         struct obd_ioctl_data data;
2299
2300         memset(&data, 0, sizeof(data));
2301         data.ioc_dev = get_mgs_device();
2302         if (argc != 3)
2303                 return CMD_HELP;
2304
2305         data.ioc_inllen1 = strlen(argv[1]) + 1;
2306         data.ioc_inlbuf1 = argv[1];
2307
2308         data.ioc_inllen2 = strlen(argv[2]) + 1;
2309         data.ioc_inlbuf2 = argv[2];
2310         memset(buf, 0, sizeof(rawbuf));
2311         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2312         if (rc) {
2313                 fprintf(stderr, "error: %s: invalid ioctl\n",
2314                         jt_cmdname(argv[0]));
2315                 return rc;
2316         }
2317
2318         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_REPLACE_NIDS, buf);
2319         if (rc < 0) {
2320                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
2321                         strerror(rc = errno));
2322         }
2323
2324         return rc;
2325 }
2326
2327 int jt_obd_deactivate(int argc, char **argv)
2328 {
2329         return do_activate(argc, argv, 0);
2330 }
2331
2332 int jt_obd_activate(int argc, char **argv)
2333 {
2334         return do_activate(argc, argv, 1);
2335 }
2336
2337 int jt_obd_recover(int argc, char **argv)
2338 {
2339         int rc;
2340         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2341         struct obd_ioctl_data data;
2342
2343         memset(&data, 0, sizeof(data));
2344         data.ioc_dev = cur_device;
2345         if (argc > 2)
2346                 return CMD_HELP;
2347
2348         if (argc == 2) {
2349                 data.ioc_inllen1 = strlen(argv[1]) + 1;
2350                 data.ioc_inlbuf1 = argv[1];
2351         }
2352
2353         memset(buf, 0, sizeof(rawbuf));
2354         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2355         if (rc) {
2356                 fprintf(stderr, "error: %s: invalid ioctl\n",
2357                         jt_cmdname(argv[0]));
2358                 return rc;
2359         }
2360         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_CLIENT_RECOVER, buf);
2361         if (rc < 0) {
2362                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
2363                         strerror(rc = errno));
2364         }
2365
2366         return rc;
2367 }
2368
2369 int jt_obd_mdc_lookup(int argc, char **argv)
2370 {
2371         struct obd_ioctl_data data;
2372         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2373         char *parent, *child;
2374         int rc, fd, verbose = 1;
2375
2376         if (argc < 3 || argc > 4)
2377                 return CMD_HELP;
2378
2379         parent = argv[1];
2380         child = argv[2];
2381         if (argc == 4)
2382                 verbose = get_verbose(argv[0], argv[3]);
2383
2384         memset(&data, 0, sizeof(data));
2385         data.ioc_dev = cur_device;
2386
2387         data.ioc_inllen1 = strlen(child) + 1;
2388         data.ioc_inlbuf1 = child;
2389
2390         memset(buf, 0, sizeof(rawbuf));
2391         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2392         if (rc) {
2393                 fprintf(stderr, "error: %s: invalid ioctl\n",
2394                         jt_cmdname(argv[0]));
2395                 return rc;
2396         }
2397
2398         fd = open(parent, O_RDONLY);
2399         if (fd < 0) {
2400                 fprintf(stderr, "open \"%s\" failed: %s\n", parent,
2401                         strerror(errno));
2402                 return -1;
2403         }
2404
2405         rc = ioctl(fd, IOC_MDC_LOOKUP, buf);
2406         if (rc < 0) {
2407                 fprintf(stderr, "error: %s: ioctl error: %s\n",
2408                         jt_cmdname(argv[0]), strerror(rc = errno));
2409         }
2410         close(fd);
2411
2412         if (verbose) {
2413                 rc = obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
2414                 if (rc) {
2415                         fprintf(stderr, "error: %s: invalid reply\n",
2416                                 jt_cmdname(argv[0]));
2417                         return rc;
2418                 }
2419                 printf("%s: mode %o uid %d gid %d\n", child,
2420                        data.ioc_obdo1.o_mode, data.ioc_obdo1.o_uid,
2421                        data.ioc_obdo1.o_gid);
2422         }
2423
2424         return rc;
2425 }
2426
2427 int jt_llog_catlist(int argc, char **argv)
2428 {
2429         struct obd_ioctl_data data;
2430         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2431         int rc;
2432
2433         if (argc != 1)
2434                 return CMD_HELP;
2435
2436         memset(&data, 0, sizeof(data));
2437         data.ioc_dev = cur_device;
2438         data.ioc_inllen1 = sizeof(rawbuf) - cfs_size_round(sizeof(data));
2439         memset(buf, 0, sizeof(rawbuf));
2440         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2441         if (rc) {
2442                 fprintf(stderr, "error: %s: invalid ioctl\n",
2443                         jt_cmdname(argv[0]));
2444                 return rc;
2445         }
2446         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_CATLOGLIST, buf);
2447         if (rc == 0)
2448                 fprintf(stdout, "%s", ((struct obd_ioctl_data*)buf)->ioc_bulk);
2449         else
2450                 fprintf(stderr, "OBD_IOC_CATLOGLIST failed: %s\n",
2451                         strerror(errno));
2452
2453         return rc;
2454 }
2455
2456 int jt_llog_info(int argc, char **argv)
2457 {
2458         struct obd_ioctl_data data;
2459         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2460         int rc;
2461
2462         if (argc != 2)
2463                 return CMD_HELP;
2464
2465         memset(&data, 0, sizeof(data));
2466         data.ioc_dev = cur_device;
2467         data.ioc_inllen1 = strlen(argv[1]) + 1;
2468         data.ioc_inlbuf1 = argv[1];
2469         data.ioc_inllen2 = sizeof(rawbuf) - cfs_size_round(sizeof(data)) -
2470                 cfs_size_round(data.ioc_inllen1);
2471         memset(buf, 0, sizeof(rawbuf));
2472         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2473         if (rc) {
2474                 fprintf(stderr, "error: %s: invalid ioctl\n",
2475                         jt_cmdname(argv[0]));
2476                 return rc;
2477         }
2478
2479         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_INFO, buf);
2480         if (rc == 0)
2481                 fprintf(stdout, "%s", ((struct obd_ioctl_data*)buf)->ioc_bulk);
2482         else
2483                 fprintf(stderr, "OBD_IOC_LLOG_INFO failed: %s\n",
2484                         strerror(errno));
2485
2486         return rc;
2487 }
2488
2489 int jt_llog_print(int argc, char **argv)
2490 {
2491         struct obd_ioctl_data data;
2492         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2493         int rc;
2494
2495         if (argc != 2 && argc != 4)
2496                 return CMD_HELP;
2497
2498         memset(&data, 0, sizeof(data));
2499         data.ioc_dev = cur_device;
2500         data.ioc_inllen1 = strlen(argv[1]) + 1;
2501         data.ioc_inlbuf1 = argv[1];
2502         if (argc == 4) {
2503                 data.ioc_inllen2 = strlen(argv[2]) + 1;
2504                 data.ioc_inlbuf2 = argv[2];
2505                 data.ioc_inllen3 = strlen(argv[3]) + 1;
2506                 data.ioc_inlbuf3 = argv[3];
2507         } else {
2508                 char from[2] = "1", to[3] = "-1";
2509                 data.ioc_inllen2 = strlen(from) + 1;
2510                 data.ioc_inlbuf2 = from;
2511                 data.ioc_inllen3 = strlen(to) + 1;
2512                 data.ioc_inlbuf3 = to;
2513         }
2514         data.ioc_inllen4 = sizeof(rawbuf) - cfs_size_round(sizeof(data)) -
2515                 cfs_size_round(data.ioc_inllen1) -
2516                 cfs_size_round(data.ioc_inllen2) -
2517                 cfs_size_round(data.ioc_inllen3);
2518         memset(buf, 0, sizeof(rawbuf));
2519         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2520         if (rc) {
2521                 fprintf(stderr, "error: %s: invalid ioctl\n",
2522                         jt_cmdname(argv[0]));
2523                 return rc;
2524         }
2525
2526         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_PRINT, buf);
2527         if (rc == 0)
2528                 fprintf(stdout, "%s", ((struct obd_ioctl_data*)buf)->ioc_bulk);
2529         else
2530                 fprintf(stderr, "OBD_IOC_LLOG_PRINT failed: %s\n",
2531                         strerror(errno));
2532
2533         return rc;
2534 }
2535
2536 static int llog_cancel_parse_optional(int argc, char **argv,
2537                                       struct obd_ioctl_data *data)
2538 {
2539         int cOpt;
2540         const char *const short_options = "c:l:i:h";
2541         const struct option long_options[] = {
2542                 {"catalog", required_argument, NULL, 'c'},
2543                 {"log_id", required_argument, NULL, 'l'},
2544                 {"log_idx", required_argument, NULL, 'i'},
2545                 {"help", no_argument, NULL, 'h'},
2546                 {NULL, 0, NULL, 0}
2547         };
2548
2549         /* sanity check */
2550         if (!data || argc <= 1) {
2551                 return -1;
2552         }
2553
2554         /*now process command line arguments*/
2555         while ((cOpt = getopt_long(argc, argv, short_options,
2556                                         long_options, NULL)) != -1) {
2557                 switch (cOpt) {
2558                 case 'c':
2559                         data->ioc_inllen1 = strlen(optarg) + 1;
2560                         data->ioc_inlbuf1 = optarg;
2561                         break;
2562
2563                 case 'l':
2564                         data->ioc_inllen2 = strlen(optarg) + 1;
2565                         data->ioc_inlbuf2 = optarg;
2566                         break;
2567
2568                 case 'i':
2569                         data->ioc_inllen3 = strlen(optarg) + 1;
2570                         data->ioc_inlbuf3 = optarg;
2571                         break;
2572
2573                 case 'h':
2574                 default:
2575                         return -1;
2576                 }
2577         }
2578
2579         if ((data->ioc_inlbuf1 == NULL) || (data->ioc_inlbuf3 == NULL)) {
2580                 /* missing mandatory parameters */
2581                 return -1;
2582         }
2583
2584         return 0;
2585 }
2586
2587 int jt_llog_cancel(int argc, char **argv)
2588 {
2589         struct obd_ioctl_data data;
2590         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2591         int rc, i;
2592
2593         /* check that the arguments provided are either all
2594          * optional or all positional.  No mixing allowed
2595          *
2596          * if argc is 4 or 3 then check all arguments to ensure that none
2597          * of them start with a '-'.  If so then this is invalid.
2598          * Otherwise if arg is > 4 then assume that this is optional
2599          * arguments, and parse as such ignoring any thing that's not
2600          * optional.  The result is that user must use optional arguments
2601          * for all mandatory parameters.  Code will ignore all other args
2602          *
2603          * The positional arguments option should eventually be phased out.
2604          */
2605         memset(&data, 0, sizeof(data));
2606         data.ioc_dev = cur_device;
2607
2608         if (argc == 3 || argc == 4) {
2609                 for (i = 1; i < argc; i++) {
2610                         if (argv[i][0] == '-')
2611                                 return CMD_HELP;
2612                 }
2613                 data.ioc_inllen1 = strlen(argv[1]) + 1;
2614                 data.ioc_inlbuf1 = argv[1];
2615                 if (argc == 4) {
2616                         data.ioc_inllen2 = strlen(argv[2]) + 1;
2617                         data.ioc_inlbuf2 = argv[2];
2618                         data.ioc_inllen3 = strlen(argv[3]) + 1;
2619                         data.ioc_inlbuf3 = argv[3];
2620                 } else {
2621                         data.ioc_inllen3 = strlen(argv[2]) + 1;
2622                         data.ioc_inlbuf3 = argv[2];
2623                 }
2624         } else {
2625                 if (llog_cancel_parse_optional(argc, argv, &data) != 0)
2626                         return CMD_HELP;
2627         }
2628
2629         memset(buf, 0, sizeof(rawbuf));
2630         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2631         if (rc) {
2632                 fprintf(stderr, "error: %s: invalid ioctl\n",
2633                         jt_cmdname(argv[0]));
2634                 return rc;
2635         }
2636
2637         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_CANCEL, buf);
2638         if (rc == 0)
2639                 fprintf(stdout, "index %s was canceled.\n",
2640                         argc == 4 ? argv[3] : argv[2]);
2641         else
2642                 fprintf(stderr, "OBD_IOC_LLOG_CANCEL failed: %s\n",
2643                         strerror(errno));
2644
2645         return rc;
2646 }
2647
2648 int jt_llog_check(int argc, char **argv)
2649 {
2650         struct obd_ioctl_data data;
2651         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2652         int rc;
2653
2654         if (argc != 2 && argc != 4)
2655                 return CMD_HELP;
2656
2657         memset(&data, 0, sizeof(data));
2658         data.ioc_dev = cur_device;
2659         data.ioc_inllen1 = strlen(argv[1]) + 1;
2660         data.ioc_inlbuf1 = argv[1];
2661         if (argc == 4) {
2662                 data.ioc_inllen2 = strlen(argv[2]) + 1;
2663                 data.ioc_inlbuf2 = argv[2];
2664                 data.ioc_inllen3 = strlen(argv[3]) + 1;
2665                 data.ioc_inlbuf3 = argv[3];
2666         } else {
2667                 char from[2] = "1", to[3] = "-1";
2668                 data.ioc_inllen2 = strlen(from) + 1;
2669                 data.ioc_inlbuf2 = from;
2670                 data.ioc_inllen3 = strlen(to) + 1;
2671                 data.ioc_inlbuf3 = to;
2672         }
2673         data.ioc_inllen4 = sizeof(rawbuf) - cfs_size_round(sizeof(data)) -
2674                 cfs_size_round(data.ioc_inllen1) -
2675                 cfs_size_round(data.ioc_inllen2) -
2676                 cfs_size_round(data.ioc_inllen3);
2677         memset(buf, 0, sizeof(rawbuf));
2678         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2679         if (rc) {
2680                 fprintf(stderr, "error: %s: invalid ioctl\n",
2681                         jt_cmdname(argv[0]));
2682                 return rc;
2683         }
2684
2685         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_CHECK, buf);
2686         if (rc == 0)
2687                 fprintf(stdout, "%s", ((struct obd_ioctl_data*)buf)->ioc_bulk);
2688         else
2689                 fprintf(stderr, "OBD_IOC_LLOG_CHECK failed: %s\n",
2690                         strerror(errno));
2691         return rc;
2692 }
2693
2694 int jt_llog_remove(int argc, char **argv)
2695 {
2696         struct obd_ioctl_data data;
2697         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
2698         int rc;
2699
2700         if (argc != 3 && argc != 2)
2701                 return CMD_HELP;
2702
2703         memset(&data, 0, sizeof(data));
2704         data.ioc_dev = cur_device;
2705         data.ioc_inllen1 = strlen(argv[1]) + 1;
2706         data.ioc_inlbuf1 = argv[1];
2707         if (argc == 3){
2708                 data.ioc_inllen2 = strlen(argv[2]) + 1;
2709                 data.ioc_inlbuf2 = argv[2];
2710         }
2711         memset(buf, 0, sizeof(rawbuf));
2712         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
2713         if (rc) {
2714                 fprintf(stderr, "error: %s: invalid ioctl\n",
2715                         jt_cmdname(argv[0]));
2716                 return rc;
2717         }
2718
2719         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_LLOG_REMOVE, buf);
2720         if (rc == 0) {
2721                 if (argc == 2)
2722                         fprintf(stdout, "log %s is removed.\n", argv[1]);
2723                 else
2724                         fprintf(stdout, "the log in catalog %s is removed. \n",
2725                                 argv[1]);
2726         } else
2727                 fprintf(stderr, "OBD_IOC_LLOG_REMOVE failed: %s\n",
2728                         strerror(errno));
2729
2730         return rc;
2731 }
2732
2733 /* attach a regular file to virtual block device.
2734  * return vaule:
2735  *  -1: fatal error
2736  *  1: error, it always means the command run failed
2737  *  0: success
2738  */
2739 static int jt_blockdev_run_process(const char *file, char *argv[])
2740 {
2741         pid_t pid;
2742         int rc;
2743
2744         pid = vfork();
2745         if (pid == 0) { /* child process */
2746                 /* don't print error messages */
2747                 close(1), close(2);
2748                 (void)execvp(file, argv);
2749                 exit(-1);
2750         } else if (pid > 0) {
2751                 int status;
2752
2753                 rc = waitpid(pid, &status, 0);
2754                 if (rc < 0 || !WIFEXITED(status))
2755                         return -1;
2756
2757                 return WEXITSTATUS(status);
2758         }
2759
2760         return -1;
2761 }
2762
2763 static int jt_blockdev_find_module(const char *module)
2764 {
2765         FILE *fp;
2766         int found = 0;
2767         char buf[1024];
2768         char *ptr;
2769
2770         fp = fopen("/proc/modules", "r");
2771         if (fp == NULL)
2772                 return -1;
2773
2774         while (fgets(buf, 1024, fp) != NULL) {
2775                 ptr = strchr(buf, ' ');
2776                 if (ptr != NULL)
2777                         *ptr = 0;
2778                 if (strcmp(module, buf) == 0) {
2779                         found = 1;
2780                         break;
2781                 }
2782         }
2783         fclose(fp);
2784
2785         return found;
2786 }
2787
2788 static int jt_blockdev_probe_module(const char *module)
2789 {
2790         char buf[1024];
2791         char *argv[10];
2792         int c, rc;
2793
2794         if (jt_blockdev_find_module(module) == 1)
2795                 return 0;
2796
2797         /* run modprobe first */
2798         c = 0;
2799         argv[c++] = "/sbin/modprobe";
2800         argv[c++] = "-q";
2801         argv[c++] = (char *)module;
2802         argv[c++] = NULL;
2803         rc = jt_blockdev_run_process("modprobe", argv);
2804         if (rc != 1)
2805                 return rc;
2806
2807         /* cannot find the module in default directory ... */
2808         sprintf(buf, "../llite/%s.ko", module);
2809         c = 0;
2810         argv[c++] = "/sbin/insmod";
2811         argv[c++] = buf;
2812         argv[c++] = NULL;
2813         rc = jt_blockdev_run_process("insmod", argv);
2814         return rc ? -1 : 0;
2815 }
2816
2817 int jt_blockdev_attach(int argc, char **argv)
2818 {
2819         int rc, fd;
2820         struct stat st;
2821         char *filename, *devname;
2822         unsigned long dev;
2823
2824         if (argc != 3)
2825                 return CMD_HELP;
2826
2827         if (jt_blockdev_probe_module("llite_lloop") < 0) {
2828                 fprintf(stderr, "error: cannot find module llite_lloop.(k)o\n");
2829                 return ENOENT;
2830         }
2831
2832         filename = argv[1];
2833         devname = argv[2];
2834
2835         fd = open(filename, O_RDWR);
2836         if (fd < 0) {
2837                 fprintf(stderr, "file %s can't be opened(%s)\n\n",
2838                         filename, strerror(errno));
2839                 return CMD_HELP;
2840         }
2841
2842         rc = ioctl(fd, LL_IOC_LLOOP_ATTACH, &dev);
2843         if (rc < 0) {
2844                 rc = errno;
2845                 fprintf(stderr, "attach error(%s)\n", strerror(rc));
2846                 goto out;
2847         }
2848
2849         rc = stat(devname, &st);
2850         if (rc == 0 && (!S_ISBLK(st.st_mode) || st.st_rdev != dev)) {
2851                 rc = EEXIST;
2852         } else if (rc < 0) {
2853                 if (errno == ENOENT &&
2854                     !mknod(devname, S_IFBLK|S_IRUSR|S_IWUSR, dev))
2855                         rc = 0;
2856                 else
2857                         rc = errno;
2858         }
2859
2860         if (rc) {
2861                 fprintf(stderr, "error: the file %s could be attached to block "
2862                                 "device %X but creating %s failed: %s\n"
2863                                 "now detaching the block device..",
2864                         filename, (int)dev, devname, strerror(rc));
2865
2866                 (void)ioctl(fd, LL_IOC_LLOOP_DETACH_BYDEV, dev);
2867                 fprintf(stderr, "%s\n", strerror(errno));
2868         }
2869 out:
2870         close(fd);
2871         return -rc;
2872 }
2873
2874 int jt_blockdev_detach(int argc, char **argv)
2875 {
2876         char *filename;
2877         int rc, fd;
2878
2879         if (argc != 2)
2880                 return CMD_HELP;
2881
2882         filename = argv[1];
2883         fd = open(filename, O_RDONLY);
2884         if (fd < 0) {
2885                 fprintf(stderr, "cannot open file %s error %s\n",
2886                         filename, strerror(errno));
2887                 return CMD_HELP;
2888         }
2889
2890         rc = ioctl(fd, LL_IOC_LLOOP_DETACH, 0);
2891         if (rc < 0) {
2892                 rc = errno;
2893                 fprintf(stderr, "detach error(%s)\n", strerror(rc));
2894         } else {
2895                 (void)unlink(filename);
2896         }
2897
2898         close(fd);
2899         return -rc;
2900 }
2901
2902 int jt_blockdev_info(int argc, char **argv)
2903 {
2904         char *filename;
2905         int rc, fd;
2906         struct lu_fid fid;
2907
2908         if (argc != 2)
2909                 return CMD_HELP;
2910
2911         filename = argv[1];
2912         fd = open(filename, O_RDONLY);
2913         if (fd < 0) {
2914                 fprintf(stderr, "cannot open file %s error: %s\n",
2915                         filename, strerror(errno));
2916                 return CMD_HELP;
2917         }
2918
2919         rc = ioctl(fd, LL_IOC_LLOOP_INFO, &fid);
2920         if (rc < 0) {
2921                 rc = errno;
2922                 fprintf(stderr, "error: %s\n", strerror(errno));
2923                 goto out;
2924         }
2925         fprintf(stdout, "lloop device info: ");
2926         if (fid_is_zero(&fid))
2927                 fprintf(stdout, "Not attached\n");
2928         else
2929                 fprintf(stdout, "attached to inode "DFID"\n", PFID(&fid));
2930 out:
2931         close(fd);
2932         return -rc;
2933 }
2934
2935 static void signal_server(int sig)
2936 {
2937         if (sig == SIGINT) {
2938                 do_disconnect("sigint", 1);
2939                 exit(1);
2940         } else
2941                 fprintf(stderr, "%s: got signal %d\n", jt_cmdname("sigint"), sig);
2942 }
2943
2944 int obd_initialize(int argc, char **argv)
2945 {
2946         if (shmem_setup() != 0)
2947                 return -1;
2948
2949         register_ioc_dev(OBD_DEV_ID, OBD_DEV_PATH,
2950                          OBD_DEV_MAJOR, OBD_DEV_MINOR);
2951
2952         return 0;
2953 }
2954
2955 void obd_finalize(int argc, char **argv)
2956 {
2957         struct sigaction sigact;
2958
2959         /* sigact initialization */
2960         sigact.sa_handler = signal_server;
2961         sigfillset(&sigact.sa_mask);
2962         sigact.sa_flags = SA_RESTART;
2963         /* coverity[uninit_use_in_call] */
2964         sigaction(SIGINT, &sigact, NULL);
2965
2966         shmem_cleanup();
2967         do_disconnect(argv[0], 1);
2968 }
2969
2970 static int check_pool_cmd(enum lcfg_command_type cmd,
2971                           char *fsname, char *poolname,
2972                           char *ostname)
2973 {
2974         int rc;
2975
2976         rc = llapi_search_ost(fsname, poolname, ostname);
2977         if (rc < 0 && (cmd != LCFG_POOL_NEW)) {
2978                 fprintf(stderr, "Pool %s.%s not found\n",
2979                         fsname, poolname);
2980                 return rc;
2981         }
2982
2983         switch (cmd) {
2984         case LCFG_POOL_NEW: {
2985                 LASSERT(ostname == NULL);
2986                 if (rc >= 0) {
2987                         fprintf(stderr, "Pool %s.%s already exists\n",
2988                                 fsname, poolname);
2989                         return -EEXIST;
2990                 }
2991                 return 0;
2992         }
2993         case LCFG_POOL_DEL: {
2994                 LASSERT(ostname == NULL);
2995                 if (rc == 1) {
2996                         fprintf(stderr, "Pool %s.%s not empty, "
2997                                 "please remove all members\n",
2998                                 fsname, poolname);
2999                         return -ENOTEMPTY;
3000                 }
3001                 return 0;
3002         }
3003         case LCFG_POOL_ADD: {
3004                 if (rc == 1) {
3005                         fprintf(stderr, "OST %s is already in pool %s.%s\n",
3006                                 ostname, fsname, poolname);
3007                         return -EEXIST;
3008                 }
3009                 rc = llapi_search_ost(fsname, NULL, ostname);
3010                 if (rc == 0) {
3011                         fprintf(stderr, "OST %s is not part of the '%s' fs.\n",
3012                                 ostname, fsname);
3013                         return -ENOENT;
3014                 }
3015                 return 0;
3016         }
3017         case LCFG_POOL_REM: {
3018                 if (rc == 0) {
3019                         fprintf(stderr, "OST %s not found in pool %s.%s\n",
3020                                 ostname, fsname, poolname);
3021                         return -ENOENT;
3022                 }
3023                 return 0;
3024         }
3025         default:
3026                 break;
3027         } /* switch */
3028         return -EINVAL;
3029 }
3030
3031 /* This check only verifies that the changes have been "pushed out" to
3032    the client successfully.  This involves waiting for a config update,
3033    and so may fail because of problems in that code or post-command
3034    network loss. So reporting a warning is appropriate, but not a failure.
3035 */
3036 static int check_pool_cmd_result(enum lcfg_command_type cmd,
3037                                  char *fsname, char *poolname,
3038                                  char *ostname)
3039 {
3040         int cpt = 10;
3041         int rc = 0;
3042
3043         switch (cmd) {
3044         case LCFG_POOL_NEW: {
3045                 do {
3046                         rc = llapi_search_ost(fsname, poolname, NULL);
3047                         if (rc == -ENODEV)
3048                                 return rc;
3049                         if (rc < 0)
3050                                 sleep(2);
3051                         cpt--;
3052                 } while ((rc < 0) && (cpt > 0));
3053                 if (rc >= 0) {
3054                         fprintf(stderr, "Pool %s.%s created\n",
3055                                 fsname, poolname);
3056                         return 0;
3057                 } else {
3058                         fprintf(stderr, "Warning, pool %s.%s not found\n",
3059                                 fsname, poolname);
3060                         return -ENOENT;
3061                 }
3062         }
3063         case LCFG_POOL_DEL: {
3064                 do {
3065                         rc = llapi_search_ost(fsname, poolname, NULL);
3066                         if (rc == -ENODEV)
3067                                 return rc;
3068                         if (rc >= 0)
3069                                 sleep(2);
3070                         cpt--;
3071                 } while ((rc >= 0) && (cpt > 0));
3072                 if (rc < 0) {
3073                         fprintf(stderr, "Pool %s.%s destroyed\n",
3074                                 fsname, poolname);
3075                         return 0;
3076                 } else {
3077                         fprintf(stderr, "Warning, pool %s.%s still found\n",
3078                                 fsname, poolname);
3079                         return -EEXIST;
3080                 }
3081         }
3082         case LCFG_POOL_ADD: {
3083                 do {
3084                         rc = llapi_search_ost(fsname, poolname, ostname);
3085                         if (rc == -ENODEV)
3086                                 return rc;
3087                         if (rc != 1)
3088                                 sleep(2);
3089                         cpt--;
3090                 } while ((rc != 1) && (cpt > 0));
3091                 if (rc == 1) {
3092                         fprintf(stderr, "OST %s added to pool %s.%s\n",
3093                                 ostname, fsname, poolname);
3094                         return 0;
3095                 } else {
3096                         fprintf(stderr, "Warning, OST %s not found in pool %s.%s\n",
3097                                 ostname, fsname, poolname);
3098                         return -ENOENT;
3099                 }
3100         }
3101         case LCFG_POOL_REM: {
3102                 do {
3103                         rc = llapi_search_ost(fsname, poolname, ostname);
3104                         if (rc == -ENODEV)
3105                                 return rc;
3106                         if (rc == 1)
3107                                 sleep(2);
3108                         cpt--;
3109                 } while ((rc == 1) && (cpt > 0));
3110                 if (rc != 1) {
3111                         fprintf(stderr, "OST %s removed from pool %s.%s\n",
3112                                 ostname, fsname, poolname);
3113                         return 0;
3114                 } else {
3115                         fprintf(stderr, "Warning, OST %s still found in pool %s.%s\n",
3116                                 ostname, fsname, poolname);
3117                         return -EEXIST;
3118                 }
3119         }
3120         default:
3121                 break;
3122         }
3123         return -EINVAL;
3124 }
3125
3126 static int check_and_complete_ostname(char *fsname, char *ostname)
3127 {
3128         char *ptr;
3129         char real_ostname[MAX_OBD_NAME + 1];
3130         char i;
3131
3132         /* if OST name does not start with fsname, we add it */
3133         /* if not check if the fsname is the right one */
3134         ptr = strchr(ostname, '-');
3135         if (ptr == NULL) {
3136                 sprintf(real_ostname, "%s-%s", fsname, ostname);
3137         } else if (strncmp(ostname, fsname, strlen(fsname)) != 0) {
3138                 fprintf(stderr, "%s does not start with fsname %s\n",
3139                         ostname, fsname);
3140                 return -EINVAL;
3141         } else {
3142                 if (strlen(ostname) > sizeof(real_ostname)-1)
3143                         return -E2BIG;
3144                 strncpy(real_ostname, ostname, sizeof(real_ostname));
3145         }
3146         /* real_ostname is fsname-????? */
3147         ptr = real_ostname + strlen(fsname) + 1;
3148         if (strncmp(ptr, "OST", 3) != 0) {
3149                 fprintf(stderr, "%s does not start by %s-OST nor OST\n",
3150                         ostname, fsname);
3151                 return -EINVAL;
3152         }
3153         /* real_ostname is fsname-OST????? */
3154         ptr += 3;
3155         for (i = 0; i < 4; i++) {
3156                 if (!isxdigit(*ptr)) {
3157                         fprintf(stderr,
3158                                 "ost's index in %s is not an hexa number\n",
3159                                 ostname);
3160                         return -EINVAL;
3161                 }
3162                 ptr++;
3163         }
3164         /* real_ostname is fsname-OSTXXXX????? */
3165         /* if OST name does not end with _UUID, we add it */
3166         if (*ptr == '\0') {
3167                 strcat(real_ostname, "_UUID");
3168         } else if (strcmp(ptr, "_UUID") != 0) {
3169                 fprintf(stderr,
3170                         "ostname %s does not end with _UUID\n", ostname);
3171                 return -EINVAL;
3172         }
3173         /* real_ostname is fsname-OSTXXXX_UUID */
3174         strcpy(ostname, real_ostname);
3175         return 0;
3176 }
3177
3178 /* returns 0 or -errno */
3179 static int pool_cmd(enum lcfg_command_type cmd,
3180                     char *cmdname, char *fullpoolname,
3181                     char *fsname, char *poolname, char *ostname)
3182 {
3183         int rc = 0;
3184         struct obd_ioctl_data data;
3185         struct lustre_cfg_bufs bufs;
3186         struct lustre_cfg *lcfg;
3187         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf;
3188
3189         rc = check_pool_cmd(cmd, fsname, poolname, ostname);
3190         if (rc == -ENODEV)
3191                 fprintf(stderr, "Can't verify pool command since there "
3192                         "is no local MDT or client, proceeding anyhow...\n");
3193         else if (rc)
3194                 return rc;
3195
3196         lustre_cfg_bufs_reset(&bufs, NULL);
3197         lustre_cfg_bufs_set_string(&bufs, 0, cmdname);
3198         lustre_cfg_bufs_set_string(&bufs, 1, fullpoolname);
3199         if (ostname != NULL)
3200                 lustre_cfg_bufs_set_string(&bufs, 2, ostname);
3201
3202         lcfg = lustre_cfg_new(cmd, &bufs);
3203         if (lcfg == NULL)
3204                 return -ENOMEM;
3205
3206         memset(&data, 0, sizeof(data));
3207         rc = data.ioc_dev = get_mgs_device();
3208         if (rc < 0)
3209                 goto out;
3210
3211         data.ioc_type = LUSTRE_CFG_TYPE;
3212         data.ioc_plen1 = lustre_cfg_len(lcfg->lcfg_bufcount,
3213                                         lcfg->lcfg_buflens);
3214         data.ioc_pbuf1 = (void *)lcfg;
3215
3216         memset(buf, 0, sizeof(rawbuf));
3217         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
3218         if (rc) {
3219                 fprintf(stderr, "error: %s: invalid ioctl\n",
3220                         jt_cmdname(cmdname));
3221                 lustre_cfg_free(lcfg);
3222                 return rc;
3223         }
3224         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_POOL, buf);
3225 out:
3226         if (rc)
3227                 rc = -errno;
3228         lustre_cfg_free(lcfg);
3229         return rc;
3230 }
3231
3232 /**
3233  * Format and send the ioctl to the MGS.
3234  *
3235  * \param       cmd             IOCTL to send
3236  * \param       ret_data        void pointer to return anything from
3237  *                              ioctl
3238  * \param       num_args        number of arguments to pack into the
3239  *                              ioctl buffer
3240  * \param       argv[]          variable number of string arguments
3241  *
3242  * \retval                      0 on success
3243  */
3244 static int nodemap_cmd(enum lcfg_command_type cmd, void *ret_data,
3245                        unsigned int ret_size, ...)
3246 {
3247         va_list                 ap;
3248         char                    *arg;
3249         int                     i = 0;
3250         struct lustre_cfg_bufs  bufs;
3251         struct obd_ioctl_data   data;
3252         struct lustre_cfg       *lcfg;
3253         char                    rawbuf[MAX_IOC_BUFLEN];
3254         char                    *buf = rawbuf;
3255         int                     rc = 0;
3256
3257         lustre_cfg_bufs_reset(&bufs, NULL);
3258
3259         va_start(ap, ret_size);
3260         arg = va_arg(ap, char *);
3261         while (arg != NULL) {
3262                 lustre_cfg_bufs_set_string(&bufs, i, arg);
3263                 i++;
3264                 arg = va_arg(ap, char *);
3265         }
3266         va_end(ap);
3267
3268         lcfg = lustre_cfg_new(cmd, &bufs);
3269         if (lcfg == NULL)
3270                 return -ENOMEM;
3271
3272         memset(&data, 0, sizeof(data));
3273         rc = data.ioc_dev = get_mgs_device();
3274         if (rc < 0)
3275                 goto out;
3276
3277         data.ioc_type = LUSTRE_CFG_TYPE;
3278         data.ioc_plen1 = lustre_cfg_len(lcfg->lcfg_bufcount,
3279                          lcfg->lcfg_buflens);
3280         data.ioc_pbuf1 = (void *)lcfg;
3281
3282         memset(buf, 0, sizeof(rawbuf));
3283         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
3284         if (rc != 0) {
3285                 fprintf(stderr, "error: invalid ioctl: %08x errno: %d with "
3286                                "rc=%d\n", cmd, errno, rc);
3287                 goto out;
3288         }
3289
3290         rc = l_ioctl(OBD_DEV_ID, OBD_IOC_NODEMAP, buf);
3291         if (rc != 0) {
3292                 fprintf(stderr, "error: invalid ioctl: %08x errno: %d with "
3293                                "rc=%d\n", cmd, errno, rc);
3294                 goto out;
3295         }
3296
3297         if (ret_data != NULL) {
3298                 rc = obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
3299                 if (rc != 0)
3300                         goto out;
3301
3302                 if (ret_size > data.ioc_plen1)
3303                         ret_size = data.ioc_plen1;
3304
3305                 memcpy(ret_data, data.ioc_pbuf1, ret_size);
3306         }
3307 out:
3308         lustre_cfg_free(lcfg);
3309
3310         return rc;
3311 }
3312
3313 /**
3314  * activate nodemap functions
3315  *
3316  * \param       argc            number of args
3317  * \param       argv[]          variable string arguments
3318  *
3319  * argv[0]                      1 for activate or 0 for deactivate
3320  *
3321  * \retval                      0 on success
3322  */
3323 int jt_nodemap_activate(int argc, char **argv)
3324 {
3325         int rc;
3326
3327         rc = nodemap_cmd(LCFG_NODEMAP_ACTIVATE, NULL, 0, argv[0], argv[1],
3328                          NULL);
3329
3330         if (rc != 0) {
3331                 errno = -rc;
3332                 perror(argv[0]);
3333         }
3334
3335         return rc;
3336 }
3337
3338 /**
3339  * add a nodemap
3340  *
3341  * \param       argc            number of args
3342  * \param       argv[]          variable string arguments
3343  *
3344  * argv[0]                      nodemap name
3345  *
3346  * \retval                      0 on success
3347  */
3348 int jt_nodemap_add(int argc, char **argv)
3349 {
3350         int rc;
3351
3352         rc = llapi_nodemap_exists(argv[1]);
3353         if (rc == 0) {
3354                 fprintf(stderr, "error: %s existing nodemap name\n", argv[1]);
3355                 return 1;
3356         }
3357
3358         rc = nodemap_cmd(LCFG_NODEMAP_ADD, NULL, 0, argv[0], argv[1], NULL);
3359
3360         if (rc != 0) {
3361                 errno = -rc;
3362                 perror(argv[0]);
3363         }
3364
3365         return rc;
3366 }
3367
3368 /**
3369  * delete a nodemap
3370  *
3371  * \param       argc            number of args
3372  * \param       argv[]          variable string arguments
3373  *
3374  * argv[0]                      nodemap name
3375  *
3376  * \retval                      0 on success
3377  */
3378 int jt_nodemap_del(int argc, char **argv)
3379 {
3380         int rc;
3381
3382         rc = llapi_nodemap_exists(argv[1]);
3383         if (rc != 0) {
3384                 fprintf(stderr, "error: %s not existing nodemap name\n",
3385                         argv[1]);
3386                 return rc;
3387         }
3388         rc = nodemap_cmd(LCFG_NODEMAP_DEL, NULL, 0, argv[0], argv[1], NULL);
3389
3390         if (rc != 0) {
3391                 errno = -rc;
3392                 perror(argv[0]);
3393         }
3394
3395         return rc;
3396 }
3397
3398 /**
3399  * test a nid for nodemap membership
3400  *
3401  * \param       argc            number of args
3402  * \param       argv[]          variable string arguments
3403  *
3404  * argv[0]                      properly formatted nid
3405  *
3406  * \retval                      0 on success
3407  */
3408 int jt_nodemap_test_nid(int argc, char **argv)
3409 {
3410
3411         char    rawbuf[MAX_IOC_BUFLEN];
3412         int     rc;
3413
3414         rc = nodemap_cmd(LCFG_NODEMAP_TEST_NID, &rawbuf, sizeof(rawbuf),
3415                          argv[0], argv[1], NULL);
3416         if (rc == 0)
3417                 printf("%s\n", (char *)rawbuf);
3418
3419         return rc;
3420 }
3421
3422 /**
3423  * test a nodemap id pair for mapping
3424  *
3425  * \param       argc            number of args
3426  * \param       argv[[]         variable string arguments
3427  *
3428  * \retval                      0 on success
3429  *
3430  * The argv array should contain the nodemap name, the id
3431  * to checking the mapping on, and the id type (UID or GID)
3432  *
3433  */
3434 int jt_nodemap_test_id(int argc, char **argv)
3435 {
3436         char    rawbuf[MAX_IOC_BUFLEN];
3437         char    *nidstr = NULL;
3438         char    *idstr = NULL;
3439         char    *typestr = NULL;
3440         int     rc = 0;
3441         int     c;
3442
3443         static struct option long_options[] = {
3444                 {
3445                         .name           = "nid",
3446                         .has_arg        = required_argument,
3447                         .flag           = 0,
3448                         .val            = 'n',
3449                 },
3450                 {
3451                         .name           = "idtype",
3452                         .has_arg        = required_argument,
3453                         .flag           = 0,
3454                         .val            = 't',
3455                 },
3456                 {
3457                         .name           = "id",
3458                         .has_arg        = required_argument,
3459                         .flag           = 0,
3460                         .val            = 'i',
3461                 },
3462                 {
3463                         NULL
3464                 }
3465         };
3466
3467         while ((c = getopt_long(argc, argv, "n:t:i:",
3468                                 long_options, NULL)) != -1) {
3469                 switch (c) {
3470                 case 'n':
3471                         nidstr = optarg;
3472                         break;
3473                 case 't':
3474                         typestr = optarg;
3475                         break;
3476                 case 'i':
3477                         idstr = optarg;
3478                         break;
3479                 }
3480         }
3481
3482         if (nidstr == NULL || typestr == NULL || idstr == NULL) {
3483                 fprintf(stderr, "usage: nodemap_test_id --nid <nid> "
3484                                 "--idtype [uid|gid] --id <id>\n");
3485                 return -1;
3486         }
3487
3488         rc = nodemap_cmd(LCFG_NODEMAP_TEST_ID, &rawbuf, sizeof(rawbuf),
3489                          argv[0], nidstr, typestr, idstr);
3490         if (rc == 0)
3491                 printf("%s\n", (char *)rawbuf);
3492
3493         return rc;
3494 }
3495
3496 /**
3497  * add an nid range to a nodemap
3498  *
3499  * \param       argc            number of args
3500  * \param       argv[]          variable string arguments
3501  *
3502  * --name                       nodemap name
3503  * --range                      properly formatted nid range
3504  *
3505  * \retval                      0 on success
3506  */
3507 int jt_nodemap_add_range(int argc, char **argv)
3508 {
3509         char                    *nodemap_name = NULL;
3510         char                    *nodemap_range = NULL;
3511         struct list_head        nidlist;
3512         char                    min_nid[LNET_NIDSTR_SIZE + 1];
3513         char                    max_nid[LNET_NIDSTR_SIZE + 1];
3514         char                    nid_range[2 * LNET_NIDSTR_SIZE + 2];
3515         int                     rc = 0;
3516         int                     c;
3517
3518         static struct option long_options[] = {
3519                 {
3520                         .name           = "name",
3521                         .has_arg        = required_argument,
3522                         .flag           = 0,
3523                         .val            = 'n',
3524                 },
3525                 {
3526                         .name           = "range",
3527                         .has_arg        = required_argument,
3528                         .flag           = 0,
3529                         .val            = 'r',
3530                 },
3531                 {
3532                         NULL
3533                 }
3534         };
3535
3536         INIT_LIST_HEAD(&nidlist);
3537
3538         while ((c = getopt_long(argc, argv, "n:r:",
3539                                 long_options, NULL)) != -1) {
3540                 switch (c) {
3541                 case 'n':
3542                         nodemap_name = optarg;
3543                         break;
3544                 case 'r':
3545                         nodemap_range = optarg;
3546                         break;
3547                 }
3548         }
3549
3550         if (nodemap_name == NULL || nodemap_range == NULL) {
3551                 fprintf(stderr, "usage: nodemap_add_range --name <name> "
3552                                 "--range <range>\n");
3553                 return -1;
3554         }
3555
3556         if (cfs_parse_nidlist(nodemap_range, strlen(nodemap_range),
3557                               &nidlist) <= 0) {
3558                 fprintf(stderr, "error: %s: can't parse nid range: %s\n",
3559                         jt_cmdname(argv[0]), nodemap_range);
3560                 return -1;
3561         }
3562
3563         if (!cfs_nidrange_is_contiguous(&nidlist)) {
3564                 fprintf(stderr, "error: %s: nodemap ranges must be "
3565                         "contiguous\n", jt_cmdname(argv[0]));
3566                 return -1;
3567         }
3568
3569         cfs_nidrange_find_min_max(&nidlist, &min_nid[0], &max_nid[0],
3570                                   LNET_NIDSTR_SIZE);
3571         snprintf(nid_range, sizeof(nid_range), "%s:%s", min_nid, max_nid);
3572
3573         rc = nodemap_cmd(LCFG_NODEMAP_ADD_RANGE, NULL, 0, argv[0],
3574                          nodemap_name, nid_range, NULL);
3575         if (rc != 0) {
3576                 errno = -rc;
3577                 fprintf(stderr, "error: %s: cannot add range '%s' to nodemap "
3578                                 "'%s': rc = %d\n",
3579                         jt_cmdname(argv[0]), nodemap_range, nodemap_name, rc);
3580         }
3581
3582         return rc;
3583 }
3584
3585 /**
3586  * delete an nid range to a nodemap
3587  *
3588  * \param       argc            number of args
3589  * \param       argv[]          variable string arguments
3590  *
3591  * --name                       nodemap name
3592  * --range                      properly formatted nid range
3593  *
3594  * \retval                      0 on success
3595  */
3596 int jt_nodemap_del_range(int argc, char **argv)
3597 {
3598         char                    *nodemap_name = NULL;
3599         char                    *nodemap_range = NULL;
3600         struct list_head        nidlist;
3601         char                    min_nid[LNET_NIDSTR_SIZE + 1];
3602         char                    max_nid[LNET_NIDSTR_SIZE + 1];
3603         char                    nid_range[2 * LNET_NIDSTR_SIZE + 2];
3604         int                     rc = 0;
3605         int                     c;
3606
3607         static struct option long_options[] = {
3608                 {
3609                         .name           = "name",
3610                         .has_arg        = required_argument,
3611                         .flag           = 0,
3612                         .val            = 'n',
3613                 },
3614                 {
3615                         .name           = "range",
3616                         .has_arg        = required_argument,
3617                         .flag           = 0,
3618                         .val            = 'r',
3619                 },
3620                 {
3621                         NULL
3622                 }
3623         };
3624
3625         INIT_LIST_HEAD(&nidlist);
3626
3627         while ((c = getopt_long(argc, argv, "n:r:",
3628                                 long_options, NULL)) != -1) {
3629                 switch (c) {
3630                 case 'n':
3631                         nodemap_name = optarg;
3632                         break;
3633                 case 'r':
3634                         nodemap_range = optarg;
3635                         break;
3636                 }
3637         }
3638
3639         if (nodemap_name == NULL || nodemap_range == NULL) {
3640                 fprintf(stderr, "usage: nodemap_del_range --name <name> "
3641                                 "--range <range>\n");
3642                 return -1;
3643         }
3644
3645         if (cfs_parse_nidlist(nodemap_range, strlen(nodemap_range),
3646                               &nidlist) <= 0) {
3647                 fprintf(stderr, "error: %s: can't parse nid range: %s\n",
3648                         jt_cmdname(argv[0]), nodemap_range);
3649                 return -1;
3650         }
3651
3652         if (!cfs_nidrange_is_contiguous(&nidlist)) {
3653                 fprintf(stderr, "error: %s: nodemap ranges must be "
3654                         "contiguous\n", jt_cmdname(argv[0]));
3655                 return -1;
3656         }
3657
3658         cfs_nidrange_find_min_max(&nidlist, &min_nid[0], &max_nid[0],
3659                                   LNET_NIDSTR_SIZE);
3660         snprintf(nid_range, sizeof(nid_range), "%s:%s", min_nid, max_nid);
3661
3662         rc = nodemap_cmd(LCFG_NODEMAP_DEL_RANGE, NULL, 0, argv[0],
3663                          nodemap_name, nid_range, NULL);
3664         if (rc != 0) {
3665                 errno = -rc;
3666                 fprintf(stderr, "error: %s: cannot delete range '%s' to "
3667                                "nodemap '%s': rc = %d\n",
3668                         jt_cmdname(argv[0]), nodemap_range, nodemap_name, rc);
3669         }
3670
3671         return rc;
3672 }
3673
3674 /**
3675  * set a fileset on a nodemap
3676  *
3677  * \param       argc            number of args
3678  * \param       argv[]          variable string arguments
3679  *
3680  * --name                       nodemap name
3681  * --fileset                    fileset name
3682  *
3683  * \retval                      0 on success
3684  */
3685 int jt_nodemap_set_fileset(int argc, char **argv)
3686 {
3687         char *nodemap_name = NULL;
3688         char *fileset_name = NULL;
3689         int   rc = 0;
3690         int   c;
3691
3692         static struct option long_options[] = {
3693                 {
3694                         .name           = "name",
3695                         .has_arg        = required_argument,
3696                         .flag           = 0,
3697                         .val            = 'n',
3698                 },
3699                 {
3700                         .name           = "fileset",
3701                         .has_arg        = required_argument,
3702                         .flag           = 0,
3703                         .val            = 'f',
3704                 },
3705                 {
3706                         NULL
3707                 }
3708         };
3709
3710         while ((c = getopt_long(argc, argv, "n:f:",
3711                                 long_options, NULL)) != -1) {
3712                 switch (c) {
3713                 case 'n':
3714                         nodemap_name = optarg;
3715                         break;
3716                 case 'f':
3717                         fileset_name = optarg;
3718                         break;
3719                 }
3720         }
3721
3722         if (nodemap_name == NULL || fileset_name == NULL) {
3723                 fprintf(stderr, "usage: nodemap_set_fileset --name <name> "
3724                                 "--fileset <fileset>\n");
3725                 return -1;
3726         }
3727
3728         rc = nodemap_cmd(LCFG_NODEMAP_SET_FILESET, NULL, 0, argv[0],
3729                          nodemap_name, fileset_name, NULL);
3730         if (rc != 0) {
3731                 errno = -rc;
3732                 fprintf(stderr, "error: %s: cannot set fileset '%s' on nodemap "
3733                                 "'%s': rc = %d\n",
3734                         jt_cmdname(argv[0]), fileset_name, nodemap_name, rc);
3735         }
3736
3737         return rc;
3738 }
3739
3740 /**
3741  * modify a nodemap's behavior
3742  *
3743  * \param       argc            number of args
3744  * \param       argv[]          variable string arguments
3745  *
3746  * --name                       nodemap name
3747  * --property                   nodemap property to change
3748  *                              admin, trusted, squash_uid, squash_gid)
3749  * --value                      value to set property
3750  *
3751  * \retval                      0 on success
3752  */
3753 int jt_nodemap_modify(int argc, char **argv)
3754 {
3755         int                     c;
3756         int                     rc = 0;
3757         enum lcfg_command_type  cmd = 0;
3758         char                    *nodemap_name = NULL;
3759         char                    *param = NULL;
3760         char                    *value = NULL;
3761
3762         static struct option long_options[] = {
3763                 {
3764                         .name           = "name",
3765                         .has_arg        = required_argument,
3766                         .flag           = 0,
3767                         .val            = 'n',
3768                 },
3769                 {
3770                         .name           = "property",
3771                         .has_arg        = required_argument,
3772                         .flag           = 0,
3773                         .val            = 'p',
3774                 },
3775                 {
3776                         .name           = "value",
3777                         .has_arg        = required_argument,
3778                         .flag           = 0,
3779                         .val            = 'v',
3780                 },
3781                 {
3782                         NULL
3783                 }
3784         };
3785
3786         while ((c = getopt_long(argc, argv, "n:p:v:",
3787                                 long_options, NULL)) != -1) {
3788                 switch (c) {
3789                 case 'n':
3790                         nodemap_name = optarg;
3791                         break;
3792                 case 'p':
3793                         param = optarg;
3794                         break;
3795                 case 'v':
3796                         value = optarg;
3797                         break;
3798                 }
3799         }
3800
3801         if (nodemap_name == NULL || param == NULL || value == NULL) {
3802                 fprintf(stderr, "usage: nodemap_modify --name <nodemap_name> "
3803                                 "--property <property_name> --value <value>\n");
3804                 fprintf(stderr, "valid properties: admin trusted "
3805                                 "squash_uid squash_gid deny_unknown\n");
3806                 return -1;
3807         }
3808
3809         if (strcmp("admin", param) == 0) {
3810                 cmd = LCFG_NODEMAP_ADMIN;
3811         } else if (strcmp("trusted", param) == 0) {
3812                 cmd = LCFG_NODEMAP_TRUSTED;
3813         } else if (strcmp("deny_unknown", param) == 0) {
3814                 cmd = LCFG_NODEMAP_DENY_UNKNOWN;
3815         } else if (strcmp("squash_uid", param) == 0) {
3816                 cmd = LCFG_NODEMAP_SQUASH_UID;
3817         } else if (strcmp("squash_gid", param) == 0) {
3818                 cmd = LCFG_NODEMAP_SQUASH_GID;
3819         } else {
3820                 fprintf(stderr, "error: %s: nodemap_modify invalid "
3821                                 "subcommand: %s\n",
3822                         jt_cmdname(argv[0]), param);
3823                 return -1;
3824         }
3825
3826         rc = nodemap_cmd(cmd, NULL, 0, argv[0], nodemap_name, param,
3827                          value, NULL);
3828         if (rc != 0) {
3829                 errno = -rc;
3830                 fprintf(stderr, "error: %s: cannot modify nodemap '%s' "
3831                                "to param '%s': value '%s': rc = %d\n",
3832                         jt_cmdname(argv[0]), nodemap_name, param, value, rc);
3833         }
3834
3835         return rc;
3836 }
3837
3838 int jt_nodemap_add_idmap(int argc, char **argv)
3839 {
3840         int                     c;
3841         enum                    lcfg_command_type cmd = 0;
3842         char                    *nodemap_name = NULL;
3843         char                    *idmap = NULL;
3844         char                    *idtype = NULL;
3845         int                     rc = 0;
3846
3847         static struct option long_options[] = {
3848                 {
3849                         .name           = "name",
3850                         .has_arg        = required_argument,
3851                         .flag           = 0,
3852                         .val            = 'n',
3853                 },
3854                 {
3855                         .name           = "idmap",
3856                         .has_arg        = required_argument,
3857                         .flag           = 0,
3858                         .val            = 'm',
3859                 },
3860                 {
3861                         .name           = "idtype",
3862                         .has_arg        = required_argument,
3863                         .flag           = 0,
3864                         .val            = 'i',
3865                 },
3866                 {
3867                         NULL
3868                 }
3869         };
3870
3871         while ((c = getopt_long(argc, argv, "n:m:i:",
3872                                 long_options, NULL)) != -1) {
3873                 switch (c) {
3874                 case 'n':
3875                         nodemap_name = optarg;
3876                         break;
3877                 case 'm':
3878                         idmap = optarg;
3879                         break;
3880                 case 'i':
3881                         idtype = optarg;
3882                         break;
3883                 }
3884         }
3885
3886         if (nodemap_name == NULL || idmap == NULL || idtype == NULL) {
3887                 fprintf(stderr, "usage: %s --name <name> --idtype [uid | gid]"
3888                         " --idmap <client id>:<filesystem id>\n", argv[0]);
3889                 return -1;
3890         }
3891
3892         if (strcmp("uid", idtype) == 0) {
3893                 cmd = LCFG_NODEMAP_ADD_UIDMAP;
3894         } else if (strcmp("gid", idtype) == 0) {
3895                 cmd = LCFG_NODEMAP_ADD_GIDMAP;
3896         } else {
3897                 fprintf(stderr, "usage: %s --name <name> --idtype [uid | gid]"
3898                         " --idmap <client id>:<filesystem id>\n", argv[0]);
3899                 return -1;
3900         }
3901
3902         rc = nodemap_cmd(cmd, NULL, 0, argv[0], nodemap_name, idmap, NULL);
3903         if (rc != 0) {
3904                 errno = -rc;
3905                 fprintf(stderr, "cannot add %smap '%s' to nodemap '%s'"
3906                         ": rc = %d\n", idtype, idmap, nodemap_name, rc);
3907         }
3908
3909         return rc;
3910 }
3911
3912 int jt_nodemap_del_idmap(int argc, char **argv)
3913 {
3914         int                     c;
3915         enum                    lcfg_command_type cmd = 0;
3916         char                    *nodemap_name = NULL;
3917         char                    *idmap = NULL;
3918         char                    *idtype = NULL;
3919         int                     rc = 0;
3920
3921         static struct option long_options[] = {
3922                 {
3923                         .name           = "name",
3924                         .has_arg        = required_argument,
3925                         .flag           = 0,
3926                         .val            = 'n',
3927                 },
3928                 {
3929                         .name           = "idmap",
3930                         .has_arg        = required_argument,
3931                         .flag           = 0,
3932                         .val            = 'm',
3933                 },
3934                 {
3935                         .name           = "idtype",
3936                         .has_arg        = required_argument,
3937                         .flag           = 0,
3938                         .val            = 'i',
3939                 },
3940                 {
3941                         NULL
3942                 }
3943         };
3944
3945         while ((c = getopt_long(argc, argv, "n:m:i:",
3946                                 long_options, NULL)) != -1) {
3947                 switch (c) {
3948                 case 'n':
3949                         nodemap_name = optarg;
3950                         break;
3951                 case 'm':
3952                         idmap = optarg;
3953                         break;
3954                 case 'i':
3955                         idtype = optarg;
3956                         break;
3957                 }
3958         }
3959
3960         if (nodemap_name == NULL || idmap == NULL || idtype == NULL) {
3961                 fprintf(stderr, "usage: %s --name <name> --idtype [uid | gid]"
3962                         " --idmap <client id>:<filesystem id>\n", argv[0]);
3963                 return -1;
3964         }
3965
3966         if (strcmp("uid", idtype) == 0)
3967                 cmd = LCFG_NODEMAP_DEL_UIDMAP;
3968         else
3969                 cmd = LCFG_NODEMAP_DEL_GIDMAP;
3970
3971         rc = nodemap_cmd(cmd, NULL, 0, argv[0], nodemap_name, idmap, NULL);
3972         if (rc != 0) {
3973                 errno = -rc;
3974                 fprintf(stderr, "cannot delete %smap '%s' from nodemap '%s'"
3975                         ": rc = %d\n", idtype, idmap, nodemap_name, rc);
3976         }
3977
3978         return rc;
3979 }
3980
3981 /*
3982  * this function tranforms a rule [start-end/step] into an array
3983  * of matching numbers
3984  * supported forms are:
3985  * [start]                : just this number
3986  * [start-end]            : all numbers from start to end
3987  * [start-end/step]       : numbers from start to end with increment of step
3988  * on return, format contains a printf format string which can be used
3989  * to generate all the strings
3990  */
3991 static int get_array_idx(char *rule, char *format, int **array)
3992 {
3993         char *start, *end, *ptr;
3994         unsigned int lo, hi, step;
3995         int array_sz = 0;
3996         int i, array_idx;
3997         int rc;
3998
3999         start = strchr(rule, '[');
4000         end = strchr(rule, ']');
4001         if ((start == NULL) || (end == NULL)) {
4002                 *array = malloc(sizeof(int));
4003                 if (*array == NULL)
4004                         return 0;
4005                 strcpy(format, rule);
4006                 array_sz = 1;
4007                 return array_sz;
4008         }
4009         *start = '\0';
4010         *end = '\0';
4011         end++;
4012         start++;
4013         /* put in format the printf format (the rule without the range) */
4014         sprintf(format, "%s%%.4x%s", rule, end);
4015
4016         array_idx = 0;
4017         array_sz = 0;
4018         *array = NULL;
4019         /* loop on , separator */
4020         do {
4021                 /* extract the 3 fields */
4022                 rc = sscanf(start, "%x-%x/%u", &lo, &hi, &step);
4023                 switch (rc) {
4024                 case 0:
4025                         goto err;
4026                 case 1: {
4027                         void *tmp;
4028
4029                         array_sz++;
4030                         tmp = realloc(*array, array_sz * sizeof(int));
4031                         if (tmp == NULL)
4032                                 goto err;
4033                         *array = tmp;
4034                         (*array)[array_idx] = lo;
4035                         array_idx++;
4036                         break;
4037                 }
4038                 case 2: {
4039                         step = 1;
4040                         /* do not break to share code with case 3: */
4041                 }
4042                 case 3: {
4043                         void *tmp;
4044
4045                         if ((hi < lo) || (step == 0))
4046                                 goto err;
4047                         array_sz += (hi - lo) / step + 1;
4048                         tmp = realloc(*array, array_sz * sizeof(int));
4049                         if (tmp == NULL)
4050                                 goto err;
4051                         *array = tmp;
4052                         for (i = lo; i <= hi; i+=step, array_idx++)
4053                                 (*array)[array_idx] = i;
4054                         break;
4055                 }
4056                 }
4057                 ptr = strchr(start, ',');
4058                 if (ptr != NULL)
4059                         start = ptr + 1;
4060
4061         } while (ptr != NULL);
4062         return array_sz;
4063 err:
4064         if (*array != NULL) {
4065                 free(*array);
4066                 *array = NULL;
4067         }
4068         return 0;
4069 }
4070
4071 static int extract_fsname_poolname(const char *arg, char *fsname,
4072                                    char *poolname)
4073 {
4074         char *ptr;
4075         int rc;
4076
4077         strlcpy(fsname, arg, PATH_MAX + 1);
4078         ptr = strchr(fsname, '.');
4079         if (ptr == NULL) {
4080                 fprintf(stderr, ". is missing in %s\n", fsname);
4081                 rc = -EINVAL;
4082                 goto err;
4083         }
4084
4085         if ((ptr - fsname) == 0) {
4086                 fprintf(stderr, "fsname is empty\n");
4087                 rc = -EINVAL;
4088                 goto err;
4089         }
4090
4091         *ptr = '\0';
4092         ++ptr;
4093
4094         rc = lustre_is_fsname_valid(fsname, 1, LUSTRE_MAXFSNAME);
4095         if (rc < 0) {
4096                 fprintf(stderr, "filesystem name %s must be 1-%d chars\n",
4097                         fsname, LUSTRE_MAXFSNAME);
4098                 rc = -EINVAL;
4099                 goto err;
4100         } else if (rc > 0) {
4101                 fprintf(stderr, "char '%c' not allowed in filesystem name\n",
4102                         rc);
4103                 rc = -EINVAL;
4104                 goto err;
4105         }
4106
4107         rc = lustre_is_poolname_valid(ptr, 1, LOV_MAXPOOLNAME);
4108         if (rc == -1) {
4109                 fprintf(stderr, "poolname is empty\n");
4110                 rc = -EINVAL;
4111                 goto err;
4112         } else if (rc == -2) {
4113                 fprintf(stderr,
4114                         "poolname %s is too long (max is %d)\n",
4115                         ptr, LOV_MAXPOOLNAME);
4116                 rc = -ENAMETOOLONG;
4117                 goto err;
4118         } else if (rc > 0) {
4119                 fprintf(stderr, "char '%c' not allowed in pool name '%s'\n",
4120                         rc, ptr);
4121                 rc = -EINVAL;
4122                 goto err;
4123         }
4124
4125         strncpy(poolname, ptr, LOV_MAXPOOLNAME);
4126         poolname[LOV_MAXPOOLNAME] = '\0';
4127         return 0;
4128
4129 err:
4130         fprintf(stderr, "argument %s must be <fsname>.<poolname>\n", arg);
4131         return rc;
4132 }
4133
4134 int jt_pool_cmd(int argc, char **argv)
4135 {
4136         enum lcfg_command_type cmd;
4137         char fsname[PATH_MAX + 1];
4138         char poolname[LOV_MAXPOOLNAME + 1];
4139         char *ostnames_buf = NULL;
4140         int i, rc;
4141         int *array = NULL, array_sz;
4142         struct {
4143                 int     rc;
4144                 char   *ostname;
4145         } *cmds = NULL;
4146
4147         switch (argc) {
4148         case 0:
4149         case 1: return CMD_HELP;
4150         case 2: {
4151                 if (strcmp("pool_new", argv[0]) == 0)
4152                         cmd = LCFG_POOL_NEW;
4153                 else if (strcmp("pool_destroy", argv[0]) == 0)
4154                         cmd = LCFG_POOL_DEL;
4155                 else if (strcmp("pool_list", argv[0]) == 0)
4156                          return llapi_poollist(argv[1]);
4157                 else return CMD_HELP;
4158
4159                 rc = extract_fsname_poolname(argv[1], fsname, poolname);
4160                 if (rc)
4161                         break;
4162
4163                 rc = pool_cmd(cmd, argv[0], argv[1], fsname, poolname, NULL);
4164                 if (rc)
4165                         break;
4166
4167                 check_pool_cmd_result(cmd, fsname, poolname, NULL);
4168                 break;
4169         }
4170         default: {
4171                 char format[2*MAX_OBD_NAME];
4172
4173                 if (strcmp("pool_remove", argv[0]) == 0) {
4174                         cmd = LCFG_POOL_REM;
4175                 } else if (strcmp("pool_add", argv[0]) == 0) {
4176                         cmd = LCFG_POOL_ADD;
4177                 } else {
4178                         return CMD_HELP;
4179                 }
4180
4181                 rc = extract_fsname_poolname(argv[1], fsname, poolname);
4182                 if (rc)
4183                         break;
4184
4185                 for (i = 2; i < argc; i++) {
4186                         int j;
4187
4188                         array_sz = get_array_idx(argv[i], format, &array);
4189                         if (array_sz == 0)
4190                                 return CMD_HELP;
4191
4192                         cmds = malloc(array_sz * sizeof(cmds[0]));
4193                         if (cmds != NULL) {
4194                                 ostnames_buf = malloc(array_sz *
4195                                                       (MAX_OBD_NAME + 1));
4196                         } else {
4197                                 free(array);
4198                                 rc = -ENOMEM;
4199                                 goto out;
4200                         }
4201
4202                         for (j = 0; j < array_sz; j++) {
4203                                 char ostname[MAX_OBD_NAME + 1];
4204
4205                                 snprintf(ostname, MAX_OBD_NAME, format,
4206                                          array[j]);
4207                                 ostname[MAX_OBD_NAME] = '\0';
4208
4209                                 rc = check_and_complete_ostname(fsname,ostname);
4210                                 if (rc) {
4211                                         free(array);
4212                                         free(cmds);
4213                                         if (ostnames_buf)
4214                                                 free(ostnames_buf);
4215                                         goto out;
4216                                 }
4217                                 if (ostnames_buf != NULL) {
4218                                         cmds[j].ostname =
4219                                           &ostnames_buf[(MAX_OBD_NAME + 1) * j];
4220                                         strcpy(cmds[j].ostname, ostname);
4221                                 } else {
4222                                         cmds[j].ostname = NULL;
4223                                 }
4224                                 cmds[j].rc = pool_cmd(cmd, argv[0], argv[1],
4225                                                       fsname, poolname,
4226                                                       ostname);
4227                                 /* Return an err if any of the add/dels fail */
4228                                 if (!rc)
4229                                         rc = cmds[j].rc;
4230                         }
4231                         for (j = 0; j < array_sz; j++) {
4232                                 if (!cmds[j].rc) {
4233                                         char ostname[MAX_OBD_NAME + 1];
4234
4235                                         if (!cmds[j].ostname) {
4236                                                 snprintf(ostname, MAX_OBD_NAME,
4237                                                          format, array[j]);
4238                                                 ostname[MAX_OBD_NAME] = '\0';
4239                                                 check_and_complete_ostname(
4240                                                         fsname, ostname);
4241                                         } else {
4242                                                 strcpy(ostname,
4243                                                        cmds[j].ostname);
4244                                         }
4245                                         check_pool_cmd_result(cmd, fsname,
4246                                                               poolname,ostname);
4247                                 }
4248                         }
4249                         if (array_sz > 0)
4250                                 free(array);
4251                         if (cmds)
4252                                 free(cmds);
4253                         if (ostnames_buf != NULL)
4254                                 free(ostnames_buf);
4255                 }
4256                 /* fall through */
4257         }
4258         } /* switch */
4259
4260 out:
4261         if (rc != 0) {
4262                 errno = -rc;
4263                 perror(argv[0]);
4264         }
4265
4266         return rc;
4267 }
4268
4269 int jt_get_obj_version(int argc, char **argv)
4270 {
4271         struct lu_fid fid;
4272         struct obd_ioctl_data data;
4273         __u64 version, id = ULLONG_MAX, group = ULLONG_MAX;
4274         char rawbuf[MAX_IOC_BUFLEN], *buf = rawbuf, *fidstr;
4275         int rc, c;
4276
4277         while ((c = getopt(argc, argv, "i:g:")) != -1) {
4278                 switch (c) {
4279                 case 'i':
4280                         id = strtoull(optarg, NULL, 0);
4281                         break;
4282                 case 'g':
4283                         group = strtoull(optarg, NULL, 0);
4284                         break;
4285                 default:
4286                         return CMD_HELP;
4287                 }
4288         }
4289
4290         argc -= optind;
4291         argv += optind;
4292
4293         if (!(id != ULLONG_MAX && group != ULLONG_MAX && argc == 0) &&
4294             !(id == ULLONG_MAX && group == ULLONG_MAX && argc == 1))
4295                 return CMD_HELP;
4296
4297         memset(&data, 0, sizeof data);
4298         data.ioc_dev = cur_device;
4299         if (argc == 1) {
4300                 fidstr = *argv;
4301                 while (*fidstr == '[')
4302                         fidstr++;
4303                 sscanf(fidstr, SFID, RFID(&fid));
4304
4305                 data.ioc_inlbuf1 = (char *) &fid;
4306                 data.ioc_inllen1 = sizeof fid;
4307         } else {
4308                 data.ioc_inlbuf3 = (char *) &id;
4309                 data.ioc_inllen3 = sizeof id;
4310                 data.ioc_inlbuf4 = (char *) &group;
4311                 data.ioc_inllen4 = sizeof group;
4312         }
4313         data.ioc_inlbuf2 = (char *) &version;
4314         data.ioc_inllen2 = sizeof version;
4315
4316         memset(buf, 0, sizeof *buf);
4317         rc = obd_ioctl_pack(&data, &buf, sizeof rawbuf);
4318         if (rc) {
4319                 fprintf(stderr, "error: %s: packing ioctl arguments: %s\n",
4320                         jt_cmdname(argv[0]), strerror(-rc));
4321                 return rc;
4322         }
4323
4324         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_GET_OBJ_VERSION, buf);
4325         if (rc == -1) {
4326                 fprintf(stderr, "error: %s: ioctl: %s\n",
4327                         jt_cmdname(argv[0]), strerror(errno));
4328                 return -errno;
4329         }
4330
4331         obd_ioctl_unpack(&data, buf, sizeof rawbuf);
4332         printf("%#jx\n", (uintmax_t)version);
4333         return 0;
4334 }
4335
4336 int jt_changelog_register(int argc, char **argv)
4337 {
4338         struct obd_ioctl_data    data = { 0 };
4339         char                     rawbuf[MAX_IOC_BUFLEN] = "";
4340         char                    *buf = rawbuf;
4341         char                    *device = lcfg_get_devname();
4342         bool                     print_name_only = false;
4343         int                      c;
4344         int                      rc;
4345
4346         if (argc > 2)
4347                 return CMD_HELP;
4348
4349         while ((c = getopt(argc, argv, "hn")) >= 0) {
4350                 switch (c) {
4351                 case 'n':
4352                         print_name_only = true;
4353                         break;
4354                 case 'h':
4355                 default:
4356                         return CMD_HELP;
4357                 }
4358         }
4359
4360         if (cur_device < 0 || device == NULL)
4361                 return CMD_HELP;
4362
4363         data.ioc_dev = cur_device;
4364
4365         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
4366         if (rc < 0) {
4367                 fprintf(stderr, "error: %s: cannot pack ioctl: %s\n",
4368                         jt_cmdname(argv[0]), strerror(-rc));
4369                return rc;
4370         }
4371
4372         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_CHANGELOG_REG, buf);
4373         if (rc < 0) {
4374                 rc = -errno;
4375                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
4376                         strerror(-rc));
4377                 return rc;
4378         }
4379
4380         obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
4381
4382         if (data.ioc_u32_1 == 0) {
4383                 fprintf(stderr, "received invalid userid!\n");
4384                 return -EPROTO;
4385         }
4386
4387         if (print_name_only)
4388                 printf("%s%u\n", CHANGELOG_USER_PREFIX, data.ioc_u32_1);
4389         else
4390                 printf("%s: Registered changelog userid '%s%u'\n",
4391                        device, CHANGELOG_USER_PREFIX, data.ioc_u32_1);
4392
4393         return 0;
4394 }
4395
4396 int jt_changelog_deregister(int argc, char **argv)
4397 {
4398         struct obd_ioctl_data    data = { 0 };
4399         char                     rawbuf[MAX_IOC_BUFLEN] = "";
4400         char                    *buf = rawbuf;
4401         char                    *device = lcfg_get_devname();
4402         int                      id;
4403         int                      rc;
4404
4405         if (argc != 2 || cur_device < 0 || device == NULL)
4406                 return CMD_HELP;
4407
4408         rc = sscanf(argv[1], CHANGELOG_USER_PREFIX"%d", &id);
4409         if (rc != 1 || id <= 0) {
4410                 fprintf(stderr,
4411                         "error: %s: expected id of the form %s<num> got '%s'\n",
4412                         jt_cmdname(argv[0]), CHANGELOG_USER_PREFIX, argv[1]);
4413                 return CMD_HELP;
4414         }
4415
4416         data.ioc_dev = cur_device;
4417         data.ioc_u32_1 = id;
4418
4419         rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf));
4420         if (rc < 0) {
4421                 fprintf(stderr, "error: %s: invalid ioctl\n",
4422                         jt_cmdname(argv[0]));
4423                 return rc;
4424         }
4425
4426         rc = l2_ioctl(OBD_DEV_ID, OBD_IOC_CHANGELOG_DEREG, buf);
4427         if (rc < 0) {
4428                 fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
4429                         strerror(rc = errno));
4430                 return rc;
4431         }
4432
4433         obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
4434         printf("%s: Deregistered changelog user '%s%u'\n",
4435                device, CHANGELOG_USER_PREFIX, data.ioc_u32_1);
4436
4437         return 0;
4438 }