From a939bfe3e102bfb28b4e8a5d951333d16badf80b Mon Sep 17 00:00:00 2001 From: dsinclair Date: Thu, 22 Sep 2016 13:18:45 -0700 Subject: Make creation of CPDFSDK_Document clearer Move the creation of the CPDFSDK_Document into FPDFDOC_InitFormFillEnvironment instead of hidden inside a Get method in CDPFXFA_Document. Review-Url: https://codereview.chromium.org/2353303004 --- fpdfsdk/fpdfformfill.cpp | 24 +++++++++++++++++------- 1 file changed, 17 insertions(+), 7 deletions(-) (limited to 'fpdfsdk/fpdfformfill.cpp') diff --git a/fpdfsdk/fpdfformfill.cpp b/fpdfsdk/fpdfformfill.cpp index 7e4ce36f5f..9baabc4de4 100644 --- a/fpdfsdk/fpdfformfill.cpp +++ b/fpdfsdk/fpdfformfill.cpp @@ -242,14 +242,22 @@ FPDFDOC_InitFormFillEnvironment(FPDF_DOCUMENT document, if (!pDocument) return nullptr; +#ifdef PDF_ENABLE_XFA + // If the CPDFXFA_Document has a SDKDocument already then we've done this + // and can just return the old Env. Otherwise, we'll end up setting a new + // SDKDocument into the XFADocument and, that could get weird. + if (pDocument->GetSDKDoc()) + return pDocument->GetSDKDoc()->GetEnv(); +#endif + CPDFSDK_Environment* pEnv = new CPDFSDK_Environment(pDocument, formInfo); + #ifdef PDF_ENABLE_XFA - pEnv->SetSDKDocument(pDocument->GetSDKDocument(pEnv)); - CPDFXFA_App* pApp = CPDFXFA_App::GetInstance(); - pApp->AddFormFillEnv(pEnv); -#else // PDF_ENABLE_XFA - pEnv->SetSDKDocument(new CPDFSDK_Document(pDocument, pEnv)); + // Ownership of the SDKDocument is passed to the CPDFXFA_Document. + pDocument->SetSDKDoc(WrapUnique(pEnv->GetSDKDocument())); + CPDFXFA_App::GetInstance()->AddFormFillEnv(pEnv); #endif // PDF_ENABLE_XFA + return pEnv; } @@ -257,16 +265,18 @@ DLLEXPORT void STDCALL FPDFDOC_ExitFormFillEnvironment(FPDF_FORMHANDLE hHandle) { if (!hHandle) return; + CPDFSDK_Environment* pEnv = HandleToCPDFSDKEnvironment(hHandle); + #ifdef PDF_ENABLE_XFA - CPDFXFA_App* pApp = CPDFXFA_App::GetInstance(); - pApp->RemoveFormFillEnv(pEnv); + CPDFXFA_App::GetInstance()->RemoveFormFillEnv(pEnv); #else // PDF_ENABLE_XFA if (CPDFSDK_Document* pSDKDoc = pEnv->GetSDKDocument()) { pEnv->SetSDKDocument(nullptr); delete pSDKDoc; } #endif // PDF_ENABLE_XFA + delete pEnv; } -- cgit v1.2.3