1129199Scognet 2129199Scognet/* 3129199Scognet * write.c 4129199Scognet * 5129199Scognet * Copyright (c) 2002 Archie L. Cobbs 6129199Scognet * All rights reserved. 7129199Scognet * 8129199Scognet * Subject to the following obligations and disclaimer of warranty, use and 9129199Scognet * redistribution of this software, in source or object code forms, with or 10129199Scognet * without modifications are expressly permitted by Archie L. Cobbs; 11135670Scognet * provided, however, that: 12129199Scognet * 1. Any and all reproductions of the source or object code must include the 13129199Scognet * copyright notice above and the following disclaimer of warranties 14129199Scognet * 15135670Scognet * THIS SOFTWARE IS BEING PROVIDED BY ARCHIE L. COBBS AS IS", AND TO 16129199Scognet * THE MAXIMUM EXTENT PERMITTED BY LAW, ARCHIE L. COBBS MAKES NO 17129199Scognet * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING THIS SOFTWARE, 18129199Scognet * INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED WARRANTIES OF 19129199Scognet * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT. 20144510Simp * ARCHIE L. COBBS DOES NOT WARRANT, GUARANTEE, OR MAKE ANY 21129199Scognet * REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS OF THE USE OF THIS 22129199Scognet * SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY, RELIABILITY OR OTHERWISE. 23136910Sru * IN NO EVENT SHALL ARCHIE L. COBBS BE LIABLE FOR ANY DAMAGES 24129199Scognet * RESULTING FROM OR ARISING OUT OF ANY USE OF THIS SOFTWARE, INCLUDING 25129199Scognet * WITHOUT LIMITATION, ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, 26129199Scognet * PUNITIVE, OR CONSEQUENTIAL DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR 27129199Scognet * SERVICES, LOSS OF USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY 28129199Scognet * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 29129199Scognet * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 30129199Scognet * THIS SOFTWARE, EVEN IF ARCHIE L. COBBS IS ADVISED OF THE POSSIBILITY 31129199Scognet * OF SUCH DAMAGE. 32129199Scognet * 33129199Scognet * $FreeBSD$ 34129199Scognet */ 35129199Scognet 36135670Scognet#include <sys/types.h> 37135670Scognet#include <sys/socket.h> 38140479Scognet 39140479Scognet#include <err.h> 40140479Scognet#include <stdio.h> 41140479Scognet#include <string.h> 42140479Scognet#include <unistd.h> 43140479Scognet 44147630Scognet#include <netgraph/ng_socket.h> 45147630Scognet 46147630Scognet#include "ngctl.h" 47129199Scognet 48129199Scognet#define BUF_SIZE 8192 49129199Scognet 50129199Scognetstatic int WriteCmd(int ac, char **av); 51129199Scognet 52129199Scognetconst struct ngcmd write_cmd = { 53129199Scognet WriteCmd, 54129199Scognet "write hook < -f file | byte ... >", 55129199Scognet "Send a data packet down the hook named by \"hook\".", 56129199Scognet "The data may be contained in a file, or may be described directly" 57129199Scognet " on the command line by supplying a sequence of bytes.", 58129199Scognet { "w" } 59135670Scognet}; 60135670Scognet 61135670Scognetstatic int 62129199ScognetWriteCmd(int ac, char **av) 63129199Scognet{ 64129199Scognet u_int32_t sagbuf[64]; 65 struct sockaddr_ng *sag = (struct sockaddr_ng *)sagbuf; 66 u_char buf[BUF_SIZE]; 67 const char *hook; 68 FILE *fp; 69 u_int len; 70 int byte; 71 int i; 72 73 /* Get arguments */ 74 if (ac < 3) 75 return (CMDRTN_USAGE); 76 hook = av[1]; 77 78 /* Get data */ 79 if (strcmp(av[2], "-f") == 0) { 80 if (ac != 4) 81 return (CMDRTN_USAGE); 82 if ((fp = fopen(av[3], "r")) == NULL) { 83 warn("can't read file \"%s\"", av[3]); 84 return (CMDRTN_ERROR); 85 } 86 if ((len = fread(buf, 1, sizeof(buf), fp)) == 0) { 87 if (ferror(fp)) 88 warn("can't read file \"%s\"", av[3]); 89 else 90 warnx("file \"%s\" is empty", av[3]); 91 fclose(fp); 92 return (CMDRTN_ERROR); 93 } 94 fclose(fp); 95 } else { 96 for (i = 2, len = 0; i < ac && len < sizeof(buf); i++, len++) { 97 if (sscanf(av[i], "%i", &byte) != 1 98 || (byte < -128 || byte > 255)) { 99 warnx("invalid byte \"%s\"", av[i]); 100 return (CMDRTN_ERROR); 101 } 102 buf[len] = (u_char)byte; 103 } 104 if (len == 0) 105 return (CMDRTN_USAGE); 106 } 107 108 /* Send data */ 109 sag->sg_len = 3 + strlen(hook); 110 sag->sg_family = AF_NETGRAPH; 111 strlcpy(sag->sg_data, hook, sizeof(sagbuf) - 2); 112 if (sendto(dsock, buf, len, 113 0, (struct sockaddr *)sag, sag->sg_len) == -1) { 114 warn("writing to hook \"%s\"", hook); 115 return (CMDRTN_ERROR); 116 } 117 118 /* Done */ 119 return (CMDRTN_OK); 120} 121 122