summaryrefslogtreecommitdiff
path: root/meshmc/launcher/GZip.cpp
blob: 2f63e93bd59d035d7f5f292805c53ebb12550abf (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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
/* SPDX-FileCopyrightText: 2026 Project Tick
 * SPDX-FileContributor: Project Tick
 * SPDX-License-Identifier: GPL-3.0-or-later
 *
 *   MeshMC - A Custom Launcher for Minecraft
 *   Copyright (C) 2026 Project Tick
 *
 *   This program 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.
 *
 *   This program 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 this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#include "GZip.h"
#include <zlib.h>
#include <QByteArray>

bool GZip::unzip(const QByteArray& compressedBytes,
				 QByteArray& uncompressedBytes)
{
	if (compressedBytes.size() == 0) {
		uncompressedBytes = compressedBytes;
		return true;
	}

	unsigned uncompLength = compressedBytes.size();
	uncompressedBytes.clear();
	uncompressedBytes.resize(uncompLength);

	z_stream strm;
	memset(&strm, 0, sizeof(strm));
	strm.next_in = (Bytef*)compressedBytes.data();
	strm.avail_in = compressedBytes.size();

	bool done = false;

	if (inflateInit2(&strm, (16 + MAX_WBITS)) != Z_OK) {
		return false;
	}

	int err = Z_OK;

	while (!done) {
		// If our output buffer is too small
		if (strm.total_out >= uncompLength) {
			uncompressedBytes.resize(uncompLength * 2);
			uncompLength *= 2;
		}

		strm.next_out = (Bytef*)(uncompressedBytes.data() + strm.total_out);
		strm.avail_out = uncompLength - strm.total_out;

		// Inflate another chunk.
		err = inflate(&strm, Z_SYNC_FLUSH);
		if (err == Z_STREAM_END)
			done = true;
		else if (err != Z_OK) {
			break;
		}
	}

	if (inflateEnd(&strm) != Z_OK || !done) {
		return false;
	}

	uncompressedBytes.resize(strm.total_out);
	return true;
}

bool GZip::zip(const QByteArray& uncompressedBytes, QByteArray& compressedBytes)
{
	if (uncompressedBytes.size() == 0) {
		compressedBytes = uncompressedBytes;
		return true;
	}

	unsigned compLength =
		std::min(static_cast<qsizetype>(16), uncompressedBytes.size());
	compressedBytes.clear();
	compressedBytes.resize(compLength);

	z_stream zs;
	memset(&zs, 0, sizeof(zs));

	if (deflateInit2(&zs, Z_DEFAULT_COMPRESSION, Z_DEFLATED, (16 + MAX_WBITS),
					 8, Z_DEFAULT_STRATEGY) != Z_OK) {
		return false;
	}

	zs.next_in = (Bytef*)uncompressedBytes.data();
	zs.avail_in = uncompressedBytes.size();

	int ret;
	compressedBytes.resize(uncompressedBytes.size());

	unsigned offset = 0;
	unsigned temp = 0;
	do {
		auto remaining = compressedBytes.size() - offset;
		if (remaining < 1) {
			compressedBytes.resize(compressedBytes.size() * 2);
		}
		zs.next_out = (Bytef*)(compressedBytes.data() + offset);
		temp = zs.avail_out = compressedBytes.size() - offset;
		ret = deflate(&zs, Z_FINISH);
		offset += temp - zs.avail_out;
	} while (ret == Z_OK);

	compressedBytes.resize(offset);

	if (deflateEnd(&zs) != Z_OK) {
		return false;
	}

	if (ret != Z_STREAM_END) {
		return false;
	}
	return true;
}