From 40a09958c4d01f30a2231b77bea7096f82fc3155 Mon Sep 17 00:00:00 2001 From: runner365 Date: Sun, 1 Mar 2020 19:42:32 +0800 Subject: [PATCH] add checking for pes_parse --- trunk/src/srt/ts_demux.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/trunk/src/srt/ts_demux.cpp b/trunk/src/srt/ts_demux.cpp index ead598bbe..db486d5ad 100644 --- a/trunk/src/srt/ts_demux.cpp +++ b/trunk/src/srt/ts_demux.cpp @@ -265,7 +265,11 @@ int ts_demux::decode_unit(unsigned char* data_p, std::string key_path, TS_DATA_C //callback last media data in data buffer on_callback(callback, _last_pid, key_path, _last_dts, _last_pts); - pes_parse(data_p+npos, npos, &ret_data_p, ret_size, dts, pts); + int ret = pes_parse(data_p+npos, npos, &ret_data_p, ret_size, dts, pts); + assert(ret <= 188); + if (ret > 188) { + return -1; + } _last_pts = pts; _last_dts = (dts == 0) ? pts : dts;