Whamcloud - gitweb
d63b1cfb75343c28c3752b8ad07e26373a12e743
[fs/lustre-release.git] / 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