archive_cmdline.c revision 324417 
1/*-
2 * Copyright (c) 2012 Michihiro NAKAJIMA
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
15 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
16 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
17 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
18 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
19 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
20 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
21 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
23 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#include "archive_platform.h"
27
28__FBSDID("$FreeBSD$");
29
30#ifdef HAVE_STRING_H
31#  include <string.h>
32#endif
33#ifdef HAVE_STDLIB_H
34#  include <stdlib.h>
35#endif
36
37#include "archive.h"
38#include "archive_cmdline_private.h"
39#include "archive_string.h"
40
41static int cmdline_set_path(struct archive_cmdline *, const char *);
42static int cmdline_add_arg(struct archive_cmdline *, const char *);
43
44static ssize_t
45extract_quotation(struct archive_string *as, const char *p)
46{
47	const char *s;
48
49	for (s = p + 1; *s;) {
50		if (*s == '\\') {
51			if (s[1] != '\0') {
52				archive_strappend_char(as, s[1]);
53				s += 2;
54			} else
55				s++;
56		} else if (*s == '"')
57			break;
58		else {
59			archive_strappend_char(as, s[0]);
60			s++;
61		}
62	}
63	if (*s != '"')
64		return (ARCHIVE_FAILED);/* Invalid sequence. */
65	return ((ssize_t)(s + 1 - p));
66}
67
68static ssize_t
69get_argument(struct archive_string *as, const char *p)
70{
71	const char *s = p;
72
73	archive_string_empty(as);
74
75	/* Skip beginning space characters. */
76	while (*s != '\0' && *s == ' ')
77		s++;
78	/* Copy non-space characters. */
79	while (*s != '\0' && *s != ' ') {
80		if (*s == '\\') {
81			if (s[1] != '\0') {
82				archive_strappend_char(as, s[1]);
83				s += 2;
84			} else {
85				s++;/* Ignore this character.*/
86				break;
87			}
88		} else if (*s == '"') {
89			ssize_t q = extract_quotation(as, s);
90			if (q < 0)
91				return (ARCHIVE_FAILED);/* Invalid sequence. */
92			s += q;
93		} else {
94			archive_strappend_char(as, s[0]);
95			s++;
96		}
97	}
98	return ((ssize_t)(s - p));
99}
100
101/*
102 * Set up command line arguments.
103 * Returns ARCHIVE_OK if everything okey.
104 * Returns ARCHIVE_FAILED if there is a lack of the `"' terminator or an
105 * empty command line.
106 * Returns ARCHIVE_FATAL if no memory.
107 */
108int
109__archive_cmdline_parse(struct archive_cmdline *data, const char *cmd)
110{
111	struct archive_string as;
112	const char *p;
113	ssize_t al;
114	int r;
115
116	archive_string_init(&as);
117
118	/* Get first argument as a command path. */
119	al = get_argument(&as, cmd);
120	if (al < 0) {
121		r = ARCHIVE_FAILED;/* Invalid sequence. */
122		goto exit_function;
123	}
124	if (archive_strlen(&as) == 0) {
125		r = ARCHIVE_FAILED;/* An empty command path. */
126		goto exit_function;
127	}
128	r = cmdline_set_path(data, as.s);
129	if (r != ARCHIVE_OK)
130		goto exit_function;
131	p = strrchr(as.s, '/');
132	if (p == NULL)
133		p = as.s;
134	else
135		p++;
136	r = cmdline_add_arg(data, p);
137	if (r != ARCHIVE_OK)
138		goto exit_function;
139	cmd += al;
140
141	for (;;) {
142		al = get_argument(&as, cmd);
143		if (al < 0) {
144			r = ARCHIVE_FAILED;/* Invalid sequence. */
145			goto exit_function;
146		}
147		if (al == 0)
148			break;
149		cmd += al;
150		if (archive_strlen(&as) == 0 && *cmd == '\0')
151			break;
152		r = cmdline_add_arg(data, as.s);
153		if (r != ARCHIVE_OK)
154			goto exit_function;
155	}
156	r = ARCHIVE_OK;
157exit_function:
158	archive_string_free(&as);
159	return (r);
160}
161
162/*
163 * Set the program path.
164 */
165static int
166cmdline_set_path(struct archive_cmdline *data, const char *path)
167{
168	char *newptr;
169
170	newptr = realloc(data->path, strlen(path) + 1);
171	if (newptr == NULL)
172		return (ARCHIVE_FATAL);
173	data->path = newptr;
174	strcpy(data->path, path);
175	return (ARCHIVE_OK);
176}
177
178/*
179 * Add a argument for the program.
180 */
181static int
182cmdline_add_arg(struct archive_cmdline *data, const char *arg)
183{
184	char **newargv;
185
186	if (data->path == NULL)
187		return (ARCHIVE_FAILED);
188
189	newargv = realloc(data->argv, (data->argc + 2) * sizeof(char *));
190	if (newargv == NULL)
191		return (ARCHIVE_FATAL);
192	data->argv = newargv;
193	data->argv[data->argc] = strdup(arg);
194	if (data->argv[data->argc] == NULL)
195		return (ARCHIVE_FATAL);
196	/* Set the terminator of argv. */
197	data->argv[++data->argc] = NULL;
198	return (ARCHIVE_OK);
199}
200
201struct archive_cmdline *
202__archive_cmdline_allocate(void)
203{
204	return (struct archive_cmdline *)
205		calloc(1, sizeof(struct archive_cmdline));
206}
207
208/*
209 * Release the resources.
210 */
211int
212__archive_cmdline_free(struct archive_cmdline *data)
213{
214
215	if (data) {
216		free(data->path);
217		if (data->argv != NULL) {
218			int i;
219			for (i = 0; data->argv[i] != NULL; i++)
220				free(data->argv[i]);
221			free(data->argv);
222		}
223		free(data);
224	}
225	return (ARCHIVE_OK);
226}
227
228