From 26aadb0b725ec2398b0161e57106928122ce01a5 Mon Sep 17 00:00:00 2001
From: Bill Zissimopoulos <billziss@navimatics.com>
Date: Sat, 30 Apr 2016 17:33:25 -0700
Subject: [PATCH] installer: use regsvr32 to register FSD/DLL (Wix cannot
 register file system drivers or network provider DLL's)

---
 build/VStudio/installer/Product.wxs | 54 +++++++++++++++++++++++++++++
 src/dll/library.c                   |  4 +++
 2 files changed, 58 insertions(+)

diff --git a/build/VStudio/installer/Product.wxs b/build/VStudio/installer/Product.wxs
index 362b1ea5..1186bc94 100644
--- a/build/VStudio/installer/Product.wxs
+++ b/build/VStudio/installer/Product.wxs
@@ -166,5 +166,59 @@
                 Value="WelcomeDlg"
                 Order="10">NOT Installed</Publish>
         </UI>
+        <!--
+            Use a bunch of custom actions to register our FSD/DLL.
+            Turns out that Wix cannot be used to register legacy drivers or
+            Network Provider DLL's.
+        -->
+        <CustomAction
+            Id="A.WinFsp.x64.Register"
+            Execute="deferred"
+            Impersonate="no"
+            Directory="BINDIR"
+            ExeCommand="&quot;[WindowsFolder]System32\regsvr32.exe&quot; winfsp-x64.dll"
+            Return="check" />
+        <CustomAction
+            Id="A.WinFsp.x64.Unregister"
+            Execute="deferred"
+            Impersonate="no"
+            Directory="BINDIR"
+            ExeCommand="&quot;[WindowsFolder]System32\regsvr32.exe&quot; /u winfsp-x64.dll"
+            Return="ignore" />
+        <CustomAction
+            Id="A.WinFsp.x86.Register"
+            Execute="deferred"
+            Impersonate="no"
+            Directory="BINDIR"
+            ExeCommand="&quot;[WindowsFolder]System32\regsvr32.exe&quot; winfsp-x86.dll"
+            Return="check" />
+        <CustomAction
+            Id="A.WinFsp.x86.Unregister"
+            Execute="deferred"
+            Impersonate="no"
+            Directory="BINDIR"
+            ExeCommand="&quot;[WindowsFolder]System32\regsvr32.exe&quot; /u winfsp-x86.dll"
+            Return="ignore" />
+        <InstallExecuteSequence>
+            <!--
+                We want to register our FSD/DLL in all cases except on Uninstall.
+                The Uninstall condition can be identified as follows:
+                    REMOVE ~= "ALL"
+
+                See http://stackoverflow.com/a/321874
+            -->
+            <Custom Action="A.WinFsp.x64.Register" After="InstallFiles">
+                VersionNT64 AND NOT (REMOVE ~= "ALL")
+            </Custom>
+            <Custom Action="A.WinFsp.x64.Unregister" Before="RemoveFiles">
+                VersionNT64 AND (REMOVE ~= "ALL")
+            </Custom>
+            <Custom Action="A.WinFsp.x86.Register" After="InstallFiles">
+                NOT VersionNT64 AND NOT (REMOVE ~= "ALL")
+            </Custom>
+            <Custom Action="A.WinFsp.x86.Unregister" Before="RemoveFiles">
+                NOT VersionNT64 AND (REMOVE ~= "ALL")
+            </Custom>
+        </InstallExecuteSequence>
     </Product>
 </Wix>
\ No newline at end of file
diff --git a/src/dll/library.c b/src/dll/library.c
index ae8a04a9..47fcab1e 100644
--- a/src/dll/library.c
+++ b/src/dll/library.c
@@ -53,6 +53,8 @@ HRESULT WINAPI DllRegisterServer(VOID)
 
     Result = FspNpRegister();
 
+    FspDebugLog("FspNpRegister = %ld\n", Result);
+
     return NT_SUCCESS(Result) ? S_OK : 0x80040201/*SELFREG_E_CLASS*/;
 }
 
@@ -62,5 +64,7 @@ HRESULT WINAPI DllUnregisterServer(VOID)
 
     Result = FspNpUnregister();
 
+    FspDebugLog("FspNpUnregister = %ld\n", Result);
+
     return NT_SUCCESS(Result) ? S_OK : 0x80040201/*SELFREG_E_CLASS*/;
 }