Merge commit '5edded9df31bc4712a023f89941b4c278f1bd6f5'
authorJames Almer <jamrial@gmail.com>
Wed, 1 Nov 2017 17:13:04 +0000 (14:13 -0300)
committerJames Almer <jamrial@gmail.com>
Wed, 1 Nov 2017 17:13:04 +0000 (14:13 -0300)
* commit '5edded9df31bc4712a023f89941b4c278f1bd6f5':
  smacker: Improve error handling

See c1947015b2eec65c7fbd702e1d8c22248be511e8

Merged-by: James Almer <jamrial@gmail.com>
1  2 
libavcodec/smacker.c

@@@ -279,9 -264,9 +279,10 @@@ static int smacker_decode_header_tree(S
          goto error;
      }
  
-     if (smacker_decode_bigtree(gb, &huff, &ctx, 0) < 0)
-         err = -1;
 -    if ((res = smacker_decode_bigtree(bc, &huff, &ctx)) < 0)
++    res = smacker_decode_bigtree(gb, &huff, &ctx, 0);
++    if (res < 0)
+         err = res;
 -    bitstream_skip(bc, 1);
 +    skip_bits1(gb);
      if(ctx.last[0] == -1) ctx.last[0] = huff.current++;
      if(ctx.last[1] == -1) ctx.last[1] = huff.current++;
      if(ctx.last[2] == -1) ctx.last[2] = huff.current++;
@@@ -648,26 -627,28 +649,26 @@@ static int smka_decode_frame(AVCodecCon
          *got_frame_ptr = 0;
          return 1;
      }
 -    stereo = bitstream_read_bit(&bc);
 -    bits   = bitstream_read_bit(&bc);
 +    stereo = get_bits1(&gb);
 +    bits = get_bits1(&gb);
      if (stereo ^ (avctx->channels != 1)) {
          av_log(avctx, AV_LOG_ERROR, "channels mismatch\n");
-         return AVERROR(EINVAL);
+         return AVERROR_INVALIDDATA;
      }
 -    if (bits && avctx->sample_fmt == AV_SAMPLE_FMT_U8) {
 +    if (bits == (avctx->sample_fmt == AV_SAMPLE_FMT_U8)) {
          av_log(avctx, AV_LOG_ERROR, "sample format mismatch\n");
-         return AVERROR(EINVAL);
+         return AVERROR_INVALIDDATA;
      }
 +
 +    /* get output buffer */
 +    frame->nb_samples = unp_size / (avctx->channels * (bits + 1));
      if (unp_size % (avctx->channels * (bits + 1))) {
          av_log(avctx, AV_LOG_ERROR,
                 "The buffer does not contain an integer number of samples\n");
-         return AVERROR(EINVAL);
+         return AVERROR_INVALIDDATA;
      }
 -
 -    /* get output buffer */
 -    frame->nb_samples = unp_size / (avctx->channels * (bits + 1));
 -    if ((ret = ff_get_buffer(avctx, frame, 0)) < 0) {
 -        av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
 +    if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
          return ret;
 -    }
      samples  = (int16_t *)frame->data[0];
      samples8 =            frame->data[0];