【zlm】 webrtc源码讲解

前端WEB

服务器收到请求

服务端的处理

播放

拉流

参考文章

前端WEB

服务器收到请求

POST /index/api/webrtc?app=live&stream=test&type=play HTTP/1.1

HttpSession::onRecvHeaderHttpSession::Handle_Req_POSTHttpSession::Handle_Req_POSTif (totalContentLen > 0 && (size_t)totalContentLen < maxReqSize )_contentCallBack = [this,parserCopy](const char *data,size_t len) {//恢复http头_parser = parserCopy;//设置content_parser.setContent(string(data,len));//触发http事件，emitHttpEvent内部会选择是否关闭连接emitHttpEvent(true);//清空数据,节省内存_parser.Clear();//content已经接收完毕return false;};HttpSession::onRecvContent(const char *data,size_t len)if (_contentCallBack)_contentCallBack(data,len);        HttpSession::emitHttpEvent// 广播HTTP事件NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastHttpRequest,_parser,invoker,consumed,static_cast<SockInfo &>(*this));

服务端的处理

// 主函数中调用web接口安装函数
installWebApiaddHttpListener();api_regist("/index/api/webrtc",[](API_ARGS_STRING_ASYNC){auto type = allArgs["type"];auto offer = allArgs.getArgs();WebRtcPluginManager::Instance().getAnswerSdp(*(static_cast<Session *>(&sender)), type,WebRtcArgsImp(allArgs, sender.getIdentifier()),[invoker, val, offer, headerOut](const WebRtcInterface &exchanger) mutable {headerOut["Content-Type"] = HttpFileManager::getContentType(".json");headerOut["Access-Control-Allow-Origin"] = "*";val["sdp"] = const_cast<WebRtcInterface &>(exchanger).getAnswerSdp(offer);val["id"] = exchanger.getIdentifier();val["type"] = "answer";invoker(200, headerOut, val.toStyledString());});});addHttpListener//注册监听kBroadcastHttpRequest事件NoticeCenter::Instance().addListener(&web_api_tag, Broadcast::kBroadcastHttpRequest,[](BroadcastHttpRequestArgs) {auto it = s_map_api.find(parser.Url());it->second(parser, invoker, sender);}

根据url找到对应的事件回调，最终会调用WebRtcPluginManager::Instance().getAnswerSdp。

WebRtcPluginManager::getAnswerSdpauto it = _map_creator.find(type);it->second(sender, args, cb);// 静态注册插件
WebRtcPluginManager::Instance().registerPlugin("play", play_plugin);void play_plugin(Session &sender, const WebRtcArgs &args, const WebRtcPluginManager::onCreateRtc &cb)// 使用rtsp媒体源，两者均是传输的rtp流info._schema = RTSP_SCHEMA;MediaSource::findAsync(info, session_ptr, [=](const MediaSource::Ptr &src_in) mutable {auto src = dynamic_pointer_cast<RtspMediaSource>(src_in);// 还原成rtc，目的是为了hook时识别哪种播放协议info._schema = RTC_SCHEMA;auto rtc = WebRtcPlayer::create(EventPollerPool::Instance().getPoller(), src, info, preferred_tcp);cb(*rtc); // 发送answer SDP给web端});

播放

拉流

Web端首先根据协商的IP和端口，服务端webrtc的端口是8000，发送STUN命令再次获取STUN地址。
首次连接，服务端会创建对应的session。

WebRtcSession::WebRtcSession(const Socket::Ptr &sock) : Session(sock)
socklen_t addr_len = sizeof(_peer_addr);
getpeername(sock->rawFD(), (struct sockaddr *)&_peer_addr, &addr_len);

WebRtcSession::onRecv_l(const char *data, size_t len)
// 首次进入，根据username获取之前创建的transport.
auto user_name = getUserName(data, len); // 此处的username就是之前设置的transport标识
auto transport = WebRtcTransportManager::Instance().getItem(user_name);
transport->setSession(shared_from_this());
_transport = std::move(transport);
_transport->inputSockData((char *)data, len, (struct sockaddr *)&_peer_addr);


WebRtcTransport::inputSockData
// 处理STUN消息
if (RTC::StunPacket::IsStun((const uint8_t *)buf, len))
std::unique_ptr<RTC::StunPacket> packet(RTC::StunPacket::Parse((const uint8_t *)buf, len));
_ice_server->ProcessStunPacket(packet.get(), tuple);
return;

// 处理
if (is_dtls(buf))
_dtls_transport->ProcessDtlsData((uint8_t *)buf, len);
return;

// 由于是拉流，不存在rtp数据，但是有rtcp数据
if (is_rtcp(buf))
if (_srtp_session_recv->DecryptSrtcp((uint8_t *)buf, &len))
onRtcp(buf, len);

DTLS交互完成后，接下来启动媒体传输

WebRtcTransport::OnDtlsTransportConnected
onStartWebRTC();

WebRtcPlayer::onStartWebRTC
WebRtcTransportImp::onStartWebRTC();
_reader = _play_src->getRing()->attach(getPoller(), true);
weak_ptr<WebRtcPlayer> weak_self = static_pointer_cast<WebRtcPlayer>(shared_from_this());
weak_ptr<Session> weak_session = getSession();
_reader->setReadCB([weak_self](const RtspMediaSource::RingDataType &pkt) {
size_t i = 0;
pkt->for_each([&](const RtpPacket::Ptr &rtp) {
strong_self->onSendRtp(rtp, ++i == pkt->size());
});
});