usr.sbin/ngctl/msg.c

   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  */
  39
  40 #include <sys/cdefs.h>
  41 __FBSDID("$FreeBSD$");
  42
  43 #include <err.h>
  44 #include <netgraph.h>
  45 #include <stdio.h>
  46 #include <stdlib.h>
  47 #include <string.h>
  48 #include <sysexits.h>
  49 #include <unistd.h>
  50
  51 #include "ngctl.h"
  52
  53 static int MsgCmd(int ac, char **av);
  54
  55 const struct ngcmd msg_cmd = {
  56         MsgCmd,
  57         "msg path command [args ... ]",
  58         "Send a netgraph control message to the node at \"path\"",
  59         "The msg command constructs a netgraph control message from the"
  60         " command name and ASCII arguments (if any) and sends that message"
  61         " to the node.  It does this by first asking the node to convert"
  62         " the ASCII message into binary format, and resending the result.",
  63         { "cmd" }
  64 };
  65
  66 static int
  67 MsgCmd(int ac, char **av)
  68 {
  69         char *buf;
  70         char *path, *cmdstr;
  71         int i, len;
  72
  73         /* Get arguments */
  74         if (ac < 3)
  75                 return (CMDRTN_USAGE);
  76         path = av[1];
  77         cmdstr = av[2];
  78
  79         /* Put command and arguments back together as one string */
  80         for (len = 1, i = 3; i < ac; i++)
  81                 len += strlen(av[i]) + 1;
  82         if ((buf = malloc(len)) == NULL) {
  83                 warn("malloc");
  84                 return (CMDRTN_ERROR);
  85         }
  86         for (*buf = '\0', i = 3; i < ac; i++) {
  87                 snprintf(buf + strlen(buf),
  88                     len - strlen(buf), " %s", av[i]);
  89         }
  90
  91         /* Send it */
  92         if (NgSendAsciiMsg(csock, path, "%s%s", cmdstr, buf) < 0) {
  93                 free(buf);
  94                 warn("send msg");
  95                 return (CMDRTN_ERROR);
  96         }
  97         free(buf);
  98
  99         /* See if a synchronous reply awaits */
 100         {
 101                 struct timeval tv;
 102                 fd_set rfds;
 103
 104                 FD_ZERO(&rfds);
 105                 FD_SET(csock, &rfds);
 106                 memset(&tv, 0, sizeof(tv));
 107                 switch (select(csock + 1, &rfds, NULL, NULL, &tv)) {
 108                 case -1:
 109                         err(EX_OSERR, "select");
 110                 case 0:
 111                         break;
 112                 default:
 113                         MsgRead();
 114                         break;
 115                 }
 116         }
 117
 118         /* Done */
 119         return (CMDRTN_OK);
 120 }
 121
 122 /*
 123  * Read and display the next incoming control message
 124  */
 125 void
 126 MsgRead(void)
 127 {
 128         struct ng_mesg *m, *m2;
 129         struct ng_mesg *ascii;
 130         char path[NG_PATHSIZ];
 131
 132         /* Get incoming message (in binary form) */
 133         if (NgAllocRecvMsg(csock, &m, path) < 0) {
 134                 warn("recv incoming message");
 135                 return;
 136         }
 137
 138         /* Ask originating node to convert message to ASCII */
 139         if (NgSendMsg(csock, path, NGM_GENERIC_COOKIE,
 140               NGM_BINARY2ASCII, m, sizeof(*m) + m->header.arglen) < 0
 141             || NgAllocRecvMsg(csock, &m2, NULL) < 0) {
 142                 printf("Rec'd %s %d from \"%s\":\n",
 143                     (m->header.flags & NGF_RESP) != 0 ? "response" : "command",
 144                     m->header.cmd, path);
 145                 if (m->header.arglen == 0)
 146                         printf("No arguments\n");
 147                 else
 148                         DumpAscii((const u_char *)m->data, m->header.arglen);
 149                 free(m);
 150                 return;
 151         }
 152
 153         /* Display message in ASCII form */
 154         free(m);
 155         ascii = (struct ng_mesg *)m2->data;
 156         printf("Rec'd %s \"%s\" (%d) from \"%s\":\n",
 157             (ascii->header.flags & NGF_RESP) != 0 ? "response" : "command",
 158             ascii->header.cmdstr, ascii->header.cmd, path);
 159         if (*ascii->data != '\0')
 160                 printf("Args:\t%s\n", ascii->data);
 161         else
 162                 printf("No arguments\n");
 163         free(m2);
 164 }
 165