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
|
#include "mupdf/fitz.h"
#include <zlib.h>
typedef struct fz_leech_s fz_leech;
struct fz_leech_s
{
fz_stream *chain;
fz_buffer *buffer;
};
static int
next_leech(fz_context *ctx, fz_stream *stm, size_t max)
{
fz_leech *state = stm->state;
fz_buffer *buffer = state->buffer;
size_t n = fz_available(ctx, state->chain, max);
if (n > max)
n = max;
while (buffer->cap < buffer->len + n)
{
fz_grow_buffer(ctx, state->buffer);
}
memcpy(buffer->data + buffer->len, state->chain->rp, n);
stm->rp = buffer->data + buffer->len;
stm->wp = buffer->data + buffer->len + n;
state->chain->rp += n;
buffer->len += n;
if (n == 0)
return EOF;
return *stm->rp++;
}
static void
close_leech(fz_context *ctx, void *state_)
{
fz_leech *state = (fz_leech *)state_;
fz_drop_stream(ctx, state->chain);
fz_free(ctx, state);
}
fz_stream *
fz_open_leecher(fz_context *ctx, fz_stream *chain, fz_buffer *buffer)
{
fz_leech *state = NULL;
fz_var(state);
fz_try(ctx)
{
state = fz_malloc_struct(ctx, fz_leech);
state->chain = chain;
state->buffer = buffer;
}
fz_catch(ctx)
{
fz_free(ctx, state);
fz_drop_stream(ctx, chain);
fz_rethrow(ctx);
}
return fz_new_stream(ctx, state, next_leech, close_leech);
}
|
