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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
#include "fitz.h"
#include <zlib.h>
typedef struct fz_flate_s fz_flate;
struct fz_flate_s
{
fz_stream *chain;
z_stream z;
};
static void *zalloc(void *opaque, unsigned int items, unsigned int size)
{
return fz_malloc_array_no_throw(opaque, items, size);
}
static void zfree(void *opaque, void *ptr)
{
fz_free(opaque, ptr);
}
static int
read_flated(fz_stream *stm, unsigned char *outbuf, int outlen)
{
fz_flate *state = stm->state;
fz_stream *chain = state->chain;
z_streamp zp = &state->z;
int code;
zp->next_out = outbuf;
zp->avail_out = outlen;
while (zp->avail_out > 0)
{
if (chain->rp == chain->wp)
fz_fill_buffer(chain);
zp->next_in = chain->rp;
zp->avail_in = chain->wp - chain->rp;
code = inflate(zp, Z_SYNC_FLUSH);
chain->rp = chain->wp - zp->avail_in;
if (code == Z_STREAM_END)
{
return outlen - zp->avail_out;
}
else if (code == Z_BUF_ERROR)
{
fz_warn(stm->ctx, "premature end of data in flate filter");
return outlen - zp->avail_out;
}
else if (code == Z_DATA_ERROR && zp->avail_in == 0)
{
fz_warn(stm->ctx, "ignoring zlib error: %s", zp->msg);
return outlen - zp->avail_out;
}
else if (code != Z_OK)
{
fz_throw(stm->ctx, "zlib error: %s", zp->msg);
}
}
return outlen - zp->avail_out;
}
static void
close_flated(fz_context *ctx, void *state_)
{
fz_flate *state = (fz_flate *)state_;
int code;
code = inflateEnd(&state->z);
if (code != Z_OK)
fz_warn(ctx, "zlib error: inflateEnd: %s", state->z.msg);
fz_close(state->chain);
fz_free(ctx, state);
}
fz_stream *
fz_open_flated(fz_stream *chain)
{
fz_flate *state;
int code = Z_OK;
fz_context *ctx = chain->ctx;
fz_var(code);
state = fz_malloc_struct(ctx, fz_flate);
state->chain = chain;
state->z.zalloc = zalloc;
state->z.zfree = zfree;
state->z.opaque = ctx;
state->z.next_in = NULL;
state->z.avail_in = 0;
fz_try(ctx)
{
code = inflateInit(&state->z);
if (code != Z_OK)
fz_throw(ctx, "zlib error: inflateInit: %s", state->z.msg);
}
fz_catch(ctx)
{
if (code == Z_OK)
inflateEnd(&state->z);
fz_free(ctx, state);
fz_rethrow(ctx);
}
return fz_new_stream(chain->ctx, state, read_flated, close_flated);
}
|