blob: 149ebf793c1d9d64e941a46cc17547e7334d6d4a [file] [log] [blame]
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -07001// Copyright (c) 2010 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include <limits.h>
6#include <stdio.h>
7#include <stdlib.h>
8#include <string.h>
9
10#include <list>
Tom Sepezdaa2e842015-01-29 15:44:37 -080011#include <sstream>
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -070012#include <string>
13#include <utility>
Tom Sepez5ee12d72014-12-17 16:24:01 -080014#include <vector>
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -070015
Cary Clark399be5b2016-03-14 16:51:29 -040016#if defined PDF_ENABLE_SKIA && !defined _SKIA_SUPPORT_
17#define _SKIA_SUPPORT_
18#endif
19
Lei Zhangb4e7f302015-11-06 15:52:32 -080020#include "public/fpdf_dataavail.h"
Lei Zhang453d96b2015-12-31 13:13:10 -080021#include "public/fpdf_edit.h"
Lei Zhangb4e7f302015-11-06 15:52:32 -080022#include "public/fpdf_ext.h"
23#include "public/fpdf_formfill.h"
24#include "public/fpdf_text.h"
25#include "public/fpdfview.h"
Dan Sinclairefbc1912016-02-17 16:54:43 -050026#include "samples/image_diff_png.h"
Lei Zhangbde53d22015-11-12 22:21:30 -080027#include "testing/test_support.h"
Tom Sepezd831dc72015-10-19 16:04:22 -070028
Tom Sepez452b4f32015-10-13 09:27:27 -070029#ifdef PDF_ENABLE_V8
John Abd-El-Malekb045ed22015-02-10 09:15:12 -080030#include "v8/include/libplatform/libplatform.h"
Tom Sepez1ed8a212015-05-11 15:25:39 -070031#include "v8/include/v8.h"
Lei Zhang8241df72015-11-06 14:38:48 -080032#endif // PDF_ENABLE_V8
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -070033
34#ifdef _WIN32
Tom Sepez5ee12d72014-12-17 16:24:01 -080035#define snprintf _snprintf
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -070036#endif
37
Cary Clark399be5b2016-03-14 16:51:29 -040038#ifdef PDF_ENABLE_SKIA
39#include "third_party/skia/include/core/SkPictureRecorder.h"
40#include "third_party/skia/include/core/SkStream.h"
41#endif
42
Vitaly Buka9e0177a2014-07-22 18:15:42 -070043enum OutputFormat {
44 OUTPUT_NONE,
45 OUTPUT_PPM,
Tom Sepezaf18cb32015-02-05 15:06:01 -080046 OUTPUT_PNG,
Vitaly Buka9e0177a2014-07-22 18:15:42 -070047#ifdef _WIN32
48 OUTPUT_BMP,
49 OUTPUT_EMF,
50#endif
Cary Clark399be5b2016-03-14 16:51:29 -040051#ifdef PDF_ENABLE_SKIA
52 OUTPUT_SKP,
53#endif
Vitaly Buka9e0177a2014-07-22 18:15:42 -070054};
55
Tom Sepez5ee12d72014-12-17 16:24:01 -080056struct Options {
tsepezf09bdfa2016-04-18 16:08:26 -070057 Options()
58 : show_config(false), send_events(false), output_format(OUTPUT_NONE) {}
Tom Sepez5ee12d72014-12-17 16:24:01 -080059
Tom Sepez2991d8d2016-01-15 16:02:48 -080060 bool show_config;
tsepezf09bdfa2016-04-18 16:08:26 -070061 bool send_events;
Tom Sepez5ee12d72014-12-17 16:24:01 -080062 OutputFormat output_format;
Tom Sepezdaa2e842015-01-29 15:44:37 -080063 std::string scale_factor_as_string;
Tom Sepez5ee12d72014-12-17 16:24:01 -080064 std::string exe_path;
65 std::string bin_directory;
Lei Zhang6f62d532015-09-23 15:31:44 -070066 std::string font_directory;
Tom Sepez5ee12d72014-12-17 16:24:01 -080067};
68
Tom Sepezaf18cb32015-02-05 15:06:01 -080069static bool CheckDimensions(int stride, int width, int height) {
70 if (stride < 0 || width < 0 || height < 0)
71 return false;
72 if (height > 0 && width > INT_MAX / height)
73 return false;
74 return true;
75}
76
Vitaly Buka9e0177a2014-07-22 18:15:42 -070077static void WritePpm(const char* pdf_name, int num, const void* buffer_void,
78 int stride, int width, int height) {
79 const char* buffer = reinterpret_cast<const char*>(buffer_void);
80
Tom Sepezaf18cb32015-02-05 15:06:01 -080081 if (!CheckDimensions(stride, width, height))
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -070082 return;
Tom Sepezaf18cb32015-02-05 15:06:01 -080083
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -070084 int out_len = width * height;
85 if (out_len > INT_MAX / 3)
86 return;
87 out_len *= 3;
88
89 char filename[256];
90 snprintf(filename, sizeof(filename), "%s.%d.ppm", pdf_name, num);
John Abd-El-Maleka548d302014-06-26 10:18:11 -070091 FILE* fp = fopen(filename, "wb");
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -070092 if (!fp)
93 return;
94 fprintf(fp, "P6\n# PDF test render\n%d %d\n255\n", width, height);
95 // Source data is B, G, R, unused.
96 // Dest data is R, G, B.
97 char* result = new char[out_len];
Lei Zhange00660b2015-08-13 15:40:18 -070098 for (int h = 0; h < height; ++h) {
99 const char* src_line = buffer + (stride * h);
100 char* dest_line = result + (width * h * 3);
101 for (int w = 0; w < width; ++w) {
102 // R
103 dest_line[w * 3] = src_line[(w * 4) + 2];
104 // G
105 dest_line[(w * 3) + 1] = src_line[(w * 4) + 1];
106 // B
107 dest_line[(w * 3) + 2] = src_line[w * 4];
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700108 }
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700109 }
Lei Zhange00660b2015-08-13 15:40:18 -0700110 fwrite(result, out_len, 1, fp);
111 delete[] result;
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700112 fclose(fp);
113}
114
Tom Sepezaf18cb32015-02-05 15:06:01 -0800115static void WritePng(const char* pdf_name, int num, const void* buffer_void,
116 int stride, int width, int height) {
117 if (!CheckDimensions(stride, width, height))
118 return;
119
120 std::vector<unsigned char> png_encoding;
121 const unsigned char* buffer = static_cast<const unsigned char*>(buffer_void);
122 if (!image_diff_png::EncodeBGRAPNG(
123 buffer, width, height, stride, false, &png_encoding)) {
124 fprintf(stderr, "Failed to convert bitmap to PNG\n");
125 return;
126 }
127
128 char filename[256];
129 int chars_formatted = snprintf(
130 filename, sizeof(filename), "%s.%d.png", pdf_name, num);
131 if (chars_formatted < 0 ||
132 static_cast<size_t>(chars_formatted) >= sizeof(filename)) {
Cary Clark399be5b2016-03-14 16:51:29 -0400133 fprintf(stderr, "Filename %s is too long\n", filename);
Tom Sepezaf18cb32015-02-05 15:06:01 -0800134 return;
135 }
136
137 FILE* fp = fopen(filename, "wb");
138 if (!fp) {
139 fprintf(stderr, "Failed to open %s for output\n", filename);
140 return;
141 }
142
143 size_t bytes_written = fwrite(
144 &png_encoding.front(), 1, png_encoding.size(), fp);
145 if (bytes_written != png_encoding.size())
146 fprintf(stderr, "Failed to write to %s\n", filename);
147
Lei Zhang5377ebf2015-09-23 14:52:53 -0700148 (void)fclose(fp);
Tom Sepezaf18cb32015-02-05 15:06:01 -0800149}
150
Vitaly Buka9e0177a2014-07-22 18:15:42 -0700151#ifdef _WIN32
152static void WriteBmp(const char* pdf_name, int num, const void* buffer,
153 int stride, int width, int height) {
Tom Sepezaf18cb32015-02-05 15:06:01 -0800154 if (!CheckDimensions(stride, width, height))
Vitaly Buka9e0177a2014-07-22 18:15:42 -0700155 return;
Tom Sepezaf18cb32015-02-05 15:06:01 -0800156
Vitaly Buka9e0177a2014-07-22 18:15:42 -0700157 int out_len = stride * height;
158 if (out_len > INT_MAX / 3)
159 return;
160
161 char filename[256];
162 snprintf(filename, sizeof(filename), "%s.%d.bmp", pdf_name, num);
163 FILE* fp = fopen(filename, "wb");
164 if (!fp)
165 return;
166
Nico Weber2827bdd2015-07-01 14:08:08 -0700167 BITMAPINFO bmi = {};
Vitaly Buka9e0177a2014-07-22 18:15:42 -0700168 bmi.bmiHeader.biSize = sizeof(bmi) - sizeof(RGBQUAD);
169 bmi.bmiHeader.biWidth = width;
170 bmi.bmiHeader.biHeight = -height; // top-down image
171 bmi.bmiHeader.biPlanes = 1;
172 bmi.bmiHeader.biBitCount = 32;
173 bmi.bmiHeader.biCompression = BI_RGB;
174 bmi.bmiHeader.biSizeImage = 0;
175
Nico Weber2827bdd2015-07-01 14:08:08 -0700176 BITMAPFILEHEADER file_header = {};
Vitaly Buka9e0177a2014-07-22 18:15:42 -0700177 file_header.bfType = 0x4d42;
178 file_header.bfSize = sizeof(file_header) + bmi.bmiHeader.biSize + out_len;
179 file_header.bfOffBits = file_header.bfSize - out_len;
180
181 fwrite(&file_header, sizeof(file_header), 1, fp);
182 fwrite(&bmi, bmi.bmiHeader.biSize, 1, fp);
183 fwrite(buffer, out_len, 1, fp);
184 fclose(fp);
185}
186
187void WriteEmf(FPDF_PAGE page, const char* pdf_name, int num) {
188 int width = static_cast<int>(FPDF_GetPageWidth(page));
189 int height = static_cast<int>(FPDF_GetPageHeight(page));
190
191 char filename[256];
192 snprintf(filename, sizeof(filename), "%s.%d.emf", pdf_name, num);
193
Lei Zhang5377ebf2015-09-23 14:52:53 -0700194 HDC dc = CreateEnhMetaFileA(nullptr, filename, nullptr, nullptr);
Tom Sepezaf18cb32015-02-05 15:06:01 -0800195
196 HRGN rgn = CreateRectRgn(0, 0, width, height);
197 SelectClipRgn(dc, rgn);
Vitaly Buka9e0177a2014-07-22 18:15:42 -0700198 DeleteObject(rgn);
199
200 SelectObject(dc, GetStockObject(NULL_PEN));
201 SelectObject(dc, GetStockObject(WHITE_BRUSH));
202 // If a PS_NULL pen is used, the dimensions of the rectangle are 1 pixel less.
203 Rectangle(dc, 0, 0, width + 1, height + 1);
204
205 FPDF_RenderPage(dc, page, 0, 0, width, height, 0,
206 FPDF_ANNOT | FPDF_PRINTING | FPDF_NO_CATCH);
207
208 DeleteEnhMetaFile(CloseEnhMetaFile(dc));
209}
210#endif
211
Cary Clark399be5b2016-03-14 16:51:29 -0400212#ifdef PDF_ENABLE_SKIA
213void WriteSkp(const char* pdf_name, int num, const void* recorder) {
214 char filename[256];
215 int chars_formatted =
216 snprintf(filename, sizeof(filename), "%s.%d.skp", pdf_name, num);
217
218 if (chars_formatted < 0 ||
219 static_cast<size_t>(chars_formatted) >= sizeof(filename)) {
220 fprintf(stderr, "Filename %s is too long\n", filename);
221 return;
222 }
223
224 SkPictureRecorder* r = (SkPictureRecorder*)recorder;
caryclark31735432016-05-11 10:52:53 -0700225 sk_sp<SkPicture> picture(r->finishRecordingAsPicture());
Cary Clark399be5b2016-03-14 16:51:29 -0400226 SkFILEWStream wStream(filename);
227 picture->serialize(&wStream);
228}
229#endif
230
Tom Sepez58fb36a2016-02-01 10:32:14 -0800231// These example JS platform callback handlers are entirely optional,
232// and exist here to show the flow of information from a document back
233// to the embedder.
Tom Sepezbd932572016-01-29 09:10:41 -0800234int ExampleAppAlert(IPDF_JSPLATFORM*,
235 FPDF_WIDESTRING msg,
236 FPDF_WIDESTRING title,
237 int nType,
238 int nIcon) {
239 printf("%ls", GetPlatformWString(title).c_str());
240 if (nIcon || nType)
241 printf("[icon=%d,type=%d]", nIcon, nType);
242 printf(": %ls\n", GetPlatformWString(msg).c_str());
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700243 return 0;
244}
245
Tom Sepez58fb36a2016-02-01 10:32:14 -0800246int ExampleAppResponse(IPDF_JSPLATFORM*,
247 FPDF_WIDESTRING question,
248 FPDF_WIDESTRING title,
249 FPDF_WIDESTRING defaultValue,
250 FPDF_WIDESTRING label,
251 FPDF_BOOL isPassword,
252 void* response,
253 int length) {
254 printf("%ls: %ls, defaultValue=%ls, label=%ls, isPassword=%d, length=%d\n",
255 GetPlatformWString(title).c_str(),
256 GetPlatformWString(question).c_str(),
257 GetPlatformWString(defaultValue).c_str(),
258 GetPlatformWString(label).c_str(), isPassword, length);
259
260 // UTF-16, always LE regardless of platform.
261 uint8_t* ptr = static_cast<uint8_t*>(response);
262 ptr[0] = 'N';
263 ptr[1] = 0;
264 ptr[2] = 'o';
265 ptr[3] = 0;
266 return 4;
267}
268
Tom Sepezb7cb36a2015-02-13 16:54:48 -0800269void ExampleDocGotoPage(IPDF_JSPLATFORM*, int pageNumber) {
270 printf("Goto Page: %d\n", pageNumber);
271}
272
Tom Sepeze5fbd7a2016-01-29 17:05:08 -0800273void ExampleDocMail(IPDF_JSPLATFORM*,
274 void* mailData,
275 int length,
276 FPDF_BOOL bUI,
277 FPDF_WIDESTRING To,
278 FPDF_WIDESTRING Subject,
279 FPDF_WIDESTRING CC,
280 FPDF_WIDESTRING BCC,
281 FPDF_WIDESTRING Msg) {
282 printf("Mail Msg: %d, to=%ls, cc=%ls, bcc=%ls, subject=%ls, body=%ls\n", bUI,
283 GetPlatformWString(To).c_str(), GetPlatformWString(CC).c_str(),
284 GetPlatformWString(BCC).c_str(), GetPlatformWString(Subject).c_str(),
285 GetPlatformWString(Msg).c_str());
286}
287
Tom Sepezb7cb36a2015-02-13 16:54:48 -0800288void ExampleUnsupportedHandler(UNSUPPORT_INFO*, int type) {
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700289 std::string feature = "Unknown";
290 switch (type) {
291 case FPDF_UNSP_DOC_XFAFORM:
292 feature = "XFA";
293 break;
294 case FPDF_UNSP_DOC_PORTABLECOLLECTION:
295 feature = "Portfolios_Packages";
296 break;
297 case FPDF_UNSP_DOC_ATTACHMENT:
298 case FPDF_UNSP_ANNOT_ATTACHMENT:
299 feature = "Attachment";
300 break;
301 case FPDF_UNSP_DOC_SECURITY:
302 feature = "Rights_Management";
303 break;
304 case FPDF_UNSP_DOC_SHAREDREVIEW:
305 feature = "Shared_Review";
306 break;
307 case FPDF_UNSP_DOC_SHAREDFORM_ACROBAT:
308 case FPDF_UNSP_DOC_SHAREDFORM_FILESYSTEM:
309 case FPDF_UNSP_DOC_SHAREDFORM_EMAIL:
310 feature = "Shared_Form";
311 break;
312 case FPDF_UNSP_ANNOT_3DANNOT:
313 feature = "3D";
314 break;
315 case FPDF_UNSP_ANNOT_MOVIE:
316 feature = "Movie";
317 break;
318 case FPDF_UNSP_ANNOT_SOUND:
319 feature = "Sound";
320 break;
321 case FPDF_UNSP_ANNOT_SCREEN_MEDIA:
322 case FPDF_UNSP_ANNOT_SCREEN_RICHMEDIA:
323 feature = "Screen";
324 break;
325 case FPDF_UNSP_ANNOT_SIG:
326 feature = "Digital_Signature";
327 break;
328 }
329 printf("Unsupported feature: %s.\n", feature.c_str());
330}
331
Tom Sepez5ee12d72014-12-17 16:24:01 -0800332bool ParseCommandLine(const std::vector<std::string>& args,
333 Options* options, std::list<std::string>* files) {
334 if (args.empty()) {
335 return false;
336 }
337 options->exe_path = args[0];
Bo Xud44e3922014-12-19 02:27:25 -0800338 size_t cur_idx = 1;
Tom Sepez5ee12d72014-12-17 16:24:01 -0800339 for (; cur_idx < args.size(); ++cur_idx) {
340 const std::string& cur_arg = args[cur_idx];
Tom Sepez2991d8d2016-01-15 16:02:48 -0800341 if (cur_arg == "--show-config") {
342 options->show_config = true;
tsepezf09bdfa2016-04-18 16:08:26 -0700343 } else if (cur_arg == "--send-events") {
344 options->send_events = true;
Tom Sepez2991d8d2016-01-15 16:02:48 -0800345 } else if (cur_arg == "--ppm") {
Tom Sepez5ee12d72014-12-17 16:24:01 -0800346 if (options->output_format != OUTPUT_NONE) {
347 fprintf(stderr, "Duplicate or conflicting --ppm argument\n");
348 return false;
349 }
350 options->output_format = OUTPUT_PPM;
Tom Sepezaf18cb32015-02-05 15:06:01 -0800351 } else if (cur_arg == "--png") {
352 if (options->output_format != OUTPUT_NONE) {
353 fprintf(stderr, "Duplicate or conflicting --png argument\n");
354 return false;
355 }
356 options->output_format = OUTPUT_PNG;
Cary Clark399be5b2016-03-14 16:51:29 -0400357#ifdef PDF_ENABLE_SKIA
358 } else if (cur_arg == "--skp") {
359 if (options->output_format != OUTPUT_NONE) {
360 fprintf(stderr, "Duplicate or conflicting --skp argument\n");
361 return false;
362 }
363 options->output_format = OUTPUT_SKP;
364#endif
Lei Zhang6f62d532015-09-23 15:31:44 -0700365 } else if (cur_arg.size() > 11 &&
366 cur_arg.compare(0, 11, "--font-dir=") == 0) {
367 if (!options->font_directory.empty()) {
368 fprintf(stderr, "Duplicate --font-dir argument\n");
369 return false;
370 }
371 options->font_directory = cur_arg.substr(11);
Vitaly Buka9e0177a2014-07-22 18:15:42 -0700372#ifdef _WIN32
Dan Sinclair738b08c2016-03-01 14:45:20 -0500373 } else if (cur_arg == "--emf") {
Tom Sepez5ee12d72014-12-17 16:24:01 -0800374 if (options->output_format != OUTPUT_NONE) {
375 fprintf(stderr, "Duplicate or conflicting --emf argument\n");
376 return false;
377 }
378 options->output_format = OUTPUT_EMF;
Dan Sinclair50cce602016-02-24 09:51:16 -0500379 } else if (cur_arg == "--bmp") {
Tom Sepez5ee12d72014-12-17 16:24:01 -0800380 if (options->output_format != OUTPUT_NONE) {
381 fprintf(stderr, "Duplicate or conflicting --bmp argument\n");
382 return false;
383 }
384 options->output_format = OUTPUT_BMP;
Tom Sepez5ee12d72014-12-17 16:24:01 -0800385#endif // _WIN32
Dan Sinclair738b08c2016-03-01 14:45:20 -0500386
Tom Sepez452b4f32015-10-13 09:27:27 -0700387#ifdef PDF_ENABLE_V8
Tom Sepez5ee12d72014-12-17 16:24:01 -0800388#ifdef V8_USE_EXTERNAL_STARTUP_DATA
Dan Sinclair738b08c2016-03-01 14:45:20 -0500389 } else if (cur_arg.size() > 10 &&
390 cur_arg.compare(0, 10, "--bin-dir=") == 0) {
Tom Sepez5ee12d72014-12-17 16:24:01 -0800391 if (!options->bin_directory.empty()) {
392 fprintf(stderr, "Duplicate --bin-dir argument\n");
393 return false;
394 }
395 options->bin_directory = cur_arg.substr(10);
Tom Sepez5ee12d72014-12-17 16:24:01 -0800396#endif // V8_USE_EXTERNAL_STARTUP_DATA
Tom Sepez452b4f32015-10-13 09:27:27 -0700397#endif // PDF_ENABLE_V8
Dan Sinclair738b08c2016-03-01 14:45:20 -0500398
399 } else if (cur_arg.size() > 8 && cur_arg.compare(0, 8, "--scale=") == 0) {
Tom Sepezdaa2e842015-01-29 15:44:37 -0800400 if (!options->scale_factor_as_string.empty()) {
401 fprintf(stderr, "Duplicate --scale argument\n");
402 return false;
403 }
404 options->scale_factor_as_string = cur_arg.substr(8);
Tom Sepez2991d8d2016-01-15 16:02:48 -0800405 } else if (cur_arg.size() >= 2 && cur_arg[0] == '-' && cur_arg[1] == '-') {
406 fprintf(stderr, "Unrecognized argument %s\n", cur_arg.c_str());
407 return false;
Dan Sinclair738b08c2016-03-01 14:45:20 -0500408 } else {
Vitaly Buka8f2c3dc2014-08-20 10:32:36 -0700409 break;
Dan Sinclair738b08c2016-03-01 14:45:20 -0500410 }
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700411 }
Bo Xud44e3922014-12-19 02:27:25 -0800412 for (size_t i = cur_idx; i < args.size(); i++) {
Tom Sepez5ee12d72014-12-17 16:24:01 -0800413 files->push_back(args[i]);
414 }
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700415 return true;
416}
417
Tom Sepezcf22eb82015-05-12 17:28:08 -0700418FPDF_BOOL Is_Data_Avail(FX_FILEAVAIL* pThis, size_t offset, size_t size) {
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700419 return true;
420}
421
422void Add_Segment(FX_DOWNLOADHINTS* pThis, size_t offset, size_t size) {
423}
424
tsepezf09bdfa2016-04-18 16:08:26 -0700425void SendPageEvents(const FPDF_FORMHANDLE& form,
426 const FPDF_PAGE& page,
427 const std::string& events) {
428 auto lines = StringSplit(events, '\n');
429 for (auto line : lines) {
430 auto command = StringSplit(line, '#');
431 if (command[0].empty())
432 continue;
433 auto tokens = StringSplit(command[0], ',');
434 if (tokens[0] == "keycode") {
435 if (tokens.size() == 2) {
436 int keycode = atoi(tokens[1].c_str());
437 FORM_OnKeyDown(form, page, keycode, 0);
438 FORM_OnKeyUp(form, page, keycode, 0);
439 } else {
440 fprintf(stderr, "keycode: bad args\n");
441 }
442 } else if (tokens[0] == "mousedown") {
443 if (tokens.size() == 4) {
444 int x = atoi(tokens[2].c_str());
445 int y = atoi(tokens[3].c_str());
446 if (tokens[1] == "left")
447 FORM_OnLButtonDown(form, page, 0, x, y);
448#ifdef PDF_ENABLE_XFA
449 else if (tokens[1] == "right")
450 FORM_OnRButtonDown(form, page, 0, x, y);
451#endif
452 else
453 fprintf(stderr, "mousedown: bad button name\n");
454 } else {
455 fprintf(stderr, "mousedown: bad args\n");
456 }
457 } else if (tokens[0] == "mouseup") {
458 if (tokens.size() == 4) {
459 int x = atoi(tokens[2].c_str());
460 int y = atoi(tokens[3].c_str());
461 if (tokens[1] == "left")
462 FORM_OnLButtonUp(form, page, 0, x, y);
463#ifdef PDF_ENABLE_XFA
464 else if (tokens[1] == "right")
465 FORM_OnRButtonUp(form, page, 0, x, y);
466#endif
467 else
468 fprintf(stderr, "mouseup: bad button name\n");
469 } else {
470 fprintf(stderr, "mouseup: bad args\n");
471 }
472 } else if (tokens[0] == "mousemove") {
473 if (tokens.size() == 3) {
474 int x = atoi(tokens[1].c_str());
475 int y = atoi(tokens[2].c_str());
476 FORM_OnMouseMove(form, page, 0, x, y);
477 } else {
478 fprintf(stderr, "mousemove: bad args\n");
479 }
480 } else {
481 fprintf(stderr, "Unrecognized event: %s\n", tokens[0].c_str());
482 }
483 }
484}
485
Jun Fangb553bcb2015-11-10 18:49:04 +0800486bool RenderPage(const std::string& name,
487 const FPDF_DOCUMENT& doc,
488 const FPDF_FORMHANDLE& form,
489 const int page_index,
tsepezf09bdfa2016-04-18 16:08:26 -0700490 const Options& options,
491 const std::string& events) {
Jun Fangdf7f3662015-11-10 18:29:18 +0800492 FPDF_PAGE page = FPDF_LoadPage(doc, page_index);
493 if (!page) {
Jun Fangb553bcb2015-11-10 18:49:04 +0800494 return false;
Jun Fangdf7f3662015-11-10 18:29:18 +0800495 }
496 FPDF_TEXTPAGE text_page = FPDFText_LoadPage(page);
497 FORM_OnAfterLoadPage(page, form);
498 FORM_DoPageAAction(page, form, FPDFPAGE_AACTION_OPEN);
499
tsepezf09bdfa2016-04-18 16:08:26 -0700500 if (options.send_events)
tsepez10b01bf2016-05-04 12:52:42 -0700501 SendPageEvents(form, page, events);
tsepezf09bdfa2016-04-18 16:08:26 -0700502
Jun Fangdf7f3662015-11-10 18:29:18 +0800503 double scale = 1.0;
504 if (!options.scale_factor_as_string.empty()) {
505 std::stringstream(options.scale_factor_as_string) >> scale;
506 }
507 int width = static_cast<int>(FPDF_GetPageWidth(page) * scale);
508 int height = static_cast<int>(FPDF_GetPageHeight(page) * scale);
Lei Zhang453d96b2015-12-31 13:13:10 -0800509 int alpha = FPDFPage_HasTransparency(page) ? 1 : 0;
510 FPDF_BITMAP bitmap = FPDFBitmap_Create(width, height, alpha);
Jun Fangdf7f3662015-11-10 18:29:18 +0800511 if (!bitmap) {
512 fprintf(stderr, "Page was too large to be rendered.\n");
Jun Fangb553bcb2015-11-10 18:49:04 +0800513 return false;
Jun Fangdf7f3662015-11-10 18:29:18 +0800514 }
Lei Zhang453d96b2015-12-31 13:13:10 -0800515 FPDF_DWORD fill_color = alpha ? 0x00000000 : 0xFFFFFFFF;
516 FPDFBitmap_FillRect(bitmap, 0, 0, width, height, fill_color);
Jun Fangdf7f3662015-11-10 18:29:18 +0800517 FPDF_RenderPageBitmap(bitmap, page, 0, 0, width, height, 0, 0);
518
519 FPDF_FFLDraw(form, bitmap, page, 0, 0, width, height, 0, 0);
520 int stride = FPDFBitmap_GetStride(bitmap);
521 const char* buffer =
522 reinterpret_cast<const char*>(FPDFBitmap_GetBuffer(bitmap));
523
524 switch (options.output_format) {
525#ifdef _WIN32
526 case OUTPUT_BMP:
527 WriteBmp(name.c_str(), page_index, buffer, stride, width, height);
528 break;
529
530 case OUTPUT_EMF:
531 WriteEmf(page, name.c_str(), page_index);
532 break;
533#endif
534 case OUTPUT_PNG:
535 WritePng(name.c_str(), page_index, buffer, stride, width, height);
536 break;
537
538 case OUTPUT_PPM:
539 WritePpm(name.c_str(), page_index, buffer, stride, width, height);
540 break;
541
Cary Clark399be5b2016-03-14 16:51:29 -0400542#ifdef PDF_ENABLE_SKIA
543 case OUTPUT_SKP: {
544 std::unique_ptr<SkPictureRecorder> recorder(
545 (SkPictureRecorder*)FPDF_RenderPageSkp(page, width, height));
546 FPDF_FFLRecord(form, recorder.get(), page, 0, 0, width, height, 0, 0);
547 WriteSkp(name.c_str(), page_index, recorder.get());
548 } break;
549#endif
Jun Fangdf7f3662015-11-10 18:29:18 +0800550 default:
551 break;
552 }
553
554 FPDFBitmap_Destroy(bitmap);
555 FORM_DoPageAAction(page, form, FPDFPAGE_AACTION_CLOSE);
556 FORM_OnBeforeClosePage(page, form);
557 FPDFText_ClosePage(text_page);
558 FPDF_ClosePage(page);
Jun Fangb553bcb2015-11-10 18:49:04 +0800559 return true;
Jun Fangdf7f3662015-11-10 18:29:18 +0800560}
561
tsepezf09bdfa2016-04-18 16:08:26 -0700562void RenderPdf(const std::string& name,
563 const char* pBuf,
564 size_t len,
565 const Options& options,
566 const std::string& events) {
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700567 IPDF_JSPLATFORM platform_callbacks;
568 memset(&platform_callbacks, '\0', sizeof(platform_callbacks));
Tom Sepeza72e8e22015-10-07 10:17:53 -0700569 platform_callbacks.version = 3;
Tom Sepezb7cb36a2015-02-13 16:54:48 -0800570 platform_callbacks.app_alert = ExampleAppAlert;
Tom Sepez58fb36a2016-02-01 10:32:14 -0800571 platform_callbacks.app_response = ExampleAppResponse;
Tom Sepezb7cb36a2015-02-13 16:54:48 -0800572 platform_callbacks.Doc_gotoPage = ExampleDocGotoPage;
Tom Sepeze5fbd7a2016-01-29 17:05:08 -0800573 platform_callbacks.Doc_mail = ExampleDocMail;
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700574
575 FPDF_FORMFILLINFO form_callbacks;
576 memset(&form_callbacks, '\0', sizeof(form_callbacks));
Tom Sepezc46d0002015-11-30 15:46:36 -0800577#ifdef PDF_ENABLE_XFA
Tom Sepezed631382014-11-18 14:10:25 -0800578 form_callbacks.version = 2;
Tom Sepezc46d0002015-11-30 15:46:36 -0800579#else // PDF_ENABLE_XFA
580 form_callbacks.version = 1;
581#endif // PDF_ENABLE_XFA
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700582 form_callbacks.m_pJsPlatform = &platform_callbacks;
583
584 TestLoader loader(pBuf, len);
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700585 FPDF_FILEACCESS file_access;
586 memset(&file_access, '\0', sizeof(file_access));
John Abd-El-Malek7dc51722014-05-26 12:54:31 -0700587 file_access.m_FileLen = static_cast<unsigned long>(len);
Tom Sepezd831dc72015-10-19 16:04:22 -0700588 file_access.m_GetBlock = TestLoader::GetBlock;
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700589 file_access.m_Param = &loader;
590
591 FX_FILEAVAIL file_avail;
592 memset(&file_avail, '\0', sizeof(file_avail));
593 file_avail.version = 1;
594 file_avail.IsDataAvail = Is_Data_Avail;
595
596 FX_DOWNLOADHINTS hints;
597 memset(&hints, '\0', sizeof(hints));
598 hints.version = 1;
599 hints.AddSegment = Add_Segment;
600
601 FPDF_DOCUMENT doc;
Jun Fangdf7f3662015-11-10 18:29:18 +0800602 int nRet = PDF_DATA_NOTAVAIL;
Jun Fangb553bcb2015-11-10 18:49:04 +0800603 bool bIsLinearized = false;
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700604 FPDF_AVAIL pdf_avail = FPDFAvail_Create(&file_avail, &file_access);
Tom Sepezc98895c2015-11-24 15:30:36 -0800605
Jun Fangdf7f3662015-11-10 18:29:18 +0800606 if (FPDFAvail_IsLinearized(pdf_avail) == PDF_LINEARIZED) {
Lei Zhang5377ebf2015-09-23 14:52:53 -0700607 doc = FPDFAvail_GetDocument(pdf_avail, nullptr);
Jun Fangdf7f3662015-11-10 18:29:18 +0800608 if (doc) {
609 while (nRet == PDF_DATA_NOTAVAIL) {
610 nRet = FPDFAvail_IsDocAvail(pdf_avail, &hints);
611 }
612 if (nRet == PDF_DATA_ERROR) {
613 fprintf(stderr, "Unknown error in checking if doc was available.\n");
614 return;
615 }
616 nRet = FPDFAvail_IsFormAvail(pdf_avail, &hints);
617 if (nRet == PDF_FORM_ERROR || nRet == PDF_FORM_NOTAVAIL) {
618 fprintf(stderr,
619 "Error %d was returned in checking if form was available.\n",
620 nRet);
621 return;
622 }
Jun Fangb553bcb2015-11-10 18:49:04 +0800623 bIsLinearized = true;
Jun Fangdf7f3662015-11-10 18:29:18 +0800624 }
625 } else {
Lei Zhang600d4072015-09-23 15:35:25 -0700626 doc = FPDF_LoadCustomDocument(&file_access, nullptr);
Jun Fangdf7f3662015-11-10 18:29:18 +0800627 }
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700628
Lei Zhang5377ebf2015-09-23 14:52:53 -0700629 if (!doc) {
Dan Sinclaireb815bf2015-10-27 13:08:41 -0400630 unsigned long err = FPDF_GetLastError();
631 fprintf(stderr, "Load pdf docs unsuccessful: ");
632 switch (err) {
633 case FPDF_ERR_SUCCESS:
634 fprintf(stderr, "Success");
635 break;
636 case FPDF_ERR_UNKNOWN:
637 fprintf(stderr, "Unknown error");
638 break;
639 case FPDF_ERR_FILE:
640 fprintf(stderr, "File not found or could not be opened");
641 break;
642 case FPDF_ERR_FORMAT:
643 fprintf(stderr, "File not in PDF format or corrupted");
644 break;
645 case FPDF_ERR_PASSWORD:
646 fprintf(stderr, "Password required or incorrect password");
647 break;
648 case FPDF_ERR_SECURITY:
649 fprintf(stderr, "Unsupported security scheme");
650 break;
651 case FPDF_ERR_PAGE:
652 fprintf(stderr, "Page not found or content error");
653 break;
654 default:
655 fprintf(stderr, "Unknown error %ld", err);
656 }
657 fprintf(stderr, ".\n");
658
Qin Zhao05224972015-10-20 18:31:06 -0400659 FPDFAvail_Destroy(pdf_avail);
JUN FANG827a1722015-03-05 13:39:21 -0800660 return;
661 }
662
Lei Zhang5377ebf2015-09-23 14:52:53 -0700663 (void)FPDF_GetDocPermissions(doc);
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700664
Bo Xu2b7a49d2014-11-14 17:40:50 -0800665 FPDF_FORMHANDLE form = FPDFDOC_InitFormFillEnvironment(doc, &form_callbacks);
Tom Sepezc46d0002015-11-30 15:46:36 -0800666#ifdef PDF_ENABLE_XFA
JUN FANG827a1722015-03-05 13:39:21 -0800667 int docType = DOCTYPE_PDF;
Lei Zhang5377ebf2015-09-23 14:52:53 -0700668 if (FPDF_HasXFAField(doc, &docType) && docType != DOCTYPE_PDF &&
669 !FPDF_LoadXFA(doc)) {
670 fprintf(stderr, "LoadXFA unsuccessful, continuing anyway.\n");
Tom Sepez56451382014-12-05 13:30:51 -0800671 }
Tom Sepezc46d0002015-11-30 15:46:36 -0800672#endif // PDF_ENABLE_XFA
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700673 FPDF_SetFormFieldHighlightColor(form, 0, 0xFFE4DD);
674 FPDF_SetFormFieldHighlightAlpha(form, 100);
675
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700676 FORM_DoDocumentJSAction(form);
677 FORM_DoDocumentOpenAction(form);
678
Jun Fangdf7f3662015-11-10 18:29:18 +0800679 int page_count = FPDF_GetPageCount(doc);
Tom Sepez1ed8a212015-05-11 15:25:39 -0700680 int rendered_pages = 0;
681 int bad_pages = 0;
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700682 for (int i = 0; i < page_count; ++i) {
Jun Fangdf7f3662015-11-10 18:29:18 +0800683 if (bIsLinearized) {
684 nRet = PDF_DATA_NOTAVAIL;
685 while (nRet == PDF_DATA_NOTAVAIL) {
686 nRet = FPDFAvail_IsPageAvail(pdf_avail, i, &hints);
687 }
688 if (nRet == PDF_DATA_ERROR) {
689 fprintf(stderr, "Unknown error in checking if page %d is available.\n",
690 i);
691 return;
692 }
693 }
tsepezf09bdfa2016-04-18 16:08:26 -0700694 if (RenderPage(name, doc, form, i, options, events)) {
Jun Fangdf7f3662015-11-10 18:29:18 +0800695 ++rendered_pages;
696 } else {
Lei Zhang5377ebf2015-09-23 14:52:53 -0700697 ++bad_pages;
Jun Fangaeacba42014-08-22 17:04:29 -0700698 }
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700699 }
700
701 FORM_DoDocumentAAction(form, FPDFDOC_AACTION_WC);
Lei Zhangba026912015-07-16 10:06:11 -0700702
Tom Sepezc46d0002015-11-30 15:46:36 -0800703#ifdef PDF_ENABLE_XFA
Lei Zhangba026912015-07-16 10:06:11 -0700704 // Note: The shut down order here is the reverse of the non-XFA branch order.
705 // Need to work out if this is required, and if it is, the lifetimes of
706 // objects owned by |doc| that |form| reference.
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700707 FPDF_CloseDocument(doc);
Bo Xu2b7a49d2014-11-14 17:40:50 -0800708 FPDFDOC_ExitFormFillEnvironment(form);
Tom Sepezc46d0002015-11-30 15:46:36 -0800709#else // PDF_ENABLE_XFA
710 FPDFDOC_ExitFormFillEnvironment(form);
711 FPDF_CloseDocument(doc);
712#endif // PDF_ENABLE_XFA
Lei Zhangba026912015-07-16 10:06:11 -0700713
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700714 FPDFAvail_Destroy(pdf_avail);
715
Tom Sepez1ed8a212015-05-11 15:25:39 -0700716 fprintf(stderr, "Rendered %d pages.\n", rendered_pages);
tsepez10b01bf2016-05-04 12:52:42 -0700717 if (bad_pages)
718 fprintf(stderr, "Skipped %d bad pages.\n", bad_pages);
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700719}
720
Tom Sepez2991d8d2016-01-15 16:02:48 -0800721static void ShowConfig() {
722 std::string config;
723 std::string maybe_comma;
724#if PDF_ENABLE_V8
725 config.append(maybe_comma);
726 config.append("V8");
727 maybe_comma = ",";
728#endif // PDF_ENABLE_V8
729#ifdef V8_USE_EXTERNAL_STARTUP_DATA
730 config.append(maybe_comma);
731 config.append("V8_EXTERNAL");
732 maybe_comma = ",";
733#endif // V8_USE_EXTERNAL_STARTUP_DATA
734#ifdef PDF_ENABLE_XFA
735 config.append(maybe_comma);
736 config.append("XFA");
737 maybe_comma = ",";
738#endif // PDF_ENABLE_XFA
739 printf("%s\n", config.c_str());
740}
741
Tom Sepez23b4e3f2015-02-06 16:05:23 -0800742static const char usage_string[] =
743 "Usage: pdfium_test [OPTION] [FILE]...\n"
Tom Sepez2991d8d2016-01-15 16:02:48 -0800744 " --show-config - print build options and exit\n"
tsepez10b01bf2016-05-04 12:52:42 -0700745 " --send-events - send input described by .evt file\n"
Lei Zhang6f62d532015-09-23 15:31:44 -0700746 " --bin-dir=<path> - override path to v8 external data\n"
747 " --font-dir=<path> - override path to external fonts\n"
748 " --scale=<number> - scale output size by number (e.g. 0.5)\n"
Tom Sepez23b4e3f2015-02-06 16:05:23 -0800749#ifdef _WIN32
750 " --bmp - write page images <pdf-name>.<page-number>.bmp\n"
751 " --emf - write page meta files <pdf-name>.<page-number>.emf\n"
Tom Sepezc46d0002015-11-30 15:46:36 -0800752#endif // _WIN32
Tom Sepez23b4e3f2015-02-06 16:05:23 -0800753 " --png - write page images <pdf-name>.<page-number>.png\n"
Cary Clark399be5b2016-03-14 16:51:29 -0400754#ifdef PDF_ENABLE_SKIA
755 " --skp - write page images <pdf-name>.<page-number>.skp\n"
756#endif
757 "";
Tom Sepez23b4e3f2015-02-06 16:05:23 -0800758
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700759int main(int argc, const char* argv[]) {
Tom Sepez5ee12d72014-12-17 16:24:01 -0800760 std::vector<std::string> args(argv, argv + argc);
761 Options options;
762 std::list<std::string> files;
763 if (!ParseCommandLine(args, &options, &files)) {
Tom Sepez23b4e3f2015-02-06 16:05:23 -0800764 fprintf(stderr, "%s", usage_string);
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700765 return 1;
766 }
767
Tom Sepez2991d8d2016-01-15 16:02:48 -0800768 if (options.show_config) {
769 ShowConfig();
770 return 0;
771 }
772
773 if (files.empty()) {
774 fprintf(stderr, "No input files.\n");
775 return 1;
776 }
777
Tom Sepez452b4f32015-10-13 09:27:27 -0700778#ifdef PDF_ENABLE_V8
Tom Sepezd831dc72015-10-19 16:04:22 -0700779 v8::Platform* platform;
Tom Sepez5ee12d72014-12-17 16:24:01 -0800780#ifdef V8_USE_EXTERNAL_STARTUP_DATA
781 v8::StartupData natives;
782 v8::StartupData snapshot;
Tom Sepezd831dc72015-10-19 16:04:22 -0700783 InitializeV8ForPDFium(options.exe_path, options.bin_directory, &natives,
784 &snapshot, &platform);
785#else // V8_USE_EXTERNAL_STARTUP_DATA
786 InitializeV8ForPDFium(&platform);
Tom Sepez5ee12d72014-12-17 16:24:01 -0800787#endif // V8_USE_EXTERNAL_STARTUP_DATA
Tom Sepez452b4f32015-10-13 09:27:27 -0700788#endif // PDF_ENABLE_V8
Tom Sepez5ee12d72014-12-17 16:24:01 -0800789
Tom Sepeza72e8e22015-10-07 10:17:53 -0700790 FPDF_LIBRARY_CONFIG config;
791 config.version = 2;
792 config.m_pUserFontPaths = nullptr;
793 config.m_pIsolate = nullptr;
794 config.m_v8EmbedderSlot = 0;
795
796 const char* path_array[2];
797 if (!options.font_directory.empty()) {
Lei Zhang6f62d532015-09-23 15:31:44 -0700798 path_array[0] = options.font_directory.c_str();
799 path_array[1] = nullptr;
Lei Zhang6f62d532015-09-23 15:31:44 -0700800 config.m_pUserFontPaths = path_array;
Lei Zhang6f62d532015-09-23 15:31:44 -0700801 }
Tom Sepeza72e8e22015-10-07 10:17:53 -0700802 FPDF_InitLibraryWithConfig(&config);
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700803
804 UNSUPPORT_INFO unsuppored_info;
805 memset(&unsuppored_info, '\0', sizeof(unsuppored_info));
806 unsuppored_info.version = 1;
Tom Sepezb7cb36a2015-02-13 16:54:48 -0800807 unsuppored_info.FSDK_UnSupport_Handler = ExampleUnsupportedHandler;
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700808
809 FSDK_SetUnSpObjProcessHandler(&unsuppored_info);
810
811 while (!files.empty()) {
Tom Sepez5ee12d72014-12-17 16:24:01 -0800812 std::string filename = files.front();
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700813 files.pop_front();
Tom Sepez5ee12d72014-12-17 16:24:01 -0800814 size_t file_length = 0;
Tom Sepez0aa35312016-01-06 10:16:32 -0800815 std::unique_ptr<char, pdfium::FreeDeleter> file_contents =
816 GetFileContents(filename.c_str(), &file_length);
tsepezf09bdfa2016-04-18 16:08:26 -0700817 if (!file_contents)
818 continue;
tsepez10b01bf2016-05-04 12:52:42 -0700819 fprintf(stderr, "Rendering PDF file %s.\n", filename.c_str());
tsepezf09bdfa2016-04-18 16:08:26 -0700820 std::string events;
821 if (options.send_events) {
822 std::string event_filename = filename;
823 size_t event_length = 0;
824 size_t extension_pos = event_filename.find(".pdf");
825 if (extension_pos != std::string::npos) {
826 event_filename.replace(extension_pos, 4, ".evt");
827 std::unique_ptr<char, pdfium::FreeDeleter> event_contents =
828 GetFileContents(event_filename.c_str(), &event_length);
829 if (event_contents) {
tsepez10b01bf2016-05-04 12:52:42 -0700830 fprintf(stderr, "Sending events from: %s\n", event_filename.c_str());
tsepezf09bdfa2016-04-18 16:08:26 -0700831 events = std::string(event_contents.get(), event_length);
tsepezf09bdfa2016-04-18 16:08:26 -0700832 }
833 }
834 }
835 RenderPdf(filename, file_contents.get(), file_length, options, events);
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700836 }
837
838 FPDF_DestroyLibrary();
Tom Sepez452b4f32015-10-13 09:27:27 -0700839#ifdef PDF_ENABLE_V8
John Abd-El-Malekb045ed22015-02-10 09:15:12 -0800840 v8::V8::ShutdownPlatform();
841 delete platform;
Tom Sepez452b4f32015-10-13 09:27:27 -0700842#endif // PDF_ENABLE_V8
John Abd-El-Malek3f3b45c2014-05-23 17:28:10 -0700843
844 return 0;
845}