[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