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