[pbs-devel] [PATCH backup 1/2] proxy: get rid of some unsafe pins
Wolfgang Bumiller
w.bumiller at proxmox.com
Wed May 12 09:29:30 CEST 2021
It's easier to follow by just boxing the futures with
Box::pin.
Signed-off-by: Wolfgang Bumiller <w.bumiller at proxmox.com>
---
src/bin/proxmox-backup-proxy.rs | 19 ++++++-------------
1 file changed, 6 insertions(+), 13 deletions(-)
diff --git a/src/bin/proxmox-backup-proxy.rs b/src/bin/proxmox-backup-proxy.rs
index fc773459..e8c31c4d 100644
--- a/src/bin/proxmox-backup-proxy.rs
+++ b/src/bin/proxmox-backup-proxy.rs
@@ -1,6 +1,5 @@
use std::sync::Arc;
use std::path::{Path, PathBuf};
-use std::pin::Pin;
use std::os::unix::io::AsRawFd;
use anyhow::{bail, format_err, Error};
@@ -220,22 +219,16 @@ async fn accept_connection(
) {
let accept_counter = Arc::new(());
- // Note that these must not be moved out/modified directly, they get pinned in the loop and
- // "rearmed" after waking up:
- let mut reload_tls = notify_tls_cert_reload.notified();
- let mut accept = listener.accept();
+ let mut reload_tls = Box::pin(notify_tls_cert_reload.notified());
+ let mut accept = Box::pin(listener.accept());
loop {
let sock;
- // normally we'd use `tokio::pin!()` but we need this to happen outside the loop and we
- // need to be able to "rearm" the futures:
- let reload_tls_pin = unsafe { Pin::new_unchecked(&mut reload_tls) };
- let accept_pin = unsafe { Pin::new_unchecked(&mut accept) };
tokio::select! {
- _ = reload_tls_pin => {
+ _ = &mut reload_tls => {
// rearm the notification:
- reload_tls = notify_tls_cert_reload.notified();
+ reload_tls = Box::pin(notify_tls_cert_reload.notified());
log::info!("reloading certificate");
match make_tls_acceptor() {
@@ -244,14 +237,14 @@ async fn accept_connection(
}
continue;
}
- res = accept_pin => match res {
+ res = &mut accept => match res {
Err(err) => {
eprintln!("error accepting tcp connection: {}", err);
continue;
}
Ok((new_sock, _addr)) => {
// rearm the accept future:
- accept = listener.accept();
+ accept = Box::pin(listener.accept());
sock = new_sock;
}
--
2.20.1
More information about the pbs-devel
mailing list