ZLMediaKit/src/Extension/AACRtmp.cpp

139 lines
3.9 KiB
C++
Raw Normal View History

2018-10-25 10:00:17 +08:00
/*
2020-04-04 20:30:09 +08:00
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
2018-10-25 10:00:17 +08:00
*
* This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
*
2020-04-04 20:30:09 +08:00
* Use of this source code is governed by MIT 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.
2018-10-25 10:00:17 +08:00
*/
2020-04-04 20:30:09 +08:00
2019-06-28 17:37:11 +08:00
#include "AACRtmp.h"
2020-04-04 22:54:49 +08:00
#include "Rtmp/Rtmp.h"
2018-10-24 14:21:59 +08:00
2018-10-24 17:17:55 +08:00
namespace mediakit{
2018-10-24 14:21:59 +08:00
2020-04-04 22:54:49 +08:00
static string getAacCfg(const RtmpPacket &thiz) {
string ret;
if (thiz.getMediaType() != FLV_CODEC_AAC) {
return ret;
}
if (!thiz.isCfgFrame()) {
return ret;
}
2020-08-30 10:48:34 +08:00
if (thiz.buffer.size() < 4) {
2020-04-04 22:54:49 +08:00
WarnL << "bad aac cfg!";
return ret;
}
2020-08-30 10:48:34 +08:00
ret = thiz.buffer.substr(2);
2020-04-04 22:54:49 +08:00
return ret;
}
2020-05-11 23:25:12 +08:00
bool AACRtmpDecoder::inputRtmp(const RtmpPacket::Ptr &pkt, bool) {
2018-10-24 14:21:59 +08:00
if (pkt->isCfgFrame()) {
2020-04-04 22:54:49 +08:00
_aac_cfg = getAacCfg(*pkt);
2020-06-30 21:08:52 +08:00
onGetAAC(nullptr, 0, 0);
return false;
2018-10-24 14:21:59 +08:00
}
2020-06-30 21:08:52 +08:00
2018-10-24 14:21:59 +08:00
if (!_aac_cfg.empty()) {
2020-08-30 10:48:34 +08:00
onGetAAC(pkt->buffer.data() + 2, pkt->buffer.size() - 2, pkt->time_stamp);
2018-10-24 14:21:59 +08:00
}
return false;
}
2020-05-11 23:25:12 +08:00
void AACRtmpDecoder::onGetAAC(const char* data, int len, uint32_t stamp) {
2020-08-01 10:22:12 +08:00
auto frame = ResourcePoolHelper<FrameImp>::obtainObj();
frame->_codec_id = CodecAAC;
2020-06-11 19:21:46 +08:00
//生成adts头
char adts_header[32] = {0};
auto size = dumpAacConfig(_aac_cfg, len, (uint8_t *) adts_header, sizeof(adts_header));
if (size > 0) {
frame->_buffer.assign(adts_header, size);
frame->_prefix_size = size;
} else {
frame->_buffer.clear();
frame->_prefix_size = 0;
}
2020-06-30 21:11:59 +08:00
if(len > 0){
//追加负载数据
frame->_buffer.append(data, len);
frame->_dts = stamp;
}
2018-10-24 14:21:59 +08:00
2020-06-30 21:11:59 +08:00
if(size > 0 || len > 0){
//有adts头或者实际aac负载
RtmpCodec::inputFrame(frame);
}
2018-10-24 14:21:59 +08:00
}
2020-05-11 23:25:12 +08:00
2018-10-24 14:21:59 +08:00
/////////////////////////////////////////////////////////////////////////////////////
2020-06-11 19:21:46 +08:00
AACRtmpEncoder::AACRtmpEncoder(const Track::Ptr &track) {
_track = dynamic_pointer_cast<AACTrack>(track);
}
void AACRtmpEncoder::makeConfigPacket() {
if (_track && _track->ready()) {
//从track中和获取aac配置信息
_aac_cfg = _track->getAacCfg();
}
if (!_aac_cfg.empty()) {
makeAudioConfigPkt();
}
}
2018-10-24 14:21:59 +08:00
void AACRtmpEncoder::inputFrame(const Frame::Ptr &frame) {
if (_aac_cfg.empty()) {
2020-06-11 19:21:46 +08:00
if (frame->prefixSize()) {
//包含adts头,从adts头获取aac配置信息
2020-06-11 19:21:46 +08:00
_aac_cfg = makeAacConfig((uint8_t *) (frame->data()), frame->prefixSize());
}
makeConfigPacket();
2018-10-24 14:21:59 +08:00
}
2018-10-24 14:44:36 +08:00
if(!_aac_cfg.empty()){
RtmpPacket::Ptr rtmpPkt = ResourcePoolHelper<RtmpPacket>::obtainObj();
2020-08-30 10:48:34 +08:00
rtmpPkt->buffer.clear();
2018-10-24 17:50:09 +08:00
2020-04-18 18:46:20 +08:00
//header
2018-10-24 14:44:36 +08:00
uint8_t is_config = false;
2020-08-30 10:48:34 +08:00
rtmpPkt->buffer.push_back(_audio_flv_flags);
rtmpPkt->buffer.push_back(!is_config);
2020-04-18 18:46:20 +08:00
//aac data
2020-08-30 10:48:34 +08:00
rtmpPkt->buffer.append(frame->data() + frame->prefixSize(), frame->size() - frame->prefixSize());
2018-10-24 14:44:36 +08:00
2020-08-30 10:48:34 +08:00
rtmpPkt->body_size = rtmpPkt->buffer.size();
rtmpPkt->chunk_id = CHUNK_AUDIO;
rtmpPkt->stream_index = STREAM_MEDIA;
rtmpPkt->time_stamp = frame->dts();
rtmpPkt->type_id = MSG_AUDIO;
2018-10-26 18:40:31 +08:00
RtmpCodec::inputRtmp(rtmpPkt, false);
2018-10-24 14:44:36 +08:00
}
2018-10-24 14:21:59 +08:00
}
void AACRtmpEncoder::makeAudioConfigPkt() {
2020-04-18 22:13:11 +08:00
_audio_flv_flags = getAudioRtmpFlags(std::make_shared<AACTrack>(_aac_cfg));
2018-10-24 14:21:59 +08:00
RtmpPacket::Ptr rtmpPkt = ResourcePoolHelper<RtmpPacket>::obtainObj();
2020-08-30 10:48:34 +08:00
rtmpPkt->buffer.clear();
2018-10-24 17:50:09 +08:00
2020-04-18 18:46:20 +08:00
//header
2018-10-24 14:21:59 +08:00
uint8_t is_config = true;
2020-08-30 10:48:34 +08:00
rtmpPkt->buffer.push_back(_audio_flv_flags);
rtmpPkt->buffer.push_back(!is_config);
2020-04-18 18:46:20 +08:00
//aac config
2020-08-30 10:48:34 +08:00
rtmpPkt->buffer.append(_aac_cfg);
2018-10-24 14:21:59 +08:00
2020-08-30 10:48:34 +08:00
rtmpPkt->body_size = rtmpPkt->buffer.size();
rtmpPkt->chunk_id = CHUNK_AUDIO;
rtmpPkt->stream_index = STREAM_MEDIA;
rtmpPkt->time_stamp = 0;
rtmpPkt->type_id = MSG_AUDIO;
2018-10-26 18:40:31 +08:00
RtmpCodec::inputRtmp(rtmpPkt, false);
2018-10-24 14:21:59 +08:00
}
2018-10-24 17:17:55 +08:00
}//namespace mediakit