fx/fx.compression/test/stream.c

#include <assert.h>
#include <fx/compression/compressor.h>
#include <fx/compression/cstream.h>
#include <fx/compression/zstd.h>
#include <fx/ringbuffer.h>
#include <fx/stream.h>
#include <stdio.h>
#include <string.h>

#define BUF_SIZE 32

static int compress(fx_type_id compressor_type, FILE *in, FILE *out)
{
	fx_stream *out_stream = fx_stream_open_fp(out);

	fx_cstream *cstream;
	fx_status status = fx_cstream_open(
		out_stream,
		compressor_type,
		FX_COMPRESSOR_MODE_COMPRESS,
		&cstream);

	if (!FX_OK(status)) {
		fprintf(stderr, "cannot initialise compressor\n");
		return -1;
	}

	fx_cstream_begin_compressed_section(cstream, NULL);

	char buf[4096];
	while (1) {
		size_t r = fread(buf, 1, sizeof buf, in);

		size_t w = 0;
		fx_cstream_write(cstream, buf, r, &w);

		if (r != w) {
			fprintf(stderr, "write error\n");
			return -1;
		}

		if (r < sizeof buf) {
			break;
		}
	}

	fx_cstream_end_compressed_section(cstream, NULL, NULL);
	fx_cstream_unref(cstream);
	fx_stream_unref(out_stream);

	return 0;
}

static int decompress(fx_type_id compressor_type, FILE *in, FILE *out)
{
	fx_stream *in_stream = fx_stream_open_fp(in);

	fx_cstream *cstream;
	fx_status status = fx_cstream_open(
		in_stream,
		compressor_type,
		FX_COMPRESSOR_MODE_DECOMPRESS,
		&cstream);

	if (!FX_OK(status)) {
		fprintf(stderr, "cannot initialise compressor\n");
		return -1;
	}

	fx_cstream_begin_compressed_section(cstream, NULL);

	char buf[4096];
	while (1) {
		size_t r = 0;
		fx_status status = fx_cstream_read(
			cstream,
			buf,
			sizeof buf,
			&r);
		if (!FX_OK(status)) {
			fprintf(stderr,
			        "read error: %s\n",
			        fx_status_description(status));
			return -1;
		}

		size_t w = fwrite(buf, 1, r, out);
		if (r != w) {
			fprintf(stderr, "write error\n");
			return -1;
		}

		if (r < sizeof buf) {
			break;
		}
	}

	fx_cstream_end_compressed_section(cstream, NULL, NULL);
	fx_cstream_unref(cstream);
	fx_stream_unref(in_stream);

	return 0;
}

int main(int argc, const char **argv)
{
	if (argc < 4) {
		fprintf(stderr,
		        "usage: %s <C/D> <inpath> <outpath>\n",
		        argv[0]);
		return -1;
	}

	fx_compressor_mode mode;
	if (!strcmp(argv[1], "C")) {
		mode = FX_COMPRESSOR_MODE_COMPRESS;
	} else if (!strcmp(argv[1], "D")) {
		mode = FX_COMPRESSOR_MODE_DECOMPRESS;
	} else {
		fprintf(stderr, "invalid mode %s\n", argv[1]);
		return -1;
	}

	FILE *in_fp = fopen(argv[2], "rb");
	if (!in_fp) {
		fprintf(stderr, "cannot open input file %s\n", argv[2]);
		return -1;
	}

	FILE *out_fp = fopen(argv[3], "wb");
	if (!out_fp) {
		fclose(in_fp);
		fprintf(stderr, "cannot open output file %s\n", argv[3]);
		return -1;
	}

	int ret = 0;
	switch (mode) {
	case FX_COMPRESSOR_MODE_COMPRESS:
		ret = compress(FX_TYPE_ZSTD_COMPRESSOR, in_fp, out_fp);
		break;
	case FX_COMPRESSOR_MODE_DECOMPRESS:
		ret = decompress(FX_TYPE_ZSTD_COMPRESSOR, in_fp, out_fp);
		break;
	default:
		ret = -1;
		break;
	}

	printf("Done\n");

	fclose(in_fp);
	fclose(out_fp);

	return ret;
}