3026d4b1b8
A later CL will make this return a unique_ptr<SkDocument>. Bug: skia:5972 Change-Id: Ie10d6c07d5f2524ecb71d906db0d37427827225d Reviewed-on: https://skia-review.googlesource.com/c/181660 Reviewed-by: Hal Canary <halcanary@google.com> Commit-Queue: Hal Canary <halcanary@google.com>
260 lines
11 KiB
C++
260 lines
11 KiB
C++
/*
|
|
* Copyright 2014 Google Inc.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license that can be
|
|
* found in the LICENSE file.
|
|
*/
|
|
|
|
#include "SkCanvas.h"
|
|
#include "SkData.h"
|
|
#include "SkImageGenerator.h"
|
|
#include "SkPDFDocument.h"
|
|
#include "SkStream.h"
|
|
|
|
#include "Resources.h"
|
|
#include "Test.h"
|
|
|
|
static bool is_subset_of(SkData* smaller, SkData* larger) {
|
|
SkASSERT(smaller && larger);
|
|
if (smaller->size() > larger->size()) {
|
|
return false;
|
|
}
|
|
size_t size = smaller->size();
|
|
size_t size_diff = larger->size() - size;
|
|
for (size_t i = 0; i <= size_diff; ++i) {
|
|
if (0 == memcmp(larger->bytes() + i, smaller->bytes(), size)) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
|
|
static sk_sp<SkData> load_resource(
|
|
skiatest::Reporter* r, const char* test, const char* filename) {
|
|
sk_sp<SkData> data = GetResourceAsData(filename);
|
|
if (!data) {
|
|
INFOF(r, "\n%s: Resource '%s' can not be found.\n",
|
|
test, filename);
|
|
}
|
|
return data; // May return nullptr.
|
|
}
|
|
|
|
/**
|
|
* Test that for Jpeg files that use the JFIF colorspace, they are
|
|
* directly embedded into the PDF (without re-encoding) when that
|
|
* makes sense.
|
|
*/
|
|
DEF_TEST(SkPDF_JpegEmbedTest, r) {
|
|
REQUIRE_PDF_DOCUMENT(SkPDF_JpegEmbedTest, r);
|
|
const char test[] = "SkPDF_JpegEmbedTest";
|
|
sk_sp<SkData> mandrillData(load_resource(r, test, "images/mandrill_512_q075.jpg"));
|
|
sk_sp<SkData> cmykData(load_resource(r, test, "images/CMYK.jpg"));
|
|
if (!mandrillData || !cmykData) {
|
|
return;
|
|
}
|
|
////////////////////////////////////////////////////////////////////////////
|
|
SkDynamicMemoryWStream pdf;
|
|
auto document = SkPDF::MakeDocument(&pdf);
|
|
SkCanvas* canvas = document->beginPage(642, 1028);
|
|
|
|
canvas->clear(SK_ColorLTGRAY);
|
|
|
|
sk_sp<SkImage> im1(SkImage::MakeFromEncoded(mandrillData));
|
|
canvas->drawImage(im1.get(), 65.0, 0.0, nullptr);
|
|
sk_sp<SkImage> im2(SkImage::MakeFromEncoded(cmykData));
|
|
canvas->drawImage(im2.get(), 0.0, 512.0, nullptr);
|
|
|
|
canvas->flush();
|
|
document->endPage();
|
|
document->close();
|
|
sk_sp<SkData> pdfData = pdf.detachAsData();
|
|
SkASSERT(pdfData);
|
|
|
|
#ifndef SK_PDF_BASE85_BINARY
|
|
REPORTER_ASSERT(r, is_subset_of(mandrillData.get(), pdfData.get()));
|
|
#endif
|
|
|
|
// This JPEG uses a nonstandard colorspace - it can not be
|
|
// embedded into the PDF directly.
|
|
REPORTER_ASSERT(r, !is_subset_of(cmykData.get(), pdfData.get()));
|
|
}
|
|
|
|
#ifdef SK_SUPPORT_PDF
|
|
|
|
#include "SkJpegInfo.h"
|
|
|
|
struct SkJFIFInfo {
|
|
SkISize fSize;
|
|
enum Type {
|
|
kGrayscale,
|
|
kYCbCr,
|
|
} fType;
|
|
};
|
|
bool SkIsJFIF(const SkData* data, SkJFIFInfo* info) {
|
|
SkISize jpegSize;
|
|
SkEncodedInfo::Color jpegColorType;
|
|
SkEncodedOrigin exifOrientation;
|
|
if (data && SkGetJpegInfo(data->data(), data->size(), &jpegSize,
|
|
&jpegColorType, &exifOrientation)) {
|
|
bool yuv = jpegColorType == SkEncodedInfo::kYUV_Color;
|
|
bool goodColorType = yuv || jpegColorType == SkEncodedInfo::kGray_Color;
|
|
if (goodColorType && kTopLeft_SkEncodedOrigin == exifOrientation) {
|
|
if (info) {
|
|
*info = {jpegSize, yuv ? SkJFIFInfo::kYCbCr : SkJFIFInfo::kGrayscale};
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
DEF_TEST(SkPDF_JpegIdentification, r) {
|
|
static struct {
|
|
const char* path;
|
|
bool isJfif;
|
|
SkJFIFInfo::Type type;
|
|
} kTests[] = {{"images/CMYK.jpg", false, SkJFIFInfo::kGrayscale},
|
|
{"images/color_wheel.jpg", true, SkJFIFInfo::kYCbCr},
|
|
{"images/grayscale.jpg", true, SkJFIFInfo::kGrayscale},
|
|
{"images/mandrill_512_q075.jpg", true, SkJFIFInfo::kYCbCr},
|
|
{"images/randPixels.jpg", true, SkJFIFInfo::kYCbCr}};
|
|
for (size_t i = 0; i < SK_ARRAY_COUNT(kTests); ++i) {
|
|
sk_sp<SkData> data(load_resource(r, "JpegIdentification", kTests[i].path));
|
|
if (!data) {
|
|
continue;
|
|
}
|
|
SkJFIFInfo info;
|
|
bool isJfif = SkIsJFIF(data.get(), &info);
|
|
if (isJfif != kTests[i].isJfif) {
|
|
ERRORF(r, "%s failed isJfif test", kTests[i].path);
|
|
continue;
|
|
}
|
|
if (!isJfif) {
|
|
continue; // not applicable
|
|
}
|
|
if (kTests[i].type != info.fType) {
|
|
ERRORF(r, "%s failed jfif type test", kTests[i].path);
|
|
continue;
|
|
}
|
|
INFOF(r, "\nJpegIdentification: %s [%d x %d]\n", kTests[i].path,
|
|
info.fSize.width(), info.fSize.height());
|
|
}
|
|
|
|
// Test several malformed jpegs.
|
|
SkJFIFInfo info;
|
|
{
|
|
static const char goodJpeg[] =
|
|
"\377\330\377\340\0\20JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\20\13\14"
|
|
"\16\14\n\20\16\r\16\22\21\20\23\30(\32\30\26\26\0301#%\35(:3=<9387"
|
|
"@H\\N@DWE78PmQW_bghg>Mqypdx\\egc\377\333\0C\1\21\22\22\30\25\30/\32"
|
|
"\32/cB8Bcccccccccccccccccccccccccccccccccccccccccccccccccc\377\300"
|
|
"\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"
|
|
"\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"
|
|
"\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"
|
|
"\221\241\10#B\261\301\25R\321\360$3br\202\t\n\26\27\30\31\32%&'()*"
|
|
"456789:CDEFGHIJSTUVWXYZcdefghijstuvwxyz\203\204\205\206\207\210\211"
|
|
"\212\222\223\224\225\226\227\230\231\232\242\243\244\245\246\247\250"
|
|
"\251\252\262\263\264\265\266\267\270\271\272\302\303\304\305\306\307"
|
|
"\310\311\312\322\323\324\325\326\327\330\331\332\341\342\343\344\345"
|
|
"\346\347\350\351\352\361\362\363\364\365\366\367\370\371\372\377\304"
|
|
"\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"
|
|
"\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"
|
|
"AQ\7aq\23\"2\201\10\24B\221\241\261\301\t#3R\360\25br\321\n\26$4\341"
|
|
"%\361\27\30\31\32&'()*56789:CDEFGHIJSTUVWXYZcdefghijstuvwxyz\202\203"
|
|
"\204\205\206\207\210\211\212\222\223\224\225\226\227\230\231\232\242"
|
|
"\243\244\245\246\247\250\251\252\262\263\264\265\266\267\270\271\272"
|
|
"\302\303\304\305\306\307\310\311\312\322\323\324\325\326\327\330\331"
|
|
"\332\342\343\344\345\346\347\350\351\352\362\363\364\365\366\367\370"
|
|
"\371\372\377\332\0\14\3\1\0\2\21\3\21\0?\0\216M\352\214\25\271\224"
|
|
"\262\310\253\363tl\22209\35O~\237\\\24QZ\306Mh\216\252i\364ml\177\377"
|
|
"\331";
|
|
size_t goodJpegLength = 659;
|
|
auto data = SkData::MakeWithoutCopy(goodJpeg, goodJpegLength);
|
|
REPORTER_ASSERT(r, SkIsJFIF(data.get(), &info));
|
|
REPORTER_ASSERT(r, info.fSize == (SkISize{8, 8}));
|
|
REPORTER_ASSERT(r, info.fType == SkJFIFInfo::kYCbCr);
|
|
|
|
// Not long enough to read first (SOI) segment marker.
|
|
data = SkData::MakeWithoutCopy(goodJpeg, 1);
|
|
REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
|
|
|
|
// Not long enough to read second segment (APP0) marker.
|
|
data = SkData::MakeWithoutCopy(goodJpeg, 3);
|
|
REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
|
|
|
|
// Not long enough to read second segment's length.
|
|
data = SkData::MakeWithoutCopy(goodJpeg, 5);
|
|
REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
|
|
|
|
// APP0 segment is truncated.
|
|
data = SkData::MakeWithoutCopy(goodJpeg, 7);
|
|
REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
|
|
|
|
// Missing SOF segment.
|
|
data = SkData::MakeWithoutCopy(goodJpeg, 89);
|
|
REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
|
|
}
|
|
{
|
|
// JFIF tag missing.
|
|
static const char jpeg[] =
|
|
"\377\330\377\340\0\20JFIX\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
|
|
"\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"
|
|
"\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
|
|
"\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
|
|
"22222222222222\377\300\0\21\10\2\0\2\0\3\1\"\0\2\21\1\3\21\001";
|
|
size_t jpegLength = 177;
|
|
auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
|
|
REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
|
|
}
|
|
{
|
|
// APP0 segment short (byte 6 changed).
|
|
static const char jpeg[] =
|
|
"\377\330\377\340\0\5JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
|
|
"\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"
|
|
"\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
|
|
"\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
|
|
"22222222222222\377\300\0\21\10\2\0\2\0\3\1\"\0\2\21\1\3\21\001";
|
|
size_t jpegLength = 177;
|
|
auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
|
|
REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
|
|
}
|
|
{
|
|
// SOF segment short. ('\21' replaced with '\5')
|
|
static const char jpeg[] =
|
|
"\377\330\377\340\0\20JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
|
|
"\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"
|
|
"\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
|
|
"\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
|
|
"22222222222222\377\300\0\5\10\2\0\2\0\3\1\"\0\2\21\1\3\21\001";
|
|
size_t jpegLength = 177;
|
|
auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
|
|
REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
|
|
}
|
|
{
|
|
// Unsupported 12-bit components. ('\10' replaced with '\14')
|
|
static const char jpeg[] =
|
|
"\377\330\377\340\0\20JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
|
|
"\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"
|
|
"\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
|
|
"\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
|
|
"22222222222222\377\300\0\21\14\2\0\2\0\3\1\"\0\2\21\1\3\21\001";
|
|
size_t jpegLength = 177;
|
|
auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
|
|
REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
|
|
}
|
|
{
|
|
// Two color channels. ('\3' replaced with '\2')
|
|
static const char jpeg[] =
|
|
"\377\330\377\340\0\20JFIF\0\1\1\0\0\1\0\1\0\0\377\333\0C\0\10\6\6\7"
|
|
"\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"
|
|
"\36\35\32\34\34 $.' \",#\34\34(7),01444\37'9=82<.342\377\333\0C\1\t"
|
|
"\t\t\14\13\14\30\r\r\0302!\34!222222222222222222222222222222222222"
|
|
"22222222222222\377\300\0\21\10\2\0\2\0\2\1\"\0\2\21\1\3\21\001";
|
|
size_t jpegLength = 177;
|
|
auto data = SkData::MakeWithoutCopy(jpeg, jpegLength);
|
|
REPORTER_ASSERT(r, !SkIsJFIF(data.get(), &info));
|
|
}
|
|
}
|
|
#endif
|