//! The GDB's JIT compilation interface. The low level module that exposes
//! the __jit_debug_register_code() and __jit_debug_descriptor to register
//! or unregister generated object images with debuggers.

extern crate alloc;

use alloc::{boxed::Box, vec::Vec};
use core::{pin::Pin, ptr};
use wasmtime_versioned_export_macros::versioned_link;

#[repr(C)]
struct JITCodeEntry {
    next_entry: *mut JITCodeEntry,
    prev_entry: *mut JITCodeEntry,
    symfile_addr: *const u8,
    symfile_size: u64,
}

const JIT_NOACTION: u32 = 0;
const JIT_REGISTER_FN: u32 = 1;
const JIT_UNREGISTER_FN: u32 = 2;

#[repr(C)]
struct JITDescriptor {
    version: u32,
    action_flag: u32,
    relevant_entry: *mut JITCodeEntry,
    first_entry: *mut JITCodeEntry,
}

unsafe extern "C" {
    #[versioned_link]
    fn wasmtime_jit_debug_descriptor() -> *mut JITDescriptor;
    fn __jit_debug_register_code();
}

#[cfg(feature = "std")]
mod gdb_registration {
    use std::sync::{Mutex, MutexGuard};

    /// The process controls access to the __jit_debug_descriptor by itself --
    /// the GDB/LLDB accesses this structure and its data at the process startup
    /// and when paused in __jit_debug_register_code.
    ///
    /// The GDB_REGISTRATION lock is needed for GdbJitImageRegistration to protect
    /// access to the __jit_debug_descriptor within this process.
    pub static GDB_REGISTRATION: Mutex<()> = Mutex::new(());

    /// The lock guard for the GDB registration lock.
    #[expect(
        dead_code,
        reason = "field used to hold the lock until the end of the scope"
    )]
    pub struct LockGuard<'a>(MutexGuard<'a, ()>);

    pub fn lock() -> LockGuard<'static> {
        LockGuard(GDB_REGISTRATION.lock().unwrap())
    }
}

/// For no_std there's no access to synchronization primitives so a primitive
/// fallback for now is to panic-on-contention which is, in theory, rare to come
/// up as threads are rarer in no_std mode too.
///
/// This provides the guarantee that the debugger will see consistent state at
/// least, but if this panic is hit in practice it'll require some sort of
/// no_std synchronization mechanism one way or another.
#[cfg(not(feature = "std"))]
mod gdb_registration {
    use core::sync::atomic::{AtomicBool, Ordering};

    /// Whether or not a lock is held or not.
    pub static GDB_REGISTRATION: AtomicBool = AtomicBool::new(false);

    /// The lock guard for the GDB registration lock.
    pub struct LockGuard;

    /// When the `LockGuard` is dropped, it releases the lock by setting
    /// `GDB_REGISTRATION` to false.
    impl Drop for LockGuard {
        fn drop(&mut self) {
            GDB_REGISTRATION.store(false, Ordering::Release);
        }
    }

    /// Locks the GDB registration lock. If the lock is already held, it panics
    pub fn lock() -> LockGuard {
        if GDB_REGISTRATION
            .compare_exchange(false, true, Ordering::Acquire, Ordering::Relaxed)
            .is_err()
        {
            panic!("GDB JIT registration lock contention detected in no_std mode");
        }
        LockGuard
    }
}

/// Registration for JIT image
pub struct GdbJitImageRegistration {
    entry: Pin<Box<JITCodeEntry>>,
    file: Pin<Box<[u8]>>,
}

impl GdbJitImageRegistration {
    /// Registers JIT image using __jit_debug_register_code
    pub fn register(file: Vec<u8>) -> Self {
        let file = Pin::new(file.into_boxed_slice());

        // Create a code entry for the file, which gives the start and size
        // of the symbol file.
        let mut entry = Pin::new(Box::new(JITCodeEntry {
            next_entry: ptr::null_mut(),
            prev_entry: ptr::null_mut(),
            symfile_addr: file.as_ptr(),
            symfile_size: file.len() as u64,
        }));

        unsafe {
            register_gdb_jit_image(&mut *entry);
        }

        Self { entry, file }
    }

    /// JIT image used in registration
    pub fn file(&self) -> &[u8] {
        &self.file
    }
}

impl Drop for GdbJitImageRegistration {
    fn drop(&mut self) {
        unsafe {
            unregister_gdb_jit_image(&mut *self.entry);
        }
    }
}

unsafe impl Send for GdbJitImageRegistration {}
unsafe impl Sync for GdbJitImageRegistration {}

unsafe fn register_gdb_jit_image(entry: *mut JITCodeEntry) {
    unsafe {
        let _lock = gdb_registration::lock();
        let desc = &mut *wasmtime_jit_debug_descriptor();

        // Add it to the linked list in the JIT descriptor.
        (*entry).next_entry = desc.first_entry;
        if !desc.first_entry.is_null() {
            (*desc.first_entry).prev_entry = entry;
        }
        desc.first_entry = entry;
        // Point the relevant_entry field of the descriptor at the entry.
        desc.relevant_entry = entry;
        // Set action_flag to JIT_REGISTER and call __jit_debug_register_code.
        desc.action_flag = JIT_REGISTER_FN;
        __jit_debug_register_code();

        desc.action_flag = JIT_NOACTION;
        desc.relevant_entry = ptr::null_mut();
    }
}

unsafe fn unregister_gdb_jit_image(entry: *mut JITCodeEntry) {
    unsafe {
        let _lock = gdb_registration::lock();
        let desc = &mut *wasmtime_jit_debug_descriptor();

        // Remove the code entry corresponding to the code from the linked list.
        if !(*entry).prev_entry.is_null() {
            (*(*entry).prev_entry).next_entry = (*entry).next_entry;
        } else {
            desc.first_entry = (*entry).next_entry;
        }
        if !(*entry).next_entry.is_null() {
            (*(*entry).next_entry).prev_entry = (*entry).prev_entry;
        }
        // Point the relevant_entry field of the descriptor at the code entry.
        desc.relevant_entry = entry;
        // Set action_flag to JIT_UNREGISTER and call __jit_debug_register_code.
        desc.action_flag = JIT_UNREGISTER_FN;
        __jit_debug_register_code();

        desc.action_flag = JIT_NOACTION;
        desc.relevant_entry = ptr::null_mut();
    }
}