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
|
#include "test-tool.h"
#include "git-zlib.h"
#include "strbuf.h"
static const char *zlib_usage = "test-tool zlib [inflate|deflate]";
static void do_zlib(struct git_zstream *stream,
int (*zlib_func)(git_zstream *, int),
int fd_in, int fd_out)
{
struct strbuf buf_in = STRBUF_INIT;
int status = Z_OK;
if (strbuf_read(&buf_in, fd_in, 0) < 0)
die_errno("read error");
stream->next_in = (unsigned char *)buf_in.buf;
stream->avail_in = buf_in.len;
while (status == Z_OK ||
(status == Z_BUF_ERROR && !stream->avail_out)) {
unsigned char buf_out[4096];
stream->next_out = buf_out;
stream->avail_out = sizeof(buf_out);
status = zlib_func(stream, Z_FINISH);
if (write_in_full(fd_out, buf_out,
sizeof(buf_out) - stream->avail_out) < 0)
die_errno("write error");
}
if (status != Z_STREAM_END)
die("zlib error %d", status);
strbuf_release(&buf_in);
}
int cmd__zlib(int argc, const char **argv)
{
git_zstream stream;
if (argc != 2)
usage(zlib_usage);
memset(&stream, 0, sizeof(stream));
if (!strcmp(argv[1], "inflate")) {
git_inflate_init(&stream);
do_zlib(&stream, git_inflate, 0, 1);
git_inflate_end(&stream);
} else if (!strcmp(argv[1], "deflate")) {
git_deflate_init(&stream, Z_DEFAULT_COMPRESSION);
do_zlib(&stream, git_deflate, 0, 1);
git_deflate_end(&stream);
} else {
error("unknown mode: %s", argv[1]);
usage(zlib_usage);
}
return 0;
}
|
