xref: /dragonfly/usr.sbin/ngctl/msg.c (revision 2cd2d2b5)
1 
2 /*
3  * msg.c
4  *
5  * Copyright (c) 1999 Whistle Communications, Inc.
6  * All rights reserved.
7  *
8  * Subject to the following obligations and disclaimer of warranty, use and
9  * redistribution of this software, in source or object code forms, with or
10  * without modifications are expressly permitted by Whistle Communications;
11  * provided, however, that:
12  * 1. Any and all reproductions of the source or object code must include the
13  *    copyright notice above and the following disclaimer of warranties; and
14  * 2. No rights are granted, in any manner or form, to use Whistle
15  *    Communications, Inc. trademarks, including the mark "WHISTLE
16  *    COMMUNICATIONS" on advertising, endorsements, or otherwise except as
17  *    such appears in the above copyright notice or in the software.
18  *
19  * THIS SOFTWARE IS BEING PROVIDED BY WHISTLE COMMUNICATIONS "AS IS", AND
20  * TO THE MAXIMUM EXTENT PERMITTED BY LAW, WHISTLE COMMUNICATIONS MAKES NO
21  * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING THIS SOFTWARE,
22  * INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED WARRANTIES OF
23  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT.
24  * WHISTLE COMMUNICATIONS DOES NOT WARRANT, GUARANTEE, OR MAKE ANY
25  * REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS OF THE USE OF THIS
26  * SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY, RELIABILITY OR OTHERWISE.
27  * IN NO EVENT SHALL WHISTLE COMMUNICATIONS BE LIABLE FOR ANY DAMAGES
28  * RESULTING FROM OR ARISING OUT OF ANY USE OF THIS SOFTWARE, INCLUDING
29  * WITHOUT LIMITATION, ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
30  * PUNITIVE, OR CONSEQUENTIAL DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR
31  * SERVICES, LOSS OF USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY
32  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
33  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
34  * THIS SOFTWARE, EVEN IF WHISTLE COMMUNICATIONS IS ADVISED OF THE POSSIBILITY
35  * OF SUCH DAMAGE.
36  *
37  * $Whistle: msg.c,v 1.2 1999/11/29 23:38:35 archie Exp $
38  * $FreeBSD: src/usr.sbin/ngctl/msg.c,v 1.1.4.1 2000/07/27 22:05:36 archie Exp $
39  * $DragonFly: src/usr.sbin/ngctl/msg.c,v 1.2 2003/06/17 04:29:57 dillon Exp $
40  */
41 
42 #include "ngctl.h"
43 
44 #define BUF_SIZE	4096
45 
46 static int MsgCmd(int ac, char **av);
47 
48 const struct ngcmd msg_cmd = {
49 	MsgCmd,
50 	"msg path command [args ... ]",
51 	"Send a netgraph control message to the node at \"path\"",
52 	"The msg command constructs a netgraph control message from the"
53 	" command name and ASCII arguments (if any) and sends that message"
54 	" to the node.  It does this by first asking the node to convert"
55 	" the ASCII message into binary format, and resending the result.",
56 	{ "cmd" }
57 };
58 
59 static int
60 MsgCmd(int ac, char **av)
61 {
62 	char buf[BUF_SIZE];
63 	char *path, *cmdstr;
64 	int i;
65 
66 	/* Get arguments */
67 	if (ac < 3)
68 		return(CMDRTN_USAGE);
69 	path = av[1];
70 	cmdstr = av[2];
71 
72 	/* Put command and arguments back together as one string */
73 	for (*buf = '\0', i = 3; i < ac; i++) {
74 		snprintf(buf + strlen(buf),
75 		    sizeof(buf) - strlen(buf), " %s", av[i]);
76 	}
77 
78 	/* Send it */
79 	if (NgSendAsciiMsg(csock, path, "%s%s", cmdstr, buf) < 0) {
80 		warn("send msg");
81 		return(CMDRTN_ERROR);
82 	}
83 
84 	/* See if a synchronous reply awaits */
85 	{
86 		struct timeval tv;
87 		fd_set rfds;
88 
89 		FD_ZERO(&rfds);
90 		FD_SET(csock, &rfds);
91 		memset(&tv, 0, sizeof(tv));
92 		switch (select(csock + 1, &rfds, NULL, NULL, &tv)) {
93 		case -1:
94 			err(EX_OSERR, "select");
95 		case 0:
96 			break;
97 		default:
98 			MsgRead();
99 			break;
100 		}
101 	}
102 
103 	/* Done */
104 	return(CMDRTN_OK);
105 }
106 
107 /*
108  * Read and display the next incoming control message
109  */
110 void
111 MsgRead()
112 {
113 	u_char buf[2 * sizeof(struct ng_mesg) + BUF_SIZE];
114 	struct ng_mesg *const m = (struct ng_mesg *)buf;
115 	struct ng_mesg *const ascii = (struct ng_mesg *)m->data;
116 	char path[NG_PATHLEN+1];
117 
118 	/* Get incoming message (in binary form) */
119 	if (NgRecvMsg(csock, m, sizeof(buf), path) < 0) {
120 		warn("recv incoming message");
121 		return;
122 	}
123 
124 	/* Ask originating node to convert message to ASCII */
125 	if (NgSendMsg(csock, path, NGM_GENERIC_COOKIE,
126 	      NGM_BINARY2ASCII, m, sizeof(*m) + m->header.arglen) < 0
127 	    || NgRecvMsg(csock, m, sizeof(buf), NULL) < 0) {
128 		printf("Rec'd %s %d from \"%s\":\n",
129 		    (m->header.flags & NGF_RESP) != 0 ? "response" : "command",
130 		    m->header.cmd, path);
131 		if (m->header.arglen == 0)
132 			printf("No arguments\n");
133 		else
134 			DumpAscii(m->data, m->header.arglen);
135 		return;
136 	}
137 
138 	/* Display message in ASCII form */
139 	printf("Rec'd %s \"%s\" (%d) from \"%s\":\n",
140 	    (ascii->header.flags & NGF_RESP) != 0 ? "response" : "command",
141 	    ascii->header.cmdstr, ascii->header.cmd, path);
142 	if (*ascii->data != '\0')
143 		printf("Args:\t%s\n", ascii->data);
144 	else
145 		printf("No arguments\n");
146 }
147 
148