mirror of https://git.ffmpeg.org/ffmpeg.git
rtmp: Check the buffer length of ping packets
Signed-off-by: Martin Storsjö <martin@martin.st>
This commit is contained in:
parent
e49e6b6451
commit
8ea1459bc3
|
@ -515,6 +515,12 @@ static int gen_pong(URLContext *s, RTMPContext *rt, RTMPPacket *ppkt)
|
|||
uint8_t *p;
|
||||
int ret;
|
||||
|
||||
if (ppkt->data_size < 6) {
|
||||
av_log(s, AV_LOG_ERROR, "Too short ping packet (%d)\n",
|
||||
ppkt->data_size);
|
||||
return AVERROR_INVALIDDATA;
|
||||
}
|
||||
|
||||
if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_PING,
|
||||
ppkt->timestamp + 1, 6)) < 0)
|
||||
return ret;
|
||||
|
@ -913,6 +919,12 @@ static int handle_ping(URLContext *s, RTMPPacket *pkt)
|
|||
RTMPContext *rt = s->priv_data;
|
||||
int t, ret;
|
||||
|
||||
if (pkt->data_size < 2) {
|
||||
av_log(s, AV_LOG_ERROR, "Too short ping packet (%d)\n",
|
||||
pkt->data_size);
|
||||
return AVERROR_INVALIDDATA;
|
||||
}
|
||||
|
||||
t = AV_RB16(pkt->data);
|
||||
if (t == 6) {
|
||||
if ((ret = gen_pong(s, rt, pkt)) < 0)
|
||||
|
|
Loading…
Reference in New Issue