summaryrefslogtreecommitdiff
path: root/src/compress.c
blob: c27937bed6a8c583e9a8838a32638ef145a62e42 (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
54
55
56
57
/* compress.c -- Wrappers for compression and decompression with zlib.

   Copyright (C) 2017 Jakob Tsar-Fox, All Rights Reserved.

   This file is part of Nekopack.

   Nekopack is free software: you can redistribute it and/or modify it
   under the terms of the GNU General Public License as published by the
   Free Software Foundation, either version 3 of the License, or (at
   your option) any later version.

   Nekopack is distributed in the hope that it will be useful, but
   WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
   General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with Nekopack. If not, see <http://www.gnu.org/licenses/>. */

#include <zlib.h>

#include "io.h"


/* Inflates `s` into a newly allocated stream structure. */
struct stream *inflate_stream(struct stream *s, size_t inflated_len) {
    z_stream strm;
    strm.zalloc = Z_NULL;
    strm.zfree = Z_NULL;
    strm.opaque = Z_NULL;
    strm.avail_in = 0;
    strm.next_in = Z_NULL;

    if (inflateInit(&strm) != Z_OK)
        return NULL;

    int ret;
    struct stream *n = stream_new(inflated_len);

    do {
        strm.avail_in = s->len;
        strm.next_in = (Bytef *) s->_cur;
        do {
            strm.avail_out = inflated_len;
            strm.next_out = (Bytef *) n->_cur;
            ret = inflate(&strm, Z_NO_FLUSH);
            if (ret != Z_OK) {
                stream_free(n);
                inflateEnd(&strm);
                return NULL;
            }
        } while (strm.avail_out == 0);
    } while (ret != Z_STREAM_END);

    inflateEnd(&strm);
    return n;
}