Cannot compile: &UtpStream is not Send

Author: jason-ni

examples/recv_file.rs

@@ -0,0 +1,38 @@
+use rustorrent::utp::{listener, UtpError, stream::UtpStream, UtpState};
+use std::env;
+use std::sync::Arc;
+use shared_arena::SharedArena;
+
+async fn handle_conn(conn: UtpStream, arena: Arc<SharedArena<[u8; 64]>>) -> Result<(), UtpError> {
+    log::debug!("pending on read stream");
+    let mut buf = arena.alloc_with(|b|{
+        let buf_ref = unsafe { &*(b.as_mut_ptr() as *mut [u8; 64])};
+        buf_ref
+    });
+    let _size = conn.read(buf.as_mut()).await?;
+    log::debug!("received data: {}", String::from_utf8_lossy(buf.as_ref()));
+    Ok(())
+}
+
+async fn run() -> Result<(), UtpError> {
+    let mut args = env::args();
+    args.next();
+    let listen_addr = args.next().expect("listen addr must input");
+    let l = listener::UtpListener::bind(listen_addr).await;
+    let arena: SharedArena<[u8; 64]> = SharedArena::new();
+    let arena_ref = Arc::new(arena);
+    loop {
+        let (conn, peer) = l.accept().await;
+        log::debug!("accepting connection from {:?}", peer);
+        tokio::spawn(handle_conn(conn, arena_ref.clone()));
+    }
+}
+
+fn main() {
+    env_logger::init();
+    let rt = tokio::runtime::Builder::new_current_thread().enable_all().build().unwrap();
+    let fut = async {
+        run().await.map_err(|e| log::debug!("run exit with error: {:?}", e))
+    };
+    let _ = rt.block_on(fut);
+}

examples/send_file.rs

@@ -0,0 +1,29 @@
+use std::env;
+use rustorrent::utp::{listener, UtpError};
+use std::net::SocketAddr;
+use tokio::fs;
+use tokio::io::AsyncWriteExt;
+
+async fn run() -> Result<(), UtpError> {
+    let mut args = env::args();
+    args.next();
+    let listen_ip = args.next().expect("listen ip must input");
+    let remote_addr_arg = args.next().expect("remote address must input");
+    let remote_addr: SocketAddr = remote_addr_arg.parse().unwrap();
+    let file_name = args.next().expect("send file name must input");
+    log::debug!("before open");
+    let mut f = fs::File::open(file_name).await?;
+    log::debug!("before connect");
+    let l = listener::UtpListener::bind(format!("{}:0", listen_ip)).await;
+    let mut conn = l.connect(remote_addr).await?;
+    conn.write_all("hello".as_bytes()).await?;
+    Ok(())
+}
+fn main() {
+    env_logger::init();
+    let rt = tokio::runtime::Builder::new_multi_thread().enable_all().build().unwrap();
+    let fut = async {
+        run().await.map_err(|e| log::debug!("run exit with error: {:?}", e));
+    };
+    let _ = rt.block_on(fut);
+}