X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=libavcodec%2Ffaxcompr.c;h=80df418a2910c2601969fe5f3c6abafdc9480284;hb=9d1d7b367eeb74dbfb8501580e6c3568f3fe5973;hp=eb39ae0b20d487c31e0a01e5d227c2cb0e68a361;hpb=27e61a716c893a6ae67ca8f859c735bc0b0b799a;p=ffmpeg diff --git a/libavcodec/faxcompr.c b/libavcodec/faxcompr.c index eb39ae0b20d..80df418a291 100644 --- a/libavcodec/faxcompr.c +++ b/libavcodec/faxcompr.c @@ -122,6 +122,77 @@ av_cold void ff_ccitt_unpack_init(void) initialized = 1; } +static int decode_uncompressed(AVCodecContext *avctx, GetBitContext *gb, + unsigned int *pix_left, int **runs, + const int *runend, int *mode) +{ + int eob = 0; + int newmode; + int saved_run = 0; + + do { + int cwi, k; + int cw = 0; + int codes[2]; + do { + cwi = show_bits(gb, 11); + if (!cwi) { + av_log(avctx, AV_LOG_ERROR, "Invalid uncompressed codeword\n"); + return AVERROR_INVALIDDATA; + } + cwi = 10 - av_log2(cwi); + skip_bits(gb, cwi + 1); + if (cwi > 5) { + newmode = get_bits1(gb); + eob = 1; + cwi -= 6; + } + cw += cwi; + } while(cwi == 5); + + codes[0] = cw; + codes[1] = !eob; + + for (k = 0; k < 2; k++) { + if (codes[k]) { + if (*mode == !k) { + *(*runs)++ = saved_run; + if (*runs >= runend) { + av_log(avctx, AV_LOG_ERROR, "uncompressed run overrun\n"); + return AVERROR_INVALIDDATA; + } + if (*pix_left <= saved_run) { + av_log(avctx, AV_LOG_ERROR, "uncompressed run went out of bounds\n"); + return AVERROR_INVALIDDATA; + } + *pix_left -= saved_run; + saved_run = 0; + *mode = !*mode; + } + saved_run += codes[k]; + } + } + } while (!eob); + *(*runs)++ = saved_run; + if (*runs >= runend) { + av_log(avctx, AV_LOG_ERROR, "uncompressed run overrun\n"); + return AVERROR_INVALIDDATA; + } + if (*pix_left <= saved_run) { + if (*pix_left == saved_run) + return 1; + av_log(avctx, AV_LOG_ERROR, "uncompressed run went out of boundsE\n"); + return AVERROR_INVALIDDATA; + } + *pix_left -= saved_run; + saved_run = 0; + *mode = !*mode; + if (newmode != *mode) { //FIXME CHECK + *(*runs)++ = 0; + *mode = newmode; + } + return 0; +} static int decode_group3_1d_line(AVCodecContext *avctx, GetBitContext *gb, unsigned int pix_left, int *runs, @@ -149,8 +220,18 @@ static int decode_group3_1d_line(AVCodecContext *avctx, GetBitContext *gb, run = 0; mode = !mode; } else if ((int)t == -1) { - av_log(avctx, AV_LOG_ERROR, "Incorrect code\n"); - return AVERROR_INVALIDDATA; + if (show_bits(gb, 12) == 15) { + int ret; + skip_bits(gb, 12); + ret = decode_uncompressed(avctx, gb, &pix_left, &runs, runend, &mode); + if (ret < 0) { + return ret; + } else if (ret) + break; + } else { + av_log(avctx, AV_LOG_ERROR, "Incorrect code\n"); + return AVERROR_INVALIDDATA; + } } } *runs++ = 0; @@ -211,8 +292,25 @@ static int decode_group3_2d_line(AVCodecContext *avctx, GetBitContext *gb, mode = !mode; } } else if (cmode == 9 || cmode == 10) { - avpriv_report_missing_feature(avctx, "Special modes support"); - return AVERROR_PATCHWELCOME; + int xxx = get_bits(gb, 3); + if (cmode == 9 && xxx == 7) { + int ret; + int pix_left = width - offs; + + if (saved_run) { + av_log(avctx, AV_LOG_ERROR, "saved run %d on entering uncompressed mode\n", saved_run); + return AVERROR_INVALIDDATA; + } + ret = decode_uncompressed(avctx, gb, &pix_left, &runs, runend, &mode); + offs = width - pix_left; + if (ret < 0) { + return ret; + } else if (ret) + break; + } else { + avpriv_report_missing_feature(avctx, "Special mode %d xxx=%d support", cmode, xxx); + return AVERROR_PATCHWELCOME; + } } else { //vertical mode run = run_off - offs + (cmode - 5); run_off -= *--ref;