lustre/utils/obdctl.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  *  Copyright (C) 2002 Cluster File Systems, Inc.
   5  *   Author: Peter J. Braam <braam@clusterfs.com>
   6  *   Author: Phil Schwan <phil@clusterfs.com>
   7  *
   8  *   This file is part of Lustre, http://www.lustre.org.
   9  *
  10  *   Lustre is free software; you can redistribute it and/or
  11  *   modify it under the terms of version 2 of the GNU General Public
  12  *   License as published by the Free Software Foundation.
  13  *
  14  *   Lustre is distributed in the hope that it will be useful,
  15  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  *   GNU General Public License for more details.
  18  *
  19  *   You should have received a copy of the GNU General Public License
  20  *   along with Lustre; if not, write to the Free Software
  21  *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  22  *
  23  */
  24
  25 #include <stdlib.h>
  26 #include <sys/ioctl.h>
  27 #include <fcntl.h>
  28 #include <sys/socket.h>
  29 #include <sys/types.h>
  30 #include <sys/wait.h>
  31 #include <sys/stat.h>
  32 #include <stdio.h>
  33 #include <signal.h>
  34 #define printk printf
  35
  36 #include <linux/lustre_lib.h>
  37 #include <linux/lustre_idl.h>
  38 #include <linux/lustre_dlm.h>
  39
  40 #include <unistd.h>
  41 #include <sys/un.h>
  42 #include <time.h>
  43 #include <sys/time.h>
  44 #include <netinet/in.h>
  45 #include <errno.h>
  46 #include <string.h>
  47
  48 #define __KERNEL__
  49 #include <linux/list.h>
  50 #undef __KERNEL__
  51
  52 #include "parser.h"
  53 #include <stdio.h>
  54
  55 int fd = -1;
  56 int connid = -1;
  57 char rawbuf[8192];
  58 char *buf = rawbuf;
  59 int max = 8192;
  60 int thread;
  61
  62 #define IOCINIT(data)                                                   \
  63 do {                                                                    \
  64         memset(&data, 0, sizeof(data));                                 \
  65         data.ioc_version = OBD_IOCTL_VERSION;                           \
  66         data.ioc_conn1 = connid;                                        \
  67         data.ioc_len = sizeof(data);                                    \
  68         if (fd < 0) {                                                   \
  69                 fprintf(stderr, "No device open, use device\n");        \
  70                 return 1;                                               \
  71         }                                                               \
  72 } while (0)
  73
  74 /*
  75     pack "LL LL LL LL LL LL LL L L L L L L L L L a60 a60 L L L",
  76     $obdo->{id}, 0,
  77     $obdo->{gr}, 0,
  78     $obdo->{atime}, 0,
  79     $obdo->{mtime}, 0 ,
  80     $obdo->{ctime}, 0,
  81     $obdo->{size}, 0,
  82     $obdo->{blocks}, 0,
  83     $obdo->{blksize},
  84     $obdo->{mode},
  85     $obdo->{uid},
  86     $obdo->{gid},
  87     $obdo->{flags},
  88     $obdo->{obdflags},
  89     $obdo->{nlink},
  90     $obdo->{generation},
  91     $obdo->{valid},
  92     $obdo->{inline},
  93     $obdo->{obdmd},
  94     0, 0, # struct list_head
  95     0;  #  struct obd_ops
  96 }
  97
  98 */
  99
 100 char * obdo_print(struct obdo *obd)
 101 {
 102         char buf[1024];
 103
 104         sprintf(buf, "id: %Ld\ngrp: %Ld\natime: %Ld\nmtime: %Ld\nctime: %Ld\n"
 105                 "size: %Ld\nblocks: %Ld\nblksize: %d\nmode: %o\nuid: %d\n"
 106                 "gid: %d\nflags: %x\nobdflags: %x\nnlink: %d,\nvalid %x\n",
 107                 obd->o_id,
 108                 obd->o_gr,
 109                 obd->o_atime,
 110                 obd->o_mtime,
 111                 obd->o_ctime,
 112                 obd->o_size,
 113                 obd->o_blocks,
 114                 obd->o_blksize,
 115                 obd->o_mode,
 116                 obd->o_uid,
 117                 obd->o_gid,
 118                 obd->o_flags,
 119                 obd->o_obdflags,
 120                 obd->o_nlink,
 121                 obd->o_valid);
 122         return strdup(buf);
 123 }
 124
 125 static char *cmdname(char *func)
 126 {
 127         static char buf[512];
 128
 129         if (thread) {
 130                 sprintf(buf, "%s-%d", func, thread);
 131                 return buf;
 132         }
 133
 134         return func;
 135 }
 136
 137 #define difftime(a, b)                                          \
 138         ((double)(a)->tv_sec - (b)->tv_sec +                    \
 139          ((double)((a)->tv_usec - (b)->tv_usec) / 1000000))
 140
 141 static int be_verbose(int verbose, struct timeval *next_time,
 142                       int num, int *next_num, int num_total)
 143 {
 144         struct timeval now;
 145
 146         if (!verbose)
 147                 return 0;
 148
 149         if (next_time != NULL)
 150                 gettimeofday(&now, NULL);
 151
 152         /* A positive verbosity means to print every X iterations */
 153         if (verbose > 0 &&
 154             (next_num == NULL || num >= *next_num || num >= num_total)) {
 155                 *next_num += verbose;
 156                 if (next_time) {
 157                         next_time->tv_sec = now.tv_sec - verbose;
 158                         next_time->tv_usec = now.tv_usec;
 159                 }
 160                 return 1;
 161         }
 162
 163         /* A negative verbosity means to print at most each X seconds */
 164         if (verbose < 0 && next_time != NULL && difftime(&now, next_time) >= 0){
 165                 next_time->tv_sec = now.tv_sec - verbose;
 166                 next_time->tv_usec = now.tv_usec;
 167                 if (next_num)
 168                         *next_num = num;
 169                 return 1;
 170         }
 171
 172         return 0;
 173 }
 174
 175 static int get_verbose(const char *arg)
 176 {
 177         int verbose;
 178
 179         if (!arg || arg[0] == 'v')
 180                 verbose = 1;
 181         else if (arg[0] == 's' || arg[0] == 'q')
 182                 verbose = 0;
 183         else
 184                 verbose = strtoul(arg, NULL, 0);
 185
 186         return verbose;
 187 }
 188
 189 static int do_disconnect(char *func, int verbose)
 190 {
 191         struct obd_ioctl_data data;
 192         int rc;
 193
 194         if (connid == -1)
 195                 return 0;
 196
 197         IOCINIT(data);
 198
 199         rc = ioctl(fd, OBD_IOC_DISCONNECT , &data);
 200         if (rc < 0) {
 201                 fprintf(stderr, "error: %s: %x %s\n", cmdname(func),
 202                         OBD_IOC_DISCONNECT, strerror(errno));
 203         } else {
 204                 if (verbose)
 205                         printf("%s: disconnected connid %d\n", cmdname(func),
 206                                connid);
 207                 connid = -1;
 208         }
 209
 210         return rc;
 211 }
 212
 213 extern command_t cmdlist[];
 214
 215 static int jt_device(int argc, char **argv)
 216 {
 217         struct obd_ioctl_data data;
 218         int rc;
 219
 220         do_disconnect(argv[0], 1);
 221
 222         memset(&data, 0, sizeof(data));
 223         if ( argc != 2 ) {
 224                 fprintf(stderr, "usage: %s devno\n", cmdname(argv[0]));
 225                 return -1;
 226         }
 227
 228         data.ioc_dev = strtoul(argv[1], NULL, 0);
 229
 230         if (obd_ioctl_pack(&data, &buf, max)) {
 231                 fprintf(stderr, "error: %s: invalid ioctl\n", cmdname(argv[0]));
 232                 return -2;
 233         }
 234
 235         if (fd == -1)
 236                 fd = open("/dev/obd", O_RDWR);
 237         if (fd == -1) {
 238                 fprintf(stderr, "error: %s: opening /dev/obd: %s\n",
 239                         cmdname(argv[0]), strerror(errno));
 240                 return errno;
 241         }
 242
 243         rc = ioctl(fd, OBD_IOC_DEVICE , buf);
 244         if (rc < 0)
 245                 fprintf(stderr, "error: %s: %x %s\n", cmdname(argv[0]),
 246                         OBD_IOC_DEVICE, strerror(rc = errno));
 247
 248         return rc;
 249 }
 250
 251 static int jt_connect(int argc, char **argv)
 252 {
 253         struct obd_ioctl_data data;
 254         int rc;
 255
 256         IOCINIT(data);
 257
 258         do_disconnect(argv[0], 1);
 259
 260         if (argc != 1) {
 261                 fprintf(stderr, "usage: %s\n", cmdname(argv[0]));
 262                 return -1;
 263         }
 264
 265         rc = ioctl(fd, OBD_IOC_CONNECT , &data);
 266         if (rc < 0)
 267                 fprintf(stderr, "error: %s: %x %s\n", cmdname(argv[0]),
 268                         OBD_IOC_CONNECT, strerror(rc = errno));
 269         else
 270                 connid = data.ioc_conn1;
 271
 272         return rc;
 273 }
 274
 275 static int jt_disconnect(int argc, char **argv)
 276 {
 277         if (argc != 1) {
 278                 fprintf(stderr, "usage: %s\n", cmdname(argv[0]));
 279                 return -1;
 280         }
 281
 282         return do_disconnect(argv[0], 0);
 283 }
 284
 285 static int jt__device(int argc, char **argv)
 286 {
 287         char *arg2[3];
 288         int rc, ret;
 289
 290         if (argc < 3) {
 291                 fprintf(stderr, "usage: %s devno <command [args ...]>\n",
 292                         cmdname(argv[0]));
 293                 return -1;
 294         }
 295
 296         arg2[0] = "device";
 297         arg2[1] = argv[1];
 298         arg2[2] = NULL;
 299         rc = jt_device(2, arg2);
 300
 301         if (!rc) {
 302                 arg2[0] = "connect";
 303                 arg2[1] = NULL;
 304                 rc = jt_connect(1, arg2);
 305         }
 306
 307         if (!rc)
 308                 rc = Parser_execarg(argc - 2, argv + 2, cmdlist);
 309
 310         ret = do_disconnect(argv[0], 0);
 311         if (!rc)
 312                 rc = ret;
 313
 314         return rc;
 315 }
 316
 317 static int jt__threads(int argc, char **argv)
 318 {
 319         int threads, next_thread;
 320         int verbose;
 321         int i, j;
 322         int rc;
 323
 324         if (argc < 5) {
 325                 fprintf(stderr,
 326                         "usage: %s numthreads verbose devno <cmd [args ...]>\n",
 327                         argv[0]);
 328                 return -1;
 329         }
 330
 331         threads = strtoul(argv[1], NULL, 0);
 332
 333         verbose = get_verbose(argv[2]);
 334
 335         printf("%s: starting %d threads on device %s running %s\n",
 336                argv[0], threads, argv[3], argv[4]);
 337
 338         for (i = 1, next_thread = verbose; i <= threads; i++) {
 339                 rc = fork();
 340                 if (rc < 0) {
 341                         fprintf(stderr, "error: %s: #%d - %s\n", argv[0], i,
 342                                 strerror(rc = errno));
 343                         break;
 344                 } else if (rc == 0) {
 345                         thread = i;
 346                         argv[2] = "--device";
 347                         return jt__device(argc - 2, argv + 2);
 348                 } else if (be_verbose(verbose, NULL, i, &next_thread, threads))
 349                         printf("%s: thread #%d (PID %d) started\n",
 350                                argv[0], i, rc);
 351                 rc = 0;
 352         }
 353
 354         if (!thread) { /* parent process */
 355                 if (!verbose)
 356                         printf("%s: started %d threads\n\n", argv[0], i - 1);
 357                 else
 358                         printf("\n");
 359
 360                 for (j = 1; j < i; j++) {
 361                         int status;
 362                         int ret = wait(&status);
 363
 364                         if (ret < 0) {
 365                                 fprintf(stderr, "error: %s: wait - %s\n",
 366                                         argv[0], strerror(errno));
 367                                 if (!rc)
 368                                         rc = errno;
 369                         } else {
 370                                 /*
 371                                  * This is a hack.  We _should_ be able to use
 372                                  * WIFEXITED(status) to see if there was an
 373                                  * error, but it appears to be broken and it
 374                                  * always returns 1 (OK).  See wait(2).
 375                                  */
 376                                 int err = WEXITSTATUS(status);
 377                                 if (err)
 378                                         fprintf(stderr,
 379                                                 "%s: PID %d had rc=%d\n",
 380                                                 argv[0], ret, err);
 381                                 if (!rc)
 382                                         rc = err;
 383                         }
 384                 }
 385         }
 386
 387         return rc;
 388 }
 389
 390 static int jt_detach(int argc, char **argv)
 391 {
 392         struct obd_ioctl_data data;
 393         int rc;
 394
 395         IOCINIT(data);
 396
 397         if (argc != 1) {
 398                 fprintf(stderr, "usage: %s\n", cmdname(argv[0]));
 399                 return -1;
 400         }
 401
 402         if (obd_ioctl_pack(&data, &buf, max)) {
 403                 fprintf(stderr, "error: %s: invalid ioctl\n", cmdname(argv[0]));
 404                 return -2;
 405         }
 406
 407         rc = ioctl(fd, OBD_IOC_DETACH , buf);
 408         if (rc < 0)
 409                 fprintf(stderr, "error: %s: %s\n", cmdname(argv[0]),
 410                         strerror(rc=errno));
 411
 412         return rc;
 413 }
 414
 415 static int jt_cleanup(int argc, char **argv)
 416 {
 417         struct obd_ioctl_data data;
 418         int rc;
 419
 420         IOCINIT(data);
 421
 422         if (argc != 1) {
 423                 fprintf(stderr, "usage: %s\n", cmdname(argv[0]));
 424                 return -1;
 425         }
 426
 427         rc = ioctl(fd, OBD_IOC_CLEANUP , &data);
 428         if (rc < 0)
 429                 fprintf(stderr, "error: %s: %s\n", cmdname(argv[0]),
 430                         strerror(rc=errno));
 431
 432         return rc;
 433 }
 434
 435 static int jt_attach(int argc, char **argv)
 436 {
 437         struct obd_ioctl_data data;
 438         int rc;
 439
 440         IOCINIT(data);
 441
 442         if (argc != 2 && argc != 3) {
 443                 fprintf(stderr, "usage: %s type [name [uuid]]\n", cmdname(argv[0]));
 444                 return -1;
 445         }
 446
 447         data.ioc_inllen1 =  strlen(argv[1]) + 1;
 448         data.ioc_inlbuf1 = argv[1];
 449         if (argc == 3) {
 450                 data.ioc_inllen2 = strlen(argv[2]) + 1;
 451                 data.ioc_inlbuf2 = argv[2];
 452         }
 453
 454         printf("%s: len %d addr %p type %s data %s\n",
 455                cmdname(argv[0]), data.ioc_len, buf,
 456                MKSTR(data.ioc_inlbuf1), MKSTR(data.ioc_inlbuf2));
 457
 458         if (obd_ioctl_pack(&data, &buf, max)) {
 459                 fprintf(stderr, "error: %s: invalid ioctl\n", cmdname(argv[0]));
 460                 return -2;
 461         }
 462         printf("%s: len %d addr %p raw %p type %s data %s and %s\n",
 463                cmdname(argv[0]), data.ioc_len, buf, rawbuf,
 464                MKSTR(data.ioc_inlbuf1), MKSTR(data.ioc_inlbuf2), &buf[516]);
 465
 466         rc = ioctl(fd, OBD_IOC_ATTACH , buf);
 467         if (rc < 0)
 468                 fprintf(stderr, "error: %s: %x %s\n", cmdname(argv[0]),
 469                         OBD_IOC_ATTACH, strerror(rc = errno));
 470         else if (argc == 3) {
 471                 char name[1024];
 472                 if (strlen(argv[2]) > 128) {
 473                         printf("Name too long to set environment\n");
 474                         return -EINVAL;
 475                 }
 476                 snprintf(name, 512, "LUSTRE-dev-%s", argv[2]);
 477                 rc = setenv(name, argv[1], 1);
 478                 if (rc) {
 479                         printf("error setting env variable %s\n", name);
 480                 }
 481         }
 482
 483         return rc;
 484 }
 485
 486 static int jt_name2dev(int argc, char **argv)
 487 {
 488         struct obd_ioctl_data data;
 489         int rc;
 490
 491         if (fd == -1)
 492                 fd = open("/dev/obd", O_RDWR);
 493         if (fd == -1) {
 494                 fprintf(stderr, "error: %s: opening /dev/obd: %s\n",
 495                         cmdname(argv[0]), strerror(errno));
 496                 return errno;
 497         }
 498
 499         IOCINIT(data);
 500
 501         if (argc != 2) {
 502                 fprintf(stderr, "usage: %s name\n", cmdname(argv[0]));
 503                 return -1;
 504         }
 505
 506         data.ioc_inllen1 =  strlen(argv[1]) + 1;
 507         data.ioc_inlbuf1 = argv[1];
 508
 509
 510         if (obd_ioctl_pack(&data, &buf, max)) {
 511                 fprintf(stderr, "error: %s: invalid ioctl\n", cmdname(argv[0]));
 512                 return -2;
 513         }
 514         rc = ioctl(fd, OBD_IOC_NAME2DEV , buf);
 515         if (rc < 0)
 516                 fprintf(stderr, "error: %s: %x %s\n", cmdname(argv[0]),
 517                         OBD_IOC_NAME2DEV, strerror(rc = errno));
 518         memcpy((char *)(&data), buf, sizeof(data));
 519         printf("%d\n", data.ioc_dev);
 520         return rc;
 521 }
 522
 523 static int jt_setup(int argc, char **argv)
 524 {
 525         struct obd_ioctl_data data;
 526         int rc;
 527
 528         IOCINIT(data);
 529
 530         if ( argc > 3) {
 531                 fprintf(stderr, "usage: %s [device] [fstype]\n",
 532                         cmdname(argv[0]));
 533                 return -1;
 534         }
 535
 536         if (argc > 1) {
 537                 data.ioc_inllen1 =  strlen(argv[1]) + 1;
 538                 data.ioc_inlbuf1 = argv[1];
 539                 data.ioc_dev = strtoul(argv[1], NULL, 0);
 540         } else {
 541                 data.ioc_dev = -1;
 542         }
 543         if ( argc == 3 ) {
 544                 data.ioc_inllen2 = strlen(argv[2]) + 1;
 545                 data.ioc_inlbuf2 = argv[2];
 546         }
 547
 548         printf("%s: len %d addr %p device %s type %s\n",
 549                cmdname(argv[0]), data.ioc_len, buf,
 550                MKSTR(data.ioc_inlbuf1), MKSTR(data.ioc_inlbuf2));
 551
 552         if (obd_ioctl_pack(&data, &buf, max)) {
 553                 fprintf(stderr, "error: %s: invalid ioctl\n", cmdname(argv[0]));
 554                 return -2;
 555         }
 556         printf("%s: len %d addr %p raw %p device %s type %s\n",
 557                cmdname(argv[0]), data.ioc_len, buf, rawbuf,
 558                MKSTR(data.ioc_inlbuf1), MKSTR(data.ioc_inlbuf2));
 559
 560         rc = ioctl(fd, OBD_IOC_SETUP , buf);
 561         if (rc < 0)
 562                 fprintf(stderr, "error: %s: %s\n", cmdname(argv[0]),
 563                         strerror(rc = errno));
 564
 565         return rc;
 566 }
 567
 568
 569 static int jt_create(int argc, char **argv)
 570 {
 571         struct obd_ioctl_data data;
 572         struct timeval next_time;
 573         int count = 1, next_count;
 574         int verbose;
 575         int i;
 576         int rc;
 577
 578         IOCINIT(data);
 579         if (argc < 2 || argc > 4) {
 580                 fprintf(stderr, "usage: %s num [mode] [verbose]\n",
 581                         cmdname(argv[0]));
 582                 return -1;
 583         }
 584         count = strtoul(argv[1], NULL, 0);
 585
 586         if (argc > 2)
 587                 data.ioc_obdo1.o_mode = strtoul(argv[2], NULL, 0);
 588         else
 589                 data.ioc_obdo1.o_mode = 0100644;
 590         data.ioc_obdo1.o_valid = OBD_MD_FLMODE;
 591
 592         verbose = get_verbose(argv[3]);
 593
 594         printf("%s: %d obdos\n", cmdname(argv[0]), count);
 595         gettimeofday(&next_time, NULL);
 596         next_time.tv_sec -= verbose;
 597
 598         for (i = 1, next_count = verbose; i <= count ; i++) {
 599                 rc = ioctl(fd, OBD_IOC_CREATE , &data);
 600                 if (rc < 0) {
 601                         fprintf(stderr, "error: %s: #%d - %s\n",
 602                                 cmdname(argv[0]), i, strerror(rc = errno));
 603                         break;
 604                 }
 605                 if (be_verbose(verbose, &next_time, i, &next_count, count))
 606                         printf("%s: #%d is object id %Ld\n", cmdname(argv[0]),
 607                                i, data.ioc_obdo1.o_id);
 608         }
 609         return rc;
 610 }
 611
 612 static int jt_setattr(int argc, char **argv)
 613 {
 614         struct obd_ioctl_data data;
 615         int rc;
 616
 617         IOCINIT(data);
 618         if (argc != 2) {
 619                 fprintf(stderr, "usage: %s id mode\n", cmdname(argv[0]));
 620                 return -1;
 621         }
 622
 623         data.ioc_obdo1.o_id = strtoul(argv[1], NULL, 0);
 624         data.ioc_obdo1.o_mode = S_IFREG | strtoul(argv[2], NULL, 0);
 625         data.ioc_obdo1.o_valid = OBD_MD_FLMODE;
 626
 627         rc = ioctl(fd, OBD_IOC_SETATTR , &data);
 628         if (rc < 0)
 629                 fprintf(stderr, "error: %s: %s\n", cmdname(argv[0]),
 630                         strerror(rc = errno));
 631
 632         return rc;
 633 }
 634
 635 static int jt_destroy(int argc, char **argv)
 636 {
 637         struct obd_ioctl_data data;
 638         int rc;
 639
 640         IOCINIT(data);
 641         if (argc != 2) {
 642                 fprintf(stderr, "usage: %s id\n", cmdname(argv[0]));
 643                 return -1;
 644         }
 645
 646         data.ioc_obdo1.o_id = strtoul(argv[1], NULL, 0);
 647         data.ioc_obdo1.o_mode = S_IFREG|0644;
 648
 649         rc = ioctl(fd, OBD_IOC_DESTROY , &data);
 650         if (rc < 0)
 651                 fprintf(stderr, "error: %s: %s\n", cmdname(argv[0]),
 652                         strerror(rc = errno));
 653
 654         return rc;
 655 }
 656
 657 static int jt_getattr(int argc, char **argv)
 658 {
 659         struct obd_ioctl_data data;
 660         int rc;
 661
 662         if (argc != 2) {
 663                 fprintf(stderr, "usage: %s id\n", cmdname(argv[0]));
 664                 return -1;
 665         }
 666
 667         IOCINIT(data);
 668         data.ioc_obdo1.o_id = strtoul(argv[1], NULL, 0);
 669         /* to help obd filter */
 670         data.ioc_obdo1.o_mode = 0100644;
 671         data.ioc_obdo1.o_valid = 0xffffffff;
 672         printf("%s: object id %Ld\n", cmdname(argv[0]), data.ioc_obdo1.o_id);
 673
 674         rc = ioctl(fd, OBD_IOC_GETATTR , &data);
 675         if (rc) {
 676                 fprintf(stderr, "error: %s: %s\n", cmdname(argv[0]),
 677                         strerror(rc=errno));
 678         } else {
 679                 printf("%s: object id %Ld, mode %o\n", cmdname(argv[0]),
 680                        data.ioc_obdo1.o_id, data.ioc_obdo1.o_mode);
 681         }
 682         return rc;
 683 }
 684
 685 static int jt_test_getattr(int argc, char **argv)
 686 {
 687         struct obd_ioctl_data data;
 688         struct timeval start, next_time;
 689         int i, count, next_count;
 690         int verbose;
 691         int rc;
 692
 693         if (argc != 2 && argc != 3) {
 694                 fprintf(stderr, "usage: %s count [verbose]\n",cmdname(argv[0]));
 695                 return -1;
 696         }
 697
 698         IOCINIT(data);
 699         count = strtoul(argv[1], NULL, 0);
 700
 701         verbose = get_verbose(argv[2]);
 702
 703         data.ioc_obdo1.o_valid = 0xffffffff;
 704         data.ioc_obdo1.o_id = 2;
 705         gettimeofday(&start, NULL);
 706         next_time.tv_sec = start.tv_sec - verbose;
 707         next_time.tv_usec = start.tv_usec;
 708         printf("%s: getting %d attrs (testing only): %s", cmdname(argv[0]),
 709                count, ctime(&start.tv_sec));
 710
 711         for (i = 1, next_count = verbose; i <= count; i++) {
 712                 rc = ioctl(fd, OBD_IOC_GETATTR , &data);
 713                 if (rc < 0) {
 714                         fprintf(stderr, "error: %s: #%d - %s\n",
 715                                 cmdname(argv[0]), i, strerror(rc = errno));
 716                         break;
 717                 } else if (be_verbose(verbose, &next_time, i,&next_count,count))
 718                         printf("%s: got attr #%d\n", cmdname(argv[0]), i);
 719         }
 720
 721         if (!rc) {
 722                 struct timeval end;
 723                 double diff;
 724
 725                 gettimeofday(&end, NULL);
 726
 727                 diff = difftime(&end, &start);
 728
 729                 --i;
 730                 printf("%s: %d attrs in %.4gs (%.4g attr/s): %s",
 731                        cmdname(argv[0]), i, diff, (double)i / diff,
 732                        ctime(&end.tv_sec));
 733         }
 734         return rc;
 735 }
 736
 737 static int jt_test_brw(int argc, char **argv)
 738 {
 739         struct obd_ioctl_data data;
 740         struct timeval start, next_time;
 741         char *bulk, *b;
 742         int pages = 1, obdos = 1, count, next_count;
 743         int verbose = 1, write = 0, rw;
 744         int i, o, p;
 745         int len;
 746         int rc;
 747
 748         if (argc < 2 || argc > 6) {
 749                 fprintf(stderr,
 750                         "usage: %s count [write [verbose [pages [obdos]]]]\n",
 751                         cmdname(argv[0]));
 752                 return -1;
 753         }
 754
 755         count = strtoul(argv[1], NULL, 0);
 756
 757         if (argc >= 3) {
 758                 if (argv[2][0] == 'w' || argv[2][0] == '1')
 759                         write = 1;
 760                 else if (argv[2][0] == 'r' || argv[2][0] == '0')
 761                         write = 0;
 762
 763                 verbose = get_verbose(argv[3]);
 764         }
 765
 766         if (argc >= 5)
 767                 pages = strtoul(argv[4], NULL, 0);
 768         if (argc >= 6)
 769                 obdos = strtoul(argv[5], NULL, 0);
 770
 771         if (obdos != 1 && obdos != 2) {
 772                 fprintf(stderr, "error: %s: only 1 or 2 obdos supported\n",
 773                         cmdname(argv[0]));
 774                 return -2;
 775         }
 776
 777         len = pages * PAGE_SIZE;
 778
 779         bulk = calloc(obdos, len);
 780         if (!bulk) {
 781                 fprintf(stderr, "error: %s: no memory allocating %dx%d pages\n",
 782                         cmdname(argv[0]), obdos, pages);
 783                 return -2;
 784         }
 785         IOCINIT(data);
 786         data.ioc_conn2 = connid;
 787         data.ioc_obdo1.o_id = 2;
 788         data.ioc_count = len;
 789         data.ioc_offset = 0;
 790         data.ioc_plen1 = len;
 791         data.ioc_pbuf1 = bulk;
 792         if (obdos > 1) {
 793                 data.ioc_obdo2.o_id = 2;
 794                 data.ioc_plen2 = len;
 795                 data.ioc_pbuf2 = bulk + len;
 796         }
 797
 798         gettimeofday(&start, NULL);
 799         next_time.tv_sec = start.tv_sec - verbose;
 800         next_time.tv_usec = start.tv_usec;
 801
 802         printf("%s: %s %d (%dx%d pages) (testing only): %s",
 803                cmdname(argv[0]), write ? "writing" : "reading",
 804                count, obdos, pages, ctime(&start.tv_sec));
 805
 806         /*
 807          * We will put in the start time (and loop count inside the loop)
 808          * at the beginning of each page so that we will be able to validate
 809          * (at some later time) whether the data actually made it or not.
 810          */
 811         for (o = 0, b = bulk; o < obdos; o++)
 812                 for (p = 0; p < pages; p++, b += PAGE_SIZE)
 813                         memcpy(b, &start, sizeof(start));
 814
 815         rw = write ? OBD_IOC_BRW_WRITE : OBD_IOC_BRW_READ;
 816         for (i = 1, next_count = verbose; i <= count; i++) {
 817                 if (write) {
 818                         b = bulk + sizeof(struct timeval);
 819                         for (o = 0; o < obdos; o++)
 820                                 for (p = 0; p < pages; p++, b += PAGE_SIZE)
 821                                         memcpy(b, &count, sizeof(count));
 822                 }
 823
 824                 rc = ioctl(fd, rw, &data);
 825                 if (rc) {
 826                         fprintf(stderr, "error: %s: #%d - %s on %s\n",
 827                                 cmdname(argv[0]), i, strerror(rc = errno),
 828                                 write ? "write" : "read");
 829                         break;
 830                 } else if (be_verbose(verbose, &next_time, i,&next_count,count))
 831                         printf("%s: %s number %d\n", cmdname(argv[0]),
 832                                write ? "write" : "read", i);
 833         }
 834
 835         free(bulk);
 836
 837         if (!rc) {
 838                 struct timeval end;
 839                 double diff;
 840
 841                 gettimeofday(&end, NULL);
 842
 843                 diff = difftime(&end, &start);
 844
 845                 --i;
 846                 printf("%s: %s %dx%dx%d pages in %.4gs (%.4g pg/s): %s",
 847                        cmdname(argv[0]), write ? "wrote" : "read", obdos,
 848                        pages, i, diff, (double)obdos * i * pages / diff,
 849                        ctime(&end.tv_sec));
 850         }
 851         return rc;
 852 }
 853
 854 static int jt_test_ldlm(int argc, char **argv)
 855 {
 856         struct obd_ioctl_data data;
 857         int rc;
 858
 859         IOCINIT(data);
 860         if (argc != 1) {
 861                 fprintf(stderr, "usage: %s\n", cmdname(argv[0]));
 862                 return 1;
 863         }
 864
 865         rc = ioctl(fd, IOC_LDLM_TEST, &data);
 866         if (rc)
 867                 fprintf(stderr, "error: %s: test failed: %s\n",
 868                         cmdname(argv[0]), strerror(rc = errno));
 869         return rc;
 870 }
 871
 872 command_t cmdlist[] = {
 873         /* Metacommands */
 874         {"--device", jt__device, 0, "--device <devno> <command [args ...]>"},
 875         {"--threads", jt__threads, 0,
 876                 "--threads <threads> <devno> <command [args ...]>"},
 877
 878         /* Device configuration commands */
 879         {"device", jt_device, 0, "set current device (args device_no name)"},
 880         {"name2dev", jt_name2dev, 0, "set device by name (args name)"},
 881         {"attach", jt_attach, 0, "name the type of device (args: type data"},
 882         {"setup", jt_setup, 0, "setup device (args: <blkdev> [data]"},
 883         {"detach", jt_detach, 0, "detach the current device (arg: )"},
 884         {"cleanup", jt_cleanup, 0, "cleanup the current device (arg: )"},
 885
 886         /* Session commands */
 887         {"connect", jt_connect, 0, "connect - get a connection to device"},
 888         {"disconnect", jt_disconnect, 0,
 889                 "disconnect - break connection to device"},
 890
 891         /* Session operations */
 892         {"create", jt_create, 0, "create [count [mode [verbose]]]"},
 893         {"destroy", jt_destroy, 0, "destroy <id>"},
 894         {"getattr", jt_getattr, 0, "getattr <id>"},
 895         {"setattr", jt_setattr, 0, "setattr <id> <mode>"},
 896         {"test_getattr", jt_test_getattr, 0, "test_getattr <count> [verbose]"},
 897         {"test_brw", jt_test_brw, 0, "test_brw <count> [write [verbose]]"},
 898         {"test_ldlm", jt_test_ldlm, 0, "test lock manager (no args)"},
 899
 900         /* User interface commands */
 901         {"help", Parser_help, 0, "help"},
 902         {"exit", Parser_quit, 0, "quit"},
 903         {"quit", Parser_quit, 0, "quit"},
 904         { 0, 0, 0, NULL }
 905 };
 906
 907
 908 static void signal_server(int sig)
 909 {
 910         if (sig == SIGINT) {
 911                 do_disconnect("sigint", 1);
 912                 exit(1);
 913         } else {
 914                 fprintf(stderr, "%s: got signal %d\n", cmdname("sigint"), sig);
 915         }
 916 }
 917
 918 int main(int argc, char **argv)
 919 {
 920         struct sigaction sigact;
 921         int rc = 0;
 922
 923         sigact.sa_handler = signal_server;
 924         sigfillset(&sigact.sa_mask);
 925         sigact.sa_flags = SA_RESTART;
 926         sigaction(SIGINT, &sigact, NULL);
 927
 928
 929         if (argc > 1) {
 930                 rc = Parser_execarg(argc - 1, argv + 1, cmdlist);
 931         } else {
 932                 Parser_init("obdctl > ", cmdlist);
 933                 Parser_commands();
 934         }
 935
 936         do_disconnect(argv[0], 1);
 937         return rc;
 938 }
 939