#include "PL_RTSPServer.h"
|
#include "MaterialBuffer.h"
|
|
#include <liveMedia.hh>
|
#include <BasicUsageEnvironment.hh>
|
|
#include "FFmpegRTSPServer/IEncoder.h"
|
#include "FFmpegRTSPServer/LiveRTSPServer.h"
|
#include "FFmpegRTSPServer/FFmpegH264Source.h"
|
#include "FFmpegRTSPServer/LiveServerMediaSubsession.h"
|
|
class MyEncoderStub;
|
|
struct RTSPServer_Internal
|
{
|
uint8_t buffer[1920*1080*3];
|
size_t buffSize;
|
size_t buffSizeMax;
|
|
RTSPServerConfig config;
|
|
bool payError;
|
pthread_t live_daemon_thid;
|
pthread_mutex_t* frame_mutex;
|
bool live_daemon_running;
|
|
MESAI::LiveRTSPServer * server;
|
MyEncoderStub * encoderStub;
|
|
RTSPServer_Internal() :
|
buffSize(0), buffSizeMax(sizeof(buffer)), config(),
|
payError(true), live_daemon_thid(0), frame_mutex(new pthread_mutex_t), live_daemon_running(false),
|
server(nullptr), encoderStub(nullptr)
|
{
|
pthread_mutex_init(frame_mutex, NULL);
|
}
|
|
~RTSPServer_Internal()
|
{
|
if (frame_mutex != nullptr)
|
{
|
pthread_mutex_destroy(frame_mutex);
|
delete frame_mutex;
|
frame_mutex = nullptr;
|
}
|
}
|
|
void reset()
|
{
|
buffSize = 0;
|
|
RTSPServerConfig _config;
|
config =_config;
|
|
payError = true;
|
|
if (frame_mutex != nullptr)
|
{
|
pthread_mutex_destroy(frame_mutex);
|
delete frame_mutex;
|
frame_mutex = nullptr;
|
}
|
|
frame_mutex = new pthread_mutex_t;
|
pthread_mutex_init(frame_mutex, NULL);
|
|
live_daemon_thid = 0;
|
live_daemon_running = false;
|
|
server = nullptr;
|
encoderStub = nullptr;
|
}
|
};
|
|
class MyEncoderStub : public MESAI::IEncoder
|
{
|
public:
|
MyEncoderStub(RTSPServer_Internal& _in) : in(_in)
|
{
|
}
|
|
virtual ~MyEncoderStub()
|
{
|
}
|
|
virtual void setCallbackFunctionFrameIsReady(std::function<void()> func)
|
{
|
onFrame = func;
|
}
|
|
virtual char GetFrame(u_int8_t** FrameBuffer, unsigned int *FrameSize)
|
{
|
if (in.buffer == nullptr || in.buffSize <= 0)
|
{
|
ReleaseFrame();
|
return 0;
|
}
|
|
uint8_t* pBuffer = in.buffer;
|
size_t newBufferSize = in.buffSize;
|
if (in.config.payWithAux)
|
{
|
if (newBufferSize <= 4)
|
{
|
ReleaseFrame();
|
return 0;
|
}
|
pBuffer += 4;
|
newBufferSize -= 4;
|
}
|
|
*FrameBuffer = pBuffer;
|
*FrameSize = newBufferSize;
|
|
printf("send frame size=%u\n", in.buffSize);
|
}
|
|
virtual char ReleaseFrame()
|
{
|
in.buffSize = 0;
|
|
if (in.config.syncDeliverFrame)
|
{
|
int ret = pthread_mutex_unlock(in.frame_mutex);
|
if(ret != 0)
|
{
|
printf("pthread_mutex_unlock frame_mutex: %s/n", strerror(ret));
|
return 0;
|
}
|
}
|
|
return 1;
|
}
|
|
void deliverFrame()
|
{
|
// write frame buffer of RTSPServer_Internal::buffer
|
onFrame();
|
|
if (in.config.syncDeliverFrame)
|
{
|
int ret = pthread_mutex_lock(in.frame_mutex);
|
if(ret != 0)
|
{
|
printf("pthread_mutex_lock frame_mutex: %s/n", strerror(ret));
|
return;
|
}
|
}
|
}
|
|
private:
|
RTSPServer_Internal& in;
|
std::function<void()> onFrame;
|
};
|
|
PipeLineElem* create_PL_RTSPServer()
|
{
|
return new PL_RTSPServer;
|
}
|
|
PL_RTSPServer::PL_RTSPServer() : internal(new RTSPServer_Internal)
|
{
|
}
|
|
PL_RTSPServer::~PL_RTSPServer()
|
{
|
delete (RTSPServer_Internal*)internal;
|
internal = nullptr;
|
}
|
|
static void* live_daemon_thd(void* arg)
|
{
|
RTSPServer_Internal* in = (RTSPServer_Internal*)arg;
|
|
MyEncoderStub encoder(*in);
|
in->encoderStub = &encoder;
|
in->server = new MESAI::LiveRTSPServer(&encoder, 8554, 8080);
|
|
in->live_daemon_running = true;
|
in->server->run(); // does not return
|
in->encoderStub = nullptr;
|
in->live_daemon_running = false;
|
}
|
|
bool PL_RTSPServer::init(void* args)
|
{
|
RTSPServer_Internal* in = (RTSPServer_Internal*)internal;
|
in->reset();
|
|
if (args)
|
{
|
RTSPServerConfig* config = (RTSPServerConfig*)args;
|
in->config = *config;
|
}
|
|
int ret = pthread_create(&(in->live_daemon_thid), NULL, live_daemon_thd, in);
|
if(ret != 0)
|
{
|
printf("pthread_create: %s/n", strerror(ret));
|
return false;
|
}
|
|
return true;
|
}
|
|
void PL_RTSPServer::finit()
|
{
|
RTSPServer_Internal* in = (RTSPServer_Internal*)internal;
|
|
pthread_join(in->live_daemon_thid, NULL);
|
}
|
|
bool PL_RTSPServer::pay(const PipeMaterial& pm)
|
{
|
RTSPServer_Internal* in = (RTSPServer_Internal*)internal;
|
|
if (pm.buffer == nullptr || pm.buffSize <= 0)
|
return false;
|
|
if (pm.type != PipeMaterial::PMT_FRAME)
|
{
|
printf("PL_RTSPServer::pay only support PMT_FRAME\n");
|
return false;
|
}
|
|
if (in->buffSize > 0)
|
printf("PL_RTSPServer::pay may lost data size=%u\n", in->buffSize);
|
|
MB_Frame* frame = (MB_Frame*)pm.buffer;
|
memcpy(in->buffer, frame->buffer, frame->buffSize);
|
in->buffSize = frame->buffSize;
|
|
if (in->encoderStub == nullptr)
|
return false;
|
|
in->encoderStub->deliverFrame();
|
return true;
|
}
|
|
bool PL_RTSPServer::gain(PipeMaterial& pm)
|
{
|
RTSPServer_Internal* in = (RTSPServer_Internal*)internal;
|
|
pm.buffer = nullptr;
|
pm.buffSize = 0;
|
pm.former = this;
|
return true;
|
}
|
