bsdcat.c revision 328827
1/*-
2 * Copyright (c) 2011-2014, Mike Kazantsev
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 "bsdcat_platform.h"
27__FBSDID("$FreeBSD: stable/11/contrib/libarchive/cat/bsdcat.c 328827 2018-02-03 02:17:04Z mm $");
28
29#include <stdio.h>
30#ifdef HAVE_STDLIB_H
31#include <stdlib.h>
32#endif
33#ifdef HAVE_UNISTD_H
34#include <unistd.h>
35#endif
36#ifdef HAVE_STRING_H
37#include <string.h>
38#endif
39
40#include "bsdcat.h"
41#include "err.h"
42
43#define	BYTES_PER_BLOCK	(20*512)
44
45static struct archive *a;
46static struct archive_entry *ae;
47static const char *bsdcat_current_path;
48static int exit_status = 0;
49
50
51void
52usage(FILE *stream, int eval)
53{
54	const char *p;
55	p = lafe_getprogname();
56	fprintf(stream,
57	    "Usage: %s [-h] [--help] [--version] [--] [filenames...]\n", p);
58	exit(eval);
59}
60
61static void
62version(void)
63{
64	printf("bsdcat %s - %s\n",
65	    BSDCAT_VERSION_STRING,
66	    archive_version_details());
67	exit(0);
68}
69
70void
71bsdcat_next(void)
72{
73	if (a != NULL) {
74		if (archive_read_close(a) != ARCHIVE_OK)
75			bsdcat_print_error();
76		archive_read_free(a);
77	}
78
79	a = archive_read_new();
80	archive_read_support_filter_all(a);
81	archive_read_support_format_empty(a);
82	archive_read_support_format_raw(a);
83}
84
85void
86bsdcat_print_error(void)
87{
88	lafe_warnc(0, "%s: %s",
89	    bsdcat_current_path, archive_error_string(a));
90	exit_status = 1;
91}
92
93void
94bsdcat_read_to_stdout(const char* filename)
95{
96	int r;
97
98	if (archive_read_open_filename(a, filename, BYTES_PER_BLOCK)
99	    != ARCHIVE_OK)
100		bsdcat_print_error();
101	else if (r = archive_read_next_header(a, &ae),
102		 r != ARCHIVE_OK && r != ARCHIVE_EOF)
103		bsdcat_print_error();
104	else if (r == ARCHIVE_EOF)
105		/* for empty payloads don't try and read data */
106		;
107	else if (archive_read_data_into_fd(a, 1) != ARCHIVE_OK)
108		bsdcat_print_error();
109	if (archive_read_close(a) != ARCHIVE_OK)
110		bsdcat_print_error();
111	archive_read_free(a);
112	a = NULL;
113}
114
115int
116main(int argc, char **argv)
117{
118	struct bsdcat *bsdcat, bsdcat_storage;
119	int c;
120
121	bsdcat = &bsdcat_storage;
122	memset(bsdcat, 0, sizeof(*bsdcat));
123
124	lafe_setprogname(*argv, "bsdcat");
125
126	bsdcat->argv = argv;
127	bsdcat->argc = argc;
128
129	while ((c = bsdcat_getopt(bsdcat)) != -1) {
130		switch (c) {
131		case 'h':
132			usage(stdout, 0);
133			break;
134		case OPTION_VERSION:
135			version();
136			break;
137		default:
138			usage(stderr, 1);
139		}
140	}
141
142	bsdcat_next();
143	if (*bsdcat->argv == NULL) {
144		bsdcat_current_path = "<stdin>";
145		bsdcat_read_to_stdout(NULL);
146	} else {
147		while (*bsdcat->argv) {
148			bsdcat_current_path = *bsdcat->argv++;
149			bsdcat_read_to_stdout(bsdcat_current_path);
150			bsdcat_next();
151		}
152		archive_read_free(a); /* Help valgrind & friends */
153	}
154
155	exit(exit_status);
156}
157