[linux-2.6-block.git] / tools / perf / util / zlib.c

// SPDX-License-Identifier: GPL-2.0
#include <fcntl.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <zlib.h>
#include <linux/compiler.h>
#include <unistd.h>

#include "util/compress.h"
#include "util/util.h"
#include "util/debug.h"


#define CHUNK_SIZE  16384

int gzip_decompress_to_file(const char *input, int output_fd)
{
	int ret = Z_STREAM_ERROR;
	int input_fd;
	void *ptr;
	int len;
	struct stat stbuf;
	unsigned char buf[CHUNK_SIZE];
	z_stream zs = {
		.zalloc		= Z_NULL,
		.zfree		= Z_NULL,
		.opaque		= Z_NULL,
		.avail_in	= 0,
		.next_in	= Z_NULL,
	};

	input_fd = open(input, O_RDONLY);
	if (input_fd < 0)
		return -1;

	if (fstat(input_fd, &stbuf) < 0)
		goto out_close;

	ptr = mmap(NULL, stbuf.st_size, PROT_READ, MAP_PRIVATE, input_fd, 0);
	if (ptr == MAP_FAILED)
		goto out_close;

	if (inflateInit2(&zs, 16 + MAX_WBITS) != Z_OK)
		goto out_unmap;

	zs.next_in = ptr;
	zs.avail_in = stbuf.st_size;

	do {
		zs.next_out = buf;
		zs.avail_out = CHUNK_SIZE;

		ret = inflate(&zs, Z_NO_FLUSH);
		switch (ret) {
		case Z_NEED_DICT:
			ret = Z_DATA_ERROR;
			/* fall through */
		case Z_DATA_ERROR:
		case Z_MEM_ERROR:
			goto out;
		default:
			break;
		}

		len = CHUNK_SIZE - zs.avail_out;
		if (writen(output_fd, buf, len) != len) {
			ret = Z_DATA_ERROR;
			goto out;
		}

	} while (ret != Z_STREAM_END);

out:
	inflateEnd(&zs);
out_unmap:
	munmap(ptr, stbuf.st_size);
out_close:
	close(input_fd);

	return ret == Z_STREAM_END ? 0 : -1;
}

bool gzip_is_compressed(const char *input)
{
	int fd = open(input, O_RDONLY);
	const uint8_t magic[2] = { 0x1f, 0x8b };
	char buf[2] = { 0 };
	ssize_t rc;

	if (fd < 0)
		return -1;

	rc = read(fd, buf, sizeof(buf));
	close(fd);
	return rc == sizeof(buf) ?
	       memcmp(buf, magic, sizeof(buf)) == 0 : false;
}
Commit	Line	Data
b2441318	1	// SPDX-License-Identifier: GPL-2.0
c23c2a0f	2	#include <fcntl.h>
e92ce12e NK	3	#include <stdio.h>
	4	#include <unistd.h>
	5	#include <sys/stat.h>
	6	#include <sys/mman.h>
	7	#include <zlib.h>
8b42b7e5	8	#include <linux/compiler.h>
88c74dc7	9	#include <unistd.h>
e92ce12e	10
611f0afe	11	#include "util/compress.h"
e92ce12e NK	12	#include "util/util.h"
	13	#include "util/debug.h"
	14
	15
	16	#define CHUNK_SIZE 16384
	17
	18	int gzip_decompress_to_file(const char *input, int output_fd)
	19	{
	20	int ret = Z_STREAM_ERROR;
	21	int input_fd;
	22	void *ptr;
	23	int len;
	24	struct stat stbuf;
	25	unsigned char buf[CHUNK_SIZE];
	26	z_stream zs = {
	27	.zalloc = Z_NULL,
	28	.zfree = Z_NULL,
	29	.opaque = Z_NULL,
	30	.avail_in = 0,
	31	.next_in = Z_NULL,
	32	};
	33
	34	input_fd = open(input, O_RDONLY);
	35	if (input_fd < 0)
	36	return -1;
	37
	38	if (fstat(input_fd, &stbuf) < 0)
	39	goto out_close;
	40
	41	ptr = mmap(NULL, stbuf.st_size, PROT_READ, MAP_PRIVATE, input_fd, 0);
	42	if (ptr == MAP_FAILED)
	43	goto out_close;
	44
	45	if (inflateInit2(&zs, 16 + MAX_WBITS) != Z_OK)
	46	goto out_unmap;
	47
	48	zs.next_in = ptr;
	49	zs.avail_in = stbuf.st_size;
	50
	51	do {
	52	zs.next_out = buf;
	53	zs.avail_out = CHUNK_SIZE;
	54
	55	ret = inflate(&zs, Z_NO_FLUSH);
	56	switch (ret) {
	57	case Z_NEED_DICT:
	58	ret = Z_DATA_ERROR;
	59	/* fall through */
	60	case Z_DATA_ERROR:
	61	case Z_MEM_ERROR:
	62	goto out;
	63	default:
	64	break;
	65	}
	66
	67	len = CHUNK_SIZE - zs.avail_out;
	68	if (writen(output_fd, buf, len) != len) {
	69	ret = Z_DATA_ERROR;
	70	goto out;
	71	}
	72
	73	} while (ret != Z_STREAM_END);
	74
	75	out:
76	inflateEnd(&zs);
77	out_unmap:
78	munmap(ptr, stbuf.st_size);
79	out_close:
80	close(input_fd);
81
82	return ret == Z_STREAM_END ? 0 : -1;
83	}
8b42b7e5	84
88c74dc7	85	bool gzip_is_compressed(const char *input)
8b42b7e5	86	{
88c74dc7 JO	87	int fd = open(input, O_RDONLY);
	88	const uint8_t magic[2] = { 0x1f, 0x8b };
	89	char buf[2] = { 0 };
	90	ssize_t rc;
	91
	92	if (fd < 0)
	93	return -1;
	94
	95	rc = read(fd, buf, sizeof(buf));
	96	close(fd);
	97	return rc == sizeof(buf) ?
	98	memcmp(buf, magic, sizeof(buf)) == 0 : false;
8b42b7e5	99	}