[PATCH 1/5] Introduce stream compress helpers (v2 and complete)
From: Marco Costalba <hidden>
Date: 2016-06-15 22:44:04
Subsystem:
kernel build + files below scripts/ (unless maintained elsewhere), the rest · Maintainers:
Nathan Chancellor, Nicolas Schier, Linus Torvalds
When decompressing a zlib stream use this helpers instead of calling low level zlib function. This patch introduces the necessary framework, still no code change. This is the first step in generalizing compress and decompress functions avoiding zlib directly calls. Signed-off-by: Marco Costalba <redacted> --- This is the full compress cleanup series, no more open coded zlib compress calls are around anymore New in this version is: - Renaming according to Linus suggestions - Added of a compress_free() that seems to suit nicely - git is zlib compress functions free now Next step will be decompress helpers Makefile | 4 ++-- compress.c | 58 ++++++++++++++++++++++++++++++++++++ compress.h | 12 ++++++++++++ 3 files changed, 72 insertions(+), 2 deletions(-) create mode 100644 compress.c create mode 100644 compress.h
diff --git a/Makefile b/Makefile
index 21c80e6..89bd99d 100644
--- a/Makefile
+++ b/Makefile@@ -288,7 +288,7 @@ LIB_FILE=libgit.a XDIFF_LIB=xdiff/lib.a LIB_H = \ - archive.h blob.h cache.h cache-tree.h commit.h csum-file.h delta.h grep.h \ + archive.h blob.h cache.h cache-tree.h commit.h compress.h
csum-file.h delta.h grep.h \ diff.h object.h pack.h pkt-line.h quote.h refs.h list-objects.h sideband.h \ run-command.h strbuf.h tag.h tree.h git-compat-util.h revision.h \ tree-walk.h log-tree.h dir.h path-list.h unpack-trees.h builtin.h \
@@ -301,7 +301,7 @@ DIFF_OBJS = \ diffcore-delta.o log-tree.o LIB_OBJS = \ - blob.o commit.o connect.o csum-file.o cache-tree.o base85.o \ + blob.o commit.o compress.o connect.o csum-file.o cache-tree.o base85.o \ date.o diff-delta.o entry.o exec_cmd.o ident.o \ pretty.o interpolate.o hash.o \ lockfile.o \
diff --git a/compress.c b/compress.c
new file mode 100644
index 0000000..be771a9
--- /dev/null
+++ b/compress.c@@ -0,0 +1,58 @@ +#include "cache.h" +#include "compress.h" + +unsigned long compress_alloc(z_stream *stream, int level, unsigned long size) +{ + memset(stream, 0, sizeof(*stream)); + deflateInit(stream, level); + return deflateBound(stream, size); +} + +int compress_start(z_stream *stream, + unsigned char *in, unsigned long in_size, + unsigned char *out, unsigned long out_size) +{ + stream->next_out = (out ? out : xmalloc(out_size)); + stream->avail_out = out_size; + stream->next_in = in; + stream->avail_in = in_size; + return Z_OK; +} + +int compress_next(z_stream *stream, int flush) +{ + int result; + + do { + result = deflate(stream, flush); + } while (result == Z_OK); + + return result; +} + +unsigned long compress_free(z_stream *stream) +{ + deflateEnd(stream); + return stream->total_out; +} + +unsigned long compress_all(int level, unsigned char *data, + unsigned long size, unsigned char **out) +{ + int bound, result; + z_stream stream; + + bound = compress_alloc(&stream, level, size); + compress_start(&stream, data, size, NULL, bound); + + *out = stream.next_out; + result = compress_next(&stream, Z_FINISH); + + if (result != Z_STREAM_END) { + compress_free(&stream); + free(*out); + *out = NULL; + return 0; + } + return compress_free(&stream); +}
diff --git a/compress.h b/compress.h
new file mode 100644
index 0000000..16b0147
--- /dev/null
+++ b/compress.h@@ -0,0 +1,12 @@ +#ifndef COMPRESS_H +#define COMPRESS_H + +extern unsigned long compress_alloc(z_stream *stream, int level,
unsigned long size); +extern int compress_start(z_stream *stream, unsigned char *in, unsigned long in_size, + unsigned char *out, unsigned long out_size); +extern int compress_next(z_stream *stream, int flush); +extern unsigned long compress_free(z_stream *stream); +extern unsigned long compress_all(int level, unsigned char *data, unsigned long size, + unsigned char **out); + +#endif -- 1.5.4.rc2.89.g1b3f-dirty