mirror of
https://github.com/asterinas/asterinas.git
synced 2025-06-25 10:23:23 +00:00
Remove the shim kernel crate
This commit is contained in:
committed by
Tate, Hongliang Tian
parent
d76c7a5b1e
commit
dafd16075f
85
kernel/src/syscall/accept.rs
Normal file
85
kernel/src/syscall/accept.rs
Normal file
@ -0,0 +1,85 @@
|
||||
// SPDX-License-Identifier: MPL-2.0
|
||||
|
||||
use super::SyscallReturn;
|
||||
use crate::{
|
||||
fs::{
|
||||
file_table::{FdFlags, FileDesc},
|
||||
utils::{CreationFlags, StatusFlags},
|
||||
},
|
||||
prelude::*,
|
||||
util::net::{get_socket_from_fd, write_socket_addr_to_user},
|
||||
};
|
||||
|
||||
pub fn sys_accept(
|
||||
sockfd: FileDesc,
|
||||
sockaddr_ptr: Vaddr,
|
||||
addrlen_ptr: Vaddr,
|
||||
ctx: &Context,
|
||||
) -> Result<SyscallReturn> {
|
||||
debug!("sockfd = {sockfd}, sockaddr_ptr = 0x{sockaddr_ptr:x}, addrlen_ptr = 0x{addrlen_ptr:x}");
|
||||
|
||||
let fd = do_accept(sockfd, sockaddr_ptr, addrlen_ptr, Flags::empty(), ctx)?;
|
||||
Ok(SyscallReturn::Return(fd as _))
|
||||
}
|
||||
|
||||
pub fn sys_accept4(
|
||||
sockfd: FileDesc,
|
||||
sockaddr_ptr: Vaddr,
|
||||
addrlen_ptr: Vaddr,
|
||||
flags: u32,
|
||||
ctx: &Context,
|
||||
) -> Result<SyscallReturn> {
|
||||
trace!("raw flags = 0x{:x}", flags);
|
||||
let flags = Flags::from_bits_truncate(flags);
|
||||
debug!(
|
||||
"sockfd = {}, sockaddr_ptr = 0x{:x}, addrlen_ptr = 0x{:x}, flags = {:?}",
|
||||
sockfd, sockaddr_ptr, addrlen_ptr, flags
|
||||
);
|
||||
|
||||
let fd = do_accept(sockfd, sockaddr_ptr, addrlen_ptr, flags, ctx)?;
|
||||
Ok(SyscallReturn::Return(fd as _))
|
||||
}
|
||||
|
||||
fn do_accept(
|
||||
sockfd: FileDesc,
|
||||
sockaddr_ptr: Vaddr,
|
||||
addrlen_ptr: Vaddr,
|
||||
flags: Flags,
|
||||
ctx: &Context,
|
||||
) -> Result<FileDesc> {
|
||||
let (connected_socket, socket_addr) = {
|
||||
let socket = get_socket_from_fd(sockfd)?;
|
||||
socket.accept()?
|
||||
};
|
||||
|
||||
if flags.contains(Flags::SOCK_NONBLOCK) {
|
||||
connected_socket.set_status_flags(StatusFlags::O_NONBLOCK)?;
|
||||
}
|
||||
|
||||
let fd_flags = if flags.contains(Flags::SOCK_CLOEXEC) {
|
||||
FdFlags::CLOEXEC
|
||||
} else {
|
||||
FdFlags::empty()
|
||||
};
|
||||
|
||||
if sockaddr_ptr != 0 {
|
||||
write_socket_addr_to_user(&socket_addr, sockaddr_ptr, addrlen_ptr)?;
|
||||
}
|
||||
|
||||
let fd = {
|
||||
let mut file_table = ctx.process.file_table().lock();
|
||||
file_table.insert(connected_socket, fd_flags)
|
||||
};
|
||||
|
||||
Ok(fd)
|
||||
}
|
||||
|
||||
bitflags! {
|
||||
struct Flags: u32 {
|
||||
const SOCK_NONBLOCK = NONBLOCK;
|
||||
const SOCK_CLOEXEC = CLOEXEC;
|
||||
}
|
||||
}
|
||||
|
||||
const NONBLOCK: u32 = StatusFlags::O_NONBLOCK.bits();
|
||||
const CLOEXEC: u32 = CreationFlags::O_CLOEXEC.bits();
|
Reference in New Issue
Block a user