continuing work on v2

dev
Paul Cruz 2017-07-03 14:18:46 -07:00
parent ac79928960
commit 00b5e6c512
2 changed files with 80 additions and 2 deletions

View File

@ -18,9 +18,14 @@ CFLAGS += $(DEBUGFLAGS)
CFLAGS += $(MOREFLAGS)
FLAGS = $(CPPFLAGS) $(CFLAGS) $(LDFLAGS)
adaptive: $(ZSTD_FILES) v1.c
all: clean v1 v2
v1: $(ZSTD_FILES) v1.c
$(CC) $(FLAGS) $^ -o $@
v2: $(ZSTD_FILES) v2.c
$(CC) $(FLAGS) $^ -o $@
clean:
@$(RM) -f adaptive
@$(RM) -f v1 v2
@$(RM) -rf *.dSYM
@$(RM) -f tmp*

View File

@ -0,0 +1,73 @@
#define DISPLAY(...) fprintf(stderr, __VA_ARGS__)
#define FILE_CHUNK_SIZE 4 << 20
typedef unsigned char BYTE;
#include <stdio.h>
#include <stdlib.h>
#include "zstd.h"
/* return 0 if successful, else return error */
int main(int argCount, const char* argv[])
{
const char* const srcFilename = argv[1];
const char* const dstFilename = argv[2];
FILE* const srcFile = fopen(srcFilename, "rb");
FILE* const dstFile = fopen(dstFilename, "wb");
BYTE* const src = malloc(FILE_CHUNK_SIZE);
size_t const dstSize = ZSTD_compressBound(FILE_CHUNK_SIZE);
BYTE* const dst = malloc(dstSize);
int ret = 0;
/* checking for errors */
if (!srcFilename || !dstFilename || !src || !dst) {
DISPLAY("Error: initial variables could not be allocated\n");
ret = 1;
goto cleanup;
}
/* compressing in blocks */
for ( ; ; ) {
size_t const readSize = fread(src, 1, FILE_CHUNK_SIZE, srcFile);
if (readSize != FILE_CHUNK_SIZE && !feof(srcFile)) {
DISPLAY("Error: could not read %d bytes\n", FILE_CHUNK_SIZE);
ret = 1;
goto cleanup;
}
{
size_t const compressedSize = ZSTD_compress(dst, dstSize, src, readSize, 6);
if (ZSTD_isError(compressedSize)) {
DISPLAY("Error: something went wrong during compression\n");
ret = 1;
goto cleanup;
}
{
size_t const writeSize = fwrite(dst, 1, compressedSize, dstFile);
if (writeSize != compressedSize) {
DISPLAY("Error: could not write compressed data to file\n");
ret = 1;
goto cleanup;
}
}
}
if (feof(srcFile)) {
/* reached end of file */
break;
}
}
/* file compression completed */
{
int const error = fclose(srcFile);
if (ret != 0) {
DISPLAY("Error: could not close the file\n");
ret = error;
goto cleanup;
}
}
cleanup:
if (src != NULL) free(src);
if (dst != NULL) free(dst);
return ret;
}