2 * Copyright (c) 2003-2004 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.
27 #include "archive_platform.h"
28 __FBSDID("$FreeBSD$");
38 #include "archive_private.h"
40 struct read_file_data {
44 mode_t st_mode; /* Mode bits for opened file. */
45 char filename[1]; /* Must be last! */
48 static int file_close(struct archive *, void *);
49 static int file_open(struct archive *, void *);
50 static ssize_t file_read(struct archive *, void *, const void **buff);
53 archive_read_open_file(struct archive *a, const char *filename,
56 struct read_file_data *mine;
58 if (filename == NULL || filename[0] == '\0') {
59 mine = malloc(sizeof(*mine));
61 archive_set_error(a, ENOMEM, "No memory");
62 return (ARCHIVE_FATAL);
64 mine->filename[0] = '\0';
66 mine = malloc(sizeof(*mine) + strlen(filename));
68 archive_set_error(a, ENOMEM, "No memory");
69 return (ARCHIVE_FATAL);
71 strcpy(mine->filename, filename);
73 mine->block_size = block_size;
76 return (archive_read_open(a, mine, file_open, file_read, file_close));
80 file_open(struct archive *a, void *client_data)
82 struct read_file_data *mine = client_data;
85 mine->buffer = malloc(mine->block_size);
86 if (mine->buffer == NULL) {
87 archive_set_error(a, ENOMEM, "No memory");
88 return (ARCHIVE_FATAL);
90 if (mine->filename[0] != '\0')
91 mine->fd = open(mine->filename, O_RDONLY);
93 mine->fd = 0; /* Fake "open" for stdin. */
95 archive_set_error(a, errno, "Failed to open '%s'",
97 return (ARCHIVE_FATAL);
99 if (fstat(mine->fd, &st) == 0) {
100 /* Set dev/ino of archive file so extract won't overwrite. */
101 a->skip_file_dev = st.st_dev;
102 a->skip_file_ino = st.st_ino;
103 /* Remember mode so close can decide whether to flush. */
104 mine->st_mode = st.st_mode;
106 if (mine->filename[0] == '\0')
107 archive_set_error(a, errno, "Can't stat stdin");
109 archive_set_error(a, errno, "Can't stat '%s'",
111 return (ARCHIVE_FATAL);
117 file_read(struct archive *a, void *client_data, const void **buff)
119 struct read_file_data *mine = client_data;
122 (void)a; /* UNUSED */
123 *buff = mine->buffer;
124 bytes_read = read(mine->fd, mine->buffer, mine->block_size);
125 if (bytes_read < 0) {
126 if (mine->filename[0] == '\0')
127 archive_set_error(a, errno, "Error reading stdin");
129 archive_set_error(a, errno, "Error reading '%s'",
136 file_close(struct archive *a, void *client_data)
138 struct read_file_data *mine = client_data;
140 (void)a; /* UNUSED */
143 * Sometimes, we should flush the input before closing.
144 * Regular files: faster to just close without flush.
145 * Devices: must not flush (user might need to
146 * read the "next" item on a non-rewind device).
147 * Pipes and sockets: must flush (otherwise, the
148 * program feeding the pipe or socket may complain).
149 * Here, I flush everything except for regular files and
152 if (!S_ISREG(mine->st_mode)
153 && !S_ISCHR(mine->st_mode)
154 && !S_ISBLK(mine->st_mode)) {
157 bytesRead = read(mine->fd, mine->buffer,
159 } while (bytesRead > 0);
161 /* If a named file was opened, then it needs to be closed. */
162 if (mine->filename[0] != '\0')
164 if (mine->buffer != NULL)