@@ -36,6 +36,6 @@ enum {
};
#define AVS2_ISPIC(x) ((x) == AVS2_INTRA_PIC_START_CODE || (x) == AVS2_INTER_PIC_START_CODE)
-#define AVS2_ISUNIT(x) ((x) == AVS2_SEQ_START_CODE || (x) == AVS2_SEQ_END_CODE || (x) == AVS2_USER_DATA_START_CODE || AVS2_ISPIC(x))
+#define AVS2_ISUNIT(x) ((x) == AVS2_SEQ_START_CODE || AVS2_ISPIC(x))
#endif
@@ -31,7 +31,7 @@ static int avs2_find_frame_end(ParseContext *pc, const uint8_t *buf, int buf_siz
if (!pic_found) {
for (; cur < buf_size; ++cur) {
state = (state << 8) | buf[cur];
- if (AVS2_ISUNIT(buf[cur])){
+ if ((state & 0xFFFFFF00) == 0x100 && AVS2_ISPIC(buf[cur])) {
cur++;
pic_found = 1;
break;
@@ -44,7 +44,7 @@ static int avs2_find_frame_end(ParseContext *pc, const uint8_t *buf, int buf_siz
return END_NOT_FOUND;
for (; cur < buf_size; cur++) {
state = (state << 8) | buf[cur];
- if ((state & 0xFFFFFF00) == 0x100 && state > AVS2_SLICE_MAX_START_CODE) {
+ if ((state & 0xFFFFFF00) == 0x100 && AVS2_ISUNIT(buf[cur])) {
pc->frame_start_found = 0;
pc->state = -1;
return cur - 3;