blob: ccac4f3ecf92a873222e2b2173ebe45054d28b13 [file] [log] [blame]
halcanarydaefa5b2014-08-27 13:00:54 -07001/*
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
halcanarydaefa5b2014-08-27 13:00:54 -07008#include "SkCanvas.h"
halcanarydaefa5b2014-08-27 13:00:54 -07009#include "SkData.h"
halcanary57f744e2016-09-09 11:41:59 -070010#include "SkImageGenerator.h"
Hal Canary23564b92018-09-07 14:33:14 -040011#include "SkPDFDocument.h"
halcanarydaefa5b2014-08-27 13:00:54 -070012#include "SkStream.h"
halcanarydaefa5b2014-08-27 13:00:54 -070013
14#include "Resources.h"
15#include "Test.h"
16
halcanarydaefa5b2014-08-27 13:00:54 -070017static bool is_subset_of(SkData* smaller, SkData* larger) {
18 SkASSERT(smaller && larger);
19 if (smaller->size() > larger->size()) {
20 return false;
21 }
22 size_t size = smaller->size();
23 size_t size_diff = larger->size() - size;
24 for (size_t i = 0; i <= size_diff; ++i) {
25 if (0 == memcmp(larger->bytes() + i, smaller->bytes(), size)) {
26 return true;
27 }
28 }
29 return false;
30}
31
32
bungeman38d909e2016-08-02 14:40:46 -070033static sk_sp<SkData> load_resource(
halcanarydaefa5b2014-08-27 13:00:54 -070034 skiatest::Reporter* r, const char* test, const char* filename) {
Mike Reed0933bc92017-12-09 01:27:41 +000035 sk_sp<SkData> data = GetResourceAsData(filename);
halcanary7d571242016-02-24 17:59:16 -080036 if (!data) {
37 INFOF(r, "\n%s: Resource '%s' can not be found.\n",
38 test, filename);
halcanarydaefa5b2014-08-27 13:00:54 -070039 }
halcanary96fcdcc2015-08-27 07:41:13 -070040 return data; // May return nullptr.
halcanarydaefa5b2014-08-27 13:00:54 -070041}
42
43/**
44 * Test that for Jpeg files that use the JFIF colorspace, they are
45 * directly embedded into the PDF (without re-encoding) when that
46 * makes sense.
47 */
halcanary57f744e2016-09-09 11:41:59 -070048DEF_TEST(SkPDF_JpegEmbedTest, r) {
49 REQUIRE_PDF_DOCUMENT(SkPDF_JpegEmbedTest, r);
50 const char test[] = "SkPDF_JpegEmbedTest";
Hal Canaryc465d132017-12-08 10:21:31 -050051 sk_sp<SkData> mandrillData(load_resource(r, test, "images/mandrill_512_q075.jpg"));
52 sk_sp<SkData> cmykData(load_resource(r, test, "images/CMYK.jpg"));
halcanarydaefa5b2014-08-27 13:00:54 -070053 if (!mandrillData || !cmykData) {
54 return;
55 }
halcanary7a14b312015-10-01 07:28:13 -070056 ////////////////////////////////////////////////////////////////////////////
halcanarydaefa5b2014-08-27 13:00:54 -070057 SkDynamicMemoryWStream pdf;
Hal Canary23564b92018-09-07 14:33:14 -040058 sk_sp<SkDocument> document(SkPDF::MakeDocument(&pdf));
halcanarydaefa5b2014-08-27 13:00:54 -070059 SkCanvas* canvas = document->beginPage(642, 1028);
60
61 canvas->clear(SK_ColorLTGRAY);
62
reed9ce9d672016-03-17 10:51:11 -070063 sk_sp<SkImage> im1(SkImage::MakeFromEncoded(mandrillData));
64 canvas->drawImage(im1.get(), 65.0, 0.0, nullptr);
65 sk_sp<SkImage> im2(SkImage::MakeFromEncoded(cmykData));
66 canvas->drawImage(im2.get(), 0.0, 512.0, nullptr);
halcanary7a14b312015-10-01 07:28:13 -070067
68 canvas->flush();
69 document->endPage();
70 document->close();
Mike Reed76147942016-10-25 09:57:13 -040071 sk_sp<SkData> pdfData = pdf.detachAsData();
halcanary7a14b312015-10-01 07:28:13 -070072 SkASSERT(pdfData);
halcanary7a14b312015-10-01 07:28:13 -070073
reed9ce9d672016-03-17 10:51:11 -070074 REPORTER_ASSERT(r, is_subset_of(mandrillData.get(), pdfData.get()));
halcanary7a14b312015-10-01 07:28:13 -070075
76 // This JPEG uses a nonstandard colorspace - it can not be
77 // embedded into the PDF directly.
reed9ce9d672016-03-17 10:51:11 -070078 REPORTER_ASSERT(r, !is_subset_of(cmykData.get(), pdfData.get()));
halcanarydaefa5b2014-08-27 13:00:54 -070079}
halcanary96287f72015-05-07 11:46:59 -070080
Hal Canary43fb7a02016-12-30 13:09:03 -050081#ifdef SK_SUPPORT_PDF
82
83#include "SkJpegInfo.h"
84
Hal Canary83e0f1b2018-04-05 16:58:41 -040085struct SkJFIFInfo {
86 SkISize fSize;
87 enum Type {
88 kGrayscale,
89 kYCbCr,
90 } fType;
91};
92bool SkIsJFIF(const SkData* data, SkJFIFInfo* info) {
93 SkISize jpegSize;
94 SkEncodedInfo::Color jpegColorType;
95 SkEncodedOrigin exifOrientation;
96 if (data && SkGetJpegInfo(data->data(), data->size(), &jpegSize,
97 &jpegColorType, &exifOrientation)) {
98 bool yuv = jpegColorType == SkEncodedInfo::kYUV_Color;
99 bool goodColorType = yuv || jpegColorType == SkEncodedInfo::kGray_Color;
100 if (goodColorType && kTopLeft_SkEncodedOrigin == exifOrientation) {
101 if (info) {
102 *info = {jpegSize, yuv ? SkJFIFInfo::kYCbCr : SkJFIFInfo::kGrayscale};
103 }
104 return true;
105 }
106 }
107 return false;
108}
109
halcanary57f744e2016-09-09 11:41:59 -0700110DEF_TEST(SkPDF_JpegIdentification, r) {
halcanary96287f72015-05-07 11:46:59 -0700111 static struct {
112 const char* path;
113 bool isJfif;
114 SkJFIFInfo::Type type;
Hal Canaryc465d132017-12-08 10:21:31 -0500115 } kTests[] = {{"images/CMYK.jpg", false, SkJFIFInfo::kGrayscale},
116 {"images/color_wheel.jpg", true, SkJFIFInfo::kYCbCr},
117 {"images/grayscale.jpg", true, SkJFIFInfo::kGrayscale},
118 {"images/mandrill_512_q075.jpg", true, SkJFIFInfo::kYCbCr},
119 {"images/randPixels.jpg", true, SkJFIFInfo::kYCbCr}};
halcanary96287f72015-05-07 11:46:59 -0700120 for (size_t i = 0; i < SK_ARRAY_COUNT(kTests); ++i) {
bungeman38d909e2016-08-02 14:40:46 -0700121 sk_sp<SkData> data(load_resource(r, "JpegIdentification", kTests[i].path));
halcanary96287f72015-05-07 11:46:59 -0700122 if (!data) {
123 continue;
124 }
125 SkJFIFInfo info;
bungeman38d909e2016-08-02 14:40:46 -0700126 bool isJfif = SkIsJFIF(data.get(), &info);
halcanary96287f72015-05-07 11:46:59 -0700127 if (isJfif != kTests[i].isJfif) {
128 ERRORF(r, "%s failed isJfif test", kTests[i].path);
129 continue;
130 }
131 if (!isJfif) {
132 continue; // not applicable
133 }
134 if (kTests[i].type != info.fType) {
135 ERRORF(r, "%s failed jfif type test", kTests[i].path);
136 continue;
137 }
halcanary7d571242016-02-24 17:59:16 -0800138 INFOF(r, "\nJpegIdentification: %s [%d x %d]\n", kTests[i].path,
139 info.fSize.width(), info.fSize.height());
halcanary96287f72015-05-07 11:46:59 -0700140 }
halcanary57f744e2016-09-09 11:41:59 -0700141
142 // Test several malformed jpegs.
143 SkJFIFInfo info;
144 {
145 static const char goodJpeg[] =
Hal Canary83e0f1b2018-04-05 16:58:41 -0400146 "\377\330\377\340\0\20JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\20\13\14"
147 "\16\14\n\20\16\r\16\22\21\20\23\30(\32\30\26\26\0301#%\35(:3=<9387"
148 "@H\\N@DWE78PmQW_bghg>Mqypdx\\egc\377\333\0C\1\21\22\22\30\25\30/\32"
149 "\32/cB8Bcccccccccccccccccccccccccccccccccccccccccccccccccc\377\300"
150 "\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"
151 "\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"
152 "\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"
153 "\221\241\10#B\261\301\25R\321\360$3br\202\t\n\26\27\30\31\32%&'()*"
154 "456789:CDEFGHIJSTUVWXYZcdefghijstuvwxyz\203\204\205\206\207\210\211"
155 "\212\222\223\224\225\226\227\230\231\232\242\243\244\245\246\247\250"
156 "\251\252\262\263\264\265\266\267\270\271\272\302\303\304\305\306\307"
157 "\310\311\312\322\323\324\325\326\327\330\331\332\341\342\343\344\345"
158 "\346\347\350\351\352\361\362\363\364\365\366\367\370\371\372\377\304"
159 "\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"
160 "\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"
161 "AQ\7aq\23\"2\201\10\24B\221\241\261\301\t#3R\360\25br\321\n\26$4\341"
162 "%\361\27\30\31\32&'()*56789:CDEFGHIJSTUVWXYZcdefghijstuvwxyz\202\203"
163 "\204\205\206\207\210\211\212\222\223\224\225\226\227\230\231\232\242"
164 "\243\244\245\246\247\250\251\252\262\263\264\265\266\267\270\271\272"
165 "\302\303\304\305\306\307\310\311\312\322\323\324\325\326\327\330\331"
166 "\332\342\343\344\345\346\347\350\351\352\362\363\364\365\366\367\370"
167 "\371\372\377\332\0\14\3\1\0\2\21\3\21\0?\0\216M\352\214\25\271\224"
168 "\262\310\253\363tl\22209\35O~\237\\\24QZ\306Mh\216\252i\364ml\177\377"
169 "\331";
170 size_t goodJpegLength = 659;
halcanary57f744e2016-09-09 11:41:59 -0700171 auto data = SkData::MakeWithoutCopy(goodJpeg, goodJpegLength);
172 REPORTER_ASSERT(r, SkIsJFIF(data.get(), &info));
Hal Canary83e0f1b2018-04-05 16:58:41 -0400173 REPORTER_ASSERT(r, info.fSize == (SkISize{8, 8}));
halcanary57f744e2016-09-09 11:41:59 -0700174 REPORTER_ASSERT(r, info.fType == SkJFIFInfo::kYCbCr);
175
176 // Not long enough to read first (SOI) segment marker.
177 data = SkData::MakeWithoutCopy(goodJpeg, 1);
178 REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
179
180 // Not long enough to read second segment (APP0) marker.
181 data = SkData::MakeWithoutCopy(goodJpeg, 3);
182 REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
183
184 // Not long enough to read second segment's length.
185 data = SkData::MakeWithoutCopy(goodJpeg, 5);
186 REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
187
188 // APP0 segment is truncated.
189 data = SkData::MakeWithoutCopy(goodJpeg, 7);
190 REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
191
192 // Missing SOF segment.
193 data = SkData::MakeWithoutCopy(goodJpeg, 89);
194 REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
195 }
196 {
197 // JFIF tag missing.
198 static const char jpeg[] =
199 "\377\330\377\340\0\20JFIX\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
200 "\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"
201 "\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
202 "\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
203 "22222222222222\377\300\0\21\10\2\0\2\0\3\1\"\0\2\21\1\3\21\001";
204 size_t jpegLength = 177;
205 auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
206 REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
207 }
208 {
209 // APP0 segment short (byte 6 changed).
210 static const char jpeg[] =
211 "\377\330\377\340\0\5JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
212 "\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"
213 "\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
214 "\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
215 "22222222222222\377\300\0\21\10\2\0\2\0\3\1\"\0\2\21\1\3\21\001";
216 size_t jpegLength = 177;
217 auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
218 REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
219 }
220 {
221 // SOF segment short. ('\21' replaced with '\5')
222 static const char jpeg[] =
223 "\377\330\377\340\0\20JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
224 "\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"
225 "\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
226 "\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
227 "22222222222222\377\300\0\5\10\2\0\2\0\3\1\"\0\2\21\1\3\21\001";
228 size_t jpegLength = 177;
229 auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
230 REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
231 }
232 {
233 // Unsupported 12-bit components. ('\10' replaced with '\14')
234 static const char jpeg[] =
235 "\377\330\377\340\0\20JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
236 "\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"
237 "\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
238 "\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
239 "22222222222222\377\300\0\21\14\2\0\2\0\3\1\"\0\2\21\1\3\21\001";
240 size_t jpegLength = 177;
241 auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
242 REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
243 }
244 {
245 // Two color channels. ('\3' replaced with '\2')
246 static const char jpeg[] =
247 "\377\330\377\340\0\20JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
248 "\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"
249 "\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
250 "\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
251 "22222222222222\377\300\0\21\10\2\0\2\0\2\1\"\0\2\21\1\3\21\001";
252 size_t jpegLength = 177;
253 auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
254 REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
255 }
halcanary96287f72015-05-07 11:46:59 -0700256}
Hal Canary43fb7a02016-12-30 13:09:03 -0500257#endif