Merge XFA to PDFium master at 4dc95e7 on 10/28/2014

[pdfium.git] / fpdfsdk / src / fpdf_flatten.cpp

diff --git a/fpdfsdk/src/fpdf_flatten.cpp b/fpdfsdk/src/fpdf_flatten.cpp
index 6d3440f..e07ccc9 100644 (file)
--- a/fpdfsdk/src/fpdf_flatten.cpp
+++ b/fpdfsdk/src/fpdf_flatten.cpp
@@ -6,6 +6,8 @@
 
 #include "../include/fsdk_define.h"
 #include "../include/fpdf_flatten.h"
+#include "../include/fpdfxfa/fpdfxfa_doc.h"
+#include "../include/fpdfxfa/fpdfxfa_page.h"
 
 typedef CFX_ArrayTemplate<CPDF_Dictionary*> CPDF_ObjectArray;
 typedef CFX_ArrayTemplate<CPDF_Rect> CPDF_RectArray;
@@ -332,7 +334,10 @@ DLLEXPORT int STDCALL FPDFPage_Flatten( FPDF_PAGE page, int nFlag)
                return FLATTEN_FAIL;
        }
 
-       CPDF_Page * pPage = (CPDF_Page*)( page );
+       CPDF_Page * pPage = ((CPDFXFA_Page*)( page ))->GetPDFPage();
+       if (!pPage)
+               return FLATTEN_FAIL;
+
        CPDF_Document * pDocument = pPage->m_pDocument;
        CPDF_Dictionary * pPageDict = pPage->m_pFormDict;