cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add check to avoid doing illegal memory access from an invalid iterator from std::prev()

Browse Source 
PiperOrigin-RevId: 573248334
This commit is contained in:
MediaPipe Team 2023-10-13 09:59:10 -07:00 committed by Copybara-Service
parent 652792ebaa
commit 8823046e4b
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mediapipe/framework/tool/switch_demux_calculator.cc
View File

@ -13,6 +13,7 @@
// limitations under the License.
#include <algorithm>
#include <iterator>
#include <memory>
#include <queue>
#include <set>
@ -223,8 +224,8 @@ absl::Status SwitchDemuxCalculator::RecordPackets(CalculatorContext* cc) {
// Returns the channel index for a Timestamp.
int SwitchDemuxCalculator::ChannelIndex(Timestamp timestamp) {
auto it = std::prev(channel_history_.upper_bound(timestamp));
return it->second;
auto it = channel_history_.upper_bound(timestamp);
return it == channel_history_.begin() ? -1 : std::prev(it)->second;
}
// Dispatches all queued input packets with known channels.
@ -237,10 +238,12 @@ absl::Status SwitchDemuxCalculator::SendActivePackets(CalculatorContext* cc) {
auto& queue = input_queue_[input_id];
while (!queue.empty() && queue.front().Timestamp() <= channel_settled) {
int channel_index = ChannelIndex(queue.front().Timestamp());
std::string output_tag = tool::ChannelTag(tag, channel_index);
auto output_id = cc->Outputs().GetId(output_tag, index);
if (output_id.IsValid()) {
cc->Outputs().Get(output_id).AddPacket(queue.front());
if (channel_index != -1) {
std::string output_tag = tool::ChannelTag(tag, channel_index);
auto output_id = cc->Outputs().GetId(output_tag, index);
if (output_id.IsValid()) {
cc->Outputs().Get(output_id).AddPacket(queue.front());
}
}
queue.pop();
}