Commit | Line | Data |
---|---|---|
5ca28f79 L |
1 | |
2 | #include "zfstream.h" | |
3 | ||
4 | gzfilebuf::gzfilebuf() : | |
5 | file(NULL), | |
6 | mode(0), | |
7 | own_file_descriptor(0) | |
8 | { } | |
9 | ||
10 | gzfilebuf::~gzfilebuf() { | |
11 | ||
12 | sync(); | |
13 | if ( own_file_descriptor ) | |
14 | close(); | |
15 | ||
16 | } | |
17 | ||
18 | gzfilebuf *gzfilebuf::open( const char *name, | |
19 | int io_mode ) { | |
20 | ||
21 | if ( is_open() ) | |
22 | return NULL; | |
23 | ||
24 | char char_mode[10]; | |
25 | char *p = char_mode; | |
26 | ||
27 | if ( io_mode & ios::in ) { | |
28 | mode = ios::in; | |
29 | *p++ = 'r'; | |
30 | } else if ( io_mode & ios::app ) { | |
31 | mode = ios::app; | |
32 | *p++ = 'a'; | |
33 | } else { | |
34 | mode = ios::out; | |
35 | *p++ = 'w'; | |
36 | } | |
37 | ||
38 | if ( io_mode & ios::binary ) { | |
39 | mode |= ios::binary; | |
40 | *p++ = 'b'; | |
41 | } | |
42 | ||
43 | // Hard code the compression level | |
44 | if ( io_mode & (ios::out|ios::app )) { | |
45 | *p++ = '9'; | |
46 | } | |
47 | ||
48 | // Put the end-of-string indicator | |
49 | *p = '\0'; | |
50 | ||
51 | if ( (file = gzopen(name, char_mode)) == NULL ) | |
52 | return NULL; | |
53 | ||
54 | own_file_descriptor = 1; | |
55 | ||
56 | return this; | |
57 | ||
58 | } | |
59 | ||
60 | gzfilebuf *gzfilebuf::attach( int file_descriptor, | |
61 | int io_mode ) { | |
62 | ||
63 | if ( is_open() ) | |
64 | return NULL; | |
65 | ||
66 | char char_mode[10]; | |
67 | char *p = char_mode; | |
68 | ||
69 | if ( io_mode & ios::in ) { | |
70 | mode = ios::in; | |
71 | *p++ = 'r'; | |
72 | } else if ( io_mode & ios::app ) { | |
73 | mode = ios::app; | |
74 | *p++ = 'a'; | |
75 | } else { | |
76 | mode = ios::out; | |
77 | *p++ = 'w'; | |
78 | } | |
79 | ||
80 | if ( io_mode & ios::binary ) { | |
81 | mode |= ios::binary; | |
82 | *p++ = 'b'; | |
83 | } | |
84 | ||
85 | // Hard code the compression level | |
86 | if ( io_mode & (ios::out|ios::app )) { | |
87 | *p++ = '9'; | |
88 | } | |
89 | ||
90 | // Put the end-of-string indicator | |
91 | *p = '\0'; | |
92 | ||
93 | if ( (file = gzdopen(file_descriptor, char_mode)) == NULL ) | |
94 | return NULL; | |
95 | ||
96 | own_file_descriptor = 0; | |
97 | ||
98 | return this; | |
99 | ||
100 | } | |
101 | ||
102 | gzfilebuf *gzfilebuf::close() { | |
103 | ||
104 | if ( is_open() ) { | |
105 | ||
106 | sync(); | |
107 | gzclose( file ); | |
108 | file = NULL; | |
109 | ||
110 | } | |
111 | ||
112 | return this; | |
113 | ||
114 | } | |
115 | ||
116 | int gzfilebuf::setcompressionlevel( int comp_level ) { | |
117 | ||
118 | return gzsetparams(file, comp_level, -2); | |
119 | ||
120 | } | |
121 | ||
122 | int gzfilebuf::setcompressionstrategy( int comp_strategy ) { | |
123 | ||
124 | return gzsetparams(file, -2, comp_strategy); | |
125 | ||
126 | } | |
127 | ||
128 | ||
129 | streampos gzfilebuf::seekoff( streamoff off, ios::seek_dir dir, int which ) { | |
130 | ||
131 | return streampos(EOF); | |
132 | ||
133 | } | |
134 | ||
135 | int gzfilebuf::underflow() { | |
136 | ||
137 | // If the file hasn't been opened for reading, error. | |
138 | if ( !is_open() || !(mode & ios::in) ) | |
139 | return EOF; | |
140 | ||
141 | // if a buffer doesn't exists, allocate one. | |
142 | if ( !base() ) { | |
143 | ||
144 | if ( (allocate()) == EOF ) | |
145 | return EOF; | |
146 | setp(0,0); | |
147 | ||
148 | } else { | |
149 | ||
150 | if ( in_avail() ) | |
151 | return (unsigned char) *gptr(); | |
152 | ||
153 | if ( out_waiting() ) { | |
154 | if ( flushbuf() == EOF ) | |
155 | return EOF; | |
156 | } | |
157 | ||
158 | } | |
159 | ||
160 | // Attempt to fill the buffer. | |
161 | ||
162 | int result = fillbuf(); | |
163 | if ( result == EOF ) { | |
164 | // disable get area | |
165 | setg(0,0,0); | |
166 | return EOF; | |
167 | } | |
168 | ||
169 | return (unsigned char) *gptr(); | |
170 | ||
171 | } | |
172 | ||
173 | int gzfilebuf::overflow( int c ) { | |
174 | ||
175 | if ( !is_open() || !(mode & ios::out) ) | |
176 | return EOF; | |
177 | ||
178 | if ( !base() ) { | |
179 | if ( allocate() == EOF ) | |
180 | return EOF; | |
181 | setg(0,0,0); | |
182 | } else { | |
183 | if (in_avail()) { | |
184 | return EOF; | |
185 | } | |
186 | if (out_waiting()) { | |
187 | if (flushbuf() == EOF) | |
188 | return EOF; | |
189 | } | |
190 | } | |
191 | ||
192 | int bl = blen(); | |
193 | setp( base(), base() + bl); | |
194 | ||
195 | if ( c != EOF ) { | |
196 | ||
197 | *pptr() = c; | |
198 | pbump(1); | |
199 | ||
200 | } | |
201 | ||
202 | return 0; | |
203 | ||
204 | } | |
205 | ||
206 | int gzfilebuf::sync() { | |
207 | ||
208 | if ( !is_open() ) | |
209 | return EOF; | |
210 | ||
211 | if ( out_waiting() ) | |
212 | return flushbuf(); | |
213 | ||
214 | return 0; | |
215 | ||
216 | } | |
217 | ||
218 | int gzfilebuf::flushbuf() { | |
219 | ||
220 | int n; | |
221 | char *q; | |
222 | ||
223 | q = pbase(); | |
224 | n = pptr() - q; | |
225 | ||
226 | if ( gzwrite( file, q, n) < n ) | |
227 | return EOF; | |
228 | ||
229 | setp(0,0); | |
230 | ||
231 | return 0; | |
232 | ||
233 | } | |
234 | ||
235 | int gzfilebuf::fillbuf() { | |
236 | ||
237 | int required; | |
238 | char *p; | |
239 | ||
240 | p = base(); | |
241 | ||
242 | required = blen(); | |
243 | ||
244 | int t = gzread( file, p, required ); | |
245 | ||
246 | if ( t <= 0) return EOF; | |
247 | ||
248 | setg( base(), base(), base()+t); | |
249 | ||
250 | return t; | |
251 | ||
252 | } | |
253 | ||
254 | gzfilestream_common::gzfilestream_common() : | |
255 | ios( gzfilestream_common::rdbuf() ) | |
256 | { } | |
257 | ||
258 | gzfilestream_common::~gzfilestream_common() | |
259 | { } | |
260 | ||
261 | void gzfilestream_common::attach( int fd, int io_mode ) { | |
262 | ||
263 | if ( !buffer.attach( fd, io_mode) ) | |
264 | clear( ios::failbit | ios::badbit ); | |
265 | else | |
266 | clear(); | |
267 | ||
268 | } | |
269 | ||
270 | void gzfilestream_common::open( const char *name, int io_mode ) { | |
271 | ||
272 | if ( !buffer.open( name, io_mode ) ) | |
273 | clear( ios::failbit | ios::badbit ); | |
274 | else | |
275 | clear(); | |
276 | ||
277 | } | |
278 | ||
279 | void gzfilestream_common::close() { | |
280 | ||
281 | if ( !buffer.close() ) | |
282 | clear( ios::failbit | ios::badbit ); | |
283 | ||
284 | } | |
285 | ||
286 | gzfilebuf *gzfilestream_common::rdbuf() | |
287 | { | |
288 | return &buffer; | |
289 | } | |
290 | ||
291 | gzifstream::gzifstream() : | |
292 | ios( gzfilestream_common::rdbuf() ) | |
293 | { | |
294 | clear( ios::badbit ); | |
295 | } | |
296 | ||
297 | gzifstream::gzifstream( const char *name, int io_mode ) : | |
298 | ios( gzfilestream_common::rdbuf() ) | |
299 | { | |
300 | gzfilestream_common::open( name, io_mode ); | |
301 | } | |
302 | ||
303 | gzifstream::gzifstream( int fd, int io_mode ) : | |
304 | ios( gzfilestream_common::rdbuf() ) | |
305 | { | |
306 | gzfilestream_common::attach( fd, io_mode ); | |
307 | } | |
308 | ||
309 | gzifstream::~gzifstream() { } | |
310 | ||
311 | gzofstream::gzofstream() : | |
312 | ios( gzfilestream_common::rdbuf() ) | |
313 | { | |
314 | clear( ios::badbit ); | |
315 | } | |
316 | ||
317 | gzofstream::gzofstream( const char *name, int io_mode ) : | |
318 | ios( gzfilestream_common::rdbuf() ) | |
319 | { | |
320 | gzfilestream_common::open( name, io_mode ); | |
321 | } | |
322 | ||
323 | gzofstream::gzofstream( int fd, int io_mode ) : | |
324 | ios( gzfilestream_common::rdbuf() ) | |
325 | { | |
326 | gzfilestream_common::attach( fd, io_mode ); | |
327 | } | |
328 | ||
329 | gzofstream::~gzofstream() { } |