| /* |
| * Copyright 2013 Google Inc. |
| * |
| * Use of this source code is governed by a BSD-style license that can be |
| * found in the LICENSE file. |
| */ |
| #include "include/core/SkBitmap.h" |
| #include "include/core/SkCanvas.h" |
| #include "include/core/SkColor.h" |
| #include "include/core/SkData.h" |
| #include "include/core/SkDocument.h" |
| #include "include/core/SkExecutor.h" |
| #include "include/core/SkFont.h" |
| #include "include/core/SkImage.h" // IWYU pragma: keep |
| #include "include/core/SkPaint.h" |
| #include "include/core/SkRefCnt.h" |
| #include "include/core/SkStream.h" |
| #include "include/core/SkString.h" |
| #include "include/docs/SkPDFDocument.h" |
| #include "src/utils/SkOSPath.h" |
| #include "tests/Test.h" |
| |
| #include <cstdint> |
| #include <cstdio> |
| #include <cstring> |
| #include <memory> |
| |
| static void test_empty(skiatest::Reporter* reporter) { |
| SkDynamicMemoryWStream stream; |
| |
| auto doc = SkPDF::MakeDocument(&stream); |
| |
| doc->close(); |
| |
| REPORTER_ASSERT(reporter, stream.bytesWritten() == 0); |
| } |
| |
| static void test_abort(skiatest::Reporter* reporter) { |
| SkDynamicMemoryWStream stream; |
| auto doc = SkPDF::MakeDocument(&stream); |
| |
| SkCanvas* canvas = doc->beginPage(100, 100); |
| canvas->drawColor(SK_ColorRED); |
| doc->endPage(); |
| |
| doc->abort(); |
| |
| // Test that only the header is written, not the full document. |
| REPORTER_ASSERT(reporter, stream.bytesWritten() < 256); |
| } |
| |
| static void test_abortWithFile(skiatest::Reporter* reporter) { |
| SkString tmpDir = skiatest::GetTmpDir(); |
| |
| if (tmpDir.isEmpty()) { |
| ERRORF(reporter, "missing tmpDir."); |
| return; |
| } |
| |
| SkString path = SkOSPath::Join(tmpDir.c_str(), "aborted.pdf"); |
| if (!SkFILEWStream(path.c_str()).isValid()) { |
| ERRORF(reporter, "unable to write to: %s", path.c_str()); |
| return; |
| } |
| |
| // Make sure doc's destructor is called to flush. |
| { |
| SkFILEWStream stream(path.c_str()); |
| auto doc = SkPDF::MakeDocument(&stream); |
| |
| SkCanvas* canvas = doc->beginPage(100, 100); |
| canvas->drawColor(SK_ColorRED); |
| doc->endPage(); |
| |
| doc->abort(); |
| } |
| |
| FILE* file = fopen(path.c_str(), "r"); |
| // Test that only the header is written, not the full document. |
| char buffer[256]; |
| REPORTER_ASSERT(reporter, fread(buffer, 1, sizeof(buffer), file) < sizeof(buffer)); |
| fclose(file); |
| } |
| |
| static void test_file(skiatest::Reporter* reporter) { |
| SkString tmpDir = skiatest::GetTmpDir(); |
| if (tmpDir.isEmpty()) { |
| ERRORF(reporter, "missing tmpDir."); |
| return; |
| } |
| |
| SkString path = SkOSPath::Join(tmpDir.c_str(), "file.pdf"); |
| if (!SkFILEWStream(path.c_str()).isValid()) { |
| ERRORF(reporter, "unable to write to: %s", path.c_str()); |
| return; |
| } |
| |
| { |
| SkFILEWStream stream(path.c_str()); |
| auto doc = SkPDF::MakeDocument(&stream); |
| SkCanvas* canvas = doc->beginPage(100, 100); |
| |
| canvas->drawColor(SK_ColorRED); |
| doc->endPage(); |
| doc->close(); |
| } |
| |
| FILE* file = fopen(path.c_str(), "r"); |
| REPORTER_ASSERT(reporter, file != nullptr); |
| char header[100]; |
| REPORTER_ASSERT(reporter, fread(header, 4, 1, file) != 0); |
| REPORTER_ASSERT(reporter, strncmp(header, "%PDF", 4) == 0); |
| fclose(file); |
| } |
| |
| static void test_close(skiatest::Reporter* reporter) { |
| SkDynamicMemoryWStream stream; |
| auto doc = SkPDF::MakeDocument(&stream); |
| |
| SkCanvas* canvas = doc->beginPage(100, 100); |
| canvas->drawColor(SK_ColorRED); |
| doc->endPage(); |
| |
| doc->close(); |
| |
| REPORTER_ASSERT(reporter, stream.bytesWritten() != 0); |
| } |
| |
| DEF_TEST(SkPDF_document_tests, reporter) { |
| REQUIRE_PDF_DOCUMENT(document_tests, reporter); |
| test_empty(reporter); |
| test_abort(reporter); |
| test_abortWithFile(reporter); |
| test_file(reporter); |
| test_close(reporter); |
| } |
| |
| DEF_TEST(SkPDF_document_skbug_4734, r) { |
| REQUIRE_PDF_DOCUMENT(SkPDF_document_skbug_4734, r); |
| SkDynamicMemoryWStream stream; |
| auto doc = SkPDF::MakeDocument(&stream); |
| SkCanvas* canvas = doc->beginPage(64, 64); |
| canvas->scale(10000.0f, 10000.0f); |
| canvas->translate(20.0f, 10.0f); |
| canvas->rotate(30.0f); |
| const char text[] = "HELLO"; |
| canvas->drawString(text, 0, 0, SkFont(), SkPaint()); |
| } |
| |
| static bool contains(const uint8_t* result, size_t size, const char expectation[]) { |
| size_t len = strlen(expectation); |
| size_t N = 1 + size - len; |
| for (size_t i = 0; i < N; ++i) { |
| if (0 == memcmp(result + i, expectation, len)) { |
| return true; |
| } |
| } |
| return false; |
| } |
| |
| // verify that the PDFA flag does something. |
| DEF_TEST(SkPDF_pdfa_document, r) { |
| REQUIRE_PDF_DOCUMENT(SkPDF_pdfa_document, r); |
| |
| SkPDF::Metadata pdfMetadata; |
| pdfMetadata.fTitle = "test document"; |
| pdfMetadata.fCreation = {0, 1999, 12, 5, 31, 23, 59, 59}; |
| pdfMetadata.fPDFA = true; |
| |
| SkDynamicMemoryWStream buffer; |
| auto doc = SkPDF::MakeDocument(&buffer, pdfMetadata); |
| doc->beginPage(64, 64)->drawColor(SK_ColorRED); |
| doc->close(); |
| sk_sp<SkData> data(buffer.detachAsData()); |
| |
| static const char* expectations[] = { |
| "sRGB IEC61966-2.1", |
| "<dc:title><rdf:Alt><rdf:li xml:lang=\"x-default\">test document", |
| "<xmp:CreateDate>1999-12-31T23:59:59+00:00</xmp:CreateDate>", |
| "/Subtype /XML", |
| "/CreationDate (D:19991231235959+00'00')>>", |
| }; |
| for (const char* expectation : expectations) { |
| if (!contains(data->bytes(), data->size(), expectation)) { |
| ERRORF(r, "PDFA expectation missing: '%s'.", expectation); |
| } |
| } |
| pdfMetadata.fProducer = "phoney library"; |
| pdfMetadata.fPDFA = true; |
| doc = SkPDF::MakeDocument(&buffer, pdfMetadata); |
| doc->beginPage(64, 64)->drawColor(SK_ColorRED); |
| doc->close(); |
| data = buffer.detachAsData(); |
| |
| static const char* moreExpectations[] = { |
| "/Producer (phoney library)", |
| "<pdf:Producer>phoney library</pdf:Producer>", |
| }; |
| for (const char* expectation : moreExpectations) { |
| if (!contains(data->bytes(), data->size(), expectation)) { |
| ERRORF(r, "PDFA expectation missing: '%s'.", expectation); |
| } |
| } |
| } |
| |
| |
| DEF_TEST(SkPDF_unicode_metadata, r) { |
| REQUIRE_PDF_DOCUMENT(SkPDF_unicode_metadata, r); |
| SkPDF::Metadata pdfMetadata; |
| pdfMetadata.fTitle = "πππππ πππππ"; // Out of basic multilingual plane |
| pdfMetadata.fAuthor = "ABCDE FGHIJ"; // ASCII |
| pdfMetadata.fSubject = "αβγδΡ ΢ηθικ"; // inside basic multilingual plane |
| pdfMetadata.fPDFA = true; |
| SkDynamicMemoryWStream wStream; |
| { |
| auto doc = SkPDF::MakeDocument(&wStream, pdfMetadata); |
| doc->beginPage(612, 792)->drawColor(SK_ColorCYAN); |
| } |
| sk_sp<SkData> data(wStream.detachAsData()); |
| static const char* expectations[] = { |
| ("<</Title <FEFFD835DCD0D835DCD1D835DCD2D835DCD3D835DCD40020" |
| "D835DCD5D835DCD6D835DCD7D835DCD8D835DCD9>"), |
| "/Author (ABCDE FGHIJ)", |
| "Subject <FEFF03B103B203B303B403B5002003B603B703B803B903BA>", |
| }; |
| for (const char* expectation : expectations) { |
| if (!contains(data->bytes(), data->size(), expectation)) { |
| ERRORF(r, "PDF expectation missing: '%s'.", expectation); |
| } |
| } |
| } |
| |
| // Make sure we excercise the multi-page functionality without problems. |
| // Add this to args.gn to output the PDF to a file: |
| // extra_cflags = [ "-DSK_PDF_TEST_MULTIPAGE=\"/tmp/skpdf_test_multipage.pdf\"" ] |
| DEF_TEST(SkPDF_multiple_pages, r) { |
| REQUIRE_PDF_DOCUMENT(SkPDF_multiple_pages, r); |
| int n = 100; |
| #ifdef SK_PDF_TEST_MULTIPAGE |
| SkFILEWStream wStream(SK_PDF_TEST_MULTIPAGE); |
| #else |
| SkDynamicMemoryWStream wStream; |
| #endif |
| auto doc = SkPDF::MakeDocument(&wStream); |
| for (int i = 0; i < n; ++i) { |
| doc->beginPage(612, 792)->drawColor( |
| SkColorSetARGB(0xFF, 0x00, (uint8_t)(255.0f * i / (n - 1)), 0x00)); |
| } |
| } |
| |
| // Test to make sure that jobs launched by PDF backend don't cause a segfault |
| // after calling abort(). |
| DEF_TEST(SkPDF_abort_jobs, rep) { |
| REQUIRE_PDF_DOCUMENT(SkPDF_abort_jobs, rep); |
| SkBitmap b; |
| b.allocN32Pixels(612, 792); |
| b.eraseColor(0x4F9643A0); |
| SkPDF::Metadata metadata; |
| std::unique_ptr<SkExecutor> executor = SkExecutor::MakeFIFOThreadPool(); |
| metadata.fExecutor = executor.get(); |
| SkNullWStream dst; |
| auto doc = SkPDF::MakeDocument(&dst, metadata); |
| doc->beginPage(612, 792)->drawImage(b.asImage(), 0, 0); |
| doc->abort(); |
| } |
| |