usr.sbin/ngctl/msg.c

   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.4 2005/12/05 02:40:27 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
  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
 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_PATHLEN+1];
 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 }