Head

GCC Code Coverage Report

Directory:	./		Exec	Total	Coverage
File:	usr.sbin/hostctl/hostctl.c	Lines:	0	89	0.0 %
Date:	2017-11-13	Branches:	0	70	0.0 %


/*	$OpenBSD: hostctl.c,v 1.3 2016/01/27 16:01:36 jmc Exp $	*/

/*
 * Copyright (c) 2016 Reyk Floeter <reyk@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/ioctl.h>
#include <sys/types.h>

#include <dev/pv/pvvar.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <err.h>
#include <vis.h>

#define KVBUFSZ	 4096				/* arbitrary value */

char		*path_pvbus = "/dev/pvbus0";	/* the first hv interface */
int		 qflag = 0;			/* quiet */
int		 tflag = 0;			/* show type */

__dead void	 usage(void);
int		 kvsetstr(char *, const char *, size_t);
int		 kvsetfile(char *, const char *, size_t);

__dead void
usage(void)
{
	extern char	*__progname;
	fprintf(stderr, "usage: %s [-qt] [-f device] "
	    "[-i input] [-o output] key [value]\n", __progname);
	exit(1);
}

int
kvsetstr(char *dst, const char *src, size_t dstlen)
{
	size_t	sz;

	/* Sanitize the string before sending it to the kernel and host */
	if ((sz = strnvis(dst, src, dstlen, VIS_SAFE | VIS_CSTYLE)) >= dstlen)
		return (-1);

	/* Remove trailing newline */
	if (dst[sz - 1] == '\n')
		dst[sz - 1] = '\0';

	return (0);
}

int
kvsetfile(char *dst, const char *input, size_t dstlen)
{
	char	*buf = NULL;
	int	 ret = -1;
	FILE	*fp;

	if (strcmp("-", input) == 0)
		fp = stdin;
	else if ((fp = fopen(input, "r")) == NULL)
		return (-1);

	if ((buf = calloc(1, dstlen)) == NULL)
		goto done;
	if (fread(buf, 1, dstlen - 1, fp) == 0)
		goto done;
	if (kvsetstr(dst, buf, dstlen) == -1)
		goto done;

	ret = 0;
 done:
	free(buf);
	if (fp != stdin)
		fclose(fp);
	return (ret);
}

int
main(int argc, char *argv[])
{
	const char		*key, *value, *in = NULL, *out = NULL;
	FILE			*outfp = stdout;
	int			fd, ret;
	struct pvbus_req	pvr;
	int			ch;
	unsigned long		cmd = 0;
	char			*str;

	while ((ch = getopt(argc, argv, "f:i:o:qt")) != -1) {
		switch (ch) {
		case 'f':
			path_pvbus = optarg;
			break;
		case 'i':
			in = optarg;
			break;
		case 'o':
			out = optarg;
			break;
		case 'q':
			qflag++;
			break;
		case 't':
			tflag++;
			break;
		default:
			usage();
		}
	}
	argc -= optind;
	argv += optind;

	if ((fd = open(path_pvbus, O_RDONLY)) == -1)
		err(1, "open: %s", path_pvbus);

	if (out != NULL) {
		if (strcmp("-", out) == 0)
			outfp = stdout;
		else if ((outfp = fopen(out, "w")) == NULL)
			err(1, "fopen: %s", out);
	}

	memset(&pvr, 0, sizeof(pvr));
	pvr.pvr_keylen = pvr.pvr_valuelen = KVBUFSZ;
	if ((pvr.pvr_key = calloc(1, pvr.pvr_keylen)) == NULL ||
	    (pvr.pvr_value = calloc(1, pvr.pvr_valuelen)) == NULL)
		err(1, "calloc");

	if (tflag) {
		if (ioctl(fd, PVBUSIOC_TYPE, &pvr, sizeof(pvr)) != 0)
			err(1, "ioctl");

		/* The returned type should be a simple single-line key */
		if (stravis(&str, pvr.pvr_key,
		    VIS_WHITE | VIS_DQ | VIS_CSTYLE) == -1)
			err(1, "stravis");
		fprintf(outfp, "%s: %s\n", path_pvbus, str);
		free(str);
		goto done;
	}

	if (argc < 1)
		usage();
	key = argv[0];

	if (kvsetstr(pvr.pvr_key, key, pvr.pvr_keylen) == -1)
		errx(1, "key too long");

	/* Validate command line options for reading or writing */
	if (argc == 2 && in == NULL) {
		cmd = PVBUSIOC_KVWRITE;
		value = argv[1];
		if (kvsetstr(pvr.pvr_value, value, pvr.pvr_valuelen) == -1)
			errx(1, "value too long");
	} else if (argc == 1 && in != NULL) {
		cmd = PVBUSIOC_KVWRITE;
		if (kvsetfile(pvr.pvr_value, in, pvr.pvr_valuelen) == -1)
			errx(1, "input file");
	} else if (argc == 1) {
		cmd = cmd == 0 ? PVBUSIOC_KVREAD : cmd;
	} else
		usage();

	/* Re-open read-writable */
	if (cmd == PVBUSIOC_KVWRITE) {
		close(fd);
		if ((fd = open(path_pvbus, O_RDWR)) == -1)
			err(1, "open: %s", path_pvbus);
	}

	if ((ret = ioctl(fd, cmd, &pvr, sizeof(pvr))) != 0)
		err(1, "ioctl");

	if (!qflag && strlen(pvr.pvr_value)) {
		/*
		 * The value can contain newlines and basically anything;
		 * only encode the unsafe characters that could perform
		 * unexpected functions on the terminal.
		 */
		if (stravis(&str, pvr.pvr_value, VIS_SAFE | VIS_CSTYLE) == -1)
			err(1, "stravis");
		fprintf(outfp, "%s\n", str);
		free(str);
	}

 done:
	if (outfp != stdout)
		fclose(outfp);
	free(pvr.pvr_value);
	free(pvr.pvr_key);
	close(fd);

	return (0);
}


Generated by: GCOVR (Version 3.3)

Line	Branch	Exec	Source
1			/* $OpenBSD: hostctl.c,v 1.3 2016/01/27 16:01:36 jmc Exp $ */
2
3			/*
4			* Copyright (c) 2016 Reyk Floeter <reyk@openbsd.org>
5			*
6			* Permission to use, copy, modify, and distribute this software for any
7			* purpose with or without fee is hereby granted, provided that the above
8			* copyright notice and this permission notice appear in all copies.
9			*
10			* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11			* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12			* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13			* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14			* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15			* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16			* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17			*/
18
19			#include <sys/ioctl.h>
20			#include <sys/types.h>
21
22			#include <dev/pv/pvvar.h>
23
24			#include <stdio.h>
25			#include <stdlib.h>
26			#include <string.h>
27			#include <unistd.h>
28			#include <fcntl.h>
29			#include <errno.h>
30			#include <err.h>
31			#include <vis.h>
32
33			#define KVBUFSZ 4096 /* arbitrary value */
34
35			char path_pvbus = "/dev/pvbus0"; / the first hv interface */
36			int qflag = 0; /* quiet */
37			int tflag = 0; /* show type */
38
39			__dead void usage(void);
40			int kvsetstr(char , const char , size_t);
41			int kvsetfile(char , const char , size_t);
42
43			__dead void
44			usage(void)
45			{
46			extern char *__progname;
47			fprintf(stderr, "usage: %s [-qt] [-f device] "
48			"[-i input] [-o output] key [value]\n", __progname);
49			exit(1);
50			}
51
52			int
53			kvsetstr(char dst, const char src, size_t dstlen)
54			{
55			size_t sz;
56
57			/* Sanitize the string before sending it to the kernel and host */
58			if ((sz = strnvis(dst, src, dstlen, VIS_SAFE \| VIS_CSTYLE)) >= dstlen)
59			return (-1);
60
61			/* Remove trailing newline */
62			if (dst[sz - 1] == '\n')
63			dst[sz - 1] = '\0';
64
65			return (0);
66			}
67
68			int
69			kvsetfile(char dst, const char input, size_t dstlen)
70			{
71			char *buf = NULL;
72			int ret = -1;
73			FILE *fp;
74
75			if (strcmp("-", input) == 0)
76			fp = stdin;
77			else if ((fp = fopen(input, "r")) == NULL)
78			return (-1);
79
80			if ((buf = calloc(1, dstlen)) == NULL)
81			goto done;
82			if (fread(buf, 1, dstlen - 1, fp) == 0)
83			goto done;
84			if (kvsetstr(dst, buf, dstlen) == -1)
85			goto done;
86
87			ret = 0;
88			done:
89			free(buf);
90			if (fp != stdin)
91			fclose(fp);
92			return (ret);
93			}
94
95			int
96			main(int argc, char *argv[])
97			{
98			const char key, value, in = NULL, out = NULL;
99			FILE *outfp = stdout;
100			int fd, ret;
101			struct pvbus_req pvr;
102			int ch;
103			unsigned long cmd = 0;
104			char *str;
105
106			while ((ch = getopt(argc, argv, "f:i:o:qt")) != -1) {
107			switch (ch) {
108			case 'f':
109			path_pvbus = optarg;
110			break;
111			case 'i':
112			in = optarg;
113			break;
114			case 'o':
115			out = optarg;
116			break;
117			case 'q':
118			qflag++;
119			break;
120			case 't':
121			tflag++;
122			break;
123			default:
124			usage();
125			}
126			}
127			argc -= optind;
128			argv += optind;
129
130			if ((fd = open(path_pvbus, O_RDONLY)) == -1)
131			err(1, "open: %s", path_pvbus);
132
133			if (out != NULL) {
134			if (strcmp("-", out) == 0)
135			outfp = stdout;
136			else if ((outfp = fopen(out, "w")) == NULL)
137			err(1, "fopen: %s", out);
138			}
139
140			memset(&pvr, 0, sizeof(pvr));
141			pvr.pvr_keylen = pvr.pvr_valuelen = KVBUFSZ;
142			if ((pvr.pvr_key = calloc(1, pvr.pvr_keylen)) == NULL \|\|
143			(pvr.pvr_value = calloc(1, pvr.pvr_valuelen)) == NULL)
144			err(1, "calloc");
145
146			if (tflag) {
147			if (ioctl(fd, PVBUSIOC_TYPE, &pvr, sizeof(pvr)) != 0)
148			err(1, "ioctl");
149
150			/* The returned type should be a simple single-line key */
151			if (stravis(&str, pvr.pvr_key,
152			VIS_WHITE \| VIS_DQ \| VIS_CSTYLE) == -1)
153			err(1, "stravis");
154			fprintf(outfp, "%s: %s\n", path_pvbus, str);
155			free(str);
156			goto done;
157			}
158
159			if (argc < 1)
160			usage();
161			key = argv[0];
162
163			if (kvsetstr(pvr.pvr_key, key, pvr.pvr_keylen) == -1)
164			errx(1, "key too long");
165
166			/* Validate command line options for reading or writing */
167			if (argc == 2 && in == NULL) {
168			cmd = PVBUSIOC_KVWRITE;
169			value = argv[1];
170			if (kvsetstr(pvr.pvr_value, value, pvr.pvr_valuelen) == -1)
171			errx(1, "value too long");
172			} else if (argc == 1 && in != NULL) {
173			cmd = PVBUSIOC_KVWRITE;
174			if (kvsetfile(pvr.pvr_value, in, pvr.pvr_valuelen) == -1)
175			errx(1, "input file");
176			} else if (argc == 1) {
177			cmd = cmd == 0 ? PVBUSIOC_KVREAD : cmd;
178			} else
179			usage();
180
181			/* Re-open read-writable */
182			if (cmd == PVBUSIOC_KVWRITE) {
183			close(fd);
184			if ((fd = open(path_pvbus, O_RDWR)) == -1)
185			err(1, "open: %s", path_pvbus);
186			}
187
188			if ((ret = ioctl(fd, cmd, &pvr, sizeof(pvr))) != 0)
189			err(1, "ioctl");
190
191			if (!qflag && strlen(pvr.pvr_value)) {
192			/*
193			* The value can contain newlines and basically anything;
194			* only encode the unsafe characters that could perform
195			* unexpected functions on the terminal.
196			*/
197			if (stravis(&str, pvr.pvr_value, VIS_SAFE \| VIS_CSTYLE) == -1)
198			err(1, "stravis");
199			fprintf(outfp, "%s\n", str);
200			free(str);
201			}
202
203			done:
204			if (outfp != stdout)
205			fclose(outfp);
206			free(pvr.pvr_value);
207			free(pvr.pvr_key);
208			close(fd);
209
210			return (0);
211			}