@@ -1250,7 +1250,7 @@ static int cavs_decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
buf_end = buf + buf_size;
for(;;) {
buf_ptr = avpriv_find_start_code(buf_ptr, buf_end, &stc);
- if ((stc & 0xFFFFFE00) || buf_ptr == buf_end) {
+ if (!start_code_is_valid(stc) || buf_ptr == buf_end) {
if (!h->stc)
av_log(h->avctx, AV_LOG_WARNING, "no frame decoded\n");
return FFMAX(0, buf_ptr - buf);
@@ -150,7 +150,7 @@ static int cbs_mpeg2_split_fragment(CodedBitstreamContext *ctx,
start = avpriv_find_start_code(frag->data, frag->data + frag->data_size,
&start_code);
- if (start_code >> 8 != 0x000001) {
+ if (!start_code_is_valid(start_code)) {
// No start code found.
return AVERROR_INVALIDDATA;
}
@@ -172,7 +172,7 @@ static int cbs_mpeg2_split_fragment(CodedBitstreamContext *ctx,
// (may be the last byte of fragment->data); end points to the byte
// following the byte containing the start code identifier (or to
// the end of fragment->data).
- if (start_code >> 8 == 0x000001) {
+ if (start_code_is_valid(start_code)) {
// Unit runs from start to the beginning of the start code
// pointed to by end (including any padding zeroes).
unit_size = (end - 4) - start;
@@ -189,7 +189,7 @@ static int cbs_mpeg2_split_fragment(CodedBitstreamContext *ctx,
start = end;
// Do we have a further unit to add to the fragment?
- } while ((start_code >> 8) == 0x000001);
+ } while (start_code_is_valid(start_code));
return 0;
}
@@ -239,7 +239,7 @@ static int extract_extradata_vc1(AVBSFContext *ctx, AVPacket *pkt,
ptr = avpriv_find_start_code(ptr, end, &state);
if (state == VC1_CODE_SEQHDR || state == VC1_CODE_ENTRYPOINT) {
has_extradata = 1;
- } else if (has_extradata && IS_MARKER(state)) {
+ } else if (has_extradata && start_code_is_valid(state)) {
extradata_size = ptr - 4 - pkt->data;
break;
}
@@ -213,7 +213,7 @@ int ff_mpeg1_find_frame_end(ParseContext *pc, const uint8_t *buf, int buf_size,
pc->frame_start_found = 0;
if (pc->frame_start_found < 4 && state == EXT_START_CODE)
pc->frame_start_found++;
- if (pc->frame_start_found == 4 && (state & 0xFFFFFF00) == 0x100) {
+ if (pc->frame_start_found == 4 && start_code_is_valid(state)) {
if (state < SLICE_MIN_START_CODE || state > SLICE_MAX_START_CODE) {
pc->frame_start_found = 0;
pc->state = -1;
@@ -2475,7 +2475,7 @@ static int decode_chunks(AVCodecContext *avctx, AVFrame *picture,
/* find next start code */
uint32_t start_code = -1;
buf_ptr = avpriv_find_start_code(buf_ptr, buf_end, &start_code);
- if (start_code > 0x1ff) {
+ if (!start_code_is_valid(start_code)) {
if (!skip_frame) {
if (HAVE_THREADS &&
(avctx->active_thread_type & FF_THREAD_SLICE) &&
@@ -82,7 +82,7 @@ static int mpeg1_find_frame_end(ParseContext *pc, const uint8_t *buf,
pc->frame_start_found = 0;
if (pc->frame_start_found < 4 && state == EXT_START_CODE)
pc->frame_start_found++;
- if (pc->frame_start_found == 4 && (state & 0xFFFFFF00) == 0x100) {
+ if (pc->frame_start_found == 4 && start_code_is_valid(state)) {
if (state < SLICE_MIN_START_CODE || state > SLICE_MAX_START_CODE) {
pc->frame_start_found = 0;
pc->state = -1;
@@ -35,8 +35,6 @@ enum RemoveFreq {
REMOVE_FREQ_NONKEYFRAME,
};
-#define START_CODE 0x000001
-
typedef struct RemoveExtradataContext {
const AVClass *class;
int freq;
@@ -73,7 +71,7 @@ static int h264_split(const uint8_t *buf, int buf_size)
while (ptr < end) {
ptr = avpriv_find_start_code(ptr, end, &state);
- if ((state & 0xFFFFFF00) != 0x100)
+ if (!start_code_is_valid(state))
break;
nalu_type = state & 0x1F;
if (nalu_type == H264_NAL_SPS) {
@@ -111,7 +109,7 @@ static int hevc_split(const uint8_t *buf, int buf_size)
while (ptr < end) {
ptr = avpriv_find_start_code(ptr, end, &state);
- if ((state >> 8) != START_CODE)
+ if (!start_code_is_valid(state))
break;
nut = (state >> 1) & 0x3F;
if (nut == HEVC_NAL_VPS)
@@ -171,7 +169,7 @@ static int vc1_split(const uint8_t *buf, int buf_size)
ptr = avpriv_find_start_code(ptr, end, &state);
if (state == VC1_CODE_SEQHDR || state == VC1_CODE_ENTRYPOINT) {
charged = 1;
- } else if (charged && IS_MARKER(state))
+ } else if (charged && start_code_is_valid(state))
return ptr - 4 - buf;
}
@@ -27,6 +27,23 @@
#include <stdint.h>
+#include "libavutil/attributes.h"
+
+/**
+ * @brief Test whether a start code found by avpriv_find_start_code() is valid.
+ *
+ * Use this to test the validity of a start code especially if a start code can
+ * be at the end of the buffer, where testing the return value of avpriv_find_start_code()
+ * would incorrectly imply that the start code is invalid (since the returned value
+ * equals <b>@c end </b>).
+ *
+ * @param[in] start_code The start code to test.
+ * @return A boolean that is true if and only if <b>@p start_code</b> is valid
+ */
+static av_always_inline int start_code_is_valid(uint32_t start_code) {
+ return (start_code & 0xFFFFFF00) == 0x100;
+}
+
const uint8_t *avpriv_find_start_code(const uint8_t *p,
const uint8_t *end,
uint32_t *state);
@@ -470,7 +470,7 @@ static int vaapi_vc1_decode_slice(AVCodecContext *avctx, const uint8_t *buffer,
int err;
/* Current bit buffer is beyond any marker for VC-1, so skip it */
- if (avctx->codec_id == AV_CODEC_ID_VC1 && IS_MARKER(AV_RB32(buffer))) {
+ if (avctx->codec_id == AV_CODEC_ID_VC1 && start_code_is_valid(AV_RB32(buffer))) {
buffer += 4;
size -= 4;
}
@@ -41,8 +41,6 @@ enum VC1Code {
};
//@}
-#define IS_MARKER(x) (((x) & ~0xFF) == VC1_CODE_RES0)
-
/** Available Profiles */
//@{
enum Profile {
@@ -61,7 +59,7 @@ static av_always_inline const uint8_t* find_next_marker(const uint8_t *src, cons
if (end - src >= 4) {
uint32_t mrk = 0xFFFFFFFF;
src = avpriv_find_start_code(src, end, &mrk);
- if (IS_MARKER(mrk))
+ if (start_code_is_valid(mrk))
return src - 4;
}
return end;
@@ -664,7 +664,7 @@ static int vc1_decode_frame(AVCodecContext *avctx, void *data,
if (!buf2)
return AVERROR(ENOMEM);
- if (IS_MARKER(AV_RB32(buf))) { /* frame starts with marker and needs to be parsed */
+ if (start_code_is_valid(AV_RB32(buf))) { /* frame starts with marker and needs to be parsed */
const uint8_t *start, *end, *next;
int size;
@@ -42,8 +42,8 @@ static int avs2_probe(const AVProbeData *p)
while (ptr < end) {
ptr = avpriv_find_start_code(ptr, end, &code);
- state = code & 0xFF;
- if ((code & 0xffffff00) == 0x100) {
+ if (start_code_is_valid(code)) {
+ state = code & 0xFF;
if (AVS2_ISUNIT(state)) {
if (sqb && !hds) {
hds = ptr - sqb;
@@ -36,8 +36,8 @@ static int avs3video_probe(const AVProbeData *p)
while (ptr < end) {
ptr = avpriv_find_start_code(ptr, end, &code);
- state = code & 0xFF;
- if ((code & 0xFFFFFF00) == 0x100) {
+ if (start_code_is_valid(code)) {
+ state = code & 0xFF;
if (state < AVS3_SEQ_START_CODE) {
if (code < slice_pos)
return 0;
@@ -38,7 +38,7 @@ static int cavsvideo_probe(const AVProbeData *p)
while (ptr < end) {
ptr = avpriv_find_start_code(ptr, end, &code);
- if ((code & 0xffffff00) == 0x100) {
+ if (start_code_is_valid(code)) {
if(code < CAVS_SEQ_START_CODE) {
/* slices have to be consecutive */
if(code < slice_pos)
@@ -44,7 +44,7 @@ static int mpegvideo_probe(const AVProbeData *p)
while (ptr < end) {
ptr = avpriv_find_start_code(ptr, end, &code);
- if ((code & 0xffffff00) == 0x100) {
+ if (start_code_is_valid(code)) {
switch(code){
case SEQ_START_CODE:
if (!(ptr[3 + 1 + 2] & 0x20))
@@ -57,7 +57,7 @@ void ff_rtp_send_mpegvideo(AVFormatContext *s1, const uint8_t *buf1, int size)
while (1) {
start_code = -1;
r = avpriv_find_start_code(r1, end, &start_code);
- if((start_code & 0xFFFFFF00) == 0x100) {
+ if (start_code_is_valid(start_code)) {
/* New start code found */
if (start_code == 0x100) {
frame_type = (r[1] & 0x38) >> 3;