lustre/utils/obd.c

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