From b7ff0d1ffe707d593584db7fb32283d494f40255 Mon Sep 17 00:00:00 2001
From: Mark Lodato <mlodato517@gmail.com>
Date: Fri, 19 Jun 2020 15:38:11 -0400
Subject: [PATCH] Add bare-bones WS server for autobahn testing (#331)

---
 Cargo.toml                                   |  1 +
 websocket-autobahn/Cargo.toml                | 16 ++++++
 websocket-autobahn/README.md                 | 33 ++++++++++++
 websocket-autobahn/config/fuzzingclient.json | 15 ++++++
 websocket-autobahn/reports/.gitignore        |  2 +
 websocket-autobahn/src/main.rs               | 55 ++++++++++++++++++++
 6 files changed, 122 insertions(+)
 create mode 100644 websocket-autobahn/Cargo.toml
 create mode 100644 websocket-autobahn/README.md
 create mode 100644 websocket-autobahn/config/fuzzingclient.json
 create mode 100644 websocket-autobahn/reports/.gitignore
 create mode 100644 websocket-autobahn/src/main.rs

diff --git a/Cargo.toml b/Cargo.toml
index b669c145..cb65c0fa 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -46,6 +46,7 @@ members = [
    "unix-socket",
    "web-cors/backend",
    "websocket",
+   "websocket-autobahn",
    "websocket-chat",
    "websocket-chat-broker",
    "websocket-tcp-chat",
diff --git a/websocket-autobahn/Cargo.toml b/websocket-autobahn/Cargo.toml
new file mode 100644
index 00000000..04111d71
--- /dev/null
+++ b/websocket-autobahn/Cargo.toml
@@ -0,0 +1,16 @@
+[package]
+name = "websocket-autobahn"
+version = "2.0.0"
+authors = ["Mark Lodato <mlodato517@gmail.com"]
+edition = "2018"
+
+[[bin]]
+name = "websocket-autobahn-server"
+path = "src/main.rs"
+
+[dependencies]
+actix = "0.9.0"
+actix-web = "2.0.0"
+actix-web-actors = "2.0.0"
+actix-rt = "1.0.0"
+env_logger = "0.7"
diff --git a/websocket-autobahn/README.md b/websocket-autobahn/README.md
new file mode 100644
index 00000000..d4564ebd
--- /dev/null
+++ b/websocket-autobahn/README.md
@@ -0,0 +1,33 @@
+# websocket
+
+Websocket server for autobahn suite testing.
+
+## Usage
+
+### server
+
+```bash
+cd examples/websocket-autobahn
+cargo run --bin websocket-autobahn-server
+```
+
+### Running Autobahn Test Suite
+
+Running the autobahn test suite is easiest using the docker image
+as explained on the [autobahn-testsuite repo](https://github.com/crossbario/autobahn-testsuite#using-the-testsuite-docker-image).
+
+First, start a server (see above). Then, run the test suite in fuzzingclient mode:
+
+```bash
+docker run -it --rm \
+    -v "${PWD}/config:/config" \
+    -v "${PWD}/reports:/reports" \
+    --network host \
+    --name autobahn \
+    crossbario/autobahn-testsuite \
+    wstest \
+    --spec /config/fuzzingclient.json \
+    --mode fuzzingclient
+```
+
+Results are written to the `reports/servers` directory for viewing.
diff --git a/websocket-autobahn/config/fuzzingclient.json b/websocket-autobahn/config/fuzzingclient.json
new file mode 100644
index 00000000..d4b765fb
--- /dev/null
+++ b/websocket-autobahn/config/fuzzingclient.json
@@ -0,0 +1,15 @@
+{
+  "outdir": "./reports/servers",
+  "servers": [
+    {
+      "agent": "actix-web-actors",
+      "url": "ws://host.docker.internal:9001"
+    }
+  ],
+  "cases": ["*"],
+  "exclude-cases": [
+    "12.*",
+    "13.*"
+  ],
+  "exclude-agent-cases": {}
+}
diff --git a/websocket-autobahn/reports/.gitignore b/websocket-autobahn/reports/.gitignore
new file mode 100644
index 00000000..d6b7ef32
--- /dev/null
+++ b/websocket-autobahn/reports/.gitignore
@@ -0,0 +1,2 @@
+*
+!.gitignore
diff --git a/websocket-autobahn/src/main.rs b/websocket-autobahn/src/main.rs
new file mode 100644
index 00000000..85522be8
--- /dev/null
+++ b/websocket-autobahn/src/main.rs
@@ -0,0 +1,55 @@
+use actix::prelude::*;
+use actix_web::{middleware, web, App, Error, HttpRequest, HttpResponse, HttpServer};
+use actix_web_actors::ws;
+
+async fn ws_index(r: HttpRequest, stream: web::Payload) -> Result<HttpResponse, Error> {
+    ws::start(WebSocket::new(), &r, stream)
+}
+
+struct WebSocket {}
+
+impl Actor for WebSocket {
+    type Context = ws::WebsocketContext<Self>;
+
+    fn started(&mut self, _ctx: &mut Self::Context) {}
+}
+
+impl StreamHandler<Result<ws::Message, ws::ProtocolError>> for WebSocket {
+    fn handle(
+        &mut self,
+        msg: Result<ws::Message, ws::ProtocolError>,
+        ctx: &mut Self::Context,
+    ) {
+        if let Ok(msg) = msg {
+            match msg {
+                ws::Message::Text(text) => ctx.text(text),
+                ws::Message::Binary(bin) => ctx.binary(bin),
+                ws::Message::Ping(bytes) => ctx.pong(&bytes),
+                _ => {}
+            }
+        } else {
+            ctx.stop();
+        }
+    }
+}
+
+impl WebSocket {
+    fn new() -> Self {
+        Self {}
+    }
+}
+
+#[actix_rt::main]
+async fn main() -> std::io::Result<()> {
+    std::env::set_var("RUST_LOG", "actix_server=info,actix_web=info");
+    env_logger::init();
+
+    HttpServer::new(|| {
+        App::new()
+            .wrap(middleware::Logger::default())
+            .service(web::resource("/").route(web::get().to(ws_index)))
+    })
+    .bind("127.0.0.1:9001")?
+    .run()
+    .await
+}