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