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