diff options
Diffstat (limited to 'testing/embedder_test.cpp')
-rw-r--r-- | testing/embedder_test.cpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/testing/embedder_test.cpp b/testing/embedder_test.cpp index 03eebe3abc..e4ac4ad2c7 100644 --- a/testing/embedder_test.cpp +++ b/testing/embedder_test.cpp @@ -546,7 +546,12 @@ int EmbedderTest::WriteBlockCallback(FPDF_FILEWRITE* pFileWrite, const void* data, unsigned long size) { EmbedderTest* pThis = static_cast<EmbedderTest*>(pFileWrite); + pThis->data_string_.append(static_cast<const char*>(data), size); + + if (pThis->filestream_.is_open()) + pThis->filestream_.write(static_cast<const char*>(data), size); + return 1; } @@ -587,3 +592,11 @@ int EmbedderTest::GetPageNumberForLoadedPage(FPDF_PAGE page) const { int EmbedderTest::GetPageNumberForSavedPage(FPDF_PAGE page) const { return GetPageNumberForPage(saved_page_map_, page); } + +void EmbedderTest::OpenPDFFileForWrite(const char* filename) { + filestream_.open(filename, std::ios_base::binary); +} + +void EmbedderTest::ClosePDFFileForWrite() { + filestream_.close(); +} |