From 1c67ff50e39b0650f93acdaba35dfd29110a0537 Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Fri, 17 Apr 2020 14:43:14 +0800
Subject: [PATCH] close connections

---
 runtime/src/comms.rs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/runtime/src/comms.rs b/runtime/src/comms.rs
index cf5cc88..651ff7e 100644
--- a/runtime/src/comms.rs
+++ b/runtime/src/comms.rs
@@ -141,7 +141,7 @@ async fn send_header(stream: &TcpStream, reply: Reply) -> Result<()> {
     Ok(())
 }
 
-async fn handle_connection(stream: TcpStream, control: Rc<RefCell<Option<kernel::Control>>>) -> Result<()> {
+async fn handle_connection(stream: &TcpStream, control: Rc<RefCell<kernel::Control>>) -> Result<()> {
     expect(&stream, b"ARTIQ coredev\n").await?;
     loop {
         expect(&stream, &[0x5a, 0x5a, 0x5a, 0x5a]).await?;
@@ -224,9 +224,11 @@ pub fn main() {
             while let stream = TcpStream::accept(1381, 2048, 2048).await.unwrap() {
                 let control = control.clone();
                 task::spawn(async {
-                    let _ = handle_connection(stream, control)
+                    let _ = handle_connection(&stream, control)
                         .await
                         .map_err(|e| println!("Connection: {}", e));
+                    let _ = stream.flush().await;
+                    let _ = stream.abort().await;
                 });
             }
         }