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