Whamcloud - gitweb
b=23663 fix buffer overflow in liblustre
[fs/lustre-release.git] / lnet / utils / portals.c
1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
2  * vim:expandtab:shiftwidth=8:tabstop=8:
3  *
4  * Copyright  2008 Sun Microsystems, Inc. All rights reserved
5  *
6  *   This file is part of Portals, http://www.sf.net/projects/lustre/
7  *
8  *   Portals is free software; you can redistribute it and/or
9  *   modify it under the terms of version 2 of the GNU General Public
10  *   License as published by the Free Software Foundation.
11  *
12  *   Portals is distributed in the hope that it will be useful,
13  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
14  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  *   GNU General Public License for more details.
16  *
17  *   You should have received a copy of the GNU General Public License
18  *   along with Portals; if not, write to the Free Software
19  *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20  *
21  */
22
23 #include <stdio.h>
24 #include <sys/types.h>
25 #ifdef HAVE_NETDB_H
26 #include <netdb.h>
27 #endif
28 #include <sys/socket.h>
29 #ifdef HAVE_NETINET_TCP_H
30 #include <netinet/tcp.h>
31 #endif
32 #include <stdlib.h>
33 #include <string.h>
34 #include <fcntl.h>
35 #ifdef HAVE_SYS_IOCTL_H
36 #include <sys/ioctl.h>
37 #endif
38 #ifndef _IOWR
39 #include "ioctl.h"
40 #endif
41 #include <errno.h>
42 #include <unistd.h>
43 #include <time.h>
44 #include <stdarg.h>
45 #ifdef HAVE_ENDIAN_H
46 #include <endian.h>
47 #endif
48
49 #include <libcfs/portals_utils.h>
50 #include <lnet/api-support.h>
51 #include <lnet/lnetctl.h>
52 #include <lnet/socklnd.h>
53 #include "parser.h"
54
55 unsigned int libcfs_debug;
56 unsigned int libcfs_printk = D_CANTMASK;
57
58 static int   g_net_set;
59 static __u32 g_net;
60
61 /* Convert a string boolean to an int; "enable" -> 1 */
62 int
63 lnet_parse_bool (int *b, char *str)
64 {
65         if (!strcasecmp (str, "no") ||
66             !strcasecmp (str, "n") ||
67             !strcasecmp (str, "off") ||
68             !strcasecmp (str, "down") ||
69             !strcasecmp (str, "disable"))
70         {
71                 *b = 0;
72                 return (0);
73         }
74
75         if (!strcasecmp (str, "yes") ||
76             !strcasecmp (str, "y") ||
77             !strcasecmp (str, "on") ||
78             !strcasecmp (str, "up") ||
79             !strcasecmp (str, "enable"))
80         {
81                 *b = 1;
82                 return (0);
83         }
84
85         return (-1);
86 }
87
88 int
89 lnet_parse_port (int *port, char *str)
90 {
91         char      *end;
92
93         *port = strtol (str, &end, 0);
94
95         if (*end == 0 &&                        /* parsed whole string */
96             *port > 0 && *port < 65536)         /* minimal sanity check */
97                 return (0);
98
99         return (-1);
100 }
101
102 #ifdef HAVE_GETHOSTBYNAME
103 static struct hostent *
104 ptl_gethostbyname(char * hname) {
105         struct hostent *he;
106         he = gethostbyname(hname);
107         if (!he) {
108                 switch(h_errno) {
109                 case HOST_NOT_FOUND:
110                 case NO_ADDRESS:
111                         fprintf(stderr, "Unable to resolve hostname: %s\n",
112                                 hname);
113                         break;
114                 default:
115                         fprintf(stderr, "gethostbyname error for %s: %s\n",
116                                 hname, strerror(h_errno));
117                         break;
118                 }
119                 return NULL;
120         }
121         return he;
122 }
123 #endif
124
125 int
126 lnet_parse_ipquad (__u32 *ipaddrp, char *str)
127 {
128         int             a;
129         int             b;
130         int             c;
131         int             d;
132
133         if (sscanf (str, "%d.%d.%d.%d", &a, &b, &c, &d) == 4 &&
134             (a & ~0xff) == 0 && (b & ~0xff) == 0 &&
135             (c & ~0xff) == 0 && (d & ~0xff) == 0)
136         {
137                 *ipaddrp = (a<<24)|(b<<16)|(c<<8)|d;
138                 return (0);
139         }
140
141         return (-1);
142 }
143
144 int
145 lnet_parse_ipaddr (__u32 *ipaddrp, char *str)
146 {
147 #ifdef HAVE_GETHOSTBYNAME
148         struct hostent *he;
149 #endif
150
151         if (!strcmp (str, "_all_")) {
152                 *ipaddrp = 0;
153                 return (0);
154         }
155
156         if (lnet_parse_ipquad(ipaddrp, str) == 0)
157                 return (0);
158
159 #ifdef HAVE_GETHOSTBYNAME
160         if ((('a' <= str[0] && str[0] <= 'z') ||
161              ('A' <= str[0] && str[0] <= 'Z')) &&
162              (he = ptl_gethostbyname (str)) != NULL) {
163                 __u32 addr = *(__u32 *)he->h_addr;
164
165                 *ipaddrp = ntohl(addr);         /* HOST byte order */
166                 return (0);
167         }
168 #endif
169
170         return (-1);
171 }
172
173 char *
174 ptl_ipaddr_2_str (__u32 ipaddr, char *str, int lookup)
175 {
176 #ifdef HAVE_GETHOSTBYNAME
177         __u32           net_ip;
178         struct hostent *he;
179
180         if (lookup) {
181                 net_ip = htonl (ipaddr);
182                 he = gethostbyaddr (&net_ip, sizeof (net_ip), AF_INET);
183                 if (he != NULL) {
184                         strcpy(str, he->h_name);
185                         return (str);
186                 }
187         }
188 #endif
189
190         sprintf (str, "%d.%d.%d.%d",
191                  (ipaddr >> 24) & 0xff, (ipaddr >> 16) & 0xff,
192                  (ipaddr >> 8) & 0xff, ipaddr & 0xff);
193         return (str);
194 }
195
196 int
197 lnet_parse_time (time_t *t, char *str)
198 {
199         char          *end;
200         int            n;
201         struct tm      tm;
202
203         *t = strtol (str, &end, 0);
204         if (*end == 0) /* parsed whole string */
205                 return (0);
206
207         memset (&tm, 0, sizeof (tm));
208         n = sscanf (str, "%d-%d-%d-%d:%d:%d",
209                     &tm.tm_year, &tm.tm_mon, &tm.tm_mday,
210                     &tm.tm_hour, &tm.tm_min, &tm.tm_sec);
211         if (n != 6)
212                 return (-1);
213
214         tm.tm_mon--;                    /* convert to 0 == Jan */
215         tm.tm_year -= 1900;             /* y2k quirk */
216         tm.tm_isdst = -1;               /* dunno if it's daylight savings... */
217
218         *t = mktime (&tm);
219         if (*t == (time_t)-1)
220                 return (-1);
221
222         return (0);
223 }
224
225 int g_net_is_set (char *cmd)
226 {
227         if (g_net_set)
228                 return 1;
229
230         if (cmd != NULL)
231                 fprintf(stderr,
232                         "You must run the 'network' command before '%s'.\n",
233                         cmd);
234         return 0;
235 }
236
237 int g_net_is_compatible (char *cmd, ...)
238 {
239         va_list       ap;
240         int           nal;
241
242         if (!g_net_is_set(cmd))
243                 return 0;
244
245         va_start(ap, cmd);
246
247         do {
248                 nal = va_arg (ap, int);
249                 if (nal == LNET_NETTYP(g_net)) {
250                         va_end (ap);
251                         return 1;
252                 }
253         } while (nal != 0);
254
255         va_end (ap);
256
257         if (cmd != NULL)
258                 fprintf (stderr,
259                          "Command %s not compatible with %s NAL\n",
260                          cmd,
261                          libcfs_lnd2str(LNET_NETTYP(g_net)));
262         return 0;
263 }
264
265 int ptl_initialize(int argc, char **argv)
266 {
267         register_ioc_dev(LNET_DEV_ID, LNET_DEV_PATH,
268                          LNET_DEV_MAJOR, LNET_DEV_MINOR);
269         return 0;
270 }
271
272
273 int jt_ptl_network(int argc, char **argv)
274 {
275         struct libcfs_ioctl_data data;
276         __u32                    net = LNET_NIDNET(LNET_NID_ANY);
277         int                      rc;
278
279         if (argc < 2) {
280                 fprintf(stderr, "usage: %s <net>|up|down\n", argv[0]);
281                 return 0;
282         }
283
284         if (!strcmp(argv[1], "unconfigure") ||
285             !strcmp(argv[1], "down")) {
286                 LIBCFS_IOC_INIT(data);
287                 rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_UNCONFIGURE, &data);
288
289                 if (rc == 0) {
290                         printf ("LNET ready to unload\n");
291                         return 0;
292                 }
293
294                 if (errno == EBUSY)
295                         fprintf(stderr, "LNET busy\n");
296                 else
297                         fprintf(stderr, "LNET unconfigure error %d: %s\n",
298                                 errno, strerror(errno));
299                 return -1;
300         }
301
302         if (!strcmp(argv[1], "configure") ||
303             !strcmp(argv[1], "up")) {
304                 LIBCFS_IOC_INIT(data);
305                 rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_CONFIGURE, &data);
306
307                 if (rc == 0) {
308                         printf ("LNET configured\n");
309                         return 0;
310                 }
311
312                 fprintf(stderr, "LNET configure error %d: %s\n",
313                         errno, strerror(errno));
314                 return -1;
315         }
316
317         net = libcfs_str2net(argv[1]);
318         if (net == LNET_NIDNET(LNET_NID_ANY)) {
319                 fprintf(stderr, "Can't parse net %s\n", argv[1]);
320                 return -1;
321         }
322
323         g_net_set = 1;
324         g_net = net;
325         return 0;
326 }
327
328 int
329 jt_ptl_list_nids(int argc, char **argv)
330 {
331         struct libcfs_ioctl_data data;
332         int                      all = 0, return_nid = 0;
333         int                      count;
334         int                      rc;
335
336         all = (argc == 2) && (strcmp(argv[1], "all") == 0);
337         /* Hack to pass back value */
338         return_nid = (argc == 2) && (argv[1][0] == 1);
339
340         if ((argc > 2) && !(all || return_nid)) {
341                 fprintf(stderr, "usage: %s [all]\n", argv[0]);
342                 return 0;
343         }
344
345         for (count = 0;; count++) {
346                 LIBCFS_IOC_INIT (data);
347                 data.ioc_count = count;
348                 rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_GET_NI, &data);
349
350                 if (rc < 0) {
351                         if ((count > 0) && (errno == ENOENT))
352                                 /* We found them all */
353                                 break;
354                         fprintf(stderr,"IOC_LIBCFS_GET_NI error %d: %s\n",
355                                 errno, strerror(errno));
356                         return -1;
357                 }
358
359                 if (all || (LNET_NETTYP(LNET_NIDNET(data.ioc_nid)) != LOLND)) {
360                         printf("%s\n", libcfs_nid2str(data.ioc_nid));
361                         if (return_nid) {
362                                 *(__u64 *)(argv[1]) = data.ioc_nid;
363                                 return_nid--;
364                         }
365                 }
366         }
367
368         return 0;
369 }
370
371 int
372 jt_ptl_which_nid (int argc, char **argv)
373 {
374         struct libcfs_ioctl_data data;
375         int          best_dist = 0;
376         int          best_order = 0;
377         lnet_nid_t   best_nid = LNET_NID_ANY;
378         int          dist;
379         int          order;
380         lnet_nid_t   nid;
381         char        *nidstr;
382         int          rc;
383         int          i;
384
385         if (argc < 2) {
386                 fprintf(stderr, "usage: %s NID [NID...]\n", argv[0]);
387                 return 0;
388         }
389
390         for (i = 1; i < argc; i++) {
391                 nidstr = argv[i];
392                 nid = libcfs_str2nid(nidstr);
393                 if (nid == LNET_NID_ANY) {
394                         fprintf(stderr, "Can't parse NID %s\n", nidstr);
395                         return -1;
396                 }
397
398                 LIBCFS_IOC_INIT(data);
399                 data.ioc_nid = nid;
400
401                 rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_LNET_DIST, &data);
402                 if (rc != 0) {
403                         fprintf(stderr, "Can't get distance to %s: %s\n",
404                                 nidstr, strerror(errno));
405                         return -1;
406                 }
407
408                 dist = data.ioc_u32[0];
409                 order = data.ioc_u32[1];
410
411                 if (dist < 0) {
412                         if (dist == -EHOSTUNREACH)
413                                 continue;
414
415                         fprintf(stderr, "Unexpected distance to %s: %d\n",
416                                 nidstr, dist);
417                         return -1;
418                 }
419
420                 if (best_nid == LNET_NID_ANY ||
421                     dist < best_dist ||
422                     (dist == best_dist && order < best_order)) {
423                         best_dist = dist;
424                         best_order = order;
425                         best_nid = nid;
426                 }
427         }
428
429         if (best_nid == LNET_NID_ANY) {
430                 fprintf(stderr, "No reachable NID\n");
431                 return -1;
432         }
433
434         printf("%s\n", libcfs_nid2str(best_nid));
435         return 0;
436 }
437
438 int
439 jt_ptl_print_interfaces (int argc, char **argv)
440 {
441         struct libcfs_ioctl_data data;
442         char                     buffer[3][64];
443         int                      index;
444         int                      rc;
445
446         if (!g_net_is_compatible (argv[0], SOCKLND, 0))
447                 return -1;
448
449         for (index = 0;;index++) {
450                 LIBCFS_IOC_INIT(data);
451                 data.ioc_net   = g_net;
452                 data.ioc_count = index;
453
454                 rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_GET_INTERFACE, &data);
455                 if (rc != 0)
456                         break;
457
458                 printf ("%s: (%s/%s) npeer %d nroute %d\n",
459                         ptl_ipaddr_2_str(data.ioc_u32[0], buffer[2], 1),
460                         ptl_ipaddr_2_str(data.ioc_u32[0], buffer[0], 0),
461                         ptl_ipaddr_2_str(data.ioc_u32[1], buffer[1], 0),
462                         data.ioc_u32[2], data.ioc_u32[3]);
463         }
464
465         if (index == 0) {
466                 if (errno == ENOENT) {
467                         printf ("<no interfaces>\n");
468                 } else {
469                         fprintf(stderr, "Error getting interfaces: %s: "
470                                 "check dmesg.\n",
471                                 strerror(errno));
472                 }
473         }
474
475         return 0;
476 }
477
478 int
479 jt_ptl_add_interface (int argc, char **argv)
480 {
481         struct libcfs_ioctl_data data;
482         __u32                    ipaddr;
483         int                      rc;
484         __u32                    netmask = 0xffffff00;
485         int                      i;
486         int                      count;
487         char                    *end;
488
489         if (argc < 2 || argc > 3) {
490                 fprintf (stderr, "usage: %s ipaddr [netmask]\n", argv[0]);
491                 return 0;
492         }
493
494         if (!g_net_is_compatible(argv[0], SOCKLND, 0))
495                 return -1;
496
497         if (lnet_parse_ipaddr(&ipaddr, argv[1]) != 0) {
498                 fprintf (stderr, "Can't parse ip: %s\n", argv[1]);
499                 return -1;
500         }
501
502         if (argc > 2 ) {
503                 count = strtol(argv[2], &end, 0);
504                 if (count > 0 && count < 32 && *end == 0) {
505                         netmask = 0;
506                         for (i = count; i > 0; i--)
507                                 netmask = netmask|(1<<(32-i));
508                 } else if (lnet_parse_ipquad(&netmask, argv[2]) != 0) {
509                         fprintf (stderr, "Can't parse netmask: %s\n", argv[2]);
510                         return -1;
511                 }
512         }
513
514         LIBCFS_IOC_INIT(data);
515         data.ioc_net    = g_net;
516         data.ioc_u32[0] = ipaddr;
517         data.ioc_u32[1] = netmask;
518
519         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_ADD_INTERFACE, &data);
520         if (rc != 0) {
521                 fprintf (stderr, "failed to add interface: %s\n",
522                          strerror (errno));
523                 return -1;
524         }
525
526         return 0;
527 }
528
529 int
530 jt_ptl_del_interface (int argc, char **argv)
531 {
532         struct libcfs_ioctl_data data;
533         int                      rc;
534         __u32                    ipaddr = 0;
535
536         if (argc > 2) {
537                 fprintf (stderr, "usage: %s [ipaddr]\n", argv[0]);
538                 return 0;
539         }
540
541         if (!g_net_is_compatible(argv[0], SOCKLND, 0))
542                 return -1;
543
544         if (argc == 2 &&
545             lnet_parse_ipaddr(&ipaddr, argv[1]) != 0) {
546                 fprintf (stderr, "Can't parse ip: %s\n", argv[1]);
547                 return -1;
548         }
549
550         LIBCFS_IOC_INIT(data);
551         data.ioc_net    = g_net;
552         data.ioc_u32[0] = ipaddr;
553
554         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_DEL_INTERFACE, &data);
555         if (rc != 0) {
556                 fprintf (stderr, "failed to delete interface: %s\n",
557                          strerror (errno));
558                 return -1;
559         }
560
561         return 0;
562 }
563
564 int
565 jt_ptl_print_peers (int argc, char **argv)
566 {
567         struct libcfs_ioctl_data data;
568         lnet_process_id_t        id;
569         char                     buffer[2][64];
570         int                      index;
571         int                      rc;
572
573         if (!g_net_is_compatible (argv[0], SOCKLND, RALND, PTLLND, MXLND,
574                                   OPENIBLND, CIBLND, IIBLND, VIBLND, O2IBLND, 0))
575                 return -1;
576
577         for (index = 0;;index++) {
578                 LIBCFS_IOC_INIT(data);
579                 data.ioc_net     = g_net;
580                 data.ioc_count   = index;
581
582                 rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_GET_PEER, &data);
583                 if (rc != 0)
584                         break;
585
586                 if (g_net_is_compatible(NULL, SOCKLND, 0)) {
587                         id.nid = data.ioc_nid;
588                         id.pid = data.ioc_u32[4];
589                         printf ("%-20s [%d]%s->%s:%d #%d\n",
590                                 libcfs_id2str(id), 
591                                 data.ioc_count, /* persistence */
592                                 ptl_ipaddr_2_str (data.ioc_u32[2], buffer[0], 1), /* my ip */
593                                 ptl_ipaddr_2_str (data.ioc_u32[0], buffer[1], 1), /* peer ip */
594                                 data.ioc_u32[1], /* peer port */
595                                 data.ioc_u32[3]); /* conn_count */
596                 } else if (g_net_is_compatible(NULL, PTLLND, 0)) {
597                         id.nid = data.ioc_nid;
598                         id.pid = data.ioc_u32[4];
599                         printf ("%-20s s %d%s [%d] "LPD64".%06d"
600                                 " m "LPD64"/"LPD64" q %d/%d c %d/%d\n",
601                                 libcfs_id2str(id),
602                                 data.ioc_net,   /* state */
603                                 data.ioc_flags ? "" : " ~!h", /* sent_hello */
604                                 data.ioc_count, /* refcount */
605                                 data.ioc_u64[0]/1000000, /* incarnation secs */
606                                 (int)(data.ioc_u64[0]%1000000), /* incarnation usecs */
607                                 (((__u64)data.ioc_u32[1])<<32) |
608                                 ((__u64)data.ioc_u32[0]), /* next_matchbits */
609                                 (((__u64)data.ioc_u32[3])<<32) |
610                                 ((__u64)data.ioc_u32[2]), /* last_matchbits_seen */
611                                 data.ioc_u32[5] >> 16, /* nsendq */
612                                 data.ioc_u32[5] & 0xffff, /* nactiveq */
613                                 data.ioc_u32[6] >> 16, /* credits */
614                                 data.ioc_u32[6] & 0xffff); /* outstanding_credits */
615                 } else if (g_net_is_compatible(NULL, RALND, OPENIBLND, CIBLND, VIBLND, 0)) {
616                         printf ("%-20s [%d]@%s:%d\n",
617                                 libcfs_nid2str(data.ioc_nid), /* peer nid */
618                                 data.ioc_count,   /* peer persistence */
619                                 ptl_ipaddr_2_str (data.ioc_u32[0], buffer[1], 1), /* peer ip */
620                                 data.ioc_u32[1]); /* peer port */
621                 } else {
622                         printf ("%-20s [%d]\n",
623                                 libcfs_nid2str(data.ioc_nid), data.ioc_count);
624                 }
625         }
626
627         if (index == 0) {
628                 if (errno == ENOENT) {
629                         printf ("<no peers>\n");
630                 } else {
631                         fprintf(stderr, "Error getting peer list: %s: "
632                                 "check dmesg.\n",
633                                 strerror(errno));
634                 }
635         }
636         return 0;
637 }
638
639 int
640 jt_ptl_add_peer (int argc, char **argv)
641 {
642         struct libcfs_ioctl_data data;
643         lnet_nid_t               nid;
644         __u32                    ip = 0;
645         int                      port = 0;
646         int                      rc;
647
648         if (!g_net_is_compatible (argv[0], SOCKLND, RALND,
649                                   OPENIBLND, CIBLND, IIBLND, VIBLND, 0))
650                 return -1;
651
652         if (g_net_is_compatible(NULL, SOCKLND, OPENIBLND, CIBLND, RALND, 0)) {
653                 if (argc != 4) {
654                         fprintf (stderr, "usage(tcp,openib,cib,ra): %s nid ipaddr port\n",
655                                  argv[0]);
656                         return 0;
657                 }
658         } else if (g_net_is_compatible(NULL, VIBLND, 0)) {
659                 if (argc != 3) {
660                         fprintf (stderr, "usage(vib): %s nid ipaddr\n",
661                                  argv[0]);
662                         return 0;
663                 }
664         } else if (argc != 2) {
665                 fprintf (stderr, "usage(iib): %s nid\n", argv[0]);
666                 return 0;
667         }
668
669         nid = libcfs_str2nid(argv[1]);
670         if (nid == LNET_NID_ANY) {
671                 fprintf (stderr, "Can't parse NID: %s\n", argv[1]);
672                 return -1;
673         }
674
675         if (g_net_is_compatible (NULL, SOCKLND, OPENIBLND, CIBLND, VIBLND, RALND, 0) &&
676             lnet_parse_ipaddr (&ip, argv[2]) != 0) {
677                 fprintf (stderr, "Can't parse ip addr: %s\n", argv[2]);
678                 return -1;
679         }
680
681         if (g_net_is_compatible (NULL, SOCKLND, OPENIBLND, CIBLND, RALND, 0) &&
682             lnet_parse_port (&port, argv[3]) != 0) {
683                 fprintf (stderr, "Can't parse port: %s\n", argv[3]);
684                 return -1;
685         }
686
687         LIBCFS_IOC_INIT(data);
688         data.ioc_net    = g_net;
689         data.ioc_nid    = nid;
690         data.ioc_u32[0] = ip;
691         data.ioc_u32[1] = port;
692
693         rc = l_ioctl (LNET_DEV_ID, IOC_LIBCFS_ADD_PEER, &data);
694         if (rc != 0) {
695                 fprintf (stderr, "failed to add peer: %s\n",
696                          strerror (errno));
697                 return -1;
698         }
699
700         return 0;
701 }
702
703 int
704 jt_ptl_del_peer (int argc, char **argv)
705 {
706         struct libcfs_ioctl_data data;
707         lnet_nid_t               nid = LNET_NID_ANY;
708         lnet_pid_t               pid = LNET_PID_ANY;
709         __u32                    ip = 0;
710         char                    *end;
711         int                      rc;
712
713         if (!g_net_is_compatible (argv[0], SOCKLND, RALND, MXLND, PTLLND,
714                                   OPENIBLND, CIBLND, IIBLND, VIBLND, O2IBLND, 0))
715                 return -1;
716
717         if (g_net_is_compatible(NULL, SOCKLND, 0)) {
718                 if (argc > 3) {
719                         fprintf (stderr, "usage: %s [nid] [ipaddr]\n",
720                                  argv[0]);
721                         return 0;
722                 }
723         } else if (g_net_is_compatible(NULL, PTLLND, 0)) {
724                 if (argc > 3) {
725                         fprintf (stderr, "usage: %s [nid] [pid]\n",
726                                  argv[0]);
727                         return 0;
728                 }
729         } else if (argc > 2) {
730                 fprintf (stderr, "usage: %s [nid]\n", argv[0]);
731                 return 0;
732         }
733
734         if (argc > 1 &&
735             !libcfs_str2anynid(&nid, argv[1])) {
736                 fprintf (stderr, "Can't parse nid: %s\n", argv[1]);
737                 return -1;
738         }
739
740         if (g_net_is_compatible(NULL, SOCKLND, 0)) {
741                 if (argc > 2 &&
742                     lnet_parse_ipaddr (&ip, argv[2]) != 0) {
743                         fprintf (stderr, "Can't parse ip addr: %s\n",
744                                  argv[2]);
745                         return -1;
746                 }
747         } else if (g_net_is_compatible(NULL, PTLLND, 0)) {
748                 if (argc > 2) {
749                         pid = strtol(argv[2], &end, 0);
750                         if (end == argv[2] || *end == 0) {
751                                 fprintf(stderr, "Can't parse pid %s\n",
752                                         argv[2]);
753                                 return -1;
754                         }
755                 }
756         }
757
758         LIBCFS_IOC_INIT(data);
759         data.ioc_net    = g_net;
760         data.ioc_nid    = nid;
761         data.ioc_u32[0] = ip;
762         data.ioc_u32[1] = pid;
763
764         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_DEL_PEER, &data);
765         if (rc != 0) {
766                 fprintf (stderr, "failed to remove peer: %s\n",
767                          strerror (errno));
768                 return -1;
769         }
770
771         return 0;
772 }
773
774 int
775 jt_ptl_print_connections (int argc, char **argv)
776 {
777         struct libcfs_ioctl_data data;
778         lnet_process_id_t        id;
779         char                     buffer[2][64];
780         int                      index;
781         int                      rc;
782
783         if (!g_net_is_compatible (argv[0], SOCKLND, RALND, MXLND,
784                                   OPENIBLND, CIBLND, IIBLND, VIBLND, O2IBLND, 0))
785                 return -1;
786
787         for (index = 0; ; index++) {
788                 LIBCFS_IOC_INIT(data);
789                 data.ioc_net     = g_net;
790                 data.ioc_count   = index;
791
792                 rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_GET_CONN, &data);
793                 if (rc != 0)
794                         break;
795
796                 if (g_net_is_compatible (NULL, SOCKLND, 0)) {
797                         id.nid = data.ioc_nid;
798                         id.pid = data.ioc_u32[6];
799                         printf ("%-20s %s[%d]%s->%s:%d %d/%d %s\n",
800                                 libcfs_id2str(id),
801                                 (data.ioc_u32[3] == SOCKLND_CONN_ANY) ? "A" :
802                                 (data.ioc_u32[3] == SOCKLND_CONN_CONTROL) ? "C" :
803                                 (data.ioc_u32[3] == SOCKLND_CONN_BULK_IN) ? "I" :
804                                 (data.ioc_u32[3] == SOCKLND_CONN_BULK_OUT) ? "O" : "?",
805                                 data.ioc_u32[4], /* scheduler */
806                                 ptl_ipaddr_2_str (data.ioc_u32[2], buffer[0], 1), /* local IP addr */
807                                 ptl_ipaddr_2_str (data.ioc_u32[0], buffer[1], 1), /* remote IP addr */
808                                 data.ioc_u32[1],         /* remote port */
809                                 data.ioc_count, /* tx buffer size */
810                                 data.ioc_u32[5], /* rx buffer size */
811                                 data.ioc_flags ? "nagle" : "nonagle");
812                 } else if (g_net_is_compatible (NULL, RALND, 0)) {
813                         printf ("%-20s [%d]\n",
814                                 libcfs_nid2str(data.ioc_nid),
815                                 data.ioc_u32[0] /* device id */);
816                 } else if (g_net_is_compatible (NULL, O2IBLND, 0)) {
817                         printf ("%s mtu %d\n",
818                                 libcfs_nid2str(data.ioc_nid),
819                                 data.ioc_u32[0]); /* path MTU */
820                 } else {
821                         printf ("%s\n", libcfs_nid2str(data.ioc_nid));
822                 }
823         }
824
825         if (index == 0) {
826                 if (errno == ENOENT) {
827                         printf ("<no connections>\n");
828                 } else {
829                         fprintf(stderr, "Error getting connection list: %s: "
830                                 "check dmesg.\n",
831                                 strerror(errno));
832                 }
833         }
834         return 0;
835 }
836
837 int jt_ptl_disconnect(int argc, char **argv)
838 {
839         struct libcfs_ioctl_data data;
840         lnet_nid_t               nid = LNET_NID_ANY;
841         __u32                    ipaddr = 0;
842         int                      rc;
843
844         if (argc > 3) {
845                 fprintf(stderr, "usage: %s [nid] [ipaddr]\n", argv[0]);
846                 return 0;
847         }
848
849         if (!g_net_is_compatible (NULL, SOCKLND, RALND, MXLND,
850                                   OPENIBLND, CIBLND, IIBLND, VIBLND, O2IBLND, 0))
851                 return 0;
852
853         if (argc >= 2 &&
854             !libcfs_str2anynid(&nid, argv[1])) {
855                 fprintf (stderr, "Can't parse nid %s\n", argv[1]);
856                 return -1;
857         }
858
859         if (g_net_is_compatible (NULL, SOCKLND, 0) &&
860             argc >= 3 &&
861             lnet_parse_ipaddr (&ipaddr, argv[2]) != 0) {
862                 fprintf (stderr, "Can't parse ip addr %s\n", argv[2]);
863                 return -1;
864         }
865
866         LIBCFS_IOC_INIT(data);
867         data.ioc_net     = g_net;
868         data.ioc_nid     = nid;
869         data.ioc_u32[0]  = ipaddr;
870
871         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_CLOSE_CONNECTION, &data);
872         if (rc != 0) {
873                 fprintf(stderr, "failed to remove connection: %s\n",
874                         strerror(errno));
875                 return -1;
876         }
877
878         return 0;
879 }
880
881 int jt_ptl_push_connection (int argc, char **argv)
882 {
883         struct libcfs_ioctl_data data;
884         int                      rc;
885         lnet_nid_t               nid = LNET_NID_ANY;
886
887         if (argc > 2) {
888                 fprintf(stderr, "usage: %s [nid]\n", argv[0]);
889                 return 0;
890         }
891
892         if (!g_net_is_compatible (argv[0], SOCKLND, 0))
893                 return -1;
894
895         if (argc > 1 &&
896             !libcfs_str2anynid(&nid, argv[1])) {
897                 fprintf(stderr, "Can't parse nid: %s\n", argv[1]);
898                 return -1;
899         }
900
901         LIBCFS_IOC_INIT(data);
902         data.ioc_net     = g_net;
903         data.ioc_nid     = nid;
904
905         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_PUSH_CONNECTION, &data);
906         if (rc != 0) {
907                 fprintf(stderr, "failed to push connection: %s\n",
908                         strerror(errno));
909                 return -1;
910         }
911
912         return 0;
913 }
914
915 int
916 jt_ptl_print_active_txs (int argc, char **argv)
917 {
918         struct libcfs_ioctl_data data;
919         int                      index;
920         int                      rc;
921
922         if (!g_net_is_compatible (argv[0], QSWLND, 0))
923                 return -1;
924
925         for (index = 0;;index++) {
926                 LIBCFS_IOC_INIT(data);
927                 data.ioc_net   = g_net;
928                 data.ioc_count = index;
929
930                 rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_GET_TXDESC, &data);
931                 if (rc != 0)
932                         break;
933
934                 printf ("type %u payload %6d to %s via %s by pid %6d: "
935                         "%s, %s, state %d\n",
936                         data.ioc_u32[0],
937                         data.ioc_count,
938                         libcfs_nid2str(data.ioc_nid),
939                         libcfs_nid2str(data.ioc_u64[0]),
940                         data.ioc_u32[1],
941                         (data.ioc_flags & 1) ? "delayed" : "immediate",
942                         (data.ioc_flags & 2) ? "nblk"    : "normal",
943                         data.ioc_flags >> 2);
944         }
945
946         if (index == 0) {
947                 if (errno == ENOENT) {
948                         printf ("<no active descs>\n");
949                 } else {
950                         fprintf(stderr, "Error getting active transmits list: "
951                                 "%s: check dmesg.\n",
952                                 strerror(errno));
953                 }
954         }
955         return 0;
956 }
957
958 int jt_ptl_ping(int argc, char **argv)
959 {
960         int                      rc;
961         int                      timeout;
962         lnet_process_id_t        id;
963         lnet_process_id_t        ids[16];
964         int                      maxids = sizeof(ids)/sizeof(ids[0]);
965         struct libcfs_ioctl_data data;
966         char                    *sep;
967         int                      i;
968
969         if (argc < 2) {
970                 fprintf(stderr, "usage: %s id [timeout (secs)]\n", argv[0]);
971                 return 0;
972         }
973
974         sep = strchr(argv[1], '-');
975         if (sep == NULL) {
976                 id.pid = LNET_PID_ANY;
977                 id.nid = libcfs_str2nid(argv[1]);
978                 if (id.nid == LNET_NID_ANY) {
979                         fprintf (stderr, "Can't parse nid \"%s\"\n", argv[1]);
980                         return -1;
981                 }
982         } else {
983                 char   *end;
984
985                 if (argv[1][0] == 'u' ||
986                     argv[1][0] == 'U')
987                         id.pid = strtoul(&argv[1][1], &end, 0) | LNET_PID_USERFLAG;
988                 else
989                         id.pid = strtoul(argv[1], &end, 0);
990
991                 id.nid = libcfs_str2nid(sep + 1);
992
993                 if (end != sep ||
994                     id.nid == LNET_NID_ANY) {
995                         fprintf(stderr, "Can't parse process id \"%s\"\n", argv[1]);
996                         return -1;
997                 }
998         }
999
1000         if (argc > 2)
1001                 timeout = 1000 * atol(argv[2]);
1002         else
1003                 timeout = 1000;                 /* default 1 second timeout */
1004
1005         LIBCFS_IOC_INIT (data);
1006         data.ioc_nid     = id.nid;
1007         data.ioc_u32[0]  = id.pid;
1008         data.ioc_u32[1]  = timeout;
1009         data.ioc_plen1   = sizeof(ids);
1010         data.ioc_pbuf1   = (char *)ids;
1011
1012         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_PING, &data);
1013         if (rc != 0) {
1014                 fprintf(stderr, "failed to ping %s: %s\n",
1015                         id.pid == LNET_PID_ANY ?
1016                         libcfs_nid2str(id.nid) : libcfs_id2str(id),
1017                         strerror(errno));
1018                 return -1;
1019         }
1020
1021         for (i = 0; i < data.ioc_count && i < maxids; i++)
1022                 printf("%s\n", libcfs_id2str(ids[i]));
1023
1024         if (data.ioc_count > maxids)
1025                 printf("%d out of %d ids listed\n", maxids, data.ioc_count);
1026
1027         return 0;
1028 }
1029
1030 int jt_ptl_mynid(int argc, char **argv)
1031 {
1032         struct libcfs_ioctl_data data;
1033         lnet_nid_t               nid;
1034         int rc;
1035
1036         if (argc != 2) {
1037                 fprintf(stderr, "usage: %s NID\n", argv[0]);
1038                 return 0;
1039         }
1040
1041         nid = libcfs_str2nid(argv[1]);
1042         if (nid == LNET_NID_ANY) {
1043                 fprintf(stderr, "Can't parse NID '%s'\n", argv[1]);
1044                 return -1;
1045         }
1046
1047         LIBCFS_IOC_INIT(data);
1048         data.ioc_net = LNET_NIDNET(nid);
1049         data.ioc_nid = nid;
1050
1051         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_REGISTER_MYNID, &data);
1052         if (rc < 0)
1053                 fprintf(stderr, "setting my NID failed: %s\n",
1054                        strerror(errno));
1055         else
1056                 printf("registered my nid %s\n", libcfs_nid2str(nid));
1057
1058         return 0;
1059 }
1060
1061 int
1062 jt_ptl_fail_nid (int argc, char **argv)
1063 {
1064         int                      rc;
1065         lnet_nid_t               nid;
1066         unsigned int             threshold;
1067         struct libcfs_ioctl_data data;
1068
1069         if (argc < 2 || argc > 3)
1070         {
1071                 fprintf (stderr, "usage: %s nid|\"*\" [count (0 == mend)]\n", argv[0]);
1072                 return (0);
1073         }
1074
1075         if (!libcfs_str2anynid(&nid, argv[1]))
1076         {
1077                 fprintf (stderr, "Can't parse nid \"%s\"\n", argv[1]);
1078                 return (-1);
1079         }
1080
1081         if (argc < 3) {
1082                 threshold = LNET_MD_THRESH_INF;
1083         } else if (sscanf (argv[2], "%i", &threshold) != 1) {
1084                 fprintf (stderr, "Can't parse count \"%s\"\n", argv[2]);
1085                 return (-1);
1086         }
1087
1088         LIBCFS_IOC_INIT (data);
1089         data.ioc_nid = nid;
1090         data.ioc_count = threshold;
1091
1092         rc = l_ioctl (LNET_DEV_ID, IOC_LIBCFS_FAIL_NID, &data);
1093         if (rc < 0)
1094                 fprintf (stderr, "IOC_LIBCFS_FAIL_NID failed: %s\n",
1095                          strerror (errno));
1096         else
1097                 printf ("%s %s\n", threshold == 0 ? "Unfailing" : "Failing", argv[1]);
1098
1099         return (0);
1100 }
1101
1102 int
1103 jt_ptl_add_route (int argc, char **argv)
1104 {
1105         struct libcfs_ioctl_data data;
1106         lnet_nid_t               gateway_nid;
1107         unsigned int             hops = 1;
1108         char                    *end;
1109         int                      rc;
1110
1111         if (argc < 2 || argc > 3)
1112         {
1113                 fprintf (stderr, "usage: %s gateway [hopcount]\n", argv[0]);
1114                 return (0);
1115         }
1116
1117         if (!g_net_is_set(argv[0]))
1118                 return (-1);
1119
1120         gateway_nid = libcfs_str2nid(argv[1]);
1121         if (gateway_nid == LNET_NID_ANY) {
1122                 fprintf (stderr, "Can't parse gateway NID \"%s\"\n", argv[1]);
1123                 return (-1);
1124         }
1125
1126         if (argc == 3) {
1127                 hops = strtoul(argv[2], &end, 0);
1128                 if (hops >= 256 || *end != 0) {
1129                         fprintf (stderr, "Can't parse hopcount \"%s\"\n", argv[2]);
1130                         return -1;
1131                 }
1132         }
1133
1134         LIBCFS_IOC_INIT(data);
1135         data.ioc_net = g_net;
1136         data.ioc_count = hops;
1137         data.ioc_nid = gateway_nid;
1138
1139         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_ADD_ROUTE, &data);
1140         if (rc != 0) {
1141                 fprintf (stderr, "IOC_LIBCFS_ADD_ROUTE failed: %s\n", strerror (errno));
1142                 return (-1);
1143         }
1144
1145         return (0);
1146 }
1147
1148 int
1149 jt_ptl_del_route (int argc, char **argv)
1150 {
1151         struct libcfs_ioctl_data data;
1152         lnet_nid_t               nid;
1153         int                      rc;
1154
1155         if (argc != 2) {
1156                 fprintf (stderr, "usage: %s gatewayNID\n", argv[0]);
1157                 return (0);
1158         }
1159
1160         if (!libcfs_str2anynid(&nid, argv[1])) {
1161                 fprintf (stderr, "Can't parse gateway NID "
1162                          "\"%s\"\n", argv[1]);
1163                 return -1;
1164         }
1165
1166         LIBCFS_IOC_INIT(data);
1167         data.ioc_net = g_net_set ? g_net : LNET_NIDNET(LNET_NID_ANY);
1168         data.ioc_nid = nid;
1169
1170         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_DEL_ROUTE, &data);
1171         if (rc != 0) {
1172                 fprintf (stderr, "IOC_LIBCFS_DEL_ROUTE (%s) failed: %s\n",
1173                          libcfs_nid2str(nid), strerror (errno));
1174                 return (-1);
1175         }
1176
1177         return (0);
1178 }
1179
1180 int
1181 jt_ptl_notify_router (int argc, char **argv)
1182 {
1183         struct libcfs_ioctl_data data;
1184         int                      enable;
1185         lnet_nid_t               nid;
1186         int                      rc;
1187         struct timeval           now;
1188         time_t                   when;
1189
1190         if (argc < 3)
1191         {
1192                 fprintf (stderr, "usage: %s targetNID <up/down> [<time>]\n", 
1193                          argv[0]);
1194                 return (0);
1195         }
1196
1197         nid = libcfs_str2nid(argv[1]);
1198         if (nid == LNET_NID_ANY) {
1199                 fprintf (stderr, "Can't parse target NID \"%s\"\n", argv[1]);
1200                 return (-1);
1201         }
1202
1203         if (lnet_parse_bool (&enable, argv[2]) != 0) {
1204                 fprintf (stderr, "Can't parse boolean %s\n", argv[2]);
1205                 return (-1);
1206         }
1207
1208         gettimeofday(&now, NULL);
1209
1210         if (argc < 4) {
1211                 when = now.tv_sec;
1212         } else if (lnet_parse_time (&when, argv[3]) != 0) {
1213                 fprintf(stderr, "Can't parse time %s\n"
1214                         "Please specify either 'YYYY-MM-DD-HH:MM:SS'\n"
1215                         "or an absolute unix time in seconds\n", argv[3]);
1216                 return (-1);
1217         } else if (when > now.tv_sec) {
1218                 fprintf (stderr, "%s specifies a time in the future\n",
1219                          argv[3]);
1220                 return (-1);
1221         }
1222
1223         LIBCFS_IOC_INIT(data);
1224         data.ioc_nid = nid;
1225         data.ioc_flags = enable;
1226         /* Yeuch; 'cept I need a __u64 on 64 bit machines... */
1227         data.ioc_u64[0] = (__u64)when;
1228
1229         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_NOTIFY_ROUTER, &data);
1230         if (rc != 0) {
1231                 fprintf (stderr, "IOC_LIBCFS_NOTIFY_ROUTER (%s) failed: %s\n",
1232                          libcfs_nid2str(nid), strerror (errno));
1233                 return (-1);
1234         }
1235
1236         return (0);
1237 }
1238
1239 int
1240 jt_ptl_print_routes (int argc, char **argv)
1241 {
1242         struct libcfs_ioctl_data  data;
1243         int                       rc;
1244         int                       index;
1245         __u32                     net;
1246         lnet_nid_t                nid;
1247         unsigned int              hops;
1248         int                       alive;
1249
1250         for (index = 0;;index++)
1251         {
1252                 LIBCFS_IOC_INIT(data);
1253                 data.ioc_count = index;
1254
1255                 rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_GET_ROUTE, &data);
1256                 if (rc != 0)
1257                         break;
1258
1259                 net     = data.ioc_net;
1260                 hops    = data.ioc_count;
1261                 nid     = data.ioc_nid;
1262                 alive   = data.ioc_flags;
1263
1264                 printf ("net %18s hops %u gw %32s %s\n", 
1265                         libcfs_net2str(net), hops,
1266                         libcfs_nid2str(nid), alive ? "up" : "down");
1267         }
1268
1269         if (errno != ENOENT)
1270                 fprintf(stderr, "Error getting routes: %s: check dmesg.\n",
1271                         strerror(errno));
1272
1273         return (0);
1274 }
1275
1276 static int
1277 lwt_control(int enable, int clear)
1278 {
1279         struct libcfs_ioctl_data data;
1280         int                      rc;
1281
1282         LIBCFS_IOC_INIT(data);
1283         data.ioc_flags = (enable ? 1 : 0) | (clear ? 2 : 0);
1284
1285         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_LWT_CONTROL, &data);
1286         if (rc == 0)
1287                 return (0);
1288
1289         fprintf(stderr, "IOC_LIBCFS_LWT_CONTROL failed: %s\n",
1290                 strerror(errno));
1291         return (-1);
1292 }
1293
1294 static int
1295 lwt_snapshot(cycles_t *now, int *ncpu, int *totalsize,
1296              lwt_event_t *events, int size)
1297 {
1298         struct libcfs_ioctl_data data;
1299         int                      rc;
1300
1301         LIBCFS_IOC_INIT(data);
1302         data.ioc_pbuf1 = (char *)events;
1303         data.ioc_plen1 = size;
1304
1305         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_LWT_SNAPSHOT, &data);
1306         if (rc != 0) {
1307                 fprintf(stderr, "IOC_LIBCFS_LWT_SNAPSHOT failed: %s\n",
1308                         strerror(errno));
1309                 return (-1);
1310         }
1311
1312         /* crappy overloads */
1313         if (data.ioc_u32[2] != sizeof(lwt_event_t) ||
1314             data.ioc_u32[3] != offsetof(lwt_event_t, lwte_where)) {
1315                 fprintf(stderr,"kernel/user LWT event mismatch %d(%d),%d(%d)\n",
1316                         (int)data.ioc_u32[2], (int)sizeof(lwt_event_t),
1317                         (int)data.ioc_u32[3],
1318                         (int)offsetof(lwt_event_t, lwte_where));
1319                 return (-1);
1320         }
1321
1322         if (now != NULL)
1323                 *now = data.ioc_u64[0];
1324
1325         LASSERT (data.ioc_u32[0] != 0);
1326         if (ncpu != NULL)
1327                 *ncpu = data.ioc_u32[0];
1328
1329         LASSERT (data.ioc_u32[1] != 0);
1330         if (totalsize != NULL)
1331                 *totalsize = data.ioc_u32[1];
1332
1333         return (0);
1334 }
1335
1336 static char *
1337 lwt_get_string(char *kstr)
1338 {
1339         char                     *ustr;
1340         struct libcfs_ioctl_data  data;
1341         int                       size;
1342         int                       rc;
1343
1344         /* FIXME: this could maintain a symbol table since we expect to be
1345          * looking up the same strings all the time... */
1346
1347         LIBCFS_IOC_INIT(data);
1348         data.ioc_pbuf1 = kstr;
1349         data.ioc_plen1 = 1;        /* non-zero just to fool portal_ioctl_is_invalid() */
1350         data.ioc_pbuf2 = NULL;
1351         data.ioc_plen2 = 0;
1352
1353         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_LWT_LOOKUP_STRING, &data);
1354         if (rc != 0) {
1355                 fprintf(stderr, "IOC_LIBCFS_LWT_LOOKUP_STRING failed: %s\n",
1356                         strerror(errno));
1357                 return (NULL);
1358         }
1359
1360         size = data.ioc_count;
1361         ustr = (char *)malloc(size);
1362         if (ustr == NULL) {
1363                 fprintf(stderr, "Can't allocate string storage of size %d\n",
1364                         size);
1365                 return (NULL);
1366         }
1367
1368         LIBCFS_IOC_INIT(data);
1369         data.ioc_pbuf1 = kstr;
1370         data.ioc_plen1 = 1;        /* non-zero just to fool portal_ioctl_is_invalid() */
1371         data.ioc_pbuf2 = ustr;
1372         data.ioc_plen2 = size;
1373
1374         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_LWT_LOOKUP_STRING, &data);
1375         if (rc != 0) {
1376                 fprintf(stderr, "IOC_LIBCFS_LWT_LOOKUP_STRING failed: %s\n",
1377                         strerror(errno));
1378                 return (NULL);
1379         }
1380
1381         LASSERT(strlen(ustr) == size - 1);
1382         return (ustr);
1383 }
1384
1385 static void
1386 lwt_put_string(char *ustr)
1387 {
1388         free(ustr);
1389 }
1390
1391 static int
1392 lwt_print(FILE *f, cycles_t t0, cycles_t tlast, double mhz, int cpu, lwt_event_t *e)
1393 {
1394 #ifndef __WORDSIZE
1395 # error "__WORDSIZE not defined"
1396 #elif __WORDSIZE == 32
1397 # define XFMT "%#010lx"
1398 #elif __WORDSIZE== 64
1399 # define XFMT "%#018lx"
1400 #else
1401 # error "Unexpected __WORDSIZE"
1402 #endif
1403         char           *where = lwt_get_string(e->lwte_where);
1404
1405         if (where == NULL)
1406                 return (-1);
1407
1408         fprintf(f, XFMT" "XFMT" "XFMT" "XFMT": "XFMT" %2d %10.6f %10.2f %s\n",
1409                 e->lwte_p1, e->lwte_p2, e->lwte_p3, e->lwte_p4,
1410                 (long)e->lwte_task, cpu, (e->lwte_when - t0) / (mhz * 1000000.0),
1411                 (t0 == e->lwte_when) ? 0.0 : (e->lwte_when - tlast) / mhz,
1412                 where);
1413
1414         lwt_put_string(where);
1415
1416         return (0);
1417 #undef XFMT
1418 }
1419
1420 double
1421 get_cycles_per_usec ()
1422 {
1423         FILE      *f = fopen ("/proc/cpuinfo", "r");
1424         double     mhz;
1425         char      line[64];
1426
1427         if (f != NULL) {
1428                 while (fgets (line, sizeof (line), f) != NULL)
1429                         if (sscanf (line, "cpu MHz : %lf", &mhz) == 1) {
1430                                 fclose (f);
1431                                 return (mhz);
1432                         }
1433                 fclose (f);
1434         }
1435
1436         fprintf (stderr, "Can't read/parse /proc/cpuinfo\n");
1437         return (1000.0);
1438 }
1439
1440 int
1441 jt_ptl_lwt(int argc, char **argv)
1442 {
1443         const int       lwt_max_cpus = 32;
1444         int             ncpus;
1445         int             totalspace;
1446         int             nevents_per_cpu;
1447         lwt_event_t    *events;
1448         lwt_event_t    *cpu_event[lwt_max_cpus + 1];
1449         lwt_event_t    *next_event[lwt_max_cpus];
1450         lwt_event_t    *first_event[lwt_max_cpus];
1451         int             cpu;
1452         lwt_event_t    *e;
1453         int             rc;
1454         int             i;
1455         double          mhz;
1456         cycles_t        t0;
1457         cycles_t        tlast;
1458         cycles_t        tnow;
1459         struct timeval  tvnow;
1460         int             printed_date = 0;
1461         int             nlines = 0;
1462         FILE           *f = stdout;
1463
1464         if (argc < 2 ||
1465             (strcmp(argv[1], "start") &&
1466              strcmp(argv[1], "stop"))) {
1467                 fprintf(stderr, 
1468                         "usage:  %s start\n"
1469                         "        %s stop [fname]\n", argv[0], argv[0]);
1470                 return (-1);
1471         }
1472
1473         if (!strcmp(argv[1], "start")) {
1474                 /* disable */
1475                 if (lwt_control(0, 0) != 0)
1476                         return (-1);
1477
1478                 /* clear */
1479                 if (lwt_control(0, 1) != 0)
1480                         return (-1);
1481
1482                 /* enable */
1483                 if (lwt_control(1, 0) != 0)
1484                         return (-1);
1485
1486                 return (0);
1487         }
1488
1489         if (lwt_snapshot(NULL, &ncpus, &totalspace, NULL, 0) != 0)
1490                 return (-1);
1491
1492         if (ncpus > lwt_max_cpus) {
1493                 fprintf(stderr, "Too many cpus: %d (%d)\n",
1494                         ncpus, lwt_max_cpus);
1495                 return (-1);
1496         }
1497
1498         events = (lwt_event_t *)malloc(totalspace);
1499         if (events == NULL) {
1500                 fprintf(stderr, "Can't allocate %d\n", totalspace);
1501                 return (-1);
1502         }
1503
1504         if (lwt_control(0, 0) != 0) {           /* disable */
1505                 free(events);
1506                 return (-1);
1507         }
1508
1509         if (lwt_snapshot(&tnow, NULL, NULL, events, totalspace)) {
1510                 free(events);
1511                 return (-1);
1512         }
1513
1514         /* we want this time to be sampled at snapshot time */
1515         gettimeofday(&tvnow, NULL);
1516
1517         if (argc > 2) {
1518                 f = fopen (argv[2], "w");
1519                 if (f == NULL) {
1520                         fprintf(stderr, "Can't open %s for writing: %s\n", argv[2], strerror (errno));
1521                         free(events);
1522                         return (-1);
1523                 }
1524         }
1525
1526         mhz = get_cycles_per_usec();
1527
1528         /* carve events into per-cpu slices */
1529         nevents_per_cpu = totalspace / (ncpus * sizeof(lwt_event_t));
1530         for (cpu = 0; cpu <= ncpus; cpu++)
1531                 cpu_event[cpu] = &events[cpu * nevents_per_cpu];
1532
1533         /* find the earliest event on each cpu */
1534         for (cpu = 0; cpu < ncpus; cpu++) {
1535                 first_event[cpu] = NULL;
1536
1537                 for (e = cpu_event[cpu]; e < cpu_event[cpu + 1]; e++) {
1538
1539                         if (e->lwte_where == NULL) /* not an event */
1540                                 continue;
1541
1542                         if (first_event[cpu] == NULL ||
1543                             first_event[cpu]->lwte_when > e->lwte_when)
1544                                 first_event[cpu] = e;
1545                 }
1546
1547                 next_event[cpu] = first_event[cpu];
1548         }
1549
1550         t0 = tlast = 0;
1551         for (cpu = 0; cpu < ncpus; cpu++) {
1552                 e = first_event[cpu];
1553                 if (e == NULL)                  /* no events this cpu */
1554                         continue;
1555
1556                 if (e == cpu_event[cpu])
1557                         e = cpu_event[cpu + 1] - 1;
1558                 else
1559                         e = e - 1;
1560
1561                 /* If there's an event immediately before the first one, this
1562                  * cpu wrapped its event buffer */
1563                 if (e->lwte_where == NULL)
1564                         continue;
1565
1566                 /* We should only start outputting events from the most recent
1567                  * first event in any wrapped cpu.  Events before this time on
1568                  * other cpus won't have any events from this CPU to interleave
1569                  * with. */
1570                 if (t0 < first_event[cpu]->lwte_when)
1571                         t0 = first_event[cpu]->lwte_when;
1572         }
1573
1574         for (;;) {
1575                 /* find which cpu has the next event */
1576                 cpu = -1;
1577                 for (i = 0; i < ncpus; i++) {
1578
1579                         if (next_event[i] == NULL) /* this cpu exhausted */
1580                                 continue;
1581
1582                         if (cpu < 0 ||
1583                             next_event[i]->lwte_when < next_event[cpu]->lwte_when)
1584                                 cpu = i;
1585                 }
1586
1587                 if (cpu < 0)                    /* all cpus exhausted */
1588                         break;
1589
1590                 if (t0 == 0) {
1591                         /* no wrapped cpus and this is he first ever event */
1592                         t0 = next_event[cpu]->lwte_when;
1593                 }
1594
1595                 if (t0 <= next_event[cpu]->lwte_when) {
1596                         /* on or after the first event */
1597                         if (!printed_date) {
1598                                 cycles_t du = (tnow - t0) / mhz;
1599                                 time_t   then = tvnow.tv_sec - du/1000000;
1600
1601                                 if (du % 1000000 > tvnow.tv_usec)
1602                                         then--;
1603
1604                                 fprintf(f, "%s", ctime(&then));
1605                                 printed_date = 1;
1606                         }
1607
1608                         rc = lwt_print(f, t0, tlast, mhz, cpu, next_event[cpu]);
1609                         if (rc != 0)
1610                                 break;
1611
1612                         if (++nlines % 10000 == 0 && f != stdout) {
1613                                 /* show some activity... */
1614                                 printf(".");
1615                                 fflush (stdout);
1616                         }
1617                 }
1618
1619                 tlast = next_event[cpu]->lwte_when;
1620
1621                 next_event[cpu]++;
1622                 if (next_event[cpu] == cpu_event[cpu + 1])
1623                         next_event[cpu] = cpu_event[cpu];
1624
1625                 if (next_event[cpu]->lwte_where == NULL ||
1626                     next_event[cpu] == first_event[cpu])
1627                         next_event[cpu] = NULL;
1628         }
1629
1630         if (f != stdout) {
1631                 printf("\n");
1632                 fclose(f);
1633         }
1634
1635         free(events);
1636         return (0);
1637 }
1638
1639 int jt_ptl_memhog(int argc, char **argv)
1640 {
1641         static int                gfp = 0;        /* sticky! */
1642
1643         struct libcfs_ioctl_data  data;
1644         int                       rc;
1645         int                       count;
1646         char                     *end;
1647
1648         if (argc < 2)  {
1649                 fprintf(stderr, "usage: %s <npages> [<GFP flags>]\n", argv[0]);
1650                 return 0;
1651         }
1652
1653         count = strtol(argv[1], &end, 0);
1654         if (count < 0 || *end != 0) {
1655                 fprintf(stderr, "Can't parse page count '%s'\n", argv[1]);
1656                 return -1;
1657         }
1658
1659         if (argc >= 3) {
1660                 rc = strtol(argv[2], &end, 0);
1661                 if (*end != 0) {
1662                         fprintf(stderr, "Can't parse gfp flags '%s'\n", argv[2]);
1663                         return -1;
1664                 }
1665                 gfp = rc;
1666         }
1667
1668         LIBCFS_IOC_INIT(data);
1669         data.ioc_count = count;
1670         data.ioc_flags = gfp;
1671         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_MEMHOG, &data);
1672
1673         if (rc != 0) {
1674                 fprintf(stderr, "memhog %d failed: %s\n", count, strerror(errno));
1675                 return -1;
1676         }
1677
1678         printf("memhog %d OK\n", count);
1679         return 0;
1680 }
1681
1682 int jt_ptl_testprotocompat(int argc, char **argv)
1683 {
1684         struct libcfs_ioctl_data  data;
1685         int                       rc;
1686         int                       flags;
1687         char                     *end;
1688
1689         if (argc < 2)  {
1690                 fprintf(stderr, "usage: %s <number>\n", argv[0]);
1691                 return 0;
1692         }
1693
1694         flags = strtol(argv[1], &end, 0);
1695         if (flags < 0 || *end != 0) {
1696                 fprintf(stderr, "Can't parse flags '%s'\n", argv[1]);
1697                 return -1;
1698         }
1699
1700         LIBCFS_IOC_INIT(data);
1701         data.ioc_flags = flags;
1702         rc = l_ioctl(LNET_DEV_ID, IOC_LIBCFS_TESTPROTOCOMPAT, &data);
1703
1704         if (rc != 0) {
1705                 fprintf(stderr, "test proto compat %x failed: %s\n",
1706                         flags, strerror(errno));
1707                 return -1;
1708         }
1709
1710         printf("test proto compat %x OK\n", flags);
1711         return 0;
1712 }
1713
1714