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

Fix atomic counting in socketpair's bind_until_success

Browse Source
This commit is contained in:
He Sun 2019-12-02 20:14:10 +08:00 committed by Tate, Hongliang Tian
parent 3c1378b7eb
commit b91566d486
1 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/libos/src/fs/unix_socket.rs
View File

@ -140,19 +140,13 @@ impl UnixSocketFile {
}
fn bind_until_success(&self) -> String {
let mut path = SOCK_PATH_PREFIX.to_string();
let mut index = SOCKETPAIR_NUM.load(Ordering::SeqCst);
path.push_str(&index.to_string());
while self.bind(&path).is_err() {
if index == std::usize::MAX {
SOCKETPAIR_NUM.store(0, Ordering::SeqCst); //flip SOCKETPAIR_NUM
loop {
let sock_path_suffix = SOCKETPAIR_NUM.fetch_add(1, Ordering::SeqCst);
let sock_path = format!("{}{}", SOCK_PATH_PREFIX, sock_path_suffix);
if self.bind(&sock_path).is_ok() {
return sock_path;
}
index += 1;
path = SOCK_PATH_PREFIX.to_string();
path.push_str(&index.to_string());
}
SOCKETPAIR_NUM.fetch_max(index + 1, Ordering::SeqCst);
path
}
}