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.

728 lines
22KB

  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. #if CONFIG_ZLIB
  27. #include <zlib.h>
  28. #endif
  29. #include "lzw.h"
  30. #include "tiff.h"
  31. #include "faxcompr.h"
  32. #include "libavutil/common.h"
  33. #include "libavutil/intreadwrite.h"
  34. #include "libavutil/imgutils.h"
  35. typedef struct TiffContext {
  36. AVCodecContext *avctx;
  37. AVFrame picture;
  38. int width, height;
  39. unsigned int bpp, bppcount;
  40. uint32_t palette[256];
  41. int palette_is_set;
  42. int le;
  43. enum TiffCompr compr;
  44. int invert;
  45. int fax_opts;
  46. int predictor;
  47. int fill_order;
  48. int strips, rps, sstype;
  49. int sot;
  50. const uint8_t* stripdata;
  51. const uint8_t* stripsizes;
  52. int stripsize, stripoff;
  53. LZWState *lzw;
  54. } TiffContext;
  55. static int tget_short(const uint8_t **p, int le){
  56. int v = le ? AV_RL16(*p) : AV_RB16(*p);
  57. *p += 2;
  58. return v;
  59. }
  60. static int tget_long(const uint8_t **p, int le){
  61. int v = le ? AV_RL32(*p) : AV_RB32(*p);
  62. *p += 4;
  63. return v;
  64. }
  65. static int tget(const uint8_t **p, int type, int le){
  66. switch(type){
  67. case TIFF_BYTE : return *(*p)++;
  68. case TIFF_SHORT: return tget_short(p, le);
  69. case TIFF_LONG : return tget_long (p, le);
  70. default : return -1;
  71. }
  72. }
  73. #if CONFIG_ZLIB
  74. static int tiff_uncompress(uint8_t *dst, unsigned long *len, const uint8_t *src, int size)
  75. {
  76. z_stream zstream;
  77. int zret;
  78. memset(&zstream, 0, sizeof(zstream));
  79. zstream.next_in = src;
  80. zstream.avail_in = size;
  81. zstream.next_out = dst;
  82. zstream.avail_out = *len;
  83. zret = inflateInit(&zstream);
  84. if (zret != Z_OK) {
  85. av_log(NULL, AV_LOG_ERROR, "Inflate init error: %d\n", zret);
  86. return zret;
  87. }
  88. zret = inflate(&zstream, Z_SYNC_FLUSH);
  89. inflateEnd(&zstream);
  90. *len = zstream.total_out;
  91. return zret == Z_STREAM_END ? Z_OK : zret;
  92. }
  93. #endif
  94. static void av_always_inline horizontal_fill(unsigned int bpp, uint8_t* dst,
  95. int usePtr, const uint8_t *src,
  96. uint8_t c, int width, int offset)
  97. {
  98. switch (bpp) {
  99. case 1:
  100. while (--width >= 0) {
  101. dst[(width+offset)*8+7] = (usePtr ? src[width] : c) & 0x1;
  102. dst[(width+offset)*8+6] = (usePtr ? src[width] : c) >> 1 & 0x1;
  103. dst[(width+offset)*8+5] = (usePtr ? src[width] : c) >> 2 & 0x1;
  104. dst[(width+offset)*8+4] = (usePtr ? src[width] : c) >> 3 & 0x1;
  105. dst[(width+offset)*8+3] = (usePtr ? src[width] : c) >> 4 & 0x1;
  106. dst[(width+offset)*8+2] = (usePtr ? src[width] : c) >> 5 & 0x1;
  107. dst[(width+offset)*8+1] = (usePtr ? src[width] : c) >> 6 & 0x1;
  108. dst[(width+offset)*8+0] = (usePtr ? src[width] : c) >> 7;
  109. }
  110. break;
  111. case 2:
  112. while (--width >= 0) {
  113. dst[(width+offset)*4+3] = (usePtr ? src[width] : c) & 0x3;
  114. dst[(width+offset)*4+2] = (usePtr ? src[width] : c) >> 2 & 0x3;
  115. dst[(width+offset)*4+1] = (usePtr ? src[width] : c) >> 4 & 0x3;
  116. dst[(width+offset)*4+0] = (usePtr ? src[width] : c) >> 6;
  117. }
  118. break;
  119. case 4:
  120. while (--width >= 0) {
  121. dst[(width+offset)*2+1] = (usePtr ? src[width] : c) & 0xF;
  122. dst[(width+offset)*2+0] = (usePtr ? src[width] : c) >> 4;
  123. }
  124. break;
  125. default:
  126. if (usePtr) {
  127. memcpy(dst + offset, src, width);
  128. } else {
  129. memset(dst + offset, c, width);
  130. }
  131. }
  132. }
  133. static int tiff_unpack_strip(TiffContext *s, uint8_t* dst, int stride, const uint8_t *src, int size, int lines){
  134. int c, line, pixels, code;
  135. const uint8_t *ssrc = src;
  136. int width = ((s->width * s->bpp) + 7) >> 3;
  137. #if CONFIG_ZLIB
  138. uint8_t *zbuf; unsigned long outlen;
  139. if(s->compr == TIFF_DEFLATE || s->compr == TIFF_ADOBE_DEFLATE){
  140. int ret;
  141. outlen = width * lines;
  142. zbuf = av_malloc(outlen);
  143. ret = tiff_uncompress(zbuf, &outlen, src, size);
  144. if(ret != Z_OK){
  145. av_log(s->avctx, AV_LOG_ERROR, "Uncompressing failed (%lu of %lu) with error %d\n", outlen, (unsigned long)width * lines, ret);
  146. av_free(zbuf);
  147. return -1;
  148. }
  149. src = zbuf;
  150. for(line = 0; line < lines; line++){
  151. if(s->bpp < 8 && s->avctx->pix_fmt == PIX_FMT_PAL8){
  152. horizontal_fill(s->bpp, dst, 1, src, 0, width, 0);
  153. }else{
  154. memcpy(dst, src, width);
  155. }
  156. dst += stride;
  157. src += width;
  158. }
  159. av_free(zbuf);
  160. return 0;
  161. }
  162. #endif
  163. if(s->compr == TIFF_LZW){
  164. if(ff_lzw_decode_init(s->lzw, 8, src, size, FF_LZW_TIFF) < 0){
  165. av_log(s->avctx, AV_LOG_ERROR, "Error initializing LZW decoder\n");
  166. return -1;
  167. }
  168. }
  169. if(s->compr == TIFF_CCITT_RLE || s->compr == TIFF_G3 || s->compr == TIFF_G4){
  170. int i, ret = 0;
  171. uint8_t *src2 = av_malloc(size + FF_INPUT_BUFFER_PADDING_SIZE);
  172. if(!src2 || (unsigned)size + FF_INPUT_BUFFER_PADDING_SIZE < (unsigned)size){
  173. av_log(s->avctx, AV_LOG_ERROR, "Error allocating temporary buffer\n");
  174. return -1;
  175. }
  176. if(s->fax_opts & 2){
  177. av_log(s->avctx, AV_LOG_ERROR, "Uncompressed fax mode is not supported (yet)\n");
  178. av_free(src2);
  179. return -1;
  180. }
  181. if(!s->fill_order){
  182. memcpy(src2, src, size);
  183. }else{
  184. for(i = 0; i < size; i++)
  185. src2[i] = av_reverse[src[i]];
  186. }
  187. memset(src2+size, 0, FF_INPUT_BUFFER_PADDING_SIZE);
  188. switch(s->compr){
  189. case TIFF_CCITT_RLE:
  190. case TIFF_G3:
  191. case TIFF_G4:
  192. ret = ff_ccitt_unpack(s->avctx, src2, size, dst, lines, stride, s->compr, s->fax_opts);
  193. break;
  194. }
  195. if (s->bpp < 8 && s->avctx->pix_fmt == PIX_FMT_PAL8)
  196. for (line = 0; line < lines; line++) {
  197. horizontal_fill(s->bpp, dst, 1, dst, 0, width, 0);
  198. dst += stride;
  199. }
  200. av_free(src2);
  201. return ret;
  202. }
  203. for(line = 0; line < lines; line++){
  204. if(src - ssrc > size){
  205. av_log(s->avctx, AV_LOG_ERROR, "Source data overread\n");
  206. return -1;
  207. }
  208. switch(s->compr){
  209. case TIFF_RAW:
  210. if (ssrc + size - src < width)
  211. return AVERROR_INVALIDDATA;
  212. if (!s->fill_order) {
  213. horizontal_fill(s->bpp * (s->avctx->pix_fmt == PIX_FMT_PAL8),
  214. dst, 1, src, 0, width, 0);
  215. } else {
  216. int i;
  217. for (i = 0; i < width; i++)
  218. dst[i] = av_reverse[src[i]];
  219. }
  220. src += width;
  221. break;
  222. case TIFF_PACKBITS:
  223. for(pixels = 0; pixels < width;){
  224. code = (int8_t)*src++;
  225. if(code >= 0){
  226. code++;
  227. if(pixels + code > width){
  228. av_log(s->avctx, AV_LOG_ERROR, "Copy went out of bounds\n");
  229. return -1;
  230. }
  231. horizontal_fill(s->bpp * (s->avctx->pix_fmt == PIX_FMT_PAL8),
  232. dst, 1, src, 0, code, pixels);
  233. src += code;
  234. pixels += code;
  235. }else if(code != -128){ // -127..-1
  236. code = (-code) + 1;
  237. if(pixels + code > width){
  238. av_log(s->avctx, AV_LOG_ERROR, "Run went out of bounds\n");
  239. return -1;
  240. }
  241. c = *src++;
  242. horizontal_fill(s->bpp * (s->avctx->pix_fmt == PIX_FMT_PAL8),
  243. dst, 0, NULL, c, code, pixels);
  244. pixels += code;
  245. }
  246. }
  247. break;
  248. case TIFF_LZW:
  249. pixels = ff_lzw_decode(s->lzw, dst, width);
  250. if(pixels < width){
  251. av_log(s->avctx, AV_LOG_ERROR, "Decoded only %i bytes of %i\n", pixels, width);
  252. return -1;
  253. }
  254. if (s->bpp < 8 && s->avctx->pix_fmt == PIX_FMT_PAL8)
  255. horizontal_fill(s->bpp, dst, 1, dst, 0, width, 0);
  256. break;
  257. }
  258. dst += stride;
  259. }
  260. return 0;
  261. }
  262. static int init_image(TiffContext *s)
  263. {
  264. int i, ret;
  265. uint32_t *pal;
  266. switch (s->bpp * 10 + s->bppcount) {
  267. case 11:
  268. if (!s->palette_is_set) {
  269. s->avctx->pix_fmt = PIX_FMT_MONOBLACK;
  270. break;
  271. }
  272. case 21:
  273. case 41:
  274. case 81:
  275. s->avctx->pix_fmt = PIX_FMT_PAL8;
  276. break;
  277. case 243:
  278. s->avctx->pix_fmt = PIX_FMT_RGB24;
  279. break;
  280. case 161:
  281. s->avctx->pix_fmt = PIX_FMT_GRAY16BE;
  282. break;
  283. case 162:
  284. s->avctx->pix_fmt = PIX_FMT_GRAY8A;
  285. break;
  286. case 324:
  287. s->avctx->pix_fmt = PIX_FMT_RGBA;
  288. break;
  289. case 483:
  290. s->avctx->pix_fmt = s->le ? PIX_FMT_RGB48LE : PIX_FMT_RGB48BE;
  291. break;
  292. case 644:
  293. s->avctx->pix_fmt = s->le ? PIX_FMT_RGBA64LE : PIX_FMT_RGBA64BE;
  294. break;
  295. default:
  296. av_log(s->avctx, AV_LOG_ERROR,
  297. "This format is not supported (bpp=%d, bppcount=%d)\n",
  298. s->bpp, s->bppcount);
  299. return AVERROR_INVALIDDATA;
  300. }
  301. if (s->width != s->avctx->width || s->height != s->avctx->height) {
  302. if ((ret = av_image_check_size(s->width, s->height, 0, s->avctx)) < 0)
  303. return ret;
  304. avcodec_set_dimensions(s->avctx, s->width, s->height);
  305. }
  306. if (s->picture.data[0])
  307. s->avctx->release_buffer(s->avctx, &s->picture);
  308. if ((ret = s->avctx->get_buffer(s->avctx, &s->picture)) < 0) {
  309. av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed\n");
  310. return ret;
  311. }
  312. if (s->avctx->pix_fmt == PIX_FMT_PAL8) {
  313. if (s->palette_is_set) {
  314. memcpy(s->picture.data[1], s->palette, sizeof(s->palette));
  315. } else {
  316. /* make default grayscale pal */
  317. pal = (uint32_t *) s->picture.data[1];
  318. for (i = 0; i < 1<<s->bpp; i++)
  319. pal[i] = 0xFF << 24 | i * 255 / ((1<<s->bpp) - 1) * 0x010101;
  320. }
  321. }
  322. return 0;
  323. }
  324. static int tiff_decode_tag(TiffContext *s, const uint8_t *start, const uint8_t *buf, const uint8_t *end_buf)
  325. {
  326. int tag, type, count, off, value = 0;
  327. int i, j;
  328. uint32_t *pal;
  329. const uint8_t *rp, *gp, *bp;
  330. if (end_buf - buf < 12)
  331. return -1;
  332. tag = tget_short(&buf, s->le);
  333. type = tget_short(&buf, s->le);
  334. count = tget_long(&buf, s->le);
  335. off = tget_long(&buf, s->le);
  336. if(count == 1){
  337. switch(type){
  338. case TIFF_BYTE:
  339. case TIFF_SHORT:
  340. buf -= 4;
  341. value = tget(&buf, type, s->le);
  342. buf = NULL;
  343. break;
  344. case TIFF_LONG:
  345. value = off;
  346. buf = NULL;
  347. break;
  348. case TIFF_STRING:
  349. if(count <= 4){
  350. buf -= 4;
  351. break;
  352. }
  353. default:
  354. value = -1;
  355. buf = start + off;
  356. }
  357. }else if(type_sizes[type] * count <= 4){
  358. buf -= 4;
  359. }else{
  360. buf = start + off;
  361. }
  362. if(buf && (buf < start || buf > end_buf)){
  363. av_log(s->avctx, AV_LOG_ERROR, "Tag referencing position outside the image\n");
  364. return -1;
  365. }
  366. switch(tag){
  367. case TIFF_WIDTH:
  368. s->width = value;
  369. break;
  370. case TIFF_HEIGHT:
  371. s->height = value;
  372. break;
  373. case TIFF_BPP:
  374. s->bppcount = count;
  375. if(count > 4){
  376. av_log(s->avctx, AV_LOG_ERROR, "This format is not supported (bpp=%d, %d components)\n", s->bpp, count);
  377. return -1;
  378. }
  379. if(count == 1) s->bpp = value;
  380. else{
  381. switch(type){
  382. case TIFF_BYTE:
  383. s->bpp = (off & 0xFF) + ((off >> 8) & 0xFF) + ((off >> 16) & 0xFF) + ((off >> 24) & 0xFF);
  384. break;
  385. case TIFF_SHORT:
  386. case TIFF_LONG:
  387. s->bpp = 0;
  388. for(i = 0; i < count && buf < end_buf; i++) s->bpp += tget(&buf, type, s->le);
  389. break;
  390. default:
  391. s->bpp = -1;
  392. }
  393. }
  394. break;
  395. case TIFF_SAMPLES_PER_PIXEL:
  396. if (count != 1) {
  397. av_log(s->avctx, AV_LOG_ERROR,
  398. "Samples per pixel requires a single value, many provided\n");
  399. return AVERROR_INVALIDDATA;
  400. }
  401. if (s->bppcount == 1)
  402. s->bpp *= value;
  403. s->bppcount = value;
  404. break;
  405. case TIFF_COMPR:
  406. s->compr = value;
  407. s->predictor = 0;
  408. switch(s->compr){
  409. case TIFF_RAW:
  410. case TIFF_PACKBITS:
  411. case TIFF_LZW:
  412. case TIFF_CCITT_RLE:
  413. break;
  414. case TIFF_G3:
  415. case TIFF_G4:
  416. s->fax_opts = 0;
  417. break;
  418. case TIFF_DEFLATE:
  419. case TIFF_ADOBE_DEFLATE:
  420. #if CONFIG_ZLIB
  421. break;
  422. #else
  423. av_log(s->avctx, AV_LOG_ERROR, "Deflate: ZLib not compiled in\n");
  424. return -1;
  425. #endif
  426. case TIFF_JPEG:
  427. case TIFF_NEWJPEG:
  428. av_log(s->avctx, AV_LOG_ERROR, "JPEG compression is not supported\n");
  429. return -1;
  430. default:
  431. av_log(s->avctx, AV_LOG_ERROR, "Unknown compression method %i\n", s->compr);
  432. return -1;
  433. }
  434. break;
  435. case TIFF_ROWSPERSTRIP:
  436. if(type == TIFF_LONG && value == -1)
  437. value = s->avctx->height;
  438. if(value < 1){
  439. av_log(s->avctx, AV_LOG_ERROR, "Incorrect value of rows per strip\n");
  440. return -1;
  441. }
  442. s->rps = value;
  443. break;
  444. case TIFF_STRIP_OFFS:
  445. if(count == 1){
  446. s->stripdata = NULL;
  447. s->stripoff = value;
  448. }else
  449. s->stripdata = start + off;
  450. s->strips = count;
  451. if(s->strips == 1) s->rps = s->height;
  452. s->sot = type;
  453. if(s->stripdata > end_buf){
  454. av_log(s->avctx, AV_LOG_ERROR, "Tag referencing position outside the image\n");
  455. return -1;
  456. }
  457. break;
  458. case TIFF_STRIP_SIZE:
  459. if(count == 1){
  460. s->stripsizes = NULL;
  461. s->stripsize = value;
  462. s->strips = 1;
  463. }else{
  464. s->stripsizes = start + off;
  465. }
  466. s->strips = count;
  467. s->sstype = type;
  468. if(s->stripsizes > end_buf){
  469. av_log(s->avctx, AV_LOG_ERROR, "Tag referencing position outside the image\n");
  470. return -1;
  471. }
  472. break;
  473. case TIFF_TILE_BYTE_COUNTS:
  474. case TIFF_TILE_LENGTH:
  475. case TIFF_TILE_OFFSETS:
  476. case TIFF_TILE_WIDTH:
  477. av_log(s->avctx, AV_LOG_ERROR, "Tiled images are not supported\n");
  478. return AVERROR_PATCHWELCOME;
  479. break;
  480. case TIFF_PREDICTOR:
  481. s->predictor = value;
  482. break;
  483. case TIFF_INVERT:
  484. switch(value){
  485. case 0:
  486. s->invert = 1;
  487. break;
  488. case 1:
  489. s->invert = 0;
  490. break;
  491. case 2:
  492. case 3:
  493. break;
  494. default:
  495. av_log(s->avctx, AV_LOG_ERROR, "Color mode %d is not supported\n", value);
  496. return -1;
  497. }
  498. break;
  499. case TIFF_FILL_ORDER:
  500. if(value < 1 || value > 2){
  501. av_log(s->avctx, AV_LOG_ERROR, "Unknown FillOrder value %d, trying default one\n", value);
  502. value = 1;
  503. }
  504. s->fill_order = value - 1;
  505. break;
  506. case TIFF_PAL:
  507. pal = (uint32_t *) s->palette;
  508. off = type_sizes[type];
  509. if (count / 3 > 256 || end_buf - buf < count / 3 * off * 3)
  510. return -1;
  511. rp = buf;
  512. gp = buf + count / 3 * off;
  513. bp = buf + count / 3 * off * 2;
  514. off = (type_sizes[type] - 1) << 3;
  515. for(i = 0; i < count / 3; i++){
  516. j = 0xff << 24;
  517. j |= (tget(&rp, type, s->le) >> off) << 16;
  518. j |= (tget(&gp, type, s->le) >> off) << 8;
  519. j |= tget(&bp, type, s->le) >> off;
  520. pal[i] = j;
  521. }
  522. s->palette_is_set = 1;
  523. break;
  524. case TIFF_PLANAR:
  525. if(value == 2){
  526. av_log(s->avctx, AV_LOG_ERROR, "Planar format is not supported\n");
  527. return -1;
  528. }
  529. break;
  530. case TIFF_T4OPTIONS:
  531. if(s->compr == TIFF_G3)
  532. s->fax_opts = value;
  533. break;
  534. case TIFF_T6OPTIONS:
  535. if(s->compr == TIFF_G4)
  536. s->fax_opts = value;
  537. break;
  538. default:
  539. av_log(s->avctx, AV_LOG_DEBUG, "Unknown or unsupported tag %d/0X%0X\n", tag, tag);
  540. }
  541. return 0;
  542. }
  543. static int decode_frame(AVCodecContext *avctx,
  544. void *data, int *data_size,
  545. AVPacket *avpkt)
  546. {
  547. const uint8_t *buf = avpkt->data;
  548. int buf_size = avpkt->size;
  549. TiffContext * const s = avctx->priv_data;
  550. AVFrame *picture = data;
  551. AVFrame * const p= (AVFrame*)&s->picture;
  552. const uint8_t *orig_buf = buf, *end_buf = buf + buf_size;
  553. unsigned off;
  554. int id, le, ret;
  555. int i, j, entries;
  556. int stride;
  557. unsigned soff, ssize;
  558. uint8_t *dst;
  559. //parse image header
  560. if (end_buf - buf < 8)
  561. return AVERROR_INVALIDDATA;
  562. id = AV_RL16(buf); buf += 2;
  563. if(id == 0x4949) le = 1;
  564. else if(id == 0x4D4D) le = 0;
  565. else{
  566. av_log(avctx, AV_LOG_ERROR, "TIFF header not found\n");
  567. return -1;
  568. }
  569. s->le = le;
  570. s->invert = 0;
  571. s->compr = TIFF_RAW;
  572. s->fill_order = 0;
  573. // As TIFF 6.0 specification puts it "An arbitrary but carefully chosen number
  574. // that further identifies the file as a TIFF file"
  575. if(tget_short(&buf, le) != 42){
  576. av_log(avctx, AV_LOG_ERROR, "The answer to life, universe and everything is not correct!\n");
  577. return -1;
  578. }
  579. /* parse image file directory */
  580. off = tget_long(&buf, le);
  581. if (off >= UINT_MAX - 14 || end_buf - orig_buf < off + 14) {
  582. av_log(avctx, AV_LOG_ERROR, "IFD offset is greater than image size\n");
  583. return AVERROR_INVALIDDATA;
  584. }
  585. buf = orig_buf + off;
  586. entries = tget_short(&buf, le);
  587. for(i = 0; i < entries; i++){
  588. if(tiff_decode_tag(s, orig_buf, buf, end_buf) < 0)
  589. return -1;
  590. buf += 12;
  591. }
  592. if(!s->stripdata && !s->stripoff){
  593. av_log(avctx, AV_LOG_ERROR, "Image data is missing\n");
  594. return -1;
  595. }
  596. /* now we have the data and may start decoding */
  597. if ((ret = init_image(s)) < 0)
  598. return ret;
  599. if(s->strips == 1 && !s->stripsize){
  600. av_log(avctx, AV_LOG_WARNING, "Image data size missing\n");
  601. s->stripsize = buf_size - s->stripoff;
  602. }
  603. stride = p->linesize[0];
  604. dst = p->data[0];
  605. for(i = 0; i < s->height; i += s->rps){
  606. if(s->stripsizes) {
  607. if (s->stripsizes >= end_buf)
  608. return AVERROR_INVALIDDATA;
  609. ssize = tget(&s->stripsizes, s->sstype, s->le);
  610. } else
  611. ssize = s->stripsize;
  612. if(s->stripdata){
  613. if (s->stripdata >= end_buf)
  614. return AVERROR_INVALIDDATA;
  615. soff = tget(&s->stripdata, s->sot, s->le);
  616. }else
  617. soff = s->stripoff;
  618. if (soff > buf_size || ssize > buf_size - soff) {
  619. av_log(avctx, AV_LOG_ERROR, "Invalid strip size/offset\n");
  620. return -1;
  621. }
  622. if(tiff_unpack_strip(s, dst, stride, orig_buf + soff, ssize, FFMIN(s->rps, s->height - i)) < 0)
  623. break;
  624. dst += s->rps * stride;
  625. }
  626. if(s->predictor == 2){
  627. dst = p->data[0];
  628. soff = s->bpp >> 3;
  629. ssize = s->width * soff;
  630. if (s->avctx->pix_fmt == PIX_FMT_RGB48LE ||
  631. s->avctx->pix_fmt == PIX_FMT_RGBA64LE) {
  632. for (i = 0; i < s->height; i++) {
  633. for (j = soff; j < ssize; j += 2)
  634. AV_WL16(dst + j, AV_RL16(dst + j) + AV_RL16(dst + j - soff));
  635. dst += stride;
  636. }
  637. } else if (s->avctx->pix_fmt == PIX_FMT_RGB48BE ||
  638. s->avctx->pix_fmt == PIX_FMT_RGBA64BE) {
  639. for (i = 0; i < s->height; i++) {
  640. for (j = soff; j < ssize; j += 2)
  641. AV_WB16(dst + j, AV_RB16(dst + j) + AV_RB16(dst + j - soff));
  642. dst += stride;
  643. }
  644. } else {
  645. for(i = 0; i < s->height; i++) {
  646. for(j = soff; j < ssize; j++)
  647. dst[j] += dst[j - soff];
  648. dst += stride;
  649. }
  650. }
  651. }
  652. if(s->invert){
  653. dst = s->picture.data[0];
  654. for(i = 0; i < s->height; i++){
  655. for(j = 0; j < s->picture.linesize[0]; j++)
  656. dst[j] = (s->avctx->pix_fmt == PIX_FMT_PAL8 ? (1<<s->bpp) - 1 : 255) - dst[j];
  657. dst += s->picture.linesize[0];
  658. }
  659. }
  660. *picture= *(AVFrame*)&s->picture;
  661. *data_size = sizeof(AVPicture);
  662. return buf_size;
  663. }
  664. static av_cold int tiff_init(AVCodecContext *avctx){
  665. TiffContext *s = avctx->priv_data;
  666. s->width = 0;
  667. s->height = 0;
  668. s->avctx = avctx;
  669. avcodec_get_frame_defaults((AVFrame*)&s->picture);
  670. avctx->coded_frame= (AVFrame*)&s->picture;
  671. ff_lzw_decode_open(&s->lzw);
  672. ff_ccitt_unpack_init();
  673. return 0;
  674. }
  675. static av_cold int tiff_end(AVCodecContext *avctx)
  676. {
  677. TiffContext * const s = avctx->priv_data;
  678. ff_lzw_decode_close(&s->lzw);
  679. if(s->picture.data[0])
  680. avctx->release_buffer(avctx, &s->picture);
  681. return 0;
  682. }
  683. AVCodec ff_tiff_decoder = {
  684. .name = "tiff",
  685. .type = AVMEDIA_TYPE_VIDEO,
  686. .id = CODEC_ID_TIFF,
  687. .priv_data_size = sizeof(TiffContext),
  688. .init = tiff_init,
  689. .close = tiff_end,
  690. .decode = decode_frame,
  691. .capabilities = CODEC_CAP_DR1,
  692. .long_name = NULL_IF_CONFIG_SMALL("TIFF image"),
  693. };