75 lines
2.3 KiB
C++
75 lines
2.3 KiB
C++
#include "RtspServer.h"
|
|
#include "Core/Logger.h"
|
|
#include <boost/asio/io_context.hpp>
|
|
#include <boost/asio/strand.hpp>
|
|
#include <chrono>
|
|
#include <cstring>
|
|
#include <mk_common.h>
|
|
#include <mk_h264_splitter.h>
|
|
#include <mk_media.h>
|
|
#include <thread>
|
|
|
|
constexpr auto iniConfig = R"(
|
|
[rtsp]
|
|
lowLatency=1
|
|
)";
|
|
class RtspServerPrivate {
|
|
public:
|
|
RtspServerPrivate(boost::asio::io_context &ioContext) : strand{ioContext.get_executor()} {
|
|
}
|
|
boost::asio::strand<boost::asio::io_context::executor_type> strand;
|
|
mk_media media;
|
|
mk_h264_splitter splitter;
|
|
};
|
|
|
|
static void on_h264_frame(void *user_data, mk_h264_splitter splitter, const char *data, int size) {
|
|
using namespace std::chrono;
|
|
using namespace std::chrono_literals;
|
|
int pts = duration_cast<milliseconds>(system_clock::now().time_since_epoch()).count();
|
|
mk_frame frame = mk_frame_create(MKCodecH264, pts, pts, data, size, NULL, NULL);
|
|
mk_media_input_frame((mk_media)user_data, frame);
|
|
mk_frame_unref(frame);
|
|
}
|
|
|
|
RtspServer::RtspServer(boost::asio::io_context &ioContext) : m_d(new RtspServerPrivate(ioContext)) {
|
|
mk_config config;
|
|
std::memset(&config, 0, sizeof(mk_config));
|
|
config.ini = iniConfig;
|
|
config.ini_is_path = 0;
|
|
config.log_mask = LOG_CONSOLE;
|
|
config.ssl_is_path = 1;
|
|
mk_env_init(&config);
|
|
|
|
uint16_t status = mk_rtsp_server_start(554, 0);
|
|
|
|
status = mk_rtc_server_start(7764);
|
|
if (status == 0) {
|
|
LOG(error) << "mk_rtc_server_start() failed.";
|
|
}
|
|
|
|
m_d->media = mk_media_create("__defaultVhost__", "live", "video", 0, 0, 0);
|
|
|
|
codec_args v_args = {0};
|
|
mk_track v_track = mk_track_create(MKCodecH264, &v_args);
|
|
mk_media_init_track(m_d->media, v_track);
|
|
mk_media_init_complete(m_d->media);
|
|
mk_track_unref(v_track);
|
|
|
|
m_d->splitter = mk_h264_splitter_create(on_h264_frame, m_d->media, 0);
|
|
}
|
|
|
|
RtspServer::~RtspServer() {
|
|
mk_h264_splitter_release(m_d->splitter);
|
|
mk_media_release(m_d->media);
|
|
mk_stop_all_server();
|
|
if (m_d != nullptr) {
|
|
delete m_d;
|
|
}
|
|
}
|
|
|
|
void RtspServer::push(const uint8_t *data, uint32_t size) {
|
|
boost::asio::post(m_d->strand, [this, frame = std::vector<uint8_t>(data, data + size)]() {
|
|
mk_h264_splitter_input_data(m_d->splitter, reinterpret_cast<const char *>(frame.data()), frame.size());
|
|
});
|
|
}
|