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.

1386 lines
45KB

  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. * @file
  22. * TIFF image decoder
  23. * @author Konstantin Shishkov
  24. */
  25. #include "config.h"
  26. #if CONFIG_ZLIB
  27. #include <zlib.h>
  28. #endif
  29. #if CONFIG_LZMA
  30. #include <lzma.h>
  31. #endif
  32. #include "libavutil/attributes.h"
  33. #include "libavutil/avstring.h"
  34. #include "libavutil/intreadwrite.h"
  35. #include "libavutil/imgutils.h"
  36. #include "avcodec.h"
  37. #include "bytestream.h"
  38. #include "faxcompr.h"
  39. #include "internal.h"
  40. #include "lzw.h"
  41. #include "mathops.h"
  42. #include "tiff.h"
  43. #include "tiff_data.h"
  44. #include "thread.h"
  45. typedef struct TiffContext {
  46. AVCodecContext *avctx;
  47. GetByteContext gb;
  48. int width, height;
  49. unsigned int bpp, bppcount;
  50. uint32_t palette[256];
  51. int palette_is_set;
  52. int le;
  53. enum TiffCompr compr;
  54. enum TiffPhotometric photometric;
  55. int planar;
  56. int subsampling[2];
  57. int fax_opts;
  58. int predictor;
  59. int fill_order;
  60. uint32_t res[4];
  61. int strips, rps, sstype;
  62. int sot;
  63. int stripsizesoff, stripsize, stripoff, strippos;
  64. LZWState *lzw;
  65. uint8_t *deinvert_buf;
  66. int deinvert_buf_size;
  67. uint8_t *yuv_line;
  68. unsigned int yuv_line_size;
  69. int geotag_count;
  70. TiffGeoTag *geotags;
  71. } TiffContext;
  72. static void free_geotags(TiffContext *const s)
  73. {
  74. int i;
  75. for (i = 0; i < s->geotag_count; i++) {
  76. if (s->geotags[i].val)
  77. av_freep(&s->geotags[i].val);
  78. }
  79. av_freep(&s->geotags);
  80. s->geotag_count = 0;
  81. }
  82. #define RET_GEOKEY(TYPE, array, element)\
  83. if (key >= TIFF_##TYPE##_KEY_ID_OFFSET &&\
  84. key - TIFF_##TYPE##_KEY_ID_OFFSET < FF_ARRAY_ELEMS(ff_tiff_##array##_name_type_map))\
  85. return ff_tiff_##array##_name_type_map[key - TIFF_##TYPE##_KEY_ID_OFFSET].element;
  86. static const char *get_geokey_name(int key)
  87. {
  88. RET_GEOKEY(VERT, vert, name);
  89. RET_GEOKEY(PROJ, proj, name);
  90. RET_GEOKEY(GEOG, geog, name);
  91. RET_GEOKEY(CONF, conf, name);
  92. return NULL;
  93. }
  94. static int get_geokey_type(int key)
  95. {
  96. RET_GEOKEY(VERT, vert, type);
  97. RET_GEOKEY(PROJ, proj, type);
  98. RET_GEOKEY(GEOG, geog, type);
  99. RET_GEOKEY(CONF, conf, type);
  100. return AVERROR_INVALIDDATA;
  101. }
  102. static int cmp_id_key(const void *id, const void *k)
  103. {
  104. return *(const int*)id - ((const TiffGeoTagKeyName*)k)->key;
  105. }
  106. static const char *search_keyval(const TiffGeoTagKeyName *keys, int n, int id)
  107. {
  108. TiffGeoTagKeyName *r = bsearch(&id, keys, n, sizeof(keys[0]), cmp_id_key);
  109. if(r)
  110. return r->name;
  111. return NULL;
  112. }
  113. static char *get_geokey_val(int key, int val)
  114. {
  115. char *ap;
  116. if (val == TIFF_GEO_KEY_UNDEFINED)
  117. return av_strdup("undefined");
  118. if (val == TIFF_GEO_KEY_USER_DEFINED)
  119. return av_strdup("User-Defined");
  120. #define RET_GEOKEY_VAL(TYPE, array)\
  121. if (val >= TIFF_##TYPE##_OFFSET &&\
  122. val - TIFF_##TYPE##_OFFSET < FF_ARRAY_ELEMS(ff_tiff_##array##_codes))\
  123. return av_strdup(ff_tiff_##array##_codes[val - TIFF_##TYPE##_OFFSET]);
  124. switch (key) {
  125. case TIFF_GT_MODEL_TYPE_GEOKEY:
  126. RET_GEOKEY_VAL(GT_MODEL_TYPE, gt_model_type);
  127. break;
  128. case TIFF_GT_RASTER_TYPE_GEOKEY:
  129. RET_GEOKEY_VAL(GT_RASTER_TYPE, gt_raster_type);
  130. break;
  131. case TIFF_GEOG_LINEAR_UNITS_GEOKEY:
  132. case TIFF_PROJ_LINEAR_UNITS_GEOKEY:
  133. case TIFF_VERTICAL_UNITS_GEOKEY:
  134. RET_GEOKEY_VAL(LINEAR_UNIT, linear_unit);
  135. break;
  136. case TIFF_GEOG_ANGULAR_UNITS_GEOKEY:
  137. case TIFF_GEOG_AZIMUTH_UNITS_GEOKEY:
  138. RET_GEOKEY_VAL(ANGULAR_UNIT, angular_unit);
  139. break;
  140. case TIFF_GEOGRAPHIC_TYPE_GEOKEY:
  141. RET_GEOKEY_VAL(GCS_TYPE, gcs_type);
  142. RET_GEOKEY_VAL(GCSE_TYPE, gcse_type);
  143. break;
  144. case TIFF_GEOG_GEODETIC_DATUM_GEOKEY:
  145. RET_GEOKEY_VAL(GEODETIC_DATUM, geodetic_datum);
  146. RET_GEOKEY_VAL(GEODETIC_DATUM_E, geodetic_datum_e);
  147. break;
  148. case TIFF_GEOG_ELLIPSOID_GEOKEY:
  149. RET_GEOKEY_VAL(ELLIPSOID, ellipsoid);
  150. break;
  151. case TIFF_GEOG_PRIME_MERIDIAN_GEOKEY:
  152. RET_GEOKEY_VAL(PRIME_MERIDIAN, prime_meridian);
  153. break;
  154. case TIFF_PROJECTED_CS_TYPE_GEOKEY:
  155. ap = av_strdup(search_keyval(ff_tiff_proj_cs_type_codes, FF_ARRAY_ELEMS(ff_tiff_proj_cs_type_codes), val));
  156. if(ap) return ap;
  157. break;
  158. case TIFF_PROJECTION_GEOKEY:
  159. ap = av_strdup(search_keyval(ff_tiff_projection_codes, FF_ARRAY_ELEMS(ff_tiff_projection_codes), val));
  160. if(ap) return ap;
  161. break;
  162. case TIFF_PROJ_COORD_TRANS_GEOKEY:
  163. RET_GEOKEY_VAL(COORD_TRANS, coord_trans);
  164. break;
  165. case TIFF_VERTICAL_CS_TYPE_GEOKEY:
  166. RET_GEOKEY_VAL(VERT_CS, vert_cs);
  167. RET_GEOKEY_VAL(ORTHO_VERT_CS, ortho_vert_cs);
  168. break;
  169. }
  170. ap = av_malloc(14);
  171. if (ap)
  172. snprintf(ap, 14, "Unknown-%d", val);
  173. return ap;
  174. }
  175. static char *doubles2str(double *dp, int count, const char *sep)
  176. {
  177. int i;
  178. char *ap, *ap0;
  179. uint64_t component_len;
  180. if (!sep) sep = ", ";
  181. component_len = 24LL + strlen(sep);
  182. if (count >= (INT_MAX - 1)/component_len)
  183. return NULL;
  184. ap = av_malloc(component_len * count + 1);
  185. if (!ap)
  186. return NULL;
  187. ap0 = ap;
  188. ap[0] = '\0';
  189. for (i = 0; i < count; i++) {
  190. unsigned l = snprintf(ap, component_len, "%.15g%s", dp[i], sep);
  191. if(l >= component_len) {
  192. av_free(ap0);
  193. return NULL;
  194. }
  195. ap += l;
  196. }
  197. ap0[strlen(ap0) - strlen(sep)] = '\0';
  198. return ap0;
  199. }
  200. static int add_metadata(int count, int type,
  201. const char *name, const char *sep, TiffContext *s, AVFrame *frame)
  202. {
  203. switch(type) {
  204. case TIFF_DOUBLE: return ff_tadd_doubles_metadata(count, name, sep, &s->gb, s->le, avpriv_frame_get_metadatap(frame));
  205. case TIFF_SHORT : return ff_tadd_shorts_metadata(count, name, sep, &s->gb, s->le, 0, avpriv_frame_get_metadatap(frame));
  206. case TIFF_STRING: return ff_tadd_string_metadata(count, name, &s->gb, s->le, avpriv_frame_get_metadatap(frame));
  207. default : return AVERROR_INVALIDDATA;
  208. };
  209. }
  210. static void av_always_inline horizontal_fill(unsigned int bpp, uint8_t* dst,
  211. int usePtr, const uint8_t *src,
  212. uint8_t c, int width, int offset)
  213. {
  214. switch (bpp) {
  215. case 1:
  216. while (--width >= 0) {
  217. dst[(width+offset)*8+7] = (usePtr ? src[width] : c) & 0x1;
  218. dst[(width+offset)*8+6] = (usePtr ? src[width] : c) >> 1 & 0x1;
  219. dst[(width+offset)*8+5] = (usePtr ? src[width] : c) >> 2 & 0x1;
  220. dst[(width+offset)*8+4] = (usePtr ? src[width] : c) >> 3 & 0x1;
  221. dst[(width+offset)*8+3] = (usePtr ? src[width] : c) >> 4 & 0x1;
  222. dst[(width+offset)*8+2] = (usePtr ? src[width] : c) >> 5 & 0x1;
  223. dst[(width+offset)*8+1] = (usePtr ? src[width] : c) >> 6 & 0x1;
  224. dst[(width+offset)*8+0] = (usePtr ? src[width] : c) >> 7;
  225. }
  226. break;
  227. case 2:
  228. while (--width >= 0) {
  229. dst[(width+offset)*4+3] = (usePtr ? src[width] : c) & 0x3;
  230. dst[(width+offset)*4+2] = (usePtr ? src[width] : c) >> 2 & 0x3;
  231. dst[(width+offset)*4+1] = (usePtr ? src[width] : c) >> 4 & 0x3;
  232. dst[(width+offset)*4+0] = (usePtr ? src[width] : c) >> 6;
  233. }
  234. break;
  235. case 4:
  236. while (--width >= 0) {
  237. dst[(width+offset)*2+1] = (usePtr ? src[width] : c) & 0xF;
  238. dst[(width+offset)*2+0] = (usePtr ? src[width] : c) >> 4;
  239. }
  240. break;
  241. default:
  242. if (usePtr) {
  243. memcpy(dst + offset, src, width);
  244. } else {
  245. memset(dst + offset, c, width);
  246. }
  247. }
  248. }
  249. static int deinvert_buffer(TiffContext *s, const uint8_t *src, int size)
  250. {
  251. int i;
  252. av_fast_padded_malloc(&s->deinvert_buf, &s->deinvert_buf_size, size);
  253. if (!s->deinvert_buf)
  254. return AVERROR(ENOMEM);
  255. for (i = 0; i < size; i++)
  256. s->deinvert_buf[i] = ff_reverse[src[i]];
  257. return 0;
  258. }
  259. static void unpack_yuv(TiffContext *s, AVFrame *p,
  260. const uint8_t *src, int lnum)
  261. {
  262. int i, j, k;
  263. int w = (s->width - 1) / s->subsampling[0] + 1;
  264. uint8_t *pu = &p->data[1][lnum / s->subsampling[1] * p->linesize[1]];
  265. uint8_t *pv = &p->data[2][lnum / s->subsampling[1] * p->linesize[2]];
  266. if (s->width % s->subsampling[0] || s->height % s->subsampling[1]) {
  267. for (i = 0; i < w; i++) {
  268. for (j = 0; j < s->subsampling[1]; j++)
  269. for (k = 0; k < s->subsampling[0]; k++)
  270. p->data[0][FFMIN(lnum + j, s->height-1) * p->linesize[0] +
  271. FFMIN(i * s->subsampling[0] + k, s->width-1)] = *src++;
  272. *pu++ = *src++;
  273. *pv++ = *src++;
  274. }
  275. }else{
  276. for (i = 0; i < w; i++) {
  277. for (j = 0; j < s->subsampling[1]; j++)
  278. for (k = 0; k < s->subsampling[0]; k++)
  279. p->data[0][(lnum + j) * p->linesize[0] +
  280. i * s->subsampling[0] + k] = *src++;
  281. *pu++ = *src++;
  282. *pv++ = *src++;
  283. }
  284. }
  285. }
  286. #if CONFIG_ZLIB
  287. static int tiff_uncompress(uint8_t *dst, unsigned long *len, const uint8_t *src,
  288. int size)
  289. {
  290. z_stream zstream = { 0 };
  291. int zret;
  292. zstream.next_in = (uint8_t *)src;
  293. zstream.avail_in = size;
  294. zstream.next_out = dst;
  295. zstream.avail_out = *len;
  296. zret = inflateInit(&zstream);
  297. if (zret != Z_OK) {
  298. av_log(NULL, AV_LOG_ERROR, "Inflate init error: %d\n", zret);
  299. return zret;
  300. }
  301. zret = inflate(&zstream, Z_SYNC_FLUSH);
  302. inflateEnd(&zstream);
  303. *len = zstream.total_out;
  304. return zret == Z_STREAM_END ? Z_OK : zret;
  305. }
  306. static int tiff_unpack_zlib(TiffContext *s, AVFrame *p, uint8_t *dst, int stride,
  307. const uint8_t *src, int size, int width, int lines,
  308. int strip_start, int is_yuv)
  309. {
  310. uint8_t *zbuf;
  311. unsigned long outlen;
  312. int ret, line;
  313. outlen = width * lines;
  314. zbuf = av_malloc(outlen);
  315. if (!zbuf)
  316. return AVERROR(ENOMEM);
  317. if (s->fill_order) {
  318. if ((ret = deinvert_buffer(s, src, size)) < 0) {
  319. av_free(zbuf);
  320. return ret;
  321. }
  322. src = s->deinvert_buf;
  323. }
  324. ret = tiff_uncompress(zbuf, &outlen, src, size);
  325. if (ret != Z_OK) {
  326. av_log(s->avctx, AV_LOG_ERROR,
  327. "Uncompressing failed (%lu of %lu) with error %d\n", outlen,
  328. (unsigned long)width * lines, ret);
  329. av_free(zbuf);
  330. return AVERROR_UNKNOWN;
  331. }
  332. src = zbuf;
  333. for (line = 0; line < lines; line++) {
  334. if (s->bpp < 8 && s->avctx->pix_fmt == AV_PIX_FMT_PAL8) {
  335. horizontal_fill(s->bpp, dst, 1, src, 0, width, 0);
  336. } else {
  337. memcpy(dst, src, width);
  338. }
  339. if (is_yuv) {
  340. unpack_yuv(s, p, dst, strip_start + line);
  341. line += s->subsampling[1] - 1;
  342. }
  343. dst += stride;
  344. src += width;
  345. }
  346. av_free(zbuf);
  347. return 0;
  348. }
  349. #endif
  350. #if CONFIG_LZMA
  351. static int tiff_uncompress_lzma(uint8_t *dst, uint64_t *len, const uint8_t *src,
  352. int size)
  353. {
  354. lzma_stream stream = LZMA_STREAM_INIT;
  355. lzma_ret ret;
  356. stream.next_in = (uint8_t *)src;
  357. stream.avail_in = size;
  358. stream.next_out = dst;
  359. stream.avail_out = *len;
  360. ret = lzma_stream_decoder(&stream, UINT64_MAX, 0);
  361. if (ret != LZMA_OK) {
  362. av_log(NULL, AV_LOG_ERROR, "LZMA init error: %d\n", ret);
  363. return ret;
  364. }
  365. ret = lzma_code(&stream, LZMA_RUN);
  366. lzma_end(&stream);
  367. *len = stream.total_out;
  368. return ret == LZMA_STREAM_END ? LZMA_OK : ret;
  369. }
  370. static int tiff_unpack_lzma(TiffContext *s, AVFrame *p, uint8_t *dst, int stride,
  371. const uint8_t *src, int size, int width, int lines,
  372. int strip_start, int is_yuv)
  373. {
  374. uint64_t outlen = width * lines;
  375. int ret, line;
  376. uint8_t *buf = av_malloc(outlen);
  377. if (!buf)
  378. return AVERROR(ENOMEM);
  379. if (s->fill_order) {
  380. if ((ret = deinvert_buffer(s, src, size)) < 0) {
  381. av_free(buf);
  382. return ret;
  383. }
  384. src = s->deinvert_buf;
  385. }
  386. ret = tiff_uncompress_lzma(buf, &outlen, src, size);
  387. if (ret != LZMA_OK) {
  388. av_log(s->avctx, AV_LOG_ERROR,
  389. "Uncompressing failed (%"PRIu64" of %"PRIu64") with error %d\n", outlen,
  390. (uint64_t)width * lines, ret);
  391. av_free(buf);
  392. return AVERROR_UNKNOWN;
  393. }
  394. src = buf;
  395. for (line = 0; line < lines; line++) {
  396. if (s->bpp < 8 && s->avctx->pix_fmt == AV_PIX_FMT_PAL8) {
  397. horizontal_fill(s->bpp, dst, 1, src, 0, width, 0);
  398. } else {
  399. memcpy(dst, src, width);
  400. }
  401. if (is_yuv) {
  402. unpack_yuv(s, p, dst, strip_start + line);
  403. line += s->subsampling[1] - 1;
  404. }
  405. dst += stride;
  406. src += width;
  407. }
  408. av_free(buf);
  409. return 0;
  410. }
  411. #endif
  412. static int tiff_unpack_fax(TiffContext *s, uint8_t *dst, int stride,
  413. const uint8_t *src, int size, int width, int lines)
  414. {
  415. int i, ret = 0;
  416. int line;
  417. uint8_t *src2 = av_malloc((unsigned)size +
  418. FF_INPUT_BUFFER_PADDING_SIZE);
  419. if (!src2) {
  420. av_log(s->avctx, AV_LOG_ERROR,
  421. "Error allocating temporary buffer\n");
  422. return AVERROR(ENOMEM);
  423. }
  424. if (s->fax_opts & 2) {
  425. avpriv_request_sample(s->avctx, "Uncompressed fax mode");
  426. av_free(src2);
  427. return AVERROR_PATCHWELCOME;
  428. }
  429. if (!s->fill_order) {
  430. memcpy(src2, src, size);
  431. } else {
  432. for (i = 0; i < size; i++)
  433. src2[i] = ff_reverse[src[i]];
  434. }
  435. memset(src2 + size, 0, FF_INPUT_BUFFER_PADDING_SIZE);
  436. ret = ff_ccitt_unpack(s->avctx, src2, size, dst, lines, stride,
  437. s->compr, s->fax_opts);
  438. if (s->bpp < 8 && s->avctx->pix_fmt == AV_PIX_FMT_PAL8)
  439. for (line = 0; line < lines; line++) {
  440. horizontal_fill(s->bpp, dst, 1, dst, 0, width, 0);
  441. dst += stride;
  442. }
  443. av_free(src2);
  444. return ret;
  445. }
  446. static int tiff_unpack_strip(TiffContext *s, AVFrame *p, uint8_t *dst, int stride,
  447. const uint8_t *src, int size, int strip_start, int lines)
  448. {
  449. PutByteContext pb;
  450. int c, line, pixels, code, ret;
  451. const uint8_t *ssrc = src;
  452. int width = ((s->width * s->bpp) + 7) >> 3;
  453. const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(p->format);
  454. int is_yuv = !(desc->flags & AV_PIX_FMT_FLAG_RGB) &&
  455. (desc->flags & AV_PIX_FMT_FLAG_PLANAR) &&
  456. desc->nb_components >= 3;
  457. if (s->planar)
  458. width /= s->bppcount;
  459. if (size <= 0)
  460. return AVERROR_INVALIDDATA;
  461. if (is_yuv) {
  462. int bytes_per_row = (((s->width - 1) / s->subsampling[0] + 1) * s->bpp *
  463. s->subsampling[0] * s->subsampling[1] + 7) >> 3;
  464. av_fast_padded_malloc(&s->yuv_line, &s->yuv_line_size, bytes_per_row);
  465. if (s->yuv_line == NULL) {
  466. av_log(s->avctx, AV_LOG_ERROR, "Not enough memory\n");
  467. return AVERROR(ENOMEM);
  468. }
  469. dst = s->yuv_line;
  470. stride = 0;
  471. width = s->width * s->subsampling[1] + 2*(s->width / s->subsampling[0]);
  472. av_assert0(width <= bytes_per_row);
  473. av_assert0(s->bpp == 24);
  474. }
  475. if (s->compr == TIFF_DEFLATE || s->compr == TIFF_ADOBE_DEFLATE) {
  476. #if CONFIG_ZLIB
  477. return tiff_unpack_zlib(s, p, dst, stride, src, size, width, lines,
  478. strip_start, is_yuv);
  479. #else
  480. av_log(s->avctx, AV_LOG_ERROR,
  481. "zlib support not enabled, "
  482. "deflate compression not supported\n");
  483. return AVERROR(ENOSYS);
  484. #endif
  485. }
  486. if (s->compr == TIFF_LZMA) {
  487. #if CONFIG_LZMA
  488. return tiff_unpack_lzma(s, p, dst, stride, src, size, width, lines,
  489. strip_start, is_yuv);
  490. #else
  491. av_log(s->avctx, AV_LOG_ERROR,
  492. "LZMA support not enabled\n");
  493. return AVERROR(ENOSYS);
  494. #endif
  495. }
  496. if (s->compr == TIFF_LZW) {
  497. if (s->fill_order) {
  498. if ((ret = deinvert_buffer(s, src, size)) < 0)
  499. return ret;
  500. ssrc = src = s->deinvert_buf;
  501. }
  502. if (size > 1 && !src[0] && (src[1]&1)) {
  503. av_log(s->avctx, AV_LOG_ERROR, "Old style LZW is unsupported\n");
  504. }
  505. if ((ret = ff_lzw_decode_init(s->lzw, 8, src, size, FF_LZW_TIFF)) < 0) {
  506. av_log(s->avctx, AV_LOG_ERROR, "Error initializing LZW decoder\n");
  507. return ret;
  508. }
  509. for (line = 0; line < lines; line++) {
  510. pixels = ff_lzw_decode(s->lzw, dst, width);
  511. if (pixels < width) {
  512. av_log(s->avctx, AV_LOG_ERROR, "Decoded only %i bytes of %i\n",
  513. pixels, width);
  514. return AVERROR_INVALIDDATA;
  515. }
  516. if (s->bpp < 8 && s->avctx->pix_fmt == AV_PIX_FMT_PAL8)
  517. horizontal_fill(s->bpp, dst, 1, dst, 0, width, 0);
  518. if (is_yuv) {
  519. unpack_yuv(s, p, dst, strip_start + line);
  520. line += s->subsampling[1] - 1;
  521. }
  522. dst += stride;
  523. }
  524. return 0;
  525. }
  526. if (s->compr == TIFF_CCITT_RLE ||
  527. s->compr == TIFF_G3 ||
  528. s->compr == TIFF_G4) {
  529. if (is_yuv)
  530. return AVERROR_INVALIDDATA;
  531. return tiff_unpack_fax(s, dst, stride, src, size, width, lines);
  532. }
  533. bytestream2_init(&s->gb, src, size);
  534. bytestream2_init_writer(&pb, dst, is_yuv ? s->yuv_line_size : (stride * lines));
  535. for (line = 0; line < lines; line++) {
  536. if (src - ssrc > size) {
  537. av_log(s->avctx, AV_LOG_ERROR, "Source data overread\n");
  538. return AVERROR_INVALIDDATA;
  539. }
  540. if (bytestream2_get_bytes_left(&s->gb) == 0 || bytestream2_get_eof(&pb))
  541. break;
  542. bytestream2_seek_p(&pb, stride * line, SEEK_SET);
  543. switch (s->compr) {
  544. case TIFF_RAW:
  545. if (ssrc + size - src < width)
  546. return AVERROR_INVALIDDATA;
  547. if (!s->fill_order) {
  548. horizontal_fill(s->bpp * (s->avctx->pix_fmt == AV_PIX_FMT_PAL8),
  549. dst, 1, src, 0, width, 0);
  550. } else {
  551. int i;
  552. for (i = 0; i < width; i++)
  553. dst[i] = ff_reverse[src[i]];
  554. }
  555. src += width;
  556. break;
  557. case TIFF_PACKBITS:
  558. for (pixels = 0; pixels < width;) {
  559. if (ssrc + size - src < 2) {
  560. av_log(s->avctx, AV_LOG_ERROR, "Read went out of bounds\n");
  561. return AVERROR_INVALIDDATA;
  562. }
  563. code = s->fill_order ? (int8_t) ff_reverse[*src++]: (int8_t) *src++;
  564. if (code >= 0) {
  565. code++;
  566. if (pixels + code > width ||
  567. ssrc + size - src < code) {
  568. av_log(s->avctx, AV_LOG_ERROR,
  569. "Copy went out of bounds\n");
  570. return AVERROR_INVALIDDATA;
  571. }
  572. horizontal_fill(s->bpp * (s->avctx->pix_fmt == AV_PIX_FMT_PAL8),
  573. dst, 1, src, 0, code, pixels);
  574. src += code;
  575. pixels += code;
  576. } else if (code != -128) { // -127..-1
  577. code = (-code) + 1;
  578. if (pixels + code > width) {
  579. av_log(s->avctx, AV_LOG_ERROR,
  580. "Run went out of bounds\n");
  581. return AVERROR_INVALIDDATA;
  582. }
  583. c = *src++;
  584. horizontal_fill(s->bpp * (s->avctx->pix_fmt == AV_PIX_FMT_PAL8),
  585. dst, 0, NULL, c, code, pixels);
  586. pixels += code;
  587. }
  588. }
  589. if (s->fill_order) {
  590. int i;
  591. for (i = 0; i < width; i++)
  592. dst[i] = ff_reverse[dst[i]];
  593. }
  594. break;
  595. }
  596. if (is_yuv) {
  597. unpack_yuv(s, p, dst, strip_start + line);
  598. line += s->subsampling[1] - 1;
  599. }
  600. dst += stride;
  601. }
  602. return 0;
  603. }
  604. static int init_image(TiffContext *s, ThreadFrame *frame)
  605. {
  606. int ret;
  607. int create_gray_palette = 0;
  608. // make sure there is no aliasing in the following switch
  609. if (s->bpp >= 100 || s->bppcount >= 10) {
  610. av_log(s->avctx, AV_LOG_ERROR,
  611. "Unsupported image parameters: bpp=%d, bppcount=%d\n",
  612. s->bpp, s->bppcount);
  613. return AVERROR_INVALIDDATA;
  614. }
  615. switch (s->planar * 1000 + s->bpp * 10 + s->bppcount) {
  616. case 11:
  617. if (!s->palette_is_set) {
  618. s->avctx->pix_fmt = AV_PIX_FMT_MONOBLACK;
  619. break;
  620. }
  621. case 21:
  622. case 41:
  623. s->avctx->pix_fmt = AV_PIX_FMT_PAL8;
  624. if (!s->palette_is_set) {
  625. create_gray_palette = 1;
  626. }
  627. break;
  628. case 81:
  629. s->avctx->pix_fmt = s->palette_is_set ? AV_PIX_FMT_PAL8 : AV_PIX_FMT_GRAY8;
  630. break;
  631. case 243:
  632. if (s->photometric == TIFF_PHOTOMETRIC_YCBCR) {
  633. if (s->subsampling[0] == 1 && s->subsampling[1] == 1) {
  634. s->avctx->pix_fmt = AV_PIX_FMT_YUV444P;
  635. } else if (s->subsampling[0] == 2 && s->subsampling[1] == 1) {
  636. s->avctx->pix_fmt = AV_PIX_FMT_YUV422P;
  637. } else if (s->subsampling[0] == 4 && s->subsampling[1] == 1) {
  638. s->avctx->pix_fmt = AV_PIX_FMT_YUV411P;
  639. } else if (s->subsampling[0] == 1 && s->subsampling[1] == 2) {
  640. s->avctx->pix_fmt = AV_PIX_FMT_YUV440P;
  641. } else if (s->subsampling[0] == 2 && s->subsampling[1] == 2) {
  642. s->avctx->pix_fmt = AV_PIX_FMT_YUV420P;
  643. } else if (s->subsampling[0] == 4 && s->subsampling[1] == 4) {
  644. s->avctx->pix_fmt = AV_PIX_FMT_YUV410P;
  645. } else {
  646. av_log(s->avctx, AV_LOG_ERROR, "Unsupported YCbCr subsampling\n");
  647. return AVERROR_PATCHWELCOME;
  648. }
  649. } else
  650. s->avctx->pix_fmt = AV_PIX_FMT_RGB24;
  651. break;
  652. case 161:
  653. s->avctx->pix_fmt = s->le ? AV_PIX_FMT_GRAY16LE : AV_PIX_FMT_GRAY16BE;
  654. break;
  655. case 162:
  656. s->avctx->pix_fmt = AV_PIX_FMT_YA8;
  657. break;
  658. case 322:
  659. s->avctx->pix_fmt = s->le ? AV_PIX_FMT_YA16LE : AV_PIX_FMT_YA16BE;
  660. break;
  661. case 324:
  662. s->avctx->pix_fmt = AV_PIX_FMT_RGBA;
  663. break;
  664. case 483:
  665. s->avctx->pix_fmt = s->le ? AV_PIX_FMT_RGB48LE : AV_PIX_FMT_RGB48BE;
  666. break;
  667. case 644:
  668. s->avctx->pix_fmt = s->le ? AV_PIX_FMT_RGBA64LE : AV_PIX_FMT_RGBA64BE;
  669. break;
  670. case 1243:
  671. s->avctx->pix_fmt = AV_PIX_FMT_GBRP;
  672. break;
  673. case 1324:
  674. s->avctx->pix_fmt = AV_PIX_FMT_GBRAP;
  675. break;
  676. case 1483:
  677. s->avctx->pix_fmt = s->le ? AV_PIX_FMT_GBRP16LE : AV_PIX_FMT_GBRP16BE;
  678. break;
  679. case 1644:
  680. s->avctx->pix_fmt = s->le ? AV_PIX_FMT_GBRAP16LE : AV_PIX_FMT_GBRAP16BE;
  681. break;
  682. default:
  683. av_log(s->avctx, AV_LOG_ERROR,
  684. "This format is not supported (bpp=%d, bppcount=%d)\n",
  685. s->bpp, s->bppcount);
  686. return AVERROR_INVALIDDATA;
  687. }
  688. if (s->photometric == TIFF_PHOTOMETRIC_YCBCR) {
  689. const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(s->avctx->pix_fmt);
  690. if((desc->flags & AV_PIX_FMT_FLAG_RGB) ||
  691. !(desc->flags & AV_PIX_FMT_FLAG_PLANAR) ||
  692. desc->nb_components < 3) {
  693. av_log(s->avctx, AV_LOG_ERROR, "Unsupported YCbCr variant\n");
  694. return AVERROR_INVALIDDATA;
  695. }
  696. }
  697. if (s->width != s->avctx->width || s->height != s->avctx->height) {
  698. ret = ff_set_dimensions(s->avctx, s->width, s->height);
  699. if (ret < 0)
  700. return ret;
  701. }
  702. if ((ret = ff_thread_get_buffer(s->avctx, frame, 0)) < 0)
  703. return ret;
  704. if (s->avctx->pix_fmt == AV_PIX_FMT_PAL8) {
  705. if (!create_gray_palette)
  706. memcpy(frame->f->data[1], s->palette, sizeof(s->palette));
  707. else {
  708. /* make default grayscale pal */
  709. int i;
  710. uint32_t *pal = (uint32_t *)frame->f->data[1];
  711. for (i = 0; i < 1<<s->bpp; i++)
  712. pal[i] = 0xFFU << 24 | i * 255 / ((1<<s->bpp) - 1) * 0x010101;
  713. }
  714. }
  715. return 0;
  716. }
  717. static void set_sar(TiffContext *s, unsigned tag, unsigned num, unsigned den)
  718. {
  719. int offset = tag == TIFF_YRES ? 2 : 0;
  720. s->res[offset++] = num;
  721. s->res[offset] = den;
  722. if (s->res[0] && s->res[1] && s->res[2] && s->res[3])
  723. av_reduce(&s->avctx->sample_aspect_ratio.num, &s->avctx->sample_aspect_ratio.den,
  724. s->res[2] * (uint64_t)s->res[1], s->res[0] * (uint64_t)s->res[3], INT32_MAX);
  725. }
  726. static int tiff_decode_tag(TiffContext *s, AVFrame *frame)
  727. {
  728. unsigned tag, type, count, off, value = 0, value2 = 0;
  729. int i, start;
  730. int pos;
  731. int ret;
  732. double *dp;
  733. ret = ff_tread_tag(&s->gb, s->le, &tag, &type, &count, &start);
  734. if (ret < 0) {
  735. goto end;
  736. }
  737. off = bytestream2_tell(&s->gb);
  738. if (count == 1) {
  739. switch (type) {
  740. case TIFF_BYTE:
  741. case TIFF_SHORT:
  742. case TIFF_LONG:
  743. value = ff_tget(&s->gb, type, s->le);
  744. break;
  745. case TIFF_RATIONAL:
  746. value = ff_tget(&s->gb, TIFF_LONG, s->le);
  747. value2 = ff_tget(&s->gb, TIFF_LONG, s->le);
  748. break;
  749. case TIFF_STRING:
  750. if (count <= 4) {
  751. break;
  752. }
  753. default:
  754. value = UINT_MAX;
  755. }
  756. }
  757. switch (tag) {
  758. case TIFF_WIDTH:
  759. s->width = value;
  760. break;
  761. case TIFF_HEIGHT:
  762. s->height = value;
  763. break;
  764. case TIFF_BPP:
  765. if (count > 4U) {
  766. av_log(s->avctx, AV_LOG_ERROR,
  767. "This format is not supported (bpp=%d, %d components)\n",
  768. value, count);
  769. return AVERROR_INVALIDDATA;
  770. }
  771. s->bppcount = count;
  772. if (count == 1)
  773. s->bpp = value;
  774. else {
  775. switch (type) {
  776. case TIFF_BYTE:
  777. case TIFF_SHORT:
  778. case TIFF_LONG:
  779. s->bpp = 0;
  780. if (bytestream2_get_bytes_left(&s->gb) < type_sizes[type] * count)
  781. return AVERROR_INVALIDDATA;
  782. for (i = 0; i < count; i++)
  783. s->bpp += ff_tget(&s->gb, type, s->le);
  784. break;
  785. default:
  786. s->bpp = -1;
  787. }
  788. }
  789. if (s->bpp > 64U) {
  790. av_log(s->avctx, AV_LOG_ERROR,
  791. "This format is not supported (bpp=%d, %d components)\n",
  792. s->bpp, count);
  793. s->bpp = 0;
  794. return AVERROR_INVALIDDATA;
  795. }
  796. break;
  797. case TIFF_SAMPLES_PER_PIXEL:
  798. if (count != 1) {
  799. av_log(s->avctx, AV_LOG_ERROR,
  800. "Samples per pixel requires a single value, many provided\n");
  801. return AVERROR_INVALIDDATA;
  802. }
  803. if (value > 4U) {
  804. av_log(s->avctx, AV_LOG_ERROR,
  805. "Samples per pixel %d is too large\n", value);
  806. return AVERROR_INVALIDDATA;
  807. }
  808. if (s->bppcount == 1)
  809. s->bpp *= value;
  810. s->bppcount = value;
  811. break;
  812. case TIFF_COMPR:
  813. s->compr = value;
  814. s->predictor = 0;
  815. switch (s->compr) {
  816. case TIFF_RAW:
  817. case TIFF_PACKBITS:
  818. case TIFF_LZW:
  819. case TIFF_CCITT_RLE:
  820. break;
  821. case TIFF_G3:
  822. case TIFF_G4:
  823. s->fax_opts = 0;
  824. break;
  825. case TIFF_DEFLATE:
  826. case TIFF_ADOBE_DEFLATE:
  827. #if CONFIG_ZLIB
  828. break;
  829. #else
  830. av_log(s->avctx, AV_LOG_ERROR, "Deflate: ZLib not compiled in\n");
  831. return AVERROR(ENOSYS);
  832. #endif
  833. case TIFF_JPEG:
  834. case TIFF_NEWJPEG:
  835. avpriv_report_missing_feature(s->avctx, "JPEG compression");
  836. return AVERROR_PATCHWELCOME;
  837. case TIFF_LZMA:
  838. #if CONFIG_LZMA
  839. break;
  840. #else
  841. av_log(s->avctx, AV_LOG_ERROR, "LZMA not compiled in\n");
  842. return AVERROR(ENOSYS);
  843. #endif
  844. default:
  845. av_log(s->avctx, AV_LOG_ERROR, "Unknown compression method %i\n",
  846. s->compr);
  847. return AVERROR_INVALIDDATA;
  848. }
  849. break;
  850. case TIFF_ROWSPERSTRIP:
  851. if (!value || (type == TIFF_LONG && value == UINT_MAX))
  852. value = s->height;
  853. s->rps = FFMIN(value, s->height);
  854. break;
  855. case TIFF_STRIP_OFFS:
  856. if (count == 1) {
  857. s->strippos = 0;
  858. s->stripoff = value;
  859. } else
  860. s->strippos = off;
  861. s->strips = count;
  862. if (s->strips == 1)
  863. s->rps = s->height;
  864. s->sot = type;
  865. break;
  866. case TIFF_STRIP_SIZE:
  867. if (count == 1) {
  868. s->stripsizesoff = 0;
  869. s->stripsize = value;
  870. s->strips = 1;
  871. } else {
  872. s->stripsizesoff = off;
  873. }
  874. s->strips = count;
  875. s->sstype = type;
  876. break;
  877. case TIFF_XRES:
  878. case TIFF_YRES:
  879. set_sar(s, tag, value, value2);
  880. break;
  881. case TIFF_TILE_BYTE_COUNTS:
  882. case TIFF_TILE_LENGTH:
  883. case TIFF_TILE_OFFSETS:
  884. case TIFF_TILE_WIDTH:
  885. av_log(s->avctx, AV_LOG_ERROR, "Tiled images are not supported\n");
  886. return AVERROR_PATCHWELCOME;
  887. break;
  888. case TIFF_PREDICTOR:
  889. s->predictor = value;
  890. break;
  891. case TIFF_PHOTOMETRIC:
  892. switch (value) {
  893. case TIFF_PHOTOMETRIC_WHITE_IS_ZERO:
  894. case TIFF_PHOTOMETRIC_BLACK_IS_ZERO:
  895. case TIFF_PHOTOMETRIC_RGB:
  896. case TIFF_PHOTOMETRIC_PALETTE:
  897. case TIFF_PHOTOMETRIC_YCBCR:
  898. s->photometric = value;
  899. break;
  900. case TIFF_PHOTOMETRIC_ALPHA_MASK:
  901. case TIFF_PHOTOMETRIC_SEPARATED:
  902. case TIFF_PHOTOMETRIC_CIE_LAB:
  903. case TIFF_PHOTOMETRIC_ICC_LAB:
  904. case TIFF_PHOTOMETRIC_ITU_LAB:
  905. case TIFF_PHOTOMETRIC_CFA:
  906. case TIFF_PHOTOMETRIC_LOG_L:
  907. case TIFF_PHOTOMETRIC_LOG_LUV:
  908. case TIFF_PHOTOMETRIC_LINEAR_RAW:
  909. avpriv_report_missing_feature(s->avctx,
  910. "PhotometricInterpretation 0x%04X",
  911. value);
  912. return AVERROR_PATCHWELCOME;
  913. default:
  914. av_log(s->avctx, AV_LOG_ERROR, "PhotometricInterpretation %u is "
  915. "unknown\n", value);
  916. return AVERROR_INVALIDDATA;
  917. }
  918. break;
  919. case TIFF_FILL_ORDER:
  920. if (value < 1 || value > 2) {
  921. av_log(s->avctx, AV_LOG_ERROR,
  922. "Unknown FillOrder value %d, trying default one\n", value);
  923. value = 1;
  924. }
  925. s->fill_order = value - 1;
  926. break;
  927. case TIFF_PAL: {
  928. GetByteContext pal_gb[3];
  929. off = type_sizes[type];
  930. if (count / 3 > 256 ||
  931. bytestream2_get_bytes_left(&s->gb) < count / 3 * off * 3)
  932. return AVERROR_INVALIDDATA;
  933. pal_gb[0] = pal_gb[1] = pal_gb[2] = s->gb;
  934. bytestream2_skip(&pal_gb[1], count / 3 * off);
  935. bytestream2_skip(&pal_gb[2], count / 3 * off * 2);
  936. off = (type_sizes[type] - 1) << 3;
  937. for (i = 0; i < count / 3; i++) {
  938. uint32_t p = 0xFF000000;
  939. p |= (ff_tget(&pal_gb[0], type, s->le) >> off) << 16;
  940. p |= (ff_tget(&pal_gb[1], type, s->le) >> off) << 8;
  941. p |= ff_tget(&pal_gb[2], type, s->le) >> off;
  942. s->palette[i] = p;
  943. }
  944. s->palette_is_set = 1;
  945. break;
  946. }
  947. case TIFF_PLANAR:
  948. s->planar = value == 2;
  949. break;
  950. case TIFF_YCBCR_SUBSAMPLING:
  951. if (count != 2) {
  952. av_log(s->avctx, AV_LOG_ERROR, "subsample count invalid\n");
  953. return AVERROR_INVALIDDATA;
  954. }
  955. for (i = 0; i < count; i++)
  956. s->subsampling[i] = ff_tget(&s->gb, type, s->le);
  957. break;
  958. case TIFF_T4OPTIONS:
  959. if (s->compr == TIFF_G3)
  960. s->fax_opts = value;
  961. break;
  962. case TIFF_T6OPTIONS:
  963. if (s->compr == TIFF_G4)
  964. s->fax_opts = value;
  965. break;
  966. #define ADD_METADATA(count, name, sep)\
  967. if ((ret = add_metadata(count, type, name, sep, s, frame)) < 0) {\
  968. av_log(s->avctx, AV_LOG_ERROR, "Error allocating temporary buffer\n");\
  969. goto end;\
  970. }
  971. case TIFF_MODEL_PIXEL_SCALE:
  972. ADD_METADATA(count, "ModelPixelScaleTag", NULL);
  973. break;
  974. case TIFF_MODEL_TRANSFORMATION:
  975. ADD_METADATA(count, "ModelTransformationTag", NULL);
  976. break;
  977. case TIFF_MODEL_TIEPOINT:
  978. ADD_METADATA(count, "ModelTiepointTag", NULL);
  979. break;
  980. case TIFF_GEO_KEY_DIRECTORY:
  981. ADD_METADATA(1, "GeoTIFF_Version", NULL);
  982. ADD_METADATA(2, "GeoTIFF_Key_Revision", ".");
  983. s->geotag_count = ff_tget_short(&s->gb, s->le);
  984. if (s->geotag_count > count / 4 - 1) {
  985. s->geotag_count = count / 4 - 1;
  986. av_log(s->avctx, AV_LOG_WARNING, "GeoTIFF key directory buffer shorter than specified\n");
  987. }
  988. if (bytestream2_get_bytes_left(&s->gb) < s->geotag_count * sizeof(int16_t) * 4) {
  989. s->geotag_count = 0;
  990. return -1;
  991. }
  992. s->geotags = av_mallocz_array(s->geotag_count, sizeof(TiffGeoTag));
  993. if (!s->geotags) {
  994. av_log(s->avctx, AV_LOG_ERROR, "Error allocating temporary buffer\n");
  995. s->geotag_count = 0;
  996. goto end;
  997. }
  998. for (i = 0; i < s->geotag_count; i++) {
  999. s->geotags[i].key = ff_tget_short(&s->gb, s->le);
  1000. s->geotags[i].type = ff_tget_short(&s->gb, s->le);
  1001. s->geotags[i].count = ff_tget_short(&s->gb, s->le);
  1002. if (!s->geotags[i].type)
  1003. s->geotags[i].val = get_geokey_val(s->geotags[i].key, ff_tget_short(&s->gb, s->le));
  1004. else
  1005. s->geotags[i].offset = ff_tget_short(&s->gb, s->le);
  1006. }
  1007. break;
  1008. case TIFF_GEO_DOUBLE_PARAMS:
  1009. if (count >= INT_MAX / sizeof(int64_t))
  1010. return AVERROR_INVALIDDATA;
  1011. if (bytestream2_get_bytes_left(&s->gb) < count * sizeof(int64_t))
  1012. return AVERROR_INVALIDDATA;
  1013. dp = av_malloc_array(count, sizeof(double));
  1014. if (!dp) {
  1015. av_log(s->avctx, AV_LOG_ERROR, "Error allocating temporary buffer\n");
  1016. goto end;
  1017. }
  1018. for (i = 0; i < count; i++)
  1019. dp[i] = ff_tget_double(&s->gb, s->le);
  1020. for (i = 0; i < s->geotag_count; i++) {
  1021. if (s->geotags[i].type == TIFF_GEO_DOUBLE_PARAMS) {
  1022. if (s->geotags[i].count == 0
  1023. || s->geotags[i].offset + s->geotags[i].count > count) {
  1024. av_log(s->avctx, AV_LOG_WARNING, "Invalid GeoTIFF key %d\n", s->geotags[i].key);
  1025. } else {
  1026. char *ap = doubles2str(&dp[s->geotags[i].offset], s->geotags[i].count, ", ");
  1027. if (!ap) {
  1028. av_log(s->avctx, AV_LOG_ERROR, "Error allocating temporary buffer\n");
  1029. av_freep(&dp);
  1030. return AVERROR(ENOMEM);
  1031. }
  1032. s->geotags[i].val = ap;
  1033. }
  1034. }
  1035. }
  1036. av_freep(&dp);
  1037. break;
  1038. case TIFF_GEO_ASCII_PARAMS:
  1039. pos = bytestream2_tell(&s->gb);
  1040. for (i = 0; i < s->geotag_count; i++) {
  1041. if (s->geotags[i].type == TIFF_GEO_ASCII_PARAMS) {
  1042. if (s->geotags[i].count == 0
  1043. || s->geotags[i].offset + s->geotags[i].count > count) {
  1044. av_log(s->avctx, AV_LOG_WARNING, "Invalid GeoTIFF key %d\n", s->geotags[i].key);
  1045. } else {
  1046. char *ap;
  1047. bytestream2_seek(&s->gb, pos + s->geotags[i].offset, SEEK_SET);
  1048. if (bytestream2_get_bytes_left(&s->gb) < s->geotags[i].count)
  1049. return AVERROR_INVALIDDATA;
  1050. ap = av_malloc(s->geotags[i].count);
  1051. if (!ap) {
  1052. av_log(s->avctx, AV_LOG_ERROR, "Error allocating temporary buffer\n");
  1053. return AVERROR(ENOMEM);
  1054. }
  1055. bytestream2_get_bufferu(&s->gb, ap, s->geotags[i].count);
  1056. ap[s->geotags[i].count - 1] = '\0'; //replace the "|" delimiter with a 0 byte
  1057. s->geotags[i].val = ap;
  1058. }
  1059. }
  1060. }
  1061. break;
  1062. case TIFF_ARTIST:
  1063. ADD_METADATA(count, "artist", NULL);
  1064. break;
  1065. case TIFF_COPYRIGHT:
  1066. ADD_METADATA(count, "copyright", NULL);
  1067. break;
  1068. case TIFF_DATE:
  1069. ADD_METADATA(count, "date", NULL);
  1070. break;
  1071. case TIFF_DOCUMENT_NAME:
  1072. ADD_METADATA(count, "document_name", NULL);
  1073. break;
  1074. case TIFF_HOST_COMPUTER:
  1075. ADD_METADATA(count, "computer", NULL);
  1076. break;
  1077. case TIFF_IMAGE_DESCRIPTION:
  1078. ADD_METADATA(count, "description", NULL);
  1079. break;
  1080. case TIFF_MAKE:
  1081. ADD_METADATA(count, "make", NULL);
  1082. break;
  1083. case TIFF_MODEL:
  1084. ADD_METADATA(count, "model", NULL);
  1085. break;
  1086. case TIFF_PAGE_NAME:
  1087. ADD_METADATA(count, "page_name", NULL);
  1088. break;
  1089. case TIFF_PAGE_NUMBER:
  1090. ADD_METADATA(count, "page_number", " / ");
  1091. break;
  1092. case TIFF_SOFTWARE_NAME:
  1093. ADD_METADATA(count, "software", NULL);
  1094. break;
  1095. default:
  1096. if (s->avctx->err_recognition & AV_EF_EXPLODE) {
  1097. av_log(s->avctx, AV_LOG_ERROR,
  1098. "Unknown or unsupported tag %d/0X%0X\n",
  1099. tag, tag);
  1100. return AVERROR_INVALIDDATA;
  1101. }
  1102. }
  1103. end:
  1104. bytestream2_seek(&s->gb, start, SEEK_SET);
  1105. return 0;
  1106. }
  1107. static int decode_frame(AVCodecContext *avctx,
  1108. void *data, int *got_frame, AVPacket *avpkt)
  1109. {
  1110. TiffContext *const s = avctx->priv_data;
  1111. AVFrame *const p = data;
  1112. ThreadFrame frame = { .f = data };
  1113. unsigned off;
  1114. int le, ret, plane, planes;
  1115. int i, j, entries, stride;
  1116. unsigned soff, ssize;
  1117. uint8_t *dst;
  1118. GetByteContext stripsizes;
  1119. GetByteContext stripdata;
  1120. bytestream2_init(&s->gb, avpkt->data, avpkt->size);
  1121. // parse image header
  1122. if ((ret = ff_tdecode_header(&s->gb, &le, &off))) {
  1123. av_log(avctx, AV_LOG_ERROR, "Invalid TIFF header\n");
  1124. return ret;
  1125. } else if (off >= UINT_MAX - 14 || avpkt->size < off + 14) {
  1126. av_log(avctx, AV_LOG_ERROR, "IFD offset is greater than image size\n");
  1127. return AVERROR_INVALIDDATA;
  1128. }
  1129. s->le = le;
  1130. // TIFF_BPP is not a required tag and defaults to 1
  1131. s->bppcount = s->bpp = 1;
  1132. s->photometric = TIFF_PHOTOMETRIC_NONE;
  1133. s->compr = TIFF_RAW;
  1134. s->fill_order = 0;
  1135. free_geotags(s);
  1136. // Reset these offsets so we can tell if they were set this frame
  1137. s->stripsizesoff = s->strippos = 0;
  1138. /* parse image file directory */
  1139. bytestream2_seek(&s->gb, off, SEEK_SET);
  1140. entries = ff_tget_short(&s->gb, le);
  1141. if (bytestream2_get_bytes_left(&s->gb) < entries * 12)
  1142. return AVERROR_INVALIDDATA;
  1143. for (i = 0; i < entries; i++) {
  1144. if ((ret = tiff_decode_tag(s, p)) < 0)
  1145. return ret;
  1146. }
  1147. for (i = 0; i<s->geotag_count; i++) {
  1148. const char *keyname = get_geokey_name(s->geotags[i].key);
  1149. if (!keyname) {
  1150. av_log(avctx, AV_LOG_WARNING, "Unknown or unsupported GeoTIFF key %d\n", s->geotags[i].key);
  1151. continue;
  1152. }
  1153. if (get_geokey_type(s->geotags[i].key) != s->geotags[i].type) {
  1154. av_log(avctx, AV_LOG_WARNING, "Type of GeoTIFF key %d is wrong\n", s->geotags[i].key);
  1155. continue;
  1156. }
  1157. ret = av_dict_set(avpriv_frame_get_metadatap(p), keyname, s->geotags[i].val, 0);
  1158. if (ret<0) {
  1159. av_log(avctx, AV_LOG_ERROR, "Writing metadata with key '%s' failed\n", keyname);
  1160. return ret;
  1161. }
  1162. }
  1163. if (!s->strippos && !s->stripoff) {
  1164. av_log(avctx, AV_LOG_ERROR, "Image data is missing\n");
  1165. return AVERROR_INVALIDDATA;
  1166. }
  1167. /* now we have the data and may start decoding */
  1168. if ((ret = init_image(s, &frame)) < 0)
  1169. return ret;
  1170. if (s->strips == 1 && !s->stripsize) {
  1171. av_log(avctx, AV_LOG_WARNING, "Image data size missing\n");
  1172. s->stripsize = avpkt->size - s->stripoff;
  1173. }
  1174. if (s->stripsizesoff) {
  1175. if (s->stripsizesoff >= (unsigned)avpkt->size)
  1176. return AVERROR_INVALIDDATA;
  1177. bytestream2_init(&stripsizes, avpkt->data + s->stripsizesoff,
  1178. avpkt->size - s->stripsizesoff);
  1179. }
  1180. if (s->strippos) {
  1181. if (s->strippos >= (unsigned)avpkt->size)
  1182. return AVERROR_INVALIDDATA;
  1183. bytestream2_init(&stripdata, avpkt->data + s->strippos,
  1184. avpkt->size - s->strippos);
  1185. }
  1186. if (s->rps <= 0) {
  1187. av_log(avctx, AV_LOG_ERROR, "rps %d invalid\n", s->rps);
  1188. return AVERROR_INVALIDDATA;
  1189. }
  1190. planes = s->planar ? s->bppcount : 1;
  1191. for (plane = 0; plane < planes; plane++) {
  1192. stride = p->linesize[plane];
  1193. dst = p->data[plane];
  1194. for (i = 0; i < s->height; i += s->rps) {
  1195. if (s->stripsizesoff)
  1196. ssize = ff_tget(&stripsizes, s->sstype, le);
  1197. else
  1198. ssize = s->stripsize;
  1199. if (s->strippos)
  1200. soff = ff_tget(&stripdata, s->sot, le);
  1201. else
  1202. soff = s->stripoff;
  1203. if (soff > avpkt->size || ssize > avpkt->size - soff) {
  1204. av_log(avctx, AV_LOG_ERROR, "Invalid strip size/offset\n");
  1205. return AVERROR_INVALIDDATA;
  1206. }
  1207. if ((ret = tiff_unpack_strip(s, p, dst, stride, avpkt->data + soff, ssize, i,
  1208. FFMIN(s->rps, s->height - i))) < 0) {
  1209. if (avctx->err_recognition & AV_EF_EXPLODE)
  1210. return ret;
  1211. break;
  1212. }
  1213. dst += s->rps * stride;
  1214. }
  1215. if (s->predictor == 2) {
  1216. if (s->photometric == TIFF_PHOTOMETRIC_YCBCR) {
  1217. av_log(s->avctx, AV_LOG_ERROR, "predictor == 2 with YUV is unsupported");
  1218. return AVERROR_PATCHWELCOME;
  1219. }
  1220. dst = p->data[plane];
  1221. soff = s->bpp >> 3;
  1222. if (s->planar)
  1223. soff = FFMAX(soff / s->bppcount, 1);
  1224. ssize = s->width * soff;
  1225. if (s->avctx->pix_fmt == AV_PIX_FMT_RGB48LE ||
  1226. s->avctx->pix_fmt == AV_PIX_FMT_RGBA64LE ||
  1227. s->avctx->pix_fmt == AV_PIX_FMT_GBRP16LE ||
  1228. s->avctx->pix_fmt == AV_PIX_FMT_GBRAP16LE) {
  1229. for (i = 0; i < s->height; i++) {
  1230. for (j = soff; j < ssize; j += 2)
  1231. AV_WL16(dst + j, AV_RL16(dst + j) + AV_RL16(dst + j - soff));
  1232. dst += stride;
  1233. }
  1234. } else if (s->avctx->pix_fmt == AV_PIX_FMT_RGB48BE ||
  1235. s->avctx->pix_fmt == AV_PIX_FMT_RGBA64BE ||
  1236. s->avctx->pix_fmt == AV_PIX_FMT_GBRP16BE ||
  1237. s->avctx->pix_fmt == AV_PIX_FMT_GBRAP16BE) {
  1238. for (i = 0; i < s->height; i++) {
  1239. for (j = soff; j < ssize; j += 2)
  1240. AV_WB16(dst + j, AV_RB16(dst + j) + AV_RB16(dst + j - soff));
  1241. dst += stride;
  1242. }
  1243. } else {
  1244. for (i = 0; i < s->height; i++) {
  1245. for (j = soff; j < ssize; j++)
  1246. dst[j] += dst[j - soff];
  1247. dst += stride;
  1248. }
  1249. }
  1250. }
  1251. if (s->photometric == TIFF_PHOTOMETRIC_WHITE_IS_ZERO) {
  1252. dst = p->data[plane];
  1253. for (i = 0; i < s->height; i++) {
  1254. for (j = 0; j < p->linesize[plane]; j++)
  1255. dst[j] = (s->avctx->pix_fmt == AV_PIX_FMT_PAL8 ? (1<<s->bpp) - 1 : 255) - dst[j];
  1256. dst += stride;
  1257. }
  1258. }
  1259. }
  1260. if (s->planar && s->bppcount > 2) {
  1261. FFSWAP(uint8_t*, p->data[0], p->data[2]);
  1262. FFSWAP(int, p->linesize[0], p->linesize[2]);
  1263. FFSWAP(uint8_t*, p->data[0], p->data[1]);
  1264. FFSWAP(int, p->linesize[0], p->linesize[1]);
  1265. }
  1266. *got_frame = 1;
  1267. return avpkt->size;
  1268. }
  1269. static av_cold int tiff_init(AVCodecContext *avctx)
  1270. {
  1271. TiffContext *s = avctx->priv_data;
  1272. s->width = 0;
  1273. s->height = 0;
  1274. s->subsampling[0] =
  1275. s->subsampling[1] = 1;
  1276. s->avctx = avctx;
  1277. ff_lzw_decode_open(&s->lzw);
  1278. ff_ccitt_unpack_init();
  1279. return 0;
  1280. }
  1281. static av_cold int tiff_end(AVCodecContext *avctx)
  1282. {
  1283. TiffContext *const s = avctx->priv_data;
  1284. free_geotags(s);
  1285. ff_lzw_decode_close(&s->lzw);
  1286. av_freep(&s->deinvert_buf);
  1287. return 0;
  1288. }
  1289. AVCodec ff_tiff_decoder = {
  1290. .name = "tiff",
  1291. .long_name = NULL_IF_CONFIG_SMALL("TIFF image"),
  1292. .type = AVMEDIA_TYPE_VIDEO,
  1293. .id = AV_CODEC_ID_TIFF,
  1294. .priv_data_size = sizeof(TiffContext),
  1295. .init = tiff_init,
  1296. .close = tiff_end,
  1297. .decode = decode_frame,
  1298. .init_thread_copy = ONLY_IF_THREADS_ENABLED(tiff_init),
  1299. .capabilities = CODEC_CAP_DR1 | CODEC_CAP_FRAME_THREADS,
  1300. };