f2ed511a2f
It's occasionally useful when testing or debugging to be able to do raw zlib inflate/deflate operations (e.g., to check the bytes of a specific loose or packed object). Even though zlib's deflate algorithm is used by many other programs, this is surprisingly hard to do in a portable way. E.g., gzip can do this if you manually munge some header bytes. But the result is somewhat arcane, and we don't assume gzip is available anyway. Likewise, pigz will handle raw zlib, but we can't assume it is available. So let's introduce a short test helper for just doing zlib operations. We'll use it in subsequent patches to add some new tests, but it would also have come in handy a few times in the past: - The hard-coded pack data from3b910d0c5e(add tests for indexing packs with delta cycles, 2013-08-23) could probably be generated on the fly. - Likewise we could avoid the hard-coded data from0b1493c2d4(git_inflate(): skip zlib_post_call() sanity check on Z_NEED_DICT, 2025-02-25). Though note this would require support for more zlib options. - It would have helped with the debugging documented in41dfbb2dbe(howto: add article on recovering a corrupted object, 2013-10-25). I'll leave refactoring existing tests for another day, but I hope the examples above show the general utility. I aimed for simplicity in the code. In particular, it will read all input into a memory buffer, rather than streaming. That makes the zlib loops harder to get wrong (which has been a source of subtle bugs in the past). Signed-off-by: Jeff King <peff@peff.net> Signed-off-by: Junio C Hamano <gitster@pobox.com>
63 lines
1.4 KiB
C
63 lines
1.4 KiB
C
#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;
|
|
}
|