2 * Copyright (c) 2003-2007 Tim Kientzle
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer
10 * in this position and unchanged.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 #include "cpio_platform.h"
29 __FBSDID("$FreeBSD$");
52 * Short options for cpio. Please keep this sorted.
54 static const char *short_options = "0AaBC:cdE:F:f:H:hI:iJjLlmnO:opR:rtuVvW:yZz";
57 * Long options for cpio. Please keep this sorted.
59 static const struct option {
61 int required; /* 1 if this option requires an argument */
62 int equivalent; /* Equivalent short option. */
64 { "b64encode", 0, OPTION_B64ENCODE },
67 { "extract", 0, 'i' },
70 { "grzip", 0, OPTION_GRZIP },
72 { "insecure", 0, OPTION_INSECURE },
75 { "lrzip", 0, OPTION_LRZIP },
76 { "lzma", 0, OPTION_LZMA },
77 { "lzop", 0, OPTION_LZOP },
78 { "make-directories", 0, 'd' },
79 { "no-preserve-owner", 0, OPTION_NO_PRESERVE_OWNER },
81 { "numeric-uid-gid", 0, 'n' },
83 { "pass-through", 0, 'p' },
84 { "preserve-modification-time", 0, 'm' },
85 { "preserve-owner", 0, OPTION_PRESERVE_OWNER },
86 { "quiet", 0, OPTION_QUIET },
87 { "unconditional", 0, 'u' },
88 { "uuencode", 0, OPTION_UUENCODE },
89 { "verbose", 0, 'v' },
90 { "version", 0, OPTION_VERSION },
96 * I used to try to select platform-provided getopt() or
97 * getopt_long(), but that caused a lot of headaches. In particular,
98 * I couldn't consistently use long options in the test harness
99 * because not all platforms have getopt_long(). That in turn led to
100 * overuse of the -W hack in the test harness, which made it rough to
101 * run the test harness against GNU cpio. (I periodically run the
102 * test harness here against GNU cpio as a sanity-check. Yes,
103 * I've found a couple of bugs in GNU cpio that way.)
106 cpio_getopt(struct cpio *cpio)
108 enum { state_start = 0, state_next_word, state_short, state_long };
109 static int state = state_start;
110 static char *opt_word;
112 const struct option *popt, *match = NULL, *match2 = NULL;
113 const char *p, *long_prefix = "--";
118 cpio->argument = NULL;
120 /* First time through, initialize everything. */
121 if (state == state_start) {
122 /* Skip program name. */
125 state = state_next_word;
129 * We're ready to look at the next word in argv.
131 if (state == state_next_word) {
132 /* No more arguments, so no more options. */
133 if (cpio->argv[0] == NULL)
135 /* Doesn't start with '-', so no more options. */
136 if (cpio->argv[0][0] != '-')
138 /* "--" marks end of options; consume it and return. */
139 if (strcmp(cpio->argv[0], "--") == 0) {
144 /* Get next word for parsing. */
145 opt_word = *cpio->argv++;
147 if (opt_word[1] == '-') {
148 /* Set up long option parser. */
150 opt_word += 2; /* Skip leading '--' */
152 /* Set up short option parser. */
154 ++opt_word; /* Skip leading '-' */
159 * We're parsing a group of POSIX-style single-character options.
161 if (state == state_short) {
162 /* Peel next option off of a group of short options. */
165 /* End of this group; recurse to get next option. */
166 state = state_next_word;
167 return cpio_getopt(cpio);
170 /* Does this option take an argument? */
171 p = strchr(short_options, opt);
177 /* If it takes an argument, parse that. */
179 /* If arg is run-in, opt_word already points to it. */
180 if (opt_word[0] == '\0') {
181 /* Otherwise, pick up the next word. */
182 opt_word = *cpio->argv;
183 if (opt_word == NULL) {
185 "Option -%c requires an argument",
194 long_prefix = "-W "; /* For clearer errors. */
196 state = state_next_word;
197 cpio->argument = opt_word;
202 /* We're reading a long option, including -W long=arg convention. */
203 if (state == state_long) {
204 /* After this long option, we'll be starting a new word. */
205 state = state_next_word;
207 /* Option name ends at '=' if there is one. */
208 p = strchr(opt_word, '=');
210 optlength = (size_t)(p - opt_word);
211 cpio->argument = (char *)(uintptr_t)(p + 1);
213 optlength = strlen(opt_word);
216 /* Search the table for an unambiguous match. */
217 for (popt = cpio_longopts; popt->name != NULL; popt++) {
218 /* Short-circuit if first chars don't match. */
219 if (popt->name[0] != opt_word[0])
221 /* If option is a prefix of name in table, record it.*/
222 if (strncmp(opt_word, popt->name, optlength) == 0) {
223 match2 = match; /* Record up to two matches. */
225 /* If it's an exact match, we're done. */
226 if (strlen(popt->name) == optlength) {
227 match2 = NULL; /* Forget the others. */
233 /* Fail if there wasn't a unique match. */
236 "Option %s%s is not supported",
237 long_prefix, opt_word);
240 if (match2 != NULL) {
242 "Ambiguous option %s%s (matches --%s and --%s)",
243 long_prefix, opt_word, match->name, match2->name);
247 /* We've found a unique match; does it need an argument? */
248 if (match->required) {
249 /* Argument required: get next word if necessary. */
250 if (cpio->argument == NULL) {
251 cpio->argument = *cpio->argv;
252 if (cpio->argument == NULL) {
254 "Option %s%s requires an argument",
255 long_prefix, match->name);
262 /* Argument forbidden: fail if there is one. */
263 if (cpio->argument != NULL) {
265 "Option %s%s does not allow an argument",
266 long_prefix, match->name);
270 return (match->equivalent);
278 * Parse the argument to the -R or --owner flag.
280 * The format is one of the following:
281 * <username|uid> - Override user but not group
282 * <username>: - Override both, group is user's default group
283 * <uid>: - Override user but not group
284 * <username|uid>:<groupname|gid> - Override both
285 * :<groupname|gid> - Override group but not user
287 * Where uid/gid are decimal representations and groupname/username
288 * are names to be looked up in system database. Note that we try
289 * to look up an argument as a name first, then try numeric parsing.
291 * A period can be used instead of the colon.
293 * Sets uid/gid return as appropriate, -1 indicates uid/gid not specified.
294 * TODO: If the spec uses uname/gname, then return those to the caller
295 * as well. If the spec provides uid/gid, just return names as NULL.
297 * Returns NULL if no error, otherwise returns error string for display.
301 owner_parse(const char *spec, int *uid, int *gid)
303 static char errbuff[128];
304 const char *u, *ue, *g;
310 return ("Invalid empty user/group spec");
313 * Split spec into [user][:.][group]
314 * u -> first char of username, NULL if no username
315 * ue -> first char after username (colon, period, or \0)
316 * g -> first char of group name
318 if (*spec == ':' || *spec == '.') {
319 /* If spec starts with ':' or '.', then just group. */
323 /* Otherwise, [user] or [user][:] or [user][:][group] */
325 while (*ue != ':' && *ue != '.' && *ue != '\0')
328 if (*g != '\0') /* Skip : or . to find first char of group. */
333 /* Look up user: ue is first char after end of user. */
335 struct passwd *pwent;
337 user = (char *)malloc(ue - u + 1);
339 return ("Couldn't allocate memory");
340 memcpy(user, u, ue - u);
342 if ((pwent = getpwnam(user)) != NULL) {
343 *uid = pwent->pw_uid;
345 *gid = pwent->pw_gid;
349 *uid = (int)strtoul(user, &end, 10);
350 if (errno || *end != '\0') {
351 snprintf(errbuff, sizeof(errbuff),
352 "Couldn't lookup user ``%s''", user);
353 errbuff[sizeof(errbuff) - 1] = '\0';
363 if ((grp = getgrnam(g)) != NULL) {
368 *gid = (int)strtoul(g, &end, 10);
369 if (errno || *end != '\0') {
370 snprintf(errbuff, sizeof(errbuff),
371 "Couldn't lookup group ``%s''", g);
372 errbuff[sizeof(errbuff) - 1] = '\0';