PDFJpegEmbedTest.cpp 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258
  1. /*
  2. * Copyright 2014 Google Inc.
  3. *
  4. * Use of this source code is governed by a BSD-style license that can be
  5. * found in the LICENSE file.
  6. */
  7. #include "include/core/SkCanvas.h"
  8. #include "include/core/SkData.h"
  9. #include "include/core/SkImageGenerator.h"
  10. #include "include/core/SkStream.h"
  11. #include "include/docs/SkPDFDocument.h"
  12. #include "tests/Test.h"
  13. #include "tools/Resources.h"
  14. static bool is_subset_of(SkData* smaller, SkData* larger) {
  15. SkASSERT(smaller && larger);
  16. if (smaller->size() > larger->size()) {
  17. return false;
  18. }
  19. size_t size = smaller->size();
  20. size_t size_diff = larger->size() - size;
  21. for (size_t i = 0; i <= size_diff; ++i) {
  22. if (0 == memcmp(larger->bytes() + i, smaller->bytes(), size)) {
  23. return true;
  24. }
  25. }
  26. return false;
  27. }
  28. static sk_sp<SkData> load_resource(
  29. skiatest::Reporter* r, const char* test, const char* filename) {
  30. sk_sp<SkData> data = GetResourceAsData(filename);
  31. if (!data) {
  32. INFOF(r, "\n%s: Resource '%s' can not be found.\n",
  33. test, filename);
  34. }
  35. return data; // May return nullptr.
  36. }
  37. /**
  38. * Test that for Jpeg files that use the JFIF colorspace, they are
  39. * directly embedded into the PDF (without re-encoding) when that
  40. * makes sense.
  41. */
  42. DEF_TEST(SkPDF_JpegEmbedTest, r) {
  43. REQUIRE_PDF_DOCUMENT(SkPDF_JpegEmbedTest, r);
  44. const char test[] = "SkPDF_JpegEmbedTest";
  45. sk_sp<SkData> mandrillData(load_resource(r, test, "images/mandrill_512_q075.jpg"));
  46. sk_sp<SkData> cmykData(load_resource(r, test, "images/CMYK.jpg"));
  47. if (!mandrillData || !cmykData) {
  48. return;
  49. }
  50. ////////////////////////////////////////////////////////////////////////////
  51. SkDynamicMemoryWStream pdf;
  52. auto document = SkPDF::MakeDocument(&pdf);
  53. SkCanvas* canvas = document->beginPage(642, 1028);
  54. canvas->clear(SK_ColorLTGRAY);
  55. sk_sp<SkImage> im1(SkImage::MakeFromEncoded(mandrillData));
  56. canvas->drawImage(im1.get(), 65.0, 0.0, nullptr);
  57. sk_sp<SkImage> im2(SkImage::MakeFromEncoded(cmykData));
  58. canvas->drawImage(im2.get(), 0.0, 512.0, nullptr);
  59. document->endPage();
  60. document->close();
  61. sk_sp<SkData> pdfData = pdf.detachAsData();
  62. SkASSERT(pdfData);
  63. #ifndef SK_PDF_BASE85_BINARY
  64. REPORTER_ASSERT(r, is_subset_of(mandrillData.get(), pdfData.get()));
  65. #endif
  66. // This JPEG uses a nonstandard colorspace - it can not be
  67. // embedded into the PDF directly.
  68. REPORTER_ASSERT(r, !is_subset_of(cmykData.get(), pdfData.get()));
  69. }
  70. #ifdef SK_SUPPORT_PDF
  71. #include "src/pdf/SkJpegInfo.h"
  72. struct SkJFIFInfo {
  73. SkISize fSize;
  74. enum Type {
  75. kGrayscale,
  76. kYCbCr,
  77. } fType;
  78. };
  79. bool SkIsJFIF(const SkData* data, SkJFIFInfo* info) {
  80. SkISize jpegSize;
  81. SkEncodedInfo::Color jpegColorType;
  82. SkEncodedOrigin exifOrientation;
  83. if (data && SkGetJpegInfo(data->data(), data->size(), &jpegSize,
  84. &jpegColorType, &exifOrientation)) {
  85. bool yuv = jpegColorType == SkEncodedInfo::kYUV_Color;
  86. bool goodColorType = yuv || jpegColorType == SkEncodedInfo::kGray_Color;
  87. if (goodColorType && kTopLeft_SkEncodedOrigin == exifOrientation) {
  88. if (info) {
  89. *info = {jpegSize, yuv ? SkJFIFInfo::kYCbCr : SkJFIFInfo::kGrayscale};
  90. }
  91. return true;
  92. }
  93. }
  94. return false;
  95. }
  96. DEF_TEST(SkPDF_JpegIdentification, r) {
  97. static struct {
  98. const char* path;
  99. bool isJfif;
  100. SkJFIFInfo::Type type;
  101. } kTests[] = {{"images/CMYK.jpg", false, SkJFIFInfo::kGrayscale},
  102. {"images/color_wheel.jpg", true, SkJFIFInfo::kYCbCr},
  103. {"images/grayscale.jpg", true, SkJFIFInfo::kGrayscale},
  104. {"images/mandrill_512_q075.jpg", true, SkJFIFInfo::kYCbCr},
  105. {"images/randPixels.jpg", true, SkJFIFInfo::kYCbCr}};
  106. for (size_t i = 0; i < SK_ARRAY_COUNT(kTests); ++i) {
  107. sk_sp<SkData> data(load_resource(r, "JpegIdentification", kTests[i].path));
  108. if (!data) {
  109. continue;
  110. }
  111. SkJFIFInfo info;
  112. bool isJfif = SkIsJFIF(data.get(), &info);
  113. if (isJfif != kTests[i].isJfif) {
  114. ERRORF(r, "%s failed isJfif test", kTests[i].path);
  115. continue;
  116. }
  117. if (!isJfif) {
  118. continue; // not applicable
  119. }
  120. if (kTests[i].type != info.fType) {
  121. ERRORF(r, "%s failed jfif type test", kTests[i].path);
  122. continue;
  123. }
  124. INFOF(r, "\nJpegIdentification: %s [%d x %d]\n", kTests[i].path,
  125. info.fSize.width(), info.fSize.height());
  126. }
  127. // Test several malformed jpegs.
  128. SkJFIFInfo info;
  129. {
  130. static const char goodJpeg[] =
  131. "\377\330\377\340\0\20JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\20\13\14"
  132. "\16\14\n\20\16\r\16\22\21\20\23\30(\32\30\26\26\0301#%\35(:3=<9387"
  133. "@H\\N@DWE78PmQW_bghg>Mqypdx\\egc\377\333\0C\1\21\22\22\30\25\30/\32"
  134. "\32/cB8Bcccccccccccccccccccccccccccccccccccccccccccccccccc\377\300"
  135. "\0\21\10\0\10\0\10\3\1\"\0\2\21\1\3\21\1\377\304\0\37\0\0\1\5\1\1\1"
  136. "\1\1\1\0\0\0\0\0\0\0\0\1\2\3\4\5\6\7\10\t\n\13\377\304\0\265\20\0\2"
  137. "\1\3\3\2\4\3\5\5\4\4\0\0\1}\1\2\3\0\4\21\5\22!1A\6\23Qa\7\"q\0242\201"
  138. "\221\241\10#B\261\301\25R\321\360$3br\202\t\n\26\27\30\31\32%&'()*"
  139. "456789:CDEFGHIJSTUVWXYZcdefghijstuvwxyz\203\204\205\206\207\210\211"
  140. "\212\222\223\224\225\226\227\230\231\232\242\243\244\245\246\247\250"
  141. "\251\252\262\263\264\265\266\267\270\271\272\302\303\304\305\306\307"
  142. "\310\311\312\322\323\324\325\326\327\330\331\332\341\342\343\344\345"
  143. "\346\347\350\351\352\361\362\363\364\365\366\367\370\371\372\377\304"
  144. "\0\37\1\0\3\1\1\1\1\1\1\1\1\1\0\0\0\0\0\0\1\2\3\4\5\6\7\10\t\n\13\377"
  145. "\304\0\265\21\0\2\1\2\4\4\3\4\7\5\4\4\0\1\2w\0\1\2\3\21\4\5!1\6\22"
  146. "AQ\7aq\23\"2\201\10\24B\221\241\261\301\t#3R\360\25br\321\n\26$4\341"
  147. "%\361\27\30\31\32&'()*56789:CDEFGHIJSTUVWXYZcdefghijstuvwxyz\202\203"
  148. "\204\205\206\207\210\211\212\222\223\224\225\226\227\230\231\232\242"
  149. "\243\244\245\246\247\250\251\252\262\263\264\265\266\267\270\271\272"
  150. "\302\303\304\305\306\307\310\311\312\322\323\324\325\326\327\330\331"
  151. "\332\342\343\344\345\346\347\350\351\352\362\363\364\365\366\367\370"
  152. "\371\372\377\332\0\14\3\1\0\2\21\3\21\0?\0\216M\352\214\25\271\224"
  153. "\262\310\253\363tl\22209\35O~\237\\\24QZ\306Mh\216\252i\364ml\177\377"
  154. "\331";
  155. size_t goodJpegLength = 659;
  156. auto data = SkData::MakeWithoutCopy(goodJpeg, goodJpegLength);
  157. REPORTER_ASSERT(r, SkIsJFIF(data.get(), &info));
  158. REPORTER_ASSERT(r, info.fSize == (SkISize{8, 8}));
  159. REPORTER_ASSERT(r, info.fType == SkJFIFInfo::kYCbCr);
  160. // Not long enough to read first (SOI) segment marker.
  161. data = SkData::MakeWithoutCopy(goodJpeg, 1);
  162. REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
  163. // Not long enough to read second segment (APP0) marker.
  164. data = SkData::MakeWithoutCopy(goodJpeg, 3);
  165. REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
  166. // Not long enough to read second segment's length.
  167. data = SkData::MakeWithoutCopy(goodJpeg, 5);
  168. REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
  169. // APP0 segment is truncated.
  170. data = SkData::MakeWithoutCopy(goodJpeg, 7);
  171. REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
  172. // Missing SOF segment.
  173. data = SkData::MakeWithoutCopy(goodJpeg, 89);
  174. REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
  175. }
  176. {
  177. // JFIF tag missing.
  178. static const char jpeg[] =
  179. "\377\330\377\340\0\20JFIX\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
  180. "\6\5\10\7\7\7\t\t\10\n\14\24\r\14\13\13\14\31\22\23\17\24\35\32\37"
  181. "\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
  182. "\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
  183. "22222222222222\377\300\0\21\10\2\0\2\0\3\1\"\0\2\21\1\3\21\001";
  184. size_t jpegLength = 177;
  185. auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
  186. REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
  187. }
  188. {
  189. // APP0 segment short (byte 6 changed).
  190. static const char jpeg[] =
  191. "\377\330\377\340\0\5JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
  192. "\6\5\10\7\7\7\t\t\10\n\14\24\r\14\13\13\14\31\22\23\17\24\35\32\37"
  193. "\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
  194. "\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
  195. "22222222222222\377\300\0\21\10\2\0\2\0\3\1\"\0\2\21\1\3\21\001";
  196. size_t jpegLength = 177;
  197. auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
  198. REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
  199. }
  200. {
  201. // SOF segment short. ('\21' replaced with '\5')
  202. static const char jpeg[] =
  203. "\377\330\377\340\0\20JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
  204. "\6\5\10\7\7\7\t\t\10\n\14\24\r\14\13\13\14\31\22\23\17\24\35\32\37"
  205. "\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
  206. "\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
  207. "22222222222222\377\300\0\5\10\2\0\2\0\3\1\"\0\2\21\1\3\21\001";
  208. size_t jpegLength = 177;
  209. auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
  210. REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
  211. }
  212. {
  213. // Unsupported 12-bit components. ('\10' replaced with '\14')
  214. static const char jpeg[] =
  215. "\377\330\377\340\0\20JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
  216. "\6\5\10\7\7\7\t\t\10\n\14\24\r\14\13\13\14\31\22\23\17\24\35\32\37"
  217. "\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
  218. "\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
  219. "22222222222222\377\300\0\21\14\2\0\2\0\3\1\"\0\2\21\1\3\21\001";
  220. size_t jpegLength = 177;
  221. auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
  222. REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
  223. }
  224. {
  225. // Two color channels. ('\3' replaced with '\2')
  226. static const char jpeg[] =
  227. "\377\330\377\340\0\20JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
  228. "\6\5\10\7\7\7\t\t\10\n\14\24\r\14\13\13\14\31\22\23\17\24\35\32\37"
  229. "\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
  230. "\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
  231. "22222222222222\377\300\0\21\10\2\0\2\0\2\1\"\0\2\21\1\3\21\001";
  232. size_t jpegLength = 177;
  233. auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
  234. REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
  235. }
  236. }
  237. #endif