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