Head

GCC Code Coverage Report

Directory:	./		Exec	Total	Coverage
File:	usr.bin/tmux/cmd-string.c	Lines:	0	150	0.0 %
Date:	2017-11-13	Branches:	0	114	0.0 %


/* $OpenBSD: cmd-string.c,v 1.29 2017/06/14 07:42:41 nicm Exp $ */

/*
 * Copyright (c) 2008 Nicholas Marriott <nicholas.marriott@gmail.com>
 *
 * 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 MIND, 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/types.h>

#include <errno.h>
#include <pwd.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>

#include "tmux.h"

/*
 * Parse a command from a string.
 */

static int	 cmd_string_getc(const char *, size_t *);
static void	 cmd_string_ungetc(size_t *);
static void	 cmd_string_copy(char **, char *, size_t *);
static char	*cmd_string_string(const char *, size_t *, char, int);
static char	*cmd_string_variable(const char *, size_t *);
static char	*cmd_string_expand_tilde(const char *, size_t *);

static int
cmd_string_getc(const char *s, size_t *p)
{
	const u_char	*ucs = s;

	if (ucs[*p] == '\0')
		return (EOF);
	return (ucs[(*p)++]);
}

static void
cmd_string_ungetc(size_t *p)
{
	(*p)--;
}

int
cmd_string_split(const char *s, int *rargc, char ***rargv)
{
	size_t		p = 0;
	int		ch, argc = 0, append = 0;
	char	      **argv = NULL, *buf = NULL, *t;
	const char     *whitespace, *equals;
	size_t		len = 0;

	for (;;) {
		ch = cmd_string_getc(s, &p);
		switch (ch) {
		case '\'':
			if ((t = cmd_string_string(s, &p, '\'', 0)) == NULL)
				goto error;
			cmd_string_copy(&buf, t, &len);
			break;
		case '"':
			if ((t = cmd_string_string(s, &p, '"', 1)) == NULL)
				goto error;
			cmd_string_copy(&buf, t, &len);
			break;
		case '$':
			if ((t = cmd_string_variable(s, &p)) == NULL)
				goto error;
			cmd_string_copy(&buf, t, &len);
			break;
		case '#':
			/* Comment: discard rest of line. */
			while ((ch = cmd_string_getc(s, &p)) != EOF)
				;
			/* FALLTHROUGH */
		case EOF:
		case ' ':
		case '\t':
			if (buf != NULL) {
				buf = xrealloc(buf, len + 1);
				buf[len] = '\0';

				argv = xreallocarray(argv, argc + 1,
				    sizeof *argv);
				argv[argc++] = buf;

				buf = NULL;
				len = 0;
			}

			if (ch != EOF)
				break;

			while (argc != 0) {
				equals = strchr(argv[0], '=');
				whitespace = argv[0] + strcspn(argv[0], " \t");
				if (equals == NULL || equals > whitespace)
					break;
				environ_put(global_environ, argv[0]);
				argc--;
				memmove(argv, argv + 1, argc * (sizeof *argv));
			}
			goto done;
		case '~':
			if (buf != NULL) {
				append = 1;
				break;
			}
			t = cmd_string_expand_tilde(s, &p);
			if (t == NULL)
				goto error;
			cmd_string_copy(&buf, t, &len);
			break;
		default:
			append = 1;
			break;
		}
		if (append) {
			if (len >= SIZE_MAX - 2)
				goto error;
			buf = xrealloc(buf, len + 1);
			buf[len++] = ch;
		}
		append = 0;
	}

done:
	*rargc = argc;
	*rargv = argv;

	free(buf);
	return (0);

error:
	if (argv != NULL)
		cmd_free_argv(argc, argv);
	free(buf);
	return (-1);
}

struct cmd_list *
cmd_string_parse(const char *s, const char *file, u_int line, char **cause)
{
	struct cmd_list	 *cmdlist = NULL;
	int		  argc;
	char		**argv;

	*cause = NULL;
	if (cmd_string_split(s, &argc, &argv) != 0) {
		xasprintf(cause, "invalid or unknown command: %s", s);
		return (NULL);
	}
	if (argc != 0) {
		cmdlist = cmd_list_parse(argc, argv, file, line, cause);
		if (cmdlist == NULL) {
			cmd_free_argv(argc, argv);
			return (NULL);
		}
	}
	cmd_free_argv(argc, argv);
	return (cmdlist);
}

static void
cmd_string_copy(char **dst, char *src, size_t *len)
{
	size_t srclen;

	srclen = strlen(src);

	*dst = xrealloc(*dst, *len + srclen + 1);
	strlcpy(*dst + *len, src, srclen + 1);

	*len += srclen;
	free(src);
}

static char *
cmd_string_string(const char *s, size_t *p, char endch, int esc)
{
	int	ch;
	char   *buf, *t;
	size_t	len;

	buf = NULL;
	len = 0;

	while ((ch = cmd_string_getc(s, p)) != endch) {
		switch (ch) {
		case EOF:
			goto error;
		case '\\':
			if (!esc)
				break;
			switch (ch = cmd_string_getc(s, p)) {
			case EOF:
				goto error;
			case 'e':
				ch = '\033';
				break;
			case 'r':
				ch = '\r';
				break;
			case 'n':
				ch = '\n';
				break;
			case 't':
				ch = '\t';
				break;
			}
			break;
		case '$':
			if (!esc)
				break;
			if ((t = cmd_string_variable(s, p)) == NULL)
				goto error;
			cmd_string_copy(&buf, t, &len);
			continue;
		}

		if (len >= SIZE_MAX - 2)
			goto error;
		buf = xrealloc(buf, len + 1);
		buf[len++] = ch;
	}

	buf = xrealloc(buf, len + 1);
	buf[len] = '\0';
	return (buf);

error:
	free(buf);
	return (NULL);
}

static char *
cmd_string_variable(const char *s, size_t *p)
{
	int			ch, fch;
	char		       *buf, *t;
	size_t			len;
	struct environ_entry   *envent;

#define cmd_string_first(ch) ((ch) == '_' || \
	((ch) >= 'a' && (ch) <= 'z') || ((ch) >= 'A' && (ch) <= 'Z'))
#define cmd_string_other(ch) ((ch) == '_' || \
	((ch) >= 'a' && (ch) <= 'z') || ((ch) >= 'A' && (ch) <= 'Z') || \
	((ch) >= '0' && (ch) <= '9'))

	buf = NULL;
	len = 0;

	fch = EOF;
	switch (ch = cmd_string_getc(s, p)) {
	case EOF:
		goto error;
	case '{':
		fch = '{';

		ch = cmd_string_getc(s, p);
		if (!cmd_string_first(ch))
			goto error;
		/* FALLTHROUGH */
	default:
		if (!cmd_string_first(ch)) {
			xasprintf(&t, "$%c", ch);
			return (t);
		}

		buf = xrealloc(buf, len + 1);
		buf[len++] = ch;

		for (;;) {
			ch = cmd_string_getc(s, p);
			if (ch == EOF || !cmd_string_other(ch))
				break;
			else {
				if (len >= SIZE_MAX - 3)
					goto error;
				buf = xrealloc(buf, len + 1);
				buf[len++] = ch;
			}
		}
	}

	if (fch == '{' && ch != '}')
		goto error;
	if (ch != EOF && fch != '{')
		cmd_string_ungetc(p); /* ch */

	buf = xrealloc(buf, len + 1);
	buf[len] = '\0';

	envent = environ_find(global_environ, buf);
	free(buf);
	if (envent == NULL)
		return (xstrdup(""));
	return (xstrdup(envent->value));

error:
	free(buf);
	return (NULL);
}

static char *
cmd_string_expand_tilde(const char *s, size_t *p)
{
	struct passwd		*pw;
	struct environ_entry	*envent;
	char			*home, *path, *user, *cp;
	int			 last;

	home = NULL;

	last = cmd_string_getc(s, p);
	if (last == EOF || last == '/' || last == ' '|| last == '\t') {
		envent = environ_find(global_environ, "HOME");
		if (envent != NULL && *envent->value != '\0')
			home = envent->value;
		else if ((pw = getpwuid(getuid())) != NULL)
			home = pw->pw_dir;
	} else {
		cmd_string_ungetc(p);

		cp = user = xmalloc(strlen(s));
		for (;;) {
			last = cmd_string_getc(s, p);
			if (last == EOF ||
			    last == '/' ||
			    last == ' '||
			    last == '\t')
				break;
			*cp++ = last;
		}
		*cp = '\0';

		if ((pw = getpwnam(user)) != NULL)
			home = pw->pw_dir;
		free(user);
	}

	if (home == NULL)
		return (NULL);

	if (last != EOF)
		xasprintf(&path, "%s%c", home, last);
	else
		xasprintf(&path, "%s", home);
	return (path);
}


Generated by: GCOVR (Version 3.3)

Line	Branch	Exec	Source
1			/* $OpenBSD: cmd-string.c,v 1.29 2017/06/14 07:42:41 nicm Exp $ */
2
3			/*
4			* Copyright (c) 2008 Nicholas Marriott <nicholas.marriott@gmail.com>
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 MIND, USE, DATA OR PROFITS, WHETHER
15			* IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
16			* OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17			*/
18
19			#include <sys/types.h>
20
21			#include <errno.h>
22			#include <pwd.h>
23			#include <stdio.h>
24			#include <string.h>
25			#include <stdlib.h>
26			#include <unistd.h>
27
28			#include "tmux.h"
29
30			/*
31			* Parse a command from a string.
32			*/
33
34			static int cmd_string_getc(const char , size_t );
35			static void cmd_string_ungetc(size_t *);
36			static void cmd_string_copy(char *, char , size_t *);
37			static char cmd_string_string(const char , size_t *, char, int);
38			static char cmd_string_variable(const char , size_t *);
39			static char cmd_string_expand_tilde(const char , size_t *);
40
41			static int
42			cmd_string_getc(const char s, size_t p)
43			{
44			const u_char *ucs = s;
45
46			if (ucs[*p] == '\0')
47			return (EOF);
48			return (ucs[(*p)++]);
49			}
50
51			static void
52			cmd_string_ungetc(size_t *p)
53			{
54			(*p)--;
55			}
56
57			int
58			cmd_string_split(const char s, int rargc, char ***rargv)
59			{
60			size_t p = 0;
61			int ch, argc = 0, append = 0;
62			char *argv = NULL, buf = NULL, *t;
63			const char whitespace, equals;
64			size_t len = 0;
65
66			for (;;) {
67			ch = cmd_string_getc(s, &p);
68			switch (ch) {
69			case '\'':
70			if ((t = cmd_string_string(s, &p, '\'', 0)) == NULL)
71			goto error;
72			cmd_string_copy(&buf, t, &len);
73			break;
74			case '"':
75			if ((t = cmd_string_string(s, &p, '"', 1)) == NULL)
76			goto error;
77			cmd_string_copy(&buf, t, &len);
78			break;
79			case '$':
80			if ((t = cmd_string_variable(s, &p)) == NULL)
81			goto error;
82			cmd_string_copy(&buf, t, &len);
83			break;
84			case '#':
85			/* Comment: discard rest of line. */
86			while ((ch = cmd_string_getc(s, &p)) != EOF)
87			;
88			/* FALLTHROUGH */
89			case EOF:
90			case ' ':
91			case '\t':
92			if (buf != NULL) {
93			buf = xrealloc(buf, len + 1);
94			buf[len] = '\0';
95
96			argv = xreallocarray(argv, argc + 1,
97			sizeof *argv);
98			argv[argc++] = buf;
99
100			buf = NULL;
101			len = 0;
102			}
103
104			if (ch != EOF)
105			break;
106
107			while (argc != 0) {
108			equals = strchr(argv[0], '=');
109			whitespace = argv[0] + strcspn(argv[0], " \t");
110			if (equals == NULL \|\| equals > whitespace)
111			break;
112			environ_put(global_environ, argv[0]);
113			argc--;
114			memmove(argv, argv + 1, argc * (sizeof *argv));
115			}
116			goto done;
117			case '~':
118			if (buf != NULL) {
119			append = 1;
120			break;
121			}
122			t = cmd_string_expand_tilde(s, &p);
123			if (t == NULL)
124			goto error;
125			cmd_string_copy(&buf, t, &len);
126			break;
127			default:
128			append = 1;
129			break;
130			}
131			if (append) {
132			if (len >= SIZE_MAX - 2)
133			goto error;
134			buf = xrealloc(buf, len + 1);
135			buf[len++] = ch;
136			}
137			append = 0;
138			}
139
140			done:
141			*rargc = argc;
142			*rargv = argv;
143
144			free(buf);
145			return (0);
146
147			error:
148			if (argv != NULL)
149			cmd_free_argv(argc, argv);
150			free(buf);
151			return (-1);
152			}
153
154			struct cmd_list *
155			cmd_string_parse(const char s, const char file, u_int line, char **cause)
156			{
157			struct cmd_list *cmdlist = NULL;
158			int argc;
159			char **argv;
160
161			*cause = NULL;
162			if (cmd_string_split(s, &argc, &argv) != 0) {
163			xasprintf(cause, "invalid or unknown command: %s", s);
164			return (NULL);
165			}
166			if (argc != 0) {
167			cmdlist = cmd_list_parse(argc, argv, file, line, cause);
168			if (cmdlist == NULL) {
169			cmd_free_argv(argc, argv);
170			return (NULL);
171			}
172			}
173			cmd_free_argv(argc, argv);
174			return (cmdlist);
175			}
176
177			static void
178			cmd_string_copy(char *dst, char src, size_t *len)
179			{
180			size_t srclen;
181
182			srclen = strlen(src);
183
184			dst = xrealloc(dst, *len + srclen + 1);
185			strlcpy(dst + len, src, srclen + 1);
186
187			*len += srclen;
188			free(src);
189			}
190
191			static char *
192			cmd_string_string(const char s, size_t p, char endch, int esc)
193			{
194			int ch;
195			char buf, t;
196			size_t len;
197
198			buf = NULL;
199			len = 0;
200
201			while ((ch = cmd_string_getc(s, p)) != endch) {
202			switch (ch) {
203			case EOF:
204			goto error;
205			case '\\':
206			if (!esc)
207			break;
208			switch (ch = cmd_string_getc(s, p)) {
209			case EOF:
210			goto error;
211			case 'e':
212			ch = '\033';
213			break;
214			case 'r':
215			ch = '\r';
216			break;
217			case 'n':
218			ch = '\n';
219			break;
220			case 't':
221			ch = '\t';
222			break;
223			}
224			break;
225			case '$':
226			if (!esc)
227			break;
228			if ((t = cmd_string_variable(s, p)) == NULL)
229			goto error;
230			cmd_string_copy(&buf, t, &len);
231			continue;
232			}
233
234			if (len >= SIZE_MAX - 2)
235			goto error;
236			buf = xrealloc(buf, len + 1);
237			buf[len++] = ch;
238			}
239
240			buf = xrealloc(buf, len + 1);
241			buf[len] = '\0';
242			return (buf);
243
244			error:
245			free(buf);
246			return (NULL);
247			}
248
249			static char *
250			cmd_string_variable(const char s, size_t p)
251			{
252			int ch, fch;
253			char buf, t;
254			size_t len;
255			struct environ_entry *envent;
256
257			#define cmd_string_first(ch) ((ch) == '_' \|\| \
258			((ch) >= 'a' && (ch) <= 'z') \|\| ((ch) >= 'A' && (ch) <= 'Z'))
259			#define cmd_string_other(ch) ((ch) == '_' \|\| \
260			((ch) >= 'a' && (ch) <= 'z') \|\| ((ch) >= 'A' && (ch) <= 'Z') \|\| \
261			((ch) >= '0' && (ch) <= '9'))
262
263			buf = NULL;
264			len = 0;
265
266			fch = EOF;
267			switch (ch = cmd_string_getc(s, p)) {
268			case EOF:
269			goto error;
270			case '{':
271			fch = '{';
272
273			ch = cmd_string_getc(s, p);
274			if (!cmd_string_first(ch))
275			goto error;
276			/* FALLTHROUGH */
277			default:
278			if (!cmd_string_first(ch)) {
279			xasprintf(&t, "$%c", ch);
280			return (t);
281			}
282
283			buf = xrealloc(buf, len + 1);
284			buf[len++] = ch;
285
286			for (;;) {
287			ch = cmd_string_getc(s, p);
288			if (ch == EOF \|\| !cmd_string_other(ch))
289			break;
290			else {
291			if (len >= SIZE_MAX - 3)
292			goto error;
293			buf = xrealloc(buf, len + 1);
294			buf[len++] = ch;
295			}
296			}
297			}
298
299			if (fch == '{' && ch != '}')
300			goto error;
301			if (ch != EOF && fch != '{')
302			cmd_string_ungetc(p); /* ch */
303
304			buf = xrealloc(buf, len + 1);
305			buf[len] = '\0';
306
307			envent = environ_find(global_environ, buf);
308			free(buf);
309			if (envent == NULL)
310			return (xstrdup(""));
311			return (xstrdup(envent->value));
312
313			error:
314			free(buf);
315			return (NULL);
316			}
317
318			static char *
319			cmd_string_expand_tilde(const char s, size_t p)
320			{
321			struct passwd *pw;
322			struct environ_entry *envent;
323			char home, path, user, cp;
324			int last;
325
326			home = NULL;
327
328			last = cmd_string_getc(s, p);
329			if (last == EOF \|\| last == '/' \|\| last == ' '\|\| last == '\t') {
330			envent = environ_find(global_environ, "HOME");
331			if (envent != NULL && *envent->value != '\0')
332			home = envent->value;
333			else if ((pw = getpwuid(getuid())) != NULL)
334			home = pw->pw_dir;
335			} else {
336			cmd_string_ungetc(p);
337
338			cp = user = xmalloc(strlen(s));
339			for (;;) {
340			last = cmd_string_getc(s, p);
341			if (last == EOF \|\|
342			last == '/' \|\|
343			last == ' '\|\|
344			last == '\t')
345			break;
346			*cp++ = last;
347			}
348			*cp = '\0';
349
350			if ((pw = getpwnam(user)) != NULL)
351			home = pw->pw_dir;
352			free(user);
353			}
354
355			if (home == NULL)
356			return (NULL);
357
358			if (last != EOF)
359			xasprintf(&path, "%s%c", home, last);
360			else
361			xasprintf(&path, "%s", home);
362			return (path);
363			}