rmenc: limit packet size

The chunk size is limited to UINT16_MAX (written by avio_wb16), so make
sure that the packet size is not too large.

Such large frames need to be split into slices smaller than 64 kB, but
that is currently supported neither by the rv10/rv20 encoders nor the rm
muxer.

Signed-off-by: Andreas Cadhalpun <Andreas.Cadhalpun@googlemail.com>
Signed-off-by: Anton Khirnov <anton@khirnov.net>
pull/122/head
Andreas Cadhalpun 10 years ago committed by Anton Khirnov
parent 4f2ee9daee
commit bb4edddd93
  1. 8
      libavformat/rmenc.c

@ -44,6 +44,10 @@ typedef struct RMMuxContext {
/* in ms */ /* in ms */
#define BUFFER_DURATION 0 #define BUFFER_DURATION 0
/* the header needs at most 7 + 4 + 12 B */
#define MAX_HEADER_SIZE (7 + 4 + 12)
/* UINT16_MAX is the maximal chunk size */
#define MAX_PACKET_SIZE (UINT16_MAX - MAX_HEADER_SIZE)
static void put_str(AVIOContext *s, const char *tag) static void put_str(AVIOContext *s, const char *tag)
@ -389,6 +393,10 @@ static int rm_write_video(AVFormatContext *s, const uint8_t *buf, int size, int
/* Well, I spent some time finding the meaning of these bits. I am /* Well, I spent some time finding the meaning of these bits. I am
not sure I understood everything, but it works !! */ not sure I understood everything, but it works !! */
#if 1 #if 1
if (size > MAX_PACKET_SIZE) {
avpriv_report_missing_feature(s, "Muxing packets larger than 64 kB");
return AVERROR(ENOSYS);
}
write_packet_header(s, stream, size + 7 + (size >= 0x4000)*4, key_frame); write_packet_header(s, stream, size + 7 + (size >= 0x4000)*4, key_frame);
/* bit 7: '1' if final packet of a frame converted in several packets */ /* bit 7: '1' if final packet of a frame converted in several packets */
avio_w8(pb, 0x81); avio_w8(pb, 0x81);

Loading…
Cancel
Save