falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39825 Commits
32 Branches
283MB
Tree: 7432bcfe5a
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '7432bcfe5a'
${ noResults }
FFmpeg/libavcodec/tiff.c

780 lines
23KB
Raw Blame History 

  1. /*

  2.  * Copyright (c) 2006 Konstantin Shishkov

  3.  *

  4.  * This file is part of FFmpeg.

  5.  *

  6.  * FFmpeg is free software; you can redistribute it and/or

  7.  * modify it under the terms of the GNU Lesser General Public

  8.  * License as published by the Free Software Foundation; either

  9.  * version 2.1 of the License, or (at your option) any later version.

  10.  *

  11.  * FFmpeg is distributed in the hope that it will be useful,

  12.  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

  14.  * Lesser General Public License for more details.

  15.  *

  16.  * You should have received a copy of the GNU Lesser General Public

  17.  * License along with FFmpeg; if not, write to the Free Software

  18.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

  19.  */

  20. 

  21. /**

  22.  * @file

  23.  * TIFF image decoder

  24.  * @author Konstantin Shishkov

  25.  */

  26. 

  27. #include "avcodec.h"

  28. #if CONFIG_ZLIB

  29. #include <zlib.h>

  30. #endif

  31. #include "lzw.h"

  32. #include "tiff.h"

  33. #include "faxcompr.h"

  34. #include "libavutil/common.h"

  35. #include "libavutil/intreadwrite.h"

  36. #include "libavutil/imgutils.h"

  37. 

  38. typedef struct TiffContext {

  39.     AVCodecContext *avctx;

  40.     AVFrame picture;

  41. 

  42.     int width, height;

  43.     unsigned int bpp, bppcount;

  44.     uint32_t palette[256];

  45.     int palette_is_set;

  46.     int le;

  47.     enum TiffCompr compr;

  48.     int invert;

  49.     int fax_opts;

  50.     int predictor;

  51.     int fill_order;

  52. 

  53.     int strips, rps, sstype;

  54.     int sot;

  55.     const uint8_t *stripdata;

  56.     const uint8_t *stripsizes;

  57.     int stripsize, stripoff;

  58.     LZWState *lzw;

  59. } TiffContext;

  60. 

  61. static unsigned tget_short(const uint8_t **p, int le)

  62. {

  63.     unsigned v = le ? AV_RL16(*p) : AV_RB16(*p);

  64.     *p += 2;

  65.     return v;

  66. }

  67. 

  68. static unsigned tget_long(const uint8_t **p, int le)

  69. {

  70.     unsigned v = le ? AV_RL32(*p) : AV_RB32(*p);

  71.     *p += 4;

  72.     return v;

  73. }

  74. 

  75. static unsigned tget(const uint8_t **p, int type, int le)

  76. {

  77.     switch (type) {

  78.     case TIFF_BYTE : return *(*p)++;

  79.     case TIFF_SHORT: return tget_short(p, le);

  80.     case TIFF_LONG : return tget_long(p, le);

  81.     default        : return UINT_MAX;

  82.     }

  83. }

  84. 

  85. #if CONFIG_ZLIB

  86. static int tiff_uncompress(uint8_t *dst, unsigned long *len, const uint8_t *src,

  87.                            int size)

  88. {

  89.     z_stream zstream = { 0 };

  90.     int zret;

  91. 

  92.     zstream.next_in = src;

  93.     zstream.avail_in = size;

  94.     zstream.next_out = dst;

  95.     zstream.avail_out = *len;

  96.     zret = inflateInit(&zstream);

  97.     if (zret != Z_OK) {

  98.         av_log(NULL, AV_LOG_ERROR, "Inflate init error: %d\n", zret);

  99.         return zret;

  100.     }

  101.     zret = inflate(&zstream, Z_SYNC_FLUSH);

  102.     inflateEnd(&zstream);

  103.     *len = zstream.total_out;

  104.     return zret == Z_STREAM_END ? Z_OK : zret;

  105. }

  106. #endif

  107. 

  108. static void av_always_inline horizontal_fill(unsigned int bpp, uint8_t* dst,

  109.                                              int usePtr, const uint8_t *src,

  110.                                              uint8_t c, int width, int offset)

  111. {

  112.     switch (bpp) {

  113.     case 1:

  114.         while (--width >= 0) {

  115.             dst[(width+offset)*8+7] = (usePtr ? src[width] : c)      & 0x1;

  116.             dst[(width+offset)*8+6] = (usePtr ? src[width] : c) >> 1 & 0x1;

  117.             dst[(width+offset)*8+5] = (usePtr ? src[width] : c) >> 2 & 0x1;

  118.             dst[(width+offset)*8+4] = (usePtr ? src[width] : c) >> 3 & 0x1;

  119.             dst[(width+offset)*8+3] = (usePtr ? src[width] : c) >> 4 & 0x1;

  120.             dst[(width+offset)*8+2] = (usePtr ? src[width] : c) >> 5 & 0x1;

  121.             dst[(width+offset)*8+1] = (usePtr ? src[width] : c) >> 6 & 0x1;

  122.             dst[(width+offset)*8+0] = (usePtr ? src[width] : c) >> 7;

  123.         }

  124.         break;

  125.     case 2:

  126.         while (--width >= 0) {

  127.             dst[(width+offset)*4+3] = (usePtr ? src[width] : c) & 0x3;

  128.             dst[(width+offset)*4+2] = (usePtr ? src[width] : c) >> 2 & 0x3;

  129.             dst[(width+offset)*4+1] = (usePtr ? src[width] : c) >> 4 & 0x3;

  130.             dst[(width+offset)*4+0] = (usePtr ? src[width] : c) >> 6;

  131.         }

  132.         break;

  133.     case 4:

  134.         while (--width >= 0) {

  135.             dst[(width+offset)*2+1] = (usePtr ? src[width] : c) & 0xF;

  136.             dst[(width+offset)*2+0] = (usePtr ? src[width] : c) >> 4;

  137.         }

  138.         break;

  139.     default:

  140.         if (usePtr) {

  141.             memcpy(dst + offset, src, width);

  142.         } else {

  143.             memset(dst + offset, c, width);

  144.         }

  145.     }

  146. }

  147. 

  148. static int tiff_unpack_strip(TiffContext *s, uint8_t *dst, int stride,

  149.                              const uint8_t *src, int size, int lines)

  150. {

  151.     int c, line, pixels, code;

  152.     const uint8_t *ssrc = src;

  153.     int width = ((s->width * s->bpp) + 7) >> 3;

  154. 

  155.     if (size <= 0)

  156.         return AVERROR_INVALIDDATA;

  157. 

  158. #if CONFIG_ZLIB

  159.     if (s->compr == TIFF_DEFLATE || s->compr == TIFF_ADOBE_DEFLATE) {

  160.         uint8_t *zbuf;

  161.         unsigned long outlen;

  162.         int ret;

  163.         outlen = width * lines;

  164.         zbuf = av_malloc(outlen);

  165.         if (!zbuf)

  166.             return AVERROR(ENOMEM);

  167.         ret = tiff_uncompress(zbuf, &outlen, src, size);

  168.         if (ret != Z_OK) {

  169.             av_log(s->avctx, AV_LOG_ERROR,

  170.                    "Uncompressing failed (%lu of %lu) with error %d\n", outlen,

  171.                    (unsigned long)width * lines, ret);

  172.             av_free(zbuf);

  173.             return -1;

  174.         }

  175.         src = zbuf;

  176.         for (line = 0; line < lines; line++) {

  177.             if(s->bpp < 8 && s->avctx->pix_fmt == PIX_FMT_PAL8){

  178.                 horizontal_fill(s->bpp, dst, 1, src, 0, width, 0);

  179.             }else{

  180.                 memcpy(dst, src, width);

  181.             }

  182.             dst += stride;

  183.             src += width;

  184.         }

  185.         av_free(zbuf);

  186.         return 0;

  187.     }

  188. #endif

  189.     if (s->compr == TIFF_LZW) {

  190.         if (ff_lzw_decode_init(s->lzw, 8, src, size, FF_LZW_TIFF) < 0) {

  191.             av_log(s->avctx, AV_LOG_ERROR, "Error initializing LZW decoder\n");

  192.             return -1;

  193.         }

  194.     }

  195.     if (s->compr == TIFF_CCITT_RLE || s->compr == TIFF_G3

  196.         || s->compr == TIFF_G4) {

  197.         int i, ret = 0;

  198.         uint8_t *src2 = av_malloc((unsigned)size +

  199.                                   FF_INPUT_BUFFER_PADDING_SIZE);

  200. 

  201.         if (!src2) {

  202.             av_log(s->avctx, AV_LOG_ERROR,

  203.                    "Error allocating temporary buffer\n");

  204.             return AVERROR(ENOMEM);

  205.         }

  206.         if (s->fax_opts & 2) {

  207.             av_log(s->avctx, AV_LOG_ERROR,

  208.                    "Uncompressed fax mode is not supported (yet)\n");

  209.             av_free(src2);

  210.             return -1;

  211.         }

  212.         if (!s->fill_order) {

  213.             memcpy(src2, src, size);

  214.         } else {

  215.             for (i = 0; i < size; i++)

  216.                 src2[i] = av_reverse[src[i]];

  217.         }

  218.         memset(src2 + size, 0, FF_INPUT_BUFFER_PADDING_SIZE);

  219.         switch (s->compr) {

  220.         case TIFF_CCITT_RLE:

  221.         case TIFF_G3:

  222.         case TIFF_G4:

  223.             ret = ff_ccitt_unpack(s->avctx, src2, size, dst, lines, stride,

  224.                                   s->compr, s->fax_opts);

  225.             break;

  226.         }

  227.         if (s->bpp < 8 && s->avctx->pix_fmt == PIX_FMT_PAL8)

  228.             for (line = 0; line < lines; line++) {

  229.                 horizontal_fill(s->bpp, dst, 1, dst, 0, width, 0);

  230.                 dst += stride;

  231.             }

  232.         av_free(src2);

  233.         return ret;

  234.     }

  235.     for (line = 0; line < lines; line++) {

  236.         if (src - ssrc > size) {

  237.             av_log(s->avctx, AV_LOG_ERROR, "Source data overread\n");

  238.             return -1;

  239.         }

  240.         switch (s->compr) {

  241.         case TIFF_RAW:

  242.             if (ssrc + size - src < width)

  243.                 return AVERROR_INVALIDDATA;

  244.             if (!s->fill_order) {

  245.                 horizontal_fill(s->bpp * (s->avctx->pix_fmt == PIX_FMT_PAL8),

  246.                                 dst, 1, src, 0, width, 0);

  247.             } else {

  248.                 int i;

  249.                 for (i = 0; i < width; i++)

  250.                     dst[i] = av_reverse[src[i]];

  251.             }

  252.             src += width;

  253.             break;

  254.         case TIFF_PACKBITS:

  255.             for (pixels = 0; pixels < width;) {

  256.                 code = (int8_t) * src++;

  257.                 if (code >= 0) {

  258.                     code++;

  259.                     if (pixels + code > width) {

  260.                         av_log(s->avctx, AV_LOG_ERROR,

  261.                                "Copy went out of bounds\n");

  262.                         return -1;

  263.                     }

  264.                     horizontal_fill(s->bpp * (s->avctx->pix_fmt == PIX_FMT_PAL8),

  265.                                     dst, 1, src, 0, code, pixels);

  266.                     src += code;

  267.                     pixels += code;

  268.                 } else if (code != -128) { // -127..-1

  269.                     code = (-code) + 1;

  270.                     if (pixels + code > width) {

  271.                         av_log(s->avctx, AV_LOG_ERROR,

  272.                                "Run went out of bounds\n");

  273.                         return -1;

  274.                     }

  275.                     c = *src++;

  276.                     horizontal_fill(s->bpp * (s->avctx->pix_fmt == PIX_FMT_PAL8),

  277.                                     dst, 0, NULL, c, code, pixels);

  278.                     pixels += code;

  279.                 }

  280.             }

  281.             break;

  282.         case TIFF_LZW:

  283.             pixels = ff_lzw_decode(s->lzw, dst, width);

  284.             if (pixels < width) {

  285.                 av_log(s->avctx, AV_LOG_ERROR, "Decoded only %i bytes of %i\n",

  286.                        pixels, width);

  287.                 return -1;

  288.             }

  289.             if (s->bpp < 8 && s->avctx->pix_fmt == PIX_FMT_PAL8)

  290.                 horizontal_fill(s->bpp, dst, 1, dst, 0, width, 0);

  291.             break;

  292.         }

  293.         dst += stride;

  294.     }

  295.     return 0;

  296. }

  297. 

  298. static int init_image(TiffContext *s)

  299. {

  300.     int i, ret;

  301.     uint32_t *pal;

  302. 

  303.     switch (s->bpp * 10 + s->bppcount) {

  304.     case 11:

  305.         if (!s->palette_is_set) {

  306.             s->avctx->pix_fmt = PIX_FMT_MONOBLACK;

  307.             break;

  308.         }

  309.     case 21:

  310.     case 41:

  311.     case 81:

  312.         s->avctx->pix_fmt = PIX_FMT_PAL8;

  313.         break;

  314.     case 243:

  315.         s->avctx->pix_fmt = PIX_FMT_RGB24;

  316.         break;

  317.     case 161:

  318.         s->avctx->pix_fmt = PIX_FMT_GRAY16BE;

  319.         break;

  320.     case 162:

  321.         s->avctx->pix_fmt = PIX_FMT_GRAY8A;

  322.         break;

  323.     case 324:

  324.         s->avctx->pix_fmt = PIX_FMT_RGBA;

  325.         break;

  326.     case 483:

  327.         s->avctx->pix_fmt = s->le ? PIX_FMT_RGB48LE : PIX_FMT_RGB48BE;

  328.         break;

  329.     case 644:

  330.         s->avctx->pix_fmt = s->le ? PIX_FMT_RGBA64LE : PIX_FMT_RGBA64BE;

  331.         break;

  332.     default:

  333.         av_log(s->avctx, AV_LOG_ERROR,

  334.                "This format is not supported (bpp=%d, bppcount=%d)\n",

  335.                s->bpp, s->bppcount);

  336.         return AVERROR_INVALIDDATA;

  337.     }

  338.     if (s->width != s->avctx->width || s->height != s->avctx->height) {

  339.         if ((ret = av_image_check_size(s->width, s->height, 0, s->avctx)) < 0)

  340.             return ret;

  341.         avcodec_set_dimensions(s->avctx, s->width, s->height);

  342.     }

  343.     if (s->picture.data[0])

  344.         s->avctx->release_buffer(s->avctx, &s->picture);

  345.     if ((ret = s->avctx->get_buffer(s->avctx, &s->picture)) < 0) {

  346.         av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed\n");

  347.         return ret;

  348.     }

  349.     if (s->avctx->pix_fmt == PIX_FMT_PAL8) {

  350.         if (s->palette_is_set) {

  351.             memcpy(s->picture.data[1], s->palette, sizeof(s->palette));

  352.         } else {

  353.             /* make default grayscale pal */

  354.             pal = (uint32_t *) s->picture.data[1];

  355.             for (i = 0; i < 1<<s->bpp; i++)

  356.                 pal[i] = 0xFF << 24 | i * 255 / ((1<<s->bpp) - 1) * 0x010101;

  357.         }

  358.     }

  359.     return 0;

  360. }

  361. 

  362. static int tiff_decode_tag(TiffContext *s, const uint8_t *start,

  363.                            const uint8_t *buf, const uint8_t *end_buf)

  364. {

  365.     unsigned tag, type, count, off, value = 0;

  366.     int i, j;

  367.     uint32_t *pal;

  368.     const uint8_t *rp, *gp, *bp;

  369. 

  370.     if (end_buf - buf < 12)

  371.         return -1;

  372.     tag = tget_short(&buf, s->le);

  373.     type = tget_short(&buf, s->le);

  374.     count = tget_long(&buf, s->le);

  375.     off = tget_long(&buf, s->le);

  376. 

  377.     if (type == 0 || type >= FF_ARRAY_ELEMS(type_sizes)) {

  378.         av_log(s->avctx, AV_LOG_DEBUG, "Unknown tiff type (%u) encountered\n",

  379.                type);

  380.         return 0;

  381.     }

  382. 

  383.     if (count == 1) {

  384.         switch (type) {

  385.         case TIFF_BYTE:

  386.         case TIFF_SHORT:

  387.             buf -= 4;

  388.             value = tget(&buf, type, s->le);

  389.             buf = NULL;

  390.             break;

  391.         case TIFF_LONG:

  392.             value = off;

  393.             buf = NULL;

  394.             break;

  395.         case TIFF_STRING:

  396.             if (count <= 4) {

  397.                 buf -= 4;

  398.                 break;

  399.             }

  400.         default:

  401.             value = UINT_MAX;

  402.             buf = start + off;

  403.         }

  404.     } else {

  405.         if (count <= 4 && type_sizes[type] * count <= 4) {

  406.             buf -= 4;

  407.         } else {

  408.             buf = start + off;

  409.         }

  410.     }

  411. 

  412.     if (buf && (buf < start || buf > end_buf)) {

  413.         av_log(s->avctx, AV_LOG_ERROR,

  414.                "Tag referencing position outside the image\n");

  415.         return -1;

  416.     }

  417. 

  418.     switch (tag) {

  419.     case TIFF_WIDTH:

  420.         s->width = value;

  421.         break;

  422.     case TIFF_HEIGHT:

  423.         s->height = value;

  424.         break;

  425.     case TIFF_BPP:

  426.         s->bppcount = count;

  427.         if (count > 4) {

  428.             av_log(s->avctx, AV_LOG_ERROR,

  429.                    "This format is not supported (bpp=%d, %d components)\n",

  430.                    s->bpp, count);

  431.             return -1;

  432.         }

  433.         if (count == 1)

  434.             s->bpp = value;

  435.         else {

  436.             switch (type) {

  437.             case TIFF_BYTE:

  438.                 s->bpp = (off & 0xFF) + ((off >> 8) & 0xFF) +

  439.                          ((off >> 16) & 0xFF) + ((off >> 24) & 0xFF);

  440.                 break;

  441.             case TIFF_SHORT:

  442.             case TIFF_LONG:

  443.                 s->bpp = 0;

  444.                 for (i = 0; i < count && buf < end_buf; i++)

  445.                     s->bpp += tget(&buf, type, s->le);

  446.                 break;

  447.             default:

  448.                 s->bpp = -1;

  449.             }

  450.         }

  451.         break;

  452.     case TIFF_SAMPLES_PER_PIXEL:

  453.         if (count != 1) {

  454.             av_log(s->avctx, AV_LOG_ERROR,

  455.                    "Samples per pixel requires a single value, many provided\n");

  456.             return AVERROR_INVALIDDATA;

  457.         }

  458.         if (s->bppcount == 1)

  459.             s->bpp *= value;

  460.         s->bppcount = value;

  461.         break;

  462.     case TIFF_COMPR:

  463.         s->compr = value;

  464.         s->predictor = 0;

  465.         switch (s->compr) {

  466.         case TIFF_RAW:

  467.         case TIFF_PACKBITS:

  468.         case TIFF_LZW:

  469.         case TIFF_CCITT_RLE:

  470.             break;

  471.         case TIFF_G3:

  472.         case TIFF_G4:

  473.             s->fax_opts = 0;

  474.             break;

  475.         case TIFF_DEFLATE:

  476.         case TIFF_ADOBE_DEFLATE:

  477. #if CONFIG_ZLIB

  478.             break;

  479. #else

  480.             av_log(s->avctx, AV_LOG_ERROR, "Deflate: ZLib not compiled in\n");

  481.             return -1;

  482. #endif

  483.         case TIFF_JPEG:

  484.         case TIFF_NEWJPEG:

  485.             av_log(s->avctx, AV_LOG_ERROR,

  486.                    "JPEG compression is not supported\n");

  487.             return -1;

  488.         default:

  489.             av_log(s->avctx, AV_LOG_ERROR, "Unknown compression method %i\n",

  490.                    s->compr);

  491.             return -1;

  492.         }

  493.         break;

  494.     case TIFF_ROWSPERSTRIP:

  495.         if (type == TIFF_LONG && value == UINT_MAX)

  496.             value = s->avctx->height;

  497.         if (value < 1) {

  498.             av_log(s->avctx, AV_LOG_ERROR,

  499.                    "Incorrect value of rows per strip\n");

  500.             return -1;

  501.         }

  502.         s->rps = value;

  503.         break;

  504.     case TIFF_STRIP_OFFS:

  505.         if (count == 1) {

  506.             s->stripdata = NULL;

  507.             s->stripoff = value;

  508.         } else

  509.             s->stripdata = start + off;

  510.         s->strips = count;

  511.         if (s->strips == 1)

  512.             s->rps = s->height;

  513.         s->sot = type;

  514.         if (s->stripdata > end_buf) {

  515.             av_log(s->avctx, AV_LOG_ERROR,

  516.                    "Tag referencing position outside the image\n");

  517.             return -1;

  518.         }

  519.         break;

  520.     case TIFF_STRIP_SIZE:

  521.         if (count == 1) {

  522.             s->stripsizes = NULL;

  523.             s->stripsize = value;

  524.             s->strips = 1;

  525.         } else {

  526.             s->stripsizes = start + off;

  527.         }

  528.         s->strips = count;

  529.         s->sstype = type;

  530.         if (s->stripsizes > end_buf) {

  531.             av_log(s->avctx, AV_LOG_ERROR,

  532.                    "Tag referencing position outside the image\n");

  533.             return -1;

  534.         }

  535.         break;

  536.     case TIFF_TILE_BYTE_COUNTS:

  537.     case TIFF_TILE_LENGTH:

  538.     case TIFF_TILE_OFFSETS:

  539.     case TIFF_TILE_WIDTH:

  540.         av_log(s->avctx, AV_LOG_ERROR, "Tiled images are not supported\n");

  541.         return AVERROR_PATCHWELCOME;

  542.         break;

  543.     case TIFF_PREDICTOR:

  544.         s->predictor = value;

  545.         break;

  546.     case TIFF_INVERT:

  547.         switch (value) {

  548.         case 0:

  549.             s->invert = 1;

  550.             break;

  551.         case 1:

  552.             s->invert = 0;

  553.             break;

  554.         case 2:

  555.         case 3:

  556.             break;

  557.         default:

  558.             av_log(s->avctx, AV_LOG_ERROR, "Color mode %d is not supported\n",

  559.                    value);

  560.             return -1;

  561.         }

  562.         break;

  563.     case TIFF_FILL_ORDER:

  564.         if (value < 1 || value > 2) {

  565.             av_log(s->avctx, AV_LOG_ERROR,

  566.                    "Unknown FillOrder value %d, trying default one\n", value);

  567.             value = 1;

  568.         }

  569.         s->fill_order = value - 1;

  570.         break;

  571.     case TIFF_PAL:

  572.         pal = (uint32_t *) s->palette;

  573.         off = type_sizes[type];

  574.         if (count / 3 > 256 || end_buf - buf < count / 3 * off * 3)

  575.             return -1;

  576.         rp = buf;

  577.         gp = buf + count / 3 * off;

  578.         bp = buf + count / 3 * off * 2;

  579.         off = (type_sizes[type] - 1) << 3;

  580.         for (i = 0; i < count / 3; i++) {

  581.             j = 0xff << 24;

  582.             j |= (tget(&rp, type, s->le) >> off) << 16;

  583.             j |= (tget(&gp, type, s->le) >> off) << 8;

  584.             j |=  tget(&bp, type, s->le) >> off;

  585.             pal[i] = j;

  586.         }

  587.         s->palette_is_set = 1;

  588.         break;

  589.     case TIFF_PLANAR:

  590.         if (value == 2) {

  591.             av_log(s->avctx, AV_LOG_ERROR, "Planar format is not supported\n");

  592.             return -1;

  593.         }

  594.         break;

  595.     case TIFF_T4OPTIONS:

  596.         if (s->compr == TIFF_G3)

  597.             s->fax_opts = value;

  598.         break;

  599.     case TIFF_T6OPTIONS:

  600.         if (s->compr == TIFF_G4)

  601.             s->fax_opts = value;

  602.         break;

  603.     default:

  604.         av_log(s->avctx, AV_LOG_DEBUG, "Unknown or unsupported tag %d/0X%0X\n",

  605.                tag, tag);

  606.     }

  607.     return 0;

  608. }

  609. 

  610. static int decode_frame(AVCodecContext *avctx,

  611.                         void *data, int *data_size, AVPacket *avpkt)

  612. {

  613.     const uint8_t *buf = avpkt->data;

  614.     int buf_size = avpkt->size;

  615.     TiffContext *const s = avctx->priv_data;

  616.     AVFrame *picture = data;

  617.     AVFrame *const p = &s->picture;

  618.     const uint8_t *orig_buf = buf, *end_buf = buf + buf_size;

  619.     unsigned off;

  620.     int id, le, ret;

  621.     int i, j, entries;

  622.     int stride;

  623.     unsigned soff, ssize;

  624.     uint8_t *dst;

  625. 

  626.     //parse image header

  627.     if (end_buf - buf < 8)

  628.         return AVERROR_INVALIDDATA;

  629.     id = AV_RL16(buf);

  630.     buf += 2;

  631.     if (id == 0x4949)

  632.         le = 1;

  633.     else if (id == 0x4D4D)

  634.         le = 0;

  635.     else {

  636.         av_log(avctx, AV_LOG_ERROR, "TIFF header not found\n");

  637.         return -1;

  638.     }

  639.     s->le = le;

  640.     s->invert = 0;

  641.     s->compr = TIFF_RAW;

  642.     s->fill_order = 0;

  643.     // As TIFF 6.0 specification puts it "An arbitrary but carefully chosen number

  644.     // that further identifies the file as a TIFF file"

  645.     if (tget_short(&buf, le) != 42) {

  646.         av_log(avctx, AV_LOG_ERROR,

  647.                "The answer to life, universe and everything is not correct!\n");

  648.         return -1;

  649.     }

  650.     // Reset these pointers so we can tell if they were set this frame

  651.     s->stripsizes = s->stripdata = NULL;

  652.     /* parse image file directory */

  653.     off = tget_long(&buf, le);

  654.     if (off >= UINT_MAX - 14 || end_buf - orig_buf < off + 14) {

  655.         av_log(avctx, AV_LOG_ERROR, "IFD offset is greater than image size\n");

  656.         return AVERROR_INVALIDDATA;

  657.     }

  658.     buf = orig_buf + off;

  659.     entries = tget_short(&buf, le);

  660.     for (i = 0; i < entries; i++) {

  661.         if (tiff_decode_tag(s, orig_buf, buf, end_buf) < 0)

  662.             return -1;

  663.         buf += 12;

  664.     }

  665.     if (!s->stripdata && !s->stripoff) {

  666.         av_log(avctx, AV_LOG_ERROR, "Image data is missing\n");

  667.         return -1;

  668.     }

  669.     /* now we have the data and may start decoding */

  670.     if ((ret = init_image(s)) < 0)

  671.         return ret;

  672. 

  673.     if (s->strips == 1 && !s->stripsize) {

  674.         av_log(avctx, AV_LOG_WARNING, "Image data size missing\n");

  675.         s->stripsize = buf_size - s->stripoff;

  676.     }

  677.     stride = p->linesize[0];

  678.     dst = p->data[0];

  679.     for (i = 0; i < s->height; i += s->rps) {

  680.         if (s->stripsizes) {

  681.             if (s->stripsizes >= end_buf)

  682.                 return AVERROR_INVALIDDATA;

  683.             ssize = tget(&s->stripsizes, s->sstype, s->le);

  684.         } else

  685.             ssize = s->stripsize;

  686. 

  687.         if (s->stripdata) {

  688.             if (s->stripdata >= end_buf)

  689.                 return AVERROR_INVALIDDATA;

  690.             soff = tget(&s->stripdata, s->sot, s->le);

  691.         } else

  692.             soff = s->stripoff;

  693. 

  694.         if (soff > buf_size || ssize > buf_size - soff) {

  695.             av_log(avctx, AV_LOG_ERROR, "Invalid strip size/offset\n");

  696.             return -1;

  697.         }

  698.         if (tiff_unpack_strip(s, dst, stride, orig_buf + soff, ssize,

  699.                               FFMIN(s->rps, s->height - i)) < 0)

  700.             break;

  701.         dst += s->rps * stride;

  702.     }

  703.     if (s->predictor == 2) {

  704.         dst = p->data[0];

  705.         soff = s->bpp >> 3;

  706.         ssize = s->width * soff;

  707.         if (s->avctx->pix_fmt == PIX_FMT_RGB48LE ||

  708.             s->avctx->pix_fmt == PIX_FMT_RGBA64LE) {

  709.             for (i = 0; i < s->height; i++) {

  710.                 for (j = soff; j < ssize; j += 2)

  711.                     AV_WL16(dst + j, AV_RL16(dst + j) + AV_RL16(dst + j - soff));

  712.                 dst += stride;

  713.             }

  714.         } else if (s->avctx->pix_fmt == PIX_FMT_RGB48BE ||

  715.                    s->avctx->pix_fmt == PIX_FMT_RGBA64BE) {

  716.             for (i = 0; i < s->height; i++) {

  717.                 for (j = soff; j < ssize; j += 2)

  718.                     AV_WB16(dst + j, AV_RB16(dst + j) + AV_RB16(dst + j - soff));

  719.                 dst += stride;

  720.             }

  721.         } else {

  722.             for (i = 0; i < s->height; i++) {

  723.                 for (j = soff; j < ssize; j++)

  724.                     dst[j] += dst[j - soff];

  725.                 dst += stride;

  726.             }

  727.         }

  728.     }

  729. 

  730.     if (s->invert) {

  731.         dst = s->picture.data[0];

  732.         for (i = 0; i < s->height; i++) {

  733.             for (j = 0; j < s->picture.linesize[0]; j++)

  734.                 dst[j] = (s->avctx->pix_fmt == PIX_FMT_PAL8 ? (1<<s->bpp) - 1 : 255) - dst[j];

  735.             dst += s->picture.linesize[0];

  736.         }

  737.     }

  738.     *picture   = s->picture;

  739.     *data_size = sizeof(AVPicture);

  740. 

  741.     return buf_size;

  742. }

  743. 

  744. static av_cold int tiff_init(AVCodecContext *avctx)

  745. {

  746.     TiffContext *s = avctx->priv_data;

  747. 

  748.     s->width = 0;

  749.     s->height = 0;

  750.     s->avctx = avctx;

  751.     avcodec_get_frame_defaults(&s->picture);

  752.     avctx->coded_frame = &s->picture;

  753.     ff_lzw_decode_open(&s->lzw);

  754.     ff_ccitt_unpack_init();

  755. 

  756.     return 0;

  757. }

  758. 

  759. static av_cold int tiff_end(AVCodecContext *avctx)

  760. {

  761.     TiffContext *const s = avctx->priv_data;

  762. 

  763.     ff_lzw_decode_close(&s->lzw);

  764.     if (s->picture.data[0])

  765.         avctx->release_buffer(avctx, &s->picture);

  766.     return 0;

  767. }

  768. 

  769. AVCodec ff_tiff_decoder = {

  770.     .name           = "tiff",

  771.     .type           = AVMEDIA_TYPE_VIDEO,

  772.     .id             = CODEC_ID_TIFF,

  773.     .priv_data_size = sizeof(TiffContext),

  774.     .init           = tiff_init,

  775.     .close          = tiff_end,

  776.     .decode         = decode_frame,

  777.     .capabilities   = CODEC_CAP_DR1,

  778.     .long_name      = NULL_IF_CONFIG_SMALL("TIFF image"),

  779. };