lib.rs

#![no_std]
#![feature(abi_x86_interrupt)]
#![feature(allocator_api)]
#![allow(clippy::assertions_on_constants)]
#![allow(clippy::manual_is_multiple_of)]
#![allow(clippy::result_unit_err)]
#![allow(clippy::missing_safety_doc)]
#![allow(clippy::manual_contains)]
#![allow(clippy::doc_lazy_continuation)]
extern crate alloc;

// 导入 drivers crate 的宏
extern crate drivers;
#[macro_use]
extern crate klog;

/// R180-7: fallible retained containers bound to the whole-heap byte ledger.
pub mod admitted;
pub mod buddy_allocator;
/// 3.3 bounded slab/NUMA/swap/THP capability primitives.
pub mod capabilities;
pub mod dma;
/// R169-11: `FallibleOrderedMap` lives in `mm` (a leaf crate both `kernel_core`
/// and `net` depend on) so the `net` fragment reassembler can use it for an
/// allocation-fallible per-fragment map without a `net -> kernel_core` dependency
/// cycle. Re-exported from `kernel_core::fallible_map` for source compatibility.
pub mod fallible_map;
/// R180-7..13: whole-kernel heap reservation and lifetime accounting.
pub mod heap_admission;
/// P2-A: single registry of kernel-heap hard floors / transient peaks / headroom.
pub mod heap_budget;
pub mod memory;
pub mod oom_killer;
pub mod page_cache;
pub mod page_table;
pub mod tlb_shootdown;

pub use admitted::{
    AdmittedAllocError, AdmittedDeque, AdmittedMap, AdmittedMapBuildError, AdmittedSet,
    AdmittedString, AdmittedVec, AdmittedVecIntoIter, PreparedAdmittedDequeCapacity,
    PreparedAdmittedMapCapacity, PreparedAdmittedSetCapacity, PreparedAdmittedVecCapacity,
    RetiredAdmittedDequeCapacity, RetiredAdmittedMapCapacity, RetiredAdmittedSetCapacity,
    RetiredAdmittedVecCapacity,
};
pub use capabilities::{
    run_memory_capability_self_test, NumaAllocation, NumaAllocator, NumaError, SlabCache,
    SlabError, SwapEntry, SwapError, SwapSlotTable, ThpError, ThpMapping, MAX_NUMA_NODES,
    THP_PAGE_COUNT, THP_SIZE,
};
pub use heap_admission::{
    allocation_charge_bytes, arc_charge_bytes, check_coexistence,
    class_snapshot as heap_class_snapshot, run_heap_admission_boundary_self_test,
    run_heap_admission_self_test, snapshot as heap_admission_snapshot,
    try_reserve as try_reserve_heap, vec_charge_bytes, vec_growth_charge_bytes, CoexistenceVerdict,
    HeapAdmissionError, HeapAdmissionSnapshot, HeapCharge, HeapClass, HeapReservation,
    NsBudgetError, NsBudgetLease, NsBudgetSnapshot, NsByteBudget, ADMITTED_HEAP_BYTES,
    BOOT_PEAK_USED_MAX_BYTES, BOOT_UNLEDGERED_FOOTPRINT_MAX_BYTES, HEAP_CLASS_COUNT,
    LARGEST_SINGLE_ALLOCATION_BYTES, NORMAL_UNADMITTED_RESERVE_BYTES,
    REGISTERED_FIXED_RESERVE_BYTES,
};
pub use heap_budget::{
    budget_bytes, general_residual_bytes, hard_floor_bytes, hard_floors_sum_bytes,
    is_published as heap_budgets_published, publish_and_assert as publish_heap_budgets,
    reserved_headroom_bytes, run_heap_budget_self_test, snapshot as heap_budget_snapshot,
    transient_peak_bytes, transient_peak_holders, HeapBudgetId, HeapBudgetSnapshot,
    TransientPeakGuard, AUDIT_RING_HARD_BYTES, CONNTRACK_HARD_BYTES, EXEC_IMAGE_PEAK_BYTES,
    EXEC_TRANSACTION_PEAK_BYTES, FUTEX_HARD_BYTES, HARD_FLOORS_SUM_BYTES, HARD_FLOOR_BYTES,
    HARD_FLOOR_COUNT, HARD_FLOOR_NAMES, PAGE_CACHE_META_HARD_BYTES, RESERVED_HEADROOM_BYTES,
    TRANSIENT_PEAK_BYTES,
};
pub use memory::{
    emergency_heap_base, emergency_heap_free_bytes, heap_free_bytes, heap_peak_used_bytes,
    run_emergency_heap_self_test, validate_framebuffer_region, BootInfo, EmergencyAllocator,
    FrameAllocator, MemoryMapInfo, EMERGENCY_HEAP_SIZE_BYTES, NORMAL_HEAP_SIZE_BYTES,
};
pub use oom_killer::{
    get_stats as get_oom_stats, on_allocation_failure as oom_allocation_failed,
    register_audit_callback as register_oom_audit_callback,
    register_callbacks as register_oom_callbacks, score_process as score_oom_process,
    OomProcessInfo, OomStats,
};
pub use page_cache::{
    find_or_create_page, init as init_page_cache, read_page, reclaim_pages,
    run_page_cache_policy_self_test, writeback_dirty_pages, writeback_page, GlobalPageCache,
    InodeId, MemoryPressureHandler, PageCacheEntry, PageCacheStats, PageCacheUnchargeFn, PageIndex,
    PageState, WritebackStats, PAGE_CACHE, PAGE_CACHE_CGROUP_CHARGE_BYTES, PAGE_CACHE_MAX_PAGES,
    PAGE_SIZE, PRESSURE_HANDLER,
};
pub use page_table::{
    checked_physical_range, map_mmio, phys_to_virt, pt_lock_owner_cpu, unmap_mmio,
    with_current_manager, MapError, PageTableManager, UnmapError, UpdateFlagsError,
    PHYSICAL_MEMORY_OFFSET,
};
pub use tlb_shootdown::{
    flush_all_address_spaces, flush_current_as_all, flush_current_as_page, flush_current_as_range,
    force_init_tlb_shootdown_locals, get_stats as get_tlb_stats,
};

// Host harness for fuzzing - only available when host_harness feature is enabled
#[cfg(feature = "host_harness")]
pub mod test_harness;

pub fn init() {
    klog_always!("Memory management module initialized");
}