Remove ownership of CPDFSDK_Document from CPDFXFA_Document

fpdfsdk/fpdfformfill.cpp

 // Copyright 2014 PDFium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
 
 #include "public/fpdf_formfill.h"
 
 #include <memory>
 #include <vector>
@@ skipping 236 matching lines @@
   // 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
-  // Ownership of the SDKDocument is passed to the CPDFXFA_Document.
-  pDocument->SetSDKDoc(pdfium::WrapUnique(pEnv->GetSDKDocument()));
+  pDocument->SetSDKDoc(pEnv->GetSDKDocument());
   CPDFXFA_App::GetInstance()->AddFormFillEnv(pEnv);
 #endif  // PDF_ENABLE_XFA
 
   return pEnv;
 }
 
 DLLEXPORT void STDCALL
 FPDFDOC_ExitFormFillEnvironment(FPDF_FORMHANDLE hHandle) {
   if (!hHandle)
     return;
 
   CPDFSDK_Environment* pEnv = HandleToCPDFSDKEnvironment(hHandle);
 
 #ifdef PDF_ENABLE_XFA
   CPDFXFA_App::GetInstance()->RemoveFormFillEnv(pEnv);
-#else   // PDF_ENABLE_XFA
+
+  // Reset the focused annotations and remove the SDK document from the
+  // XFA document.
   if (CPDFSDK_Document* pSDKDoc = pEnv->GetSDKDocument()) {
-    pEnv->SetSDKDocument(nullptr);
-    delete pSDKDoc;
+    pSDKDoc->ClearAllFocusedAnnots();
+    pSDKDoc->GetXFADocument()->SetSDKDoc(nullptr);
   }
 #endif  // PDF_ENABLE_XFA
 
   delete pEnv;
 }
 
 DLLEXPORT FPDF_BOOL STDCALL FORM_OnMouseMove(FPDF_FORMHANDLE hHandle,
                                              FPDF_PAGE page,
                                              int modifier,
                                              double page_x,
@@ skipping 445 matching lines @@
       CPDF_Action action = aa.GetAction(CPDF_AAction::OpenPage);
       pActionHandler->DoAction_Page(action, CPDF_AAction::OpenPage, pSDKDoc);
     }
   } else {
     if (aa.ActionExist(CPDF_AAction::ClosePage)) {
       CPDF_Action action = aa.GetAction(CPDF_AAction::ClosePage);
       pActionHandler->DoAction_Page(action, CPDF_AAction::ClosePage, pSDKDoc);
     }
   }
 }