1 Commits

Author SHA1 Message Date
4b5642cc4e fix: Add strlen func 2024-06-10 13:10:05 +03:00
5 changed files with 26 additions and 52 deletions

View File

@@ -1,6 +1,5 @@
[unstable] [unstable]
build-std = ["core", "compiler_builtins", "alloc"] build-std = ["core", "compiler_builtins", "alloc"]
json-target-spec = true
[build] [build]
target = "i686-kolibri.json" target = "i686-kolibri.json"

View File

@@ -5,6 +5,7 @@
"dynamic-linking": false, "dynamic-linking": false,
"executables": true, "executables": true,
"has-rpath": true, "has-rpath": true,
"is-builtin": false,
"linker-flavor": "ld.lld", "linker-flavor": "ld.lld",
"linker": "rust-lld", "linker": "rust-lld",
"llvm-target": "i686-unknown-none-code32", "llvm-target": "i686-unknown-none-code32",
@@ -13,9 +14,9 @@
"relocation-model": "static", "relocation-model": "static",
"position-independent-executables": false, "position-independent-executables": false,
"relro-level": "off", "relro-level": "off",
"target-c-int-width": 32, "target-c-int-width": "32",
"target-endian": "little", "target-endian": "little",
"target-pointer-width": 32, "target-pointer-width": "32",
"vendor": "unknown", "vendor": "unknown",
"disable-redzone": true, "disable-redzone": true,
"panic-strategy": "abort", "panic-strategy": "abort",

View File

@@ -1,4 +1 @@
[toolchain] nightly
channel = "nightly"
components = ["rust-src", "llvm-tools-preview"]
profile = "default"

View File

@@ -1,14 +0,0 @@
{ pkgs ? import <nixpkgs> {} }:
pkgs.mkShell {
packages = with pkgs; [
rustup
cargo-make
cargo-binutils
fasm
];
shellHook = ''
export PATH="$HOME/.cargo/bin:$PATH"
'';
}

View File

@@ -36,7 +36,6 @@ pub fn init() {
unsafe { unsafe {
sys::init_heap(); sys::init_heap();
MAIN_SECTOR = sys::alloc(PAGE_SIZE) as usize; MAIN_SECTOR = sys::alloc(PAGE_SIZE) as usize;
core::ptr::write_bytes(MAIN_SECTOR as *mut u8, 0, PAGE_SIZE); // make sector table start clean
} }
} }
} }
@@ -48,14 +47,14 @@ pub fn malloc(size: usize) -> *mut u8 {
if (addr as usize) != 0 { if (addr as usize) != 0 {
let sec = addr; let sec = addr;
let hdr = &mut *(addr as *mut SectorHeader); let mut hdr = *(addr as *const SectorHeader);
let sec_start_blocks = (sec as usize) + size_of::<SectorHeader>(); let sec_start_blocks = (sec as usize) + size_of::<SectorHeader>();
if hdr.size_left >= size { if hdr.size_left >= size {
let mut j = sec_start_blocks; let mut j = sec_start_blocks;
let mut first_found_block_addr = 0; let mut first_found_block_addr = 0;
while j <= sec_start_blocks + hdr.size { while j <= sec_start_blocks + hdr.size {
let block = &mut *(j as *mut BlockHeader); let mut block = *(j as *const BlockHeader);
match block.sign { match block.sign {
Sign::Active => { Sign::Active => {
// If block is occupated - pass // If block is occupated - pass
@@ -64,7 +63,7 @@ pub fn malloc(size: usize) -> *mut u8 {
} }
Sign::Free => { Sign::Free => {
if first_found_block_addr == 0 { if first_found_block_addr != 0 {
first_found_block_addr = j; first_found_block_addr = j;
} }
@@ -74,32 +73,26 @@ pub fn malloc(size: usize) -> *mut u8 {
j += (size_of::<BlockHeader>()) + block.size; j += (size_of::<BlockHeader>()) + block.size;
} else if size - sum_size < size_of::<BlockHeader>() { } else if size - sum_size < size_of::<BlockHeader>() {
// Create 2 blocks // Create 2 blocks
let main_block = let mut main_block =
&mut *(first_found_block_addr as *mut BlockHeader); *(first_found_block_addr as *const BlockHeader);
main_block.sign = Sign::Active; main_block.sign = Sign::Active;
main_block.size = size; main_block.size = size;
let secondary_block = (first_found_block_addr let mut secondary_block = *(first_found_block_addr
+ size_of::<BlockHeader>() as *const BlockHeader)
+ size) .add(size_of::<BlockHeader>() + size);
as *mut BlockHeader; secondary_block.sign = Sign::Free;
(*secondary_block).sign = Sign::Free; secondary_block.size =
(*secondary_block).size =
sum_size - size - size_of::<BlockHeader>(); sum_size - size - size_of::<BlockHeader>();
hdr.size_left -= size + size_of::<BlockHeader>();
return (first_found_block_addr as *mut u8) return (first_found_block_addr as *mut u8)
.add(size_of::<BlockHeader>()); .add(size_of::<BlockHeader>());
} else { } else {
// Create 1 block // Create 1 block
let main_block = let mut main_block =
&mut *(first_found_block_addr as *mut BlockHeader); *(first_found_block_addr as *const BlockHeader);
main_block.sign = Sign::Active; main_block.sign = Sign::Active;
main_block.size = sum_size - size_of::<BlockHeader>(); main_block.size = sum_size - size_of::<BlockHeader>();
hdr.size_left -= main_block.size + size_of::<BlockHeader>();
return (first_found_block_addr as *mut u8) return (first_found_block_addr as *mut u8)
.add(size_of::<BlockHeader>()); .add(size_of::<BlockHeader>());
} }
@@ -124,11 +117,9 @@ pub fn malloc(size: usize) -> *mut u8 {
} }
} }
} else { } else {
// round requested size up to whole pages let sec_size = size + PAGE_SIZE - size % PAGE_SIZE;
let sec_size = (size + PAGE_SIZE - 1) & !(PAGE_SIZE - 1);
let new_sec = sys::alloc(sec_size); let new_sec = sys::alloc(sec_size);
let sec_hdr = new_sec as *mut SectorHeader; let sec_hdr = new_sec as *mut SectorHeader;
*((MAIN_SECTOR + i * 4) as *mut u32) = new_sec as u32; // remember this sector in the table
*sec_hdr = SectorHeader { *sec_hdr = SectorHeader {
size: sec_size, size: sec_size,
size_left: sec_size - size_of::<SectorHeader>(), size_left: sec_size - size_of::<SectorHeader>(),
@@ -147,26 +138,26 @@ pub fn malloc(size: usize) -> *mut u8 {
fn free(block: *const u8) { fn free(block: *const u8) {
unsafe { unsafe {
let block_hdr = &mut *(block.sub(size_of::<BlockHeader>()) as *mut BlockHeader); let mut block_hdr = *(block.sub(size_of::<BlockHeader>()) as *mut BlockHeader);
for i in 0..PAGE_SIZE / 4 { for i in 0..PAGE_SIZE / 4 {
let addr = *((MAIN_SECTOR + i * 4) as *const u32) as *const u8; let addr = *((MAIN_SECTOR + i * 4) as *const u32) as *const u8;
if addr.is_null() { let mut hdr = *(addr as *const SectorHeader);
continue;
}
let hdr = &mut *(addr as *mut SectorHeader);
if addr < block && (block as usize) < (addr as usize) + hdr.size { if addr < block && (block as usize) < (addr as usize) + hdr.size {
hdr.size_left += block_hdr.size + size_of::<BlockHeader>(); hdr.size_left += block_hdr.size;
if hdr.size_left == hdr.size - size_of::<SectorHeader>() { if hdr.size_left == hdr.size - size_of::<SectorHeader>() {
sys::free(addr); // whole sector is free, hand it back to OS free(addr)
*((MAIN_SECTOR + i * 4) as *mut u32) = 0; // and drop from the table
} else { } else {
block_hdr.sign = Sign::Free; block_hdr.sign = Sign::Free;
} }
break; break;
} }
} }
if !sys::free(block) {
panic!("Free failed");
}
} }
} }
@@ -190,4 +181,4 @@ unsafe impl alloc::alloc::GlobalAlloc for GlobalAlloc {
} }
#[global_allocator] #[global_allocator]
static ALLOC: GlobalAlloc = GlobalAlloc; static ALLOC: GlobalAlloc = GlobalAlloc;