more bug fixes

4 years ago · 28bfa671a1
parent af9a5d9592
commit 28bfa671a1
2 changed files with 2 additions and 2 deletions
--- a/reflector/M17Protocol.cpp
+++ b/reflector/M17Protocol.cpp
@ -170,7 +170,7 @@ void CM17Protocol::Task(void)
 void CM17Protocol::OnDvHeaderPacketIn(std::unique_ptr<CDvHeaderPacket> &Header, const CIp &Ip)
 {
 	// find the stream
-	auto stream = GetStream(Header->GetStreamId());
+	auto stream = GetStream(Header->GetStreamId(), &Ip);
 	if ( stream )
 	{
 		// stream already open
--- a/reflector/Protocol.cpp
+++ b/reflector/Protocol.cpp
@ -156,7 +156,7 @@ std::shared_ptr<CPacketStream> CProtocol::GetStream(uint16_t uiStreamId, const C
 	if (it == m_Streams.end())
 		return nullptr;
-	if (it->second->GetOwnerIp() != nullptr)
+	if (Ip != nullptr && it->second->GetOwnerIp() != nullptr)
 	{
 		if (*Ip == *it->second->GetOwnerIp())
 		{