mirror of https://github.com/nostar/urfd.git
Tom Early
4 years ago
parent
436e0b00d9
commit
065be9bf90
@ -1,386 +0,0 @@
|
|||||||
// Copyright © 2015 Jean-Luc Deltombe (LX3JL). All rights reserved.
|
|
||||||
|
|
||||||
// urfd -- The universal reflector
|
|
||||||
// Copyright © 2021 Thomas A. Early N7TAE
|
|
||||||
//
|
|
||||||
// This program is free software: you can redistribute it and/or modify
|
|
||||||
// it under the terms of the GNU General Public License as published by
|
|
||||||
// the Free Software Foundation, either version 3 of the License, or
|
|
||||||
// (at your option) any later version.
|
|
||||||
//
|
|
||||||
// This program is distributed in the hope that it will be useful,
|
|
||||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
// GNU General Public License for more details.
|
|
||||||
//
|
|
||||||
// You should have received a copy of the GNU General Public License
|
|
||||||
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
||||||
|
|
||||||
#include "Main.h"
|
|
||||||
#include "Reflector.h"
|
|
||||||
#include "Transcoder.h"
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
// define
|
|
||||||
|
|
||||||
// status
|
|
||||||
#define STATUS_IDLE 0
|
|
||||||
#define STATUS_LOGGED 1
|
|
||||||
|
|
||||||
// timeout
|
|
||||||
#define AMBED_OPENSTREAM_TIMEOUT 200 // in ms
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
|
|
||||||
CTranscoder g_Transcoder;
|
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
// constructor
|
|
||||||
|
|
||||||
CTranscoder::CTranscoder()
|
|
||||||
{
|
|
||||||
keep_running = true;
|
|
||||||
m_bConnected = false;
|
|
||||||
m_LastKeepaliveTime.start();
|
|
||||||
m_LastActivityTime.start();
|
|
||||||
m_bStreamOpened = false;
|
|
||||||
m_StreamidOpenStream = 0;
|
|
||||||
m_PortOpenStream = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
// destructor
|
|
||||||
|
|
||||||
CTranscoder::~CTranscoder()
|
|
||||||
{
|
|
||||||
Close();
|
|
||||||
}
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
// initialization
|
|
||||||
|
|
||||||
bool CTranscoder::Init(void)
|
|
||||||
{
|
|
||||||
// create address to the transcoder
|
|
||||||
auto s = g_Reflector.GetTranscoderIp();
|
|
||||||
m_Ip.Initialize(strchr(s, ':') ? AF_INET6 : AF_INET, TRANSCODER_PORT, s);
|
|
||||||
|
|
||||||
// does the user not want to use a transcoder?
|
|
||||||
if (0 == strncasecmp(s, "none", 4))
|
|
||||||
{
|
|
||||||
std::cout << "Transcoder will not be enabled because the transcoder IP addess is 'none'" << std::endl;
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
// now open the transcoder port
|
|
||||||
#ifdef LISTEN_IPV4
|
|
||||||
#ifdef LISTEN_IPV6
|
|
||||||
auto paddr = (AF_INET == m_Ip.GetFamily()) ? LISTEN_IPV4 : LISTEN_IPV6;
|
|
||||||
#else
|
|
||||||
auto paddr = LISTEN_IPV4;
|
|
||||||
#endif
|
|
||||||
#else
|
|
||||||
auto paddr = LISTEN_IPV6;
|
|
||||||
#endif
|
|
||||||
CIp tc(m_Ip.GetFamily(), TRANSCODER_PORT, paddr);
|
|
||||||
|
|
||||||
// create our socket
|
|
||||||
if (tc.IsSet())
|
|
||||||
{
|
|
||||||
if (! m_Socket.Open(tc))
|
|
||||||
{
|
|
||||||
std::cerr << "Error opening socket on port UDP" << TRANSCODER_PORT << " on ip " << m_Ip << std::endl;
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
// something bad was specified for the transcoder IP?
|
|
||||||
std::cerr << "Error initializing transcoder port using " << ((AF_INET == m_Ip.GetFamily()) ? "IPv4" : "IPv6") << " on " << paddr << std::endl;
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// start thread
|
|
||||||
keep_running = true;
|
|
||||||
m_Future = std::async(std::launch::async, &CTranscoder::Thread, this);
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
void CTranscoder::Close(void)
|
|
||||||
{
|
|
||||||
// close socket
|
|
||||||
m_Socket.Close();
|
|
||||||
|
|
||||||
// close all streams
|
|
||||||
m_Mutex.lock();
|
|
||||||
m_Streams.clear();
|
|
||||||
m_Mutex.unlock();
|
|
||||||
|
|
||||||
// kill threads
|
|
||||||
keep_running = false;
|
|
||||||
if ( m_Future.valid() )
|
|
||||||
{
|
|
||||||
m_Future.get();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
// thread
|
|
||||||
|
|
||||||
void CTranscoder::Thread()
|
|
||||||
{
|
|
||||||
while (keep_running)
|
|
||||||
{
|
|
||||||
Task();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void CTranscoder::Task(void)
|
|
||||||
{
|
|
||||||
CBuffer Buffer;
|
|
||||||
CIp Ip;
|
|
||||||
uint16_t StreamId;
|
|
||||||
uint16_t Port;
|
|
||||||
|
|
||||||
// anything coming in from codec server ?
|
|
||||||
//if ( (m_Socket.Receive(&Buffer, &Ip, 20) != -1) && (Ip == m_Ip) )
|
|
||||||
if ( m_Socket.Receive(Buffer, Ip, 20) )
|
|
||||||
{
|
|
||||||
m_LastActivityTime.start();
|
|
||||||
|
|
||||||
// crack packet
|
|
||||||
if ( IsValidStreamDescrPacket(Buffer, &StreamId, &Port) )
|
|
||||||
{
|
|
||||||
//std::cout << "Transcoder stream " << (int) StreamId << " descr packet " << std::endl;
|
|
||||||
m_bStreamOpened = true;
|
|
||||||
m_StreamidOpenStream = StreamId;
|
|
||||||
m_PortOpenStream = Port;
|
|
||||||
m_SemaphoreOpenStream.Notify();
|
|
||||||
}
|
|
||||||
else if ( IsValidNoStreamAvailablePacket(Buffer) )
|
|
||||||
{
|
|
||||||
m_bStreamOpened = false;
|
|
||||||
m_SemaphoreOpenStream.Notify();
|
|
||||||
}
|
|
||||||
else if ( IsValidKeepAlivePacket(Buffer) )
|
|
||||||
{
|
|
||||||
if ( !m_bConnected )
|
|
||||||
{
|
|
||||||
std::cout << "Transcoder connected at " << Ip << std::endl;
|
|
||||||
}
|
|
||||||
m_bConnected = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// keep client alive
|
|
||||||
if ( m_LastKeepaliveTime.time() > TRANSCODER_KEEPALIVE_PERIOD )
|
|
||||||
{
|
|
||||||
//
|
|
||||||
HandleKeepalives();
|
|
||||||
|
|
||||||
// update time
|
|
||||||
m_LastKeepaliveTime.start();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
// manage streams
|
|
||||||
|
|
||||||
std::shared_ptr<CCodecStream> CTranscoder::GetCodecStream(CPacketStream *PacketStream, uint8_t uiCodecIn)
|
|
||||||
{
|
|
||||||
CBuffer Buffer;
|
|
||||||
|
|
||||||
// do we need transcoding
|
|
||||||
if ( uiCodecIn != CODEC_NONE )
|
|
||||||
{
|
|
||||||
// are we connected to server
|
|
||||||
if ( m_bConnected )
|
|
||||||
{
|
|
||||||
// yes, post openstream request
|
|
||||||
EncodeOpenstreamPacket(&Buffer, uiCodecIn, (uiCodecIn == CODEC_AMBEPLUS) ? CODEC_AMBE2PLUS : CODEC_AMBEPLUS);
|
|
||||||
m_Socket.Send(Buffer, m_Ip, TRANSCODER_PORT);
|
|
||||||
|
|
||||||
// wait relpy here
|
|
||||||
if ( m_SemaphoreOpenStream.WaitFor(AMBED_OPENSTREAM_TIMEOUT) )
|
|
||||||
{
|
|
||||||
if ( m_bStreamOpened )
|
|
||||||
{
|
|
||||||
std::cout << "ambed openstream ok" << std::endl;
|
|
||||||
|
|
||||||
// create stream object
|
|
||||||
auto stream = std::make_shared<CCodecStream>(PacketStream, m_StreamidOpenStream, uiCodecIn, (uiCodecIn == CODEC_AMBEPLUS) ? CODEC_AMBE2PLUS : CODEC_AMBEPLUS);
|
|
||||||
|
|
||||||
if ( stream ) {
|
|
||||||
if ( stream->Init(m_PortOpenStream) )
|
|
||||||
{
|
|
||||||
// and append to list
|
|
||||||
Lock();
|
|
||||||
m_Streams.push_back(stream);
|
|
||||||
Unlock();
|
|
||||||
return stream;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
// send close packet
|
|
||||||
EncodeClosestreamPacket(&Buffer, stream->GetStreamId());
|
|
||||||
m_Socket.Send(Buffer, m_Ip, TRANSCODER_PORT);
|
|
||||||
stream.reset();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else
|
|
||||||
std::cerr << "ERROR: Unable to make a new CCodecStream" << std::endl;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
std::cerr << "ERROR: Ambed openstream failed (no suitable channel available)" << std::endl;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
std::cout << "ambed openstream timeout" << std::endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
void CTranscoder::ReleaseStream(std::shared_ptr<CCodecStream> stream)
|
|
||||||
{
|
|
||||||
CBuffer Buffer;
|
|
||||||
|
|
||||||
if ( stream )
|
|
||||||
{
|
|
||||||
// look for the stream
|
|
||||||
bool found = false;
|
|
||||||
Lock();
|
|
||||||
{
|
|
||||||
for ( auto it=m_Streams.begin(); it!=m_Streams.end(); it++ )
|
|
||||||
{
|
|
||||||
// compare object pointers
|
|
||||||
if ( (*it) == stream )
|
|
||||||
{
|
|
||||||
// send close packet
|
|
||||||
EncodeClosestreamPacket(&Buffer, (*it)->GetStreamId());
|
|
||||||
m_Socket.Send(Buffer, m_Ip, TRANSCODER_PORT);
|
|
||||||
|
|
||||||
// display stats
|
|
||||||
if ( (*it)->GetPingMin() >= 0.0 )
|
|
||||||
{
|
|
||||||
char sz[256];
|
|
||||||
sprintf(sz, "ambed stats (ms) : %.1f/%.1f/%.1f",
|
|
||||||
(*it)->GetPingMin() * 1000.0,
|
|
||||||
(*it)->GetPingAve() * 1000.0,
|
|
||||||
(*it)->GetPingMax() * 1000.0);
|
|
||||||
std::cout << sz << std::endl;
|
|
||||||
}
|
|
||||||
if ( (*it)->GetTimeoutPackets() > 0 )
|
|
||||||
{
|
|
||||||
char sz[256];
|
|
||||||
sprintf(sz, "ambed %d of %d packets timed out",
|
|
||||||
(*it)->GetTimeoutPackets(),
|
|
||||||
(*it)->GetTotalPackets());
|
|
||||||
std::cout << sz << std::endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
// and close it
|
|
||||||
(*it)->Close();
|
|
||||||
m_Streams.erase(it);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Unlock();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
// keepalive helpers
|
|
||||||
|
|
||||||
void CTranscoder::HandleKeepalives(void)
|
|
||||||
{
|
|
||||||
CBuffer keepalive;
|
|
||||||
|
|
||||||
// send keepalive
|
|
||||||
EncodeKeepAlivePacket(&keepalive);
|
|
||||||
m_Socket.Send(keepalive, m_Ip, TRANSCODER_PORT);
|
|
||||||
|
|
||||||
// check if still with us
|
|
||||||
if ( m_bConnected && (m_LastActivityTime.time() >= TRANSCODER_KEEPALIVE_TIMEOUT) )
|
|
||||||
{
|
|
||||||
// no, disconnect
|
|
||||||
m_bConnected = false;
|
|
||||||
std::cout << "Transcoder keepalive timeout" << std::endl;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
// packet decoding helpers
|
|
||||||
|
|
||||||
bool CTranscoder::IsValidKeepAlivePacket(const CBuffer &Buffer)
|
|
||||||
{
|
|
||||||
uint8_t tag[] = { 'A','M','B','E','D','P','O','N','G' };
|
|
||||||
|
|
||||||
bool valid = false;
|
|
||||||
if ( (Buffer.size() == 9) && (Buffer.Compare(tag, sizeof(tag)) == 0) )
|
|
||||||
{
|
|
||||||
valid = true;
|
|
||||||
}
|
|
||||||
return valid;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool CTranscoder::IsValidStreamDescrPacket(const CBuffer &Buffer, uint16_t *Id, uint16_t *Port)
|
|
||||||
{
|
|
||||||
uint8_t tag[] = { 'A','M','B','E','D','S','T','D' };
|
|
||||||
|
|
||||||
bool valid = false;
|
|
||||||
if ( (Buffer.size() == 14) && (Buffer.Compare(tag, sizeof(tag)) == 0) )
|
|
||||||
{
|
|
||||||
*Id = *(uint16_t *)(&Buffer.data()[8]);
|
|
||||||
*Port = *(uint16_t *)(&Buffer.data()[10]);
|
|
||||||
// uint8_t CodecIn = Buffer.data()[12];
|
|
||||||
// uint8_t CodecOut = Buffer.data()[13];
|
|
||||||
valid = true;
|
|
||||||
}
|
|
||||||
return valid;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool CTranscoder::IsValidNoStreamAvailablePacket(const CBuffer&Buffer)
|
|
||||||
{
|
|
||||||
uint8_t tag[] = { 'A','M','B','E','D','B','U','S','Y' };
|
|
||||||
|
|
||||||
return ( (Buffer.size() == 9) && (Buffer.Compare(tag, sizeof(tag)) == 0) );
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
// packet encoding helpers
|
|
||||||
|
|
||||||
void CTranscoder::EncodeKeepAlivePacket(CBuffer *Buffer)
|
|
||||||
{
|
|
||||||
uint8_t tag[] = { 'A','M','B','E','D','P','I','N','G' };
|
|
||||||
|
|
||||||
Buffer->Set(tag, sizeof(tag));
|
|
||||||
Buffer->Append((uint8_t *)(const char *)g_Reflector.GetCallsign(), CALLSIGN_LEN);
|
|
||||||
}
|
|
||||||
|
|
||||||
void CTranscoder::EncodeOpenstreamPacket(CBuffer *Buffer, uint8_t uiCodecIn, uint8_t uiCodecOut)
|
|
||||||
{
|
|
||||||
uint8_t tag[] = { 'A','M','B','E','D','O','S' };
|
|
||||||
|
|
||||||
Buffer->Set(tag, sizeof(tag));
|
|
||||||
Buffer->Append((uint8_t *)(const char *)g_Reflector.GetCallsign(), CALLSIGN_LEN);
|
|
||||||
Buffer->Append((uint8_t)uiCodecIn);
|
|
||||||
Buffer->Append((uint8_t)uiCodecOut);
|
|
||||||
}
|
|
||||||
|
|
||||||
void CTranscoder::EncodeClosestreamPacket(CBuffer *Buffer, uint16_t uiStreamId)
|
|
||||||
{
|
|
||||||
uint8_t tag[] = { 'A','M','B','E','D','C','S' };
|
|
||||||
|
|
||||||
Buffer->Set(tag, sizeof(tag));
|
|
||||||
Buffer->Append((uint16_t)uiStreamId);
|
|
||||||
}
|
|
||||||
@ -1,92 +0,0 @@
|
|||||||
// Copyright © 2015 Jean-Luc Deltombe (LX3JL). All rights reserved.
|
|
||||||
|
|
||||||
// urfd -- The universal reflector
|
|
||||||
// Copyright © 2021 Thomas A. Early N7TAE
|
|
||||||
//
|
|
||||||
// This program is free software: you can redistribute it and/or modify
|
|
||||||
// it under the terms of the GNU General Public License as published by
|
|
||||||
// the Free Software Foundation, either version 3 of the License, or
|
|
||||||
// (at your option) any later version.
|
|
||||||
//
|
|
||||||
// This program is distributed in the hope that it will be useful,
|
|
||||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
// GNU General Public License for more details.
|
|
||||||
//
|
|
||||||
// You should have received a copy of the GNU General Public License
|
|
||||||
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
||||||
|
|
||||||
#pragma once
|
|
||||||
|
|
||||||
#include "Semaphore.h"
|
|
||||||
#include "CodecStream.h"
|
|
||||||
#include "UDPSocket.h"
|
|
||||||
|
|
||||||
class CPacketStream;
|
|
||||||
|
|
||||||
class CTranscoder
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
// constructor
|
|
||||||
CTranscoder();
|
|
||||||
|
|
||||||
// destructor
|
|
||||||
~CTranscoder();
|
|
||||||
|
|
||||||
// initialization
|
|
||||||
bool Init(void);
|
|
||||||
void Close(void);
|
|
||||||
|
|
||||||
// locks
|
|
||||||
void Lock(void) { m_Mutex.lock(); }
|
|
||||||
void Unlock(void) { m_Mutex.unlock(); }
|
|
||||||
|
|
||||||
// status
|
|
||||||
bool IsConnected(void) const { return m_bConnected; }
|
|
||||||
|
|
||||||
// manage streams
|
|
||||||
std::shared_ptr<CCodecStream> GetCodecStream(CPacketStream *, uint8_t);
|
|
||||||
void ReleaseStream(std::shared_ptr<CCodecStream>);
|
|
||||||
|
|
||||||
// task
|
|
||||||
void Thread(void);
|
|
||||||
void Task(void);
|
|
||||||
|
|
||||||
protected:
|
|
||||||
// keepalive helpers
|
|
||||||
void HandleKeepalives(void);
|
|
||||||
|
|
||||||
// packet decoding helpers
|
|
||||||
bool IsValidKeepAlivePacket(const CBuffer &);
|
|
||||||
bool IsValidStreamDescrPacket(const CBuffer &, uint16_t *, uint16_t *);
|
|
||||||
bool IsValidNoStreamAvailablePacket(const CBuffer&);
|
|
||||||
|
|
||||||
// packet encoding helpers
|
|
||||||
void EncodeKeepAlivePacket(CBuffer *);
|
|
||||||
void EncodeOpenstreamPacket(CBuffer *, uint8_t, uint8_t);
|
|
||||||
void EncodeClosestreamPacket(CBuffer *, uint16_t);
|
|
||||||
|
|
||||||
protected:
|
|
||||||
// streams
|
|
||||||
std::mutex m_Mutex;
|
|
||||||
std::list<std::shared_ptr<CCodecStream>> m_Streams;
|
|
||||||
|
|
||||||
// sync objects for Openstream
|
|
||||||
CSemaphore m_SemaphoreOpenStream;
|
|
||||||
bool m_bStreamOpened;
|
|
||||||
uint16_t m_StreamidOpenStream;
|
|
||||||
uint16_t m_PortOpenStream;
|
|
||||||
|
|
||||||
// thread
|
|
||||||
std::atomic<bool> keep_running;
|
|
||||||
std::future<void> m_Future;
|
|
||||||
|
|
||||||
// socket
|
|
||||||
CIp m_Ip;
|
|
||||||
CUdpSocket m_Socket;
|
|
||||||
bool m_bConnected;
|
|
||||||
|
|
||||||
// time
|
|
||||||
CTimer m_LastKeepaliveTime;
|
|
||||||
CTimer m_LastActivityTime;
|
|
||||||
};
|
|
||||||
Loading…
Reference in new issue