some checks to avoid segfault

Originally committed as revision 1740 to svn://svn.ffmpeg.org/ffmpeg/trunk
pull/126/head
Michael Niedermayer 22 years ago
parent 9f2e61b6a3
commit 8b92b792d1
  1. 30
      libavcodec/h264.c

@ -2826,7 +2826,16 @@ static int decode_slice_header(H264Context *h){
return -1;
}
h->pps= h->pps_buffer[pps_id];
if(h->pps.slice_group_count == 0){
fprintf(stderr, "non existing PPS referenced\n");
return -1;
}
h->sps= h->sps_buffer[ h->pps.sps_id ];
if(h->sps.log2_max_frame_num == 0){
fprintf(stderr, "non existing SPS referenced\n");
return -1;
}
s->mb_width= h->sps.mb_width;
s->mb_height= h->sps.mb_height;
@ -4036,7 +4045,7 @@ static int decode_nal_units(H264Context *h, uint8_t *buf, int buf_size){
init_get_bits(&h->inter_gb, ptr, bit_length);
h->inter_gb_ptr= &h->inter_gb;
if(h->redundant_pic_count==0)
if(h->redundant_pic_count==0 && h->intra_gb_ptr && s->data_partitioning)
decode_slice(h);
break;
case NAL_SEI:
@ -4067,6 +4076,8 @@ static int decode_nal_units(H264Context *h, uint8_t *buf, int buf_size){
s->current_picture.key_frame= s->pict_type == I_TYPE;
}
if(!s->current_picture_ptr) return buf_index; //no frame
h->prev_frame_num_offset= h->frame_num_offset;
h->prev_frame_num= h->frame_num;
if(s->current_picture_ptr->reference){
@ -4147,21 +4158,14 @@ static int decode_frame(AVCodecContext *avctx,
*pict= *(AVFrame*)&s->last_picture;
}
#endif
if(!s->current_picture_ptr){
fprintf(stderr, "error, NO frame\n");
return -1;
}
*pict= *(AVFrame*)&s->current_picture; //FIXME
assert(pict->data[0]);
//printf("out %d\n", (int)pict->data[0]);
if(avctx->debug&FF_DEBUG_QP){
int8_t *qtab= pict->qscale_table;
int x,y;
for(y=0; y<s->mb_height; y++){
for(x=0; x<s->mb_width; x++){
printf("%2d ", qtab[x + y*s->mb_width]);
}
printf("\n");
}
printf("\n");
}
#if 0 //?
/* Return the Picture timestamp as the frame number */

Loading…
Cancel
Save