Samuka007/asterinas
1
0
Fork 0
mirror of https://github.com/asterinas/asterinas.git synced 2025-06-17 12:47:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Avoid nested locks

Browse Source
This commit is contained in:
Ruihan Li 2024-07-26 22:42:49 +08:00 committed by Tate, Hongliang Tian
parent 5445a26ec5
commit 8bcb5a2a5f
2 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
kernel/aster-nix/src/net/socket/unix/stream/init.rs
View File

@ -14,21 +14,20 @@ use crate::{
};
pub(super) struct Init {
addr: Mutex<Option<UnixSocketAddrBound>>,
addr: Option<UnixSocketAddrBound>,
pollee: Pollee,
}
impl Init {
pub(super) fn new() -> Self {
Self {
addr: Mutex::new(None),
addr: None,
pollee: Pollee::new(IoEvents::empty()),
}
}
pub(super) fn bind(&self, addr_to_bind: &UnixSocketAddr) -> Result<()> {
let mut addr = self.addr.lock();
if addr.is_some() {
pub(super) fn bind(&mut self, addr_to_bind: &UnixSocketAddr) -> Result<()> {
if self.addr.is_some() {
return_errno_with_message!(Errno::EINVAL, "the socket is already bound");
}
@ -40,28 +39,28 @@ impl Init {
UnixSocketAddrBound::Path(dentry)
}
};
self.addr = Some(bound_addr);
*addr = Some(bound_addr);
Ok(())
}
pub(super) fn connect(&self, remote_addr: &UnixSocketAddrBound) -> Result<Connected> {
let addr = self.addr();
if let Some(ref addr) = addr {
if let Some(addr) = addr {
if *addr == *remote_addr {
return_errno_with_message!(Errno::EINVAL, "try to connect to self is invalid");
}
}
let (this_end, remote_end) = Endpoint::new_pair(addr, Some(remote_addr.clone()));
let (this_end, remote_end) = Endpoint::new_pair(addr.cloned(), Some(remote_addr.clone()));
push_incoming(remote_addr, remote_end)?;
Ok(Connected::new(this_end))
}
pub(super) fn addr(&self) -> Option<UnixSocketAddrBound> {
self.addr.lock().clone()
pub(super) fn addr(&self) -> Option<&UnixSocketAddrBound> {
self.addr.as_ref()
}
pub(super) fn poll(&self, mask: IoEvents, poller: Option<&mut Poller>) -> IoEvents {

6
kernel/aster-nix/src/net/socket/unix/stream/socket.rs
View File

@ -74,7 +74,7 @@ impl UnixStreamSocket {
fn bound_addr(&self) -> Option<UnixSocketAddrBound> {
let state = self.state.read();
match &*state {
State::Init(init) => init.addr(),
State::Init(init) => init.addr().cloned(),
State::Listen(listen) => Some(listen.addr().clone()),
State::Connected(connected) => connected.addr().cloned(),
}
@ -197,7 +197,7 @@ impl Socket for UnixStreamSocket {
fn bind(&self, socket_addr: SocketAddr) -> Result<()> {
let addr = UnixSocketAddr::try_from(socket_addr)?;
match &*self.state.read() {
match &mut *self.state.write() {
State::Init(init) => init.bind(&addr),
_ => return_errno_with_message!(
Errno::EINVAL,
@ -273,7 +273,7 @@ impl Socket for UnixStreamSocket {
fn addr(&self) -> Result<SocketAddr> {
let addr = match &*self.state.read() {
State::Init(init) => init.addr(),
State::Init(init) => init.addr().cloned(),
State::Listen(listen) => Some(listen.addr().clone()),
State::Connected(connected) => connected.addr().cloned(),
};