/* This is a fuzz test of the permessage-deflate module */

#define WIN32_EXPORT

#include <cstdio>
#include <string>

/* We test the permessage deflate module */
#include "../src/PerMessageDeflate.h"

#include "helpers.h"

extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {

    /* First byte determines what compressor to use */
    if (size < 1) {
        return 0;
    }

    int compressors[] = {
        //uWS::DEDICATED_COMPRESSOR_3KB, -- 3kb has msan issues on OSS-Fuzz (bug in zlib)
        uWS::DEDICATED_COMPRESSOR_4KB,
        uWS::DEDICATED_COMPRESSOR_8KB,
        uWS::DEDICATED_COMPRESSOR_16KB,
        uWS::DEDICATED_COMPRESSOR_32KB,
        uWS::DEDICATED_COMPRESSOR_64KB,
        uWS::DEDICATED_COMPRESSOR_128KB,
        uWS::DEDICATED_COMPRESSOR_256KB
    };

    auto compressor = compressors[data[0] % 7];
    data++;
    size--;

    /* If we could specify LARGE_BUFFER_SIZE small here we could force it to inflate in chunks,
     * triggering more line coverage. Currently it is set to 16kb which is always too much */
    struct StaticData {
        uWS::DeflationStream deflationStream;
        uWS::ZlibContext zlibContext;

        uWS::InflationStream inflationStream;
    } staticData = {compressor};

    /* Why is this padded? */
    makeChunked(makePadded(data, size), size, [&staticData](const uint8_t *data, size_t size) {
        auto [inflation, valid] = staticData.inflationStream.inflate(&staticData.zlibContext, std::string_view((char *) data, size), 256);
        if (inflation.length() > 256) {
            /* Cause ASAN to freak out */
            delete (int *) (void *) 1;
        }
    });

    makeChunked(makePadded(data, size), size, [&staticData](const uint8_t *data, size_t size) {
        /* Always reset */
        staticData.deflationStream.deflate(&staticData.zlibContext, std::string_view((char *) data, size), true);
    });

    return 0;
}