Make HTTP server more robust

Signed-off-by: Nick Spinale <[email protected]>
seL4 · Feb 5, 2024 · 6f1de06 · 6f1de06
1 parent e9dd69c
commit 6f1de06
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/crates/examples/microkit/http-server/pds/server/core/src/lib.rs b/crates/examples/microkit/http-server/pds/server/core/src/lib.rs
@@ -129,13 +129,12 @@ async fn use_socket_for_https<D: fat::BlockDevice + 'static, T: fat::TimeSource
     tls_config: Arc<ServerConfig>,
     mut socket: TcpSocket,
 ) -> Result<(), ClosedError<AsyncRustlsError<TcpSocketError>>> {
-    socket.accept(HTTPS_PORT).await.unwrap(); // TODO
-
-    let mut conn = ServerConnector::from(tls_config)
-        .connect(socket)
-        .unwrap()
+    socket
+        .accept(HTTPS_PORT)
         .await
-        .unwrap();
+        .map_err(AsyncRustlsError::TransitError)?;
+
+    let mut conn = ServerConnector::from(tls_config).connect(socket)?.await?;
 
     server.handle_connection(&mut conn).await?;
 

diff --git a/crates/examples/microkit/http-server/pds/server/core/src/server.rs b/crates/examples/microkit/http-server/pds/server/core/src/server.rs
@@ -38,7 +38,9 @@ impl<D: fat::BlockDevice + 'static, T: fat::TimeSource + 'static> Server<D, T> {
             let mut i = 0;
             loop {
                 let n = conn.read(&mut buf[i..]).await?;
-                assert_ne!(n, 0);
+                if n == 0 {
+                    return Err(ClosedError::Closed);
+                }
                 i += n;
                 if is_request_complete(&buf[..i]).unwrap_or(false) {
                     break;