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.

1192 lines
37KB

  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 "avcodec.h"
  26. #include "bytestream.h"
  27. #include "config.h"
  28. #if CONFIG_ZLIB
  29. #include <zlib.h>
  30. #endif
  31. #include "lzw.h"
  32. #include "tiff.h"
  33. #include "tiff_data.h"
  34. #include "faxcompr.h"
  35. #include "mathops.h"
  36. #include "libavutil/attributes.h"
  37. #include "libavutil/intreadwrite.h"
  38. #include "libavutil/imgutils.h"
  39. #include "libavutil/avstring.h"
  40. typedef struct TiffContext {
  41. AVCodecContext *avctx;
  42. AVFrame picture;
  43. GetByteContext gb;
  44. int width, height;
  45. unsigned int bpp, bppcount;
  46. uint32_t palette[256];
  47. int palette_is_set;
  48. int le;
  49. enum TiffCompr compr;
  50. int invert;
  51. int fax_opts;
  52. int predictor;
  53. int fill_order;
  54. int strips, rps, sstype;
  55. int sot;
  56. int stripsizesoff, stripsize, stripoff, strippos;
  57. LZWState *lzw;
  58. int geotag_count;
  59. TiffGeoTag *geotags;
  60. } TiffContext;
  61. static unsigned tget_short(GetByteContext *gb, int le)
  62. {
  63. unsigned v = le ? bytestream2_get_le16u(gb) : bytestream2_get_be16u(gb);
  64. return v;
  65. }
  66. static unsigned tget_long(GetByteContext *gb, int le)
  67. {
  68. unsigned v = le ? bytestream2_get_le32u(gb) : bytestream2_get_be32u(gb);
  69. return v;
  70. }
  71. static double tget_double(GetByteContext *gb, int le)
  72. {
  73. av_alias64 i = { .u64 = le ? bytestream2_get_le64u(gb) : bytestream2_get_be64u(gb)};
  74. return i.f64;
  75. }
  76. static unsigned tget(GetByteContext *gb, int type, int le)
  77. {
  78. switch (type) {
  79. case TIFF_BYTE : return bytestream2_get_byteu(gb);
  80. case TIFF_SHORT: return tget_short(gb, le);
  81. case TIFF_LONG : return tget_long(gb, le);
  82. default : return UINT_MAX;
  83. }
  84. }
  85. static void free_geotags(TiffContext *const s)
  86. {
  87. int i;
  88. for (i = 0; i < s->geotag_count; i++) {
  89. if (s->geotags[i].val)
  90. av_freep(&s->geotags[i].val);
  91. }
  92. av_freep(&s->geotags);
  93. }
  94. #define RET_GEOKEY(TYPE, array, element)\
  95. if (key >= TIFF_##TYPE##_KEY_ID_OFFSET &&\
  96. key - TIFF_##TYPE##_KEY_ID_OFFSET < FF_ARRAY_ELEMS(ff_tiff_##array##_name_type_map))\
  97. return ff_tiff_##array##_name_type_map[key - TIFF_##TYPE##_KEY_ID_OFFSET].element;
  98. static const char *get_geokey_name(int key)
  99. {
  100. RET_GEOKEY(VERT, vert, name);
  101. RET_GEOKEY(PROJ, proj, name);
  102. RET_GEOKEY(GEOG, geog, name);
  103. RET_GEOKEY(CONF, conf, name);
  104. return NULL;
  105. }
  106. static int get_geokey_type(int key)
  107. {
  108. RET_GEOKEY(VERT, vert, type);
  109. RET_GEOKEY(PROJ, proj, type);
  110. RET_GEOKEY(GEOG, geog, type);
  111. RET_GEOKEY(CONF, conf, type);
  112. return AVERROR_INVALIDDATA;
  113. }
  114. static int cmp_id_key(const void *id, const void *k)
  115. {
  116. return *(const int*)id - ((const TiffGeoTagKeyName*)k)->key;
  117. }
  118. static const char *search_keyval(const TiffGeoTagKeyName *keys, int n, int id)
  119. {
  120. return ((TiffGeoTagKeyName*)bsearch(&id, keys, n, sizeof(keys[0]), cmp_id_key))->name;
  121. }
  122. static char *get_geokey_val(int key, int val)
  123. {
  124. char *ap;
  125. if (val == TIFF_GEO_KEY_UNDEFINED)
  126. return av_strdup("undefined");
  127. if (val == TIFF_GEO_KEY_USER_DEFINED)
  128. return av_strdup("User-Defined");
  129. #define RET_GEOKEY_VAL(TYPE, array)\
  130. if (val >= TIFF_##TYPE##_OFFSET &&\
  131. val - TIFF_##TYPE##_OFFSET < FF_ARRAY_ELEMS(ff_tiff_##array##_codes))\
  132. return av_strdup(ff_tiff_##array##_codes[val - TIFF_##TYPE##_OFFSET]);
  133. switch (key) {
  134. case TIFF_GT_MODEL_TYPE_GEOKEY:
  135. RET_GEOKEY_VAL(GT_MODEL_TYPE, gt_model_type);
  136. break;
  137. case TIFF_GT_RASTER_TYPE_GEOKEY:
  138. RET_GEOKEY_VAL(GT_RASTER_TYPE, gt_raster_type);
  139. break;
  140. case TIFF_GEOG_LINEAR_UNITS_GEOKEY:
  141. case TIFF_PROJ_LINEAR_UNITS_GEOKEY:
  142. case TIFF_VERTICAL_UNITS_GEOKEY:
  143. RET_GEOKEY_VAL(LINEAR_UNIT, linear_unit);
  144. break;
  145. case TIFF_GEOG_ANGULAR_UNITS_GEOKEY:
  146. case TIFF_GEOG_AZIMUTH_UNITS_GEOKEY:
  147. RET_GEOKEY_VAL(ANGULAR_UNIT, angular_unit);
  148. break;
  149. case TIFF_GEOGRAPHIC_TYPE_GEOKEY:
  150. RET_GEOKEY_VAL(GCS_TYPE, gcs_type);
  151. RET_GEOKEY_VAL(GCSE_TYPE, gcse_type);
  152. break;
  153. case TIFF_GEOG_GEODETIC_DATUM_GEOKEY:
  154. RET_GEOKEY_VAL(GEODETIC_DATUM, geodetic_datum);
  155. RET_GEOKEY_VAL(GEODETIC_DATUM_E, geodetic_datum_e);
  156. break;
  157. case TIFF_GEOG_ELLIPSOID_GEOKEY:
  158. RET_GEOKEY_VAL(ELLIPSOID, ellipsoid);
  159. break;
  160. case TIFF_GEOG_PRIME_MERIDIAN_GEOKEY:
  161. RET_GEOKEY_VAL(PRIME_MERIDIAN, prime_meridian);
  162. break;
  163. case TIFF_PROJECTED_CS_TYPE_GEOKEY:
  164. return av_strdup(search_keyval(ff_tiff_proj_cs_type_codes, FF_ARRAY_ELEMS(ff_tiff_proj_cs_type_codes), val));
  165. break;
  166. case TIFF_PROJECTION_GEOKEY:
  167. return av_strdup(search_keyval(ff_tiff_projection_codes, FF_ARRAY_ELEMS(ff_tiff_projection_codes), val));
  168. break;
  169. case TIFF_PROJ_COORD_TRANS_GEOKEY:
  170. RET_GEOKEY_VAL(COORD_TRANS, coord_trans);
  171. break;
  172. case TIFF_VERTICAL_CS_TYPE_GEOKEY:
  173. RET_GEOKEY_VAL(VERT_CS, vert_cs);
  174. RET_GEOKEY_VAL(ORTHO_VERT_CS, ortho_vert_cs);
  175. break;
  176. }
  177. ap = av_malloc(14);
  178. if (ap)
  179. snprintf(ap, 14, "Unknown-%d", val);
  180. return ap;
  181. }
  182. static char *doubles2str(double *dp, int count, const char *sep)
  183. {
  184. int i;
  185. char *ap, *ap0;
  186. int component_len;
  187. if (!sep) sep = ", ";
  188. component_len = 15 + strlen(sep);
  189. ap = av_malloc(component_len * count);
  190. if (!ap)
  191. return NULL;
  192. ap0 = ap;
  193. ap[0] = '\0';
  194. for (i = 0; i < count; i++) {
  195. unsigned l = snprintf(ap, component_len, "%f%s", dp[i], sep);
  196. if(l >= component_len) {
  197. av_free(ap0);
  198. return NULL;
  199. }
  200. ap += l;
  201. }
  202. ap0[strlen(ap0) - strlen(sep)] = '\0';
  203. return ap0;
  204. }
  205. static char *shorts2str(int16_t *sp, int count, const char *sep)
  206. {
  207. int i;
  208. char *ap, *ap0;
  209. if (!sep) sep = ", ";
  210. ap = av_malloc((5 + strlen(sep)) * count);
  211. if (!ap)
  212. return NULL;
  213. ap0 = ap;
  214. ap[0] = '\0';
  215. for (i = 0; i < count; i++) {
  216. int l = snprintf(ap, 5 + strlen(sep), "%d%s", sp[i], sep);
  217. ap += l;
  218. }
  219. ap0[strlen(ap0) - strlen(sep)] = '\0';
  220. return ap0;
  221. }
  222. static int add_doubles_metadata(int count,
  223. const char *name, const char *sep,
  224. TiffContext *s)
  225. {
  226. char *ap;
  227. int i;
  228. double *dp;
  229. if (count >= INT_MAX / sizeof(int64_t))
  230. return AVERROR_INVALIDDATA;
  231. if (bytestream2_get_bytes_left(&s->gb) < count * sizeof(int64_t))
  232. return AVERROR_INVALIDDATA;
  233. dp = av_malloc(count * sizeof(double));
  234. if (!dp)
  235. return AVERROR(ENOMEM);
  236. for (i = 0; i < count; i++)
  237. dp[i] = tget_double(&s->gb, s->le);
  238. ap = doubles2str(dp, count, sep);
  239. av_freep(&dp);
  240. if (!ap)
  241. return AVERROR(ENOMEM);
  242. av_dict_set(&s->picture.metadata, name, ap, AV_DICT_DONT_STRDUP_VAL);
  243. return 0;
  244. }
  245. static int add_shorts_metadata(int count, const char *name,
  246. const char *sep, TiffContext *s)
  247. {
  248. char *ap;
  249. int i;
  250. int16_t *sp;
  251. if (count >= INT_MAX / sizeof(int16_t))
  252. return AVERROR_INVALIDDATA;
  253. if (bytestream2_get_bytes_left(&s->gb) < count * sizeof(int16_t))
  254. return AVERROR_INVALIDDATA;
  255. sp = av_malloc(count * sizeof(int16_t));
  256. if (!sp)
  257. return AVERROR(ENOMEM);
  258. for (i = 0; i < count; i++)
  259. sp[i] = tget_short(&s->gb, s->le);
  260. ap = shorts2str(sp, count, sep);
  261. av_freep(&sp);
  262. if (!ap)
  263. return AVERROR(ENOMEM);
  264. av_dict_set(&s->picture.metadata, name, ap, AV_DICT_DONT_STRDUP_VAL);
  265. return 0;
  266. }
  267. static int add_string_metadata(int count, const char *name,
  268. TiffContext *s)
  269. {
  270. char *value;
  271. if (bytestream2_get_bytes_left(&s->gb) < count)
  272. return AVERROR_INVALIDDATA;
  273. value = av_malloc(count + 1);
  274. if (!value)
  275. return AVERROR(ENOMEM);
  276. bytestream2_get_bufferu(&s->gb, value, count);
  277. value[count] = 0;
  278. av_dict_set(&s->picture.metadata, name, value, AV_DICT_DONT_STRDUP_VAL);
  279. return 0;
  280. }
  281. static int add_metadata(int count, int type,
  282. const char *name, const char *sep, TiffContext *s)
  283. {
  284. switch(type) {
  285. case TIFF_DOUBLE: return add_doubles_metadata(count, name, sep, s);
  286. case TIFF_SHORT : return add_shorts_metadata(count, name, sep, s);
  287. case TIFF_STRING: return add_string_metadata(count, name, s);
  288. default : return AVERROR_INVALIDDATA;
  289. };
  290. }
  291. #if CONFIG_ZLIB
  292. static int tiff_uncompress(uint8_t *dst, unsigned long *len, const uint8_t *src,
  293. int size)
  294. {
  295. z_stream zstream = { 0 };
  296. int zret;
  297. zstream.next_in = (uint8_t *)src;
  298. zstream.avail_in = size;
  299. zstream.next_out = dst;
  300. zstream.avail_out = *len;
  301. zret = inflateInit(&zstream);
  302. if (zret != Z_OK) {
  303. av_log(NULL, AV_LOG_ERROR, "Inflate init error: %d\n", zret);
  304. return zret;
  305. }
  306. zret = inflate(&zstream, Z_SYNC_FLUSH);
  307. inflateEnd(&zstream);
  308. *len = zstream.total_out;
  309. return zret == Z_STREAM_END ? Z_OK : zret;
  310. }
  311. #endif
  312. static void av_always_inline horizontal_fill(unsigned int bpp, uint8_t* dst,
  313. int usePtr, const uint8_t *src,
  314. uint8_t c, int width, int offset)
  315. {
  316. switch (bpp) {
  317. case 1:
  318. while (--width >= 0) {
  319. dst[(width+offset)*8+7] = (usePtr ? src[width] : c) & 0x1;
  320. dst[(width+offset)*8+6] = (usePtr ? src[width] : c) >> 1 & 0x1;
  321. dst[(width+offset)*8+5] = (usePtr ? src[width] : c) >> 2 & 0x1;
  322. dst[(width+offset)*8+4] = (usePtr ? src[width] : c) >> 3 & 0x1;
  323. dst[(width+offset)*8+3] = (usePtr ? src[width] : c) >> 4 & 0x1;
  324. dst[(width+offset)*8+2] = (usePtr ? src[width] : c) >> 5 & 0x1;
  325. dst[(width+offset)*8+1] = (usePtr ? src[width] : c) >> 6 & 0x1;
  326. dst[(width+offset)*8+0] = (usePtr ? src[width] : c) >> 7;
  327. }
  328. break;
  329. case 2:
  330. while (--width >= 0) {
  331. dst[(width+offset)*4+3] = (usePtr ? src[width] : c) & 0x3;
  332. dst[(width+offset)*4+2] = (usePtr ? src[width] : c) >> 2 & 0x3;
  333. dst[(width+offset)*4+1] = (usePtr ? src[width] : c) >> 4 & 0x3;
  334. dst[(width+offset)*4+0] = (usePtr ? src[width] : c) >> 6;
  335. }
  336. break;
  337. case 4:
  338. while (--width >= 0) {
  339. dst[(width+offset)*2+1] = (usePtr ? src[width] : c) & 0xF;
  340. dst[(width+offset)*2+0] = (usePtr ? src[width] : c) >> 4;
  341. }
  342. break;
  343. default:
  344. if (usePtr) {
  345. memcpy(dst + offset, src, width);
  346. } else {
  347. memset(dst + offset, c, width);
  348. }
  349. }
  350. }
  351. static int tiff_unpack_strip(TiffContext *s, uint8_t *dst, int stride,
  352. const uint8_t *src, int size, int lines)
  353. {
  354. int c, line, pixels, code;
  355. const uint8_t *ssrc = src;
  356. int width = ((s->width * s->bpp) + 7) >> 3;
  357. if (size <= 0)
  358. return AVERROR_INVALIDDATA;
  359. #if CONFIG_ZLIB
  360. if (s->compr == TIFF_DEFLATE || s->compr == TIFF_ADOBE_DEFLATE) {
  361. uint8_t *zbuf;
  362. unsigned long outlen;
  363. int ret;
  364. outlen = width * lines;
  365. zbuf = av_malloc(outlen);
  366. if (!zbuf)
  367. return AVERROR(ENOMEM);
  368. ret = tiff_uncompress(zbuf, &outlen, src, size);
  369. if (ret != Z_OK) {
  370. av_log(s->avctx, AV_LOG_ERROR,
  371. "Uncompressing failed (%lu of %lu) with error %d\n", outlen,
  372. (unsigned long)width * lines, ret);
  373. av_free(zbuf);
  374. return -1;
  375. }
  376. src = zbuf;
  377. for (line = 0; line < lines; line++) {
  378. if(s->bpp < 8 && s->avctx->pix_fmt == AV_PIX_FMT_PAL8){
  379. horizontal_fill(s->bpp, dst, 1, src, 0, width, 0);
  380. }else{
  381. memcpy(dst, src, width);
  382. }
  383. dst += stride;
  384. src += width;
  385. }
  386. av_free(zbuf);
  387. return 0;
  388. }
  389. #endif
  390. if (s->compr == TIFF_LZW) {
  391. if (ff_lzw_decode_init(s->lzw, 8, src, size, FF_LZW_TIFF) < 0) {
  392. av_log(s->avctx, AV_LOG_ERROR, "Error initializing LZW decoder\n");
  393. return -1;
  394. }
  395. }
  396. if (s->compr == TIFF_CCITT_RLE || s->compr == TIFF_G3
  397. || s->compr == TIFF_G4) {
  398. int i, ret = 0;
  399. uint8_t *src2 = av_malloc((unsigned)size +
  400. FF_INPUT_BUFFER_PADDING_SIZE);
  401. if (!src2) {
  402. av_log(s->avctx, AV_LOG_ERROR,
  403. "Error allocating temporary buffer\n");
  404. return AVERROR(ENOMEM);
  405. }
  406. if (s->fax_opts & 2) {
  407. av_log(s->avctx, AV_LOG_ERROR,
  408. "Uncompressed fax mode is not supported (yet)\n");
  409. av_free(src2);
  410. return -1;
  411. }
  412. if (!s->fill_order) {
  413. memcpy(src2, src, size);
  414. } else {
  415. for (i = 0; i < size; i++)
  416. src2[i] = ff_reverse[src[i]];
  417. }
  418. memset(src2 + size, 0, FF_INPUT_BUFFER_PADDING_SIZE);
  419. switch (s->compr) {
  420. case TIFF_CCITT_RLE:
  421. case TIFF_G3:
  422. case TIFF_G4:
  423. ret = ff_ccitt_unpack(s->avctx, src2, size, dst, lines, stride,
  424. s->compr, s->fax_opts);
  425. break;
  426. }
  427. if (s->bpp < 8 && s->avctx->pix_fmt == AV_PIX_FMT_PAL8)
  428. for (line = 0; line < lines; line++) {
  429. horizontal_fill(s->bpp, dst, 1, dst, 0, width, 0);
  430. dst += stride;
  431. }
  432. av_free(src2);
  433. return ret;
  434. }
  435. for (line = 0; line < lines; line++) {
  436. if (src - ssrc > size) {
  437. av_log(s->avctx, AV_LOG_ERROR, "Source data overread\n");
  438. return -1;
  439. }
  440. switch (s->compr) {
  441. case TIFF_RAW:
  442. if (ssrc + size - src < width)
  443. return AVERROR_INVALIDDATA;
  444. if (!s->fill_order) {
  445. horizontal_fill(s->bpp * (s->avctx->pix_fmt == AV_PIX_FMT_PAL8),
  446. dst, 1, src, 0, width, 0);
  447. } else {
  448. int i;
  449. for (i = 0; i < width; i++)
  450. dst[i] = ff_reverse[src[i]];
  451. }
  452. src += width;
  453. break;
  454. case TIFF_PACKBITS:
  455. for (pixels = 0; pixels < width;) {
  456. if (ssrc + size - src < 2) {
  457. av_log(s->avctx, AV_LOG_ERROR, "Read went out of bounds\n");
  458. return AVERROR_INVALIDDATA;
  459. }
  460. code = (int8_t) * src++;
  461. if (code >= 0) {
  462. code++;
  463. if (pixels + code > width) {
  464. av_log(s->avctx, AV_LOG_ERROR,
  465. "Copy went out of bounds\n");
  466. return -1;
  467. }
  468. if (ssrc + size - src < code) {
  469. av_log(s->avctx, AV_LOG_ERROR, "Read went out of bounds\n");
  470. return AVERROR_INVALIDDATA;
  471. }
  472. horizontal_fill(s->bpp * (s->avctx->pix_fmt == AV_PIX_FMT_PAL8),
  473. dst, 1, src, 0, code, pixels);
  474. src += code;
  475. pixels += code;
  476. } else if (code != -128) { // -127..-1
  477. code = (-code) + 1;
  478. if (pixels + code > width) {
  479. av_log(s->avctx, AV_LOG_ERROR,
  480. "Run went out of bounds\n");
  481. return -1;
  482. }
  483. c = *src++;
  484. horizontal_fill(s->bpp * (s->avctx->pix_fmt == AV_PIX_FMT_PAL8),
  485. dst, 0, NULL, c, code, pixels);
  486. pixels += code;
  487. }
  488. }
  489. break;
  490. case TIFF_LZW:
  491. pixels = ff_lzw_decode(s->lzw, dst, width);
  492. if (pixels < width) {
  493. av_log(s->avctx, AV_LOG_ERROR, "Decoded only %i bytes of %i\n",
  494. pixels, width);
  495. return -1;
  496. }
  497. if (s->bpp < 8 && s->avctx->pix_fmt == AV_PIX_FMT_PAL8)
  498. horizontal_fill(s->bpp, dst, 1, dst, 0, width, 0);
  499. break;
  500. }
  501. dst += stride;
  502. }
  503. return 0;
  504. }
  505. static int init_image(TiffContext *s)
  506. {
  507. int i, ret;
  508. uint32_t *pal;
  509. switch (s->bpp * 10 + s->bppcount) {
  510. case 11:
  511. if (!s->palette_is_set) {
  512. s->avctx->pix_fmt = AV_PIX_FMT_MONOBLACK;
  513. break;
  514. }
  515. case 21:
  516. case 41:
  517. case 81:
  518. s->avctx->pix_fmt = AV_PIX_FMT_PAL8;
  519. break;
  520. case 243:
  521. s->avctx->pix_fmt = AV_PIX_FMT_RGB24;
  522. break;
  523. case 161:
  524. s->avctx->pix_fmt = s->le ? AV_PIX_FMT_GRAY16LE : AV_PIX_FMT_GRAY16BE;
  525. break;
  526. case 162:
  527. s->avctx->pix_fmt = AV_PIX_FMT_GRAY8A;
  528. break;
  529. case 324:
  530. s->avctx->pix_fmt = AV_PIX_FMT_RGBA;
  531. break;
  532. case 483:
  533. s->avctx->pix_fmt = s->le ? AV_PIX_FMT_RGB48LE : AV_PIX_FMT_RGB48BE;
  534. break;
  535. case 644:
  536. s->avctx->pix_fmt = s->le ? AV_PIX_FMT_RGBA64LE : AV_PIX_FMT_RGBA64BE;
  537. break;
  538. default:
  539. av_log(s->avctx, AV_LOG_ERROR,
  540. "This format is not supported (bpp=%d, bppcount=%d)\n",
  541. s->bpp, s->bppcount);
  542. return AVERROR_INVALIDDATA;
  543. }
  544. if (s->width != s->avctx->width || s->height != s->avctx->height) {
  545. if ((ret = av_image_check_size(s->width, s->height, 0, s->avctx)) < 0)
  546. return ret;
  547. avcodec_set_dimensions(s->avctx, s->width, s->height);
  548. }
  549. if (s->picture.data[0])
  550. s->avctx->release_buffer(s->avctx, &s->picture);
  551. if ((ret = s->avctx->get_buffer(s->avctx, &s->picture)) < 0) {
  552. av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed\n");
  553. return ret;
  554. }
  555. if (s->avctx->pix_fmt == AV_PIX_FMT_PAL8) {
  556. if (s->palette_is_set) {
  557. memcpy(s->picture.data[1], s->palette, sizeof(s->palette));
  558. } else {
  559. /* make default grayscale pal */
  560. pal = (uint32_t *) s->picture.data[1];
  561. for (i = 0; i < 1<<s->bpp; i++)
  562. pal[i] = 0xFFU << 24 | i * 255 / ((1<<s->bpp) - 1) * 0x010101;
  563. }
  564. }
  565. return 0;
  566. }
  567. static int tiff_decode_tag(TiffContext *s)
  568. {
  569. unsigned tag, type, count, off, value = 0;
  570. int i, j, k, pos, start;
  571. int ret;
  572. uint32_t *pal;
  573. double *dp;
  574. tag = tget_short(&s->gb, s->le);
  575. type = tget_short(&s->gb, s->le);
  576. count = tget_long(&s->gb, s->le);
  577. off = tget_long(&s->gb, s->le);
  578. start = bytestream2_tell(&s->gb);
  579. if (type == 0 || type >= FF_ARRAY_ELEMS(type_sizes)) {
  580. av_log(s->avctx, AV_LOG_DEBUG, "Unknown tiff type (%u) encountered\n",
  581. type);
  582. return 0;
  583. }
  584. if (count == 1) {
  585. switch (type) {
  586. case TIFF_BYTE:
  587. case TIFF_SHORT:
  588. bytestream2_seek(&s->gb, -4, SEEK_CUR);
  589. value = tget(&s->gb, type, s->le);
  590. break;
  591. case TIFF_LONG:
  592. value = off;
  593. break;
  594. case TIFF_STRING:
  595. if (count <= 4) {
  596. bytestream2_seek(&s->gb, -4, SEEK_CUR);
  597. break;
  598. }
  599. default:
  600. value = UINT_MAX;
  601. bytestream2_seek(&s->gb, off, SEEK_SET);
  602. }
  603. } else {
  604. if (count <= 4 && type_sizes[type] * count <= 4) {
  605. bytestream2_seek(&s->gb, -4, SEEK_CUR);
  606. } else {
  607. bytestream2_seek(&s->gb, off, SEEK_SET);
  608. }
  609. }
  610. switch (tag) {
  611. case TIFF_WIDTH:
  612. s->width = value;
  613. break;
  614. case TIFF_HEIGHT:
  615. s->height = value;
  616. break;
  617. case TIFF_BPP:
  618. s->bppcount = count;
  619. if (count > 4) {
  620. av_log(s->avctx, AV_LOG_ERROR,
  621. "This format is not supported (bpp=%d, %d components)\n",
  622. s->bpp, count);
  623. return -1;
  624. }
  625. if (count == 1)
  626. s->bpp = value;
  627. else {
  628. switch (type) {
  629. case TIFF_BYTE:
  630. s->bpp = (off & 0xFF) + ((off >> 8) & 0xFF) +
  631. ((off >> 16) & 0xFF) + ((off >> 24) & 0xFF);
  632. break;
  633. case TIFF_SHORT:
  634. case TIFF_LONG:
  635. s->bpp = 0;
  636. if (bytestream2_get_bytes_left(&s->gb) < type_sizes[type] * count)
  637. return -1;
  638. for (i = 0; i < count; i++)
  639. s->bpp += tget(&s->gb, type, s->le);
  640. break;
  641. default:
  642. s->bpp = -1;
  643. }
  644. }
  645. break;
  646. case TIFF_SAMPLES_PER_PIXEL:
  647. if (count != 1) {
  648. av_log(s->avctx, AV_LOG_ERROR,
  649. "Samples per pixel requires a single value, many provided\n");
  650. return AVERROR_INVALIDDATA;
  651. }
  652. if (s->bppcount == 1)
  653. s->bpp *= value;
  654. s->bppcount = value;
  655. break;
  656. case TIFF_COMPR:
  657. s->compr = value;
  658. s->predictor = 0;
  659. switch (s->compr) {
  660. case TIFF_RAW:
  661. case TIFF_PACKBITS:
  662. case TIFF_LZW:
  663. case TIFF_CCITT_RLE:
  664. break;
  665. case TIFF_G3:
  666. case TIFF_G4:
  667. s->fax_opts = 0;
  668. break;
  669. case TIFF_DEFLATE:
  670. case TIFF_ADOBE_DEFLATE:
  671. #if CONFIG_ZLIB
  672. break;
  673. #else
  674. av_log(s->avctx, AV_LOG_ERROR, "Deflate: ZLib not compiled in\n");
  675. return -1;
  676. #endif
  677. case TIFF_JPEG:
  678. case TIFF_NEWJPEG:
  679. av_log(s->avctx, AV_LOG_ERROR,
  680. "JPEG compression is not supported\n");
  681. return -1;
  682. default:
  683. av_log(s->avctx, AV_LOG_ERROR, "Unknown compression method %i\n",
  684. s->compr);
  685. return -1;
  686. }
  687. break;
  688. case TIFF_ROWSPERSTRIP:
  689. if (type == TIFF_LONG && value == UINT_MAX)
  690. value = s->avctx->height;
  691. if (value < 1) {
  692. av_log(s->avctx, AV_LOG_ERROR,
  693. "Incorrect value of rows per strip\n");
  694. return -1;
  695. }
  696. s->rps = value;
  697. break;
  698. case TIFF_STRIP_OFFS:
  699. if (count == 1) {
  700. s->strippos = 0;
  701. s->stripoff = value;
  702. } else
  703. s->strippos = off;
  704. s->strips = count;
  705. if (s->strips == 1)
  706. s->rps = s->height;
  707. s->sot = type;
  708. if (s->strippos > bytestream2_size(&s->gb)) {
  709. av_log(s->avctx, AV_LOG_ERROR,
  710. "Tag referencing position outside the image\n");
  711. return -1;
  712. }
  713. break;
  714. case TIFF_STRIP_SIZE:
  715. if (count == 1) {
  716. s->stripsizesoff = 0;
  717. s->stripsize = value;
  718. s->strips = 1;
  719. } else {
  720. s->stripsizesoff = off;
  721. }
  722. s->strips = count;
  723. s->sstype = type;
  724. if (s->stripsizesoff > bytestream2_size(&s->gb)) {
  725. av_log(s->avctx, AV_LOG_ERROR,
  726. "Tag referencing position outside the image\n");
  727. return -1;
  728. }
  729. break;
  730. case TIFF_TILE_BYTE_COUNTS:
  731. case TIFF_TILE_LENGTH:
  732. case TIFF_TILE_OFFSETS:
  733. case TIFF_TILE_WIDTH:
  734. av_log(s->avctx, AV_LOG_ERROR, "Tiled images are not supported\n");
  735. return AVERROR_PATCHWELCOME;
  736. break;
  737. case TIFF_PREDICTOR:
  738. s->predictor = value;
  739. break;
  740. case TIFF_INVERT:
  741. switch (value) {
  742. case 0:
  743. s->invert = 1;
  744. break;
  745. case 1:
  746. s->invert = 0;
  747. break;
  748. case 2:
  749. case 3:
  750. break;
  751. default:
  752. av_log(s->avctx, AV_LOG_ERROR, "Color mode %d is not supported\n",
  753. value);
  754. return -1;
  755. }
  756. break;
  757. case TIFF_FILL_ORDER:
  758. if (value < 1 || value > 2) {
  759. av_log(s->avctx, AV_LOG_ERROR,
  760. "Unknown FillOrder value %d, trying default one\n", value);
  761. value = 1;
  762. }
  763. s->fill_order = value - 1;
  764. break;
  765. case TIFF_PAL:
  766. pal = (uint32_t *) s->palette;
  767. off = type_sizes[type];
  768. if (count / 3 > 256 || bytestream2_get_bytes_left(&s->gb) < count / 3 * off * 3)
  769. return -1;
  770. off = (type_sizes[type] - 1) << 3;
  771. for (k = 2; k >= 0; k--) {
  772. for (i = 0; i < count / 3; i++) {
  773. if (k == 2)
  774. pal[i] = 0xFFU << 24;
  775. j = (tget(&s->gb, type, s->le) >> off) << (k * 8);
  776. pal[i] |= j;
  777. }
  778. }
  779. s->palette_is_set = 1;
  780. break;
  781. case TIFF_PLANAR:
  782. if (value == 2) {
  783. av_log(s->avctx, AV_LOG_ERROR, "Planar format is not supported\n");
  784. return -1;
  785. }
  786. break;
  787. case TIFF_T4OPTIONS:
  788. if (s->compr == TIFF_G3)
  789. s->fax_opts = value;
  790. break;
  791. case TIFF_T6OPTIONS:
  792. if (s->compr == TIFF_G4)
  793. s->fax_opts = value;
  794. break;
  795. #define ADD_METADATA(count, name, sep)\
  796. if (ret = add_metadata(count, type, name, sep, s) < 0) {\
  797. av_log(s->avctx, AV_LOG_ERROR, "Error allocating temporary buffer\n");\
  798. return ret;\
  799. }
  800. case TIFF_MODEL_PIXEL_SCALE:
  801. ADD_METADATA(count, "ModelPixelScaleTag", NULL);
  802. break;
  803. case TIFF_MODEL_TRANSFORMATION:
  804. ADD_METADATA(count, "ModelTransformationTag", NULL);
  805. break;
  806. case TIFF_MODEL_TIEPOINT:
  807. ADD_METADATA(count, "ModelTiepointTag", NULL);
  808. break;
  809. case TIFF_GEO_KEY_DIRECTORY:
  810. ADD_METADATA(1, "GeoTIFF_Version", NULL);
  811. ADD_METADATA(2, "GeoTIFF_Key_Revision", ".");
  812. s->geotag_count = tget_short(&s->gb, s->le);
  813. if (s->geotag_count > count / 4 - 1) {
  814. s->geotag_count = count / 4 - 1;
  815. av_log(s->avctx, AV_LOG_WARNING, "GeoTIFF key directory buffer shorter than specified\n");
  816. }
  817. if (bytestream2_get_bytes_left(&s->gb) < s->geotag_count * sizeof(int16_t) * 4)
  818. return -1;
  819. s->geotags = av_mallocz(sizeof(TiffGeoTag) * s->geotag_count);
  820. if (!s->geotags) {
  821. av_log(s->avctx, AV_LOG_ERROR, "Error allocating temporary buffer\n");
  822. return AVERROR(ENOMEM);
  823. }
  824. for (i = 0; i < s->geotag_count; i++) {
  825. s->geotags[i].key = tget_short(&s->gb, s->le);
  826. s->geotags[i].type = tget_short(&s->gb, s->le);
  827. s->geotags[i].count = tget_short(&s->gb, s->le);
  828. if (!s->geotags[i].type)
  829. s->geotags[i].val = get_geokey_val(s->geotags[i].key, tget_short(&s->gb, s->le));
  830. else
  831. s->geotags[i].offset = tget_short(&s->gb, s->le);
  832. }
  833. break;
  834. case TIFF_GEO_DOUBLE_PARAMS:
  835. if (count >= INT_MAX / sizeof(int64_t))
  836. return AVERROR_INVALIDDATA;
  837. if (bytestream2_get_bytes_left(&s->gb) < count * sizeof(int64_t))
  838. return AVERROR_INVALIDDATA;
  839. dp = av_malloc(count * sizeof(double));
  840. if (!dp) {
  841. av_log(s->avctx, AV_LOG_ERROR, "Error allocating temporary buffer\n");
  842. return AVERROR(ENOMEM);
  843. }
  844. for (i = 0; i < count; i++)
  845. dp[i] = tget_double(&s->gb, s->le);
  846. for (i = 0; i < s->geotag_count; i++) {
  847. if (s->geotags[i].type == TIFF_GEO_DOUBLE_PARAMS) {
  848. if (s->geotags[i].count == 0
  849. || s->geotags[i].offset + s->geotags[i].count > count) {
  850. av_log(s->avctx, AV_LOG_WARNING, "Invalid GeoTIFF key %d\n", s->geotags[i].key);
  851. } else {
  852. char *ap = doubles2str(&dp[s->geotags[i].offset], s->geotags[i].count, ", ");
  853. if (!ap) {
  854. av_log(s->avctx, AV_LOG_ERROR, "Error allocating temporary buffer\n");
  855. av_freep(&dp);
  856. return AVERROR(ENOMEM);
  857. }
  858. s->geotags[i].val = ap;
  859. }
  860. }
  861. }
  862. av_freep(&dp);
  863. break;
  864. case TIFF_GEO_ASCII_PARAMS:
  865. pos = bytestream2_tell(&s->gb);
  866. for (i = 0; i < s->geotag_count; i++) {
  867. if (s->geotags[i].type == TIFF_GEO_ASCII_PARAMS) {
  868. if (s->geotags[i].count == 0
  869. || s->geotags[i].offset + s->geotags[i].count > count) {
  870. av_log(s->avctx, AV_LOG_WARNING, "Invalid GeoTIFF key %d\n", s->geotags[i].key);
  871. } else {
  872. char *ap;
  873. bytestream2_seek(&s->gb, pos + s->geotags[i].offset, SEEK_SET);
  874. if (bytestream2_get_bytes_left(&s->gb) < s->geotags[i].count)
  875. return -1;
  876. ap = av_malloc(s->geotags[i].count);
  877. if (!ap) {
  878. av_log(s->avctx, AV_LOG_ERROR, "Error allocating temporary buffer\n");
  879. return AVERROR(ENOMEM);
  880. }
  881. bytestream2_get_bufferu(&s->gb, ap, s->geotags[i].count);
  882. ap[s->geotags[i].count - 1] = '\0'; //replace the "|" delimiter with a 0 byte
  883. s->geotags[i].val = ap;
  884. }
  885. }
  886. }
  887. break;
  888. case TIFF_ARTIST:
  889. ADD_METADATA(count, "artist", NULL);
  890. break;
  891. case TIFF_COPYRIGHT:
  892. ADD_METADATA(count, "copyright", NULL);
  893. break;
  894. case TIFF_DATE:
  895. ADD_METADATA(count, "date", NULL);
  896. break;
  897. case TIFF_DOCUMENT_NAME:
  898. ADD_METADATA(count, "document_name", NULL);
  899. break;
  900. case TIFF_HOST_COMPUTER:
  901. ADD_METADATA(count, "computer", NULL);
  902. break;
  903. case TIFF_IMAGE_DESCRIPTION:
  904. ADD_METADATA(count, "description", NULL);
  905. break;
  906. case TIFF_MAKE:
  907. ADD_METADATA(count, "make", NULL);
  908. break;
  909. case TIFF_MODEL:
  910. ADD_METADATA(count, "model", NULL);
  911. break;
  912. case TIFF_PAGE_NAME:
  913. ADD_METADATA(count, "page_name", NULL);
  914. break;
  915. case TIFF_PAGE_NUMBER:
  916. ADD_METADATA(count, "page_number", " / ");
  917. break;
  918. case TIFF_SOFTWARE_NAME:
  919. ADD_METADATA(count, "software", NULL);
  920. break;
  921. default:
  922. av_log(s->avctx, AV_LOG_DEBUG, "Unknown or unsupported tag %d/0X%0X\n",
  923. tag, tag);
  924. }
  925. bytestream2_seek(&s->gb, start, SEEK_SET);
  926. return 0;
  927. }
  928. static int decode_frame(AVCodecContext *avctx,
  929. void *data, int *data_size, AVPacket *avpkt)
  930. {
  931. TiffContext *const s = avctx->priv_data;
  932. AVFrame *picture = data;
  933. AVFrame *const p = &s->picture;
  934. unsigned off;
  935. int id, le, ret;
  936. int i, j, entries;
  937. int stride;
  938. unsigned soff, ssize;
  939. uint8_t *dst;
  940. GetByteContext stripsizes;
  941. GetByteContext stripdata;
  942. bytestream2_init(&s->gb, avpkt->data, avpkt->size);
  943. //parse image header
  944. if (avpkt->size < 8)
  945. return AVERROR_INVALIDDATA;
  946. id = bytestream2_get_le16u(&s->gb);
  947. if (id == 0x4949)
  948. le = 1;
  949. else if (id == 0x4D4D)
  950. le = 0;
  951. else {
  952. av_log(avctx, AV_LOG_ERROR, "TIFF header not found\n");
  953. return -1;
  954. }
  955. s->le = le;
  956. // TIFF_BPP is not a required tag and defaults to 1
  957. s->bppcount = s->bpp = 1;
  958. s->invert = 0;
  959. s->compr = TIFF_RAW;
  960. s->fill_order = 0;
  961. free_geotags(s);
  962. /* free existing metadata */
  963. av_dict_free(&s->picture.metadata);
  964. // As TIFF 6.0 specification puts it "An arbitrary but carefully chosen number
  965. // that further identifies the file as a TIFF file"
  966. if (tget_short(&s->gb, le) != 42) {
  967. av_log(avctx, AV_LOG_ERROR,
  968. "The answer to life, universe and everything is not correct!\n");
  969. return -1;
  970. }
  971. // Reset these offsets so we can tell if they were set this frame
  972. s->stripsizesoff = s->strippos = 0;
  973. /* parse image file directory */
  974. off = tget_long(&s->gb, le);
  975. if (off >= UINT_MAX - 14 || avpkt->size < off + 14) {
  976. av_log(avctx, AV_LOG_ERROR, "IFD offset is greater than image size\n");
  977. return AVERROR_INVALIDDATA;
  978. }
  979. bytestream2_seek(&s->gb, off, SEEK_SET);
  980. entries = tget_short(&s->gb, le);
  981. if (bytestream2_get_bytes_left(&s->gb) < entries * 12)
  982. return AVERROR_INVALIDDATA;
  983. for (i = 0; i < entries; i++) {
  984. if (tiff_decode_tag(s) < 0)
  985. return -1;
  986. }
  987. for (i = 0; i<s->geotag_count; i++) {
  988. const char *keyname = get_geokey_name(s->geotags[i].key);
  989. if (!keyname) {
  990. av_log(avctx, AV_LOG_WARNING, "Unknown or unsupported GeoTIFF key %d\n", s->geotags[i].key);
  991. continue;
  992. }
  993. if (get_geokey_type(s->geotags[i].key) != s->geotags[i].type) {
  994. av_log(avctx, AV_LOG_WARNING, "Type of GeoTIFF key %d is wrong\n", s->geotags[i].key);
  995. continue;
  996. }
  997. ret = av_dict_set(&s->picture.metadata, keyname, s->geotags[i].val, 0);
  998. if (ret<0) {
  999. av_log(avctx, AV_LOG_ERROR, "Writing metadata with key '%s' failed\n", keyname);
  1000. return ret;
  1001. }
  1002. }
  1003. if (!s->strippos && !s->stripoff) {
  1004. av_log(avctx, AV_LOG_ERROR, "Image data is missing\n");
  1005. return -1;
  1006. }
  1007. /* now we have the data and may start decoding */
  1008. if ((ret = init_image(s)) < 0)
  1009. return ret;
  1010. if (s->strips == 1 && !s->stripsize) {
  1011. av_log(avctx, AV_LOG_WARNING, "Image data size missing\n");
  1012. s->stripsize = avpkt->size - s->stripoff;
  1013. }
  1014. stride = p->linesize[0];
  1015. dst = p->data[0];
  1016. if (s->stripsizesoff) {
  1017. if (s->stripsizesoff >= avpkt->size)
  1018. return AVERROR_INVALIDDATA;
  1019. bytestream2_init(&stripsizes, avpkt->data + s->stripsizesoff, avpkt->size - s->stripsizesoff);
  1020. }
  1021. if (s->strippos) {
  1022. if (s->strippos >= avpkt->size)
  1023. return AVERROR_INVALIDDATA;
  1024. bytestream2_init(&stripdata, avpkt->data + s->strippos, avpkt->size - s->strippos);
  1025. }
  1026. for (i = 0; i < s->height; i += s->rps) {
  1027. if (s->stripsizesoff)
  1028. ssize = tget(&stripsizes, s->sstype, s->le);
  1029. else
  1030. ssize = s->stripsize;
  1031. if (s->strippos)
  1032. soff = tget(&stripdata, s->sot, s->le);
  1033. else
  1034. soff = s->stripoff;
  1035. if (soff > avpkt->size || ssize > avpkt->size - soff) {
  1036. av_log(avctx, AV_LOG_ERROR, "Invalid strip size/offset\n");
  1037. return -1;
  1038. }
  1039. if (tiff_unpack_strip(s, dst, stride, avpkt->data + soff, ssize,
  1040. FFMIN(s->rps, s->height - i)) < 0)
  1041. break;
  1042. dst += s->rps * stride;
  1043. }
  1044. if (s->predictor == 2) {
  1045. dst = p->data[0];
  1046. soff = s->bpp >> 3;
  1047. ssize = s->width * soff;
  1048. if (s->avctx->pix_fmt == AV_PIX_FMT_RGB48LE ||
  1049. s->avctx->pix_fmt == AV_PIX_FMT_RGBA64LE) {
  1050. for (i = 0; i < s->height; i++) {
  1051. for (j = soff; j < ssize; j += 2)
  1052. AV_WL16(dst + j, AV_RL16(dst + j) + AV_RL16(dst + j - soff));
  1053. dst += stride;
  1054. }
  1055. } else if (s->avctx->pix_fmt == AV_PIX_FMT_RGB48BE ||
  1056. s->avctx->pix_fmt == AV_PIX_FMT_RGBA64BE) {
  1057. for (i = 0; i < s->height; i++) {
  1058. for (j = soff; j < ssize; j += 2)
  1059. AV_WB16(dst + j, AV_RB16(dst + j) + AV_RB16(dst + j - soff));
  1060. dst += stride;
  1061. }
  1062. } else {
  1063. for (i = 0; i < s->height; i++) {
  1064. for (j = soff; j < ssize; j++)
  1065. dst[j] += dst[j - soff];
  1066. dst += stride;
  1067. }
  1068. }
  1069. }
  1070. if (s->invert) {
  1071. dst = s->picture.data[0];
  1072. for (i = 0; i < s->height; i++) {
  1073. for (j = 0; j < s->picture.linesize[0]; j++)
  1074. dst[j] = (s->avctx->pix_fmt == AV_PIX_FMT_PAL8 ? (1<<s->bpp) - 1 : 255) - dst[j];
  1075. dst += s->picture.linesize[0];
  1076. }
  1077. }
  1078. *picture = s->picture;
  1079. *data_size = sizeof(AVPicture);
  1080. return avpkt->size;
  1081. }
  1082. static av_cold int tiff_init(AVCodecContext *avctx)
  1083. {
  1084. TiffContext *s = avctx->priv_data;
  1085. s->width = 0;
  1086. s->height = 0;
  1087. s->avctx = avctx;
  1088. avcodec_get_frame_defaults(&s->picture);
  1089. avctx->coded_frame = &s->picture;
  1090. ff_lzw_decode_open(&s->lzw);
  1091. ff_ccitt_unpack_init();
  1092. return 0;
  1093. }
  1094. static av_cold int tiff_end(AVCodecContext *avctx)
  1095. {
  1096. TiffContext *const s = avctx->priv_data;
  1097. free_geotags(s);
  1098. if (avctx->coded_frame && avctx->coded_frame->metadata)
  1099. av_dict_free(&avctx->coded_frame->metadata);
  1100. ff_lzw_decode_close(&s->lzw);
  1101. if (s->picture.data[0])
  1102. avctx->release_buffer(avctx, &s->picture);
  1103. return 0;
  1104. }
  1105. AVCodec ff_tiff_decoder = {
  1106. .name = "tiff",
  1107. .type = AVMEDIA_TYPE_VIDEO,
  1108. .id = AV_CODEC_ID_TIFF,
  1109. .priv_data_size = sizeof(TiffContext),
  1110. .init = tiff_init,
  1111. .close = tiff_end,
  1112. .decode = decode_frame,
  1113. .capabilities = CODEC_CAP_DR1,
  1114. .long_name = NULL_IF_CONFIG_SMALL("TIFF image"),
  1115. };