Whamcloud - gitweb
4c1e2bb9ca7e19800bdbafc975713317414489ca
[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/stat.h>
31 #include <stdio.h>
32 #include <signal.h>
33 #define printk printf
34
35 #include <linux/lustre_lib.h>
36 #include <linux/lustre_idl.h>
37 #include <linux/lustre_dlm.h>
38
39 #include <unistd.h>
40 #include <sys/un.h>
41 #include <time.h>
42 #include <sys/time.h>
43 #include <netinet/in.h>
44 #include <errno.h>
45 #include <string.h>
46
47 #define __KERNEL__
48 #include <linux/list.h>
49 #undef __KERNEL__
50
51 #include "parser.h"
52 #include <stdio.h>
53
54 int fd = -1;
55 int connid = -1;
56 char rawbuf[8192];
57 char *buf = rawbuf;
58 int max = 8192;
59
60 #define IOCINIT(data)                                                   \
61 do {                                                                    \
62         memset(&data, 0, sizeof(data));                                 \
63         data.ioc_version = OBD_IOCTL_VERSION;                           \
64         data.ioc_conn1 = connid;                                        \
65         data.ioc_len = sizeof(data);                                    \
66         if (fd < 0) {                                                   \
67                 fprintf(stderr, "No device open, use device\n");        \
68                 return 1;                                               \
69         }                                                               \
70 } while (0)
71
72 /*
73     pack "LL LL LL LL LL LL LL L L L L L L L L L a60 a60 L L L",
74     $obdo->{id}, 0,
75     $obdo->{gr}, 0,
76     $obdo->{atime}, 0,
77     $obdo->{mtime}, 0 ,
78     $obdo->{ctime}, 0,
79     $obdo->{size}, 0,
80     $obdo->{blocks}, 0,
81     $obdo->{blksize},
82     $obdo->{mode},
83     $obdo->{uid},
84     $obdo->{gid},
85     $obdo->{flags},
86     $obdo->{obdflags},
87     $obdo->{nlink},
88     $obdo->{generation},
89     $obdo->{valid},
90     $obdo->{inline},
91     $obdo->{obdmd},
92     0, 0, # struct list_head
93     0;  #  struct obd_ops
94 }
95
96 */
97
98 char * obdo_print(struct obdo *obd)
99 {
100         char buf[1024];
101
102         sprintf(buf, "id: %Ld\ngrp: %Ld\natime: %Ld\nmtime: %Ld\nctime: %Ld\n"
103                 "size: %Ld\nblocks: %Ld\nblksize: %d\nmode: %o\nuid: %d\n"
104                 "gid: %d\nflags: %x\nobdflags: %x\nnlink: %d,\nvalid %x\n",
105                 obd->o_id,
106                 obd->o_gr,
107                 obd->o_atime,
108                 obd->o_mtime,
109                 obd->o_ctime,
110                 obd->o_size,
111                 obd->o_blocks,
112                 obd->o_blksize,
113                 obd->o_mode,
114                 obd->o_uid,
115                 obd->o_gid,
116                 obd->o_flags,
117                 obd->o_obdflags,
118                 obd->o_nlink,
119                 obd->o_valid);
120         return strdup(buf);
121 }
122
123 static int do_disconnect(const char *func)
124 {
125         struct obd_ioctl_data data;
126         int rc;
127
128         if (connid == -1)
129                 return 0;
130
131         IOCINIT(data);
132
133         rc = ioctl(fd, OBD_IOC_DISCONNECT , &data);
134         if (rc < 0) {
135                 fprintf(stderr, "error: %s: %x %s\n", func,
136                         OBD_IOC_DISCONNECT, strerror(errno));
137         } else {
138                 printf("Disconnected connid %d\n", connid);
139                 connid = -1;
140         }
141
142         return rc;
143 }
144
145 extern command_t cmdlist[];
146
147 static int jt_device(int argc, char **argv)
148 {
149         struct obd_ioctl_data data;
150         int rc;
151
152         do_disconnect(argv[0]);
153
154         memset(&data, 0, sizeof(data));
155         if ( argc != 2 ) {
156                 fprintf(stderr, "usage: %s devno\n", argv[0]);
157                 return -1;
158         }
159
160         data.ioc_dev = atoi(argv[1]);
161
162         if (obd_ioctl_pack(&data, &buf, max)) {
163                 fprintf(stderr, "error: %s: invalid ioctl\n", argv[0]);
164                 return -2;
165         }
166
167         if (fd == -1)
168                 fd = open("/dev/obd", O_RDWR);
169         if (fd == -1) {
170                 fprintf(stderr, "error: %s: opening /dev/obd: %s\n", argv[0],
171                         strerror(errno));
172                 return errno;
173         }
174
175         rc = ioctl(fd, OBD_IOC_DEVICE , buf);
176         if (rc < 0)
177                 fprintf(stderr, "error: %s: %x %s\n", argv[0],
178                         OBD_IOC_DEVICE, strerror(rc = errno));
179
180         return rc;
181 }
182
183 static int jt_connect(int argc, char **argv)
184 {
185         struct obd_ioctl_data data;
186         int rc;
187
188         IOCINIT(data);
189
190         do_disconnect(argv[0]);
191
192         if (argc != 1) {
193                 fprintf(stderr, "usage: %s\n", argv[0]);
194                 return -1;
195         }
196
197         rc = ioctl(fd, OBD_IOC_CONNECT , &data);
198         if (rc < 0)
199                 fprintf(stderr, "error: %s: %x %s\n", argv[0],
200                         OBD_IOC_CONNECT, strerror(rc = errno));
201         else
202                 connid = data.ioc_conn1;
203
204         return rc;
205 }
206
207 static int jt_disconnect(int argc, char **argv)
208 {
209         struct obd_ioctl_data data;
210         int rc;
211
212         IOCINIT(data);
213
214         if (argc != 1) {
215                 fprintf(stderr, "usage: %s\n", argv[0]);
216                 return -1;
217         }
218
219         rc = ioctl(fd, OBD_IOC_DISCONNECT , &data);
220         if (rc < 0)
221                 fprintf(stderr, "error: %s: %x %s\n", argv[0],
222                         OBD_IOC_DISCONNECT, strerror(rc = errno));
223         else
224                 connid = -1;
225
226         return rc;
227 }
228
229 static int jt_detach(int argc, char **argv)
230 {
231         struct obd_ioctl_data data;
232         int rc;
233
234         IOCINIT(data);
235
236         if (argc != 1) {
237                 fprintf(stderr, "usage: %s\n", argv[0]);
238                 return -1;
239         }
240
241         if (obd_ioctl_pack(&data, &buf, max)) {
242                 fprintf(stderr, "error: %s: invalid ioctl\n", argv[0]);
243                 return -2;
244         }
245
246         rc = ioctl(fd, OBD_IOC_DETACH , buf);
247         if (rc < 0)
248                 fprintf(stderr, "error: %s: %s\n", argv[0], strerror(rc=errno));
249
250         return rc;
251 }
252
253 static int jt_cleanup(int argc, char **argv)
254 {
255         struct obd_ioctl_data data;
256         int rc;
257
258         IOCINIT(data);
259
260         if (argc != 1) {
261                 fprintf(stderr, "usage: %s\n", argv[0]);
262                 return -1;
263         }
264
265         rc = ioctl(fd, OBD_IOC_CLEANUP , &data);
266         if (rc < 0)
267                 fprintf(stderr, "error: %s: %s\n", argv[0], strerror(rc=errno));
268
269         return rc;
270 }
271
272 static int jt_attach(int argc, char **argv)
273 {
274         struct obd_ioctl_data data;
275         int rc;
276
277         IOCINIT(data);
278
279         if (argc != 2 && argc != 3) {
280                 fprintf(stderr, "usage: %s type [data]\n", argv[0]);
281                 return -1;
282         }
283
284         data.ioc_inllen1 =  strlen(argv[1]) + 1;
285         data.ioc_inlbuf1 = argv[1];
286         if (argc == 3) {
287                 data.ioc_inllen2 = strlen(argv[2]) + 1;
288                 data.ioc_inlbuf2 = argv[2];
289         }
290
291         printf("attach len %d addr %p type %s data %s\n", data.ioc_len, buf,
292                MKSTR(data.ioc_inlbuf1), MKSTR(data.ioc_inlbuf2));
293
294         if (obd_ioctl_pack(&data, &buf, max)) {
295                 fprintf(stderr, "error: %s: invalid ioctl\n", argv[0]);
296                 return -2;
297         }
298         printf("attach len %d addr %p raw %p type %s data %s and %s\n",
299                data.ioc_len, buf, rawbuf,
300                MKSTR(data.ioc_inlbuf1), MKSTR(data.ioc_inlbuf2), &buf[516]);
301
302         rc = ioctl(fd, OBD_IOC_ATTACH , buf);
303         if (rc < 0)
304                 fprintf(stderr, "error: %s: %x %s\n", argv[0],
305                         OBD_IOC_ATTACH, strerror(rc = errno));
306
307         return rc;
308 }
309
310 static int jt_setup(int argc, char **argv)
311 {
312         struct obd_ioctl_data data;
313         int rc;
314
315         IOCINIT(data);
316
317         if ( argc > 3) {
318                 fprintf(stderr, "usage: %s [device] [fstype]\n", argv[0]);
319                 return -1;
320         }
321
322         if (argc > 1) {
323                 data.ioc_inllen1 =  strlen(argv[1]) + 1;
324                 data.ioc_inlbuf1 = argv[1];
325                 data.ioc_dev = strtoul(argv[1], NULL, 0);
326         } else {
327                 data.ioc_dev = -1;
328         }
329         if ( argc == 3 ) {
330                 data.ioc_inllen2 = strlen(argv[2]) + 1;
331                 data.ioc_inlbuf2 = argv[2];
332         }
333
334         printf("setup len %d addr %p device %s type %s\n", data.ioc_len, buf,
335                MKSTR(data.ioc_inlbuf1), MKSTR(data.ioc_inlbuf2));
336
337         if (obd_ioctl_pack(&data, &buf, max)) {
338                 fprintf(stderr, "error: %s: invalid ioctl\n", argv[0]);
339                 return -2;
340         }
341         printf("setup len %d addr %p raw %p device %s type %s\n", 
342                data.ioc_len, buf, rawbuf,
343                MKSTR(data.ioc_inlbuf1), MKSTR(data.ioc_inlbuf2));
344
345         rc = ioctl(fd, OBD_IOC_SETUP , buf);
346         if (rc < 0)
347                 fprintf(stderr, "error: %s: %x %s\n", argv[0],
348                         OBD_IOC_SETUP, strerror(rc = errno));
349
350         return rc;
351 }
352
353
354 static int jt_create(int argc, char **argv)
355 {
356         struct obd_ioctl_data data;
357         int num = 1;
358         int silent = 0;
359         int i;
360         int rc;
361
362         IOCINIT(data);
363         if (argc < 2 || argc > 4) {
364                 fprintf(stderr, "usage: %s num [mode] [silent]\n", argv[0]);
365                 return -1;
366         }
367         num = strtoul(argv[1], NULL, 0);
368
369         if (argc > 2)
370                 data.ioc_obdo1.o_mode = strtoul(argv[2], NULL, 0);
371         else
372                 data.ioc_obdo1.o_mode = 0100644;
373         data.ioc_obdo1.o_valid = OBD_MD_FLMODE;
374
375         if (argc > 3)
376                 silent = strtoul(argv[3], NULL, 0);
377
378         printf("Creating %d obdos\n", num);
379
380         for (i = 0 ; i < num ; i++) {
381                 rc = ioctl(fd, OBD_IOC_CREATE , &data);
382                 if (rc < 0) {
383                         fprintf(stderr, "error: %s: #%d - %x %s\n", argv[0],
384                                 i, OBD_IOC_CREATE, strerror(rc = errno));
385                         break;
386                 }
387                 if (!silent)
388                         printf("created obdo %Ld\n", data.ioc_obdo1.o_id);
389         }
390         return rc;
391 }
392
393 static int jt_setattr(int argc, char **argv)
394 {
395         struct obd_ioctl_data data;
396         int rc;
397
398         IOCINIT(data);
399         if (argc != 2) {
400                 fprintf(stderr, "usage: %s id mode\n", argv[0]);
401                 return -1;
402         }
403
404         data.ioc_obdo1.o_id = strtoul(argv[1], NULL, 0);
405         data.ioc_obdo1.o_mode = S_IFREG | strtoul(argv[2], NULL, 0);
406         data.ioc_obdo1.o_valid = OBD_MD_FLMODE; 
407
408         rc = ioctl(fd, OBD_IOC_SETATTR , &data);
409         if (rc < 0)
410                 fprintf(stderr, "error: %s: %x %s\n", argv[0],
411                         OBD_IOC_SETATTR, strerror(rc = errno));
412
413         return rc;
414 }
415
416 static int jt_destroy(int argc, char **argv)
417 {
418         struct obd_ioctl_data data;
419         int rc;
420
421         IOCINIT(data);
422         if (argc != 2) {
423                 fprintf(stderr, "usage: %s id\n", argv[0]);
424                 return -1;
425         }
426
427         data.ioc_obdo1.o_id = strtoul(argv[1], NULL, 0);
428         data.ioc_obdo1.o_mode = S_IFREG|0644;
429
430         rc = ioctl(fd, OBD_IOC_DESTROY , &data);
431         if (rc < 0)
432                 fprintf(stderr, "error: %s: %x %s\n", argv[0],
433                         OBD_IOC_DESTROY, strerror(rc = errno));
434
435         return rc;
436 }
437
438 static int jt_getattr(int argc, char **argv)
439 {
440         struct obd_ioctl_data data;
441         int rc;
442
443         if (argc != 2) {
444                 fprintf(stderr, "usage: %s id\n", argv[0]);
445                 return -1;
446         }
447
448         IOCINIT(data);
449         data.ioc_obdo1.o_id = strtoul(argv[1], NULL, 0);
450         /* to help obd filter */ 
451         data.ioc_obdo1.o_mode = 0100644;
452         data.ioc_obdo1.o_valid = 0xffffffff;
453         printf("getting attr for %Ld\n", data.ioc_obdo1.o_id);
454
455         rc = ioctl(fd, OBD_IOC_GETATTR , &data);
456         if (rc) {
457                 fprintf(stderr, "error: %s: %s\n", argv[0], strerror(rc=errno));
458         } else {
459                 printf("attr obdo %Ld, mode %o\n", data.ioc_obdo1.o_id,
460                        data.ioc_obdo1.o_mode);
461         }
462         return rc;
463 }
464
465 static int jt_test_getattr(int argc, char **argv)
466 {
467         struct obd_ioctl_data data;
468         struct timeval start;
469         int count, i;
470         int silent = 0;
471         int rc;
472
473         if (argc != 2 && argc != 3) {
474                 fprintf(stderr, "usage: %s count [silent]\n", argv[0]);
475                 return -1;
476         }
477
478         IOCINIT(data);
479         count = strtoul(argv[1], NULL, 0);
480
481         if (argc == 3) {
482                 if (argv[2][0] == 's' || argv[2][0] == 'q')
483                         silent = 1;
484                 else
485                         silent = strtoul(argv[2], NULL, 0);
486         }
487         data.ioc_obdo1.o_valid = 0xffffffff;
488         data.ioc_obdo1.o_id = 2;
489         gettimeofday(&start, NULL);
490         printf("getting %d attrs (testing only): %s", count,
491                ctime(&start.tv_sec));
492
493         for (i = 0 ; i < count; i++) {
494                 rc = ioctl(fd, OBD_IOC_GETATTR , &data);
495                 if (rc) {
496                         fprintf(stderr, "error: %s: #%d - %s\n", argv[0], i,
497                                 strerror(rc = errno));
498                         break;
499                 } else if (!silent) {
500                         printf("attr number %d\n", i);
501                 }
502         }
503         if (!rc) {
504                 struct timeval end;
505                 double diff;
506
507                 gettimeofday(&end, NULL);
508
509                 diff = ((double)(end.tv_sec - start.tv_sec) +
510                         (double)(end.tv_usec - start.tv_usec) / 1000000);
511
512                 printf("got %d attrs in %gs (%g attr/s): %s", i,
513                        diff, (double)i / diff, ctime(&end.tv_sec));
514         }
515         return rc;
516 }
517
518 static int jt_test_brw(int argc, char **argv)
519 {
520         struct obd_ioctl_data data;
521         struct timeval start;
522         char *bulk, *b;
523         int pages = 1, obdos = 1, count;
524         int silent = 0, write = 0, rw;
525         int i, o, p;
526         int len;
527         int rc;
528
529         if (argc < 2 || argc > 6) {
530                 fprintf(stderr, "usage: \n"
531                         "%s count [write [silent [pages [obdos]]]]\n", argv[0]);
532                 return -1;
533         }
534
535         count = strtoul(argv[1], NULL, 0);
536
537         if (argc >= 3) {
538                 if (argv[2][0] == 'w')
539                         write = 1;
540                 else if (argv[2][0] == 'r')
541                         write = 0;
542                 else
543                         write = strtoul(argv[2], NULL, 0);
544         }
545         if (argc >= 4) {
546                 if (argv[3][0] == 's' || argv[3][0] == 'q')
547                         silent = 1;
548                 else
549                         silent = strtoul(argv[3], NULL, 0);
550         }
551         if (argc >= 5)
552                 pages = strtoul(argv[4], NULL, 0);
553         if (argc >= 6)
554                 obdos = strtoul(argv[5], NULL, 0);
555
556         if (obdos != 1 && obdos != 2) {
557                 fprintf(stderr, "error: %s: only 1 or 2 obdos supported\n",
558                         argv[0]);
559                 return -2;
560         }
561
562         len = pages * PAGE_SIZE;
563
564         bulk = calloc(obdos, len);
565         if (!bulk) {
566                 fprintf(stderr, "error: %s: no memory allocating %dx%d pages\n",
567                         argv[0], obdos, pages);
568                 return -2;
569         }
570         IOCINIT(data);
571         data.ioc_conn2 = connid;
572         data.ioc_obdo1.o_id = 2;
573         data.ioc_count = len;
574         data.ioc_offset = 0;
575         data.ioc_plen1 = len;
576         data.ioc_pbuf1 = bulk;
577         if (obdos > 1) {
578                 data.ioc_obdo2.o_id = 2;
579                 data.ioc_plen2 = len;
580                 data.ioc_pbuf2 = bulk + len;
581         }
582
583         gettimeofday(&start, NULL);
584         printf("%s %d (%dx%d pages) (testing only): %s",
585                write ? "writing" : "reading", count, obdos, pages,
586                ctime(&start.tv_sec));
587
588         /*
589          * We will put in the start time (and loop count inside the loop)
590          * at the beginning of each page so that we will be able to validate
591          * (at some later time) whether the data actually made it or not.
592          */
593         for (o = 0, b = bulk; o < obdos; o++)
594                 for (p = 0; p < pages; p++, b += PAGE_SIZE)
595                         memcpy(b, &start, sizeof(start));
596
597         rw = write ? OBD_IOC_BRW_WRITE : OBD_IOC_BRW_READ;
598         for (i = 0 ; i < count; i++) {
599                 if (write) {
600                         b = bulk + sizeof(struct timeval);
601                         for (o = 0; o < obdos; o++)
602                                 for (p = 0; p < pages; p++, b += PAGE_SIZE)
603                                         memcpy(b, &count, sizeof(count));
604                 }
605
606                 rc = ioctl(fd, rw, &data);
607                 if (rc) {
608                         fprintf(stderr, "error: %s: #%d - %s on %s\n", argv[0],
609                                 i, strerror(rc = errno),
610                                 write ? "write" : "read");
611                         break;
612                 } else if (!silent)
613                         printf("%s number %d\n", write ? "write" : "read", i);
614         }
615
616         free(bulk);
617
618         if (!rc) {
619                 struct timeval end;
620                 double diff;
621
622                 gettimeofday(&end, NULL);
623
624                 diff = ((double)(end.tv_sec - start.tv_sec) +
625                         (double)(end.tv_usec - start.tv_usec) / 1000000);
626
627                 printf("%s %dx%dx%d pages in %gs (%g pg/s): %s",
628                        write ? "wrote" : "read", obdos, pages, i, diff,
629                        (double)obdos * i * pages / diff, ctime(&end.tv_sec));
630         }
631         return rc;
632 }
633
634 static int jt_test_ldlm(int argc, char **argv)
635 {
636         struct obd_ioctl_data data;
637         int rc;
638
639         IOCINIT(data);
640         if (argc != 1) {
641                 fprintf(stderr, "usage: %s\n", argv[0]);
642                 return 1;
643         }
644
645         rc = ioctl(fd, IOC_LDLM_TEST, &data);
646         if (rc)
647                 fprintf(stderr, "error: %s: test failed: %s\n", argv[0],
648                         strerror(rc = errno));
649         return rc;
650 }
651
652 command_t cmdlist[] = {
653         {"device", jt_device, 0, "set current device (args device no)"},
654         {"attach", jt_attach, 0, "name the type of device (args: type data"},
655         {"setup", jt_setup, 0, "setup device (args: <blkdev> [data]"},
656         {"detach", jt_detach, 0, "detach the current device (arg: )"},
657         {"cleanup", jt_cleanup, 0, "cleanup the current device (arg: )"},
658         {"create", jt_create, 0, "create [count [mode [silent]]]"},
659         {"destroy", jt_destroy, 0, "destroy <id>"},
660         {"getattr", jt_getattr, 0, "getattr <id>"},
661         {"setattr", jt_setattr, 0, "setattr <id> <mode>"},
662         {"connect", jt_connect, 0, "connect - get a connection to device"},
663         {"disconnect", jt_disconnect, 0,
664                 "disconnect - break connection to device"},
665         {"test_getattr", jt_test_getattr, 0, "test_getattr <count> [silent]"},
666         {"test_brw", jt_test_brw, 0, "test_brw <count> [write [silent]]"},
667         {"test_ldlm", jt_test_ldlm, 0, "test lock manager (no args)"},
668         {"help", Parser_help, 0, "help"},
669         {"exit", Parser_quit, 0, "quit"},
670         {"quit", Parser_quit, 0, "quit"},
671         { 0, 0, 0, NULL }
672 };
673
674
675 static void signal_server(int sig)
676 {
677         if (sig == SIGINT) { 
678                 fprintf(stderr,"\nDisconnecting any existing connections...\n");
679                 do_disconnect("sigint");
680                 exit(1);
681         }
682 }
683
684 int main(int argc, char **argv)
685 {
686         struct sigaction sigact;
687         int rc = 0;
688
689         sigact.sa_handler = signal_server;
690         sigfillset(&sigact.sa_mask);
691         sigact.sa_flags = SA_RESTART;
692         sigaction(SIGINT, &sigact, NULL);
693
694
695         if (argc > 1) {
696                 rc = Parser_execarg(argc - 1, &argv[1], cmdlist);
697         } else {
698                 Parser_init("obdctl > ", cmdlist);
699                 Parser_commands();
700         }
701
702         do_disconnect(argv[0]);
703         return rc;
704 }
705