tests/ossfuzz/standaloneengine.cc - master - niceSOFT/libssh2

tests/ossfuzz/standaloneengine.cc
71 строка2 KB
Viktor Szakats
tidy-up: drop most `{}` blocks around single-statements
21 июн 2026, 05:59
21 июн 2026, 05:5933f6508
/* Copyright (C) The libssh2 project and its contributors.
 *
 * SPDX-License-Identifier: BSD-3-Clause
 */
#include <stdio.h>
#include <stdlib.h>

#include "testinput.h"

/**
 * Main procedure for standalone fuzzing engine.
 *
 * Reads filenames from the argument array. For each filename, read the file
 * into memory and then call the fuzzing interface with the data.
 */
int main(int argc, char **argv)
{
    int ii;
    for(ii = 1; ii < argc; ii++) {
        FILE *infile;
        printf("[%s] ", argv[ii]);

        /* Try and open the file. */
        infile = fopen(argv[ii], "rb");
        if(infile) {
            uint8_t *buffer = NULL;
            size_t buffer_len;

            printf("Opened.. ");

            /* Get the length of the file. */
            fseek(infile, 0L, SEEK_END);
            buffer_len = (size_t)ftell(infile);

            /* Reset the file indicator to the beginning of the file. */
            fseek(infile, 0L, SEEK_SET);

            /* Allocate a buffer for the file contents. */
            buffer = (uint8_t *)calloc(buffer_len, sizeof(uint8_t));
            if(buffer) {
                /* Read all the text from the file into the buffer. */
                fread(buffer, sizeof(uint8_t), buffer_len, infile);
                printf("Read %zu bytes, fuzzing.. ", buffer_len);

                /* Call the fuzzer with the data. */
                LLVMFuzzerTestOneInput(buffer, buffer_len);

                printf("complete.");

                /* Free the buffer as it is no longer needed. */
                free(buffer);
                buffer = NULL;
            }
            else
                fprintf(stderr,
                        "[%s] Failed to allocate %zu bytes \n",
                        argv[ii],
                        buffer_len);

            /* Close the file as it is no longer needed. */
            fclose(infile);
            infile = NULL;
        }
        else
            /* Failed to open the file.
               Maybe wrong name or wrong permissions? */
            fprintf(stderr, "[%s] Open failed. \n", argv[ii]);

        printf("\n");
    }
}