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