| OLD | NEW | 
|---|
| (Empty) |  | 
|  | 1 // 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> | 
|  | 11 #include <string> | 
|  | 12 #include <utility> | 
|  | 13 | 
|  | 14 #include "fpdfsdk/include/fpdf_dataavail.h" | 
|  | 15 #include "fpdfsdk/include/fpdf_ext.h" | 
|  | 16 #include "fpdfsdk/include/fpdfformfill.h" | 
|  | 17 #include "fpdfsdk/include/fpdftext.h" | 
|  | 18 #include "fpdfsdk/include/fpdfview.h" | 
|  | 19 #include "v8/include/v8.h" | 
|  | 20 | 
|  | 21 #ifdef _WIN32 | 
|  | 22   #define snprintf _snprintf | 
|  | 23   /* in Windows, rb for open and read binary file */ | 
|  | 24   #define FOPEN_READ "rb" | 
|  | 25 #else | 
|  | 26   #define FOPEN_READ "r" | 
|  | 27 #endif | 
|  | 28 | 
|  | 29 static void WritePpm(const char* pdf_name, int num, | 
|  | 30                      const char* buffer, int stride, int width, int height) { | 
|  | 31   if (stride < 0 || width < 0 || height < 0) | 
|  | 32     return; | 
|  | 33   if (height > 0 && width > INT_MAX / height) | 
|  | 34     return; | 
|  | 35   int out_len = width * height; | 
|  | 36   if (out_len > INT_MAX / 3) | 
|  | 37     return; | 
|  | 38   out_len *= 3; | 
|  | 39 | 
|  | 40   char filename[256]; | 
|  | 41   snprintf(filename, sizeof(filename), "%s.%d.ppm", pdf_name, num); | 
|  | 42   FILE* fp = fopen(filename, "w"); | 
|  | 43   if (!fp) | 
|  | 44     return; | 
|  | 45   fprintf(fp, "P6\n# PDF test render\n%d %d\n255\n", width, height); | 
|  | 46   // Source data is B, G, R, unused. | 
|  | 47   // Dest data is R, G, B. | 
|  | 48   char* result = new char[out_len]; | 
|  | 49   if (result) { | 
|  | 50     for (int h = 0; h < height; ++h) { | 
|  | 51       const char* src_line = buffer + (stride * h); | 
|  | 52       char* dest_line = result + (width * h * 3); | 
|  | 53       for (int w = 0; w < width; ++w) { | 
|  | 54         // R | 
|  | 55         dest_line[w * 3] = src_line[(w * 4) + 2]; | 
|  | 56         // G | 
|  | 57         dest_line[(w * 3) + 1] = src_line[(w * 4) + 1]; | 
|  | 58         // B | 
|  | 59         dest_line[(w * 3) + 2] = src_line[w * 4]; | 
|  | 60       } | 
|  | 61     } | 
|  | 62     fwrite(result, out_len, 1, fp); | 
|  | 63     delete [] result; | 
|  | 64   } | 
|  | 65   fclose(fp); | 
|  | 66 } | 
|  | 67 | 
|  | 68 int Form_Alert(IPDF_JSPLATFORM*, FPDF_WIDESTRING, FPDF_WIDESTRING, int, int) { | 
|  | 69   printf("Form_Alert called.\n"); | 
|  | 70   return 0; | 
|  | 71 } | 
|  | 72 | 
|  | 73 void Unsupported_Handler(UNSUPPORT_INFO*, int type) { | 
|  | 74   std::string feature = "Unknown"; | 
|  | 75   switch (type) { | 
|  | 76     case FPDF_UNSP_DOC_XFAFORM: | 
|  | 77       feature = "XFA"; | 
|  | 78       break; | 
|  | 79     case FPDF_UNSP_DOC_PORTABLECOLLECTION: | 
|  | 80       feature = "Portfolios_Packages"; | 
|  | 81       break; | 
|  | 82     case FPDF_UNSP_DOC_ATTACHMENT: | 
|  | 83     case FPDF_UNSP_ANNOT_ATTACHMENT: | 
|  | 84       feature = "Attachment"; | 
|  | 85       break; | 
|  | 86     case FPDF_UNSP_DOC_SECURITY: | 
|  | 87       feature = "Rights_Management"; | 
|  | 88       break; | 
|  | 89     case FPDF_UNSP_DOC_SHAREDREVIEW: | 
|  | 90       feature = "Shared_Review"; | 
|  | 91       break; | 
|  | 92     case FPDF_UNSP_DOC_SHAREDFORM_ACROBAT: | 
|  | 93     case FPDF_UNSP_DOC_SHAREDFORM_FILESYSTEM: | 
|  | 94     case FPDF_UNSP_DOC_SHAREDFORM_EMAIL: | 
|  | 95       feature = "Shared_Form"; | 
|  | 96       break; | 
|  | 97     case FPDF_UNSP_ANNOT_3DANNOT: | 
|  | 98       feature = "3D"; | 
|  | 99       break; | 
|  | 100     case FPDF_UNSP_ANNOT_MOVIE: | 
|  | 101       feature = "Movie"; | 
|  | 102       break; | 
|  | 103     case FPDF_UNSP_ANNOT_SOUND: | 
|  | 104       feature = "Sound"; | 
|  | 105       break; | 
|  | 106     case FPDF_UNSP_ANNOT_SCREEN_MEDIA: | 
|  | 107     case FPDF_UNSP_ANNOT_SCREEN_RICHMEDIA: | 
|  | 108       feature = "Screen"; | 
|  | 109       break; | 
|  | 110     case FPDF_UNSP_ANNOT_SIG: | 
|  | 111       feature = "Digital_Signature"; | 
|  | 112       break; | 
|  | 113   } | 
|  | 114   printf("Unsupported feature: %s.\n", feature.c_str()); | 
|  | 115 } | 
|  | 116 | 
|  | 117 bool ParseCommandLine(int argc, const char* argv[], bool* write_images, | 
|  | 118                       std::list<const char*>* files) { | 
|  | 119   *write_images = false; | 
|  | 120   files->clear(); | 
|  | 121 | 
|  | 122   int cur_arg = 1; | 
|  | 123   if (cur_arg < argc && | 
|  | 124       strcmp(argv[cur_arg], "--write_images") == 0) { | 
|  | 125     *write_images = true; | 
|  | 126     cur_arg++; | 
|  | 127   } | 
|  | 128 | 
|  | 129   if (cur_arg >= argc) | 
|  | 130     return false; | 
|  | 131 | 
|  | 132   for (int i = cur_arg; i < argc; i++) | 
|  | 133     files->push_back(argv[i]); | 
|  | 134 | 
|  | 135   return true; | 
|  | 136 } | 
|  | 137 | 
|  | 138 class TestLoader { | 
|  | 139  public: | 
|  | 140   TestLoader(const char* pBuf, size_t len); | 
|  | 141 | 
|  | 142   const char* m_pBuf; | 
|  | 143   size_t m_Len; | 
|  | 144 }; | 
|  | 145 | 
|  | 146 TestLoader::TestLoader(const char* pBuf, size_t len) | 
|  | 147     : m_pBuf(pBuf), m_Len(len) { | 
|  | 148 } | 
|  | 149 | 
|  | 150 int Get_Block(void* param, unsigned long pos, unsigned char* pBuf, | 
|  | 151               unsigned long size) { | 
|  | 152   TestLoader* pLoader = (TestLoader*) param; | 
|  | 153   if (pos + size < pos || pos + size > pLoader->m_Len) return 0; | 
|  | 154   memcpy(pBuf, pLoader->m_pBuf + pos, size); | 
|  | 155   return 1; | 
|  | 156 } | 
|  | 157 | 
|  | 158 bool Is_Data_Avail(FX_FILEAVAIL* pThis, size_t offset, size_t size) { | 
|  | 159   return true; | 
|  | 160 } | 
|  | 161 | 
|  | 162 void Add_Segment(FX_DOWNLOADHINTS* pThis, size_t offset, size_t size) { | 
|  | 163 } | 
|  | 164 | 
|  | 165 void RenderPdf(const char* name, const char* pBuf, size_t len, | 
|  | 166                bool write_images) { | 
|  | 167   printf("Rendering PDF file %s.\n", name); | 
|  | 168 | 
|  | 169   IPDF_JSPLATFORM platform_callbacks; | 
|  | 170   memset(&platform_callbacks, '\0', sizeof(platform_callbacks)); | 
|  | 171   platform_callbacks.version = 1; | 
|  | 172   platform_callbacks.app_alert = Form_Alert; | 
|  | 173 | 
|  | 174   FPDF_FORMFILLINFO form_callbacks; | 
|  | 175   memset(&form_callbacks, '\0', sizeof(form_callbacks)); | 
|  | 176   form_callbacks.version = 1; | 
|  | 177   form_callbacks.m_pJsPlatform = &platform_callbacks; | 
|  | 178 | 
|  | 179   TestLoader loader(pBuf, len); | 
|  | 180 | 
|  | 181   FPDF_FILEACCESS file_access; | 
|  | 182   memset(&file_access, '\0', sizeof(file_access)); | 
|  | 183   file_access.m_FileLen = len; | 
|  | 184   file_access.m_GetBlock = Get_Block; | 
|  | 185   file_access.m_Param = &loader; | 
|  | 186 | 
|  | 187   FX_FILEAVAIL file_avail; | 
|  | 188   memset(&file_avail, '\0', sizeof(file_avail)); | 
|  | 189   file_avail.version = 1; | 
|  | 190   file_avail.IsDataAvail = Is_Data_Avail; | 
|  | 191 | 
|  | 192   FX_DOWNLOADHINTS hints; | 
|  | 193   memset(&hints, '\0', sizeof(hints)); | 
|  | 194   hints.version = 1; | 
|  | 195   hints.AddSegment = Add_Segment; | 
|  | 196 | 
|  | 197   FPDF_DOCUMENT doc; | 
|  | 198   FPDF_AVAIL pdf_avail = FPDFAvail_Create(&file_avail, &file_access); | 
|  | 199 | 
|  | 200   (void) FPDFAvail_IsDocAvail(pdf_avail, &hints); | 
|  | 201 | 
|  | 202   if (!FPDFAvail_IsLinearized(pdf_avail)) { | 
|  | 203     printf("Non-linearized path...\n"); | 
|  | 204     doc = FPDF_LoadCustomDocument(&file_access, NULL); | 
|  | 205   } else { | 
|  | 206     printf("Linearized path...\n"); | 
|  | 207     doc = FPDFAvail_GetDocument(pdf_avail, NULL); | 
|  | 208   } | 
|  | 209 | 
|  | 210   (void) FPDF_GetDocPermissions(doc); | 
|  | 211   (void) FPDFAvail_IsFormAvail(pdf_avail, &hints); | 
|  | 212 | 
|  | 213   FPDF_FORMHANDLE form = FPDFDOC_InitFormFillEnviroument(doc, &form_callbacks); | 
|  | 214   FPDF_SetFormFieldHighlightColor(form, 0, 0xFFE4DD); | 
|  | 215   FPDF_SetFormFieldHighlightAlpha(form, 100); | 
|  | 216 | 
|  | 217   int first_page = FPDFAvail_GetFirstPageNum(doc); | 
|  | 218   (void) FPDFAvail_IsPageAvail(pdf_avail, first_page, &hints); | 
|  | 219 | 
|  | 220   int page_count = FPDF_GetPageCount(doc); | 
|  | 221   for (int i = 0; i < page_count; ++i) { | 
|  | 222     (void) FPDFAvail_IsPageAvail(pdf_avail, i, &hints); | 
|  | 223   } | 
|  | 224 | 
|  | 225   FORM_DoDocumentJSAction(form); | 
|  | 226   FORM_DoDocumentOpenAction(form); | 
|  | 227 | 
|  | 228   for (int i = 0; i < page_count; ++i) { | 
|  | 229     FPDF_PAGE page = FPDF_LoadPage(doc, i); | 
|  | 230     FPDF_TEXTPAGE text_page = FPDFText_LoadPage(page); | 
|  | 231     FORM_OnAfterLoadPage(page, form); | 
|  | 232     FORM_DoPageAAction(page, form, FPDFPAGE_AACTION_OPEN); | 
|  | 233 | 
|  | 234     int width = static_cast<int>(FPDF_GetPageWidth(page)); | 
|  | 235     int height = static_cast<int>(FPDF_GetPageHeight(page)); | 
|  | 236     FPDF_BITMAP bitmap = FPDFBitmap_Create(width, height, 0); | 
|  | 237     FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 255, 255, 255, 255); | 
|  | 238 | 
|  | 239     FPDF_RenderPageBitmap(bitmap, page, 0, 0, width, height, 0, 0); | 
|  | 240     FPDF_FFLDraw(form, bitmap, page, 0, 0, width, height, 0, 0); | 
|  | 241     if (write_images) { | 
|  | 242       const char* buffer = reinterpret_cast<const char*>( | 
|  | 243           FPDFBitmap_GetBuffer(bitmap)); | 
|  | 244       int stride = FPDFBitmap_GetStride(bitmap); | 
|  | 245       WritePpm(name, i, buffer, stride, width, height); | 
|  | 246     } | 
|  | 247 | 
|  | 248     FPDFBitmap_Destroy(bitmap); | 
|  | 249 | 
|  | 250     FORM_DoPageAAction(page, form, FPDFPAGE_AACTION_CLOSE); | 
|  | 251     FORM_OnBeforeClosePage(page, form); | 
|  | 252     FPDFText_ClosePage(text_page); | 
|  | 253     FPDF_ClosePage(page); | 
|  | 254   } | 
|  | 255 | 
|  | 256   FORM_DoDocumentAAction(form, FPDFDOC_AACTION_WC); | 
|  | 257   FPDFDOC_ExitFormFillEnviroument(form); | 
|  | 258   FPDF_CloseDocument(doc); | 
|  | 259   FPDFAvail_Destroy(pdf_avail); | 
|  | 260 | 
|  | 261   printf("Loaded, parsed and rendered %d pages.\n", page_count); | 
|  | 262 } | 
|  | 263 | 
|  | 264 int main(int argc, const char* argv[]) { | 
|  | 265   v8::V8::InitializeICU(); | 
|  | 266   bool write_images = false; | 
|  | 267   std::list<const char*> files; | 
|  | 268   if (!ParseCommandLine(argc, argv, &write_images, &files)) { | 
|  | 269     printf("Usage is: test [--write_images] /path/to/pdf\n"); | 
|  | 270     printf("--write_images - to write page images <pdf-name>.<page-number>.ppm\n
     "); | 
|  | 271     return 1; | 
|  | 272   } | 
|  | 273 | 
|  | 274   FPDF_InitLibrary(NULL); | 
|  | 275 | 
|  | 276   UNSUPPORT_INFO unsuppored_info; | 
|  | 277   memset(&unsuppored_info, '\0', sizeof(unsuppored_info)); | 
|  | 278   unsuppored_info.version = 1; | 
|  | 279   unsuppored_info.FSDK_UnSupport_Handler = Unsupported_Handler; | 
|  | 280 | 
|  | 281   FSDK_SetUnSpObjProcessHandler(&unsuppored_info); | 
|  | 282 | 
|  | 283   while (!files.empty()) { | 
|  | 284     const char* filename = files.front(); | 
|  | 285     files.pop_front(); | 
|  | 286     FILE* file = fopen(filename, FOPEN_READ); | 
|  | 287     if (!file) { | 
|  | 288       fprintf(stderr, "Failed to open: %s\n", filename); | 
|  | 289       continue; | 
|  | 290     } | 
|  | 291     (void) fseek(file, 0, SEEK_END); | 
|  | 292     size_t len = ftell(file); | 
|  | 293     (void) fseek(file, 0, SEEK_SET); | 
|  | 294     char* pBuf = (char*) malloc(len); | 
|  | 295     size_t ret = fread(pBuf, 1, len, file); | 
|  | 296     (void) fclose(file); | 
|  | 297     if (ret != len) { | 
|  | 298       fprintf(stderr, "Failed to read: %s\n", filename); | 
|  | 299     } else { | 
|  | 300       RenderPdf(filename, pBuf, len, write_images); | 
|  | 301     } | 
|  | 302     free(pBuf); | 
|  | 303   } | 
|  | 304 | 
|  | 305   FPDF_DestroyLibrary(); | 
|  | 306 | 
|  | 307   return 0; | 
|  | 308 } | 
|  | 309 | 
| OLD | NEW | 
|---|