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