GCC Code Coverage Report

Directory: cvmfs/
File: cvmfs/compression/compression.cc
Date: 2026-05-19 11:45:12
Exec Total Coverage
Lines: 409 547 74.8%
Branches: 221 448 49.3%
Line Branch Exec Source
1 /**
2 * This file is part of the CernVM File System.
3 *
4 * This is a wrapper around zlib. It provides
5 * a set of functions to conveniently compress and decompress stuff.
6 * Almost all of the functions return true on success, otherwise false.
7 *
8 * TODO: think about code deduplication
9 */
10
11
12 #include "compression/compression.h"
13
14 #include <alloca.h>
15 #include <stdlib.h>
16 #include <sys/stat.h>
17
18 #include <algorithm>
19 #include <cassert>
20 #include <cstring>
21
22 #include "crypto/hash.h"
23 #include "util/exception.h"
24 #include "util/logging.h"
25 #include "util/platform.h"
26 #include "util/posix.h"
27 #include "util/smalloc.h"
28
29 using namespace std; // NOLINT
30
31
32 317 static bool CopyFile2File(FILE *fsrc, FILE *fdest) {
33 unsigned char buf[1024];
34
1/2
✓ Branch 1 taken 317 times.
✗ Branch 2 not taken.
 317 rewind(fsrc);
35
1/2
✓ Branch 1 taken 317 times.
✗ Branch 2 not taken.
 317 rewind(fdest);
36
37 size_t have;
38 do {
39
1/2
✓ Branch 1 taken 707 times.
✗ Branch 2 not taken.
 707 have = fread(buf, 1, 1024, fsrc);
40
2/4
✓ Branch 1 taken 707 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 707 times.
 707 if (fwrite(buf, 1, have, fdest) != have)
41 ✗ return false;
42
2/2
✓ Branch 0 taken 390 times.
✓ Branch 1 taken 317 times.
 707 } while (have == 1024);
43 317 return true;
44 }
45
46 78 bool CopyPath2File(const std::string &src, FILE *fdest) {
47 78 int retval = -1;
48 platform_stat64 info;
49
50
1/2
✓ Branch 2 taken 78 times.
✗ Branch 3 not taken.
 78 FILE *fsrc = fopen(src.c_str(), "r");
51
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 78 times.
 78 if (!fsrc)
52 ✗ goto file_copy_final;
53
54
2/4
✓ Branch 1 taken 78 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 78 times.
 78 if (!CopyFile2File(fsrc, fdest))
55 ✗ goto file_copy_final;
56 78 retval = platform_fstat(fileno(fsrc), &info);
57 78 retval |= fchmod(fileno(fdest), info.st_mode);
58
59 78 file_copy_final:
60
1/2
✓ Branch 0 taken 78 times.
✗ Branch 1 not taken.
 78 if (fsrc)
61
1/2
✓ Branch 1 taken 78 times.
✗ Branch 2 not taken.
 78 fclose(fsrc);
62 78 return retval == 0;
63 }
64
65
66 239 bool CopyPath2Path(const string &src, const string &dest) {
67 239 FILE *fsrc = NULL;
68 239 FILE *fdest = NULL;
69 239 int retval = -1;
70 platform_stat64 info;
71
72
1/2
✓ Branch 2 taken 239 times.
✗ Branch 3 not taken.
 239 fsrc = fopen(src.c_str(), "r");
73
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 239 times.
 239 if (!fsrc)
74 ✗ goto file_copy_final;
75
76
1/2
✓ Branch 2 taken 239 times.
✗ Branch 3 not taken.
 239 fdest = fopen(dest.c_str(), "w");
77
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 239 times.
 239 if (!fdest)
78 ✗ goto file_copy_final;
79
80
2/4
✓ Branch 1 taken 239 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 239 times.
 239 if (!CopyFile2File(fsrc, fdest))
81 ✗ goto file_copy_final;
82 239 retval = platform_fstat(fileno(fsrc), &info);
83 239 retval |= fchmod(fileno(fdest), info.st_mode);
84
85 239 file_copy_final:
86
1/2
✓ Branch 0 taken 239 times.
✗ Branch 1 not taken.
 239 if (fsrc)
87
1/2
✓ Branch 1 taken 239 times.
✗ Branch 2 not taken.
 239 fclose(fsrc);
88
1/2
✓ Branch 0 taken 239 times.
✗ Branch 1 not taken.
 239 if (fdest)
89
1/2
✓ Branch 1 taken 239 times.
✗ Branch 2 not taken.
 239 fclose(fdest);
90 239 return retval == 0;
91 }
92
93
94 ✗ bool CopyMem2File(const unsigned char *buffer, const unsigned buffer_size,
95 FILE *fdest) {
96 ✗ int written = fwrite(buffer, 1, buffer_size, fdest);
97 ✗ return (written >= 0) && (unsigned(written) == buffer_size);
98 }
99
100
101 2248 bool CopyMem2Path(const unsigned char *buffer, const unsigned buffer_size,
102 const string &path) {
103 2248 int fd = open(path.c_str(), O_WRONLY | O_CREAT | O_TRUNC, kDefaultFileMode);
104
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2248 times.
 2248 if (fd < 0)
105 ✗ return false;
106
107 2248 int written = write(fd, buffer, buffer_size);
108 2248 close(fd);
109
110
2/4
✓ Branch 0 taken 2248 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 2248 times.
✗ Branch 3 not taken.
 2248 return (written >= 0) && (unsigned(written) == buffer_size);
111 }
112
113
114 49 bool CopyPath2Mem(const string &path, unsigned char **buffer,
115 unsigned *buffer_size) {
116 49 const int fd = open(path.c_str(), O_RDONLY);
117
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 49 times.
 49 if (fd < 0)
118 ✗ return false;
119
120 49 *buffer_size = 512;
121 49 *buffer = reinterpret_cast<unsigned char *>(smalloc(*buffer_size));
122 49 unsigned total_bytes = 0;
123 while (true) {
124 98 int num_bytes = read(fd, *buffer + total_bytes, *buffer_size - total_bytes);
125
2/2
✓ Branch 0 taken 49 times.
✓ Branch 1 taken 49 times.
 98 if (num_bytes == 0)
126 49 break;
127
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 49 times.
 49 if (num_bytes < 0) {
128 ✗ close(fd);
129 ✗ free(*buffer);
130 ✗ *buffer_size = 0;
131 ✗ return false;
132 }
133 49 total_bytes += num_bytes;
134
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 49 times.
 49 if (total_bytes >= *buffer_size) {
135 ✗ *buffer_size *= 2;
136 ✗ *buffer = reinterpret_cast<unsigned char *>(
137 ✗ srealloc(*buffer, *buffer_size));
138 }
139 49 }
140
141 49 close(fd);
142 49 *buffer_size = total_bytes;
143 49 return true;
144 }
145
146 namespace zlib {
147
148 const unsigned kBufferSize = 32768;
149
150 /**
151 * Aborts if string doesn't match any of the algorithms.
152 */
153 ✗ Algorithms ParseCompressionAlgorithm(const std::string &algorithm_option) {
154 ✗ if ((algorithm_option == "default") || (algorithm_option == "zlib"))
155 ✗ return kZlibDefault;
156 ✗ if (algorithm_option == "none")
157 ✗ return kNoCompression;
158 ✗ PANIC(kLogStderr, "unknown compression algorithms: %s",
159 algorithm_option.c_str());
160 }
161
162
163 ✗ std::string AlgorithmName(const zlib::Algorithms alg) {
164 ✗ switch (alg) {
165 ✗ case kZlibDefault:
166 ✗ return "zlib";
167 break;
168 ✗ case kNoCompression:
169 ✗ return "none";
170 break;
171 // Purposely did not add a 'default' statement here: this will
172 // cause the compiler to generate a warning if a new algorithm
173 // is added but this function is not updated.
174 }
175 ✗ return "unknown";
176 }
177
178
179 6939 void CompressInit(z_stream *strm) {
180 6939 strm->zalloc = Z_NULL;
181 6939 strm->zfree = Z_NULL;
182 6939 strm->opaque = Z_NULL;
183 6939 strm->next_in = Z_NULL;
184 6939 strm->avail_in = 0;
185 6939 int retval = deflateInit(strm, Z_DEFAULT_COMPRESSION);
186
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6939 times.
 6939 assert(retval == 0);
187 6939 }
188
189
190 7200 void DecompressInit(z_stream *strm) {
191 7200 strm->zalloc = Z_NULL;
192 7200 strm->zfree = Z_NULL;
193 7200 strm->opaque = Z_NULL;
194 7200 strm->avail_in = 0;
195 7200 strm->next_in = Z_NULL;
196 7200 int retval = inflateInit(strm);
197
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7200 times.
 7200 assert(retval == 0);
198 7200 }
199
200
201 6939 void CompressFini(z_stream *strm) { (void)deflateEnd(strm); }
202
203
204 7010 void DecompressFini(z_stream *strm) { (void)inflateEnd(strm); }
205
206
207 196 StreamStates CompressZStream2Null(const void *buf,
208 const int64_t size,
209 const bool eof,
210 z_stream *strm,
211 shash::ContextPtr *hash_context) {
212 unsigned char out[kZChunk];
213 int z_ret;
214
215 196 strm->avail_in = size;
216 196 strm->next_in = static_cast<unsigned char *>(const_cast<void *>(buf));
217 // Run deflate() on input until output buffer not full, finish
218 // compression if all of source has been read in
219 do {
220 196 strm->avail_out = kZChunk;
221 196 strm->next_out = out;
222
3/4
✓ Branch 0 taken 147 times.
✓ Branch 1 taken 49 times.
✓ Branch 3 taken 196 times.
✗ Branch 4 not taken.
 196 z_ret = deflate(strm, eof ? Z_FINISH : Z_NO_FLUSH); // no bad return value
223
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 196 times.
 196 if (z_ret == Z_STREAM_ERROR)
224 ✗ return kStreamDataError;
225 196 size_t have = kZChunk - strm->avail_out;
226
1/2
✓ Branch 1 taken 196 times.
✗ Branch 2 not taken.
 196 shash::Update(out, have, *hash_context);
227
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 196 times.
 196 } while (strm->avail_out == 0);
228
229
2/2
✓ Branch 0 taken 147 times.
✓ Branch 1 taken 49 times.
 196 return (z_ret == Z_STREAM_END ? kStreamEnd : kStreamContinue);
230 }
231
232
233 3810 StreamStates DecompressZStream2Sink(const void *buf,
234 const int64_t size,
235 z_stream *strm,
236 cvmfs::Sink *sink) {
237 unsigned char out[kZChunk];
238 int z_ret;
239 3810 int64_t pos = 0;
240
241 do {
242 3810 strm->avail_in = (kZChunk > (size - pos)) ? size - pos : kZChunk;
243 3810 strm->next_in = ((unsigned char *)buf) + pos;
244
245 // Run inflate() on input until output buffer not full
246 do {
247 6269 strm->avail_out = kZChunk;
248 6269 strm->next_out = out;
249
1/2
✓ Branch 1 taken 6269 times.
✗ Branch 2 not taken.
 6269 z_ret = inflate(strm, Z_NO_FLUSH);
250
1/4
✗ Branch 0 not taken.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 6269 times.
 6269 switch (z_ret) {
251 ✗ case Z_NEED_DICT:
252 ✗ z_ret = Z_DATA_ERROR; // and fall through
253 ✗ case Z_STREAM_ERROR:
254 case Z_DATA_ERROR:
255 ✗ return kStreamDataError;
256 ✗ case Z_MEM_ERROR:
257 ✗ return kStreamIOError;
258 }
259 6269 size_t have = kZChunk - strm->avail_out;
260
1/2
✓ Branch 1 taken 6269 times.
✗ Branch 2 not taken.
 6269 int64_t written = sink->Write(out, have);
261
2/4
✓ Branch 0 taken 6269 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 6269 times.
 6269 if ((written < 0) || (static_cast<uint64_t>(written) != have))
262 ✗ return kStreamIOError;
263
2/2
✓ Branch 0 taken 2459 times.
✓ Branch 1 taken 3810 times.
 6269 } while (strm->avail_out == 0);
264
265 3810 pos += kZChunk;
266
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3810 times.
 3810 } while (pos < size);
267
268
2/2
✓ Branch 0 taken 3568 times.
✓ Branch 1 taken 242 times.
 3810 return (z_ret == Z_STREAM_END ? kStreamEnd : kStreamContinue);
269 }
270
271
272 208 StreamStates DecompressZStream2File(const void *buf,
273 const int64_t size,
274 z_stream *strm,
275 FILE *f) {
276 unsigned char out[kZChunk];
277 int z_ret;
278 208 int64_t pos = 0;
279
280 do {
281 208 strm->avail_in = (kZChunk > (size - pos)) ? size - pos : kZChunk;
282 208 strm->next_in = ((unsigned char *)buf) + pos;
283
284 // Run inflate() on input until output buffer not full
285 do {
286 286 strm->avail_out = kZChunk;
287 286 strm->next_out = out;
288
1/2
✓ Branch 1 taken 286 times.
✗ Branch 2 not taken.
 286 z_ret = inflate(strm, Z_NO_FLUSH);
289
1/4
✗ Branch 0 not taken.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 286 times.
 286 switch (z_ret) {
290 ✗ case Z_NEED_DICT:
291 ✗ z_ret = Z_DATA_ERROR; // and fall through
292 ✗ case Z_STREAM_ERROR:
293 case Z_DATA_ERROR:
294 ✗ return kStreamDataError;
295 ✗ case Z_MEM_ERROR:
296 ✗ return kStreamIOError;
297 }
298 286 size_t have = kZChunk - strm->avail_out;
299
4/9
✓ Branch 1 taken 286 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 286 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 286 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 286 times.
 286 if (fwrite(out, 1, have, f) != have || ferror(f)) {
300 ✗ LogCvmfs(kLogCompress, kLogDebug,
301 "Inflate to file failed with %s "
302 "(errno=%d)",
303 ✗ strerror(errno), errno);
304 ✗ return kStreamIOError;
305 }
306
2/2
✓ Branch 0 taken 78 times.
✓ Branch 1 taken 208 times.
 286 } while (strm->avail_out == 0);
307
308 208 pos += kZChunk;
309
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 208 times.
 208 } while (pos < size);
310
311
1/2
✓ Branch 0 taken 208 times.
✗ Branch 1 not taken.
 208 return (z_ret == Z_STREAM_END ? kStreamEnd : kStreamContinue);
312 }
313
314
315 1206 bool CompressPath2Path(const string &src, const string &dest) {
316 1206 FILE *fsrc = fopen(src.c_str(), "r");
317
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1206 times.
 1206 if (!fsrc) {
318 ✗ LogCvmfs(kLogCompress, kLogDebug, "open %s as compression source failed",
319 src.c_str());
320 ✗ return false;
321 }
322
323 1206 FILE *fdest = fopen(dest.c_str(), "w");
324
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1206 times.
 1206 if (!fdest) {
325 ✗ LogCvmfs(kLogCompress, kLogDebug,
326 "open %s as compression destination "
327 "failed with errno=%d",
328 ✗ dest.c_str(), errno);
329 ✗ fclose(fsrc);
330 ✗ return false;
331 }
332
333 1206 LogCvmfs(kLogCompress, kLogDebug, "opened %s and %s for compression",
334 src.c_str(), dest.c_str());
335 1206 const bool result = CompressFile2File(fsrc, fdest);
336
337 1206 fclose(fsrc);
338 1206 fclose(fdest);
339 1206 return result;
340 }
341
342
343 2963 bool CompressPath2Path(const string &src, const string &dest,
344 shash::Any *compressed_hash) {
345
1/2
✓ Branch 2 taken 2963 times.
✗ Branch 3 not taken.
 2963 FILE *fsrc = fopen(src.c_str(), "r");
346
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2963 times.
 2963 if (!fsrc) {
347 ✗ LogCvmfs(kLogCompress, kLogDebug, "open %s as compression source failed",
348 src.c_str());
349 ✗ return false;
350 }
351
352
1/2
✓ Branch 2 taken 2963 times.
✗ Branch 3 not taken.
 2963 FILE *fdest = fopen(dest.c_str(), "w");
353
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2963 times.
 2963 if (!fdest) {
354 ✗ LogCvmfs(kLogCompress, kLogDebug,
355 "open %s as compression destination "
356 "failed with errno=%d",
357 ✗ dest.c_str(), errno);
358 ✗ fclose(fsrc);
359 ✗ return false;
360 }
361
362
1/2
✓ Branch 3 taken 2963 times.
✗ Branch 4 not taken.
 2963 LogCvmfs(kLogCompress, kLogDebug, "opened %s and %s for compression",
363 src.c_str(), dest.c_str());
364 2963 bool result = false;
365
2/4
✓ Branch 1 taken 2963 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 2963 times.
 2963 if (!CompressFile2File(fsrc, fdest, compressed_hash))
366 ✗ goto compress_path2path_final;
367 platform_stat64 info;
368
1/2
✗ Branch 2 not taken.
✓ Branch 3 taken 2963 times.
 2963 if (platform_fstat(fileno(fsrc), &info) != 0)
369 ✗ goto compress_path2path_final;
370 // TODO(jakob): open in the right mode from the beginning
371
1/2
✗ Branch 2 not taken.
✓ Branch 3 taken 2963 times.
 2963 if (fchmod(fileno(fdest), info.st_mode) != 0)
372 ✗ goto compress_path2path_final;
373
374 2963 result = true;
375
376 2963 compress_path2path_final:
377
1/2
✓ Branch 1 taken 2963 times.
✗ Branch 2 not taken.
 2963 fclose(fsrc);
378
1/2
✓ Branch 1 taken 2963 times.
✗ Branch 2 not taken.
 2963 fclose(fdest);
379 2963 return result;
380 }
381
382
383 ✗ bool DecompressPath2Path(const string &src, const string &dest) {
384 ✗ FILE *fsrc = NULL;
385 ✗ FILE *fdest = NULL;
386 ✗ int result = false;
387
388 ✗ fsrc = fopen(src.c_str(), "r");
389 ✗ if (!fsrc)
390 ✗ goto decompress_path2path_final;
391
392 ✗ fdest = fopen(dest.c_str(), "w");
393 ✗ if (!fdest)
394 ✗ goto decompress_path2path_final;
395
396 ✗ result = DecompressFile2File(fsrc, fdest);
397
398 ✗ decompress_path2path_final:
399 ✗ if (fsrc)
400 ✗ fclose(fsrc);
401 ✗ if (fdest)
402 ✗ fclose(fdest);
403 ✗ return result;
404 }
405
406
407 1206 bool CompressFile2Null(FILE *fsrc, shash::Any *compressed_hash) {
408 1206 int z_ret = 0;
409 1206 int flush = 0;
410 1206 bool result = -1;
411 unsigned have;
412 z_stream strm;
413 unsigned char in[kZChunk];
414 unsigned char out[kZChunk];
415
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 shash::ContextPtr hash_context(compressed_hash->algorithm);
416
417
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 CompressInit(&strm);
418 1206 hash_context.buffer = alloca(hash_context.size);
419
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 shash::Init(hash_context);
420
421 // Compress until end of file
422 do {
423
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 strm.avail_in = fread(in, 1, kZChunk, fsrc);
424
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1206 times.
 1206 if (ferror(fsrc))
425 ✗ goto compress_file2null_final;
426
427
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 flush = feof(fsrc) ? Z_FINISH : Z_NO_FLUSH;
428 1206 strm.next_in = in;
429
430 // Run deflate() on input until output buffer not full, finish
431 // compression if all of source has been read in
432 do {
433 1206 strm.avail_out = kZChunk;
434 1206 strm.next_out = out;
435
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 z_ret = deflate(&strm, flush); // no bad return value
436
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1206 times.
 1206 if (z_ret == Z_STREAM_ERROR)
437 ✗ goto compress_file2null_final; // state not clobbered
438 1206 have = kZChunk - strm.avail_out;
439
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 shash::Update(out, have, hash_context);
440
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1206 times.
 1206 } while (strm.avail_out == 0);
441
442 // Done when last data in file processed
443
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1206 times.
 1206 } while (flush != Z_FINISH);
444
445 // stream will be complete
446
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1206 times.
 1206 if (z_ret != Z_STREAM_END)
447 ✗ goto compress_file2null_final;
448
449
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 shash::Final(hash_context, compressed_hash);
450 1206 result = true;
451
452 // Clean up and return
453 1206 compress_file2null_final:
454
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 CompressFini(&strm);
455
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 LogCvmfs(kLogCompress, kLogDebug, "file compression finished with result %d",
456 result);
457 1206 return result;
458 }
459
460
461 62 bool CompressFd2Null(int fd_src, shash::Any *compressed_hash,
462 uint64_t *processed_bytes) {
463 62 int z_ret = 0;
464 62 int flush = 0;
465 62 bool result = false;
466 unsigned have;
467 z_stream strm;
468 unsigned char in[kZChunk];
469 unsigned char out[kZChunk];
470 62 off_t cksum_bytes = 0;
471
1/2
✓ Branch 1 taken 62 times.
✗ Branch 2 not taken.
 62 shash::ContextPtr hash_context(compressed_hash->algorithm);
472
473
1/2
✓ Branch 1 taken 62 times.
✗ Branch 2 not taken.
 62 CompressInit(&strm);
474 62 hash_context.buffer = alloca(hash_context.size);
475
1/2
✓ Branch 1 taken 62 times.
✗ Branch 2 not taken.
 62 shash::Init(hash_context);
476
477 // Compress until end of file
478 do {
479
1/2
✓ Branch 1 taken 62 times.
✗ Branch 2 not taken.
 62 ssize_t bytes_read = read(fd_src, in, kZChunk);
480
2/2
✓ Branch 0 taken 31 times.
✓ Branch 1 taken 31 times.
 62 if (bytes_read < 0) {
481
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
 31 if (errno == EINTR) {
482 ✗ continue;
483 }
484 31 goto compress_fd2null_final;
485 }
486 31 cksum_bytes += bytes_read;
487 31 strm.avail_in = bytes_read;
488
489
1/2
✓ Branch 0 taken 31 times.
✗ Branch 1 not taken.
 31 flush = (static_cast<size_t>(bytes_read) < kZChunk) ? Z_FINISH : Z_NO_FLUSH;
490 31 strm.next_in = in;
491
492 // Run deflate() on input until output buffer not full, finish
493 // compression if all of source has been read in
494 do {
495 31 strm.avail_out = kZChunk;
496 31 strm.next_out = out;
497
1/2
✓ Branch 1 taken 31 times.
✗ Branch 2 not taken.
 31 z_ret = deflate(&strm, flush); // no bad return value
498
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
 31 if (z_ret == Z_STREAM_ERROR)
499 ✗ goto compress_fd2null_final; // state not clobbered
500 31 have = kZChunk - strm.avail_out;
501
1/2
✓ Branch 1 taken 31 times.
✗ Branch 2 not taken.
 31 shash::Update(out, have, hash_context);
502
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
 31 } while (strm.avail_out == 0);
503
504 // Done when last data in file processed
505
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
 31 } while (flush != Z_FINISH);
506
507 // stream will be complete
508
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
 31 if (z_ret != Z_STREAM_END)
509 ✗ goto compress_fd2null_final;
510
511
1/2
✓ Branch 1 taken 31 times.
✗ Branch 2 not taken.
 31 shash::Final(hash_context, compressed_hash);
512
1/2
✓ Branch 0 taken 31 times.
✗ Branch 1 not taken.
 31 if (processed_bytes) {
513 31 *processed_bytes = cksum_bytes;
514 }
515 31 result = true;
516
517 // Clean up and return
518 62 compress_fd2null_final:
519
1/2
✓ Branch 1 taken 62 times.
✗ Branch 2 not taken.
 62 CompressFini(&strm);
520
1/2
✓ Branch 1 taken 62 times.
✗ Branch 2 not taken.
 62 LogCvmfs(kLogCompress, kLogDebug, "file compression finished with result %d",
521 result);
522 62 return result;
523 }
524
525
526 1206 bool CompressPath2Null(const string &src, shash::Any *compressed_hash) {
527 1206 FILE *fsrc = fopen(src.c_str(), "r");
528
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1206 times.
 1206 if (fsrc == NULL)
529 ✗ return false;
530 1206 bool retval = CompressFile2Null(fsrc, compressed_hash);
531 1206 fclose(fsrc);
532 1206 return retval;
533 }
534
535
536 1206 bool CompressFile2File(FILE *fsrc, FILE *fdest) {
537 1206 int z_ret = 0;
538 1206 int flush = 0;
539 1206 bool result = false;
540 unsigned have;
541 z_stream strm;
542 unsigned char in[kZChunk];
543 unsigned char out[kZChunk];
544
545
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 CompressInit(&strm);
546
547 // Compress until end of file
548 do {
549
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 strm.avail_in = fread(in, 1, kZChunk, fsrc);
550
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1206 times.
 1206 if (ferror(fsrc))
551 ✗ goto compress_file2file_final;
552
553
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 flush = feof(fsrc) ? Z_FINISH : Z_NO_FLUSH;
554 1206 strm.next_in = in;
555
556 // Run deflate() on input until output buffer not full, finish
557 // compression if all of source has been read in
558 do {
559 1206 strm.avail_out = kZChunk;
560 1206 strm.next_out = out;
561
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 z_ret = deflate(&strm, flush); // no bad return value
562
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1206 times.
 1206 if (z_ret == Z_STREAM_ERROR)
563 ✗ goto compress_file2file_final; // state not clobbered
564 1206 have = kZChunk - strm.avail_out;
565
4/9
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1206 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 1206 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 1206 times.
 1206 if (fwrite(out, 1, have, fdest) != have || ferror(fdest))
566 ✗ goto compress_file2file_final;
567
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1206 times.
 1206 } while (strm.avail_out == 0);
568
569 // Done when last data in file processed
570
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1206 times.
 1206 } while (flush != Z_FINISH);
571
572 // stream will be complete
573
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1206 times.
 1206 if (z_ret != Z_STREAM_END)
574 ✗ goto compress_file2file_final;
575
576 1206 result = true;
577
578 // Clean up and return
579 1206 compress_file2file_final:
580
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 CompressFini(&strm);
581
1/2
✓ Branch 1 taken 1206 times.
✗ Branch 2 not taken.
 1206 LogCvmfs(kLogCompress, kLogDebug, "file compression finished with result %d",
582 result);
583 1206 return result;
584 }
585
586 ✗ bool CompressPath2File(const string &src, FILE *fdest,
587 shash::Any *compressed_hash) {
588 ✗ FILE *fsrc = fopen(src.c_str(), "r");
589 ✗ if (!fsrc)
590 ✗ return false;
591
592 ✗ bool retval = CompressFile2File(fsrc, fdest, compressed_hash);
593 ✗ fclose(fsrc);
594 ✗ return retval;
595 }
596
597
598 2963 bool CompressFile2File(FILE *fsrc, FILE *fdest, shash::Any *compressed_hash) {
599 2963 int z_ret = 0;
600 2963 int flush = 0;
601 2963 bool result = false;
602 unsigned have;
603 z_stream strm;
604 unsigned char in[kZChunk];
605 unsigned char out[kZChunk];
606
1/2
✓ Branch 1 taken 2963 times.
✗ Branch 2 not taken.
 2963 shash::ContextPtr hash_context(compressed_hash->algorithm);
607
608
1/2
✓ Branch 1 taken 2963 times.
✗ Branch 2 not taken.
 2963 CompressInit(&strm);
609 2963 hash_context.buffer = alloca(hash_context.size);
610
1/2
✓ Branch 1 taken 2963 times.
✗ Branch 2 not taken.
 2963 shash::Init(hash_context);
611
612 // Compress until end of file
613 do {
614
1/2
✓ Branch 1 taken 4156 times.
✗ Branch 2 not taken.
 4156 strm.avail_in = fread(in, 1, kZChunk, fsrc);
615
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 4156 times.
 4156 if (ferror(fsrc))
616 ✗ goto compress_file2file_hashed_final;
617
618
2/2
✓ Branch 1 taken 2963 times.
✓ Branch 2 taken 1193 times.
 4156 flush = feof(fsrc) ? Z_FINISH : Z_NO_FLUSH;
619 4156 strm.next_in = in;
620
621 // Run deflate() on input until output buffer not full, finish
622 // compression if all of source has been read in
623 do {
624 4156 strm.avail_out = kZChunk;
625 4156 strm.next_out = out;
626
1/2
✓ Branch 1 taken 4156 times.
✗ Branch 2 not taken.
 4156 z_ret = deflate(&strm, flush); // no bad return value
627
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4156 times.
 4156 if (z_ret == Z_STREAM_ERROR)
628 ✗ goto compress_file2file_hashed_final; // state not clobbered
629 4156 have = kZChunk - strm.avail_out;
630
4/9
✓ Branch 1 taken 4156 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 4156 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 4156 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 4156 times.
 4156 if (fwrite(out, 1, have, fdest) != have || ferror(fdest))
631 ✗ goto compress_file2file_hashed_final;
632
1/2
✓ Branch 1 taken 4156 times.
✗ Branch 2 not taken.
 4156 shash::Update(out, have, hash_context);
633
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4156 times.
 4156 } while (strm.avail_out == 0);
634
635 // Done when last data in file processed
636
2/2
✓ Branch 0 taken 1193 times.
✓ Branch 1 taken 2963 times.
 4156 } while (flush != Z_FINISH);
637
638 // Stream will be complete
639
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2963 times.
 2963 if (z_ret != Z_STREAM_END)
640 ✗ goto compress_file2file_hashed_final;
641
642
1/2
✓ Branch 1 taken 2963 times.
✗ Branch 2 not taken.
 2963 shash::Final(hash_context, compressed_hash);
643 2963 result = true;
644
645 // Clean up and return
646 2963 compress_file2file_hashed_final:
647
1/2
✓ Branch 1 taken 2963 times.
✗ Branch 2 not taken.
 2963 CompressFini(&strm);
648
1/2
✓ Branch 1 taken 2963 times.
✗ Branch 2 not taken.
 2963 LogCvmfs(kLogCompress, kLogDebug, "file compression finished with result %d",
649 result);
650 2963 return result;
651 }
652
653
654 260 bool DecompressFile2File(FILE *fsrc, FILE *fdest) {
655 260 bool result = false;
656 260 StreamStates stream_state = kStreamIOError;
657 z_stream strm;
658 size_t have;
659 unsigned char buf[kBufferSize];
660
661
1/2
✓ Branch 1 taken 260 times.
✗ Branch 2 not taken.
 260 DecompressInit(&strm);
662
663
3/4
✓ Branch 1 taken 468 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 208 times.
✓ Branch 4 taken 260 times.
 468 while ((have = fread(buf, 1, kBufferSize, fsrc)) > 0) {
664
1/2
✓ Branch 1 taken 208 times.
✗ Branch 2 not taken.
 208 stream_state = DecompressZStream2File(buf, have, &strm, fdest);
665
2/4
✓ Branch 0 taken 208 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 208 times.
 208 if ((stream_state == kStreamDataError) || (stream_state == kStreamIOError))
666 ✗ goto decompress_file2file_final;
667 }
668
1/2
✓ Branch 2 taken 260 times.
✗ Branch 3 not taken.
 260 LogCvmfs(kLogCompress, kLogDebug, "end of decompression, state=%d, error=%d",
669 stream_state, ferror(fsrc));
670
5/6
✓ Branch 0 taken 208 times.
✓ Branch 1 taken 52 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 208 times.
✓ Branch 5 taken 52 times.
✓ Branch 6 taken 208 times.
 260 if ((stream_state != kStreamEnd) || ferror(fsrc))
671 52 goto decompress_file2file_final;
672
673 208 result = true;
674
675 260 decompress_file2file_final:
676
1/2
✓ Branch 1 taken 260 times.
✗ Branch 2 not taken.
 260 DecompressFini(&strm);
677 260 return result;
678 }
679
680
681 260 bool DecompressPath2File(const string &src, FILE *fdest) {
682 260 FILE *fsrc = fopen(src.c_str(), "r");
683
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 260 times.
 260 if (!fsrc)
684 ✗ return false;
685
686 260 bool retval = DecompressFile2File(fsrc, fdest);
687 260 fclose(fsrc);
688 260 return retval;
689 }
690
691
692 49 bool CompressMem2File(const unsigned char *buf, const size_t size, FILE *fdest,
693 shash::Any *compressed_hash) {
694 49 int z_ret = 0;
695 49 int flush = 0;
696 49 bool result = false;
697 unsigned have;
698 z_stream strm;
699 49 size_t offset = 0;
700 49 size_t used = 0;
701 unsigned char out[kZChunk];
702
1/2
✓ Branch 1 taken 49 times.
✗ Branch 2 not taken.
 49 shash::ContextPtr hash_context(compressed_hash->algorithm);
703
704
1/2
✓ Branch 1 taken 49 times.
✗ Branch 2 not taken.
 49 CompressInit(&strm);
705 49 hash_context.buffer = alloca(hash_context.size);
706
1/2
✓ Branch 1 taken 49 times.
✗ Branch 2 not taken.
 49 shash::Init(hash_context);
707
708 // Compress the given memory buffer
709 do {
710 245 used = min(static_cast<size_t>(kZChunk), size - offset);
711 245 strm.avail_in = used;
712
713
2/2
✓ Branch 0 taken 49 times.
✓ Branch 1 taken 196 times.
 245 flush = (strm.avail_in < kZChunk) ? Z_FINISH : Z_NO_FLUSH;
714 245 strm.next_in = const_cast<unsigned char *>(buf + offset);
715
716 // Run deflate() on input until output buffer not full, finish
717 // compression if all of source has been read in
718 do {
719 392 strm.avail_out = kZChunk;
720 392 strm.next_out = out;
721
1/2
✓ Branch 1 taken 392 times.
✗ Branch 2 not taken.
 392 z_ret = deflate(&strm, flush); // no bad return value
722
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 392 times.
 392 if (z_ret == Z_STREAM_ERROR)
723 ✗ goto compress_file2file_hashed_final; // state not clobbered
724 392 have = kZChunk - strm.avail_out;
725
4/9
✓ Branch 1 taken 392 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 392 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 392 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 392 times.
 392 if (fwrite(out, 1, have, fdest) != have || ferror(fdest))
726 ✗ goto compress_file2file_hashed_final;
727
1/2
✓ Branch 1 taken 392 times.
✗ Branch 2 not taken.
 392 shash::Update(out, have, hash_context);
728
2/2
✓ Branch 0 taken 147 times.
✓ Branch 1 taken 245 times.
 392 } while (strm.avail_out == 0);
729
730 245 offset += used;
731
732 // Done when last data in file processed
733
2/2
✓ Branch 0 taken 196 times.
✓ Branch 1 taken 49 times.
 245 } while (flush != Z_FINISH);
734
735 // Stream will be complete
736
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 49 times.
 49 if (z_ret != Z_STREAM_END)
737 ✗ goto compress_file2file_hashed_final;
738
739
1/2
✓ Branch 1 taken 49 times.
✗ Branch 2 not taken.
 49 shash::Final(hash_context, compressed_hash);
740 49 result = true;
741
742 // Clean up and return
743 49 compress_file2file_hashed_final:
744
1/2
✓ Branch 1 taken 49 times.
✗ Branch 2 not taken.
 49 CompressFini(&strm);
745
1/2
✓ Branch 1 taken 49 times.
✗ Branch 2 not taken.
 49 LogCvmfs(kLogCompress, kLogDebug, "file compression finished with result %d",
746 result);
747 49 return result;
748 }
749
750
751 /**
752 * User of this function has to free out_buf.
753 */
754 1257 bool CompressMem2Mem(const void *buf, const int64_t size, void **out_buf,
755 uint64_t *out_size) {
756 unsigned char out[kZChunk];
757 int z_ret;
758 int flush;
759 z_stream strm;
760 1257 int64_t pos = 0;
761 1257 uint64_t alloc_size = kZChunk;
762
763
1/2
✓ Branch 1 taken 1257 times.
✗ Branch 2 not taken.
 1257 CompressInit(&strm);
764 1257 *out_buf = smalloc(alloc_size);
765 1257 *out_size = 0;
766
767 do {
768 51384 strm.avail_in = (kZChunk > (size - pos)) ? size - pos : kZChunk;
769
2/2
✓ Branch 0 taken 1257 times.
✓ Branch 1 taken 50127 times.
 51384 flush = (pos + kZChunk) >= size ? Z_FINISH : Z_NO_FLUSH;
770 51384 strm.next_in = ((unsigned char *)buf) + pos;
771
772 // Run deflate() on input until output buffer not full
773 do {
774 52168 strm.avail_out = kZChunk;
775 52168 strm.next_out = out;
776
1/2
✓ Branch 1 taken 52168 times.
✗ Branch 2 not taken.
 52168 z_ret = deflate(&strm, flush);
777
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 52168 times.
 52168 if (z_ret == Z_STREAM_ERROR) {
778 ✗ CompressFini(&strm);
779 ✗ free(*out_buf);
780 ✗ *out_buf = NULL;
781 ✗ *out_size = 0;
782 ✗ return false;
783 }
784 52168 size_t have = kZChunk - strm.avail_out;
785
2/2
✓ Branch 0 taken 245 times.
✓ Branch 1 taken 51923 times.
 52168 if (*out_size + have > alloc_size) {
786 245 alloc_size *= 2;
787 245 *out_buf = srealloc(*out_buf, alloc_size);
788 }
789 52168 memcpy(static_cast<unsigned char *>(*out_buf) + *out_size, out, have);
790 52168 *out_size += have;
791
2/2
✓ Branch 0 taken 784 times.
✓ Branch 1 taken 51384 times.
 52168 } while (strm.avail_out == 0);
792
793 51384 pos += kZChunk;
794
2/2
✓ Branch 0 taken 50127 times.
✓ Branch 1 taken 1257 times.
 51384 } while (flush != Z_FINISH);
795
796
1/2
✓ Branch 1 taken 1257 times.
✗ Branch 2 not taken.
 1257 CompressFini(&strm);
797
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1257 times.
 1257 if (z_ret != Z_STREAM_END) {
798 ✗ free(*out_buf);
799 ✗ *out_buf = NULL;
800 ✗ *out_size = 0;
801 ✗ return false;
802 } else {
803 1257 return true;
804 }
805 }
806
807
808 /**
809 * User of this function has to free out_buf.
810 */
811 2813 bool DecompressMem2Mem(const void *buf, const int64_t size, void **out_buf,
812 uint64_t *out_size) {
813 unsigned char out[kZChunk];
814 int z_ret;
815 z_stream strm;
816 2813 int64_t pos = 0;
817 2813 uint64_t alloc_size = kZChunk;
818
819
1/2
✓ Branch 1 taken 2813 times.
✗ Branch 2 not taken.
 2813 DecompressInit(&strm);
820 2813 *out_buf = smalloc(alloc_size);
821 2813 *out_size = 0;
822
823 do {
824 36093 strm.avail_in = (kZChunk > (size - pos)) ? size - pos : kZChunk;
825 36093 strm.next_in = ((unsigned char *)buf) + pos;
826
827 // Run inflate() on input until output buffer not full
828 do {
829 36119 strm.avail_out = kZChunk;
830 36119 strm.next_out = out;
831
1/2
✓ Branch 1 taken 36119 times.
✗ Branch 2 not taken.
 36119 z_ret = inflate(&strm, Z_NO_FLUSH);
832
1/3
✗ Branch 0 not taken.
✗ Branch 1 not taken.
✓ Branch 2 taken 36119 times.
 36119 switch (z_ret) {
833 ✗ case Z_NEED_DICT:
834 ✗ z_ret = Z_DATA_ERROR; // and fall through
835 ✗ case Z_STREAM_ERROR:
836 case Z_DATA_ERROR:
837 case Z_MEM_ERROR:
838 ✗ DecompressFini(&strm);
839 ✗ free(*out_buf);
840 ✗ *out_buf = NULL;
841 ✗ *out_size = 0;
842 ✗ return false;
843 }
844 36119 size_t have = kZChunk - strm.avail_out;
845
2/2
✓ Branch 0 taken 286 times.
✓ Branch 1 taken 35833 times.
 36119 if (*out_size + have > alloc_size) {
846 286 alloc_size *= 2;
847 286 *out_buf = srealloc(*out_buf, alloc_size);
848 }
849 36119 memcpy(static_cast<unsigned char *>(*out_buf) + *out_size, out, have);
850 36119 *out_size += have;
851
2/2
✓ Branch 0 taken 26 times.
✓ Branch 1 taken 36093 times.
 36119 } while (strm.avail_out == 0);
852
853 36093 pos += kZChunk;
854
2/2
✓ Branch 0 taken 33280 times.
✓ Branch 1 taken 2813 times.
 36093 } while (pos < size);
855
856
1/2
✓ Branch 1 taken 2813 times.
✗ Branch 2 not taken.
 2813 DecompressFini(&strm);
857
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2813 times.
 2813 if (z_ret != Z_STREAM_END) {
858 ✗ free(*out_buf);
859 ✗ *out_buf = NULL;
860 ✗ *out_size = 0;
861 ✗ return false;
862 }
863
864 2813 return true;
865 }
866
867
868 //------------------------------------------------------------------------------
869
870
871 49 void Compressor::RegisterPlugins() {
872 49 RegisterPlugin<ZlibCompressor>();
873 49 RegisterPlugin<EchoCompressor>();
874 49 }
875
876
877 //------------------------------------------------------------------------------
878
879
880 6989926 bool ZlibCompressor::WillHandle(const zlib::Algorithms &alg) {
881 6989926 return alg == kZlibDefault;
882 }
883
884
885 6991981 ZlibCompressor::ZlibCompressor(const Algorithms &alg) : Compressor(alg) {
886 6991897 stream_.zalloc = Z_NULL;
887 6991897 stream_.zfree = Z_NULL;
888 6991897 stream_.opaque = Z_NULL;
889 6991897 stream_.next_in = Z_NULL;
890 6991897 stream_.avail_in = 0;
891
1/2
✓ Branch 1 taken 7012981 times.
✗ Branch 2 not taken.
 6991897 const int zlib_retval = deflateInit(&stream_, Z_DEFAULT_COMPRESSION);
892
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7012981 times.
 7012981 assert(zlib_retval == 0);
893 7012981 }
894
895
896 ✗ Compressor *ZlibCompressor::Clone() {
897 ✗ ZlibCompressor *other = new ZlibCompressor(zlib::kZlibDefault);
898 ✗ assert(stream_.avail_in == 0);
899 // Delete the other stream
900 ✗ int retcode = deflateEnd(&other->stream_);
901 ✗ assert(retcode == Z_OK);
902 ✗ retcode = deflateCopy(const_cast<z_streamp>(&other->stream_), &stream_);
903 ✗ assert(retcode == Z_OK);
904 ✗ return other;
905 }
906
907 23622707 bool ZlibCompressor::Deflate(const bool flush, unsigned char **inbuf,
908 size_t *inbufsize, unsigned char **outbuf,
909 size_t *outbufsize) {
910 // Adding compression
911 23622707 stream_.avail_in = *inbufsize;
912 23622707 stream_.next_in = *inbuf;
913
2/2
✓ Branch 0 taken 12467193 times.
✓ Branch 1 taken 11155514 times.
 23622707 const int flush_int = (flush) ? Z_FINISH : Z_NO_FLUSH;
914 23622707 int retcode = 0;
915
916 // TODO(jblomer) Figure out what exactly behaves differently with zlib 1.2.10
917 // if ((*inbufsize == 0) && !flush)
918 // return true;
919
920 23622707 stream_.avail_out = *outbufsize;
921 23622707 stream_.next_out = *outbuf;
922
923 // Deflate in zlib!
924 23622707 retcode = deflate(&stream_, flush_int);
925
3/4
✓ Branch 0 taken 7012617 times.
✓ Branch 1 taken 16664494 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 7012617 times.
 23677111 assert(retcode == Z_OK || retcode == Z_STREAM_END);
926
927 23677111 *outbufsize -= stream_.avail_out;
928 23677111 *inbuf = stream_.next_in;
929 23677111 *inbufsize = stream_.avail_in;
930
931
3/4
✓ Branch 0 taken 11200538 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1892541 times.
✓ Branch 3 taken 9307997 times.
 11200538 return (flush_int == Z_NO_FLUSH && retcode == Z_OK && stream_.avail_in == 0)
932
6/6
✓ Branch 0 taken 11200538 times.
✓ Branch 1 taken 12476573 times.
✓ Branch 2 taken 12477553 times.
✓ Branch 3 taken 1891561 times.
✓ Branch 4 taken 7003209 times.
✓ Branch 5 taken 5474344 times.
 34877649 || (flush_int == Z_FINISH && retcode == Z_STREAM_END);
933 }
934
935
936 27991780 ZlibCompressor::~ZlibCompressor() {
937 13996002 int retcode = deflateEnd(&stream_);
938
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7012113 times.
 14024226 assert(retcode == Z_OK);
939 28020004 }
940
941
942 26 size_t ZlibCompressor::DeflateBound(const size_t bytes) {
943 // Call zlib's deflate bound
944 26 return deflateBound(&stream_, bytes);
945 }
946
947
948 //------------------------------------------------------------------------------
949
950
951 101 EchoCompressor::EchoCompressor(const zlib::Algorithms &alg)
952 101 : Compressor(alg) { }
953
954
955 101 bool EchoCompressor::WillHandle(const zlib::Algorithms &alg) {
956 101 return alg == kNoCompression;
957 }
958
959
960 ✗ Compressor *EchoCompressor::Clone() {
961 ✗ return new EchoCompressor(zlib::kNoCompression);
962 }
963
964
965 5452691 bool EchoCompressor::Deflate(const bool flush, unsigned char **inbuf,
966 size_t *inbufsize, unsigned char **outbuf,
967 size_t *outbufsize) {
968 5452691 size_t bytes_to_copy = min(*outbufsize, *inbufsize);
969 5452691 memcpy(*outbuf, *inbuf, bytes_to_copy);
970 5452691 const bool done = (bytes_to_copy == *inbufsize);
971
972 // Update the return variables
973 5452691 *inbuf += bytes_to_copy;
974 5452691 *outbufsize = bytes_to_copy;
975 5452691 *inbufsize -= bytes_to_copy;
976
977 5452691 return done;
978 }
979
980
981 26 size_t EchoCompressor::DeflateBound(const size_t bytes) {
982 // zero bytes as an upper bound is no good because some callers want to
983 // allocate buffers according to this value
984
1/2
✓ Branch 0 taken 26 times.
✗ Branch 1 not taken.
 26 return (bytes == 0) ? 1 : bytes;
985 }
986
987 } // namespace zlib
988