Zebediah Figura : msi: Return ERROR_INVALID_HANDLE when MsiDatabaseExport( ) is called from a custom action.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 3ff7d30520e4450fd9efbf5569eb28b82fb81aeb
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=3ff7d30520e4450fd9efbf5569eb28b82fb81aeb

Author: Zebediah Figura <z.figura12 at gmail.com>
Date:   Mon Mar 25 22:42:03 2019 -0500

msi: Return ERROR_INVALID_HANDLE when MsiDatabaseExport() is called from a custom action.

Signed-off-by: Zebediah Figura <z.figura12 at gmail.com>
Signed-off-by: Hans Leidekker <hans at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 dlls/msi/database.c     | 13 ++-----------
 dlls/msi/tests/custom.c |  2 +-
 2 files changed, 3 insertions(+), 12 deletions(-)

diff --git a/dlls/msi/database.c b/dlls/msi/database.c
index 1992661..d039903 100644
--- a/dlls/msi/database.c
+++ b/dlls/msi/database.c
@@ -1194,17 +1194,8 @@ UINT WINAPI MsiDatabaseExportW( MSIHANDLE handle, LPCWSTR szTable,
     TRACE("%x %s %s %s\n", handle, debugstr_w(szTable),
           debugstr_w(szFolder), debugstr_w(szFilename));
 
-    db = msihandle2msiinfo( handle, MSIHANDLETYPE_DATABASE );
-    if( !db )
-    {
-        MSIHANDLE remote_database = msi_get_remote(handle);
-        if ( !remote_database )
-            return ERROR_INVALID_HANDLE;
-
-        WARN("MsiDatabaseExport not allowed during a custom action!\n");
-
-        return ERROR_SUCCESS;
-    }
+    if (!(db = msihandle2msiinfo(handle, MSIHANDLETYPE_DATABASE)))
+        return ERROR_INVALID_HANDLE;
 
     r = MSI_DatabaseExport( db, szTable, szFolder, szFilename );
     msiobj_release( &db->hdr );
diff --git a/dlls/msi/tests/custom.c b/dlls/msi/tests/custom.c
index 526c4a2..d9d4689 100644
--- a/dlls/msi/tests/custom.c
+++ b/dlls/msi/tests/custom.c
@@ -1135,7 +1135,7 @@ static void test_invalid_functions(MSIHANDLE hinst)
 
     GetCurrentDirectoryA(sizeof(path), path);
     r = MsiDatabaseExportA(db, "Test", path, "bogus.idt");
-    todo_wine ok(hinst, r == ERROR_INVALID_HANDLE, "got %u\n", r);
+    ok(hinst, r == ERROR_INVALID_HANDLE, "got %u\n", r);
 
     r = MsiDatabaseImportA(db, path, "bogus.idt");
     todo_wine ok(hinst, r == ERROR_INVALID_HANDLE, "got %u\n", r);