From 0b47ac75f04afae0aa520152676fe3fb3cc3f966 Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Mon, 20 Jul 2020 19:09:56 +0800
Subject: [PATCH] comms,mgmt: log connections as soon as they arrive

---
 src/runtime/src/comms.rs | 2 +-
 src/runtime/src/mgmt.rs  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/runtime/src/comms.rs b/src/runtime/src/comms.rs
index 26edddd..882998a 100644
--- a/src/runtime/src/comms.rs
+++ b/src/runtime/src/comms.rs
@@ -266,7 +266,6 @@ async fn handle_connection(stream: &TcpStream, control: Rc<RefCell<kernel::Contr
     if !expect(stream, b"ARTIQ coredev\n").await? {
         return Err(Error::UnexpectedPattern);
     }
-    info!("received connection");
     loop {
         let request = read_request(stream, true).await?;
         if request.is_none() {
@@ -355,6 +354,7 @@ pub fn main(timer: GlobalTimer, cfg: &config::Config) {
             let stream = TcpStream::accept(1381, 2048, 2048).await.unwrap();
             let control = control.clone();
             task::spawn(async {
+                info!("received connection");
                 let _ = handle_connection(&stream, control)
                     .await
                     .map_err(|e| warn!("connection terminated: {}", e));
diff --git a/src/runtime/src/mgmt.rs b/src/runtime/src/mgmt.rs
index c53b022..c07f99e 100644
--- a/src/runtime/src/mgmt.rs
+++ b/src/runtime/src/mgmt.rs
@@ -33,7 +33,6 @@ async fn get_logger_buffer() -> LogBufferRef<'static> {
 
 async fn handle_connection(stream: &mut TcpStream, pull_id: Rc<RefCell<u32>>) -> Result<(), Error> {
     Request::read_magic(stream).await?;
-    info!("received connection");
 
     loop {
         let req = Request::read_from(stream).await;
@@ -104,6 +103,7 @@ pub fn start() {
             let mut stream = TcpStream::accept(1380, 2048, 2048).await.unwrap();
             let pull_id = pull_id.clone();
             task::spawn(async move {
+                info!("received connection");
                 let _ = handle_connection(&mut stream, pull_id)
                     .await
                     .map_err(|e| warn!("connection terminated: {:?}", e));