/*

 * Copyright (C) 2019-2023 by Sukchan Lee <acetcom@gmail.com>

 *

 * This file is part of Open5GS.

 *

 * This program is free software: you can redistribute it and/or modify

 * it under the terms of the GNU Affero General Public License as published by

 * the Free Software Foundation, either version 3 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program.  If not, see <https://www.gnu.org/licenses/>.

 */

#include <stdio.h>

#include <stdint.h>

#include "fuzzing.h"

#include "ogs-gtp.h"

#define kMinInputLength 5

#define kMaxInputLength 1024

#define GTP_VERSION(byte) (((byte) >> 5) & 0x07)

extern int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) 

{ /* open5gs/tests/non3gpp/gtp-path.c */

    if (Size < kMinInputLength || Size > kMaxInputLength) {

        return 1;

    }

    if (!initialized) {

        initialize();

        ogs_log_install_domain(&__ogs_gtp_domain, "gtp", OGS_LOG_NONE);

        ogs_log_install_domain(&__ogs_tlv_domain, "tlv", OGS_LOG_NONE);

    }

    ogs_pkbuf_t *pkbuf;

    pkbuf = ogs_pkbuf_alloc(NULL, OGS_MAX_SDU_LEN);

    if (pkbuf == NULL) {

        return 1;

    }

    ogs_pkbuf_put_data(pkbuf, Data, Size);

    if (GTP_VERSION(Data[0]) == OGS_GTP1_VERSION_1) {

        /* GTPv1 */

        ogs_gtp1_message_t gtp1_message;

        ogs_gtp1_parse_msg(&gtp1_message, pkbuf);

    } else {

        /* GTPv2 */

        ogs_gtp2_message_t gtp2_message;

        ogs_gtp2_parse_msg(&gtp2_message, pkbuf);

    }

    ogs_pkbuf_free(pkbuf);

    return 0;

}