#ifndef UTSM_DSK_H
#define UTSM_DSK_H

#include <utsm/types.h>

#define DSK_BOOT_MAGIC 0x44534B31424F4F54ULL /* "DSK1BOOT" */
#define DSK_BOOT_ABI_VERSION 1u

#define DSK_BOOT_FLAG_FROM_UTSM       (1ULL << 0)
#define DSK_BOOT_FLAG_DKM_READY       (1ULL << 1)
#define DSK_BOOT_FLAG_FAT32_PATH      (1ULL << 2)
#define DSK_BOOT_FLAG_PROBE_INFO      (1ULL << 3)

/* ---- Probe 信息结构（通过 reserved[6] 传递给 shell/desktop） ---- */
#define PROBE_INFO_MAGIC 0x50524F42u  /* "PRFB" — Probe inFo Buffer */

typedef struct probe_info {
    u32 magic;              /* PROBE_INFO_MAGIC */
    u32 version;            /* ABI 版本，当前 1 */
    u64 probe_buf_addr;     /* probe_entry 缓冲基地址（物理/HHDM 地址） */
    u32 probe_cap;          /* 缓冲容量（条目数） */
    u32 probe_head;         /* 当前写入位置（递增取模） */
    u32 probe_count;        /* 已写入总条目数 */
    u32 probe_dropped;      /* 因缓冲满而覆盖的条目数 */
    u32 instr_enabled;      /* g_instr_enabled 当前值 */
    u32 instr_probe_enable; /* g_instr_probe_enable 当前值 */
    u64 reserved[4];
} probe_info;

typedef struct dsk_boot_context {
    u64 magic;
    u32 abi_version;
    u32 size;

    u64 flags;

    u64 hhdm_offset;
    u64 rsdp_address;

    u64 framebuffer_address;
    u64 framebuffer_width;
    u64 framebuffer_height;
    u64 framebuffer_pitch;
    u32 framebuffer_bpp;
    u32 reserved_fb;

    u64 boot_modules_response;

    u64 dkm_kernel_api;
    u64 dkm_driver_table;
    u64 dkm_driver_count;

    u64 utsm_state;
    u64 drr_state;

    u64 memory_map;
    u64 memory_map_count;
    u64 memory_map_entry_size;

    u64 kernel_stack_top;
    u64 reserved[8];
} dsk_boot_context;

typedef void (*dsk_entry_fn)(const dsk_boot_context *context);

int dsk_load_and_jump(void);

#endif