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