move test binaries to examples/

These aren't installed and are only used for manual testing, so there's no reason to force them to be built all the time. Signed-off-by: Wolfgang Bumiller <w.bumiller@proxmox.com>
2020-06-16 13:32:23 +02:00
parent 929a13b357
commit d19c96d507
15 changed files with 0 additions and 0 deletions
--- a/examples/h2s-server.rs
+++ b/examples/h2s-server.rs
@ -0,0 +1,81 @@
+use std::sync::Arc;
+
+use anyhow::{format_err, Error};
+use futures::*;
+use hyper::{Request, Response, Body};
+use openssl::ssl::{SslMethod, SslAcceptor, SslFiletype};
+use tokio::net::{TcpListener, TcpStream};
+
+use proxmox_backup::configdir;
+
+// Simple H2 server to test H2 speed with h2s-client.rs
+
+fn main() -> Result<(), Error> {
+    proxmox_backup::tools::runtime::main(run())
+}
+
+async fn run() -> Result<(), Error> {
+    let key_path = configdir!("/proxy.key");
+    let cert_path = configdir!("/proxy.pem");
+
+    let mut acceptor = SslAcceptor::mozilla_intermediate(SslMethod::tls()).unwrap();
+    acceptor.set_private_key_file(key_path, SslFiletype::PEM)
+        .map_err(|err| format_err!("unable to read proxy key {} - {}", key_path, err))?;
+    acceptor.set_certificate_chain_file(cert_path)
+        .map_err(|err| format_err!("unable to read proxy cert {} - {}", cert_path, err))?;
+    acceptor.check_private_key().unwrap();
+
+    let acceptor = Arc::new(acceptor.build());
+
+    let mut listener = TcpListener::bind(std::net::SocketAddr::from(([127,0,0,1], 8008))).await?;
+
+    println!("listening on {:?}", listener.local_addr());
+
+    loop {
+        let (socket, _addr) = listener.accept().await?;
+        tokio::spawn(handle_connection(socket, Arc::clone(&acceptor))
+            .map(|res| {
+                if let Err(err) = res {
+                    eprintln!("Error: {}", err);
+                }
+            }));
+    }
+}
+
+async fn handle_connection(
+    socket: TcpStream,
+    acceptor: Arc<SslAcceptor>,
+) -> Result<(), Error> {
+    socket.set_nodelay(true).unwrap();
+    socket.set_send_buffer_size(1024*1024).unwrap();
+    socket.set_recv_buffer_size(1024*1024).unwrap();
+
+    let socket = tokio_openssl::accept(acceptor.as_ref(), socket).await?;
+
+    let mut http = hyper::server::conn::Http::new();
+    http.http2_only(true);
+    // increase window size: todo - find optiomal size
+    let max_window_size = (1 << 31) - 2;
+    http.http2_initial_stream_window_size(max_window_size);
+    http.http2_initial_connection_window_size(max_window_size);
+
+    let service = hyper::service::service_fn(|_req: Request<Body>| {
+        println!("Got request");
+        let buffer = vec![65u8; 1024*1024]; // nonsense [A,A,A,A...]
+        let body = Body::from(buffer);
+
+        let response = Response::builder()
+            .status(http::StatusCode::OK)
+            .header(http::header::CONTENT_TYPE, "application/octet-stream")
+            .body(body)
+            .unwrap();
+        future::ok::<_, Error>(response)
+    });
+
+    http.serve_connection(socket, service)
+        .map_err(Error::from)
+        .await?;
+
+    println!("H2 connection CLOSE !");
+    Ok(())
+}