Alexandre Julliard : setupapi: Run registration scripts of fake dlls as they are copied.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 5ec4e8f192a3642d9b982f24392e303ef1e63957
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=5ec4e8f192a3642d9b982f2439...

Author: Alexandre Julliard 
Date:   Tue Aug  2 13:47:43 2011 +0200

setupapi: Run registration scripts of fake dlls as they are copied.

---

dlls/setupapi/fakedll.c |   74 +++++++++++++++++++++++++++++++++++++++++++++-
 tools/wine.inf.in       |   63 +++++++++-------------------------------
 2 files changed, 86 insertions(+), 51 deletions(-)

diff --git a/dlls/setupapi/fakedll.c b/dlls/setupapi/fakedll.c
index 27d4b8a..0c39935 100644
--- a/dlls/setupapi/fakedll.c
+++ b/dlls/setupapi/fakedll.c
@@ -33,6 +33,8 @@
 # include <unistd.h>
 #endif

+#define COBJMACROS
+#define ATL_INITGUID
 #define NONAMELESSSTRUCT
 #define NONAMELESSUNION
 #include "ntstatus.h"
@@ -45,6 +47,8 @@
 #include "wine/unicode.h"
 #include "wine/library.h"
 #include "wine/debug.h"
+#include "ole2.h"
+#include "atliface.h"

WINE_DEFAULT_DEBUG_CHANNEL(setupapi);

@@ -59,6 +63,7 @@ static SIZE_T file_buffer_size;
 static unsigned int handled_count;
 static unsigned int handled_total;
 static char **handled_dlls;
+static IRegistrar *registrar;

struct dll_info
 {
@@ -465,6 +470,67 @@ static HANDLE create_dest_file( const WCHAR *name )
     return h;
 }

+static BOOL CALLBACK register_resource( HMODULE module, LPCWSTR type, LPWSTR name, LONG_PTR arg )
+{
+    HRESULT *hr = (HRESULT *)arg;
+    WCHAR *buffer;
+    HRSRC rsrc = FindResourceW( module, name, type );
+    char *str = LoadResource( module, rsrc );
+    DWORD lenW, lenA = SizeofResource( module, rsrc );
+
+    if (!str) return FALSE;
+    lenW = MultiByteToWideChar( CP_UTF8, 0, str, lenA, NULL, 0 ) + 1;
+    if (!(buffer = HeapAlloc( GetProcessHeap(), 0, lenW * sizeof(WCHAR) ))) return FALSE;
+    MultiByteToWideChar( CP_UTF8, 0, str, lenA, buffer, lenW );
+    buffer[lenW - 1] = 0;
+    *hr = IRegistrar_StringRegister( registrar, buffer );
+    HeapFree( GetProcessHeap(), 0, buffer );
+    return TRUE;
+}
+
+static void register_fake_dll( const WCHAR *name, const void *data, size_t size )
+{
+    static const WCHAR atlW[] = {'a','t','l','.','d','l','l',0};
+    static const WCHAR moduleW[] = {'M','O','D','U','L','E',0};
+    static const WCHAR regtypeW[] = {'W','I','N','E','_','R','E','G','I','S','T','R','Y',0};
+    const IMAGE_RESOURCE_DIRECTORY *resdir;
+    LDR_RESOURCE_INFO info;
+    HRESULT hr = S_OK;
+    HMODULE module = (HMODULE)((ULONG_PTR)data | 1);
+
+    info.Type = (ULONG_PTR)regtypeW;
+    if (LdrFindResourceDirectory_U( module, &info, 1, &resdir )) return;
+
+    if (!registrar)
+    {
+        /* create the object by hand since we can't guarantee that atl and ole32 are registered */
+        IClassFactory *cf;
+        HRESULT (WINAPI *pDllGetClassObject)( REFCLSID clsid, REFIID iid, LPVOID *ppv );
+        HMODULE atl = LoadLibraryW( atlW );
+
+        if ((pDllGetClassObject = (void *)GetProcAddress( atl, "DllGetClassObject" )))
+        {
+            hr = pDllGetClassObject( &CLSID_Registrar, &IID_IClassFactory, (void **)&cf );
+            if (SUCCEEDED( hr ))
+            {
+                hr = IClassFactory_CreateInstance( cf, NULL, &IID_IRegistrar, (void **)&registrar );
+                IClassFactory_Release( cf );
+            }
+        }
+        if (!registrar)
+        {
+            ERR( "failed to create IRegistrar: %x\n", hr );
+            return;
+        }
+    }
+
+    TRACE( "registering %s\n", debugstr_w(name) );
+    IRegistrar_ClearReplacements( registrar );
+    IRegistrar_AddReplacement( registrar, moduleW, name );
+    EnumResourceNamesW( module, regtypeW, register_resource, (LONG_PTR)&hr );
+    if (FAILED(hr)) ERR( "failed to register %s: %x\n", debugstr_w(name), hr );
+}
+
 /* copy a fake dll file to the dest directory */
 static void install_fake_dll( WCHAR *dest, char *file, const char *ext )
 {
@@ -494,7 +560,8 @@ static void install_fake_dll( WCHAR *dest, char *file, const char *ext )
             ret = (WriteFile( h, data, size, &written, NULL ) && written == size);
             if (!ret) ERR( "failed to write to %s (error=%u)\n", debugstr_w(dest), GetLastError() );
             CloseHandle( h );
-            if (!ret) DeleteFileW( dest );
+            if (ret) register_fake_dll( dest, data, size );
+            else DeleteFileW( dest );
         }
     }
     *destname = 0;  /* restore it for next file */
@@ -608,7 +675,8 @@ BOOL create_fake_dll( const WCHAR *name, const WCHAR *source )
         DWORD written;

ret = (WriteFile( h, buffer, size, &written, NULL ) && written == size);
-        if (!ret) ERR( "failed to write to %s (error=%u)\n", debugstr_w(name), GetLastError() );
+        if (ret) register_fake_dll( name, buffer, size );
+        else ERR( "failed to write to %s (error=%u)\n", debugstr_w(name), GetLastError() );
     }
     else
     {
@@ -632,4 +700,6 @@ void cleanup_fake_dlls(void)
     HeapFree( GetProcessHeap(), 0, handled_dlls );
     handled_dlls = NULL;
     handled_count = handled_total = 0;
+    if (registrar) IRegistrar_Release( registrar );
+    registrar = NULL;
 }
diff --git a/tools/wine.inf.in b/tools/wine.inf.in
index 47c5824..cfefbe4 100644
--- a/tools/wine.inf.in
+++ b/tools/wine.inf.in
@@ -2489,85 +2489,36 @@ HKLM,%CurrentVersion%\Telephony\Country List\998,"SameAreaRule",,"G"

[RegisterDllsSection]
 ;;some dlls have to be registered first
-11,,atl.dll,1
-11,,ole32.dll,1
 11,,oleaut32.dll,1
 11,,shell32.dll,1

-11,,actxprxy.dll,1
-11,,amstream.dll,1
-11,,avifil32.dll,1
-11,,browseui.dll,1
 11,,comctl32.dll,2
-11,,comdlg32.dll,1
 11,,cryptdlg.dll,1
 11,,cryptnet.dll,1
-11,,d3dxof.dll,1
-11,,ddraw.dll,1
-11,,ddrawex.dll,1
 11,,devenum.dll,1
-11,,dinput.dll,1
-11,,dinput8.dll,1
-11,,dispex.dll,1
-11,,dmband.dll,1
-11,,dmcompos.dll,1
-11,,dmime.dll,1
-11,,dmloader.dll,1
-11,,dmscript.dll,1
-11,,dmstyle.dll,1
-11,,dmsynth.dll,1
-11,,dmusic.dll,1
-11,,dplayx.dll,1
-11,,dpnet.dll,1
-11,,dsound.dll,1
-11,,dswave.dll,1
-11,,dxdiagn.dll,1
-11,,explorerframe.dll,1
 11,,gameux.dll,1
 11,,hhctrl.ocx,1
-11,,hlink.dll,1
-11,,hnetcfg.dll,1
 11,,ieframe.dll,1
-11,,inetcomm.dll,1
-11,,infosoft.dll,1
-11,,inseng.dll,1
-11,,itss.dll,1
 11,,jscript.dll,1
-11,,mlang.dll,1
-11,,mmdevapi.dll,1
-11,,msctf.dll,1
-11,,msdaps.dll,1
 11,,mshtml.dll,1
 11,,msi.dll,1
 11,,msiexec.exe,1
-11,,msimtf.dll,1
 11,,msisip.dll,1
-11,,mstask.dll,1
 11,,msxml.dll,1
 11,,msxml2.dll,1
 11,,msxml3.dll,1
 11,,msxml4.dll,1
 11,,msxml6.dll,1
-11,,objsel.dll,1
-11,,oledb32.dll,1
 11,,qcap.dll,1
-11,,qedit.dll,1
 11,,qmgr.dll,1
-11,,qmgrprxy.dll,1
 11,,quartz.dll,1
-11,,rsaenh.dll,1
-11,,shdocvw.dll,1
-11,,sti.dll,1
 11,,urlmon.dll,1
-11,,vbscript.dll,1
-11,,wbemprox.dll,1
 11,,wiaservc.dll,1
 11,,windowscodecs.dll,1
 11,,winegstreamer.dll,1
 11,,wineqtdecoder.dll,1
 11,,winhttp.dll,1
 11,,wintrust.dll,1
-11,,wmiutils.dll,1
 11,,wuapi.dll,1
 11,,iexplore.exe,1

@@ -2595,6 +2546,13 @@ HKLM,%CurrentVersion%\Telephony\Country List\998,"SameAreaRule",,"G"
 11,,iexplore.exe,-
 11,,winetest.exe,-
 12,,mountmgr.sys,-
+; registration order matters for these
+11,,msxml.dll
+11,,msxml2.dll
+11,,msxml3.dll
+11,,msxml4.dll
+11,,msxml6.dll
+11,,shdocvw.dll
 11,,*

[FakeDlls]
@@ -2614,6 +2572,13 @@ HKLM,%CurrentVersion%\Telephony\Country List\998,"SameAreaRule",,"G"
 11,,notepad.exe
 11,,winetest.exe,-
 12,,mountmgr.sys
+; registration order matters for these
+11,,msxml.dll
+11,,msxml2.dll
+11,,msxml3.dll
+11,,msxml4.dll
+11,,msxml6.dll
+11,,shdocvw.dll
 10,Microsoft.NET\Framework\v1.1.4322,aspnet_regiis.exe
 10,Microsoft.NET\Framework\v1.1.4322,ngen.exe
 10,Microsoft.NET\Framework\v1.1.4322,fusion.dll