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