githubmirror/zstd

contrib/seqBench/seqBench.c
53 строки2 KB
Danielle Rozenblit
use ZSTD_sequenceBound in seqBench
09 сен 2022, 23:04
09 сен 2022, 23:041613caf
#define ZSTD_STATIC_LINKING_ONLY
#include <zstd.h>
#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
#include <string.h>

int main(int argc, char *argv[]) {
    ZSTD_CCtx* zc = ZSTD_createCCtx();

    if (argc != 2) {
        printf("Usage: seqBench <file>\n"); // TODO provide the block delim option here
        return 1;
    }

    FILE *f = fopen(argv[1], "rb");
    fseek(f, 0, SEEK_END);
    long inBufSize = ftell(f);
    fseek(f, 0, SEEK_SET);

    char *inBuf = malloc(inBufSize + 1);
    fread(inBuf, inBufSize, 1, f);
    fclose(f);

    size_t seqsSize = ZSTD_sequenceBound(inBufSize);
    ZSTD_Sequence *seqs = (ZSTD_Sequence*)malloc(seqsSize * sizeof(ZSTD_Sequence));
    char *outBuf = malloc(ZSTD_compressBound(inBufSize));

    ZSTD_generateSequences(zc, seqs, seqsSize, inBuf, inBufSize);
    ZSTD_CCtx_setParameter(zc, ZSTD_c_blockDelimiters, ZSTD_sf_explicitBlockDelimiters);
    size_t outBufSize = ZSTD_compressSequences(zc, outBuf, inBufSize, seqs, seqsSize, inBuf, inBufSize);
    if (ZSTD_isError(outBufSize)) {
        printf("ERROR: %lu\n", outBufSize);
        return 1;
    }

    char *validationBuf = malloc(inBufSize);
    ZSTD_decompress(validationBuf, inBufSize, outBuf, outBufSize);

    if (memcmp(inBuf, validationBuf, inBufSize) == 0) {
        printf("Compression and decompression were successful!\n");
    } else {
        printf("ERROR: input and validation buffers don't match!\n");
        for (int i = 0; i < inBufSize; i++) {
            if (inBuf[i] != validationBuf[i]) {
                printf("First bad index: %d\n", i);
                break;
            }
        }
    }

    return 0;
}