1 2 /* 3 * write.c 4 * 5 * Copyright (c) 2002 Archie L. Cobbs 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 Archie L. Cobbs; 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 14 * 15 * THIS SOFTWARE IS BEING PROVIDED BY ARCHIE L. COBBS AS IS", AND TO 16 * THE MAXIMUM EXTENT PERMITTED BY LAW, ARCHIE L. COBBS MAKES NO 17 * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING THIS SOFTWARE, 18 * INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED WARRANTIES OF 19 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT. 20 * ARCHIE L. COBBS DOES NOT WARRANT, GUARANTEE, OR MAKE ANY 21 * REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS OF THE USE OF THIS 22 * SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY, RELIABILITY OR OTHERWISE. 23 * IN NO EVENT SHALL ARCHIE L. COBBS BE LIABLE FOR ANY DAMAGES 24 * RESULTING FROM OR ARISING OUT OF ANY USE OF THIS SOFTWARE, INCLUDING 25 * WITHOUT LIMITATION, ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, 26 * PUNITIVE, OR CONSEQUENTIAL DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR 27 * SERVICES, LOSS OF USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY 28 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 29 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 30 * THIS SOFTWARE, EVEN IF ARCHIE L. COBBS IS ADVISED OF THE POSSIBILITY 31 * OF SUCH DAMAGE. 32 */ 33 34 #include <sys/types.h> 35 #include <sys/socket.h> 36 37 #include <err.h> 38 #include <stdio.h> 39 #include <string.h> 40 #include <unistd.h> 41 42 #include <netgraph/ng_socket.h> 43 44 #include "ngctl.h" 45 46 #define BUF_SIZE 8192 47 48 static int WriteCmd(int ac, char **av); 49 50 const struct ngcmd write_cmd = { 51 WriteCmd, 52 "write hook < -f file | byte ... >", 53 "Send a data packet down the hook named by \"hook\".", 54 "The data may be contained in a file, or may be described directly" 55 " on the command line by supplying a sequence of bytes.", 56 { "w" } 57 }; 58 59 static int 60 WriteCmd(int ac, char **av) 61 { 62 u_int32_t sagbuf[64]; 63 struct sockaddr_ng *sag = (struct sockaddr_ng *)sagbuf; 64 u_char buf[BUF_SIZE]; 65 const char *hook; 66 FILE *fp; 67 u_int len; 68 int byte; 69 int i; 70 71 /* Get arguments */ 72 if (ac < 3) 73 return (CMDRTN_USAGE); 74 hook = av[1]; 75 76 /* Get data */ 77 if (strcmp(av[2], "-f") == 0) { 78 if (ac != 4) 79 return (CMDRTN_USAGE); 80 if ((fp = fopen(av[3], "r")) == NULL) { 81 warn("can't read file \"%s\"", av[3]); 82 return (CMDRTN_ERROR); 83 } 84 if ((len = fread(buf, 1, sizeof(buf), fp)) == 0) { 85 if (ferror(fp)) 86 warn("can't read file \"%s\"", av[3]); 87 else 88 warnx("file \"%s\" is empty", av[3]); 89 fclose(fp); 90 return (CMDRTN_ERROR); 91 } 92 fclose(fp); 93 } else { 94 for (i = 2, len = 0; i < ac && len < sizeof(buf); i++, len++) { 95 if (sscanf(av[i], "%i", &byte) != 1 96 || (byte < -128 || byte > 255)) { 97 warnx("invalid byte \"%s\"", av[i]); 98 return (CMDRTN_ERROR); 99 } 100 buf[len] = (u_char)byte; 101 } 102 if (len == 0) 103 return (CMDRTN_USAGE); 104 } 105 106 /* Send data */ 107 sag->sg_len = 3 + strlen(hook); 108 sag->sg_family = AF_NETGRAPH; 109 strlcpy(sag->sg_data, hook, sizeof(sagbuf) - 2); 110 if (sendto(dsock, buf, len, 111 0, (struct sockaddr *)sag, sag->sg_len) == -1) { 112 warn("writing to hook \"%s\"", hook); 113 return (CMDRTN_ERROR); 114 } 115 116 /* Done */ 117 return (CMDRTN_OK); 118 } 119 120