lustre/utils/obd.c

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