Changeset 36455 in vbox for trunk/src/VBox/Additions

Timestamp:

Mar 29, 2011 11:37:01 AM (14 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

70837

Message:

Windows Guest Additions installer: Try to replace WFP protected files on the fly.

Location:

trunk/src/VBox/Additions/WINNT/Installer

Files:

• InstallHelper/VBoxGuestInstallHelper.cpp (modified) (3 diffs)
• VBoxGuestAdditions.nsi (modified) (3 diffs)

trunk/src/VBox/Additions/WINNT/Installer/InstallHelper/VBoxGuestInstallHelper.cpp

@@ -5 +5 @@
 
 /*
- * Copyright (C) 2010 Oracle Corporation
+ * Copyright (C) 2011 Oracle Corporation
  *
  * This file is part of VirtualBox Open Source Edition (OSE), as
@@ -26 +26 @@
 HINSTANCE g_hInstance;
 HWND g_hwndParent;
+
+typedef DWORD (__stdcall *fnSfcFileException) (DWORD param1, PWCHAR param2, DWORD param3);
+fnSfcFileException g_pfnSfcFileException = NULL;
 
 #define VBOXINSTALLHELPER_EXPORT extern "C" void __declspec(dllexport)
@@ -181 +184 @@
 }
 
+/**
+ * Disables the Windows File Protection for a specified file
+ * using an undocumented SFC API call. Don't try this at home!
+ *
+ * @param   hwndParent          Window handle of parent.
+ * @param   string_size         Size of variable string.
+ * @param   variables           The actual variable string.
+ * @param   stacktop            Pointer to a pointer to the current stack.
+ */
+VBOXINSTALLHELPER_EXPORT DisableWFP(HWND hwndParent, int string_size,
+                                    TCHAR *variables, stack_t **stacktop)
+{
+    TCHAR szFile[MAX_PATH + 1];
+    HRESULT hr = VBoxPopString(szFile, sizeof(szFile) / sizeof(TCHAR));
+    if (SUCCEEDED(hr))
+    {
+        HMODULE hSFC = LoadLibrary("sfc_os.dll");
+        if (NULL != hSFC)
+        {
+            g_pfnSfcFileException = (fnSfcFileException)GetProcAddress(hSFC, "SfcFileException");
+            if (g_pfnSfcFileException == NULL)
+            {
+                /* If we didn't get the proc address with the call above, try it harder with
+                 * the (zero based) index of the function list. */
+                g_pfnSfcFileException = (fnSfcFileException)GetProcAddress(hSFC, (LPCSTR)5);
+                if (g_pfnSfcFileException == NULL)
+                    hr = HRESULT_FROM_WIN32(ERROR_PROC_NOT_FOUND);
+            }
+        }
+        else
+            hr = HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
+
+        if (SUCCEEDED(hr))
+        {
+            WCHAR wszFile[MAX_PATH + 1];
+            hr = StringCchPrintfW(wszFile, sizeof(wszFile), L"%s", szFile);
+            if (SUCCEEDED(hr))
+                hr = HRESULT_FROM_WIN32(g_pfnSfcFileException(0, wszFile, -1));
+        }
+
+        if (hSFC)
+            FreeLibrary(hSFC);
+    }
+
+    /* Push simple return value on stack. */
+    SUCCEEDED(hr) ? pushstring("0") : pushstring("1");
+}
+

trunk/src/VBox/Additions/WINNT/Installer/VBoxGuestAdditions.nsi

@@ -27 +27 @@
 
 ; Defines for special functions
-!define WHQL_FAKE    ; Turns on the faking of non WHQL signed / approved drivers
-                     ; Needs the VBoxWHQLFake.exe in the additions output directory!
+!define WHQL_FAKE                   ; Enables faking of non WHQL signed / approved drivers
+                                    ; Needs the VBoxWHQLFake.exe in the additions output directory!
+!define WFP_FILE_EXCEPTION          ; Enables setting a temporary file exception for WFP proctected files
 
 !define VENDOR_ROOT_KEY             "SOFTWARE\$%VBOX_VENDOR_SHORT%"
 
+; Product defines
 !define PRODUCT_NAME                "$%VBOX_PRODUCT% Guest Additions"
 !define PRODUCT_DESC                "$%VBOX_PRODUCT% Guest Additions"
@@ -730 +732 @@
 
   Pop $0
+
+!if $%VBOX_WITH_GUEST_INSTALL_HELPER% == "1"
+  !ifdef WFP_FILE_EXCEPTION
+    VBoxGuestInstallHelper::DisableWFP "$0"
+    Pop $1 ; Get return value (ignored for now)
+    DetailPrint "Setting WFP exception for '$0': $1"
+  !endif
+!endif
+
   IfFileExists "$g_strSystemDir\takeown.exe" 0 +2
     nsExec::ExecToLog '"$g_strSystemDir\takeown.exe" /F "$0"'
@@ -871 +882 @@
 
 done:
+
+!ifndef WFP_FILE_EXCEPTION
   MessageBox MB_ICONINFORMATION|MB_OK $(VBOX_WFP_WARN_REPLACE) /SD IDOK
+!endif
   Goto exit