DragonOS-Community/DragonOS
1
1
Fork 0
mirror of https://github.com/DragonOS-Community/DragonOS.git synced 2025-06-08 14:16:47 +00:00
Code Issues Releases Wiki Activity

修复get_ramdom的长度错误问题() (#677)

Browse Source
This commit is contained in:
Val213 2024-03-31 18:01:32 +08:00 committed by GitHub
parent 56cc4dbe27
commit 7d580ef99d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
kernel/src/syscall/misc.rs
View File

@ -1,11 +1,11 @@
use alloc::vec::Vec;
use system_error::SystemError;
use crate::{
arch::{mm::LockedFrameAllocator, rand::rand},
libs::rand::GRandFlags,
mm::allocator::page_frame::FrameAllocator,
};
use alloc::vec::Vec;
use core::cmp;
use system_error::SystemError;
use super::{user_access::UserBufferWriter, Syscall};
@ -63,7 +63,6 @@ impl Syscall {
}
/// ## 将随机字节填入buf
///
/// ### 该系统调用与linux不一致，因为目前没有其他随机源
pub fn get_random(buf: *mut u8, len: usize, flags: GRandFlags) -> Result<usize, SystemError> {
if flags.bits() == (GRandFlags::GRND_INSECURE.bits() | GRandFlags::GRND_RANDOM.bits()) {
@ -75,8 +74,11 @@ impl Syscall {
let mut ret = Vec::new();
let mut count = 0;
while count < len {
// 对 len - count 的长度进行判断，remain_len 小于4则循环次数和 remain_len 相等
let remain_len = len - count;
let step = cmp::min(remain_len, 4);
let rand = rand();
for offset in 0..4 {
for offset in 0..step {
ret.push((rand >> (offset * 2)) as u8);
count += 1;
}