blob: 14b79404c85a6993b648c3539175a57b0df2905a (
plain)
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
  | 
/*
 * Copyright (C) the libgit2 contributors. All rights reserved.
 *
 * This file is part of libgit2, distributed under the GNU GPL v2 with
 * a Linking Exception. For full terms see the included COPYING file.
 */
#include "compress.h"
#include <zlib.h>
#define BUFFER_SIZE (1024 * 1024)
int git__compress(git_buf *buf, const void *buff, size_t len)
{
	z_stream zs;
	char *zb;
	size_t have;
	memset(&zs, 0, sizeof(zs));
	if (deflateInit(&zs, Z_DEFAULT_COMPRESSION) != Z_OK)
		return -1;
	zb = git__malloc(BUFFER_SIZE);
	GITERR_CHECK_ALLOC(zb);
	zs.next_in = (void *)buff;
	zs.avail_in = (uInt)len;
	do {
		zs.next_out = (unsigned char *)zb;
		zs.avail_out = BUFFER_SIZE;
		if (deflate(&zs, Z_FINISH) == Z_STREAM_ERROR) {
			git__free(zb);
			return -1;
		}
		have = BUFFER_SIZE - (size_t)zs.avail_out;
		if (git_buf_put(buf, zb, have) < 0) {
			git__free(zb);
			return -1;
		}
	} while (zs.avail_out == 0);
	assert(zs.avail_in == 0);
	deflateEnd(&zs);
	git__free(zb);
	return 0;
}
  |