Merge from vendor branch LIBARCHIVE:
[dragonfly.git] / contrib / libarchive-1.3.1 / libarchive / archive_read_open_file.c
1 /*-
2  * Copyright (c) 2003-2006 Tim Kientzle
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  *    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.
14  *
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.
25  */
26
27 #include "archive_platform.h"
28 __FBSDID("$FreeBSD: src/lib/libarchive/archive_read_open_file.c,v 1.11 2006/09/05 05:59:46 kientzle Exp $");
29
30 #include <sys/stat.h>
31 #include <errno.h>
32 #include <stdio.h>
33 #include <stdlib.h>
34 #include <string.h>
35
36 #include "archive.h"
37
38 struct read_FILE_data {
39         FILE    *f;
40         size_t   block_size;
41         void    *buffer;
42 };
43
44 static int      file_close(struct archive *, void *);
45 static int      file_open(struct archive *, void *);
46 static ssize_t  file_read(struct archive *, void *, const void **buff);
47 static ssize_t  file_skip(struct archive *, void *, size_t request);
48
49 int
50 archive_read_open_FILE(struct archive *a, FILE *f)
51 {
52         struct read_FILE_data *mine;
53
54         mine = malloc(sizeof(*mine));
55         if (mine == NULL) {
56                 archive_set_error(a, ENOMEM, "No memory");
57                 return (ARCHIVE_FATAL);
58         }
59         mine->block_size = 128 * 1024;
60         mine->buffer = malloc(mine->block_size);
61         if (mine->buffer == NULL) {
62                 archive_set_error(a, ENOMEM, "No memory");
63                 free(mine);
64                 return (ARCHIVE_FATAL);
65         }
66         mine->f = f;
67         return (archive_read_open2(a, mine, file_open, file_read,
68                     file_skip, file_close));
69 }
70
71 static int
72 file_open(struct archive *a, void *client_data)
73 {
74         struct read_FILE_data *mine = client_data;
75         struct stat st;
76
77         /*
78          * If we can't fstat() the file, it may just be that
79          * it's not a file.  (FILE * objects can wrap many kinds
80          * of I/O streams.)
81          */
82         if (fstat(fileno(mine->f), &st) == 0 && S_ISREG(st.st_mode))
83                 archive_read_extract_set_skip_file(a, st.st_dev, st.st_ino);
84
85         return (ARCHIVE_OK);
86 }
87
88 static ssize_t
89 file_read(struct archive *a, void *client_data, const void **buff)
90 {
91         struct read_FILE_data *mine = client_data;
92         ssize_t bytes_read;
93
94         *buff = mine->buffer;
95         bytes_read = fread(mine->buffer, 1, mine->block_size, mine->f);
96         if (bytes_read < 0) {
97                 archive_set_error(a, errno, "Error reading file");
98         }
99         return (bytes_read);
100 }
101
102 static ssize_t
103 file_skip(struct archive *a, void *client_data, size_t request)
104 {
105         struct read_FILE_data *mine = client_data;
106         off_t old_offset, new_offset;
107
108         /* Reduce request to the next smallest multiple of block_size */
109         request = (request / mine->block_size) * mine->block_size;
110         /*
111          * Note: the 'fd' and 'filename' versions round the request
112          * down to a multiple of the block size to ensure proper
113          * operation on block-oriented media such as tapes.  But stdio
114          * doesn't work with such media (it doesn't ensure blocking),
115          * so we don't need to bother.
116          */
117         old_offset = ftello(mine->f);
118         fseeko(mine->f, request, SEEK_CUR);
119         new_offset = ftello(mine->f);
120         if (old_offset < 0 || new_offset < 0) {
121                 archive_set_error(a, errno, "Error skipping forward");
122                 return (ARCHIVE_FATAL);
123         }
124         return (new_offset - old_offset);
125 }
126
127 static int
128 file_close(struct archive *a, void *client_data)
129 {
130         struct read_FILE_data *mine = client_data;
131
132         (void)a; /* UNUSED */
133         if (mine->buffer != NULL)
134                 free(mine->buffer);
135         free(mine);
136         return (ARCHIVE_OK);
137 }