From ca83987f14690905401ef0dc9ded9ee47e199b88 Mon Sep 17 00:00:00 2001
From: xia-chu <771730766@qq.com>
Date: Sat, 9 Dec 2023 09:42:26 +0800
Subject: [PATCH] Fix buf

---
 src/Extension/AAC.cpp     | 3 ++-
 src/Record/MP4Demuxer.cpp | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/Extension/AAC.cpp b/src/Extension/AAC.cpp
index 01774b8b..e53c7ec3 100644
--- a/src/Extension/AAC.cpp
+++ b/src/Extension/AAC.cpp
@@ -280,7 +280,8 @@ static Frame::Ptr addADTSHeader(const Frame::Ptr &frame_in, const std::string &a
 
 bool AACTrack::inputFrame(const Frame::Ptr &frame) {
     if (!frame->prefixSize()) {
-        return inputFrame_l(addADTSHeader(frame, getExtraData()->toString()));
+        CHECK(ready());
+        return inputFrame_l(addADTSHeader(frame, _cfg));
     }
 
     bool ret = false;
diff --git a/src/Record/MP4Demuxer.cpp b/src/Record/MP4Demuxer.cpp
index 9b674924..49020bc0 100644
--- a/src/Record/MP4Demuxer.cpp
+++ b/src/Record/MP4Demuxer.cpp
@@ -67,11 +67,12 @@ void MP4Demuxer::onVideoTrack(uint32_t track, uint8_t object, int width, int hei
     }
 }
 
-void MP4Demuxer::onAudioTrack(uint32_t track_id, uint8_t object, int channel_count, int bit_per_sample, int sample_rate, const void *extra, size_t bytes) {
+void MP4Demuxer::onAudioTrack(uint32_t track, uint8_t object, int channel_count, int bit_per_sample, int sample_rate, const void *extra, size_t bytes) {
     auto audio = Factory::getTrackByCodecId(getCodecByMovId(object), sample_rate, channel_count, bit_per_sample / channel_count);
     if (!audio) {
         return;
     }
+    _track_to_codec.emplace(track, audio);
     if (extra && bytes) {
         audio->setExtraData((uint8_t *)extra, bytes);
     }