/* Copyright (C) 2021 Open Information Security Foundation

 *

 * You can copy, redistribute or modify this Program under the terms of

 * the GNU General Public License version 2 as published by the Free

 * Software Foundation.

 *

 * 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

 * version 2 along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA

 * 02110-1301, USA.

 */

use crate::core::{STREAM_TOCLIENT, STREAM_TOSERVER};

use crate::quic::quic::QuicTransaction;

use std::os::raw::c_void;

use std::ptr;

use suricata_sys::sys::DetectEngineThreadCtx;

#[no_mangle]

pub unsafe extern "C" fn SCQuicTxGetUa(

    tx: &QuicTransaction, buffer: *mut *const u8, buffer_len: *mut u32,

) -> u8 {

    if let Some(ua) = &tx.ua {

        *buffer = ua.as_ptr();

        *buffer_len = ua.len() as u32;

        1

    } else {

        *buffer = ptr::null();

        *buffer_len = 0;

        0

    }

}

#[no_mangle]

pub unsafe extern "C" fn SCQuicTxGetSni(

    tx: &QuicTransaction, buffer: *mut *const u8, buffer_len: *mut u32,

) -> u8 {

    if let Some(sni) = &tx.sni {

        *buffer = sni.as_ptr();

        *buffer_len = sni.len() as u32;

        1

    } else {

        *buffer = ptr::null();

        *buffer_len = 0;

        0

    }

}

#[no_mangle]

pub unsafe extern "C" fn SCQuicTxGetJa3(

    tx: *const c_void, dir: u8, buffer: *mut *const u8, buffer_len: *mut u32,

) -> bool {

    if tx.is_null() {

        *buffer = ptr::null();

        *buffer_len = 0;

        return false;

    }

    let tx = cast_pointer!(tx, QuicTransaction);

    if tx.client {

        if dir & STREAM_TOSERVER == 0 {

            return false;

        }

    } else if dir & STREAM_TOCLIENT == 0 {

        return false;

    }

    if let Some(ja3) = &tx.ja3 {

        *buffer = ja3.as_ptr();

        *buffer_len = ja3.len() as u32;

        return true;

    }

    return false;

}

#[no_mangle]

pub unsafe extern "C" fn SCQuicTxGetJa4(

    tx: *const c_void, buffer: *mut *const u8, buffer_len: *mut u32,

) -> u8 {

    if tx.is_null() {

        *buffer = ptr::null();

        *buffer_len = 0;

        return 0;

    }

    let tx = cast_pointer!(tx, QuicTransaction);

    if let Some(ja4) = &tx.ja4 {

        *buffer = ja4.as_ref().as_ptr();

        *buffer_len = ja4.as_ref().len() as u32;

        1

    } else {

        *buffer = ptr::null();

        *buffer_len = 0;

        0

    }

}

#[no_mangle]

pub unsafe extern "C" fn SCQuicTxGetVersion(

    tx: &QuicTransaction, buffer: *mut *const u8, buffer_len: *mut u32,

) -> u8 {

    if tx.header.flags.is_long {

        let s = &tx.header.version_buf;

        *buffer = s.as_ptr();

        *buffer_len = s.len() as u32;

        1

    } else {

        *buffer = ptr::null();

        *buffer_len = 0;

        0

    }

}

#[no_mangle]

pub unsafe extern "C" fn SCQuicTxGetCyuHash(

    _de: *mut DetectEngineThreadCtx, tx: *const c_void, _flags: u8, i: u32, buffer: *mut *const u8,

    buffer_len: *mut u32,

) -> bool {

    let tx = cast_pointer!(tx, QuicTransaction);

    if (i as usize) < tx.cyu.len() {

        let cyu = &tx.cyu[i as usize];

        let p = &cyu.hash;

        *buffer = p.as_ptr();

        *buffer_len = p.len() as u32;

        true

    } else {

        *buffer = ptr::null();

        *buffer_len = 0;

        false

    }

}

#[no_mangle]

pub unsafe extern "C" fn SCQuicTxGetCyuString(

    _de: *mut DetectEngineThreadCtx, tx: *const c_void, _flags: u8, i: u32, buffer: *mut *const u8,

    buffer_len: *mut u32,

) -> bool {

    let tx = cast_pointer!(tx, QuicTransaction);

    if (i as usize) < tx.cyu.len() {

        let cyu = &tx.cyu[i as usize];

        let p = &cyu.string;

        *buffer = p.as_ptr();

        *buffer_len = p.len() as u32;

        true

    } else {

        *buffer = ptr::null();

        *buffer_len = 0;

        false

    }

}