From 6caa02c63ff406237fea5a217c010c66849201cd Mon Sep 17 00:00:00 2001
From: Thulinma <jaron@vietors.com>
Date: Tue, 24 May 2016 11:12:54 +0200
Subject: [PATCH] Prevent sessions from becoming unset.

---
 src/controller/controller_statistics.cpp | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/controller/controller_statistics.cpp b/src/controller/controller_statistics.cpp
index 513d24bf..8f494ced 100644
--- a/src/controller/controller_statistics.cpp
+++ b/src/controller/controller_statistics.cpp
@@ -297,6 +297,15 @@ void Controller::statSession::update(unsigned long index, IPC::statExchange & da
     }else{
       streamStats[streamName].upBytes += currUp - prevUp;
       streamStats[streamName].downBytes += currDown - prevDown;
+      if (sessionType == SESS_UNSET){
+        if (data.connector() == "INPUT"){
+          sessionType = SESS_INPUT;
+        }else if (data.connector() == "OUTPUT"){
+          sessionType = SESS_OUTPUT;
+        }else{
+          sessionType = SESS_VIEWER;
+        }
+      }
     }
   }
 }