1/*-
2 * Copyright (c) 2007 Kai Wang
3 * Copyright (c) 2007 Tim Kientzle
4 * Copyright (c) 2007 Joseph Koshy
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer
12 *    in this position and unchanged.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29/*-
30 * Copyright (c) 1990, 1993, 1994
31 *	The Regents of the University of California.  All rights reserved.
32 *
33 * This code is derived from software contributed to Berkeley by
34 * Hugh Smith at The University of Guelph.
35 *
36 * Redistribution and use in source and binary forms, with or without
37 * modification, are permitted provided that the following conditions
38 * are met:
39 * 1. Redistributions of source code must retain the above copyright
40 *    notice, this list of conditions and the following disclaimer.
41 * 2. Redistributions in binary form must reproduce the above copyright
42 *    notice, this list of conditions and the following disclaimer in the
43 *    documentation and/or other materials provided with the distribution.
44 * 3. Neither the name of the University nor the names of its contributors
45 *    may be used to endorse or promote products derived from this software
46 *    without specific prior written permission.
47 *
48 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
49 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
50 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
51 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
52 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
53 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
54 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
55 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
56 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
57 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
58 * SUCH DAMAGE.
59 */
60
61#include <sys/cdefs.h>
62__FBSDID("$FreeBSD$");
63
64#include <sys/queue.h>
65#include <sys/types.h>
66#include <archive.h>
67#include <errno.h>
68#include <getopt.h>
69#include <libgen.h>
70#include <stdio.h>
71#include <stdlib.h>
72#include <string.h>
73#include <sysexits.h>
74
75#include "ar.h"
76
77enum options
78{
79	OPTION_HELP
80};
81
82static struct option longopts[] =
83{
84	{"help", no_argument, NULL, OPTION_HELP},
85	{"version", no_argument, NULL, 'V'},
86	{NULL, 0, NULL, 0}
87};
88
89static void	bsdar_usage(void);
90static void	ranlib_usage(void);
91static void	set_mode(struct bsdar *bsdar, char opt);
92static void	only_mode(struct bsdar *bsdar, const char *opt,
93		    const char *valid_modes);
94static void	bsdar_version(void);
95static void	ranlib_version(void);
96
97int
98main(int argc, char **argv)
99{
100	struct bsdar	*bsdar, bsdar_storage;
101	char		*p;
102	size_t		 len;
103	int		 i, opt;
104
105	bsdar = &bsdar_storage;
106	memset(bsdar, 0, sizeof(*bsdar));
107
108	if ((bsdar->progname = getprogname()) == NULL)
109		bsdar->progname = "ar";
110
111	/* Act like ranlib if our name ends in "ranlib"; this
112	 * accommodates arm-freebsd7.1-ranlib, bsdranlib, etc. */
113	len = strlen(bsdar->progname);
114	if (len >= strlen("ranlib") &&
115	    strcmp(bsdar->progname + len - strlen("ranlib"), "ranlib") == 0) {
116		while ((opt = getopt_long(argc, argv, "tDV", longopts,
117		    NULL)) != -1) {
118			switch(opt) {
119			case 't':
120				/* Ignored. */
121				break;
122			case 'D':
123				bsdar->options |= AR_D;
124				break;
125			case 'V':
126				ranlib_version();
127				break;
128			case OPTION_HELP:
129				ranlib_usage();
130			default:
131				ranlib_usage();
132			}
133		}
134		argv += optind;
135		argc -= optind;
136
137		if (*argv == NULL)
138			ranlib_usage();
139
140		bsdar->options |= AR_S;
141		for (;(bsdar->filename = *argv++) != NULL;)
142			ar_mode_s(bsdar);
143
144		exit(EX_OK);
145	} else {
146		if (argc < 2)
147			bsdar_usage();
148
149		if (*argv[1] != '-') {
150			len = strlen(argv[1]) + 2;
151			if ((p = malloc(len)) == NULL)
152				bsdar_errc(bsdar, EX_SOFTWARE, errno,
153				    "malloc failed");
154			*p = '-';
155			(void)strlcpy(p + 1, argv[1], len - 1);
156			argv[1] = p;
157		}
158	}
159
160	while ((opt = getopt_long(argc, argv, "abCcdDfijlMmopqrSsTtuVvxz",
161	    longopts, NULL)) != -1) {
162		switch(opt) {
163		case 'a':
164			bsdar->options |= AR_A;
165			break;
166		case 'b':
167		case 'i':
168			bsdar->options |= AR_B;
169			break;
170		case 'C':
171			bsdar->options |= AR_CC;
172			break;
173		case 'c':
174			bsdar->options |= AR_C;
175			break;
176		case 'd':
177			set_mode(bsdar, opt);
178			break;
179		case 'D':
180			bsdar->options |= AR_D;
181			break;
182		case 'f':
183		case 'T':
184			bsdar->options |= AR_TR;
185			break;
186		case 'j':
187			/* ignored */
188			break;
189		case 'l':
190			/* ignored, for GNU ar comptibility */
191			break;
192		case 'M':
193			set_mode(bsdar, opt);
194			break;
195		case 'm':
196			set_mode(bsdar, opt);
197			break;
198		case 'o':
199			bsdar->options |= AR_O;
200			break;
201		case 'p':
202			set_mode(bsdar, opt);
203			break;
204		case 'q':
205			set_mode(bsdar, opt);
206			break;
207		case 'r':
208			set_mode(bsdar, opt);
209			break;
210		case 'S':
211			bsdar->options |= AR_SS;
212			break;
213		case 's':
214			bsdar->options |= AR_S;
215			break;
216		case 't':
217			set_mode(bsdar, opt);
218			break;
219		case 'u':
220			bsdar->options |= AR_U;
221			break;
222		case 'V':
223			bsdar_version();
224			break;
225		case 'v':
226			bsdar->options |= AR_V;
227			break;
228		case 'x':
229			set_mode(bsdar, opt);
230			break;
231		case 'z':
232			/* ignored */
233			break;
234		case OPTION_HELP:
235			bsdar_usage();
236		default:
237			bsdar_usage();
238		}
239	}
240
241	argv += optind;
242	argc -= optind;
243
244	if (*argv == NULL && bsdar->mode != 'M')
245		bsdar_usage();
246
247	if (bsdar->options & AR_A && bsdar->options & AR_B)
248		bsdar_errc(bsdar, EX_USAGE, 0,
249		    "only one of -a and -[bi] options allowed");
250
251	if (bsdar->options & AR_J && bsdar->options & AR_Z)
252		bsdar_errc(bsdar, EX_USAGE, 0,
253		    "only one of -j and -z options allowed");
254
255	if (bsdar->options & AR_S && bsdar->options & AR_SS)
256		bsdar_errc(bsdar, EX_USAGE, 0,
257		    "only one of -s and -S options allowed");
258
259	if (bsdar->options & (AR_A | AR_B)) {
260		if ((bsdar->posarg = *argv) == NULL)
261			bsdar_errc(bsdar, EX_USAGE, 0,
262			    "no position operand specified");
263		if ((bsdar->posarg = basename(bsdar->posarg)) == NULL)
264			bsdar_errc(bsdar, EX_SOFTWARE, errno,
265			    "basename failed");
266		argc--;
267		argv++;
268	}
269
270	if (bsdar->options & AR_A)
271		only_mode(bsdar, "-a", "mqr");
272	if (bsdar->options & AR_B)
273		only_mode(bsdar, "-b", "mqr");
274	if (bsdar->options & AR_C)
275		only_mode(bsdar, "-c", "qr");
276	if (bsdar->options & AR_CC)
277		only_mode(bsdar, "-C", "x");
278	if (bsdar->options & AR_D)
279		only_mode(bsdar, "-D", "qr");
280	if (bsdar->options & AR_O)
281		only_mode(bsdar, "-o", "x");
282	if (bsdar->options & AR_SS)
283		only_mode(bsdar, "-S", "mqr");
284	if (bsdar->options & AR_U)
285		only_mode(bsdar, "-u", "qrx");
286
287	if (bsdar->mode == 'M') {
288		ar_mode_script(bsdar);
289		exit(EX_OK);
290	}
291
292	if ((bsdar->filename = *argv) == NULL)
293		bsdar_usage();
294
295	bsdar->argc = --argc;
296	bsdar->argv = ++argv;
297
298	if ((!bsdar->mode || strchr("ptx", bsdar->mode)) &&
299	    bsdar->options & AR_S) {
300		ar_mode_s(bsdar);
301		if (!bsdar->mode)
302			exit(EX_OK);
303	}
304
305	switch(bsdar->mode) {
306	case 'd':
307		ar_mode_d(bsdar);
308		break;
309	case 'm':
310		ar_mode_m(bsdar);
311		break;
312	case 'p':
313		ar_mode_p(bsdar);
314		break;
315	case 'q':
316		ar_mode_q(bsdar);
317		break;
318	case 'r':
319		ar_mode_r(bsdar);
320		break;
321	case 't':
322		ar_mode_t(bsdar);
323		break;
324	case 'x':
325		ar_mode_x(bsdar);
326		break;
327	default:
328		bsdar_usage();
329		/* NOTREACHED */
330	}
331
332	for (i = 0; i < bsdar->argc; i++)
333		if (bsdar->argv[i] != NULL)
334			bsdar_warnc(bsdar, 0, "%s: not found in archive",
335			    bsdar->argv[i]);
336
337	exit(EX_OK);
338}
339
340static void
341set_mode(struct bsdar *bsdar, char opt)
342{
343
344	if (bsdar->mode != '\0' && bsdar->mode != opt)
345		bsdar_errc(bsdar, EX_USAGE, 0,
346		    "Can't specify both -%c and -%c", opt, bsdar->mode);
347	bsdar->mode = opt;
348}
349
350static void
351only_mode(struct bsdar *bsdar, const char *opt, const char *valid_modes)
352{
353
354	if (strchr(valid_modes, bsdar->mode) == NULL)
355		bsdar_errc(bsdar, EX_USAGE, 0,
356		    "Option %s is not permitted in mode -%c", opt, bsdar->mode);
357}
358
359static void
360bsdar_usage(void)
361{
362
363	(void)fprintf(stderr, "usage:  ar -d [-Tjsvz] archive file ...\n");
364	(void)fprintf(stderr, "\tar -m [-Tjsvz] archive file ...\n");
365	(void)fprintf(stderr, "\tar -m [-Tabijsvz] position archive file ...\n");
366	(void)fprintf(stderr, "\tar -p [-Tv] archive [file ...]\n");
367	(void)fprintf(stderr, "\tar -q [-TcDjsvz] archive file ...\n");
368	(void)fprintf(stderr, "\tar -r [-TcDjsuvz] archive file ...\n");
369	(void)fprintf(stderr, "\tar -r [-TabcDijsuvz] position archive file ...\n");
370	(void)fprintf(stderr, "\tar -s [-jz] archive\n");
371	(void)fprintf(stderr, "\tar -t [-Tv] archive [file ...]\n");
372	(void)fprintf(stderr, "\tar -x [-CTouv] archive [file ...]\n");
373	(void)fprintf(stderr, "\tar -V\n");
374	exit(EX_USAGE);
375}
376
377static void
378ranlib_usage(void)
379{
380
381	(void)fprintf(stderr, "usage:	ranlib [-t] archive ...\n");
382	(void)fprintf(stderr, "\tranlib -V\n");
383	exit(EX_USAGE);
384}
385
386static void
387bsdar_version(void)
388{
389	(void)printf("BSD ar %s - %s\n", BSDAR_VERSION, archive_version_string());
390	exit(EX_OK);
391}
392
393static void
394ranlib_version(void)
395{
396	(void)printf("ranlib %s - %s\n", BSDAR_VERSION, archive_version_string());
397	exit(EX_OK);
398}
399