From 7b1ef6c76e4cb7f49022f15b5f1bc69b0db03364 Mon Sep 17 00:00:00 2001
From: Thulinma <jaron@vietors.com>
Date: Thu, 2 Feb 2017 11:31:00 +0100
Subject: [PATCH] Fixed RTSP input not counting as input, fixed UDP-style input
 disconnect bug

# Conflicts:
#	src/output/output.h
---
 src/output/output.h        | 4 ++--
 src/output/output_rtsp.cpp | 9 +++++++++
 src/output/output_rtsp.h   | 1 +
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/output/output.h b/src/output/output.h
index d7d4025e..0a730acf 100644
--- a/src/output/output.h
+++ b/src/output/output.h
@@ -95,12 +95,12 @@ namespace Mist {
       int pageNumMax(long unsigned int trackId);
       bool isRecordingToFile;
       unsigned int lastStats;///<Time of last sending of stats.
-      long long unsigned int firstTime;///< Time of first packet after last seek. Used for real-time sending.
       std::map<unsigned long, unsigned long> nxtKeyNum;///< Contains the number of the next key, for page seeking purposes.
       std::set<sortedPageInfo> buffer;///< A sorted list of next-to-be-loaded packets.
       bool sought;///<If a seek has been done, this is set to true. Used for seeking on prepareNext().
-      uint64_t lastRecv;
     protected://these are to be messed with by child classes
+      uint64_t lastRecv;
+      long long unsigned int firstTime;///< Time of first packet after last seek. Used for real-time sending.
       virtual std::string getConnectedHost();
       virtual std::string getConnectedBinHost();
       virtual std::string getStatsName();
diff --git a/src/output/output_rtsp.cpp b/src/output/output_rtsp.cpp
index a329ff82..96e64053 100644
--- a/src/output/output_rtsp.cpp
+++ b/src/output/output_rtsp.cpp
@@ -147,6 +147,14 @@ namespace Mist {
     
   }
 
+  std::string OutRTSP::getStatsName(){
+    if (isPushing){
+      return "INPUT";
+    }else{
+      return Output::getStatsName();
+    }
+  }
+
   /// This request handler also checks for UDP packets
   void OutRTSP::requestHandler(){
     if (!expectTCP){
@@ -460,6 +468,7 @@ namespace Mist {
           //wrong sending port, ignore packet
           continue;
         }
+        lastRecv = Util::epoch();//prevent disconnect of idle TCP connection when using UDP
         RTP::Packet pack(s.data, s.data_len);
         if (!it->second.rtpSeq){it->second.rtpSeq = pack.getSequence();}
         //packet is very early - assume dropped after 10 packets
diff --git a/src/output/output_rtsp.h b/src/output/output_rtsp.h
index 4d427abd..eb8137be 100644
--- a/src/output/output_rtsp.h
+++ b/src/output/output_rtsp.h
@@ -170,6 +170,7 @@ namespace Mist {
       void handleIncomingRTP(const uint64_t track, const RTP::Packet & pkt);
       void h264MultiParse(uint64_t ts, const uint64_t track, char * buffer, const uint32_t len);
       void h264Packet(uint64_t ts, const uint64_t track, const char * buffer, const uint32_t len, bool isKey);
+      std::string getStatsName();
   };
 }