Add boot options with on-screen debugging

The static local variable RingFlag in HlComPortReadLsr caused shared state across multiple calls and ports, leading to race conditions and incorrect behavior.

.github/workflows/build.yml

@@ -7,6 +7,7 @@ jobs:
     strategy:
       matrix:
         arch: [amd64, i686]
+        build: [debug, release]
     runs-on: oscw
     container:
       image: codingworkshop/oscw-runner:latest
@@ -17,7 +18,7 @@ jobs:
           fetch-depth: 0
       - name: Build ExectOS
         run: |
-          echo "charch ${{ matrix.arch }} && ./configure.sh && cd build-${{ matrix.arch }}-xtchain && xbuild -v && xbuild diskimg -v" > build.cmds
+          echo "charch ${{ matrix.arch }} && chbuild ${{ matrix.build }} && ./configure.sh && cd build-${{ matrix.arch }}-${{ matrix.build }} && xbuild -v && xbuild diskimg -v" > build.cmds
           xtchain < build.cmds
       - name: Publish binaries
         if: ${{ github.ref == 'refs/heads/master' }}
@@ -26,8 +27,8 @@ jobs:
           OSCW_ARTIFACTS_USERNAME: ${{ secrets.OSCW_ARTIFACTS_USERNAME }}
           OSCW_ARTIFACTS_USERKEY: ${{ secrets.OSCW_ARTIFACTS_USERKEY }}
         run: |
-          tar -I 'gzip' -cpf ExectOS-$(date +'%Y%m%d')-${GITHUB_SHA:0:10}-${{ matrix.arch }}-bin.tar.gz -C build-${{ matrix.arch }}-xtchain/output/binaries .
-          tar -I 'gzip' -cpf ExectOS-$(date +'%Y%m%d')-${GITHUB_SHA:0:10}-${{ matrix.arch }}-lib.tar.gz -C build-${{ matrix.arch }}-xtchain/output/library .
-          tar -I 'gzip' -cpf ExectOS-$(date +'%Y%m%d')-${GITHUB_SHA:0:10}-${{ matrix.arch }}-sym.tar.gz -C build-${{ matrix.arch }}-xtchain/output/symbols .
-          gzip -c build-${{ matrix.arch }}-xtchain/output/disk.img > ExectOS-$(date +'%Y%m%d')-${GITHUB_SHA:0:10}-${{ matrix.arch }}.img.gz
-          artifact_publish "ExectOS-$(date +'%Y%m%d')-${GITHUB_SHA:0:10}-${{ matrix.arch }}*.gz" ExectOS
+          tar -I 'gzip' -cpf ExectOS-$(date +'%Y%m%d')-${GITHUB_SHA:0:10}-${{ matrix.arch }}-${{ matrix.build }}-bin.tar.gz -C build-${{ matrix.arch }}-${{ matrix.build }}/output/binaries .
+          tar -I 'gzip' -cpf ExectOS-$(date +'%Y%m%d')-${GITHUB_SHA:0:10}-${{ matrix.arch }}-${{ matrix.build }}-lib.tar.gz -C build-${{ matrix.arch }}-${{ matrix.build }}/output/library .
+          tar -I 'gzip' -cpf ExectOS-$(date +'%Y%m%d')-${GITHUB_SHA:0:10}-${{ matrix.arch }}-${{ matrix.build }}-sym.tar.gz -C build-${{ matrix.arch }}-${{ matrix.build }}/output/symbols .
+          gzip -c build-${{ matrix.arch }}-${{ matrix.build }}/output/disk.img > ExectOS-$(date +'%Y%m%d')-${GITHUB_SHA:0:10}-${{ matrix.arch }}-${{ matrix.build }}.img.gz
+          artifact_publish "ExectOS-$(date +'%Y%m%d')-${GITHUB_SHA:0:10}-${{ matrix.arch }}-${{ matrix.build }}*.gz" ExectOS

CMakeLists.txt

@@ -1,8 +1,5 @@
-# Detect XTChain toolchain
+# Minimum CMake version requirement
 cmake_minimum_required(VERSION 3.19.0)
-if(NOT CMAKE_VERSION MATCHES "XTC")
-    message(FATAL_ERROR "XTChain not detected or corrupted!")
-endif()
 
 # Lowercase target architecture
 string(TOLOWER ${ARCH} ARCH)
@@ -44,7 +41,7 @@ set(CMAKE_TOOLCHAIN_FILE "sdk/cmake/toolchain.cmake")
 project(EXECTOS)
 
 # Load all the CMake SDK
-include(sdk/cmake/baseaddress.cmake)
+include(sdk/cmake/baseaddress/${ARCH}.cmake)
 include(sdk/cmake/emulation.cmake)
 include(sdk/cmake/functions.cmake)
 include(sdk/cmake/version.cmake)

IDEAS.md

@@ -11,15 +11,13 @@ This is a list of ideas that migh but not must be realized.
 ### XTLDR
  - [ ] Rewrite memory mapping and paging support in bootloader to make it more flexible and architecture independent.
        This should support paging levels, thus allowing to make a use of PML5 on modern AMD64 processors and increasing
-       the addressable virtual memory from 256TB to 128PB.
- - [ ] Find graphics card from all PCI devices and identify its framebuffer address when GOP is not supported by UEFI
-       firmware and UGA has to be used instead.
+       the addressable virtual memory from 256TB to 128PB. This is partially done.
+ - [ ] Implement editing boot menu entries directly from the boot menu. Changes should be runtime only (not stored on
+       disk).
 
 ### XTOSKRNL
  - [ ] Implement mechanism for detecting CPU features and checking hardware requirements. If CPU does not meet
        requirements, it should cause a kernel panic before any non-supported instruction is being used.
- - [ ] Design a mechanism of sharing common code between some architectures (i.e. both i686 and amd64 supports APIC,
-       while it is not available on ARM).
  - [ ] Finish framebuffer and terminal implementation. Initialization code is already prepared as well as routines for
        clearing the screen and drawing single points. Terminal should be instantiable (should be able to create many
        terminals and switch between them) and work on top of FB. It should define ANSI colors and scrollback buffer.

README.md

@@ -8,20 +8,20 @@
   <a href="https://git.codingworkshop.eu.org/xt-sys/exectos/actions">
     <img alt="Build Status" src="https://codingworkshop.eu.org/actions.php?project=xt-sys/exectos">
   </a>
-  <a href="https://artifacts.codingworkshop.eu.org/ExectOS">
+  <a href="https://artifacts.codingworkshop.eu.org/ExectOS/?C=M&O=D">
     <img alt="CI/CD Artifacts" src="https://img.shields.io/badge/Download-%F0%9F%A1%87-blueviolet">
   </a>
   <a href="https://git.codingworkshop.eu.org/xt-sys/exectos/src/branch/master/COPYING.md">
     <img alt="License" src="https://img.shields.io/badge/License-GPLv3-blue.svg">
   </a>
-  <a href="https://codeium.com/">
-    <img alt="Codeium" src="https://img.shields.io/badge/Powered%20By-Codeium-09B6A2?logo=Codeium">
+  <a href="https://exectos.eu.org/ai-assisted">
+    <img alt="AI Assisted" src="https://img.shields.io/badge/AI-Assisted-darkcyan">
   </a>
   <a href="https://github.com/sponsors/xt-sys/">
     <img alt="Sponsor" src="https://img.shields.io/badge/Sponsor-%E2%9D%A4-red?logo=GitHub">
   </a>
   <a href="https://discord.com/invite/zBzJ5qMGX7">
-    <img alt="Discord" src="https://img.shields.io/discord/723186294540206100?label=Chat">
+    <img alt="Discord" src="https://img.shields.io/badge/Chat-Join%20Discord-success">
   </a>
 </p>
 
@@ -69,15 +69,12 @@ design, it requires a modern EFI enabled hardware. It is not possible currently
 | xtldr       | XTOS boot loader source code                             |
 
 # Build
-XTOS can be built only by using [XTChain](https://git.codingworkshop.eu.org/xt-sys/xtchain), a special toolchain
-prepared for compiling XT software. Currently, there is only a Linux version available, so a Linux distribution or WSL
-is needed. If XTChain is already installed and available, then building ExectOS is quiet easy. First, open a terminal
-or WSL console and type the following command to launch XTChain build console:
-```
-xtchain
-```
-While the console is already running, navigate to the directory containing ExectOS source code and use the following
-commands to first set target build architecture and configure the sources:
+XTOS can only be built using [XTchain](https://git.codingworkshop.eu.org/xt-sys/xtchain), a dedicated toolchain designed
+specifically for compiling XT software. XTChain is currently available for both Linux and Windows. Detailed instructions
+on how to configure and run XTChain can be found [here](https://exectos.eu.org/contributing/setting-up-xtchain).
+
+With the XTchain environment already running, navigate to the directory containing the ExectOS source code and use the
+following commands to set the target build architecture and configure the sources:
 ```
 charch [i686|amd64]
 chbuild [DEBUG|RELEASE]

bootdata/xtldr/xtldr.ini

@@ -40,6 +40,30 @@ SystemPath=multi(0)disk(0)rdisk(0)partition(1)/ExectOS
 KernelFile=xtoskrnl.exe
 Parameters=DEBUG=COM1,115200
 
+[ExectOS_FBDEBUG]
+SystemName="ExectOS Operating System (FBDEBUG)"
+SystemType=XTOS
+BootModules=xtos_o
+SystemPath=multi(0)disk(0)rdisk(0)partition(1)/ExectOS
+KernelFile=xtoskrnl.exe
+Parameters=DEBUG=COM1,115200;SCREEN
+
+[ExectOS_NOXPA]
+SystemName="ExectOS Operating System (NOXPA)"
+SystemType=XTOS
+BootModules=xtos_o
+SystemPath=multi(0)disk(0)rdisk(0)partition(1)/ExectOS
+KernelFile=xtoskrnl.exe
+Parameters=DEBUG=COM1,115200 NOXPA
+
+[ExectOS_NOXPA_FBDEBUG]
+SystemName="ExectOS Operating System (NOXPA / FBDEBUG)"
+SystemType=XTOS
+BootModules=xtos_o
+SystemPath=multi(0)disk(0)rdisk(0)partition(1)/ExectOS
+KernelFile=xtoskrnl.exe
+Parameters=DEBUG=COM1,115200;SCREEN NOXPA
+
 [Windows]
 SystemName="Microsoft Windows 2000"
 SystemType=NT50

configure.ps1

@@ -0,0 +1,51 @@
+# PROJECT:     ExectOS
+# LICENSE:     See the COPYING.md in the top level directory
+# FILE:        configure.ps1
+# DESCRIPTION: Project configuration script for preparing the build environment
+# DEVELOPERS:  Aiken Harris <harraiken91@gmail.com>
+
+
+# Check XTchain
+if (-not $env:XTCVER) {
+    Write-Host "XTChain not detected or corrupted!"
+    exit 1
+}
+
+# Set target architecture
+if ($env:TARGET) {
+    $ARCH = $env:TARGET
+} else {
+    $ARCH = "amd64"
+}
+
+# Set target build type
+if (-not $env:BUILD_TYPE) {
+    $env:BUILD_TYPE = "DEBUG"
+}
+
+# Set variables
+$EXECTOS_SOURCE_DIR = (Get-Location).Path
+$EXECTOS_BINARY_DIR = "build-$($ARCH)-$($env:BUILD_TYPE.ToLower())"
+
+# Create directories if needed
+if ($EXECTOS_SOURCE_DIR -eq (Get-Location).Path) {
+    Write-Host "Creating directories in $EXECTOS_BINARY_DIR"
+    New-Item -ItemType Directory -Path $EXECTOS_BINARY_DIR -Force | Out-Null
+    Set-Location -Path $EXECTOS_BINARY_DIR
+}
+
+# Delete old cache
+Remove-Item -Path "CMakeCache.txt" -ErrorAction SilentlyContinue
+Remove-Item -Path "host-tools/CMakeCache.txt" -ErrorAction SilentlyContinue
+
+# Configure project using CMake
+& cmake -G Ninja -DARCH:STRING=$($ARCH) -DBUILD_TYPE:STRING=$($env:BUILD_TYPE) $EXECTOS_SOURCE_DIR
+
+# Check if configuration succeeded
+if ($LASTEXITCODE -ne 0) {
+    Write-Host "Configure script failed."
+    exit 1
+} else {
+    "$($ARCH)" | Out-File -Encoding ASCII -NoNewline build.arch
+    Write-Host "Configure script completed. Enter '$EXECTOS_BINARY_DIR' directory and execute 'xbuild' to build ExectOS."
+}

configure.sh

@@ -1,4 +1,10 @@
 #!/bin/bash
+# PROJECT:     ExectOS
+# LICENSE:     See the COPYING.md in the top level directory
+# FILE:        configure.sh
+# DESCRIPTION: Project configuration script for preparing the build environment
+# DEVELOPERS:  Rafal Kupiec <belliash@codingworkshop.eu.org>
+
 
 # Check XTCHain
 if [ "x${XTCVER}" = "x" ]; then
@@ -16,13 +22,12 @@ fi
 
 # Set variables
 EXECTOS_SOURCE_DIR=$(cd `dirname ${0}` && pwd)
-EXECTOS_BINARY_DIR=build-${ARCH}-xtchain
+EXECTOS_BINARY_DIR=build-${ARCH}-${BUILD_TYPE,,}
 
 # Create directories if needed
 if [ "${EXECTOS_SOURCE_DIR}" = "${PWD}" ]; then
 	echo Creating directories in ${EXECTOS_BINARY_DIR}
 	mkdir -p "${EXECTOS_BINARY_DIR}"
-	ln -sf ${EXECTOS_BINARY_DIR} build
 	cd "${EXECTOS_BINARY_DIR}"
 fi
 

sdk/cmake/baseaddress/amd64.cmake

@@ -0,0 +1,2 @@
+# Set base addresses for all modules
+set(BASEADDRESS_XTOSKRNL 0x0000000140000000)

sdk/cmake/baseaddress/i686.cmake

@@ -1,3 +1,2 @@
 # Set base addresses for all modules
-set(BASEADDRESS_XTLDR    0x00010000)
 set(BASEADDRESS_XTOSKRNL 0x00400000)

sdk/cmake/emulation.cmake

@@ -27,22 +27,22 @@ add_custom_target(bochsvm
 
 # This target starts up a QEMU+OVMF virtual machine using KVM accelerator
 add_custom_target(testkvm
-                  DEPENDS diskimg
+                  DEPENDS install
                   COMMAND ${QEMU_COMMAND} -name "ExectOS-${ARCH}-KVM" -machine type=q35,kernel_irqchip=on,accel=kvm,mem-merge=off,vmport=off -enable-kvm -cpu host,-hypervisor,+topoext
                                           -smp 2,sockets=1,cores=1,threads=2 -m 4G -overcommit mem-lock=off -rtc clock=host,base=localtime,driftfix=none
                                           -drive file=${EXECTOS_SOURCE_DIR}/sdk/firmware/ovmf_code_${ARCH}.fd,if=pflash,format=raw,unit=0,readonly=on
                                           -drive file=${EXECTOS_SOURCE_DIR}/sdk/firmware/ovmf_vars_${ARCH}.fd,if=pflash,format=raw,unit=1
-                                          -hda ${EXECTOS_BINARY_DIR}/output/disk.img
+                                          -hda fat:rw:${EXECTOS_BINARY_DIR}/output/binaries
                                           -boot menu=on -d int -M smm=off -no-reboot -no-shutdown -serial stdio
                   VERBATIM USES_TERMINAL)
 
 # This target starts up a QEMU+OVMF virtual machine using TCG accelerator
 add_custom_target(testtcg
-                  DEPENDS diskimg
+                  DEPENDS install
                   COMMAND ${QEMU_COMMAND} -name "ExectOS-${ARCH}-TCG" -machine type=q35,accel=tcg -cpu max,-hypervisor
                                           -smp 2,sockets=1,cores=1,threads=2 -m 4G -overcommit mem-lock=off -rtc clock=host,base=localtime,driftfix=none
                                           -drive file=${EXECTOS_SOURCE_DIR}/sdk/firmware/ovmf_code_${ARCH}.fd,if=pflash,format=raw,unit=0,readonly=on
                                           -drive file=${EXECTOS_SOURCE_DIR}/sdk/firmware/ovmf_vars_${ARCH}.fd,if=pflash,format=raw,unit=1
-                                          -hda ${EXECTOS_BINARY_DIR}/output/disk.img
+                                          -hda fat:rw:${EXECTOS_BINARY_DIR}/output/binaries
                                           -boot menu=on -d int -M smm=off -no-reboot -no-shutdown -serial stdio
                   VERBATIM USES_TERMINAL)

sdk/cmake/xtchain.cmake

@@ -11,11 +11,10 @@ endif()
 
 # Set build optimisation
 if(BUILD_TYPE STREQUAL "DEBUG")
-    add_compiler_ccxxflags("/Zi")
-    add_compiler_ccxxflags("-Ob0 -Od")
+    add_compiler_ccxxflags("/GS- /Zi /Ob0 /Od")
     add_linker_flags("/DEBUG /INCREMENTAL /OPT:NOREF /OPT:NOICF /PDBSOURCEPATH:build")
 else()
-    add_compiler_ccxxflags("-Ob2 -Oy")
+    add_compiler_ccxxflags("/GS- /Ob2 /Ot /Ox /Oy")
     add_linker_flags("/INCREMENTAL:NO /OPT:REF /OPT:ICF")
 endif()
 
@@ -48,6 +47,9 @@ add_compiler_ccxxflags("-nostdinc -Wno-char-subscripts -Wno-incompatible-library
 add_compiler_ccxxflags("-Wno-microsoft-enum-forward-reference -Wno-multichar -Wno-parentheses-equality -Wno-undefined-inline")
 add_compiler_ccxxflags("-Wno-gnu-folding-constant")
 
+# Disable compiler builtins
+add_compiler_ccxxflags("-fno-builtin")
+
 # Set debugging symbols output directory
 set(CMAKE_PDB_OUTPUT_DIRECTORY "${EXECTOS_BINARY_DIR}/output/symbols")
 

sdk/xtdk/amd64/arfuncs.h

@@ -24,6 +24,10 @@ XTCDECL
 BOOLEAN
 ArCpuId(IN OUT PCPUID_REGISTERS Registers);
 
+XTCDECL
+VOID
+ArEnableExtendedPhysicalAddressing(IN ULONG_PTR PageMap);
+
 XTCDECL
 VOID
 ArHalt(VOID);

sdk/xtdk/amd64/artypes.h

@@ -39,13 +39,22 @@
 #define CR4_PCE                                         0x00000100
 #define CR4_FXSR                                        0x00000200
 #define CR4_XMMEXCPT                                    0x00000400
+#define CR4_UMIP                                        0x00000800
 #define CR4_LA57                                        0x00001000
-#define CR4_RESERVED1                                   0x00001800
 #define CR4_VMXE                                        0x00002000
 #define CR4_SMXE                                        0x00004000
-#define CR4_RESERVED2                                   0x00018000
-#define CR4_XSAVE                                       0x00020000
-#define CR4_RESERVED3                                   0xFFFC0000
+#define CR4_FSGSBASE                                    0x00010000
+#define CR4_PCIDE                                       0x00020000
+#define CR4_XSAVE                                       0x00040000
+#define CR4_KL                                          0x00080000
+#define CR4_SMEP                                        0x00100000
+#define CR4_SMAP                                        0x00200000
+#define CR4_PKE                                         0x00400000
+#define CR4_CET                                         0x00800000
+#define CR4_PKS                                         0x01000000
+#define CR4_UINTR                                       0x02000000
+#define CR4_LASS                                        0x08000000
+#define CR4_LAM_SUP                                     0x10000000
 
 /* Descriptors size */
 #define GDT_ENTRIES                                     128
@@ -84,6 +93,7 @@
 #define X86_MSR_FSBASE                                  0xC0000100
 #define X86_MSR_GSBASE                                  0xC0000101
 #define X86_MSR_KERNEL_GSBASE                           0xC0000102
+#define X86_MSR_TSC_AUX                                 0xC0000103
 
 /* Processor features in the EFER MSR */
 #define X86_MSR_EFER_SCE                                (1 <<  0)
@@ -91,107 +101,289 @@
 #define X86_MSR_EFER_LMA                                (1 << 10)
 #define X86_MSR_EFER_NXE                                (1 << 11)
 #define X86_MSR_EFER_SVME                               (1 << 12)
+#define X86_EFER_LMSLE                                  (1 << 13)
+#define X86_EFER_FFXSR                                  (1 << 14)
+#define X86_EFER_TCE                                    (1 << 15)
+#define X86_EFER_AUTOIBRS                               (1 << 21)
 
 /* X86 EFLAG bit masks definitions */
-#define X86_EFLAGS_CF_MASK                              0x00000001
-#define X86_EFLAGS_PF_MASK                              0x00000004
-#define X86_EFALGS_AF_MASK                              0x00000010
-#define X86_EFLAGS_ZF_MASK                              0x00000040
-#define X86_EFLAGS_SF_MASK                              0x00000080
-#define X86_EFLAGS_TF_MASK                              0x00000100
-#define X86_EFLAGS_IF_MASK                              0x00000200
-#define X86_EFLAGS_DF_MASK                              0x00000400
-#define X86_EFLAGS_OF_MASK                              0x00000800
-#define X86_EFLAGS_IOPL_MASK                            0x00003000
-#define X86_EFLAGS_NT_MASK                              0x00004000
-#define X86_EFLAGS_RF_MASK                              0x00010000
-#define X86_EFLAGS_VM_MASK                              0x00020000
-#define X86_EFLAGS_AC_MASK                              0x00040000
-#define X86_EFLAGS_VIF_MASK                             0x00080000
-#define X86_EFLAGS_VIP_MASK                             0x00100000
-#define X86_EFLAGS_ID_MASK                              0x00200000
+#define X86_EFLAGS_NF_MASK                              0x00000000 /* None */
+#define X86_EFLAGS_CF_MASK                              0x00000001 /* Carry */
+#define X86_EFLAGS_PF_MASK                              0x00000004 /* Parity */
+#define X86_EFALGS_AF_MASK                              0x00000010 /* Aux Carry */
+#define X86_EFLAGS_ZF_MASK                              0x00000040 /* Zero */
+#define X86_EFLAGS_SF_MASK                              0x00000080 /* Sign */
+#define X86_EFLAGS_TF_MASK                              0x00000100 /* Trap */
+#define X86_EFLAGS_IF_MASK                              0x00000200 /* Interrupt */
+#define X86_EFLAGS_DF_MASK                              0x00000400 /* Direction */
+#define X86_EFLAGS_OF_MASK                              0x00000800 /* Overflow */
+#define X86_EFLAGS_IOPL_MASK                            0x00003000 /* I/O Privilege */
+#define X86_EFLAGS_NT_MASK                              0x00004000 /* Nested Task */
+#define X86_EFLAGS_SIGN_MASK                            0x00008000 /* Sign */
+#define X86_EFLAGS_RF_MASK                              0x00010000 /* Resume */
+#define X86_EFLAGS_V86_MASK                             0x00020000 /* Virtual 8086 */
+#define X86_EFLAGS_AC_MASK                              0x00040000 /* Alignment Check */
+#define X86_EFLAGS_VIF_MASK                             0x00080000 /* Virtual Interrupt */
+#define X86_EFLAGS_VIP_MASK                             0x00100000 /* Virtual Interrupt Pending */
+#define X86_EFLAGS_ID_MASK                              0x00200000 /* Identification */
 
 /* CPU vendor enumeration list */
 typedef enum _CPU_VENDOR
 {
     CPU_VENDOR_AMD = 0x68747541,
-    CPU_VENDOR_INTEL = 0x756e6547,
+    CPU_VENDOR_INTEL = 0x756E6547,
     CPU_VENDOR_UNKNOWN = 0xFFFFFFFF
 } CPU_VENDOR, *PCPU_VENDOR;
 
-/* CPUID features enumeration list */
-typedef enum _CPUID_FEATURES
+/* CPUID extended features (0x80000001) enumeration list */
+typedef enum _CPUID_FEATURES_EXTENDED
 {
-    CPUID_FEATURES_ECX_SSE3         = 1 << 0,
-    CPUID_FEATURES_ECX_PCLMUL       = 1 << 1,
-    CPUID_FEATURES_ECX_DTES64       = 1 << 2,
-    CPUID_FEATURES_ECX_MONITOR      = 1 << 3,
-    CPUID_FEATURES_ECX_DS_CPL       = 1 << 4,
-    CPUID_FEATURES_ECX_VMX          = 1 << 5,
-    CPUID_FEATURES_ECX_SMX          = 1 << 6,
-    CPUID_FEATURES_ECX_EST          = 1 << 7,
-    CPUID_FEATURES_ECX_TM2          = 1 << 8,
-    CPUID_FEATURES_ECX_SSSE3        = 1 << 9,
-    CPUID_FEATURES_ECX_CID          = 1 << 10,
-    CPUID_FEATURES_ECX_SDBG         = 1 << 11,
-    CPUID_FEATURES_ECX_FMA          = 1 << 12,
-    CPUID_FEATURES_ECX_CX16         = 1 << 13,
-    CPUID_FEATURES_ECX_XTPR         = 1 << 14,
-    CPUID_FEATURES_ECX_PDCM         = 1 << 15,
-    CPUID_FEATURES_ECX_PCID         = 1 << 17,
-    CPUID_FEATURES_ECX_DCA          = 1 << 18,
-    CPUID_FEATURES_ECX_SSE4_1       = 1 << 19,
-    CPUID_FEATURES_ECX_SSE4_2       = 1 << 20,
-    CPUID_FEATURES_ECX_X2APIC       = 1 << 21,
-    CPUID_FEATURES_ECX_MOVBE        = 1 << 22,
-    CPUID_FEATURES_ECX_POPCNT       = 1 << 23,
-    CPUID_FEATURES_ECX_TSC          = 1 << 24,
-    CPUID_FEATURES_ECX_AES          = 1 << 25,
-    CPUID_FEATURES_ECX_XSAVE        = 1 << 26,
-    CPUID_FEATURES_ECX_OSXSAVE      = 1 << 27,
-    CPUID_FEATURES_ECX_AVX          = 1 << 28,
-    CPUID_FEATURES_ECX_F16C         = 1 << 29,
-    CPUID_FEATURES_ECX_RDRAND       = 1 << 30,
-    CPUID_FEATURES_ECX_HYPERVISOR   = 1 << 31,
-    CPUID_FEATURES_EDX_FPU          = 1 << 0,
-    CPUID_FEATURES_EDX_VME          = 1 << 1,
-    CPUID_FEATURES_EDX_DE           = 1 << 2,
-    CPUID_FEATURES_EDX_PSE          = 1 << 3,
-    CPUID_FEATURES_EDX_TSC          = 1 << 4,
-    CPUID_FEATURES_EDX_MSR          = 1 << 5,
-    CPUID_FEATURES_EDX_PAE          = 1 << 6,
-    CPUID_FEATURES_EDX_MCE          = 1 << 7,
-    CPUID_FEATURES_EDX_CX8          = 1 << 8,
-    CPUID_FEATURES_EDX_APIC         = 1 << 9,
-    CPUID_FEATURES_EDX_SEP          = 1 << 11,
-    CPUID_FEATURES_EDX_MTRR         = 1 << 12,
-    CPUID_FEATURES_EDX_PGE          = 1 << 13,
-    CPUID_FEATURES_EDX_MCA          = 1 << 14,
-    CPUID_FEATURES_EDX_CMOV         = 1 << 15,
-    CPUID_FEATURES_EDX_PAT          = 1 << 16,
-    CPUID_FEATURES_EDX_PSE36        = 1 << 17,
-    CPUID_FEATURES_EDX_PSN          = 1 << 18,
-    CPUID_FEATURES_EDX_CLFLUSH      = 1 << 19,
-    CPUID_FEATURES_EDX_DS           = 1 << 21,
-    CPUID_FEATURES_EDX_ACPI         = 1 << 22,
-    CPUID_FEATURES_EDX_MMX          = 1 << 23,
-    CPUID_FEATURES_EDX_FXSR         = 1 << 24,
-    CPUID_FEATURES_EDX_SSE          = 1 << 25,
-    CPUID_FEATURES_EDX_SSE2         = 1 << 26,
-    CPUID_FEATURES_EDX_SS           = 1 << 27,
-    CPUID_FEATURES_EDX_HTT          = 1 << 28,
-    CPUID_FEATURES_EDX_TM           = 1 << 29,
-    CPUID_FEATURES_EDX_IA64         = 1 << 30,
-    CPUID_FEATURES_EDX_PBE          = 1 << 31
-} CPUID_FEATURES, *PCPUID_FEATURES;
+    CPUID_FEATURES_ECX_LAHF_SAHF              = 1 << 0,
+    CPUID_FEATURES_ECX_CMP_LEGACY             = 1 << 1,
+    CPUID_FEATURES_ECX_SVM                    = 1 << 2,
+    CPUID_FEATURES_ECX_EXT_APIC_SPACE         = 1 << 3,
+    CPUID_FEATURES_ECX_ALT_MOV_CR8            = 1 << 4,
+    CPUID_FEATURES_ECX_LZCNT                  = 1 << 5,
+    CPUID_FEATURES_ECX_SSE4A                  = 1 << 6,
+    CPUID_FEATURES_ECX_MISALIGNED_SSE         = 1 << 7,
+    CPUID_FEATURES_ECX_PREFETCHW              = 1 << 8,
+    CPUID_FEATURES_ECX_OSVW                   = 1 << 9,
+    CPUID_FEATURES_ECX_IBS                    = 1 << 10,
+    CPUID_FEATURES_ECX_XOP                    = 1 << 11,
+    CPUID_FEATURES_ECX_SKINIT                 = 1 << 12,
+    CPUID_FEATURES_ECX_WDT                    = 1 << 13,
+    CPUID_FEATURES_ECX_LWP                    = 1 << 15,
+    CPUID_FEATURES_ECX_FMA4                   = 1 << 16,
+    CPUID_FEATURES_ECX_TCE                    = 1 << 17,
+    CPUID_FEATURES_ECX_NODEID                 = 1 << 19,
+    CPUID_FEATURES_ECX_TBM                    = 1 << 21,
+    CPUID_FEATURES_ECX_TOPOLOGY_EXTENSIONS    = 1 << 22,
+    CPUID_FEATURES_ECX_PERFCTR_EXT_CORE       = 1 << 23,
+    CPUID_FEATURES_ECX_PERFCTR_EXT_NB         = 1 << 24,
+    CPUID_FEATURES_ECX_DATA_BREAKPOINT_EXT    = 1 << 26,
+    CPUID_FEATURES_ECX_PERF_TSC               = 1 << 27,
+    CPUID_FEATURES_ECX_PERFCTR_EXT_L2I        = 1 << 28,
+    CPUID_FEATURES_ECX_MONITORX_MWAITX        = 1 << 29,
+    CPUID_FEATURES_ECX_CODEBP_ADDRMASK_EXT    = 1 << 30,
+    CPUID_FEATURES_EDX_SYSCALL_SYSRET         = 1 << 11,
+    CPUID_FEATURES_EDX_NX                     = 1 << 20,
+    CPUID_FEATURES_EDX_AMD_MMX_EXT            = 1 << 22,
+    CPUID_FEATURES_EDX_FFXSR                  = 1 << 25,
+    CPUID_FEATURES_EDX_1G_PAGES               = 1 << 26,
+    CPUID_FEATURES_EDX_RDTSCP                 = 1 << 27,
+    CPUID_FEATURES_EDX_LONG_MODE              = 1 << 29,
+    CPUID_FEATURES_EDX_3DNOW_EXT              = 1 << 30,
+    CPUID_FEATURES_EDX_3DNOW                  = 1 << 31
+} CPUID_FEATURES_EXTENDED, *PCPUID_FEATURES_EXTENDED;
+
+/* CPUID STD1 features (0x00000001) enumeration list */
+typedef enum _CPUID_FEATURES_STANDARD1
+{
+    CPUID_FEATURES_ECX_SSE3                   = 1 << 0,
+    CPUID_FEATURES_ECX_PCLMUL                 = 1 << 1,
+    CPUID_FEATURES_ECX_DTES64                 = 1 << 2,
+    CPUID_FEATURES_ECX_MONITOR                = 1 << 3,
+    CPUID_FEATURES_ECX_DS_CPL                 = 1 << 4,
+    CPUID_FEATURES_ECX_VMX                    = 1 << 5,
+    CPUID_FEATURES_ECX_SMX                    = 1 << 6,
+    CPUID_FEATURES_ECX_EST                    = 1 << 7,
+    CPUID_FEATURES_ECX_TM2                    = 1 << 8,
+    CPUID_FEATURES_ECX_SSSE3                  = 1 << 9,
+    CPUID_FEATURES_ECX_CID                    = 1 << 10,
+    CPUID_FEATURES_ECX_SDBG                   = 1 << 11,
+    CPUID_FEATURES_ECX_FMA                    = 1 << 12,
+    CPUID_FEATURES_ECX_CX16                   = 1 << 13,
+    CPUID_FEATURES_ECX_XTPR                   = 1 << 14,
+    CPUID_FEATURES_ECX_PDCM                   = 1 << 15,
+    CPUID_FEATURES_ECX_PCID                   = 1 << 17,
+    CPUID_FEATURES_ECX_DCA                    = 1 << 18,
+    CPUID_FEATURES_ECX_SSE4_1                 = 1 << 19,
+    CPUID_FEATURES_ECX_SSE4_2                 = 1 << 20,
+    CPUID_FEATURES_ECX_X2APIC                 = 1 << 21,
+    CPUID_FEATURES_ECX_MOVBE                  = 1 << 22,
+    CPUID_FEATURES_ECX_POPCNT                 = 1 << 23,
+    CPUID_FEATURES_ECX_TSC                    = 1 << 24,
+    CPUID_FEATURES_ECX_AES                    = 1 << 25,
+    CPUID_FEATURES_ECX_XSAVE                  = 1 << 26,
+    CPUID_FEATURES_ECX_OSXSAVE                = 1 << 27,
+    CPUID_FEATURES_ECX_AVX                    = 1 << 28,
+    CPUID_FEATURES_ECX_F16C                   = 1 << 29,
+    CPUID_FEATURES_ECX_RDRAND                 = 1 << 30,
+    CPUID_FEATURES_ECX_HYPERVISOR             = 1 << 31,
+    CPUID_FEATURES_EDX_FPU                    = 1 << 0,
+    CPUID_FEATURES_EDX_VME                    = 1 << 1,
+    CPUID_FEATURES_EDX_DE                     = 1 << 2,
+    CPUID_FEATURES_EDX_PSE                    = 1 << 3,
+    CPUID_FEATURES_EDX_TSC                    = 1 << 4,
+    CPUID_FEATURES_EDX_MSR                    = 1 << 5,
+    CPUID_FEATURES_EDX_PAE                    = 1 << 6,
+    CPUID_FEATURES_EDX_MCE                    = 1 << 7,
+    CPUID_FEATURES_EDX_CX8                    = 1 << 8,
+    CPUID_FEATURES_EDX_APIC                   = 1 << 9,
+    CPUID_FEATURES_EDX_SEP                    = 1 << 11,
+    CPUID_FEATURES_EDX_MTRR                   = 1 << 12,
+    CPUID_FEATURES_EDX_PGE                    = 1 << 13,
+    CPUID_FEATURES_EDX_MCA                    = 1 << 14,
+    CPUID_FEATURES_EDX_CMOV                   = 1 << 15,
+    CPUID_FEATURES_EDX_PAT                    = 1 << 16,
+    CPUID_FEATURES_EDX_PSE36                  = 1 << 17,
+    CPUID_FEATURES_EDX_PSN                    = 1 << 18,
+    CPUID_FEATURES_EDX_CLFLUSH                = 1 << 19,
+    CPUID_FEATURES_EDX_DS                     = 1 << 21,
+    CPUID_FEATURES_EDX_ACPI                   = 1 << 22,
+    CPUID_FEATURES_EDX_MMX                    = 1 << 23,
+    CPUID_FEATURES_EDX_FXSR                   = 1 << 24,
+    CPUID_FEATURES_EDX_SSE                    = 1 << 25,
+    CPUID_FEATURES_EDX_SSE2                   = 1 << 26,
+    CPUID_FEATURES_EDX_SS                     = 1 << 27,
+    CPUID_FEATURES_EDX_HTT                    = 1 << 28,
+    CPUID_FEATURES_EDX_TM                     = 1 << 29,
+    CPUID_FEATURES_EDX_IA64                   = 1 << 30,
+    CPUID_FEATURES_EDX_PBE                    = 1 << 31
+} CPUID_FEATURES_STANDARD1, *PCPUID_FEATURES_STANDARD1;
+
+/* CPUID STD7 features (0x00000007, subleaf 0) enumeration list */
+typedef enum _CPUID_FEATURES_STANDARD7_LEAF0
+{
+    CPUID_FEATURES_EBX_FSGSBASE               = 1 << 0,
+    CPUID_FEATURES_EBX_TSC_ADJUST             = 1 << 1,
+    CPUID_FEATURES_EBX_SGX                    = 1 << 2,
+    CPUID_FEATURES_EBX_BMI1                   = 1 << 3,
+    CPUID_FEATURES_EBX_HLE                    = 1 << 4,
+    CPUID_FEATURES_EBX_AVX2                   = 1 << 5,
+    CPUID_FEATURES_EBX_FDP_DEPRECATION        = 1 << 6,
+    CPUID_FEATURES_EBX_SMEP                   = 1 << 7,
+    CPUID_FEATURES_EBX_BMI2                   = 1 << 8,
+    CPUID_FEATURES_EBX_ERMS                   = 1 << 9,
+    CPUID_FEATURES_EBX_INVPCID                = 1 << 10,
+    CPUID_FEATURES_EBX_RTM                    = 1 << 11,
+    CPUID_FEATURES_EBX_QOS_MONITORING         = 1 << 12,
+    CPUID_FEATURES_EBX_DEPRECATE_FCS_FDS      = 1 << 13,
+    CPUID_FEATURES_EBX_MPX                    = 1 << 14,
+    CPUID_FEATURES_EBX_QOS_ENFORCEMENT        = 1 << 15,
+    CPUID_FEATURES_EBX_AVX512F                = 1 << 16,
+    CPUID_FEATURES_EBX_AVX512DQ               = 1 << 17,
+    CPUID_FEATURES_EBX_RDSEED                 = 1 << 18,
+    CPUID_FEATURES_EBX_ADX                    = 1 << 19,
+    CPUID_FEATURES_EBX_SMAP                   = 1 << 20,
+    CPUID_FEATURES_EBX_AVX512IFMA52           = 1 << 21,
+    CPUID_FEATURES_EBX_CLFLUSHOPT             = 1 << 23,
+    CPUID_FEATURES_EBX_CLWB                   = 1 << 24,
+    CPUID_FEATURES_EBX_PROCESSOR_TRACE        = 1 << 25,
+    CPUID_FEATURES_EBX_AVX512PF               = 1 << 26,
+    CPUID_FEATURES_EBX_AVX512ER               = 1 << 27,
+    CPUID_FEATURES_EBX_AVX512CD               = 1 << 28,
+    CPUID_FEATURES_EBX_SHA                    = 1 << 29,
+    CPUID_FEATURES_EBX_AVX512BW               = 1 << 30,
+    CPUID_FEATURES_EBX_AVX512VL               = 1 << 31,
+    CPUID_FEATURES_ECX_PREFETCHWT1            = 1 << 0,
+    CPUID_FEATURES_ECX_AVX512_VBMI            = 1 << 1,
+    CPUID_FEATURES_ECX_UMIP                   = 1 << 2,
+    CPUID_FEATURES_ECX_PKU                    = 1 << 3,
+    CPUID_FEATURES_ECX_OSPKE                  = 1 << 4,
+    CPUID_FEATURES_ECX_WAITPKG                = 1 << 5,
+    CPUID_FEATURES_ECX_AVX512_VBMI2           = 1 << 6,
+    CPUID_FEATURES_ECX_CET_SS                 = 1 << 7,
+    CPUID_FEATURES_ECX_GFNI                   = 1 << 8,
+    CPUID_FEATURES_ECX_VAES                   = 1 << 9,
+    CPUID_FEATURES_ECX_VPCLMULQDQ             = 1 << 10,
+    CPUID_FEATURES_ECX_AVX512_VNNI            = 1 << 11,
+    CPUID_FEATURES_ECX_AVX512_BITALG          = 1 << 12,
+    CPUID_FEATURES_ECX_TME                    = 1 << 13,
+    CPUID_FEATURES_ECX_AVX512_VPOPCNTDQ       = 1 << 14,
+    CPUID_FEATURES_ECX_LA57                   = 1 << 16,
+    CPUID_FEATURES_ECX_RDPID                  = 1 << 22,
+    CPUID_FEATURES_ECX_KEYLOCKER              = 1 << 23,
+    CPUID_FEATURES_ECX_BUS_LOCK_DETECT        = 1 << 24,
+    CPUID_FEATURES_ECX_CLDEMOTE               = 1 << 25,
+    CPUID_FEATURES_ECX_MOVDIRI                = 1 << 27,
+    CPUID_FEATURES_ECX_MOVDIR64B              = 1 << 28,
+    CPUID_FEATURES_ECX_ENQCMD                 = 1 << 29,
+    CPUID_FEATURES_ECX_SGX_LAUNCH_CONFIG      = 1 << 30,
+    CPUID_FEATURES_ECX_PKS                    = 1 << 31,
+    CPUID_FEATURES_EDX_SGX_KEYS               = 1 << 1,
+    CPUID_FEATURES_EDX_AVX512_4VNNIW          = 1 << 2,
+    CPUID_FEATURES_EDX_AVX512_4FMAPS          = 1 << 3,
+    CPUID_FEATURES_EDX_FAST_SHORT_REP_MOV     = 1 << 4,
+    CPUID_FEATURES_EDX_UINTR                  = 1 << 5,
+    CPUID_FEATURES_EDX_AVX512_VPINTERSECT     = 1 << 8,
+    CPUID_FEATURES_EDX_SRBDS_CTRL             = 1 << 9,
+    CPUID_FEATURES_EDX_MD_CLEAR               = 1 << 10,
+    CPUID_FEATURES_EDX_RTM_ALWAYS_ABORT       = 1 << 11,
+    CPUID_FEATURES_EDX_RTM_FORCE_ABORT        = 1 << 13,
+    CPUID_FEATURES_EDX_SERIALIZE              = 1 << 14,
+    CPUID_FEATURES_EDX_HYBRID                 = 1 << 15,
+    CPUID_FEATURES_EDX_TSXLDTRK               = 1 << 16,
+    CPUID_FEATURES_EDX_PCONFIG                = 1 << 18,
+    CPUID_FEATURES_EDX_ARCH_LBR               = 1 << 19,
+    CPUID_FEATURES_EDX_CET_IBT                = 1 << 20,
+    CPUID_FEATURES_EDX_AMX_BF16               = 1 << 22,
+    CPUID_FEATURES_EDX_AVX512_FP16            = 1 << 23,
+    CPUID_FEATURES_EDX_AMX_TILE               = 1 << 24,
+    CPUID_FEATURES_EDX_AMX_INT8               = 1 << 25,
+    CPUID_FEATURES_EDX_SCA_IBRS_IBPB          = 1 << 26,
+    CPUID_FEATURES_EDX_SCA_STIBP              = 1 << 27,
+    CPUID_FEATURES_EDX_L1D_FLUSH              = 1 << 28,
+    CPUID_FEATURES_EDX_ARCH_CAPABILITIES_MSR  = 1 << 29,
+    CPUID_FEATURES_EDX_CORE_CAPABILITIES_MSR  = 1 << 30,
+    CPUID_FEATURES_EDX_SCA_SSBD               = 1 << 31
+} CPUID_FEATURES_STANDARD7_LEAF0, *PCPUID_FEATURES_STANDARD7_LEAF0;
+
+/* CPUID STD7 features (0x00000007, subleaf 1) enumeration list */
+typedef enum _CPUID_FEATURES_STANDARD7_LEAF1
+{
+    CPUID_FEATURES_EAX_SHA512                 = 1 << 0,
+    CPUID_FEATURES_EAX_SM3                    = 1 << 1,
+    CPUID_FEATURES_EAX_SM4                    = 1 << 2,
+    CPUID_FEATURES_EAX_RAO_INT                = 1 << 3,
+    CPUID_FEATURES_EAX_AVX_VNNI               = 1 << 4,
+    CPUID_FEATURES_EAX_AVX512_BF16            = 1 << 5,
+    CPUID_FEATURES_EAX_LASS                   = 1 << 6,
+    CPUID_FEATURES_EAX_CMPCCXADD              = 1 << 7,
+    CPUID_FEATURES_EAX_ARCH_PERFMON           = 1 << 8,
+    CPUID_FEATURES_EAX_FAST_ZEROLEN_REP_MOVSB = 1 << 10,
+    CPUID_FEATURES_EAX_FAST_ZEROLEN_REP_STOSB = 1 << 11,
+    CPUID_FEATURES_EAX_FAST_ZEROLEN_REP_CMPSB = 1 << 12,
+    CPUID_FEATURES_EAX_FRED                   = 1 << 17,
+    CPUID_FEATURES_EAX_LKGS                   = 1 << 18,
+    CPUID_FEATURES_EAX_WRMSRNS                = 1 << 19,
+    CPUID_FEATURES_EAX_NMI_SOURCE_REPORTING   = 1 << 20,
+    CPUID_FEATURES_EAX_AMX_FP16               = 1 << 21,
+    CPUID_FEATURES_EAX_HRESET                 = 1 << 22,
+    CPUID_FEATURES_EAX_AVX_IFMA               = 1 << 23,
+    CPUID_FEATURES_EAX_LAM                    = 1 << 26,
+    CPUID_FEATURES_EAX_MSRLIST                = 1 << 27,
+    CPUID_FEATURES_EAX_INVD_DISABLE           = 1 << 30,
+    CPUID_FEATURES_EAX_MOVRS                  = 1 << 31,
+    CPUID_FEATURES_EBX_PPIN                   = 1 << 0,
+    CPUID_FEATURES_EBX_TSE                    = 1 << 1,
+    CPUID_FEATURES_EBX_CPUIDMAXVAL_LIM_RMV    = 1 << 3,
+    CPUID_FEATURES_ECX_MSR_IMM                = 1 << 5,
+    CPUID_FEATURES_EDX_AVX_VNNI_INT8          = 1 << 4,
+    CPUID_FEATURES_EDX_AVX_NE_CONVERT         = 1 << 5,
+    CPUID_FEATURES_EDX_AMX_COMPLEX            = 1 << 8,
+    CPUID_FEATURES_EDX_AVX_VNNI_INT16         = 1 << 10,
+    CPUID_FEATURES_EDX_USER_TIMER             = 1 << 13,
+    CPUID_FEATURES_EDX_PREFETCHI              = 1 << 14,
+    CPUID_FEATURES_EDX_USER_MSR               = 1 << 15,
+    CPUID_FEATURES_EDX_UIRET_UIF              = 1 << 17,
+    CPUID_FEATURES_EDX_CET_SSS                = 1 << 18,
+    CPUID_FEATURES_EDX_AVX10                  = 1 << 19,
+    CPUID_FEATURES_EDX_APX                    = 1 << 21,
+    CPUID_FEATURES_EDX_MWAIT_AND_LEAF5        = 1 << 23
+} CPUID_FEATURES_STANDARD7_LEAF1, *PCPUID_FEATURES_STANDARD7_LEAF1;
 
 /* CPUID requests */
 typedef enum _CPUID_REQUESTS
 {
     CPUID_GET_VENDOR_STRING,
-    CPUID_GET_CPU_FEATURES,
-    CPUID_GET_TLB,
-    CPUID_GET_SERIAL
+    CPUID_GET_STANDARD1_FEATURES,
+    CPUID_GET_TLB_CACHE,
+    CPUID_GET_SERIAL,
+    CPUID_GET_CACHE_TOPOLOGY,
+    CPUID_GET_MONITOR_MWAIT,
+    CPUID_GET_POWER_MANAGEMENT,
+    CPUID_GET_STANDARD7_FEATURES
 } CPUID_REQUESTS, *PCPUID_REQUESTS;
 
 /* Processor identification information */

sdk/xtdk/amd64/hltypes.h

@@ -42,37 +42,143 @@
 #define APIC_DF_FLAT                                    0xFFFFFFFF
 #define APIC_DF_CLUSTER                                 0x0FFFFFFF
 
-/* APIC delivery modes */
-#define APIC_DM_FIXED                                   0
-#define APIC_DM_LOWPRIO                                 1
-#define APIC_DM_SMI                                     2
-#define APIC_DM_REMOTE                                  3
-#define APIC_DM_NMI                                     4
-#define APIC_DM_INIT                                    5
-#define APIC_DM_STARTUP                                 6
-#define APIC_DM_EXTINT                                  7
-
 /* APIC trigger modes */
 #define APIC_TGM_EDGE                                   0
 #define APIC_TGM_LEVEL                                  1
 
+/* APIC LDR (Logical Destination Register) shifts */
+#define APIC_X2APIC_LDR_SHIFT                           16
+#define APIC_XAPIC_LDR_SHIFT                            24
+
+/* Maximum number of I/O APICs */
+#define APIC_MAX_IOAPICS                                64
+
 /* 8259/ISP PIC ports definitions */
 #define PIC1_CONTROL_PORT                               0x20
 #define PIC1_DATA_PORT                                  0x21
-#define PIC1_ELCR_PORT                                  0x04D0
 #define PIC2_CONTROL_PORT                               0xA0
 #define PIC2_DATA_PORT                                  0xA1
-#define PIC2_ELCR_PORT                                  0x04D1
 
 /* PIC vector definitions */
 #define PIC1_VECTOR_SPURIOUS                            0x37
 
-/* Serial port I/O addresses */
-#define COMPORT_ADDRESSES                               {0x000, 0x3F8, 0x2F8, 0x3E8, 0x2E8, 0x5F8, 0x4F8, 0x5E8, 0x4E8}
+/* Serial ports information */
+#define COMPORT_ADDRESS                                 {0x3F8, 0x2F8, 0x3E8, 0x2E8, 0x5F8, 0x4F8, 0x5E8, 0x4E8}
+#define COMPORT_COUNT                                   8
 
 /* Initial stall factor */
 #define INITIAL_STALL_FACTOR                            100
 
+/* APIC delivery mode enumeration list */
+typedef enum _APIC_DM
+{
+    APIC_DM_FIXED,
+    APIC_DM_LOWPRIO,
+    APIC_DM_SMI,
+    APIC_DM_REMOTE,
+    APIC_DM_NMI,
+    APIC_DM_INIT,
+    APIC_DM_STARTUP,
+    APIC_DM_EXTINT,
+} APIC_DM, *PAPIC_DM;
+
+/* APIC destination short-hand enumeration list */
+typedef enum _APIC_DSH
+{
+    APIC_DSH_Destination,
+    APIC_DSH_Self,
+    APIC_DSH_AllIncludingSelf,
+    APIC_DSH_AllExclusingSelf
+} APIC_DSH, *PAPIC_DSH;
+
+/* APIC mode list */
+typedef enum _APIC_MODE
+{
+    APIC_MODE_COMPAT,
+    APIC_MODE_X2APIC
+} APIC_MODE, *PAPIC_MODE;
+
+/* APIC Register Address Map */
+typedef enum _APIC_REGISTER
+{
+    APIC_ID       = 0x02, /* APIC ID Register */
+    APIC_VER      = 0x03, /* APIC Version Register */
+    APIC_TPR      = 0x08, /* Task Priority Register */
+    APIC_APR      = 0x09, /* Arbitration Priority Register */
+    APIC_PPR      = 0x0A, /* Processor Priority Register (R) */
+    APIC_EOI      = 0x0B, /* EOI Register */
+    APIC_RRR      = 0x0C, /* Remote Read Register */
+    APIC_LDR      = 0x0D, /* Logical Destination Register */
+    APIC_DFR      = 0x0E, /* Destination Format Register (not available in extended mode) */
+    APIC_SIVR     = 0x0F, /* Spurious Interrupt Vector Register */
+    APIC_ISR      = 0x10, /* Interrupt Service Register*/
+    APIC_TMR      = 0x18, /* Trigger Mode Register */
+    APIC_IRR      = 0x20, /* Interrupt Request Register */
+    APIC_ESR      = 0x28, /* Error Status Register */
+    APIC_ICR0     = 0x30, /* Interrupt Command Register */
+    APIC_ICR1     = 0x31, /* Interrupt Command Register (not available in extended mode) */
+    APIC_TMRLVTR  = 0x32, /* Timer Local Vector Table */
+    APIC_THRMLVTR = 0x33, /* Thermal Local Vector Table */
+    APIC_PCLVTR   = 0x34, /* Performance Counter Local Vector Table */
+    APIC_LINT0    = 0x35, /* LINT0 Local Vector Table */
+    APIC_LINT1    = 0x36, /* LINT1 Local Vector Table */
+    APIC_ERRLVTR  = 0x37, /* Error Local Vector Table */
+    APIC_TICR     = 0x38, /* Initial Count Register for Timer */
+    APIC_TCCR     = 0x39, /* Current Count Register for Timer */
+    APIC_TDCR     = 0x3E, /* Timer Divide Configuration Register */
+    APIC_EAFR     = 0x40, /* extended APIC Feature register */
+    APIC_EACR     = 0x41, /* Extended APIC Control Register */
+    APIC_SEOI     = 0x42, /* Specific End Of Interrupt Register */
+    APIC_EXT0LVTR = 0x50, /* Extended Interrupt 0 Local Vector Table */
+    APIC_EXT1LVTR = 0x51, /* Extended Interrupt 1 Local Vector Table */
+    APIC_EXT2LVTR = 0x52, /* Extended Interrupt 2 Local Vector Table */
+    APIC_EXT3LVTR = 0x53  /* Extended Interrupt 3 Local Vector Table */
+} APIC_REGISTER, *PAPIC_REGISTER;
+
+/* I8259 PIC interrupt mode enumeration list */
+typedef enum _PIC_I8259_ICW1_INTERRUPT_MODE
+{
+    EdgeTriggered,
+    LevelTriggered
+} PIC_I8259_ICW1_INTERRUPT_MODE, *PPIC_I8259_ICW1_INTERRUPT_MODE;
+
+/* I8259 PIC interval enumeration list */
+typedef enum _PIC_I8259_ICW1_INTERVAL
+{
+    Interval8,
+    Interval4
+} PIC_I8259_ICW1_INTERVAL, *PPIC_I8259_ICW1_INTERVAL;
+
+/* I8259 PIC operating mode enumeration list */
+typedef enum _PIC_I8259_ICW1_OPERATING_MODE
+{
+    Cascade,
+    Single
+} PIC_I8259_ICW1_OPERATING_MODE, *PPIC_I8259_ICW1_OPERATING_MODE;
+
+/* I8259 PIC buffered mode enumeration list */
+typedef enum _PIC_I8259_ICW4_BUFFERED_MODE
+{
+    NonBuffered,
+    NonBuffered2,
+    BufferedSlave,
+    BufferedMaster
+} PIC_I8259_ICW4_BUFFERED_MODE, *PPIC_I8259_ICW4_BUFFERED_MODE;
+
+/* I8259 PIC End Of Interrupt (EOI) mode enumeration list */
+typedef enum _PIC_I8259_ICW4_EOI_MODE
+{
+    NormalEoi,
+    AutomaticEoi
+} PIC_I8259_ICW4_EOI_MODE, *PPIC_I8259_ICW4_EOI_MODE;
+
+/* I8259 PIC system mode enumeration list */
+typedef enum _PIC_I8259_ICW4_SYSTEM_MODE
+{
+    Mcs8085Mode,
+    New8086Mode
+} PIC_I8259_ICW4_SYSTEM_MODE, *PPIC_I8259_ICW4_SYSTEM_MODE;
+
 /* APIC Base Register */
 typedef union _APIC_BASE_REGISTER
 {
@@ -89,6 +195,31 @@ typedef union _APIC_BASE_REGISTER
     };
 } APIC_BASE_REGISTER, *PAPIC_BASE_REGISTER;
 
+/* APIC Command Register */
+typedef union _APIC_COMMAND_REGISTER
+{
+    ULONGLONG LongLong;
+    struct
+    {
+        ULONG Long0;
+        ULONG Long1;
+    };
+    struct
+    {
+        ULONGLONG Vector:8;
+        ULONGLONG DeliveryMode:3;
+        ULONGLONG DestinationMode:1;
+        ULONGLONG DeliveryStatus:1;
+        ULONGLONG ReservedMBZ:1;
+        ULONGLONG Level:1;
+        ULONGLONG TriggerMode:1;
+        ULONGLONG RemoteReadStatus:2;
+        ULONGLONG DestinationShortHand:2;
+        ULONGLONG Reserved2MBZ:36;
+        ULONGLONG Destination:8;
+    };
+} APIC_COMMAND_REGISTER, *PAPIC_COMMAND_REGISTER;
+
 /* APIC Local Vector Table (LVT) Register */
 typedef union _APIC_LVT_REGISTER
 {
@@ -96,7 +227,7 @@ typedef union _APIC_LVT_REGISTER
     struct
     {
         ULONG Vector:8;
-        ULONG MessageType:3;
+        ULONG DeliveryMode:3;
         ULONG Reserved1:1;
         ULONG DeliveryStatus:1;
         ULONG Reserved2:1;
@@ -121,14 +252,69 @@ typedef union _APIC_SPURIOUS_REGISTER
     };
 } APIC_SPURIOUS_REGISTER, *PAPIC_SPURIOUS_REGISTER;
 
-/* Processor identity structure */
-typedef struct _HAL_PROCESSOR_IDENTITY
+/* I8259 PIC register structure */
+typedef union _PIC_I8259_ICW1
 {
-    UCHAR ProcessorId;
-    UCHAR LApicId;
-    BOOLEAN Bsp;
-    BOOLEAN Started;
-    PKPROCESSOR_BLOCK ProcessorBlock;
-} HAL_PROCESSOR_IDENTITY, *PHAL_PROCESSOR_IDENTITY;
+    struct
+    {
+        UCHAR NeedIcw4:1;
+        UCHAR OperatingMode:1;
+        UCHAR Interval:1;
+        UCHAR InterruptMode:1;
+        UCHAR Init:1;
+        UCHAR InterruptVectorAddress:3;
+    };
+    UCHAR Bits;
+} PIC_I8259_ICW1, *PPIC_I8259_ICW1;
+
+/* I8259 PIC register structure */
+typedef union _PIC_I8259_ICW2
+{
+    struct
+    {
+        UCHAR Sbz:3;
+        UCHAR InterruptVector:5;
+    };
+    UCHAR Bits;
+} PIC_I8259_ICW2, *PPIC_I8259_ICW2;
+
+/* I8259 PIC register structure */
+typedef union _PIC_I8259_ICW3
+{
+    union
+    {
+        struct
+        {
+            UCHAR SlaveIrq0:1;
+            UCHAR SlaveIrq1:1;
+            UCHAR SlaveIrq2:1;
+            UCHAR SlaveIrq3:1;
+            UCHAR SlaveIrq4:1;
+            UCHAR SlaveIrq5:1;
+            UCHAR SlaveIrq6:1;
+            UCHAR SlaveIrq7:1;
+        };
+        struct
+        {
+            UCHAR SlaveId:3;
+            UCHAR Reserved:5;
+        };
+    };
+    UCHAR Bits;
+} PIC_I8259_ICW3, *PPIC_I8259_ICW3;
+
+/* I8259 PIC register structure */
+typedef union _PIC_I8259_ICW4
+{
+    struct
+    {
+        UCHAR SystemMode:1;
+        UCHAR EoiMode:1;
+        UCHAR BufferedMode:2;
+        UCHAR SpecialFullyNestedMode:1;
+        UCHAR Reserved:3;
+    };
+    UCHAR Bits;
+} PIC_I8259_ICW4, *PPIC_I8259_ICW4;
 
 #endif /* __XTDK_AMD64_HLTYPES_H */

sdk/xtdk/amd64/ketypes.h

@@ -67,31 +67,6 @@
 #define AMD64_INTERRUPT_GATE                      0xE
 #define AMD64_TRAP_GATE                           0xF
 
-/* EFLAGS bits definitions */
-#define EFLAGS_NF_MASK                            0x00000000L /* None */
-#define EFLAGS_CF_MASK                            0x00000001L /* Carry */
-#define EFLAGS_PF_MASK                            0x00000004L /* Parity */
-#define EFLAGS_AF_MASK                            0x00000010L /* Aux Carry */
-#define EFLAGS_ZF_MASK                            0x00000040L /* Zero */
-#define EFLAGS_SF_MASK                            0x00000080L /* Sign */
-#define EFLAGS_TF                                 0x00000100L /* Trap */
-#define EFLAGS_INTERRUPT_MASK                     0x00000200L /* Interrupt */
-#define EFLAGS_DF_MASK                            0x00000400L /* Direction */
-#define EFLAGS_OF_MASK                            0x00000800L /* Overflow */
-#define EFLAGS_IOPL_MASK                          0x00003000L /* I/O Privilege */
-#define EFLAGS_NT                                 0x00004000L /* Nested Task */
-#define EFLAGS_SIGN_MASK                          0x00008000L /* Sign */
-#define EFLAGS_RF                                 0x00010000L /* Resume */
-#define EFLAGS_V86_MASK                           0x00020000L /* Virtual 8086 */
-#define EFLAGS_ALIGN_CHECK                        0x00040000L /* Alignment Check */
-#define EFLAGS_VIF                                0x00080000L /* Virtual Interrupt */
-#define EFLAGS_VIP                                0x00100000L /* Virtual Interrupt Pending */
-#define EFLAGS_ID_MASK                            0x00200000L /* Identification */
-
-/* EFLAGS sanitize masks */
-#define EFLAGS_KERNELMODE                         0x00210FD5L
-#define EFLAGS_USERMODE                           0x00010DD5L
-
 /* Context control flags */
 #define CONTEXT_ARCHITECTURE                      0x00100000
 #define CONTEXT_CONTROL                           (CONTEXT_ARCHITECTURE | 0x01)
@@ -142,6 +117,10 @@
 /* XTOS Kernel stack guard pages */
 #define KERNEL_STACK_GUARD_PAGES          1
 
+/* Processor structures size */
+#define KPROCESSOR_STRUCTURES_SIZE        ((2 * KERNEL_STACK_SIZE) + sizeof(ArInitialGdt) + sizeof(ArInitialTss) + \
+                                          sizeof(ArInitialProcessorBlock) + MM_PAGE_SIZE)
+
 /* Kernel frames */
 #define KEXCEPTION_FRAME_SIZE             sizeof(KEXCEPTION_FRAME)
 #define KSWITCH_FRAME_SIZE                sizeof(KSWITCH_FRAME)
@@ -314,7 +293,7 @@ typedef struct _KTSS
     ULONG64 Reserved1;
     USHORT Reserved2;
     USHORT IoMapBase;
-} PACK KTSS, *PKTSS;
+} PACKED KTSS, *PKTSS;
 
 /* Exception frame definition */
 typedef struct _KEXCEPTION_FRAME
@@ -491,7 +470,7 @@ typedef struct _KPROCESSOR_STATE
 typedef struct _KPROCESSOR_CONTROL_BLOCK
 {
     ULONG MxCsr;
-    UCHAR Number;
+    UCHAR CpuNumber;
     PKTHREAD CurrentThread;
     PKTHREAD IdleThread;
     PKTHREAD NextThread;
@@ -526,8 +505,13 @@ typedef struct _KPROCESSOR_BLOCK
     PKIDTENTRY IdtBase;
     KRUNLEVEL RunLevel;
     KPROCESSOR_CONTROL_BLOCK Prcb;
+    ULONG Irr;
+    ULONG IrrActive;
+    ULONG Idr;
     ULONG ContextSwitches;
+    KAFFINITY SetMember;
     ULONG StallScaleFactor;
+    UCHAR CpuNumber;
 } KPROCESSOR_BLOCK, *PKPROCESSOR_BLOCK;
 
 /* Thread Environment Block (TEB) structure definition */

sdk/xtdk/amd64/mmtypes.h

@@ -14,22 +14,35 @@
 
 
 /* Pages related definitions */
-#define MM_PAGE_MASK                               0xFFF
+#define MM_PAGE_MASK                               (MM_PAGE_SIZE - 1)
 #define MM_PAGE_SHIFT                              12L
 #define MM_PAGE_SIZE                               4096
 
-#define MM_PTE_BASE                                0xFFFFF68000000000UI64
-#define MM_PDE_BASE                                0xFFFFF6FB40000000UI64
-#define MM_PPE_BASE                                0xFFFFF6FB7DA00000UI64
-#define MM_PXE_BASE                                0xFFFFF6FB7DBED000UI64
+/* Page directory and page base addresses for 4-level paging */
+#define MM_PTE_BASE                                0xFFFFF68000000000ULL
+#define MM_PDE_BASE                                0xFFFFF6FB40000000ULL
+#define MM_PPE_BASE                                0xFFFFF6FB7DA00000ULL
+#define MM_PXE_BASE                                0xFFFFF6FB7DBED000ULL
 
+/* Page directory and page base addresses for 5-level paging */
+#define MM_PTE_LA57_BASE                           0xFFFF000000000000ULL
+#define MM_PDE_LA57_BASE                           0xFFFF010000000000ULL
+#define MM_PPE_LA57_BASE                           0xFFFF010800000000ULL
+#define MM_PXE_LA57_BASE                           0xFFFF010840000000ULL
+#define MM_P5E_LA57_BASE                           0xFFFF010840200000ULL
+
+/* PTE shift values */
+#define MM_PTE_SHIFT                               3
 #define MM_PTI_SHIFT                               12
 #define MM_PDI_SHIFT                               21
 #define MM_PPI_SHIFT                               30
 #define MM_PXI_SHIFT                               39
+#define MM_P5I_SHIFT                               48
 
-#define MM_PTE_SHIFT                               3
-
+/* Number of PTEs per page */
+#define MM_PTE_PER_PAGE                            512
+#define MM_PDE_PER_PAGE                            512
+#define MM_PPE_PER_PAGE                            512
 #define MM_PXE_PER_PAGE                            512
 
 /* Minimum number of physical pages needed by the system */
@@ -38,6 +51,21 @@
 /* Default number of secondary colors */
 #define MM_DEFAULT_SECONDARY_COLORS                64
 
+/* Number of HAL allocation descriptors */
+#define MM_HARDWARE_ALLOCATION_DESCRIPTORS         64
+
+/* Kernel HAL heap initial start address */
+#define MM_HARDWARE_HEAP_START_ADDRESS             ((PVOID)(((ULONG_PTR)MM_HARDWARE_VA_START) + 1024 * 1024))
+
+/* HAL memory pool virtual address start */
+#define MM_HARDWARE_VA_START                       0xFFFFFFFFFFC00000ULL
+
+/* Maximum physical address used by HAL allocations */
+#define MM_MAXIMUM_PHYSICAL_ADDRESS                0x00000000FFFFFFFFULL
+
+/* Trampoline code address */
+#define MM_TRAMPOLINE_ADDRESS                      0x80000
+
 /* Page size enumeration list */
 typedef enum _PAGE_SIZE
 {
@@ -67,6 +95,18 @@ typedef struct _HARDWARE_PTE
     ULONGLONG NoExecute:1;
 } HARDWARE_PTE, *PHARDWARE_PTE;
 
+/* Page map information structure definition */
+typedef struct _MMPAGEMAP_INFO
+{
+    BOOLEAN Xpa;
+    ULONGLONG PteBase;
+    ULONGLONG PdeBase;
+    ULONGLONG PpeBase;
+    ULONGLONG PxeBase;
+    ULONGLONG P5eBase;
+    ULONG VaBits;
+} MMPAGEMAP_INFO, *PMMPAGEMAP_INFO;
+
 /* A Page Table Entry on AMD64 system */
 typedef struct _MMPTE_HARDWARE
 {

sdk/xtdk/amd64/xtstruct.h

@@ -13,10 +13,23 @@
 
 
 /* Architecture-specific enumeration lists forward references */
+typedef enum _APIC_DM APIC_DM, *PAPIC_DM;
+typedef enum _APIC_DSH APIC_DSH, *PAPIC_DSH;
+typedef enum _APIC_MODE APIC_MODE, *PAPIC_MODE;
+typedef enum _APIC_REGISTER APIC_REGISTER, *PAPIC_REGISTER;
 typedef enum _CPU_VENDOR CPU_VENDOR, *PCPU_VENDOR;
-typedef enum _CPUID_FEATURES CPUID_FEATURES, *PCPUID_FEATURES;
+typedef enum _CPUID_FEATURES_EXTENDED CPUID_FEATURES_EXTENDED, *PCPUID_FEATURES_EXTENDED;
+typedef enum _CPUID_FEATURES_STANDARD1 CPUID_FEATURES_STANDARD1, *PCPUID_FEATURES_STANDARD1;
+typedef enum _CPUID_FEATURES_STANDARD7_LEAF0 CPUID_FEATURES_STANDARD7_LEAF0, *PCPUID_FEATURES_STANDARD7_LEAF0;
+typedef enum _CPUID_FEATURES_STANDARD7_LEAF1 CPUID_FEATURES_STANDARD7_LEAF1, *PCPUID_FEATURES_STANDARD7_LEAF1;
 typedef enum _CPUID_REQUESTS CPUID_REQUESTS, *PCPUID_REQUESTS;
 typedef enum _PAGE_SIZE PAGE_SIZE, *PPAGE_SIZE;
+typedef enum _PIC_I8259_ICW1_INTERRUPT_MODE PIC_I8259_ICW1_INTERRUPT_MODE, *PPIC_I8259_ICW1_INTERRUPT_MODE;
+typedef enum _PIC_I8259_ICW1_INTERVAL PIC_I8259_ICW1_INTERVAL, *PPIC_I8259_ICW1_INTERVAL;
+typedef enum _PIC_I8259_ICW1_OPERATING_MODE PIC_I8259_ICW1_OPERATING_MODE, *PPIC_I8259_ICW1_OPERATING_MODE;
+typedef enum _PIC_I8259_ICW4_BUFFERED_MODE PIC_I8259_ICW4_BUFFERED_MODE, *PPIC_I8259_ICW4_BUFFERED_MODE;
+typedef enum _PIC_I8259_ICW4_EOI_MODE PIC_I8259_ICW4_EOI_MODE, *PPIC_I8259_ICW4_EOI_MODE;
+typedef enum _PIC_I8259_ICW4_SYSTEM_MODE PIC_I8259_ICW4_SYSTEM_MODE, *PPIC_I8259_ICW4_SYSTEM_MODE;
 
 /* Architecture-specific structures forward references */
 typedef struct _CONTEXT CONTEXT, *PCONTEXT;
@@ -38,6 +51,7 @@ typedef struct _KSWITCH_FRAME KSWITCH_FRAME, *PKSWITCH_FRAME;
 typedef struct _KTHREAD_INIT_FRAME KTHREAD_INIT_FRAME, *PKTHREAD_INIT_FRAME;
 typedef struct _KTRAP_FRAME KTRAP_FRAME, *PKTRAP_FRAME;
 typedef struct _KTSS KTSS, *PKTSS;
+typedef struct _MMPAGEMAP_INFO MMPAGEMAP_INFO, *PMMPAGEMAP_INFO;
 typedef struct _MMPFN MMPFN, *PMMPFN;
 typedef struct _MMPTE_HARDWARE MMPTE_HARDWARE, *PMMPTE_HARDWARE;
 typedef struct _MMPTE_HARDWARE_LARGEPAGE MMPTE_HARDWARE_LARGEPAGE, *PMMPTE_HARDWARE_LARGEPAGE;
@@ -49,9 +63,18 @@ typedef struct _MMPTE_TRANSITION MMPTE_TRANSITION, *PMMPTE_TRANSITION;
 typedef struct _THREAD_ENVIRONMENT_BLOCK THREAD_ENVIRONMENT_BLOCK, *PTHREAD_ENVIRONMENT_BLOCK;
 
 /* Unions forward references */
+typedef union _APIC_BASE_REGISTER APIC_BASE_REGISTER, *PAPIC_BASE_REGISTER;
+typedef union _APIC_COMMAND_REGISTER APIC_COMMAND_REGISTER, *PAPIC_COMMAND_REGISTER;
+typedef union _APIC_LVT_REGISTER APIC_LVT_REGISTER, *PAPIC_LVT_REGISTER;
+typedef union _APIC_SPURIOUS_REGISTER APIC_SPURIOUS_REGISTER, *PAPIC_SPURIOUS_REGISTER;
+typedef union _MMPTE MMP5E, *PMMP5E;
 typedef union _MMPTE MMPDE, *PMMPDE;
 typedef union _MMPTE MMPPE, *PMMPPE;
 typedef union _MMPTE MMPTE, *PMMPTE;
 typedef union _MMPTE MMPXE, *PMMPXE;
+typedef union _PIC_I8259_ICW1 PIC_I8259_ICW1, *PPIC_I8259_ICW1;
+typedef union _PIC_I8259_ICW2 PIC_I8259_ICW2, *PPIC_I8259_ICW2;
+typedef union _PIC_I8259_ICW3 PIC_I8259_ICW3, *PPIC_I8259_ICW3;
+typedef union _PIC_I8259_ICW4 PIC_I8259_ICW4, *PPIC_I8259_ICW4;
 
 #endif /* __XTDK_AMD64_XTSTRUCT_H */

sdk/xtdk/bltypes.h

@@ -18,10 +18,11 @@
 
 /* EFI XT boot devices */
 #define XTBL_BOOT_DEVICE_UNKNOWN                                    0x00
-#define XTBL_BOOT_DEVICE_CDROM                                      0x01
-#define XTBL_BOOT_DEVICE_FLOPPY                                     0x02
-#define XTBL_BOOT_DEVICE_HARDDISK                                   0x03
-#define XTBL_BOOT_DEVICE_RAMDISK                                    0x04
+#define XTBL_BOOT_DEVICE_ESP                                        0x01
+#define XTBL_BOOT_DEVICE_CDROM                                      0x02
+#define XTBL_BOOT_DEVICE_FLOPPY                                     0x04
+#define XTBL_BOOT_DEVICE_HARDDISK                                   0x08
+#define XTBL_BOOT_DEVICE_RAMDISK                                    0x10
 
 /* XTLDR Debug Port type definitions */
 #define XTBL_DEBUGPORT_SCREEN                                       1
@@ -40,30 +41,34 @@
 #define XTBL_TUI_MAX_DIALOG_WIDTH                                   100
 
 /* XTLDR Routine pointers */
-typedef LONG (*PBL_GET_MEMTYPE_ROUTINE)(IN LONG EfiMemoryType);
+typedef LONG (*PBL_GET_MEMTYPE_ROUTINE)(IN EFI_MEMORY_TYPE EfiMemoryType);
 
 /* Boot Loader protocol routine pointers */
-typedef EFI_STATUS (*PBL_ALLOCATE_PAGES)(IN ULONGLONG Size, OUT PEFI_PHYSICAL_ADDRESS Memory);
+typedef EFI_STATUS (*PBL_ALLOCATE_PAGES)(IN EFI_ALLOCATE_TYPE AllocationType, IN ULONGLONG Size, OUT PEFI_PHYSICAL_ADDRESS Memory);
 typedef EFI_STATUS (*PBL_ALLOCATE_POOL)(IN UINT_PTR Size, OUT PVOID *Memory);
-typedef EFI_STATUS (*PBL_BOOTMENU_INITIALIZE_OS_LIST)(OUT PXTBL_BOOTMENU_ITEM *MenuEntries, OUT PULONG EntriesCount, OUT PULONG DefaultId);
+typedef EFI_STATUS (*PBL_BOOTMENU_INITIALIZE_OS_LIST)(IN ULONG MaxNameLength, OUT PXTBL_BOOTMENU_ITEM *MenuEntries, OUT PULONG EntriesCount, OUT PULONG DefaultId);
+typedef BOOLEAN (*PBL_BOOTUTIL_GET_BOOLEAN_PARAMETER)(IN CONST PWCHAR Parameters, IN CONST PWCHAR Needle);
 typedef EFI_STATUS (*PBL_BUILD_PAGE_MAP)(IN PXTBL_PAGE_MAPPING PageMap, IN ULONG_PTR SelfMapAddress);
 typedef EFI_STATUS (*PBL_CLOSE_VOLUME)(IN PEFI_HANDLE VolumeHandle);
 typedef VOID (*PBL_CLEAR_CONSOLE_LINE)(IN ULONGLONG LineNo);
 typedef EFI_STATUS (*PBL_CLOSE_XT_PROTOCOL)(IN PEFI_HANDLE Handle, IN PEFI_GUID ProtocolGuid);
 typedef BOOLEAN (*PBL_CONFIG_GET_BOOLEAN_VALUE)(IN CONST PWCHAR ConfigName);
-typedef PWCHAR (*PBL_CONFIG_GET_VALUE)(IN CONST PWCHAR ConfigName);
+typedef EFI_STATUS (*PBL_CONFIG_GET_BOOT_OPTION_VALUE)(IN PLIST_ENTRY Options, IN CONST PWCHAR OptionName, OUT PWCHAR *OptionValue);
+typedef VOID (*PBL_CONFIG_GET_EDITABLE_OPTIONS)(OUT CONST PWCHAR **OptionsArray, OUT PULONG OptionsCount);
+typedef EFI_STATUS (*PBL_CONFIG_GET_VALUE)(IN CONST PWCHAR ConfigName, OUT PWCHAR *ConfigValue);
+typedef EFI_STATUS (*PBL_CONFIG_SET_BOOT_OPTION_VALUE)(IN PLIST_ENTRY Options, IN CONST PWCHAR OptionName, IN CONST PWCHAR OptionValue);
 typedef VOID (*PBL_CONSOLE_CLEAR_SCREEN)();
 typedef VOID (*PBL_CONSOLE_DISABLE_CURSOR)();
 typedef VOID (*PBL_CONSOLE_ENABLE_CURSOR)();
-typedef VOID (*PBL_CONSOLE_PRINT)(IN PUSHORT Format, IN ...);
+typedef VOID (*PBL_CONSOLE_PRINT)(IN PCWSTR Format, IN ...);
 typedef VOID (*PBL_CONSOLE_QUERY_MODE)(OUT PUINT_PTR ResX, OUT PUINT_PTR ResY);
 typedef VOID (*PBL_CONSOLE_READ_KEY_STROKE)(OUT PEFI_INPUT_KEY Key);
 typedef VOID (*PBL_CONSOLE_RESET_INPUT_BUFFER)();
 typedef VOID (*PBL_CONSOLE_SET_ATTRIBUTES)(IN ULONGLONG Attributes);
 typedef VOID (*PBL_CONSOLE_SET_CURSOR_POSITION)(IN ULONGLONG PosX, IN ULONGLONG PosY);
-typedef VOID (*PBL_CONSOLE_WRITE)(IN PUSHORT String);
+typedef VOID (*PBL_CONSOLE_WRITE)(IN PCWSTR String);
 typedef VOID (XTAPI *PBL_COPY_MEMORY)(OUT PVOID Destination, IN PCVOID Source, IN SIZE_T Length);
-typedef VOID (*PBL_DEBUG_PRINT)(IN PUSHORT Format, IN ...);
+typedef VOID (*PBL_DEBUG_PRINT)(IN PCWSTR Format, IN ...);
 typedef EFI_STATUS (*PBL_ENTER_FIRMWARE_SETUP)();
 typedef EFI_STATUS (*PBL_EXIT_BOOT_SERVICES)();
 typedef EFI_STATUS (*PBL_FIND_BOOT_PROTOCOL)(IN PWCHAR SystemType, OUT PEFI_GUID BootProtocolGuid);
@@ -84,7 +89,7 @@ typedef EFI_STATUS (*PBL_INVOKE_BOOT_PROTOCOL)(IN PWCHAR ShortName, IN PLIST_ENT
 typedef EFI_STATUS (*PBL_LOCATE_PROTOCOL_HANDLES)(OUT PEFI_HANDLE *Handles, OUT PUINT_PTR Count, IN PEFI_GUID ProtocolGuid);
 typedef EFI_STATUS (*PBL_LOAD_EFI_IMAGE)(IN PEFI_DEVICE_PATH_PROTOCOL DevicePath, IN PVOID ImageData, IN SIZE_T ImageSize, OUT PEFI_HANDLE ImageHandle);
 typedef EFI_STATUS (*PBL_MAP_EFI_MEMORY)(IN OUT PXTBL_PAGE_MAPPING PageMap, IN OUT PVOID *MemoryMapAddress, IN PBL_GET_MEMTYPE_ROUTINE GetMemoryTypeRoutine);
-typedef EFI_STATUS (*PBL_MAP_PAGE)(IN PXTBL_PAGE_MAPPING PageMap, IN UINT_PTR VirtualAddress, IN UINT_PTR PhysicalAddress, IN UINT NumberOfPages);
+typedef EFI_STATUS (*PBL_MAP_PAGE)(IN PXTBL_PAGE_MAPPING PageMap, IN ULONG_PTR VirtualAddress, IN ULONG_PTR PhysicalAddress, IN ULONG NumberOfPages);
 typedef EFI_STATUS (*PBL_MAP_VIRTUAL_MEMORY)(IN OUT PXTBL_PAGE_MAPPING PageMap, IN PVOID VirtualAddress, IN PVOID PhysicalAddress, IN ULONGLONG NumberOfPages, IN LOADER_MEMORY_TYPE MemoryType);
 typedef EFI_STATUS (*PBL_OPEN_VOLUME)(IN PEFI_DEVICE_PATH_PROTOCOL DevicePath, OUT PEFI_HANDLE DiskHandle, OUT PEFI_FILE_HANDLE *FsHandle);
 typedef EFI_STATUS (*PBL_OPEN_PROTOCOL)(OUT PEFI_HANDLE Handle, OUT PVOID *ProtocolHandler, IN PEFI_GUID ProtocolGuid);
@@ -110,6 +115,7 @@ typedef VOID (XTAPI *PBL_ZERO_MEMORY)(OUT PVOID Destination, IN SIZE_T Length);
 
 /* Module protocols routine pointers */
 typedef EFI_STATUS (*PBL_ACPI_GET_ACPI_DESCRIPTION_POINTER)(OUT PVOID *AcpiTable);
+typedef EFI_STATUS (*PBL_ACPI_GET_ACPI_TABLE)(IN CONST UINT Signature, IN PVOID PreviousTable, OUT PVOID *AcpiTable);
 typedef EFI_STATUS (*PBL_ACPI_GET_APIC_BASE)(OUT PVOID *ApicBase);
 typedef EFI_STATUS (*PBL_ACPI_GET_RSDP_TABLE)(OUT PVOID *AcpiTable);
 typedef EFI_STATUS (*PBL_ACPI_GET_SMBIOS_TABLE)(OUT PVOID *SmBiosTable);
@@ -151,6 +157,7 @@ typedef struct _XTBL_BOOT_PARAMETERS
 typedef struct _XTBL_BOOTMENU_ITEM
 {
     PWCHAR EntryName;
+    PWCHAR FullName;
     PWCHAR ShortName;
     PLIST_ENTRY Options;
 } XTBL_BOOTMENU_ITEM, *PXTBL_BOOTMENU_ITEM;
@@ -269,14 +276,14 @@ typedef struct _XTBL_FRAMEBUFFER_MODE_INFORMATION
     EFI_GRAPHICS_PIXEL_FORMAT PixelFormat;
     struct
     {
-        USHORT BlueMask;
         USHORT BlueShift;
-        USHORT GreenMask;
+        USHORT BlueSize;
         USHORT GreenShift;
-        USHORT RedMask;
+        USHORT GreenSize;
         USHORT RedShift;
-        USHORT ReservedMask;
+        USHORT RedSize;
         USHORT ReservedShift;
+        USHORT ReservedSize;
     } PixelInformation;
 } XTBL_FRAMEBUFFER_MODE_INFORMATION, *PXTBL_FRAMEBUFFER_MODE_INFORMATION;
 
@@ -301,6 +308,7 @@ typedef struct _XTBL_FRAMEBUFFER_INFORMATION
 typedef struct _XTBL_ACPI_PROTOCOL
 {
     PBL_ACPI_GET_ACPI_DESCRIPTION_POINTER GetAcpiDescriptionPointer;
+    PBL_ACPI_GET_ACPI_TABLE GetAcpiTable;
     PBL_ACPI_GET_APIC_BASE GetApicBase;
     PBL_ACPI_GET_RSDP_TABLE GetRsdpTable;
     PBL_ACPI_GET_SMBIOS_TABLE GetSMBiosTable;
@@ -352,9 +360,16 @@ typedef struct _XTBL_LOADER_PROTOCOL
         PBL_REGISTER_BOOT_PROTOCOL RegisterProtocol;
     } Boot;
     struct
+    {
+        PBL_BOOTUTIL_GET_BOOLEAN_PARAMETER GetBooleanParameter;
+    } BootUtil;
+    struct
     {
         PBL_CONFIG_GET_BOOLEAN_VALUE GetBooleanValue;
+        PBL_CONFIG_GET_BOOT_OPTION_VALUE GetBootOptionValue;
+        PBL_CONFIG_GET_EDITABLE_OPTIONS GetEditableOptions;
         PBL_CONFIG_GET_VALUE GetValue;
+        PBL_CONFIG_SET_BOOT_OPTION_VALUE SetBootOptionValue;
     } Config;
     struct
     {

sdk/xtdk/hlfuncs.h

@@ -23,7 +23,6 @@ HlComPortPutByte(IN PCPPORT Port,
 XTCDECL
 XTSTATUS
 HlInitializeComPort(IN OUT PCPPORT Port,
-                    IN ULONG PortNumber,
                     IN PUCHAR PortAddress,
                     IN ULONG BaudRate);
 

sdk/xtdk/hltypes.h

@@ -9,141 +9,365 @@
 #ifndef __XTDK_HLTYPES_H
 #define __XTDK_HLTYPES_H
 
+#include <xtbase.h>
+#include <xtdefs.h>
 #include <xttypes.h>
+#include ARCH_HEADER(hltypes.h)
 
 
+/* ACPI Root System Description Pointer (RSDP) signature */
+#define ACPI_RSDP_SIGNATURE                         0x2052545020445352
+
+/* ACPI table signatures */
+#define ACPI_BERT_SIGNATURE                         0x54524542 /* Boot Error Record Table */
+#define ACPI_BGRT_SIGNATURE                         0x54524742 /* Boot Graphics Record Table */
+#define ACPI_BOOT_SIGNATURE                         0x544F4F42 /* ACPI BOOT Table */
+#define ACPI_CPEP_SIGNATURE                         0x50455043 /* Corrected Platform Error Polling Table */
+#define ACPI_DBG2_SIGNATURE                         0x32474244 /* Debug Port Table v2 */
+#define ACPI_DBGP_SIGNATURE                         0x50474244 /* Debug Port Table */
+#define ACPI_DMAR_SIGNATURE                         0x52414D44 /* DMA Remapping Table */
+#define ACPI_DSDT_SIGNATURE                         0x54445344 /* Differentiated System Description Table */
+#define ACPI_ECDT_SIGNATURE                         0x54444345 /* Embedded Controller Description Table */
+#define ACPI_ERST_SIGNATURE                         0x54535245 /* Error Record Serialization Table */
+#define ACPI_FACS_SIGNATURE                         0x53434146 /* Firmware ACPI Control Structure */
+#define ACPI_FADT_SIGNATURE                         0x50434146 /* Fixed ACPI Description Table */
+#define ACPI_FBPT_SIGNATURE                         0x54504246 /* Firmware Boot Performance Table */
+#define ACPI_FPDT_SIGNATURE                         0x54445046 /* Firmware Performance Data Table */
+#define ACPI_GTDT_SIGNATURE                         0x54445447 /* Generic Timer Description Table */
+#define ACPI_HPET_SIGNATURE                         0x54455048 /* High Precision Event Timer */
+#define ACPI_IVRS_SIGNATURE                         0x53525649 /* AMD IOMMU Resource Table */
+#define ACPI_MADT_SIGNATURE                         0x43495041 /* MADT/APIC Description Table */
+#define ACPI_MCFG_SIGNATURE                         0x4746434D /* Memory Mapped Configuration Space Access Table */
+#define ACPI_MPST_SIGNATURE                         0x5453504D /* Memory Power State Table*/
+#define ACPI_MSCT_SIGNATURE                         0x5443534D /* Maximum System Characteristics Table */
+#define ACPI_NFIT_SIGNATURE                         0x5449464E /* NVDIMM Firmware Interface Table */
+#define ACPI_PMMT_SIGNATURE                         0x544D4D50 /* Platform Memory Topology Table */
+#define ACPI_PSDT_SIGNATURE                         0x54445350 /* Persistent System Description Table */
+#define ACPI_RAS2_SIGNATURE                         0x32534152 /* ACPI RAS2 Feature Table */
+#define ACPI_RASF_SIGNATURE                         0x46534152 /* ACPI RAS Feature Table */
+#define ACPI_RSDT_SIGNATURE                         0x54445352 /* Root System Description Table */
+#define ACPI_SBST_SIGNATURE                         0x54534253 /* Smart Battery Subsystem Table */
+#define ACPI_SDEV_SIGNATURE                         0x56454453 /* Secure Device Table */
+#define ACPI_SLIT_SIGNATURE                         0x54494C53 /* System Locality Distance Information Table */
+#define ACPI_SPCR_SIGNATURE                         0x52435053 /* Serial Port Console Redirection Table */
+#define ACPI_SRAT_SIGNATURE                         0x54415253 /* Static Resource Affinity Table */
+#define ACPI_SSDT_SIGNATURE                         0x54445353 /* Secondary System Descriptor Table */
+#define ACPI_TPM2_SIGNATURE                         0x324D5054 /* ACPI TPM 2.0 Table */
+#define ACPI_WAET_SIGNATURE                         0x54454157 /* Windows ACPI Enlightenment Table */
+#define ACPI_WDAT_SIGNATURE                         0x54414457 /* Watch Dog Action Table */
+#define ACPI_WDTT_SIGNATURE                         0x54524457 /* Watchdog Timer Resource Table */
+#define ACPI_WPBT_SIGNATURE                         0x54425057 /* Windows Platform Binary Table */
+#define ACPI_WSMT_SIGNATURE                         0x544D5357 /* Windows SMM Security Mitigation Table */
+#define ACPI_XSDT_SIGNATURE                         0x54445358 /* eXtended System Descriptor Table */
+
+/* ACPI FADT flags masks */
+#define ACPI_FADT_32BIT_TIMER                       (1<<8)
+
+/* ACPI Timer bit masks */
+#define ACPI_FADT_TIMER_32BIT                       0x80000000
+#define ACPI_FADT_TIMER_24BIT                       0x00800000
+
+/* ACPI MADT subtable type definitions */
+#define ACPI_MADT_TYPE_LOCAL_APIC                   0
+#define ACPI_MADT_TYPE_IOAPIC                       1
+#define ACPI_MADT_TYPE_INT_OVERRIDE                 2
+#define ACPI_MADT_TYPE_NMI_SOURCE                   3
+#define ACPI_MADT_TYPE_LOCAL_APIC_NMI               4
+#define ACPI_MADT_TYPE_LOCAL_APIC_OVERRIDE          5
+#define ACPI_MADT_TYPE_IO_SAPIC                     6
+#define ACPI_MADT_TYPE_LOCAL_SAPIC                  7
+#define ACPI_MADT_TYPE_INTERRUPT_SOURCE             8
+#define ACPI_MADT_TYPE_LOCAL_X2APIC                 9
+#define ACPI_MADT_TYPE_LOCAL_X2APIC_NMI             10
+#define ACPI_MADT_TYPE_GENERIC_INTERRUPT            11
+#define ACPI_MADT_TYPE_GENERIC_DISTRIBUTOR          12
+#define ACPI_MADT_TYPE_GENERIC_MSI_FRAME            13
+#define ACPI_MADT_TYPE_GENERIC_REDISTRIBUTOR        14
+#define ACPI_MADT_TYPE_GENERIC_TRANSLATOR           15
+#define ACPI_MADT_TYPE_MULTIPROC_WAKEUP             16
+#define ACPI_MADT_TYPE_CORE_PIC                     17
+#define ACPI_MADT_TYPE_LIO_PIC                      18
+#define ACPI_MADT_TYPE_HT_PIC                       19
+#define ACPI_MADT_TYPE_EIO_PIC                      20
+#define ACPI_MADT_TYPE_MSI_PIC                      21
+#define ACPI_MADT_TYPE_BIO_PIC                      22
+#define ACPI_MADT_TYPE_LPC_PIC                      23
+#define ACPI_MADT_TYPE_RINTC                        24
+#define ACPI_MADT_TYPE_IMSIC                        25
+#define ACPI_MADT_TYPE_APLIC                        26
+#define ACPI_MADT_TYPE_PLIC                         27
+
+/* ACPI MADT Processor Local APIC Flags */
+#define ACPI_MADT_PLACE_ENABLED                     0 /* Processor Local APIC CPU Enabled */
+#define ACPI_MADT_PLAOC_ENABLED                     1 /* Processor Local APIC Online Capable */
+
 /* Default serial port settings */
-#define COMPORT_CLOCK_RATE          0x1C200
-#define COMPORT_WAIT_TIMEOUT        204800
+#define COMPORT_CLOCK_RATE                          0x1C200
+#define COMPORT_WAIT_TIMEOUT                        204800
 
 /* Serial port divisors */
-#define COMPORT_DIV_DLL             0x00 /* Divisor Latch Least */
-#define COMPORT_DIV_DLM             0x01 /* Divisor Latch Most */
+#define COMPORT_DIV_DLL                             0x00 /* Divisor Latch Least */
+#define COMPORT_DIV_DLM                             0x01 /* Divisor Latch Most */
 
 /* Serial port control flags */
-#define COMPORT_FLAG_INIT           0x01 /* Port Initialized */
-#define COMPORT_FLAG_DBR            0x02 /* Default Baud Rate */
-#define COMPORT_FLAG_MC             0x04 /* Modem Control */
+#define COMPORT_FLAG_INIT                           0x01 /* Port Initialized */
+#define COMPORT_FLAG_DBR                            0x02 /* Default Baud Rate */
+#define COMPORT_FLAG_MC                             0x04 /* Modem Control */
 
 /* Serial port Fifo Control Register (FCR) access masks */
-#define COMPORT_FCR_DISABLE         0x00 /* Disable */
-#define COMPORT_FCR_ENABLE          0x01 /* Enable */
-#define COMPORT_FCR_RCVR_RESET      0x02 /* Receiver Reset */
-#define COMPORT_FCR_TXMT_RESET      0x04 /* Transmitter Reset */
+#define COMPORT_FCR_DISABLE                         0x00 /* Disable */
+#define COMPORT_FCR_ENABLE                          0x01 /* Enable */
+#define COMPORT_FCR_RCVR_RESET                      0x02 /* Receiver Reset */
+#define COMPORT_FCR_TXMT_RESET                      0x04 /* Transmitter Reset */
 
 /* Serial port Line Control Register (LCR) access masks */
-#define COMPORT_LCR_1STOP           0x00 /* 1 Stop Bit */
-#define COMPORT_LCR_2STOP           0x04 /* 2 Stop Bits */
-#define COMPORT_LCR_5DATA           0x00 /* 5 Data Bits */
-#define COMPORT_LCR_6DATA           0x01 /* 6 Data Bits */
-#define COMPORT_LCR_7DATA           0x02 /* 7 Data Bits */
-#define COMPORT_LCR_8DATA           0x03 /* 8 Data Bits */
-#define COMPORT_LCR_PARN            0x00 /* None Parity */
-#define COMPORT_LCR_PARO            0x08 /* Odd Parity */
-#define COMPORT_LCR_PARE            0x18 /* Even Parity */
-#define COMPORT_LCR_PARM            0x28 /* Mark Parity */
-#define COMPORT_LCR_PARS            0x38 /* Space Parity */
-#define COMPORT_LCR_BREAK           0x40 /* Break */
-#define COMPORT_LCR_DLAB            0x80 /* Divisor Latch Access Bit */
+#define COMPORT_LCR_1STOP                           0x00 /* 1 Stop Bit */
+#define COMPORT_LCR_2STOP                           0x04 /* 2 Stop Bits */
+#define COMPORT_LCR_5DATA                           0x00 /* 5 Data Bits */
+#define COMPORT_LCR_6DATA                           0x01 /* 6 Data Bits */
+#define COMPORT_LCR_7DATA                           0x02 /* 7 Data Bits */
+#define COMPORT_LCR_8DATA                           0x03 /* 8 Data Bits */
+#define COMPORT_LCR_PARN                            0x00 /* None Parity */
+#define COMPORT_LCR_PARO                            0x08 /* Odd Parity */
+#define COMPORT_LCR_PARE                            0x18 /* Even Parity */
+#define COMPORT_LCR_PARM                            0x28 /* Mark Parity */
+#define COMPORT_LCR_PARS                            0x38 /* Space Parity */
+#define COMPORT_LCR_BREAK                           0x40 /* Break */
+#define COMPORT_LCR_DLAB                            0x80 /* Divisor Latch Access Bit */
 
 /* Serial port Line Status Register (LSR) access masks */
-#define COMPORT_LSR_DIS             0x00 /* Disable */
-#define COMPORT_LSR_DR              0x01 /* Data Ready */
-#define COMPORT_LSR_OE              0x02 /* Overrun Error */
-#define COMPORT_LSR_PE              0x04 /* Parity Error */
-#define COMPORT_LSR_FE              0x08 /* Framing Error */
-#define COMPORT_LSR_BI              0x10 /* Break Interrupt */
-#define COMPORT_LSR_THRE            0x20 /* Transmit Holding Register Empty */
-#define COMPORT_LSR_TEMPTY          0x40 /* Transmitter Empty */
-#define COMPORT_LSR_FIFOE           0x80 /* FIFO Error */
+#define COMPORT_LSR_DIS                             0x00 /* Disable */
+#define COMPORT_LSR_DR                              0x01 /* Data Ready */
+#define COMPORT_LSR_OE                              0x02 /* Overrun Error */
+#define COMPORT_LSR_PE                              0x04 /* Parity Error */
+#define COMPORT_LSR_FE                              0x08 /* Framing Error */
+#define COMPORT_LSR_BI                              0x10 /* Break Interrupt */
+#define COMPORT_LSR_THRE                            0x20 /* Transmit Holding Register Empty */
+#define COMPORT_LSR_TEMPTY                          0x40 /* Transmitter Empty */
+#define COMPORT_LSR_FIFOE                           0x80 /* FIFO Error */
 
 /* Serial port Modem Control Register (MCR) access masks */
-#define COMPORT_MCR_DTR             0x01 /* Data Terminal Ready */
-#define COMPORT_MCR_RTS             0x02 /* Ready To Send */
-#define COMPORT_MCR_OUT1            0x04 /* Generic Output 1 */
-#define COMPORT_MCR_OUT2            0x08 /* Generic Output 2 */
-#define COMPORT_MCR_NOM             0x0F /* Normal Operation Mode */
-#define COMPORT_MCR_LOOP            0x10 /* Loopback Testing Mode */
+#define COMPORT_MCR_DTR                             0x01 /* Data Terminal Ready */
+#define COMPORT_MCR_RTS                             0x02 /* Ready To Send */
+#define COMPORT_MCR_OUT1                            0x04 /* Generic Output 1 */
+#define COMPORT_MCR_OUT2                            0x08 /* Generic Output 2 */
+#define COMPORT_MCR_NOM                             0x0F /* Normal Operation Mode */
+#define COMPORT_MCR_LOOP                            0x10 /* Loopback Testing Mode */
 
 /* Serial port Modem Status Register (MSR) access masks */
-#define COMPORT_MSR_DCTS            0x01 /* Delta Clear To Send */
-#define COMPORT_MSR_DDSR            0x02 /* Delta Data Set Ready */
-#define COMPORT_MSR_DTRRTS          0x03 /* DTR and RTS */
-#define COMPORT_MSR_TERI            0x04 /* Trailing Edge Ring Indicator */
-#define COMPORT_MSR_DDCD            0x08 /* Delta Data Carrier Detect */
-#define COMPORT_MSR_CTS             0x10 /* Clear To Send */
-#define COMPORT_MSR_DSR             0x20 /* Data Set Ready */
-#define COMPORT_MSR_RI              0x40 /* Ring Indicator */
-#define COMPORT_MSR_DCD             0x80 /* Data Carrier Detect */
-#define COMPORT_MSR_DSRCTSCD        0xB0 /* DSR, CTS and CD */
-#define COMPORT_MSR_TST             0xAE /* Test Pattern */
+#define COMPORT_MSR_DCTS                            0x01 /* Delta Clear To Send */
+#define COMPORT_MSR_DDSR                            0x02 /* Delta Data Set Ready */
+#define COMPORT_MSR_DTRRTS                          0x03 /* DTR and RTS */
+#define COMPORT_MSR_TERI                            0x04 /* Trailing Edge Ring Indicator */
+#define COMPORT_MSR_DDCD                            0x08 /* Delta Data Carrier Detect */
+#define COMPORT_MSR_CTS                             0x10 /* Clear To Send */
+#define COMPORT_MSR_DSR                             0x20 /* Data Set Ready */
+#define COMPORT_MSR_RI                              0x40 /* Ring Indicator */
+#define COMPORT_MSR_DCD                             0x80 /* Data Carrier Detect */
+#define COMPORT_MSR_DSRCTSCD                        0xB0 /* DSR, CTS and CD */
+#define COMPORT_MSR_TST                             0xAE /* Test Pattern */
 
 /* Serial port offsets of the various registers */
-#define COMPORT_REG_RBR             0x00 /* Receive Buffer Register */
-#define COMPORT_REG_THR             0x00 /* Transmit Holding Register */
-#define COMPORT_REG_IER             0x01 /* Interrupt Enable Register */
-#define COMPORT_REG_IIR             0x02 /* Interrupt Identity Register */
-#define COMPORT_REG_FCR             0x02 /* FIFO Control Register */
-#define COMPORT_REG_LCR             0x03 /* Line Control Register */
-#define COMPORT_REG_MCR             0x04 /* Modem Control Register */
-#define COMPORT_REG_LSR             0x05 /* Line Status Register */
-#define COMPORT_REG_MSR             0x06 /* Modem Status Register */
-#define COMPORT_REG_SR              0x07 /* Scratch Register */
+#define COMPORT_REG_RBR                             0x00 /* Receive Buffer Register */
+#define COMPORT_REG_THR                             0x00 /* Transmit Holding Register */
+#define COMPORT_REG_IER                             0x01 /* Interrupt Enable Register */
+#define COMPORT_REG_IIR                             0x02 /* Interrupt Identity Register */
+#define COMPORT_REG_FCR                             0x02 /* FIFO Control Register */
+#define COMPORT_REG_LCR                             0x03 /* Line Control Register */
+#define COMPORT_REG_MCR                             0x04 /* Modem Control Register */
+#define COMPORT_REG_LSR                             0x05 /* Line Status Register */
+#define COMPORT_REG_MSR                             0x06 /* Modem Status Register */
+#define COMPORT_REG_SR                              0x07 /* Scratch Register */
 
-/* APIC Register Address Map */
-typedef enum _APIC_REGISTER
+/* Generic Address structure */
+typedef struct _GENERIC_ADDRESS
 {
-    APIC_ID                       = 0x02, /* APIC ID Register */
-    APIC_VER                      = 0x03, /* APIC Version Register */
-    APIC_TPR                      = 0x08, /* Task Priority Register */
-    APIC_APR                      = 0x09, /* Arbitration Priority Register */
-    APIC_PPR                      = 0x0A, /* Processor Priority Register (R) */
-    APIC_EOI                      = 0x0B, /* EOI Register */
-    APIC_RRR                      = 0x0C, /* Remote Read Register */
-    APIC_LDR                      = 0x0D, /* Logical Destination Register */
-    APIC_DFR                      = 0x0E, /* Destination Format Register (not available in extended mode) */
-    APIC_SIVR                     = 0x0F, /* Spurious Interrupt Vector Register */
-    APIC_ISR                      = 0x10, /* Interrupt Service Register*/
-    APIC_TMR                      = 0x18, /* Trigger Mode Register */
-    APIC_IRR                      = 0x20, /* Interrupt Request Register */
-    APIC_ESR                      = 0x28, /* Error Status Register */
-    APIC_ICR0                     = 0x30, /* Interrupt Command Register */
-    APIC_ICR1                     = 0x31, /* Interrupt Command Register (not available in extended mode) */
-    APIC_TMRLVTR                  = 0x32, /* Timer Local Vector Table */
-    APIC_THRMLVTR                 = 0x33, /* Thermal Local Vector Table */
-    APIC_PCLVTR                   = 0x34, /* Performance Counter Local Vector Table */
-    APIC_LINT0                    = 0x35, /* LINT0 Local Vector Table */
-    APIC_LINT1                    = 0x36, /* LINT1 Local Vector Table */
-    APIC_ERRLVTR                  = 0x37, /* Error Local Vector Table */
-    APIC_TICR                     = 0x38, /* Initial Count Register for Timer */
-    APIC_TCCR                     = 0x39, /* Current Count Register for Timer */
-    APIC_TDCR                     = 0x3E, /* Timer Divide Configuration Register */
-    APIC_EAFR                     = 0x40, /* extended APIC Feature register */
-    APIC_EACR                     = 0x41, /* Extended APIC Control Register */
-    APIC_SEOI                     = 0x42, /* Specific End Of Interrupt Register */
-    APIC_EXT0LVTR                 = 0x50, /* Extended Interrupt 0 Local Vector Table */
-    APIC_EXT1LVTR                 = 0x51, /* Extended Interrupt 1 Local Vector Table */
-    APIC_EXT2LVTR                 = 0x52, /* Extended Interrupt 2 Local Vector Table */
-    APIC_EXT3LVTR                 = 0x53  /* Extended Interrupt 3 Local Vector Table */
-} APIC_REGISTER, *PAPIC_REGISTER;
+    UCHAR AddressSpaceID;
+    UCHAR BitWidth;
+    UCHAR BitOffset;
+    UCHAR Reserved;
+    PHYSICAL_ADDRESS Address;
+} PACKED GENERIC_ADDRESS, *PGENERIC_ADDRESS;
 
-/* APIC mode list */
-typedef enum _HAL_APIC_MODE
+/* Each ACPI table description header structure */
+typedef struct _ACPI_DESCRIPTION_HEADER
 {
-    APIC_MODE_COMPAT,
-    APIC_MODE_X2APIC
-} HAL_APIC_MODE, *PHAL_APIC_MODE;
+    ULONG Signature;
+    ULONG Length;
+    UCHAR Revision;
+    UCHAR Checksum;
+    UCHAR OemId[6];
+    UCHAR OemTableID[8];
+    ULONG OemRevision;
+    UCHAR CreatorID[4];
+    ULONG CreatorRev;
+} PACKED ACPI_DESCRIPTION_HEADER, *PACPI_DESCRIPTION_HEADER;
 
-/* Serial (COM) port initial state */
+/* Each ACPI subtable description header structure */
+typedef struct _ACPI_SUBTABLE_HEADER
+{
+    UCHAR Type;
+    UCHAR Length;
+} PACKED ACPI_SUBTABLE_HEADER, *PACPI_SUBTABLE_HEADER;
+
+/* ACPI cache list structure */
+typedef struct _ACPI_CACHE_LIST
+{
+    LIST_ENTRY ListEntry;
+    ACPI_DESCRIPTION_HEADER Header;
+} ACPI_CACHE_LIST, *PACPI_CACHE_LIST;
+
+/* ACPI Root System Description Table Pointer (RSDP) structure */
+typedef struct _ACPI_RSDP
+{
+    ULONGLONG Signature;
+    UCHAR Checksum;
+    UCHAR OemId[6];
+    UCHAR Revision;
+    ULONG RsdtAddress;
+    ULONG Length;
+    ULONGLONG XsdtAddress;
+    UCHAR XChecksum;
+    UCHAR Reserved[3];
+} PACKED ACPI_RSDP, *PACPI_RSDP;
+
+/* ACPI Root System Description Table (RSDT) structure */
+typedef struct _ACPI_RSDT
+{
+    ACPI_DESCRIPTION_HEADER Header;
+    ULONG Tables[];
+} PACKED ACPI_RSDT, *PACPI_RSDT;
+
+/* ACPI eXtended Root System Description Table (XSDT) structure */
+typedef struct _ACPI_XSDT
+{
+    ACPI_DESCRIPTION_HEADER Header;
+    ULONGLONG Tables[];
+} PACKED ACPI_XSDT, *PACPI_XSDT;
+
+/* Fixed ACPI Description Table (FADT) structure */
+typedef struct _ACPI_FADT
+{
+    ACPI_DESCRIPTION_HEADER Header;
+    ULONG FirmwareCtrl;
+    ULONG Dsdt;
+    UCHAR IntModel;
+    UCHAR PmProfile;
+    USHORT SciIntVector;
+    ULONG SmiCmdIoPort;
+    UCHAR AcpiOnValue;
+    UCHAR AcpiOffValue;
+    UCHAR S4BiosReq;
+    UCHAR PStateControl;
+    ULONG Pm1aEvtBlkIoPort;
+    ULONG Pm1bEvtBlkIoPort;
+    ULONG Pm1aCtrlBlkIoPort;
+    ULONG Pm1bCtrlBlkIoPort;
+    ULONG Pm2CtrlBlkIoPort;
+    ULONG PmTmrBlkIoPort;
+    ULONG Gp0BlkIoPort;
+    ULONG Gp1BlkIoPort;
+    UCHAR Pm1EvtLen;
+    UCHAR Pm1CtrlLen;
+    UCHAR Pm2CtrlLen;
+    UCHAR PmTmrLen;
+    UCHAR Gp0BlkLen;
+    UCHAR Gp1BlkLen;
+    UCHAR Gp1Base;
+    UCHAR CStateControl;
+    USHORT Lvl2Latency;
+    USHORT Lvl3Latency;
+    USHORT FlushSize;
+    USHORT FlushStride;
+    UCHAR DutyOffset;
+    UCHAR DutyWidth;
+    UCHAR DayAlarmIndex;
+    UCHAR MonthAlarmIndex;
+    UCHAR CenturyAlarmIndex;
+    USHORT BootArch;
+    UCHAR Reserved0;
+    ULONG Flags;
+    GENERIC_ADDRESS ResetReg;
+    UCHAR ResetVal;
+    USHORT ArmBootArch;
+    UCHAR Reserved1;
+    PHYSICAL_ADDRESS XFirmwareCtrl;
+    PHYSICAL_ADDRESS XDsdt;
+    GENERIC_ADDRESS XPm1aEvtBlk;
+    GENERIC_ADDRESS XPm1bEvtBlk;
+    GENERIC_ADDRESS XPm1aCtrlBlk;
+    GENERIC_ADDRESS XPm1bCtrlBlk;
+    GENERIC_ADDRESS XPm2CtrlBlk;
+    GENERIC_ADDRESS XPmTmrBlk;
+    GENERIC_ADDRESS XGp0Blk;
+    GENERIC_ADDRESS XGp1Blk;
+    GENERIC_ADDRESS SleepControlReg;
+    GENERIC_ADDRESS SleepStatusReg;
+} PACKED ACPI_FADT, *PACPI_FADT;
+
+/* ACPI Multiple APIC Description Table (MADT) structure */
+typedef struct _ACPI_MADT
+{
+    ACPI_DESCRIPTION_HEADER Header;
+    ULONG LocalApicAddress;
+    ULONG Flags;
+    ULONG ApicTables[];
+} PACKED ACPI_MADT, *PACPI_MADT;
+
+/* ACPI Local APIC MADT subtable structure */
+typedef struct _ACPI_MADT_LOCAL_APIC
+{
+    ACPI_SUBTABLE_HEADER Header;
+    UCHAR AcpiId;
+    UCHAR ApicId;
+    ULONG Flags;
+} PACKED ACPI_MADT_LOCAL_APIC, *PACPI_MADT_LOCAL_APIC;
+
+/* ACPI Local X2APIC MADT subtable structure */
+typedef struct _ACPI_MADT_LOCAL_X2APIC
+{
+    ACPI_SUBTABLE_HEADER Header;
+    USHORT Reserved;
+    ULONG ApicId;
+    ULONG Flags;
+    ULONG AcpiId;
+} PACKED ACPI_MADT_LOCAL_X2APIC, *PACPI_MADT_LOCAL_X2APIC;
+
+/* ACPI System Information structure */
+typedef struct _ACPI_SYSTEM_INFO
+{
+    ULONG CpuCount;
+    ULONG RunningCpus;
+    ULONG BusCount;
+    ULONG IoApicCount;
+    ULONG IntiCount;
+    ULONG LintiCount;
+    BOOLEAN ImcrPresent;
+    ULONG ApicBase;
+    PPROCESSOR_IDENTITY CpuInfo;
+    ULONG IoApicPhysicalBase[APIC_MAX_IOAPICS];
+    ULONG IoApicVirtualBase[APIC_MAX_IOAPICS];
+    ULONG IoApicVectorBase[APIC_MAX_IOAPICS];
+} ACPI_SYSTEM_INFO, *PACPI_SYSTEM_INFO;
+
+/* ACPI Timer information structure */
+typedef struct _ACPI_TIMER_INFO
+{
+    ULONG TimerPort;
+    ULONG MsbMask;
+} ACPI_TIMER_INFO, *PACPI_TIMER_INFO;
+
+/* Serial (COM) port state */
 typedef struct _CPPORT
 {
     PUCHAR Address;
     ULONG Baud;
     USHORT Flags;
+    UCHAR Ring;
 } CPPORT, *PCPPORT;
 
-/* HAL framebuffer data structure */
-typedef struct _HAL_FRAMEBUFFER_DATA
+/* Framebuffer data structure */
+typedef struct _HL_FRAMEBUFFER_DATA
 {
     BOOLEAN Initialized;
     PVOID Address;
@@ -151,10 +375,46 @@ typedef struct _HAL_FRAMEBUFFER_DATA
     UINT Width;
     UINT Height;
     UINT PixelsPerScanLine;
-    UINT BitsPerPixel;
+    UINT BytesPerPixel;
     UINT Pitch;
     PVOID Font;
-} HAL_FRAMEBUFFER_DATA, *PHAL_FRAMEBUFFER_DATA;
+    struct
+    {
+        USHORT BlueShift;
+        USHORT BlueSize;
+        USHORT GreenShift;
+        USHORT GreenSize;
+        USHORT RedShift;
+        USHORT RedSize;
+        USHORT ReservedShift;
+        USHORT ReservedSize;
+    } Pixels;
+} HL_FRAMEBUFFER_DATA, *PHL_FRAMEBUFFER_DATA;
+
+/* Scroll region data structure */
+typedef struct _HL_SCROLL_REGION_DATA
+{
+    ULONG Left;
+    ULONG Top;
+    ULONG Right;
+    ULONG Bottom;
+    ULONG WidthInChars;
+    ULONG HeightInChars;
+    ULONG CursorX;
+    ULONG CursorY;
+    ULONG BackgroundColor;
+    ULONG TextColor;
+} HL_SCROLL_REGION_DATA, *PHL_SCROLL_REGION_DATA;
+
+/* Processor identity structure */
+typedef struct _PROCESSOR_IDENTITY
+{
+    ULONG AcpiId;
+    ULONG ApicId;
+    USHORT CpuNumber;
+    BOOLEAN Bsp;
+    BOOLEAN Started;
+} PROCESSOR_IDENTITY, *PPROCESSOR_IDENTITY;
 
 /* SMBIOS table header structure */
 typedef struct _SMBIOS_TABLE_HEADER

sdk/xtdk/i686/artypes.h

@@ -39,13 +39,22 @@
 #define CR4_PCE                                         0x00000100
 #define CR4_FXSR                                        0x00000200
 #define CR4_XMMEXCPT                                    0x00000400
+#define CR4_UMIP                                        0x00000800
 #define CR4_LA57                                        0x00001000
-#define CR4_RESERVED1                                   0x00001800
 #define CR4_VMXE                                        0x00002000
 #define CR4_SMXE                                        0x00004000
-#define CR4_RESERVED2                                   0x00018000
-#define CR4_XSAVE                                       0x00020000
-#define CR4_RESERVED3                                   0xFFFC0000
+#define CR4_FSGSBASE                                    0x00010000
+#define CR4_PCIDE                                       0x00020000
+#define CR4_XSAVE                                       0x00040000
+#define CR4_KL                                          0x00080000
+#define CR4_SMEP                                        0x00100000
+#define CR4_SMAP                                        0x00200000
+#define CR4_PKE                                         0x00400000
+#define CR4_CET                                         0x00800000
+#define CR4_PKS                                         0x01000000
+#define CR4_UINTR                                       0x02000000
+#define CR4_LASS                                        0x08000000
+#define CR4_LAM_SUP                                     0x10000000
 
 /* Descriptors size */
 #define GDT_ENTRIES                                     128
@@ -62,6 +71,27 @@
 #define SEGMENT_FS                                      0x64
 #define SEGMENT_GS                                      0x65
 
+/* X86 EFLAG bit masks definitions */
+#define X86_EFLAGS_NF_MASK                              0x00000000 /* None */
+#define X86_EFLAGS_CF_MASK                              0x00000001 /* Carry */
+#define X86_EFLAGS_PF_MASK                              0x00000004 /* Parity */
+#define X86_EFALGS_AF_MASK                              0x00000010 /* Aux Carry */
+#define X86_EFLAGS_ZF_MASK                              0x00000040 /* Zero */
+#define X86_EFLAGS_SF_MASK                              0x00000080 /* Sign */
+#define X86_EFLAGS_TF_MASK                              0x00000100 /* Trap */
+#define X86_EFLAGS_IF_MASK                              0x00000200 /* Interrupt */
+#define X86_EFLAGS_DF_MASK                              0x00000400 /* Direction */
+#define X86_EFLAGS_OF_MASK                              0x00000800 /* Overflow */
+#define X86_EFLAGS_IOPL_MASK                            0x00003000 /* I/O Privilege */
+#define X86_EFLAGS_NT_MASK                              0x00004000 /* Nested Task */
+#define X86_EFLAGS_SIGN_MASK                            0x00008000 /* Sign */
+#define X86_EFLAGS_RF_MASK                              0x00010000 /* Resume */
+#define X86_EFLAGS_V86_MASK                             0x00020000 /* Virtual 8086 */
+#define X86_EFLAGS_AC_MASK                              0x00040000 /* Alignment Check */
+#define X86_EFLAGS_VIF_MASK                             0x00080000 /* Virtual Interrupt */
+#define X86_EFLAGS_VIP_MASK                             0x00100000 /* Virtual Interrupt Pending */
+#define X86_EFLAGS_ID_MASK                              0x00200000 /* Identification */
+
 /* CPU vendor enumeration list */
 typedef enum _CPU_VENDOR
 {
@@ -70,79 +100,255 @@ typedef enum _CPU_VENDOR
     CPU_VENDOR_UNKNOWN = 0xFFFFFFFF
 } CPU_VENDOR, *PCPU_VENDOR;
 
-/* CPUID features enumeration list */
-typedef enum _CPUID_FEATURES
+/* CPUID extended features (0x80000001) enumeration list */
+typedef enum _CPUID_FEATURES_EXTENDED
 {
-    CPUID_FEATURES_ECX_SSE3         = 1 << 0,
-    CPUID_FEATURES_ECX_PCLMUL       = 1 << 1,
-    CPUID_FEATURES_ECX_DTES64       = 1 << 2,
-    CPUID_FEATURES_ECX_MONITOR      = 1 << 3,
-    CPUID_FEATURES_ECX_DS_CPL       = 1 << 4,
-    CPUID_FEATURES_ECX_VMX          = 1 << 5,
-    CPUID_FEATURES_ECX_SMX          = 1 << 6,
-    CPUID_FEATURES_ECX_EST          = 1 << 7,
-    CPUID_FEATURES_ECX_TM2          = 1 << 8,
-    CPUID_FEATURES_ECX_SSSE3        = 1 << 9,
-    CPUID_FEATURES_ECX_CID          = 1 << 10,
-    CPUID_FEATURES_ECX_SDBG         = 1 << 11,
-    CPUID_FEATURES_ECX_FMA          = 1 << 12,
-    CPUID_FEATURES_ECX_CX16         = 1 << 13,
-    CPUID_FEATURES_ECX_XTPR         = 1 << 14,
-    CPUID_FEATURES_ECX_PDCM         = 1 << 15,
-    CPUID_FEATURES_ECX_PCID         = 1 << 17,
-    CPUID_FEATURES_ECX_DCA          = 1 << 18,
-    CPUID_FEATURES_ECX_SSE4_1       = 1 << 19,
-    CPUID_FEATURES_ECX_SSE4_2       = 1 << 20,
-    CPUID_FEATURES_ECX_X2APIC       = 1 << 21,
-    CPUID_FEATURES_ECX_MOVBE        = 1 << 22,
-    CPUID_FEATURES_ECX_POPCNT       = 1 << 23,
-    CPUID_FEATURES_ECX_TSC          = 1 << 24,
-    CPUID_FEATURES_ECX_AES          = 1 << 25,
-    CPUID_FEATURES_ECX_XSAVE        = 1 << 26,
-    CPUID_FEATURES_ECX_OSXSAVE      = 1 << 27,
-    CPUID_FEATURES_ECX_AVX          = 1 << 28,
-    CPUID_FEATURES_ECX_F16C         = 1 << 29,
-    CPUID_FEATURES_ECX_RDRAND       = 1 << 30,
-    CPUID_FEATURES_ECX_HYPERVISOR   = 1 << 31,
-    CPUID_FEATURES_EDX_FPU          = 1 << 0,
-    CPUID_FEATURES_EDX_VME          = 1 << 1,
-    CPUID_FEATURES_EDX_DE           = 1 << 2,
-    CPUID_FEATURES_EDX_PSE          = 1 << 3,
-    CPUID_FEATURES_EDX_TSC          = 1 << 4,
-    CPUID_FEATURES_EDX_MSR          = 1 << 5,
-    CPUID_FEATURES_EDX_PAE          = 1 << 6,
-    CPUID_FEATURES_EDX_MCE          = 1 << 7,
-    CPUID_FEATURES_EDX_CX8          = 1 << 8,
-    CPUID_FEATURES_EDX_APIC         = 1 << 9,
-    CPUID_FEATURES_EDX_SEP          = 1 << 11,
-    CPUID_FEATURES_EDX_MTRR         = 1 << 12,
-    CPUID_FEATURES_EDX_PGE          = 1 << 13,
-    CPUID_FEATURES_EDX_MCA          = 1 << 14,
-    CPUID_FEATURES_EDX_CMOV         = 1 << 15,
-    CPUID_FEATURES_EDX_PAT          = 1 << 16,
-    CPUID_FEATURES_EDX_PSE36        = 1 << 17,
-    CPUID_FEATURES_EDX_PSN          = 1 << 18,
-    CPUID_FEATURES_EDX_CLFLUSH      = 1 << 19,
-    CPUID_FEATURES_EDX_DS           = 1 << 21,
-    CPUID_FEATURES_EDX_ACPI         = 1 << 22,
-    CPUID_FEATURES_EDX_MMX          = 1 << 23,
-    CPUID_FEATURES_EDX_FXSR         = 1 << 24,
-    CPUID_FEATURES_EDX_SSE          = 1 << 25,
-    CPUID_FEATURES_EDX_SSE2         = 1 << 26,
-    CPUID_FEATURES_EDX_SS           = 1 << 27,
-    CPUID_FEATURES_EDX_HTT          = 1 << 28,
-    CPUID_FEATURES_EDX_TM           = 1 << 29,
-    CPUID_FEATURES_EDX_IA64         = 1 << 30,
-    CPUID_FEATURES_EDX_PBE          = 1 << 31
-} CPUID_FEATURES, *PCPUID_FEATURES;
+    CPUID_FEATURES_ECX_LAHF_SAHF              = 1 << 0,
+    CPUID_FEATURES_ECX_CMP_LEGACY             = 1 << 1,
+    CPUID_FEATURES_ECX_SVM                    = 1 << 2,
+    CPUID_FEATURES_ECX_EXT_APIC_SPACE         = 1 << 3,
+    CPUID_FEATURES_ECX_ALT_MOV_CR8            = 1 << 4,
+    CPUID_FEATURES_ECX_LZCNT                  = 1 << 5,
+    CPUID_FEATURES_ECX_SSE4A                  = 1 << 6,
+    CPUID_FEATURES_ECX_MISALIGNED_SSE         = 1 << 7,
+    CPUID_FEATURES_ECX_PREFETCHW              = 1 << 8,
+    CPUID_FEATURES_ECX_OSVW                   = 1 << 9,
+    CPUID_FEATURES_ECX_IBS                    = 1 << 10,
+    CPUID_FEATURES_ECX_XOP                    = 1 << 11,
+    CPUID_FEATURES_ECX_SKINIT                 = 1 << 12,
+    CPUID_FEATURES_ECX_WDT                    = 1 << 13,
+    CPUID_FEATURES_ECX_LWP                    = 1 << 15,
+    CPUID_FEATURES_ECX_FMA4                   = 1 << 16,
+    CPUID_FEATURES_ECX_TCE                    = 1 << 17,
+    CPUID_FEATURES_ECX_NODEID                 = 1 << 19,
+    CPUID_FEATURES_ECX_TBM                    = 1 << 21,
+    CPUID_FEATURES_ECX_TOPOLOGY_EXTENSIONS    = 1 << 22,
+    CPUID_FEATURES_ECX_PERFCTR_EXT_CORE       = 1 << 23,
+    CPUID_FEATURES_ECX_PERFCTR_EXT_NB         = 1 << 24,
+    CPUID_FEATURES_ECX_DATA_BREAKPOINT_EXT    = 1 << 26,
+    CPUID_FEATURES_ECX_PERF_TSC               = 1 << 27,
+    CPUID_FEATURES_ECX_PERFCTR_EXT_L2I        = 1 << 28,
+    CPUID_FEATURES_ECX_MONITORX_MWAITX        = 1 << 29,
+    CPUID_FEATURES_ECX_CODEBP_ADDRMASK_EXT    = 1 << 30,
+    CPUID_FEATURES_EDX_SYSCALL_SYSRET         = 1 << 11,
+    CPUID_FEATURES_EDX_NX                     = 1 << 20,
+    CPUID_FEATURES_EDX_AMD_MMX_EXT            = 1 << 22,
+    CPUID_FEATURES_EDX_FFXSR                  = 1 << 25,
+    CPUID_FEATURES_EDX_1G_PAGES               = 1 << 26,
+    CPUID_FEATURES_EDX_RDTSCP                 = 1 << 27,
+    CPUID_FEATURES_EDX_LONG_MODE              = 1 << 29,
+    CPUID_FEATURES_EDX_3DNOW_EXT              = 1 << 30,
+    CPUID_FEATURES_EDX_3DNOW                  = 1 << 31
+} CPUID_FEATURES_EXTENDED, *PCPUID_FEATURES_EXTENDED;
+
+/* CPUID STD1 features (0x00000001) enumeration list */
+typedef enum _CPUID_FEATURES_STANDARD1
+{
+    CPUID_FEATURES_ECX_SSE3                   = 1 << 0,
+    CPUID_FEATURES_ECX_PCLMUL                 = 1 << 1,
+    CPUID_FEATURES_ECX_DTES64                 = 1 << 2,
+    CPUID_FEATURES_ECX_MONITOR                = 1 << 3,
+    CPUID_FEATURES_ECX_DS_CPL                 = 1 << 4,
+    CPUID_FEATURES_ECX_VMX                    = 1 << 5,
+    CPUID_FEATURES_ECX_SMX                    = 1 << 6,
+    CPUID_FEATURES_ECX_EST                    = 1 << 7,
+    CPUID_FEATURES_ECX_TM2                    = 1 << 8,
+    CPUID_FEATURES_ECX_SSSE3                  = 1 << 9,
+    CPUID_FEATURES_ECX_CID                    = 1 << 10,
+    CPUID_FEATURES_ECX_SDBG                   = 1 << 11,
+    CPUID_FEATURES_ECX_FMA                    = 1 << 12,
+    CPUID_FEATURES_ECX_CX16                   = 1 << 13,
+    CPUID_FEATURES_ECX_XTPR                   = 1 << 14,
+    CPUID_FEATURES_ECX_PDCM                   = 1 << 15,
+    CPUID_FEATURES_ECX_PCID                   = 1 << 17,
+    CPUID_FEATURES_ECX_DCA                    = 1 << 18,
+    CPUID_FEATURES_ECX_SSE4_1                 = 1 << 19,
+    CPUID_FEATURES_ECX_SSE4_2                 = 1 << 20,
+    CPUID_FEATURES_ECX_X2APIC                 = 1 << 21,
+    CPUID_FEATURES_ECX_MOVBE                  = 1 << 22,
+    CPUID_FEATURES_ECX_POPCNT                 = 1 << 23,
+    CPUID_FEATURES_ECX_TSC                    = 1 << 24,
+    CPUID_FEATURES_ECX_AES                    = 1 << 25,
+    CPUID_FEATURES_ECX_XSAVE                  = 1 << 26,
+    CPUID_FEATURES_ECX_OSXSAVE                = 1 << 27,
+    CPUID_FEATURES_ECX_AVX                    = 1 << 28,
+    CPUID_FEATURES_ECX_F16C                   = 1 << 29,
+    CPUID_FEATURES_ECX_RDRAND                 = 1 << 30,
+    CPUID_FEATURES_ECX_HYPERVISOR             = 1 << 31,
+    CPUID_FEATURES_EDX_FPU                    = 1 << 0,
+    CPUID_FEATURES_EDX_VME                    = 1 << 1,
+    CPUID_FEATURES_EDX_DE                     = 1 << 2,
+    CPUID_FEATURES_EDX_PSE                    = 1 << 3,
+    CPUID_FEATURES_EDX_TSC                    = 1 << 4,
+    CPUID_FEATURES_EDX_MSR                    = 1 << 5,
+    CPUID_FEATURES_EDX_PAE                    = 1 << 6,
+    CPUID_FEATURES_EDX_MCE                    = 1 << 7,
+    CPUID_FEATURES_EDX_CX8                    = 1 << 8,
+    CPUID_FEATURES_EDX_APIC                   = 1 << 9,
+    CPUID_FEATURES_EDX_SEP                    = 1 << 11,
+    CPUID_FEATURES_EDX_MTRR                   = 1 << 12,
+    CPUID_FEATURES_EDX_PGE                    = 1 << 13,
+    CPUID_FEATURES_EDX_MCA                    = 1 << 14,
+    CPUID_FEATURES_EDX_CMOV                   = 1 << 15,
+    CPUID_FEATURES_EDX_PAT                    = 1 << 16,
+    CPUID_FEATURES_EDX_PSE36                  = 1 << 17,
+    CPUID_FEATURES_EDX_PSN                    = 1 << 18,
+    CPUID_FEATURES_EDX_CLFLUSH                = 1 << 19,
+    CPUID_FEATURES_EDX_DS                     = 1 << 21,
+    CPUID_FEATURES_EDX_ACPI                   = 1 << 22,
+    CPUID_FEATURES_EDX_MMX                    = 1 << 23,
+    CPUID_FEATURES_EDX_FXSR                   = 1 << 24,
+    CPUID_FEATURES_EDX_SSE                    = 1 << 25,
+    CPUID_FEATURES_EDX_SSE2                   = 1 << 26,
+    CPUID_FEATURES_EDX_SS                     = 1 << 27,
+    CPUID_FEATURES_EDX_HTT                    = 1 << 28,
+    CPUID_FEATURES_EDX_TM                     = 1 << 29,
+    CPUID_FEATURES_EDX_IA64                   = 1 << 30,
+    CPUID_FEATURES_EDX_PBE                    = 1 << 31
+} CPUID_FEATURES_STANDARD1, *PCPUID_FEATURES_STANDARD1;
+
+/* CPUID STD7 features (0x00000007, subleaf 0) enumeration list */
+typedef enum _CPUID_FEATURES_STANDARD7_LEAF0
+{
+    CPUID_FEATURES_EBX_FSGSBASE               = 1 << 0,
+    CPUID_FEATURES_EBX_TSC_ADJUST             = 1 << 1,
+    CPUID_FEATURES_EBX_SGX                    = 1 << 2,
+    CPUID_FEATURES_EBX_BMI1                   = 1 << 3,
+    CPUID_FEATURES_EBX_HLE                    = 1 << 4,
+    CPUID_FEATURES_EBX_AVX2                   = 1 << 5,
+    CPUID_FEATURES_EBX_FDP_DEPRECATION        = 1 << 6,
+    CPUID_FEATURES_EBX_SMEP                   = 1 << 7,
+    CPUID_FEATURES_EBX_BMI2                   = 1 << 8,
+    CPUID_FEATURES_EBX_ERMS                   = 1 << 9,
+    CPUID_FEATURES_EBX_INVPCID                = 1 << 10,
+    CPUID_FEATURES_EBX_RTM                    = 1 << 11,
+    CPUID_FEATURES_EBX_QOS_MONITORING         = 1 << 12,
+    CPUID_FEATURES_EBX_DEPRECATE_FCS_FDS      = 1 << 13,
+    CPUID_FEATURES_EBX_MPX                    = 1 << 14,
+    CPUID_FEATURES_EBX_QOS_ENFORCEMENT        = 1 << 15,
+    CPUID_FEATURES_EBX_AVX512F                = 1 << 16,
+    CPUID_FEATURES_EBX_AVX512DQ               = 1 << 17,
+    CPUID_FEATURES_EBX_RDSEED                 = 1 << 18,
+    CPUID_FEATURES_EBX_ADX                    = 1 << 19,
+    CPUID_FEATURES_EBX_SMAP                   = 1 << 20,
+    CPUID_FEATURES_EBX_AVX512IFMA52           = 1 << 21,
+    CPUID_FEATURES_EBX_CLFLUSHOPT             = 1 << 23,
+    CPUID_FEATURES_EBX_CLWB                   = 1 << 24,
+    CPUID_FEATURES_EBX_PROCESSOR_TRACE        = 1 << 25,
+    CPUID_FEATURES_EBX_AVX512PF               = 1 << 26,
+    CPUID_FEATURES_EBX_AVX512ER               = 1 << 27,
+    CPUID_FEATURES_EBX_AVX512CD               = 1 << 28,
+    CPUID_FEATURES_EBX_SHA                    = 1 << 29,
+    CPUID_FEATURES_EBX_AVX512BW               = 1 << 30,
+    CPUID_FEATURES_EBX_AVX512VL               = 1 << 31,
+    CPUID_FEATURES_ECX_PREFETCHWT1            = 1 << 0,
+    CPUID_FEATURES_ECX_AVX512_VBMI            = 1 << 1,
+    CPUID_FEATURES_ECX_UMIP                   = 1 << 2,
+    CPUID_FEATURES_ECX_PKU                    = 1 << 3,
+    CPUID_FEATURES_ECX_OSPKE                  = 1 << 4,
+    CPUID_FEATURES_ECX_WAITPKG                = 1 << 5,
+    CPUID_FEATURES_ECX_AVX512_VBMI2           = 1 << 6,
+    CPUID_FEATURES_ECX_CET_SS                 = 1 << 7,
+    CPUID_FEATURES_ECX_GFNI                   = 1 << 8,
+    CPUID_FEATURES_ECX_VAES                   = 1 << 9,
+    CPUID_FEATURES_ECX_VPCLMULQDQ             = 1 << 10,
+    CPUID_FEATURES_ECX_AVX512_VNNI            = 1 << 11,
+    CPUID_FEATURES_ECX_AVX512_BITALG          = 1 << 12,
+    CPUID_FEATURES_ECX_TME                    = 1 << 13,
+    CPUID_FEATURES_ECX_AVX512_VPOPCNTDQ       = 1 << 14,
+    CPUID_FEATURES_ECX_LA57                   = 1 << 16,
+    CPUID_FEATURES_ECX_RDPID                  = 1 << 22,
+    CPUID_FEATURES_ECX_KEYLOCKER              = 1 << 23,
+    CPUID_FEATURES_ECX_BUS_LOCK_DETECT        = 1 << 24,
+    CPUID_FEATURES_ECX_CLDEMOTE               = 1 << 25,
+    CPUID_FEATURES_ECX_MOVDIRI                = 1 << 27,
+    CPUID_FEATURES_ECX_MOVDIR64B              = 1 << 28,
+    CPUID_FEATURES_ECX_ENQCMD                 = 1 << 29,
+    CPUID_FEATURES_ECX_SGX_LAUNCH_CONFIG      = 1 << 30,
+    CPUID_FEATURES_ECX_PKS                    = 1 << 31,
+    CPUID_FEATURES_EDX_SGX_KEYS               = 1 << 1,
+    CPUID_FEATURES_EDX_AVX512_4VNNIW          = 1 << 2,
+    CPUID_FEATURES_EDX_AVX512_4FMAPS          = 1 << 3,
+    CPUID_FEATURES_EDX_FAST_SHORT_REP_MOV     = 1 << 4,
+    CPUID_FEATURES_EDX_UINTR                  = 1 << 5,
+    CPUID_FEATURES_EDX_AVX512_VPINTERSECT     = 1 << 8,
+    CPUID_FEATURES_EDX_SRBDS_CTRL             = 1 << 9,
+    CPUID_FEATURES_EDX_MD_CLEAR               = 1 << 10,
+    CPUID_FEATURES_EDX_RTM_ALWAYS_ABORT       = 1 << 11,
+    CPUID_FEATURES_EDX_RTM_FORCE_ABORT        = 1 << 13,
+    CPUID_FEATURES_EDX_SERIALIZE              = 1 << 14,
+    CPUID_FEATURES_EDX_HYBRID                 = 1 << 15,
+    CPUID_FEATURES_EDX_TSXLDTRK               = 1 << 16,
+    CPUID_FEATURES_EDX_PCONFIG                = 1 << 18,
+    CPUID_FEATURES_EDX_ARCH_LBR               = 1 << 19,
+    CPUID_FEATURES_EDX_CET_IBT                = 1 << 20,
+    CPUID_FEATURES_EDX_AMX_BF16               = 1 << 22,
+    CPUID_FEATURES_EDX_AVX512_FP16            = 1 << 23,
+    CPUID_FEATURES_EDX_AMX_TILE               = 1 << 24,
+    CPUID_FEATURES_EDX_AMX_INT8               = 1 << 25,
+    CPUID_FEATURES_EDX_SCA_IBRS_IBPB          = 1 << 26,
+    CPUID_FEATURES_EDX_SCA_STIBP              = 1 << 27,
+    CPUID_FEATURES_EDX_L1D_FLUSH              = 1 << 28,
+    CPUID_FEATURES_EDX_ARCH_CAPABILITIES_MSR  = 1 << 29,
+    CPUID_FEATURES_EDX_CORE_CAPABILITIES_MSR  = 1 << 30,
+    CPUID_FEATURES_EDX_SCA_SSBD               = 1 << 31
+} CPUID_FEATURES_STANDARD7_LEAF0, *PCPUID_FEATURES_STANDARD7_LEAF0;
+
+/* CPUID STD7 features (0x00000007, subleaf 1) enumeration list */
+typedef enum _CPUID_FEATURES_STANDARD7_LEAF1
+{
+    CPUID_FEATURES_EAX_SHA512                 = 1 << 0,
+    CPUID_FEATURES_EAX_SM3                    = 1 << 1,
+    CPUID_FEATURES_EAX_SM4                    = 1 << 2,
+    CPUID_FEATURES_EAX_RAO_INT                = 1 << 3,
+    CPUID_FEATURES_EAX_AVX_VNNI               = 1 << 4,
+    CPUID_FEATURES_EAX_AVX512_BF16            = 1 << 5,
+    CPUID_FEATURES_EAX_LASS                   = 1 << 6,
+    CPUID_FEATURES_EAX_CMPCCXADD              = 1 << 7,
+    CPUID_FEATURES_EAX_ARCH_PERFMON           = 1 << 8,
+    CPUID_FEATURES_EAX_FAST_ZEROLEN_REP_MOVSB = 1 << 10,
+    CPUID_FEATURES_EAX_FAST_ZEROLEN_REP_STOSB = 1 << 11,
+    CPUID_FEATURES_EAX_FAST_ZEROLEN_REP_CMPSB = 1 << 12,
+    CPUID_FEATURES_EAX_FRED                   = 1 << 17,
+    CPUID_FEATURES_EAX_LKGS                   = 1 << 18,
+    CPUID_FEATURES_EAX_WRMSRNS                = 1 << 19,
+    CPUID_FEATURES_EAX_NMI_SOURCE_REPORTING   = 1 << 20,
+    CPUID_FEATURES_EAX_AMX_FP16               = 1 << 21,
+    CPUID_FEATURES_EAX_HRESET                 = 1 << 22,
+    CPUID_FEATURES_EAX_AVX_IFMA               = 1 << 23,
+    CPUID_FEATURES_EAX_LAM                    = 1 << 26,
+    CPUID_FEATURES_EAX_MSRLIST                = 1 << 27,
+    CPUID_FEATURES_EAX_INVD_DISABLE           = 1 << 30,
+    CPUID_FEATURES_EAX_MOVRS                  = 1 << 31,
+    CPUID_FEATURES_EBX_PPIN                   = 1 << 0,
+    CPUID_FEATURES_EBX_TSE                    = 1 << 1,
+    CPUID_FEATURES_EBX_CPUIDMAXVAL_LIM_RMV    = 1 << 3,
+    CPUID_FEATURES_ECX_MSR_IMM                = 1 << 5,
+    CPUID_FEATURES_EDX_AVX_VNNI_INT8          = 1 << 4,
+    CPUID_FEATURES_EDX_AVX_NE_CONVERT         = 1 << 5,
+    CPUID_FEATURES_EDX_AMX_COMPLEX            = 1 << 8,
+    CPUID_FEATURES_EDX_AVX_VNNI_INT16         = 1 << 10,
+    CPUID_FEATURES_EDX_USER_TIMER             = 1 << 13,
+    CPUID_FEATURES_EDX_PREFETCHI              = 1 << 14,
+    CPUID_FEATURES_EDX_USER_MSR               = 1 << 15,
+    CPUID_FEATURES_EDX_UIRET_UIF              = 1 << 17,
+    CPUID_FEATURES_EDX_CET_SSS                = 1 << 18,
+    CPUID_FEATURES_EDX_AVX10                  = 1 << 19,
+    CPUID_FEATURES_EDX_APX                    = 1 << 21,
+    CPUID_FEATURES_EDX_MWAIT_AND_LEAF5        = 1 << 23
+} CPUID_FEATURES_STANDARD7_LEAF1, *PCPUID_FEATURES_STANDARD7_LEAF1;
 
 /* CPUID requests */
 typedef enum _CPUID_REQUESTS
 {
     CPUID_GET_VENDOR_STRING,
-    CPUID_GET_CPU_FEATURES,
-    CPUID_GET_TLB,
-    CPUID_GET_SERIAL
+    CPUID_GET_STANDARD1_FEATURES,
+    CPUID_GET_TLB_CACHE,
+    CPUID_GET_SERIAL,
+    CPUID_GET_CACHE_TOPOLOGY,
+    CPUID_GET_MONITOR_MWAIT,
+    CPUID_GET_POWER_MANAGEMENT,
+    CPUID_GET_STANDARD7_FEATURES
 } CPUID_REQUESTS, *PCPUID_REQUESTS;
 
 /* Processor identification information */

sdk/xtdk/i686/hltypes.h

@@ -47,20 +47,17 @@
 #define APIC_DF_FLAT                                    0xFFFFFFFF
 #define APIC_DF_CLUSTER                                 0x0FFFFFFF
 
-/* APIC delivery modes */
-#define APIC_DM_FIXED                                   0
-#define APIC_DM_LOWPRIO                                 1
-#define APIC_DM_SMI                                     2
-#define APIC_DM_REMOTE                                  3
-#define APIC_DM_NMI                                     4
-#define APIC_DM_INIT                                    5
-#define APIC_DM_STARTUP                                 6
-#define APIC_DM_EXTINT                                  7
-
 /* APIC trigger modes */
 #define APIC_TGM_EDGE                                   0
 #define APIC_TGM_LEVEL                                  1
 
+/* APIC LDR (Logical Destination Register) shifts */
+#define APIC_X2APIC_LDR_SHIFT                           16
+#define APIC_XAPIC_LDR_SHIFT                            24
+
+/* Maximum number of I/O APICs */
+#define APIC_MAX_IOAPICS                                64
+
 /* 8259/ISP PIC ports definitions */
 #define PIC1_CONTROL_PORT                               0x20
 #define PIC1_DATA_PORT                                  0x21
@@ -72,12 +69,123 @@
 /* PIC vector definitions */
 #define PIC1_VECTOR_SPURIOUS                            0x37
 
-/* Serial port I/O addresses */
-#define COMPORT_ADDRESSES                               {0x000, 0x3F8, 0x2F8, 0x3E8, 0x2E8, 0x5F8, 0x4F8, 0x5E8, 0x4E8}
+/* Serial ports information */
+#define COMPORT_ADDRESS                                 {0x3F8, 0x2F8, 0x3E8, 0x2E8, 0x5F8, 0x4F8, 0x5E8, 0x4E8}
+#define COMPORT_COUNT                                   8
 
 /* Initial stall factor */
 #define INITIAL_STALL_FACTOR                            100
 
+/* APIC delivery mode enumeration list */
+typedef enum _APIC_DM
+{
+    APIC_DM_FIXED,
+    APIC_DM_LOWPRIO,
+    APIC_DM_SMI,
+    APIC_DM_REMOTE,
+    APIC_DM_NMI,
+    APIC_DM_INIT,
+    APIC_DM_STARTUP,
+    APIC_DM_EXTINT,
+} APIC_DM, *PAPIC_DM;
+
+/* APIC destination short-hand enumeration list */
+typedef enum _APIC_DSH
+{
+    APIC_DSH_Destination,
+    APIC_DSH_Self,
+    APIC_DSH_AllIncludingSelf,
+    APIC_DSH_AllExclusingSelf
+} APIC_DSH, *PAPIC_DSH;
+
+/* APIC mode list */
+typedef enum _APIC_MODE
+{
+    APIC_MODE_COMPAT,
+    APIC_MODE_X2APIC
+} APIC_MODE, *PAPIC_MODE;
+
+/* APIC Register Address Map */
+typedef enum _APIC_REGISTER
+{
+    APIC_ID       = 0x02, /* APIC ID Register */
+    APIC_VER      = 0x03, /* APIC Version Register */
+    APIC_TPR      = 0x08, /* Task Priority Register */
+    APIC_APR      = 0x09, /* Arbitration Priority Register */
+    APIC_PPR      = 0x0A, /* Processor Priority Register (R) */
+    APIC_EOI      = 0x0B, /* EOI Register */
+    APIC_RRR      = 0x0C, /* Remote Read Register */
+    APIC_LDR      = 0x0D, /* Logical Destination Register */
+    APIC_DFR      = 0x0E, /* Destination Format Register (not available in extended mode) */
+    APIC_SIVR     = 0x0F, /* Spurious Interrupt Vector Register */
+    APIC_ISR      = 0x10, /* Interrupt Service Register*/
+    APIC_TMR      = 0x18, /* Trigger Mode Register */
+    APIC_IRR      = 0x20, /* Interrupt Request Register */
+    APIC_ESR      = 0x28, /* Error Status Register */
+    APIC_ICR0     = 0x30, /* Interrupt Command Register */
+    APIC_ICR1     = 0x31, /* Interrupt Command Register (not available in extended mode) */
+    APIC_TMRLVTR  = 0x32, /* Timer Local Vector Table */
+    APIC_THRMLVTR = 0x33, /* Thermal Local Vector Table */
+    APIC_PCLVTR   = 0x34, /* Performance Counter Local Vector Table */
+    APIC_LINT0    = 0x35, /* LINT0 Local Vector Table */
+    APIC_LINT1    = 0x36, /* LINT1 Local Vector Table */
+    APIC_ERRLVTR  = 0x37, /* Error Local Vector Table */
+    APIC_TICR     = 0x38, /* Initial Count Register for Timer */
+    APIC_TCCR     = 0x39, /* Current Count Register for Timer */
+    APIC_TDCR     = 0x3E, /* Timer Divide Configuration Register */
+    APIC_EAFR     = 0x40, /* extended APIC Feature register */
+    APIC_EACR     = 0x41, /* Extended APIC Control Register */
+    APIC_SEOI     = 0x42, /* Specific End Of Interrupt Register */
+    APIC_EXT0LVTR = 0x50, /* Extended Interrupt 0 Local Vector Table */
+    APIC_EXT1LVTR = 0x51, /* Extended Interrupt 1 Local Vector Table */
+    APIC_EXT2LVTR = 0x52, /* Extended Interrupt 2 Local Vector Table */
+    APIC_EXT3LVTR = 0x53  /* Extended Interrupt 3 Local Vector Table */
+} APIC_REGISTER, *PAPIC_REGISTER;
+
+/* I8259 PIC interrupt mode enumeration list */
+typedef enum _PIC_I8259_ICW1_INTERRUPT_MODE
+{
+    EdgeTriggered,
+    LevelTriggered
+} PIC_I8259_ICW1_INTERRUPT_MODE, *PPIC_I8259_ICW1_INTERRUPT_MODE;
+
+/* I8259 PIC interval enumeration list */
+typedef enum _PIC_I8259_ICW1_INTERVAL
+{
+    Interval8,
+    Interval4
+} PIC_I8259_ICW1_INTERVAL, *PPIC_I8259_ICW1_INTERVAL;
+
+/* I8259 PIC operating mode enumeration list */
+typedef enum _PIC_I8259_ICW1_OPERATING_MODE
+{
+    Cascade,
+    Single
+} PIC_I8259_ICW1_OPERATING_MODE, *PPIC_I8259_ICW1_OPERATING_MODE;
+
+/* I8259 PIC buffered mode enumeration list */
+typedef enum _PIC_I8259_ICW4_BUFFERED_MODE
+{
+    NonBuffered,
+    NonBuffered2,
+    BufferedSlave,
+    BufferedMaster
+} PIC_I8259_ICW4_BUFFERED_MODE, *PPIC_I8259_ICW4_BUFFERED_MODE;
+
+/* I8259 PIC End Of Interrupt (EOI) mode enumeration list */
+typedef enum _PIC_I8259_ICW4_EOI_MODE
+{
+    NormalEoi,
+    AutomaticEoi
+} PIC_I8259_ICW4_EOI_MODE, *PPIC_I8259_ICW4_EOI_MODE;
+
+/* I8259 PIC system mode enumeration list */
+typedef enum _PIC_I8259_ICW4_SYSTEM_MODE
+{
+    Mcs8085Mode,
+    New8086Mode
+} PIC_I8259_ICW4_SYSTEM_MODE, *PPIC_I8259_ICW4_SYSTEM_MODE;
+
 /* APIC Base Register */
 typedef union _APIC_BASE_REGISTER
 {
@@ -94,6 +202,31 @@ typedef union _APIC_BASE_REGISTER
     };
 } APIC_BASE_REGISTER, *PAPIC_BASE_REGISTER;
 
+/* APIC Command Register */
+typedef union _APIC_COMMAND_REGISTER
+{
+    ULONGLONG LongLong;
+    struct
+    {
+        ULONG Long0;
+        ULONG Long1;
+    };
+    struct
+    {
+        ULONGLONG Vector:8;
+        ULONGLONG DeliveryMode:3;
+        ULONGLONG DestinationMode:1;
+        ULONGLONG DeliveryStatus:1;
+        ULONGLONG ReservedMBZ:1;
+        ULONGLONG Level:1;
+        ULONGLONG TriggerMode:1;
+        ULONGLONG RemoteReadStatus:2;
+        ULONGLONG DestinationShortHand:2;
+        ULONGLONG Reserved2MBZ:36;
+        ULONGLONG Destination:8;
+    };
+} APIC_COMMAND_REGISTER, *PAPIC_COMMAND_REGISTER;
+
 /* APIC Local Vector Table (LVT) Register */
 typedef union _APIC_LVT_REGISTER
 {
@@ -101,7 +234,7 @@ typedef union _APIC_LVT_REGISTER
     struct
     {
         ULONG Vector:8;
-        ULONG MessageType:3;
+        ULONG DeliveryMode:3;
         ULONG Reserved1:1;
         ULONG DeliveryStatus:1;
         ULONG Reserved2:1;
@@ -126,14 +259,69 @@ typedef union _APIC_SPURIOUS_REGISTER
     };
 } APIC_SPURIOUS_REGISTER, *PAPIC_SPURIOUS_REGISTER;
 
-/* Processor identity structure */
-typedef struct _HAL_PROCESSOR_IDENTITY
+/* I8259 PIC register structure */
+typedef union _PIC_I8259_ICW1
 {
-    UCHAR ProcessorId;
-    UCHAR LApicId;
-    BOOLEAN Bsp;
-    BOOLEAN Started;
-    PKPROCESSOR_BLOCK ProcessorBlock;
-} HAL_PROCESSOR_IDENTITY, *PHAL_PROCESSOR_IDENTITY;
+    struct
+    {
+        UCHAR NeedIcw4:1;
+        UCHAR OperatingMode:1;
+        UCHAR Interval:1;
+        UCHAR InterruptMode:1;
+        UCHAR Init:1;
+        UCHAR InterruptVectorAddress:3;
+    };
+    UCHAR Bits;
+} PIC_I8259_ICW1, *PPIC_I8259_ICW1;
+
+/* I8259 PIC register structure */
+typedef union _PIC_I8259_ICW2
+{
+    struct
+    {
+        UCHAR Sbz:3;
+        UCHAR InterruptVector:5;
+    };
+    UCHAR Bits;
+} PIC_I8259_ICW2, *PPIC_I8259_ICW2;
+
+/* I8259 PIC register structure */
+typedef union _PIC_I8259_ICW3
+{
+    union
+    {
+        struct
+        {
+            UCHAR SlaveIrq0:1;
+            UCHAR SlaveIrq1:1;
+            UCHAR SlaveIrq2:1;
+            UCHAR SlaveIrq3:1;
+            UCHAR SlaveIrq4:1;
+            UCHAR SlaveIrq5:1;
+            UCHAR SlaveIrq6:1;
+            UCHAR SlaveIrq7:1;
+        };
+        struct
+        {
+            UCHAR SlaveId:3;
+            UCHAR Reserved:5;
+        };
+    };
+    UCHAR Bits;
+} PIC_I8259_ICW3, *PPIC_I8259_ICW3;
+
+/* I8259 PIC register structure */
+typedef union _PIC_I8259_ICW4
+{
+    struct
+    {
+        UCHAR SystemMode:1;
+        UCHAR EoiMode:1;
+        UCHAR BufferedMode:2;
+        UCHAR SpecialFullyNestedMode:1;
+        UCHAR Reserved:3;
+    };
+    UCHAR Bits;
+} PIC_I8259_ICW4, *PPIC_I8259_ICW4;
 
 #endif /* __XTDK_I686_HLTYPES_H */

sdk/xtdk/i686/ketypes.h

@@ -95,31 +95,6 @@
 #define I686_INTERRUPT_GATE               0xE
 #define I686_TRAP_GATE                    0xF
 
-/* EFlags bits definitions */
-#define EFLAGS_NF_MASK                    0x00000000L /* None */
-#define EFLAGS_CF_MASK                    0x00000001L /* Carry */
-#define EFLAGS_PF_MASK                    0x00000004L /* Parity */
-#define EFLAGS_AF_MASK                    0x00000010L /* Aux Carry */
-#define EFLAGS_ZF_MASK                    0x00000040L /* Zero */
-#define EFLAGS_SF_MASK                    0x00000080L /* Sign */
-#define EFLAGS_TF                         0x00000100L /* Trap */
-#define EFLAGS_INTERRUPT_MASK             0x00000200L /* Interrupt */
-#define EFLAGS_DF_MASK                    0x00000400L /* Direction */
-#define EFLAGS_OF_MASK                    0x00000800L /* Overflow */
-#define EFLAGS_IOPL_MASK                  0x00003000L /* I/O Privilege */
-#define EFLAGS_NT                         0x00004000L /* Nested Task */
-#define EFLAGS_SIGN_MASK                  0x00008000L /* Sign */
-#define EFLAGS_RF                         0x00010000L /* Resume */
-#define EFLAGS_V86_MASK                   0x00020000L /* Virtual 8086 */
-#define EFLAGS_ALIGN_CHECK                0x00040000L /* Alignment Check */
-#define EFLAGS_VIF                        0x00080000L /* Virtual Interrupt */
-#define EFLAGS_VIP                        0x00100000L /* Virtual Interrupt Pending */
-#define EFLAGS_ID_MASK                    0x00200000L /* Identification */
-
-/* EFLAGS sanitize masks */
-#define EFLAGS_KERNELMODE                 0x003F0FD7L
-#define EFLAGS_USERMODE                   0x003F4DD7L
-
 /* Context control flags */
 #define CONTEXT_ARCHITECTURE              0x00010000
 #define CONTEXT_CONTROL                   (CONTEXT_ARCHITECTURE | 0x01)
@@ -168,6 +143,10 @@
 /* XTOS Kernel stack guard pages */
 #define KERNEL_STACK_GUARD_PAGES          1
 
+/* Processor structures size */
+#define KPROCESSOR_STRUCTURES_SIZE        ((2 * KERNEL_STACK_SIZE) + sizeof(ArInitialGdt) + sizeof(ArInitialTss) + \
+                                          sizeof(ArInitialProcessorBlock) + MM_PAGE_SIZE)
+
 /* Kernel frames */
 #define KTRAP_FRAME_ALIGN                 0x08
 #define KTRAP_FRAME_SIZE                  sizeof(KTRAP_FRAME)
@@ -453,7 +432,7 @@ typedef struct _KPROCESSOR_CONTROL_BLOCK
     PKTHREAD CurrentThread;
     PKTHREAD IdleThread;
     PKTHREAD NextThread;
-    UCHAR Number;
+    UCHAR CpuNumber;
     ULONG_PTR SetMember;
     CPU_IDENTIFICATION CpuId;
     KPROCESSOR_STATE ProcessorState;
@@ -470,16 +449,27 @@ typedef struct _KPROCESSOR_CONTROL_BLOCK
 /* Processor Block structure definition */
 typedef struct _KPROCESSOR_BLOCK
 {
-    THREAD_INFORMATION_BLOCK ThreadInformationBlock;
-    PKPROCESSOR_BLOCK Self;
-    PKPROCESSOR_CONTROL_BLOCK CurrentPrcb;
-    KRUNLEVEL RunLevel;
+    union
+    {
+        THREAD_INFORMATION_BLOCK ThreadInformationBlock;
+        struct
+        {
+            PKGDTENTRY GdtBase;
+            PKTSS TssBase;
+            PKPROCESSOR_BLOCK Self;
+            PKPROCESSOR_CONTROL_BLOCK CurrentPrcb;
+        };
+    };
     PKIDTENTRY IdtBase;
-    PKGDTENTRY GdtBase;
-    PKTSS TssBase;
+    KRUNLEVEL RunLevel;
     KPROCESSOR_CONTROL_BLOCK Prcb;
+    ULONG Irr;
+    ULONG IrrActive;
+    ULONG Idr;
     ULONG ContextSwitches;
+    KAFFINITY SetMember;
     ULONG StallScaleFactor;
+    UCHAR CpuNumber;
 } KPROCESSOR_BLOCK, *PKPROCESSOR_BLOCK;
 
 /* Thread Environment Block (TEB) structure definition */

sdk/xtdk/i686/mmtypes.h

@@ -14,16 +14,48 @@
 
 
 /* Pages related definitions */
-#define MM_PAGE_MASK                               0xFFF
+#define MM_PAGE_MASK                               (MM_PAGE_SIZE - 1)
 #define MM_PAGE_SHIFT                              12
 #define MM_PAGE_SIZE                               4096
 
+/* Page directory and page base addresses */
+#define MM_PTE_BASE                                0xC0000000
+#define MM_PDE_BASE                                0xC0600000
+
+/* PTE shift values */
+#define MM_PTE_SHIFT                               3
+#define MM_PTI_SHIFT                               12
+#define MM_PDI_SHIFT                               21
+#define MM_PPI_SHIFT                               30
+
+/* Page directory and page base legacy address */
+#define MM_PDE_LEGACY_BASE                         0xC0300000
+
+/* PTE legacy shift values */
+#define MM_PTE_LEGACY_SHIFT                        2
+#define MM_PDI_LEGACY_SHIFT                        22
+
 /* Minimum number of physical pages needed by the system */
 #define MM_MINIMUM_PHYSICAL_PAGES                  1100
 
 /* Default number of secondary colors */
 #define MM_DEFAULT_SECONDARY_COLORS                64
 
+/* Number of HAL allocation descriptors */
+#define MM_HARDWARE_ALLOCATION_DESCRIPTORS         64
+
+/* Kernel HAL heap initial start address */
+#define MM_HARDWARE_HEAP_START_ADDRESS             ((PVOID)(((ULONG_PTR)MM_HARDWARE_VA_START) + 1024 * 1024))
+
+/* HAL memory pool virtual address start */
+#define MM_HARDWARE_VA_START                       0xFFC00000
+
+/* Maximum physical address used by HAL allocations */
+#define MM_MAXIMUM_PHYSICAL_ADDRESS                0xFFFFFFFF
+
+/* Trampoline code address */
+#define MM_TRAMPOLINE_ADDRESS                      0x80000
+
 /* Page size enumeration list */
 typedef enum _PAGE_SIZE
 {
@@ -32,123 +64,7 @@ typedef enum _PAGE_SIZE
     Size4M
 } PAGE_SIZE, *PPAGE_SIZE;
 
-/* Page Table entry structure definition (with PAE support) */
-typedef struct _HARDWARE_PTE
-{
-    ULONGLONG Valid:1;
-    ULONGLONG Writable:1;
-    ULONGLONG Owner:1;
-    ULONGLONG WriteThrough:1;
-    ULONGLONG CacheDisable:1;
-    ULONGLONG Accessed:1;
-    ULONGLONG Dirty:1;
-    ULONGLONG LargePage:1;
-    ULONGLONG Global:1;
-    ULONGLONG CopyOnWrite:1;
-    ULONGLONG Prototype:1;
-    ULONGLONG Reserved0:1;
-    ULONGLONG PageFrameNumber:26;
-    ULONGLONG Reserved1:14;
-    ULONGLONG SoftwareWsIndex:11;
-    ULONGLONG NoExecute:1;
-} HARDWARE_PTE, *PHARDWARE_PTE;
-
-/* Page Table Entry on PAE enabled system */
-typedef struct _MMPTE_HARDWARE
-{
-    ULONGLONG Valid:1;
-    ULONGLONG Writable:1;
-    ULONGLONG Owner:1;
-    ULONGLONG WriteThrough:1;
-    ULONGLONG CacheDisable:1;
-    ULONGLONG Accessed:1;
-    ULONGLONG Dirty:1;
-    ULONGLONG LargePage:1;
-    ULONGLONG Global:1;
-    ULONGLONG CopyOnWrite:1;
-    ULONGLONG Prototype:1;
-    ULONGLONG Write:1;
-    ULONGLONG PageFrameNumber:26;
-    ULONGLONG Reserved1:25;
-    ULONGLONG NoExecute:1;
-} MMPTE_HARDWARE, *PMMPTE_HARDWARE;
-
-/* Page Table Entry list structure definition (with PAE support) */
-typedef struct _MMPTE_LIST
-{
-    ULONGLONG Valid:1;
-    ULONGLONG OneEntry:1;
-    ULONGLONG Reserved1:8;
-    ULONGLONG Prototype:1;
-    ULONGLONG Reserved2:21;
-    ULONGLONG NextEntry:32;
-} MMPTE_LIST, *PMMPTE_LIST;
-
-/* Page Table Entry subsection structure definition (with PAE support) */
-typedef struct _MMPTE_PROTOTYPE
-{
-    ULONGLONG Valid:1;
-    ULONGLONG Reserved1:7;
-    ULONGLONG ReadOnly:1;
-    ULONGLONG Reserved2:1;
-    ULONGLONG Prototype:1;
-    ULONGLONG Protection:5;
-    ULONGLONG Reserved3:16;
-    ULONGLONG ProtoAddress:32;
-} MMPTE_PROTOTYPE, *PMMPTE_PROTOTYPE;
-
-/* Page Table Entry software structure definition (with PAE support) */
-typedef struct _MMPTE_SOFTWARE
-{
-    ULONGLONG Valid:1;
-    ULONGLONG PageFileLow:4;
-    ULONGLONG Protection:5;
-    ULONGLONG Prototype:1;
-    ULONGLONG Transition:1;
-    ULONGLONG Reserved1:20;
-    ULONGLONG PageFileHigh:32;
-} MMPTE_SOFTWARE, *PMMPTE_SOFTWARE;
-
-/* Page Table Entry subsection structure definition (with PAE support) */
-typedef struct _MMPTE_SUBSECTION
-{
-    ULONGLONG Valid:1;
-    ULONGLONG Reserved1:4;
-    ULONGLONG Protection:5;
-    ULONGLONG Prototype:1;
-    ULONGLONG Reserved2:21;
-    ULONGLONG SubsectionAddress:32;
-} MMPTE_SUBSECTION, *PMMPTE_SUBSECTION;
-
-/* Page Table Entry transition structure definition (with PAE support) */
-typedef struct _MMPTE_TRANSITION
-{
-    ULONGLONG Valid:1;
-    ULONGLONG Write:1;
-    ULONGLONG Owner:1;
-    ULONGLONG WriteThrough:1;
-    ULONGLONG CacheDisable:1;
-    ULONGLONG Protection:5;
-    ULONGLONG Prototype:1;
-    ULONGLONG Transition:1;
-    ULONGLONG PageFrameNumber:26;
-    ULONGLONG Unused:26;
-} MMPTE_TRANSITION, *PMMPTE_TRANSITION;
-
-/* Page Table Entry structure definition (with PAE support) */
-typedef union _MMPTE
-{
-    ULONGLONG Long;
-    HARDWARE_PTE Flush;
-    MMPTE_HARDWARE Hardware;
-    MMPTE_PROTOTYPE Prototype;
-    MMPTE_SOFTWARE Software;
-    MMPTE_TRANSITION Transition;
-    MMPTE_SUBSECTION Subsection;
-    MMPTE_LIST List;
-} MMPTE, *PMMPTE;
-
-/* Legacy Page Table entry structure definition (without PAE support) */
+/* Legacy Page Table entry structure definition (PML2) */
 typedef struct _HARDWARE_LEGACY_PTE
 {
     ULONG Valid:1;
@@ -166,8 +82,47 @@ typedef struct _HARDWARE_LEGACY_PTE
     ULONG PageFrameNumber:20;
 } HARDWARE_LEGACY_PTE, *PHARDWARE_LEGACY_PTE;
 
-/* Legacy Page Table Entry on non-PAE system */
-typedef struct _MMPTE_LEGACY_HARDWARE
+/* Page Table entry structure definition (PML3) */
+typedef struct _HARDWARE_MODERN_PTE
+{
+    ULONGLONG Valid:1;
+    ULONGLONG Writable:1;
+    ULONGLONG Owner:1;
+    ULONGLONG WriteThrough:1;
+    ULONGLONG CacheDisable:1;
+    ULONGLONG Accessed:1;
+    ULONGLONG Dirty:1;
+    ULONGLONG LargePage:1;
+    ULONGLONG Global:1;
+    ULONGLONG CopyOnWrite:1;
+    ULONGLONG Prototype:1;
+    ULONGLONG Reserved0:1;
+    ULONGLONG PageFrameNumber:26;
+    ULONGLONG Reserved1:14;
+    ULONGLONG SoftwareWsIndex:11;
+    ULONGLONG NoExecute:1;
+} HARDWARE_MODERN_PTE, *PHARDWARE_MODERN_PTE;
+
+/* Generic Page Table entry union to abstract PML2 and PML3 formats */
+typedef union _HARDWARE_PTE
+{
+    ULONGLONG Long;
+    HARDWARE_LEGACY_PTE Pml2;
+    HARDWARE_MODERN_PTE Pml3;
+} HARDWARE_PTE, *PHARDWARE_PTE;
+
+/* Page map information structure definition */
+typedef struct _MMPAGEMAP_INFO
+{
+    BOOLEAN Xpa;
+    ULONG PteBase;
+    ULONG PdeBase;
+    ULONG PdiShift;
+    ULONG PteShift;
+} MMPAGEMAP_INFO, *PMMPAGEMAP_INFO;
+
+/* Legacy Page Table Entry hardware structure definition (PML2) */
+typedef struct _MMPML2_PTE_HARDWARE
 {
     ULONG Valid:1;
     ULONG Writable:1;
@@ -182,21 +137,21 @@ typedef struct _MMPTE_LEGACY_HARDWARE
     ULONG Prototype:1;
     ULONG Write:1;
     ULONG PageFrameNumber:20;
-} MMPTE_LEGACY_HARDWARE, *PMMPTE_LEGACY_HARDWARE;
+} MMPML2_PTE_HARDWARE, *PMMPML2_PTE_HARDWARE;
 
-/* Legacy Page Table Entry list structure definition (without PAE support) */
-typedef struct _MMPTE_LEGACY_LIST
+/* Legacy Page Table Entry list structure definition (PML2) */
+typedef struct _MMPML2_PTE_LIST
 {
     ULONG Valid:1;
     ULONG OneEntry:1;
-    ULONG Reserved1:8;
+    ULONG Reserved0:8;
     ULONG Prototype:1;
-    ULONG Reserved2:1;
+    ULONG Reserved1:1;
     ULONG NextEntry:20;
-} MMPTE_LEGACY_LIST, *PMMPTE_LEGACY_LIST;
+} MMPML2_PTE_LIST, *PMMPML2_PTE_LIST;
 
-/* Legacy Page Table Entry prototype structure definition (without PAE support) */
-typedef struct _MMPTE_LEGACY_PROTOTYPE
+/* Legacy Page Table Entry subsection structure definition (PML2) */
+typedef struct _MMPML2_PTE_PROTOTYPE
 {
     ULONG Valid:1;
     ULONG ProtoAddressLow:7;
@@ -204,10 +159,10 @@ typedef struct _MMPTE_LEGACY_PROTOTYPE
     ULONG WhichPool:1;
     ULONG Prototype:1;
     ULONG ProtoAddressHigh:21;
-} MMPTE_LEGACY_PROTOTYPE, *PMMPTE_LEGACY_PROTOTYPE;
+} MMPML2_PTE_PROTOTYPE, *PMMPML2_PTE_PROTOTYPE;
 
-/* Legacy Page Table Entry software structure definition (without PAE support) */
-typedef struct _MMPTE_LEGACY_SOFTWARE
+/* Legacy Page Table Entry software structure definition (PML2) */
+typedef struct _MMPML2_PTE_SOFTWARE
 {
     ULONG Valid:1;
     ULONG PageFileLow:4;
@@ -215,10 +170,10 @@ typedef struct _MMPTE_LEGACY_SOFTWARE
     ULONG Prototype:1;
     ULONG Transition:1;
     ULONG PageFileHigh:20;
-} MMPTE_LEGACY_SOFTWARE, *PMMPTE_LEGACY_SOFTWARE;
+} MMPML2_PTE_SOFTWARE, *PMMPML2_PTE_SOFTWARE;
 
-/* Legacy Page Table Entry subsection structure definition (without PAE support) */
-typedef struct _MMPTE_LEGACY_SUBSECTION
+/* Legacy Page Table Entry subsection structure definition (PML2) */
+typedef struct _MMPML2_PTE_SUBSECTION
 {
     ULONG Valid:1;
     ULONG SubsectionAddressLow:4;
@@ -226,10 +181,10 @@ typedef struct _MMPTE_LEGACY_SUBSECTION
     ULONG Prototype:1;
     ULONG SubsectionAddressHigh:20;
     ULONG WhichPool:1;
-} MMPTE_LEGACY_SUBSECTION, *PMMPTE_LEGACY_SUBSECTION;
+} MMPML2_PTE_SUBSECTION, *PMMPML2_PTE_SUBSECTION;
 
-/* Legacy Page Table Entry transition structure definition (without PAE support) */
-typedef struct _MMPTE_LEGACY_TRANSITION
+/* Legacy Page Table Entry transition structure definition (PML2) */
+typedef struct _MMPML2_PTE_TRANSITION
 {
     ULONG Valid:1;
     ULONG Write:1;
@@ -240,20 +195,123 @@ typedef struct _MMPTE_LEGACY_TRANSITION
     ULONG Prototype:1;
     ULONG Transition:1;
     ULONG PageFrameNumber:20;
-} MMPTE_LEGACY_TRANSITION, *PMMPTE_LEGACY_TRANSITION;
+} MMPML2_PTE_TRANSITION, *PMMPML2_PTE_TRANSITION;
 
-/* Legacy Page Table Entry structure definition (without PAE support) */
-typedef union _MMPTE_LEGACY
+/* Legacy Page Table Entry union definition (PML2) */
+typedef union _MMPML2_PTE
 {
     ULONG Long;
-    HARDWARE_LEGACY_PTE Flush;
-    MMPTE_LEGACY_HARDWARE Hardware;
-    MMPTE_LEGACY_PROTOTYPE Prototype;
-    MMPTE_LEGACY_SOFTWARE Software;
-    MMPTE_LEGACY_TRANSITION Transition;
-    MMPTE_LEGACY_SUBSECTION Subsection;
-    MMPTE_LEGACY_LIST List;
-} MMPTE_LEGACY, *PMMPTE_LEGACY;
+    HARDWARE_PTE Flush;
+    MMPML2_PTE_HARDWARE Hard;
+    MMPML2_PTE_PROTOTYPE Proto;
+    MMPML2_PTE_SOFTWARE Soft;
+    MMPML2_PTE_TRANSITION Trans;
+    MMPML2_PTE_SUBSECTION Subsect;
+    MMPML2_PTE_LIST List;
+} MMPML2_PTE, *PMMPML2_PTE;
+
+/* Page Table Entry hardware structure definition (PML3) */
+typedef struct _MMPML3_PTE_HARDWARE
+{
+    ULONGLONG Valid:1;
+    ULONGLONG Writable:1;
+    ULONGLONG Owner:1;
+    ULONGLONG WriteThrough:1;
+    ULONGLONG CacheDisable:1;
+    ULONGLONG Accessed:1;
+    ULONGLONG Dirty:1;
+    ULONGLONG LargePage:1;
+    ULONGLONG Global:1;
+    ULONGLONG CopyOnWrite:1;
+    ULONGLONG Prototype:1;
+    ULONGLONG Write:1;
+    ULONGLONG PageFrameNumber:26;
+    ULONGLONG Reserved0:25;
+    ULONGLONG NoExecute:1;
+} MMPML3_PTE_HARDWARE, *PMMPML3_PTE_HARDWARE;
+
+/* Page Table Entry list structure definition (PML3) */
+typedef struct _MMPML3_PTE_LIST
+{
+    ULONGLONG Valid:1;
+    ULONGLONG OneEntry:1;
+    ULONGLONG Reserved0:8;
+    ULONGLONG Prototype:1;
+    ULONGLONG Reserved1:21;
+    ULONGLONG NextEntry:32;
+} MMPML3_PTE_LIST, *PMMPML3_PTE_LIST;
+
+/* Page Table Entry subsection structure definition (PML3) */
+typedef struct _MMPML3_PTE_PROTOTYPE
+{
+    ULONGLONG Valid:1;
+    ULONGLONG Reserved0:7;
+    ULONGLONG ReadOnly:1;
+    ULONGLONG Reserved1:1;
+    ULONGLONG Prototype:1;
+    ULONGLONG Protection:5;
+    ULONGLONG Reserved2:16;
+    ULONGLONG ProtoAddress:32;
+} MMPML3_PTE_PROTOTYPE, *PMMPML3_PTE_PROTOTYPE;
+
+/* Page Table Entry software structure definition (PML3) */
+typedef struct _MMPML3_PTE_SOFTWARE
+{
+    ULONGLONG Valid:1;
+    ULONGLONG PageFileLow:4;
+    ULONGLONG Protection:5;
+    ULONGLONG Prototype:1;
+    ULONGLONG Transition:1;
+    ULONGLONG Reserved0:20;
+    ULONGLONG PageFileHigh:32;
+} MMPML3_PTE_SOFTWARE, *PMMPML3_PTE_SOFTWARE;
+
+/* Page Table Entry subsection structure definition (PML3) */
+typedef struct _MMPML3_PTE_SUBSECTION
+{
+    ULONGLONG Valid:1;
+    ULONGLONG Reserved0:4;
+    ULONGLONG Protection:5;
+    ULONGLONG Prototype:1;
+    ULONGLONG Reserved1:21;
+    ULONGLONG SubsectionAddress:32;
+} MMPML3_PTE_SUBSECTION, *PMMPML3_PTE_SUBSECTION;
+
+/* Page Table Entry transition structure definition (PML3) */
+typedef struct _MMPML3_PTE_TRANSITION
+{
+    ULONGLONG Valid:1;
+    ULONGLONG Write:1;
+    ULONGLONG Owner:1;
+    ULONGLONG WriteThrough:1;
+    ULONGLONG CacheDisable:1;
+    ULONGLONG Protection:5;
+    ULONGLONG Prototype:1;
+    ULONGLONG Transition:1;
+    ULONGLONG PageFrameNumber:26;
+    ULONGLONG Unused:26;
+} MMPML3_PTE_TRANSITION, *PMMPML3_PTE_TRANSITION;
+
+/* Page Table Entry union definition (PML3) */
+typedef union _MMPML3_PTE
+{
+    ULONGLONG Long;
+    HARDWARE_PTE Flush;
+    MMPML3_PTE_HARDWARE Hardware;
+    MMPML3_PTE_PROTOTYPE Prototype;
+    MMPML3_PTE_SOFTWARE Software;
+    MMPML3_PTE_TRANSITION Transition;
+    MMPML3_PTE_SUBSECTION Subsection;
+    MMPML3_PTE_LIST List;
+} MMPML3_PTE, *PMMPML3_PTE;
+
+/* Generic Page Table Entry union to abstract PML2 and PML3 formats */
+typedef union _MMPTE
+{
+    ULONGLONG Long;
+    MMPML2_PTE Pml2;
+    MMPML3_PTE Pml3;
+} MMPTE, *PMMPTE;
 
 /* Page Frame Number structure definition */
 typedef struct _MMPFN

sdk/xtdk/i686/xtstruct.h

@@ -13,10 +13,23 @@
 
 
 /* Architecture-specific enumeration lists forward references */
+typedef enum _APIC_DM APIC_DM, *PAPIC_DM;
+typedef enum _APIC_DSH APIC_DSH, *PAPIC_DSH;
+typedef enum _APIC_MODE APIC_MODE, *PAPIC_MODE;
+typedef enum _APIC_REGISTER APIC_REGISTER, *PAPIC_REGISTER;
 typedef enum _CPU_VENDOR CPU_VENDOR, *PCPU_VENDOR;
-typedef enum _CPUID_FEATURES CPUID_FEATURES, *PCPUID_FEATURES;
+typedef enum _CPUID_FEATURES_EXTENDED CPUID_FEATURES_EXTENDED, *PCPUID_FEATURES_EXTENDED;
+typedef enum _CPUID_FEATURES_STANDARD1 CPUID_FEATURES_STANDARD1, *PCPUID_FEATURES_STANDARD1;
+typedef enum _CPUID_FEATURES_STANDARD7_LEAF0 CPUID_FEATURES_STANDARD7_LEAF0, *PCPUID_FEATURES_STANDARD7_LEAF0;
+typedef enum _CPUID_FEATURES_STANDARD7_LEAF1 CPUID_FEATURES_STANDARD7_LEAF1, *PCPUID_FEATURES_STANDARD7_LEAF1;
 typedef enum _CPUID_REQUESTS CPUID_REQUESTS, *PCPUID_REQUESTS;
 typedef enum _PAGE_SIZE PAGE_SIZE, *PPAGE_SIZE;
+typedef enum _PIC_I8259_ICW1_INTERRUPT_MODE PIC_I8259_ICW1_INTERRUPT_MODE, *PPIC_I8259_ICW1_INTERRUPT_MODE;
+typedef enum _PIC_I8259_ICW1_INTERVAL PIC_I8259_ICW1_INTERVAL, *PPIC_I8259_ICW1_INTERVAL;
+typedef enum _PIC_I8259_ICW1_OPERATING_MODE PIC_I8259_ICW1_OPERATING_MODE, *PPIC_I8259_ICW1_OPERATING_MODE;
+typedef enum _PIC_I8259_ICW4_BUFFERED_MODE PIC_I8259_ICW4_BUFFERED_MODE, *PPIC_I8259_ICW4_BUFFERED_MODE;
+typedef enum _PIC_I8259_ICW4_EOI_MODE PIC_I8259_ICW4_EOI_MODE, *PPIC_I8259_ICW4_EOI_MODE;
+typedef enum _PIC_I8259_ICW4_SYSTEM_MODE PIC_I8259_ICW4_SYSTEM_MODE, *PPIC_I8259_ICW4_SYSTEM_MODE;
 
 /* Architecture-specific structures forward references */
 typedef struct _CONTEXT CONTEXT, *PCONTEXT;
@@ -27,7 +40,7 @@ typedef struct _FN_SAVE_FORMAT FN_SAVE_FORMAT, *PFN_SAVE_FORMAT;
 typedef struct _FX_SAVE_AREA FX_SAVE_AREA, *PFX_SAVE_AREA;
 typedef struct _FX_SAVE_FORMAT FX_SAVE_FORMAT, *PFX_SAVE_FORMAT;
 typedef struct _HARDWARE_LEGACY_PTE HARDWARE_LEGACY_PTE, *PHARDWARE_LEGACY_PTE;
-typedef struct _HARDWARE_PTE HARDWARE_PTE, *PHARDWARE_PTE;
+typedef struct _HARDWARE_MODERN_PTE HARDWARE_MODERN_PTE, *PHARDWARE_MODERN_PTE;
 typedef struct _KDESCRIPTOR KDESCRIPTOR, *PKDESCRIPTOR;
 typedef struct _KEXCEPTION_FRAME KEXCEPTION_FRAME, *PKEXCEPTION_FRAME;
 typedef struct _KGDTENTRY KGDTENTRY, *PKGDTENTRY;
@@ -42,25 +55,36 @@ typedef struct _KSWITCH_FRAME KSWITCH_FRAME, *PKSWITCH_FRAME;
 typedef struct _KTHREAD_INIT_FRAME KTHREAD_INIT_FRAME, *PKTHREAD_INIT_FRAME;
 typedef struct _KTRAP_FRAME KTRAP_FRAME, *PKTRAP_FRAME;
 typedef struct _KTSS KTSS, *PKTSS;
+typedef struct _MMPAGEMAP_INFO MMPAGEMAP_INFO, *PMMPAGEMAP_INFO;
 typedef struct _MMPFN MMPFN, *PMMPFN;
-typedef struct _MMPTE_HARDWARE MMPTE_HARDWARE, *PMMPTE_HARDWARE;
-typedef struct _MMPTE_LEGACY_HARDWARE MMPTE_LEGACY_HARDWARE, *PMMPTE_LEGACY_HARDWARE;
-typedef struct _MMPTE_LEGACY_LIST MMPTE_LEGACY_LIST, *PMMPTE_LEGACY_LIST;
-typedef struct _MMPTE_LEGACY_PROTOTYPE MMPTE_LEGACY_PROTOTYPE, *PMMPTE_LEGACY_PROTOTYPE;
-typedef struct _MMPTE_LEGACY_SOFTWARE MMPTE_LEGACY_SOFTWARE, *PMMPTE_LEGACY_SOFTWARE;
-typedef struct _MMPTE_LEGACY_SUBSECTION MMPTE_LEGACY_SUBSECTION, *PMMPTE_LEGACY_SUBSECTION;
-typedef struct _MMPTE_LEGACY_TRANSITION MMPTE_LEGACY_TRANSITION, *PMMPTE_LEGACY_TRANSITION;
-typedef struct _MMPTE_LIST MMPTE_LIST, *PMMPTE_LIST;
-typedef struct _MMPTE_PROTOTYPE MMPTE_PROTOTYPE, *PMMPTE_PROTOTYPE;
-typedef struct _MMPTE_SOFTWARE MMPTE_SOFTWARE, *PMMPTE_SOFTWARE;
-typedef struct _MMPTE_SUBSECTION MMPTE_SUBSECTION, *PMMPTE_SUBSECTION;
-typedef struct _MMPTE_TRANSITION MMPTE_TRANSITION, *PMMPTE_TRANSITION;
+typedef struct _MMPML2_PTE_HARDWARE MMPML2_PTE_HARDWARE, *PMMPML2_PTE_HARDWARE;
+typedef struct _MMPML2_PTE_LIST MMPML2_PTE_LIST, *PMMPML2_PTE_LIST;
+typedef struct _MMPML2_PTE_PROTOTYPE MMPML2_PTE_PROTOTYPE, *PMMPML2_PTE_PROTOTYPE;
+typedef struct _MMPML2_PTE_SOFTWARE MMPML2_PTE_SOFTWARE, *PMMPML2_PTE_SOFTWARE;
+typedef struct _MMPML2_PTE_SUBSECTION MMPML2_PTE_SUBSECTION, *PMMPML2_PTE_SUBSECTION;
+typedef struct _MMPML2_PTE_TRANSITION MMPML2_PTE_TRANSITION, *PMMPML2_PTE_TRANSITION;
+typedef struct _MMPML3_PTE_HARDWARE MMPML3_PTE_HARDWARE, *PMMPML3_PTE_HARDWARE;
+typedef struct _MMPML3_PTE_LIST MMPML3_PTE_LIST, *PMMPML3_PTE_LIST;
+typedef struct _MMPML3_PTE_PROTOTYPE MMPML3_PTE_PROTOTYPE, *PMMPML3_PTE_PROTOTYPE;
+typedef struct _MMPML3_PTE_SOFTWARE MMPML3_PTE_SOFTWARE, *PMMPML3_PTE_SOFTWARE;
+typedef struct _MMPML3_PTE_SUBSECTION MMPML3_PTE_SUBSECTION, *PMMPML3_PTE_SUBSECTION;
+typedef struct _MMPML3_PTE_TRANSITION MMPML3_PTE_TRANSITION, *PMMPML3_PTE_TRANSITION;
 typedef struct _THREAD_ENVIRONMENT_BLOCK THREAD_ENVIRONMENT_BLOCK, *PTHREAD_ENVIRONMENT_BLOCK;
 
 /* Unions forward references */
+typedef union _APIC_BASE_REGISTER APIC_BASE_REGISTER, *PAPIC_BASE_REGISTER;
+typedef union _APIC_COMMAND_REGISTER APIC_COMMAND_REGISTER, *PAPIC_COMMAND_REGISTER;
+typedef union _APIC_LVT_REGISTER APIC_LVT_REGISTER, *PAPIC_LVT_REGISTER;
+typedef union _APIC_SPURIOUS_REGISTER APIC_SPURIOUS_REGISTER, *PAPIC_SPURIOUS_REGISTER;
+typedef union _HARDWARE_PTE HARDWARE_PTE, *PHARDWARE_PTE;
+typedef union _MMPML2_PTE MMPML2_PTE, *PMMPML2_PTE;
+typedef union _MMPML3_PTE MMPML3_PTE, *PMMPML3_PTE;
 typedef union _MMPTE MMPDE, *PMMPDE;
+typedef union _MMPTE MMPPE, *PMMPPE;
 typedef union _MMPTE MMPTE, *PMMPTE;
-typedef union _MMPTE_LEGACY MMPDE_LEGACY, *PMMPDE_LEGACY;
-typedef union _MMPTE_LEGACY MMPTE_LEGACY, *PMMPTE_LEGACY;
+typedef union _PIC_I8259_ICW1 PIC_I8259_ICW1, *PPIC_I8259_ICW1;
+typedef union _PIC_I8259_ICW2 PIC_I8259_ICW2, *PPIC_I8259_ICW2;
+typedef union _PIC_I8259_ICW3 PIC_I8259_ICW3, *PPIC_I8259_ICW3;
+typedef union _PIC_I8259_ICW4 PIC_I8259_ICW4, *PPIC_I8259_ICW4;
 
 #endif /* __XTDK_I686_XTSTRUCT_H */

sdk/xtdk/kdtypes.h

@@ -0,0 +1,45 @@
+/**
+ * PROJECT:         ExectOS
+ * COPYRIGHT:       See COPYING.md in the top level directory
+ * FILE:            sdk/xtdk/kdtypes.h
+ * DESCRIPTION:     Kernel Debugger data structures
+ * DEVELOPERS:      Aiken Harris <harraiken91@gmail.com>
+ */
+
+#ifndef __XTDK_KDTYPES_H
+#define __XTDK_KDTYPES_H
+
+#include <xtbase.h>
+#include <xtstruct.h>
+#include <rtltypes.h>
+
+
+/* Number of debug providers */
+#define KDBG_PROVIDERS_COUNT                                2
+
+/* Debug providers bitmask definitions */
+#define DEBUG_PROVIDER_COMPORT                              0x00000001
+#define DEBUG_PROVIDER_FRAMEBUFFER                          0x00000002
+
+/* Kernel routine callbacks */
+typedef XTSTATUS (XTAPI *PKD_INIT_ROUTINE)();
+typedef VOID (*PKD_PRINT_ROUTINE)(IN PCWSTR Format, IN ...);
+
+/* Debug mode structure definition */
+typedef struct _KD_DEBUG_MODE
+{
+    BOOLEAN Enabled;
+    ULONG Mode;
+    ULONG ComPortAddress;
+    ULONG ComPortNumber;
+    ULONG ComPortBaudRate;
+} KD_DEBUG_MODE, *PKD_DEBUG_MODE;
+
+/* Kernel debugger dispatch table structure definition */
+typedef struct _KD_DISPATCH_TABLE
+{
+    LIST_ENTRY ListEntry;
+    RTL_PRINT_CONTEXT PrintContext;
+} KD_DISPATCH_TABLE, *PKD_DISPATCH_TABLE;
+
+#endif /* __XTDK_KDTYPES_H */

sdk/xtdk/kefuncs.h

@@ -24,6 +24,11 @@ XTFASTCALL
 VOID
 KeAcquireSpinLock(IN OUT PKSPIN_LOCK SpinLock);
 
+XTAPI
+XTSTATUS
+KeAcquireSystemResource(IN SYSTEM_RESOURCE_TYPE ResourceType,
+                        OUT PSYSTEM_RESOURCE_HEADER *ResourceHeader);
+
 XTAPI
 BOOLEAN
 KeCancelTimer(IN PKTIMER Timer);
@@ -32,6 +37,11 @@ XTFASTCALL
 KRUNLEVEL
 KeGetCurrentRunLevel(VOID);
 
+XTAPI
+XTSTATUS
+KeGetSystemResource(IN SYSTEM_RESOURCE_TYPE ResourceType,
+                    OUT PSYSTEM_RESOURCE_HEADER *ResourceHeader);
+
 XTAPI
 BOOLEAN
 KeGetTimerState(IN PKTIMER Timer);
@@ -101,6 +111,10 @@ XTFASTCALL
 VOID
 KeReleaseSpinLock(IN OUT PKSPIN_LOCK SpinLock);
 
+XTAPI
+VOID
+KeReleaseSystemResource(IN PSYSTEM_RESOURCE_HEADER ResourceHeader);
+
 XTAPI
 VOID
 KeSetTargetProcessorDpc(IN PKDPC Dpc,

sdk/xtdk/ketypes.h

@@ -183,6 +183,14 @@ typedef enum _MODE
     MaximumMode
 } MODE, *PMODE;
 
+/* System resource types */
+typedef enum _SYSTEM_RESOURCE_TYPE
+{
+    SystemResourceInvalid,
+    SystemResourceAcpi,
+    SystemResourceFrameBuffer
+} SYSTEM_RESOURCE_TYPE, *PSYSTEM_RESOURCE_TYPE;
+
 /* Wait type */
 typedef enum _WAIT_TYPE
 {
@@ -190,6 +198,14 @@ typedef enum _WAIT_TYPE
     WaitAny
 } WAIT_TYPE, *PWAIT_TYPE;
 
+/* Kernel UBSAN data types enumeration list */
+typedef enum _KUBSAN_DATA_TYPE
+{
+    DataTypeInt,
+    DataTypeFloat,
+    DataTypeUnknown = 0xFFFF
+} KUBSAN_DATA_TYPE, *PKUBSAN_DATA_TYPE;
+
 /* Kernel routine callbacks */
 typedef EXCEPTION_DISPOSITION (XTCDECL *PEXCEPTION_ROUTINE)(IN PEXCEPTION_RECORD ExceptionRecord, IN PVOID EstablisherFrame, IN OUT PCONTEXT ContextRecord, IN OUT PVOID DispatcherContext);
 typedef VOID (XTAPI *PKDEFERRED_ROUTINE)(IN PKDPC Dpc, IN PVOID DeferredContext, IN PVOID SystemArgument1, IN PVOID SystemArgument2);
@@ -365,25 +381,35 @@ typedef struct _KPROCESS
     LIST_ENTRY ProfileListHead;
     ULONG_PTR DirectoryTable[2];
     USHORT IopmOffset;
+    UCHAR Iopl;
     VOLATILE KAFFINITY ActiveProcessors;
+    ULONG KernelTime;
+    ULONG UserTime;
     LIST_ENTRY ReadyListHead;
+    SINGLE_LIST_ENTRY SwapListEntry;
+    PVOID VdmTrapHandler;
     LIST_ENTRY ThreadListHead;
+    KSPIN_LOCK ProcessLock;
     KAFFINITY Affinity;
     union
     {
         struct
         {
-            LONG AutoAlignment:1;
-            LONG DisableBoost:1;
-            LONG DisableQuantum:1;
+            BOOLEAN AutoAlignment;
+            BOOLEAN DisableBoost;
+            BOOLEAN DisableQuantum;
             LONG ReservedFlags:29;
         };
         LONG ProcessFlags;
     };
+    ULONG_PTR StackCount;
     SCHAR BasePriority;
     SCHAR Quantum;
     UCHAR State;
-    ULONG_PTR StackCount;
+    UCHAR ThreadSeed;
+    UCHAR PowerState;
+    UCHAR IdealNode;
+    UCHAR Spare;
 } KPROCESS, *PKPROCESS;
 
 /* Thread control block structure definition */
@@ -396,7 +422,13 @@ typedef struct _KTHREAD
     PVOID StackBase;
     PVOID StackLimit;
     KSPIN_LOCK ThreadLock;
-    volatile UCHAR State;
+
+    ULONG ContextSwitches;
+    VOLATILE UCHAR State;
+    UCHAR NpxState;
+    KRUNLEVEL WaitRunLevel;
+    KPROCESSOR_MODE WaitMode;
+    PTHREAD_ENVIRONMENT_BLOCK EnvironmentBlock;
     union
     {
         KAPC_STATE ApcState;
@@ -411,13 +443,8 @@ typedef struct _KTHREAD
         };
     };
     KSPIN_LOCK ApcQueueLock;
-    ULONG ContextSwitches;
     LONG_PTR WaitStatus;
-    union
-    {
-        PKWAIT_BLOCK WaitBlockList;
-        PKGATE GateObject;
-    };
+    PKWAIT_BLOCK WaitBlockList;
     BOOLEAN Alertable;
     BOOLEAN WaitNext;
     UCHAR WaitReason;
@@ -431,43 +458,191 @@ typedef struct _KTHREAD
         SINGLE_LIST_ENTRY SwapListEntry;
     };
     PKQUEUE Queue;
-    CHAR PreviousMode;
-    SHORT SpecialApcDisable;
-    PTHREAD_ENVIRONMENT_BLOCK EnvironmentBlock;
+    ULONG WaitTime;
     union
     {
-        KTIMER Timer;
         struct
         {
-            UCHAR TimerFill[KTIMER_LENGTH];
-            union
-            {
-                struct
-                {
-                    LONG AutoAlignment:1;
-                    LONG DisableBoost:1;
-                    LONG ReservedFlags:30;
-                };
-                LONG ThreadFlags;
-            };
+            SHORT KernelApcDisable;
+            SHORT SpecialApcDisable;
         };
+        ULONG CombinedApcDisable;
     };
+    KTIMER Timer;
     KWAIT_BLOCK WaitBlock[KTHREAD_WAIT_BLOCK + 1];
-    UCHAR NpxState;
-    KRUNLEVEL WaitRunLevel;
     LIST_ENTRY QueueListEntry;
-    PKTRAP_FRAME TrapFrame;
-    PVOID CallbackStack;
-    PVOID ServiceTable;
-    ULONG KernelLimit;
     UCHAR ApcStateIndex;
-    BOOLEAN StackResident;
+    BOOLEAN Preempted;
+    BOOLEAN ProcessReadyQueue;
+    BOOLEAN KernelStackResident;
+    CHAR Saturation;
+    UCHAR IdealProcessor;
+    SCHAR BasePriority;
+    UCHAR Spare4;
+    SCHAR PriorityDecrement;
+    SCHAR Quantum;
+    BOOLEAN SystemAffinityActive;
+    CHAR PreviousMode;
+    UCHAR ResourceIndex;
+    UCHAR DisableBoost;
+    KAFFINITY UserAffinity;
     PKPROCESS Process;
     KAFFINITY Affinity;
+    PVOID ServiceTable;
     PKAPC_STATE ApcStatePointer[2];
     KAPC_STATE SavedApcState;
+    PVOID CallbackStack;
+    PVOID SubSystemThread;
+    PKTRAP_FRAME TrapFrame;
+    ULONG KernelTime;
+    ULONG UserTime;
     KAPC SuspendApc;
     KSEMAPHORE SuspendSemaphore;
+    PVOID TlsArray;
+    PVOID LegoData;
+    LIST_ENTRY ThreadListEntry;
+    UCHAR LargeStack;
+    UCHAR PowerState;
+    UCHAR NpxIrql;
+    UCHAR Spare5;
+    BOOLEAN AutoAlignment;
+    UCHAR Iopl;
+    CCHAR FreezeCount;
+    CCHAR SuspendCount;
+    UCHAR Spare0[1];
+    UCHAR UserIdealProcessor;
+    UCHAR Spare2[3];
+    ULONG KernelLimit;
+    BOOLEAN StackResident;
 } KTHREAD, *PKTHREAD;
 
+/* System resource common header structure definition */
+typedef struct _SYSTEM_RESOURCE_HEADER
+{
+    LIST_ENTRY ListEntry;
+    SYSTEM_RESOURCE_TYPE ResourceType;
+    BOOLEAN ResourceLocked;
+    ULONG ResourceSize;
+    PVOID PhysicalAddress;
+    PVOID VirtualAddress;
+} SYSTEM_RESOURCE_HEADER, *PSYSTEM_RESOURCE_HEADER;
+
+/* ACPI system resource structure definition */
+typedef struct _SYSTEM_RESOURCE_ACPI
+{
+    SYSTEM_RESOURCE_HEADER Header;
+    PVOID ApicBase;
+} SYSTEM_RESOURCE_ACPI, *PSYSTEM_RESOURCE_ACPI;
+
+/* FrameBuffer system resource structure definition */
+typedef struct _SYSTEM_RESOURCE_FRAMEBUFFER
+{
+    SYSTEM_RESOURCE_HEADER Header;
+    ULONG_PTR BufferSize;
+    UINT Width;
+    UINT Height;
+    UINT Depth;
+    UINT PixelsPerScanLine;
+    UINT BitsPerPixel;
+    UINT Pitch;
+    PVOID Font;
+    struct
+    {
+        USHORT BlueShift;
+        USHORT BlueSize;
+        USHORT GreenShift;
+        USHORT GreenSize;
+        USHORT RedShift;
+        USHORT RedSize;
+        USHORT ReservedShift;
+        USHORT ReservedSize;
+    } Pixels;
+} SYSTEM_RESOURCE_FRAMEBUFFER, *PSYSTEM_RESOURCE_FRAMEBUFFER;
+
+/* Kernel UBSAN source location structure definition */
+typedef struct _KUBSAN_SOURCE_LOCATION
+{
+    PCCHAR FileName;
+    union
+    {
+        ULONG Reported;
+        struct
+        {
+            UINT Line;
+            UINT Column;
+        };
+    };
+} KUBSAN_SOURCE_LOCATION, *PKUBSAN_SOURCE_LOCATION;
+
+/* Kernel UBSAN type descriptor structure definition */
+typedef struct _KUBSAN_TYPE_DESCRIPTOR
+{
+    USHORT DataType;
+    USHORT TypeInfo;
+    CHAR TypeName[1];
+} KUBSAN_TYPE_DESCRIPTOR, *PKUBSAN_TYPE_DESCRIPTOR;
+
+/* Kernel UBSAN float cast overflow data structure definition */
+typedef struct _KUBSAN_FLOAT_CAST_OVERFLOW_DATA
+{
+    KUBSAN_SOURCE_LOCATION Location;
+    PKUBSAN_TYPE_DESCRIPTOR LhsType;
+    PKUBSAN_TYPE_DESCRIPTOR RhsType;
+} KUBSAN_FLOAT_CAST_OVERFLOW_DATA, *PKUBSAN_FLOAT_CAST_OVERFLOW_DATA;
+
+/* Kernel UBSAN function type mismatch data structure definition */
+typedef struct _KUBSAN_FUNCTION_TYPE_MISMATCH_DATA
+{
+    KUBSAN_SOURCE_LOCATION Location;
+    PKUBSAN_TYPE_DESCRIPTOR Type;
+} KUBSAN_FUNCTION_TYPE_MISMATCH_DATA, *PKUBSAN_FUNCTION_TYPE_MISMATCH_DATA;
+
+/* Kernel UBSAN invalid builtin data structure definition */
+typedef struct _KUBSAN_INVALID_BUILTIN_DATA
+{
+    KUBSAN_SOURCE_LOCATION Location;
+    UCHAR Kind;
+} KUBSAN_INVALID_BUILTIN_DATA, *PKUBSAN_INVALID_BUILTIN_DATA;
+
+/* Kernel UBSAN shift out of bounds data structure definition */
+typedef struct _KUBSAN_SHIFT_OUT_OF_BOUNDS_DATA
+{
+    KUBSAN_SOURCE_LOCATION Location;
+    PKUBSAN_TYPE_DESCRIPTOR LhsType;
+    PKUBSAN_TYPE_DESCRIPTOR RhsType;
+} KUBSAN_SHIFT_OUT_OF_BOUNDS_DATA, *PKUBSAN_SHIFT_OUT_OF_BOUNDS_DATA;
+
+/* Kernel UBSAN out of bounds data structure definition */
+typedef struct _KUBSAN_OUT_OF_BOUNDS_DATA
+{
+    KUBSAN_SOURCE_LOCATION Location;
+    PKUBSAN_TYPE_DESCRIPTOR ArrayType;
+    PKUBSAN_TYPE_DESCRIPTOR IndexType;
+} KUBSAN_OUT_OF_BOUNDS_DATA, *PKUBSAN_OUT_OF_BOUNDS_DATA;
+
+/* Kernel UBSAN overflow data structure definition */
+typedef struct _KUBSAN_OVERFLOW_DATA
+{
+    KUBSAN_SOURCE_LOCATION Location;
+    PKUBSAN_TYPE_DESCRIPTOR Type;
+} KUBSAN_OVERFLOW_DATA, *PKUBSAN_OVERFLOW_DATA;
+
+/* Kernel UBSAN type mismatch data structure definition */
+typedef struct _KUBSAN_TYPE_MISMATCH_DATA
+{
+    KUBSAN_SOURCE_LOCATION Location;
+    PKUBSAN_TYPE_DESCRIPTOR Type;
+    ULONG Alignment;
+    UCHAR TypeCheckKind;
+} KUBSAN_TYPE_MISMATCH_DATA, *PKUBSAN_TYPE_MISMATCH_DATA;
+
+/* Kernel UBSAN type mismatch data structure definition */
+typedef struct _KUBSAN_TYPE_MISMATCH_DATA_V1
+{
+    KUBSAN_SOURCE_LOCATION Location;
+    PKUBSAN_TYPE_DESCRIPTOR Type;
+    UCHAR LogAlignment;
+    UCHAR TypeCheckKind;
+} KUBSAN_TYPE_MISMATCH_DATA_V1, *PKUBSAN_TYPE_MISMATCH_DATA_V1;
+
 #endif /* __XTDK_KEFUNCS_H */

sdk/xtdk/mmtypes.h

@@ -10,8 +10,18 @@
 #define __XTDK_MMTYPES_H
 
 #include <xtbase.h>
+#include ARCH_HEADER(xtstruct.h)
 
 
+/* Page map routines structure definition */
+typedef CONST STRUCT _CMMPAGEMAP_ROUTINES
+{
+    VOID (XTAPI *ClearPte)(PHARDWARE_PTE PtePointer);
+    BOOLEAN (XTAPI *PteValid)(PHARDWARE_PTE PtePointer);
+    VOID (XTAPI *SetPteCaching)(PHARDWARE_PTE PtePointer, BOOLEAN CacheDisable, BOOLEAN WriteThrough);
+    VOID (XTAPI *SetPte)(PHARDWARE_PTE PtePointer, PFN_NUMBER PageFrameNumber, BOOLEAN Writable);
+} CMMPAGEMAP_ROUTINES, *PCMMPAGEMAP_ROUTINES;
+
 /* Color tables structure definition */
 typedef struct _MMCOLOR_TABLES
 {

sdk/xtdk/potypes.h

@@ -15,16 +15,77 @@
 
 
 /* Power Manager routine callbacks */
-typedef VOID (XTFASTCALL *PPROCESSOR_IDLE_FUNCTION)(PPROCESSOR_POWER_STATE PowerState);
+typedef VOID (XTFASTCALL *PPROCESSOR_IDLE_FUNCTION)(IN PPROCESSOR_POWER_STATE PowerState);
+typedef XTSTATUS (XTFASTCALL *PSET_PROCESSOR_THROTTLE)(IN UCHAR Throttle);
+
+/* Processor IDLE times structure definition */
+typedef struct _PROCESSOR_IDLE_TIMES
+{
+    ULONGLONG StartTime;
+    ULONGLONG EndTime;
+    ULONG IdleHandlerReserved[4];
+} PROCESSOR_IDLE_TIMES, *PPROCESSOR_IDLE_TIMES;
+
+/* Processor performance state structure definition */
+typedef struct _PROCESSOR_PERF_STATE
+{
+    UCHAR PercentFrequency;
+    UCHAR MinCapacity;
+    USHORT Power;
+    UCHAR IncreaseLevel;
+    UCHAR DecreaseLevel;
+    USHORT Flags;
+    ULONG IncreaseTime;
+    ULONG DecreaseTime;
+    ULONG IncreaseCount;
+    ULONG DecreaseCount;
+    ULONGLONG PerformanceTime;
+} PROCESSOR_PERF_STATE, *PPROCESSOR_PERF_STATE;
 
 /* Processor power state structure definition */
 typedef struct _PROCESSOR_POWER_STATE
 {
     PPROCESSOR_IDLE_FUNCTION IdleFunction;
     ULONG Idle0TimeLimit;
+    ULONG Idle0LastTime;
+    PVOID IdleHandlers;
+    PVOID IdleState;
+    ULONG IdleHandlersCount;
+    ULONGLONG LastCheck;
+    PROCESSOR_IDLE_TIMES IdleTimes;
+    ULONG IdleTime1;
+    ULONG PromotionCheck;
+    ULONG IdleTime2;
     UCHAR CurrentThrottle;
+    UCHAR ThermalThrottleLimit;
+    UCHAR CurrentThrottleIndex;
+    UCHAR ThermalThrottleIndex;
+    ULONG PerfSystemTime;
+    ULONG PerfIdleTime;
+    ULONG LastSysTime;
+    ULONGLONG TotalIdleStateTime[3];
+    ULONG TotalIdleTransitions[3];
+    ULONGLONG PreviousC3StateTime;
+    UCHAR KneeThrottleIndex;
+    UCHAR ThrottleLimitIndex;
+    UCHAR PerfStatesCount;
+    UCHAR ProcessorMinThrottle;
+    UCHAR ProcessorMaxThrottle;
+    UCHAR LastBusyPercentage;
+    UCHAR LastC3Percentage;
+    UCHAR LastAdjustedBusyPercentage;
+    ULONG PromotionCount;
+    ULONG DemotionCount;
+    ULONG ErrorCount;
+    ULONG RetryCount;
+    ULONG Flags;
+    LARGE_INTEGER PerfCounterFrequency;
+    ULONG PerfTickCount;
     KTIMER PerfTimer;
     KDPC PerfDpc;
+    PPROCESSOR_PERF_STATE PerfStates;
+    PSET_PROCESSOR_THROTTLE PerfSetThrottle;
+    ULONG LastC3UserTime;
 } PROCESSOR_POWER_STATE, *PPROCESSOR_POWER_STATE;
 
 #endif /* __XTDK_POTYPES_H */

sdk/xtdk/rtlfuncs.h

@@ -83,26 +83,26 @@ RtlCompareMemory(IN PCVOID LeftBuffer,
 
 XTAPI
 SIZE_T
-RtlCompareString(IN CONST PCHAR String1,
-                 IN CONST PCHAR String2,
+RtlCompareString(IN PCSTR String1,
+                 IN PCSTR String2,
                  IN SIZE_T Length);
 
 XTAPI
 SIZE_T
-RtlCompareStringInsensitive(IN CONST PCHAR String1,
-                            IN CONST PCHAR String2,
+RtlCompareStringInsensitive(IN PCSTR String1,
+                            IN PCSTR String2,
                             IN SIZE_T Length);
 
 XTAPI
 SIZE_T
-RtlCompareWideString(IN CONST PWCHAR String1,
-                     IN CONST PWCHAR String2,
+RtlCompareWideString(IN PCWSTR String1,
+                     IN PCWSTR String2,
                      IN SIZE_T Length);
 
 XTAPI
 SIZE_T
-RtlCompareWideStringInsensitive(IN CONST PWCHAR String1,
-                                IN CONST PWCHAR String2,
+RtlCompareWideStringInsensitive(IN PCWSTR String1,
+                                IN PCWSTR String2,
                                 IN SIZE_T Length);
 
 XTAPI
@@ -134,13 +134,13 @@ RtlCopyMemory(OUT PVOID Destination,
 XTAPI
 VOID
 RtlCopyString(IN PCHAR Destination,
-              IN PCCHAR Source,
+              IN PCSTR Source,
               IN ULONG Length);
 
 XTAPI
 VOID
 RtlCopyWideString(IN PWCHAR Destination,
-                  IN CONST PWCHAR Source,
+                  IN PCWSTR Source,
                   IN ULONG Length);
 
 XTAPI
@@ -161,6 +161,26 @@ RtlFindSetBits(IN PRTL_BITMAP BitMap,
                IN ULONG_PTR Length,
                IN ULONG_PTR Index);
 
+XTAPI
+PCSTR
+RtlFindString(IN PCSTR Source,
+              IN PCSTR Search);
+
+XTAPI
+PCSTR
+RtlFindStringInsensitive(IN PCSTR Source,
+                         IN PCSTR Search);
+
+XTAPI
+PCWSTR
+RtlFindWideString(IN PCWSTR Source,
+                  IN PCWSTR Search);
+
+XTAPI
+PCWSTR
+RtlFindWideStringInsensitive(IN PCWSTR Source,
+                             IN PCWSTR Search);
+
 XTAPI
 XTSTATUS
 RtlFormatWideString(IN PRTL_PRINT_CONTEXT Context,
@@ -229,13 +249,13 @@ RtlSetMemory(OUT PVOID Destination,
 
 XTAPI
 SIZE_T
-RtlStringLength(IN CONST PCHAR String,
+RtlStringLength(IN PCSTR String,
                 IN SIZE_T MaxLength);
 
 XTAPI
 SIZE_T
 RtlStringToWideString(OUT PWCHAR Destination,
-                      IN CONST PCHAR *Source,
+                      IN PCSTR *Source,
                       IN SIZE_T Length);
 
 XTAPI
@@ -246,42 +266,58 @@ RtlTestBit(IN PRTL_BITMAP BitMap,
 XTAPI
 PCHAR
 RtlTokenizeString(IN PCHAR String,
-                  IN CONST PCHAR Delimiter,
+                  IN PCSTR Delimiter,
                   IN OUT PCHAR *SavePtr);
 
 XTAPI
 PWCHAR
 RtlTokenizeWideString(IN PWCHAR String,
-                      IN CONST PWCHAR Delimiter,
+                      IN PCWSTR Delimiter,
                       IN OUT PWCHAR *SavePtr);
 
 XTAPI
-PCHAR
-RtlTrimLeftString(IN CONST PCHAR String);
+CHAR
+RtlToLowerCharacter(IN CHAR Character);
 
 XTAPI
-PWCHAR
-RtlTrimLeftWideString(IN CONST PWCHAR String);
+WCHAR
+RtlToLowerWideCharacter(IN WCHAR Character);
+
+XTAPI
+CHAR
+RtlToUpperCharacter(IN CHAR Character);
+
+XTAPI
+WCHAR
+RtlToUpperWideCharacter(IN WCHAR Character);
 
 XTAPI
 PCHAR
-RtlTrimRightString(IN CONST PCHAR String);
+RtlTrimLeftString(IN PCHAR String);
 
 XTAPI
 PWCHAR
-RtlTrimRightWideString(IN CONST PWCHAR String);
+RtlTrimLeftWideString(IN PWCHAR String);
 
 XTAPI
 PCHAR
-RtlTrimString(IN CONST PCHAR String);
+RtlTrimRightString(IN PCHAR String);
 
 XTAPI
 PWCHAR
-RtlTrimWideString(IN CONST PWCHAR String);
+RtlTrimRightWideString(IN PWCHAR String);
+
+XTAPI
+PCHAR
+RtlTrimString(IN PCHAR String);
+
+XTAPI
+PWCHAR
+RtlTrimWideString(IN PWCHAR String);
 
 XTAPI
 SIZE_T
-RtlWideStringLength(IN CONST PWCHAR String,
+RtlWideStringLength(IN PCWSTR String,
                     IN SIZE_T MaxLength);
 
 XTAPI

sdk/xtdk/xtbase.h

@@ -32,6 +32,9 @@ typedef ULONG_PTR KSPIN_LOCK, *PKSPIN_LOCK;
 /* Page Frame Number */
 typedef ULONG_PTR PFN_NUMBER, *PPFN_NUMBER;
 
+/* Physical address */
+typedef LARGE_INTEGER PHYSICAL_ADDRESS, *PPHYSICAL_ADDRESS;
+
 /* 128-bit buffer containing a unique identifier value */
 typedef struct _GUID
 {

sdk/xtdk/xtdebug.h

@@ -11,16 +11,16 @@
 
 
 /* Debugging macros */
-#define CHECKPOINT                      DebugPrint(L"Checkpoint reached at %s:%d\n", __FILE__, __LINE__);
+#define CHECKPOINT                      DebugPrint(L"Checkpoint reached at %s:%d\n", __RELFILE__, __LINE__);
 #define DEPRECATED                      DebugPrint(L"Called deprecated routine '%s()' at %s:%d\n", \
-                                                 __FUNCTION__, __FILE__, __LINE__);
+                                                 __FUNCTION__, __RELFILE__, __LINE__);
 #define UNIMPLEMENTED                   DebugPrint(L"Called unimplemented routine '%s()' at %s:%d\n", \
-                                                 __FUNCTION__, __FILE__, __LINE__);
+                                                 __FUNCTION__, __RELFILE__, __LINE__);
 
 /* XTOS platform debugging macros */
 #ifdef DBG
     #define DEBUG                       1
-    #define DebugPrint(Format, ...)     if(KeDbgPrint) KeDbgPrint(Format, __VA_ARGS__);
+    #define DebugPrint(Format, ...)     if(KdPrint) KdPrint(Format, __VA_ARGS__);
 #else
     #define DEBUG                       0
     #define DebugPrint(Format, ...)     ((VOID)NULL)

sdk/xtdk/xtdefs.h

@@ -16,6 +16,7 @@
 #define XTAPI                                  __stdcall
 #define XTCDECL                                __cdecl
 #define XTFASTCALL                             __fastcall
+#define XTVECTORCALL                           __vectorcall
 #define XTINLINE                               __inline
 #define XTASSEMBLY                             __attribute__((naked))
 #define XTINTERRUPT                            __attribute__((interrupt))
@@ -54,32 +55,29 @@
 
 /* Preprocessor macros for defining an additional compiler attributes */
 #define ALIGN(Alignment)                       __attribute__((aligned(Alignment)))
-#define PACK                                   __attribute__((packed))
+#define PACKED                                 __attribute__((packed))
 #define SEGMENT(Segment)                       __attribute__((section(Segment)))
 #define USED                                   __attribute__((__used__))
 
 /* Macro for calculating size of an array */
 #define ARRAY_SIZE(Array)                      (sizeof(Array) / sizeof(*Array))
 
-/* Macro for concatenating two strings */
+/* Macros for concatenating two strings */
 #define CONCAT_STRING(Str1, Str2)              Str1##Str2
 #define CONCATENATE(Str1, Str2)                CONCAT_STRING(Str1, Str2)
 
 /* Macro for accessing the base address of a structure from a structure member */
-#define CONTAIN_RECORD(Address, Type, Field)   ((Type *)(((ULONG_PTR)Address) - (ULONG_PTR)(&(((Type *)0)->Field))))
+#define CONTAIN_RECORD(Address, Type, Field)   ((Type *)((char *)(Address) - FIELD_OFFSET(Type, Field)))
 
 /* EFI size to pages conversion macro */
 #define EFI_SIZE_TO_PAGES(Size)                (((Size) >> EFI_PAGE_SHIFT) + (((Size) & EFI_PAGE_MASK) ? 1 : 0))
 
 /* Macro for calculating byte offset of a field in the structure */
-#define FIELD_OFFSET(Structure, Field)         ((LONG)(LONG_PTR)&(((Structure *)0)->Field))
+#define FIELD_OFFSET(Structure, Field)         __builtin_offsetof(Structure, Field)
 
 /* Macro for calculating size of a field in the structure */
 #define FIELD_SIZE(Structure, Field)           (sizeof(((Structure *)0)->Field))
 
-/* Macro that yields field type in the structure */
-#define FIELD_TYPE(Structure, Field)           (((Structure*)0)->Field)
-
 /* Macro that page-aligns a virtual address */
 #define PAGE_ALIGN(VirtualAddress)             ((PVOID)((ULONG_PTR)VirtualAddress & ~MM_PAGE_MASK))
 
@@ -97,7 +95,8 @@
 #define SIGNATURE32(A, B, C, D)                (SIGNATURE16(A, B) | (SIGNATURE16(C, D) << 16))
 #define SIGNATURE64(A, B, C, D, E, F, G, H)    (SIGNATURE32(A, B, C, D) | ((UINT64)(SIGNATURE32(E, F, G, H)) << 32))
 
-/* XT size to pages conversion macro */
+/* XT size <-> pages conversion macro */
+#define PAGES_TO_SIZE(Pages)                   ((Pages) << MM_PAGE_SHIFT)
 #define SIZE_TO_PAGES(Size)                    (((Size) >> MM_PAGE_SHIFT) + (((Size) & (MM_PAGE_MASK)) ? 1 : 0))
 
 /* Macros for concatenating strings */

sdk/xtdk/xtfw.h

@@ -48,7 +48,7 @@ typedef enum _LOADER_MEMORY_TYPE
     LoaderBBTMemory,
     LoaderReserve,
     LoaderXIPRom,
-    LoaderHALCachedMemory,
+    LoaderHardwareCachedMemory,
     LoaderMaximum
 } LOADER_MEMORY_TYPE, *PLOADER_MEMORY_TYPE;
 
@@ -61,11 +61,6 @@ typedef enum _SYSTEM_FIRMWARE_TYPE
     SystemFirmwarePcat
 } SYSTEM_FIRMWARE_TYPE, *PSYSTEM_FIRMWARE_TYPE;
 
-/* Hardware information block */
-typedef struct _HARDWARE_INFORMATION_BLOCK
-{
-} HARDWARE_INFORMATION_BLOCK, *PHARDWARE_INFORMATION_BLOCK;
-
 /* PCAT Firmware information block */
 typedef struct _PCAT_FIRMWARE_INFORMATION
 {
@@ -90,36 +85,20 @@ typedef struct _FIRMWARE_INFORMATION_BLOCK
     };
 } FIRMWARE_INFORMATION_BLOCK, *PFIRMWARE_INFORMATION_BLOCK;
 
-/* Boot Loader FrameBuffer information block */
-typedef struct _LOADER_GRAPHICS_INFORMATION_BLOCK
-{
-    BOOLEAN Initialized;
-    PVOID Address;
-    ULONG_PTR BufferSize;
-    UINT Width;
-    UINT Height;
-    UINT PixelsPerScanLine;
-    UINT BitsPerPixel;
-    UINT Pitch;
-    PVOID Font;
-} LOADER_GRAPHICS_INFORMATION_BLOCK, *PLOADER_GRAPHICS_INFORMATION_BLOCK;
-
 /* Boot Loader information block */
 typedef struct _LOADER_INFORMATION_BLOCK
 {
     PVOID DbgPrint;
-    ULONG PageMapLevel;
-    LOADER_GRAPHICS_INFORMATION_BLOCK FrameBuffer;
 } LOADER_INFORMATION_BLOCK, *PLOADER_INFORMATION_BLOCK;
 
 /* Boot Loader memory mapping information */
-typedef struct _LOADER_MEMORY_MAPPING
+typedef struct _LOADER_MEMORY_DESCRIPTOR
 {
     LIST_ENTRY ListEntry;
     LOADER_MEMORY_TYPE MemoryType;
     ULONG BasePage;
     ULONG PageCount;
-} LOADER_MEMORY_MAPPING, *PLOADER_MEMORY_MAPPING;
+} LOADER_MEMORY_DESCRIPTOR, *PLOADER_MEMORY_DESCRIPTOR;
 
 /* Loader provided information needed by the kernel to initialize */
 typedef struct _KERNEL_INITIALIZATION_BLOCK
@@ -131,7 +110,7 @@ typedef struct _KERNEL_INITIALIZATION_BLOCK
     LIST_ENTRY LoadOrderListHead;
     LIST_ENTRY MemoryDescriptorListHead;
     LIST_ENTRY BootDriverListHead;
-    HARDWARE_INFORMATION_BLOCK HardwareInformation;
+    LIST_ENTRY SystemResourcesListHead;
     LOADER_INFORMATION_BLOCK LoaderInformation;
     FIRMWARE_INFORMATION_BLOCK FirmwareInformation;
 } KERNEL_INITIALIZATION_BLOCK, *PKERNEL_INITIALIZATION_BLOCK;

sdk/xtdk/xtkmapi.h

@@ -30,6 +30,7 @@
 #include <extypes.h>
 #include <hltypes.h>
 #include <iotypes.h>
+#include <kdtypes.h>
 #include <ketypes.h>
 #include <ldrtypes.h>
 #include <mmtypes.h>

sdk/xtdk/xtstatus.h

@@ -48,11 +48,22 @@
 
 /* XT status code definitions */
 #define STATUS_SUCCESS                                                     ((XTSTATUS) 0x00000000L)
+#define STATUS_END_OF_MEDIA                                                ((XTSTATUS) 0x8000001EL)
+#define STATUS_RESOURCE_LOCKED                                             ((XTSTATUS) 0xC0000000L)
 #define STATUS_UNSUCCESSFUL                                                ((XTSTATUS) 0xC0000001L)
 #define STATUS_NOT_IMPLEMENTED                                             ((XTSTATUS) 0xC0000002L)
+#define STATUS_ACCESS_VIOLATION                                            ((XTSTATUS) 0xC0000005L)
+#define STATUS_IN_PAGE_ERROR                                               ((XTSTATUS) 0xC0000006L)
+#define STATUS_INVALID_HANDLE                                              ((XTSTATUS) 0xC0000008L)
+#define STATUS_BAD_INITIAL_STACK                                           ((XTSTATUS) 0xC0000009L)
 #define STATUS_INVALID_PARAMETER                                           ((XTSTATUS) 0xC000000DL)
+#define STATUS_END_OF_FILE                                                 ((XTSTATUS) 0xC0000011L)
+#define STATUS_NO_MEMORY                                                   ((XTSTATUS) 0xC0000017L)
+#define STATUS_PORT_DISCONNECTED                                           ((XTSTATUS) 0xC0000037L)
+#define STATUS_CRC_ERROR                                                   ((XTSTATUS) 0xC000003FL)
 #define STATUS_INSUFFICIENT_RESOURCES                                      ((XTSTATUS) 0xC000009AL)
 #define STATUS_DEVICE_NOT_READY                                            ((XTSTATUS) 0xC00000A3L)
+#define STATUS_NOT_SUPPORTED                                               ((XTSTATUS) 0xC00000BBL)
 #define STATUS_TIMEOUT                                                     ((XTSTATUS) 0x00000102L)
 #define STATUS_IO_DEVICE_ERROR                                             ((XTSTATUS) 0xC0000185L)
 #define STATUS_NOT_FOUND                                                   ((XTSTATUS) 0xC0000225L)

sdk/xtdk/xtstruct.h

@@ -39,6 +39,7 @@ typedef enum _EFI_TIMER_DELAY EFI_TIMER_DELAY, *PEFI_TIMER_DELAY;
 typedef enum _EFI_UART_PARITY_TYPE EFI_UART_PARITY_TYPE, *PEFI_UART_PARITY_TYPE;
 typedef enum _EFI_UART_STOP_BITS_TYPE EFI_UART_STOP_BITS_TYPE, *PEFI_UART_STOP_BITS_TYPE;
 typedef enum _EFI_UNIVERSA_GRAPHICS_BLT_OPERATION EFI_UNIVERSA_GRAPHICS_BLT_OPERATION, *PEFI_UNIVERSA_GRAPHICS_BLT_OPERATION;
+typedef enum _HAL_APIC_MODE HAL_APIC_MODE, *PHAL_APIC_MODE;
 typedef enum _KAPC_ENVIRONMENT KAPC_ENVIRONMENT, *PKAPC_ENVIRONMENT;
 typedef enum _KDPC_IMPORTANCE KDPC_IMPORTANCE, *PKDPC_IMPORTANCE;
 typedef enum _KEVENT_TYPE KEVENT_TYPE, *PKEVENT_TYPE;
@@ -46,16 +47,30 @@ typedef enum _KOBJECTS KOBJECTS, *PKOBJECTS;
 typedef enum _KPROCESS_STATE KPROCESS_STATE, *PKPROCESS_STATE;
 typedef enum _KTHREAD_STATE KTHREAD_STATE, *PKTHREAD_STATE;
 typedef enum _KTIMER_TYPE KTIMER_TYPE, *PKTIMER_TYPE;
+typedef enum _KUBSAN_DATA_TYPE KUBSAN_DATA_TYPE, *PKUBSAN_DATA_TYPE;
 typedef enum _LOADER_MEMORY_TYPE LOADER_MEMORY_TYPE, *PLOADER_MEMORY_TYPE;
 typedef enum _MODE MODE, *PMODE;
+typedef enum _RTL_VARIABLE_TYPE RTL_VARIABLE_TYPE, *PRTL_VARIABLE_TYPE;
 typedef enum _SYSTEM_FIRMWARE_TYPE SYSTEM_FIRMWARE_TYPE, *PSYSTEM_FIRMWARE_TYPE;
 typedef enum _WAIT_TYPE WAIT_TYPE, *PWAIT_TYPE;
 
 /* Structures forward references */
+typedef struct _ACPI_CACHE_LIST ACPI_CACHE_LIST, *PACPI_CACHE_LIST;
+typedef struct _ACPI_DESCRIPTION_HEADER ACPI_DESCRIPTION_HEADER, *PACPI_DESCRIPTION_HEADER;
+typedef struct _ACPI_FADT ACPI_FADT, *PACPI_FADT;
+typedef struct _ACPI_MADT ACPI_MADT, *PACPI_MADT;
+typedef struct _ACPI_MADT_TABLE_LOCAL_APIC ACPI_MADT_TABLE_LOCAL_APIC, *PACPI_MADT_TABLE_LOCAL_APIC;
+typedef struct _ACPI_RSDP ACPI_RSDP, *PACPI_RSDP;
+typedef struct _ACPI_RSDT ACPI_RSDT, *PACPI_RSDT;
+typedef struct _ACPI_SUBTABLE_HEADER ACPI_SUBTABLE_HEADER, *PACPI_SUBTABLE_HEADER;
+typedef struct _ACPI_SYSTEM_INFO ACPI_SYSTEM_INFO, *PACPI_SYSTEM_INFO;
+typedef struct _ACPI_TIMER_INFO ACPI_TIMER_INFO, *PACPI_TIMER_INFO;
+typedef struct _ACPI_XSDT ACPI_XSDT, *PACPI_XSDT;
 typedef struct _ANSI_STRING ANSI_STRING, *PANSI_STRING;
 typedef struct _ANSI_STRING32 ANSI_STRING32, *PANSI_STRING32;
 typedef struct _ANSI_STRING64 ANSI_STRING64, *PANSI_STRING64;
 typedef struct _CPPORT CPPORT, *PCPPORT;
+typedef const struct _CMMPAGEMAP_ROUTINES CMMPAGEMAP_ROUTINES, *PCMMPAGEMAP_ROUTINES;
 typedef struct _CSTRING CSTRING, *PCSTRING;
 typedef struct _EFI_1394_DEVICE_PATH EFI_1394_DEVICE_PATH, *PEFI_1394_DEVICE_PATH;
 typedef struct _EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR, *PEFI_ACPI_ADDRESS_SPACE_DESCRIPTOR;
@@ -218,9 +233,14 @@ typedef struct _EXCEPTION_RECORD EXCEPTION_RECORD, *PEXCEPTION_RECORD;
 typedef struct _EXCEPTION_REGISTRATION_RECORD EXCEPTION_REGISTRATION_RECORD, *PEXCEPTION_REGISTRATION_RECORD;
 typedef struct _FIRMWARE_INFORMATION_BLOCK FIRMWARE_INFORMATION_BLOCK, *PFIRMWARE_INFORMATION_BLOCK;
 typedef struct _FLOAT128 FLOAT128, *PFLOAT128;
+typedef struct _GENERIC_ADDRESS GENERIC_ADDRESS, *PGENERIC_ADDRESS;
 typedef struct _GUID GUID, *PGUID;
+typedef struct _HL_FRAMEBUFFER_DATA HL_FRAMEBUFFER_DATA, *PHL_FRAMEBUFFER_DATA;
+typedef struct _HL_SCROLL_REGION_DATA HL_SCROLL_REGION_DATA, *PHL_SCROLL_REGION_DATA;
 typedef struct _KAPC KAPC, *PKAPC;
 typedef struct _KAPC_STATE KAPC_STATE, *PKAPC_STATE;
+typedef struct _KD_DEBUG_MODE KD_DEBUG_MODE, *PKD_DEBUG_MODE;
+typedef struct _KD_DISPATCH_TABLE KD_DISPATCH_TABLE, *PKD_DISPATCH_TABLE;
 typedef struct _KDPC KDPC, *PKDPC;
 typedef struct _KDPC_DATA KDPC_DATA, *PKDPC_DATA;
 typedef struct _KERNEL_INITIALIZATION_BLOCK KERNEL_INITIALIZATION_BLOCK, *PKERNEL_INITIALIZATION_BLOCK;
@@ -234,6 +254,16 @@ typedef struct _KSERVICE_DESCRIPTOR_TABLE KSERVICE_DESCRIPTOR_TABLE, *PKSERVICE_
 typedef struct _KSPIN_LOCK_QUEUE KSPIN_LOCK_QUEUE, *PKSPIN_LOCK_QUEUE;
 typedef struct _KTHREAD KTHREAD, *PKTHREAD;
 typedef struct _KTIMER KTIMER, *PKTIMER;
+typedef struct _KUBSAN_FLOAT_CAST_OVERFLOW_DATA KUBSAN_FLOAT_CAST_OVERFLOW_DATA, *PKUBSAN_FLOAT_CAST_OVERFLOW_DATA;
+typedef struct _KUBSAN_FUNCTION_TYPE_MISMATCH_DATA KUBSAN_FUNCTION_TYPE_MISMATCH_DATA, *PKUBSAN_FUNCTION_TYPE_MISMATCH_DATA;
+typedef struct _KUBSAN_INVALID_BUILTIN_DATA KUBSAN_INVALID_BUILTIN_DATA, *PKUBSAN_INVALID_BUILTIN_DATA;
+typedef struct _KUBSAN_OUT_OF_BOUNDS_DATA KUBSAN_OUT_OF_BOUNDS_DATA, *PKUBSAN_OUT_OF_BOUNDS_DATA;
+typedef struct _KUBSAN_OVERFLOW_DATA KUBSAN_OVERFLOW_DATA, *PKUBSAN_OVERFLOW_DATA;
+typedef struct _KUBSAN_SHIFT_OUT_OF_BOUNDS_DATA KUBSAN_SHIFT_OUT_OF_BOUNDS_DATA, *PKUBSAN_SHIFT_OUT_OF_BOUNDS_DATA;
+typedef struct _KUBSAN_SOURCE_LOCATION KUBSAN_SOURCE_LOCATION, *PKUBSAN_SOURCE_LOCATION;
+typedef struct _KUBSAN_TYPE_DESCRIPTOR KUBSAN_TYPE_DESCRIPTOR, *PKUBSAN_TYPE_DESCRIPTOR;
+typedef struct _KUBSAN_TYPE_MISMATCH_DATA KUBSAN_TYPE_MISMATCH_DATA, *PKUBSAN_TYPE_MISMATCH_DATA;
+typedef struct _KUBSAN_TYPE_MISMATCH_DATA_V1 KUBSAN_TYPE_MISMATCH_DATA_V1, *PKUBSAN_TYPE_MISMATCH_DATA_V1;
 typedef struct _KWAIT_BLOCK KWAIT_BLOCK, *PKWAIT_BLOCK;
 typedef struct _LDR_DATA_TABLE_ENTRY LDR_DATA_TABLE_ENTRY, *PLDR_DATA_TABLE_ENTRY;
 typedef struct _LIST_ENTRY LIST_ENTRY, *PLIST_ENTRY;
@@ -241,7 +271,7 @@ typedef struct _LIST_ENTRY32 LIST_ENTRY32, *PLIST_ENTRY32;
 typedef struct _LIST_ENTRY64 LIST_ENTRY64, *PLIST_ENTRY64;
 typedef struct _LOADER_GRAPHICS_INFORMATION_BLOCK LOADER_GRAPHICS_INFORMATION_BLOCK, *PLOADER_GRAPHICS_INFORMATION_BLOCK;
 typedef struct _LOADER_INFORMATION_BLOCK LOADER_INFORMATION_BLOCK, *PLOADER_INFORMATION_BLOCK;
-typedef struct _LOADER_MEMORY_MAPPING LOADER_MEMORY_MAPPING, *PLOADER_MEMORY_MAPPING;
+typedef struct _LOADER_MEMORY_DESCRIPTOR LOADER_MEMORY_DESCRIPTOR, *PLOADER_MEMORY_DESCRIPTOR;
 typedef struct _M128 M128, *PM128;
 typedef struct _MMCOLOR_TABLES MMCOLOR_TABLES, *PMMCOLOR_TABLES;
 typedef struct _MMPFNENTRY MMPFNENTRY, *PMMPFNENTRY;
@@ -273,8 +303,14 @@ typedef struct _PECOFF_IMAGE_ROM_HEADER PECOFF_IMAGE_ROM_HEADER, *PPECOFF_IMAGE_
 typedef struct _PECOFF_IMAGE_ROM_OPTIONAL_HEADER PECOFF_IMAGE_ROM_OPTIONAL_HEADER, *PPECOFF_IMAGE_ROM_OPTIONAL_HEADER;
 typedef struct _PECOFF_IMAGE_SECTION_HEADER PECOFF_IMAGE_SECTION_HEADER, *PPECOFF_IMAGE_SECTION_HEADER;
 typedef struct _PECOFF_IMAGE_VXD_HEADER PECOFF_IMAGE_VXD_HEADER, *PPECOFF_IMAGE_VXD_HEADER;
+typedef struct _PROCESSOR_IDENTITY PROCESSOR_IDENTITY, *PPROCESSOR_IDENTITY;
 typedef struct _PROCESSOR_POWER_STATE PROCESSOR_POWER_STATE, *PPROCESSOR_POWER_STATE;
+typedef struct _RTL_BITMAP RTL_BITMAP, *PRTL_BITMAP;
+typedef struct _RTL_PRINT_CONTEXT RTL_PRINT_CONTEXT, *PRTL_PRINT_CONTEXT;
+typedef struct _RTL_PRINT_FORMAT_PROPERTIES RTL_PRINT_FORMAT_PROPERTIES, *PRTL_PRINT_FORMAT_PROPERTIES;
 typedef struct _SINGLE_LIST_ENTRY SINGLE_LIST_ENTRY, *PSINGLE_LIST_ENTRY;
+typedef struct _SMBIOS_TABLE_HEADER SMBIOS_TABLE_HEADER, *PSMBIOS_TABLE_HEADER;
+typedef struct _SMBIOS3_TABLE_HEADER SMBIOS3_TABLE_HEADER, *PSMBIOS3_TABLE_HEADER;
 typedef struct _STRING STRING, *PSTRING;
 typedef struct _STRING32 STRING32, *PSTRING32;
 typedef struct _STRING64 STRING64, *PSTRING64;

sdk/xtdk/xttarget.h

@@ -29,6 +29,7 @@
     #define MEMORY_ALIGNMENT                        8
     #define MM_USERPAGE_TABLES                      1536
     #define MM_VIRTUAL_PAGESIZE                     20
+    #define STACK_ALIGNMENT                         4
 #elif defined(__amd64__) || defined(__x86_64__)
     #define _ARCH                                   amd64
     #define _ARCH_AMD64                             1
@@ -43,6 +44,7 @@
     #define MEMORY_ALIGNMENT                        16
     #define MM_USERPAGE_TABLES                      4194304
     #define MM_VIRTUAL_PAGESIZE                     52
+    #define STACK_ALIGNMENT                         16
 #else
     #error Unknown architecture
 #endif

xtldr/CMakeLists.txt

@@ -16,6 +16,7 @@ list(APPEND LIBXTLDR_SOURCE
 # Specify list of source code files
 list(APPEND XTLDR_SOURCE
     ${XTLDR_SOURCE_DIR}/arch/${ARCH}/memory.c
+    ${XTLDR_SOURCE_DIR}/bootutil.c
     ${XTLDR_SOURCE_DIR}/config.c
     ${XTLDR_SOURCE_DIR}/console.c
     ${XTLDR_SOURCE_DIR}/debug.c
@@ -49,6 +50,5 @@ set_install_target(xtldr efi/boot)
 
 # Set loader entrypoint and subsystem
 set_entrypoint(xtldr "BlStartXtLoader")
-set_imagebase(xtldr ${BASEADDRESS_XTLDR})
 set_linker_map(xtldr TRUE)
 set_subsystem(xtldr efi_application)

xtldr/arch/amd64/memory.c

@@ -4,6 +4,7 @@
  * FILE:            xtldr/arch/amd64/memory.c
  * DESCRIPTION:     XT Boot Loader AMD64 specific memory management
  * DEVELOPERS:      Rafal Kupiec <belliash@codingworkshop.eu.org>
+ *                  Aiken Harris <harraiken91@gmail.com>
  */
 
 #include <xtldr.h>
@@ -34,7 +35,7 @@ BlBuildPageMap(IN PXTBL_PAGE_MAPPING PageMap,
     EFI_STATUS Status;
 
     /* Allocate pages for the Page Map */
-    Status = BlAllocateMemoryPages(1, &Address);
+    Status = BlAllocateMemoryPages(AllocateAnyPages, 1, &Address);
     if(Status != STATUS_EFI_SUCCESS)
     {
         /* Memory allocation failure */
@@ -53,6 +54,15 @@ BlBuildPageMap(IN PXTBL_PAGE_MAPPING PageMap,
         return Status;
     }
 
+    /* Map the trampoline code area */
+    Status = BlMapVirtualMemory(PageMap, (PVOID)MM_TRAMPOLINE_ADDRESS,(PVOID)MM_TRAMPOLINE_ADDRESS,
+                                1, LoaderFirmwareTemporary);
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        /* Mapping trampoline code failed */
+        return Status;
+    }
+
     /* Get list of XTLDR modules */
     ModulesList = BlGetModulesList();
     ModulesListEntry = ModulesList->Flink;
@@ -149,12 +159,13 @@ BlBuildPageMap(IN PXTBL_PAGE_MAPPING PageMap,
 XTCDECL
 EFI_STATUS
 BlMapPage(IN PXTBL_PAGE_MAPPING PageMap,
-          IN UINT_PTR VirtualAddress,
-          IN UINT_PTR PhysicalAddress,
-          IN UINT NumberOfPages)
+          IN ULONG_PTR VirtualAddress,
+          IN ULONG_PTR PhysicalAddress,
+          IN ULONG NumberOfPages)
 {
+    PVOID Pml1, Pml2, Pml3, Pml4, Pml5;
     SIZE_T Pml1Entry, Pml2Entry, Pml3Entry, Pml4Entry, Pml5Entry;
-    PHARDWARE_PTE Pml1, Pml2, Pml3, Pml4, Pml5;
+    PHARDWARE_PTE PmlTable;
     SIZE_T PageFrameNumber;
     EFI_STATUS Status;
 
@@ -165,17 +176,17 @@ BlMapPage(IN PXTBL_PAGE_MAPPING PageMap,
     while(NumberOfPages > 0)
     {
         /* Calculate the indices in the various Page Tables from the virtual address */
-        Pml5Entry = (VirtualAddress & ((ULONGLONG)0x1FF << 48)) >> 48;
-        Pml4Entry = (VirtualAddress & ((ULONGLONG)0x1FF << 39)) >> 39;
-        Pml3Entry = (VirtualAddress & ((ULONGLONG)0x1FF << 30)) >> 30;
-        Pml2Entry = (VirtualAddress & ((ULONGLONG)0x1FF << 21)) >> 21;
-        Pml1Entry = (VirtualAddress & ((ULONGLONG)0x1FF << 12)) >> 12;
+        Pml5Entry = (VirtualAddress & ((ULONGLONG)0x1FF << MM_P5I_SHIFT)) >> MM_P5I_SHIFT;
+        Pml4Entry = (VirtualAddress & ((ULONGLONG)0x1FF << MM_PXI_SHIFT)) >> MM_PXI_SHIFT;
+        Pml3Entry = (VirtualAddress & ((ULONGLONG)0x1FF << MM_PPI_SHIFT)) >> MM_PPI_SHIFT;
+        Pml2Entry = (VirtualAddress & ((ULONGLONG)0x1FF << MM_PDI_SHIFT)) >> MM_PDI_SHIFT;
+        Pml1Entry = (VirtualAddress & ((ULONGLONG)0x1FF << MM_PTI_SHIFT)) >> MM_PTI_SHIFT;
 
         /* Check page map level */
         if(PageMap->PageMapLevel == 5)
         {
             /* Five level Page Map */
-            Pml5 = ((PHARDWARE_PTE)(PageMap->PtePointer));
+            Pml5 = PageMap->PtePointer;
 
             /* Get PML4 */
             Status = BlpGetNextPageTable(PageMap, Pml5, Pml5Entry, &Pml4);
@@ -188,7 +199,7 @@ BlMapPage(IN PXTBL_PAGE_MAPPING PageMap,
         else
         {
             /* Four level Page Map */
-            Pml4 = ((PHARDWARE_PTE)(PageMap->PtePointer));
+            Pml4 = PageMap->PtePointer;
         }
 
         /* Get PML3 */
@@ -216,9 +227,11 @@ BlMapPage(IN PXTBL_PAGE_MAPPING PageMap,
         }
 
         /* Set paging entry settings */
-        Pml1[Pml1Entry].PageFrameNumber = PageFrameNumber;
-        Pml1[Pml1Entry].Valid = 1;
-        Pml1[Pml1Entry].Writable = 1;
+        PmlTable = (PHARDWARE_PTE)Pml1;
+        RtlZeroMemory(&PmlTable[Pml1Entry], sizeof(HARDWARE_PTE));
+        PmlTable[Pml1Entry].PageFrameNumber = PageFrameNumber;
+        PmlTable[Pml1Entry].Valid = 1;
+        PmlTable[Pml1Entry].Writable = 1;
 
         /* Take next virtual address and PFN */
         VirtualAddress += EFI_PAGE_SIZE;
@@ -232,6 +245,81 @@ BlMapPage(IN PXTBL_PAGE_MAPPING PageMap,
     return STATUS_EFI_SUCCESS;
 }
 
+/**
+ * Returns next level of the Page Table.
+ *
+ * @param PageMap
+ *        Supplies a pointer to the page mapping structure.
+ *
+ * @param PageTable
+ *        Supplies a pointer to the current Page Table.
+ *
+ * @param Entry
+ *        Supplies an index of the current Page Table entry.
+ *
+ * @param NextPageTable
+ *        Supplies a pointer to the memory area where the next Page Table level is returned.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+EFI_STATUS
+BlpGetNextPageTable(IN PXTBL_PAGE_MAPPING PageMap,
+                    IN PVOID PageTable,
+                    IN SIZE_T Entry,
+                    OUT PVOID *NextPageTable)
+{
+    EFI_PHYSICAL_ADDRESS Address;
+    ULONGLONG PmlPointer = 0;
+    PHARDWARE_PTE PmlTable;
+    EFI_STATUS Status;
+
+    PmlTable = (PHARDWARE_PTE)PageTable;
+
+    /* Check if this is a valid table */
+    if(PmlTable[Entry].Valid)
+    {
+        /* Get PML pointer */
+        PmlPointer = PmlTable[Entry].PageFrameNumber;
+        PmlPointer <<= EFI_PAGE_SHIFT;
+    }
+    else
+    {
+        /* Allocate pages for new PML entry */
+        Status = BlAllocateMemoryPages(AllocateAnyPages, 1, &Address);
+        if(Status != STATUS_EFI_SUCCESS)
+        {
+            /* Memory allocation failure */
+            return Status;
+        }
+
+        /* Add new memory mapping */
+        Status = BlMapVirtualMemory(PageMap, NULL, (PVOID)(UINT_PTR)Address, 1, LoaderMemoryData);
+        if(Status != STATUS_EFI_SUCCESS)
+        {
+            /* Memory mapping failure */
+            return Status;
+        }
+
+        /* Fill allocated memory with zeros */
+        RtlZeroMemory((PVOID)(ULONGLONG)Address, EFI_PAGE_SIZE);
+
+        /* Set paging entry settings */
+        PmlTable[Entry].PageFrameNumber = Address / EFI_PAGE_SIZE;
+        PmlTable[Entry].Valid = 1;
+        PmlTable[Entry].Writable = 1;
+        PmlPointer = (ULONGLONG)Address;
+    }
+
+    /* Set next Page Map Level (PML) */
+    *NextPageTable = (PVOID)(ULONGLONG)PmlPointer;
+
+    /* Return success */
+    return STATUS_EFI_SUCCESS;
+}
+
 /**
  * Creates a recursive self mapping for all PML levels.
  *
@@ -250,26 +338,30 @@ EFI_STATUS
 BlpSelfMapPml(IN PXTBL_PAGE_MAPPING PageMap,
               IN ULONG_PTR SelfMapAddress)
 {
+    PHARDWARE_PTE PmlBase;
     ULONGLONG PmlIndex;
 
+    /* Initialize PML base pointer */
+    PmlBase = (PHARDWARE_PTE)PageMap->PtePointer;
+
     /* Check page map level */
     if(PageMap->PageMapLevel == 5)
     {
-        /* Self-mapping for PML5 is not supported */
-        BlDebugPrint(L"PML5 self-mapping not supported yet!\n");
-        return STATUS_EFI_UNSUPPORTED;
+        /* Calculate PML index based on provided self map address for PML5 */
+        PmlIndex = (SelfMapAddress >> MM_P5I_SHIFT) & 0x1FF;
     }
     else
     {
-        /* Calculate PML index based on provided self map address */
-        PmlIndex = (SelfMapAddress >> 39) & 0x1FF;
-
-        /* Add self-mapping for PML4 */
-        ((PHARDWARE_PTE)PageMap->PtePointer)[PmlIndex].PageFrameNumber = (UINT_PTR)PageMap->PtePointer / EFI_PAGE_SIZE;
-        ((PHARDWARE_PTE)PageMap->PtePointer)[PmlIndex].Valid = 1;
-        ((PHARDWARE_PTE)PageMap->PtePointer)[PmlIndex].Writable = 1;
+        /* Calculate PML index based on provided self map address for PML4 */
+        PmlIndex = (SelfMapAddress >> MM_PXI_SHIFT) & 0x1FF;
     }
 
+    /* Add self-mapping */
+    RtlZeroMemory(&PmlBase[PmlIndex], sizeof(HARDWARE_PTE));
+    PmlBase[PmlIndex].PageFrameNumber = (UINT_PTR)PageMap->PtePointer / EFI_PAGE_SIZE;
+    PmlBase[PmlIndex].Valid = 1;
+    PmlBase[PmlIndex].Writable = 1;
+
     /* Return success */
     return STATUS_EFI_SUCCESS;
 }

xtldr/arch/i686/memory.c

@@ -4,6 +4,7 @@
  * FILE:            xtldr/arch/i686/memory.c
  * DESCRIPTION:     XT Boot Loader i686 specific memory management
  * DEVELOPERS:      Rafal Kupiec <belliash@codingworkshop.eu.org>
+ *                  Aiken Harris <harraiken91@gmail.com>
  */
 
 #include <xtldr.h>
@@ -31,38 +32,54 @@ BlBuildPageMap(IN PXTBL_PAGE_MAPPING PageMap,
     EFI_STATUS Status;
     ULONG Index;
 
-    /* Allocate pages for the Page Map */
-    Status = BlAllocateMemoryPages(1, &Address);
-    if(Status != STATUS_EFI_SUCCESS)
+    /* Check the page map level to determine which paging structure to create */
+    if(PageMap->PageMapLevel == 3)
     {
-        /* Memory allocation failure */
-        return Status;
+        /* Allocate a page for the 3-level page map structure (PAE enabled) */
+        Status = BlAllocateMemoryPages(AllocateAnyPages, 1, &Address);
+        if(Status != STATUS_EFI_SUCCESS)
+        {
+            /* Memory allocation failed, cannot proceed with page map creation */
+            return Status;
+        }
+
+        /* Assign the allocated page to the page map and zero it out */
+        PageMap->PtePointer = (PVOID)(UINT_PTR)Address;
+        RtlZeroMemory(PageMap->PtePointer, EFI_PAGE_SIZE);
+
+        /* Allocate 4 pages for the Page Directories (PDs) */
+        Status = BlAllocateMemoryPages(AllocateAnyPages, 4, &DirectoryAddress);
+        if(Status != STATUS_EFI_SUCCESS)
+        {
+            /* Memory allocation failed, cannot proceed with page map creation */
+            return Status;
+        }
+
+        /* Zero-fill the allocated memory for the Page Directories */
+        RtlZeroMemory((PVOID)DirectoryAddress, EFI_PAGE_SIZE * 4);
+
+        /* Fill the PDPT with pointers to the Page Directories */
+        for(Index = 0; Index < 4; Index++)
+        {
+            RtlZeroMemory(&((PHARDWARE_MODERN_PTE)PageMap->PtePointer)[Index], sizeof(HARDWARE_MODERN_PTE));
+            ((PHARDWARE_MODERN_PTE)PageMap->PtePointer)[Index].PageFrameNumber = DirectoryAddress / EFI_PAGE_SIZE;
+            ((PHARDWARE_MODERN_PTE)PageMap->PtePointer)[Index].Valid = 1;
+            DirectoryAddress += EFI_PAGE_SIZE;
+        }
     }
-
-    /* Assign and zero-fill memory used by page mappings */
-    PageMap->PtePointer = (PVOID)(UINT_PTR)Address;
-    RtlZeroMemory(PageMap->PtePointer, EFI_PAGE_SIZE);
-
-    /* Allocate pages for the Page Directory */
-    Status = BlAllocateMemoryPages(4, &DirectoryAddress);
-    if(Status != STATUS_EFI_SUCCESS)
+    else
     {
-        /* Memory allocation failure */
-        return Status;
-    }
+        /* Allocate a page for the 2-level page map structure (PAE disabled) */
+        Status = BlAllocateMemoryPages(AllocateAnyPages, 1, &Address);
+        if(Status != STATUS_EFI_SUCCESS)
+        {
+            /* Memory allocation failed, cannot proceed with page map creation */
+            return Status;
+        }
 
-    /* Zero fill memory used by Page Directory */
-    RtlZeroMemory((PVOID)DirectoryAddress, EFI_PAGE_SIZE * 4);
-
-    /* Set the page directory into the PDPT and mark it present */
-    for(Index = 0; Index < 4; Index++)
-    {
-        /* Set paging entry settings */
-        ((PHARDWARE_PTE)PageMap->PtePointer)[Index].PageFrameNumber = DirectoryAddress / EFI_PAGE_SIZE;
-        ((PHARDWARE_PTE)PageMap->PtePointer)[Index].Valid = 1;
-
-        /* Next valid PFN address */
-        DirectoryAddress += EFI_PAGE_SIZE;
+        /* Assign the allocated page to the page map and zero it out */
+        PageMap->PtePointer = (PVOID)(UINT_PTR)Address;
+        RtlZeroMemory(PageMap->PtePointer, EFI_PAGE_SIZE);
     }
 
     /* Add page mapping itself to memory mapping */
@@ -73,6 +90,15 @@ BlBuildPageMap(IN PXTBL_PAGE_MAPPING PageMap,
         return Status;
     }
 
+    /* Map the trampoline code area */
+    Status = BlMapVirtualMemory(PageMap, (PVOID)MM_TRAMPOLINE_ADDRESS,(PVOID)MM_TRAMPOLINE_ADDRESS,
+                                1, LoaderFirmwareTemporary);
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        /* Mapping trampoline code failed */
+        return Status;
+    }
+
     /* Get list of XTLDR modules */
     ModulesList = BlGetModulesList();
     ModulesListEntry = ModulesList->Flink;
@@ -169,58 +195,81 @@ BlBuildPageMap(IN PXTBL_PAGE_MAPPING PageMap,
 XTCDECL
 EFI_STATUS
 BlMapPage(IN PXTBL_PAGE_MAPPING PageMap,
-          IN UINT_PTR VirtualAddress,
-          IN UINT_PTR PhysicalAddress,
-          IN UINT NumberOfPages)
+          IN ULONG_PTR VirtualAddress,
+          IN ULONG_PTR PhysicalAddress,
+          IN ULONG NumberOfPages)
 {
-    SIZE_T Pml1Entry, Pml2Entry, Pml3Entry;
-    PHARDWARE_PTE Pml1, Pml2, Pml3;
     SIZE_T PageFrameNumber;
+    PVOID Pml1, Pml2, Pml3;
+    SIZE_T Pml1Entry, Pml2Entry, Pml3Entry;
+    PHARDWARE_LEGACY_PTE LegacyPmlTable;
+    PHARDWARE_MODERN_PTE PmlTable;
     EFI_STATUS Status;
 
     /* Set the Page Frame Number (PFN) */
     PageFrameNumber = PhysicalAddress >> EFI_PAGE_SHIFT;
 
-    /* Do the recursive mapping */
+    /* Map all requested pages */
     while(NumberOfPages > 0)
     {
-        /* Calculate the indices in the various Page Tables from the virtual address */
-        Pml3Entry = (VirtualAddress & ((ULONGLONG)0x1FF << 30)) >> 30;
-        Pml2Entry = (VirtualAddress & ((ULONGLONG)0x1FF << 21)) >> 21;
-        Pml1Entry = (VirtualAddress & ((ULONGLONG)0x1FF << 12)) >> 12;
-
-        /* Check page map level */
+        /* Check the paging mode to use the correct page table structure */
         if(PageMap->PageMapLevel == 3)
         {
-            /* Three level Page Map */
-            Pml3 = ((PHARDWARE_PTE)(PageMap->PtePointer));
+            /* Calculate the indices for PAE page tables */
+            Pml3Entry = (VirtualAddress >> 30) & 0x3;
+            Pml2Entry = (VirtualAddress >> 21) & 0x1FF;
+            Pml1Entry = (VirtualAddress >> 12) & 0x1FF;
 
-            /* Get PML2 */
+            /* Get Page Directory Pointer Table (PML3) */
+            Pml3 = PageMap->PtePointer;
+
+            /* Get Page Directory (PML2) */
             Status = BlpGetNextPageTable(PageMap, Pml3, Pml3Entry, &Pml2);
             if(Status != STATUS_EFI_SUCCESS)
             {
-                /* Memory mapping failure */
+                /* Failed to get the Page Table, abort mapping */
                 return Status;
             }
+
+            /* Get Page Table (PML1) */
+            Status = BlpGetNextPageTable(PageMap, Pml2, Pml2Entry, &Pml1);
+            if(Status != STATUS_EFI_SUCCESS)
+            {
+                /* Failed to get the Page Table, abort mapping */
+                return Status;
+            }
+
+            /* Set the 64-bit PTE entry */
+            PmlTable = (PHARDWARE_MODERN_PTE)Pml1;
+            RtlZeroMemory(&PmlTable[Pml1Entry], sizeof(HARDWARE_MODERN_PTE));
+            PmlTable[Pml1Entry].PageFrameNumber = PageFrameNumber;
+            PmlTable[Pml1Entry].Valid = 1;
+            PmlTable[Pml1Entry].Writable = 1;
         }
         else
         {
-            /* Two level Page Map */
-            Pml2 = ((PHARDWARE_PTE)(PageMap->PtePointer));
-        }
+            /* Calculate the indices for non-PAE page tables */
+            Pml2Entry = (VirtualAddress >> 22) & 0x3FF;
+            Pml1Entry = (VirtualAddress >> 12) & 0x3FF;
 
-        /* Get PML1 */
-        Status = BlpGetNextPageTable(PageMap, Pml2, Pml2Entry, &Pml1);
-        if(Status != STATUS_EFI_SUCCESS)
-        {
-            /* Memory mapping failure */
-            return Status;
-        }
+            /* Get Page Directory (PML2) */
+            Pml2 = PageMap->PtePointer;
 
-        /* Set paging entry settings */
-        Pml1[Pml1Entry].PageFrameNumber = PageFrameNumber;
-        Pml1[Pml1Entry].Valid = 1;
-        Pml1[Pml1Entry].Writable = 1;
+            /* Get Page Table (PML1) */
+            Status = BlpGetNextPageTable(PageMap, Pml2, Pml2Entry, &Pml1);
+            if(Status != STATUS_EFI_SUCCESS)
+            {
+                /* Failed to get the Page Table, abort mapping */
+                return Status;
+            }
+
+            /* Set the 32-bit PTE entry */
+            LegacyPmlTable = (PHARDWARE_LEGACY_PTE)Pml1;
+            RtlZeroMemory(&LegacyPmlTable[Pml1Entry], sizeof(HARDWARE_LEGACY_PTE));
+            LegacyPmlTable[Pml1Entry].PageFrameNumber = (UINT32)PageFrameNumber;
+            LegacyPmlTable[Pml1Entry].Valid = 1;
+            LegacyPmlTable[Pml1Entry].Writable = 1;
+        }
 
         /* Take next virtual address and PFN */
         VirtualAddress += EFI_PAGE_SIZE;
@@ -234,6 +283,119 @@ BlMapPage(IN PXTBL_PAGE_MAPPING PageMap,
     return STATUS_EFI_SUCCESS;
 }
 
+/**
+ * Returns next level of the Page Table.
+ *
+ * @param PageMap
+ *        Supplies a pointer to the page mapping structure.
+ *
+ * @param PageTable
+ *        Supplies a pointer to the current Page Table.
+ *
+ * @param Entry
+ *        Supplies an index of the current Page Table entry.
+ *
+ * @param NextPageTable
+ *        Supplies a pointer to the memory area where the next Page Table level is returned.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+EFI_STATUS
+BlpGetNextPageTable(IN PXTBL_PAGE_MAPPING PageMap,
+                    IN PVOID PageTable,
+                    IN SIZE_T Entry,
+                    OUT PVOID *NextPageTable)
+{
+    EFI_PHYSICAL_ADDRESS Address;
+    ULONGLONG PmlPointer = 0;
+    EFI_STATUS Status;
+    PHARDWARE_LEGACY_PTE LegacyPmlTable;
+    PHARDWARE_MODERN_PTE PmlTable;
+    BOOLEAN ValidPte = FALSE;
+
+    /* Check page map level to determine PTE size */
+    if(PageMap->PageMapLevel >= 3)
+    {
+        /* 64-bit PTE for PML3 (PAE enabled) */
+        PmlTable = (PHARDWARE_MODERN_PTE)PageTable;
+        if(PmlTable[Entry].Valid)
+        {
+            /* Get page frame number from page table entry */
+            PmlPointer = PmlTable[Entry].PageFrameNumber;
+            ValidPte = TRUE;
+        }
+    }
+    else
+    {
+        /* 32-bit PTE for PML2 (PAE disabled) */
+        LegacyPmlTable = (PHARDWARE_LEGACY_PTE)PageTable;
+        if(LegacyPmlTable[Entry].Valid)
+        {
+            /* Get page frame number from page table entry */
+            PmlPointer = LegacyPmlTable[Entry].PageFrameNumber;
+            ValidPte = TRUE;
+        }
+    }
+
+    /* Check if page table entry is valid */
+    if(ValidPte)
+    {
+        /* Calculate the base address of the next page table */
+        PmlPointer <<= EFI_PAGE_SHIFT;
+    }
+    else
+    {
+        /* Allocate pages for new PML entry */
+        Status = BlAllocateMemoryPages(AllocateAnyPages, 1, &Address);
+        if(Status != STATUS_EFI_SUCCESS)
+        {
+            /* Memory allocation failure */
+            return Status;
+        }
+
+        /* Add new memory mapping */
+        Status = BlMapVirtualMemory(PageMap, NULL, (PVOID)(UINT_PTR)Address, 1, LoaderMemoryData);
+        if(Status != STATUS_EFI_SUCCESS)
+        {
+            /* Memory mapping failure */
+            return Status;
+        }
+
+        /* Fill allocated memory with zeros */
+        RtlZeroMemory((PVOID)(ULONGLONG)Address, EFI_PAGE_SIZE);
+
+        /* Set paging entry settings based on level */
+        if(PageMap->PageMapLevel >= 3)
+        {
+            /* 64-bit PTE for PML3 (PAE enabled) */
+            PmlTable = (PHARDWARE_MODERN_PTE)PageTable;
+            PmlTable[Entry].PageFrameNumber = Address / EFI_PAGE_SIZE;
+            PmlTable[Entry].Valid = 1;
+            PmlTable[Entry].Writable = 1;
+        }
+        else
+        {
+            /* 32-bit PTE for PML2 (PAE disabled) */
+            LegacyPmlTable = (PHARDWARE_LEGACY_PTE)PageTable;
+            LegacyPmlTable[Entry].PageFrameNumber = (UINT32)(Address / EFI_PAGE_SIZE);
+            LegacyPmlTable[Entry].Valid = 1;
+            LegacyPmlTable[Entry].Writable = 1;
+        }
+
+        /* Return the address of the new page table */
+        PmlPointer = (ULONGLONG)Address;
+    }
+
+    /* Set next Page Map Level (PML) */
+    *NextPageTable = (PVOID)(ULONGLONG)PmlPointer;
+
+    /* Return success */
+    return STATUS_EFI_SUCCESS;
+}
+
 /**
  * Creates a recursive self mapping for all PML levels.
  *
@@ -252,36 +414,41 @@ EFI_STATUS
 BlpSelfMapPml(IN PXTBL_PAGE_MAPPING PageMap,
               IN ULONG_PTR SelfMapAddress)
 {
+    PHARDWARE_LEGACY_PTE LegacyPml;
+    PHARDWARE_MODERN_PTE Pml;
     ULONGLONG PmlIndex;
-    PHARDWARE_PTE Pml;
     ULONG Index;
 
     /* Check page map level */
     if(PageMap->PageMapLevel == 3)
     {
         /* Calculate PML index based on provided self map address */
-        PmlIndex = (SelfMapAddress >> 21) & 0x1FF;
+        PmlIndex = (SelfMapAddress >> MM_PDI_SHIFT) & 0x1FF;
 
         /* Get Page Directory */
-        Pml = (PHARDWARE_PTE)(((PHARDWARE_PTE)PageMap->PtePointer)[SelfMapAddress >> 30].PageFrameNumber * EFI_PAGE_SIZE);
+        Pml = (PHARDWARE_MODERN_PTE)(((PHARDWARE_MODERN_PTE)PageMap->PtePointer)[SelfMapAddress >> MM_PPI_SHIFT].PageFrameNumber * EFI_PAGE_SIZE);
 
         /* Add self-mapping for PML3 (PAE enabled) */
         for(Index = 0; Index < 4; Index++)
         {
-            Pml[PmlIndex + Index].PageFrameNumber = ((PHARDWARE_PTE)PageMap->PtePointer)[Index].PageFrameNumber;
+            RtlZeroMemory(&Pml[PmlIndex + Index], sizeof(HARDWARE_MODERN_PTE));
+            Pml[PmlIndex + Index].PageFrameNumber = ((PHARDWARE_MODERN_PTE)PageMap->PtePointer)[Index].PageFrameNumber;
             Pml[PmlIndex + Index].Valid = 1;
             Pml[PmlIndex + Index].Writable = 1;
         }
     }
     else
     {
+        LegacyPml = (PHARDWARE_LEGACY_PTE)PageMap->PtePointer;
+
         /* Calculate PML index based on provided self map address */
-        PmlIndex = (SelfMapAddress >> 22);
+        PmlIndex = (SelfMapAddress >> MM_PDI_LEGACY_SHIFT);
 
         /* Add self-mapping for PML2 (PAE disabled) */
-        ((PHARDWARE_LEGACY_PTE)PageMap->PtePointer)[PmlIndex].PageFrameNumber = (UINT_PTR)PageMap->PtePointer / EFI_PAGE_SIZE;
-        ((PHARDWARE_LEGACY_PTE)PageMap->PtePointer)[PmlIndex].Valid = 1;
-        ((PHARDWARE_LEGACY_PTE)PageMap->PtePointer)[PmlIndex].Writable = 1;
+        RtlZeroMemory(&LegacyPml[PmlIndex], sizeof(HARDWARE_LEGACY_PTE));
+        LegacyPml[PmlIndex].PageFrameNumber = (UINT_PTR)PageMap->PtePointer / EFI_PAGE_SIZE;
+        LegacyPml[PmlIndex].Valid = 1;
+        LegacyPml[PmlIndex].Writable = 1;
     }
 
     /* Return success */

xtldr/bootutil.c

@@ -0,0 +1,87 @@
+/**
+ * PROJECT:         ExectOS
+ * COPYRIGHT:       See COPYING.md in the top level directory
+ * FILE:            xtldr/bootutil.c
+ * DESCRIPTION:     Helper functions used by the boot protocol during system startup
+ * DEVELOPERS:      Aiken Harris <harraiken91@gmail.com>
+ */
+
+#include <xtldr.h>
+
+
+/**
+ * Checks if a specific option exists in the list of provided boot parameters.
+ *
+ * @param Parameters
+ *        A pointer to the wide-character string containing the boot parameters, separated by spaces.
+ *
+ * @param Needle
+ *        A pointer to the wide-character string representing the kernel option to find.
+ *
+ * @return This routine returns TRUE if the option is found, otherwise FALSE.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+BOOLEAN
+BlGetBooleanParameter(IN CONST PWCHAR Parameters,
+                      IN CONST PWCHAR Needle)
+{
+    PWCHAR CurrentPosition, TokenEnd, TokenStart;
+    SIZE_T NeedleLength, TokenLength;
+
+    /* Validate input data and ensure the option is not an empty string */
+    if(Parameters == NULL || Needle == NULL || *Needle == L'\0')
+    {
+        /* One of the parameters was invalid */
+        return FALSE;
+    }
+
+    CurrentPosition = Parameters;
+    NeedleLength = RtlWideStringLength(Needle, 0);
+
+    /* Iterate through the entire parameters string */
+    while(*CurrentPosition != L'\0')
+    {
+        /* Skip any leading whitespace to find the start of the token */
+        while(*CurrentPosition == L' ')
+        {
+            CurrentPosition++;
+        }
+
+        /* Check if end of the string has been reached */
+        if(*CurrentPosition == L'\0')
+        {
+            /* End of string reached, no more tokens */
+            break;
+        }
+
+        /* Identify the boundaries of the current token */
+        TokenStart = CurrentPosition;
+        TokenEnd = TokenStart;
+        while(*TokenEnd != L'\0' && *TokenEnd != L' ')
+        {
+            TokenEnd++;
+        }
+
+        /* Calculate the length of the token found */
+        TokenLength = TokenEnd - TokenStart;
+
+         /* Compare the token length */
+        if(TokenLength == NeedleLength)
+        {
+            /* Length matches, compare the strings */
+            if(RtlCompareWideStringInsensitive(TokenStart, Needle, NeedleLength) == 0)
+            {
+                /* A match was found */
+                return TRUE;
+            }
+        }
+
+        /* Move the position past the current token to continue the search */
+        CurrentPosition = TokenEnd;
+    }
+
+    /* No match was found */
+    return FALSE;
+}

xtldr/config.c

@@ -4,11 +4,84 @@
  * FILE:            xtldr/config.c
  * DESCRIPTION:     XT Boot Loader Configuration
  * DEVELOPERS:      Rafal Kupiec <belliash@codingworkshop.eu.org>
+ *                  Aiken Harris <harraiken91@gmail.com>
  */
 
 #include <xtldr.h>
 
 
+/**
+ * @brief Retrieves the value of a specific OS boot option from a list.
+ *
+ * @param Options
+ *        A pointer to the head of a list of XTBL_CONFIG_ENTRY structures.
+ *
+ * @param OptionName
+ *        A pointer to wide string that contains the name of the boot option to retrieve.
+ *
+ * @param OptionValue
+ *        A pointer to a variable that receives a pointer to the retrieved boot option's value.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+EFI_STATUS
+BlGetBootOptionValue(IN PLIST_ENTRY Options,
+                     IN CONST PWCHAR OptionName,
+                     OUT PWCHAR *OptionValue)
+{
+    PXTBL_CONFIG_ENTRY ConfigEntry;
+    PLIST_ENTRY ConfigList;
+    ULONG KeyLength, ValueLength;
+    EFI_STATUS Status;
+
+    /* Assume the option will not be found */
+    *OptionValue = NULL;
+
+    /* Get the length of the option name we are looking for */
+    KeyLength = RtlWideStringLength(OptionName, 0);
+
+    /* Start iterating from the first entry in the options list */
+    ConfigList = Options->Flink;
+    while(ConfigList != Options)
+    {
+        /* Get the container record for the current config entry */
+        ConfigEntry = CONTAIN_RECORD(ConfigList, XTBL_CONFIG_ENTRY, Flink);
+
+        /* Compare the current entry's name with the requested option name */
+        if(RtlCompareWideStringInsensitive(ConfigEntry->Name, OptionName, KeyLength) == 0)
+        {
+            /* Found the option, now prepare to copy its value */
+            ValueLength = RtlWideStringLength(ConfigEntry->Value, 0);
+
+            /* Allocate memory for the output value string */
+            Status = BlAllocateMemoryPool((ValueLength + 1) * sizeof(WCHAR), (PVOID *)OptionValue);
+            if(Status != STATUS_EFI_SUCCESS)
+            {
+                /* Memory allocation failure, print debug message and return status code */
+                BlDebugPrint(L"ERROR: Memory allocation failure (Status Code: 0x%zX)\n", Status);
+                *OptionValue = NULL;
+                return Status;
+            }
+
+            /* Copy the value and NULL-terminate the new string */
+            RtlCopyMemory(*OptionValue, ConfigEntry->Value, ValueLength * sizeof(WCHAR));
+            (*OptionValue)[ValueLength] = L'\0';
+
+            /* Successfully retrieved the option value, return success */
+            return STATUS_EFI_SUCCESS;
+        }
+
+        /* Move to the next entry in the list */
+        ConfigList = ConfigList->Flink;
+    }
+
+    /* Option not found */
+    return STATUS_EFI_NOT_FOUND;
+}
+
 /**
  * Returns a boolean value of the specified configuration key.
  *
@@ -26,7 +99,7 @@ BlGetConfigBooleanValue(IN CONST PWCHAR ConfigName)
     PWCHAR Value;
 
     /* Get config value */
-    Value = BlGetConfigValue(ConfigName);
+    BlGetConfigValue(ConfigName, &Value);
 
     /* Check if option is enabled */
     if(RtlCompareWideStringInsensitive(Value, L"ENABLED", 0) == 0 ||
@@ -53,8 +126,9 @@ BlGetConfigBooleanValue(IN CONST PWCHAR ConfigName)
  * @since XT 1.0
  */
 XTCDECL
-PWCHAR
-BlGetConfigValue(IN CONST PWCHAR ConfigName)
+EFI_STATUS
+BlGetConfigValue(IN CONST PWCHAR ConfigName,
+                 OUT PWCHAR *ConfigValue)
 {
     PXTBL_CONFIG_ENTRY ConfigEntry;
     PLIST_ENTRY ConfigListEntry;
@@ -62,6 +136,9 @@ BlGetConfigValue(IN CONST PWCHAR ConfigName)
     EFI_STATUS Status;
     PWCHAR Value;
 
+    /* Assume the option will not be found */
+    *ConfigValue = NULL;
+
     /* Get config entry name length */
     KeyLength = RtlWideStringLength(ConfigName, 0);
 
@@ -84,13 +161,14 @@ BlGetConfigValue(IN CONST PWCHAR ConfigName)
             {
                 /* Memory allocation failure, return NULL */
                 BlDebugPrint(L"ERROR: Memory allocation failure (Status Code: 0x%zX)\n", Status);
-                return NULL;
+                return Status;
             }
 
             /* Copy value and return it */
             RtlCopyMemory(Value, ConfigEntry->Value, ValueLength * sizeof(WCHAR));
             Value[ValueLength] = L'\0';
-            return Value;
+            *ConfigValue = Value;
+            return STATUS_EFI_SUCCESS;
         }
 
         /* Move to the next config entry */
@@ -98,7 +176,157 @@ BlGetConfigValue(IN CONST PWCHAR ConfigName)
     }
 
     /* Config entry not found, return NULL */
-    return NULL;
+    return STATUS_EFI_NOT_FOUND;
+}
+
+/**
+ * Retrieves the list of user-editable boot options.
+ *
+ * @param OptionsArray
+ *        A pointer to a variable that will receive the pointer to the array of editable option names.
+ *
+ * @param OptionsCount
+ *        A pointer to a variable that will be updated with the number of elements in the OptionsArray.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+VOID
+BlGetEditableOptions(OUT CONST PWCHAR **OptionsArray,
+                     OUT PULONG OptionsCount)
+{
+    ULONG Count = 0;
+
+    /* Return a pointer to the global array of editable options */
+    *OptionsArray = BlpEditableConfigOptions;
+
+    /* Calculate the number of elements in the array */
+    while(BlpEditableConfigOptions[Count])
+    {
+        Count++;
+    }
+
+    /* Return the number of elements */
+    *OptionsCount = Count;
+}
+
+/**
+ * Sets the value of a specific OS boot option in a list, or adds it if it doesn't exist.
+ *
+ * @param Options
+ *       A pointer to the head of a list of XTBL_CONFIG_ENTRY structures.
+ *
+ * @param OptionName
+ *       A pointer to a wide string that contains the name of the boot option to set.
+ *
+ * @param OptionValue
+ *       A pointer to a wide string that contains the new value for the boot option.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+EFI_STATUS
+BlSetBootOptionValue(IN PLIST_ENTRY Options,
+                     IN CONST PWCHAR OptionName,
+                     IN CONST PWCHAR OptionValue)
+{
+    PXTBL_CONFIG_ENTRY ConfigEntry;
+    PLIST_ENTRY ConfigList;
+    ULONG Length;
+    EFI_STATUS Status;
+
+    /* Get the length of the option name we are looking for */
+    Length = RtlWideStringLength(OptionName, 0);
+
+    /* Start iterating from the first entry in the options list */
+    ConfigList = Options->Flink;
+    while(ConfigList != Options)
+    {
+        /* Get the container record for the current config entry */
+        ConfigEntry = CONTAIN_RECORD(ConfigList, XTBL_CONFIG_ENTRY, Flink);
+
+        /* Compare the current entry's name with the requested option name */
+        if(RtlCompareWideStringInsensitive(ConfigEntry->Name, OptionName, Length) == 0)
+        {
+            /* Found the option, get its length */
+            Length = RtlWideStringLength(OptionValue, 0);
+
+            /* Reallocate memory for the new value */
+            Status = BlFreeMemoryPool(ConfigEntry->Value);
+            if(Status != STATUS_EFI_SUCCESS)
+            {
+                /* Failed to free memory, return status code */
+                return Status;
+            }
+
+            /* Allocate new memory for the updated value */
+            Status = BlAllocateMemoryPool((Length + 1) * sizeof(WCHAR), (PVOID *)&ConfigEntry->Value);
+            if(Status != STATUS_EFI_SUCCESS)
+            {
+                /* Memory allocation failure, print debug message and return status code */
+                BlDebugPrint(L"ERROR: Memory allocation failure (Status Code: 0x%zX)\\n", Status);
+                return Status;
+            }
+
+            /* Copy the value and NULL-terminate the new string */
+            RtlCopyMemory(ConfigEntry->Value, OptionValue, Length * sizeof(WCHAR));
+            ConfigEntry->Value[Length] = L'\0';
+            return STATUS_EFI_SUCCESS;
+        }
+
+        /* Move to the next entry in the list */
+        ConfigList = ConfigList->Flink;
+    }
+
+    /* Option not found, allocate memory for the new one */
+    Status = BlAllocateMemoryPool(sizeof(XTBL_CONFIG_ENTRY), (PVOID *)&ConfigEntry);
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        /* Memory allocation failure, print debug message and return status code */
+        BlDebugPrint(L"ERROR: Memory allocation failure (Status Code: 0x%zX)\\n", Status);
+        return Status;
+    }
+
+    /* Allocate memory for the option name */
+    Length = RtlWideStringLength(OptionName, 0);
+    Status = BlAllocateMemoryPool((Length + 1) * sizeof(WCHAR), (PVOID *)&ConfigEntry->Name);
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        /* Memory allocation failure, print debug message and return status code */
+        BlDebugPrint(L"ERROR: Memory allocation failure (Status Code: 0x%zX)\\n", Status);
+        BlFreeMemoryPool(ConfigEntry);
+        return Status;
+    }
+
+    /* Copy the option name and NULL-terminate the new string */
+    RtlCopyMemory(ConfigEntry->Name, OptionName, Length * sizeof(WCHAR));
+    ConfigEntry->Name[Length] = L'\0';
+
+    /* Allocate memory for the option value */
+    Length = RtlWideStringLength(OptionValue, 0);
+    Status = BlAllocateMemoryPool((Length + 1) * sizeof(WCHAR), (PVOID *)&ConfigEntry->Value);
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        /* Memory allocation failure, print debug message and return status code */
+        BlDebugPrint(L"ERROR: Memory allocation failure (Status Code: 0x%zX)\\n", Status);
+        BlFreeMemoryPool(ConfigEntry->Name);
+        BlFreeMemoryPool(ConfigEntry);
+        return Status;
+    }
+
+    /* Copy the value and NULL-terminate the new string */
+    RtlCopyMemory(ConfigEntry->Value, OptionValue, Length * sizeof(WCHAR));
+    ConfigEntry->Value[Length] = L'\0';
+
+    /* Insert the new config entry at the end of the options list */
+    RtlInsertTailList(Options, &ConfigEntry->Flink);
+
+    /* Return success */
+    return STATUS_EFI_SUCCESS;
 }
 
 /**
@@ -110,7 +338,7 @@ BlGetConfigValue(IN CONST PWCHAR ConfigName)
  * @param ConfigValue
  *        Specifies the new configuration value.
  *
- * @return This routine returns status code.
+ * @return This routine returns a status code.
  *
  * @since XT 1.0
  */
@@ -174,7 +402,7 @@ BlSetConfigValue(IN CONST PWCHAR ConfigName,
 /**
  * Loads and parses XTLDR configuration file.
  *
- * @return This routine returns status code.
+ * @return This routine returns a status code.
  *
  * @since XT 1.0
  */
@@ -453,7 +681,7 @@ BlpParseConfigFile(IN CONST PCHAR RawConfig,
 
             /* Initialize new section and convert its name to wide string */
             RtlInitializeListHead(&Section->Options);
-            RtlStringToWideString(Section->SectionName, &SectionName, SectionLength);
+            RtlStringToWideString(Section->SectionName, (PCSTR*)&SectionName, SectionLength);
 
             /* Ensure string is NULL-terminated and add new section to the configuration list */
             Section->SectionName[SectionLength] = L'\0';
@@ -542,8 +770,8 @@ BlpParseConfigFile(IN CONST PCHAR RawConfig,
             }
 
             /* Convert key and value to wide strings */
-            RtlStringToWideString(Option->Name, &Key, RtlStringLength(Key, 0) + 1);
-            RtlStringToWideString(Option->Value, &Value, RtlStringLength(Value, 0) + 1);
+            RtlStringToWideString(Option->Name, (PCSTR*)&Key, RtlStringLength(Key, 0) + 1);
+            RtlStringToWideString(Option->Value, (PCSTR*)&Value, RtlStringLength(Value, 0) + 1);
 
             /* Ensure strings are NULL-terminated and add new option to the list */
             Option->Name[KeyLength] = L'\0';
@@ -568,7 +796,7 @@ BlpParseConfigFile(IN CONST PCHAR RawConfig,
  * @param ConfigData
  *        Provides a buffer to store the data read from the configuration file.
  *
- * @return This routine returns status code.
+ * @return This routine returns a status code.
  *
  * @since XT 1.0
  */
@@ -629,6 +857,7 @@ VOID
 BlpUpdateConfiguration(IN PLIST_ENTRY NewConfig)
 {
     PXTBL_CONFIG_ENTRY ConfigEntry;
+    PWCHAR ConfigValue;
     PLIST_ENTRY ConfigListEntry, NextListEntry;
 
     /* Iterate through new config entries */
@@ -642,7 +871,8 @@ BlpUpdateConfiguration(IN PLIST_ENTRY NewConfig)
         NextListEntry = ConfigListEntry->Flink;
 
         /* Make sure config entry does not exist yet */
-        if(BlGetConfigValue(ConfigEntry->Name) == NULL)
+        BlGetConfigValue(ConfigEntry->Name, &ConfigValue);
+        if(ConfigValue == NULL)
         {
             /* Remove new config entry from input list and put it into global config list */
             RtlRemoveEntryList(&ConfigEntry->Flink);

xtldr/console.c

@@ -95,7 +95,7 @@ BlEnableConsoleCursor()
  */
 XTCDECL
 VOID
-BlConsolePrint(IN PUSHORT Format,
+BlConsolePrint(IN PCWSTR Format,
                IN ...)
 {
     RTL_PRINT_CONTEXT ConsolePrintContext, SerialPrintContext;
@@ -138,9 +138,9 @@ BlConsolePrint(IN PUSHORT Format,
  */
 XTCDECL
 VOID
-BlConsoleWrite(IN PUSHORT String)
+BlConsoleWrite(IN PCWSTR String)
 {
-    EfiSystemTable->ConOut->OutputString(EfiSystemTable->ConOut, String);
+    EfiSystemTable->ConOut->OutputString(EfiSystemTable->ConOut, (PWSTR)String);
 }
 
 /**
@@ -292,9 +292,9 @@ BlSetCursorPosition(IN ULONGLONG PosX,
  */
 XTCDECL
 XTSTATUS
-BlpConsolePutChar(IN USHORT Character)
+BlpConsolePutChar(IN WCHAR Character)
 {
-    USHORT Buffer[2];
+    WCHAR Buffer[2];
 
     /* Check if character is a newline ('\n') */
     if(Character == L'\n')

xtldr/debug.c

@@ -24,7 +24,7 @@
  */
 XTCDECL
 VOID
-BlDebugPrint(IN PUSHORT Format,
+BlDebugPrint(IN PCWSTR Format,
              IN ...)
 {
     RTL_PRINT_CONTEXT ConsolePrintContext, SerialPrintContext;
@@ -71,17 +71,14 @@ BlDebugPrint(IN PUSHORT Format,
  */
 XTCDECL
 XTSTATUS
-BlpDebugPutChar(IN USHORT Character)
+BlpDebugPutChar(IN WCHAR Character)
 {
-    USHORT Buffer[2];
+    WCHAR Buffer[2];
 
     /* Write character to the serial console */
     Buffer[0] = Character;
     Buffer[1] = 0;
-    HlComPortPutByte(&BlpStatus.SerialPort, Buffer[0]);
-
-    /* Return success */
-    return STATUS_EFI_SUCCESS;
+    return HlComPortPutByte(&BlpStatus.SerialPort, Buffer[0]);
 }
 
 /**
@@ -105,7 +102,7 @@ BlpInitializeDebugConsole()
     BaudRate = 0;
 
     /* Get debug configuration */
-    DebugConfiguration = BlGetConfigValue(L"DEBUG");
+    BlGetConfigValue(L"DEBUG", &DebugConfiguration);
 
     /* Make sure any debug options are provided and debug console is not initialized yet */
     if(DebugConfiguration && BlpStatus.DebugPort == 0)
@@ -232,18 +229,35 @@ BlpInitializeSerialPort(IN ULONG PortNumber,
     EFI_STATUS EfiStatus;
     XTSTATUS Status;
 
-    /* Print debug message depending on port settings */
-    if(PortAddress)
+    /* Check if custom COM port address supplied */
+    if(!PortAddress)
     {
-        BlConsolePrint(L"Initializing serial console at COM port address: 0x%lX\n", PortAddress);
+        /* We support only a pre-defined number of ports */
+        if(PortNumber > COMPORT_COUNT)
+        {
+            /* Fail if wrong/unsupported port used */
+            return STATUS_INVALID_PARAMETER;
+        }
+
+        /* Check if serial port is set */
+        if(PortNumber == 0)
+        {
+            /* Use COM1 by default */
+            PortNumber = 1;
+        }
+
+        /* Set custom port address based on the port number and print debug message */
+        PortAddress = BlComPortList[PortNumber - 1];
+        BlConsolePrint(L"Initializing serial console at port COM%d\n", PortNumber);
     }
     else
     {
-        BlConsolePrint(L"Initializing serial console at port COM%d\n", PortNumber);
+        /* Custom port address supplied, print debug message */
+        BlConsolePrint(L"Initializing serial console at COM port address: 0x%lX\n", PortAddress);
     }
 
     /* Initialize COM port */
-    Status = HlInitializeComPort(&BlpStatus.SerialPort, PortNumber, UlongToPtr(PortAddress), BaudRate);
+    Status = HlInitializeComPort(&BlpStatus.SerialPort, UlongToPtr(PortAddress), BaudRate);
 
     /* Port not found under supplied address */
     if(Status == STATUS_NOT_FOUND && PortAddress)
@@ -254,7 +268,7 @@ BlpInitializeSerialPort(IN ULONG PortNumber,
         {
             /* Try to reinitialize COM port */
             BlConsolePrint(L"Enabled I/O space access for all PCI(E) serial controllers found\n");
-            Status = HlInitializeComPort(&BlpStatus.SerialPort, PortNumber, UlongToPtr(PortAddress), BaudRate);
+            Status = HlInitializeComPort(&BlpStatus.SerialPort, UlongToPtr(PortAddress), BaudRate);
         }
     }
 

xtldr/globals.c

@@ -12,12 +12,22 @@
 /* XT Boot Loader registered boot protocol list */
 LIST_ENTRY BlpBootProtocols;
 
+/* XT Boot Loader serial ports list */
+ULONG BlComPortList[COMPORT_COUNT] = COMPORT_ADDRESS;
+
 /* XT Boot Loader configuration list */
 LIST_ENTRY BlpConfig;
 
 /* XT Boot Loader loaded configuration */
 LIST_ENTRY BlpConfigSections;
 
+/* List of user-editable boot options */
+PWCHAR BlpEditableConfigOptions[] = {
+    L"BootModules", L"SystemType", L"SystemPath",
+    L"KernelFile",  L"InitrdFile", L"HalFile",
+    L"Parameters", NULL
+};
+
 /* XT Boot Loader protocol */
 XTBL_LOADER_PROTOCOL BlpLdrProtocol;
 
@@ -38,3 +48,4 @@ EFI_HANDLE EfiImageHandle;
 
 /* EFI System Table */
 PEFI_SYSTEM_TABLE EfiSystemTable;
+

xtldr/includes/globals.h

@@ -15,12 +15,18 @@
 /* XT Boot Loader registered boot protocol list */
 EXTERN LIST_ENTRY BlpBootProtocols;
 
+/* XT Boot Loader serial ports list */
+EXTERN ULONG BlComPortList[COMPORT_COUNT];
+
 /* XT Boot Loader configuration list */
 EXTERN LIST_ENTRY BlpConfig;
 
 /* XT Boot Loader loaded configuration */
 EXTERN LIST_ENTRY BlpConfigSections;
 
+/* List of user-editable boot options */
+EXTERN PWCHAR BlpEditableConfigOptions[];
+
 /* XT Boot Loader protocol */
 EXTERN XTBL_LOADER_PROTOCOL BlpLdrProtocol;
 

xtldr/includes/xtldr.h

@@ -18,7 +18,8 @@
 /* XTLDR routines forward references */
 XTCDECL
 EFI_STATUS
-BlAllocateMemoryPages(IN ULONGLONG Pages,
+BlAllocateMemoryPages(IN EFI_ALLOCATE_TYPE AllocationType,
+                      IN ULONGLONG NumberOfPages,
                       OUT PEFI_PHYSICAL_ADDRESS Memory);
 
 XTCDECL
@@ -50,16 +51,16 @@ BlCloseVolume(IN PEFI_HANDLE VolumeHandle);
 
 XTCDECL
 VOID
-BlConsolePrint(IN PUSHORT Format,
+BlConsolePrint(IN PCWSTR Format,
                IN ...);
 
 XTCDECL
 VOID
-BlConsoleWrite(IN PUSHORT String);
+BlConsoleWrite(IN PCWSTR String);
 
 XTCDECL
 VOID
-BlDebugPrint(IN PUSHORT Format,
+BlDebugPrint(IN PCWSTR Format,
              IN ...);
 
 XTCDECL
@@ -70,6 +71,10 @@ XTCDECL
 VOID
 BlDisplayBootMenu();
 
+XTCDECL
+VOID
+BlDisplayEditMenu(IN PXTBL_BOOTMENU_ITEM MenuEntry);
+
 XTCDECL
 VOID
 BlDisplayErrorDialog(IN PWCHAR Caption,
@@ -121,26 +126,43 @@ BlFindVolumeDevicePath(IN PEFI_DEVICE_PATH_PROTOCOL FsHandle,
 
 XTCDECL
 EFI_STATUS
-BlFreeMemoryPages(IN ULONGLONG Pages,
+BlFreeMemoryPages(IN ULONGLONG NumberOfPages,
                   IN EFI_PHYSICAL_ADDRESS Memory);
 
 XTCDECL
 EFI_STATUS
 BlFreeMemoryPool(IN PVOID Memory);
 
+XTCDECL
+BOOLEAN
+BlGetBooleanParameter(IN CONST PWCHAR Parameters,
+                      IN CONST PWCHAR Needle);
+
+XTCDECL
+EFI_STATUS
+BlGetBootOptionValue(IN PLIST_ENTRY Options,
+                     IN CONST PWCHAR OptionName,
+                     OUT PWCHAR *OptionValue);
+
 XTCDECL
 BOOLEAN
 BlGetConfigBooleanValue(IN CONST PWCHAR ConfigName);
 
 XTCDECL
-PWCHAR
-BlGetConfigValue(IN CONST PWCHAR ConfigName);
+EFI_STATUS
+BlGetConfigValue(IN CONST PWCHAR ConfigName,
+                 OUT PWCHAR *ConfigValue);
 
 XTCDECL
 EFI_STATUS
 BlGetConfigurationTable(IN PEFI_GUID TableGuid,
                         OUT PVOID *Table);
 
+XTCDECL
+VOID
+BlGetEditableOptions(OUT CONST PWCHAR **OptionsArray,
+                     OUT PULONG OptionsCount);
+
 XTCDECL
 EFI_STATUS
 BlGetEfiPath(IN PWCHAR SystemPath,
@@ -191,7 +213,8 @@ BlInitializeBootLoader();
 
 XTCDECL
 EFI_STATUS
-BlInitializeBootMenuList(OUT PXTBL_BOOTMENU_ITEM *MenuEntries,
+BlInitializeBootMenuList(IN ULONG MaxNameLength,
+                         OUT PXTBL_BOOTMENU_ITEM *MenuEntries,
                          OUT PULONG EntriesCount,
                          OUT PULONG DefaultId);
 
@@ -249,9 +272,9 @@ BlMapEfiMemory(IN OUT PXTBL_PAGE_MAPPING PageMap,
 XTCDECL
 EFI_STATUS
 BlMapPage(IN PXTBL_PAGE_MAPPING PageMap,
-          IN UINT_PTR VirtualAddress,
-          IN UINT_PTR PhysicalAddress,
-          IN UINT NumberOfPages);
+          IN ULONG_PTR VirtualAddress,
+          IN ULONG_PTR PhysicalAddress,
+          IN ULONG NumberOfPages);
 
 XTCDECL
 EFI_STATUS
@@ -325,6 +348,12 @@ XTCDECL
 VOID
 BlResetConsoleInputBuffer();
 
+XTCDECL
+EFI_STATUS
+BlSetBootOptionValue(IN PLIST_ENTRY Options,
+                     IN CONST PWCHAR OptionName,
+                     IN CONST PWCHAR OptionValue);
+
 XTCDECL
 EFI_STATUS
 BlSetConfigValue(IN CONST PWCHAR ConfigName,
@@ -389,11 +418,11 @@ BlpActivateSerialIOController();
 
 XTCDECL
 XTSTATUS
-BlpConsolePutChar(IN USHORT Character);
+BlpConsolePutChar(IN WCHAR Character);
 
 XTCDECL
 XTSTATUS
-BlpDebugPutChar(IN USHORT Character);
+BlpDebugPutChar(IN WCHAR Character);
 
 XTCDECL
 VOID
@@ -449,6 +478,18 @@ VOID
 BlpDrawDialogProgressBar(IN PXTBL_DIALOG_HANDLE Handle,
                          IN UCHAR Percentage);
 
+XTCDECL
+VOID
+BlpDrawEditMenu(OUT PXTBL_DIALOG_HANDLE Handle);
+
+XTCDECL
+EFI_STATUS
+BlpDrawEditMenuEntry(IN PXTBL_DIALOG_HANDLE Handle,
+                     IN PWCHAR OptionName,
+                     IN PWCHAR OptionValue,
+                     IN UINT Position,
+                     IN BOOLEAN Highlighted);
+
 XTCDECL
 PEFI_DEVICE_PATH_PROTOCOL
 BlpDuplicateDevicePath(IN PEFI_DEVICE_PATH_PROTOCOL DevicePath);
@@ -462,10 +503,10 @@ XTCDECL
 BOOLEAN
 BlpFindParentBlockDevice(IN PLIST_ENTRY BlockDevices,
                          IN PEFI_BLOCK_DEVICE_DATA ChildNode,
-                         OUT PEFI_BLOCK_DEVICE_DATA ParentNode);
+                         OUT PEFI_BLOCK_DEVICE_DATA *ParentNode);
 
 XTCDECL
-LOADER_MEMORY_TYPE
+LONG
 BlpGetLoaderMemoryType(IN EFI_MEMORY_TYPE EfiMemoryType);
 
 XTCDECL
@@ -484,9 +525,9 @@ BlpGetModuleInfoStrings(IN PWCHAR SectionData,
 XTCDECL
 EFI_STATUS
 BlpGetNextPageTable(IN PXTBL_PAGE_MAPPING PageMap,
-                    IN PHARDWARE_PTE PageTable,
+                    IN PVOID PageTable,
                     IN SIZE_T Entry,
-                    OUT PHARDWARE_PTE *NextPageTable);
+                    OUT PVOID *NextPageTable);
 
 XTCDECL
 EFI_STATUS

xtldr/memory.c

@@ -12,7 +12,7 @@
 /**
  * This routine allocates one or more 4KB pages.
  *
- * @param Pages
+ * @param NumberOfPages
  *        The number of contiguous 4KB pages to allocate.
  *
  * @param Memory
@@ -24,10 +24,11 @@
  */
 XTCDECL
 EFI_STATUS
-BlAllocateMemoryPages(IN ULONGLONG Pages,
+BlAllocateMemoryPages(IN EFI_ALLOCATE_TYPE AllocationType,
+                      IN ULONGLONG NumberOfPages,
                       OUT PEFI_PHYSICAL_ADDRESS Memory)
 {
-    return EfiSystemTable->BootServices->AllocatePages(AllocateAnyPages, EfiLoaderData, Pages, Memory);
+    return EfiSystemTable->BootServices->AllocatePages(AllocationType, EfiLoaderData, NumberOfPages, Memory);
 }
 
 /**
@@ -55,7 +56,7 @@ BlAllocateMemoryPool(IN UINT_PTR Size,
 /**
  * This routine frees memory pages.
  *
- * @param Pages
+ * @param NumberOfPages
  *        The number of contiguous 4 KB pages to free.
  *
  * @param Memory
@@ -67,10 +68,10 @@ BlAllocateMemoryPool(IN UINT_PTR Size,
  */
 XTCDECL
 EFI_STATUS
-BlFreeMemoryPages(IN ULONGLONG Pages,
+BlFreeMemoryPages(IN ULONGLONG NumberOfPages,
                   IN EFI_PHYSICAL_ADDRESS Memory)
 {
-    return EfiSystemTable->BootServices->FreePages(Memory, Pages);
+    return EfiSystemTable->BootServices->FreePages(Memory, NumberOfPages);
 }
 
 /**
@@ -292,7 +293,7 @@ BlMapEfiMemory(IN OUT PXTBL_PAGE_MAPPING PageMap,
     if(GetMemoryTypeRoutine == NULL)
     {
         /* Use default memory type routine */
-        GetMemoryTypeRoutine = (PBL_GET_MEMTYPE_ROUTINE)BlpGetLoaderMemoryType;
+        GetMemoryTypeRoutine = BlpGetLoaderMemoryType;
     }
 
     /* Allocate and zero-fill buffer for EFI memory map */
@@ -454,7 +455,7 @@ BlMapVirtualMemory(IN OUT PXTBL_PAGE_MAPPING PageMap,
     Mapping1->MemoryType = MemoryType;
 
     /* Calculate the end of the physical address */
-    PhysicalAddressEnd = (PUCHAR)PhysicalAddress + (NumberOfPages * EFI_PAGE_SIZE) - 1;
+    PhysicalAddressEnd = (PVOID)((ULONG_PTR)PhysicalAddress + (NumberOfPages * EFI_PAGE_SIZE) - 1);
 
     /* Iterate through all the mappings already set to insert new mapping at the correct place */
     ListEntry = PageMap->MemoryMap.Flink;
@@ -462,7 +463,7 @@ BlMapVirtualMemory(IN OUT PXTBL_PAGE_MAPPING PageMap,
     {
         /* Take a mapping from the list and calculate its end of physical address */
         Mapping2 = CONTAIN_RECORD(ListEntry, XTBL_MEMORY_MAPPING, ListEntry);
-        PhysicalAddress2End = (PUCHAR)Mapping2->PhysicalAddress + (Mapping2->NumberOfPages * EFI_PAGE_SIZE) - 1 ;
+        PhysicalAddress2End = (PVOID)((ULONG_PTR)Mapping2->PhysicalAddress + (Mapping2->NumberOfPages * EFI_PAGE_SIZE) - 1);
 
         /* Check if new mapping is a subset of an existing mapping */
         if(Mapping1->PhysicalAddress >= Mapping2->PhysicalAddress && PhysicalAddressEnd <= PhysicalAddress2End)
@@ -508,7 +509,7 @@ BlMapVirtualMemory(IN OUT PXTBL_PAGE_MAPPING PageMap,
             /* Calculate number of pages and the end of the physical address */
             Mapping2->NumberOfPages = ((PUCHAR)PhysicalAddressEnd + 1 -
                                        (PUCHAR)Mapping2->PhysicalAddress) / EFI_PAGE_SIZE;
-            PhysicalAddress2End = (PUCHAR)Mapping2->PhysicalAddress + (Mapping2->NumberOfPages * EFI_PAGE_SIZE) - 1;
+            PhysicalAddress2End = (PVOID)((ULONG_PTR)Mapping2->PhysicalAddress + (Mapping2->NumberOfPages * EFI_PAGE_SIZE) - 1);
         }
 
         /* Check if they overlap */
@@ -544,7 +545,7 @@ BlMapVirtualMemory(IN OUT PXTBL_PAGE_MAPPING PageMap,
             /* Calculate number of pages and the end of the physical address */
             Mapping2->NumberOfPages = ((PUCHAR)Mapping1->PhysicalAddress -
                                        (PUCHAR)Mapping2->PhysicalAddress) / EFI_PAGE_SIZE;
-            PhysicalAddress2End = (PUCHAR)Mapping2->PhysicalAddress + (Mapping2->NumberOfPages * EFI_PAGE_SIZE) - 1;
+            PhysicalAddress2End = (PVOID)((ULONG_PTR)Mapping2->PhysicalAddress + (Mapping2->NumberOfPages * EFI_PAGE_SIZE) - 1);
         }
 
         /* Check if mapping is really needed */
@@ -659,16 +660,6 @@ BlPhysicalListToVirtual(IN PXTBL_PAGE_MAPPING PageMap,
         NextEntry = ListEntry->Flink;
 
         /* Convert the address of this element to VirtualAddress */
-        if(ListEntry->Flink == ListHead)
-        {
-            /* Convert list head */
-            ListEntry->Flink = ListHead->Flink->Blink;
-        }
-        else
-        {
-            /* Convert list entry */
-            ListEntry->Flink = BlPhysicalAddressToVirtual(ListEntry->Flink, (PVOID)PhysicalBase, VirtualBase);
-        }
         if(ListEntry->Blink == ListHead)
         {
             /* Find virtual address of list head */
@@ -679,14 +670,24 @@ BlPhysicalListToVirtual(IN PXTBL_PAGE_MAPPING PageMap,
             /* Convert list entry */
             ListEntry->Blink = BlPhysicalAddressToVirtual(ListEntry->Blink, (PVOID)PhysicalBase, VirtualBase);
         }
+        if(ListEntry->Flink == ListHead)
+        {
+            /* Convert list head */
+            ListEntry->Flink = ListHead->Flink->Blink;
+        }
+        else
+        {
+            /* Convert list entry */
+            ListEntry->Flink = BlPhysicalAddressToVirtual(ListEntry->Flink, (PVOID)PhysicalBase, VirtualBase);
+        }
 
         /* Get to the next element*/
         ListEntry = NextEntry;
     }
 
     /* Convert list head */
-    ListHead->Flink = BlPhysicalAddressToVirtual(ListHead->Flink, (PVOID)PhysicalBase, VirtualBase);
     ListHead->Blink = BlPhysicalAddressToVirtual(ListHead->Blink, (PVOID)PhysicalBase, VirtualBase);
+    ListHead->Flink = BlPhysicalAddressToVirtual(ListHead->Flink, (PVOID)PhysicalBase, VirtualBase);
 
     /* Return success */
     return STATUS_EFI_SUCCESS;
@@ -703,7 +704,7 @@ BlPhysicalListToVirtual(IN PXTBL_PAGE_MAPPING PageMap,
  * @since XT 1.0
  */
 XTCDECL
-LOADER_MEMORY_TYPE
+LONG
 BlpGetLoaderMemoryType(IN EFI_MEMORY_TYPE EfiMemoryType)
 {
     LOADER_MEMORY_TYPE MemoryType;
@@ -739,75 +740,3 @@ BlpGetLoaderMemoryType(IN EFI_MEMORY_TYPE EfiMemoryType)
     /* Return XTLDR memory type */
     return MemoryType;
 }
-
-/**
- * Returns next level of the Page Table.
- *
- * @param PageMap
- *        Supplies a pointer to the page mapping structure.
- *
- * @param PageTable
- *        Supplies a pointer to the current Page Table.
- *
- * @param Entry
- *        Supplies an index of the current Page Table entry.
- *
- * @param NextPageTable
- *        Supplies a pointer to the memory area where the next Page Table level is returned.
- *
- * @return This routine returns a status code.
- *
- * @since XT 1.0
- */
-XTCDECL
-EFI_STATUS
-BlpGetNextPageTable(IN PXTBL_PAGE_MAPPING PageMap,
-                    IN PHARDWARE_PTE PageTable,
-                    IN SIZE_T Entry,
-                    OUT PHARDWARE_PTE *NextPageTable)
-{
-    EFI_PHYSICAL_ADDRESS Address;
-    ULONGLONG PmlPointer;
-    EFI_STATUS Status;
-
-    /* Check if this is a valid table */
-    if(PageTable[Entry].Valid)
-    {
-        /* Get PML pointer */
-        PmlPointer = PageTable[Entry].PageFrameNumber;
-        PmlPointer <<= EFI_PAGE_SHIFT;
-    }
-    else
-    {
-        /* Allocate pages for new PML entry */
-        Status = BlAllocateMemoryPages(1, &Address);
-        if(Status != STATUS_EFI_SUCCESS)
-        {
-            /* Memory allocation failure */
-            return Status;
-        }
-
-        /* Add new memory mapping */
-        Status = BlMapVirtualMemory(PageMap, NULL, (PVOID)(UINT_PTR)Address, 1, LoaderMemoryData);
-        if(Status != STATUS_EFI_SUCCESS)
-        {
-            /* Memory mapping failure */
-            return Status;
-        }
-
-        /* Fill allocated memory with zeros */
-        RtlZeroMemory((PVOID)(ULONGLONG)Address, EFI_PAGE_SIZE);
-
-        /* Set paging entry settings */
-        PageTable[Entry].PageFrameNumber = Address / EFI_PAGE_SIZE;
-        PageTable[Entry].Valid = 1;
-        PageTable[Entry].Writable = 1;
-        PmlPointer = (ULONGLONG)Address;
-    }
-
-    /* Set next Page Map Level (PML) */
-    *NextPageTable = (PHARDWARE_PTE)(ULONGLONG)PmlPointer;
-
-    /* Return success */
-    return STATUS_EFI_SUCCESS;
-}

xtldr/modules/acpi/acpi.c

@@ -51,6 +51,126 @@ AcGetAcpiDescriptionPointer(OUT PVOID *AcpiTable)
     return STATUS_EFI_NOT_FOUND;
 }
 
+/**
+ * Finds ACPI description table with given signature.
+ *
+ * @param Signature
+ *        Supplies the signature of the desired ACPI table.
+ *
+ * @param PreviousTable
+ *        Supplies a pointer to the table to start searching from.
+ *
+ * @param AcpiTable
+ *        Supplies a pointer to memory area where ACPI table address will be stored, or NULL if not found.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+EFI_STATUS
+AcGetAcpiTable(IN CONST UINT Signature,
+               IN PVOID PreviousTable,
+               OUT PVOID *AcpiTable)
+{
+    PACPI_DESCRIPTION_HEADER TableHeader;
+    SIZE_T RsdtIndex, TableIndex;
+    EFI_STATUS Status;
+    SIZE_T TableCount;
+    PACPI_RSDP Rsdp;
+    PACPI_RSDT Rsdt;
+    BOOLEAN Xsdp;
+
+    /* Return NULL address by default if requested table not found */
+    *AcpiTable = NULL;
+
+    /* Get Root System Description Table Pointer */
+    Status = AcGetAcpiDescriptionPointer((PVOID)&Rsdp);
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        /* ACPI tables not found, return error */
+        return Status;
+    }
+
+    /* Check if it is XSDP (ACPI 2.0) or RSDP (ACPI 1.0) */
+    if(Rsdp->Revision >= 2 && Rsdp->XsdtAddress)
+    {
+        /* XSDP (ACPI 2.0) */
+        Xsdp = TRUE;
+        Rsdt = (PACPI_RSDT)(UINT_PTR)Rsdp->XsdtAddress;
+        TableCount = (Rsdt->Header.Length - sizeof(ACPI_DESCRIPTION_HEADER)) / 8;
+    }
+    else
+    {
+        /* RSDP (ACPI 1.0) */
+        Xsdp = FALSE;
+        Rsdt = (PACPI_RSDT)(UINT_PTR)Rsdp->RsdtAddress;
+        TableCount = (Rsdt->Header.Length - sizeof(ACPI_DESCRIPTION_HEADER)) / 4;
+    }
+
+    /* Iterate over all ACPI tables */
+    for(TableIndex = 0; TableIndex < TableCount; TableIndex++)
+    {
+        /* Get table headers in reverse order */
+        RsdtIndex = TableCount - TableIndex - 1;
+
+        /* Check if XSDP or RSDT is used */
+        if(Xsdp)
+        {
+            /* Get table header from XSDT */
+            TableHeader = (PACPI_DESCRIPTION_HEADER)(ULONG_PTR)((PULONGLONG)Rsdt->Tables)[RsdtIndex];
+        }
+        else
+        {
+            /* Get table header from RSDT */
+            TableHeader = (PACPI_DESCRIPTION_HEADER)(ULONG_PTR)((PULONG)Rsdt->Tables)[RsdtIndex];
+        }
+
+        /* Check if previous table provided */
+        if(PreviousTable != NULL)
+        {
+            /* Check if this is a table previously found */
+            if(TableHeader == (PVOID)PreviousTable)
+            {
+                /* Unset previous table */
+                PreviousTable = NULL;
+            }
+
+            /* Skip to next ACPI table */
+            continue;
+        }
+
+        /* Verify table signature */
+        if((TableHeader->Signature == Signature))
+        {
+            /* Found requested ACPI table */
+            break;
+        }
+    }
+
+    /* Make sure table was found */
+    if(TableHeader->Signature != Signature)
+    {
+        /* ACPI table not found, return error */
+        return STATUS_EFI_NOT_FOUND;
+    }
+
+    /* Don't validate FADT on old, broken firmwares with ACPI 2.0 or older */
+    if(TableHeader->Signature != ACPI_FADT_SIGNATURE || TableHeader->Revision > 2)
+    {
+        /* Validate table checksum */
+        if(!AcpValidateAcpiTable(TableHeader, TableHeader->Length))
+        {
+            /* Checksum mismatch, return error */
+            return STATUS_EFI_CRC_ERROR;
+        }
+    }
+
+    /* Found valid ACPI table, return success */
+    *AcpiTable = TableHeader;
+    return STATUS_EFI_SUCCESS;
+}
+
 /**
  * Gets the Advanced Programmable Interrupt Controller (APIC) base address.
  *
@@ -65,19 +185,17 @@ XTCDECL
 EFI_STATUS
 AcGetApicBase(OUT PVOID *ApicBase)
 {
-    PCPUID_REGISTERS CpuRegisters = NULL;
+    CPUID_REGISTERS CpuRegisters;
 
-    /* Get CPU features list */
-    CpuRegisters->Leaf = CPUID_GET_CPU_FEATURES;
-    CpuRegisters->SubLeaf = 0;
-    CpuRegisters->Eax = 0;
-    CpuRegisters->Ebx = 0;
-    CpuRegisters->Ecx = 0;
-    CpuRegisters->Edx = 0;
-    ArCpuId(CpuRegisters);
+    /* Prepare CPUID registers to query for APIC support */
+    RtlZeroMemory(&CpuRegisters, sizeof(CPUID_REGISTERS));
+    CpuRegisters.Leaf = CPUID_GET_STANDARD1_FEATURES;
+
+    /* Query CPUID */
+    ArCpuId(&CpuRegisters);
 
     /* Check if APIC present */
-    if((CpuRegisters->Edx & CPUID_FEATURES_EDX_APIC) == 0)
+    if((CpuRegisters.Edx & CPUID_FEATURES_EDX_APIC) == 0)
     {
         /* APIC is not supported by the CPU */
         return STATUS_EFI_UNSUPPORTED;
@@ -110,7 +228,7 @@ AcGetRsdpTable(OUT PVOID *AcpiTable)
 
     /* Get RSDP (ACPI 1.0) table from system configuration tables */
     Status = XtLdrProtocol->Util.GetConfigurationTable(&AcpiGuid, &RsdpTable);
-    if(Status != STATUS_EFI_SUCCESS || AcpChecksumTable(RsdpTable, 20) != 0)
+    if(Status != STATUS_EFI_SUCCESS || !AcpValidateAcpiTable(RsdpTable, 20))
     {
         /* RSDP not found or checksum mismatch */
         *AcpiTable = NULL;
@@ -142,7 +260,7 @@ AcGetSMBiosTable(OUT PVOID *SmBiosTable)
 
     /* Get SMBIOS table from system configuration tables */
     Status = XtLdrProtocol->Util.GetConfigurationTable(&SmBiosGuid, (PVOID)&SmBios);
-    if(Status != STATUS_EFI_SUCCESS || AcpChecksumTable(SmBios, SmBios->Length) != 0)
+    if(Status != STATUS_EFI_SUCCESS || !AcpValidateAcpiTable(SmBios, SmBios->Length))
     {
         /* SMBIOS not found or checksum mismatch */
         *SmBiosTable = NULL;
@@ -174,7 +292,7 @@ AcGetSMBios3Table(OUT PVOID *SmBiosTable)
 
     /* Get SMBIOS3 table from system configuration tables */
     Status = XtLdrProtocol->Util.GetConfigurationTable(&SmBios3Guid, (PVOID)&SmBios);
-    if(Status != STATUS_EFI_SUCCESS || AcpChecksumTable(SmBios, SmBios->Length) != 0)
+    if(Status != STATUS_EFI_SUCCESS || !AcpValidateAcpiTable(SmBios, SmBios->Length))
     {
         /* SMBIOS3 not found or checksum mismatch */
         *SmBiosTable = NULL;
@@ -206,7 +324,7 @@ AcGetXsdpTable(OUT PVOID *AcpiTable)
 
     /* Get XSDP (ACPI 2.0) from system configuration tables */
     Status = XtLdrProtocol->Util.GetConfigurationTable(&AcpiGuid, &XsdpTable);
-    if(Status != STATUS_EFI_SUCCESS || AcpChecksumTable(XsdpTable, 36) != 0)
+    if(Status != STATUS_EFI_SUCCESS || !AcpValidateAcpiTable(XsdpTable, 36))
     {
         /* XSDP not found or checksum mismatch */
         *AcpiTable = NULL;
@@ -219,7 +337,7 @@ AcGetXsdpTable(OUT PVOID *AcpiTable)
 }
 
 /**
- * Checksums a given ACPI table.
+ * Validates given ACPI table by calculating its checksum.
  *
  * @param Buffer
  *        Supplies a pointer to the table to checksum.
@@ -227,12 +345,14 @@ AcGetXsdpTable(OUT PVOID *AcpiTable)
  * @param Size
  *        Supplies the size of the table, in bytes.
  *
- * @return This routine returns the calculated checksum.
+ * @return This routine returns TRUE if the table is valid, or FALSE otherwise.
+ *
+ * @since XT 1.0
  */
 XTCDECL
-UCHAR
-AcpChecksumTable(IN PVOID Buffer,
-                 IN UINT_PTR Size)
+BOOLEAN
+AcpValidateAcpiTable(IN PVOID Buffer,
+                     IN UINT_PTR Size)
 {
     PUCHAR Pointer;
     UCHAR Sum;
@@ -250,7 +370,7 @@ AcpChecksumTable(IN PVOID Buffer,
     }
 
     /* Return calculated checksum */
-    return Sum;
+    return (Sum == 0) ? TRUE : FALSE;
 }
 
 /**
@@ -284,6 +404,7 @@ XtLdrModuleMain(IN EFI_HANDLE ImageHandle,
 
     /* Set routines available via ACPI protocol */
     AcpAcpiProtocol.GetAcpiDescriptionPointer = AcGetAcpiDescriptionPointer;
+    AcpAcpiProtocol.GetAcpiTable = AcGetAcpiTable;
     AcpAcpiProtocol.GetApicBase = AcGetApicBase;
     AcpAcpiProtocol.GetRsdpTable = AcGetRsdpTable;
     AcpAcpiProtocol.GetSMBiosTable = AcGetSMBiosTable;

xtldr/modules/acpi/includes/acpi.h

@@ -18,6 +18,12 @@ XTCDECL
 EFI_STATUS
 AcGetAcpiDescriptionPointer(OUT PVOID *AcpiTable);
 
+XTCDECL
+EFI_STATUS
+AcGetAcpiTable(IN CONST UINT Signature,
+               IN PVOID PreviousTable,
+               OUT PVOID *AcpiTable);
+
 XTCDECL
 EFI_STATUS
 AcGetApicBase(OUT PVOID *ApicBase);
@@ -39,9 +45,9 @@ EFI_STATUS
 AcGetXsdpTable(OUT PVOID *AcpiTable);
 
 XTCDECL
-UCHAR
-AcpChecksumTable(IN PVOID Buffer,
-                 IN UINT_PTR Size);
+BOOLEAN
+AcpValidateAcpiTable(IN PVOID Buffer,
+                     IN UINT_PTR Size);
 
 XTCDECL
 EFI_STATUS

xtldr/modules/beep/beep.c

@@ -196,6 +196,7 @@ XtLdrModuleMain(IN EFI_HANDLE ImageHandle,
                 IN PEFI_SYSTEM_TABLE SystemTable)
 {
     EFI_STATUS Status;
+    PWCHAR Tune;
 
     /* Open the XTLDR protocol */
     Status = BlGetXtLdrProtocol(SystemTable, ImageHandle, &XtLdrProtocol);
@@ -206,7 +207,8 @@ XtLdrModuleMain(IN EFI_HANDLE ImageHandle,
     }
 
     /* Play the tune set in the configuration */
-    BpPlayTune(XtLdrProtocol->Config.GetValue(L"TUNE"));
+    XtLdrProtocol->Config.GetValue(L"TUNE", &Tune);
+    BpPlayTune(Tune);
 
     /* Return success */
     return STATUS_EFI_SUCCESS;

xtldr/modules/chainldr/chainldr.c

@@ -47,7 +47,7 @@ ChBootSystem(IN PXTBL_BOOT_PARAMETERS Parameters)
     }
 
     /* Open EFI volume */
-    Status = XtLdrProtocol->Disk.OpenVolume(NULL, &DiskHandle, &FsHandle);
+    Status = XtLdrProtocol->Disk.OpenVolume(Parameters->DevicePath, &DiskHandle, &FsHandle);
     if(Status != STATUS_EFI_SUCCESS)
     {
         /* Failed to open a volume, return error code */

xtldr/modules/framebuf/framebuf.c

@@ -524,8 +524,8 @@ FbpFindFramebufferAddress(OUT PEFI_PHYSICAL_ADDRESS Address)
  * @param PixelBitMask
  *        Provides a pixel bit mask.
  *
- * @param ColorMask
- *        Supplies a pointer to the memory area where the color mask will be stored.
+ * @param ColorSize
+ *        Supplies a pointer to the memory area where the color size will be stored.
  *
  * @param ColorShift
  *        Supplies a pointer to the memory area where the color shift (position) will be stored.
@@ -537,34 +537,37 @@ FbpFindFramebufferAddress(OUT PEFI_PHYSICAL_ADDRESS Address)
 XTCDECL
 VOID
 FbpGetColorMask(IN UINT PixelBitMask,
-                OUT PUSHORT ColorMask,
+                OUT PUSHORT ColorSize,
                 OUT PUSHORT ColorShift)
 {
-    UINT Index, Mask;
+    UINT Shift, Size;
 
     /* Initialize variables */
-    Index = 0;
-    Mask = 1;
+    Shift = 0;
+    Size = 0;
 
     /* Make sure EfiMask is not zero */
     if(PixelBitMask)
     {
-        while((Index < 32) && ((PixelBitMask & Mask) == 0))
+        /* Get color shift */
+        while((PixelBitMask & 1) == 0)
         {
-            Index++;
-            Mask <<= 1;
+            Shift++;
+            PixelBitMask >>= 1;
+        }
+
+        /* Get color size */
+        while((PixelBitMask & 1) == 1)
+        {
+            Size++;
+            PixelBitMask >>= 1;
         }
 
-        /* Set color mask and shift */
-        *ColorShift = Index;
-        *ColorMask = (Mask >> Index);
-    }
-    else
-    {
-        /* Set default color mask and shift */
-        *ColorMask = 0;
-        *ColorShift = 0;
     }
+
+    /* Set color mask and shift */
+    *ColorShift = Shift;
+    *ColorSize = Size;
 }
 
 /**
@@ -672,26 +675,26 @@ FbpGetPixelInformation(IN PEFI_PIXEL_BITMASK PixelsBitMask)
         case PixelBlueGreenRedReserved8BitPerColor:
             /* BGRR, 32 bits per pixel */
             FbpDisplayInfo.ModeInfo.BitsPerPixel = 32;
-            FbpDisplayInfo.ModeInfo.PixelInformation.BlueMask = 0xFF;
             FbpDisplayInfo.ModeInfo.PixelInformation.BlueShift = 0;
-            FbpDisplayInfo.ModeInfo.PixelInformation.GreenMask = 0xFF;
+            FbpDisplayInfo.ModeInfo.PixelInformation.BlueSize = 8;
             FbpDisplayInfo.ModeInfo.PixelInformation.GreenShift = 8;
-            FbpDisplayInfo.ModeInfo.PixelInformation.RedMask = 0xFF;
+            FbpDisplayInfo.ModeInfo.PixelInformation.GreenSize = 8;
             FbpDisplayInfo.ModeInfo.PixelInformation.RedShift = 16;
-            FbpDisplayInfo.ModeInfo.PixelInformation.ReservedMask = 0xFF;
+            FbpDisplayInfo.ModeInfo.PixelInformation.RedSize = 8;
             FbpDisplayInfo.ModeInfo.PixelInformation.ReservedShift = 24;
+            FbpDisplayInfo.ModeInfo.PixelInformation.ReservedSize = 8;
             break;
         case PixelRedGreenBlueReserved8BitPerColor:
             /* RGBR, 32 bits per pixel */
             FbpDisplayInfo.ModeInfo.BitsPerPixel = 32;
-            FbpDisplayInfo.ModeInfo.PixelInformation.BlueMask = 0xFF;
             FbpDisplayInfo.ModeInfo.PixelInformation.BlueShift = 16;
-            FbpDisplayInfo.ModeInfo.PixelInformation.GreenMask = 0xFF;
+            FbpDisplayInfo.ModeInfo.PixelInformation.BlueSize = 8;
             FbpDisplayInfo.ModeInfo.PixelInformation.GreenShift = 8;
-            FbpDisplayInfo.ModeInfo.PixelInformation.RedMask = 0xFF;
+            FbpDisplayInfo.ModeInfo.PixelInformation.GreenSize = 8;
             FbpDisplayInfo.ModeInfo.PixelInformation.RedShift = 0;
-            FbpDisplayInfo.ModeInfo.PixelInformation.ReservedMask = 0xFF;
+            FbpDisplayInfo.ModeInfo.PixelInformation.RedSize = 8;
             FbpDisplayInfo.ModeInfo.PixelInformation.ReservedShift = 24;
+            FbpDisplayInfo.ModeInfo.PixelInformation.ReservedSize = 8;
             break;
         case PixelBitMask:
             /* Assume 32 bits per pixel */
@@ -711,26 +714,26 @@ FbpGetPixelInformation(IN PEFI_PIXEL_BITMASK PixelsBitMask)
             }
 
             /* Set pixel information */
-            FbpGetColorMask(PixelsBitMask->RedMask, &FbpDisplayInfo.ModeInfo.PixelInformation.RedMask,
+            FbpGetColorMask(PixelsBitMask->RedMask, &FbpDisplayInfo.ModeInfo.PixelInformation.RedSize,
                             &FbpDisplayInfo.ModeInfo.PixelInformation.RedShift);
-            FbpGetColorMask(PixelsBitMask->GreenMask, &FbpDisplayInfo.ModeInfo.PixelInformation.GreenMask,
+            FbpGetColorMask(PixelsBitMask->GreenMask, &FbpDisplayInfo.ModeInfo.PixelInformation.GreenSize,
                             &FbpDisplayInfo.ModeInfo.PixelInformation.GreenShift);
-            FbpGetColorMask(PixelsBitMask->BlueMask, &FbpDisplayInfo.ModeInfo.PixelInformation.BlueMask,
+            FbpGetColorMask(PixelsBitMask->BlueMask, &FbpDisplayInfo.ModeInfo.PixelInformation.BlueSize,
                             &FbpDisplayInfo.ModeInfo.PixelInformation.BlueShift);
-            FbpGetColorMask(PixelsBitMask->ReservedMask, &FbpDisplayInfo.ModeInfo.PixelInformation.ReservedMask,
+            FbpGetColorMask(PixelsBitMask->ReservedMask, &FbpDisplayInfo.ModeInfo.PixelInformation.ReservedSize,
                             &FbpDisplayInfo.ModeInfo.PixelInformation.ReservedShift);
             break;
         default:
             /* Unknown pixel format */
             FbpDisplayInfo.ModeInfo.BitsPerPixel = 0;
-            FbpDisplayInfo.ModeInfo.PixelInformation.BlueMask = 0x0;
             FbpDisplayInfo.ModeInfo.PixelInformation.BlueShift = 0;
-            FbpDisplayInfo.ModeInfo.PixelInformation.GreenMask = 0x0;
+            FbpDisplayInfo.ModeInfo.PixelInformation.BlueSize = 0;
             FbpDisplayInfo.ModeInfo.PixelInformation.GreenShift = 0;
-            FbpDisplayInfo.ModeInfo.PixelInformation.RedMask = 0x0;
+            FbpDisplayInfo.ModeInfo.PixelInformation.GreenSize = 0;
             FbpDisplayInfo.ModeInfo.PixelInformation.RedShift = 0;
-            FbpDisplayInfo.ModeInfo.PixelInformation.ReservedMask = 0x0;
+            FbpDisplayInfo.ModeInfo.PixelInformation.RedSize = 0;
             FbpDisplayInfo.ModeInfo.PixelInformation.ReservedShift = 0;
+            FbpDisplayInfo.ModeInfo.PixelInformation.ReservedSize = 0;
             break;
     }
 

xtldr/modules/framebuf/includes/framebuf.h

@@ -45,7 +45,7 @@ FbpFindFramebufferAddress(OUT PEFI_PHYSICAL_ADDRESS Address);
 XTCDECL
 VOID
 FbpGetColorMask(IN UINT EfiMask,
-                OUT PUSHORT ColorMask,
+                OUT PUSHORT ColorSize,
                 OUT PUSHORT ColorShift);
 
 XTCDECL

xtldr/modules/pecoff/pecoff.c

@@ -409,7 +409,7 @@ PeLoadImage(IN PEFI_FILE_HANDLE FileHandle,
     Pages = EFI_SIZE_TO_PAGES(ImageData->FileSize);
 
     /* Allocate pages */
-    Status = XtLdrProtocol->Memory.AllocatePages(Pages, &Address);
+    Status = XtLdrProtocol->Memory.AllocatePages(AllocateAnyPages, Pages, &Address);
     if(Status != STATUS_EFI_SUCCESS)
     {
         /* Pages allocation failure */
@@ -472,7 +472,7 @@ PeLoadImage(IN PEFI_FILE_HANDLE FileHandle,
     ImageData->ImagePages = EFI_SIZE_TO_PAGES(ImageData->ImageSize);
 
     /* Allocate image pages */
-    Status = XtLdrProtocol->Memory.AllocatePages(ImageData->ImagePages, &Address);
+    Status = XtLdrProtocol->Memory.AllocatePages(AllocateAnyPages, ImageData->ImagePages, &Address);
     if(Status != STATUS_EFI_SUCCESS)
     {
         /* Pages reallocation failure */

xtldr/modules/xtos_o/amd64/memory.c

@@ -4,25 +4,204 @@
  * FILE:            xtldr/amd64/memory.c
  * DESCRIPTION:     EFI memory management for AMD64 target
  * DEVELOPERS:      Rafal Kupiec <belliash@codingworkshop.eu.org>
+ *                  Aiken Harris <harraiken91@gmail.com>
  */
 
 #include <xtos.h>
 
 
+/**
+ * Determines the appropriate paging level (PML) for the AMD64 architecture.
+ *
+ * @param Parameters
+ *        A pointer to the wide character string containing the kernel boot parameters.
+ *
+ * @return This routine returns the appropriate page map level (5 if LA57 is enabled, 4 otherwise).
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+ULONG
+XtpDeterminePagingLevel(IN CONST PWCHAR Parameters)
+{
+    CPUID_REGISTERS CpuRegisters;
+
+    /* Prepare CPUID registers to query for STD7 features */
+    RtlZeroMemory(&CpuRegisters, sizeof(CPUID_REGISTERS));
+    CpuRegisters.Leaf = CPUID_GET_VENDOR_STRING;
+
+    /* Query CPUID */
+    ArCpuId(&CpuRegisters);
+
+    /* Verify if the CPU supports the STD7 feature leaf (0x00000007) */
+    if(CpuRegisters.Eax >= CPUID_GET_STANDARD7_FEATURES)
+    {
+        /* Prepare CPUID registers to query for LA57 support */
+        RtlZeroMemory(&CpuRegisters, sizeof(CPUID_REGISTERS));
+        CpuRegisters.Leaf = CPUID_GET_STANDARD7_FEATURES;
+
+        /* Query CPUID */
+        ArCpuId(&CpuRegisters);
+
+        /* Check if eXtended Physical Addressing (XPA) is enabled and if LA57 is supported by the CPU */
+        if((CpuRegisters.Ecx & CPUID_FEATURES_ECX_LA57) &&
+           !(XtLdrProtocol->BootUtil.GetBooleanParameter(Parameters, L"NOXPA")))
+        {
+            /* Enable LA57 (PML5) */
+            return 5;
+        }
+    }
+
+    /* Disable LA57 and use PML4 by default */
+    return 4;
+}
+
+/**
+ * Maps the page table for hardware layer addess space.
+ *
+ * @param PageMap
+ *        Supplies a pointer to the page mapping structure.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+EFI_STATUS
+XtpMapHardwareMemoryPool(IN PXTBL_PAGE_MAPPING PageMap)
+{
+    PHARDWARE_PTE P5eBase, PdeBase, PpeBase, PxeBase;
+    EFI_PHYSICAL_ADDRESS Address;
+    EFI_STATUS Status;
+
+    if(PageMap->PageMapLevel == 5)
+    {
+        /* Get P5E (PML5) base address */
+        P5eBase = (PHARDWARE_PTE)PageMap->PtePointer;
+
+        /* Check if P5E entry already exists */
+        if(!P5eBase[(MM_HARDWARE_VA_START >> MM_P5I_SHIFT) & 0x1FF].Valid)
+        {
+            /* No valid P5E, allocate memory */
+            Status = XtLdrProtocol->Memory.AllocatePages(AllocateAnyPages, 1, &Address);
+            if(Status != STATUS_EFI_SUCCESS)
+            {
+                /* Memory allocation failure, return error */
+                return Status;
+            }
+
+            /* Zero fill memory used by P5E */
+            RtlZeroMemory((PVOID)Address, EFI_PAGE_SIZE);
+
+            /* Make P5E valid */
+            P5eBase[(MM_HARDWARE_VA_START >> MM_P5I_SHIFT) & 0x1FF].Valid = 1;
+            P5eBase[(MM_HARDWARE_VA_START >> MM_P5I_SHIFT) & 0x1FF].PageFrameNumber = Address / EFI_PAGE_SIZE;
+            P5eBase[(MM_HARDWARE_VA_START >> MM_P5I_SHIFT) & 0x1FF].Writable = 1;
+
+            /* Set PXE base address */
+            PxeBase = (PHARDWARE_PTE)(UINT_PTR)Address;
+        }
+        else
+        {
+            /* Set PXE base address based on existing P5E */
+            PxeBase = (PHARDWARE_PTE)((P5eBase[(MM_HARDWARE_VA_START >> MM_P5I_SHIFT) & 0x1FF].PageFrameNumber) << EFI_PAGE_SHIFT);
+        }
+    }
+    else
+    {
+        /* Get PXE (PML4) base address */
+        PxeBase = (PHARDWARE_PTE)PageMap->PtePointer;
+    }
+
+    /* Check if PXE entry already exists */
+    if(!PxeBase[(MM_HARDWARE_VA_START >> MM_PXI_SHIFT) & 0x1FF].Valid)
+    {
+        /* No valid PXE, allocate memory */
+        Status = XtLdrProtocol->Memory.AllocatePages(AllocateAnyPages, 1, &Address);
+        if(Status != STATUS_EFI_SUCCESS)
+        {
+            /* Memory allocation failure, return error */
+            return Status;
+        }
+
+        /* Zero fill memory used by PXE */
+        RtlZeroMemory((PVOID)Address, EFI_PAGE_SIZE);
+
+        /* Make PXE valid */
+        PxeBase[(MM_HARDWARE_VA_START >> MM_PXI_SHIFT) & 0x1FF].Valid = 1;
+        PxeBase[(MM_HARDWARE_VA_START >> MM_PXI_SHIFT) & 0x1FF].PageFrameNumber = Address / EFI_PAGE_SIZE;
+        PxeBase[(MM_HARDWARE_VA_START >> MM_PXI_SHIFT) & 0x1FF].Writable = 1;
+
+        /* Set PPE base address */
+        PpeBase = (PHARDWARE_PTE)(UINT_PTR)Address;
+    }
+    else
+    {
+        /* Set PPE base address based on existing PXE */
+        PpeBase = (PHARDWARE_PTE)((PxeBase[(MM_HARDWARE_VA_START >> MM_PXI_SHIFT) & 0x1FF].PageFrameNumber) << EFI_PAGE_SHIFT);
+    }
+
+    /* Check if PPE entry already exists */
+    if(!PpeBase[(MM_HARDWARE_VA_START >> MM_PPI_SHIFT) & 0x1FF].Valid)
+    {
+        /* No valid PPE, allocate memory */
+        Status = XtLdrProtocol->Memory.AllocatePages(AllocateAnyPages, 1, &Address);
+        if(Status != STATUS_EFI_SUCCESS)
+        {
+            /* Memory allocation failure, return error */
+            return Status;
+        }
+
+        /* Zero fill memory used by PPE */
+        RtlZeroMemory((PVOID)Address, EFI_PAGE_SIZE);
+
+        /* Make PPE valid */
+        PpeBase[(MM_HARDWARE_VA_START >> MM_PPI_SHIFT) & 0x1FF].Valid = 1;
+        PpeBase[(MM_HARDWARE_VA_START >> MM_PPI_SHIFT) & 0x1FF].PageFrameNumber = Address / EFI_PAGE_SIZE;
+        PpeBase[(MM_HARDWARE_VA_START >> MM_PPI_SHIFT) & 0x1FF].Writable = 1;
+
+        /* Set PDE base address */
+        PdeBase = (PHARDWARE_PTE)Address;
+    }
+    else
+    {
+        /* Set PDE base address, based on existing PPE */
+        PdeBase = (PHARDWARE_PTE)((PpeBase[(MM_HARDWARE_VA_START >> MM_PPI_SHIFT) & 0x1FF].PageFrameNumber) << EFI_PAGE_SHIFT);
+    }
+
+    /* Loop through 2 PDE entries */
+    for(UINT Index = 0 ; Index < 2 ; Index++)
+    {
+        /* Check if PDE entry already exists */
+        if(!PdeBase[((MM_HARDWARE_VA_START >> MM_PDI_SHIFT) & 0x1FF) + Index].Valid)
+        {
+            /* No valid PDE, allocate memory */
+            Status = XtLdrProtocol->Memory.AllocatePages(AllocateAnyPages, 1, &Address);
+            if(Status != STATUS_EFI_SUCCESS)
+            {
+                /* Memory allocation failure, return error */
+                return Status;
+            }
+
+            /* Zero fill memory used by PDE */
+            RtlZeroMemory((PVOID)Address, EFI_PAGE_SIZE);
+
+            /* Make PDE valid */
+            PdeBase[((MM_HARDWARE_VA_START >> MM_PDI_SHIFT) & 0x1FF) + Index].Valid = 1;
+            PdeBase[((MM_HARDWARE_VA_START >> MM_PDI_SHIFT) & 0x1FF) + Index].PageFrameNumber = Address / EFI_PAGE_SIZE;
+            PdeBase[((MM_HARDWARE_VA_START >> MM_PDI_SHIFT) & 0x1FF) + Index].Writable = 1;
+        }
+    }
+
+    /* Return success */
+    return STATUS_EFI_SUCCESS;
+}
+
 /**
  * Builds the actual memory mapping page table and enables paging. This routine exits EFI boot services as well.
  *
- * @param MemoryMappings
- *        Supplies a pointer to linked list containing all memory mappings.
- *
- * @param VirtualAddress
- *        Supplies a pointer to the next valid, free and available virtual address.
- *
- * @param ImageProtocol
- *        A pointer to the EFI loaded image protocol with information about where in memory the loader code was placed.
- *
- * @param PtePointer
- *        Supplies a pointer to memory area containing a Page Table Entries (PTE).
+ * @param PageMap
+ *        Supplies a pointer to the page mapping structure.
  *
  * @return This routine returns a status code.
  *
@@ -33,9 +212,12 @@ EFI_STATUS
 XtEnablePaging(IN PXTBL_PAGE_MAPPING PageMap)
 {
     EFI_STATUS Status;
+    EFI_PHYSICAL_ADDRESS TrampolineAddress;
+    PXT_TRAMPOLINE_ENTRY TrampolineEntry;
+    ULONG_PTR TrampolineSize;
 
     /* Build page map */
-    Status = XtLdrProtocol->Memory.BuildPageMap(PageMap, 0xFFFFF6FB7DBED000);
+    Status = XtLdrProtocol->Memory.BuildPageMap(PageMap, (PageMap->PageMapLevel > 4) ? MM_P5E_LA57_BASE : MM_PXE_BASE);
     if(Status != STATUS_EFI_SUCCESS)
     {
         /* Failed to build page map */
@@ -43,6 +225,38 @@ XtEnablePaging(IN PXTBL_PAGE_MAPPING PageMap)
         return Status;
     }
 
+    /* Map memory for hardware layer */
+    Status = XtpMapHardwareMemoryPool(PageMap);
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        /* Failed to map memory for hardware layer */
+        XtLdrProtocol->Debug.Print(L"Failed to map memory for hardware leyer (Status code: %zX)\n", Status);
+        return Status;
+    }
+
+    /* Check the configured page map level to set the LA57 state accordingly */
+    if(PageMap->PageMapLevel == 5)
+    {
+        /* Set the address of the trampoline code below 1MB */
+        TrampolineAddress = MM_TRAMPOLINE_ADDRESS;
+
+        /* Calculate the size of the trampoline code */
+        TrampolineSize = (ULONG_PTR)ArEnableExtendedPhysicalAddressingEnd - (ULONG_PTR)ArEnableExtendedPhysicalAddressing;
+
+        /* Allocate pages for the trampoline */
+        Status = XtLdrProtocol->Memory.AllocatePages(AllocateAddress, EFI_SIZE_TO_PAGES(TrampolineSize), &TrampolineAddress);
+        if(Status != STATUS_EFI_SUCCESS)
+        {
+            /* Failed to allocate memory for trampoline code */
+            XtLdrProtocol->Debug.Print(L"Failed to allocate memory for trampoline code (Status code: %zX)\n", Status);
+            return Status;
+        }
+
+        /* Set the trampoline entry point and copy its code into the allocated buffer */
+        TrampolineEntry = (PXT_TRAMPOLINE_ENTRY)(UINT_PTR)TrampolineAddress;
+        RtlCopyMemory(TrampolineEntry, ArEnableExtendedPhysicalAddressing, TrampolineSize);
+    }
+
     /* Exit EFI Boot Services */
     XtLdrProtocol->Debug.Print(L"Exiting EFI boot services\n");
     Status = XtLdrProtocol->Util.ExitBootServices();
@@ -53,8 +267,24 @@ XtEnablePaging(IN PXTBL_PAGE_MAPPING PageMap)
         return STATUS_EFI_ABORTED;
     }
 
-    /* Write PML4 to CR3 */
-    ArWriteControlRegister(3, (UINT_PTR)PageMap->PtePointer);
+    /* Check the configured page map level to set the LA57 state accordingly */
+    if(PageMap->PageMapLevel == 5)
+    {
+        /* Enable Linear Address 57-bit (LA57) extension */
+        XtLdrProtocol->Debug.Print(L"Enabling Linear Address 57-bit (LA57)\n");
+
+        /* Execute the trampoline to enable LA57 and write PML5 to CR3 */
+        TrampolineEntry((UINT64)PageMap->PtePointer);
+    }
+    else
+    {
+        /* Disable Linear Address 57-bit (LA57) extension */
+        XtLdrProtocol->Debug.Print(L"Disabling Linear Address 57-bit (LA57)\n");
+
+        /* Write PML4 to CR3 and enable paging */
+        ArWriteControlRegister(3, (UINT_PTR)PageMap->PtePointer);
+        ArWriteControlRegister(0, ArReadControlRegister(0) | CR0_PG);
+    }
 
     /* Return success */
     return STATUS_EFI_SUCCESS;

xtldr/modules/xtos_o/i686/memory.c

@@ -9,20 +9,111 @@
 #include <xtos.h>
 
 
+/**
+ * Determines the appropriate paging level (PML) for the i686 architecture.
+ *
+ * @param Parameters
+ *        A pointer to the wide character string containing the kernel boot parameters.
+ *
+ * @return This routine returns the appropriate page map level (3 if PAE is enabled, 2 otherwise).
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+ULONG
+XtpDeterminePagingLevel(IN CONST PWCHAR Parameters)
+{
+    CPUID_REGISTERS CpuRegisters;
+
+    /* Prepare CPUID registers to query for PAE support */
+    RtlZeroMemory(&CpuRegisters, sizeof(CPUID_REGISTERS));
+    CpuRegisters.Leaf = CPUID_GET_STANDARD1_FEATURES;
+
+    /* Query CPUID */
+    ArCpuId(&CpuRegisters);
+
+    /* Check if eXtended Physical Addressing (XPA) is enabled and if PAE is supported by the CPU */
+    if((CpuRegisters.Edx & CPUID_FEATURES_EDX_PAE) &&
+       !(XtLdrProtocol->BootUtil.GetBooleanParameter(Parameters, L"NOXPA")))
+    {
+        /* Enable PAE (PML3) */
+        return 3;
+    }
+
+    /* Disable PAE and use PML2 by default */
+    return 2;
+}
+
+/**
+ * Maps the page table for hardware layer addess space.
+ *
+ * @param PageMap
+ *        Supplies a pointer to the page mapping structure.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+EFI_STATUS
+XtpMapHardwareMemoryPool(IN PXTBL_PAGE_MAPPING PageMap)
+{
+    EFI_PHYSICAL_ADDRESS Address;
+    PHARDWARE_LEGACY_PTE LegacyPdeBase;
+    PHARDWARE_MODERN_PTE PdeBase;
+    EFI_STATUS Status;
+
+    /* Allocate memory */
+    Status = XtLdrProtocol->Memory.AllocatePages(AllocateAnyPages, 1, &Address);
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        /* Memory allocation failure, return error */
+        return Status;
+    }
+
+    /* Zero fill allocated memory */
+    RtlZeroMemory((PVOID)Address, EFI_PAGE_SIZE);
+
+    /* Check if PAE is enabled (3-level paging) */
+    if(PageMap->PageMapLevel == 3)
+    {
+        /* Get PDE base address (PAE enabled) */
+        PdeBase = (PHARDWARE_MODERN_PTE)(((PHARDWARE_MODERN_PTE)PageMap->PtePointer)[MM_HARDWARE_VA_START >> MM_PPI_SHIFT].PageFrameNumber << MM_PAGE_SHIFT);
+
+        /* Make PDE valid */
+        RtlZeroMemory(&PdeBase[(MM_HARDWARE_VA_START >> MM_PDI_SHIFT) & 0x1FF], sizeof(HARDWARE_MODERN_PTE));
+        PdeBase[(MM_HARDWARE_VA_START >> MM_PDI_SHIFT) & 0x1FF].PageFrameNumber = Address >> MM_PAGE_SHIFT;
+        PdeBase[(MM_HARDWARE_VA_START >> MM_PDI_SHIFT) & 0x1FF].Valid = 1;
+        PdeBase[(MM_HARDWARE_VA_START >> MM_PDI_SHIFT) & 0x1FF].Writable = 1;
+    }
+    else
+    {
+        /* Get PDE base address (PAE disabled) */
+        LegacyPdeBase = (PHARDWARE_LEGACY_PTE)PageMap->PtePointer;
+
+        /* Check for a conflicting PDE */
+        if(LegacyPdeBase[MM_HARDWARE_VA_START >> MM_PDI_LEGACY_SHIFT].Valid)
+        {
+            /* PDE already exists and is valid, nothing to do */
+            return STATUS_EFI_SUCCESS;
+        }
+
+        /* Make PDE valid  */
+        RtlZeroMemory(&LegacyPdeBase[MM_HARDWARE_VA_START >> MM_PDI_LEGACY_SHIFT], sizeof(HARDWARE_LEGACY_PTE));
+        LegacyPdeBase[MM_HARDWARE_VA_START >> MM_PDI_LEGACY_SHIFT].Valid = 1;
+        LegacyPdeBase[MM_HARDWARE_VA_START >> MM_PDI_LEGACY_SHIFT].PageFrameNumber = Address >> MM_PAGE_SHIFT;
+        LegacyPdeBase[MM_HARDWARE_VA_START >> MM_PDI_LEGACY_SHIFT].Writable = 1;
+    }
+
+    /* Return success */
+    return STATUS_EFI_SUCCESS;
+}
+
 /**
  * Builds the actual memory mapping page table and enables paging. This routine exits EFI boot services as well.
  *
- * @param MemoryMappings
- *        Supplies a pointer to linked list containing all memory mappings.
- *
- * @param VirtualAddress
- *        Supplies a pointer to the next valid, free and available virtual address.
- *
- * @param ImageProtocol
- *        A pointer to the EFI loaded image protocol with information about where in memory the loader code was placed.
- *
- * @param PtePointer
- *        Supplies a pointer to memory area containing a Page Table Entries (PTE).
+ * @param PageMap
+ *        Supplies a pointer to the page mapping structure.
  *
  * @return This routine returns a status code.
  *
@@ -32,30 +123,10 @@ XTCDECL
 EFI_STATUS
 XtEnablePaging(IN PXTBL_PAGE_MAPPING PageMap)
 {
-    PCPUID_REGISTERS CpuRegisters = NULL;
     EFI_STATUS Status;
 
-    /* Prepare CPUID registers */
-    CpuRegisters->Leaf = CPUID_GET_CPU_FEATURES;
-    CpuRegisters->SubLeaf = 0;
-    CpuRegisters->Eax = 0;
-    CpuRegisters->Ebx = 0;
-    CpuRegisters->Ecx = 0;
-    CpuRegisters->Edx = 0;
-
-    /* Get CPUID */
-    ArCpuId(CpuRegisters);
-
-    /* Store PAE status from the CPUID results */
-    if(!(CpuRegisters->Edx & CPUID_FEATURES_EDX_PAE))
-    {
-        /* No PAE support */
-        XtLdrProtocol->Debug.Print(L"ERROR: PAE extension not supported by the CPU\n");
-        return STATUS_EFI_UNSUPPORTED;
-    }
-
     /* Build page map */
-    Status = XtLdrProtocol->Memory.BuildPageMap(PageMap, 0xC0000000);
+    Status = XtLdrProtocol->Memory.BuildPageMap(PageMap, MM_PTE_BASE);
     if(Status != STATUS_EFI_SUCCESS)
     {
         /* Failed to build page map */
@@ -63,6 +134,15 @@ XtEnablePaging(IN PXTBL_PAGE_MAPPING PageMap)
         return Status;
     }
 
+    /* Map memory for hardware layer */
+    Status = XtpMapHardwareMemoryPool(PageMap);
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        /* Failed to map memory for hardware layer */
+        XtLdrProtocol->Debug.Print(L"Failed to map memory for hardware layer (Status code: %zX)\n", Status);
+        return Status;
+    }
+
     /* Exit EFI Boot Services */
     XtLdrProtocol->Debug.Print(L"Exiting EFI boot services\n");
     Status = XtLdrProtocol->Util.ExitBootServices();
@@ -73,8 +153,22 @@ XtEnablePaging(IN PXTBL_PAGE_MAPPING PageMap)
         return STATUS_EFI_ABORTED;
     }
 
-    /* Enable Physical Address Extension (PAE) */
-    ArWriteControlRegister(4, ArReadControlRegister(4) | CR4_PAE);
+    /* Disable paging */
+    ArWriteControlRegister(0, ArReadControlRegister(0) & ~CR0_PG);
+
+    /* Check the configured page map level to set the PAE state accordingly */
+    if(PageMap->PageMapLevel == 3)
+    {
+        /* Enable Physical Address Extension (PAE) */
+        XtLdrProtocol->Debug.Print(L"Enabling Physical Address Extension (PAE)\n");
+        ArWriteControlRegister(4, ArReadControlRegister(4) | CR4_PAE);
+    }
+    else
+    {
+        /* Disable Physical Address Extension (PAE) */
+        XtLdrProtocol->Debug.Print(L"Disabling Physical Address Extension (PAE)\n");
+        ArWriteControlRegister(4, ArReadControlRegister(4) & ~CR4_PAE);
+    }
 
     /* Write page mappings to CR3 */
     ArWriteControlRegister(3, (UINT_PTR)PageMap->PtePointer);

xtldr/modules/xtos_o/includes/xtos.h

@@ -29,9 +29,15 @@ typedef struct _XT_FRAMEBUFFER_PROTOCOL
 /* EFI XT Loader Protocol */
 EXTERN PXTBL_LOADER_PROTOCOL XtLdrProtocol;
 
+/* XTOS trampoline end address to calculate trampoline size */
+EXTERN PVOID ArEnableExtendedPhysicalAddressingEnd[];
+
 /* XTOS kernel entry point */
 typedef VOID (XTAPI *PXT_ENTRY_POINT)(IN PKERNEL_INITIALIZATION_BLOCK BootParameters);
 
+/* XTOS trampoline entry point */
+typedef VOID (*PXT_TRAMPOLINE_ENTRY)(UINT64 PageMap);
+
 /* XTOS boot protocol related routines forward references */
 XTCDECL
 EFI_STATUS
@@ -49,6 +55,10 @@ XTCDECL
 LOADER_MEMORY_TYPE
 XtConvertEfiMemoryType(IN EFI_MEMORY_TYPE EfiMemoryType);
 
+XTCDECL
+ULONG
+XtpDeterminePagingLevel(IN CONST PWCHAR Parameters);
+
 XTCDECL
 EFI_STATUS
 XtEnablePaging(IN PXTBL_PAGE_MAPPING PageMap);
@@ -96,6 +106,10 @@ XtpLoadModule(IN PEFI_FILE_HANDLE BootDir,
               IN LOADER_MEMORY_TYPE MemoryType,
               OUT PPECOFF_IMAGE_CONTEXT *ImageContext);
 
+XTCDECL
+EFI_STATUS
+XtpMapHardwareMemoryPool(IN PXTBL_PAGE_MAPPING PageMap);
+
 XTCDECL
 EFI_STATUS
 BlXtLdrModuleMain(IN EFI_HANDLE ImageHandle,

xtldr/modules/xtos_o/xtos.c

@@ -37,40 +37,51 @@ XTBL_BOOT_PROTOCOL XtBootProtocol;
  */
 XTCDECL
 VOID
-XtGetDisplayInformation(OUT PLOADER_GRAPHICS_INFORMATION_BLOCK InformationBlock,
+XtGetDisplayInformation(OUT PSYSTEM_RESOURCE_FRAMEBUFFER FrameBufferResource,
                         IN PEFI_PHYSICAL_ADDRESS FrameBufferBase,
                         IN PULONG_PTR FrameBufferSize,
                         IN PXTBL_FRAMEBUFFER_MODE_INFORMATION FrameBufferModeInfo)
 {
-    InformationBlock->Initialized = TRUE;
-    InformationBlock->Address = (PVOID)*FrameBufferBase;
-    InformationBlock->BufferSize = *FrameBufferSize;
-    InformationBlock->Width = FrameBufferModeInfo->Width;
-    InformationBlock->Height = FrameBufferModeInfo->Height;
-    InformationBlock->BitsPerPixel = FrameBufferModeInfo->BitsPerPixel;
-    InformationBlock->PixelsPerScanLine = FrameBufferModeInfo->PixelsPerScanLine;
-    InformationBlock->Pitch = FrameBufferModeInfo->Pitch;
+    /* Fill in frame buffer resource */
+    FrameBufferResource->Header.PhysicalAddress = (PVOID)*FrameBufferBase;
+    FrameBufferResource->Header.ResourceType = SystemResourceFrameBuffer;
+    FrameBufferResource->Header.ResourceSize = sizeof(SYSTEM_RESOURCE_FRAMEBUFFER);
+    FrameBufferResource->BufferSize = *FrameBufferSize;
+    FrameBufferResource->Width = FrameBufferModeInfo->Width;
+    FrameBufferResource->Height = FrameBufferModeInfo->Height;
+    FrameBufferResource->Depth = FrameBufferModeInfo->Depth;
+    FrameBufferResource->BitsPerPixel = FrameBufferModeInfo->BitsPerPixel;
+    FrameBufferResource->PixelsPerScanLine = FrameBufferModeInfo->PixelsPerScanLine;
+    FrameBufferResource->Pitch = FrameBufferModeInfo->Pitch;
+    FrameBufferResource->Pixels.BlueShift = FrameBufferModeInfo->PixelInformation.BlueShift;
+    FrameBufferResource->Pixels.BlueSize = FrameBufferModeInfo->PixelInformation.BlueSize;
+    FrameBufferResource->Pixels.GreenShift = FrameBufferModeInfo->PixelInformation.GreenShift;
+    FrameBufferResource->Pixels.GreenSize = FrameBufferModeInfo->PixelInformation.GreenSize;
+    FrameBufferResource->Pixels.RedShift = FrameBufferModeInfo->PixelInformation.RedShift;
+    FrameBufferResource->Pixels.RedSize = FrameBufferModeInfo->PixelInformation.RedSize;
+    FrameBufferResource->Pixels.ReservedShift = FrameBufferModeInfo->PixelInformation.ReservedShift;
+    FrameBufferResource->Pixels.ReservedSize = FrameBufferModeInfo->PixelInformation.ReservedSize;
 }
 
 XTCDECL
 EFI_STATUS
 XtGetMemoryDescriptorList(IN PXTBL_PAGE_MAPPING PageMap,
-                          IN PVOID VirtualAddress,
+                          IN PVOID *VirtualAddress,
                           OUT PLIST_ENTRY MemoryDescriptorList)
 {
     EFI_PHYSICAL_ADDRESS Address;
     EFI_STATUS Status;
     ULONGLONG Pages;
 
-    Pages = (ULONGLONG)EFI_SIZE_TO_PAGES((PageMap->MapSize + 1) * sizeof(LOADER_MEMORY_MAPPING));
+    Pages = (ULONGLONG)EFI_SIZE_TO_PAGES((PageMap->MapSize + 1) * sizeof(LOADER_MEMORY_DESCRIPTOR));
 
-    Status = XtLdrProtocol->Memory.AllocatePages(Pages, &Address);
+    Status = XtLdrProtocol->Memory.AllocatePages(AllocateAnyPages, Pages, &Address);
     if(Status != STATUS_EFI_SUCCESS)
     {
         return Status;
     }
 
-    Status = XtLdrProtocol->Memory.MapVirtualMemory(PageMap, VirtualAddress, (PVOID)Address, Pages, LoaderMemoryData);
+    Status = XtLdrProtocol->Memory.MapVirtualMemory(PageMap, *VirtualAddress, (PVOID)Address, Pages, LoaderMemoryData);
     if(Status != STATUS_EFI_SUCCESS)
     {
         XtLdrProtocol->Memory.FreePages(Address, Pages);
@@ -84,7 +95,7 @@ XtGetMemoryDescriptorList(IN PXTBL_PAGE_MAPPING PageMap,
     while(ListEntry != &PageMap->MemoryMap)
     {
         PXTBL_MEMORY_MAPPING MemoryMapping = CONTAIN_RECORD(ListEntry, XTBL_MEMORY_MAPPING, ListEntry);
-        PLOADER_MEMORY_MAPPING MemoryDescriptor = (PLOADER_MEMORY_MAPPING)Address;
+        PLOADER_MEMORY_DESCRIPTOR MemoryDescriptor = (PLOADER_MEMORY_DESCRIPTOR)Address;
 
         MemoryDescriptor->MemoryType = MemoryMapping->MemoryType;
         MemoryDescriptor->BasePage = (UINT_PTR)MemoryMapping->PhysicalAddress / EFI_PAGE_SIZE;
@@ -92,11 +103,126 @@ XtGetMemoryDescriptorList(IN PXTBL_PAGE_MAPPING PageMap,
 
         RtlInsertTailList(MemoryDescriptorList, &MemoryDescriptor->ListEntry);
 
-        Address = Address + sizeof(LOADER_MEMORY_MAPPING);
+        Address = Address + sizeof(LOADER_MEMORY_DESCRIPTOR);
         ListEntry = ListEntry->Flink;
     }
 
-    XtLdrProtocol->Memory.PhysicalListToVirtual(PageMap, MemoryDescriptorList, PhysicalBase, VirtualAddress);
+    XtLdrProtocol->Memory.PhysicalListToVirtual(PageMap, MemoryDescriptorList, PhysicalBase, *VirtualAddress);
+
+    return STATUS_EFI_SUCCESS;
+}
+
+XTCDECL
+EFI_STATUS
+XtGetSystemResourcesList(IN PXTBL_PAGE_MAPPING PageMap,
+                         IN PVOID *VirtualAddress,
+                         OUT PLIST_ENTRY SystemResourcesList)
+{
+    XTSTATUS Status;
+    EFI_HANDLE ProtocolHandle;
+    EFI_GUID AcpiGuid = XT_ACPI_PROTOCOL_GUID;
+    EFI_GUID FrameBufGuid = XT_FRAMEBUFFER_PROTOCOL_GUID;
+    PXTBL_ACPI_PROTOCOL AcpiProtocol;
+    PXTBL_FRAMEBUFFER_PROTOCOL FrameBufProtocol;
+    XTBL_FRAMEBUFFER_MODE_INFORMATION FbModeInfo;
+    EFI_PHYSICAL_ADDRESS FbAddress;
+    ULONG_PTR FbSize;
+    UINT FrameBufferPages;
+    PSYSTEM_RESOURCE_FRAMEBUFFER FrameBufferResource;
+    PSYSTEM_RESOURCE_ACPI AcpiResource;
+    ULONGLONG Pages;
+    EFI_PHYSICAL_ADDRESS Address;
+    PVOID PhysicalBase, VirtualBase;
+
+    Pages = (ULONGLONG)EFI_SIZE_TO_PAGES(sizeof(SYSTEM_RESOURCE_ACPI) + sizeof(SYSTEM_RESOURCE_FRAMEBUFFER));
+
+    Status = XtLdrProtocol->Memory.AllocatePages(AllocateAnyPages, Pages, &Address);
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        return Status;
+    }
+    Status = XtLdrProtocol->Memory.MapVirtualMemory(PageMap, *VirtualAddress, (PVOID)Address, Pages, LoaderFirmwarePermanent);
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        XtLdrProtocol->Memory.FreePages(Address, Pages);
+        return Status;
+    }
+
+    PhysicalBase = (PVOID)Address;
+    VirtualBase = *VirtualAddress;
+
+    /* Calculate next valid virtual address */
+    *VirtualAddress += (UINT_PTR)(Pages * EFI_PAGE_SIZE);
+
+    AcpiResource = (PSYSTEM_RESOURCE_ACPI)Address;
+
+    RtlZeroMemory(AcpiResource, sizeof(SYSTEM_RESOURCE_ACPI));
+
+    /* Load FrameBuffer protocol */
+    Status = XtLdrProtocol->Protocol.Open(&ProtocolHandle, (PVOID*)&AcpiProtocol, &AcpiGuid);
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        return Status;
+    }
+
+    AcpiResource->Header.ResourceType = SystemResourceAcpi;
+    AcpiResource->Header.ResourceSize = sizeof(SYSTEM_RESOURCE_ACPI);
+
+    /* Get APIC and XSDP/RSDP addresses */
+    AcpiProtocol->GetApicBase(&AcpiResource->ApicBase);
+    AcpiProtocol->GetAcpiDescriptionPointer(&AcpiResource->Header.PhysicalAddress);
+
+    /* No need to map ACPI */
+    AcpiResource->Header.VirtualAddress = 0;
+
+    RtlInsertTailList(SystemResourcesList, &AcpiResource->Header.ListEntry);
+
+    /* Close FrameBuffer protocol */
+    XtLdrProtocol->Protocol.Close(ProtocolHandle, &FrameBufGuid);
+
+    Address = Address + sizeof(SYSTEM_RESOURCE_ACPI);
+
+    FrameBufferResource = (PSYSTEM_RESOURCE_FRAMEBUFFER)Address;
+
+    RtlZeroMemory(FrameBufferResource, sizeof(SYSTEM_RESOURCE_FRAMEBUFFER));
+
+    /* Load FrameBuffer protocol */
+    Status = XtLdrProtocol->Protocol.Open(&ProtocolHandle, (PVOID*)&FrameBufProtocol, &FrameBufGuid);
+    if(Status == STATUS_EFI_SUCCESS)
+    {
+        /* Get FrameBuffer information */
+        Status = FrameBufProtocol->GetDisplayInformation(&FbAddress, &FbSize, &FbModeInfo);
+        if(Status == STATUS_EFI_SUCCESS)
+        {
+
+            /* Store information about FrameBuffer device */
+            XtGetDisplayInformation(FrameBufferResource, &FbAddress, &FbSize, &FbModeInfo);
+        }
+    }
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        return Status;
+    }
+
+    /* Calculate pages needed to map framebuffer */
+    FrameBufferPages = EFI_SIZE_TO_PAGES(FbSize);
+
+    /* Rewrite framebuffer address by using virtual address */
+    FrameBufferResource->Header.VirtualAddress = *VirtualAddress;
+
+    /* Map frame buffer memory */
+    XtLdrProtocol->Memory.MapVirtualMemory(PageMap, FrameBufferResource->Header.VirtualAddress,
+                                           FrameBufferResource->Header.PhysicalAddress,
+                                           FrameBufferPages, LoaderFirmwarePermanent);
+
+    /* Close FrameBuffer protocol */
+    XtLdrProtocol->Protocol.Close(ProtocolHandle, &FrameBufGuid);
+
+    *VirtualAddress += (UINT_PTR)(FrameBufferPages * EFI_PAGE_SIZE);
+
+    RtlInsertTailList(SystemResourcesList, &FrameBufferResource->Header.ListEntry);
+
+    XtLdrProtocol->Memory.PhysicalListToVirtual(PageMap, SystemResourcesList, PhysicalBase, VirtualBase);
 
     return STATUS_EFI_SUCCESS;
 }
@@ -197,7 +323,7 @@ XtBootSystem(IN PXTBL_BOOT_PARAMETERS Parameters)
                                Parameters->KernelFile, Parameters->Parameters);
 
     /* Open EFI volume */
-    Status = XtLdrProtocol->Disk.OpenVolume(NULL, &DiskHandle, &FsHandle);
+    Status = XtLdrProtocol->Disk.OpenVolume(Parameters->DevicePath, &DiskHandle, &FsHandle);
     if(Status != STATUS_EFI_SUCCESS)
     {
         /* Failed to open a volume */
@@ -284,7 +410,7 @@ XtpBootSequence(IN PEFI_FILE_HANDLE BootDir,
     VirtualAddress = (PVOID)(KSEG0_BASE + KSEG0_KERNEL_BASE);
 
     /* Initialize virtual memory mappings */
-    XtLdrProtocol->Memory.InitializePageMap(&PageMap, 3, Size4K);
+    XtLdrProtocol->Memory.InitializePageMap(&PageMap, XtpDeterminePagingLevel(Parameters->Parameters), Size4K);
 
     Status = XtLdrProtocol->Memory.MapEfiMemory(&PageMap, &VirtualMemoryArea, NULL);
     if(Status != STATUS_EFI_SUCCESS)
@@ -416,22 +542,21 @@ XtpInitializeLoaderBlock(IN PXTBL_PAGE_MAPPING PageMap,
                          IN PVOID *VirtualAddress,
                          IN PXTBL_BOOT_PARAMETERS Parameters)
 {
-    EFI_GUID FrameBufGuid = XT_FRAMEBUFFER_PROTOCOL_GUID;
-    PXTBL_FRAMEBUFFER_PROTOCOL FrameBufProtocol;
-    XTBL_FRAMEBUFFER_MODE_INFORMATION FbModeInfo;
     PKERNEL_INITIALIZATION_BLOCK LoaderBlock;
-    EFI_PHYSICAL_ADDRESS Address, FbAddress;
+    EFI_PHYSICAL_ADDRESS Address;
     // PVOID RuntimeServices;
-    ULONG_PTR FbSize;
     EFI_STATUS Status;
-    EFI_HANDLE ProtocolHandle;
-    UINT BlockPages, FrameBufferPages;
+    UINT BlockPages;
+    UINT ParametersSize;
+
+    /* Calculate size of parameters */
+    ParametersSize = (RtlWideStringLength(Parameters->Parameters, 0) + 1) * sizeof(WCHAR);
 
     /* Calculate number of pages needed for initialization block */
-    BlockPages = EFI_SIZE_TO_PAGES(sizeof(KERNEL_INITIALIZATION_BLOCK));
+    BlockPages = EFI_SIZE_TO_PAGES(sizeof(KERNEL_INITIALIZATION_BLOCK) + ParametersSize);
 
     /* Allocate memory for kernel initialization block */
-    Status = XtLdrProtocol->Memory.AllocatePages(BlockPages, &Address);
+    Status = XtLdrProtocol->Memory.AllocatePages(AllocateAnyPages, BlockPages, &Address);
     if(Status != STATUS_EFI_SUCCESS)
     {
         /* Memory allocation failure */
@@ -440,7 +565,7 @@ XtpInitializeLoaderBlock(IN PXTBL_PAGE_MAPPING PageMap,
 
     /* Initialize and zero-fill kernel initialization block */
     LoaderBlock = (PKERNEL_INITIALIZATION_BLOCK)(UINT_PTR)Address;
-    RtlZeroMemory(LoaderBlock, sizeof(KERNEL_INITIALIZATION_BLOCK));
+    RtlZeroMemory(LoaderBlock, sizeof(KERNEL_INITIALIZATION_BLOCK) + ParametersSize);
 
     /* Set basic loader block properties */
     LoaderBlock->BlockSize = sizeof(KERNEL_INITIALIZATION_BLOCK);
@@ -450,30 +575,6 @@ XtpInitializeLoaderBlock(IN PXTBL_PAGE_MAPPING PageMap,
     /* Set LoaderInformation block properties */
     LoaderBlock->LoaderInformation.DbgPrint = XtLdrProtocol->Debug.Print;
 
-    /* Load FrameBuffer protocol */
-    Status = XtLdrProtocol->Protocol.Open(&ProtocolHandle, (PVOID*)&FrameBufProtocol, &FrameBufGuid);
-    if(Status == STATUS_EFI_SUCCESS)
-    {
-        /* Get FrameBuffer information */
-        Status = FrameBufProtocol->GetDisplayInformation(&FbAddress, &FbSize, &FbModeInfo);
-        if(Status == STATUS_EFI_SUCCESS)
-        {
-            /* Store information about FrameBuffer device */
-            XtGetDisplayInformation(&LoaderBlock->LoaderInformation.FrameBuffer, &FbAddress, &FbSize, &FbModeInfo);
-        }
-    }
-    if(Status != STATUS_EFI_SUCCESS)
-    {
-        /* No FrameBuffer available */
-        LoaderBlock->LoaderInformation.FrameBuffer.Initialized = FALSE;
-    }
-
-    /* Store page map level */
-    LoaderBlock->LoaderInformation.PageMapLevel = 3;
-
-    /* Close FrameBuffer protocol */
-    XtLdrProtocol->Protocol.Close(ProtocolHandle, &FrameBufGuid);
-
     /* Attempt to find virtual address of the EFI Runtime Services */
     // Status = XtLdrProtocol->GetVirtualAddress(MemoryMappings, &EfiSystemTable->RuntimeServices->Hdr, &RuntimeServices);
     // if(Status == STATUS_EFI_SUCCESS)
@@ -490,8 +591,10 @@ XtpInitializeLoaderBlock(IN PXTBL_PAGE_MAPPING PageMap,
     // }
 
     /* Copy parameters to kernel initialization block */
-    RtlCopyMemory(LoaderBlock->KernelParameters, Parameters->Parameters,
-                  (RtlWideStringLength(Parameters->Parameters, 0) + 1) * sizeof(WCHAR));
+    LoaderBlock->KernelParameters = (PWCHAR)((UINT_PTR)*VirtualAddress + sizeof(KERNEL_INITIALIZATION_BLOCK));
+    RtlCopyMemory((PVOID)((UINT_PTR)LoaderBlock + sizeof(KERNEL_INITIALIZATION_BLOCK)),
+                  Parameters->Parameters,
+                  ParametersSize);
 
     /* Map kernel initialization block */
     XtLdrProtocol->Memory.MapVirtualMemory(PageMap, *VirtualAddress, (PVOID)LoaderBlock,
@@ -500,27 +603,12 @@ XtpInitializeLoaderBlock(IN PXTBL_PAGE_MAPPING PageMap,
     /* Calculate next valid virtual address */
     *VirtualAddress += (UINT_PTR)(BlockPages * EFI_PAGE_SIZE);
 
-    /* Check if framebuffer initialized */
-    if(LoaderBlock->LoaderInformation.FrameBuffer.Initialized)
-    {
-        /* Calculate pages needed to map framebuffer */
-        FrameBufferPages = EFI_SIZE_TO_PAGES(LoaderBlock->LoaderInformation.FrameBuffer.BufferSize);
-
-        /* Map frame buffer memory */
-        XtLdrProtocol->Memory.MapVirtualMemory(PageMap, *VirtualAddress,
-                                               LoaderBlock->LoaderInformation.FrameBuffer.Address,
-                                               FrameBufferPages, LoaderFirmwarePermanent);
-
-        /* Rewrite framebuffer address by using virtual address */
-        LoaderBlock->LoaderInformation.FrameBuffer.Address = *VirtualAddress;
-
-        /* Calcualate next valid virtual address */
-        *VirtualAddress += (UINT_PTR)(FrameBufferPages * EFI_PAGE_SIZE);
-    }
+    RtlInitializeListHead(&LoaderBlock->SystemResourcesListHead);
+    XtGetSystemResourcesList(PageMap, VirtualAddress, &LoaderBlock->SystemResourcesListHead);
 
     /* Initialize memory descriptor list */
     RtlInitializeListHead(&LoaderBlock->MemoryDescriptorListHead);
-    XtGetMemoryDescriptorList(PageMap, *VirtualAddress, &LoaderBlock->MemoryDescriptorListHead);
+    XtGetMemoryDescriptorList(PageMap, VirtualAddress, &LoaderBlock->MemoryDescriptorListHead);
 
     /* Return success */
     return STATUS_EFI_SUCCESS;

xtldr/protocol.c

@@ -749,100 +749,96 @@ BlpGetModuleInfoStrings(IN PWCHAR SectionData,
     EFI_STATUS Status;
     PWCHAR *Array;
     PWCHAR String;
+    ULONG DataSize;
 
     /* Check input parameters */
     InfoStrings = SectionData;
     if(!InfoStrings || !SectionSize)
     {
         /* Invalid input parameters */
+        *ModInfo = NULL;
+        *InfoCount = 0;
         return STATUS_EFI_INVALID_PARAMETER;
     }
 
-    /* Skip zero padding */
-    while(InfoStrings[0] == L'\0' && SectionSize > 1)
+    /* Calculate the size of the data based on the size of the section */
+    DataSize = SectionSize / sizeof(WCHAR);
+
+    /* Skip zero padding at the beginning */
+    while(DataSize > 0 && *InfoStrings == L'\0')
     {
-        /* Get next character and decrement section size */
         InfoStrings++;
-        SectionSize--;
+        DataSize--;
     }
 
     /* Make sure there is at least one string available */
-    if(SectionSize <= 1)
+    if(DataSize < 1)
     {
         /* No strings found */
+        *ModInfo = NULL;
+        *InfoCount = 0;
         return STATUS_EFI_END_OF_FILE;
     }
 
     /* Count number of strings */
     Index = 0;
     Count = 0;
-    while(Index < SectionSize)
+    while(Index < DataSize)
     {
-        /* Get to the next string */
-        if(InfoStrings[Index] != L'\0')
-        {
-            /* Get next character */
-            Index++;
-            continue;
-        }
+        /* Found start of a new string */
+        Count++;
 
-        /* Skip zero padding */
-        while(InfoStrings[Index] == L'\0' && Index < SectionSize)
+        /* Go to the end of the string */
+        while(Index < DataSize && InfoStrings[Index] != L'\0')
+        {
+            Index++;
+        }
+        /* Skip all null terminators */
+        while(Index < DataSize && InfoStrings[Index] == L'\0')
         {
-            /* Get next character */
             Index++;
         }
-
-        /* New string found, increment counter */
-        Count++;
     }
 
-    /* Make sure there is no missing string */
-    if(InfoStrings[Index - 1] != L'\0')
-    {
-        /* One more string available */
-        Count++;
-    }
-
-    /* Allocate memory for array of strings */
-    Status = BlAllocateMemoryPool(SectionSize + 1 + sizeof(PWCHAR) * (Count + 1), (PVOID *)&Array);
+    /* Allocate memory for the pointer array and the string data */
+    Status = BlAllocateMemoryPool(sizeof(PWCHAR) * (Count + 1) + (DataSize + 1) * sizeof(WCHAR), (PVOID *)&Array);
     if(Status != STATUS_EFI_SUCCESS)
     {
         /* Failed to allocate memory */
         return STATUS_EFI_OUT_OF_RESOURCES;
     }
 
-    /* Allocate memory and copy strings read from '.modinfo' section */
-    BlAllocateMemoryPool(SectionSize, (PVOID*)&String);
-    RtlCopyMemory(String, InfoStrings, SectionSize);
+    /* The string buffer is located right after the pointer array */
+    String = (PWCHAR)(Array + Count + 1);
 
-    /* Make sure last string is NULL-terminated */
+    /* Copy the raw string data */
+    RtlCopyMemory(String, InfoStrings, DataSize * sizeof(WCHAR));
+
+    /* Ensure the entire buffer is null-terminated for safety */
+    String[DataSize] = L'\0';
+
+    /* Set the last element of the pointer array to NULL */
     Array[Count] = NULL;
-    Array[0] = String;
 
-    /* Parse strings into array */
+    /* Populate the array with pointers to the strings within the buffer */
     Index = 0;
-    ArrayIndex = 1;
-    while(Index < SectionSize && ArrayIndex < Count)
+    ArrayIndex = 0;
+    while(Index < DataSize && ArrayIndex < Count)
     {
-        /* Get to the next string */
-        if(String[Index] != L'\0')
-        {
-            /* Get next character */
-            Index++;
-            continue;
-        }
+        /* Set pointer to the beginning of the string */
+        Array[ArrayIndex++] = &String[Index];
 
-        /* Skip zero padding */
-        while(InfoStrings[Index] == L'\0' && Index < SectionSize)
+        /* Find the end of the current string */
+        while(Index < DataSize && String[Index] != L'\0')
         {
-            /* Get next character */
             Index++;
         }
 
-        /* Push string into array */
-        Array[ArrayIndex] = &String[Index];
-        ArrayIndex++;
+        /* Skip all null terminators to find the beginning of the next string */
+        while(Index < DataSize && String[Index] == L'\0')
+        {
+            Index++;
+        }
     }
 
     /* Return array of strings and its size */
@@ -872,8 +868,12 @@ BlpInstallXtLoaderProtocol()
     BlpLdrProtocol.Boot.InvokeProtocol = BlInvokeBootProtocol;
     BlpLdrProtocol.Boot.RegisterMenu = BlRegisterBootMenu;
     BlpLdrProtocol.Boot.RegisterProtocol = BlRegisterBootProtocol;
+    BlpLdrProtocol.BootUtil.GetBooleanParameter = BlGetBooleanParameter;
     BlpLdrProtocol.Config.GetBooleanValue = BlGetConfigBooleanValue;
+    BlpLdrProtocol.Config.GetBootOptionValue = BlGetBootOptionValue;
+    BlpLdrProtocol.Config.GetEditableOptions = BlGetEditableOptions;
     BlpLdrProtocol.Config.GetValue = BlGetConfigValue;
+    BlpLdrProtocol.Config.SetBootOptionValue = BlSetBootOptionValue;
     BlpLdrProtocol.Console.ClearLine = BlClearConsoleLine;
     BlpLdrProtocol.Console.ClearScreen = BlClearConsoleScreen;
     BlpLdrProtocol.Console.DisableCursor = BlDisableConsoleCursor;

xtldr/textui.c

@@ -4,6 +4,7 @@
  * FILE:            xtldr/textui.c
  * DESCRIPTION:     Text console User Interface (TUI) support for XT Boot Loader
  * DEVELOPERS:      Rafal Kupiec <belliash@codingworkshop.eu.org>
+ *                  Aiken Harris <harraiken91@gmail.com>
  */
 
 #include <xtldr.h>
@@ -22,7 +23,9 @@ BlDisplayBootMenu()
 {
     XTBL_DIALOG_HANDLE Handle;
     PXTBL_BOOTMENU_ITEM MenuEntries = NULL;
-    ULONG Index, NumberOfEntries, HighligtedEntryId;
+    ULONG Index;
+    ULONG HighligtedEntryId, OldHighligtedEntryId, NumberOfEntries, TopVisibleEntry, VisibleEntries;
+    BOOLEAN RedrawBootMenu, RedrawEntries;
     UINT_PTR EventIndex;
     EFI_EVENT Events[2];
     EFI_INPUT_KEY Key;
@@ -31,16 +34,30 @@ BlDisplayBootMenu()
     LONG TimeOut;
     PWCHAR TimeOutString;
 
+    /* Draw boot menu */
+    BlpDrawBootMenu(&Handle);
+
     /* Initialize boot menu list */
-    Status = BlInitializeBootMenuList(&MenuEntries, &NumberOfEntries, &HighligtedEntryId);
+    TopVisibleEntry = 0;
+    Status = BlInitializeBootMenuList(Handle.Width - 4, &MenuEntries, &NumberOfEntries, &HighligtedEntryId);
     if(Status != STATUS_EFI_SUCCESS)
     {
         /* Failed to initialize boot menu list, exit into XTLDR shell */
         return;
     }
 
+    /* Calculate how many entries can be visible in the menu box */
+    VisibleEntries = Handle.Height - 2;
+
+    /* Adjust the view if the default entry is not initially visible */
+    if(HighligtedEntryId >= VisibleEntries)
+    {
+        /* Scroll the view to make the highlighted entry the last visible one */
+        TopVisibleEntry = HighligtedEntryId - VisibleEntries + 1;
+    }
+
     /* Get timeout from the configuration */
-    TimeOutString = BlGetConfigValue(L"TIMEOUT");
+    BlGetConfigValue(L"TIMEOUT", &TimeOutString);
     TimeOut = -1;
 
     /* Check if timeout is specified */
@@ -56,26 +73,43 @@ BlDisplayBootMenu()
         }
     }
 
+    /* Set redraw flags to not redraw the menu itself, but fill it with entries */
+    RedrawBootMenu = FALSE;
+    RedrawEntries = TRUE;
+
     /* Infinite boot menu loop */
     while(TRUE)
     {
-        /* Draw boot menu */
-        BlpDrawBootMenu(&Handle);
+        /* Redraw boot menu frame if requested */
+        if(RedrawBootMenu)
+        {
+            BlpDrawBootMenu(&Handle);
+            RedrawBootMenu = FALSE;
+            RedrawEntries = TRUE;
+        }
+
+        /* Sanity check to ensure we do not display more entries than possible */
+        if(VisibleEntries > NumberOfEntries)
+        {
+            VisibleEntries = NumberOfEntries;
+        }
 
         /* Check if there is anything to show in the boot menu */
-        if(NumberOfEntries > 0) {
-            /* Check if all menu entries will fit into the menu box */
-            if(NumberOfEntries > Handle.Height - 2)
+        if(NumberOfEntries > 0)
+        {
+            /* Check if we need to redraw boot menu entries */
+            if(RedrawEntries)
             {
-                /* Too many menu entries, limit entries to match box height (-2 for upper and bottom borders) */
-                NumberOfEntries = Handle.Height - 2;
-            }
+                /* Iterate through all menu entries */
+                for(Index = 0; Index < VisibleEntries; Index++)
+                {
+                    /* Draw menu entry */
+                    BlpDrawBootMenuEntry(&Handle, MenuEntries[TopVisibleEntry + Index].EntryName,
+                                         Index, (TopVisibleEntry + Index) == HighligtedEntryId);
+                }
 
-            /* Iterate through all menu entries */
-            for(Index = 0; Index < NumberOfEntries; Index++)
-            {
-                /* Draw menu entry */
-                BlpDrawBootMenuEntry(&Handle, MenuEntries[Index].EntryName, Index, Index == HighligtedEntryId);
+                /* Clear redraw entries flag */
+                RedrawEntries = FALSE;
             }
         }
         else
@@ -109,6 +143,9 @@ BlDisplayBootMenu()
         /* Flush keyboard buffer out of any keystrokes */
         EfiSystemTable->ConIn->Reset(EfiSystemTable->ConIn, FALSE);
 
+        /* Store old highlighted entry */
+        OldHighligtedEntryId = HighligtedEntryId;
+
         /* Infinite boot menu event loop */
         while(TRUE)
         {
@@ -140,7 +177,7 @@ BlDisplayBootMenu()
                     BlSetConsoleAttributes(Handle.DialogColor | Handle.TextColor);
                     BlClearConsoleLine(Handle.PosY + Handle.Height + 4);
                     BlSetCursorPosition(4, Handle.PosY + Handle.Height + 4);
-                    BlConsolePrint(L"Booting '%S' now...\n", MenuEntries[HighligtedEntryId].EntryName);
+                    BlConsolePrint(L"Booting '%S' now...\n", MenuEntries[HighligtedEntryId].FullName);
 
                     /* Boot the highlighted (chosen) OS */
                     Status = BlInvokeBootProtocol(MenuEntries[HighligtedEntryId].ShortName,
@@ -149,8 +186,9 @@ BlDisplayBootMenu()
                     {
                         /* Failed to boot OS */
                         BlDebugPrint(L"ERROR: Failed to boot '%S' (Status Code: 0x%zX)\n",
-                                     MenuEntries[HighligtedEntryId].EntryName, Status);
+                                     MenuEntries[HighligtedEntryId].FullName, Status);
                         BlDisplayErrorDialog(L"XTLDR", L"Failed to startup the selected Operating System.");
+                        RedrawBootMenu = TRUE;
                     }
 
                     /* Break from boot menu event loop to redraw whole boot menu */
@@ -162,11 +200,23 @@ BlDisplayBootMenu()
                     if(HighligtedEntryId > 0)
                     {
                         /* Highlight previous entry */
+                        OldHighligtedEntryId = HighligtedEntryId;
                         HighligtedEntryId--;
-                        BlpDrawBootMenuEntry(&Handle, MenuEntries[HighligtedEntryId + 1].EntryName,
-                                             HighligtedEntryId + 1, FALSE);
+
+                        /* Check if we need to scroll the view */
+                        if(HighligtedEntryId < TopVisibleEntry)
+                        {
+                            /* Scroll the view */
+                            TopVisibleEntry = HighligtedEntryId;
+                            RedrawEntries = TRUE;
+                            break;
+                        }
+
+                        /* Redraw new highlighted entry and the old one */
+                        BlpDrawBootMenuEntry(&Handle, MenuEntries[OldHighligtedEntryId].EntryName,
+                                             OldHighligtedEntryId - TopVisibleEntry, FALSE);
                         BlpDrawBootMenuEntry(&Handle, MenuEntries[HighligtedEntryId].EntryName,
-                                             HighligtedEntryId, TRUE);
+                                             HighligtedEntryId - TopVisibleEntry, TRUE);
                     }
                 }
                 else if(Key.ScanCode == 0x02)
@@ -175,11 +225,23 @@ BlDisplayBootMenu()
                     if(HighligtedEntryId < NumberOfEntries - 1)
                     {
                         /* Highlight next entry */
+                        OldHighligtedEntryId = HighligtedEntryId;
                         HighligtedEntryId++;
-                        BlpDrawBootMenuEntry(&Handle, MenuEntries[HighligtedEntryId - 1].EntryName,
-                                             HighligtedEntryId - 1, FALSE);
+
+                        /* Check if we need to scroll the view */
+                        if(HighligtedEntryId >= TopVisibleEntry + VisibleEntries)
+                        {
+                            /* Scroll the view */
+                            TopVisibleEntry = HighligtedEntryId - VisibleEntries + 1;
+                            RedrawEntries = TRUE;
+                            break;
+                        }
+
+                        /* Redraw new highlighted entry and the old one */
+                        BlpDrawBootMenuEntry(&Handle, MenuEntries[OldHighligtedEntryId].EntryName,
+                                             OldHighligtedEntryId - TopVisibleEntry, FALSE);
                         BlpDrawBootMenuEntry(&Handle, MenuEntries[HighligtedEntryId].EntryName,
-                                             HighligtedEntryId, TRUE);
+                                             HighligtedEntryId - TopVisibleEntry, TRUE);
                     }
                 }
                 else if(Key.ScanCode == 0x09)
@@ -188,12 +250,10 @@ BlDisplayBootMenu()
                     if(HighligtedEntryId != 0)
                     {
                         /* Highlight first entry */
-                        BlpDrawBootMenuEntry(&Handle, MenuEntries[HighligtedEntryId].EntryName,
-                                             HighligtedEntryId, FALSE);
-                        BlpDrawBootMenuEntry(&Handle, MenuEntries[0].EntryName, 0, TRUE);
-
-                        /* Update highlighted entry ID */
                         HighligtedEntryId = 0;
+                        TopVisibleEntry = 0;
+                        RedrawEntries = TRUE;
+                        break;
                     }
                 }
                 else if(Key.ScanCode == 0x0A)
@@ -202,13 +262,10 @@ BlDisplayBootMenu()
                     if(HighligtedEntryId != NumberOfEntries - 1)
                     {
                         /* Highlight last entry */
-                        BlpDrawBootMenuEntry(&Handle, MenuEntries[HighligtedEntryId].EntryName,
-                                             HighligtedEntryId, FALSE);
-                        BlpDrawBootMenuEntry(&Handle, MenuEntries[NumberOfEntries - 1].EntryName,
-                                             NumberOfEntries - 1, TRUE);
-
-                        /* Update highlighted entry ID */
                         HighligtedEntryId = NumberOfEntries - 1;
+                        TopVisibleEntry = (NumberOfEntries > VisibleEntries) ? (NumberOfEntries - VisibleEntries) : 0;
+                        RedrawEntries = TRUE;
+                        break;
                     }
                 }
                 else if(Key.ScanCode == 0x0B)
@@ -231,6 +288,7 @@ BlDisplayBootMenu()
                                                   L"Visit https://exectos.eu.org/ for more information.");
 
                     /* Break from boot menu event loop to redraw whole boot menu */
+                    RedrawBootMenu = TRUE;
                     break;
                 }
                 else if(Key.ScanCode == 0x14)
@@ -238,6 +296,7 @@ BlDisplayBootMenu()
                     /* F10 key pressed, reboot into UEFI setup interface */
                     BlEnterFirmwareSetup();
                     BlDisplayErrorDialog(L"XTLDR", L"Reboot into firmware setup interface not supported!");
+                    RedrawBootMenu = TRUE;
 
                     /* Break from boot menu event loop to redraw whole boot menu */
                     break;
@@ -247,6 +306,7 @@ BlDisplayBootMenu()
                     /* F11 key pressed, reboot the machine */
                     BlRebootSystem();
                     BlDisplayErrorDialog(L"XTLDR", L"Failed to reboot the machine!");
+                    RedrawBootMenu = TRUE;
 
                     /* Break from boot menu event loop to redraw whole boot menu */
                     break;
@@ -256,6 +316,7 @@ BlDisplayBootMenu()
                     /* F12 key pressed, shutdown the machine */
                     BlShutdownSystem();
                     BlDisplayErrorDialog(L"XTLDR", L"Failed to shutdown the machine!");
+                    RedrawBootMenu = TRUE;
 
                     /* Break from boot menu event loop to redraw whole boot menu */
                     break;
@@ -263,7 +324,8 @@ BlDisplayBootMenu()
                 else if(Key.UnicodeChar == 0x65)
                 {
                     /* 'e' key pressed, edit the highlighted entry */
-                    BlDisplayErrorDialog(L"XTLDR", L"Editing boot menu entries is not implemented yet!");
+                    BlDisplayEditMenu(&MenuEntries[HighligtedEntryId]);
+                    RedrawBootMenu = TRUE;
 
                     /* Break from boot menu event loop to redraw whole boot menu */
                     break;
@@ -292,7 +354,7 @@ BlDisplayBootMenu()
                     BlSetConsoleAttributes(Handle.DialogColor | Handle.TextColor);
                     BlClearConsoleLine(Handle.PosY + Handle.Height + 4);
                     BlSetCursorPosition(4, Handle.PosY + Handle.Height + 4);
-                    BlConsolePrint(L"Booting '%S' now...\n", MenuEntries[HighligtedEntryId].EntryName);
+                    BlConsolePrint(L"Booting '%S' now...\n", MenuEntries[HighligtedEntryId].FullName);
 
                     /* Disable the timer just in case booting OS fails */
                     TimeOut = -1;
@@ -304,8 +366,9 @@ BlDisplayBootMenu()
                     {
                         /* Failed to boot OS */
                         BlDebugPrint(L"ERROR: Failed to boot '%S' (Status Code: 0x%zX)\n",
-                                     MenuEntries[HighligtedEntryId].EntryName, Status);
+                                     MenuEntries[HighligtedEntryId].FullName, Status);
                         BlDisplayErrorDialog(L"XTLDR", L"Failed to startup the selected Operating System.");
+                        RedrawBootMenu = TRUE;
                     }
                     break;
                 }
@@ -314,6 +377,254 @@ BlDisplayBootMenu()
     }
 }
 
+/**
+ * Displays a simple TUI-based edit menu.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+VOID
+BlDisplayEditMenu(IN PXTBL_BOOTMENU_ITEM MenuEntry)
+{
+    ULONG HighligtedOptionId, Index, NumberOfOptions, OldHighligtedOptionId, TopVisibleEntry, VisibleEntries;
+    XTBL_DIALOG_HANDLE Handle;
+    BOOLEAN RedrawEditMenu, RedrawEntries;
+    EFI_INPUT_KEY Key;
+    UINT_PTR EventIndex;
+    PWCHAR NewValue, OptionName, OriginalValue, Value, ValueToEdit;
+    CONST PWCHAR *EditableOptions;
+    EFI_STATUS Status;
+
+    /* Draw edit menu */
+    BlpDrawEditMenu(&Handle);
+
+    /* Get the list of user editable options */
+    BlGetEditableOptions(&EditableOptions, &NumberOfOptions);
+
+    /* Calculate how many entries can be visible in the menu box */
+    VisibleEntries = Handle.Height - 2;
+
+    /* Assume the first option is highlighted by default */
+    HighligtedOptionId = 0;
+    OldHighligtedOptionId = 0;
+    TopVisibleEntry = 0;
+
+    /* Set redraw flags to not redraw the menu itself, but fill it with entries */
+    RedrawEditMenu = FALSE;
+    RedrawEntries = TRUE;
+
+    /* Infinite edit menu loop */
+    while(TRUE)
+    {
+        /* Redraw edit menu frame if requested */
+        if(RedrawEditMenu)
+        {
+            BlpDrawEditMenu(&Handle);
+            RedrawEditMenu = FALSE;
+            RedrawEntries = TRUE;
+        }
+
+        /* Sanity check to ensure we do not display more entries than possible */
+        if(VisibleEntries > NumberOfOptions)
+        {
+            VisibleEntries = NumberOfOptions;
+        }
+
+        /* Check if we need to redraw boot menu entries */
+        if(RedrawEntries)
+        {
+            /* Iterate through all menu entries */
+            for(Index = 0; Index < VisibleEntries; Index++)
+            {
+                /* Draw menu entry */
+                BlGetBootOptionValue(MenuEntry->Options, EditableOptions[TopVisibleEntry + Index], &Value);
+                BlpDrawEditMenuEntry(&Handle, EditableOptions[TopVisibleEntry + Index], Value, Index,
+                                     (TopVisibleEntry + Index) == HighligtedOptionId);
+
+                /* Free allocated value string if needed */
+                if(Value != NULL)
+                {
+                    BlFreeMemoryPool(Value);
+                }
+            }
+
+            /* Clear redraw entries flag */
+            RedrawEntries = FALSE;
+        }
+
+        /* Wait for EFI event and read key stroke */
+        BlWaitForEfiEvent(1, &EfiSystemTable->ConIn->WaitForKey, &EventIndex);
+        BlReadKeyStroke(&Key);
+
+        /* Check key press scan code */
+        if(Key.UnicodeChar == 0x0D || Key.UnicodeChar == 0x65)
+        {
+            /* ENTER or 'e' key pressed, edit the highlighted option */
+            OptionName = EditableOptions[HighligtedOptionId];
+            BlGetBootOptionValue(MenuEntry->Options, OptionName, &OriginalValue);
+
+            /* If the original value is NULL, use an empty string for editing */
+            if(OriginalValue == NULL)
+            {
+                ValueToEdit = L"";
+            }
+            else
+            {
+                ValueToEdit = OriginalValue;
+            }
+
+            /* Display input dialog to edit the option value */
+            NewValue = ValueToEdit;
+            BlDisplayInputDialog(OptionName, L"Enter new value:", &NewValue);
+
+            /* Check if the value was changed */
+            if(NewValue != ValueToEdit)
+            {
+                /* Update the boot option with the new value and free the old value */
+                BlSetBootOptionValue(MenuEntry->Options, OptionName, NewValue);
+                BlFreeMemoryPool(NewValue);
+            }
+
+            /* Free the original value if it was allocated */
+            if(OriginalValue != NULL)
+            {
+                BlFreeMemoryPool(OriginalValue);
+            }
+
+            /* Mark the edit menu for redraw */
+            RedrawEditMenu = TRUE;
+        }
+        else if(Key.ScanCode == 0x01)
+        {
+            /* UpArrow key pressed, go to previous entry if possible */
+            if(HighligtedOptionId > 0)
+            {
+                /* Highlight previous entry */
+                OldHighligtedOptionId = HighligtedOptionId;
+                HighligtedOptionId--;
+
+                /* Check if we need to scroll the view */
+                if(HighligtedOptionId < TopVisibleEntry)
+                {
+                    /* Scroll the view */
+                    TopVisibleEntry = HighligtedOptionId;
+                    RedrawEntries = TRUE;
+                    continue;
+                }
+
+                /* Redraw old highlighted entry */
+                BlGetBootOptionValue(MenuEntry->Options, EditableOptions[OldHighligtedOptionId], &Value);
+                BlpDrawEditMenuEntry(&Handle, EditableOptions[OldHighligtedOptionId], Value, OldHighligtedOptionId - TopVisibleEntry, FALSE);
+
+                /* Free allocated value string if needed */
+                if(Value != NULL)
+                {
+                    BlFreeMemoryPool(Value);
+                }
+
+                /* Redraw new highlighted entry */
+                BlGetBootOptionValue(MenuEntry->Options, EditableOptions[HighligtedOptionId], &Value);
+                BlpDrawEditMenuEntry(&Handle, EditableOptions[HighligtedOptionId], Value, HighligtedOptionId - TopVisibleEntry, TRUE);
+
+                /* Free allocated value string if needed */
+                if(Value != NULL)
+                {
+                    BlFreeMemoryPool(Value);
+                }
+            }
+        }
+        else if(Key.ScanCode == 0x02)
+        {
+            /* DownArrow key pressed, go to next entry if possible */
+            if(HighligtedOptionId < NumberOfOptions - 1)
+            {
+                /* Highlight next entry */
+                OldHighligtedOptionId = HighligtedOptionId;
+                HighligtedOptionId++;
+
+                /* Check if we need to scroll the view */
+                if(HighligtedOptionId >= TopVisibleEntry + VisibleEntries)
+                {
+                    /* Scroll the view */
+                    TopVisibleEntry = HighligtedOptionId - VisibleEntries + 1;
+                    RedrawEntries = TRUE;
+                    continue;
+                }
+
+                /* Redraw old highlighted entry */
+                BlGetBootOptionValue(MenuEntry->Options, EditableOptions[OldHighligtedOptionId], &Value);
+                BlpDrawEditMenuEntry(&Handle, EditableOptions[OldHighligtedOptionId], Value, OldHighligtedOptionId - TopVisibleEntry, FALSE);
+
+                /* Free allocated value string if needed */
+                if(Value != NULL)
+                {
+                    BlFreeMemoryPool(Value);
+                }
+
+                /* Redraw new highlighted entry */
+                BlGetBootOptionValue(MenuEntry->Options, EditableOptions[HighligtedOptionId], &Value);
+                BlpDrawEditMenuEntry(&Handle, EditableOptions[HighligtedOptionId], Value, HighligtedOptionId - TopVisibleEntry, TRUE);
+
+                /* Free allocated value string if needed */
+                if(Value != NULL)
+                {
+                    BlFreeMemoryPool(Value);
+                }
+            }
+        }
+        else if(Key.ScanCode == 0x09)
+        {
+            /* PageUp key pressed, go to top entry */
+            if(HighligtedOptionId != 0)
+            {
+                /* Highlight first entry */
+                HighligtedOptionId = 0;
+                TopVisibleEntry = 0;
+                RedrawEntries = TRUE;
+            }
+        }
+        else if(Key.ScanCode == 0x0A)
+        {
+            /* PageDown key pressed, go to bottom entry */
+            if(HighligtedOptionId != NumberOfOptions - 1)
+            {
+                /* Highlight last entry */
+                HighligtedOptionId = NumberOfOptions - 1;
+                TopVisibleEntry = (NumberOfOptions > VisibleEntries) ? (NumberOfOptions - VisibleEntries) : 0;
+                RedrawEntries = TRUE;
+            }
+        }
+        else if(Key.UnicodeChar == 0x02)
+        {
+            /* CTRL-B key pressed, boot the OS */
+            BlSetConsoleAttributes(Handle.DialogColor | Handle.TextColor);
+            BlClearConsoleLine(Handle.PosY + Handle.Height + 4);
+            BlSetCursorPosition(4, Handle.PosY + Handle.Height + 4);
+            BlConsolePrint(L"Booting '%S' now...\n", MenuEntry->FullName);
+
+            /* Boot the OS */
+            Status = BlInvokeBootProtocol(MenuEntry->ShortName, MenuEntry->Options);
+            if(Status != STATUS_SUCCESS)
+            {
+                /* Failed to boot OS */
+                BlDebugPrint(L"ERROR: Failed to boot '%S' (Status Code: 0x%zX)\n", MenuEntry->FullName, Status);
+                BlDisplayErrorDialog(L"XTLDR", L"Failed to startup the selected Operating System.");
+                RedrawEditMenu = TRUE;
+            }
+
+            /* Return to the edit menu */
+            continue;
+        }
+        else if(Key.ScanCode == 0x17)
+        {
+            /* ESC key pressed, exit edit menu */
+            break;
+        }
+    }
+}
+
 /**
  * Displays a red error dialog box with the specified caption and message.
  *
@@ -440,12 +751,11 @@ XTCDECL
 VOID
 BlDisplayInputDialog(IN PWCHAR Caption,
                      IN PWCHAR Message,
-                     IN PWCHAR *InputFieldText)
+                     IN OUT PWCHAR *InputFieldText)
 {
     SIZE_T InputFieldLength, TextCursorPosition, TextIndex, TextPosition;
     XTBL_DIALOG_HANDLE Handle;
     PWCHAR InputFieldBuffer;
-    SIZE_T BufferLength;
     EFI_INPUT_KEY Key;
     EFI_STATUS Status;
     UINT_PTR Index;
@@ -470,9 +780,11 @@ BlDisplayInputDialog(IN PWCHAR Caption,
     Key.ScanCode = 0;
     Key.UnicodeChar = 0;
 
-    /* Get initial input text length and allocate a buffer */
-    BufferLength = RtlWideStringLength(*InputFieldText, 0);
-    Status = BlAllocateMemoryPool(BufferLength * sizeof(WCHAR), (PVOID *)&InputFieldBuffer);
+    /* Determine input field length */
+    InputFieldLength = RtlWideStringLength(*InputFieldText, 0);
+
+    /* Allocate a buffer for storing the input field text */
+    Status = BlAllocateMemoryPool(2048 * sizeof(WCHAR), (PVOID *)&InputFieldBuffer);
     if(Status != STATUS_EFI_SUCCESS)
     {
         /* Memory allocation failure, print error message and return */
@@ -482,15 +794,8 @@ BlDisplayInputDialog(IN PWCHAR Caption,
     }
 
     /* Copy input text into edit buffer */
-    RtlCopyMemory(InputFieldBuffer, *InputFieldText, BufferLength * sizeof(WCHAR));
-    InputFieldBuffer[BufferLength] = L'\0';
-
-    /* Determine input field length */
-    InputFieldLength = BufferLength;
-    if(InputFieldLength > Handle.Width - 8)
-    {
-        InputFieldLength = Handle.Width - 8;
-    }
+    RtlCopyMemory(InputFieldBuffer, *InputFieldText, InputFieldLength * sizeof(WCHAR));
+    InputFieldBuffer[InputFieldLength] = L'\0';
 
     /* Start at first character */
     TextPosition = 0;
@@ -502,7 +807,6 @@ BlDisplayInputDialog(IN PWCHAR Caption,
         /* Wait for key press and read key stroke */
         BlWaitForEfiEvent(1, &EfiSystemTable->ConIn->WaitForKey, &Index);
         BlReadKeyStroke(&Key);
-        BlResetConsoleInputBuffer();
 
         /* Check key press scan code */
         if(Key.ScanCode == 0x17)
@@ -553,12 +857,16 @@ BlDisplayInputDialog(IN PWCHAR Caption,
             /* DELETE key pressed, delete character */
             if(Handle.Attributes & XTBL_TUI_DIALOG_ACTIVE_INPUT)
             {
+                /* Check if buffer is not empty */
                 if(InputFieldLength > 0 && TextPosition < InputFieldLength)
                 {
+                    /* Delete character */
                     RtlMoveMemory(InputFieldBuffer + TextPosition, InputFieldBuffer + TextPosition + 1,
                                   (InputFieldLength - TextPosition) * sizeof(WCHAR));
+
+                    /* Decrement length and null terminate string */
                     InputFieldLength--;
-                    InputFieldBuffer[InputFieldLength] = 0;
+                    InputFieldBuffer[InputFieldLength] = L'\0';
                 }
             }
         }
@@ -567,13 +875,17 @@ BlDisplayInputDialog(IN PWCHAR Caption,
             /* BACKSPACE key pressed, delete character */
             if(Handle.Attributes & XTBL_TUI_DIALOG_ACTIVE_INPUT)
             {
+                /* Check if buffer is not empty */
                 if(InputFieldLength > 0 && TextPosition > 0 && TextPosition <= InputFieldLength)
                 {
+                    /* Move memory to overwrite the character to the left of the cursor */
+                    RtlMoveMemory(InputFieldBuffer + TextPosition - 1, InputFieldBuffer + TextPosition,
+                                  (InputFieldLength - TextPosition + 1) * sizeof(WCHAR));
+
+                    /* Decrement length, position and null terminate string */
                     TextPosition--;
-                    RtlMoveMemory(InputFieldBuffer + TextPosition, InputFieldBuffer + TextPosition + 1,
-                                  (InputFieldLength - TextPosition) * sizeof(WCHAR));
                     InputFieldLength--;
-                    InputFieldBuffer[InputFieldLength] = 0;
+                    InputFieldBuffer[InputFieldLength] = L'\0';
                 }
             }
         }
@@ -588,15 +900,23 @@ BlDisplayInputDialog(IN PWCHAR Caption,
             /* Other key pressed, add character to the buffer */
             if(Handle.Attributes & XTBL_TUI_DIALOG_ACTIVE_INPUT && Key.UnicodeChar != 0)
             {
-                RtlMoveMemory(InputFieldBuffer + TextPosition + 1, InputFieldBuffer + TextPosition,
-                              (InputFieldLength - TextPosition) * sizeof(WCHAR));
-                InputFieldBuffer[TextPosition] = Key.UnicodeChar;
-                TextPosition++;
-                InputFieldLength++;
-                InputFieldBuffer[InputFieldLength] = 0;
+                /* Check if buffer is full */
+                if(InputFieldLength < 2047)
+                {
+                    /* Insert character at current position */
+                    RtlMoveMemory(InputFieldBuffer + TextPosition + 1, InputFieldBuffer + TextPosition,
+                                  (InputFieldLength - TextPosition) * sizeof(WCHAR));
+                    InputFieldBuffer[TextPosition] = Key.UnicodeChar;
+
+                    /* Increment length, position and null terminate string */
+                    TextPosition++;
+                    InputFieldLength++;
+                    InputFieldBuffer[InputFieldLength] = L'\0';
+                }
             }
         }
 
+        /* Calculate text index and cursor position */
         if(TextPosition > (Handle.Width - 9))
         {
             TextIndex = TextPosition - (Handle.Width - 9);
@@ -612,6 +932,7 @@ BlDisplayInputDialog(IN PWCHAR Caption,
         BlpDrawDialogButton(&Handle);
         BlpDrawDialogInputField(&Handle, &InputFieldBuffer[TextIndex]);
 
+        /* Set cursor position if input field is active */
         if(Handle.Attributes & XTBL_TUI_DIALOG_ACTIVE_INPUT)
         {
             BlSetCursorPosition(Handle.PosX + 4 + TextCursorPosition, Handle.PosY + Handle.Height - 4);
@@ -870,8 +1191,8 @@ BlpDrawBootMenu(OUT PXTBL_DIALOG_HANDLE Handle)
     BlSetCursorPosition(0, Handle->PosY + Handle->Height);
     BlSetConsoleAttributes(EFI_TEXT_BGCOLOR_BLACK | EFI_TEXT_FGCOLOR_LIGHTGRAY);
     BlConsolePrint(L"    Use cursors to change the selection. Press ENTER key to boot the chosen\n"
-                    "    Operating System, 'e' to edit it before booting or 's' for XTLDR shell.\n"
-                    "    Additional help available after pressing F1 key.");
+                   L"    Operating System, 'e' to edit it before booting or 's' for XTLDR shell.\n"
+                   L"    Additional help available after pressing F1 key.");
 }
 
 /**
@@ -1330,3 +1651,164 @@ BlpDrawDialogProgressBar(IN PXTBL_DIALOG_HANDLE Handle,
     BlDisableConsoleCursor();
     BlConsoleWrite(ProgressBar);
 }
+
+/**
+ * Draws a text-based boot edition menu.
+ *
+ * @param Handle
+ *        Supplies a pointer to the edition menu handle.
+ *
+ * @return This function does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+VOID
+BlpDrawEditMenu(OUT PXTBL_DIALOG_HANDLE Handle)
+{
+    /* Query console screen resolution */
+    BlQueryConsoleMode(&Handle->ResX, &Handle->ResY);
+
+    /* Set boot menu parameters */
+    Handle->Attributes = 0;
+    Handle->DialogColor = EFI_TEXT_BGCOLOR_BLACK;
+    Handle->TextColor = EFI_TEXT_FGCOLOR_LIGHTGRAY;
+    Handle->PosX = 3;
+    Handle->PosY = 3;
+    Handle->Width = Handle->ResX - 6;
+    Handle->Height = Handle->ResY - 10;
+
+    /* Clear screen and disable cursor */
+    BlSetConsoleAttributes(Handle->DialogColor | Handle->TextColor);
+    BlClearConsoleScreen();
+    BlDisableConsoleCursor();
+
+    /* Check if debugging enabled */
+    if(DEBUG)
+    {
+        /* Print debug version of XTLDR banner */
+        BlSetCursorPosition((Handle->ResX - 44) / 2, 1);
+        BlConsolePrint(L"XTLDR Boot Loader v%d.%d (%s-%s)\n",
+                       XTLDR_VERSION_MAJOR, XTLDR_VERSION_MINOR, XTOS_VERSION_DATE, XTOS_VERSION_HASH);
+    }
+    else
+    {
+        /* Print standard XTLDR banner */
+        BlSetCursorPosition((Handle->ResX - 22) / 2, 1);
+        BlConsolePrint(L"XTLDR Boot Loader v%d.%d\n", XTLDR_VERSION_MAJOR, XTLDR_VERSION_MINOR);
+    }
+
+    /* Draw empty dialog box for boot menu */
+    BlpDrawDialogBox(Handle, L"Edit Options", NULL);
+
+    /* Print help message below the edit menu */
+    BlSetCursorPosition(0, Handle->PosY + Handle->Height);
+    BlSetConsoleAttributes(EFI_TEXT_BGCOLOR_BLACK | EFI_TEXT_FGCOLOR_LIGHTGRAY);
+    BlConsolePrint(L"    Use cursors to change the selection. Press ENTER key to edit the chosen\n"
+                   L"    option, ESC to return to the main boot menu or CTRL-B to boot.\n");
+}
+
+/**
+ * Draws edit menu entry at the specified position.
+ *
+ * @param Handle
+ *        Supplies a pointer to the boot menu handle.
+ *
+ * @param OptionName
+ *        Supplies a pointer to the buffer containing a part of the menu entry name (an option name).
+ *
+ * @param OptionValue
+ *        Supplies a pointer to the buffer containing a part of the menu entry name (an option value).
+ *
+ * @param Position
+ *        Specifies entry position on the list in the boot menu.
+ *
+ * @param Highlighted
+ *        Specifies whether this entry should be highlighted or not.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+EFI_STATUS
+BlpDrawEditMenuEntry(IN PXTBL_DIALOG_HANDLE Handle,
+                     IN PWCHAR OptionName,
+                     IN PWCHAR OptionValue,
+                     IN UINT Position,
+                     IN BOOLEAN Highlighted)
+{
+    BOOLEAN Allocation;
+    PWCHAR DisplayValue, ShortValue;
+    UINT Index;
+    ULONG OptionNameLength, OptionValueLength, OptionWidth;
+    EFI_STATUS Status;
+
+    /* Assume no allocation was made */
+    Allocation = FALSE;
+
+    /* Set display value depending on input */
+    DisplayValue = (OptionValue != NULL) ? OptionValue : L"";
+
+    /* Determine lengths */
+    OptionNameLength = RtlWideStringLength(OptionName, 0);
+    OptionValueLength = RtlWideStringLength(DisplayValue, 0);
+    OptionWidth = Handle->Width - 4 - (OptionNameLength + 2);
+
+    /* Check if value needs to be truncated */
+    if(OptionValueLength > OptionWidth)
+    {
+        /* Allocate buffer for new, shortened value */
+        Status = BlAllocateMemoryPool((OptionWidth + 1) * sizeof(WCHAR), (PVOID *)&ShortValue);
+        if(Status != STATUS_EFI_SUCCESS)
+        {
+            /* Memory allocation failure, print debug message and return */
+            BlDebugPrint(L"ERROR: Memory allocation failure (Status Code: 0x%zX)\n", Status);
+            return Status;
+        }
+
+        /* Copy a desired value length into the allocated buffer and append "..." */
+        RtlCopyMemory(ShortValue, DisplayValue, (OptionWidth - 3) * sizeof(WCHAR));
+        RtlCopyMemory(ShortValue + OptionWidth - 3, L"...", 3 * sizeof(WCHAR));
+        ShortValue[OptionWidth] = L'\0';
+
+        /* Mark that allocation was made and set new display value */
+        Allocation = TRUE;
+        DisplayValue = ShortValue;
+    }
+
+    /* Move cursor to the right position */
+    BlSetCursorPosition(5, 4 + Position);
+
+    /* Check whether this entry should be highlighted */
+    if(Highlighted)
+    {
+        /* Highlight this entry */
+        BlSetConsoleAttributes(EFI_TEXT_BGCOLOR_LIGHTGRAY | EFI_TEXT_FGCOLOR_BLACK);
+    }
+    else
+    {
+        /* Set default colors */
+        BlSetConsoleAttributes(EFI_TEXT_BGCOLOR_BLACK | EFI_TEXT_FGCOLOR_LIGHTGRAY);
+    }
+
+    /* Clear menu entry */
+    for(Index = 0; Index < Handle->Width - 4; Index++)
+    {
+        BlConsolePrint(L" ");
+    }
+
+    /* Print menu entry */
+    BlSetCursorPosition(5, 4 + Position);
+    BlConsolePrint(L"%S: %S", OptionName, DisplayValue);
+
+    /* Check if allocation was made */
+    if(Allocation)
+    {
+        /* Free allocated memory */
+        BlFreeMemoryPool(DisplayValue);
+    }
+
+    /* Return success */
+    return STATUS_EFI_SUCCESS;
+}

xtldr/volume.c

@@ -4,6 +4,7 @@
  * FILE:            xtldr/volume.c
  * DESCRIPTION:     XTLDR volume support
  * DEVELOPERS:      Rafal Kupiec <belliash@codingworkshop.eu.org>
+ *                  Aiken Harris <harraiken91@gmail.com>
  */
 
 #include <xtldr.h>
@@ -47,7 +48,11 @@ XTCDECL
 EFI_STATUS
 BlEnumerateBlockDevices()
 {
-    PEFI_DEVICE_PATH_PROTOCOL LastNode = NULL;
+    EFI_GUID LoadedImageProtocolGuid = EFI_LOADED_IMAGE_PROTOCOL_GUID;
+    EFI_GUID BlockIoGuid = EFI_BLOCK_IO_PROTOCOL_GUID;
+    EFI_HANDLE BootDeviceHandle = NULL, DeviceHandle = NULL;
+    EFI_LOADED_IMAGE_PROTOCOL* LoadedImage;
+    PEFI_DEVICE_PATH_PROTOCOL DevicePath = NULL, LastNode = NULL;
     PEFI_BLOCK_DEVICE_DATA ParentNode = NULL;
     PEFI_BLOCK_DEVICE_DATA BlockDeviceData;
     PEFI_BLOCK_DEVICE BlockDevice;
@@ -62,6 +67,18 @@ BlEnumerateBlockDevices()
     USHORT DriveType;
     ULONG CDCount = 0, FDCount = 0, HDCount = 0, RDCount = 0;
 
+    /* Get the device handle of the image that is running */
+    Status = EfiSystemTable->BootServices->HandleProtocol(EfiImageHandle, &LoadedImageProtocolGuid, (VOID**)&LoadedImage);
+    if(Status != STATUS_EFI_SUCCESS)
+    {
+        /* Failed to get boot device handle */
+        BlDebugPrint(L"ERROR: Failed to get boot device handle (Status Code: 0x%zX)\n", Status);
+        return Status;
+    }
+
+    /* Save the boot device handle */
+    BootDeviceHandle = LoadedImage->DeviceHandle;
+
     /* Initialize list entries */
     RtlInitializeListHead(&BlockDevices);
     RtlInitializeListHead(&EfiBlockDevices);
@@ -78,21 +95,42 @@ BlEnumerateBlockDevices()
     ListEntry = BlockDevices.Flink;
     while(ListEntry != &BlockDevices)
     {
-        /* Take block device from the list */
+        /* Get data for the next discovered device. */
         BlockDeviceData = CONTAIN_RECORD(ListEntry, EFI_BLOCK_DEVICE_DATA, ListEntry);
+        PartitionGuid = NULL;
 
         /* Find last node */
         Status = BlpFindLastBlockDeviceNode(BlockDeviceData->DevicePath, &LastNode);
         if(Status != STATUS_EFI_SUCCESS)
         {
+            /* Skip this device if its last node cannot be found, as it is required for classification */
             BlDebugPrint(L"WARNING: Block device last node not found\n");
             ListEntry = ListEntry->Flink;
             continue;
         }
 
-        /* Set drive type to 'unknown' by default */
+        /* Initialize drive type before attempting to classify the device */
         DriveType = XTBL_BOOT_DEVICE_UNKNOWN;
 
+        /* Locate the parent for this block device to ensure it is not an orphaned entry */
+        if(!BlpFindParentBlockDevice(&BlockDevices, BlockDeviceData, &ParentNode))
+        {
+            /* Orphaned device found. Log a warning and skip it as it cannot be properly classified */
+            BlDebugPrint(L"WARNING: No parent device found, skipping orphaned media device path\n");
+            ListEntry = ListEntry->Flink;
+            continue;
+        }
+
+        /* Verify that media information is available, as some devices may not report it */
+        if(!BlockDeviceData->BlockIo->Media)
+        {
+            /* The device is unusable without media info, log a warning and skip it */
+            BlDebugPrint(L"WARNING: Block device is missing media information\n");
+            ListEntry = ListEntry->Flink;
+            continue;
+        }
+        Media = BlockDeviceData->BlockIo->Media;
+
         /* Check last node type */
         if(LastNode->Type == EFI_ACPI_DEVICE_PATH && LastNode->SubType == EFI_ACPI_DP)
         {
@@ -101,65 +139,73 @@ BlEnumerateBlockDevices()
             if(AcpiDevice->HID == 0x60441D0 || AcpiDevice->HID == 0x70041D0 || AcpiDevice->HID == 0x70141D1)
             {
                 /* Floppy drive found */
-                Media = BlockDeviceData->BlockIo->Media;
                 DriveType = XTBL_BOOT_DEVICE_FLOPPY;
                 DriveNumber = FDCount++;
                 PartitionNumber = 0;
 
                 /* Print debug message */
                 BlDebugPrint(L"Found Floppy Disk (DiskNumber: %lu, MediaPresent: %u, RO: %u)\n",
-                           DriveNumber, Media->MediaPresent, Media->ReadOnly);
+                             DriveNumber, Media->MediaPresent, Media->ReadOnly);
             }
         }
-        else if(LastNode->Type == EFI_MEDIA_DEVICE_PATH)
+        else if((LastNode->Type == EFI_MEDIA_DEVICE_PATH && LastNode->SubType == EFI_MEDIA_CDROM_DP) ||
+                (LastNode->Type == EFI_MESSAGING_DEVICE_PATH &&
+                 (LastNode->SubType == EFI_MESSAGING_ATAPI_DP || LastNode->SubType == EFI_MESSAGING_SATA_DP) &&
+                 Media->MediaPresent && Media->RemovableMedia))
         {
-            /* Media device path found */
-            if(LastNode->SubType == EFI_MEDIA_CDROM_DP)
-            {
-                /* Optical drive found */
-                Media = BlockDeviceData->BlockIo->Media;
-                DriveType = XTBL_BOOT_DEVICE_CDROM;
-                DriveNumber = CDCount++;
-                PartitionNumber = 0;
+            /* Optical drive found */
+            DriveType = XTBL_BOOT_DEVICE_CDROM;
+            DriveNumber = CDCount++;
+            PartitionNumber = 0;
 
-                /* Print debug message */
-                BlDebugPrint(L"Found CD-ROM drive (DriveNumber: %lu, MediaPresent: %u, RemovableMedia: %u, RO: %u)\n",
-                           DriveNumber, Media->MediaPresent, Media->RemovableMedia, Media->ReadOnly);
-            }
-            else if(LastNode->SubType == EFI_MEDIA_HARDDRIVE_DP)
-            {
-                /* Hard disk partition found */
-                Media = BlockDeviceData->BlockIo->Media;
-                HDPath = (PEFI_HARDDRIVE_DEVICE_PATH)LastNode;
-                DriveType = XTBL_BOOT_DEVICE_HARDDISK;
-                DriveNumber = (HDPath->PartitionNumber == 1) ? HDCount++ : HDCount - 1;
-                PartitionNumber = HDPath->PartitionNumber;
-                PartitionGuid = (PEFI_GUID)HDPath->Signature;
+            /* Print debug message */
+            BlDebugPrint(L"Found CD-ROM drive (DriveNumber: %lu, MediaPresent: %u, RemovableMedia: %u, RO: %u)\n",
+                         DriveNumber, Media->MediaPresent, Media->RemovableMedia, Media->ReadOnly);
+        }
+        else if(LastNode->Type == EFI_MEDIA_DEVICE_PATH && LastNode->SubType == EFI_MEDIA_HARDDRIVE_DP)
+        {
+            /* Hard disk partition found */
+            HDPath = (PEFI_HARDDRIVE_DEVICE_PATH)LastNode;
+            DriveType = XTBL_BOOT_DEVICE_HARDDISK;
+            DriveNumber = (HDPath->PartitionNumber == 1) ? HDCount++ : HDCount - 1;
+            PartitionNumber = HDPath->PartitionNumber;
+            PartitionGuid = (PEFI_GUID)HDPath->Signature;
 
-                /* Print debug message */
-                BlDebugPrint(L"Found Hard Disk partition (DiskNumber: %lu, PartNumber: %lu, "
-                           L"MBRType: %u, GUID: {%V}, PartSize: %uB)\n",
-                           DriveNumber, PartitionNumber, HDPath->MBRType,
-                           PartitionGuid, HDPath->PartitionSize * Media->BlockSize);
-            }
-            else if(LastNode->SubType == EFI_MEDIA_RAMDISK_DP)
+            /* Check if this is the EFI System Partition (ESP) */
+            if(BootDeviceHandle != NULL)
             {
-                /* RAM disk found */
-                Media = BlockDeviceData->BlockIo->Media;
-                DriveType = XTBL_BOOT_DEVICE_RAMDISK;
-                DriveNumber = RDCount++;
-                PartitionNumber = 0;
-
-                /* Print debug message */
-                BlDebugPrint(L"Found RAM Disk (DiskNumber: %lu, MediaPresent: %u)\n",
-                           DriveNumber, Media->MediaPresent);
+                /* Allocate memory for device path */
+                DevicePath = BlpDuplicateDevicePath(BlockDeviceData->DevicePath);
+                if(DevicePath != NULL)
+                {
+                    /* Check if this is the boot device */
+                    Status = EfiSystemTable->BootServices->LocateDevicePath(&BlockIoGuid, &DevicePath,
+                                                                            &DeviceHandle);
+                    if(Status == STATUS_EFI_SUCCESS && DeviceHandle == BootDeviceHandle)
+                    {
+                        /* Mark partition as ESP */
+                        DriveType |= XTBL_BOOT_DEVICE_ESP;
+                    }
+                }
             }
 
-            if(!BlpFindParentBlockDevice(&BlockDevices, BlockDeviceData, ParentNode))
-            {
-                BlDebugPrint(L"WARNING: No parent device found, skipping orphaned media device path\n");
-                continue;
-            }
+            /* Print debug message */
+            BlDebugPrint(L"Found Hard Disk partition (DiskNumber: %lu, PartNumber: %lu, "
+                         L"MBRType: %u, GUID: {%V}, PartSize: %lluB) %S\n",
+                         DriveNumber, PartitionNumber, HDPath->MBRType,
+                         PartitionGuid, HDPath->PartitionSize * Media->BlockSize,
+                         (DriveType & XTBL_BOOT_DEVICE_ESP) ? L"(ESP)" : L"");
+        }
+        else if(LastNode->Type == EFI_MEDIA_DEVICE_PATH && LastNode->SubType == EFI_MEDIA_RAMDISK_DP)
+        {
+            /* RAM disk found */
+            DriveType = XTBL_BOOT_DEVICE_RAMDISK;
+            DriveNumber = RDCount++;
+            PartitionNumber = 0;
+
+            /* Print debug message */
+            BlDebugPrint(L"Found RAM Disk (DiskNumber: %lu, MediaPresent: %u)\n",
+                         DriveNumber, Media->MediaPresent);
         }
 
         /* Make sure the device found has valid type set */
@@ -418,12 +464,26 @@ BlGetVolumeDevicePath(IN PWCHAR SystemPath,
     {
         /* Check if this is the volume we are looking for */
         Device = CONTAIN_RECORD(ListEntry, EFI_BLOCK_DEVICE, ListEntry);
-        if((Device->DriveType == DriveType && Device->DriveNumber == DriveNumber &&
-           Device->PartitionNumber == PartNumber))
+        if(DriveType == XTBL_BOOT_DEVICE_ESP)
         {
-            /* Found volume */
-            *DevicePath = Device->DevicePath;
-            break;
+            /* ESP requested, verify if flag is set for this device */
+            if((Device->DriveType & XTBL_BOOT_DEVICE_ESP) != 0)
+            {
+                /* Found volume */
+                *DevicePath = Device->DevicePath;
+                break;
+            }
+        }
+        else
+        {
+            if(((Device->DriveType & DriveType) == DriveType) &&
+               (Device->DriveNumber == DriveNumber) &&
+               (Device->PartitionNumber == PartNumber))
+            {
+                /* Found volume */
+                *DevicePath = Device->DevicePath;
+                break;
+            }
         }
         ListEntry = ListEntry->Flink;
     }
@@ -431,9 +491,9 @@ BlGetVolumeDevicePath(IN PWCHAR SystemPath,
     /* Check if volume was found */
     if(*DevicePath == NULL)
     {
-        /* Failed to find volume */
+        /* Volume not found */
         BlDebugPrint(L"ERROR: Volume (DriveType: %u, DriveNumber: %lu, PartNumber: %lu) not found\n",
-                   DriveType, DriveNumber, PartNumber);
+                     DriveType, DriveNumber, PartNumber);
         return STATUS_EFI_NOT_FOUND;
     }
 
@@ -606,7 +666,7 @@ BlReadFile(IN PEFI_FILE_HANDLE DirHandle,
     Pages = EFI_SIZE_TO_PAGES(FileInfo->FileSize);
 
     /* Allocate pages */
-    Status = BlAllocateMemoryPages(Pages, &Address);
+    Status = BlAllocateMemoryPages(AllocateAnyPages, Pages, &Address);
     if(Status != STATUS_EFI_SUCCESS)
     {
         /* Pages allocation failure */
@@ -776,6 +836,12 @@ BlpDissectVolumeArcPath(IN PWCHAR SystemPath,
         ArcLength = 10;
         *DriveType = XTBL_BOOT_DEVICE_RAMDISK;
     }
+    else if(RtlCompareWideStringInsensitive(SystemPath, L"multi(0)esp(0)", 0) == 0)
+    {
+        /* This is ESP */
+        ArcLength = 14;
+        *DriveType = XTBL_BOOT_DEVICE_ESP;
+    }
     else if(RtlCompareWideStringInsensitive(SystemPath, L"multi(0)disk(0)", 0) == 0)
     {
         /* This is a multi-disk port */
@@ -917,17 +983,25 @@ BlpDuplicateDevicePath(IN PEFI_DEVICE_PATH_PROTOCOL DevicePath)
     EFI_STATUS Status;
     UINT Length = 0;
 
+    /* Check if the input device path is NULL */
+    if(!DevicePath)
+    {
+        /* Nothing to duplicate */
+        return NULL;
+    }
+
+    /* Start iterating from the beginning of the device path */
     DevicePathNode = DevicePath;
 
     /* Get the device path length */
     while(TRUE)
     {
-        Length += *(PUSHORT)DevicePath->Length;
+        Length += *(PUSHORT)DevicePathNode->Length;
         if(DevicePathNode->Type == EFI_END_DEVICE_PATH)
         {
             break;
         }
-        DevicePathNode = (PEFI_DEVICE_PATH_PROTOCOL)((PUCHAR)DevicePathNode + *(PUSHORT)DevicePath->Length);
+        DevicePathNode = (PEFI_DEVICE_PATH_PROTOCOL)((PUCHAR)DevicePathNode + *(PUSHORT)DevicePathNode->Length);
     }
 
     /* Check length */
@@ -1016,7 +1090,7 @@ XTCDECL
 BOOLEAN
 BlpFindParentBlockDevice(IN PLIST_ENTRY BlockDevices,
                          IN PEFI_BLOCK_DEVICE_DATA ChildNode,
-                         OUT PEFI_BLOCK_DEVICE_DATA ParentNode)
+                         OUT PEFI_BLOCK_DEVICE_DATA *ParentNode)
 {
     PEFI_DEVICE_PATH_PROTOCOL ChildDevicePath, ParentDevicePath;
     PEFI_BLOCK_DEVICE_DATA BlockDeviceData;
@@ -1029,6 +1103,14 @@ BlpFindParentBlockDevice(IN PLIST_ENTRY BlockDevices,
         /* Take block device from the list */
         BlockDeviceData = CONTAIN_RECORD(ListEntry, EFI_BLOCK_DEVICE_DATA, ListEntry);
 
+        /* A device cannot be its own parent */
+        if (BlockDeviceData == ChildNode)
+        {
+            /* Move to the next device */
+            ListEntry = ListEntry->Flink;
+            continue;
+        }
+
         ChildDevicePath = ChildNode->DevicePath;
         ParentDevicePath = BlockDeviceData->DevicePath;
 
@@ -1039,7 +1121,7 @@ BlpFindParentBlockDevice(IN PLIST_ENTRY BlockDevices,
             if(ParentDevicePath->Type == EFI_END_DEVICE_PATH)
             {
                 /* Parent device is a match */
-                ParentNode = BlockDeviceData;
+                *ParentNode = BlockDeviceData;
                 return TRUE;
             }
 
@@ -1047,10 +1129,11 @@ BlpFindParentBlockDevice(IN PLIST_ENTRY BlockDevices,
             ChildLength = *(PUSHORT)ChildDevicePath->Length;
             ParentLength = *(PUSHORT)ParentDevicePath->Length;
 
-            /* Check if lengths match */
-            if(ChildLength != ParentLength)
+            /* Check if nodes match */
+            if((ChildLength != ParentLength) ||
+               (RtlCompareMemory(ChildDevicePath, ParentDevicePath, ParentLength) != ParentLength))
             {
-                /* Lengths do not match, this is not a valid parent */
+                /* Nodes do not match, this is not a valid parent */
                 break;
             }
 

xtldr/xtldr.c

@@ -90,16 +90,17 @@ BlInitializeBootLoader()
  */
 XTCDECL
 EFI_STATUS
-BlInitializeBootMenuList(OUT PXTBL_BOOTMENU_ITEM *MenuEntries,
+BlInitializeBootMenuList(IN ULONG MaxNameLength,
+                         OUT PXTBL_BOOTMENU_ITEM *MenuEntries,
                          OUT PULONG EntriesCount,
                          OUT PULONG DefaultId)
 {
     EFI_GUID VendorGuid = XT_BOOT_LOADER_PROTOCOL_GUID;
-    PWCHAR DefaultMenuEntry, LastBooted, MenuEntryName;
+    PWCHAR DefaultMenuEntry, LastBooted, MenuEntryName, VisibleName;
     PLIST_ENTRY MenuEntrySectionList, MenuEntryList;
     PXTBL_CONFIG_SECTION MenuEntrySection;
     PXTBL_CONFIG_ENTRY MenuEntryOption;
-    ULONG DefaultOS, NumberOfEntries;
+    ULONG DefaultOS, NameLength,NumberOfEntries;
     PXTBL_BOOTMENU_ITEM OsList;
     EFI_STATUS Status;
 
@@ -108,7 +109,7 @@ BlInitializeBootMenuList(OUT PXTBL_BOOTMENU_ITEM *MenuEntries,
     NumberOfEntries = 0;
 
     /* Get default menu entry from configuration */
-    DefaultMenuEntry = BlGetConfigValue(L"DEFAULT");
+    BlGetConfigValue(L"DEFAULT", &DefaultMenuEntry);
 
     /* Check if configuration allows to use last booted OS */
     if(BlGetConfigBooleanValue(L"KEEPLASTBOOT"))
@@ -151,7 +152,8 @@ BlInitializeBootMenuList(OUT PXTBL_BOOTMENU_ITEM *MenuEntries,
         MenuEntrySection = CONTAIN_RECORD(MenuEntrySectionList, XTBL_CONFIG_SECTION, Flink);
 
         /* Check if this is the default menu entry */
-        if(RtlCompareWideStringInsensitive(MenuEntrySection->SectionName, DefaultMenuEntry, 0) == 0)
+        if((RtlWideStringLength(MenuEntrySection->SectionName, 0) == RtlWideStringLength(DefaultMenuEntry, 0)) &&
+           (RtlCompareWideStringInsensitive(MenuEntrySection->SectionName, DefaultMenuEntry, 0) == 0))
         {
             /* Set default OS ID */
             DefaultOS = NumberOfEntries;
@@ -176,10 +178,36 @@ BlInitializeBootMenuList(OUT PXTBL_BOOTMENU_ITEM *MenuEntries,
         }
 
         /* Add OS to the boot menu list */
-        OsList[NumberOfEntries].EntryName = MenuEntryName;
+        OsList[NumberOfEntries].FullName = MenuEntryName;
         OsList[NumberOfEntries].ShortName = MenuEntrySection->SectionName;
         OsList[NumberOfEntries].Options = &MenuEntrySection->Options;
 
+        /* Check if the menu entry name fits the maximum length */
+        NameLength = RtlWideStringLength(MenuEntryName, 0);
+        if(NameLength > MaxNameLength)
+        {
+            /* Menu entry name is too long, allocate memory for shorter name visible in the boot menu */
+            Status = BlAllocateMemoryPool((MaxNameLength + 1) * sizeof(WCHAR), (PVOID*)&VisibleName);
+            if(Status != STATUS_EFI_SUCCESS)
+            {
+                /* Memory allocation failure */
+                return STATUS_EFI_OUT_OF_RESOURCES;
+            }
+
+            /* Copy shorter name and append "..." at the end */
+            RtlCopyMemory(VisibleName, MenuEntryName, (MaxNameLength - 3) * sizeof(WCHAR));
+            RtlCopyMemory(VisibleName + MaxNameLength - 3, L"...", 3 * sizeof(WCHAR));
+            VisibleName[MaxNameLength] = L'\0';
+
+            /* Set visible menu entry name */
+            OsList[NumberOfEntries].EntryName = VisibleName;
+        }
+        else
+        {
+            /* Menu entry name fits the maximum length, use it as is */
+            OsList[NumberOfEntries].EntryName = MenuEntryName;
+        }
+
         /* Get next menu entry */
         MenuEntrySectionList = MenuEntrySectionList->Flink;
         NumberOfEntries++;
@@ -363,6 +391,7 @@ EFI_STATUS
 BlStartXtLoader(IN EFI_HANDLE ImageHandle,
                 IN PEFI_SYSTEM_TABLE SystemTable)
 {
+    PWCHAR Modules;
     EFI_STATUS Status;
 
     /* Set the system table and image handle */
@@ -427,8 +456,9 @@ BlStartXtLoader(IN EFI_HANDLE ImageHandle,
         return Status;
     }
 
-    /* Load boot loader modules */
-    Status = BlLoadModules(BlGetConfigValue(L"MODULES"));
+    /* Load all necessary modules */
+    BlGetConfigValue(L"MODULES", &Modules);
+    Status = BlLoadModules(Modules);
     if(Status != STATUS_EFI_SUCCESS)
     {
         /* Failed to load modules */

xtoskrnl/CMakeLists.txt

@@ -1,4 +1,5 @@
-# XT Kernel
+# XT Kernel and library
+PROJECT(LIBXTOS)
 PROJECT(XTOSKRNL)
 
 # Specify include directories
@@ -6,42 +7,69 @@ include_directories(
     ${EXECTOS_SOURCE_DIR}/sdk/xtdk
     ${XTOSKRNL_SOURCE_DIR}/includes)
 
-# Specify list of source code files
+# Specify list of library source code files
+list(APPEND LIBXTOS_SOURCE
+    ${XTOSKRNL_SOURCE_DIR}/ar/${ARCH}/boot.S
+    ${XTOSKRNL_SOURCE_DIR}/ar/${ARCH}/cpufunc.c
+    ${XTOSKRNL_SOURCE_DIR}/hl/cport.c
+    ${XTOSKRNL_SOURCE_DIR}/hl/${ARCH}/ioport.c
+    ${XTOSKRNL_SOURCE_DIR}/rtl/globals.c
+    ${XTOSKRNL_SOURCE_DIR}/rtl/guid.c
+    ${XTOSKRNL_SOURCE_DIR}/rtl/math.c
+    ${XTOSKRNL_SOURCE_DIR}/rtl/memory.c
+    ${XTOSKRNL_SOURCE_DIR}/rtl/plist.c
+    ${XTOSKRNL_SOURCE_DIR}/rtl/string.c
+    ${XTOSKRNL_SOURCE_DIR}/rtl/widestr.c)
+
+# Specify list of kernel source code files
 list(APPEND XTOSKRNL_SOURCE
     ${XTOSKRNL_SOURCE_DIR}/ar/${ARCH}/archsup.S
+    ${XTOSKRNL_SOURCE_DIR}/ar/${ARCH}/boot.S
     ${XTOSKRNL_SOURCE_DIR}/ar/${ARCH}/cpufunc.c
     ${XTOSKRNL_SOURCE_DIR}/ar/${ARCH}/globals.c
     ${XTOSKRNL_SOURCE_DIR}/ar/${ARCH}/procsup.c
     ${XTOSKRNL_SOURCE_DIR}/ar/${ARCH}/traps.c
     ${XTOSKRNL_SOURCE_DIR}/ex/rundown.c
+    ${XTOSKRNL_SOURCE_DIR}/hl/acpi.c
     ${XTOSKRNL_SOURCE_DIR}/hl/cport.c
-    ${XTOSKRNL_SOURCE_DIR}/hl/efifb.c
+    ${XTOSKRNL_SOURCE_DIR}/hl/fbdev.c
     ${XTOSKRNL_SOURCE_DIR}/hl/globals.c
+    ${XTOSKRNL_SOURCE_DIR}/hl/init.c
     ${XTOSKRNL_SOURCE_DIR}/hl/${ARCH}/cpu.c
     ${XTOSKRNL_SOURCE_DIR}/hl/${ARCH}/pic.c
     ${XTOSKRNL_SOURCE_DIR}/hl/${ARCH}/ioport.c
     ${XTOSKRNL_SOURCE_DIR}/hl/${ARCH}/runlevel.c
+    ${XTOSKRNL_SOURCE_DIR}/kd/dbginit.c
+    ${XTOSKRNL_SOURCE_DIR}/kd/dbgio.c
+    ${XTOSKRNL_SOURCE_DIR}/kd/globals.c
     ${XTOSKRNL_SOURCE_DIR}/ke/apc.c
     ${XTOSKRNL_SOURCE_DIR}/ke/dpc.c
     ${XTOSKRNL_SOURCE_DIR}/ke/event.c
     ${XTOSKRNL_SOURCE_DIR}/ke/globals.c
+    ${XTOSKRNL_SOURCE_DIR}/ke/info.c
     ${XTOSKRNL_SOURCE_DIR}/ke/kprocess.c
     ${XTOSKRNL_SOURCE_DIR}/ke/krnlinit.c
     ${XTOSKRNL_SOURCE_DIR}/ke/kthread.c
+    ${XTOSKRNL_SOURCE_DIR}/ke/kubsan.c
     ${XTOSKRNL_SOURCE_DIR}/ke/panic.c
     ${XTOSKRNL_SOURCE_DIR}/ke/runlevel.c
     ${XTOSKRNL_SOURCE_DIR}/ke/semphore.c
     ${XTOSKRNL_SOURCE_DIR}/ke/spinlock.c
+    ${XTOSKRNL_SOURCE_DIR}/ke/sysres.c
     ${XTOSKRNL_SOURCE_DIR}/ke/timer.c
     ${XTOSKRNL_SOURCE_DIR}/ke/${ARCH}/irqs.c
     ${XTOSKRNL_SOURCE_DIR}/ke/${ARCH}/krnlinit.c
     ${XTOSKRNL_SOURCE_DIR}/ke/${ARCH}/kthread.c
     ${XTOSKRNL_SOURCE_DIR}/ke/${ARCH}/proc.c
     ${XTOSKRNL_SOURCE_DIR}/mm/globals.c
+    ${XTOSKRNL_SOURCE_DIR}/mm/hlpool.c
     ${XTOSKRNL_SOURCE_DIR}/mm/init.c
     ${XTOSKRNL_SOURCE_DIR}/mm/kpools.c
+    ${XTOSKRNL_SOURCE_DIR}/mm/pages.c
+    ${XTOSKRNL_SOURCE_DIR}/mm/${ARCH}/globals.c
     ${XTOSKRNL_SOURCE_DIR}/mm/${ARCH}/init.c
     ${XTOSKRNL_SOURCE_DIR}/mm/${ARCH}/pages.c
+    ${XTOSKRNL_SOURCE_DIR}/mm/${ARCH}/pmap.c
     ${XTOSKRNL_SOURCE_DIR}/po/idle.c
     ${XTOSKRNL_SOURCE_DIR}/rtl/atomic.c
     ${XTOSKRNL_SOURCE_DIR}/rtl/bitmap.c
@@ -61,7 +89,7 @@ list(APPEND XTOSKRNL_SOURCE
 set_specfile(xtoskrnl.spec xtoskrnl.exe)
 
 # Link static XTOS library
-add_library(libxtos ${XTOSKRNL_SOURCE})
+add_library(libxtos ${LIBXTOS_SOURCE})
 
 # Link kernel executable
 add_executable(xtoskrnl

xtoskrnl/README.md

@@ -1,58 +1,68 @@
 ## XTOSKRNL
-XTOSKRNL is an XT system kernel executable, providing the kernel and executive layers for XTOS kernel space. It is
-a fundamental part of ExectOS, responsible for various core services like hardware abstraction or memory management.
-This kernel, contains the scheduler (called sometimes as Dispatcher), cache, object and memory managers, security
-manager and other kernel executives described below.
+XTOSKRNL is the core kernel executable for ExectOS, providing the fundamental kernel and executive layers that operate
+within the XTOS kernel space. It is responsible for various core services, such as hardware abstraction, memory
+management, and process scheduling. The kernel contains the scheduler (sometimes referred to as the Dispatcher), the
+cache, object, and memory managers, the security manager, and other executive components described below.
 
-All routines in kernel are prefixed to indicate the kernel subsystem they belong and put inside separate directory.
-This is a list of them:
+All routines in the kernel are prefixed to indicate the subsystem they belong to, and their source code is organized
+into corresponding directories. These subsystems include:
 
- * Ar - Architecture library
+ * Ar - Architecture-specific Library
  * Ex - Kernel Executive
- * Hl - Hardware Abstraction Layer (HAL)
- * Ke - Core kernel library
- * Mm - Memory manager
+ * Hl - Hardware Layer
+ * Kd - Kernel Debugger
+ * Ke - Core Kernel Library
+ * Mm - Memory Manager
  * Po - Plug&Play and Power Manager
  * Rtl - Runtime library
 
 ### AR: Architecture Library
-This module contains processor architecture specific functions. This includes enabling and disabling interrupts at
-the processor, getting the address of a page fault, getting CPUID information, and performing very early processor
-initialization. This module does not contain any manufacturer or board-specific code, only CPU architecture specific
-code.
+This module contains functions specific to the processor architecture. These include routines for enabling and disabling
+interrupts, retrieving the faulting address on a page fault, querying CPUID information, and performing very early
+processor initialization. This module contains only CPU architecture-specific code, with no manufacturer or
+board-specific implementations.
 
 ### EX: Kernel Executive
-The kernel executive supplies heap management, including support for allocating system memory from paged/non-paged
-pools, as well as synchronization primitives like push locks and fast mutexes, interlocked memory access, and worker
-threads.
+The Kernel Executive provides services for allocating system memory from paged and non-paged pools. It also supplies
+synchronization primitives such as pushlocks and fast mutexes, routines for interlocked memory access, and support for
+worker threads.
 
-### HL: Hardware Abstraction Layer
-Hardware Abstraction Layer (HAL), is a layer between the physical hardware of the computer and the rest of the operating
-system. It was designed to hide differences in hardware and therefore it provides a consistent platform on which
-the system and applications may run.
+### HL: Hardware Layer
+The Hardware Layer is an abstraction layer between the physical hardware and the rest of the operating system. It is
+designed to abstract away hardware differences, providing a consistent platform on which the kernel and applications
+can run.
+
+### KD: Kernel Debugger
+The Kernel Debugger (KD) subsystem provides debugging support for the kernel. The KD is initialized early in the boot
+process to facilitate debugging from the very beginning of the kernel's execution.
 
 ### KE: Kernel Library
-The kernel implements its core functionality that everything else in the system depends upon. This includes basic
-low-level operations such as routing hardware interrupts.
+The Core Kernel Library implements the core functionality upon which the rest of the system depends. This includes
+fundamental low-level operations, such as routing hardware interrupts and managing dispatcher objects.
 
 ### MM: Memory Manager
-Memory Manager is one of core subsystems. It manages virtual memory, controls memory protection and the paging of memory
-in and out of physical memory to secondary storage. It also implements a general-purpose allocator of physical memory.
+The Memory Manager is one of the core subsystems. It manages virtual memory, controls memory protection, and
+handles paging memory between physical RAM and secondary storage. It also implements a general-purpose allocator for
+physical memory.
 
 ### PO: Plug&Play and Power Manager
-This subsystem deals with power events, such as power-off, stand-by, or hibernate. It also handles Plug&Play and
-supports device detection and installation at boot time. It is responsible for starting and stopping devices on demand.
+This subsystem handles power management events, such as shutdown or standby. It also manages Plug and Play (PnP),
+supporting device detection and installation at boot time. Furthermore, it is responsible for starting and stopping
+devices on demand.
 
 ### RTL: Runtime Library
-This is a required static copy of C runtime objects. It includes many utility functions that can be used by native
-applications.
+The Runtime Library provides a kernel-mode implementation of common C library functions. It includes many utility
+routines, for use by other kernel components.
 
-## Functions Naming Convention
-When naming a kernel functions, certain conventions are used. The function name is usually structured with
-<Prefix><Operation><Object>. The prefix denotes the module to which it belongs, thus module
-can be identified simply by the name of the function. Additionally, the prefix identifies the function visibility
-as well. Thus all private functions, that should not be used from outside of the module has added "p" suffix to
-the prefix. For example, KepInitializeStack() routine:
- * Kep - prefix meaning this is private routine belonging to Kernel library (Ke) module,
- * Initialize - operation this function does with the object,
- * Stack - the object is stack.
+## Function Naming Convention
+All kernel functions adhere to a strict naming convention to enhance code readability and maintainability. The structure
+of a function name is generally composed of three parts: <Prefix><Operation><Object>
+
+The prefix identifies the component to which the function belongs. Additionally, the prefix indicates the function's
+visibility. Private functions, which should not be called from outside their own module, have a 'p' appended to their
+prefix.
+
+For example, consider the **KepInitializeStack()** routine:
+ * **Kep** - The prefix indicates a private (p) routine belonging to the Core Kernel Library (Ke).
+ * **Initialize** - The operation performed by the function.
+ * **Stack** - The object on which the operation is performed.

xtoskrnl/ar/amd64/archsup.S

@@ -13,7 +13,7 @@
 
 
 /**
- * This macro creates a trap handler for the specified vector.
+ * Creates a trap handler for the specified vector.
  *
  * @param Vector
  *        Supplies a trap vector number.

xtoskrnl/ar/amd64/boot.S

@@ -0,0 +1,133 @@
+/**
+ * PROJECT:         ExectOS
+ * COPYRIGHT:       See COPYING.md in the top level directory
+ * FILE:            xtoskrnl/ar/amd64/boot.S
+ * DESCRIPTION:     AMD64-specific boot code for setting up the low-level CPU environment
+ * DEVELOPERS:      Aiken Harris <harraiken91@gmail.com>
+ */
+
+#include <amd64/asmsup.h>
+
+.altmacro
+.text
+
+
+/**
+ * Enables eXtended Physical Addressing (XPA).
+ *
+ * @param PageMap
+ *        Supplies a pointer to the page map to be used.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+.global ArEnableExtendedPhysicalAddressing
+ArEnableExtendedPhysicalAddressing:
+    /* Save the original CR4 register */
+    movq %cr4, %rax
+
+    /* Save the state of stack pointer and non-volatile registers */
+    movq %rsp, XpaRegisterSaveArea(%rip)
+    movq %rbp, XpaRegisterSaveArea+0x08(%rip)
+    movq %rax, XpaRegisterSaveArea+0x10(%rip)
+    movq %rbx, XpaRegisterSaveArea+0x18(%rip)
+
+    /* Save the original CR0 register */
+    movq %cr0, %rbp
+
+    /* Load temporary GDT required for mode transitions */
+    leaq XpaTemporaryGdtDesc(%rip), %rax
+    movq %rax, XpaTemporaryGdtBase(%rip)
+    lgdtq XpaTemporaryGdtSize(%rip)
+
+    /* Load addresses for entering compatibility mode and re-entering long mode */
+    leaq XpaEnterCompatMode(%rip), %rax
+    leaq XpaEnterLongMode(%rip), %rbx
+
+    /* Push the 32-bit code segment selector and the target address for a far jump */
+    pushq $GDT_R0_CMCODE
+    pushq %rax
+
+    /* Perform a far return to switch to 32-bit compatibility mode */
+    lretq
+
+XpaEnterCompatMode:
+    /* Enter 32-bit compatibility mode */
+    .code32
+
+    /* Store the PageMap pointer on the stack for future use */
+    pushl %ecx
+
+    /* Set the stack segment to the 32-bit data segment selector */
+    movl $GDT_R0_DATA, %eax
+    movl %eax, %ss
+
+    /* Disable PGE and PCIDE to ensure all TLB entries will be flushed */
+    movl %cr4, %eax
+    andl $~(CR4_PGE | CR4_PCIDE), %eax
+    movl %eax, %cr4
+
+    /* Temporarily disable paging */
+    movl %ebp, %eax
+    andl $~CR0_PG, %eax
+    movl %eax, %cr0
+
+    /* Disable Long Mode as prerequisite for enabling 5-level paging */
+    movl $X86_MSR_EFER, %ecx
+    rdmsr
+    andl $~X86_MSR_EFER_LME, %eax
+    wrmsr
+
+    /* Transition to 5-level paging (PML5/LA57) */
+    movl %cr4, %eax
+    orl $CR4_LA57, %eax
+    movl %eax, %cr4
+
+    /* Restore the PageMap pointer from the stack and load it into CR3 */
+    popl %ecx
+    movl %ecx, %cr3
+
+    /* Re-enable Long Mode */
+    movl $X86_MSR_EFER, %ecx
+    rdmsr
+    orl $X86_MSR_EFER_LME, %eax
+    wrmsr
+
+    /* Restore CR0 with paging enabled and flush the instruction pipeline */
+    movl %ebp, %cr0
+    call XpaFlushInstructions
+
+XpaFlushInstructions:
+    /* Push the 64-bit code segment selector and the target address for a far jump */
+    pushl $GDT_R0_CODE
+    pushl %ebx
+
+    /* Perform a far return to switch to 64-bit long mode */
+    lretl
+
+XpaEnterLongMode:
+    /* Enter 64-bit long mode */
+    .code64
+
+    /* Restore the stack pointer and non-volatile registers */
+    movq XpaRegisterSaveArea(%rip), %rsp
+    movq XpaRegisterSaveArea+8(%rip), %rbp
+    movq XpaRegisterSaveArea+0x10(%rip), %rax
+    movq XpaRegisterSaveArea+0x18(%rip), %rbx
+
+    /* Restore the original CR4 register with LA57 bit set */
+    orq $CR4_LA57, %rax
+    movq %rax, %cr4
+
+    /* Return to the caller */
+    retq
+
+/* Data section for saving registers and temporary GDT */
+XpaRegisterSaveArea: .quad 0x0000000000000000, 0x0000000000000000, 0x0000000000000000, 0x0000000000000000
+XpaTemporaryGdtSize: .short ArEnableExtendedPhysicalAddressingEnd - XpaTemporaryGdtDesc - 1
+XpaTemporaryGdtBase: .quad 0x0000000000000000
+XpaTemporaryGdtDesc: .quad 0x0000000000000000, 0x00CF9A000000FFFF, 0x00AF9A000000FFFF, 0x00CF92000000FFFF
+
+.global ArEnableExtendedPhysicalAddressingEnd
+ArEnableExtendedPhysicalAddressingEnd:

xtoskrnl/ar/amd64/cpufunc.c

@@ -20,7 +20,7 @@ XTCDECL
 VOID
 ArClearInterruptFlag(VOID)
 {
-    asm volatile("cli");
+    __asm__ volatile("cli");
 }
 
 /**
@@ -40,12 +40,12 @@ ArCpuId(IN OUT PCPUID_REGISTERS Registers)
     UINT32 MaxLeaf;
 
     /* Get highest function ID available */
-    asm volatile("cpuid"
-                 : "=a" (MaxLeaf)
-                 : "a" (Registers->Leaf & 0x80000000)
-                 : "rbx",
-                   "rcx",
-                   "rdx");
+    __asm__ volatile("cpuid"
+                     : "=a" (MaxLeaf)
+                     : "a" (Registers->Leaf & 0x80000000)
+                     : "rbx",
+                       "rcx",
+                       "rdx");
 
     /* Check if CPU supports this command */
     if(Registers->Leaf > MaxLeaf)
@@ -55,13 +55,13 @@ ArCpuId(IN OUT PCPUID_REGISTERS Registers)
     }
 
     /* Execute CPUID function */
-    asm volatile("cpuid"
-                 : "=a" (Registers->Eax),
-                   "=b" (Registers->Ebx),
-                   "=c" (Registers->Ecx),
-                   "=d" (Registers->Edx)
-                 : "a" (Registers->Leaf),
-                   "c" (Registers->SubLeaf));
+    __asm__ volatile("cpuid"
+                     : "=a" (Registers->Eax),
+                       "=b" (Registers->Ebx),
+                       "=c" (Registers->Ecx),
+                       "=d" (Registers->Edx)
+                     : "a" (Registers->Leaf),
+                       "c" (Registers->SubLeaf));
 
     /* Return TRUE */
     return TRUE;
@@ -82,6 +82,30 @@ ArFlushTlb(VOID)
     ArWriteControlRegister(3, ArReadControlRegister(3));
 }
 
+/**
+ * Gets the RFLAGS register.
+ *
+ * @return This routine returns the RFLAGS register.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+ULONG
+ArGetCpuFlags(VOID)
+{
+    ULONG_PTR Flags;
+
+    /* Get RFLAGS register */
+    __asm__ volatile("pushf\n"
+                     "pop %0\n"
+                     : "=rm" (Flags)
+                     :
+                     : "memory");
+
+    /* Return flags */
+    return Flags;
+}
+
 /**
  * Gets the address of the current stack register.
  *
@@ -95,11 +119,11 @@ ULONG_PTR
 ArGetStackPointer(VOID)
 {
     /* Get current stack pointer */
-    asm volatile("movq %%rsp, %%rax\n"
-                 "retq\n"
-                  :
-                  :
-                  :);
+    __asm__ volatile("movq %%rsp, %%rax\n"
+                     "retq\n"
+                     :
+                     :
+                     :);
 }
 
 /**
@@ -113,7 +137,27 @@ XTCDECL
 VOID
 ArHalt(VOID)
 {
-    asm volatile("hlt");
+    __asm__ volatile("hlt");
+}
+
+/**
+ * Checks whether interrupts are enabled or not.
+ *
+ * @return This routine returns TRUE if interrupts are enabled, or FALSE otherwise.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+BOOLEAN
+ArInterruptsEnabled(VOID)
+{
+    ULONG_PTR Flags;
+
+    /* Get RFLAGS register */
+    Flags = ArGetCpuFlags();
+
+    /* Check if interrupts are enabled and return result */
+    return (Flags & X86_EFLAGS_IF_MASK) ? TRUE : FALSE;
 }
 
 /**
@@ -130,10 +174,10 @@ XTCDECL
 VOID
 ArInvalidateTlbEntry(IN PVOID Address)
 {
-    asm volatile("invlpg (%0)"
-                 :
-                 : "b" (Address)
-                 : "memory");
+    __asm__ volatile("invlpg (%0)"
+                     :
+                     : "b" (Address)
+                     : "memory");
 }
 
 /**
@@ -150,10 +194,10 @@ XTCDECL
 VOID
 ArLoadGlobalDescriptorTable(IN PVOID Source)
 {
-    asm volatile("lgdt %0"
-                 :
-                 : "m" (*(PSHORT)Source)
-                 : "memory");
+    __asm__ volatile("lgdt %0"
+                     :
+                     : "m" (*(PSHORT)Source)
+                     : "memory");
 }
 
 /**
@@ -170,10 +214,10 @@ XTCDECL
 VOID
 ArLoadInterruptDescriptorTable(IN PVOID Source)
 {
-    asm volatile("lidt %0"
-                 :
-                 : "m" (*(PSHORT)Source)
-                 : "memory");
+    __asm__ volatile("lidt %0"
+                     :
+                     : "m" (*(PSHORT)Source)
+                     : "memory");
 }
 
 /**
@@ -190,9 +234,9 @@ XTCDECL
 VOID
 ArLoadLocalDescriptorTable(IN USHORT Source)
 {
-    asm volatile("lldtw %0"
-                 :
-                 : "g" (Source));
+    __asm__ volatile("lldtw %0"
+                     :
+                     : "g" (Source));
 }
 
 /**
@@ -209,9 +253,9 @@ XTCDECL
 VOID
 ArLoadMxcsrRegister(IN ULONG Source)
 {
-    asm volatile("ldmxcsr %0"
-                 :
-                 : "m" (Source));
+    __asm__ volatile("ldmxcsr %0"
+                     :
+                     : "m" (Source));
 }
 
 /**
@@ -236,45 +280,45 @@ ArLoadSegment(IN USHORT Segment,
     {
         case SEGMENT_CS:
             /* Load CS Segment */
-            asm volatile("mov %0, %%rax\n"
-                         "push %%rax\n"
-                         "lea label(%%rip), %%rax\n"
-                         "push %%rax\n"
-                         "lretq\n"
-                         "label:"
-                         :
-                         : "ri" ((ULONGLONG)Source)
-                         : "rax");
+            __asm__ volatile("mov %0, %%rax\n"
+                             "push %%rax\n"
+                             "lea label(%%rip), %%rax\n"
+                             "push %%rax\n"
+                             "lretq\n"
+                             "label:"
+                             :
+                             : "ri" ((ULONGLONG)Source)
+                             : "rax");
             break;
         case SEGMENT_DS:
             /* Load DS Segment */
-            asm volatile("movl %0, %%ds"
-                         :
-                         : "r" (Source));
+            __asm__ volatile("movl %0, %%ds"
+                             :
+                             : "r" (Source));
             break;
         case SEGMENT_ES:
             /* Load ES Segment */
-            asm volatile("movl %0, %%es"
-                         :
-                         : "r" (Source));
+            __asm__ volatile("movl %0, %%es"
+                             :
+                             : "r" (Source));
             break;
         case SEGMENT_FS:
             /* Load FS Segment */
-            asm volatile("movl %0, %%fs"
-                         :
-                         : "r" (Source));
+            __asm__ volatile("movl %0, %%fs"
+                             :
+                             : "r" (Source));
             break;
         case SEGMENT_GS:
             /* Load GS Segment */
-            asm volatile("movl %0, %%gs"
-                         :
-                         : "r" (Source));
+            __asm__ volatile("movl %0, %%gs"
+                             :
+                             : "r" (Source));
             break;
             /* Load SS Segment */
         case SEGMENT_SS:
-            asm volatile("movl %0, %%ss"
-                         :
-                         : "r" (Source));
+            __asm__ volatile("movl %0, %%ss"
+                             :
+                             : "r" (Source));
             break;
     }
 }
@@ -293,9 +337,9 @@ XTCDECL
 VOID
 ArLoadTaskRegister(USHORT Source)
 {
-    asm volatile("ltr %0"
-                 :
-                 : "rm" (Source));
+    __asm__ volatile("ltr %0"
+                     :
+                     : "rm" (Source));
 }
 
 /**
@@ -310,9 +354,9 @@ VOID
 ArMemoryBarrier(VOID)
 {
     LONG Barrier;
-    asm volatile("lock; orl $0, %0;"
-                 :
-                 : "m"(Barrier));
+    __asm__ volatile("lock; orl $0, %0;"
+                     :
+                     : "m"(Barrier));
 }
 
 /**
@@ -336,38 +380,38 @@ ArReadControlRegister(IN USHORT ControlRegister)
     {
         case 0:
             /* Read value from CR0 */
-            asm volatile("mov %%cr0, %0"
-                         : "=r" (Value)
-                         :
-                         : "memory");
+            __asm__ volatile("mov %%cr0, %0"
+                             : "=r" (Value)
+                             :
+                             : "memory");
             break;
         case 2:
             /* Read value from CR2 */
-            asm volatile("mov %%cr2, %0"
-                         : "=r" (Value)
-                         :
-                         : "memory");
+            __asm__ volatile("mov %%cr2, %0"
+                             : "=r" (Value)
+                             :
+                             : "memory");
             break;
         case 3:
             /* Read value from CR3 */
-            asm volatile("mov %%cr3, %0"
-                         : "=r" (Value)
-                         :
-                         : "memory");
+            __asm__ volatile("mov %%cr3, %0"
+                             : "=r" (Value)
+                             :
+                             : "memory");
             break;
         case 4:
             /* Read value from CR4 */
-            asm volatile("mov %%cr4, %0"
-                         : "=r" (Value)
-                         :
-                         : "memory");
+            __asm__ volatile("mov %%cr4, %0"
+                             : "=r" (Value)
+                             :
+                             : "memory");
             break;
         case 8:
             /* Read value from CR8 */
-            asm volatile("mov %%cr8, %0"
-                         : "=r" (Value)
-                         :
-                         : "memory");
+            __asm__ volatile("mov %%cr8, %0"
+                             : "=r" (Value)
+                             :
+                             : "memory");
             break;
         default:
             /* Invalid control register set */
@@ -400,43 +444,43 @@ ArReadDebugRegister(IN USHORT DebugRegister)
     {
         case 0:
             /* Read value from DR0 */
-            asm volatile("mov %%dr0, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr0, %0"
+                             : "=r" (Value));
             break;
         case 1:
             /* Read value from DR1 */
-            asm volatile("mov %%dr1, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr1, %0"
+                             : "=r" (Value));
             break;
         case 2:
             /* Read value from DR2 */
-            asm volatile("mov %%dr2, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr2, %0"
+                             : "=r" (Value));
             break;
         case 3:
             /* Read value from DR3 */
-            asm volatile("mov %%dr3, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr3, %0"
+                             : "=r" (Value));
             break;
         case 4:
             /* Read value from DR4 */
-            asm volatile("mov %%dr4, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr4, %0"
+                             : "=r" (Value));
             break;
         case 5:
             /* Read value from DR5 */
-            asm volatile("mov %%dr5, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr5, %0"
+                             : "=r" (Value));
             break;
         case 6:
             /* Read value from DR6 */
-            asm volatile("mov %%dr6, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr6, %0"
+                             : "=r" (Value));
             break;
         case 7:
             /* Read value from DR7 */
-            asm volatile("mov %%dr7, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr7, %0"
+                             : "=r" (Value));
             break;
         default:
             /* Invalid debug register set */
@@ -465,9 +509,9 @@ ArReadGSQuadWord(ULONG Offset)
     ULONGLONG Value;
 
     /* Read quadword from GS segment */
-    asm volatile("movq %%gs:%a[Offset], %q[Value]"
-                 : [Value] "=r" (Value)
-                 : [Offset] "ir" (Offset));
+    __asm__ volatile("movq %%gs:%a[Offset], %q[Value]"
+                     : [Value] "=r" (Value)
+                     : [Offset] "ir" (Offset));
     return Value;
 }
 
@@ -487,10 +531,10 @@ ArReadModelSpecificRegister(IN ULONG Register)
 {
     ULONG Low, High;
 
-    asm volatile("rdmsr"
-                 : "=a" (Low),
-                   "=d" (High)
-                 : "c" (Register));
+    __asm__ volatile("rdmsr"
+                     : "=a" (Low),
+                       "=d" (High)
+                     : "c" (Register));
 
     return ((ULONGLONG)High << 32) | Low;
 }
@@ -522,9 +566,9 @@ ArReadTimeStampCounter(VOID)
 {
     ULONGLONG Low, High;
 
-    asm volatile("rdtsc"
-                 : "=a" (Low),
-                   "=d" (High));
+    __asm__ volatile("rdtsc"
+                     : "=a" (Low),
+                       "=d" (High));
 
     return ((ULONGLONG)High << 32) | Low;
 }
@@ -540,10 +584,10 @@ XTCDECL
 VOID
 ArReadWriteBarrier(VOID)
 {
-    asm volatile(""
-                 :
-                 :
-                 : "memory");
+    __asm__ volatile(""
+                     :
+                     :
+                     : "memory");
 }
 
 /**
@@ -557,7 +601,7 @@ XTCDECL
 VOID
 ArSetInterruptFlag(VOID)
 {
-    asm volatile("sti");
+    __asm__ volatile("sti");
 }
 
 /**
@@ -574,10 +618,10 @@ XTCDECL
 VOID
 ArStoreGlobalDescriptorTable(OUT PVOID Destination)
 {
-    asm volatile("sgdt %0"
-                 : "=m" (*(PSHORT)Destination)
-                 :
-                 : "memory");
+    __asm__ volatile("sgdt %0"
+                     : "=m" (*(PSHORT)Destination)
+                     :
+                     : "memory");
 }
 
 /**
@@ -594,10 +638,10 @@ XTCDECL
 VOID
 ArStoreInterruptDescriptorTable(OUT PVOID Destination)
 {
-    asm volatile("sidt %0"
-                 : "=m" (*(PSHORT)Destination)
-                 :
-                 : "memory");
+    __asm__ volatile("sidt %0"
+                     : "=m" (*(PSHORT)Destination)
+                     :
+                     : "memory");
 }
 
 /**
@@ -614,10 +658,10 @@ XTCDECL
 VOID
 ArStoreLocalDescriptorTable(OUT PVOID Destination)
 {
-    asm volatile("sldt %0"
-                 : "=m" (*(PSHORT)Destination)
-                 :
-                 : "memory");
+    __asm__ volatile("sldt %0"
+                     : "=m" (*(PSHORT)Destination)
+                     :
+                     : "memory");
 }
 
 /**
@@ -641,28 +685,28 @@ ArStoreSegment(IN USHORT Segment,
     switch(Segment)
     {
         case SEGMENT_CS:
-            asm volatile("movl %%cs, %0"
-                         : "=r" (*(PUINT)Destination));
+            __asm__ volatile("movl %%cs, %0"
+                             : "=r" (*(PUINT)Destination));
             break;
         case SEGMENT_DS:
-            asm volatile("movl %%ds, %0"
-                         : "=r" (*(PUINT)Destination));
+            __asm__ volatile("movl %%ds, %0"
+                             : "=r" (*(PUINT)Destination));
             break;
         case SEGMENT_ES:
-            asm volatile("movl %%es, %0"
-                         : "=r" (*(PUINT)Destination));
+            __asm__ volatile("movl %%es, %0"
+                             : "=r" (*(PUINT)Destination));
             break;
         case SEGMENT_FS:
-            asm volatile("movl %%fs, %0"
-                         : "=r" (*(PUINT)Destination));
+            __asm__ volatile("movl %%fs, %0"
+                             : "=r" (*(PUINT)Destination));
             break;
         case SEGMENT_GS:
-            asm volatile("movl %%gs, %0"
-                         : "=r" (*(PUINT)Destination));
+            __asm__ volatile("movl %%gs, %0"
+                             : "=r" (*(PUINT)Destination));
             break;
         case SEGMENT_SS:
-            asm volatile("movl %%ss, %0"
-                         : "=r" (*(PUINT)Destination));
+            __asm__ volatile("movl %%ss, %0"
+                             : "=r" (*(PUINT)Destination));
             break;
         default:
             Destination = NULL;
@@ -684,10 +728,10 @@ XTCDECL
 VOID
 ArStoreTaskRegister(OUT PVOID Destination)
 {
-    asm volatile("str %0"
-                 : "=m" (*(PULONG)Destination)
-                 :
-                 : "memory");
+    __asm__ volatile("str %0"
+                     : "=m" (*(PULONG)Destination)
+                     :
+                     : "memory");
 }
 
 /**
@@ -713,38 +757,38 @@ ArWriteControlRegister(IN USHORT ControlRegister,
     {
         case 0:
             /* Write value to CR0 */
-            asm volatile("mov %0, %%cr0"
-                         :
-                         : "r"(Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%cr0"
+                             :
+                             : "r"(Value)
+                             : "memory");
             break;
         case 2:
             /* Write value to CR2 */
-            asm volatile("mov %0, %%cr2"
-                         :
-                         : "r"(Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%cr2"
+                             :
+                             : "r"(Value)
+                             : "memory");
             break;
         case 3:
             /* Write value to CR3 */
-            asm volatile("mov %0, %%cr3"
-                         :
-                         : "r"(Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%cr3"
+                             :
+                             : "r"(Value)
+                             : "memory");
             break;
         case 4:
             /* Write value to CR4 */
-            asm volatile("mov %0, %%cr4"
-                         :
-                         : "r"(Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%cr4"
+                             :
+                             : "r"(Value)
+                             : "memory");
             break;
         case 8:
             /* Write value to CR8 */
-            asm volatile("mov %0, %%cr8"
-                         :
-                         : "r"(Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%cr8"
+                             :
+                             : "r"(Value)
+                             : "memory");
             break;
     }
 }
@@ -772,52 +816,52 @@ ArWriteDebugRegister(IN USHORT DebugRegister,
     {
         case 0:
             /* Write value to DR0 */
-            asm volatile("mov %0, %%dr0"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr0"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 1:
             /* Write value to DR1 */
-            asm volatile("mov %0, %%dr1"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr1"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 2:
             /* Write value to DR2 */
-            asm volatile("mov %0, %%dr2"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr2"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 3:
             /* Write value to DR3 */
-            asm volatile("mov %0, %%dr3"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr3"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 4:
             /* Write value to DR4 */
-            asm volatile("mov %0, %%dr4"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr4"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 5:
             /* Write value to DR5 */
-            asm volatile("mov %0, %%dr5"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr5"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 6:
             /* Write value to DR6 */
-            asm volatile("mov %0, %%dr6"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr6"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 7:
             /* Write value to DR7 */
-            asm volatile("mov %0, %%dr7"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr7"
+                             :
+                             : "r" (Value)
+                             : "memory");
     }
 }
 
@@ -835,10 +879,10 @@ XTCDECL
 VOID
 ArWriteEflagsRegister(IN UINT_PTR Value)
 {
-    asm volatile("push %0\n"
-                 "popf"
-                 :
-                 : "rim" (Value));
+    __asm__ volatile("push %0\n"
+                     "popf"
+                     :
+                     : "rim" (Value));
 }
 
 /**
@@ -862,11 +906,11 @@ ArWriteModelSpecificRegister(IN ULONG Register,
     ULONG Low = Value & 0xFFFFFFFF;
     ULONG High = Value >> 32;
 
-    asm volatile("wrmsr"
-                 :
-                 : "c" (Register),
-                   "a" (Low),
-                   "d" (High));
+    __asm__ volatile("wrmsr"
+                     :
+                     : "c" (Register),
+                       "a" (Low),
+                       "d" (High));
 }
 
 /**
@@ -880,8 +924,8 @@ XTCDECL
 VOID
 ArYieldProcessor(VOID)
 {
-    asm volatile("pause"
-                 :
-                 :
-                 : "memory");
+    __asm__ volatile("pause"
+                     :
+                     :
+                     : "memory");
 }

xtoskrnl/ar/amd64/procsup.c

@@ -18,31 +18,43 @@
  */
 XTAPI
 VOID
-ArInitializeProcessor(VOID)
+ArInitializeProcessor(IN PVOID ProcessorStructures)
 {
     KDESCRIPTOR GdtDescriptor, IdtDescriptor;
+    PVOID KernelBootStack, KernelFaultStack;
     PKPROCESSOR_BLOCK ProcessorBlock;
-    PVOID KernelFaultStack;
     PKGDTENTRY Gdt;
     PKIDTENTRY Idt;
     PKTSS Tss;
 
-    /* Use initial structures */
-    Gdt = ArInitialGdt;
-    Idt = ArInitialIdt;
-    Tss = &ArInitialTss;
-    KernelFaultStack = &ArKernelFaultStack;
+    /* Check if processor structures buffer provided */
+    if(ProcessorStructures)
+    {
+        /* Assign CPU structures from provided buffer */
+        ArpInitializeProcessorStructures(ProcessorStructures, &Gdt, &Tss, &ProcessorBlock,
+                                         &KernelBootStack, &KernelFaultStack);
 
-    /* Load processor block */
-    ProcessorBlock = CONTAIN_RECORD(&ArInitialProcessorBlock.Prcb, KPROCESSOR_BLOCK, Prcb);
+        /* Use global IDT */
+        Idt = ArInitialIdt;
+    }
+    else
+    {
+        /* Use initial structures */
+        Gdt = ArInitialGdt;
+        Idt = ArInitialIdt;
+        Tss = &ArInitialTss;
+        KernelBootStack = &ArKernelBootStack;
+        KernelFaultStack = &ArKernelFaultStack;
+        ProcessorBlock = &ArInitialProcessorBlock;
+    }
 
     /* Initialize processor block */
     ArpInitializeProcessorBlock(ProcessorBlock, Gdt, Idt, Tss, KernelFaultStack);
 
     /* Initialize GDT, IDT and TSS */
     ArpInitializeGdt(ProcessorBlock);
-    ArpInitializeTss(ProcessorBlock);
     ArpInitializeIdt(ProcessorBlock);
+    ArpInitializeTss(ProcessorBlock, KernelBootStack, KernelFaultStack);
 
     /* Set GDT and IDT descriptors */
     GdtDescriptor.Base = Gdt;
@@ -135,14 +147,14 @@ ArpIdentifyProcessor(VOID)
 
     /* Store CPU vendor in processor control block */
     Prcb->CpuId.Vendor = CpuRegisters.Ebx;
-    Prcb->CpuId.VendorName[0] = CpuRegisters.Ebx;
-    Prcb->CpuId.VendorName[4] = CpuRegisters.Edx;
-    Prcb->CpuId.VendorName[8] = CpuRegisters.Ecx;
+    *(PULONG)&Prcb->CpuId.VendorName[0] = CpuRegisters.Ebx;
+    *(PULONG)&Prcb->CpuId.VendorName[4] = CpuRegisters.Edx;
+    *(PULONG)&Prcb->CpuId.VendorName[8] = CpuRegisters.Ecx;
     Prcb->CpuId.VendorName[12] = '\0';
 
-    /* Get CPU features */
+    /* Get CPU standard features */
     RtlZeroMemory(&CpuRegisters, sizeof(CPUID_REGISTERS));
-    CpuRegisters.Leaf = CPUID_GET_CPU_FEATURES;
+    CpuRegisters.Leaf = CPUID_GET_STANDARD1_FEATURES;
     ArCpuId(&CpuRegisters);
 
     /* Store CPU signature in processor control block */
@@ -155,9 +167,9 @@ ArpIdentifyProcessor(VOID)
     if(Prcb->CpuId.Vendor == CPU_VENDOR_AMD)
     {
         /* AMD CPU */
-        Prcb->CpuId.Family = Prcb->CpuId.Family + CpuSignature.ExtendedFamily;
-        if(Prcb->CpuId.Model == 0xF)
+        if(Prcb->CpuId.Family >= 0xF)
         {
+            Prcb->CpuId.Family = Prcb->CpuId.Family + CpuSignature.ExtendedFamily;
             Prcb->CpuId.Model = Prcb->CpuId.Model + (CpuSignature.ExtendedModel << 4);
         }
     }
@@ -169,7 +181,7 @@ ArpIdentifyProcessor(VOID)
             Prcb->CpuId.Family = Prcb->CpuId.Family + CpuSignature.ExtendedFamily;
         }
 
-        if((Prcb->CpuId.Family == 0xF) || (Prcb->CpuId.Family == 0x6))
+        if((Prcb->CpuId.Family == 0x6) || (Prcb->CpuId.Family == 0xF))
         {
             Prcb->CpuId.Model = Prcb->CpuId.Model + (CpuSignature.ExtendedModel << 4);
         }
@@ -204,7 +216,7 @@ ArpInitializeGdt(IN PKPROCESSOR_BLOCK ProcessorBlock)
     ArpSetGdtEntry(ProcessorBlock->GdtBase, KGDT_R3_CMCODE, 0x0, 0xFFFFFFFF, KGDT_TYPE_CODE, KGDT_DPL_USER, 2);
     ArpSetGdtEntry(ProcessorBlock->GdtBase, KGDT_R3_DATA, 0x0, 0xFFFFFFFF, KGDT_TYPE_DATA, KGDT_DPL_USER, 2);
     ArpSetGdtEntry(ProcessorBlock->GdtBase, KGDT_R3_CODE, 0x0, 0x0, KGDT_TYPE_CODE, KGDT_DPL_USER, 1);
-    ArpSetGdtEntry(ProcessorBlock->GdtBase, KGDT_SYS_TSS, (ULONG_PTR)ProcessorBlock->TssBase, sizeof(KTSS), AMD64_TSS, KGDT_DPL_SYSTEM, 0);
+    ArpSetGdtEntry(ProcessorBlock->GdtBase, KGDT_SYS_TSS, (ULONG_PTR)ProcessorBlock->TssBase, sizeof(KTSS) - 1, AMD64_TSS, KGDT_DPL_SYSTEM, 0);
     ArpSetGdtEntry(ProcessorBlock->GdtBase, KGDT_R3_CMTEB, 0x0, 0x0FFF, KGDT_TYPE_DATA, KGDT_DPL_USER, 2);
     ArpSetGdtEntry(ProcessorBlock->GdtBase, KGDT_R0_LDT, 0x0, 0xFFFFFFFF, KGDT_TYPE_CODE, KGDT_DPL_SYSTEM, 0);
     ArpSetGdtEntry(ProcessorBlock->GdtBase, KGDT_ALIAS, (ULONG_PTR)ProcessorBlock->GdtBase, (GDT_ENTRIES * sizeof(KGDTENTRY)) - 1, KGDT_TYPE_DATA, KGDT_DPL_SYSTEM, 0);
@@ -287,9 +299,6 @@ ArpInitializeProcessorBlock(OUT PKPROCESSOR_BLOCK ProcessorBlock,
                             IN PKTSS Tss,
                             IN PVOID DpcStack)
 {
-    /* Fill processor block with zeroes */
-    RtlZeroMemory(ProcessorBlock, sizeof(KPROCESSOR_BLOCK));
-
     /* Set processor block and processor control block */
     ProcessorBlock->Self = ProcessorBlock;
     ProcessorBlock->CurrentPrcb = &ProcessorBlock->Prcb;
@@ -304,9 +313,9 @@ ArpInitializeProcessorBlock(OUT PKPROCESSOR_BLOCK ProcessorBlock,
     ProcessorBlock->Prcb.DpcStack = DpcStack;
 
     /* Setup processor control block */
-    ProcessorBlock->Prcb.Number = 0;
-    ProcessorBlock->Prcb.SetMember = 1ULL;
-    ProcessorBlock->Prcb.MultiThreadProcessorSet = 1ULL;
+    ProcessorBlock->Prcb.CpuNumber = ProcessorBlock->CpuNumber;
+    ProcessorBlock->Prcb.SetMember = 1ULL << ProcessorBlock->CpuNumber;
+    ProcessorBlock->Prcb.MultiThreadProcessorSet = 1ULL << ProcessorBlock->CpuNumber;
 
     /* Clear DR6 and DR7 registers */
     ProcessorBlock->Prcb.ProcessorState.SpecialRegisters.KernelDr6 = 0;
@@ -386,6 +395,64 @@ ArpInitializeProcessorRegisters(VOID)
     ArLoadMxcsrRegister(INITIAL_MXCSR);
 }
 
+/**
+ * Initializes i686 processor specific structures with provided memory buffer.
+ *
+ * @param ProcessorStructures
+ *        Supplies a pointer to the allocated buffer with processor structures.
+ *
+ * @param Gdt
+ *        Supplies a pointer to the GDT.
+ *
+ * @param Tss
+ *        Supplies a pointer to the TSS.
+ *
+ * @param ProcessorBlock
+ *        Supplies a pointer to the processor block.
+ *
+ * @param KernelBootStack
+ *        Supplies a pointer to the kernel boot stack.
+ *
+ * @param KernelFaultStack
+ *        Supplies a pointer to the kernel fault stack.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+VOID
+ArpInitializeProcessorStructures(IN PVOID ProcessorStructures,
+                                 OUT PKGDTENTRY *Gdt,
+                                 OUT PKTSS *Tss,
+                                 OUT PKPROCESSOR_BLOCK *ProcessorBlock,
+                                 OUT PVOID *KernelBootStack,
+                                 OUT PVOID *KernelFaultStack)
+{
+    UINT_PTR Address;
+
+    /* Align address to page size boundary and move to kernel boot stack */
+    Address = ROUND_UP((UINT_PTR)ProcessorStructures, MM_PAGE_SIZE) + KERNEL_STACK_SIZE;
+
+    /* Assign a space for kernel boot stack and advance */
+    *KernelBootStack = (PVOID)Address;
+    Address += KERNEL_STACK_SIZE;
+
+    /* Assign a space for kernel fault stack, no advance needed as stack grows down */
+    *KernelFaultStack = (PVOID)Address;
+
+    /* Assign a space for GDT and advance */
+    *Gdt = (PVOID)Address;
+    Address += sizeof(ArInitialGdt);
+
+    /* Assign a space for Processor Block and advance */
+    *ProcessorBlock = (PVOID)Address;
+    Address += sizeof(ArInitialProcessorBlock);
+
+    /* Assign a space for TSS */
+    *Tss = (PVOID)Address;
+}
+
 /**
  * Initializes segment registers.
  *
@@ -421,16 +488,18 @@ ArpInitializeSegments(VOID)
  */
 XTAPI
 VOID
-ArpInitializeTss(IN PKPROCESSOR_BLOCK ProcessorBlock)
+ArpInitializeTss(IN PKPROCESSOR_BLOCK ProcessorBlock,
+                 IN PVOID KernelBootStack,
+                 IN PVOID KernelFaultStack)
 {
     /* Fill TSS with zeroes */
     RtlZeroMemory(ProcessorBlock->TssBase, sizeof(KTSS));
 
     /* Setup I/O map and stacks for ring0 & traps */
     ProcessorBlock->TssBase->IoMapBase = sizeof(KTSS);
-    ProcessorBlock->TssBase->Rsp0 = (ULONG_PTR)&ArKernelBootStack;
-    ProcessorBlock->TssBase->Ist[KIDT_IST_PANIC] = (ULONG_PTR)&ArKernelFaultStack;
-    ProcessorBlock->TssBase->Ist[KIDT_IST_MCA] = (ULONG_PTR)&ArKernelFaultStack;
+    ProcessorBlock->TssBase->Rsp0 = (ULONG_PTR)KernelBootStack;
+    ProcessorBlock->TssBase->Ist[KIDT_IST_PANIC] = (ULONG_PTR)KernelFaultStack;
+    ProcessorBlock->TssBase->Ist[KIDT_IST_MCA] = (ULONG_PTR)KernelFaultStack;
 }
 
 /**

xtoskrnl/ar/i686/boot.S

@@ -0,0 +1,14 @@
+/**
+ * PROJECT:         ExectOS
+ * COPYRIGHT:       See COPYING.md in the top level directory
+ * FILE:            xtoskrnl/ar/i686/boot.S
+ * DESCRIPTION:     i686-specific boot code for setting up the low-level CPU environment
+ * DEVELOPERS:      Aiken Harris <harraiken91@gmail.com>
+ */
+
+#include <amd64/asmsup.h>
+
+.altmacro
+.text
+
+

xtoskrnl/ar/i686/cpufunc.c

@@ -20,7 +20,7 @@ XTCDECL
 VOID
 ArClearInterruptFlag(VOID)
 {
-    asm volatile("cli");
+    __asm__ volatile("cli");
 }
 
 /**
@@ -40,12 +40,12 @@ ArCpuId(IN OUT PCPUID_REGISTERS Registers)
     UINT32 MaxLeaf;
 
     /* Get highest function ID available */
-    asm volatile("cpuid"
-                 : "=a" (MaxLeaf)
-                 : "a" (Registers->Leaf & 0x80000000)
-                 : "rbx",
-                   "rcx",
-                   "rdx");
+    __asm__ volatile("cpuid"
+                     : "=a" (MaxLeaf)
+                     : "a" (Registers->Leaf & 0x80000000)
+                     : "rbx",
+                       "rcx",
+                       "rdx");
 
     /* Check if CPU supports this command */
     if(Registers->Leaf > MaxLeaf)
@@ -55,13 +55,13 @@ ArCpuId(IN OUT PCPUID_REGISTERS Registers)
     }
 
     /* Execute CPUID function */
-    asm volatile("cpuid"
-                 : "=a" (Registers->Eax),
-                   "=b" (Registers->Ebx),
-                   "=c" (Registers->Ecx),
-                   "=d" (Registers->Edx)
-                 : "a" (Registers->Leaf),
-                   "c" (Registers->SubLeaf));
+    __asm__ volatile("cpuid"
+                     : "=a" (Registers->Eax),
+                       "=b" (Registers->Ebx),
+                       "=c" (Registers->Ecx),
+                       "=d" (Registers->Edx)
+                     : "a" (Registers->Leaf),
+                       "c" (Registers->SubLeaf));
 
     /* Return TRUE */
     return TRUE;
@@ -82,6 +82,30 @@ ArFlushTlb(VOID)
     ArWriteControlRegister(3, ArReadControlRegister(3));
 }
 
+/**
+ * Gets the EFLAGS register.
+ *
+ * @return This routine returns the EFLAGS register.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+ULONG
+ArGetCpuFlags(VOID)
+{
+    ULONG_PTR Flags;
+
+    /* Get EFLAGS register */
+    __asm__ volatile("pushf\n"
+                     "pop %0\n"
+                     : "=rm" (Flags)
+                     :
+                     : "memory");
+
+    /* Return flags */
+    return Flags;
+}
+
 /**
  * Gets the address of the current stack register.
  *
@@ -95,11 +119,11 @@ ULONG_PTR
 ArGetStackPointer(VOID)
 {
     /* Get current stack pointer */
-    asm volatile("mov %%esp, %%eax\n"
-                 "ret\n"
-                  :
-                  :
-                  :);
+    __asm__ volatile("mov %%esp, %%eax\n"
+                     "ret\n"
+                      :
+                      :
+                      :);
 }
 
 /**
@@ -113,7 +137,27 @@ XTCDECL
 VOID
 ArHalt(VOID)
 {
-    asm volatile("hlt");
+    __asm__ volatile("hlt");
+}
+
+/**
+ * Checks whether interrupts are enabled or not.
+ *
+ * @return This routine returns TRUE if interrupts are enabled, or FALSE otherwise.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+BOOLEAN
+ArInterruptsEnabled(VOID)
+{
+    ULONG_PTR Flags;
+
+    /* Get RFLAGS register */
+    Flags = ArGetCpuFlags();
+
+    /* Check if interrupts are enabled and return result */
+    return (Flags & X86_EFLAGS_IF_MASK) ? TRUE : FALSE;
 }
 
 /**
@@ -130,10 +174,10 @@ XTCDECL
 VOID
 ArInvalidateTlbEntry(PVOID Address)
 {
-    asm volatile("invlpg (%0)"
-                 :
-                 : "b" (Address)
-                 : "memory");
+    __asm__ volatile("invlpg (%0)"
+                     :
+                     : "b" (Address)
+                     : "memory");
 }
 
 /**
@@ -150,10 +194,10 @@ XTCDECL
 VOID
 ArLoadGlobalDescriptorTable(IN PVOID Source)
 {
-    asm volatile("lgdt %0"
-                 :
-                 : "m" (*(PSHORT)Source)
-                 : "memory");
+    __asm__ volatile("lgdt %0"
+                     :
+                     : "m" (*(PSHORT)Source)
+                     : "memory");
 }
 
 /**
@@ -170,10 +214,10 @@ XTCDECL
 VOID
 ArLoadInterruptDescriptorTable(IN PVOID Source)
 {
-    asm volatile("lidt %0"
-                 :
-                 : "m" (*(PSHORT)Source)
-                 : "memory");
+    __asm__ volatile("lidt %0"
+                     :
+                     : "m" (*(PSHORT)Source)
+                     : "memory");
 }
 
 /**
@@ -190,9 +234,9 @@ XTCDECL
 VOID
 ArLoadLocalDescriptorTable(IN USHORT Source)
 {
-    asm volatile("lldtw %0"
-                 :
-                 : "g" (Source));
+    __asm__ volatile("lldtw %0"
+                     :
+                     : "g" (Source));
 }
 
 /**
@@ -217,45 +261,45 @@ ArLoadSegment(IN USHORT Segment,
     {
         case SEGMENT_CS:
             /* Load CS Segment */
-            asm volatile("mov %0, %%eax\n"
-                         "push %%eax\n"
-                         "lea label, %%eax\n"
-                         "push %%eax\n"
-                         "lret\n"
-                         "label:"
-                         :
-                         : "ri" (Source)
-                         : "eax");
+            __asm__ volatile("mov %0, %%eax\n"
+                             "push %%eax\n"
+                             "lea label, %%eax\n"
+                             "push %%eax\n"
+                             "lret\n"
+                             "label:"
+                             :
+                             : "ri" (Source)
+                             : "eax");
             break;
         case SEGMENT_DS:
             /* Load DS Segment */
-            asm volatile("movl %0, %%ds"
-                         :
-                         : "r" (Source));
+            __asm__ volatile("movl %0, %%ds"
+                             :
+                             : "r" (Source));
             break;
         case SEGMENT_ES:
             /* Load ES Segment */
-            asm volatile("movl %0, %%es"
-                         :
-                         : "r" (Source));
+            __asm__ volatile("movl %0, %%es"
+                             :
+                             : "r" (Source));
             break;
         case SEGMENT_FS:
             /* Load FS Segment */
-            asm volatile("movl %0, %%fs"
-                         :
-                         : "r" (Source));
+            __asm__ volatile("movl %0, %%fs"
+                             :
+                             : "r" (Source));
             break;
         case SEGMENT_GS:
             /* Load GS Segment */
-            asm volatile("movl %0, %%gs"
-                         :
-                         : "r" (Source));
+            __asm__ volatile("movl %0, %%gs"
+                             :
+                             : "r" (Source));
             break;
         case SEGMENT_SS:
             /* Load SS Segment */
-            asm volatile("movl %0, %%ss"
-                         :
-                         : "r" (Source));
+            __asm__ volatile("movl %0, %%ss"
+                             :
+                             : "r" (Source));
             break;
     }
 }
@@ -274,9 +318,9 @@ XTCDECL
 VOID
 ArLoadTaskRegister(USHORT Source)
 {
-    asm volatile("ltr %0"
-                 :
-                 : "rm" (Source));
+    __asm__ volatile("ltr %0"
+                     :
+                     : "rm" (Source));
 }
 
 /**
@@ -291,10 +335,10 @@ VOID
 ArMemoryBarrier(VOID)
 {
    LONG Barrier;
-   asm volatile("xchg %%eax, %0"
-                :
-                : "m" (Barrier)
-                : "%eax");
+   __asm__ volatile("xchg %%eax, %0"
+                    :
+                    : "m" (Barrier)
+                    : "%eax");
 }
 
 /**
@@ -318,31 +362,31 @@ ArReadControlRegister(IN USHORT ControlRegister)
     {
         case 0:
             /* Read value from CR0 */
-            asm volatile("mov %%cr0, %0"
-                         : "=r" (Value)
-                         :
-                         : "memory");
+            __asm__ volatile("mov %%cr0, %0"
+                             : "=r" (Value)
+                             :
+                             : "memory");
             break;
         case 2:
             /* Read value from CR2 */
-            asm volatile("mov %%cr2, %0"
-                         : "=r" (Value)
-                         :
-                         : "memory");
+            __asm__ volatile("mov %%cr2, %0"
+                             : "=r" (Value)
+                             :
+                             : "memory");
             break;
         case 3:
             /* Read value from CR3 */
-            asm volatile("mov %%cr3, %0"
-                         : "=r" (Value)
-                         :
-                         : "memory");
+            __asm__ volatile("mov %%cr3, %0"
+                             : "=r" (Value)
+                             :
+                             : "memory");
             break;
         case 4:
             /* Read value from CR4 */
-            asm volatile("mov %%cr4, %0"
-                         : "=r" (Value)
-                         :
-                         : "memory");
+            __asm__ volatile("mov %%cr4, %0"
+                             : "=r" (Value)
+                             :
+                             : "memory");
             break;
         default:
             /* Invalid control register set */
@@ -375,43 +419,43 @@ ArReadDebugRegister(IN USHORT DebugRegister)
     {
         case 0:
             /* Read value from DR0 */
-            asm volatile("mov %%dr0, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr0, %0"
+                             : "=r" (Value));
             break;
         case 1:
             /* Read value from DR1 */
-            asm volatile("mov %%dr1, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr1, %0"
+                             : "=r" (Value));
             break;
         case 2:
             /* Read value from DR2 */
-            asm volatile("mov %%dr2, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr2, %0"
+                             : "=r" (Value));
             break;
         case 3:
             /* Read value from DR3 */
-            asm volatile("mov %%dr3, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr3, %0"
+                             : "=r" (Value));
             break;
         case 4:
             /* Read value from DR4 */
-            asm volatile("mov %%dr4, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr4, %0"
+                             : "=r" (Value));
             break;
         case 5:
             /* Read value from DR5 */
-            asm volatile("mov %%dr5, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr5, %0"
+                             : "=r" (Value));
             break;
         case 6:
             /* Read value from DR6 */
-            asm volatile("mov %%dr6, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr6, %0"
+                             : "=r" (Value));
             break;
         case 7:
             /* Read value from DR7 */
-            asm volatile("mov %%dr7, %0"
-                         : "=r" (Value));
+            __asm__ volatile("mov %%dr7, %0"
+                             : "=r" (Value));
             break;
         default:
             /* Invalid debug register set */
@@ -438,9 +482,9 @@ ULONG
 ArReadFSDualWord(ULONG Offset)
 {
     ULONG Value;
-    asm volatile("movl %%fs:%a[Offset], %k[Value]"
-                 : [Value] "=r" (Value)
-                 : [Offset] "ir" (Offset));
+    __asm__ volatile("movl %%fs:%a[Offset], %k[Value]"
+                     : [Value] "=r" (Value)
+                     : [Offset] "ir" (Offset));
     return Value;
 }
 
@@ -460,9 +504,9 @@ ArReadModelSpecificRegister(IN ULONG Register)
 {
     ULONGLONG Value;
 
-    asm volatile("rdmsr"
-                 : "=A" (Value)
-                 : "c" (Register));
+    __asm__ volatile("rdmsr"
+                     : "=A" (Value)
+                     : "c" (Register));
     return Value;
 }
 
@@ -493,8 +537,8 @@ ArReadTimeStampCounter(VOID)
 {
     ULONGLONG Value;
 
-    asm volatile("rdtsc"
-                 : "=A" (Value));
+    __asm__ volatile("rdtsc"
+                     : "=A" (Value));
 
     return Value;
 }
@@ -510,10 +554,10 @@ XTCDECL
 VOID
 ArReadWriteBarrier(VOID)
 {
-    asm volatile(""
-                 :
-                 :
-                 : "memory");
+    __asm__ volatile(""
+                     :
+                     :
+                     : "memory");
 }
 
 /**
@@ -527,7 +571,7 @@ XTCDECL
 VOID
 ArSetInterruptFlag(VOID)
 {
-    asm volatile("sti");
+    __asm__ volatile("sti");
 }
 
 /**
@@ -544,10 +588,10 @@ XTCDECL
 VOID
 ArStoreGlobalDescriptorTable(OUT PVOID Destination)
 {
-    asm volatile("sgdt %0"
-                 : "=m" (*(PSHORT)Destination)
-                 :
-                 : "memory");
+    __asm__ volatile("sgdt %0"
+                     : "=m" (*(PSHORT)Destination)
+                     :
+                     : "memory");
 }
 
 /**
@@ -564,10 +608,10 @@ XTCDECL
 VOID
 ArStoreInterruptDescriptorTable(OUT PVOID Destination)
 {
-    asm volatile("sidt %0"
-                 : "=m" (*(PSHORT)Destination)
-                 :
-                 : "memory");
+    __asm__ volatile("sidt %0"
+                     : "=m" (*(PSHORT)Destination)
+                     :
+                     : "memory");
 }
 
 /**
@@ -584,10 +628,10 @@ XTCDECL
 VOID
 ArStoreLocalDescriptorTable(OUT PVOID Destination)
 {
-    asm volatile("sldt %0"
-                 : "=m" (*(PSHORT)Destination)
-                 :
-                 : "memory");
+    __asm__ volatile("sldt %0"
+                     : "=m" (*(PSHORT)Destination)
+                     :
+                     : "memory");
 }
 
 /**
@@ -611,28 +655,28 @@ ArStoreSegment(IN USHORT Segment,
     switch(Segment)
     {
         case SEGMENT_CS:
-            asm volatile("movl %%cs, %0"
-                         : "=r" (*(PUINT)Destination));
+            __asm__ volatile("movl %%cs, %0"
+                             : "=r" (*(PUINT)Destination));
             break;
         case SEGMENT_DS:
-            asm volatile("movl %%ds, %0"
-                         : "=r" (*(PUINT)Destination));
+            __asm__ volatile("movl %%ds, %0"
+                             : "=r" (*(PUINT)Destination));
             break;
         case SEGMENT_ES:
-            asm volatile("movl %%es, %0"
-                         : "=r" (*(PUINT)Destination));
+            __asm__ volatile("movl %%es, %0"
+                             : "=r" (*(PUINT)Destination));
             break;
         case SEGMENT_FS:
-            asm volatile("movl %%fs, %0"
-                         : "=r" (*(PUINT)Destination));
+            __asm__ volatile("movl %%fs, %0"
+                             : "=r" (*(PUINT)Destination));
             break;
         case SEGMENT_GS:
-            asm volatile("movl %%gs, %0"
-                         : "=r" (*(PUINT)Destination));
+            __asm__ volatile("movl %%gs, %0"
+                             : "=r" (*(PUINT)Destination));
             break;
         case SEGMENT_SS:
-            asm volatile("movl %%ss, %0"
-                         : "=r" (*(PUINT)Destination));
+            __asm__ volatile("movl %%ss, %0"
+                             : "=r" (*(PUINT)Destination));
             break;
         default:
             Destination = NULL;
@@ -654,10 +698,10 @@ XTCDECL
 VOID
 ArStoreTaskRegister(OUT PVOID Destination)
 {
-    asm volatile("str %0"
-                 : "=m" (*(PULONG)Destination)
-                 :
-                 : "memory");
+    __asm__ volatile("str %0"
+                     : "=m" (*(PULONG)Destination)
+                     :
+                     : "memory");
 }
 
 /**
@@ -683,31 +727,31 @@ ArWriteControlRegister(IN USHORT ControlRegister,
     {
         case 0:
             /* Write value to CR0 */
-            asm volatile("mov %0, %%cr0"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%cr0"
+                             :
+                             : "r" (Value)
+                             : "memory");
             break;
         case 2:
             /* Write value to CR2 */
-            asm volatile("mov %0, %%cr2"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%cr2"
+                             :
+                             : "r" (Value)
+                             : "memory");
             break;
         case 3:
             /* Write value to CR3 */
-            asm volatile("mov %0, %%cr3"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%cr3"
+                             :
+                             : "r" (Value)
+                             : "memory");
             break;
         case 4:
             /* Write value to CR4 */
-            asm volatile("mov %0, %%cr4"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%cr4"
+                             :
+                             : "r" (Value)
+                             : "memory");
             break;
     }
 }
@@ -735,52 +779,52 @@ ArWriteDebugRegister(IN USHORT DebugRegister,
     {
         case 0:
             /* Write value to DR0 */
-            asm volatile("mov %0, %%dr0"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr0"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 1:
             /* Write value to DR1 */
-            asm volatile("mov %0, %%dr1"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr1"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 2:
             /* Write value to DR2 */
-            asm volatile("mov %0, %%dr2"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr2"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 3:
             /* Write value to DR3 */
-            asm volatile("mov %0, %%dr3"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr3"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 4:
             /* Write value to DR4 */
-            asm volatile("mov %0, %%dr4"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr4"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 5:
             /* Write value to DR5 */
-            asm volatile("mov %0, %%dr5"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr5"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 6:
             /* Write value to DR6 */
-            asm volatile("mov %0, %%dr6"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr6"
+                             :
+                             : "r" (Value)
+                             : "memory");
         case 7:
             /* Write value to DR7 */
-            asm volatile("mov %0, %%dr7"
-                         :
-                         : "r" (Value)
-                         : "memory");
+            __asm__ volatile("mov %0, %%dr7"
+                             :
+                             : "r" (Value)
+                             : "memory");
     }
 }
 
@@ -798,10 +842,10 @@ XTCDECL
 VOID
 ArWriteEflagsRegister(IN UINT_PTR Value)
 {
-    asm volatile("push %0\n"
-                 "popf"
-                 :
-                 : "rim" (Value));
+    __asm__ volatile("push %0\n"
+                     "popf"
+                     :
+                     : "rim" (Value));
 }
 
 /**
@@ -822,10 +866,10 @@ VOID
 ArWriteModelSpecificRegister(IN ULONG Register,
                              IN ULONGLONG Value)
 {
-    asm volatile("wrmsr"
-                 :
-                 : "c" (Register),
-                   "A" (Value));
+    __asm__ volatile("wrmsr"
+                     :
+                     : "c" (Register),
+                       "A" (Value));
 }
 
 /**
@@ -839,8 +883,8 @@ XTCDECL
 VOID
 ArYieldProcessor(VOID)
 {
-    asm volatile("pause"
-                 :
-                 :
-                 : "memory");
+    __asm__ volatile("pause"
+                     :
+                     :
+                     : "memory");
 }

xtoskrnl/ar/i686/procsup.c

@@ -18,31 +18,43 @@
  */
 XTAPI
 VOID
-ArInitializeProcessor(VOID)
+ArInitializeProcessor(IN PVOID ProcessorStructures)
 {
     KDESCRIPTOR GdtDescriptor, IdtDescriptor;
+    PVOID KernelBootStack, KernelFaultStack;
     PKPROCESSOR_BLOCK ProcessorBlock;
-    PVOID KernelFaultStack;
     PKGDTENTRY Gdt;
     PKIDTENTRY Idt;
     PKTSS Tss;
 
-    /* Use initial structures */
-    Gdt = ArInitialGdt;
-    Idt = ArInitialIdt;
-    Tss = &ArInitialTss;
-    KernelFaultStack = &ArKernelFaultStack;
+    /* Check if processor structures buffer provided */
+    if(ProcessorStructures)
+    {
+        /* Assign CPU structures from provided buffer */
+        ArpInitializeProcessorStructures(ProcessorStructures, &Gdt, &Tss, &ProcessorBlock,
+                                         &KernelBootStack, &KernelFaultStack);
 
-    /* Load processor block */
-    ProcessorBlock = CONTAIN_RECORD(&ArInitialProcessorBlock.Prcb, KPROCESSOR_BLOCK, Prcb);
+        /* Use global IDT */
+        Idt = ArInitialIdt;
+    }
+    else
+    {
+        /* Use initial structures */
+        Gdt = ArInitialGdt;
+        Idt = ArInitialIdt;
+        Tss = &ArInitialTss;
+        KernelBootStack = &ArKernelBootStack;
+        KernelFaultStack = &ArKernelFaultStack;
+        ProcessorBlock = &ArInitialProcessorBlock;
+    }
 
     /* Initialize processor block */
     ArpInitializeProcessorBlock(ProcessorBlock, Gdt, Idt, Tss, KernelFaultStack);
 
     /* Initialize GDT, IDT and TSS */
     ArpInitializeGdt(ProcessorBlock);
-    ArpInitializeTss(ProcessorBlock);
     ArpInitializeIdt(ProcessorBlock);
+    ArpInitializeTss(ProcessorBlock, KernelBootStack, KernelFaultStack);
 
     /* Set GDT and IDT descriptors */
     GdtDescriptor.Base = Gdt;
@@ -130,14 +142,14 @@ ArpIdentifyProcessor(VOID)
 
     /* Store CPU vendor in processor control block */
     Prcb->CpuId.Vendor = CpuRegisters.Ebx;
-    Prcb->CpuId.VendorName[0] = CpuRegisters.Ebx;
-    Prcb->CpuId.VendorName[4] = CpuRegisters.Edx;
-    Prcb->CpuId.VendorName[8] = CpuRegisters.Ecx;
+    *(PULONG)&Prcb->CpuId.VendorName[0] = CpuRegisters.Ebx;
+    *(PULONG)&Prcb->CpuId.VendorName[4] = CpuRegisters.Edx;
+    *(PULONG)&Prcb->CpuId.VendorName[8] = CpuRegisters.Ecx;
     Prcb->CpuId.VendorName[12] = '\0';
 
-    /* Get CPU features */
+    /* Get CPU standard features */
     RtlZeroMemory(&CpuRegisters, sizeof(CPUID_REGISTERS));
-    CpuRegisters.Leaf = CPUID_GET_CPU_FEATURES;
+    CpuRegisters.Leaf = CPUID_GET_STANDARD1_FEATURES;
     ArCpuId(&CpuRegisters);
 
     /* Store CPU signature in processor control block */
@@ -150,9 +162,9 @@ ArpIdentifyProcessor(VOID)
     if(Prcb->CpuId.Vendor == CPU_VENDOR_AMD)
     {
         /* AMD CPU */
-        Prcb->CpuId.Family = Prcb->CpuId.Family + CpuSignature.ExtendedFamily;
-        if(Prcb->CpuId.Model == 0xF)
+        if(Prcb->CpuId.Family >= 0xF)
         {
+            Prcb->CpuId.Family = Prcb->CpuId.Family + CpuSignature.ExtendedFamily;
             Prcb->CpuId.Model = Prcb->CpuId.Model + (CpuSignature.ExtendedModel << 4);
         }
     }
@@ -164,7 +176,7 @@ ArpIdentifyProcessor(VOID)
             Prcb->CpuId.Family = Prcb->CpuId.Family + CpuSignature.ExtendedFamily;
         }
 
-        if((Prcb->CpuId.Family == 0xF) || (Prcb->CpuId.Family == 0x6))
+        if((Prcb->CpuId.Family == 0x6) || (Prcb->CpuId.Family == 0xF))
         {
             Prcb->CpuId.Model = Prcb->CpuId.Model + (CpuSignature.ExtendedModel << 4);
         }
@@ -286,9 +298,6 @@ ArpInitializeProcessorBlock(OUT PKPROCESSOR_BLOCK ProcessorBlock,
                             IN PKTSS Tss,
                             IN PVOID DpcStack)
 {
-    /* Fill processor block with zeroes */
-    RtlZeroMemory(ProcessorBlock, sizeof(KPROCESSOR_BLOCK));
-
     /* Set processor block and processor control block */
     ProcessorBlock->Self = ProcessorBlock;
     ProcessorBlock->CurrentPrcb = &ProcessorBlock->Prcb;
@@ -302,9 +311,9 @@ ArpInitializeProcessorBlock(OUT PKPROCESSOR_BLOCK ProcessorBlock,
     ProcessorBlock->Prcb.DpcStack = DpcStack;
 
     /* Setup processor control block */
-    ProcessorBlock->Prcb.Number = 0;
-    ProcessorBlock->Prcb.SetMember = 1;
-    ProcessorBlock->Prcb.MultiThreadProcessorSet = 1;
+    ProcessorBlock->Prcb.CpuNumber = ProcessorBlock->CpuNumber;
+    ProcessorBlock->Prcb.SetMember = 1 << ProcessorBlock->CpuNumber;
+    ProcessorBlock->Prcb.MultiThreadProcessorSet = 1 << ProcessorBlock->CpuNumber;
 
     /* Clear DR6 and DR7 registers */
     ProcessorBlock->Prcb.ProcessorState.SpecialRegisters.KernelDr6 = 0;
@@ -338,6 +347,64 @@ ArpInitializeProcessorRegisters(VOID)
     ArWriteControlRegister(0, ArReadControlRegister(0) | CR0_WP);
 }
 
+/**
+ * Initializes i686 processor specific structures with provided memory buffer.
+ *
+ * @param ProcessorStructures
+ *        Supplies a pointer to the allocated buffer with processor structures.
+ *
+ * @param Gdt
+ *        Supplies a pointer to the GDT.
+ *
+ * @param Tss
+ *        Supplies a pointer to the TSS.
+ *
+ * @param ProcessorBlock
+ *        Supplies a pointer to the processor block.
+ *
+ * @param KernelBootStack
+ *        Supplies a pointer to the kernel boot stack.
+ *
+ * @param KernelFaultStack
+ *        Supplies a pointer to the kernel fault stack.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+VOID
+ArpInitializeProcessorStructures(IN PVOID ProcessorStructures,
+                                 OUT PKGDTENTRY *Gdt,
+                                 OUT PKTSS *Tss,
+                                 OUT PKPROCESSOR_BLOCK *ProcessorBlock,
+                                 OUT PVOID *KernelBootStack,
+                                 OUT PVOID *KernelFaultStack)
+{
+    UINT_PTR Address;
+
+    /* Align address to page size boundary and move to kernel boot stack */
+    Address = ROUND_UP((UINT_PTR)ProcessorStructures, MM_PAGE_SIZE) + KERNEL_STACK_SIZE;
+
+    /* Assign a space for kernel boot stack and advance */
+    *KernelBootStack = (PVOID)Address;
+    Address += KERNEL_STACK_SIZE;
+
+    /* Assign a space for kernel fault stack, no advance needed as stack grows down */
+    *KernelFaultStack = (PVOID)Address;
+
+    /* Assign a space for GDT and advance */
+    *Gdt = (PVOID)Address;
+    Address += sizeof(ArInitialGdt);
+
+    /* Assign a space for Processor Block and advance */
+    *ProcessorBlock = (PVOID)Address;
+    Address += sizeof(ArInitialProcessorBlock);
+
+    /* Assign a space for TSS */
+    *Tss = (PVOID)Address;
+}
+
 /**
  * Initializes segment registers.
  *
@@ -368,7 +435,9 @@ ArpInitializeSegments(VOID)
  */
 XTAPI
 VOID
-ArpInitializeTss(IN PKPROCESSOR_BLOCK ProcessorBlock)
+ArpInitializeTss(IN PKPROCESSOR_BLOCK ProcessorBlock,
+                 IN PVOID KernelBootStack,
+                 IN PVOID KernelFaultStack)
 {
     /* Clear I/O map */
     RtlSetMemory(ProcessorBlock->TssBase->IoMaps[0].IoMap, 0xFF, IOPM_FULL_SIZE);
@@ -391,6 +460,7 @@ ArpInitializeTss(IN PKPROCESSOR_BLOCK ProcessorBlock)
 
     /* Set I/O map base and disable traps */
     ProcessorBlock->TssBase->IoMapBase = sizeof(KTSS);
+    ProcessorBlock->TssBase->Esp0 = (ULONG_PTR)KernelBootStack;
     ProcessorBlock->TssBase->Flags = 0;
 
     /* Set LDT and SS */
@@ -398,8 +468,8 @@ ArpInitializeTss(IN PKPROCESSOR_BLOCK ProcessorBlock)
     ProcessorBlock->TssBase->Ss0 = KGDT_R0_DATA;
 
     /* Initialize task gates for DoubleFault and NMI traps */
-    ArpSetDoubleFaultTssEntry(ProcessorBlock);
-    ArpSetNonMaskableInterruptTssEntry(ProcessorBlock);
+    ArpSetDoubleFaultTssEntry(ProcessorBlock, KernelFaultStack);
+    ArpSetNonMaskableInterruptTssEntry(ProcessorBlock, KernelFaultStack);
 }
 
 /**
@@ -414,7 +484,8 @@ ArpInitializeTss(IN PKPROCESSOR_BLOCK ProcessorBlock)
  */
 XTAPI
 VOID
-ArpSetDoubleFaultTssEntry(IN PKPROCESSOR_BLOCK ProcessorBlock)
+ArpSetDoubleFaultTssEntry(IN PKPROCESSOR_BLOCK ProcessorBlock,
+                          IN PVOID KernelFaultStack)
 {
     PKGDTENTRY TaskGateEntry, TssEntry;
     PKTSS Tss;
@@ -432,8 +503,8 @@ ArpSetDoubleFaultTssEntry(IN PKPROCESSOR_BLOCK ProcessorBlock)
     Tss->Flags = 0;
     Tss->LDT = KGDT_R0_LDT;
     Tss->CR3 = ArReadControlRegister(3);
-    Tss->Esp = (ULONG_PTR)&ArKernelFaultStack;
-    Tss->Esp0 = (ULONG_PTR)&ArKernelFaultStack;
+    Tss->Esp = (ULONG_PTR)KernelFaultStack;
+    Tss->Esp0 = (ULONG_PTR)KernelFaultStack;
     Tss->Eip = PtrToUlong(ArpHandleTrap08);
     Tss->Cs = KGDT_R0_CODE;
     Tss->Ds = KGDT_R3_DATA | RPL_MASK;
@@ -583,7 +654,8 @@ ArpSetIdtGate(IN PKIDTENTRY Idt,
  */
 XTAPI
 VOID
-ArpSetNonMaskableInterruptTssEntry(IN PKPROCESSOR_BLOCK ProcessorBlock)
+ArpSetNonMaskableInterruptTssEntry(IN PKPROCESSOR_BLOCK ProcessorBlock,
+                                   IN PVOID KernelFaultStack)
 {
     PKGDTENTRY TaskGateEntry, TssEntry;
     PKTSS Tss;
@@ -601,8 +673,8 @@ ArpSetNonMaskableInterruptTssEntry(IN PKPROCESSOR_BLOCK ProcessorBlock)
     Tss->Flags = 0;
     Tss->LDT = KGDT_R0_LDT;
     Tss->CR3 = ArReadControlRegister(3);
-    Tss->Esp = (ULONG_PTR)&ArKernelFaultStack;
-    Tss->Esp0 = (ULONG_PTR)&ArKernelFaultStack;
+    Tss->Esp = (ULONG_PTR)KernelFaultStack;
+    Tss->Esp0 = (ULONG_PTR)KernelFaultStack;
     Tss->Eip = PtrToUlong(ArpHandleTrap02);
     Tss->Cs = KGDT_R0_CODE;
     Tss->Ds = KGDT_R3_DATA | RPL_MASK;

xtoskrnl/hl/acpi.c

@@ -0,0 +1,774 @@
+/**
+ * PROJECT:         ExectOS
+ * COPYRIGHT:       See COPYING.md in the top level directory
+ * FILE:            xtoskrnl/hl/x86/acpi.c
+ * DESCRIPTION:     Advanced Configuration and Power Interface (ACPI) support
+ * DEVELOPERS:      Rafal Kupiec <belliash@codingworkshop.eu.org>
+ */
+
+#include <xtos.h>
+
+
+/**
+ * Gets a pointer to the ACPI system description pointer (RSDP).
+ *
+ * @param Rsdp
+ *        Supplies a pointer to the memory area, where RSDP virtual address will be stored.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+XTSTATUS
+HlGetAcpiSystemDescriptionPointer(OUT PACPI_RSDP *Rsdp)
+{
+    /* Get RSDP and return success */
+    *Rsdp = HlpAcpiRsdp;
+    return STATUS_SUCCESS;
+}
+
+/**
+ * Gets an ACPI description table with given signature, available in the XSDT/RSDT.
+ *
+ * @param Signature
+ *        Supplies the signature of the desired ACPI table.
+ *
+ * @param AcpiTable
+ *        Supplies a pointer to memory area where ACPI table virtual address will be stored.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+XTSTATUS
+HlGetAcpiTable(IN ULONG Signature,
+               OUT PACPI_DESCRIPTION_HEADER *AcpiTable)
+{
+    PACPI_DESCRIPTION_HEADER Table;
+    XTSTATUS Status;
+
+    /* Assume ACPI table not found */
+    *AcpiTable = NULL;
+
+    /* Attempt to get ACPI table from the cache */
+    Status = HlpQueryAcpiCache(Signature, &Table);
+    if(Status != STATUS_SUCCESS)
+    {
+        /* Table not found in the cache, query ACPI tables */
+        Status = HlpQueryAcpiTables(Signature, &Table);
+        if(Status != STATUS_SUCCESS)
+        {
+            /* ACPI table not found, return error */
+            return STATUS_NOT_FOUND;
+        }
+    }
+
+    /* Store ACPI table and return success */
+    *AcpiTable = Table;
+    return STATUS_SUCCESS;
+}
+
+/**
+ * Stores given ACPI table in the kernel local cache.
+ *
+ * @param AcpiTable
+ *        Supplies a pointer to ACPI table that will be stored in the cache.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+VOID
+HlpCacheAcpiTable(IN PACPI_DESCRIPTION_HEADER AcpiTable)
+{
+    PACPI_CACHE_LIST AcpiCache;
+
+    /* Create new ACPI table cache entry */
+    AcpiCache = CONTAIN_RECORD(AcpiTable, ACPI_CACHE_LIST, Header);
+    RtlInsertTailList(&HlpAcpiCacheList, &AcpiCache->ListEntry);
+}
+
+/**
+ * Performs an initialization of the ACPI subsystem.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+XTSTATUS
+HlpInitializeAcpi(VOID)
+{
+    PACPI_FADT Fadt;
+    XTSTATUS Status;
+
+    /* Initialize ACPI cache */
+    Status = HlpInitializeAcpiCache();
+    if(Status != STATUS_SUCCESS)
+    {
+        /* ACPI cache initialization failed, return error */
+        return Status;
+    }
+
+    /* Get Fixed ACPI Description Table (FADT) */
+    Status = HlGetAcpiTable(ACPI_FADT_SIGNATURE, (PACPI_DESCRIPTION_HEADER*)&Fadt);
+    if(Status != STATUS_SUCCESS || !Fadt)
+    {
+        /* Failed to get FADT, return error */
+        return STATUS_NOT_FOUND;
+    }
+
+    /* Initialize ACPI timer */
+    HlpInitializeAcpiTimer();
+
+    /* Return success */
+    return STATUS_SUCCESS;
+}
+
+/**
+ * Initializes the kernel's local ACPI cache storage.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+XTSTATUS
+HlpInitializeAcpiCache(VOID)
+{
+    PACPI_DESCRIPTION_HEADER Rsdt;
+    XTSTATUS Status;
+
+    /* Initialize ACPI cache list */
+    RtlInitializeListHead(&HlpAcpiCacheList);
+
+    /* Get XSDT/RSDT */
+    Status = HlpInitializeAcpiSystemDescriptionTable(&Rsdt);
+    if(Status != STATUS_SUCCESS)
+    {
+        /* Failed to get XSDT/RSDT, return error */
+        return Status;
+    }
+
+    /* Cache XSDT/RSDT table */
+    HlpCacheAcpiTable(Rsdt);
+
+    /* Return success */
+    return STATUS_SUCCESS;
+}
+
+/**
+ * Initializes ACPI System Description Table (XSDT/RSDT) by finding proper table and mapping its virtual address.
+ *
+ * @param AcpiTable
+ *        Supplies a pointer to memory area where ACPI table virtual address will be stored.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+XTSTATUS
+HlpInitializeAcpiSystemDescriptionTable(OUT PACPI_DESCRIPTION_HEADER *AcpiTable)
+{
+    PHYSICAL_ADDRESS RsdpAddress, RsdtAddress;
+    PSYSTEM_RESOURCE_HEADER ResourceHeader;
+    PSYSTEM_RESOURCE_ACPI AcpiResource;
+    ULONG RsdtPages;
+    PACPI_RSDT Rsdt;
+    XTSTATUS Status;
+
+    /* Assume ACPI table not found */
+    *AcpiTable = NULL;
+
+    /* Get ACPI system resource */
+    Status = KeGetSystemResource(SystemResourceAcpi, &ResourceHeader);
+    if(Status != STATUS_SUCCESS)
+    {
+        /* Resource not found */
+        return STATUS_NOT_FOUND;
+    }
+
+    /* Cast system resource to ACPI resource and store RSDP physical address */
+    AcpiResource = (PSYSTEM_RESOURCE_ACPI)ResourceHeader;
+    RsdpAddress.QuadPart = (LONGLONG)AcpiResource->Header.PhysicalAddress;
+
+    /* Map RSDP and mark it as CD/WT to avoid delays in write-back cache */
+    Status = MmMapHardwareMemory(RsdpAddress, 1, TRUE, (PVOID *)&HlpAcpiRsdp);
+    MmMarkHardwareMemoryWriteThrough(HlpAcpiRsdp, 1);
+
+    /* Validate RSDP signature */
+    if(Status != STATUS_SUCCESS || HlpAcpiRsdp->Signature != ACPI_RSDP_SIGNATURE)
+    {
+        /* Not mapped correctly or invalid RSDP signature, return error */
+        return STATUS_INVALID_PARAMETER;
+    }
+
+    /* Check RSDP revision to determine RSDT/XSDT address */
+    if(HlpAcpiRsdp->Revision >= 2)
+    {
+        /* Get XSDT address */
+        RsdtAddress.QuadPart = (LONGLONG)HlpAcpiRsdp->XsdtAddress;
+    }
+    else
+    {
+        /* Get RSDT address */
+        RsdtAddress.QuadPart = (LONGLONG)HlpAcpiRsdp->RsdtAddress;
+    }
+
+    /* Map RSDT/XSDT as CD/WT */
+    Status = MmMapHardwareMemory(RsdtAddress, 2, TRUE, (PVOID *)&Rsdt);
+    MmMarkHardwareMemoryWriteThrough(Rsdt, 2);
+
+    /* Validate RSDT/XSDT signature */
+    if((Status != STATUS_SUCCESS) ||
+       (Rsdt->Header.Signature != ACPI_RSDT_SIGNATURE &&
+        Rsdt->Header.Signature != ACPI_XSDT_SIGNATURE))
+    {
+        /* Not mapped correctly or invalid RSDT/XSDT signature, return error */
+        return STATUS_INVALID_PARAMETER;
+    }
+
+    /* Calculate the length of all available ACPI tables and remap it if needed */
+    RsdtPages = ((RsdtAddress.LowPart & (MM_PAGE_SIZE - 1)) + Rsdt->Header.Length + (MM_PAGE_SIZE - 1)) >> MM_PAGE_SHIFT;
+    if(RsdtPages != 2)
+    {
+        /* RSDT/XSDT needs less or more than 2 pages, remap it */
+        MmUnmapHardwareMemory(Rsdt, 2, TRUE);
+        Status = MmMapHardwareMemory(RsdtAddress, RsdtPages, TRUE, (PVOID *)&Rsdt);
+        MmMarkHardwareMemoryWriteThrough(Rsdt, RsdtPages);
+
+        /* Make sure remapping was successful */
+        if(Status != STATUS_SUCCESS)
+        {
+            /* Remapping failed, return error */
+            return STATUS_INSUFFICIENT_RESOURCES;
+        }
+    }
+
+    /* Get ACPI table header and return success */
+    *AcpiTable = &Rsdt->Header;
+    return STATUS_SUCCESS;
+}
+
+/**
+ * Initializes System Information structure based on the ACPI provided data.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+XTSTATUS
+HlpInitializeAcpiSystemInformation(VOID)
+{
+    PACPI_MADT_LOCAL_X2APIC LocalX2Apic;
+    PACPI_MADT_LOCAL_APIC LocalApic;
+    ULONG_PTR MadtTable;
+    PACPI_MADT Madt;
+    XTSTATUS Status;
+    USHORT CpuCount;
+
+    /* Allocate memory for ACPI system information structure */
+    Status = HlpInitializeAcpiSystemStructure();
+    if(Status != STATUS_SUCCESS)
+    {
+        /* Failed to allocate memory, return error */
+        return STATUS_INSUFFICIENT_RESOURCES;
+    }
+
+    /* Get Multiple APIC Description Table (MADT) */
+    Status = HlGetAcpiTable(ACPI_MADT_SIGNATURE, (PACPI_DESCRIPTION_HEADER*)&Madt);
+    if(Status != STATUS_SUCCESS)
+    {
+        /* Failed to get MADT, return error */
+        return STATUS_NOT_FOUND;
+    }
+
+    /* Set APIC table traverse pointer and initialize number of CPUs */
+    MadtTable = (ULONG_PTR)Madt->ApicTables;
+    CpuCount = 0;
+
+    /* Traverse all MADT tables to get system information */
+    while(MadtTable <= ((ULONG_PTR)Madt + Madt->Header.Length))
+    {
+        /* Check if this is a local APIC subtable */
+        if((((PACPI_SUBTABLE_HEADER)MadtTable)->Type == ACPI_MADT_TYPE_LOCAL_APIC) &&
+           (((PACPI_SUBTABLE_HEADER)MadtTable)->Length == sizeof(ACPI_MADT_LOCAL_APIC)))
+        {
+            /* Get local APIC subtable */
+            LocalApic = (PACPI_MADT_LOCAL_APIC)MadtTable;
+
+            /* Make sure, this CPU can be enabled */
+            if(LocalApic->Flags & ACPI_MADT_PLAOC_ENABLED)
+            {
+                /* Store CPU number, APIC ID and CPU ID */
+                HlpSystemInfo.CpuInfo[CpuCount].AcpiId = LocalApic->AcpiId;
+                HlpSystemInfo.CpuInfo[CpuCount].ApicId = LocalApic->ApicId;
+                HlpSystemInfo.CpuInfo[CpuCount].CpuNumber = CpuCount;
+
+                /* Increment number of CPUs */
+                CpuCount++;
+            }
+
+            /* Go to the next MADT table */
+            MadtTable += ((PACPI_SUBTABLE_HEADER)MadtTable)->Length;
+        }
+        else if((((PACPI_SUBTABLE_HEADER)MadtTable)->Type == ACPI_MADT_TYPE_LOCAL_X2APIC) &&
+                (((PACPI_SUBTABLE_HEADER)MadtTable)->Length == sizeof(ACPI_MADT_LOCAL_X2APIC)))
+        {
+            /* Get local X2APIC subtable */
+            LocalX2Apic = (PACPI_MADT_LOCAL_X2APIC)MadtTable;
+
+            /* Make sure, this CPU can be enabled */
+            if(LocalX2Apic->Flags & ACPI_MADT_PLAOC_ENABLED)
+            {
+                /* Store CPU number, APIC ID and CPU ID */
+                HlpSystemInfo.CpuInfo[CpuCount].AcpiId = LocalX2Apic->AcpiId;
+                HlpSystemInfo.CpuInfo[CpuCount].ApicId = LocalX2Apic->ApicId;
+                HlpSystemInfo.CpuInfo[CpuCount].CpuNumber = CpuCount;
+
+                /* Increment number of CPUs */
+                CpuCount++;
+            }
+
+            /* Go to the next MADT table */
+            MadtTable += ((PACPI_SUBTABLE_HEADER)MadtTable)->Length;
+        }
+        else
+        {
+            /* Any other MADT table, try to go to the next one byte-by-byte */
+            MadtTable += 1;
+        }
+    }
+
+    /* Store number of CPUs */
+    HlpSystemInfo.CpuCount = CpuCount;
+
+    /* Return success */
+    return STATUS_SUCCESS;
+}
+
+/**
+ * Initializes ACPI System Information data structure based on the size of available ACPI data.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+XTSTATUS
+HlpInitializeAcpiSystemStructure(VOID)
+{
+    PHYSICAL_ADDRESS PhysicalAddress;
+    PFN_NUMBER PageCount;
+    ULONG_PTR MadtTable;
+    PACPI_MADT Madt;
+    XTSTATUS Status;
+    ULONG CpuCount;
+
+    /* Get Multiple APIC Description Table (MADT) */
+    Status = HlGetAcpiTable(ACPI_MADT_SIGNATURE, (PACPI_DESCRIPTION_HEADER*)&Madt);
+    if(Status != STATUS_SUCCESS)
+    {
+        /* Failed to get MADT, return error */
+        return STATUS_NOT_FOUND;
+    }
+
+    /* Set APIC table traverse pointer and initialize number of CPUs */
+    MadtTable = (ULONG_PTR)Madt->ApicTables;
+    CpuCount = 0;
+
+    /* Traverse all MADT tables to get number of processors */
+    while(MadtTable <= ((ULONG_PTR)Madt + Madt->Header.Length))
+    {
+        /* Check if this is a local APIC subtable */
+        if((((PACPI_SUBTABLE_HEADER)MadtTable)->Type == ACPI_MADT_TYPE_LOCAL_APIC) &&
+           (((PACPI_SUBTABLE_HEADER)MadtTable)->Length == sizeof(ACPI_MADT_LOCAL_APIC)))
+        {
+            /* Make sure, this CPU can be enabled */
+            if(((PACPI_MADT_LOCAL_APIC)MadtTable)->Flags & ACPI_MADT_PLAOC_ENABLED)
+            {
+                /* Increment number of CPUs */
+                CpuCount++;
+            }
+
+            /* Go to the next MADT table */
+            MadtTable += ((PACPI_SUBTABLE_HEADER)MadtTable)->Length;
+        }
+        else if((((PACPI_SUBTABLE_HEADER)MadtTable)->Type == ACPI_MADT_TYPE_LOCAL_X2APIC) &&
+                (((PACPI_SUBTABLE_HEADER)MadtTable)->Length == sizeof(ACPI_MADT_LOCAL_X2APIC)))
+        {
+            /* Make sure, this CPU can be enabled */
+            if(((PACPI_MADT_LOCAL_X2APIC)MadtTable)->Flags & ACPI_MADT_PLAOC_ENABLED)
+            {
+                /* Increment number of CPUs */
+                CpuCount++;
+            }
+
+            /* Go to the next MADT table */
+            MadtTable += ((PACPI_SUBTABLE_HEADER)MadtTable)->Length;
+        }
+        else
+        {
+            /* Any other MADT table, try to go to the next one byte-by-byte */
+            MadtTable += 1;
+        }
+    }
+
+    /* Zero the ACPI system information structure */
+    RtlZeroMemory(&HlpSystemInfo, sizeof(ACPI_SYSTEM_INFO));
+
+    /* Calculate number of pages needed to store CPU information */
+    PageCount = SIZE_TO_PAGES(CpuCount * sizeof(PROCESSOR_IDENTITY));
+
+    /* Allocate memory for CPU information */
+    Status = MmAllocateHardwareMemory(PageCount, TRUE, &PhysicalAddress);
+    if(Status != STATUS_SUCCESS)
+    {
+        /* Failed to allocate memory, return error */
+        return Status;
+    }
+
+    /* Map physical address to the virtual memory area */
+    Status = MmMapHardwareMemory(PhysicalAddress, PageCount, TRUE, (PVOID *)&HlpSystemInfo.CpuInfo);
+    if(Status != STATUS_SUCCESS)
+    {
+        /* Failed to map memory, return error */
+        return Status;
+    }
+
+    /* Zero the CPU information structure */
+    RtlZeroMemory(HlpSystemInfo.CpuInfo, PAGES_TO_SIZE(PageCount));
+
+    /* Return success */
+    return STATUS_SUCCESS;
+}
+
+/**
+ * Initializes the ACPI Timer.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+XTSTATUS
+HlpInitializeAcpiTimer(VOID)
+{
+    PACPI_FADT Fadt;
+    XTSTATUS Status;
+
+    /* Get Fixed ACPI Description Table (FADT) */
+    Status = HlGetAcpiTable(ACPI_FADT_SIGNATURE, (PACPI_DESCRIPTION_HEADER*)&Fadt);
+    if(Status != STATUS_SUCCESS || !Fadt)
+    {
+        /* Failed to get FADT, return error */
+        return STATUS_NOT_FOUND;
+    }
+
+    /* Set ACPI timer port address */
+    HlpAcpiTimerInfo.TimerPort = Fadt->PmTmrBlkIoPort;
+
+    /* Determine whether 32-bit or 24-bit timer is used */
+    if(Fadt->Flags & ACPI_FADT_32BIT_TIMER)
+    {
+        /* 32-bit timer */
+        HlpAcpiTimerInfo.MsbMask = ACPI_FADT_TIMER_32BIT;
+    }
+    else
+    {
+        /* 24-bit timer */
+        HlpAcpiTimerInfo.MsbMask = ACPI_FADT_TIMER_24BIT;
+    }
+
+    /* Return success */
+    return STATUS_SUCCESS;
+}
+
+/**
+ * Queries kernel local ACPI cache in attempt to find a requested ACPI table.
+ *
+ * @param Signature
+ *        Supplies the signature of the desired ACPI table.
+ *
+ * @param AcpiTable
+ *        Supplies a pointer to memory area where ACPI table virtual address will be stored.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+XTSTATUS
+HlpQueryAcpiCache(IN ULONG Signature,
+                  OUT PACPI_DESCRIPTION_HEADER *AcpiTable)
+{
+    PACPI_DESCRIPTION_HEADER TableHeader;
+    PACPI_CACHE_LIST AcpiCache;
+    PLIST_ENTRY ListEntry;
+
+    /* Initialize variables */
+    TableHeader = NULL;
+
+    /* Iterate through ACPI tables cache list */
+    ListEntry = HlpAcpiCacheList.Flink;
+    while(ListEntry != &HlpAcpiCacheList)
+    {
+        /* Get cached ACPI table header */
+        AcpiCache = CONTAIN_RECORD(ListEntry, ACPI_CACHE_LIST, ListEntry);
+
+        /* Check if ACPI table signature matches */
+        if(AcpiCache->Header.Signature == Signature)
+        {
+            /* ACPI table found in cache, return it */
+            TableHeader = &AcpiCache->Header;
+            break;
+        }
+
+        /* Go to the next cache entry */
+        ListEntry = ListEntry->Flink;
+    }
+
+    /* Check if the requested ACPI table was found in the cache */
+    if(TableHeader == NULL)
+    {
+        /* ACPI table not found in cache, return error */
+        return STATUS_NOT_FOUND;
+    }
+
+    /* Return table header and status code */
+    *AcpiTable = TableHeader;
+    return STATUS_SUCCESS;
+}
+
+/**
+ * Queries XSDT/RSDT in order to find a requested ACPI table.
+ * Once the desired ACPI table is found, it is being mapped and cached.
+ *
+ * @param Signature
+ *        Supplies the signature of the desired ACPI table.
+ *
+ * @param AcpiTable
+ *        Supplies a pointer to memory area where ACPI table virtual address will be stored.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+XTSTATUS
+HlpQueryAcpiTables(IN ULONG Signature,
+                   OUT PACPI_DESCRIPTION_HEADER *AcpiTable)
+{
+    ULONG TableCount, TableIndex, TablePages;
+    PACPI_DESCRIPTION_HEADER TableHeader;
+    PHYSICAL_ADDRESS TableAddress;
+    PACPI_FADT Fadt;
+    PACPI_RSDT Rsdt;
+    PACPI_XSDT Xsdt;
+    XTSTATUS Status;
+
+    /* Check if requesting a valid ACPI table */
+    if(Signature == ACPI_RSDT_SIGNATURE || Signature == ACPI_XSDT_SIGNATURE)
+    {
+        /* Cannot provide RSDP/XSDP table, it should be cached; return error */
+        return STATUS_INVALID_PARAMETER;
+    }
+
+    /* Ensure that table header is not set before attempting to find ACPI table */
+    TableHeader = NULL;
+
+    /* Check if DSDT or FACS table requested */
+    if(Signature == ACPI_DSDT_SIGNATURE || Signature == ACPI_FACS_SIGNATURE)
+    {
+        /* Get FADT as it contains a pointer to DSDT and FACS */
+        Status = HlGetAcpiTable(ACPI_FADT_SIGNATURE, (PACPI_DESCRIPTION_HEADER*)&Fadt);
+        if(Status != STATUS_SUCCESS)
+        {
+            /* Failed to get FADT, return error */
+            return Status;
+        }
+
+        /* Check if DSDT or FACS table requested */
+        if(Signature == ACPI_DSDT_SIGNATURE)
+        {
+            /* Get DSDT address */
+            TableAddress.LowPart = Fadt->Dsdt;
+        }
+        else
+        {
+            /* Get FACS address */
+            TableAddress.LowPart = Fadt->FirmwareCtrl;
+        }
+
+        /* Fill in high part of ACPI table address */
+        TableAddress.HighPart = 0;
+
+        /* Map table using hardware memory pool */
+        Status = MmMapHardwareMemory(TableAddress, 2, TRUE, (PVOID*)&TableHeader);
+        if(Status != STATUS_SUCCESS)
+        {
+            /* Failed to map table, return error */
+            return Status;
+        }
+    }
+    else
+    {
+        /* Query cache for XSDP table */
+        Status = HlpQueryAcpiCache(ACPI_XSDT_SIGNATURE, (PACPI_DESCRIPTION_HEADER*)&Xsdt);
+        if(Status != STATUS_SUCCESS)
+        {
+            /* XSDP not found, query cache for RSDP table */
+            Status = HlpQueryAcpiCache(ACPI_RSDT_SIGNATURE, (PACPI_DESCRIPTION_HEADER*)&Rsdt);
+        }
+
+        /* Check if XSDT or RSDT table found */
+        if(Status != STATUS_SUCCESS)
+        {
+            /* Failed to get XSDT/RSDT, return error */
+            return Status;
+        }
+
+        /* Get table count depending on root table type */
+        if(Xsdt != NULL)
+        {
+            /* Get table count from XSDT */
+            TableCount = (Xsdt->Header.Length - sizeof(ACPI_DESCRIPTION_HEADER)) / 8;
+        }
+        else
+        {
+            /* Get table count from RSDT */
+            TableCount = (Rsdt->Header.Length - sizeof(ACPI_DESCRIPTION_HEADER)) / 4;
+        }
+
+        /* Iterate over all ACPI tables */
+        for(TableIndex = 0; TableIndex < TableCount; TableIndex++)
+        {
+            /* Check if XSDP or RSDT is used */
+            if(Xsdt != NULL)
+            {
+                /* Get table header physical address from XSDT */
+                TableAddress.QuadPart = Xsdt->Tables[TableIndex];
+            }
+            else
+            {
+                /* Get table header physical address from RSDT */
+                TableAddress.LowPart = Rsdt->Tables[TableIndex];
+                TableAddress.HighPart = 0;
+            }
+
+            /* Check whether some table is already mapped */
+            if(TableHeader != NULL)
+            {
+                /* Unmap previous table */
+                MmUnmapHardwareMemory(TableHeader, 2, TRUE);
+            }
+
+            /* Map table using hardware memory pool */
+            Status = MmMapHardwareMemory(TableAddress, 2, TRUE, (PVOID*)&TableHeader);
+            if(Status != STATUS_SUCCESS)
+            {
+                /* Failed to map table, return error */
+                return Status;
+            }
+
+            /* Verify table signature */
+            if(TableHeader->Signature == Signature)
+            {
+                /* Found requested ACPI table */
+                break;
+            }
+        }
+    }
+
+    /* Make sure table was found */
+    if(TableHeader->Signature != Signature)
+    {
+        /* ACPI table not found, check if cleanup is needed */
+        if(TableHeader != NULL)
+        {
+            /* Unmap non-matching ACPI table */
+            MmUnmapHardwareMemory(TableHeader, 2, TRUE);
+        }
+
+        /* Return error */
+        return STATUS_NOT_FOUND;
+    }
+
+    /* Don't validate FADT on old, broken firmwares with ACPI 2.0 or older */
+    if(TableHeader->Signature != ACPI_FADT_SIGNATURE || TableHeader->Revision > 2)
+    {
+        /* Validate table checksum */
+        if(!HlpValidateAcpiTable(TableHeader, TableHeader->Length))
+        {
+            /* Checksum mismatch, unmap table and return error */
+            MmUnmapHardwareMemory(TableHeader, 2, TRUE);
+            return STATUS_CRC_ERROR;
+        }
+    }
+
+    /* Calculate the length of ACPI table and remap it if needed */
+    TablePages = (((ULONG_PTR)TableHeader & (MM_PAGE_SIZE - 1)) + TableHeader->Length + (MM_PAGE_SIZE - 1)) >> MM_PAGE_SHIFT;
+    if(TablePages != 2)
+    {
+        /* ACPI table needs less or more than 2 pages, remap it */
+        MmUnmapHardwareMemory(TableHeader, 2, FALSE);
+        Status = MmMapHardwareMemory(TableAddress, TablePages, TRUE, (PVOID*)&TableHeader);
+
+        /* Make sure remapping was successful */
+        if(Status != STATUS_SUCCESS)
+        {
+            /* Remapping failed, return error */
+            return STATUS_INSUFFICIENT_RESOURCES;
+        }
+    }
+
+    /* Mark table as write through and store it in local cache */
+    MmMarkHardwareMemoryWriteThrough(TableHeader, TablePages);
+    HlpCacheAcpiTable(TableHeader);
+
+    /* Store ACPI table and return success */
+    *AcpiTable = TableHeader;
+    return STATUS_SUCCESS;
+}
+
+/**
+ * Validates given ACPI table by calculating its checksum.
+ *
+ * @param Buffer
+ *        Supplies a pointer to the table to checksum.
+ *
+ * @param Size
+ *        Supplies the size of the table, in bytes.
+ *
+ * @return This routine returns TRUE if the table is valid, or FALSE otherwise.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+BOOLEAN
+HlpValidateAcpiTable(IN PVOID Buffer,
+                     IN UINT_PTR Size)
+{
+    PUCHAR Pointer;
+    UCHAR Sum;
+
+    /* Initialize variables */
+    Sum = 0;
+    Pointer = Buffer;
+
+    /* Calculate checksum of given table */
+    while(Size != 0)
+    {
+        Sum = (UCHAR)(Sum + *Pointer);
+        Pointer += 1;
+        Size -= 1;
+    }
+
+    /* Return calculated checksum */
+    return (Sum == 0) ? TRUE : FALSE;
+}

xtoskrnl/hl/amd64/ioport.c

@@ -24,9 +24,9 @@ UCHAR
 HlIoPortInByte(IN USHORT Port)
 {
     UCHAR Value;
-    asm volatile("inb %1, %0"
-                 : "=a" (Value)
-                 : "Nd" (Port));
+    __asm__ volatile("inb %1, %0"
+                     : "=a" (Value)
+                     : "Nd" (Port));
     return Value;
 }
 
@@ -45,9 +45,9 @@ ULONG
 HlIoPortInLong(IN USHORT Port)
 {
     ULONG Value;
-    asm volatile("inl %1, %0"
-                 : "=a" (Value)
-                 : "Nd" (Port));
+    __asm__ volatile("inl %1, %0"
+                     : "=a" (Value)
+                     : "Nd" (Port));
     return Value;
 }
 
@@ -66,9 +66,9 @@ USHORT
 HlIoPortInShort(IN USHORT Port)
 {
     USHORT Value;
-    asm volatile("inw %1, %0"
-                 : "=a" (Value)
-                 : "Nd" (Port));
+    __asm__ volatile("inw %1, %0"
+                     : "=a" (Value)
+                     : "Nd" (Port));
     return Value;
 }
 
@@ -90,10 +90,10 @@ VOID
 HlIoPortOutByte(IN USHORT Port,
                 IN UCHAR Value)
 {
-    asm volatile("outb %0, %1"
-                 :
-                 : "a" (Value),
-                   "Nd" (Port));
+    __asm__ volatile("outb %0, %1"
+                     :
+                     : "a" (Value),
+                       "Nd" (Port));
 }
 
 /**
@@ -114,10 +114,10 @@ VOID
 HlIoPortOutLong(IN USHORT Port,
                 IN ULONG Value)
 {
-    asm volatile("outl %0, %1"
-                 :
-                 : "a" (Value),
-                   "Nd" (Port));
+    __asm__ volatile("outl %0, %1"
+                     :
+                     : "a" (Value),
+                       "Nd" (Port));
 }
 
 /**
@@ -138,8 +138,8 @@ VOID
 HlIoPortOutShort(IN USHORT Port,
                  IN USHORT Value)
 {
-    asm volatile("outw %0, %1"
-                 :
-                 : "a" (Value),
-                   "Nd" (Port));
+    __asm__ volatile("outw %0, %1"
+                     :
+                     : "a" (Value),
+                       "Nd" (Port));
 }

xtoskrnl/hl/cport.c

@@ -9,9 +9,6 @@
 #include <xtos.h>
 
 
-/* COM port I/O addresses */
-ULONG ComPortAddress[] = COMPORT_ADDRESSES;
-
 /**
  * This routine gets a byte from serial port.
  *
@@ -168,7 +165,6 @@ HlComPortReadLsr(IN PCPPORT Port,
                  IN UCHAR Byte)
 {
     UCHAR Lsr, Msr;
-    STATIC UCHAR RingFlag;
 
     /* Read the Line Status Register (LSR) */
     Lsr = HlIoPortInByte(PtrToUshort(Port->Address + (ULONG)COMPORT_REG_LSR));
@@ -178,8 +174,8 @@ HlComPortReadLsr(IN PCPPORT Port,
     {
         /* Check Modem Status Register (MSR) for ring indicator */
         Msr = HlIoPortInByte(PtrToUshort(Port->Address + (ULONG)COMPORT_REG_MSR));
-        RingFlag |= (Msr & COMPORT_MSR_RI) ? 1 : 2;
-        if(RingFlag == 3)
+        Port->Ring |= (Msr & COMPORT_MSR_RI) ? 1 : 2;
+        if(Port->Ring == 3)
         {
             /* Ring indicator toggled, use modem control */
             Port->Flags |= COMPORT_FLAG_MC;
@@ -212,37 +208,16 @@ HlComPortReadLsr(IN PCPPORT Port,
 XTCDECL
 XTSTATUS
 HlInitializeComPort(IN OUT PCPPORT Port,
-                    IN ULONG PortNumber,
                     IN PUCHAR PortAddress,
                     IN ULONG BaudRate)
 {
-    USHORT Flags = 0;
-    UCHAR Byte = 0;
-    PUCHAR Address;
+    USHORT Flags;
+    UCHAR Byte;
     ULONG Mode;
 
-    /* We support only a pre-defined number of ports */
-    if(PortNumber > ARRAY_SIZE(ComPortAddress))
-    {
-        /* Fail if wrong/unsupported port used */
-        return STATUS_INVALID_PARAMETER;
-    }
-
-    /* Check if serial port is set */
-    if(PortNumber == 0)
-    {
-        /* Check if port address supplied instead */
-        if(PortAddress)
-        {
-            /* Set custom port address */
-            ComPortAddress[PortNumber] = PtrToUlong(PortAddress);
-        }
-        else
-        {
-            /* Use COM1 by default */
-            PortNumber = 1;
-        }
-    }
+    /* Initialize variables */
+    Byte = 0;
+    Flags = 0;
 
     /* Check if baud rate is set */
     if(BaudRate == 0)
@@ -252,11 +227,8 @@ HlInitializeComPort(IN OUT PCPPORT Port,
         Flags |= COMPORT_FLAG_DBR;
     }
 
-    /* Store COM pointer */
-    Address = UlongToPtr(ComPortAddress[PortNumber]);
-
     /* Check whether this port is not already initialized */
-    if((Port->Address == Address) && (Port->Baud == BaudRate))
+    if((Port->Address == PortAddress) && (Port->Baud == BaudRate))
     {
         return STATUS_SUCCESS;
     }
@@ -265,8 +237,8 @@ HlInitializeComPort(IN OUT PCPPORT Port,
     do
     {
         /* Check whether the 16450/16550 scratch register exists */
-        HlIoPortOutByte(PtrToUshort(Address + (ULONG)COMPORT_REG_SR), Byte);
-        if(HlIoPortInByte(PtrToUshort(Address + (ULONG)COMPORT_REG_SR)) != Byte)
+        HlIoPortOutByte(PtrToUshort(PortAddress + (ULONG)COMPORT_REG_SR), Byte);
+        if(HlIoPortInByte(PtrToUshort(PortAddress + (ULONG)COMPORT_REG_SR)) != Byte)
         {
             return STATUS_NOT_FOUND;
         }
@@ -274,42 +246,43 @@ HlInitializeComPort(IN OUT PCPPORT Port,
     while(++Byte != 0);
 
     /* Disable interrupts */
-    HlIoPortOutByte(PtrToUshort(Address + (ULONG)COMPORT_REG_LCR), COMPORT_LSR_DIS);
-    HlIoPortOutByte(PtrToUshort(Address + (ULONG)COMPORT_REG_IER), COMPORT_LSR_DIS);
+    HlIoPortOutByte(PtrToUshort(PortAddress + (ULONG)COMPORT_REG_LCR), COMPORT_LSR_DIS);
+    HlIoPortOutByte(PtrToUshort(PortAddress + (ULONG)COMPORT_REG_IER), COMPORT_LSR_DIS);
 
     /* Enable Divisor Latch Access Bit (DLAB) */
-    HlIoPortOutByte(PtrToUshort(Address + (ULONG)COMPORT_REG_LCR), COMPORT_LCR_DLAB);
+    HlIoPortOutByte(PtrToUshort(PortAddress + (ULONG)COMPORT_REG_LCR), COMPORT_LCR_DLAB);
 
     /* Set baud rate */
     Mode = COMPORT_CLOCK_RATE / BaudRate;
-    HlIoPortOutByte(PtrToUshort(Address + (ULONG)COMPORT_DIV_DLL), (UCHAR)(Mode & 0xFF));
-    HlIoPortOutByte(PtrToUshort(Address + (ULONG)COMPORT_DIV_DLM), (UCHAR)((Mode >> 8) & 0xFF));
+    HlIoPortOutByte(PtrToUshort(PortAddress + (ULONG)COMPORT_DIV_DLL), (UCHAR)(Mode & 0xFF));
+    HlIoPortOutByte(PtrToUshort(PortAddress + (ULONG)COMPORT_DIV_DLM), (UCHAR)((Mode >> 8) & 0xFF));
 
     /* Set 8 data bits, 1 stop bits, no parity (8n1) */
-    HlIoPortOutByte(PtrToUshort(Address + (ULONG)COMPORT_REG_LCR),
+    HlIoPortOutByte(PtrToUshort(PortAddress + (ULONG)COMPORT_REG_LCR),
                        COMPORT_LCR_8DATA | COMPORT_LCR_1STOP | COMPORT_LCR_PARN);
 
     /* Enable DTR, RTS and OUT2 */
-    HlIoPortOutByte(PtrToUshort(Address + (ULONG)COMPORT_REG_MCR),
+    HlIoPortOutByte(PtrToUshort(PortAddress + (ULONG)COMPORT_REG_MCR),
                        COMPORT_MCR_DTR | COMPORT_MCR_RTS | COMPORT_MCR_OUT2);
 
     /* Enable FIFO */
-    HlIoPortOutByte(PtrToUshort(Address + (ULONG)COMPORT_REG_FCR),
+    HlIoPortOutByte(PtrToUshort(PortAddress + (ULONG)COMPORT_REG_FCR),
                        COMPORT_FCR_ENABLE | COMPORT_FCR_RCVR_RESET | COMPORT_FCR_TXMT_RESET);
 
     /* Mark port as fully initialized */
     Flags |= COMPORT_FLAG_INIT;
 
     /* Make sure port works in Normal Operation Mode (NOM) */
-    HlIoPortOutByte(PtrToUshort(Address + (ULONG)COMPORT_REG_MCR), COMPORT_MCR_NOM);
+    HlIoPortOutByte(PtrToUshort(PortAddress + (ULONG)COMPORT_REG_MCR), COMPORT_MCR_NOM);
 
     /* Read junk data out of the Receive Buffer Register (RBR) */
-    HlIoPortInByte(PtrToUshort(Port->Address + (ULONG)COMPORT_REG_RBR));
+    HlIoPortInByte(PtrToUshort(PortAddress + (ULONG)COMPORT_REG_RBR));
 
     /* Store port details */
-    Port->Address = Address;
+    Port->Address = PortAddress;
     Port->Baud = BaudRate;
     Port->Flags = Flags;
+    Port->Ring = 0;
 
     /* Return success */
     return STATUS_SUCCESS;

xtoskrnl/hl/efifb.c

@@ -1,291 +0,0 @@
-/**
- * PROJECT:         ExectOS
- * COPYRIGHT:       See COPYING.md in the top level directory
- * FILE:            xtoskrnl/hl/efifb.c
- * DESCRIPTION:     EFI framebuffer support
- * DEVELOPERS:      Rafal Kupiec <belliash@codingworkshop.eu.org>
- */
-
-#include <xtos.h>
-#include <xtfont.h>
-
-
-/**
- * Clears the screen by drawing a filled black box.
- *
- * @return This routine does not return any value.
- *
- * @since XT 1.0
- */
-XTAPI
-VOID
-HlClearScreen(VOID)
-{
-    SIZE_T Line, PositionX, PositionY;
-    PULONG FrameBuf;
-
-    /* Get pointer to frame buffer */
-    FrameBuf = HlpFrameBufferData.Address;
-
-    /* Fill the screen with a black box */
-    for(PositionY = 0; PositionY < HlpFrameBufferData.Height; PositionY++)
-    {
-        Line = PositionY * HlpFrameBufferData.PixelsPerScanLine;
-        for(PositionX = 0; PositionX < HlpFrameBufferData.Width; PositionX++)
-        {
-            FrameBuf[Line + PositionX] = 0x00000000;
-        }
-    }
-}
-
-/**
- * Draw a pixel on the screen at the given position and color.
- *
- * @param PositionX
- *        Supplies the X coordinate of the pixel.
- *
- * @param PositionY
- *        Supplies the Y coordinate of the pixel.
- *
- * @param Color
- *        Specifies the color of the pixel.
- *
- * @return This routine does not return any value.
- *
- * @since XT 1.0
- */
-XTAPI
-VOID
-HlDrawPixel(IN ULONG PositionX,
-            IN ULONG PositionY,
-            IN ULONG Color)
-{
-    SIZE_T FrameBufferIndex;
-
-    /* Make sure frame buffer is already initialized */
-    if(HlpFrameBufferData.Initialized == FALSE)
-    {
-        /* Unable to operate on non-initialized frame buffer */
-        return;
-    }
-
-    /* Make sure point is not offscreen */
-    if(PositionX >= HlpFrameBufferData.Width || PositionY >= HlpFrameBufferData.Height || Color > 0xFFFFFFFF)
-    {
-        /* Invalid pixel position or color given */
-        return;
-    }
-
-    /* Calculate the index of the pixel in the frame buffer memory using the provided x and y coordinates */
-    FrameBufferIndex = 4 * HlpFrameBufferData.PixelsPerScanLine * PositionY + 4 * PositionX;
-
-    /* Set the color of the pixel by writing to the corresponding memory location */
-    *((PULONG)(HlpFrameBufferData.Address + FrameBufferIndex)) = Color;
-}
-
-/**
- * Initializes frame buffer display.
- *
- * @return This routine returns a status code.
- *
- * @since XT 1.0
- */
-XTAPI
-XTSTATUS
-HlInitializeFrameBuffer(VOID)
-{
-    /* Check if display already initialized */
-    if(HlpFrameBufferData.Initialized)
-    {
-        /* Nothing to do */
-        return STATUS_SUCCESS;
-    }
-
-    /* Check if framebuffer initialized by bootloader */
-    if(!KeInitializationBlock->LoaderInformation.FrameBuffer.Initialized ||
-       !KeInitializationBlock->LoaderInformation.FrameBuffer.Address)
-    {
-        /* Display not initialized */
-        return STATUS_DEVICE_NOT_READY;
-    }
-
-    /* Check if custom font provided by bootloader */
-    if(KeInitializationBlock->LoaderInformation.FrameBuffer.Font)
-    {
-        /* Use custom font */
-        HlpFrameBufferData.Font = KeInitializationBlock->LoaderInformation.FrameBuffer.Font;
-    }
-    else
-    {
-        /* Use default font */
-        HlpFrameBufferData.Font = (PVOID)&XtFbDefaultFont;
-    }
-
-    /* Save framebuffer information and mark display as initialized */
-    HlpFrameBufferData.Address = KeInitializationBlock->LoaderInformation.FrameBuffer.Address;
-    HlpFrameBufferData.Width = KeInitializationBlock->LoaderInformation.FrameBuffer.Width;
-    HlpFrameBufferData.Height = KeInitializationBlock->LoaderInformation.FrameBuffer.Height;
-    HlpFrameBufferData.BitsPerPixel = KeInitializationBlock->LoaderInformation.FrameBuffer.BitsPerPixel;
-    HlpFrameBufferData.PixelsPerScanLine = KeInitializationBlock->LoaderInformation.FrameBuffer.PixelsPerScanLine;
-    HlpFrameBufferData.Pitch = KeInitializationBlock->LoaderInformation.FrameBuffer.Pitch;
-    HlpFrameBufferData.Initialized = TRUE;
-
-    /* Clear screen */
-    HlClearScreen();
-
-    /* Return success */
-    return STATUS_SUCCESS;
-}
-
-/**
- * Puts a wide character on the framebuffer at the given position and color using the SSFN font.
- *
- * @param PositionX
- *        Supplies the X coordinate of the character.
- *
- * @param PositionY
- *        Supplies the Y coordinate of the character.
- *
- * @param Color
- *        Supplies the font color.
- *
- * @param WideCharacter
- *        Supplies the wide character to be drawn on the framebuffer.
- *
- * @return This routine does not return any value.
- *
- * @since XT 1.0
- */
-XTAPI
-VOID
-HlPutCharacter(IN ULONG PositionX,
-               IN ULONG PositionY,
-               IN ULONG Color,
-               IN WCHAR WideCharacter)
-{
-    UINT CurrentFragment, Glyph, GlyphLimit, Index, Line, Mapping;
-    PUCHAR Character, CharacterMapping, Fragment;
-    UINT_PTR GlyphPixel, Pixel;
-    PSSFN_FONT_HEADER FbFont;
-
-    /* Get pointers to font data */
-    FbFont = (PSSFN_FONT_HEADER)HlpFrameBufferData.Font;
-    CharacterMapping = (PUCHAR)FbFont + FbFont->CharactersOffset;
-
-    /* Find the character in the font's character table */
-    Character = 0;
-    for(Index = 0; Index < 0x110000; Index++)
-    {
-        if(CharacterMapping[0] == 0xFF)
-        {
-            /* Skip 65535 code points */
-            Index += 65535;
-            CharacterMapping++;
-        }
-        else if((CharacterMapping[0] & 0xC0) == 0xC0)
-        {
-            /* Skip (N << 8 + additional byte) + 1 code points (up to 16128) */
-            Index += (((CharacterMapping[0] & 0x3F) << 8) | CharacterMapping[1]);
-            CharacterMapping += 2;
-        }
-        else if((CharacterMapping[0] & 0xC0) == 0x80)
-        {
-            /* Skip N + 1 code points (up to 64) */
-            Index += (CharacterMapping[0] & 0x3F);
-            CharacterMapping++;
-        }
-        else
-        {
-            /* There's a glyph for this character, check if it matches */
-            if(Index == WideCharacter)
-            {
-                /* Found the character, break loop */
-                Character = CharacterMapping;
-                break;
-            }
-
-            /* Move to next character table entry */
-            CharacterMapping += (6 + CharacterMapping[1] * (CharacterMapping[0] & 0x40 ? 6 : 5));
-        }
-    }
-
-    /* Make sure the character has been found in the font */
-    if(!Character)
-    {
-        /* Character not found, don't draw anything */
-        return;
-    }
-
-    /* Find the glyph position on the frame buffer */
-    GlyphPixel = (UINT_PTR)HlpFrameBufferData.Address + PositionY * HlpFrameBufferData.Pitch + PositionX * 4;
-
-    /* Check all kerning fragments */
-    Mapping = 0;
-    CharacterMapping = Character + 6;
-    for(Index = 0; Index < Character[1]; Index++)
-    {
-        /* Check if number of fragments is not exceeded */
-        if(CharacterMapping[0] == 255 && CharacterMapping[1] == 255)
-        {
-            /* Get next mapping */
-            continue;
-        }
-
-        /* Get pointer to fragment */
-        Fragment = (PUCHAR)FbFont + (CharacterMapping[2] |
-                                     (CharacterMapping[3] << 8) |
-                                     (CharacterMapping[4] << 16) |
-                                     ((Character[0] & 0x40) ? (CharacterMapping[5] << 24) : 0));
-
-        /* Check if fragment is printable */
-        if((Fragment[0] & 0xE0) != 0x80)
-        {
-            /* Skip fragment */
-            continue;
-        }
-
-        /* Get initial glyph line */
-        GlyphPixel += (CharacterMapping[1] - Mapping) * HlpFrameBufferData.Pitch;
-        Mapping = CharacterMapping[1];
-
-        /* Extract glyph data from fragments table and advance */
-        Glyph = ((Fragment[0] & 0x1F) + 1) << 3;
-        GlyphLimit = Fragment[1] + 1;
-        Fragment += 2;
-
-        /* Look for kerning group for next code point */
-        CurrentFragment = 1;
-        while(GlyphLimit--)
-        {
-            Pixel = GlyphPixel;
-            for(Line = 0; Line < Glyph; Line++)
-            {
-                /* Decode compressed offsets */
-                if(CurrentFragment > 0x80)
-                {
-                    /* Advance to next fragment */
-                    Fragment++;
-                    CurrentFragment = 1;
-                }
-
-                /* Check if pixel should be drawn */
-                if(*Fragment & CurrentFragment)
-                {
-                    /* Draw glyph pixel */
-                    *((PULONG)Pixel) = Color;
-                }
-
-                /* Advance pixel pointer */
-                Pixel += 4;
-                CurrentFragment <<= 1;
-            }
-
-            /* Advance to next line and increase mapping */
-            GlyphPixel += HlpFrameBufferData.Pitch;
-            Mapping++;
-        }
-
-        /* Get next mapping */
-        CharacterMapping += Character[0] & 0x40 ? 6 : 5;
-    }
-}

xtoskrnl/hl/fbdev.c

@@ -0,0 +1,592 @@
+/**
+ * PROJECT:         ExectOS
+ * COPYRIGHT:       See COPYING.md in the top level directory
+ * FILE:            xtoskrnl/hl/fbdev.c
+ * DESCRIPTION:     FrameBuffer support
+ * DEVELOPERS:      Rafal Kupiec <belliash@codingworkshop.eu.org>
+ *                  Aiken Harris <harraiken91@gmail.com>
+ */
+
+#include <xtos.h>
+#include <xtfont.h>
+
+
+/**
+ * Clears the screen by drawing a box filled with specified color.
+ *
+ * @param Color
+ *        Specifies the color of the box used to fill the screen in (A)RGB format.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+VOID
+HlClearScreen(IN ULONG Color)
+{
+    ULONG PositionX, PositionY;
+    ULONG BackgroundColor;
+    PCHAR CurrentLine;
+    PULONG Pixel;
+
+    /* Make sure frame buffer is already initialized */
+    if(HlpFrameBufferData.Initialized == FALSE)
+    {
+        /* Unable to operate on non-initialized frame buffer */
+        return;
+    }
+
+    /* Convert background color and get pointer to frame buffer */
+    BackgroundColor = HlpRGBColor(Color);
+    CurrentLine = HlpFrameBufferData.Address;
+
+    /* Fill the screen with the specified color */
+    for(PositionY = 0; PositionY < HlpFrameBufferData.Height; PositionY++, CurrentLine += HlpFrameBufferData.Pitch)
+    {
+        /* Fill the current line with the specified color */
+        Pixel = (PULONG)CurrentLine;
+        for(PositionX = 0; PositionX < HlpFrameBufferData.Width; PositionX++)
+        {
+            /* Set the color of the pixel */
+            Pixel[PositionX] = BackgroundColor;
+        }
+    }
+}
+
+/**
+ * Displays a single character at the current cursor position inside the scroll region.
+ *
+ * @param Character
+ *        Supplies the character to be displayed.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTCDECL
+XTSTATUS
+HlDisplayCharacter(IN WCHAR Character)
+{
+    PSSFN_FONT_HEADER FbFont;
+
+    /* Make sure frame buffer is already initialized */
+    if(HlpFrameBufferData.Initialized == FALSE)
+    {
+        /* Unable to operate on non-initialized frame buffer */
+        return STATUS_DEVICE_NOT_READY;
+    }
+
+    /* Get font information */
+    FbFont = (PSSFN_FONT_HEADER)HlpFrameBufferData.Font;
+
+    /* Handle special characters */
+    switch(Character)
+    {
+        case L'\n':
+            /* Move cursor to the beginning of the next line */
+            HlpScrollRegionData.CursorX = HlpScrollRegionData.Left;
+            HlpScrollRegionData.CursorY += FbFont->Height;
+            break;
+        case L'\t':
+            /* Move cursor to the next tab stop */
+            HlpScrollRegionData.CursorX += (8 - (HlpScrollRegionData.CursorX - HlpScrollRegionData.Left) / FbFont->Width % 8) * FbFont->Width;
+            if (HlpScrollRegionData.CursorX >= HlpScrollRegionData.Right)
+            {
+                HlpScrollRegionData.CursorX = HlpScrollRegionData.Left;
+                HlpScrollRegionData.CursorY += FbFont->Height;
+            }
+            break;
+        default:
+            /* Draw the character */
+            HlpDrawCharacter(HlpScrollRegionData.CursorX, HlpScrollRegionData.CursorY, HlpScrollRegionData.TextColor, Character);
+
+            /* Advance cursor */
+            HlpScrollRegionData.CursorX += FbFont->Width;
+
+            /* Check if cursor reached end of line */
+            if(HlpScrollRegionData.CursorX >= HlpScrollRegionData.Right)
+            {
+                HlpScrollRegionData.CursorX = HlpScrollRegionData.Left;
+                HlpScrollRegionData.CursorY += FbFont->Height;
+            }
+            break;
+    }
+
+    /* Check if cursor reached end of scroll region */
+    if(HlpScrollRegionData.CursorY >= HlpScrollRegionData.Bottom)
+    {
+        /* Scroll one line up */
+        HlpScrollRegion();
+        HlpScrollRegionData.CursorY = HlpScrollRegionData.Bottom - FbFont->Height;
+    }
+
+    /* Return success */
+    return STATUS_SUCCESS;
+}
+
+/**
+ * Returns the current resolution of the frame buffer display.
+ *
+ * @param Width
+ *        A pointer to memory area where the screen width will be stored.
+ *
+ * @param Height
+ *        A pointer to memory area where the screen height will be stored.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+VOID
+HlGetFrameBufferResolution(OUT PULONG Width, OUT PULONG Height)
+{
+    *Width = HlpFrameBufferData.Width;
+    *Height = HlpFrameBufferData.Height;
+}
+
+/**
+ * Initializes frame buffer display.
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+XTSTATUS
+HlInitializeFrameBuffer(VOID)
+{
+    PSYSTEM_RESOURCE_FRAMEBUFFER FrameBufferResource;
+    PSYSTEM_RESOURCE_HEADER SystemResource;
+    XTSTATUS Status;
+
+    /* Check if display already initialized */
+    if(HlpFrameBufferData.Initialized)
+    {
+        /* Nothing to do */
+        return STATUS_SUCCESS;
+    }
+
+    /* Get FrameBuffer system resource */
+    Status = KeGetSystemResource(SystemResourceFrameBuffer, &SystemResource);
+    if(Status != STATUS_SUCCESS)
+    {
+        /* Resource not found */
+        return STATUS_NOT_FOUND;
+    }
+
+    /* Cast system resource to FrameBuffer resource */
+    FrameBufferResource = (PSYSTEM_RESOURCE_FRAMEBUFFER)SystemResource;
+
+    /* Check if bootloader provided a framebuffer address */
+    if(!FrameBufferResource->Header.VirtualAddress)
+    {
+        /* Display probably not initialized */
+        return STATUS_DEVICE_NOT_READY;
+    }
+
+    /* Check if bootloader provided a custom font */
+    if(FrameBufferResource->Font)
+    {
+        /* Use custom font */
+        HlpFrameBufferData.Font = FrameBufferResource->Font;
+    }
+    else
+    {
+        /* Use default font */
+        HlpFrameBufferData.Font = (PVOID)&XtFbDefaultFont;
+    }
+
+    /* Save framebuffer information and mark display as initialized */
+    HlpFrameBufferData.Address = FrameBufferResource->Header.VirtualAddress;
+    HlpFrameBufferData.Width = FrameBufferResource->Width;
+    HlpFrameBufferData.Height = FrameBufferResource->Height;
+    HlpFrameBufferData.BytesPerPixel = FrameBufferResource->BitsPerPixel / 8;
+    HlpFrameBufferData.PixelsPerScanLine = FrameBufferResource->PixelsPerScanLine;
+    HlpFrameBufferData.Pitch = FrameBufferResource->Pitch;
+    HlpFrameBufferData.Pixels.BlueShift = FrameBufferResource->Pixels.BlueShift;
+    HlpFrameBufferData.Pixels.BlueSize = FrameBufferResource->Pixels.BlueSize;
+    HlpFrameBufferData.Pixels.GreenShift = FrameBufferResource->Pixels.GreenShift;
+    HlpFrameBufferData.Pixels.GreenSize = FrameBufferResource->Pixels.GreenSize;
+    HlpFrameBufferData.Pixels.RedShift = FrameBufferResource->Pixels.RedShift;
+    HlpFrameBufferData.Pixels.RedSize = FrameBufferResource->Pixels.RedSize;
+    HlpFrameBufferData.Pixels.ReservedShift = FrameBufferResource->Pixels.ReservedShift;
+    HlpFrameBufferData.Pixels.ReservedSize = FrameBufferResource->Pixels.ReservedSize;
+    HlpFrameBufferData.Initialized = TRUE;
+
+    /* Clear screen */
+    HlClearScreen(0x00000000);
+
+    /* Return success */
+    return STATUS_SUCCESS;
+}
+
+/**
+ * Sets the scrollable region of the screen and calculates character dimensions.
+ *
+ * @param Left
+ *        Supplies the left pixel coordinate of the scroll region.
+ *
+ * @param Top
+ *        Supplies the top pixel coordinate of the scroll region.
+ *
+ * @param Right
+ *        Supplies the right pixel coordinate of the scroll region.
+ *
+ * @param Bottom
+ *        Supplies the bottom pixel coordinate of the scroll region.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+VOID
+HlInitializeScrollRegion(IN ULONG Left,
+                         IN ULONG Top,
+                         IN ULONG Right,
+                         IN ULONG Bottom,
+                         IN ULONG FontColor)
+{
+    PSSFN_FONT_HEADER FbFont;
+    PCHAR PixelAddress;
+
+    /* Make sure frame buffer is already initialized */
+    if(HlpFrameBufferData.Initialized == FALSE)
+    {
+        /* Unable to operate on non-initialized frame buffer */
+        return;
+    }
+
+    /* Store pixel coordinates of the scroll region */
+    HlpScrollRegionData.Left = Left;
+    HlpScrollRegionData.Top = Top;
+    HlpScrollRegionData.Right = Right;
+    HlpScrollRegionData.Bottom = Bottom;
+
+    /* Get font information */
+    FbFont = (PSSFN_FONT_HEADER)HlpFrameBufferData.Font;
+
+    /* Validate font information */
+    if(FbFont && FbFont->Width > 0 && FbFont->Height > 0)
+    {
+        /* Calculate character dimensions */
+        HlpScrollRegionData.WidthInChars = (Right - Left) / FbFont->Width;
+        HlpScrollRegionData.HeightInChars = (Bottom - Top) / FbFont->Height;
+
+        /* Ensure the bottom of the scroll region is an exact multiple of the font height */
+        HlpScrollRegionData.Bottom = HlpScrollRegionData.Top + (HlpScrollRegionData.HeightInChars * FbFont->Height);
+    }
+    else
+    {
+        /* Fallback to 0 if font info is not available or invalid */
+        HlpScrollRegionData.WidthInChars = 0;
+        HlpScrollRegionData.HeightInChars = 0;
+    }
+
+    /* Initialize cursor position and font color */
+    HlpScrollRegionData.CursorX = HlpScrollRegionData.Left;
+    HlpScrollRegionData.CursorY = HlpScrollRegionData.Top;
+    HlpScrollRegionData.TextColor = FontColor;
+
+    /* Get the background color by reading the pixel at the top-left corner of the scroll region */
+    PixelAddress = (PCHAR)HlpFrameBufferData.Address + (Top * HlpFrameBufferData.Pitch) +
+                          (Left * HlpFrameBufferData.BytesPerPixel);
+    HlpScrollRegionData.BackgroundColor = *((PULONG)PixelAddress);
+}
+
+/**
+ * Draws a character on the framebuffer at the given position and color using the SSFN font.
+ *
+ * @param PositionX
+ *        Supplies the X coordinate of the character.
+ *
+ * @param PositionY
+ *        Supplies the Y coordinate of the character.
+ *
+ * @param Color
+ *        Supplies the font color in (A)RGB format.
+ *
+ * @param WideCharacter
+ *        Supplies the wide character to be drawn on the framebuffer.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+VOID
+HlpDrawCharacter(IN ULONG PositionX,
+                 IN ULONG PositionY,
+                 IN ULONG Color,
+                 IN WCHAR WideCharacter)
+{
+    UINT CurrentFragment, Glyph, GlyphLimit, Index, Line, Mapping;
+    PUCHAR Character, CharacterMapping, Fragment;
+    UINT_PTR GlyphPixel, Pixel;
+    PSSFN_FONT_HEADER FbFont;
+    ULONG FontColor;
+
+    /* Make sure frame buffer is already initialized */
+    if(HlpFrameBufferData.Initialized == FALSE)
+    {
+        /* Unable to operate on non-initialized frame buffer */
+        return;
+    }
+
+    /* Get pointers to font data */
+    FbFont = (PSSFN_FONT_HEADER)HlpFrameBufferData.Font;
+    CharacterMapping = (PUCHAR)FbFont + FbFont->CharactersOffset;
+
+    /* Find the character in the font's character table */
+    Character = 0;
+    for(Index = 0; Index < 0x110000; Index++)
+    {
+        if(CharacterMapping[0] == 0xFF)
+        {
+            /* Skip 65535 code points */
+            Index += 65535;
+            CharacterMapping++;
+        }
+        else if((CharacterMapping[0] & 0xC0) == 0xC0)
+        {
+            /* Skip (N << 8 + additional byte) + 1 code points (up to 16128) */
+            Index += (((CharacterMapping[0] & 0x3F) << 8) | CharacterMapping[1]);
+            CharacterMapping += 2;
+        }
+        else if((CharacterMapping[0] & 0xC0) == 0x80)
+        {
+            /* Skip N + 1 code points (up to 64) */
+            Index += (CharacterMapping[0] & 0x3F);
+            CharacterMapping++;
+        }
+        else
+        {
+            /* There's a glyph for this character, check if it matches */
+            if(Index == WideCharacter)
+            {
+                /* Found the character, break loop */
+                Character = CharacterMapping;
+                break;
+            }
+
+            /* Move to next character table entry */
+            CharacterMapping += (6 + CharacterMapping[1] * (CharacterMapping[0] & 0x40 ? 6 : 5));
+        }
+    }
+
+    /* Make sure the character has been found in the font */
+    if(!Character)
+    {
+        /* Character not found, don't draw anything */
+        return;
+    }
+
+    /* Find the glyph position on the frame buffer and set font color */
+    GlyphPixel = (UINT_PTR)HlpFrameBufferData.Address + PositionY * HlpFrameBufferData.Pitch +
+                           PositionX * HlpFrameBufferData.BytesPerPixel;
+    FontColor = HlpRGBColor(Color);
+
+    /* Check all kerning fragments */
+    Mapping = 0;
+    CharacterMapping = Character + 6;
+    for(Index = 0; Index < Character[1]; Index++)
+    {
+        /* Check if number of fragments is not exceeded */
+        if(CharacterMapping[0] == 255 && CharacterMapping[1] == 255)
+        {
+            /* Get next mapping */
+            continue;
+        }
+
+        /* Get pointer to fragment */
+        Fragment = (PUCHAR)FbFont + (CharacterMapping[2] |
+                                     (CharacterMapping[3] << 8) |
+                                     (CharacterMapping[4] << 16) |
+                                     ((Character[0] & 0x40) ? (CharacterMapping[5] << 24) : 0));
+
+        /* Check if fragment is printable */
+        if((Fragment[0] & 0xE0) != 0x80)
+        {
+            /* Skip fragment */
+            continue;
+        }
+
+        /* Get initial glyph line */
+        GlyphPixel += (CharacterMapping[1] - Mapping) * HlpFrameBufferData.Pitch;
+        Mapping = CharacterMapping[1];
+
+        /* Extract glyph data from fragments table and advance */
+        Glyph = ((Fragment[0] & 0x1F) + 1) << 3;
+        GlyphLimit = Fragment[1] + 1;
+        Fragment += 2;
+
+        /* Look for kerning group for next code point */
+        CurrentFragment = 1;
+        while(GlyphLimit--)
+        {
+            Pixel = GlyphPixel;
+            for(Line = 0; Line < Glyph; Line++)
+            {
+                /* Decode compressed offsets */
+                if(CurrentFragment > 0x80)
+                {
+                    /* Advance to next fragment */
+                    Fragment++;
+                    CurrentFragment = 1;
+                }
+
+                /* Check if pixel should be drawn */
+                if(*Fragment & CurrentFragment)
+                {
+                    /* Draw glyph pixel */
+                    *((PULONG)Pixel) = FontColor;
+                }
+
+                /* Advance pixel pointer */
+                Pixel += HlpFrameBufferData.BytesPerPixel;
+                CurrentFragment <<= 1;
+            }
+
+            /* Advance to next line and increase mapping */
+            GlyphPixel += HlpFrameBufferData.Pitch;
+            Mapping++;
+        }
+
+        /* Get next mapping */
+        CharacterMapping += Character[0] & 0x40 ? 6 : 5;
+    }
+}
+
+/**
+ * Draw a pixel on the screen at the given position and color.
+ *
+ * @param PositionX
+ *        Supplies the X coordinate of the pixel.
+ *
+ * @param PositionY
+ *        Supplies the Y coordinate of the pixel.
+ *
+ * @param Color
+ *        Specifies the color of the pixel in (A)RGB format.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+VOID
+HlpDrawPixel(IN ULONG PositionX,
+             IN ULONG PositionY,
+             IN ULONG Color)
+{
+    PCHAR PixelAddress;
+
+    /* Make sure frame buffer is already initialized */
+    if(HlpFrameBufferData.Initialized == FALSE)
+    {
+        /* Unable to operate on non-initialized frame buffer */
+        return;
+    }
+
+    /* Make sure point is not offscreen */
+    if(PositionX >= HlpFrameBufferData.Width || PositionY >= HlpFrameBufferData.Height || Color > 0xFFFFFFFF)
+    {
+        /* Invalid pixel position or color given */
+        return;
+    }
+
+    /* Calculate the address of the pixel in the frame buffer memory */
+    PixelAddress = (PCHAR)HlpFrameBufferData.Address + (PositionY * HlpFrameBufferData.Pitch) +
+                          (PositionX * HlpFrameBufferData.BytesPerPixel);
+
+    /* Set the color of the pixel by writing to the corresponding memory location */
+    *((PULONG)PixelAddress) = HlpRGBColor(Color);
+}
+
+/**
+ * Converts color format from (A)RGB one expected by current FrameBuffer.
+ *
+ * @param Color
+ *        Specifies the color in (A)RGB format.
+ *
+ * @return Returns the color in FrameBuffer format.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+ULONG
+HlpRGBColor(IN ULONG Color)
+{
+    USHORT Blue, Green, Red, Reserved;
+
+    /* Extract color components from (A)RGB value */
+    Blue = (USHORT)(Color & 0xFF);
+    Green = (USHORT)((Color >> 8) & 0xFF);
+    Red = (USHORT)((Color >> 16) & 0xFF);
+    Reserved = (USHORT)((Color >> 24) & 0xFF);
+
+    /* Return color in FrameBuffer pixel format */
+    return (ULONG)((Blue << HlpFrameBufferData.Pixels.BlueShift) | (Green << HlpFrameBufferData.Pixels.GreenShift) |
+                   (Red << HlpFrameBufferData.Pixels.RedShift) | (Reserved << HlpFrameBufferData.Pixels.ReservedShift));
+}
+
+/**
+ * Scrolls the content of the scroll region up by one line.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+VOID
+HlpScrollRegion(VOID)
+{
+    PCHAR Destination, Source;
+    PSSFN_FONT_HEADER FbFont;
+    ULONG Line, PositionX;
+    ULONG LineBytes;
+    PULONG Pixel;
+
+    /* Make sure frame buffer is already initialized */
+    if(HlpFrameBufferData.Initialized == FALSE)
+    {
+        /* Unable to operate on non-initialized frame buffer */
+        return;
+    }
+
+    /* Get font information */
+    FbFont = (PSSFN_FONT_HEADER)HlpFrameBufferData.Font;
+
+    /* Calculate bytes per line in the scroll region */
+    LineBytes = (HlpScrollRegionData.Right - HlpScrollRegionData.Left) * HlpFrameBufferData.BytesPerPixel;
+
+    /* Scroll up each scan line in the scroll region */
+    for(Line = HlpScrollRegionData.Top; Line < HlpScrollRegionData.Bottom - FbFont->Height; Line++)
+    {
+        Destination = (PCHAR)HlpFrameBufferData.Address + Line * HlpFrameBufferData.Pitch +
+                      HlpScrollRegionData.Left * HlpFrameBufferData.BytesPerPixel;
+
+        /* The source is one full text line (FbFont->Height) below the destination */
+        Source = (PCHAR)HlpFrameBufferData.Address + (Line + FbFont->Height) * HlpFrameBufferData.Pitch +
+                 HlpScrollRegionData.Left * HlpFrameBufferData.BytesPerPixel;
+
+        /* Move each scan line in the scroll region up */
+        RtlMoveMemory(Destination, Source, LineBytes);
+    }
+
+    /* Clear the last text line */
+    for(Line = HlpScrollRegionData.Bottom - FbFont->Height; Line < HlpScrollRegionData.Bottom; Line++)
+    {
+        /* Get pointer to the start of the scan line to clear */
+        Pixel = (PULONG)((PCHAR)HlpFrameBufferData.Address + Line * HlpFrameBufferData.Pitch +
+                         HlpScrollRegionData.Left * HlpFrameBufferData.BytesPerPixel);
+
+        /* Clear each pixel in the scan line with the background color */
+        for(PositionX = 0; PositionX < (HlpScrollRegionData.Right - HlpScrollRegionData.Left); PositionX++)
+        {
+            Pixel[PositionX] = HlpScrollRegionData.BackgroundColor;
+        }
+    }
+}

xtoskrnl/hl/globals.c

@@ -9,14 +9,26 @@
 #include <xtos.h>
 
 
+/* ACPI tables cache list */
+LIST_ENTRY HlpAcpiCacheList;
+
+/* ACPI Root System Description Pointer (RSDP) */
+PACPI_RSDP HlpAcpiRsdp;
+
+/* ACPI timer information */
+ACPI_TIMER_INFO HlpAcpiTimerInfo;
+
 /* Active processors count */
 KAFFINITY HlpActiveProcessors;
 
 /* APIC mode */
-HAL_APIC_MODE HlpApicMode;
+APIC_MODE HlpApicMode;
 
 /* FrameBuffer information */
-HAL_FRAMEBUFFER_DATA HlpFrameBufferData;
+HL_FRAMEBUFFER_DATA HlpFrameBufferData;
 
-/* Processors identity table */
-HAL_PROCESSOR_IDENTITY HlpProcessorsIdentity[MAXIMUM_PROCESSORS] = {{0}};
+/* Scroll region information */
+HL_SCROLL_REGION_DATA HlpScrollRegionData;
+
+/* System information */
+ACPI_SYSTEM_INFO HlpSystemInfo;

xtoskrnl/hl/i686/ioport.c

@@ -24,9 +24,9 @@ UCHAR
 HlIoPortInByte(IN USHORT Port)
 {
     UCHAR Value;
-    asm volatile("inb %1, %0"
-                 : "=a" (Value)
-                 : "Nd" (Port));
+    __asm__ volatile("inb %1, %0"
+                     : "=a" (Value)
+                     : "Nd" (Port));
     return Value;
 }
 
@@ -45,9 +45,9 @@ ULONG
 HlIoPortInLong(IN USHORT Port)
 {
     ULONG Value;
-    asm volatile("inl %1, %0"
-                 : "=a" (Value)
-                 : "Nd" (Port));
+    __asm__ volatile("inl %1, %0"
+                     : "=a" (Value)
+                     : "Nd" (Port));
     return Value;
 }
 
@@ -66,9 +66,9 @@ USHORT
 HlIoPortInShort(IN USHORT Port)
 {
     USHORT Value;
-    asm volatile("inw %1, %0"
-                 : "=a" (Value)
-                 : "Nd" (Port));
+    __asm__ volatile("inw %1, %0"
+                     : "=a" (Value)
+                     : "Nd" (Port));
     return Value;
 }
 
@@ -90,10 +90,10 @@ VOID
 HlIoPortOutByte(IN USHORT Port,
                 IN UCHAR Value)
 {
-    asm volatile("outb %0, %1"
-                 :
-                 : "a" (Value),
-                   "Nd" (Port));
+    __asm__ volatile("outb %0, %1"
+                     :
+                     : "a" (Value),
+                       "Nd" (Port));
 }
 
 /**
@@ -114,10 +114,10 @@ VOID
 HlIoPortOutLong(IN USHORT Port,
                 IN ULONG Value)
 {
-    asm volatile("outl %0, %1"
-                 :
-                 : "a" (Value),
-                   "Nd" (Port));
+    __asm__ volatile("outl %0, %1"
+                     :
+                     : "a" (Value),
+                       "Nd" (Port));
 }
 
 /**
@@ -138,8 +138,8 @@ VOID
 HlIoPortOutShort(IN USHORT Port,
                  IN USHORT Value)
 {
-    asm volatile("outw %0, %1"
-                 :
-                 : "a" (Value),
-                   "Nd" (Port));
+    __asm__ volatile("outw %0, %1"
+                     :
+                     : "a" (Value),
+                       "Nd" (Port));
 }

xtoskrnl/hl/init.c

@@ -0,0 +1,41 @@
+/**
+ * PROJECT:         ExectOS
+ * COPYRIGHT:       See COPYING.md in the top level directory
+ * FILE:            xtoskrnl/hl/init.c
+ * DESCRIPTION:     Hardware layer initialization code
+ * DEVELOPERS:      Rafal Kupiec <belliash@codingworkshop.eu.org>
+ */
+
+#include <xtos.h>
+
+
+/**
+ * Initializes the hardware layer subsystem
+ *
+ * @return This routine returns a status code.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+XTSTATUS
+HlInitializeSystem(VOID)
+{
+    XTSTATUS Status;
+
+    /* Initialize ACPI */
+    Status = HlpInitializeAcpi();
+    if(Status != STATUS_SUCCESS)
+    {
+        return Status;
+    }
+
+    /* Get system information from ACPI */
+    Status = HlpInitializeAcpiSystemInformation();
+    if(Status != STATUS_SUCCESS)
+    {
+        return Status;
+    }
+
+    /* Return success */
+    return STATUS_SUCCESS;
+}

xtoskrnl/hl/x86/cpu.c

@@ -12,6 +12,9 @@
 /**
  * Initializes the processor.
  *
+ * @param CpuNumber
+ *        Supplies the number of the CPU, that is being initialized.
+ *
  * @return This routine does not return any value.
  *
  * @since XT 1.0
@@ -22,25 +25,22 @@ HlInitializeProcessor(VOID)
 {
     PKPROCESSOR_BLOCK ProcessorBlock;
     KAFFINITY Affinity;
-    ULONG CpuNumber = 0;
 
     /* Get current processor block */
     ProcessorBlock = KeGetCurrentProcessorBlock();
 
-    /* Set initial stall factor */
+    /* Set initial stall factor, CPU number and mask interrupts */
     ProcessorBlock->StallScaleFactor = INITIAL_STALL_FACTOR;
-
-    /* Record processor block in the processors table */
-    HlpProcessorsIdentity[CpuNumber].ProcessorBlock = ProcessorBlock;
+    ProcessorBlock->Idr = 0xFFFFFFFF;
 
     /* Set processor affinity */
-    Affinity = (KAFFINITY) 1 << CpuNumber;
+    Affinity = (KAFFINITY) 1 << ProcessorBlock->CpuNumber;
 
     /* Apply affinity to a set of processors */
     HlpActiveProcessors |= Affinity;
 
     /* Initialize APIC for this processor */
-    HlpInitializeApic();
+    HlpInitializePic();
 
     /* Set the APIC running level */
     HlSetRunLevel(KeGetCurrentProcessorBlock()->RunLevel);

xtoskrnl/hl/x86/pic.c

@@ -4,7 +4,6 @@
  * FILE:            xtoskrnl/hl/x86/pic.c
  * DESCRIPTION:     Programmable Interrupt Controller (PIC) for x86 (i686/AMD64) support
  * DEVELOPERS:      Rafal Kupiec <belliash@codingworkshop.eu.org>
- *                  Jozef Nagy <schkwve@gmail.com>
  */
 
 #include <xtos.h>
@@ -25,21 +24,6 @@ HlClearApicErrors(VOID)
     HlWriteApicRegister(APIC_ESR, 0);
 }
 
-/**
- * Disables the legacy 8259 Programmable Interrupt Controller (PIC).
- *
- * @return This routine does not return any value.
- *
- * @since XT 1.0
- */
-XTAPI
-VOID
-HlDisableLegacyPic(VOID)
-{
-    HlIoPortOutByte(PIC1_DATA_PORT, 0xFF);
-    HlIoPortOutByte(PIC2_DATA_PORT, 0xFF);
-}
-
 /**
  * Reads from the APIC register.
  *
@@ -51,7 +35,7 @@ HlDisableLegacyPic(VOID)
  * @since XT 1.0
  */
 XTFASTCALL
-ULONG
+ULONGLONG
 HlReadApicRegister(IN APIC_REGISTER Register)
 {
     if(HlpApicMode == APIC_MODE_X2APIC)
@@ -97,7 +81,7 @@ HlSendEoi(VOID)
 XTFASTCALL
 VOID
 HlWriteApicRegister(IN APIC_REGISTER Register,
-                    IN ULONG Value)
+                    IN ULONGLONG Value)
 {
     if(HlpApicMode == APIC_MODE_X2APIC)
     {
@@ -128,7 +112,7 @@ HlpCheckX2ApicSupport(VOID)
     CPUID_REGISTERS CpuRegisters;
 
     /* Prepare CPUID registers */
-    CpuRegisters.Leaf = CPUID_GET_CPU_FEATURES;
+    CpuRegisters.Leaf = CPUID_GET_STANDARD1_FEATURES;
     CpuRegisters.SubLeaf = 0;
     CpuRegisters.Eax = 0;
     CpuRegisters.Ebx = 0;
@@ -149,6 +133,26 @@ HlpCheckX2ApicSupport(VOID)
     return TRUE;
 }
 
+/**
+ * Gets the local APIC ID of the current processor.
+ *
+ * @return This routine returns the current processor's local APIC ID.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+ULONG
+HlpGetCpuApicId(VOID)
+{
+    ULONG ApicId;
+
+    /* Read APIC ID register */
+    ApicId = HlReadApicRegister(APIC_ID);
+
+    /* Return logical CPU ID depending on current APIC mode */
+    return (HlpApicMode == APIC_MODE_COMPAT) ? ((ApicId & 0xFFFFFFFF) >> APIC_XAPIC_LDR_SHIFT) : ApicId;
+}
+
 /**
  * Allows an APIC spurious interrupts to end up.
  *
@@ -181,88 +185,231 @@ HlpHandlePicSpuriousService(VOID)
  * @return This routine does not return any value.
  *
  * @since XT 1.0
- *
- * @todo Register interrupt handlers for spurious vectors.
  */
 XTAPI
 VOID
 HlpInitializeApic(VOID)
 {
+    APIC_SPURIOUS_REGISTER SpuriousRegister;
     APIC_BASE_REGISTER BaseRegister;
     APIC_LVT_REGISTER LvtRegister;
-    APIC_SPURIOUS_REGISTER SpuriousRegister;
-    ULONG CpuNumber = 0;
+    ULONG CpuNumber;
 
-    /* Check if this is an x2APIC compatible machine */
+    /* Determine APIC mode (xAPIC compatibility or x2APIC) */
     if(HlpCheckX2ApicSupport())
     {
-        /* Enable x2APIC */
+        /* Enable x2APIC mode */
         HlpApicMode = APIC_MODE_X2APIC;
     }
     else
     {
-        /* Use xAPIC compatibility mode */
+        /* Fall back to xAPIC compatibility mode */
         HlpApicMode = APIC_MODE_COMPAT;
     }
 
+    /* Get current processor number */
+    CpuNumber = KeGetCurrentProcessorNumber();
+
     /* Enable the APIC */
     BaseRegister.LongLong = ArReadModelSpecificRegister(APIC_LAPIC_MSR_BASE);
     BaseRegister.Enable = 1;
     BaseRegister.ExtendedMode = (HlpApicMode == APIC_MODE_X2APIC);
-    BaseRegister.BootStrapProcessor = 1;
+    BaseRegister.BootStrapProcessor = (CpuNumber == 0) ? 1 : 0;
     ArWriteModelSpecificRegister(APIC_LAPIC_MSR_BASE, BaseRegister.LongLong);
 
-    /* xAPIC compatibility mode specific initialization */
+    /* Mask all interrupts by raising Task Priority Register (TPR)  */
+    HlWriteApicRegister(APIC_TPR, 0xFF);
+
+    /* Perform initialization specific to xAPIC compatibility mode */
     if(HlpApicMode == APIC_MODE_COMPAT)
     {
-        /* Initialize Destination Format Register with flat model */
+        /* Use Flat Model for destination format (not supported in x2APIC) */
         HlWriteApicRegister(APIC_DFR, APIC_DF_FLAT);
 
-        /* Set the logical APIC ID */
+        /* Set the logical APIC ID for this processor (read-only in x2APIC) */
         HlWriteApicRegister(APIC_LDR, (1UL << CpuNumber) << 24);
     }
 
-    /* Set the spurious interrupt vector */
+    /* Configure the spurious interrupt vector */
     SpuriousRegister.Long = HlReadApicRegister(APIC_SIVR);
     SpuriousRegister.Vector = APIC_VECTOR_SPURIOUS;
     SpuriousRegister.SoftwareEnable = 1;
     SpuriousRegister.CoreChecking = 0;
     HlWriteApicRegister(APIC_SIVR, SpuriousRegister.Long);
 
-    /* Initialize Logical Vector Table */
+    /* Setup the LVT Error entry to deliver APIC errors on a dedicated vector */
+    HlWriteApicRegister(APIC_ERRLVTR, APIC_VECTOR_ERROR);
+
+    /* Program the APIC timer for periodic mode */
     LvtRegister.Long = 0;
-    LvtRegister.MessageType = APIC_DM_FIXED;
-    LvtRegister.DeliveryStatus = 0;
-    LvtRegister.RemoteIRR = 0;
+    LvtRegister.Mask = 1;
+    LvtRegister.DeliveryMode = APIC_DM_FIXED;
+    LvtRegister.TimerMode = 1;
     LvtRegister.TriggerMode = APIC_TGM_EDGE;
-    LvtRegister.Mask = 0;
-    LvtRegister.TimerMode = 0;
-
-    /* Mask LVTR_ERROR first, to prevent local APIC error */
-    LvtRegister.Vector = APIC_VECTOR_ERROR;
-    HlWriteApicRegister(APIC_ERRLVTR, LvtRegister.Long);
-
-    /* Mask LVT tables */
-    LvtRegister.Vector = APIC_VECTOR_NMI;
+    LvtRegister.Vector = APIC_VECTOR_PROFILE;
     HlWriteApicRegister(APIC_TMRLVTR, LvtRegister.Long);
-    HlWriteApicRegister(APIC_THRMLVTR, LvtRegister.Long);
+
+    /* Configure the performance counter overflow */
+    LvtRegister.Long = 0;
+    LvtRegister.Mask = 0;
+    LvtRegister.DeliveryMode = APIC_DM_FIXED;
+    LvtRegister.TimerMode = 0;
+    LvtRegister.TriggerMode = APIC_TGM_EDGE;
+    LvtRegister.Vector = APIC_VECTOR_PERF;
     HlWriteApicRegister(APIC_PCLVTR, LvtRegister.Long);
 
-    /* Mask LINT0 */
+    /* Configure the LINT0 pin */
+    LvtRegister.Long = 0;
+    LvtRegister.Mask = 1;
+    LvtRegister.DeliveryMode = APIC_DM_FIXED;
+    LvtRegister.TimerMode = 0;
+    LvtRegister.TriggerMode = APIC_TGM_EDGE;
     LvtRegister.Vector = APIC_VECTOR_SPURIOUS;
-    LvtRegister.MessageType = APIC_DM_EXTINT;
     HlWriteApicRegister(APIC_LINT0, LvtRegister.Long);
 
-    /* Mask LINT1 */
+    /* Configure the LINT1 pin */
+    LvtRegister.Long = 0;
+    LvtRegister.Mask = 0;
+    LvtRegister.DeliveryMode = APIC_DM_NMI;
+    LvtRegister.TimerMode = 0;
+    LvtRegister.TriggerMode = APIC_TGM_EDGE;
     LvtRegister.Vector = APIC_VECTOR_NMI;
-    LvtRegister.MessageType = APIC_DM_NMI;
-    LvtRegister.TriggerMode = APIC_TGM_LEVEL;
     HlWriteApicRegister(APIC_LINT1, LvtRegister.Long);
 
-    /* Clear errors after enabling vectors */
-    HlWriteApicRegister(APIC_ESR, 0);
-
-    /* Register interrupt handlers once the APIC initialization is done */
+    /* Register interrupt handlers */
     KeSetInterruptHandler(APIC_VECTOR_SPURIOUS, HlpHandleApicSpuriousService);
     KeSetInterruptHandler(PIC1_VECTOR_SPURIOUS, HlpHandlePicSpuriousService);
+
+    /* Clear any pre-existing errors */
+    HlWriteApicRegister(APIC_ESR, 0);
+
+    /* Re-enable all interrupts by lowering the Task Priority Register */
+    HlWriteApicRegister(APIC_TPR, 0x00);
+}
+
+/**
+ * Initializes the legacy PIC interrupt controller.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+VOID
+HlpInitializeLegacyPic(VOID)
+{
+    PIC_I8259_ICW1 Icw1;
+    PIC_I8259_ICW2 Icw2;
+    PIC_I8259_ICW3 Icw3;
+    PIC_I8259_ICW4 Icw4;
+
+    /* Initialize ICW1 for PIC1 port */
+    Icw1.Init = TRUE;
+    Icw1.InterruptMode = EdgeTriggered;
+    Icw1.InterruptVectorAddress = 0;
+    Icw1.Interval = Interval8;
+    Icw1.NeedIcw4 = TRUE;
+    Icw1.OperatingMode = Cascade;
+    HlIoPortOutByte(PIC1_CONTROL_PORT, Icw1.Bits);
+
+    /* Initialize ICW2 for PIC1 port */
+    Icw2.Bits = 0x00;
+    HlIoPortOutByte(PIC1_DATA_PORT, Icw2.Bits);
+
+    /* Initialize ICW3 for PIC1 port */
+    Icw3.Bits = 0;
+    Icw3.SlaveIrq2 = TRUE;
+    HlIoPortOutByte(PIC1_DATA_PORT, Icw3.Bits);
+
+    /* Initialize ICW4 for PIC1 port */
+    Icw4.BufferedMode = NonBuffered;
+    Icw4.EoiMode = NormalEoi;
+    Icw4.Reserved = 0;
+    Icw4.SpecialFullyNestedMode = FALSE;
+    Icw4.SystemMode = New8086Mode;
+    HlIoPortOutByte(PIC1_DATA_PORT, Icw4.Bits);
+
+    /* Mask all interrupts on PIC1 port */
+    HlIoPortOutByte(PIC1_DATA_PORT, 0xFF);
+
+    /* Initialize ICW1 for PIC2 port */
+    Icw1.Init = TRUE;
+    Icw1.InterruptMode = EdgeTriggered;
+    Icw1.InterruptVectorAddress = 0;
+    Icw1.Interval = Interval8;
+    Icw1.NeedIcw4 = TRUE;
+    Icw1.OperatingMode = Cascade;
+    HlIoPortOutByte(PIC2_CONTROL_PORT, Icw1.Bits);
+
+    /* Initialize ICW2 for PIC2 port */
+    Icw2.Bits = 0x08;
+    HlIoPortOutByte(PIC2_DATA_PORT, Icw2.Bits);
+
+    /* Initialize ICW3 for PIC2 port */
+    Icw3.Bits = 0;
+    Icw3.SlaveId = 2;
+    HlIoPortOutByte(PIC2_DATA_PORT, Icw3.Bits);
+
+    /* Initialize ICW4 for PIC2 port */
+    Icw4.BufferedMode = NonBuffered;
+    Icw4.EoiMode = NormalEoi;
+    Icw4.Reserved = 0;
+    Icw4.SpecialFullyNestedMode = FALSE;
+    Icw4.SystemMode = New8086Mode;
+    HlIoPortOutByte(PIC2_DATA_PORT, Icw4.Bits);
+
+    /* Mask all interrupts on PIC2 port */
+    HlIoPortOutByte(PIC2_DATA_PORT, 0xFF);
+}
+
+/**
+ * Initializes the (A)PIC interrupt controller.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ *
+ * @todo Initialize APIC only when supported, otherwise fall back to legacy PIC.
+ */
+XTAPI
+VOID
+HlpInitializePic(VOID)
+{
+    /* Initialize APIC */
+    HlpInitializeApic();
+
+    /* Initialize legacy PIC */
+    HlpInitializeLegacyPic();
+}
+
+/**
+ * Sends an IPI (Inter-Processor Interrupt) to the specified CPU.
+ *
+ * @param ApicId
+ *        Supplies a CPU APIC ID to send an IPI to.
+ *
+ * @param Vector
+ *        Supplies the IPI vector to send.
+ *
+ * @return This routine does not return any value.
+ *
+ * @since XT 1.0
+ */
+XTAPI
+VOID
+HlpSendIpi(ULONG ApicId,
+           ULONG Vector)
+{
+    /* Check current APIC mode */
+    if(HlpApicMode == APIC_MODE_X2APIC)
+    {
+        /* Send IPI using x2APIC mode */
+        HlWriteApicRegister(APIC_ICR0, ((ULONGLONG)ApicId << 32) | Vector);
+    }
+    else
+    {
+        /* Send IPI using xAPIC compatibility mode */
+        HlWriteApicRegister(APIC_ICR1, ApicId << 24);
+        HlWriteApicRegister(APIC_ICR0, Vector);
+    }
 }

xtoskrnl/includes/amd64/ari.h

@@ -25,6 +25,10 @@ XTCDECL
 VOID
 ArFlushTlb(VOID);
 
+XTCDECL
+ULONG
+ArGetCpuFlags(VOID);
+
 XTASSEMBLY
 XTCDECL
 ULONG_PTR
@@ -36,7 +40,11 @@ ArHalt(VOID);
 
 XTAPI
 VOID
-ArInitializeProcessor(VOID);
+ArInitializeProcessor(IN PVOID ProcessorStructures);
+
+XTCDECL
+BOOLEAN
+ArInterruptsEnabled(VOID);
 
 XTCDECL
 VOID
@@ -289,13 +297,24 @@ XTAPI
 VOID
 ArpInitializeProcessorRegisters(VOID);
 
+XTAPI
+VOID
+ArpInitializeProcessorStructures(IN PVOID ProcessorStructures,
+                                 OUT PKGDTENTRY *Gdt,
+                                 OUT PKTSS *Tss,
+                                 OUT PKPROCESSOR_BLOCK *ProcessorBlock,
+                                 OUT PVOID *KernelBootStack,
+                                 OUT PVOID *KernelFaultStack);
+
 XTAPI
 VOID
 ArpInitializeSegments(VOID);
 
 XTAPI
 VOID
-ArpInitializeTss(IN PKPROCESSOR_BLOCK ProcessorBlock);
+ArpInitializeTss(IN PKPROCESSOR_BLOCK ProcessorBlock,
+                 IN PVOID KernelBootStack,
+                 IN PVOID KernelFaultStack);
 
 XTAPI
 VOID

xtoskrnl/includes/amd64/asmsup.h

@@ -10,6 +10,21 @@
 #define __XTOSKRNL_AMD64_ASMSUP_H
 
 
+/* Control Register bit definitions */
+#define CR0_PG                                  0x80000000
+#define CR4_PGE                                 0x00000080
+#define CR4_LA57                                0x00001000
+#define CR4_PCIDE                               0x00020000
+
+/* GDT selectors */
+#define GDT_R0_CMCODE                           0x08
+#define GDT_R0_CODE                             0x10
+#define GDT_R0_DATA                             0x18
+
+/* MSR registers */
+#define X86_MSR_EFER                            0xC0000080
+#define X86_MSR_EFER_LME                        (1 <<  8)
+
 /* KTRAP_FRAME structure offsets */
 #define TrapXmm0                                0
 #define TrapXmm1                                16

xtoskrnl/includes/amd64/globals.h

@@ -30,4 +30,10 @@ EXTERN UCHAR ArKernelBootStack[KERNEL_STACK_SIZE];
 /* Kernel own fault stack */
 EXTERN UCHAR ArKernelFaultStack[KERNEL_STACK_SIZE];
 
+/* Page mapping routines for systems using 4-level paging (PML4) */
+EXTERN CMMPAGEMAP_ROUTINES MmpPml4Routines;
+
+/* Page mapping routines for systems using 5-level paging (PML5) */
+EXTERN CMMPAGEMAP_ROUTINES MmpPml5Routines;
+
 #endif /* __XTOSKRNL_AMD64_GLOBALS_H */

xtoskrnl/includes/amd64/hli.h

@@ -17,12 +17,8 @@ XTAPI
 VOID
 HlClearApicErrors(VOID);
 
-XTAPI
-VOID
-HlDisableLegacyPic(VOID);
-
 XTFASTCALL
-ULONG
+ULONGLONG
 HlReadApicRegister(IN APIC_REGISTER Register);
 
 XTAPI
@@ -32,12 +28,16 @@ HlSendEoi(VOID);
 XTFASTCALL
 VOID
 HlWriteApicRegister(IN APIC_REGISTER Register,
-                    IN ULONG Value);
+                    IN ULONGLONG Value);
 
 XTAPI
 BOOLEAN
 HlpCheckX2ApicSupport(VOID);
 
+XTAPI
+ULONG
+HlpGetCpuApicId(VOID);
+
 XTCDECL
 VOID
 HlpHandleApicSpuriousService(VOID);
@@ -48,7 +48,20 @@ HlpHandlePicSpuriousService(VOID);
 
 XTAPI
 VOID
-HlpInitializeApic(VOID);
+HlpInitializeApic();
+
+XTAPI
+VOID
+HlpInitializeLegacyPic(VOID);
+
+XTAPI
+VOID
+HlpInitializePic();
+
+XTAPI
+VOID
+HlpSendIpi(ULONG ApicId,
+           ULONG Vector);
 
 XTFASTCALL
 KRUNLEVEL

xtoskrnl/includes/amd64/kei.h

@@ -21,6 +21,10 @@ XTAPI
 PKPROCESSOR_CONTROL_BLOCK
 KeGetCurrentProcessorControlBlock(VOID);
 
+XTAPI
+ULONG
+KeGetCurrentProcessorNumber(VOID);
+
 XTAPI
 PKTHREAD
 KeGetCurrentThread(VOID);

xtoskrnl/includes/amd64/mmi.h

@@ -13,17 +13,33 @@
 
 
 /* AMD64 Memory Manager routines forward references */
+XTAPI
+VOID
+MmInitializePageMapSupport(VOID);
+
 XTFASTCALL
 VOID
 MmZeroPages(IN PVOID Address,
             IN ULONG Size);
 
 XTAPI
-PMMPTE
+VOID
+MmpClearPte(PHARDWARE_PTE PtePointer);
+
+XTAPI
+BOOLEAN
+MmpGetExtendedPhysicalAddressingStatus(VOID);
+
+XTAPI
+PMMP5E
+MmpGetP5eAddress(PVOID Address);
+
+XTAPI
+PMMPDE
 MmpGetPdeAddress(PVOID Address);
 
 XTAPI
-PMMPTE
+PMMPPE
 MmpGetPpeAddress(PVOID Address);
 
 XTAPI
@@ -31,11 +47,27 @@ PMMPTE
 MmpGetPteAddress(PVOID Address);
 
 XTAPI
-PMMPTE
+PMMPXE
 MmpGetPxeAddress(PVOID Address);
 
 XTAPI
 VOID
 MmpInitializeArchitecture(VOID);
 
+XTAPI
+BOOLEAN
+MmpPteValid(PHARDWARE_PTE PtePointer);
+
+XTAPI
+VOID
+MmpSetPte(PHARDWARE_PTE PtePointer,
+          PFN_NUMBER PageFrameNumber,
+          BOOLEAN Writable);
+
+XTAPI
+VOID
+MmpSetPteCaching(PHARDWARE_PTE PtePointer,
+                 BOOLEAN CacheDisable,
+                 BOOLEAN WriteThrough);
+
 #endif /* __XTOSKRNL_AMD64_MMI_H */

xtoskrnl/includes/globals.h

@@ -12,20 +12,48 @@
 #include <xtos.h>
 
 
+/* ACPI tables cache list */
+EXTERN LIST_ENTRY HlpAcpiCacheList;
+
+/* ACPI Root System Description Pointer (RSDP) */
+EXTERN PACPI_RSDP HlpAcpiRsdp;
+
+/* ACPI timer information */
+EXTERN ACPI_TIMER_INFO HlpAcpiTimerInfo;
+
 /* Active processors count */
 EXTERN KAFFINITY HlpActiveProcessors;
 
 /* APIC mode */
-EXTERN HAL_APIC_MODE HlpApicMode;
+EXTERN APIC_MODE HlpApicMode;
 
 /* FrameBuffer information */
-EXTERN HAL_FRAMEBUFFER_DATA HlpFrameBufferData;
+EXTERN HL_FRAMEBUFFER_DATA HlpFrameBufferData;
 
-/* Processors identity table */
-EXTERN HAL_PROCESSOR_IDENTITY HlpProcessorsIdentity[MAXIMUM_PROCESSORS];
+/* Scroll region information */
+EXTERN HL_SCROLL_REGION_DATA HlpScrollRegionData;
+
+/* System information */
+EXTERN ACPI_SYSTEM_INFO HlpSystemInfo;
+
+/* Pointer to DbgPrint() routine */
+EXTERN PKD_PRINT_ROUTINE KdPrint;
+
+/* Kernel Debugger mode */
+EXTERN KD_DEBUG_MODE KdpDebugMode;
+
+/* Debugger I/O providers initialization routines */
+EXTERN PKD_INIT_ROUTINE KdpIoProvidersInitRoutines[KDBG_PROVIDERS_COUNT];
+
+/* List of active I/O providers */
+EXTERN LIST_ENTRY KdpProviders;
+
+/* Debugger's serial port handle */
+EXTERN CPPORT KdpSerialPort;
+
+/* Pre-defined serial port addresses */
+EXTERN ULONG KdpSerialPortList[COMPORT_COUNT];
 
-/* Pointer to boot loader provided DbgPrint() routine */
-EXTERN VOID (*KeDbgPrint)(IN PWCHAR Format, IN ...);
 
 /* Kernel initialization block passed by boot loader */
 EXTERN PKERNEL_INITIALIZATION_BLOCK KeInitializationBlock;
@@ -42,8 +70,17 @@ EXTERN KSERVICE_DESCRIPTOR_TABLE KeServiceDescriptorTable[KSERVICE_TABLES_COUNT]
 /* Kernel process list */
 EXTERN LIST_ENTRY KepProcessListHead;
 
+/* Kernel system resources list */
+EXTERN LIST_ENTRY KepSystemResourcesListHead;
+
+/* Kernel system resources lock */
+EXTERN KSPIN_LOCK KepSystemResourcesLock;
+
+/* Kernel UBSAN active frame flag */
+EXTERN BOOLEAN KepUbsanActiveFrame;
+
 /* Biggest free memory descriptor */
-EXTERN PLOADER_MEMORY_MAPPING MmFreeDescriptor;
+EXTERN PLOADER_MEMORY_DESCRIPTOR MmFreeDescriptor;
 
 /* Highest physical page number */
 EXTERN ULONG_PTR MmHighestPhysicalPage;
@@ -55,9 +92,24 @@ EXTERN ULONG_PTR MmLowestPhysicalPage;
 EXTERN ULONG MmNumberOfPhysicalPages;
 
 /* Old biggest free memory descriptor */
-EXTERN LOADER_MEMORY_MAPPING MmOldFreeDescriptor;
+EXTERN LOADER_MEMORY_DESCRIPTOR MmOldFreeDescriptor;
 
-/* Page Map Level */
-EXTERN ULONG MmPageMapLevel;
+/* Processor structures data (THIS IS A TEMPORARY HACK) */
+EXTERN UCHAR MmProcessorStructuresData[MAXIMUM_PROCESSORS][KPROCESSOR_STRUCTURES_SIZE];
+
+/* Allocation descriptors dedicated for hardware layer */
+EXTERN LOADER_MEMORY_DESCRIPTOR MmpHardwareAllocationDescriptors[MM_HARDWARE_ALLOCATION_DESCRIPTORS];
+
+/* Live address of kernel's hardware heap */
+EXTERN PVOID MmpHardwareHeapStart;
+
+/* Information about the current page map */
+EXTERN MMPAGEMAP_INFO MmpPageMapInfo;
+
+/* Pointers to page map routines for the current paging mode */
+EXTERN PCMMPAGEMAP_ROUTINES MmpPageMapRoutines;
+
+/* Number of used hardware allocation descriptors */
+EXTERN ULONG MmpUsedHardwareAllocationDescriptors;
 
 #endif /* __XTOSKRNL_GLOBALS_H */

xtoskrnl/includes/hli.h

@@ -11,10 +11,11 @@
 
 #include <xtos.h>
 
+
 /* HAL library routines forward references */
 XTAPI
 VOID
-HlClearScreen(VOID);
+HlClearScreen(IN ULONG Color);
 
 XTCDECL
 XTSTATUS
@@ -33,21 +34,30 @@ UCHAR
 HlComPortReadLsr(IN PCPPORT Port,
                  IN UCHAR Byte);
 
+XTCDECL
+XTSTATUS
+HlDisplayCharacter(IN WCHAR Character);
+
+XTAPI
+XTSTATUS
+HlGetAcpiSystemDescriptionPointer(OUT PACPI_RSDP *Rsdp);
+
+XTAPI
+XTSTATUS
+HlGetAcpiTable(IN ULONG Signature,
+               OUT PACPI_DESCRIPTION_HEADER *AcpiTable);
+
 XTAPI
 VOID
-HlDrawPixel(IN ULONG PosX,
-            IN ULONG PosY,
-            IN ULONG Color);
+HlGetFrameBufferResolution(OUT PULONG Width, OUT PULONG Height);
 
 XTFASTCALL
 KRUNLEVEL
 HlGetRunLevel(VOID);
 
-
 XTCDECL
 XTSTATUS
 HlInitializeComPort(IN OUT PCPPORT Port,
-                    IN ULONG PortNumber,
                     IN PUCHAR PortAddress,
                     IN ULONG BaudRate);
 
@@ -57,10 +67,19 @@ HlInitializeFrameBuffer(VOID);
 
 XTAPI
 VOID
-HlPutCharacter(IN ULONG PositionX,
-               IN ULONG PositionY,
-               IN ULONG Color,
-               IN WCHAR WideCharacter);
+HlInitializeProcessor(VOID);
+
+XTAPI
+VOID
+HlInitializeScrollRegion(IN ULONG Left,
+                         IN ULONG Top,
+                         IN ULONG Right,
+                         IN ULONG Bottom,
+                         IN ULONG FontColor);
+
+XTAPI
+XTSTATUS
+HlInitializeSystem(VOID);
 
 XTFASTCALL
 VOID
@@ -68,6 +87,66 @@ HlSetRunLevel(IN KRUNLEVEL RunLevel);
 
 XTAPI
 VOID
-HlInitializeProcessor(VOID);
+HlpCacheAcpiTable(IN PACPI_DESCRIPTION_HEADER AcpiTable);
+
+XTAPI
+VOID
+HlpDrawCharacter(IN ULONG PositionX,
+                 IN ULONG PositionY,
+                 IN ULONG Color,
+                 IN WCHAR WideCharacter);
+
+XTAPI
+VOID
+HlpDrawPixel(IN ULONG PosX,
+             IN ULONG PosY,
+             IN ULONG Color);
+
+XTAPI
+XTSTATUS
+HlpInitializeAcpi(VOID);
+
+XTAPI
+XTSTATUS
+HlpInitializeAcpiCache(VOID);
+
+XTAPI
+XTSTATUS
+HlpInitializeAcpiSystemDescriptionTable(OUT PACPI_DESCRIPTION_HEADER *AcpiTable);
+
+XTAPI
+XTSTATUS
+HlpInitializeAcpiSystemInformation(VOID);
+
+XTAPI
+XTSTATUS
+HlpInitializeAcpiSystemStructure(VOID);
+
+XTAPI
+XTSTATUS
+HlpInitializeAcpiTimer(VOID);
+
+XTAPI
+XTSTATUS
+HlpQueryAcpiCache(IN ULONG Signature,
+                  OUT PACPI_DESCRIPTION_HEADER *AcpiTable);
+
+XTAPI
+XTSTATUS
+HlpQueryAcpiTables(IN ULONG Signature,
+                   OUT PACPI_DESCRIPTION_HEADER *AcpiTable);
+
+XTAPI
+ULONG
+HlpRGBColor(IN ULONG Color);
+
+XTAPI
+VOID
+HlpScrollRegion(VOID);
+
+XTAPI
+BOOLEAN
+HlpValidateAcpiTable(IN PVOID Buffer,
+                     IN UINT_PTR Size);
 
 #endif /* __XTOSKRNL_HLI_H */

xtoskrnl/includes/i686/ari.h

@@ -25,6 +25,10 @@ XTCDECL
 VOID
 ArFlushTlb(VOID);
 
+XTCDECL
+ULONG
+ArGetCpuFlags(VOID);
+
 XTASSEMBLY
 XTCDECL
 ULONG_PTR
@@ -36,7 +40,11 @@ ArHalt(VOID);
 
 XTAPI
 VOID
-ArInitializeProcessor(VOID);
+ArInitializeProcessor(IN PVOID ProcessorStructures);
+
+XTCDECL
+BOOLEAN
+ArInterruptsEnabled(VOID);
 
 XTCDECL
 VOID
@@ -277,17 +285,29 @@ XTAPI
 VOID
 ArpInitializeProcessorRegisters(VOID);
 
+XTAPI
+VOID
+ArpInitializeProcessorStructures(IN PVOID ProcessorStructures,
+                                 OUT PKGDTENTRY *Gdt,
+                                 OUT PKTSS *Tss,
+                                 OUT PKPROCESSOR_BLOCK *ProcessorBlock,
+                                 OUT PVOID *KernelBootStack,
+                                 OUT PVOID *KernelFaultStack);
+
 XTAPI
 VOID
 ArpInitializeSegments(VOID);
 
 XTAPI
 VOID
-ArpInitializeTss(IN PKPROCESSOR_BLOCK ProcessorBlock);
+ArpInitializeTss(IN PKPROCESSOR_BLOCK ProcessorBlock,
+                 IN PVOID KernelBootStack,
+                 IN PVOID KernelFaultStack);
 
 XTAPI
 VOID
-ArpSetDoubleFaultTssEntry(IN PKPROCESSOR_BLOCK ProcessorBlock);
+ArpSetDoubleFaultTssEntry(IN PKPROCESSOR_BLOCK ProcessorBlock,
+                          IN PVOID KernelFaultStack);
 
 XTAPI
 VOID
@@ -310,7 +330,8 @@ ArpSetIdtGate(IN PKIDTENTRY Idt,
 
 XTAPI
 VOID
-ArpSetNonMaskableInterruptTssEntry(IN PKPROCESSOR_BLOCK ProcessorBlock);
+ArpSetNonMaskableInterruptTssEntry(IN PKPROCESSOR_BLOCK ProcessorBlock,
+                                   IN PVOID KernelFaultStack);
 
 XTCDECL
 VOID

xtoskrnl/includes/i686/globals.h

@@ -34,4 +34,10 @@ EXTERN UCHAR ArKernelBootStack[KERNEL_STACK_SIZE];
 /* Kernel own fault stack */
 EXTERN UCHAR ArKernelFaultStack[KERNEL_STACK_SIZE];
 
+/* Page mapping routines for systems using 2-level paging (PML2) */
+EXTERN CMMPAGEMAP_ROUTINES MmpPml2Routines;
+
+/* Page mapping routines for systems using 3-level paging (PML3) */
+EXTERN CMMPAGEMAP_ROUTINES MmpPml3Routines;
+
 #endif /* __XTOSKRNL_I686_GLOBALS_H */

xtoskrnl/includes/i686/hli.h

@@ -17,12 +17,8 @@ XTAPI
 VOID
 HlClearApicErrors(VOID);
 
-XTAPI
-VOID
-HlDisableLegacyPic(VOID);
-
 XTFASTCALL
-ULONG
+ULONGLONG
 HlReadApicRegister(IN APIC_REGISTER Register);
 
 XTAPI
@@ -32,12 +28,16 @@ HlSendEoi(VOID);
 XTFASTCALL
 VOID
 HlWriteApicRegister(IN APIC_REGISTER Register,
-                    IN ULONG Value);
+                    IN ULONGLONG Value);
 
 XTAPI
 BOOLEAN
 HlpCheckX2ApicSupport(VOID);
 
+XTAPI
+ULONG
+HlpGetCpuApicId(VOID);
+
 XTCDECL
 VOID
 HlpHandleApicSpuriousService(VOID);
@@ -50,6 +50,19 @@ XTAPI
 VOID
 HlpInitializeApic(VOID);
 
+XTAPI
+VOID
+HlpInitializeLegacyPic(VOID);
+
+XTAPI
+VOID
+HlpInitializePic(VOID);
+
+XTAPI
+VOID
+HlpSendIpi(ULONG ApicId,
+           ULONG Vector);
+
 XTFASTCALL
 KRUNLEVEL
 HlpTransformApicTprToRunLevel(IN UCHAR Tpr);

xtoskrnl/includes/i686/kei.h

@@ -21,6 +21,10 @@ XTAPI
 PKPROCESSOR_CONTROL_BLOCK
 KeGetCurrentProcessorControlBlock(VOID);
 
+XTAPI
+ULONG
+KeGetCurrentProcessorNumber(VOID);
+
 XTAPI
 PKTHREAD
 KeGetCurrentThread(VOID);

xtoskrnl/includes/i686/mmi.h

@@ -12,14 +12,70 @@
 #include <xtos.h>
 
 
-/* I686 Memory Manager routines forward references */
+/* i686 Memory Manager routines forward references */
+XTAPI
+VOID
+MmInitializePageMapSupport(VOID);
+
 XTFASTCALL
 VOID
 MmZeroPages(IN PVOID Address,
             IN ULONG Size);
 
+XTAPI
+VOID
+MmpClearPte(PHARDWARE_PTE PtePointer);
+
+XTAPI
+BOOLEAN
+MmpGetExtendedPhysicalAddressingStatus(VOID);
+
+XTAPI
+PMMPDE
+MmpGetPdeAddress(PVOID Address);
+
+XTAPI
+PMMPPE
+MmpGetPpeAddress(PVOID Address);
+
+XTAPI
+PMMPTE
+MmpGetPteAddress(PVOID Address);
+
 XTAPI
 VOID
 MmpInitializeArchitecture(VOID);
 
+XTAPI
+BOOLEAN
+MmpPml2PteValid(PHARDWARE_PTE PtePointer);
+
+XTAPI
+VOID
+MmpSetPml2Pte(PHARDWARE_PTE PtePointer,
+              PFN_NUMBER PageFrameNumber,
+              BOOLEAN Writable);
+
+XTAPI
+VOID
+MmpSetPml2PteCaching(PHARDWARE_PTE PtePointer,
+                     BOOLEAN CacheDisable,
+                     BOOLEAN WriteThrough);
+
+XTAPI
+BOOLEAN
+MmpPml3PteValid(PHARDWARE_PTE PtePointer);
+
+XTAPI
+VOID
+MmpSetPml3Pte(PHARDWARE_PTE PtePointer,
+              PFN_NUMBER PageFrameNumber,
+              BOOLEAN Writable);
+
+XTAPI
+VOID
+MmpSetPml3PteCaching(PHARDWARE_PTE PtePointer,
+                     BOOLEAN CacheDisable,
+                     BOOLEAN WriteThrough);
+
 #endif /* __XTOSKRNL_I686_MMI_H */

xtoskrnl/includes/kdi.h

@@ -0,0 +1,44 @@
+/**
+ * PROJECT:         ExectOS
+ * COPYRIGHT:       See COPYING.md in the top level directory
+ * FILE:            xtoskrnl/includes/kdi.h
+ * DESCRIPTION:     Kernel Debugger routines
+ * DEVELOPERS:      Aiken Harris <harraiken91@gmail.com>
+ */
+
+#ifndef __XTOSKRNL_KDI_H
+#define __XTOSKRNL_KDI_H
+
+#include <xtos.h>
+
+
+/* Kernel Debugger routines forward references */
+XTAPI
+XTSTATUS
+KdInitializeDebugIoProviders(VOID);
+
+XTAPI
+VOID
+KdSetPrintRoutine(PVOID DebugPrintRoutine);
+
+XTCDECL
+VOID
+KdpDebugPrint(PCWSTR Format, ...);
+
+XTAPI
+XTSTATUS
+KdpDetectDebugPorts(VOID);
+
+XTAPI
+XTSTATUS
+KdpInitializeFrameBufferProvider(VOID);
+
+XTAPI
+XTSTATUS
+KdpInitializeSerialPortProvider(VOID);
+
+XTCDECL
+XTSTATUS
+KdpSerialWriteCharacter(WCHAR Character);
+
+#endif /* __XTOSKRNL_KDI_H */

xtoskrnl/includes/kei.h

@@ -21,6 +21,15 @@ XTAPI
 VOID
 KeClearTimer(IN PKTIMER Timer);
 
+XTAPI
+SYSTEM_FIRMWARE_TYPE
+KeGetFirmwareType(VOID);
+
+XTAPI
+XTSTATUS
+KeGetKernelParameter(IN PCWSTR ParameterName,
+                     OUT PCWSTR *Parameter);
+
 XTAPI
 VOID
 KeHaltSystem(VOID);
@@ -86,10 +95,115 @@ XTAPI
 VOID
 KeStartXtSystem(IN PKERNEL_INITIALIZATION_BLOCK Parameters);
 
+XTCDECL
+BOOLEAN
+KepCheckUbsanReport(PKUBSAN_SOURCE_LOCATION Location);
+
+XTCDECL
+VOID
+KepEnterUbsanFrame(PKUBSAN_SOURCE_LOCATION Location,
+                   PCCHAR Reason);
+
 XTFASTCALL
 VOID
 KepExitDispatcher(IN KRUNLEVEL OldRunLevel);
 
+XTCDECL
+LONGLONG
+KepGetSignedUbsanValue(PKUBSAN_TYPE_DESCRIPTOR Type,
+                       PVOID Value);
+
+XTAPI
+XTSTATUS
+KepGetSystemResource(IN SYSTEM_RESOURCE_TYPE ResourceType,
+                     IN BOOLEAN Acquire,
+                     OUT PSYSTEM_RESOURCE_HEADER *ResourceHeader);
+
+XTCDECL
+PCCHAR
+KepGetUbsanTypeKind(UCHAR TypeCheckKind);
+
+XTCDECL
+ULONGLONG
+KepGetUnsignedUbsanValue(PKUBSAN_TYPE_DESCRIPTOR Type,
+                         PVOID Value);
+
+XTCDECL
+VOID
+KepHandleUbsanDivisionOverflow(PKUBSAN_OVERFLOW_DATA Data,
+                               PVOID Lhs,
+                               PVOID Rhs);
+
+XTCDECL
+VOID
+KepHandleUbsanFloatCastOverflow(PKUBSAN_FLOAT_CAST_OVERFLOW_DATA Data,
+                                ULONG_PTR Lhs,
+                                ULONG_PTR Rhs);
+
+XTCDECL
+VOID
+KepHandleUbsanFunctionTypeMismatch(PKUBSAN_FUNCTION_TYPE_MISMATCH_DATA Data,
+                                   ULONG_PTR Pointer);
+
+XTCDECL
+VOID
+KepHandleUbsanIntegerOverflow(PKUBSAN_OVERFLOW_DATA Data,
+                                 ULONG_PTR Lhs,
+                                 ULONG_PTR Rhs);
+
+XTCDECL
+VOID
+KepHandleUbsanInvalidBuiltin(PKUBSAN_INVALID_BUILTIN_DATA Data);
+
+XTCDECL
+VOID
+KepHandleUbsanMisalignedAccess(PKUBSAN_TYPE_MISMATCH_DATA Data,
+                               ULONG_PTR Pointer);
+
+XTCDECL
+VOID
+KepHandleUbsanNegateOverflow(PKUBSAN_OVERFLOW_DATA Data,
+                             ULONG_PTR OldValue);
+
+XTCDECL
+VOID
+KepHandleUbsanNullPointerDereference(PKUBSAN_TYPE_MISMATCH_DATA Data);
+
+XTCDECL
+VOID
+KepHandleUbsanObjectSizeMismatch(PKUBSAN_TYPE_MISMATCH_DATA Data,
+                                 ULONG_PTR Pointer);
+
+XTCDECL
+VOID
+KepHandleUbsanOutOfBounds(PKUBSAN_OUT_OF_BOUNDS_DATA Data,
+                          ULONG_PTR Index);
+
+XTCDECL
+VOID
+KepHandleUbsanPointerOverflow(PKUBSAN_OVERFLOW_DATA Data,
+                              ULONG_PTR Lhs,
+                              ULONG_PTR Rhs);
+
+XTCDECL
+VOID
+KepHandleUbsanShiftOutOfBounds(PKUBSAN_SHIFT_OUT_OF_BOUNDS_DATA Data,
+                               ULONG_PTR Lhs,
+                               ULONG_PTR Rhs);
+
+XTCDECL
+VOID
+KepHandleUbsanTypeMismatch(PKUBSAN_TYPE_MISMATCH_DATA Data,
+                           ULONG_PTR Pointer);
+
+XTAPI
+VOID
+KepInitializeSystemResources(VOID);
+
+XTCDECL
+VOID
+KepLeaveUbsanFrame();
+
 XTAPI
 VOID
 KepRemoveTimer(IN OUT PKTIMER Timer);

xtoskrnl/includes/mmi.h

@@ -13,21 +13,64 @@
 
 
 /* Memory Manager routines forward references */
+XTAPI
+XTSTATUS
+MmAllocateHardwareMemory(IN PFN_NUMBER PageCount,
+                         IN BOOLEAN Aligned,
+                         OUT PPHYSICAL_ADDRESS Buffer);
+
 XTAPI
 XTSTATUS
 MmAllocateKernelStack(IN PVOID *Stack,
                       IN BOOLEAN LargeStack,
                       IN UCHAR SystemNode);
 
+XTAPI
+XTSTATUS
+MmAllocateProcessorStructures(IN ULONG CpuNumber,
+                              OUT PVOID *StructuresData);
+
+XTAPI
+VOID
+MmFlushTlb(VOID);
+
 XTAPI
 VOID
 MmFreeKernelStack(IN PVOID Stack,
                   IN BOOLEAN LargeStack);
 
+XTAPI
+VOID
+MmFreeProcessorStructures(IN PVOID StructuresData);
+
 XTAPI
 VOID
 MmInitializeMemoryManager(VOID);
 
+XTAPI
+XTSTATUS
+MmMapHardwareMemory(IN PHYSICAL_ADDRESS PhysicalAddress,
+                    IN PFN_NUMBER PageCount,
+                    IN BOOLEAN FlushTlb,
+                    OUT PVOID *VirtualAddress);
+
+XTAPI
+VOID
+MmMarkHardwareMemoryWriteThrough(IN PVOID VirtualAddress,
+                                 IN PFN_NUMBER PageCount);
+
+XTAPI
+VOID
+MmRemapHardwareMemory(IN PVOID VirtualAddress,
+                      IN PHYSICAL_ADDRESS PhysicalAddress,
+                      IN BOOLEAN FlushTlb);
+
+XTAPI
+XTSTATUS
+MmUnmapHardwareMemory(IN PVOID VirtualAddress,
+                      IN PFN_NUMBER PageCount,
+                      IN BOOLEAN FlushTlb);
+
 XTAPI
 VOID
 MmpScanMemoryDescriptors(VOID);