1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
/*
Copyright 1996-2014 Han The Thanh, <thanh@pdftex.org>
This file is part of pdfTeX.
pdfTeX is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
pdfTeX is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License along
with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "ptexlib.h"
#include "zlib.h"
#include <assert.h>
#define ZIP_BUF_SIZE 32768
#define check_err(f, fn) \
if (f != Z_OK) \
pdftex_fail("zlib: %s() failed (error code %d)", fn, f)
static char *zipbuf = NULL;
static z_stream c_stream; /* compression stream */
void writezip(boolean finish)
{
int err;
static int level_old = 0;
int level = getpdfcompresslevel();
assert(level > 0);
cur_file_name = NULL;
if (pdfstreamlength == 0) {
if (zipbuf == NULL) {
zipbuf = xtalloc(ZIP_BUF_SIZE, char);
c_stream.zalloc = (alloc_func) 0;
c_stream.zfree = (free_func) 0;
c_stream.opaque = (voidpf) 0;
check_err(deflateInit(&c_stream, level), "deflateInit");
} else {
if (level != level_old) { /* \pdfcompresslevel change in mid document */
check_err(deflateEnd(&c_stream), "deflateEnd");
c_stream.zalloc = (alloc_func) 0; /* these 3 lines no need, just to be safe */
c_stream.zfree = (free_func) 0;
c_stream.opaque = (voidpf) 0;
check_err(deflateInit(&c_stream, level), "deflateInit");
} else
check_err(deflateReset(&c_stream), "deflateReset");
}
level_old = level;
c_stream.next_out = (Bytef *) zipbuf;
c_stream.avail_out = ZIP_BUF_SIZE;
}
assert(zipbuf != NULL);
c_stream.next_in = pdfbuf;
c_stream.avail_in = pdfptr;
for (;;) {
if (c_stream.avail_out == 0) {
pdfgone += xfwrite(zipbuf, 1, ZIP_BUF_SIZE, pdffile);
pdflastbyte = zipbuf[ZIP_BUF_SIZE - 1]; /* not needed */
c_stream.next_out = (Bytef *) zipbuf;
c_stream.avail_out = ZIP_BUF_SIZE;
}
err = deflate(&c_stream, finish ? Z_FINISH : Z_NO_FLUSH);
if (finish && err == Z_STREAM_END)
break;
check_err(err, "deflate");
if (!finish && c_stream.avail_in == 0)
break;
}
if (finish) {
if (c_stream.avail_out < ZIP_BUF_SIZE) { /* at least one byte has been output */
pdfgone +=
xfwrite(zipbuf, 1, ZIP_BUF_SIZE - c_stream.avail_out, pdffile);
pdflastbyte = zipbuf[ZIP_BUF_SIZE - c_stream.avail_out - 1];
}
xfflush(pdffile);
}
pdfstreamlength = c_stream.total_out;
}
void zip_free(void)
{
if (zipbuf != NULL) {
check_err(deflateEnd(&c_stream), "deflateEnd");
free(zipbuf);
}
}
|
