lnet/utils/portals.c

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