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