ws load test: rewrite with glommio and futures-rustls

aquatic_ws_load_test/src/main.rs

@@ -2,6 +2,7 @@ use std::sync::{atomic::Ordering, Arc};
 use std::thread;
 use std::time::{Duration, Instant};
 
+use glommio::LocalExecutorBuilder;
 use rand::prelude::*;
 use rand_distr::Pareto;
 
@@ -48,11 +49,18 @@ fn run(config: Config) -> ::anyhow::Result<()> {
         pareto: Arc::new(pareto),
     };
 
+    let tls_config = create_tls_config().unwrap();
+
     for _ in 0..config.num_workers {
         let config = config.clone();
+        let tls_config = tls_config.clone();
         let state = state.clone();
 
-        thread::spawn(move || run_socket_thread(&config, state));
+        LocalExecutorBuilder::default()
+            .spawn(|| async move {
+                run_socket_thread(config, tls_config, state).await.unwrap();
+            })
+            .unwrap();
     }
 
     monitor_statistics(state, &config);
@@ -60,6 +68,36 @@ fn run(config: Config) -> ::anyhow::Result<()> {
     Ok(())
 }
 
+struct FakeCertificateVerifier;
+
+impl rustls::client::ServerCertVerifier for FakeCertificateVerifier {
+    fn verify_server_cert(
+        &self,
+        _end_entity: &rustls::Certificate,
+        _intermediates: &[rustls::Certificate],
+        _server_name: &rustls::ServerName,
+        _scts: &mut dyn Iterator<Item = &[u8]>,
+        _ocsp_response: &[u8],
+        _now: std::time::SystemTime,
+    ) -> Result<rustls::client::ServerCertVerified, rustls::Error> {
+        Ok(rustls::client::ServerCertVerified::assertion())
+    }
+}
+
+fn create_tls_config() -> anyhow::Result<Arc<rustls::ClientConfig>> {
+    let mut config = rustls::ClientConfig::builder()
+        .with_safe_defaults()
+        .with_root_certificates(rustls::RootCertStore::empty())
+        .with_no_client_auth();
+
+    config
+        .dangerous()
+        .set_certificate_verifier(Arc::new(FakeCertificateVerifier));
+
+    Ok(Arc::new(config))
+}
+
+
 fn monitor_statistics(state: LoadTestState, config: &Config) {
     let start_time = Instant::now();
     let mut report_avg_response_vec: Vec<f64> = Vec::new();