| OLD | NEW |
| 1 // Copyright 2015 PDFium Authors. All rights reserved. | 1 // Copyright 2015 PDFium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "testing/embedder_test.h" | 5 #include "testing/embedder_test.h" |
| 6 | 6 |
| 7 #include <limits.h> | 7 #include <limits.h> |
| 8 | 8 |
| 9 #include <list> | 9 #include <list> |
| 10 #include <string> | 10 #include <string> |
| (...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 104 bool EmbedderTest::CreateEmptyDocument() { | 104 bool EmbedderTest::CreateEmptyDocument() { |
| 105 document_ = FPDF_CreateNewDocument(); | 105 document_ = FPDF_CreateNewDocument(); |
| 106 if (!document_) | 106 if (!document_) |
| 107 return false; | 107 return false; |
| 108 | 108 |
| 109 SetupFormFillEnvironment(); | 109 SetupFormFillEnvironment(); |
| 110 return true; | 110 return true; |
| 111 } | 111 } |
| 112 | 112 |
| 113 bool EmbedderTest::OpenDocument(const std::string& filename, | 113 bool EmbedderTest::OpenDocument(const std::string& filename, |
| 114 const char* password, |
| 114 bool must_linearize) { | 115 bool must_linearize) { |
| 115 std::string file_path; | 116 std::string file_path; |
| 116 if (!PathService::GetTestFilePath(filename, &file_path)) | 117 if (!PathService::GetTestFilePath(filename, &file_path)) |
| 117 return false; | 118 return false; |
| 118 file_contents_ = GetFileContents(file_path.c_str(), &file_length_); | 119 file_contents_ = GetFileContents(file_path.c_str(), &file_length_); |
| 119 if (!file_contents_) | 120 if (!file_contents_) |
| 120 return false; | 121 return false; |
| 121 | 122 |
| 122 loader_ = new TestLoader(file_contents_.get(), file_length_); | 123 loader_ = new TestLoader(file_contents_.get(), file_length_); |
| 123 file_access_.m_FileLen = static_cast<unsigned long>(file_length_); | 124 file_access_.m_FileLen = static_cast<unsigned long>(file_length_); |
| 124 file_access_.m_GetBlock = TestLoader::GetBlock; | 125 file_access_.m_GetBlock = TestLoader::GetBlock; |
| 125 file_access_.m_Param = loader_; | 126 file_access_.m_Param = loader_; |
| 126 | 127 |
| 127 file_avail_.version = 1; | 128 file_avail_.version = 1; |
| 128 file_avail_.IsDataAvail = Is_Data_Avail; | 129 file_avail_.IsDataAvail = Is_Data_Avail; |
| 129 | 130 |
| 130 hints_.version = 1; | 131 hints_.version = 1; |
| 131 hints_.AddSegment = Add_Segment; | 132 hints_.AddSegment = Add_Segment; |
| 132 | 133 |
| 133 avail_ = FPDFAvail_Create(&file_avail_, &file_access_); | 134 avail_ = FPDFAvail_Create(&file_avail_, &file_access_); |
| 134 | 135 |
| 135 if (FPDFAvail_IsLinearized(avail_) == PDF_LINEARIZED) { | 136 if (FPDFAvail_IsLinearized(avail_) == PDF_LINEARIZED) { |
| 136 document_ = FPDFAvail_GetDocument(avail_, nullptr); | 137 document_ = FPDFAvail_GetDocument(avail_, password); |
| 137 if (!document_) { | 138 if (!document_) { |
| 138 return false; | 139 return false; |
| 139 } | 140 } |
| 140 int32_t nRet = PDF_DATA_NOTAVAIL; | 141 int32_t nRet = PDF_DATA_NOTAVAIL; |
| 141 while (nRet == PDF_DATA_NOTAVAIL) { | 142 while (nRet == PDF_DATA_NOTAVAIL) { |
| 142 nRet = FPDFAvail_IsDocAvail(avail_, &hints_); | 143 nRet = FPDFAvail_IsDocAvail(avail_, &hints_); |
| 143 } | 144 } |
| 144 if (nRet == PDF_DATA_ERROR) { | 145 if (nRet == PDF_DATA_ERROR) { |
| 145 return false; | 146 return false; |
| 146 } | 147 } |
| (...skipping 169 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 316 } | 317 } |
| 317 | 318 |
| 318 // Can't use gtest-provided main since we need to stash the path to the | 319 // Can't use gtest-provided main since we need to stash the path to the |
| 319 // executable in order to find the external V8 binary data files. | 320 // executable in order to find the external V8 binary data files. |
| 320 int main(int argc, char** argv) { | 321 int main(int argc, char** argv) { |
| 321 g_exe_path_ = argv[0]; | 322 g_exe_path_ = argv[0]; |
| 322 testing::InitGoogleTest(&argc, argv); | 323 testing::InitGoogleTest(&argc, argv); |
| 323 testing::InitGoogleMock(&argc, argv); | 324 testing::InitGoogleMock(&argc, argv); |
| 324 return RUN_ALL_TESTS(); | 325 return RUN_ALL_TESTS(); |
| 325 } | 326 } |
| OLD | NEW |