fuzz/decoder.c - master - niceSOFT/openssl

fuzz/decoder.c
101 строка3 KB
Jakub Zelenka
Fix fuzz test to handle memory failures
15 июл 2026, 19:01
15 июл 2026, 19:01c263ea5
/*
 * Copyright 2023-2024 The OpenSSL Project Authors. All Rights Reserved.
 *
 * Licensed under the Apache License 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * https://www.openssl.org/source/license.html
 * or in the file LICENSE in the source distribution.
 */

#include <openssl/decoder.h>
#include <openssl/err.h>
#include <openssl/rand.h>
#include "fuzzer.h"

static ASN1_PCTX *pctx;

int FuzzerInitialize(int *argc, char ***argv)
{
    FuzzerSetRand();

    OPENSSL_init_crypto(OPENSSL_INIT_LOAD_CRYPTO_STRINGS
            | OPENSSL_INIT_ADD_ALL_CIPHERS
            | OPENSSL_INIT_ADD_ALL_DIGESTS,
        NULL);

    pctx = ASN1_PCTX_new();
    if (pctx == NULL)
        return 0;
    ASN1_PCTX_set_flags(pctx, ASN1_PCTX_FLAGS_SHOW_ABSENT | ASN1_PCTX_FLAGS_SHOW_SEQUENCE | ASN1_PCTX_FLAGS_SHOW_SSOF | ASN1_PCTX_FLAGS_SHOW_TYPE | ASN1_PCTX_FLAGS_SHOW_FIELD_STRUCT_NAME);
    ASN1_PCTX_set_str_flags(pctx, ASN1_STRFLGS_UTF8_CONVERT | ASN1_STRFLGS_SHOW_TYPE | ASN1_STRFLGS_DUMP_ALL);

    ERR_clear_error();
    CRYPTO_free_ex_index(0, -1);
    return 1;
}

int FuzzerTestOneInput(const uint8_t *buf, size_t len)
{
    OSSL_DECODER_CTX *dctx;
    EVP_PKEY *pkey = NULL;
    EVP_PKEY_CTX *ctx = NULL;
    BIO *bio;

    bio = BIO_new(BIO_s_null());
    if (bio == NULL) {
        ERR_clear_error();
        return 0;
    }
    dctx = OSSL_DECODER_CTX_new_for_pkey(&pkey, NULL, NULL, NULL, 0, NULL,
        NULL);
    if (dctx == NULL) {
        BIO_free(bio);
        ERR_clear_error();
        return 0;
    }
    if (OSSL_DECODER_from_data(dctx, &buf, &len)) {
        EVP_PKEY *pkey2;

        EVP_PKEY_print_public(bio, pkey, 1, pctx);
        EVP_PKEY_print_private(bio, pkey, 1, pctx);
        EVP_PKEY_print_params(bio, pkey, 1, pctx);

        pkey2 = EVP_PKEY_dup(pkey);
        if (pkey2 != NULL) {
            EVP_PKEY_eq(pkey, pkey2);
            EVP_PKEY_free(pkey2);
        }

        ctx = EVP_PKEY_CTX_new(pkey, NULL);
        if (ctx != NULL) {
            /*
             * Param check will take too long time on large DH parameters.
             * Skip it.
             */
            if ((!EVP_PKEY_is_a(pkey, "DH") && !EVP_PKEY_is_a(pkey, "DHX"))
                || EVP_PKEY_get_bits(pkey) <= 2048)
                EVP_PKEY_param_check(ctx);

            EVP_PKEY_public_check(ctx);
            /* Private and pairwise checks are unbounded, skip for large keys. */
            if (EVP_PKEY_get_bits(pkey) <= 4096) {
                EVP_PKEY_private_check(ctx);
                EVP_PKEY_pairwise_check(ctx);
            }
            EVP_PKEY_CTX_free(ctx);
        }
        EVP_PKEY_free(pkey);
    }
    OSSL_DECODER_CTX_free(dctx);

    BIO_free(bio);
    ERR_clear_error();
    return 0;
}

void FuzzerCleanup(void)
{
    ASN1_PCTX_free(pctx);
    FuzzerClearRand();
}