general/occlum
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix panic if user input invalid memory arguments

Browse Source
This commit is contained in:
Hui, Chunyang 2021-12-30 06:26:54 +00:00 committed by Zongmin.Gu
parent 9d55882cd5
commit 1229de1fd1
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/libos/src/vm/free_space_manager.rs
View File

@ -48,7 +48,7 @@ impl VMFreeSpaceManager {
if free_range.size() < size {
continue;
}
unsafe { VMRange::from_unchecked(free_range.start(), free_range.end()) }
free_range.clone()
};
match addr {

2
src/libos/src/vm/user_space_vm.rs
View File

@ -29,7 +29,7 @@ impl UserSpaceVMManager {
"allocated rsrv addr is 0x{:x}, len is 0x{:x}",
addr, rsrv_mem_size
);
VMRange::from_unchecked(addr, addr + rsrv_mem_size)
VMRange::new(addr, addr + rsrv_mem_size)?
};
let vm_manager = VMManager::init(vm_range)?;

4
src/libos/src/vm/vm_manager.rs
View File

@ -74,14 +74,14 @@ impl VMManager {
match addr {
VMMapAddr::Any => {}
VMMapAddr::Hint(addr) => {
let target_range = unsafe { VMRange::from_unchecked(addr, addr + size) };
let target_range = VMRange::new(addr, addr + size)?;
let ret = self.mmap_with_addr(target_range, options);
if ret.is_ok() {
return ret;
}
}
VMMapAddr::Need(addr) | VMMapAddr::Force(addr) => {
let target_range = unsafe { VMRange::from_unchecked(addr, addr + size) };
let target_range = VMRange::new(addr, addr + size)?;
return self.mmap_with_addr(target_range, options);
}
}