time unmodified

This commit is contained in:
Tateisi 2025-08-03 15:42:14 +08:00
parent cc1a9a4751
commit 58fa67414e
7 changed files with 89 additions and 12 deletions

View file

@ -30,7 +30,7 @@ pub fn syscall(syscall_id: usize, args: [usize; 3]) -> isize {
SYSCALL_WRITE => sys_write(args[0], args[1] as *const u8, args[2]), SYSCALL_WRITE => sys_write(args[0], args[1] as *const u8, args[2]),
SYSCALL_EXIT => sys_exit(args[0] as i32), SYSCALL_EXIT => sys_exit(args[0] as i32),
SYSCALL_YIELD => sys_yield(), SYSCALL_YIELD => sys_yield(),
SYSCALL_GET_TIME => sys_get_time(), SYSCALL_GET_TIME => sys_get_time(args[0] as *mut TimeVal, args[1]),
SYSCALL_SBRK => sys_sbrk(args[0] as i32), SYSCALL_SBRK => sys_sbrk(args[0] as i32),
SYSCALL_MMAP => sys_mmap(args[0], args[1], args[2]), SYSCALL_MMAP => sys_mmap(args[0], args[1], args[2]),
SYSCALL_MUNMAP => sys_munmap(args[0], args[1]), SYSCALL_MUNMAP => sys_munmap(args[0], args[1]),

View file

@ -3,7 +3,7 @@
use crate::config::PAGE_SIZE; use crate::config::PAGE_SIZE;
use crate::mm::{MapPermission, VPNRange, VirtAddr}; use crate::mm::{MapPermission, VPNRange, VirtAddr};
use crate::task::{change_program_brk, create_new_map_area, exit_current_and_run_next, get_current_task_page_table_entry, suspend_current_and_run_next, unmap_virtual_page}; use crate::task::{change_program_brk, create_new_map_area, exit_current_and_run_next, get_current_task_page_table_entry, suspend_current_and_run_next, unmap_virtual_page};
use crate::timer::get_time_ms; use crate::timer::get_time_us;
/// task exits and submit an exit code /// task exits and submit an exit code
pub fn sys_exit(exit_code: i32) -> ! { pub fn sys_exit(exit_code: i32) -> ! {
@ -18,9 +18,24 @@ pub fn sys_yield() -> isize {
0 0
} }
#[repr(C)]
#[derive(Debug)]
pub struct TimeVal {
pub sec: usize,
pub usec: usize,
}
/// get current time /// get current time
pub fn sys_get_time() -> isize { pub fn sys_get_time(ts: *mut TimeVal, _tz: usize) -> isize {
get_time_ms() as isize let us = get_time_us();
unsafe {
*ts = TimeVal {
sec: us / 1_000_000,
usec: us % 1_000_000,
};
}
0
-1
} }
/// change data segment size /// change data segment size

View file

@ -6,16 +6,23 @@ use riscv::register::time;
const TICKS_PER_SEC: usize = 100; const TICKS_PER_SEC: usize = 100;
const MSEC_PER_SEC: usize = 1000; const MSEC_PER_SEC: usize = 1000;
const MICRO_PER_SEC: usize = 1_000_000;
pub fn get_time() -> usize { pub fn get_time() -> usize {
time::read() time::read()
} }
/// get current time in microseconds /// get current time in microseconds
#[allow(unused)]
pub fn get_time_ms() -> usize { pub fn get_time_ms() -> usize {
time::read() / (CLOCK_FREQ / MSEC_PER_SEC) time::read() / (CLOCK_FREQ / MSEC_PER_SEC)
} }
/// get current time in us
pub fn get_time_us() -> usize {
time::read() / (CLOCK_FREQ / MICRO_PER_SEC)
}
/// set the next timer interrupt /// set the next timer interrupt
pub fn set_next_trigger() { pub fn set_next_trigger() {
set_timer(get_time() + CLOCK_FREQ / TICKS_PER_SEC); set_timer(get_time() + CLOCK_FREQ / TICKS_PER_SEC);

View file

@ -6,13 +6,19 @@ extern crate user_lib;
use user_lib::{get_time, yield_}; use user_lib::{get_time, yield_};
/// 正确输出:(无报错信息)
/// get_time OK! {...}
/// Test sleep OK!
#[no_mangle] #[no_mangle]
fn main() -> i32 { fn main() -> i32 {
let current_timer = get_time(); let current_time = get_time();
let wait_for = current_timer + 3000; assert!(current_time > 0);
println!("get_time OK! {}", current_time);
let wait_for = current_time + 3000;
while get_time() < wait_for { while get_time() < wait_for {
yield_(); yield_();
} }
println!("Test sleep OK!"); println!("Test sleep OK!");
0 0
} }

22
user/src/bin/03sleep1.rs Normal file
View file

@ -0,0 +1,22 @@
#![no_std]
#![no_main]
#[macro_use]
extern crate user_lib;
use user_lib::{get_time, sleep};
#[no_mangle]
pub fn main() -> i32 {
let start = get_time();
println!("current time_msec = {}", start);
sleep(100);
let end = get_time();
println!(
"time_msec = {} after sleeping 100 ticks, delta = {}ms!",
end,
end - start
);
println!("Test sleep1 passed!");
0
}

View file

@ -20,6 +20,19 @@ fn main() -> i32 {
panic!("Cannot find main!"); panic!("Cannot find main!");
} }
#[repr(C)]
#[derive(Debug, Default)]
pub struct TimeVal {
pub sec: usize,
pub usec: usize,
}
impl TimeVal {
pub fn new() -> Self {
Self::default()
}
}
use syscall::*; use syscall::*;
pub fn write(fd: usize, buf: &[u8]) -> isize { pub fn write(fd: usize, buf: &[u8]) -> isize {
@ -32,7 +45,11 @@ pub fn yield_() -> isize {
sys_yield() sys_yield()
} }
pub fn get_time() -> isize { pub fn get_time() -> isize {
sys_get_time() let time = TimeVal::new();
match sys_get_time(&time, 0) {
0 => ((time.sec & 0xffff) * 1000 + time.usec / 1000) as isize,
_ => -1,
}
} }
pub fn sbrk(size: i32) -> isize { pub fn sbrk(size: i32) -> isize {
@ -46,3 +63,10 @@ pub fn munmap(start: usize, len: usize) -> isize {
pub fn mmap(start: usize, len: usize, prot: usize) -> isize { pub fn mmap(start: usize, len: usize, prot: usize) -> isize {
sys_mmap(start, len, prot) sys_mmap(start, len, prot)
} }
pub fn sleep(period_ms: usize) {
let start = get_time();
while get_time() < start + period_ms as isize {
sys_yield();
}
}

View file

@ -1,12 +1,15 @@
use core::arch::asm; use core::arch::asm;
use crate::TimeVal;
const SYSCALL_WRITE: usize = 64; const SYSCALL_WRITE: usize = 64;
const SYSCALL_EXIT: usize = 93; const SYSCALL_EXIT: usize = 93;
const SYSCALL_YIELD: usize = 124; const SYSCALL_YIELD: usize = 124;
const SYSCALL_GET_TIME: usize = 169; const SYSCALL_GET_TIME: usize = 169;
const SYSCALL_GETTIMEOFDAY: usize = 169;
const SYSCALL_SBRK: usize = 214; const SYSCALL_SBRK: usize = 214;
pub const SYSCALL_MMAP: usize = 222; const SYSCALL_MMAP: usize = 222;
pub const SYSCALL_MUNMAP: usize = 215; const SYSCALL_MUNMAP: usize = 215;
fn syscall(id: usize, args: [usize; 3]) -> isize { fn syscall(id: usize, args: [usize; 3]) -> isize {
let mut ret: isize; let mut ret: isize;
@ -34,8 +37,8 @@ pub fn sys_yield() -> isize {
syscall(SYSCALL_YIELD, [0, 0, 0]) syscall(SYSCALL_YIELD, [0, 0, 0])
} }
pub fn sys_get_time() -> isize { pub fn sys_get_time(time: &TimeVal, tz: usize) -> isize {
syscall(SYSCALL_GET_TIME, [0, 0, 0]) syscall(SYSCALL_GETTIMEOFDAY, [time as *const _ as usize, tz, 0])
} }
pub fn sys_sbrk(size: i32) -> isize { pub fn sys_sbrk(size: i32) -> isize {