sync from master

This commit is contained in:
xiongguangjie 2024-04-16 17:28:10 +08:00
commit 3adbad5fc4
58 changed files with 995 additions and 952 deletions

View File

@ -44,7 +44,6 @@ Xinghua Zhao <(holychaossword@hotmail.com>
[Dw9](https://github.com/Dw9) [Dw9](https://github.com/Dw9)
明月惊鹊 <mingyuejingque@gmail.com> 明月惊鹊 <mingyuejingque@gmail.com>
cgm <2958580318@qq.com> cgm <2958580318@qq.com>
hejilin <1724010622@qq.com>
alexliyu7352 <liyu7352@gmail.com> alexliyu7352 <liyu7352@gmail.com>
cgm <2958580318@qq.com> cgm <2958580318@qq.com>
[haorui wang](https://github.com/HaoruiWang) [haorui wang](https://github.com/HaoruiWang)
@ -104,3 +103,8 @@ WuPeng <wp@zafu.edu.cn>
[sandro-qiang](https://github.com/sandro-qiang) [sandro-qiang](https://github.com/sandro-qiang)
[Paul Philippov](https://github.com/themactep) [Paul Philippov](https://github.com/themactep)
[张传峰](https://github.com/zhang-chuanfeng) [张传峰](https://github.com/zhang-chuanfeng)
[lidaofu-hub](https://github.com/lidaofu-hub)
[huangcaichun](https://github.com/huangcaichun)
[jamesZHANG500](https://github.com/jamesZHANG500)
[weidelong](https://github.com/wdl1697454803)
[小强先生](https://github.com/linshangqiang)

View File

@ -57,7 +57,8 @@ option(ENABLE_SCTP "Enable SCTP" ON)
option(ENABLE_WEBRTC "Enable WebRTC" ON) option(ENABLE_WEBRTC "Enable WebRTC" ON)
option(ENABLE_X264 "Enable x264" OFF) option(ENABLE_X264 "Enable x264" OFF)
option(ENABLE_WEPOLL "Enable wepoll" ON) option(ENABLE_WEPOLL "Enable wepoll" ON)
option(DISABLE_REPORT "Disable report to report.zlmediakit.com" off) option(ENABLE_VIDEOSTACK "Enable video stack" OFF)
option(DISABLE_REPORT "Disable report to report.zlmediakit.com" OFF)
option(USE_SOLUTION_FOLDERS "Enable solution dir supported" ON) option(USE_SOLUTION_FOLDERS "Enable solution dir supported" ON)
############################################################################## ##############################################################################
# socket256k.0socket,使用系统内核默认值(设置为0仅对linux有效) # socket256k.0socket,使用系统内核默认值(设置为0仅对linux有效)
@ -141,8 +142,8 @@ if(GIT_FOUND)
endif() endif()
configure_file( configure_file(
${CMAKE_CURRENT_SOURCE_DIR}/version.h.ini ${CMAKE_CURRENT_SOURCE_DIR}/ZLMVersion.h.ini
${CMAKE_CURRENT_BINARY_DIR}/version.h ${CMAKE_CURRENT_BINARY_DIR}/ZLMVersion.h
@ONLY) @ONLY)
message(STATUS "Git version is ${BRANCH_NAME} ${COMMIT_HASH}/${COMMIT_TIME} ${BUILD_TIME}") message(STATUS "Git version is ${BRANCH_NAME} ${COMMIT_HASH}/${COMMIT_TIME} ${BUILD_TIME}")
@ -191,7 +192,7 @@ if(UNIX)
set(COMPILE_OPTIONS_DEFAULT set(COMPILE_OPTIONS_DEFAULT
"-fPIC" "-fPIC"
"-Wall;-Wextra" "-Wall;-Wextra"
"-Wno-unused-function;-Wno-unused-parameter;-Wno-unused-variable" "-Wno-unused-function;-Wno-unused-parameter;-Wno-unused-variable;-Wno-deprecated-declarations"
"-Wno-error=extra;-Wno-error=missing-field-initializers;-Wno-error=type-limits") "-Wno-error=extra;-Wno-error=missing-field-initializers;-Wno-error=type-limits")
if("${CMAKE_BUILD_TYPE}" STREQUAL "Debug") if("${CMAKE_BUILD_TYPE}" STREQUAL "Debug")
@ -535,6 +536,6 @@ file(COPY "${CMAKE_CURRENT_SOURCE_DIR}/default.pem" DESTINATION ${EXECUTABLE_OUT
# VideoStack # VideoStack
# Copy the default background image used by VideoStack when there is no video stream # Copy the default background image used by VideoStack when there is no video stream
if (ENABLE_FFMPEG AND ENABLE_X264) if (ENABLE_VIDEOSTACK AND ENABLE_FFMPEG AND ENABLE_X264)
file(COPY "${CMAKE_CURRENT_SOURCE_DIR}/conf/novideo.yuv" DESTINATION ${EXECUTABLE_OUTPUT_PATH}) file(COPY "${CMAKE_CURRENT_SOURCE_DIR}/conf/novideo.yuv" DESTINATION ${EXECUTABLE_OUTPUT_PATH})
endif () endif ()

View File

@ -358,6 +358,11 @@ bash build_docker_images.sh
[sandro-qiang](https://github.com/sandro-qiang) [sandro-qiang](https://github.com/sandro-qiang)
[Paul Philippov](https://github.com/themactep) [Paul Philippov](https://github.com/themactep)
[张传峰](https://github.com/zhang-chuanfeng) [张传峰](https://github.com/zhang-chuanfeng)
[lidaofu-hub](https://github.com/lidaofu-hub)
[huangcaichun](https://github.com/huangcaichun)
[jamesZHANG500](https://github.com/jamesZHANG500)
[weidelong](https://github.com/wdl1697454803)
[小强先生](https://github.com/linshangqiang)
同时感谢JetBrains对开源项目的支持本项目使用CLion开发与调试 同时感谢JetBrains对开源项目的支持本项目使用CLion开发与调试

View File

@ -516,6 +516,11 @@ Thanks to all those who have supported this project in various ways, including b
[sandro-qiang](https://github.com/sandro-qiang) [sandro-qiang](https://github.com/sandro-qiang)
[Paul Philippov](https://github.com/themactep) [Paul Philippov](https://github.com/themactep)
[张传峰](https://github.com/zhang-chuanfeng) [张传峰](https://github.com/zhang-chuanfeng)
[lidaofu-hub](https://github.com/lidaofu-hub)
[huangcaichun](https://github.com/huangcaichun)
[jamesZHANG500](https://github.com/jamesZHANG500)
[weidelong](https://github.com/wdl1697454803)
[小强先生](https://github.com/linshangqiang)
Also thank to JetBrains for their support for open source project, we developed and debugged zlmediakit with CLion: Also thank to JetBrains for their support for open source project, we developed and debugged zlmediakit with CLion:

View File

@ -24,6 +24,10 @@
# define API_CALL # define API_CALL
#endif #endif
#ifndef _WIN32
#define _strdup strdup
#endif
#if defined(_WIN32) && defined(_MSC_VER) #if defined(_WIN32) && defined(_MSC_VER)
# if !defined(GENERATE_EXPORT) # if !defined(GENERATE_EXPORT)
# if defined(MediaKitApi_EXPORTS) # if defined(MediaKitApi_EXPORTS)

View File

@ -103,6 +103,16 @@ API_EXPORT int API_CALL mk_media_source_get_track_count(const mk_media_source ct
API_EXPORT mk_track API_CALL mk_media_source_get_track(const mk_media_source ctx, int index); API_EXPORT mk_track API_CALL mk_media_source_get_track(const mk_media_source ctx, int index);
// MediaSource::broadcastMessage // MediaSource::broadcastMessage
API_EXPORT int API_CALL mk_media_source_broadcast_msg(const mk_media_source ctx, const char *msg, size_t len); API_EXPORT int API_CALL mk_media_source_broadcast_msg(const mk_media_source ctx, const char *msg, size_t len);
// MediaSource::getOriginUrl()
API_EXPORT const char* API_CALL mk_media_source_get_origin_url(const mk_media_source ctx);
// MediaSource::getOriginType()
API_EXPORT int API_CALL mk_media_source_get_origin_type(const mk_media_source ctx);
// MediaSource::getCreateStamp()
API_EXPORT uint64_t API_CALL mk_media_source_get_create_stamp(const mk_media_source ctx);
// MediaSource::isRecording() 0:hls,1:MP4
API_EXPORT int API_CALL mk_media_source_is_recording(const mk_media_source ctx, int type);
/** /**
* ZLMediaKit中被称作为MediaSource * ZLMediaKit中被称作为MediaSource
@ -142,7 +152,7 @@ API_EXPORT void API_CALL mk_media_source_find(const char *schema,
void *user_data, void *user_data,
on_mk_media_source_find_cb cb); on_mk_media_source_find_cb cb);
API_EXPORT const mk_media_source API_CALL mk_media_source_find2(const char *schema, API_EXPORT mk_media_source API_CALL mk_media_source_find2(const char *schema,
const char *vhost, const char *vhost,
const char *app, const char *app,
const char *stream, const char *stream,

View File

@ -304,10 +304,10 @@ API_EXPORT void API_CALL mk_webrtc_get_answer_sdp2(void *user_data, on_user_data
std::string offer_str = offer; std::string offer_str = offer;
std::shared_ptr<void> ptr(user_data, user_data_free ? user_data_free : [](void *) {}); std::shared_ptr<void> ptr(user_data, user_data_free ? user_data_free : [](void *) {});
auto args = std::make_shared<WebRtcArgsUrl>(url); auto args = std::make_shared<WebRtcArgsUrl>(url);
WebRtcPluginManager::Instance().getAnswerSdp(*session, type, *args, WebRtcPluginManager::Instance().negotiateSdp(*session, type, *args, [offer_str, session, ptr, cb](const WebRtcInterface &exchanger) mutable {
[offer_str, session, ptr, cb](const WebRtcInterface &exchanger) mutable { auto &handler = const_cast<WebRtcInterface &>(exchanger);
try { try {
auto sdp_answer = exchangeSdp(exchanger, offer_str); auto sdp_answer = handler.getAnswerSdp(offer_str);
cb(ptr.get(), sdp_answer.data(), nullptr); cb(ptr.get(), sdp_answer.data(), nullptr);
} catch (std::exception &ex) { } catch (std::exception &ex) {
cb(ptr.get(), nullptr, ex.what()); cb(ptr.get(), nullptr, ex.what());

View File

@ -14,7 +14,10 @@
#include "Http/HttpSession.h" #include "Http/HttpSession.h"
#include "Rtsp/RtspSession.h" #include "Rtsp/RtspSession.h"
#include "Record/MP4Recorder.h" #include "Record/MP4Recorder.h"
#ifdef ENABLE_WEBRTC
#include "webrtc/WebRtcTransport.h" #include "webrtc/WebRtcTransport.h"
#endif
using namespace toolkit; using namespace toolkit;
using namespace mediakit; using namespace mediakit;
@ -168,7 +171,7 @@ API_EXPORT void API_CALL mk_events_listen(const mk_events *events){
sender.getMediaTuple().stream.c_str(), ssrc.c_str(), ex.getErrCode(), ex.what()); sender.getMediaTuple().stream.c_str(), ssrc.c_str(), ex.getErrCode(), ex.what());
} }
}); });
#ifdef ENABLE_WEBRTC
NoticeCenter::Instance().addListener(&s_tag, Broadcast::kBroadcastRtcSctpConnecting,[](BroadcastRtcSctpConnectArgs){ NoticeCenter::Instance().addListener(&s_tag, Broadcast::kBroadcastRtcSctpConnecting,[](BroadcastRtcSctpConnectArgs){
if (s_events.on_mk_rtc_sctp_connecting) { if (s_events.on_mk_rtc_sctp_connecting) {
s_events.on_mk_rtc_sctp_connecting((mk_rtc_transport)&sender); s_events.on_mk_rtc_sctp_connecting((mk_rtc_transport)&sender);
@ -204,6 +207,7 @@ API_EXPORT void API_CALL mk_events_listen(const mk_events *events){
s_events.on_mk_rtc_sctp_received((mk_rtc_transport)&sender, streamId, ppid, msg, len); s_events.on_mk_rtc_sctp_received((mk_rtc_transport)&sender, streamId, ppid, msg, len);
} }
}); });
#endif
}); });
} }

View File

@ -17,7 +17,10 @@
#include "Http/HttpClient.h" #include "Http/HttpClient.h"
#include "Rtsp/RtspSession.h" #include "Rtsp/RtspSession.h"
#ifdef ENABLE_WEBRTC
#include "webrtc/WebRtcTransport.h" #include "webrtc/WebRtcTransport.h"
#endif
using namespace toolkit; using namespace toolkit;
using namespace mediakit; using namespace mediakit;
@ -127,7 +130,7 @@ API_EXPORT const char* API_CALL mk_parser_get_content(const mk_parser ctx, size_
API_EXPORT const char* API_CALL mk_media_info_get_params(const mk_media_info ctx){ API_EXPORT const char* API_CALL mk_media_info_get_params(const mk_media_info ctx){
assert(ctx); assert(ctx);
MediaInfo *info = (MediaInfo *)ctx; MediaInfo *info = (MediaInfo *)ctx;
return info->param_strs.c_str(); return info->params.c_str();
} }
API_EXPORT const char* API_CALL mk_media_info_get_schema(const mk_media_info ctx){ API_EXPORT const char* API_CALL mk_media_info_get_schema(const mk_media_info ctx){
@ -225,6 +228,30 @@ API_EXPORT int API_CALL mk_media_source_broadcast_msg(const mk_media_source ctx,
return src->broadcastMessage(any); return src->broadcastMessage(any);
} }
API_EXPORT const char* API_CALL mk_media_source_get_origin_url(const mk_media_source ctx) {
assert(ctx);
MediaSource *src = (MediaSource *)ctx;
return _strdup(src->getOriginUrl().c_str());
}
API_EXPORT int API_CALL mk_media_source_get_origin_type(const mk_media_source ctx) {
assert(ctx);
MediaSource *src = (MediaSource *)ctx;
return static_cast<int>(src->getOriginType());
}
API_EXPORT uint64_t API_CALL mk_media_source_get_create_stamp(const mk_media_source ctx) {
assert(ctx);
MediaSource *src = (MediaSource *)ctx;
return src->getCreateStamp();
}
API_EXPORT int API_CALL mk_media_source_is_recording(const mk_media_source ctx,int type) {
assert(ctx);
MediaSource *src = (MediaSource *)ctx;
return src->isRecording((Recorder::type)type);
}
API_EXPORT int API_CALL mk_media_source_close(const mk_media_source ctx,int force){ API_EXPORT int API_CALL mk_media_source_close(const mk_media_source ctx,int force){
assert(ctx); assert(ctx);
MediaSource *src = (MediaSource *)ctx; MediaSource *src = (MediaSource *)ctx;
@ -275,7 +302,7 @@ API_EXPORT void API_CALL mk_media_source_find(const char *schema,
cb(user_data, (mk_media_source)src.get()); cb(user_data, (mk_media_source)src.get());
} }
API_EXPORT const mk_media_source API_CALL mk_media_source_find2(const char *schema, API_EXPORT mk_media_source API_CALL mk_media_source_find2(const char *schema,
const char *vhost, const char *vhost,
const char *app, const char *app,
const char *stream, const char *stream,

View File

@ -21,10 +21,6 @@ using namespace std;
using namespace toolkit; using namespace toolkit;
using namespace mediakit; using namespace mediakit;
#ifndef _WIN32
#define _strdup strdup
#endif
API_EXPORT void API_CALL mk_free(void *ptr) { API_EXPORT void API_CALL mk_free(void *ptr) {
free(ptr); free(ptr);
} }

152
api/tests/h264_pusher.c Normal file
View File

@ -0,0 +1,152 @@
/*
* Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
*
* This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
*
* Use of this source code is governed by MIT-like license that can be found in the
* LICENSE file in the root of the source tree. All contributing project authors
* may be found in the AUTHORS file in the root of the source tree.
*/
#include <signal.h>
#include <stdio.h>
#include <string.h>
#ifdef _WIN32
#include "windows.h"
#else
#include "unistd.h"
#endif
#include "mk_mediakit.h"
static int exit_flag = 0;
static void s_on_exit(int sig) {
exit_flag = 1;
}
static void on_h264_frame(void *user_data, mk_h264_splitter splitter, const char *data, int size) {
#ifdef _WIN32
Sleep(40);
#else
usleep(40 * 1000);
#endif
static int dts = 0;
mk_frame frame = mk_frame_create(MKCodecH264, dts, dts, data, size, NULL, NULL);
dts += 40;
mk_media_input_frame((mk_media)user_data, frame);
mk_frame_unref(frame);
}
typedef struct {
mk_pusher pusher;
char *url;
} Context;
void release_context(void *user_data) {
Context *ptr = (Context *)user_data;
if (ptr->pusher) {
mk_pusher_release(ptr->pusher);
}
free(ptr->url);
free(ptr);
log_info("停止推流");
}
void on_push_result(void *user_data, int err_code, const char *err_msg) {
Context *ptr = (Context *)user_data;
if (err_code == 0) {
log_info("推流成功: %s", ptr->url);
} else {
log_warn("推流%s失败: %d(%s)", ptr->url, err_code, err_msg);
}
}
void on_push_shutdown(void *user_data, int err_code, const char *err_msg) {
Context *ptr = (Context *)user_data;
log_warn("推流%s中断: %d(%s)", ptr->url, err_code, err_msg);
}
void API_CALL on_regist(void *user_data, mk_media_source sender, int regist) {
Context *ptr = (Context *)user_data;
const char *schema = mk_media_source_get_schema(sender);
if (strstr(ptr->url, schema) != ptr->url) {
// 协议匹配失败
return;
}
if (!regist) {
// 注销
if (ptr->pusher) {
mk_pusher_release(ptr->pusher);
ptr->pusher = NULL;
}
} else {
// 注册
if (!ptr->pusher) {
ptr->pusher = mk_pusher_create_src(sender);
mk_pusher_set_on_result2(ptr->pusher, on_push_result, ptr, NULL);
mk_pusher_set_on_shutdown2(ptr->pusher, on_push_shutdown, ptr, NULL);
// 开始推流
mk_pusher_publish(ptr->pusher, ptr->url);
}
}
}
int main(int argc, char *argv[]) {
if (argc < 3) {
log_error("Usage: /path/to/h264/file rtsp_or_rtmp_url");
return -1;
}
mk_config config = { .ini = NULL,
.ini_is_path = 1,
.log_level = 0,
.log_mask = LOG_CONSOLE,
.log_file_path = NULL,
.log_file_days = 0,
.ssl = NULL,
.ssl_is_path = 1,
.ssl_pwd = NULL,
.thread_num = 0 };
mk_env_init(&config);
FILE *fp = fopen(argv[1], "rb");
if (!fp) {
log_error("打开文件失败!");
return -1;
}
mk_media media = mk_media_create("__defaultVhost__", "live", "test", 0, 0, 0);
// h264的codec
codec_args v_args = { 0 };
mk_track v_track = mk_track_create(MKCodecH264, &v_args);
mk_media_init_track(media, v_track);
mk_media_init_complete(media);
mk_track_unref(v_track);
Context *ctx = (Context *)malloc(sizeof(Context));
memset(ctx, 0, sizeof(Context));
ctx->url = strdup(argv[2]);
mk_media_set_on_regist2(media, on_regist, ctx, release_context);
// 创建h264分帧器
mk_h264_splitter splitter = mk_h264_splitter_create(on_h264_frame, media, 0);
signal(SIGINT, s_on_exit); // 设置退出信号
signal(SIGTERM, s_on_exit); // 设置退出信号
char buf[1024];
while (!exit_flag) {
int size = fread(buf, 1, sizeof(buf) - 1, fp);
if (size > 0) {
mk_h264_splitter_input_data(splitter, buf, size);
} else {
// 文件读完了,重新开始
fseek(fp, 0, SEEK_SET);
}
}
log_info("文件读取完毕");
mk_h264_splitter_release(splitter);
mk_media_release(media);
fclose(fp);
return 0;
}

View File

@ -357,7 +357,7 @@ tcpPort = 8000
rembBitRate=0 rembBitRate=0
#rtc支持的音频codec类型,在前面的优先级更高 #rtc支持的音频codec类型,在前面的优先级更高
#以下范例为所有支持的音频codec #以下范例为所有支持的音频codec
preferredCodecA=PCMU,PCMA,opus,mpeg4-generic preferredCodecA=PCMA,PCMU,opus,mpeg4-generic
#rtc支持的视频codec类型,在前面的优先级更高 #rtc支持的视频codec类型,在前面的优先级更高
#以下范例为所有支持的视频codec #以下范例为所有支持的视频codec
preferredCodecV=H264,H265,AV1,VP9,VP8 preferredCodecV=H264,H265,AV1,VP9,VP8

View File

@ -17,7 +17,7 @@ using namespace toolkit;
namespace mediakit { namespace mediakit {
void AACRtmpDecoder::inputRtmp(const RtmpPacket::Ptr &pkt) { void AACRtmpDecoder::inputRtmp(const RtmpPacket::Ptr &pkt) {
CHECK(pkt->size() > 2); CHECK_RET(pkt->size() > 2);
if (pkt->isConfigFrame()) { if (pkt->isConfigFrame()) {
getTrack()->setExtraData((uint8_t *)pkt->data() + 2, pkt->size() - 2); getTrack()->setExtraData((uint8_t *)pkt->data() + 2, pkt->size() - 2);
return; return;

View File

@ -14,14 +14,6 @@
using namespace std; using namespace std;
using namespace toolkit; using namespace toolkit;
#define CHECK_RET(...) \
try { \
CHECK(__VA_ARGS__); \
} catch (AssertFailedException & ex) { \
WarnL << ex.what(); \
return; \
}
namespace mediakit { namespace mediakit {
void H264RtmpDecoder::inputRtmp(const RtmpPacket::Ptr &pkt) { void H264RtmpDecoder::inputRtmp(const RtmpPacket::Ptr &pkt) {

View File

@ -18,14 +18,6 @@
using namespace std; using namespace std;
using namespace toolkit; using namespace toolkit;
#define CHECK_RET(...) \
try { \
CHECK(__VA_ARGS__); \
} catch (AssertFailedException & ex) { \
WarnL << ex.what(); \
return; \
}
namespace mediakit { namespace mediakit {
void H265RtmpDecoder::inputRtmp(const RtmpPacket::Ptr &pkt) { void H265RtmpDecoder::inputRtmp(const RtmpPacket::Ptr &pkt) {

View File

@ -1,4 +1,4 @@
#if defined(ENABLE_X264) && defined(ENABLE_FFMPEG) #if defined(ENABLE_VIDEOSTACK) && defined(ENABLE_X264) && defined(ENABLE_FFMPEG)
#include "VideoStack.h" #include "VideoStack.h"
#include "Codec/Transcode.h" #include "Codec/Transcode.h"
#include "Common/Device.h" #include "Common/Device.h"
@ -62,7 +62,8 @@ void Channel::addParam(const std::weak_ptr<Param>& p)
void Channel::onFrame(const mediakit::FFmpegFrame::Ptr& frame) void Channel::onFrame(const mediakit::FFmpegFrame::Ptr& frame)
{ {
std::weak_ptr<Channel> weakSelf = shared_from_this(); std::weak_ptr<Channel> weakSelf = shared_from_this();
// toolkit::WorkThreadPool::Instance().getFirstPoller()->async([weakSelf, frame]() { _poller = _poller ? _poller : toolkit::WorkThreadPool::Instance().getPoller();
_poller->async([weakSelf, frame]() {
auto self = weakSelf.lock(); auto self = weakSelf.lock();
if (!self) { if (!self) {
return; return;
@ -70,7 +71,7 @@ void Channel::onFrame(const mediakit::FFmpegFrame::Ptr& frame)
self->_tmp = self->_sws->inputFrame(frame); self->_tmp = self->_sws->inputFrame(frame);
self->forEachParam([self](const Param::Ptr& p) { self->fillBuffer(p); }); self->forEachParam([self](const Param::Ptr& p) { self->fillBuffer(p); });
// }); });
} }
void Channel::forEachParam(const std::function<void(const Param::Ptr&)>& func) void Channel::forEachParam(const std::function<void(const Param::Ptr&)>& func)
@ -180,7 +181,7 @@ void StackPlayer::play()
self->onFrame(frame); self->onFrame(frame);
}); });
videoTrack->addDelegate((std::function<bool(const mediakit::Frame::Ptr&)>)[decoder](const mediakit::Frame::Ptr& frame) { videoTrack->addDelegate([decoder](const mediakit::Frame::Ptr& frame) {
return decoder->inputFrame(frame, false, true); return decoder->inputFrame(frame, false, true);
}); });
} }
@ -440,6 +441,7 @@ int VideoStackManager::stopVideoStack(const std::string& id)
auto it = _stackMap.find(id); auto it = _stackMap.find(id);
if (it != _stackMap.end()) { if (it != _stackMap.end()) {
_stackMap.erase(it); _stackMap.erase(it);
InfoL << "VideoStack stop: " << id;
return 0; return 0;
} }
return -1; return -1;
@ -467,8 +469,8 @@ Params VideoStackManager::parseParams(const Json::Value& json,
float gaph = json["gaph"].asFloat(); //水平间距 float gaph = json["gaph"].asFloat(); //水平间距
//单个间距 //单个间距
int gaphPix = static_cast<int>(std::round(width * gaph)); int gaphPix = static_cast<int>(round(width * gaph));
int gapvPix = static_cast<int>(std::round(height * gapv)); int gapvPix = static_cast<int>(round(height * gapv));
// 根据间距计算格子宽高 // 根据间距计算格子宽高
int gridWidth = cols > 1 ? (width - gaphPix * (cols - 1)) / cols : width; int gridWidth = cols > 1 ? (width - gaphPix * (cols - 1)) / cols : width;

View File

@ -1,5 +1,5 @@
#pragma once #pragma once
#if defined(ENABLE_X264) && defined(ENABLE_FFMPEG) #if defined(ENABLE_VIDEOSTACK) && defined(ENABLE_X264) && defined(ENABLE_FFMPEG)
#include "Codec/Transcode.h" #include "Codec/Transcode.h"
#include "Common/Device.h" #include "Common/Device.h"
#include "Player/MediaPlayer.h" #include "Player/MediaPlayer.h"
@ -80,6 +80,7 @@ class Channel : public std::enable_shared_from_this<Channel> {
std::vector<std::weak_ptr<Param>> _params; std::vector<std::weak_ptr<Param>> _params;
mediakit::FFmpegSws::Ptr _sws; mediakit::FFmpegSws::Ptr _sws;
toolkit::EventPoller::Ptr _poller;
}; };
class StackPlayer : public std::enable_shared_from_this<StackPlayer> { class StackPlayer : public std::enable_shared_from_this<StackPlayer> {

View File

@ -59,10 +59,10 @@
#endif #endif
#if defined(ENABLE_VERSION) #if defined(ENABLE_VERSION)
#include "version.h" #include "ZLMVersion.h"
#endif #endif
#if defined(ENABLE_X264) && defined (ENABLE_FFMPEG) #if defined(ENABLE_VIDEOSTACK) && defined(ENABLE_X264) && defined (ENABLE_FFMPEG)
#include "VideoStack.h" #include "VideoStack.h"
#endif #endif
@ -119,7 +119,7 @@ static HttpApi toApi(const function<void(API_ARGS_MAP_ASYNC)> &cb) {
//参数解析成map //参数解析成map
auto args = getAllArgs(parser); auto args = getAllArgs(parser);
cb(sender, headerOut, HttpAllArgs<decltype(args)>(parser, args), val, invoker); cb(sender, headerOut, ArgsMap(parser, args), val, invoker);
}; };
} }
@ -147,7 +147,7 @@ static HttpApi toApi(const function<void(API_ARGS_JSON_ASYNC)> &cb) {
Json::Reader reader; Json::Reader reader;
reader.parse(parser.content(), args); reader.parse(parser.content(), args);
cb(sender, headerOut, HttpAllArgs<decltype(args)>(parser, args), val, invoker); cb(sender, headerOut, ArgsJson(parser, args), val, invoker);
}; };
} }
@ -167,7 +167,7 @@ static HttpApi toApi(const function<void(API_ARGS_STRING_ASYNC)> &cb) {
Json::Value val; Json::Value val;
val["code"] = API::Success; val["code"] = API::Success;
cb(sender, headerOut, HttpAllArgs<string>(parser, (string &)parser.content()), val, invoker); cb(sender, headerOut, ArgsString(parser, (string &)parser.content()), val, invoker);
}; };
} }
@ -389,6 +389,7 @@ void dumpMediaTuple(const MediaTuple &tuple, Json::Value& item) {
item[VHOST_KEY] = tuple.vhost; item[VHOST_KEY] = tuple.vhost;
item["app"] = tuple.app; item["app"] = tuple.app;
item["stream"] = tuple.stream; item["stream"] = tuple.stream;
item["params"] = tuple.params;
} }
Value makeMediaSourceJson(MediaSource &media){ Value makeMediaSourceJson(MediaSource &media){
@ -584,8 +585,10 @@ void addStreamProxy(const string &vhost, const string &app, const string &stream
//添加拉流代理 //添加拉流代理
auto player = s_player_proxy.make(key, vhost, app, stream, option, retry_count); auto player = s_player_proxy.make(key, vhost, app, stream, option, retry_count);
// 先透传参数 // 先透传拷贝参数
player->mINI::operator=(args); for (auto &pr : args) {
(*player)[pr.first] = pr.second;
}
//指定RTP over TCP(播放rtsp时有效) //指定RTP over TCP(播放rtsp时有效)
(*player)[Client::kRtpType] = rtp_type; (*player)[Client::kRtpType] = rtp_type;
@ -660,13 +663,6 @@ void addStreamPusherProxy(const string &schema,
pusher->publish(url); pusher->publish(url);
} }
template <typename Type>
static void getArgsValue(const HttpAllArgs<ApiArgsType> &allArgs, const string &key, Type &value) {
auto val = allArgs[key];
if (!val.empty()) {
value = (Type)val;
}
}
/** /**
* api接口 * api接口
@ -733,7 +729,7 @@ void installWebApi() {
CHECK_SECRET(); CHECK_SECRET();
auto &ini = mINI::Instance(); auto &ini = mINI::Instance();
int changed = API::Success; int changed = API::Success;
for (auto &pr : allArgs.getArgs()) { for (auto &pr : allArgs.args) {
if (ini.find(pr.first) == ini.end()) { if (ini.find(pr.first) == ini.end()) {
#if 1 #if 1
//没有这个key //没有这个key
@ -1091,7 +1087,7 @@ void installWebApi() {
CHECK_ARGS("vhost","app","stream","url"); CHECK_ARGS("vhost","app","stream","url");
mINI args; mINI args;
for (auto &pr : allArgs.getArgs()) { for (auto &pr : allArgs.args) {
args.emplace(pr.first, pr.second); args.emplace(pr.first, pr.second);
} }
@ -1188,7 +1184,7 @@ void installWebApi() {
//测试url http://127.0.0.1/index/api/downloadBin //测试url http://127.0.0.1/index/api/downloadBin
api_regist("/index/api/downloadBin",[](API_ARGS_MAP_ASYNC){ api_regist("/index/api/downloadBin",[](API_ARGS_MAP_ASYNC){
CHECK_SECRET(); CHECK_SECRET();
invoker.responseFile(allArgs.getParser().getHeader(),StrCaseMap(),exePath()); invoker.responseFile(allArgs.parser.getHeader(), StrCaseMap(), exePath());
}); });
#if defined(ENABLE_RTPPROXY) #if defined(ENABLE_RTPPROXY)
@ -1319,7 +1315,7 @@ void installWebApi() {
if (!src) { if (!src) {
throw ApiRetException("can not find the source stream", API::NotFound); throw ApiRetException("can not find the source stream", API::NotFound);
} }
auto type = allArgs["type"].as<int>(); auto type = allArgs["type"].empty() ? (int)MediaSourceEvent::SendRtpArgs::kRtpPS : allArgs["type"].as<int>();
if (!allArgs["use_ps"].empty()) { if (!allArgs["use_ps"].empty()) {
// 兼容之前的use_ps参数 // 兼容之前的use_ps参数
type = allArgs["use_ps"].as<int>(); type = allArgs["use_ps"].as<int>();
@ -1359,7 +1355,7 @@ void installWebApi() {
if (!src) { if (!src) {
throw ApiRetException("can not find the source stream", API::NotFound); throw ApiRetException("can not find the source stream", API::NotFound);
} }
auto type = allArgs["type"].as<int>(); auto type = allArgs["type"].empty() ? (int)MediaSourceEvent::SendRtpArgs::kRtpPS : allArgs["type"].as<int>();
if (!allArgs["use_ps"].empty()) { if (!allArgs["use_ps"].empty()) {
// 兼容之前的use_ps参数 // 兼容之前的use_ps参数
type = allArgs["use_ps"].as<int>(); type = allArgs["use_ps"].as<int>();
@ -1568,7 +1564,7 @@ void installWebApi() {
api_regist("/index/api/deleteRecordDirectory", [](API_ARGS_MAP) { api_regist("/index/api/deleteRecordDirectory", [](API_ARGS_MAP) {
CHECK_SECRET(); CHECK_SECRET();
CHECK_ARGS("vhost", "app", "stream", "period"); CHECK_ARGS("vhost", "app", "stream", "period");
auto tuple = MediaTuple{allArgs["vhost"], allArgs["app"], allArgs["stream"]}; auto tuple = MediaTuple{allArgs["vhost"], allArgs["app"], allArgs["stream"], ""};
auto record_path = Recorder::getRecordPath(Recorder::type_mp4, tuple, allArgs["customized_path"]); auto record_path = Recorder::getRecordPath(Recorder::type_mp4, tuple, allArgs["customized_path"]);
auto period = allArgs["period"]; auto period = allArgs["period"];
record_path = record_path + period + "/"; record_path = record_path + period + "/";
@ -1607,7 +1603,7 @@ void installWebApi() {
api_regist("/index/api/getMP4RecordFile", [](API_ARGS_MAP){ api_regist("/index/api/getMP4RecordFile", [](API_ARGS_MAP){
CHECK_SECRET(); CHECK_SECRET();
CHECK_ARGS("vhost", "app", "stream"); CHECK_ARGS("vhost", "app", "stream");
auto tuple = MediaTuple{allArgs["vhost"], allArgs["app"], allArgs["stream"]}; auto tuple = MediaTuple{allArgs["vhost"], allArgs["app"], allArgs["stream"], ""};
auto record_path = Recorder::getRecordPath(Recorder::type_mp4, tuple, allArgs["customized_path"]); auto record_path = Recorder::getRecordPath(Recorder::type_mp4, tuple, allArgs["customized_path"]);
auto period = allArgs["period"]; auto period = allArgs["period"];
@ -1695,7 +1691,7 @@ void installWebApi() {
//截图存在,且未过期,那么返回之 //截图存在,且未过期,那么返回之
res_old_snap = true; res_old_snap = true;
responseSnap(path, allArgs.getParser().getHeader(), invoker); responseSnap(path, allArgs.parser.getHeader(), invoker);
//中断遍历 //中断遍历
return false; return false;
}); });
@ -1726,7 +1722,7 @@ void installWebApi() {
File::delete_file(new_snap); File::delete_file(new_snap);
rename(new_snap_tmp.data(), new_snap.data()); rename(new_snap_tmp.data(), new_snap.data());
} }
responseSnap(new_snap, allArgs.getParser().getHeader(), invoker, err_msg); responseSnap(new_snap, allArgs.parser.getHeader(), invoker, err_msg);
}); });
}); });
@ -1741,7 +1737,7 @@ void installWebApi() {
#ifdef ENABLE_WEBRTC #ifdef ENABLE_WEBRTC
class WebRtcArgsImp : public WebRtcArgs { class WebRtcArgsImp : public WebRtcArgs {
public: public:
WebRtcArgsImp(const HttpAllArgs<string> &args, std::string session_id) WebRtcArgsImp(const ArgsString &args, std::string session_id)
: _args(args) : _args(args)
, _session_id(std::move(session_id)) {} , _session_id(std::move(session_id)) {}
~WebRtcArgsImp() override = default; ~WebRtcArgsImp() override = default;
@ -1759,40 +1755,26 @@ void installWebApi() {
CHECK_ARGS("app", "stream"); CHECK_ARGS("app", "stream");
return StrPrinter << "rtc://" << _args["Host"] << "/" << _args["app"] << "/" return StrPrinter << "rtc://" << _args["Host"] << "/" << _args["app"] << "/"
<< _args["stream"] << "?" << _args.getParser().params() + "&session=" + _session_id; << _args["stream"] << "?" << _args.parser.params() + "&session=" + _session_id;
} }
private: private:
HttpAllArgs<string> _args; ArgsString _args;
std::string _session_id; std::string _session_id;
}; };
api_regist("/index/api/webrtc",[](API_ARGS_STRING_ASYNC){ api_regist("/index/api/webrtc",[](API_ARGS_STRING_ASYNC){
CHECK_ARGS("type"); CHECK_ARGS("type");
auto type = allArgs["type"]; auto type = allArgs["type"];
auto offer = allArgs.getArgs(); auto offer = allArgs.args;
CHECK(!offer.empty(), "http body(webrtc offer sdp) is empty"); CHECK(!offer.empty(), "http body(webrtc offer sdp) is empty");
std::string host = allArgs.getParser()["Host"];
std::string localIp = host.substr(0, host.find(':'));
auto isVaildIP = [](std::string ip)-> bool {
int a,b,c,d;
return sscanf(ip.c_str(),"%d.%d.%d.%d", &a, &b, &c, &d) == 4;
};
if (!isVaildIP(localIp) || localIp=="127.0.0.1") {
localIp = "";
}
auto &session = static_cast<Session&>(sender);
auto args = std::make_shared<WebRtcArgsImp>(allArgs, sender.getIdentifier()); auto args = std::make_shared<WebRtcArgsImp>(allArgs, sender.getIdentifier());
WebRtcPluginManager::Instance().getAnswerSdp(static_cast<Session&>(sender), type, *args, [invoker, val, offer, headerOut, localIp](const WebRtcInterface &exchanger) mutable { WebRtcPluginManager::Instance().negotiateSdp(session, type, *args, [invoker, val, offer, headerOut](const WebRtcInterface &exchanger) mutable {
//设置返回类型 auto &handler = const_cast<WebRtcInterface &>(exchanger);
headerOut["Content-Type"] = HttpFileManager::getContentType(".json");
//设置跨域
headerOut["Access-Control-Allow-Origin"] = "*";
try { try {
setLocalIp(exchanger,localIp); val["sdp"] = handler.getAnswerSdp(offer);
val["sdp"] = exchangeSdp(exchanger, offer);
val["id"] = exchanger.getIdentifier(); val["id"] = exchanger.getIdentifier();
val["type"] = "answer"; val["type"] = "answer";
invoker(200, headerOut, val.toStyledString()); invoker(200, headerOut, val.toStyledString());
@ -1806,21 +1788,19 @@ void installWebApi() {
static constexpr char delete_webrtc_url [] = "/index/api/delete_webrtc"; static constexpr char delete_webrtc_url [] = "/index/api/delete_webrtc";
static auto whip_whep_func = [](const char *type, API_ARGS_STRING_ASYNC) { static auto whip_whep_func = [](const char *type, API_ARGS_STRING_ASYNC) {
auto offer = allArgs.getArgs(); auto offer = allArgs.args;
CHECK(!offer.empty(), "http body(webrtc offer sdp) is empty"); CHECK(!offer.empty(), "http body(webrtc offer sdp) is empty");
auto &session = static_cast<Session&>(sender); auto &session = static_cast<Session&>(sender);
auto location = std::string("http") + (session.overSsl() ? "s" : "") + "://" + allArgs["host"] + delete_webrtc_url; auto location = std::string(session.overSsl() ? "https://" : "http://") + allArgs["host"] + delete_webrtc_url;
auto args = std::make_shared<WebRtcArgsImp>(allArgs, sender.getIdentifier()); auto args = std::make_shared<WebRtcArgsImp>(allArgs, sender.getIdentifier());
WebRtcPluginManager::Instance().getAnswerSdp(session, type, *args, WebRtcPluginManager::Instance().negotiateSdp(session, type, *args, [invoker, offer, headerOut, location](const WebRtcInterface &exchanger) mutable {
[invoker, offer, headerOut, location](const WebRtcInterface &exchanger) mutable { auto &handler = const_cast<WebRtcInterface &>(exchanger);
// 设置跨域
headerOut["Access-Control-Allow-Origin"] = "*";
try { try {
// 设置返回类型 // 设置返回类型
headerOut["Content-Type"] = "application/sdp"; headerOut["Content-Type"] = "application/sdp";
headerOut["Location"] = location + "?id=" + exchanger.getIdentifier() + "&token=" + exchanger.deleteRandStr(); headerOut["Location"] = location + "?id=" + exchanger.getIdentifier() + "&token=" + exchanger.deleteRandStr();
invoker(201, headerOut, exchangeSdp(exchanger, offer)); invoker(201, headerOut, handler.getAnswerSdp(offer));
} catch (std::exception &ex) { } catch (std::exception &ex) {
headerOut["Content-Type"] = "text/plain"; headerOut["Content-Type"] = "text/plain";
invoker(406, headerOut, ex.what()); invoker(406, headerOut, ex.what());
@ -1833,7 +1813,7 @@ void installWebApi() {
api_regist(delete_webrtc_url, [](API_ARGS_MAP_ASYNC) { api_regist(delete_webrtc_url, [](API_ARGS_MAP_ASYNC) {
CHECK_ARGS("id", "token"); CHECK_ARGS("id", "token");
CHECK(allArgs.getParser().method() == "DELETE", "http method is not DELETE: " + allArgs.getParser().method()); CHECK(allArgs.parser.method() == "DELETE", "http method is not DELETE: " + allArgs.parser.method());
auto obj = WebRtcTransportManager::Instance().getItem(allArgs["id"]); auto obj = WebRtcTransportManager::Instance().getItem(allArgs["id"]);
if (!obj) { if (!obj) {
invoker(404, headerOut, "id not found"); invoker(404, headerOut, "id not found");
@ -1919,44 +1899,39 @@ void installWebApi() {
if (!save_name.empty()) { if (!save_name.empty()) {
res_header.emplace("Content-Disposition", "attachment;filename=\"" + save_name + "\""); res_header.emplace("Content-Disposition", "attachment;filename=\"" + save_name + "\"");
} }
invoker.responseFile(allArgs.getParser().getHeader(), res_header, allArgs["file_path"]); invoker.responseFile(allArgs.parser.getHeader(), res_header, allArgs["file_path"]);
} }
}; };
bool flag = NOTICE_EMIT(BroadcastHttpAccessArgs, Broadcast::kBroadcastHttpAccess, allArgs.getParser(), file_path, false, file_invoker, sender); bool flag = NOTICE_EMIT(BroadcastHttpAccessArgs, Broadcast::kBroadcastHttpAccess, allArgs.parser, file_path, false, file_invoker, sender);
if (!flag) { if (!flag) {
// 文件下载鉴权事件无人监听,不允许下载 // 文件下载鉴权事件无人监听,不允许下载
invoker(401, StrCaseMap {}, "None http access event listener"); invoker(401, StrCaseMap {}, "None http access event listener");
} }
}); });
#if defined(ENABLE_X264) && defined(ENABLE_FFMPEG) #if defined(ENABLE_VIDEOSTACK) && defined(ENABLE_X264) && defined(ENABLE_FFMPEG)
VideoStackManager::Instance().loadBgImg("novideo.yuv"); VideoStackManager::Instance().loadBgImg("novideo.yuv");
NoticeCenter::Instance().addListener(nullptr, Broadcast::kBroadcastStreamNoneReader, [](BroadcastStreamNoneReaderArgs) { NoticeCenter::Instance().addListener(nullptr, Broadcast::kBroadcastStreamNoneReader, [](BroadcastStreamNoneReaderArgs) {
auto id = sender.getMediaTuple().stream; auto id = sender.getMediaTuple().stream;
VideoStackManager::Instance().stopVideoStack(id); VideoStackManager::Instance().stopVideoStack(id);
InfoL << "VideoStack: " << id <<" stop";
}); });
api_regist("/index/api/stack/start", [](API_ARGS_JSON_ASYNC) { api_regist("/index/api/stack/start", [](API_ARGS_JSON_ASYNC) {
CHECK_SECRET(); CHECK_SECRET();
auto ret = VideoStackManager::Instance().startVideoStack(allArgs.getArgs()); auto ret = VideoStackManager::Instance().startVideoStack(allArgs.args);
if (!ret) { val["code"] = ret;
invoker(200, headerOut, "success"); val["msg"] = ret ? "failed" : "success";
} else { invoker(200, headerOut, val.toStyledString());
invoker(200, headerOut, "failed");
}
}); });
api_regist("/index/api/stack/stop", [](API_ARGS_MAP_ASYNC) { api_regist("/index/api/stack/stop", [](API_ARGS_MAP_ASYNC) {
CHECK_SECRET(); CHECK_SECRET();
CHECK_ARGS("id"); CHECK_ARGS("id");
auto ret = VideoStackManager::Instance().stopVideoStack(allArgs["id"]); auto ret = VideoStackManager::Instance().stopVideoStack(allArgs["id"]);
if (!ret) { val["code"] = ret;
invoker(200, headerOut, "success"); val["msg"] = ret ? "failed" : "success";
} else { invoker(200, headerOut, val.toStyledString());
invoker(200, headerOut, "failed");
}
}); });
#endif #endif
} }

View File

@ -115,72 +115,41 @@ std::string getValue(const mediakit::Parser &parser, Args &args, const First &fi
template<typename Args> template<typename Args>
class HttpAllArgs { class HttpAllArgs {
mediakit::Parser* _parser = nullptr;
Args* _args = nullptr;
public: public:
HttpAllArgs(const mediakit::Parser &parser, Args &args) { const mediakit::Parser& parser;
_get_args = [&args]() { Args& args;
return (void *) &args;
};
_get_parser = [&parser]() -> const mediakit::Parser & {
return parser;
};
_get_value = [](HttpAllArgs &that, const std::string &key) {
return getValue(that.getParser(), that.getArgs(), key);
};
_clone = [&](HttpAllArgs &that) {
that._get_args = [args]() {
return (void *) &args;
};
that._get_parser = [parser]() -> const mediakit::Parser & {
return parser;
};
that._get_value = [](HttpAllArgs &that, const std::string &key) {
return getValue(that.getParser(), that.getArgs(), key);
};
that._cache_able = true;
};
}
HttpAllArgs(const HttpAllArgs &that) { HttpAllArgs(const mediakit::Parser &p, Args &a): parser(p), args(a) {}
if (that._cache_able) {
_get_args = that._get_args; HttpAllArgs(const HttpAllArgs &that): _parser(new mediakit::Parser(that.parser)),
_get_parser = that._get_parser; _args(new Args(that.args)),
_get_value = that._get_value; parser(*_parser), args(*_args) {}
_cache_able = true; ~HttpAllArgs() {
} else { if (_parser) {
that._clone(*this); delete _parser;
}
if (_args) {
delete _args;
} }
} }
template<typename Key> template<typename Key>
toolkit::variant operator[](const Key &key) const { toolkit::variant operator[](const Key &key) const {
return (toolkit::variant)_get_value(*(HttpAllArgs*)this, key); return (toolkit::variant)getValue(parser, args, key);
} }
const mediakit::Parser &getParser() const {
return _get_parser();
}
Args &getArgs() {
return *((Args *) _get_args());
}
const Args &getArgs() const {
return *((Args *) _get_args());
}
private:
bool _cache_able = false;
std::function<void *() > _get_args;
std::function<const mediakit::Parser &() > _get_parser;
std::function<std::string(HttpAllArgs &that, const std::string &key)> _get_value;
std::function<void(HttpAllArgs &that) > _clone;
}; };
#define API_ARGS_MAP toolkit::SockInfo &sender, mediakit::HttpSession::KeyValue &headerOut, const HttpAllArgs<ApiArgsType> &allArgs, Json::Value &val using ArgsMap = HttpAllArgs<ApiArgsType>;
using ArgsJson = HttpAllArgs<Json::Value>;
using ArgsString = HttpAllArgs<std::string>;
#define API_ARGS_MAP toolkit::SockInfo &sender, mediakit::HttpSession::KeyValue &headerOut, const ArgsMap &allArgs, Json::Value &val
#define API_ARGS_MAP_ASYNC API_ARGS_MAP, const mediakit::HttpSession::HttpResponseInvoker &invoker #define API_ARGS_MAP_ASYNC API_ARGS_MAP, const mediakit::HttpSession::HttpResponseInvoker &invoker
#define API_ARGS_JSON toolkit::SockInfo &sender, mediakit::HttpSession::KeyValue &headerOut, const HttpAllArgs<Json::Value> &allArgs, Json::Value &val #define API_ARGS_JSON toolkit::SockInfo &sender, mediakit::HttpSession::KeyValue &headerOut, const ArgsJson &allArgs, Json::Value &val
#define API_ARGS_JSON_ASYNC API_ARGS_JSON, const mediakit::HttpSession::HttpResponseInvoker &invoker #define API_ARGS_JSON_ASYNC API_ARGS_JSON, const mediakit::HttpSession::HttpResponseInvoker &invoker
#define API_ARGS_STRING toolkit::SockInfo &sender, mediakit::HttpSession::KeyValue &headerOut, const HttpAllArgs<std::string> &allArgs, Json::Value &val #define API_ARGS_STRING toolkit::SockInfo &sender, mediakit::HttpSession::KeyValue &headerOut, const ArgsString &allArgs, Json::Value &val
#define API_ARGS_STRING_ASYNC API_ARGS_STRING, const mediakit::HttpSession::HttpResponseInvoker &invoker #define API_ARGS_STRING_ASYNC API_ARGS_STRING, const mediakit::HttpSession::HttpResponseInvoker &invoker
#define API_ARGS_VALUE sender, headerOut, allArgs, val #define API_ARGS_VALUE sender, headerOut, allArgs, val
@ -234,8 +203,6 @@ void unInstallWebApi();
#if defined(ENABLE_RTPPROXY) #if defined(ENABLE_RTPPROXY)
uint16_t openRtpServer(uint16_t local_port, const std::string &stream_id, int tcp_mode, const std::string &local_ip, bool re_use_port, uint32_t ssrc, int only_track, bool multiplex=false); uint16_t openRtpServer(uint16_t local_port, const std::string &stream_id, int tcp_mode, const std::string &local_ip, bool re_use_port, uint32_t ssrc, int only_track, bool multiplex=false);
void connectRtpServer(const std::string &stream_id, const std::string &dst_url, uint16_t dst_port, const std::function<void(const toolkit::SockException &ex)> &cb);
bool closeRtpServer(const std::string &stream_id);
#endif #endif
Json::Value makeMediaSourceJson(mediakit::MediaSource &media); Json::Value makeMediaSourceJson(mediakit::MediaSource &media);

View File

@ -225,7 +225,7 @@ static ArgsType make_json(const MediaInfo &args) {
ArgsType body; ArgsType body;
body["schema"] = args.schema; body["schema"] = args.schema;
dumpMediaTuple(args, body); dumpMediaTuple(args, body);
body["params"] = args.param_strs; body["params"] = args.params;
return body; return body;
} }
@ -286,7 +286,7 @@ static string getPullUrl(const string &origin_fmt, const MediaInfo &info) {
return ""; return "";
} }
// 告知源站这是来自边沿站的拉流请求,如果未找到流请立即返回拉流失败 // 告知源站这是来自边沿站的拉流请求,如果未找到流请立即返回拉流失败
return string(url) + '?' + kEdgeServerParam + '&' + VHOST_KEY + '=' + info.vhost + '&' + info.param_strs; return string(url) + '?' + kEdgeServerParam + '&' + VHOST_KEY + '=' + info.vhost + '&' + info.params;
} }
static void pullStreamFromOrigin(const vector<string> &urls, size_t index, size_t failed_cnt, const MediaInfo &args, const function<void()> &closePlayer) { static void pullStreamFromOrigin(const vector<string> &urls, size_t index, size_t failed_cnt, const MediaInfo &args, const function<void()> &closePlayer) {
@ -498,7 +498,7 @@ void installWebHook() {
return; return;
} }
if (start_with(args.param_strs, kEdgeServerParam)) { if (start_with(args.params, kEdgeServerParam)) {
// 源站收到来自边沿站的溯源请求,流不存在时立即返回拉流失败 // 源站收到来自边沿站的溯源请求,流不存在时立即返回拉流失败
closePlayer(); closePlayer();
return; return;

View File

@ -38,7 +38,7 @@
#endif #endif
#if defined(ENABLE_VERSION) #if defined(ENABLE_VERSION)
#include "version.h" #include "ZLMVersion.h"
#endif #endif
#if !defined(_WIN32) #if !defined(_WIN32)
@ -258,6 +258,15 @@ int start_main(int argc,char *argv[]) {
//加载配置文件,如果配置文件不存在就创建一个 //加载配置文件,如果配置文件不存在就创建一个
loadIniConfig(g_ini_file.data()); loadIniConfig(g_ini_file.data());
auto &secret = mINI::Instance()[API::kSecret];
if (secret == "035c73f7-bb6b-4889-a715-d9eb2d1925cc" || secret.empty()) {
// 使用默认secret被禁止启动
secret = makeRandStr(32, true);
mINI::Instance().dumpFile(g_ini_file);
WarnL << "The " << API::kSecret << " is invalid, modified it to: " << secret
<< ", saved config file: " << g_ini_file;
}
if (!File::is_dir(ssl_file)) { if (!File::is_dir(ssl_file)) {
// 不是文件夹,加载证书,证书包含公钥和私钥 // 不是文件夹,加载证书,证书包含公钥和私钥
SSL_Initor::Instance().loadCertificate(ssl_file.data()); SSL_Initor::Instance().loadCertificate(ssl_file.data());
@ -352,14 +361,6 @@ int start_main(int argc,char *argv[]) {
InfoL << "已启动http hook 接口"; InfoL << "已启动http hook 接口";
try { try {
auto &secret = mINI::Instance()[API::kSecret];
if (secret == "035c73f7-bb6b-4889-a715-d9eb2d1925cc" || secret.empty()) {
// 使用默认secret被禁止启动
secret = makeRandStr(32, true);
mINI::Instance().dumpFile(g_ini_file);
WarnL << "The " << API::kSecret << " is invalid, modified it to: " << secret
<< ", saved config file: " << g_ini_file;
}
//rtsp服务器端口默认554 //rtsp服务器端口默认554
if (rtspPort) { rtspSrv->start<RtspSession>(rtspPort); } if (rtspPort) { rtspSrv->start<RtspSession>(rtspPort); }
//rtsps服务器端口默认322 //rtsps服务器端口默认322

View File

@ -113,7 +113,7 @@ ProtocolOption::ProtocolOption() {
////////////////////////////////////////////////////////////////////////////////////////////////////////////// //////////////////////////////////////////////////////////////////////////////////////////////////////////////
struct MediaSourceNull : public MediaSource { struct MediaSourceNull : public MediaSource {
MediaSourceNull() : MediaSource("schema", MediaTuple{"vhost", "app", "stream"}) {}; MediaSourceNull() : MediaSource("schema", MediaTuple{"vhost", "app", "stream", ""}) {};
int readerCount() override { return 0; } int readerCount() override { return 0; }
}; };
@ -583,7 +583,7 @@ void MediaInfo::parse(const std::string &url_in){
auto url = url_in; auto url = url_in;
auto pos = url.find("?"); auto pos = url.find("?");
if (pos != string::npos) { if (pos != string::npos) {
param_strs = url.substr(pos + 1); params = url.substr(pos + 1);
url.erase(pos); url.erase(pos);
} }
@ -616,9 +616,10 @@ void MediaInfo::parse(const std::string &url_in){
stream = stream_id; stream = stream_id;
} }
auto params = Parser::parseArgs(param_strs); auto kv = Parser::parseArgs(params);
if (params.find(VHOST_KEY) != params.end()) { auto it = kv.find(VHOST_KEY);
vhost = params[VHOST_KEY]; if (it != kv.end()) {
vhost = it->second;
} }
GET_CONFIG(bool, enableVhost, General::kEnableVhost); GET_CONFIG(bool, enableVhost, General::kEnableVhost);

View File

@ -136,6 +136,15 @@ private:
toolkit::Timer::Ptr _async_close_timer; toolkit::Timer::Ptr _async_close_timer;
}; };
template <typename MAP, typename KEY, typename TYPE>
static void getArgsValue(const MAP &allArgs, const KEY &key, TYPE &value) {
auto val = ((MAP &)allArgs)[key];
if (!val.empty()) {
value = (TYPE)val;
}
}
class ProtocolOption { class ProtocolOption {
public: public:
ProtocolOption(); ProtocolOption();
@ -243,15 +252,6 @@ public:
GET_OPT_VALUE(stream_replace); GET_OPT_VALUE(stream_replace);
GET_OPT_VALUE(max_track); GET_OPT_VALUE(max_track);
} }
private:
template <typename MAP, typename KEY, typename TYPE>
static void getArgsValue(const MAP &allArgs, const KEY &key, TYPE &value) {
auto val = ((MAP &)allArgs)[key];
if (!val.empty()) {
value = (TYPE)val;
}
}
}; };
//该对象用于拦截感兴趣的MediaSourceEvent事件 //该对象用于拦截感兴趣的MediaSourceEvent事件
@ -299,7 +299,6 @@ public:
std::string full_url; std::string full_url;
std::string schema; std::string schema;
std::string host; std::string host;
std::string param_strs;
}; };
bool equalMediaTuple(const MediaTuple& a, const MediaTuple& b); bool equalMediaTuple(const MediaTuple& a, const MediaTuple& b);

View File

@ -294,8 +294,8 @@ void RtspUrl::setup(bool is_ssl, const string &url, const string &user, const st
splitUrl(ip, ip, port); splitUrl(ip, ip, port);
_url = std::move(url); _url = std::move(url);
_user = strCoding::UrlDecode(std::move(user)); _user = strCoding::UrlDecodeUserOrPass(user);
_passwd = strCoding::UrlDecode(std::move(passwd)); _passwd = strCoding::UrlDecodeUserOrPass(passwd);
_host = std::move(ip); _host = std::move(ip);
_port = port; _port = port;
_is_ssl = is_ssl; _is_ssl = is_ssl;

View File

@ -30,7 +30,7 @@ struct StrCaseCompare {
class StrCaseMap : public std::multimap<std::string, std::string, StrCaseCompare> { class StrCaseMap : public std::multimap<std::string, std::string, StrCaseCompare> {
public: public:
using Super = multimap<std::string, std::string, StrCaseCompare>; using Super = std::multimap<std::string, std::string, StrCaseCompare>;
std::string &operator[](const std::string &k) { std::string &operator[](const std::string &k) {
auto it = find(k); auto it = find(k);

View File

@ -14,7 +14,7 @@
using namespace toolkit; using namespace toolkit;
#if defined(ENABLE_VERSION) #if defined(ENABLE_VERSION)
#include "version.h" #include "ZLMVersion.h"
#endif #endif
extern "C" { extern "C" {

View File

@ -36,6 +36,16 @@
#define CHECK(exp, ...) ::mediakit::Assert_ThrowCpp(!(exp), #exp, __FUNCTION__, __FILE__, __LINE__, ##__VA_ARGS__) #define CHECK(exp, ...) ::mediakit::Assert_ThrowCpp(!(exp), #exp, __FUNCTION__, __FILE__, __LINE__, ##__VA_ARGS__)
#endif // CHECK #endif // CHECK
#ifndef CHECK_RET
#define CHECK_RET(...) \
try { \
CHECK(__VA_ARGS__); \
} catch (AssertFailedException & ex) { \
WarnL << ex.what(); \
return; \
}
#endif
#ifndef MAX #ifndef MAX
#define MAX(a, b) ((a) > (b) ? (a) : (b)) #define MAX(a, b) ((a) > (b) ? (a) : (b))
#endif // MAX #endif // MAX

View File

@ -52,25 +52,7 @@ char HexStrToBin(const char *str) {
} }
return (high << 4) | low; return (high << 4) | low;
} }
static string UrlEncodeCommon(const string &str,const char* dont_escape){
string strCoding::UrlEncode(const string &str) {
string out;
size_t len = str.size();
for (size_t i = 0; i < len; ++i) {
char ch = str[i];
if (isalnum((uint8_t) ch)) {
out.push_back(ch);
} else {
char buf[4];
sprintf(buf, "%%%X%X", (uint8_t) ch >> 4, (uint8_t) ch & 0x0F);
out.append(buf);
}
}
return out;
}
string strCoding::UrlEncodePath(const string &str) {
const char *dont_escape = "!#&'*+:=?@/._-$,;~()";
string out; string out;
size_t len = str.size(); size_t len = str.size();
for (size_t i = 0; i < len; ++i) { for (size_t i = 0; i < len; ++i) {
@ -85,52 +67,7 @@ string strCoding::UrlEncodePath(const string &str) {
} }
return out; return out;
} }
static string UrlDecodeCommon(const string &str,const char* dont_unescape){
string strCoding::UrlEncodeComponent(const string &str) {
const char *dont_escape = "!'()*-._~";
string out;
size_t len = str.size();
for (size_t i = 0; i < len; ++i) {
char ch = str[i];
if (isalnum((uint8_t) ch) || strchr(dont_escape, (uint8_t) ch) != NULL) {
out.push_back(ch);
} else {
char buf[4];
snprintf(buf, 4, "%%%X%X", (uint8_t) ch >> 4, (uint8_t) ch & 0x0F);
out.append(buf);
}
}
return out;
}
string strCoding::UrlDecode(const string &str) {
string output;
size_t i = 0, len = str.length();
while (i < len) {
if (str[i] == '%') {
if (i + 3 > len) {
// %后面必须还有两个字节才会反转义
output.append(str, i, len - i);
break;
}
char ch = HexStrToBin(&(str[i + 1]));
if (ch == -1) {
// %后面两个字节不是16进制字符串转义失败直接拼接3个原始字符
output.append(str, i, 3);
} else {
output += ch;
}
i += 3;
} else {
output += str[i];
++i;
}
}
return output;
}
string strCoding::UrlDecodePath(const string &str) {
const char *dont_unescape = "#$&+,/:;=?@";
string output; string output;
size_t i = 0, len = str.length(); size_t i = 0, len = str.length();
while (i < len) { while (i < len) {
@ -156,6 +93,36 @@ string strCoding::UrlDecodePath(const string &str) {
return output; return output;
} }
string strCoding::UrlEncodePath(const string &str) {
const char *dont_escape = "!#&'*+:=?@/._-$,;~()";
return UrlEncodeCommon(str,dont_escape);
}
string strCoding::UrlEncodeComponent(const string &str) {
const char *dont_escape = "!'()*-._~";
return UrlEncodeCommon(str,dont_escape);
}
std::string strCoding::UrlEncodeUserOrPass(const std::string &str) {
// from rfc https://datatracker.ietf.org/doc/html/rfc3986
// §2.3 Unreserved characters (mark)
//'-', '_', '.', '~'
// §2.2 Reserved characters (reserved)
// '$', '&', '+', ',', '/', ':', ';', '=', '?', '@',
// §3.2.1
// The RFC allows ';', ':', '&', '=', '+', '$', and ',' in
// userinfo, so we must escape only '@', '/', and '?'.
// The parsing of userinfo treats ':' as special so we must escape
// that too.
const char *dont_escape = "$&+,;=-._~";
return UrlEncodeCommon(str,dont_escape);
}
string strCoding::UrlDecodePath(const string &str) {
const char *dont_unescape = "#$&+,/:;=?@";
return UrlDecodeCommon(str,dont_unescape);
}
std::string strCoding::UrlDecodeComponent(const std::string &str) { std::string strCoding::UrlDecodeComponent(const std::string &str) {
string output; string output;
size_t i = 0, len = str.length(); size_t i = 0, len = str.length();
@ -185,27 +152,11 @@ std::string strCoding::UrlDecodeComponent(const std::string &str) {
return output; return output;
} }
#if 0
#include "Util/onceToken.h"
static toolkit::onceToken token([]() {
auto str0 = strCoding::UrlDecode(
"rtsp%3A%2F%2Fadmin%3AJm13317934%25jm%40111.47.84.69%3A554%2FStreaming%2FChannels%2F101%3Ftransportmode%3Dunicast%26amp%3Bprofile%3DProfile_1");
auto str1 = strCoding::UrlDecode("%j1"); // 测试%后面两个字节不是16进制字符串
auto str2 = strCoding::UrlDecode("%a"); // 测试%后面字节数不够
auto str3 = strCoding::UrlDecode("%"); // 测试只有%
auto str4 = strCoding::UrlDecode("%%%"); // 测试多个%
auto str5 = strCoding::UrlDecode("%%%%40"); // 测试多个非法%后恢复正常解析
auto str6 = strCoding::UrlDecode("Jm13317934%jm"); // 测试多个非法%后恢复正常解析
cout << str0 << endl;
cout << str1 << endl;
cout << str2 << endl;
cout << str3 << endl;
cout << str4 << endl;
cout << str5 << endl;
cout << str6 << endl;
});
#endif
std::string strCoding::UrlDecodeUserOrPass(const std::string &str) {
const char *dont_unescape = "";
return UrlDecodeCommon(str,dont_unescape);
}
///////////////////////////////windows专用/////////////////////////////////// ///////////////////////////////windows专用///////////////////////////////////
#if defined(_WIN32) #if defined(_WIN32)
void UnicodeToGB2312(char* pOut, wchar_t uData) void UnicodeToGB2312(char* pOut, wchar_t uData)

View File

@ -18,12 +18,12 @@ namespace mediakit {
class strCoding { class strCoding {
public: public:
[[deprecated]] static std::string UrlEncode(const std::string &str); //url utf8编码, deprecated
static std::string UrlEncodePath(const std::string &str); //url路径 utf8编码 static std::string UrlEncodePath(const std::string &str); //url路径 utf8编码
static std::string UrlEncodeComponent(const std::string &str); // url参数 utf8编码 static std::string UrlEncodeComponent(const std::string &str); // url参数 utf8编码
[[deprecated]] static std::string UrlDecode(const std::string &str); //url utf8解码, deprecated
static std::string UrlDecodePath(const std::string &str); //url路径 utf8解码 static std::string UrlDecodePath(const std::string &str); //url路径 utf8解码
static std::string UrlDecodeComponent(const std::string &str); // url参数 utf8解码 static std::string UrlDecodeComponent(const std::string &str); // url参数 utf8解码
static std::string UrlEncodeUserOrPass(const std::string &str); // url中用户名与密码编码
static std::string UrlDecodeUserOrPass(const std::string &str); // url中用户名与密码解码
#if defined(_WIN32) #if defined(_WIN32)
static std::string UTF8ToGB2312(const std::string &str);//utf_8转为gb2312 static std::string UTF8ToGB2312(const std::string &str);//utf_8转为gb2312
static std::string GB2312ToUTF8(const std::string &str); //gb2312 转utf_8 static std::string GB2312ToUTF8(const std::string &str); //gb2312 转utf_8

View File

@ -65,18 +65,18 @@ void HttpRequestSplitter::input(const char *data,size_t len) {
_content_len = onRecvHeader(header_ptr, header_size); _content_len = onRecvHeader(header_ptr, header_size);
} }
if(_remain_data_size <= 0){
//没有剩余数据,清空缓存
_remain_data.clear();
return;
}
/* /*
* *
* HttpRequestSplitter::reset() * HttpRequestSplitter::reset()
*/ */
tail_ref = tail_tmp; tail_ref = tail_tmp;
if(_remain_data_size <= 0){
//没有剩余数据,清空缓存
_remain_data.clear();
return;
}
if(_content_len == 0){ if(_content_len == 0){
//尚未找到http头缓存定位到剩余数据部分 //尚未找到http头缓存定位到剩余数据部分
_remain_data.assign(ptr,_remain_data_size); _remain_data.assign(ptr,_remain_data_size);

View File

@ -683,18 +683,6 @@ void HttpSession::sendResponse(int code,
AsyncSender::onSocketFlushed(data); AsyncSender::onSocketFlushed(data);
} }
string HttpSession::urlDecode(const string &str) {
auto ret = strCoding::UrlDecode(str);
#ifdef _WIN32
GET_CONFIG(string, charSet, Http::kCharSet);
bool isGb2312 = !strcasecmp(charSet.data(), "gb2312");
if (isGb2312) {
ret = strCoding::UTF8ToGB2312(ret);
}
#endif // _WIN32
return ret;
}
string HttpSession::urlDecodePath(const string &str) { string HttpSession::urlDecodePath(const string &str) {
auto ret = strCoding::UrlDecodePath(str); auto ret = strCoding::UrlDecodePath(str);
#ifdef _WIN32 #ifdef _WIN32

View File

@ -44,7 +44,6 @@ public:
void onRecv(const toolkit::Buffer::Ptr &) override; void onRecv(const toolkit::Buffer::Ptr &) override;
void onError(const toolkit::SockException &err) override; void onError(const toolkit::SockException &err) override;
void onManager() override; void onManager() override;
[[deprecated]] static std::string urlDecode(const std::string &str);
static std::string urlDecodePath(const std::string &str); static std::string urlDecodePath(const std::string &str);
static std::string urlDecodeComponent(const std::string &str); static std::string urlDecodeComponent(const std::string &str);
void setTimeoutSec(size_t second); void setTimeoutSec(size_t second);

View File

@ -195,10 +195,8 @@ std::shared_ptr<FILE> HlsMakerImp::makeFile(const string &file, bool setbuf) {
return ret; return ret;
} }
void HlsMakerImp::setMediaSource(const string &vhost, const string &app, const string &stream_id) { void HlsMakerImp::setMediaSource(const MediaTuple& tuple) {
_info.app = app; static_cast<MediaTuple &>(_info) = tuple;
_info.stream = stream_id;
_info.vhost = vhost;
_media_src = std::make_shared<HlsMediaSource>(isFmp4() ? HLS_FMP4_SCHEMA : HLS_SCHEMA, _info); _media_src = std::make_shared<HlsMediaSource>(isFmp4() ? HLS_FMP4_SCHEMA : HLS_SCHEMA, _info);
} }

View File

@ -27,11 +27,8 @@ public:
/** /**
* *
* @param vhost
* @param app
* @param stream_id id
*/ */
void setMediaSource(const std::string &vhost, const std::string &app, const std::string &stream_id); void setMediaSource(const MediaTuple& tuple);
/** /**
* MediaSource * MediaSource

View File

@ -34,7 +34,7 @@ public:
} }
void setMediaSource(const MediaTuple& tuple) { void setMediaSource(const MediaTuple& tuple) {
_hls->setMediaSource(tuple.vhost, tuple.app, tuple.stream); _hls->setMediaSource(tuple);
} }
void setListener(const std::weak_ptr<MediaSourceEvent> &listener) { void setListener(const std::weak_ptr<MediaSourceEvent> &listener) {

View File

@ -38,7 +38,7 @@ MP4Reader::MP4Reader(const std::string &vhost, const std::string &app, const std
void MP4Reader::setup(const std::string &vhost, const std::string &app, const std::string &stream_id, const std::string &file_path, const ProtocolOption &option, toolkit::EventPoller::Ptr poller) { void MP4Reader::setup(const std::string &vhost, const std::string &app, const std::string &stream_id, const std::string &file_path, const ProtocolOption &option, toolkit::EventPoller::Ptr poller) {
//读写文件建议放在后台线程 //读写文件建议放在后台线程
auto tuple = MediaTuple{vhost, app, stream_id}; auto tuple = MediaTuple{vhost, app, stream_id, ""};
_poller = poller ? std::move(poller) : WorkThreadPool::Instance().getPoller(); _poller = poller ? std::move(poller) : WorkThreadPool::Instance().getPoller();
_file_path = file_path; _file_path = file_path;
if (_file_path.empty()) { if (_file_path.empty()) {

View File

@ -22,12 +22,10 @@ using namespace toolkit;
namespace mediakit { namespace mediakit {
MP4Recorder::MP4Recorder(const string &path, const string &vhost, const string &app, const string &stream_id, size_t max_second) { MP4Recorder::MP4Recorder(const MediaTuple &tuple, const string &path, size_t max_second) {
_folder_path = path; _folder_path = path;
/////record 业务逻辑////// /////record 业务逻辑//////
_info.app = app; static_cast<MediaTuple &>(_info) = tuple;
_info.stream = stream_id;
_info.vhost = vhost;
_info.folder = path; _info.folder = path;
GET_CONFIG(uint32_t, s_max_second, Protocol::kMP4MaxSecond); GET_CONFIG(uint32_t, s_max_second, Protocol::kMP4MaxSecond);
_max_second = max_second ? max_second : s_max_second; _max_second = max_second ? max_second : s_max_second;
@ -120,7 +118,7 @@ bool MP4Recorder::inputFrame(const Frame::Ptr &frame) {
//b帧情况下dts时间戳可能回退 //b帧情况下dts时间戳可能回退
_last_dts = MAX(frame->dts(), _last_dts); _last_dts = MAX(frame->dts(), _last_dts);
} }
auto duration = 5; // 默认至少一帧5ms auto duration = 5u; // 默认至少一帧5ms
if (frame->dts() > 0 && frame->dts() > _last_dts) { if (frame->dts() > 0 && frame->dts() > _last_dts) {
duration = MAX(duration, frame->dts() - _last_dts); duration = MAX(duration, frame->dts() - _last_dts);
} }

View File

@ -26,7 +26,7 @@ class MP4Recorder final : public MediaSinkInterface {
public: public:
using Ptr = std::shared_ptr<MP4Recorder>; using Ptr = std::shared_ptr<MP4Recorder>;
MP4Recorder(const std::string &path, const std::string &vhost, const std::string &app, const std::string &stream_id, size_t max_second); MP4Recorder(const MediaTuple &tuple, const std::string &path, size_t max_second);
~MP4Recorder() override; ~MP4Recorder() override;
/** /**

View File

@ -68,8 +68,7 @@ string Recorder::getRecordPath(Recorder::type type, const MediaTuple& tuple, con
} }
return File::absolutePath(m3u8FilePath, hlsPath); return File::absolutePath(m3u8FilePath, hlsPath);
} }
default: default: return "";
return "";
} }
} }
@ -85,13 +84,12 @@ std::shared_ptr<MediaSinkInterface> Recorder::createRecorder(type type, const Me
#else #else
throw std::invalid_argument("hls相关功能未打开请开启ENABLE_HLS宏后编译再测试"); throw std::invalid_argument("hls相关功能未打开请开启ENABLE_HLS宏后编译再测试");
#endif #endif
} }
case Recorder::type_mp4: { case Recorder::type_mp4: {
#if defined(ENABLE_MP4) #if defined(ENABLE_MP4)
auto path = Recorder::getRecordPath(type, tuple, option.mp4_save_path); auto path = Recorder::getRecordPath(type, tuple, option.mp4_save_path);
return std::make_shared<MP4Recorder>(path, tuple.vhost, tuple.app, tuple.stream, option.mp4_max_second); return std::make_shared<MP4Recorder>(tuple, path, option.mp4_max_second);
#else #else
throw std::invalid_argument("mp4相关功能未打开请开启ENABLE_MP4宏后编译再测试"); throw std::invalid_argument("mp4相关功能未打开请开启ENABLE_MP4宏后编译再测试");
#endif #endif

View File

@ -22,6 +22,7 @@ struct MediaTuple {
std::string vhost; std::string vhost;
std::string app; std::string app;
std::string stream; std::string stream;
std::string params;
std::string shortUrl() const { std::string shortUrl() const {
return vhost + '/' + app + '/' + stream; return vhost + '/' + app + '/' + stream;
} }

View File

@ -29,7 +29,13 @@ public:
getRtmpRing()->setDelegate(_media_src); getRtmpRing()->setDelegate(_media_src);
} }
~RtmpMediaSourceMuxer() override { RtmpMuxer::flush(); } ~RtmpMediaSourceMuxer() override {
try {
RtmpMuxer::flush();
} catch (std::exception &ex) {
WarnL << ex.what();
}
}
void setListener(const std::weak_ptr<MediaSourceEvent> &listener){ void setListener(const std::weak_ptr<MediaSourceEvent> &listener){
setDelegate(listener); setDelegate(listener);

View File

@ -165,14 +165,7 @@ void RtmpProtocol::sendResponse(int type, const string &str) {
void RtmpProtocol::sendInvoke(const string &cmd, const AMFValue &val) { void RtmpProtocol::sendInvoke(const string &cmd, const AMFValue &val) {
AMFEncoder enc; AMFEncoder enc;
if (val.type() == AMFType::AMF_OBJECT || val.type() == AMFType::AMF_NULL)
{
enc << cmd << ++_send_req_id << val; enc << cmd << ++_send_req_id << val;
}
else
{
enc << cmd << ++_send_req_id << AMFValue() << val;
}
sendRequest(MSG_CMD, enc.data()); sendRequest(MSG_CMD, enc.data());
} }
@ -632,6 +625,7 @@ const char* RtmpProtocol::handle_rtmp(const char *data, size_t len) {
case 4: case 4:
chunk_data.ts_field = load_be24(header->time_stamp); chunk_data.ts_field = load_be24(header->time_stamp);
} }
auto time_stamp = chunk_data.ts_field; auto time_stamp = chunk_data.ts_field;
if (chunk_data.ts_field == 0xFFFFFF) { if (chunk_data.ts_field == 0xFFFFFF) {
if (len < header_len + offset + 4) { if (len < header_len + offset + 4) {

View File

@ -11,7 +11,6 @@
#ifndef SRC_RTMP_RTMPPROTOCOL_H_ #ifndef SRC_RTMP_RTMPPROTOCOL_H_
#define SRC_RTMP_RTMPPROTOCOL_H_ #define SRC_RTMP_RTMPPROTOCOL_H_
#include <cstdint>
#include <memory> #include <memory>
#include <string> #include <string>
#include <functional> #include <functional>

View File

@ -163,18 +163,6 @@ void RtmpPusher::send_connect() {
} }
void RtmpPusher::send_createStream() { void RtmpPusher::send_createStream() {
// Workaround : 兼容较旧的 FMS3.0
{
{
AMFValue obj(_stream_id);
sendInvoke("releaseStream", obj);
}
{
AMFValue obj(_stream_id);
sendInvoke("FCPublish", obj);
}
}
{
AMFValue obj(AMF_NULL); AMFValue obj(AMF_NULL);
sendInvoke("createStream", obj); sendInvoke("createStream", obj);
addOnResultCB([this](AMFDecoder &dec) { addOnResultCB([this](AMFDecoder &dec) {
@ -183,8 +171,6 @@ void RtmpPusher::send_createStream() {
_stream_index = dec.load<int>(); _stream_index = dec.load<int>();
send_publish(); send_publish();
}); });
}
} }
#define RTMP_STREAM_LIVE "live" #define RTMP_STREAM_LIVE "live"

View File

@ -63,8 +63,8 @@ void SrtTransportImp::onHandShakeFinished(std::string &streamid, struct sockaddr
return; return;
} }
auto params = Parser::parseArgs(_media_info.param_strs); auto kv = Parser::parseArgs(_media_info.params);
if (params["m"] == "publish") { if (kv["m"] == "publish") {
_is_pusher = true; _is_pusher = true;
_decoder = DecoderImp::createDecoder(DecoderImp::decoder_ts, this); _decoder = DecoderImp::createDecoder(DecoderImp::decoder_ts, this);
emitOnPublish(); emitOnPublish();
@ -98,10 +98,10 @@ bool SrtTransportImp::parseStreamid(std::string &streamid) {
app = tmps[0]; app = tmps[0];
stream_name = tmps[1]; stream_name = tmps[1];
} else { } else {
if (_media_info.param_strs.empty()) { if (_media_info.params.empty()) {
_media_info.param_strs = it.first + "=" + it.second; _media_info.params = it.first + "=" + it.second;
} else { } else {
_media_info.param_strs += "&" + it.first + "=" + it.second; _media_info.params += "&" + it.first + "=" + it.second;
} }
} }
} }
@ -118,7 +118,7 @@ bool SrtTransportImp::parseStreamid(std::string &streamid) {
_media_info.app = app; _media_info.app = app;
_media_info.stream = stream_name; _media_info.stream = stream_name;
TraceL << " mediainfo=" << _media_info.shortUrl() << " params=" << _media_info.param_strs; TraceL << " mediainfo=" << _media_info.shortUrl() << " params=" << _media_info.params;
return true; return true;
} }

View File

@ -27,7 +27,7 @@ using namespace mediakit;
class FlvSplitterImp : public FlvSplitter { class FlvSplitterImp : public FlvSplitter {
public: public:
FlvSplitterImp() { FlvSplitterImp() {
_src = std::make_shared<RtmpMediaSourceImp>(MediaTuple{DEFAULT_VHOST, "live", "test"}); _src = std::make_shared<RtmpMediaSourceImp>(MediaTuple{DEFAULT_VHOST, "live", "test", ""});
} }
~FlvSplitterImp() override = default; ~FlvSplitterImp() override = default;

View File

@ -90,7 +90,7 @@ void initEventListener() {
static onceToken s_token([]() { static onceToken s_token([]() {
//监听kBroadcastOnGetRtspRealm事件决定rtsp链接是否需要鉴权(传统的rtsp鉴权方案)才能访问 //监听kBroadcastOnGetRtspRealm事件决定rtsp链接是否需要鉴权(传统的rtsp鉴权方案)才能访问
NoticeCenter::Instance().addListener(nullptr, Broadcast::kBroadcastOnGetRtspRealm, [](BroadcastOnGetRtspRealmArgs) { NoticeCenter::Instance().addListener(nullptr, Broadcast::kBroadcastOnGetRtspRealm, [](BroadcastOnGetRtspRealmArgs) {
DebugL << "RTSP是否需要鉴权事件" << args.getUrl() << " " << args.param_strs; DebugL << "RTSP是否需要鉴权事件" << args.getUrl() << " " << args.params;
if (string("1") == args.stream) { if (string("1") == args.stream) {
// live/1需要认证 // live/1需要认证
//该流需要认证并且设置realm //该流需要认证并且设置realm
@ -104,7 +104,7 @@ void initEventListener() {
//监听kBroadcastOnRtspAuth事件返回正确的rtsp鉴权用户密码 //监听kBroadcastOnRtspAuth事件返回正确的rtsp鉴权用户密码
NoticeCenter::Instance().addListener(nullptr, Broadcast::kBroadcastOnRtspAuth, [](BroadcastOnRtspAuthArgs) { NoticeCenter::Instance().addListener(nullptr, Broadcast::kBroadcastOnRtspAuth, [](BroadcastOnRtspAuthArgs) {
DebugL << "RTSP播放鉴权:" << args.getUrl() << " " << args.param_strs; DebugL << "RTSP播放鉴权:" << args.getUrl() << " " << args.params;
DebugL << "RTSP用户" << user_name << (must_no_encrypt ? " Base64" : " MD5") << " 方式登录"; DebugL << "RTSP用户" << user_name << (must_no_encrypt ? " Base64" : " MD5") << " 方式登录";
string user = user_name; string user = user_name;
//假设我们异步读取数据库 //假设我们异步读取数据库
@ -134,14 +134,14 @@ void initEventListener() {
//监听rtsp/rtmp推流事件返回结果告知是否有推流权限 //监听rtsp/rtmp推流事件返回结果告知是否有推流权限
NoticeCenter::Instance().addListener(nullptr, Broadcast::kBroadcastMediaPublish, [](BroadcastMediaPublishArgs) { NoticeCenter::Instance().addListener(nullptr, Broadcast::kBroadcastMediaPublish, [](BroadcastMediaPublishArgs) {
DebugL << "推流鉴权:" << args.getUrl() << " " << args.param_strs; DebugL << "推流鉴权:" << args.getUrl() << " " << args.params;
invoker("", ProtocolOption());//鉴权成功 invoker("", ProtocolOption());//鉴权成功
//invoker("this is auth failed message");//鉴权失败 //invoker("this is auth failed message");//鉴权失败
}); });
//监听rtsp/rtsps/rtmp/http-flv播放事件返回结果告知是否有播放权限(rtsp通过kBroadcastOnRtspAuth或此事件都可以实现鉴权) //监听rtsp/rtsps/rtmp/http-flv播放事件返回结果告知是否有播放权限(rtsp通过kBroadcastOnRtspAuth或此事件都可以实现鉴权)
NoticeCenter::Instance().addListener(nullptr, Broadcast::kBroadcastMediaPlayed, [](BroadcastMediaPlayedArgs) { NoticeCenter::Instance().addListener(nullptr, Broadcast::kBroadcastMediaPlayed, [](BroadcastMediaPlayedArgs) {
DebugL << "播放鉴权:" << args.getUrl() << " " << args.param_strs; DebugL << "播放鉴权:" << args.getUrl() << " " << args.params;
invoker("");//鉴权成功 invoker("");//鉴权成功
//invoker("this is auth failed message");//鉴权失败 //invoker("this is auth failed message");//鉴权失败
}); });
@ -183,13 +183,13 @@ void initEventListener() {
* *
* ZLMediaKit会把其立即转发给播放器(55) * ZLMediaKit会把其立即转发给播放器(55)
*/ */
DebugL << "未找到流事件:" << args.getUrl() << " " << args.param_strs; DebugL << "未找到流事件:" << args.getUrl() << " " << args.params;
}); });
//监听播放或推流结束时消耗流量事件 //监听播放或推流结束时消耗流量事件
NoticeCenter::Instance().addListener(nullptr, Broadcast::kBroadcastFlowReport, [](BroadcastFlowReportArgs) { NoticeCenter::Instance().addListener(nullptr, Broadcast::kBroadcastFlowReport, [](BroadcastFlowReportArgs) {
DebugL << "播放器(推流器)断开连接事件:" << args.getUrl() << " " << args.param_strs << "\r\n使用流量:" << totalBytes << " bytes,连接时长:" << totalDuration << ""; DebugL << "播放器(推流器)断开连接事件:" << args.getUrl() << " " << args.params << "\r\n使用流量:" << totalBytes << " bytes,连接时长:" << totalDuration << "";
}); });

View File

@ -20,19 +20,19 @@ namespace mediakit {
namespace Rtc { namespace Rtc {
#define RTC_FIELD "rtc." #define RTC_FIELD "rtc."
const string kPreferredCodecA = RTC_FIELD"preferredCodecA"; const string kPreferredCodecA = RTC_FIELD "preferredCodecA";
const string kPreferredCodecV = RTC_FIELD"preferredCodecV"; const string kPreferredCodecV = RTC_FIELD "preferredCodecV";
static onceToken token([]() { static onceToken token([]() {
mINI::Instance()[kPreferredCodecA] = "PCMU,PCMA,opus,mpeg4-generic"; mINI::Instance()[kPreferredCodecA] = "PCMA,PCMU,opus,mpeg4-generic";
mINI::Instance()[kPreferredCodecV] = "H264,H265,AV1,VP9,VP8"; mINI::Instance()[kPreferredCodecV] = "H264,H265,AV1,VP9,VP8";
}); });
} } // namespace Rtc
using onCreateSdpItem = function<SdpItem::Ptr(const string &key, const string &value)>; using onCreateSdpItem = function<SdpItem::Ptr(const string &key, const string &value)>;
static map<string, onCreateSdpItem, StrCaseCompare> sdpItemCreator; static map<string, onCreateSdpItem, StrCaseCompare> sdpItemCreator;
template <typename Item> template <typename Item>
void registerSdpItem(){ void registerSdpItem() {
onCreateSdpItem func = [](const string &key, const string &value) { onCreateSdpItem func = [](const string &key, const string &value) {
auto ret = std::make_shared<Item>(); auto ret = std::make_shared<Item>();
ret->parse(value); ret->parse(value);
@ -47,52 +47,50 @@ public:
virtual RtpDirection getDirection() const = 0; virtual RtpDirection getDirection() const = 0;
}; };
class SdpDirectionSendonly : public SdpItem, public DirectionInterface{ class SdpDirectionSendonly : public SdpItem, public DirectionInterface {
public: public:
const char* getKey() const override { return getRtpDirectionString(getDirection());} const char *getKey() const override { return getRtpDirectionString(getDirection()); }
RtpDirection getDirection() const override {return RtpDirection::sendonly;} RtpDirection getDirection() const override { return RtpDirection::sendonly; }
}; };
class SdpDirectionRecvonly : public SdpItem, public DirectionInterface{ class SdpDirectionRecvonly : public SdpItem, public DirectionInterface {
public: public:
const char* getKey() const override { return getRtpDirectionString(getDirection());} const char *getKey() const override { return getRtpDirectionString(getDirection()); }
RtpDirection getDirection() const override {return RtpDirection::recvonly;} RtpDirection getDirection() const override { return RtpDirection::recvonly; }
}; };
class SdpDirectionSendrecv : public SdpItem, public DirectionInterface{ class SdpDirectionSendrecv : public SdpItem, public DirectionInterface {
public: public:
const char* getKey() const override { return getRtpDirectionString(getDirection());} const char *getKey() const override { return getRtpDirectionString(getDirection()); }
RtpDirection getDirection() const override {return RtpDirection::sendrecv;} RtpDirection getDirection() const override { return RtpDirection::sendrecv; }
}; };
class SdpDirectionInactive : public SdpItem, public DirectionInterface{ class SdpDirectionInactive : public SdpItem, public DirectionInterface {
public: public:
const char* getKey() const override { return getRtpDirectionString(getDirection());} const char *getKey() const override { return getRtpDirectionString(getDirection()); }
RtpDirection getDirection() const override {return RtpDirection::inactive;} RtpDirection getDirection() const override { return RtpDirection::inactive; }
}; };
class DirectionInterfaceImp : public SdpItem, public DirectionInterface{ class DirectionInterfaceImp : public SdpItem, public DirectionInterface {
public: public:
DirectionInterfaceImp(RtpDirection direct){ DirectionInterfaceImp(RtpDirection direct) { direction = direct; }
direction = direct; const char *getKey() const override { return getRtpDirectionString(getDirection()); }
} RtpDirection getDirection() const override { return direction; }
const char* getKey() const override { return getRtpDirectionString(getDirection());}
RtpDirection getDirection() const override {return direction;}
private: private:
RtpDirection direction; RtpDirection direction;
}; };
static bool registerAllItem(){ static bool registerAllItem() {
registerSdpItem<SdpString<'v'> >(); registerSdpItem<SdpString<'v'>>();
registerSdpItem<SdpString<'s'> >(); registerSdpItem<SdpString<'s'>>();
registerSdpItem<SdpString<'i'> >(); registerSdpItem<SdpString<'i'>>();
registerSdpItem<SdpString<'u'> >(); registerSdpItem<SdpString<'u'>>();
registerSdpItem<SdpString<'e'> >(); registerSdpItem<SdpString<'e'>>();
registerSdpItem<SdpString<'p'> >(); registerSdpItem<SdpString<'p'>>();
registerSdpItem<SdpString<'z'> >(); registerSdpItem<SdpString<'z'>>();
registerSdpItem<SdpString<'k'> >(); registerSdpItem<SdpString<'k'>>();
registerSdpItem<SdpString<'r'> >(); registerSdpItem<SdpString<'r'>>();
registerSdpItem<SdpTime>(); registerSdpItem<SdpTime>();
registerSdpItem<SdpOrigin>(); registerSdpItem<SdpOrigin>();
registerSdpItem<SdpConnection>(); registerSdpItem<SdpConnection>();
@ -138,11 +136,11 @@ DtlsRole getDtlsRole(const string &str) {
return it == dtls_role_map.end() ? DtlsRole::invalid : it->second; return it == dtls_role_map.end() ? DtlsRole::invalid : it->second;
} }
const char* getDtlsRoleString(DtlsRole role){ const char *getDtlsRoleString(DtlsRole role) {
switch (role) { switch (role) {
case DtlsRole::active : return "active"; case DtlsRole::active: return "active";
case DtlsRole::passive : return "passive"; case DtlsRole::passive: return "passive";
case DtlsRole::actpass : return "actpass"; case DtlsRole::actpass: return "actpass";
default: return "invalid"; default: return "invalid";
} }
} }
@ -159,12 +157,12 @@ RtpDirection getRtpDirection(const string &str) {
return it == direction_map.end() ? RtpDirection::invalid : it->second; return it == direction_map.end() ? RtpDirection::invalid : it->second;
} }
const char* getRtpDirectionString(RtpDirection val){ const char *getRtpDirectionString(RtpDirection val) {
switch (val) { switch (val) {
case RtpDirection::sendonly : return "sendonly"; case RtpDirection::sendonly: return "sendonly";
case RtpDirection::recvonly : return "recvonly"; case RtpDirection::recvonly: return "recvonly";
case RtpDirection::sendrecv : return "sendrecv"; case RtpDirection::sendrecv: return "sendrecv";
case RtpDirection::inactive : return "inactive"; case RtpDirection::inactive: return "inactive";
default: return "invalid"; default: return "invalid";
} }
} }
@ -179,7 +177,7 @@ string RtcSdpBase::toString() const {
return std::move(printer); return std::move(printer);
} }
RtpDirection RtcSdpBase::getDirection() const{ RtpDirection RtcSdpBase::getDirection() const {
for (auto &item : items) { for (auto &item : items) {
auto attr = dynamic_pointer_cast<SdpAttr>(item); auto attr = dynamic_pointer_cast<SdpAttr>(item);
if (attr) { if (attr) {
@ -200,7 +198,7 @@ SdpItem::Ptr RtcSdpBase::getItem(char key_c, const char *attr_key) const {
return item; return item;
} }
auto attr = dynamic_pointer_cast<SdpAttr>(item); auto attr = dynamic_pointer_cast<SdpAttr>(item);
if (attr && !strcasecmp(attr->detail->getKey() , attr_key)) { if (attr && !strcasecmp(attr->detail->getKey(), attr_key)) {
return attr->detail; return attr->detail;
} }
} }
@ -225,7 +223,7 @@ string RtcSessionSdp::getSessionInfo() const {
return getStringItem('i'); return getStringItem('i');
} }
SdpTime RtcSessionSdp::getSessionTime() const{ SdpTime RtcSessionSdp::getSessionTime() const {
return getItemClass<SdpTime>('t'); return getItemClass<SdpTime>('t');
} }
@ -475,7 +473,7 @@ string SdpAttrRtcp::toString() const {
return SdpItem::toString(); return SdpItem::toString();
} }
void SdpAttrIceOption::parse(const string &str){ void SdpAttrIceOption::parse(const string &str) {
auto vec = split(str, " "); auto vec = split(str, " ");
for (auto &v : vec) { for (auto &v : vec) {
if (!strcasecmp(v.data(), "trickle")) { if (!strcasecmp(v.data(), "trickle")) {
@ -489,7 +487,7 @@ void SdpAttrIceOption::parse(const string &str){
} }
} }
string SdpAttrIceOption::toString() const{ string SdpAttrIceOption::toString() const {
if (value.empty()) { if (value.empty()) {
if (trickle && renomination) { if (trickle && renomination) {
value = "trickle renomination"; value = "trickle renomination";
@ -529,8 +527,8 @@ string SdpAttrSetup::toString() const {
} }
void SdpAttrExtmap::parse(const string &str) { void SdpAttrExtmap::parse(const string &str) {
char buf[128] = {0}; char buf[128] = { 0 };
char direction_buf[32] = {0}; char direction_buf[32] = { 0 };
if (sscanf(str.data(), "%" SCNd8 "/%31[^ ] %127s", &id, direction_buf, buf) != 3) { if (sscanf(str.data(), "%" SCNd8 "/%31[^ ] %127s", &id, direction_buf, buf) != 3) {
CHECK_SDP(sscanf(str.data(), "%" SCNd8 " %127s", &id, buf) == 2); CHECK_SDP(sscanf(str.data(), "%" SCNd8 " %127s", &id, buf) == 2);
direction = RtpDirection::sendrecv; direction = RtpDirection::sendrecv;
@ -542,7 +540,7 @@ void SdpAttrExtmap::parse(const string &str) {
string SdpAttrExtmap::toString() const { string SdpAttrExtmap::toString() const {
if (value.empty()) { if (value.empty()) {
if(direction == RtpDirection::invalid || direction == RtpDirection::sendrecv){ if (direction == RtpDirection::invalid || direction == RtpDirection::sendrecv) {
value = to_string((int)id) + " " + ext; value = to_string((int)id) + " " + ext;
} else { } else {
value = to_string((int)id) + "/" + getRtpDirectionString(direction) + " " + ext; value = to_string((int)id) + "/" + getRtpDirectionString(direction) + " " + ext;
@ -552,11 +550,11 @@ string SdpAttrExtmap::toString() const {
} }
void SdpAttrRtpMap::parse(const string &str) { void SdpAttrRtpMap::parse(const string &str) {
char buf[32] = {0}; char buf[32] = { 0 };
if (sscanf(str.data(), "%" SCNu8 " %31[^/]/%" SCNd32 "/%" SCNd32, &pt, buf, &sample_rate, &channel) != 4) { if (sscanf(str.data(), "%" SCNu8 " %31[^/]/%" SCNd32 "/%" SCNd32, &pt, buf, &sample_rate, &channel) != 4) {
CHECK_SDP(sscanf(str.data(), "%" SCNu8 " %31[^/]/%" SCNd32, &pt, buf, &sample_rate) == 3); CHECK_SDP(sscanf(str.data(), "%" SCNu8 " %31[^/]/%" SCNd32, &pt, buf, &sample_rate) == 3);
if (getTrackType(getCodecId(buf)) == TrackAudio) { if (getTrackType(getCodecId(buf)) == TrackAudio) {
//未指定通道数时且为音频时那么通道数默认为1 // 未指定通道数时且为音频时那么通道数默认为1
channel = 1; channel = 1;
} }
} }
@ -576,7 +574,7 @@ string SdpAttrRtpMap::toString() const {
void SdpAttrRtcpFb::parse(const string &str_in) { void SdpAttrRtcpFb::parse(const string &str_in) {
auto str = str_in + "\n"; auto str = str_in + "\n";
char rtcp_type_buf[32] = {0}; char rtcp_type_buf[32] = { 0 };
CHECK_SDP(sscanf(str.data(), "%" SCNu8 " %31[^\n]", &pt, rtcp_type_buf) == 2); CHECK_SDP(sscanf(str.data(), "%" SCNu8 " %31[^\n]", &pt, rtcp_type_buf) == 2);
rtcp_type = rtcp_type_buf; rtcp_type = rtcp_type_buf;
} }
@ -596,7 +594,7 @@ void SdpAttrFmtp::parse(const string &str) {
for (auto &item : vec) { for (auto &item : vec) {
trim(item); trim(item);
auto pos = item.find('='); auto pos = item.find('=');
if(pos == string::npos){ if (pos == string::npos) {
fmtp.emplace(std::make_pair(item, "")); fmtp.emplace(std::make_pair(item, ""));
} else { } else {
fmtp.emplace(std::make_pair(item.substr(0, pos), item.substr(pos + 1))); fmtp.emplace(std::make_pair(item.substr(0, pos), item.substr(pos + 1)));
@ -619,8 +617,8 @@ string SdpAttrFmtp::toString() const {
void SdpAttrSSRC::parse(const string &str_in) { void SdpAttrSSRC::parse(const string &str_in) {
auto str = str_in + '\n'; auto str = str_in + '\n';
char attr_buf[32] = {0}; char attr_buf[32] = { 0 };
char attr_val_buf[128] = {0}; char attr_val_buf[128] = { 0 };
if (3 == sscanf(str.data(), "%" SCNu32 " %31[^:]:%127[^\n]", &ssrc, attr_buf, attr_val_buf)) { if (3 == sscanf(str.data(), "%" SCNu32 " %31[^:]:%127[^\n]", &ssrc, attr_buf, attr_val_buf)) {
attribute = attr_buf; attribute = attr_buf;
attribute_value = attr_val_buf; attribute_value = attr_val_buf;
@ -650,14 +648,14 @@ void SdpAttrSSRCGroup::parse(const string &str) {
CHECK(isFID() || isSIM()); CHECK(isFID() || isSIM());
vec.erase(vec.begin()); vec.erase(vec.begin());
for (auto ssrc : vec) { for (auto ssrc : vec) {
ssrcs.emplace_back((uint32_t) atoll(ssrc.data())); ssrcs.emplace_back((uint32_t)atoll(ssrc.data()));
} }
} }
string SdpAttrSSRCGroup::toString() const { string SdpAttrSSRCGroup::toString() const {
if (value.empty()) { if (value.empty()) {
value = type; value = type;
//最少要求2个ssrc // 最少要求2个ssrc
CHECK(ssrcs.size() >= 2); CHECK(ssrcs.size() >= 2);
for (auto &ssrc : ssrcs) { for (auto &ssrc : ssrcs) {
value += ' '; value += ' ';
@ -668,7 +666,7 @@ string SdpAttrSSRCGroup::toString() const {
} }
void SdpAttrSctpMap::parse(const string &str) { void SdpAttrSctpMap::parse(const string &str) {
char subtypes_buf[64] = {0}; char subtypes_buf[64] = { 0 };
CHECK_SDP(3 == sscanf(str.data(), "%" SCNu16 " %63[^ ] %" SCNd32, &port, subtypes_buf, &streams)); CHECK_SDP(3 == sscanf(str.data(), "%" SCNu16 " %63[^ ] %" SCNd32, &port, subtypes_buf, &streams));
subtypes = subtypes_buf; subtypes = subtypes_buf;
} }
@ -685,10 +683,10 @@ string SdpAttrSctpMap::toString() const {
} }
void SdpAttrCandidate::parse(const string &str) { void SdpAttrCandidate::parse(const string &str) {
char foundation_buf[40] = {0}; char foundation_buf[40] = { 0 };
char transport_buf[16] = {0}; char transport_buf[16] = { 0 };
char address_buf[64] = {0}; char address_buf[64] = { 0 };
char type_buf[16] = {0}; char type_buf[16] = { 0 };
// https://datatracker.ietf.org/doc/html/rfc5245#section-15.1 // https://datatracker.ietf.org/doc/html/rfc5245#section-15.1
CHECK_SDP(sscanf(str.data(), "%32[^ ] %" SCNu32 " %15[^ ] %" SCNu32 " %63[^ ] %" SCNu16 " typ %15[^ ]", CHECK_SDP(sscanf(str.data(), "%32[^ ] %" SCNu32 " %15[^ ] %" SCNu32 " %63[^ ] %" SCNu16 " typ %15[^ ]",
@ -717,8 +715,7 @@ void SdpAttrCandidate::parse(const string &str) {
string SdpAttrCandidate::toString() const { string SdpAttrCandidate::toString() const {
if (value.empty()) { if (value.empty()) {
value = foundation + " " + to_string(component) + " " + transport + " " + to_string(priority) + value = foundation + " " + to_string(component) + " " + transport + " " + to_string(priority) + " " + address + " " + to_string(port) + " typ " + type;
" " + address + " " + to_string(port) + " typ " + type;
for (auto &pr : arr) { for (auto &pr : arr) {
value += ' '; value += ' ';
value += pr.first; value += pr.first;
@ -730,10 +727,10 @@ string SdpAttrCandidate::toString() const {
} }
void SdpAttrSimulcast::parse(const string &str) { void SdpAttrSimulcast::parse(const string &str) {
//https://www.meetecho.com/blog/simulcast-janus-ssrc/ // https://www.meetecho.com/blog/simulcast-janus-ssrc/
//a=simulcast:send/recv q;h;f // a=simulcast:send/recv q;h;f
//a=simulcast:send/recv [rid=]q;h;f // a=simulcast:send/recv [rid=]q;h;f
//a=simulcast: recv h;m;l // a=simulcast: recv h;m;l
// //
auto vec = split(str, " "); auto vec = split(str, " ");
CHECK_SDP(vec.size() == 2); CHECK_SDP(vec.size() == 2);
@ -845,12 +842,12 @@ void RtcSession::loadFrom(const string &str) {
for (auto &group : ssrc_groups) { for (auto &group : ssrc_groups) {
if (group.isFID()) { if (group.isFID()) {
have_rtx_ssrc = true; have_rtx_ssrc = true;
//ssrc-group:FID字段必须包含rtp/rtx的ssrc // ssrc-group:FID字段必须包含rtp/rtx的ssrc
CHECK(group.ssrcs.size() == 2); CHECK(group.ssrcs.size() == 2);
//根据rtp ssrc找到对象 // 根据rtp ssrc找到对象
auto it = rtc_ssrc_map.find(group.ssrcs[0]); auto it = rtc_ssrc_map.find(group.ssrcs[0]);
CHECK(it != rtc_ssrc_map.end()); CHECK(it != rtc_ssrc_map.end());
//设置rtx ssrc // 设置rtx ssrc
it->second.rtx_ssrc = group.ssrcs[1]; it->second.rtx_ssrc = group.ssrcs[1];
rtc_media.rtp_rtx_ssrc.emplace_back(it->second); rtc_media.rtp_rtx_ssrc.emplace_back(it->second);
} else if (group.isSIM()) { } else if (group.isSIM()) {
@ -860,7 +857,7 @@ void RtcSession::loadFrom(const string &str) {
} }
if (!have_rtx_ssrc) { if (!have_rtx_ssrc) {
//按照sdp顺序依次添加ssrc // 按照sdp顺序依次添加ssrc
for (auto &attr : ssrc_attr) { for (auto &attr : ssrc_attr) {
if (attr.attribute == "cname") { if (attr.attribute == "cname") {
rtc_media.rtp_rtx_ssrc.emplace_back(rtc_ssrc_map[attr.ssrc]); rtc_media.rtp_rtx_ssrc.emplace_back(rtc_ssrc_map[attr.ssrc]);
@ -883,56 +880,56 @@ void RtcSession::loadFrom(const string &str) {
CHECK(rid.direction == simulcast.direction); CHECK(rid.direction == simulcast.direction);
CHECK(rid_map.find(rid.rid) != rid_map.end()); CHECK(rid_map.find(rid.rid) != rid_map.end());
} }
//simulcast最少要求2种方案 // simulcast最少要求2种方案
CHECK(simulcast.rids.size() >= 2); CHECK(simulcast.rids.size() >= 2);
rtc_media.rtp_rids = simulcast.rids; rtc_media.rtp_rids = simulcast.rids;
} }
if (ssrc_group_sim) { if (ssrc_group_sim) {
//指定了a=ssrc-group:SIM // 指定了a=ssrc-group:SIM
for (auto ssrc : ssrc_group_sim->ssrcs) { for (auto ssrc : ssrc_group_sim->ssrcs) {
auto it = rtc_ssrc_map.find(ssrc); auto it = rtc_ssrc_map.find(ssrc);
CHECK(it != rtc_ssrc_map.end()); CHECK(it != rtc_ssrc_map.end());
rtc_media.rtp_ssrc_sim.emplace_back(it->second); rtc_media.rtp_ssrc_sim.emplace_back(it->second);
} }
} else if (!rtc_media.rtp_rids.empty()) { } else if (!rtc_media.rtp_rids.empty()) {
//未指定a=ssrc-group:SIM, 但是指定了a=simulcast, 那么只能根据ssrc顺序来对应rid顺序 // 未指定a=ssrc-group:SIM, 但是指定了a=simulcast, 那么只能根据ssrc顺序来对应rid顺序
rtc_media.rtp_ssrc_sim = rtc_media.rtp_rtx_ssrc; rtc_media.rtp_ssrc_sim = rtc_media.rtp_rtx_ssrc;
} }
if (!rtc_media.supportSimulcast()) { if (!rtc_media.supportSimulcast()) {
//不支持simulcast的情况下最多一组ssrc // 不支持simulcast的情况下最多一组ssrc
CHECK(rtc_media.rtp_rtx_ssrc.size() <= 1); CHECK(rtc_media.rtp_rtx_ssrc.size() <= 1);
} else { } else {
//simulcast的情况下要么没有指定ssrc要么指定的ssrc个数与rid个数一致 // simulcast的情况下要么没有指定ssrc要么指定的ssrc个数与rid个数一致
//CHECK(rtc_media.rtp_ssrc_sim.empty() || rtc_media.rtp_ssrc_sim.size() == rtc_media.rtp_rids.size()); // CHECK(rtc_media.rtp_ssrc_sim.empty() || rtc_media.rtp_ssrc_sim.size() == rtc_media.rtp_rids.size());
} }
auto rtpmap_arr = media.getAllItem<SdpAttrRtpMap>('a', "rtpmap"); auto rtpmap_arr = media.getAllItem<SdpAttrRtpMap>('a', "rtpmap");
auto rtcpfb_arr = media.getAllItem<SdpAttrRtcpFb>('a', "rtcp-fb"); auto rtcpfb_arr = media.getAllItem<SdpAttrRtcpFb>('a', "rtcp-fb");
auto fmtp_aar = media.getAllItem<SdpAttrFmtp>('a', "fmtp"); auto fmtp_aar = media.getAllItem<SdpAttrFmtp>('a', "fmtp");
//方便根据pt查找rtpmap,一个pt必有一条 // 方便根据pt查找rtpmap,一个pt必有一条
map<uint8_t, SdpAttrRtpMap &> rtpmap_map; map<uint8_t, SdpAttrRtpMap &> rtpmap_map;
//方便根据pt查找rtcp-fb,一个pt可能有多条或0条 // 方便根据pt查找rtcp-fb,一个pt可能有多条或0条
multimap<uint8_t, SdpAttrRtcpFb &> rtcpfb_map; multimap<uint8_t, SdpAttrRtcpFb &> rtcpfb_map;
//方便根据pt查找fmtp一个pt最多一条 // 方便根据pt查找fmtp一个pt最多一条
map<uint8_t, SdpAttrFmtp &> fmtp_map; map<uint8_t, SdpAttrFmtp &> fmtp_map;
for (auto &rtpmap : rtpmap_arr) { for (auto &rtpmap : rtpmap_arr) {
//添加失败,有多条 // 添加失败,有多条
CHECK(rtpmap_map.emplace(rtpmap.pt, rtpmap).second, "该pt存在多条a=rtpmap:", (int)rtpmap.pt); CHECK(rtpmap_map.emplace(rtpmap.pt, rtpmap).second, "该pt存在多条a=rtpmap:", (int)rtpmap.pt);
} }
for (auto &rtpfb : rtcpfb_arr) { for (auto &rtpfb : rtcpfb_arr) {
rtcpfb_map.emplace(rtpfb.pt, rtpfb); rtcpfb_map.emplace(rtpfb.pt, rtpfb);
} }
for (auto &fmtp : fmtp_aar) { for (auto &fmtp : fmtp_aar) {
//添加失败,有多条 // 添加失败,有多条
CHECK(fmtp_map.emplace(fmtp.pt, fmtp).second, "该pt存在多条a=fmtp:", (int)fmtp.pt); CHECK(fmtp_map.emplace(fmtp.pt, fmtp).second, "该pt存在多条a=fmtp:", (int)fmtp.pt);
} }
for (auto &item : mline.fmts) { for (auto &item : mline.fmts) {
auto pt = atoi(item.c_str()); auto pt = atoi(item.c_str());
CHECK(pt < 0xFF, "invalid payload type: ", item); CHECK(pt < 0xFF, "invalid payload type: ", item);
//遍历所有编码方案的pt // 遍历所有编码方案的pt
rtc_media.plan.emplace_back(); rtc_media.plan.emplace_back();
auto &plan = rtc_media.plan.back(); auto &plan = rtc_media.plan.back();
auto rtpmap_it = rtpmap_map.find(pt); auto rtpmap_it = rtpmap_map.find(pt);
@ -952,8 +949,7 @@ void RtcSession::loadFrom(const string &str) {
if (fmtp_it != fmtp_map.end()) { if (fmtp_it != fmtp_map.end()) {
plan.fmtp = fmtp_it->second.fmtp; plan.fmtp = fmtp_it->second.fmtp;
} }
for (auto rtpfb_it = rtcpfb_map.find(pt); for (auto rtpfb_it = rtcpfb_map.find(pt); rtpfb_it != rtcpfb_map.end() && rtpfb_it->second.pt == pt; ++rtpfb_it) {
rtpfb_it != rtcpfb_map.end() && rtpfb_it->second.pt == pt; ++rtpfb_it) {
plan.rtcp_fb.emplace(rtpfb_it->second.rtcp_type); plan.rtcp_fb.emplace(rtpfb_it->second.rtcp_type);
} }
} }
@ -971,7 +967,7 @@ void RtcSdpBase::toRtsp() {
case 'i': case 'i':
case 't': case 't':
case 'c': case 'c':
case 'b':{ case 'b': {
++it; ++it;
break; break;
} }
@ -986,8 +982,7 @@ void RtcSdpBase::toRtsp() {
case 'a': { case 'a': {
auto attr = dynamic_pointer_cast<SdpAttr>(*it); auto attr = dynamic_pointer_cast<SdpAttr>(*it);
CHECK(attr); CHECK(attr);
if (!strcasecmp(attr->detail->getKey(), "rtpmap") if (!strcasecmp(attr->detail->getKey(), "rtpmap") || !strcasecmp(attr->detail->getKey(), "fmtp")) {
|| !strcasecmp(attr->detail->getKey(), "fmtp")) {
++it; ++it;
break; break;
} }
@ -1000,7 +995,7 @@ void RtcSdpBase::toRtsp() {
} }
} }
string RtcSession::toRtspSdp() const{ string RtcSession::toRtspSdp() const {
RtcSession copy = *this; RtcSession copy = *this;
copy.media.clear(); copy.media.clear();
for (auto &m : media) { for (auto &m : media) {
@ -1056,17 +1051,17 @@ void addSdpAttrSSRC(const RtcSSRC &rtp_ssrc, RtcSdpBase &media, uint32_t ssrc_nu
} }
} }
RtcSessionSdp::Ptr RtcSession::toRtcSessionSdp() const{ RtcSessionSdp::Ptr RtcSession::toRtcSessionSdp() const {
RtcSessionSdp::Ptr ret = std::make_shared<RtcSessionSdp>(); RtcSessionSdp::Ptr ret = std::make_shared<RtcSessionSdp>();
auto &sdp = *ret; auto &sdp = *ret;
sdp.addItem(std::make_shared<SdpString<'v'> >(to_string(version))); sdp.addItem(std::make_shared<SdpString<'v'>>(to_string(version)));
sdp.addItem(std::make_shared<SdpOrigin>(origin)); sdp.addItem(std::make_shared<SdpOrigin>(origin));
sdp.addItem(std::make_shared<SdpString<'s'> >(session_name)); sdp.addItem(std::make_shared<SdpString<'s'>>(session_name));
if (!session_info.empty()) { if (!session_info.empty()) {
sdp.addItem(std::make_shared<SdpString<'i'> >(session_info)); sdp.addItem(std::make_shared<SdpString<'i'>>(session_info));
} }
sdp.addItem(std::make_shared<SdpTime>(time)); sdp.addItem(std::make_shared<SdpTime>(time));
if(connection.empty()){ if (connection.empty()) {
sdp.addItem(std::make_shared<SdpConnection>(connection)); sdp.addItem(std::make_shared<SdpConnection>(connection));
} }
sdp.addAttr(std::make_shared<SdpAttrGroup>(group)); sdp.addAttr(std::make_shared<SdpAttrGroup>(group));
@ -1124,21 +1119,21 @@ RtcSessionSdp::Ptr RtcSession::toRtcSessionSdp() const{
sdp_media.addAttr(std::make_shared<SdpCommon>("rtcp-rsize")); sdp_media.addAttr(std::make_shared<SdpCommon>("rtcp-rsize"));
} }
if(m.type != TrackApplication) { if (m.type != TrackApplication) {
for (auto &p : m.plan) { for (auto &p : m.plan) {
auto rtp_map = std::make_shared<SdpAttrRtpMap>(); auto rtp_map = std::make_shared<SdpAttrRtpMap>();
rtp_map->pt = p.pt; rtp_map->pt = p.pt;
rtp_map->codec = p.codec; rtp_map->codec = p.codec;
rtp_map->sample_rate = p.sample_rate; rtp_map->sample_rate = p.sample_rate;
rtp_map->channel = p.channel; rtp_map->channel = p.channel;
//添加a=rtpmap // 添加a=rtpmap
sdp_media.addAttr(std::move(rtp_map)); sdp_media.addAttr(std::move(rtp_map));
for (auto &fb : p.rtcp_fb) { for (auto &fb : p.rtcp_fb) {
auto rtcp_fb = std::make_shared<SdpAttrRtcpFb>(); auto rtcp_fb = std::make_shared<SdpAttrRtcpFb>();
rtcp_fb->pt = p.pt; rtcp_fb->pt = p.pt;
rtcp_fb->rtcp_type = fb; rtcp_fb->rtcp_type = fb;
//添加a=rtcp-fb // 添加a=rtcp-fb
sdp_media.addAttr(std::move(rtcp_fb)); sdp_media.addAttr(std::move(rtcp_fb));
} }
@ -1146,13 +1141,13 @@ RtcSessionSdp::Ptr RtcSession::toRtcSessionSdp() const{
auto fmtp = std::make_shared<SdpAttrFmtp>(); auto fmtp = std::make_shared<SdpAttrFmtp>();
fmtp->pt = p.pt; fmtp->pt = p.pt;
fmtp->fmtp = p.fmtp; fmtp->fmtp = p.fmtp;
//添加a=fmtp // 添加a=fmtp
sdp_media.addAttr(std::move(fmtp)); sdp_media.addAttr(std::move(fmtp));
} }
} }
{ {
//添加a=msid字段 // 添加a=msid字段
if (!m.rtp_rtx_ssrc.empty()) { if (!m.rtp_rtx_ssrc.empty()) {
if (!m.rtp_rtx_ssrc[0].msid.empty()) { if (!m.rtp_rtx_ssrc[0].msid.empty()) {
auto msid = std::make_shared<SdpAttrMsid>(); auto msid = std::make_shared<SdpAttrMsid>();
@ -1164,14 +1159,14 @@ RtcSessionSdp::Ptr RtcSession::toRtcSessionSdp() const{
{ {
for (auto &ssrc : m.rtp_rtx_ssrc) { for (auto &ssrc : m.rtp_rtx_ssrc) {
//添加a=ssrc字段 // 添加a=ssrc字段
CHECK(!ssrc.empty()); CHECK(!ssrc.empty());
addSdpAttrSSRC(ssrc, sdp_media, ssrc.ssrc); addSdpAttrSSRC(ssrc, sdp_media, ssrc.ssrc);
if (ssrc.rtx_ssrc) { if (ssrc.rtx_ssrc) {
addSdpAttrSSRC(ssrc, sdp_media, ssrc.rtx_ssrc); addSdpAttrSSRC(ssrc, sdp_media, ssrc.rtx_ssrc);
//生成a=ssrc-group:FID字段 // 生成a=ssrc-group:FID字段
//有rtx ssrc // 有rtx ssrc
auto group = std::make_shared<SdpAttrSSRCGroup>(); auto group = std::make_shared<SdpAttrSSRCGroup>();
group->type = "FID"; group->type = "FID";
group->ssrcs.emplace_back(ssrc.ssrc); group->ssrcs.emplace_back(ssrc.ssrc);
@ -1183,12 +1178,12 @@ RtcSessionSdp::Ptr RtcSession::toRtcSessionSdp() const{
{ {
if (m.rtp_ssrc_sim.size() >= 2) { if (m.rtp_ssrc_sim.size() >= 2) {
//simulcast 要求 2~3路 // simulcast 要求 2~3路
auto group = std::make_shared<SdpAttrSSRCGroup>(); auto group = std::make_shared<SdpAttrSSRCGroup>();
for (auto &ssrc : m.rtp_ssrc_sim) { for (auto &ssrc : m.rtp_ssrc_sim) {
group->ssrcs.emplace_back(ssrc.ssrc); group->ssrcs.emplace_back(ssrc.ssrc);
} }
//添加a=ssrc-group:SIM字段 // 添加a=ssrc-group:SIM字段
group->type = "SIM"; group->type = "SIM";
sdp_media.addAttr(std::move(group)); sdp_media.addAttr(std::move(group));
} }
@ -1221,17 +1216,17 @@ RtcSessionSdp::Ptr RtcSession::toRtcSessionSdp() const{
} }
} }
} }
if(ice_lite) if (ice_lite) {
sdp.addAttr(std::make_shared<SdpCommon>("ice-lite")); sdp.addAttr(std::make_shared<SdpCommon>("ice-lite"));
}
return ret; return ret;
} }
string RtcSession::toString() const{ string RtcSession::toString() const {
return toRtcSessionSdp()->toString(); return toRtcSessionSdp()->toString();
} }
string RtcCodecPlan::getFmtp(const char *key) const{ string RtcCodecPlan::getFmtp(const char *key) const {
for (auto &item : fmtp) { for (auto &item : fmtp) {
if (strcasecmp(item.first.data(), key) == 0) { if (strcasecmp(item.first.data(), key) == 0) {
return item.second; return item.second;
@ -1240,7 +1235,7 @@ string RtcCodecPlan::getFmtp(const char *key) const{
return ""; return "";
} }
const RtcCodecPlan *RtcMedia::getPlan(uint8_t pt) const{ const RtcCodecPlan *RtcMedia::getPlan(uint8_t pt) const {
for (auto &item : plan) { for (auto &item : plan) {
if (item.pt == pt) { if (item.pt == pt) {
return &item; return &item;
@ -1249,7 +1244,7 @@ const RtcCodecPlan *RtcMedia::getPlan(uint8_t pt) const{
return nullptr; return nullptr;
} }
const RtcCodecPlan *RtcMedia::getPlan(const char *codec) const{ const RtcCodecPlan *RtcMedia::getPlan(const char *codec) const {
for (auto &item : plan) { for (auto &item : plan) {
if (strcasecmp(item.codec.data(), codec) == 0) { if (strcasecmp(item.codec.data(), codec) == 0) {
return &item; return &item;
@ -1258,7 +1253,7 @@ const RtcCodecPlan *RtcMedia::getPlan(const char *codec) const{
return nullptr; return nullptr;
} }
const RtcCodecPlan *RtcMedia::getRelatedRtxPlan(uint8_t pt) const{ const RtcCodecPlan *RtcMedia::getRelatedRtxPlan(uint8_t pt) const {
for (auto &item : plan) { for (auto &item : plan) {
if (strcasecmp(item.codec.data(), "rtx") == 0) { if (strcasecmp(item.codec.data(), "rtx") == 0) {
auto apt = atoi(item.getFmtp("apt").data()); auto apt = atoi(item.getFmtp("apt").data());
@ -1294,7 +1289,7 @@ bool RtcMedia::supportSimulcast() const {
return false; return false;
} }
void RtcMedia::checkValid() const{ void RtcMedia::checkValid() const {
CHECK(type != TrackInvalid); CHECK(type != TrackInvalid);
CHECK(!mid.empty()); CHECK(!mid.empty());
CHECK(!proto.empty()); CHECK(!proto.empty());
@ -1304,7 +1299,7 @@ void RtcMedia::checkValid() const{
bool send_rtp = (direction == RtpDirection::sendonly || direction == RtpDirection::sendrecv); bool send_rtp = (direction == RtpDirection::sendonly || direction == RtpDirection::sendrecv);
if (!supportSimulcast()) { if (!supportSimulcast()) {
//非simulcast时检查有没有指定rtp ssrc // 非simulcast时检查有没有指定rtp ssrc
CHECK(!rtp_rtx_ssrc.empty() || !send_rtp); CHECK(!rtp_rtx_ssrc.empty() || !send_rtp);
} }
@ -1318,7 +1313,7 @@ void RtcMedia::checkValid() const{
#endif #endif
} }
void RtcSession::checkValid() const{ void RtcSession::checkValid() const {
CHECK(version == 0); CHECK(version == 0);
CHECK(!origin.empty()); CHECK(!origin.empty());
CHECK(!session_name.empty()); CHECK(!session_name.empty());
@ -1337,15 +1332,15 @@ void RtcSession::checkValid() const{
case RtpDirection::sendrecv: case RtpDirection::sendrecv:
case RtpDirection::sendonly: case RtpDirection::sendonly:
case RtpDirection::recvonly: have_active_media = true; break; case RtpDirection::recvonly: have_active_media = true; break;
default : break; default: break;
} }
} }
CHECK(have_active_media, "必须确保最少有一个活跃的track"); CHECK(have_active_media, "必须确保最少有一个活跃的track");
} }
const RtcMedia *RtcSession::getMedia(TrackType type) const{ const RtcMedia *RtcSession::getMedia(TrackType type) const {
for(auto &m : media){ for (auto &m : media) {
if(m.type == type){ if (m.type == type) {
return &m; return &m;
} }
} }
@ -1377,7 +1372,7 @@ bool RtcSession::isOnlyDatachannel() const {
string const SdpConst::kTWCCRtcpFb = "transport-cc"; string const SdpConst::kTWCCRtcpFb = "transport-cc";
string const SdpConst::kRembRtcpFb = "goog-remb"; string const SdpConst::kRembRtcpFb = "goog-remb";
void RtcConfigure::RtcTrackConfigure::enableTWCC(bool enable){ void RtcConfigure::RtcTrackConfigure::enableTWCC(bool enable) {
if (!enable) { if (!enable) {
rtcp_fb.erase(SdpConst::kTWCCRtcpFb); rtcp_fb.erase(SdpConst::kTWCCRtcpFb);
extmap.erase(RtpExtType::transport_cc); extmap.erase(RtpExtType::transport_cc);
@ -1387,7 +1382,7 @@ void RtcConfigure::RtcTrackConfigure::enableTWCC(bool enable){
} }
} }
void RtcConfigure::RtcTrackConfigure::enableREMB(bool enable){ void RtcConfigure::RtcTrackConfigure::enableREMB(bool enable) {
if (!enable) { if (!enable) {
rtcp_fb.erase(SdpConst::kRembRtcpFb); rtcp_fb.erase(SdpConst::kRembRtcpFb);
extmap.erase(RtpExtType::abs_send_time); extmap.erase(RtpExtType::abs_send_time);
@ -1397,7 +1392,7 @@ void RtcConfigure::RtcTrackConfigure::enableREMB(bool enable){
} }
} }
static vector<CodecId> toCodecArray(const string &str){ static vector<CodecId> toCodecArray(const string &str) {
vector<CodecId> ret; vector<CodecId> ret;
auto vec = split(str, ","); auto vec = split(str, ",");
for (auto &s : vec) { for (auto &s : vec) {
@ -1409,7 +1404,7 @@ static vector<CodecId> toCodecArray(const string &str){
return ret; return ret;
} }
void RtcConfigure::RtcTrackConfigure::setDefaultSetting(TrackType type){ void RtcConfigure::RtcTrackConfigure::setDefaultSetting(TrackType type) {
rtcp_mux = true; rtcp_mux = true;
rtcp_rsize = false; rtcp_rsize = false;
group_bundle = true; group_bundle = true;
@ -1421,47 +1416,43 @@ void RtcConfigure::RtcTrackConfigure::setDefaultSetting(TrackType type){
ice_renomination = false; ice_renomination = false;
switch (type) { switch (type) {
case TrackAudio: { case TrackAudio: {
//此处调整偏好的编码格式优先级 // 此处调整偏好的编码格式优先级
GET_CONFIG_FUNC(vector<CodecId>, s_preferred_codec, Rtc::kPreferredCodecA, toCodecArray); GET_CONFIG_FUNC(vector<CodecId>, s_preferred_codec, Rtc::kPreferredCodecA, toCodecArray);
CHECK(!s_preferred_codec.empty(), "rtc音频偏好codec不能为空"); CHECK(!s_preferred_codec.empty(), "rtc音频偏好codec不能为空");
preferred_codec = s_preferred_codec; preferred_codec = s_preferred_codec;
rtcp_fb = {SdpConst::kTWCCRtcpFb, SdpConst::kRembRtcpFb}; rtcp_fb = { SdpConst::kTWCCRtcpFb, SdpConst::kRembRtcpFb };
extmap = { extmap = { { RtpExtType::ssrc_audio_level, RtpDirection::sendrecv },
{RtpExtType::ssrc_audio_level, RtpDirection::sendrecv}, { RtpExtType::csrc_audio_level, RtpDirection::sendrecv },
{RtpExtType::csrc_audio_level, RtpDirection::sendrecv}, { RtpExtType::abs_send_time, RtpDirection::sendrecv },
{RtpExtType::abs_send_time, RtpDirection::sendrecv}, { RtpExtType::transport_cc, RtpDirection::sendrecv },
{RtpExtType::transport_cc, RtpDirection::sendrecv}, // rtx重传rtp时忽略sdes_mid类型的rtp ext,实测发现Firefox在接收rtx时如果存在sdes_mid的ext,将导致无法播放
//rtx重传rtp时忽略sdes_mid类型的rtp ext,实测发现Firefox在接收rtx时如果存在sdes_mid的ext,将导致无法播放
//{RtpExtType::sdes_mid,RtpDirection::sendrecv}, //{RtpExtType::sdes_mid,RtpDirection::sendrecv},
{RtpExtType::sdes_rtp_stream_id, RtpDirection::sendrecv}, { RtpExtType::sdes_rtp_stream_id, RtpDirection::sendrecv },
{RtpExtType::sdes_repaired_rtp_stream_id, RtpDirection::sendrecv} { RtpExtType::sdes_repaired_rtp_stream_id, RtpDirection::sendrecv } };
};
break; break;
} }
case TrackVideo: { case TrackVideo: {
//此处调整偏好的编码格式优先级 // 此处调整偏好的编码格式优先级
GET_CONFIG_FUNC(vector<CodecId>, s_preferred_codec, Rtc::kPreferredCodecV, toCodecArray); GET_CONFIG_FUNC(vector<CodecId>, s_preferred_codec, Rtc::kPreferredCodecV, toCodecArray);
CHECK(!s_preferred_codec.empty(), "rtc视频偏好codec不能为空"); CHECK(!s_preferred_codec.empty(), "rtc视频偏好codec不能为空");
preferred_codec = s_preferred_codec; preferred_codec = s_preferred_codec;
rtcp_fb = {SdpConst::kTWCCRtcpFb, SdpConst::kRembRtcpFb, "nack", "ccm fir", "nack pli"}; rtcp_fb = { SdpConst::kTWCCRtcpFb, SdpConst::kRembRtcpFb, "nack", "ccm fir", "nack pli" };
extmap = { extmap = { { RtpExtType::abs_send_time, RtpDirection::sendrecv },
{RtpExtType::abs_send_time, RtpDirection::sendrecv}, { RtpExtType::transport_cc, RtpDirection::sendrecv },
{RtpExtType::transport_cc, RtpDirection::sendrecv}, // rtx重传rtp时忽略sdes_mid类型的rtp ext,实测发现Firefox在接收rtx时如果存在sdes_mid的ext,将导致无法播放
//rtx重传rtp时忽略sdes_mid类型的rtp ext,实测发现Firefox在接收rtx时如果存在sdes_mid的ext,将导致无法播放
//{RtpExtType::sdes_mid,RtpDirection::sendrecv}, //{RtpExtType::sdes_mid,RtpDirection::sendrecv},
{RtpExtType::sdes_rtp_stream_id, RtpDirection::sendrecv}, { RtpExtType::sdes_rtp_stream_id, RtpDirection::sendrecv },
{RtpExtType::sdes_repaired_rtp_stream_id, RtpDirection::sendrecv}, { RtpExtType::sdes_repaired_rtp_stream_id, RtpDirection::sendrecv },
{RtpExtType::video_timing, RtpDirection::sendrecv}, { RtpExtType::video_timing, RtpDirection::sendrecv },
{RtpExtType::color_space, RtpDirection::sendrecv}, { RtpExtType::color_space, RtpDirection::sendrecv },
{RtpExtType::video_content_type, RtpDirection::sendrecv}, { RtpExtType::video_content_type, RtpDirection::sendrecv },
{RtpExtType::playout_delay, RtpDirection::sendrecv}, { RtpExtType::playout_delay, RtpDirection::sendrecv },
//手机端推webrtc 会带有旋转角度rtc协议能正常播放 其他协议拉流画面旋转 // 手机端推webrtc 会带有旋转角度rtc协议能正常播放 其他协议拉流画面旋转
//{RtpExtType::video_orientation, RtpDirection::sendrecv}, //{RtpExtType::video_orientation, RtpDirection::sendrecv},
{RtpExtType::toffset, RtpDirection::sendrecv}, { RtpExtType::toffset, RtpDirection::sendrecv },
{RtpExtType::framemarking, RtpDirection::sendrecv} { RtpExtType::framemarking, RtpDirection::sendrecv } };
};
break; break;
} }
case TrackApplication: { case TrackApplication: {
@ -1471,8 +1462,7 @@ void RtcConfigure::RtcTrackConfigure::setDefaultSetting(TrackType type){
} }
} }
void RtcConfigure::setDefaultSetting(string ice_ufrag, string ice_pwd, RtpDirection direction, void RtcConfigure::setDefaultSetting(string ice_ufrag, string ice_pwd, RtpDirection direction, const SdpAttrFingerprint &fingerprint) {
const SdpAttrFingerprint &fingerprint) {
video.setDefaultSetting(TrackVideo); video.setDefaultSetting(TrackVideo);
audio.setDefaultSetting(TrackAudio); audio.setDefaultSetting(TrackAudio);
application.setDefaultSetting(TrackApplication); application.setDefaultSetting(TrackApplication);
@ -1512,7 +1502,7 @@ void RtcConfigure::addCandidate(const SdpAttrCandidate &candidate, TrackType typ
} }
} }
void RtcConfigure::enableTWCC(bool enable, TrackType type){ void RtcConfigure::enableTWCC(bool enable, TrackType type) {
switch (type) { switch (type) {
case TrackAudio: { case TrackAudio: {
audio.enableTWCC(enable); audio.enableTWCC(enable);
@ -1530,7 +1520,7 @@ void RtcConfigure::enableTWCC(bool enable, TrackType type){
} }
} }
void RtcConfigure::enableREMB(bool enable, TrackType type){ void RtcConfigure::enableREMB(bool enable, TrackType type) {
switch (type) { switch (type) {
case TrackAudio: { case TrackAudio: {
audio.enableREMB(enable); audio.enableREMB(enable);
@ -1559,10 +1549,10 @@ shared_ptr<RtcSession> RtcConfigure::createAnswer(const RtcSession &offer) const
matchMedia(ret, m); matchMedia(ret, m);
} }
//设置音视频端口复用 // 设置音视频端口复用
if (!offer.group.mids.empty()) { if (!offer.group.mids.empty()) {
for (auto &m : ret->media) { for (auto &m : ret->media) {
//The remote end has rejected (port 0) the m-section, so it should not be putting its mid in the group attribute. // The remote end has rejected (port 0) the m-section, so it should not be putting its mid in the group attribute.
if (m.port) { if (m.port) {
ret->group.mids.emplace_back(m.mid); ret->group.mids.emplace_back(m.mid);
} }
@ -1571,32 +1561,32 @@ shared_ptr<RtcSession> RtcConfigure::createAnswer(const RtcSession &offer) const
return ret; return ret;
} }
static RtpDirection matchDirection(RtpDirection offer_direction, RtpDirection supported){ static RtpDirection matchDirection(RtpDirection offer_direction, RtpDirection supported) {
switch (offer_direction) { switch (offer_direction) {
case RtpDirection::sendonly : { case RtpDirection::sendonly: {
if (supported != RtpDirection::recvonly && supported != RtpDirection::sendrecv) { if (supported != RtpDirection::recvonly && supported != RtpDirection::sendrecv) {
//我们不支持接收 // 我们不支持接收
return RtpDirection::inactive; return RtpDirection::inactive;
} }
return RtpDirection::recvonly; return RtpDirection::recvonly;
} }
case RtpDirection::recvonly : { case RtpDirection::recvonly: {
if (supported != RtpDirection::sendonly && supported != RtpDirection::sendrecv) { if (supported != RtpDirection::sendonly && supported != RtpDirection::sendrecv) {
//我们不支持发送 // 我们不支持发送
return RtpDirection::inactive; return RtpDirection::inactive;
} }
return RtpDirection::sendonly; return RtpDirection::sendonly;
} }
//对方支持发送接收,那么最终能力根据配置来决定 // 对方支持发送接收,那么最终能力根据配置来决定
case RtpDirection::sendrecv : return (supported == RtpDirection::invalid ? RtpDirection::inactive : supported); case RtpDirection::sendrecv: return (supported == RtpDirection::invalid ? RtpDirection::inactive : supported);
case RtpDirection::inactive : return RtpDirection::inactive; case RtpDirection::inactive: return RtpDirection::inactive;
default: return RtpDirection::invalid; default: return RtpDirection::invalid;
} }
} }
static DtlsRole mathDtlsRole(DtlsRole role){ static DtlsRole mathDtlsRole(DtlsRole role) {
switch (role) { switch (role) {
case DtlsRole::actpass: case DtlsRole::actpass:
case DtlsRole::active: return DtlsRole::passive; case DtlsRole::active: return DtlsRole::passive;
@ -1605,7 +1595,7 @@ static DtlsRole mathDtlsRole(DtlsRole role){
} }
} }
void RtcConfigure::matchMedia(const std::shared_ptr<RtcSession> &ret,const RtcMedia &offer_media) const { void RtcConfigure::matchMedia(const std::shared_ptr<RtcSession> &ret, const RtcMedia &offer_media) const {
bool check_profile = true; bool check_profile = true;
bool check_codec = true; bool check_codec = true;
const RtcTrackConfigure *cfg_ptr = nullptr; const RtcTrackConfigure *cfg_ptr = nullptr;
@ -1643,20 +1633,20 @@ RETRY:
} }
const RtcCodecPlan *selected_plan = nullptr; const RtcCodecPlan *selected_plan = nullptr;
for (auto &plan : offer_media.plan) { for (auto &plan : offer_media.plan) {
//先检查编码格式是否为偏好 // 先检查编码格式是否为偏好
if (check_codec && getCodecId(plan.codec) != codec) { if (check_codec && getCodecId(plan.codec) != codec) {
continue; continue;
} }
//命中偏好的编码格式,然后检查规格 // 命中偏好的编码格式,然后检查规格
if (check_profile && !onCheckCodecProfile(plan, codec)) { if (check_profile && !onCheckCodecProfile(plan, codec)) {
continue; continue;
} }
//找到中意的codec // 找到中意的codec
selected_plan = &plan; selected_plan = &plan;
break; break;
} }
if (!selected_plan) { if (!selected_plan) {
//offer中该媒体的所有的codec都不支持 // offer中该媒体的所有的codec都不支持
continue; continue;
} }
RtcMedia answer_media; RtcMedia answer_media;
@ -1682,24 +1672,23 @@ RETRY:
answer_media.role = mathDtlsRole(offer_media.role); answer_media.role = mathDtlsRole(offer_media.role);
//如果codec匹配失败那么禁用该track // 如果codec匹配失败那么禁用该track
answer_media.direction = check_codec ? matchDirection(offer_media.direction, configure.direction) answer_media.direction = check_codec ? matchDirection(offer_media.direction, configure.direction) : RtpDirection::inactive;
: RtpDirection::inactive;
if (answer_media.direction == RtpDirection::invalid) { if (answer_media.direction == RtpDirection::invalid) {
continue; continue;
} }
if (answer_media.direction == RtpDirection::sendrecv) { if (answer_media.direction == RtpDirection::sendrecv) {
//如果是收发双向那么我们拷贝offer sdp的ssrc确保ssrc一致 // 如果是收发双向那么我们拷贝offer sdp的ssrc确保ssrc一致
answer_media.rtp_rtx_ssrc = offer_media.rtp_rtx_ssrc; answer_media.rtp_rtx_ssrc = offer_media.rtp_rtx_ssrc;
} }
//添加媒体plan // 添加媒体plan
answer_media.plan.emplace_back(*selected_plan); answer_media.plan.emplace_back(*selected_plan);
onSelectPlan(answer_media.plan.back(), codec); onSelectPlan(answer_media.plan.back(), codec);
set<uint8_t> pt_selected = {selected_plan->pt}; set<uint8_t> pt_selected = { selected_plan->pt };
//添加rtx,red,ulpfec plan // 添加rtx,red,ulpfec plan
if (configure.support_red || configure.support_rtx || configure.support_ulpfec) { if (configure.support_red || configure.support_rtx || configure.support_ulpfec) {
for (auto &plan : offer_media.plan) { for (auto &plan : offer_media.plan) {
if (!strcasecmp(plan.codec.data(), "rtx")) { if (!strcasecmp(plan.codec.data(), "rtx")) {
@ -1726,7 +1715,7 @@ RETRY:
} }
} }
//对方和我方都支持的扩展,那么我们才支持 // 对方和我方都支持的扩展,那么我们才支持
for (auto &ext : offer_media.extmap) { for (auto &ext : offer_media.extmap) {
auto it = configure.extmap.find(RtpExt::getExtType(ext.ext)); auto it = configure.extmap.find(RtpExt::getExtType(ext.ext));
if (it != configure.extmap.end()) { if (it != configure.extmap.end()) {
@ -1743,10 +1732,10 @@ RETRY:
auto &rtcp_fb_ref = answer_media.plan[0].rtcp_fb; auto &rtcp_fb_ref = answer_media.plan[0].rtcp_fb;
rtcp_fb_ref.clear(); rtcp_fb_ref.clear();
//对方和我方都支持的rtcpfb那么我们才支持 // 对方和我方都支持的rtcpfb那么我们才支持
for (auto &fp : selected_plan->rtcp_fb) { for (auto &fp : selected_plan->rtcp_fb) {
if (configure.rtcp_fb.find(fp) != configure.rtcp_fb.end()) { if (configure.rtcp_fb.find(fp) != configure.rtcp_fb.end()) {
//对方该rtcp被我们支持 // 对方该rtcp被我们支持
rtcp_fb_ref.emplace(fp); rtcp_fb_ref.emplace(fp);
} }
} }
@ -1764,19 +1753,19 @@ RETRY:
} }
if (check_profile) { if (check_profile) {
//如果是由于检查profile导致匹配失败那么重试一次且不检查profile // 如果是由于检查profile导致匹配失败那么重试一次且不检查profile
check_profile = false; check_profile = false;
goto RETRY; goto RETRY;
} }
if (check_codec) { if (check_codec) {
//如果是由于检查codec导致匹配失败那么重试一次且不检查codec // 如果是由于检查codec导致匹配失败那么重试一次且不检查codec
check_codec = false; check_codec = false;
goto RETRY; goto RETRY;
} }
} }
void RtcConfigure::setPlayRtspInfo(const string &sdp){ void RtcConfigure::setPlayRtspInfo(const string &sdp) {
RtcSession session; RtcSession session;
video.direction = RtpDirection::inactive; video.direction = RtpDirection::inactive;
audio.direction = RtpDirection::inactive; audio.direction = RtpDirection::inactive;
@ -1784,14 +1773,14 @@ void RtcConfigure::setPlayRtspInfo(const string &sdp){
session.loadFrom(sdp); session.loadFrom(sdp);
for (auto &m : session.media) { for (auto &m : session.media) {
switch (m.type) { switch (m.type) {
case TrackVideo : { case TrackVideo: {
video.direction = RtpDirection::sendonly; video.direction = RtpDirection::sendonly;
_rtsp_video_plan = std::make_shared<RtcCodecPlan>(m.plan[0]); _rtsp_video_plan = std::make_shared<RtcCodecPlan>(m.plan[0]);
video.preferred_codec.clear(); video.preferred_codec.clear();
video.preferred_codec.emplace_back(getCodecId(_rtsp_video_plan->codec)); video.preferred_codec.emplace_back(getCodecId(_rtsp_video_plan->codec));
break; break;
} }
case TrackAudio : { case TrackAudio: {
audio.direction = RtpDirection::sendonly; audio.direction = RtpDirection::sendonly;
_rtsp_audio_plan = std::make_shared<RtcCodecPlan>(m.plan[0]); _rtsp_audio_plan = std::make_shared<RtcCodecPlan>(m.plan[0]);
audio.preferred_codec.clear(); audio.preferred_codec.clear();
@ -1803,21 +1792,21 @@ void RtcConfigure::setPlayRtspInfo(const string &sdp){
} }
} }
static const string kProfile{"profile-level-id"}; static const string kProfile { "profile-level-id" };
static const string kMode{"packetization-mode"}; static const string kMode { "packetization-mode" };
bool RtcConfigure::onCheckCodecProfile(const RtcCodecPlan &plan, CodecId codec) const { bool RtcConfigure::onCheckCodecProfile(const RtcCodecPlan &plan, CodecId codec) const {
if (_rtsp_audio_plan && codec == getCodecId(_rtsp_audio_plan->codec)) { if (_rtsp_audio_plan && codec == getCodecId(_rtsp_audio_plan->codec)) {
if (plan.sample_rate != _rtsp_audio_plan->sample_rate || plan.channel != _rtsp_audio_plan->channel) { if (plan.sample_rate != _rtsp_audio_plan->sample_rate || plan.channel != _rtsp_audio_plan->channel) {
//音频采样率和通道数必须相同 // 音频采样率和通道数必须相同
return false; return false;
} }
return true; return true;
} }
if (_rtsp_video_plan && codec == CodecH264 && getCodecId(_rtsp_video_plan->codec) == CodecH264) { if (_rtsp_video_plan && codec == CodecH264 && getCodecId(_rtsp_video_plan->codec) == CodecH264) {
//h264时profile-level-id // h264时profile-level-id
if (strcasecmp(_rtsp_video_plan->fmtp[kProfile].data(), const_cast<RtcCodecPlan &>(plan).fmtp[kProfile].data())) { if (strcasecmp(_rtsp_video_plan->fmtp[kProfile].data(), const_cast<RtcCodecPlan &>(plan).fmtp[kProfile].data())) {
//profile-level-id 不匹配 // profile-level-id 不匹配
return false; return false;
} }
return true; return true;

View File

@ -22,10 +22,10 @@
namespace mediakit { namespace mediakit {
//https://datatracker.ietf.org/doc/rfc4566/?include_text=1 // https://datatracker.ietf.org/doc/rfc4566/?include_text=1
//https://blog.csdn.net/aggresss/article/details/109850434 // https://blog.csdn.net/aggresss/article/details/109850434
//https://aggresss.blog.csdn.net/article/details/106436703 // https://aggresss.blog.csdn.net/article/details/106436703
//Session description // Session description
// v= (protocol version) // v= (protocol version)
// o= (originator and session identifier) // o= (originator and session identifier)
// s= (session name) // s= (session name)
@ -57,62 +57,52 @@ namespace mediakit {
enum class RtpDirection { enum class RtpDirection {
invalid = -1, invalid = -1,
//只发送 // 只发送
sendonly, sendonly,
//只接收 // 只接收
recvonly, recvonly,
//同时发送接收 // 同时发送接收
sendrecv, sendrecv,
//禁止发送数据 // 禁止发送数据
inactive inactive
}; };
enum class DtlsRole { enum class DtlsRole {
invalid = -1, invalid = -1,
//客户端 // 客户端
active, active,
//服务端 // 服务端
passive, passive,
//既可作做客户端也可以做服务端 // 既可作做客户端也可以做服务端
actpass, actpass,
}; };
enum class SdpType { enum class SdpType { invalid = -1, offer, answer };
invalid = -1,
offer,
answer
};
DtlsRole getDtlsRole(const std::string &str); DtlsRole getDtlsRole(const std::string &str);
const char* getDtlsRoleString(DtlsRole role); const char *getDtlsRoleString(DtlsRole role);
RtpDirection getRtpDirection(const std::string &str); RtpDirection getRtpDirection(const std::string &str);
const char* getRtpDirectionString(RtpDirection val); const char *getRtpDirectionString(RtpDirection val);
class SdpItem { class SdpItem {
public: public:
using Ptr = std::shared_ptr<SdpItem>; using Ptr = std::shared_ptr<SdpItem>;
virtual ~SdpItem() = default; virtual ~SdpItem() = default;
virtual void parse(const std::string &str) { virtual void parse(const std::string &str) { value = str; }
value = str; virtual std::string toString() const { return value; }
} virtual const char *getKey() const = 0;
virtual std::string toString() const {
return value;
}
virtual const char* getKey() const = 0;
void reset() { void reset() { value.clear(); }
value.clear();
}
protected: protected:
mutable std::string value; mutable std::string value;
}; };
template <char KEY> template <char KEY>
class SdpString : public SdpItem{ class SdpString : public SdpItem {
public: public:
SdpString() = default; SdpString() = default;
SdpString(std::string val) {value = std::move(val);} SdpString(std::string val) { value = std::move(val); }
// *=* // *=*
const char* getKey() const override { static std::string key(1, KEY); return key.data();} const char* getKey() const override { static std::string key(1, KEY); return key.data();}
}; };
@ -126,34 +116,34 @@ public:
this->value = std::move(val); this->value = std::move(val);
} }
const char* getKey() const override { return key.data();} const char *getKey() const override { return key.data(); }
}; };
class SdpTime : public SdpItem{ class SdpTime : public SdpItem {
public: public:
//5.9. Timing ("t=") // 5.9. Timing ("t=")
// t=<start-time> <stop-time> // t=<start-time> <stop-time>
uint64_t start {0}; uint64_t start { 0 };
uint64_t stop {0}; uint64_t stop { 0 };
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "t";} const char *getKey() const override { return "t"; }
}; };
class SdpOrigin : public SdpItem{ class SdpOrigin : public SdpItem {
public: public:
// 5.2. Origin ("o=") // 5.2. Origin ("o=")
// o=jdoe 2890844526 2890842807 IN IP4 10.47.16.5 // o=jdoe 2890844526 2890842807 IN IP4 10.47.16.5
// o=<username> <sess-id> <sess-version> <nettype> <addrtype> <unicast-address> // o=<username> <sess-id> <sess-version> <nettype> <addrtype> <unicast-address>
std::string username {"-"}; std::string username { "-" };
std::string session_id; std::string session_id;
std::string session_version; std::string session_version;
std::string nettype {"IN"}; std::string nettype { "IN" };
std::string addrtype {"IP4"}; std::string addrtype { "IP4" };
std::string address {"0.0.0.0"}; std::string address { "0.0.0.0" };
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "o";} const char *getKey() const override { return "o"; }
bool empty() const { bool empty() const {
return username.empty() || session_id.empty() || session_version.empty() return username.empty() || session_id.empty() || session_version.empty()
|| nettype.empty() || addrtype.empty() || address.empty(); || nettype.empty() || addrtype.empty() || address.empty();
@ -165,28 +155,28 @@ public:
// 5.7. Connection Data ("c=") // 5.7. Connection Data ("c=")
// c=IN IP4 224.2.17.12/127 // c=IN IP4 224.2.17.12/127
// c=<nettype> <addrtype> <connection-address> // c=<nettype> <addrtype> <connection-address>
std::string nettype {"IN"}; std::string nettype { "IN" };
std::string addrtype {"IP4"}; std::string addrtype { "IP4" };
std::string address {"0.0.0.0"}; std::string address { "0.0.0.0" };
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "c";} const char *getKey() const override { return "c"; }
bool empty() const {return address.empty();} bool empty() const { return address.empty(); }
}; };
class SdpBandwidth : public SdpItem { class SdpBandwidth : public SdpItem {
public: public:
//5.8. Bandwidth ("b=") // 5.8. Bandwidth ("b=")
//b=<bwtype>:<bandwidth> // b=<bwtype>:<bandwidth>
//AS、CT // AS、CT
std::string bwtype {"AS"}; std::string bwtype { "AS" };
uint32_t bandwidth {0}; uint32_t bandwidth { 0 };
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "b";} const char *getKey() const override { return "b"; }
bool empty() const {return bandwidth == 0;} bool empty() const { return bandwidth == 0; }
}; };
class SdpMedia : public SdpItem { class SdpMedia : public SdpItem {
@ -195,47 +185,47 @@ public:
// m=<media> <port> <proto> <fmt> ... // m=<media> <port> <proto> <fmt> ...
TrackType type; TrackType type;
uint16_t port; uint16_t port;
//RTP/AVP应用场景为视频/音频的 RTP 协议。参考 RFC 3551 // RTP/AVP应用场景为视频/音频的 RTP 协议。参考 RFC 3551
//RTP/SAVP应用场景为视频/音频的 SRTP 协议。参考 RFC 3711 // RTP/SAVP应用场景为视频/音频的 SRTP 协议。参考 RFC 3711
//RTP/AVPF: 应用场景为视频/音频的 RTP 协议,支持 RTCP-based Feedback。参考 RFC 4585 // RTP/AVPF: 应用场景为视频/音频的 RTP 协议,支持 RTCP-based Feedback。参考 RFC 4585
//RTP/SAVPF: 应用场景为视频/音频的 SRTP 协议,支持 RTCP-based Feedback。参考 RFC 5124 // RTP/SAVPF: 应用场景为视频/音频的 SRTP 协议,支持 RTCP-based Feedback。参考 RFC 5124
std::string proto; std::string proto;
std::vector<std::string> fmts; std::vector<std::string> fmts;
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "m";} const char *getKey() const override { return "m"; }
}; };
class SdpAttr : public SdpItem{ class SdpAttr : public SdpItem {
public: public:
using Ptr = std::shared_ptr<SdpAttr>; using Ptr = std::shared_ptr<SdpAttr>;
//5.13. Attributes ("a=") // 5.13. Attributes ("a=")
//a=<attribute> // a=<attribute>
//a=<attribute>:<value> // a=<attribute>:<value>
SdpItem::Ptr detail; SdpItem::Ptr detail;
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "a";} const char *getKey() const override { return "a"; }
}; };
class SdpAttrGroup : public SdpItem{ class SdpAttrGroup : public SdpItem {
public: public:
//a=group:BUNDLE line with all the 'mid' identifiers part of the // a=group:BUNDLE line with all the 'mid' identifiers part of the
// BUNDLE group is included at the session-level. // BUNDLE group is included at the session-level.
//a=group:LS session level attribute MUST be included wth the 'mid' // a=group:LS session level attribute MUST be included wth the 'mid'
// identifiers that are part of the same lip sync group. // identifiers that are part of the same lip sync group.
std::string type {"BUNDLE"}; std::string type { "BUNDLE" };
std::vector<std::string> mids; std::vector<std::string> mids;
void parse(const std::string &str) override ; void parse(const std::string &str) override;
std::string toString() const override ; std::string toString() const override;
const char* getKey() const override { return "group";} const char *getKey() const override { return "group"; }
}; };
class SdpAttrMsidSemantic : public SdpItem { class SdpAttrMsidSemantic : public SdpItem {
public: public:
//https://tools.ietf.org/html/draft-alvestrand-rtcweb-msid-02#section-3 // https://tools.ietf.org/html/draft-alvestrand-rtcweb-msid-02#section-3
//3. The Msid-Semantic Attribute // 3. The Msid-Semantic Attribute
// //
// In order to fully reproduce the semantics of the SDP and SSRC // In order to fully reproduce the semantics of the SDP and SSRC
// grouping frameworks, a session-level attribute is defined for // grouping frameworks, a session-level attribute is defined for
@ -253,229 +243,226 @@ public:
// The semantic field may hold values from the IANA registries // The semantic field may hold values from the IANA registries
// "Semantics for the "ssrc-group" SDP Attribute" and "Semantics for the // "Semantics for the "ssrc-group" SDP Attribute" and "Semantics for the
// "group" SDP Attribute". // "group" SDP Attribute".
//a=msid-semantic: WMS 616cfbb1-33a3-4d8c-8275-a199d6005549 // a=msid-semantic: WMS 616cfbb1-33a3-4d8c-8275-a199d6005549
std::string msid{"WMS"}; std::string msid { "WMS" };
std::string token; std::string token;
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "msid-semantic";} const char *getKey() const override { return "msid-semantic"; }
bool empty() const { bool empty() const { return msid.empty(); }
return msid.empty();
}
}; };
class SdpAttrRtcp : public SdpItem { class SdpAttrRtcp : public SdpItem {
public: public:
// a=rtcp:9 IN IP4 0.0.0.0 // a=rtcp:9 IN IP4 0.0.0.0
uint16_t port{0}; uint16_t port { 0 };
std::string nettype {"IN"}; std::string nettype { "IN" };
std::string addrtype {"IP4"}; std::string addrtype { "IP4" };
std::string address {"0.0.0.0"}; std::string address { "0.0.0.0" };
void parse(const std::string &str) override;; void parse(const std::string &str) override;
;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "rtcp";} const char *getKey() const override { return "rtcp"; }
bool empty() const { bool empty() const { return address.empty() || !port; }
return address.empty() || !port;
}
}; };
class SdpAttrIceUfrag : public SdpItem { class SdpAttrIceUfrag : public SdpItem {
public: public:
SdpAttrIceUfrag() = default; SdpAttrIceUfrag() = default;
SdpAttrIceUfrag(std::string str) {value = std::move(str);} SdpAttrIceUfrag(std::string str) { value = std::move(str); }
//a=ice-ufrag:sXJ3 // a=ice-ufrag:sXJ3
const char* getKey() const override { return "ice-ufrag";} const char *getKey() const override { return "ice-ufrag"; }
}; };
class SdpAttrIcePwd : public SdpItem { class SdpAttrIcePwd : public SdpItem {
public: public:
SdpAttrIcePwd() = default; SdpAttrIcePwd() = default;
SdpAttrIcePwd(std::string str) {value = std::move(str);} SdpAttrIcePwd(std::string str) { value = std::move(str); }
//a=ice-pwd:yEclOTrLg1gEubBFefOqtmyV // a=ice-pwd:yEclOTrLg1gEubBFefOqtmyV
const char* getKey() const override { return "ice-pwd";} const char *getKey() const override { return "ice-pwd"; }
}; };
class SdpAttrIceOption : public SdpItem { class SdpAttrIceOption : public SdpItem {
public: public:
//a=ice-options:trickle // a=ice-options:trickle
bool trickle{false}; bool trickle { false };
bool renomination{false}; bool renomination { false };
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "ice-options";} const char *getKey() const override { return "ice-options"; }
}; };
class SdpAttrFingerprint : public SdpItem { class SdpAttrFingerprint : public SdpItem {
public: public:
//a=fingerprint:sha-256 22:14:B5:AF:66:12:C7:C7:8D:EF:4B:DE:40:25:ED:5D:8F:17:54:DD:88:33:C0:13:2E:FD:1A:FA:7E:7A:1B:79 // a=fingerprint:sha-256 22:14:B5:AF:66:12:C7:C7:8D:EF:4B:DE:40:25:ED:5D:8F:17:54:DD:88:33:C0:13:2E:FD:1A:FA:7E:7A:1B:79
std::string algorithm; std::string algorithm;
std::string hash; std::string hash;
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "fingerprint";} const char *getKey() const override { return "fingerprint"; }
bool empty() const { return algorithm.empty() || hash.empty(); } bool empty() const { return algorithm.empty() || hash.empty(); }
}; };
class SdpAttrSetup : public SdpItem { class SdpAttrSetup : public SdpItem {
public: public:
//a=setup:actpass // a=setup:actpass
SdpAttrSetup() = default; SdpAttrSetup() = default;
SdpAttrSetup(DtlsRole r) { role = r; } SdpAttrSetup(DtlsRole r) { role = r; }
DtlsRole role{DtlsRole::actpass}; DtlsRole role { DtlsRole::actpass };
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "setup";} const char *getKey() const override { return "setup"; }
}; };
class SdpAttrMid : public SdpItem { class SdpAttrMid : public SdpItem {
public: public:
SdpAttrMid() = default; SdpAttrMid() = default;
SdpAttrMid(std::string val) { value = std::move(val); } SdpAttrMid(std::string val) { value = std::move(val); }
//a=mid:audio // a=mid:audio
const char* getKey() const override { return "mid";} const char *getKey() const override { return "mid"; }
}; };
class SdpAttrExtmap : public SdpItem { class SdpAttrExtmap : public SdpItem {
public: public:
//https://aggresss.blog.csdn.net/article/details/106436703 // https://aggresss.blog.csdn.net/article/details/106436703
//a=extmap:1[/sendonly] urn:ietf:params:rtp-hdrext:ssrc-audio-level // a=extmap:1[/sendonly] urn:ietf:params:rtp-hdrext:ssrc-audio-level
uint8_t id; uint8_t id;
RtpDirection direction{RtpDirection::invalid}; RtpDirection direction { RtpDirection::invalid };
std::string ext; std::string ext;
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "extmap";} const char *getKey() const override { return "extmap"; }
}; };
class SdpAttrRtpMap : public SdpItem { class SdpAttrRtpMap : public SdpItem {
public: public:
//a=rtpmap:111 opus/48000/2 // a=rtpmap:111 opus/48000/2
uint8_t pt; uint8_t pt;
std::string codec; std::string codec;
uint32_t sample_rate; uint32_t sample_rate;
uint32_t channel {0}; uint32_t channel { 0 };
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "rtpmap";} const char *getKey() const override { return "rtpmap"; }
}; };
class SdpAttrRtcpFb : public SdpItem { class SdpAttrRtcpFb : public SdpItem {
public: public:
//a=rtcp-fb:98 nack pli // a=rtcp-fb:98 nack pli
//a=rtcp-fb:120 nack 支持 nack 重传nack (Negative-Acknowledgment) 。 // a=rtcp-fb:120 nack 支持 nack 重传nack (Negative-Acknowledgment) 。
//a=rtcp-fb:120 nack pli 支持 nack 关键帧重传PLI (Picture Loss Indication) 。 // a=rtcp-fb:120 nack pli 支持 nack 关键帧重传PLI (Picture Loss Indication) 。
//a=rtcp-fb:120 ccm fir 支持编码层关键帧请求CCM (Codec Control Message)FIR (Full Intra Request ),通常与 nack pli 有同样的效果,但是 nack pli 是用于重传时的关键帧请求。 // a=rtcp-fb:120 ccm fir 支持编码层关键帧请求CCM (Codec Control Message)FIR (Full Intra Request ),通常与 nack pli 有同样的效果,但是 nack pli
//a=rtcp-fb:120 goog-remb 支持 REMB (Receiver Estimated Maximum Bitrate) 。 // 是用于重传时的关键帧请求。 a=rtcp-fb:120 goog-remb 支持 REMB (Receiver Estimated Maximum Bitrate) 。 a=rtcp-fb:120 transport-cc 支持 TCC (Transport
//a=rtcp-fb:120 transport-cc 支持 TCC (Transport Congest Control) 。 // Congest Control) 。
uint8_t pt; uint8_t pt;
std::string rtcp_type; std::string rtcp_type;
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "rtcp-fb";} const char *getKey() const override { return "rtcp-fb"; }
}; };
class SdpAttrFmtp : public SdpItem { class SdpAttrFmtp : public SdpItem {
public: public:
//fmtp:96 level-asymmetry-allowed=1;packetization-mode=0;profile-level-id=42e01f // fmtp:96 level-asymmetry-allowed=1;packetization-mode=0;profile-level-id=42e01f
uint8_t pt; uint8_t pt;
std::map<std::string/*key*/, std::string/*value*/, StrCaseCompare> fmtp; std::map<std::string /*key*/, std::string /*value*/, StrCaseCompare> fmtp;
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "fmtp";} const char *getKey() const override { return "fmtp"; }
}; };
class SdpAttrSSRC : public SdpItem { class SdpAttrSSRC : public SdpItem {
public: public:
//a=ssrc:3245185839 cname:Cx4i/VTR51etgjT7 // a=ssrc:3245185839 cname:Cx4i/VTR51etgjT7
//a=ssrc:3245185839 msid:cb373bff-0fea-4edb-bc39-e49bb8e8e3b9 0cf7e597-36a2-4480-9796-69bf0955eef5 // a=ssrc:3245185839 msid:cb373bff-0fea-4edb-bc39-e49bb8e8e3b9 0cf7e597-36a2-4480-9796-69bf0955eef5
//a=ssrc:3245185839 mslabel:cb373bff-0fea-4edb-bc39-e49bb8e8e3b9 // a=ssrc:3245185839 mslabel:cb373bff-0fea-4edb-bc39-e49bb8e8e3b9
//a=ssrc:3245185839 label:0cf7e597-36a2-4480-9796-69bf0955eef5 // a=ssrc:3245185839 label:0cf7e597-36a2-4480-9796-69bf0955eef5
//a=ssrc:<ssrc-id> <attribute> // a=ssrc:<ssrc-id> <attribute>
//a=ssrc:<ssrc-id> <attribute>:<value> // a=ssrc:<ssrc-id> <attribute>:<value>
//cname 是必须的msid/mslabel/label 这三个属性都是 WebRTC 自创的,或者说 Google 自创的,可以参考 https://tools.ietf.org/html/draft-ietf-mmusic-msid-17 // cname 是必须的msid/mslabel/label 这三个属性都是 WebRTC 自创的,或者说 Google 自创的,可以参考 https://tools.ietf.org/html/draft-ietf-mmusic-msid-17
// 理解它们三者的关系需要先了解三个概念RTP stream / MediaStreamTrack / MediaStream // 理解它们三者的关系需要先了解三个概念RTP stream / MediaStreamTrack / MediaStream
//一个 a=ssrc 代表一个 RTP stream // 一个 a=ssrc 代表一个 RTP stream
//一个 MediaStreamTrack 通常包含一个或多个 RTP stream例如一个视频 MediaStreamTrack 中通常包含两个 RTP stream一个用于常规传输一个用于 nack 重传; // 一个 MediaStreamTrack 通常包含一个或多个 RTP stream例如一个视频 MediaStreamTrack 中通常包含两个 RTP stream一个用于常规传输一个用于 nack 重传;
//一个 MediaStream 通常包含一个或多个 MediaStreamTrack ,例如 simulcast 场景下,一个 MediaStream 通常会包含三个不同编码质量的 MediaStreamTrack // 一个 MediaStream 通常包含一个或多个 MediaStreamTrack ,例如 simulcast 场景下,一个 MediaStream 通常会包含三个不同编码质量的 MediaStreamTrack
//这种标记方式并不被 Firefox 认可,在 Firefox 生成的 SDP 中一个 a=ssrc 通常只有一行,例如: // 这种标记方式并不被 Firefox 认可,在 Firefox 生成的 SDP 中一个 a=ssrc 通常只有一行,例如:
//a=ssrc:3245185839 cname:Cx4i/VTR51etgjT7 // a=ssrc:3245185839 cname:Cx4i/VTR51etgjT7
uint32_t ssrc; uint32_t ssrc;
std::string attribute; std::string attribute;
std::string attribute_value; std::string attribute_value;
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "ssrc";} const char *getKey() const override { return "ssrc"; }
}; };
class SdpAttrSSRCGroup : public SdpItem { class SdpAttrSSRCGroup : public SdpItem {
public: public:
//a=ssrc-group 定义参考 RFC 5576(https://tools.ietf.org/html/rfc5576) ,用于描述多个 ssrc 之间的关联,常见的有两种: // a=ssrc-group 定义参考 RFC 5576(https://tools.ietf.org/html/rfc5576) ,用于描述多个 ssrc 之间的关联,常见的有两种:
//a=ssrc-group:FID 2430709021 3715850271 // a=ssrc-group:FID 2430709021 3715850271
// FID (Flow Identification) 最初用在 FEC 的关联中WebRTC 中通常用于关联一组常规 RTP stream 和 重传 RTP stream 。 // FID (Flow Identification) 最初用在 FEC 的关联中WebRTC 中通常用于关联一组常规 RTP stream 和 重传 RTP stream 。
//a=ssrc-group:SIM 360918977 360918978 360918980 // a=ssrc-group:SIM 360918977 360918978 360918980
// 在 Chrome 独有的 SDP munging 风格的 simulcast 中使用,将三组编码质量由低到高的 MediaStreamTrack 关联在一起。 // 在 Chrome 独有的 SDP munging 风格的 simulcast 中使用,将三组编码质量由低到高的 MediaStreamTrack 关联在一起。
std::string type{"FID"}; std::string type { "FID" };
std::vector<uint32_t> ssrcs; std::vector<uint32_t> ssrcs;
bool isFID() const { return type == "FID"; } bool isFID() const { return type == "FID"; }
bool isSIM() const { return type == "SIM"; } bool isSIM() const { return type == "SIM"; }
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "ssrc-group";} const char *getKey() const override { return "ssrc-group"; }
}; };
class SdpAttrSctpMap : public SdpItem { class SdpAttrSctpMap : public SdpItem {
public: public:
//https://tools.ietf.org/html/draft-ietf-mmusic-sctp-sdp-05 // https://tools.ietf.org/html/draft-ietf-mmusic-sctp-sdp-05
//a=sctpmap:5000 webrtc-datachannel 1024 // a=sctpmap:5000 webrtc-datachannel 1024
//a=sctpmap: sctpmap-number media-subtypes [streams] // a=sctpmap: sctpmap-number media-subtypes [streams]
uint16_t port = 0; uint16_t port = 0;
std::string subtypes; std::string subtypes;
uint32_t streams = 0; uint32_t streams = 0;
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "sctpmap";} const char *getKey() const override { return "sctpmap"; }
bool empty() const { return port == 0 && subtypes.empty() && streams == 0; } bool empty() const { return port == 0 && subtypes.empty() && streams == 0; }
}; };
class SdpAttrCandidate : public SdpItem { class SdpAttrCandidate : public SdpItem {
public: public:
using Ptr = std::shared_ptr<SdpAttrCandidate>; using Ptr = std::shared_ptr<SdpAttrCandidate>;
//https://tools.ietf.org/html/rfc5245 // https://tools.ietf.org/html/rfc5245
//15.1. "candidate" Attribute // 15.1. "candidate" Attribute
//a=candidate:4 1 udp 2 192.168.1.7 58107 typ host // a=candidate:4 1 udp 2 192.168.1.7 58107 typ host
//a=candidate:<foundation> <component-id> <transport> <priority> <address> <port> typ <cand-type> // a=candidate:<foundation> <component-id> <transport> <priority> <address> <port> typ <cand-type>
std::string foundation; std::string foundation;
//传输媒体的类型,1代表RTP;2代表 RTCP。 // 传输媒体的类型,1代表RTP;2代表 RTCP。
uint32_t component; uint32_t component;
std::string transport {"udp"}; std::string transport { "udp" };
uint32_t priority; uint32_t priority;
std::string address; std::string address;
uint16_t port; uint16_t port;
std::string type; std::string type;
std::vector<std::pair<std::string, std::string> > arr; std::vector<std::pair<std::string, std::string>> arr;
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "candidate";} const char *getKey() const override { return "candidate"; }
}; };
class SdpAttrMsid : public SdpItem{ class SdpAttrMsid : public SdpItem {
public: public:
const char* getKey() const override { return "msid";} const char *getKey() const override { return "msid"; }
}; };
class SdpAttrExtmapAllowMixed : public SdpItem{ class SdpAttrExtmapAllowMixed : public SdpItem {
public: public:
const char* getKey() const override { return "extmap-allow-mixed";} const char *getKey() const override { return "extmap-allow-mixed"; }
}; };
class SdpAttrSimulcast : public SdpItem{ class SdpAttrSimulcast : public SdpItem {
public: public:
//https://www.meetecho.com/blog/simulcast-janus-ssrc/ // https://www.meetecho.com/blog/simulcast-janus-ssrc/
//https://tools.ietf.org/html/draft-ietf-mmusic-sdp-simulcast-14 // https://tools.ietf.org/html/draft-ietf-mmusic-sdp-simulcast-14
const char* getKey() const override { return "simulcast";} const char *getKey() const override { return "simulcast"; }
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
bool empty() const { return rids.empty(); } bool empty() const { return rids.empty(); }
@ -483,11 +470,11 @@ public:
std::vector<std::string> rids; std::vector<std::string> rids;
}; };
class SdpAttrRid : public SdpItem{ class SdpAttrRid : public SdpItem {
public: public:
void parse(const std::string &str) override; void parse(const std::string &str) override;
std::string toString() const override; std::string toString() const override;
const char* getKey() const override { return "rid";} const char *getKey() const override { return "rid"; }
std::string direction; std::string direction;
std::string rid; std::string rid;
}; };
@ -507,8 +494,8 @@ public:
RtpDirection getDirection() const; RtpDirection getDirection() const;
template<typename cls> template <typename cls>
cls getItemClass(char key, const char *attr_key = nullptr) const{ cls getItemClass(char key, const char *attr_key = nullptr) const {
auto item = std::dynamic_pointer_cast<cls>(getItem(key, attr_key)); auto item = std::dynamic_pointer_cast<cls>(getItem(key, attr_key));
if (!item) { if (!item) {
return cls(); return cls();
@ -516,7 +503,7 @@ public:
return *item; return *item;
} }
std::string getStringItem(char key, const char *attr_key = nullptr) const{ std::string getStringItem(char key, const char *attr_key = nullptr) const {
auto item = getItem(key, attr_key); auto item = getItem(key, attr_key);
if (!item) { if (!item) {
return ""; return "";
@ -526,7 +513,7 @@ public:
SdpItem::Ptr getItem(char key, const char *attr_key = nullptr) const; SdpItem::Ptr getItem(char key, const char *attr_key = nullptr) const;
template<typename cls> template <typename cls>
std::vector<cls> getAllItem(char key_c, const char *attr_key = nullptr) const { std::vector<cls> getAllItem(char key_c, const char *attr_key = nullptr) const {
std::vector<cls> ret; std::vector<cls> ret;
std::string key(1, key_c); std::string key(1, key_c);
@ -555,7 +542,7 @@ private:
std::vector<SdpItem::Ptr> items; std::vector<SdpItem::Ptr> items;
}; };
class RtcSessionSdp : public RtcSdpBase{ class RtcSessionSdp : public RtcSdpBase {
public: public:
using Ptr = std::shared_ptr<RtcSessionSdp>; using Ptr = std::shared_ptr<RtcSessionSdp>;
int getVersion() const; int getVersion() const;
@ -580,45 +567,45 @@ public:
////////////////////////////////////////////////////////////////// //////////////////////////////////////////////////////////////////
//ssrc相关信息 // ssrc相关信息
class RtcSSRC{ class RtcSSRC {
public: public:
uint32_t ssrc {0}; uint32_t ssrc { 0 };
uint32_t rtx_ssrc {0}; uint32_t rtx_ssrc { 0 };
std::string cname; std::string cname;
std::string msid; std::string msid;
std::string mslabel; std::string mslabel;
std::string label; std::string label;
bool empty() const {return ssrc == 0 && cname.empty();} bool empty() const { return ssrc == 0 && cname.empty(); }
}; };
//rtc传输编码方案 // rtc传输编码方案
class RtcCodecPlan{ class RtcCodecPlan {
public: public:
using Ptr = std::shared_ptr<RtcCodecPlan>; using Ptr = std::shared_ptr<RtcCodecPlan>;
uint8_t pt; uint8_t pt;
std::string codec; std::string codec;
uint32_t sample_rate; uint32_t sample_rate;
//音频时有效 // 音频时有效
uint32_t channel = 0; uint32_t channel = 0;
//rtcp反馈 // rtcp反馈
std::set<std::string> rtcp_fb; std::set<std::string> rtcp_fb;
std::map<std::string/*key*/, std::string/*value*/, StrCaseCompare> fmtp; std::map<std::string /*key*/, std::string /*value*/, StrCaseCompare> fmtp;
std::string getFmtp(const char *key) const; std::string getFmtp(const char *key) const;
}; };
//rtc 媒体描述 // rtc 媒体描述
class RtcMedia{ class RtcMedia {
public: public:
TrackType type{TrackType::TrackInvalid}; TrackType type { TrackType::TrackInvalid };
std::string mid; std::string mid;
uint16_t port{0}; uint16_t port { 0 };
SdpConnection addr; SdpConnection addr;
SdpBandwidth bandwidth; SdpBandwidth bandwidth;
std::string proto; std::string proto;
RtpDirection direction{RtpDirection::invalid}; RtpDirection direction { RtpDirection::invalid };
std::vector<RtcCodecPlan> plan; std::vector<RtcCodecPlan> plan;
//////// rtp //////// //////// rtp ////////
@ -629,20 +616,20 @@ public:
std::vector<std::string> rtp_rids; std::vector<std::string> rtp_rids;
//////// rtcp //////// //////// rtcp ////////
bool rtcp_mux{false}; bool rtcp_mux { false };
bool rtcp_rsize{false}; bool rtcp_rsize { false };
SdpAttrRtcp rtcp_addr; SdpAttrRtcp rtcp_addr;
//////// ice //////// //////// ice ////////
bool ice_trickle{false}; bool ice_trickle { false };
bool ice_lite{false}; bool ice_lite { false };
bool ice_renomination{false}; bool ice_renomination { false };
std::string ice_ufrag; std::string ice_ufrag;
std::string ice_pwd; std::string ice_pwd;
std::vector<SdpAttrCandidate> candidate; std::vector<SdpAttrCandidate> candidate;
//////// dtls //////// //////// dtls ////////
DtlsRole role{DtlsRole::invalid}; DtlsRole role { DtlsRole::invalid };
SdpAttrFingerprint fingerprint; SdpAttrFingerprint fingerprint;
//////// extmap //////// //////// extmap ////////
@ -650,7 +637,7 @@ public:
//////// sctp //////////// //////// sctp ////////////
SdpAttrSctpMap sctpmap; SdpAttrSctpMap sctpmap;
uint32_t sctp_port{0}; uint32_t sctp_port { 0 };
void checkValid() const; void checkValid() const;
const RtcCodecPlan *getPlan(uint8_t pt) const; const RtcCodecPlan *getPlan(uint8_t pt) const;
@ -705,7 +692,7 @@ public:
std::string ice_ufrag; std::string ice_ufrag;
std::string ice_pwd; std::string ice_pwd;
RtpDirection direction{RtpDirection::invalid}; RtpDirection direction { RtpDirection::invalid };
SdpAttrFingerprint fingerprint; SdpAttrFingerprint fingerprint;
std::set<std::string> rtcp_fb; std::set<std::string> rtcp_fb;
@ -752,6 +739,6 @@ private:
~SdpConst() = delete; ~SdpConst() = delete;
}; };
}// namespace mediakit } // namespace mediakit
#endif //ZLMEDIAKIT_SDP_H #endif // ZLMEDIAKIT_SDP_H

View File

@ -45,7 +45,7 @@ void WebRtcEchoTest::onCheckSdp(SdpType type, RtcSession &sdp) {
for (auto &m : sdp.media) { for (auto &m : sdp.media) {
for (auto &ssrc : m.rtp_rtx_ssrc) { for (auto &ssrc : m.rtp_rtx_ssrc) {
if (!ssrc.msid.empty()) { if (!ssrc.msid.empty()) {
ssrc.msid = "zlmediakit msid"; ssrc.msid = "zlmediakit-mslabel zlmediakit-label-" + m.mid;
} }
} }
} }

View File

@ -27,7 +27,6 @@ protected:
void onRtp(const char *buf, size_t len, uint64_t stamp_ms) override; void onRtp(const char *buf, size_t len, uint64_t stamp_ms) override;
void onRtcp(const char *buf, size_t len) override; void onRtcp(const char *buf, size_t len) override;
void onRecvRtp(MediaTrack &track, const std::string &rid, RtpPacket::Ptr rtp) override {};
void onBeforeEncryptRtp(const char *buf, int &len, void *ctx) override {}; void onBeforeEncryptRtp(const char *buf, int &len, void *ctx) override {};
void onBeforeEncryptRtcp(const char *buf, int &len, void *ctx) override {}; void onBeforeEncryptRtcp(const char *buf, int &len, void *ctx) override {};

View File

@ -17,9 +17,8 @@ namespace mediakit {
WebRtcPlayer::Ptr WebRtcPlayer::create(const EventPoller::Ptr &poller, WebRtcPlayer::Ptr WebRtcPlayer::create(const EventPoller::Ptr &poller,
const RtspMediaSource::Ptr &src, const RtspMediaSource::Ptr &src,
const MediaInfo &info, const MediaInfo &info) {
bool preferred_tcp) { WebRtcPlayer::Ptr ret(new WebRtcPlayer(poller, src, info), [](WebRtcPlayer *ptr) {
WebRtcPlayer::Ptr ret(new WebRtcPlayer(poller, src, info, preferred_tcp), [](WebRtcPlayer *ptr) {
ptr->onDestory(); ptr->onDestory();
delete ptr; delete ptr;
}); });
@ -29,8 +28,7 @@ WebRtcPlayer::Ptr WebRtcPlayer::create(const EventPoller::Ptr &poller,
WebRtcPlayer::WebRtcPlayer(const EventPoller::Ptr &poller, WebRtcPlayer::WebRtcPlayer(const EventPoller::Ptr &poller,
const RtspMediaSource::Ptr &src, const RtspMediaSource::Ptr &src,
const MediaInfo &info, const MediaInfo &info) : WebRtcTransportImp(poller) {
bool preferred_tcp) : WebRtcTransportImp(poller,preferred_tcp) {
_media_info = info; _media_info = info;
_play_src = src; _play_src = src;
CHECK(src); CHECK(src);

View File

@ -19,7 +19,7 @@ namespace mediakit {
class WebRtcPlayer : public WebRtcTransportImp { class WebRtcPlayer : public WebRtcTransportImp {
public: public:
using Ptr = std::shared_ptr<WebRtcPlayer>; using Ptr = std::shared_ptr<WebRtcPlayer>;
static Ptr create(const EventPoller::Ptr &poller, const RtspMediaSource::Ptr &src, const MediaInfo &info, bool preferred_tcp = false); static Ptr create(const EventPoller::Ptr &poller, const RtspMediaSource::Ptr &src, const MediaInfo &info);
MediaInfo getMediaInfo() { return _media_info; } MediaInfo getMediaInfo() { return _media_info; }
protected: protected:
@ -27,10 +27,9 @@ protected:
void onStartWebRTC() override; void onStartWebRTC() override;
void onDestory() override; void onDestory() override;
void onRtcConfigure(RtcConfigure &configure) const override; void onRtcConfigure(RtcConfigure &configure) const override;
void onRecvRtp(MediaTrack &track, const std::string &rid, RtpPacket::Ptr rtp) override {};
private: private:
WebRtcPlayer(const EventPoller::Ptr &poller, const RtspMediaSource::Ptr &src, const MediaInfo &info, bool preferred_tcp); WebRtcPlayer(const EventPoller::Ptr &poller, const RtspMediaSource::Ptr &src, const MediaInfo &info);
private: private:
//媒体相关元数据 //媒体相关元数据

View File

@ -20,9 +20,8 @@ WebRtcPusher::Ptr WebRtcPusher::create(const EventPoller::Ptr &poller,
const RtspMediaSource::Ptr &src, const RtspMediaSource::Ptr &src,
const std::shared_ptr<void> &ownership, const std::shared_ptr<void> &ownership,
const MediaInfo &info, const MediaInfo &info,
const ProtocolOption &option, const ProtocolOption &option) {
bool preferred_tcp) { WebRtcPusher::Ptr ret(new WebRtcPusher(poller, src, ownership, info, option), [](WebRtcPusher *ptr) {
WebRtcPusher::Ptr ret(new WebRtcPusher(poller, src, ownership, info, option,preferred_tcp), [](WebRtcPusher *ptr) {
ptr->onDestory(); ptr->onDestory();
delete ptr; delete ptr;
}); });
@ -34,8 +33,7 @@ WebRtcPusher::WebRtcPusher(const EventPoller::Ptr &poller,
const RtspMediaSource::Ptr &src, const RtspMediaSource::Ptr &src,
const std::shared_ptr<void> &ownership, const std::shared_ptr<void> &ownership,
const MediaInfo &info, const MediaInfo &info,
const ProtocolOption &option, const ProtocolOption &option) : WebRtcTransportImp(poller) {
bool preferred_tcp) : WebRtcTransportImp(poller,preferred_tcp) {
_media_info = info; _media_info = info;
_push_src = src; _push_src = src;
_push_src_ownership = ownership; _push_src_ownership = ownership;

View File

@ -20,8 +20,7 @@ class WebRtcPusher : public WebRtcTransportImp, public MediaSourceEvent {
public: public:
using Ptr = std::shared_ptr<WebRtcPusher>; using Ptr = std::shared_ptr<WebRtcPusher>;
static Ptr create(const EventPoller::Ptr &poller, const RtspMediaSource::Ptr &src, static Ptr create(const EventPoller::Ptr &poller, const RtspMediaSource::Ptr &src,
const std::shared_ptr<void> &ownership, const MediaInfo &info, const ProtocolOption &option, bool preferred_tcp = false); const std::shared_ptr<void> &ownership, const MediaInfo &info, const ProtocolOption &option);
protected: protected:
///////WebRtcTransportImp override/////// ///////WebRtcTransportImp override///////
@ -53,7 +52,7 @@ protected:
private: private:
WebRtcPusher(const EventPoller::Ptr &poller, const RtspMediaSource::Ptr &src, WebRtcPusher(const EventPoller::Ptr &poller, const RtspMediaSource::Ptr &src,
const std::shared_ptr<void> &ownership, const MediaInfo &info, const ProtocolOption &option, bool preferred_tcp); const std::shared_ptr<void> &ownership, const MediaInfo &info, const ProtocolOption &option);
private: private:
bool _simulcast = false; bool _simulcast = false;

View File

@ -378,6 +378,12 @@ void WebRtcTransport::setRemoteDtlsFingerprint(const RtcSession &remote) {
} }
void WebRtcTransport::onRtcConfigure(RtcConfigure &configure) const { void WebRtcTransport::onRtcConfigure(RtcConfigure &configure) const {
SdpAttrFingerprint fingerprint;
fingerprint.algorithm = _offer_sdp->media[0].fingerprint.algorithm;
fingerprint.hash = getFingerprint(fingerprint.algorithm, _dtls_transport);
configure.setDefaultSetting(
_ice_server->GetUsernameFragment(), _ice_server->GetPassword(), RtpDirection::sendrecv, fingerprint);
// 开启remb后关闭twcc因为开启twcc后remb无效 // 开启remb后关闭twcc因为开启twcc后remb无效
GET_CONFIG(size_t, remb_bit_rate, Rtc::kRembBitRate); GET_CONFIG(size_t, remb_bit_rate, Rtc::kRembBitRate);
configure.enableTWCC(!remb_bit_rate); configure.enableTWCC(!remb_bit_rate);
@ -407,12 +413,7 @@ std::string WebRtcTransport::getAnswerSdp(const string &offer) {
setRemoteDtlsFingerprint(*_offer_sdp); setRemoteDtlsFingerprint(*_offer_sdp);
//// sdp 配置 //// //// sdp 配置 ////
SdpAttrFingerprint fingerprint;
fingerprint.algorithm = _offer_sdp->media[0].fingerprint.algorithm;
fingerprint.hash = getFingerprint(fingerprint.algorithm, _dtls_transport);
RtcConfigure configure; RtcConfigure configure;
configure.setDefaultSetting(
_ice_server->GetUsernameFragment(), _ice_server->GetPassword(), RtpDirection::sendrecv, fingerprint);
onRtcConfigure(configure); onRtcConfigure(configure);
//// 生成answer sdp //// //// 生成answer sdp ////
@ -431,10 +432,6 @@ static bool isDtls(char *buf) {
return ((*buf > 19) && (*buf < 64)); return ((*buf > 19) && (*buf < 64));
} }
static string getPeerAddress(RTC::TransportTuple *tuple) {
return tuple->get_peer_ip();
}
void WebRtcTransport::inputSockData(char *buf, int len, RTC::TransportTuple *tuple) { void WebRtcTransport::inputSockData(char *buf, int len, RTC::TransportTuple *tuple) {
if (RTC::StunPacket::IsStun((const uint8_t *)buf, len)) { if (RTC::StunPacket::IsStun((const uint8_t *)buf, len)) {
std::unique_ptr<RTC::StunPacket> packet(RTC::StunPacket::Parse((const uint8_t *)buf, len)); std::unique_ptr<RTC::StunPacket> packet(RTC::StunPacket::Parse((const uint8_t *)buf, len));
@ -451,7 +448,7 @@ void WebRtcTransport::inputSockData(char *buf, int len, RTC::TransportTuple *tup
} }
if (isRtp(buf, len)) { if (isRtp(buf, len)) {
if (!_srtp_session_recv) { if (!_srtp_session_recv) {
WarnL << "received rtp packet when dtls not completed from:" << getPeerAddress(tuple); WarnL << "received rtp packet when dtls not completed from:" << tuple->get_peer_ip();
return; return;
} }
if (_srtp_session_recv->DecryptSrtp((uint8_t *)buf, &len)) { if (_srtp_session_recv->DecryptSrtp((uint8_t *)buf, &len)) {
@ -461,7 +458,7 @@ void WebRtcTransport::inputSockData(char *buf, int len, RTC::TransportTuple *tup
} }
if (isRtcp(buf, len)) { if (isRtcp(buf, len)) {
if (!_srtp_session_recv) { if (!_srtp_session_recv) {
WarnL << "received rtcp packet when dtls not completed from:" << getPeerAddress(tuple); WarnL << "received rtcp packet when dtls not completed from:" << tuple->get_peer_ip();
return; return;
} }
if (_srtp_session_recv->DecryptSrtcp((uint8_t *)buf, &len)) { if (_srtp_session_recv->DecryptSrtcp((uint8_t *)buf, &len)) {
@ -533,8 +530,7 @@ void WebRtcTransportImp::OnDtlsTransportApplicationDataReceived(const RTC::DtlsT
#endif #endif
} }
WebRtcTransportImp::WebRtcTransportImp(const EventPoller::Ptr &poller,bool preferred_tcp) WebRtcTransportImp::WebRtcTransportImp(const EventPoller::Ptr &poller) : WebRtcTransport(poller) {
: WebRtcTransport(poller), _preferred_tcp(preferred_tcp) {
InfoL << getIdentifier(); InfoL << getIdentifier();
} }
@ -674,7 +670,7 @@ void WebRtcTransportImp::onCheckAnswer(RtcSession &sdp) {
}); });
for (auto &m : sdp.media) { for (auto &m : sdp.media) {
m.addr.reset(); m.addr.reset();
m.addr.address = extern_ips.empty() ? _localIp.empty() ? SockUtil::get_local_ip() : _localIp : extern_ips[0]; m.addr.address = extern_ips.empty() ? _local_ip.empty() ? SockUtil::get_local_ip() : _local_ip : extern_ips[0];
m.rtcp_addr.reset(); m.rtcp_addr.reset();
m.rtcp_addr.address = m.addr.address; m.rtcp_addr.address = m.addr.address;
@ -769,7 +765,7 @@ void WebRtcTransportImp::onRtcConfigure(RtcConfigure &configure) const {
return ret; return ret;
}); });
if (extern_ips.empty()) { if (extern_ips.empty()) {
std::string local_ip = _localIp.empty() ? SockUtil::get_local_ip() : _localIp; std::string local_ip = _local_ip.empty() ? SockUtil::get_local_ip() : _local_ip;
if (local_udp_port) { configure.addCandidate(*makeIceCandidate(local_ip, local_udp_port, 120, "udp")); } if (local_udp_port) { configure.addCandidate(*makeIceCandidate(local_ip, local_udp_port, 120, "udp")); }
if (local_tcp_port) { configure.addCandidate(*makeIceCandidate(local_ip, local_tcp_port, _preferred_tcp ? 125 : 115, "tcp")); } if (local_tcp_port) { configure.addCandidate(*makeIceCandidate(local_ip, local_tcp_port, _preferred_tcp ? 125 : 115, "tcp")); }
} else { } else {
@ -783,12 +779,16 @@ void WebRtcTransportImp::onRtcConfigure(RtcConfigure &configure) const {
} }
} }
void WebRtcTransportImp::setIceCandidate(vector<SdpAttrCandidate> cands) { void WebRtcTransportImp::setPreferredTcp(bool flag) {
_cands = std::move(cands); _preferred_tcp = flag;
} }
void WebRtcTransportImp::setLocalIp(const std::string &localIp) { void WebRtcTransportImp::setLocalIp(std::string local_ip) {
_localIp = localIp; _local_ip = std::move(local_ip);
}
void WebRtcTransportImp::setIceCandidate(vector<SdpAttrCandidate> cands) {
_cands = std::move(cands);
} }
/////////////////////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////////
@ -1278,21 +1278,14 @@ void WebRtcPluginManager::registerPlugin(const string &type, Plugin cb) {
_map_creator[type] = std::move(cb); _map_creator[type] = std::move(cb);
} }
std::string exchangeSdp(const WebRtcInterface &exchanger, const std::string& offer) {
return const_cast<WebRtcInterface &>(exchanger).getAnswerSdp(offer);
}
void setLocalIp(const WebRtcInterface& exchanger, const std::string& localIp) {
return const_cast<WebRtcInterface &>(exchanger).setLocalIp(localIp);
}
void WebRtcPluginManager::setListener(Listener cb) { void WebRtcPluginManager::setListener(Listener cb) {
lock_guard<mutex> lck(_mtx_creator); lock_guard<mutex> lck(_mtx_creator);
_listener = std::move(cb); _listener = std::move(cb);
} }
void WebRtcPluginManager::getAnswerSdp(Session &sender, const string &type, const WebRtcArgs &args, const onCreateRtc &cb_in) { void WebRtcPluginManager::negotiateSdp(Session &sender, const string &type, const WebRtcArgs &args, const onCreateWebRtc &cb_in) {
onCreateRtc cb; onCreateWebRtc cb;
lock_guard<mutex> lck(_mtx_creator); lock_guard<mutex> lck(_mtx_creator);
if (_listener) { if (_listener) {
auto listener = _listener; auto listener = _listener;
@ -1308,21 +1301,19 @@ void WebRtcPluginManager::getAnswerSdp(Session &sender, const string &type, cons
auto it = _map_creator.find(type); auto it = _map_creator.find(type);
if (it == _map_creator.end()) { if (it == _map_creator.end()) {
cb(WebRtcException(SockException(Err_other, "the type can not supported"))); cb_in(WebRtcException(SockException(Err_other, "the type can not supported")));
return; return;
} }
it->second(sender, args, cb); it->second(sender, args, cb);
} }
void echo_plugin(Session &sender, const WebRtcArgs &args, const WebRtcPluginManager::onCreateRtc &cb) { void echo_plugin(Session &sender, const WebRtcArgs &args, const onCreateWebRtc &cb) {
cb(*WebRtcEchoTest::create(EventPollerPool::Instance().getPoller())); cb(*WebRtcEchoTest::create(EventPollerPool::Instance().getPoller()));
} }
void push_plugin(Session &sender, const WebRtcArgs &args, const WebRtcPluginManager::onCreateRtc &cb) { void push_plugin(Session &sender, const WebRtcArgs &args, const onCreateWebRtc &cb) {
MediaInfo info(args["url"]); MediaInfo info(args["url"]);
bool preferred_tcp = args["preferred_tcp"]; Broadcast::PublishAuthInvoker invoker = [cb, info](const string &err, const ProtocolOption &option) mutable {
Broadcast::PublishAuthInvoker invoker = [cb, info, preferred_tcp](const string &err, const ProtocolOption &option) mutable {
if (!err.empty()) { if (!err.empty()) {
cb(WebRtcException(SockException(Err_other, err))); cb(WebRtcException(SockException(Err_other, err)));
return; return;
@ -1361,7 +1352,7 @@ void push_plugin(Session &sender, const WebRtcArgs &args, const WebRtcPluginMana
push_src_ownership = push_src->getOwnership(); push_src_ownership = push_src->getOwnership();
push_src->setProtocolOption(option); push_src->setProtocolOption(option);
} }
auto rtc = WebRtcPusher::create(EventPollerPool::Instance().getPoller(), push_src, push_src_ownership, info, option, preferred_tcp); auto rtc = WebRtcPusher::create(EventPollerPool::Instance().getPoller(), push_src, push_src_ownership, info, option);
push_src->setListener(rtc); push_src->setListener(rtc);
cb(*rtc); cb(*rtc);
}; };
@ -1374,12 +1365,10 @@ void push_plugin(Session &sender, const WebRtcArgs &args, const WebRtcPluginMana
} }
} }
void play_plugin(Session &sender, const WebRtcArgs &args, const WebRtcPluginManager::onCreateRtc &cb) { void play_plugin(Session &sender, const WebRtcArgs &args, const onCreateWebRtc &cb) {
MediaInfo info(args["url"]); MediaInfo info(args["url"]);
bool preferred_tcp = args["preferred_tcp"];
auto session_ptr = static_pointer_cast<Session>(sender.shared_from_this()); auto session_ptr = static_pointer_cast<Session>(sender.shared_from_this());
Broadcast::AuthInvoker invoker = [cb, info, session_ptr, preferred_tcp](const string &err) mutable { Broadcast::AuthInvoker invoker = [cb, info, session_ptr](const string &err) mutable {
if (!err.empty()) { if (!err.empty()) {
cb(WebRtcException(SockException(Err_other, err))); cb(WebRtcException(SockException(Err_other, err)));
return; return;
@ -1395,7 +1384,7 @@ void play_plugin(Session &sender, const WebRtcArgs &args, const WebRtcPluginMana
} }
// 还原成rtc目的是为了hook时识别哪种播放协议 // 还原成rtc目的是为了hook时识别哪种播放协议
info.schema = "rtc"; info.schema = "rtc";
auto rtc = WebRtcPlayer::create(EventPollerPool::Instance().getPoller(), src, info, preferred_tcp); auto rtc = WebRtcPlayer::create(EventPollerPool::Instance().getPoller(), src, info);
cb(*rtc); cb(*rtc);
}); });
}; };
@ -1408,14 +1397,35 @@ void play_plugin(Session &sender, const WebRtcArgs &args, const WebRtcPluginMana
} }
} }
static void set_webrtc_cands(const WebRtcArgs &args, const WebRtcInterface &rtc) { static void setWebRtcArgs(const WebRtcArgs &args, WebRtcInterface &rtc) {
{
static auto is_vaild_ip = [](const std::string &ip) -> bool {
int a, b, c, d;
return sscanf(ip.c_str(), "%d.%d.%d.%d", &a, &b, &c, &d) == 4;
};
std::string host = args["Host"];
if (!host.empty()) {
auto local_ip = host.substr(0, host.find(':'));
if (!is_vaild_ip(local_ip) || local_ip == "127.0.0.1") {
local_ip = "";
}
rtc.setLocalIp(std::move(local_ip));
}
}
bool preferred_tcp = args["preferred_tcp"];
{
rtc.setPreferredTcp(preferred_tcp);
}
{
vector<SdpAttrCandidate> cands; vector<SdpAttrCandidate> cands;
{ {
auto cand_str = trim(args["cand_udp"]); auto cand_str = trim(args["cand_udp"]);
auto ip_port = toolkit::split(cand_str, ":"); auto ip_port = toolkit::split(cand_str, ":");
if (ip_port.size() == 2) { if (ip_port.size() == 2) {
// udp优先 // udp优先
auto ice_cand = makeIceCandidate(ip_port[0], atoi(ip_port[1].data()), 120, "udp"); auto ice_cand = makeIceCandidate(ip_port[0], atoi(ip_port[1].data()), preferred_tcp ? 100 : 120, "udp");
cands.emplace_back(std::move(*ice_cand)); cands.emplace_back(std::move(*ice_cand));
} }
} }
@ -1424,23 +1434,26 @@ static void set_webrtc_cands(const WebRtcArgs &args, const WebRtcInterface &rtc)
auto ip_port = toolkit::split(cand_str, ":"); auto ip_port = toolkit::split(cand_str, ":");
if (ip_port.size() == 2) { if (ip_port.size() == 2) {
// tcp模式 // tcp模式
auto ice_cand = makeIceCandidate(ip_port[0], atoi(ip_port[1].data()), 100, "tcp"); auto ice_cand = makeIceCandidate(ip_port[0], atoi(ip_port[1].data()), preferred_tcp ? 120 : 100, "tcp");
cands.emplace_back(std::move(*ice_cand)); cands.emplace_back(std::move(*ice_cand));
} }
} }
if (!cands.empty()) { if (!cands.empty()) {
// udp优先 // udp优先
const_cast<WebRtcInterface &>(rtc).setIceCandidate(std::move(cands)); rtc.setIceCandidate(std::move(cands));
}
} }
} }
static onceToken s_rtc_auto_register([]() { static onceToken s_rtc_auto_register([]() {
#if !defined (NDEBUG)
// debug模式才开启echo插件
WebRtcPluginManager::Instance().registerPlugin("echo", echo_plugin); WebRtcPluginManager::Instance().registerPlugin("echo", echo_plugin);
#endif
WebRtcPluginManager::Instance().registerPlugin("push", push_plugin); WebRtcPluginManager::Instance().registerPlugin("push", push_plugin);
WebRtcPluginManager::Instance().registerPlugin("play", play_plugin); WebRtcPluginManager::Instance().registerPlugin("play", play_plugin);
WebRtcPluginManager::Instance().setListener([](Session &sender, const std::string &type, const WebRtcArgs &args, const WebRtcInterface &rtc) { WebRtcPluginManager::Instance().setListener([](Session &sender, const std::string &type, const WebRtcArgs &args, const WebRtcInterface &rtc) {
set_webrtc_cands(args, rtc); setWebRtcArgs(args, const_cast<WebRtcInterface&>(rtc));
}); });
}); });

View File

@ -42,13 +42,10 @@ public:
virtual const std::string& getIdentifier() const = 0; virtual const std::string& getIdentifier() const = 0;
virtual const std::string& deleteRandStr() const { static std::string s_null; return s_null; } virtual const std::string& deleteRandStr() const { static std::string s_null; return s_null; }
virtual void setIceCandidate(std::vector<SdpAttrCandidate> cands) {} virtual void setIceCandidate(std::vector<SdpAttrCandidate> cands) {}
virtual void setLocalIp(const std::string &localIp) {} virtual void setLocalIp(std::string localIp) {}
virtual void setPreferredTcp(bool flag) {}
}; };
std::string exchangeSdp(const WebRtcInterface &exchanger, const std::string& offer);
void setLocalIp(const WebRtcInterface &exchanger, const std::string &localIp);
class WebRtcException : public WebRtcInterface { class WebRtcException : public WebRtcInterface {
public: public:
WebRtcException(const SockException &ex) : _ex(ex) {}; WebRtcException(const SockException &ex) : _ex(ex) {};
@ -88,7 +85,7 @@ public:
* @param offer offer sdp * @param offer offer sdp
* @return answer sdp * @return answer sdp
*/ */
std::string getAnswerSdp(const std::string &offer) override; std::string getAnswerSdp(const std::string &offer) override final;
/** /**
* id * id
@ -252,14 +249,16 @@ public:
void onSendRtp(const RtpPacket::Ptr &rtp, bool flush, bool rtx = false); void onSendRtp(const RtpPacket::Ptr &rtp, bool flush, bool rtx = false);
void createRtpChannel(const std::string &rid, uint32_t ssrc, MediaTrack &track); void createRtpChannel(const std::string &rid, uint32_t ssrc, MediaTrack &track);
void setIceCandidate(std::vector<SdpAttrCandidate> cands) override;
void removeTuple(RTC::TransportTuple* tuple); void removeTuple(RTC::TransportTuple* tuple);
void safeShutdown(const SockException &ex); void safeShutdown(const SockException &ex);
void setLocalIp(const std::string &localIp) override; void setPreferredTcp(bool flag) override;
void setLocalIp(std::string local_ip) override;
void setIceCandidate(std::vector<SdpAttrCandidate> cands) override;
protected: protected:
void OnIceServerSelectedTuple(const RTC::IceServer *iceServer, RTC::TransportTuple *tuple) override; void OnIceServerSelectedTuple(const RTC::IceServer *iceServer, RTC::TransportTuple *tuple) override;
WebRtcTransportImp(const EventPoller::Ptr &poller,bool preferred_tcp = false); WebRtcTransportImp(const EventPoller::Ptr &poller);
void OnDtlsTransportApplicationDataReceived(const RTC::DtlsTransport *dtlsTransport, const uint8_t *data, size_t len) override; void OnDtlsTransportApplicationDataReceived(const RTC::DtlsTransport *dtlsTransport, const uint8_t *data, size_t len) override;
void onStartWebRTC() override; void onStartWebRTC() override;
void onSendSockData(Buffer::Ptr buf, bool flush = true, RTC::TransportTuple *tuple = nullptr) override; void onSendSockData(Buffer::Ptr buf, bool flush = true, RTC::TransportTuple *tuple = nullptr) override;
@ -273,7 +272,7 @@ protected:
void onCreate() override; void onCreate() override;
void onDestory() override; void onDestory() override;
void onShutdown(const SockException &ex) override; void onShutdown(const SockException &ex) override;
virtual void onRecvRtp(MediaTrack &track, const std::string &rid, RtpPacket::Ptr rtp) = 0; virtual void onRecvRtp(MediaTrack &track, const std::string &rid, RtpPacket::Ptr rtp) {}
void updateTicker(); void updateTicker();
float getLossRate(TrackType type); float getLossRate(TrackType type);
void onRtcpBye() override; void onRtcpBye() override;
@ -289,7 +288,7 @@ private:
void onCheckAnswer(RtcSession &sdp); void onCheckAnswer(RtcSession &sdp);
private: private:
bool _preferred_tcp; bool _preferred_tcp = false;
uint16_t _rtx_seq[2] = {0, 0}; uint16_t _rtx_seq[2] = {0, 0};
//用掉的总流量 //用掉的总流量
uint64_t _bytes_usage = 0; uint64_t _bytes_usage = 0;
@ -310,8 +309,8 @@ private:
//根据接收rtp的pt获取相关信息 //根据接收rtp的pt获取相关信息
std::unordered_map<uint8_t/*pt*/, std::unique_ptr<WrappedMediaTrack>> _pt_to_track; std::unordered_map<uint8_t/*pt*/, std::unique_ptr<WrappedMediaTrack>> _pt_to_track;
std::vector<SdpAttrCandidate> _cands; std::vector<SdpAttrCandidate> _cands;
//源访问的hostip //http访问时的host ip
std::string _localIp; std::string _local_ip;
}; };
class WebRtcTransportManager { class WebRtcTransportManager {
@ -333,21 +332,20 @@ private:
class WebRtcArgs : public std::enable_shared_from_this<WebRtcArgs> { class WebRtcArgs : public std::enable_shared_from_this<WebRtcArgs> {
public: public:
virtual ~WebRtcArgs() = default; virtual ~WebRtcArgs() = default;
virtual variant operator[](const std::string &key) const = 0; virtual variant operator[](const std::string &key) const = 0;
}; };
using onCreateWebRtc = std::function<void(const WebRtcInterface &rtc)>;
class WebRtcPluginManager { class WebRtcPluginManager {
public: public:
using onCreateRtc = std::function<void(const WebRtcInterface &rtc)>; using Plugin = std::function<void(Session &sender, const WebRtcArgs &args, const onCreateWebRtc &cb)>;
using Plugin = std::function<void(Session &sender, const WebRtcArgs &args, const onCreateRtc &cb)>;
using Listener = std::function<void(Session &sender, const std::string &type, const WebRtcArgs &args, const WebRtcInterface &rtc)>; using Listener = std::function<void(Session &sender, const std::string &type, const WebRtcArgs &args, const WebRtcInterface &rtc)>;
static WebRtcPluginManager &Instance(); static WebRtcPluginManager &Instance();
void registerPlugin(const std::string &type, Plugin cb); void registerPlugin(const std::string &type, Plugin cb);
void getAnswerSdp(Session &sender, const std::string &type, const WebRtcArgs &args, const onCreateRtc &cb);
void setListener(Listener cb); void setListener(Listener cb);
void negotiateSdp(Session &sender, const std::string &type, const WebRtcArgs &args, const onCreateWebRtc &cb);
private: private:
WebRtcPluginManager() = default; WebRtcPluginManager() = default;