1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
2 * vim:expandtab:shiftwidth=8:tabstop=8:
4 * Copyright (C) 2001, 2002 Cluster File Systems, Inc.
6 * This file is a modified version of the ptlctl tool which is
7 * part of Portals, http://www.sf.net/projects/lustre/
9 * Portals is free software; you can redistribute it and/or
10 * modify it under the terms of version 2 of the GNU General Public
11 * License as published by the Free Software Foundation.
13 * Portals is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with Portals; if not, write to the Free Software
20 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
25 #include <sys/types.h>
26 #include <sys/socket.h>
27 #include <netinet/tcp.h>
32 #include <sys/ioctl.h>
36 #include <portals/api-support.h>
40 static char rawbuf[8192];
41 static char *buf = rawbuf;
42 static int max = 8192;
44 static unsigned int g_nid = 0;
45 static unsigned int g_nal = 0;
46 static unsigned short g_port = 0;
47 static int g_pfd = -1;
49 static int g_socket_txmem = 0;
50 static int g_socket_rxmem = 0;
51 static int g_socket_nonagle = 1;
53 static name2num_t nalnames[] = {
60 int network_setup(int argc, char **argv) {
65 static name2num_t *name2num_lookup_name(name2num_t *table, char *str) {
66 while (table->name != NULL)
67 if (!strcmp (str, table->name))
74 static name2num_t *name2num_lookup_num(name2num_t *table, int num) {
75 while (table->name != NULL)
76 if (num == table->num)
83 static int name2nal(char *str) {
84 name2num_t *e = name2num_lookup_name (nalnames, str);
86 return ((e == NULL) ? 0 : e->num);
89 static char *nal2name (int nal) {
90 name2num_t *e = name2num_lookup_num (nalnames, nal);
92 return ((e == NULL) ? "???" : e->name);
95 static int nid2nal(ptl_nid_t nid) {
96 /* BIG pragmatic assumption */
97 return ((((__u32)nid) & 0xffff0000) != 0 ? SOCKNAL : QSWNAL);
100 static int parse_nid(ptl_nid_t *nidp, char *str) {
107 if (sscanf (str, "%d.%d.%d.%d", &a, &b, &c, &d) == 4 &&
108 (a & ~0xff) == 0 && (b & ~0xff) == 0 &&
109 (c & ~0xff) == 0 && (d & ~0xff) == 0) {
110 __u32 addr = (a<<24)|(b<<16)|(c<<8)|d;
111 *nidp = (ptl_nid_t)addr;
115 if ((('a' <= str[0] && str[0] <= 'z') ||
116 ('A' <= str[0] && str[0] <= 'Z')) &&
117 (he = gethostbyname (str)) != NULL) {
118 __u32 addr = *(__u32 *)he->h_addr;
119 *nidp = (ptl_nid_t)ntohl(addr); /* HOST byte order */
123 if (sscanf (str, "%i", &a) == 1) {
124 *nidp = (ptl_nid_t)a;
131 static char *nid2str (char *buffer, ptl_nid_t nid) {
132 switch (nid2nal(nid)) {
134 sprintf (buffer, "%Ld", nid);
138 __u32 addr = htonl((__u32)nid); /* back to NETWORK byte order*/
139 struct hostent *he = gethostbyaddr ((const char *)&addr,
140 sizeof(addr), AF_INET);
143 strcpy (buffer, he->h_name);
146 sprintf(buffer, "%d.%d.%d.%d", (addr>>24)&0xff,
147 (addr>>16)&0xff, (addr>>8)&0xff, addr&0xff);
153 sprintf (buffer, "nid2nal broken");
158 int jt_net_network(int argc, char **argv) {
161 if (argc != 2 || (nal = name2nal (argv[1])) == 0)
168 int jt_net_connect(int argc, char **argv) {
173 fprintf(stderr, "Error: you must run the 'setup' command "
178 if (g_nal == SOCKNAL) {
180 struct portal_ioctl_data data;
181 struct sockaddr_in srvaddr;
192 he = gethostbyname(argv[1]);
194 fprintf(stderr, "gethostbyname error: %s\n",
199 g_port = atol(argv[2]);
201 memset(&srvaddr, 0, sizeof(srvaddr));
202 srvaddr.sin_family = AF_INET;
203 srvaddr.sin_port = htons(g_port);
204 srvaddr.sin_addr.s_addr = *(__u32 *)he->h_addr;
206 fd = socket(PF_INET, SOCK_STREAM, 0);
208 fprintf(stderr, "socket() failed: %s\n",
213 if (g_socket_nonagle) {
215 if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY,
216 &o, sizeof (o)) != 0) {
217 fprintf(stderr, "cannot disable nagle: %s\n",
223 if (g_socket_rxmem != 0) {
225 if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF,
226 &o, sizeof (o)) != 0) {
227 fprintf(stderr, "cannot set receive buffer "
228 "size: %s\n", strerror(errno));
233 if (g_socket_txmem != 0) {
235 if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF,
236 &o, sizeof (o)) != 0) {
237 fprintf(stderr, "cannot set send buffer "
238 "size: %s\n", strerror(errno));
243 rc = connect(fd, (struct sockaddr *)&srvaddr, sizeof(srvaddr));
245 fprintf(stderr, "connect() failed: %s\n",
250 olen = sizeof (txmem);
251 if (getsockopt (fd, SOL_SOCKET, SO_SNDBUF, &txmem, &olen) != 0)
252 fprintf(stderr, "Can't get send buffer size: %s\n",
254 olen = sizeof (rxmem);
255 if (getsockopt (fd, SOL_SOCKET, SO_RCVBUF, &rxmem, &olen) != 0)
256 fprintf(stderr, "Can't get receive buffer size: %s\n",
258 olen = sizeof (nonagle);
259 if (getsockopt(fd,IPPROTO_TCP,TCP_NODELAY,&nonagle,&olen) != 0)
260 fprintf(stderr, "Can't get nagle: %s\n",
263 printf("Connected to %s (snd %d, rcv %d, nagle %s)\n",
264 argv[1],txmem,rxmem,nonagle ? "Disabled" : "Enabled");
266 PORTAL_IOC_INIT(data);
268 /* HOST byte order */
269 data.ioc_nid = ntohl(srvaddr.sin_addr.s_addr);
271 rc = ioctl(g_pfd, IOC_PORTAL_REGISTER_CLIENT_FD, &data);
273 fprintf(stderr, "failed to register fd with portals: "
274 "%s\n", strerror(errno));
278 g_nid = ntohl (srvaddr.sin_addr.s_addr); /* HOST byte order */
279 printf("Connection to 0x%x registered with socknal\n",
284 fprintf(stderr, "close failed: %d\n", rc);
286 } else if (g_nal == QSWNAL) {
287 g_nid = atoi(argv[1]);
288 } else if (g_nal == GMNAL) {
289 g_nid = atoi(argv[1]);
291 fprintf(stderr, "This should never happen. Also it is very "
298 int jt_net_disconnect(int argc, char **argv) {
303 fprintf(stderr, "Error: you must run the 'network' command "
308 if (g_nal == SOCKNAL) {
310 struct portal_ioctl_data data;
313 PORTAL_IOC_INIT(data);
315 he = gethostbyname(argv[1]);
317 fprintf(stderr, "gethostbyname error: %s\n",
322 /* HOST byte order */
323 data.ioc_nid = ntohl (*(__u32 *)he->h_addr);
326 fprintf(stderr, "Disconnecting ALL connections.\n");
327 /* leave ioc_nid zeroed == disconnect all */
329 rc = ioctl(g_pfd, IOC_PORTAL_CLOSE_CONNECTION, &data);
331 fprintf(stderr, "failed to remove connection: %s\n",
335 } else if (g_nal == QSWNAL) {
336 printf("'disconnect' doesn't make any sense for "
338 } else if (g_nal == GMNAL) {
339 printf("'disconnect' doesn't make any sense for "
342 fprintf(stderr, "This should never happen. Also it is very "
350 int jt_net_ping(int argc, char **argv) {
356 struct portal_ioctl_data data;
359 fprintf(stderr, "usage: %s nid [count] [size] [timeout "
360 (secs)]\n", argv[0]);
365 fprintf(stderr, "Error: you must run the 'setup' command "
370 if (parse_nid (&nid, argv[1]) != 0) {
371 fprintf (stderr, "Can\'t parse nid \"%s\"\n", argv[1]);
376 count = atol(argv[2]);
378 if (count < 0 || count > 20000) {
379 fprintf(stderr, "are you insane? "
380 "%ld is a crazy count.\n", count);
389 timeout = atol (argv[4]);
391 PORTAL_IOC_INIT (data);
392 data.ioc_count = count;
393 data.ioc_size = size;
394 data.ioc_nid = (__u32)nid;
395 data.ioc_nal = g_nal;
396 data.ioc_timeout = timeout;
398 rc = ioctl(g_pfd, IOC_PORTAL_PING, &data);
400 fprintf(stderr, "failed to start pinger: %s\n",
404 fprintf(stderr, "Pinger started, take cover...\n");
409 int jt_net_mynid(int argc, char **argv) {
412 char buf[1024], *hostname;
413 struct portal_ioctl_data data;
420 fprintf(stderr, "Error: you must run the 'setup' command "
425 if (g_nal == QSWNAL) {
426 fprintf(stderr, "'mynid' doesn't make any sense for elan.\n");
428 } else if (g_nal == GMNAL) {
429 fprintf(stderr, "'mynid' doesn't make any sense for GM.\n");
433 if (g_nal != SOCKNAL) {
434 fprintf(stderr, "This should never happen. Also it is very "
440 if (gethostname(buf, sizeof(buf)) != 0) {
441 fprintf(stderr, "gethostname failed: %s\n",
450 h = gethostbyname(hostname);
453 fprintf(stderr, "cannot get address for host '%s': %d\n",
457 mynid = (ptl_nid_t)ntohl (*(__u32 *)h->h_addr); /* HOST byte order */
459 PORTAL_IOC_INIT(data);
460 data.ioc_nid = (__u32)mynid;
462 rc = ioctl(g_pfd, IOC_PORTAL_REGISTER_MYNID, &data);
464 fprintf(stderr, "IOC_PORTAL_REGISTER_MYNID failed: %s\n",
467 printf("registered my nid 0x%Lx (%s)\n",
472 int jt_net_add_uuid(int argc, char **argv) {
475 struct portal_ioctl_data data;
476 ptl_nid_t nid = g_nid;
481 if (parse_nid(&nid, argv[2]) != 0) {
482 fprintf (stderr, "Can't parse NID %s\n", argv[2]);
487 fprintf(stderr, "Error: you must run the 'setup' command "
492 memset(&data, 0, sizeof(data));
494 data.ioc_inllen1 = strlen(argv[1]) + 1;
495 data.ioc_inlbuf1 = argv[1];
496 data.ioc_nal = g_nal;
497 if (portal_ioctl_pack(&data, &buf, max) != 0) {
498 fprintf(stderr, "portal_ioctl_pack failed.\n");
502 rc = ioctl(g_pfd, IOC_PORTAL_ADD_UUID, buf);
504 fprintf(stderr, "IOC_PORTAL_ADD_UUID failed: %s\n",
509 printf ("Added uuid %s: %s\n", argv[1], nid2str (tmp, nid));
514 static int jt_close_uuid(int argc, char **argv)
517 struct portal_ioctl_data data;
520 fprintf(stderr, "usage: %s <uuid>\n", argv[0]);
525 fprintf(stderr, "Error: you must run the 'setup' command "
530 memset(&data, 0, sizeof(data));
531 data.ioc_inllen1 = strlen(argv[1]) + 1;
532 data.ioc_inlbuf1 = argv[1];
533 data.ioc_nal = g_nal;
534 if (portal_ioctl_pack(&data, &buf, max) != 0) {
535 fprintf(stderr, "portal_ioctl_pack failed.\n");
539 rc = ioctl(g_pfd, IOC_PORTAL_CLOSE_UUID, buf);
541 fprintf(stderr, "IOC_PORTAL_CLOSE_UUID failed: %s\n",
549 int jt_net_del_uuid(int argc, char **argv) {
551 struct portal_ioctl_data data;
557 fprintf(stderr, "Error: you must run the 'setup' command "
562 memset(&data, 0, sizeof(data));
563 data.ioc_inllen1 = strlen(argv[1]) + 1;
564 data.ioc_inlbuf1 = argv[1];
565 data.ioc_nal = g_nal;
566 if (portal_ioctl_pack(&data, &buf, max) != 0) {
567 fprintf(stderr, "portal_ioctl_pack failed.\n");
571 rc = ioctl(g_pfd, IOC_PORTAL_DEL_UUID, buf);
573 fprintf(stderr, "IOC_PORTAL_DEL_UUID failed: %s\n",
580 int jt_net_add_route (int argc, char **argv) {
581 struct portal_ioctl_data data;
584 ptl_nid_t gateway_nid;
591 if (parse_nid (&gateway_nid, argv[1]) != 0) {
592 fprintf (stderr, "Can't parse gateway NID \"%s\"\n", argv[1]);
596 gateway_nal = nid2nal (gateway_nid);
597 if (parse_nid (&nid1, argv[2]) != 0) {
598 fprintf(stderr, "Can't parse first target NID \"%s\"\n",
605 } else if (parse_nid (&nid2, argv[3]) != 0) {
606 fprintf(stderr, "Can't parse second target NID \"%s\"\n",
611 PORTAL_IOC_INIT(data);
612 data.ioc_nid = gateway_nid;
613 data.ioc_nal = gateway_nal;
614 data.ioc_nid2 = MIN (nid1, nid2);
615 data.ioc_nid3 = MAX (nid1, nid2);
617 rc = ioctl (g_pfd, IOC_PORTAL_ADD_ROUTE, &data);
619 fprintf(stderr, "IOC_PORTAL_ADD_ROUTE failed: %s\n",
627 int jt_net_del_route(int argc, char **argv) {
628 struct portal_ioctl_data data;
635 if (parse_nid (&nid, argv[1]) != 0) {
636 fprintf (stderr, "Can't parse target NID \"%s\"\n", argv[1]);
640 PORTAL_IOC_INIT(data);
643 rc = ioctl (g_pfd, IOC_PORTAL_DEL_ROUTE, &data);
645 fprintf(stderr, "IOC_PORTAL_DEL_ROUTE (0x%Lx) failed: %s\n",
646 nid, strerror (errno));
653 int jt_net_route_list(int argc, char **argv) {
655 struct portal_ioctl_data data;
659 ptl_nid_t gateway_nid;
663 for (index = 0;;index++) {
664 PORTAL_IOC_INIT(data);
665 data.ioc_count = index;
667 rc = ioctl (g_pfd, IOC_PORTAL_GET_ROUTE, &data);
671 gateway_nal = data.ioc_nal;
672 gateway_nid = data.ioc_nid;
673 nid1 = data.ioc_nid2;
674 nid2 = data.ioc_nid3;
676 printf ("%8s %18s : %s - %s\n",
677 nal2name (gateway_nal),
678 nid2str (buffer[0], gateway_nid),
679 nid2str (buffer[1], nid1),
680 nid2str (buffer[2], nid2));
685 int jt_net_recv_mem(int argc, char **argv) {
689 if (Parser_size (&size, argv[1]) != 0 || size < 0) {
690 fprintf (stderr, "Can't parse size %s\n", argv[1]);
694 g_socket_rxmem = size;
697 printf ("Socket rxmem = %d\n", g_socket_rxmem);
701 int jt_net_send_mem(int argc, char **argv) {
705 if (Parser_size (&size, argv[1]) != 0 || size < 0) {
706 fprintf (stderr, "Can't parse size %s\n", argv[1]);
709 g_socket_txmem = size;
712 printf ("Socket txmem = %d\n", g_socket_txmem);
716 int jt_net_nagle(int argc, char **argv) {
720 if (Parser_bool (&enable, argv[1]) != 0) {
721 fprintf (stderr, "Can't parse boolean %s\n", argv[1]);
724 g_socket_nonagle = !enable;
727 printf ("Nagle %s\n", g_socket_nonagle ? "disabled" : "enabled");