1 files changed, 74 insertions, 0 deletions

diff --git a/src/test/msi/TestData/ComPlusExtensionTests/Components/TestComponentNative/TestComponentNative.cpp b/src/test/msi/TestData/ComPlusExtensionTests/Components/TestComponentNative/TestComponentNative.cpp
new file mode 100644
index 00000000..1a9fcb32
--- /dev/null
+++ b/src/test/msi/TestData/ComPlusExtensionTests/Components/TestComponentNative/TestComponentNative.cpp
@@ -0,0 +1,74 @@
+// TestComponentNative.cpp : Implementation of DLL Exports.
+
+
+#include "pch.h"
+#include "framework.h"
+#include "resource.h"
+#include "TestComponentNative_i.h"
+#include "dllmain.h"
+
+
+using namespace ATL;
+
+// Used to determine whether the DLL can be unloaded by OLE.
+_Use_decl_annotations_
+STDAPI DllCanUnloadNow(void)
+{
+        return _AtlModule.DllCanUnloadNow();
+}
+
+// Returns a class factory to create an object of the requested type.
+_Use_decl_annotations_
+STDAPI DllGetClassObject(_In_ REFCLSID rclsid, _In_ REFIID riid, _Outptr_ LPVOID* ppv)
+{
+        return _AtlModule.DllGetClassObject(rclsid, riid, ppv);
+}
+
+// DllRegisterServer - Adds entries to the system registry.
+_Use_decl_annotations_
+STDAPI DllRegisterServer(void)
+{
+        // registers object, typelib and all interfaces in typelib
+        HRESULT hr = _AtlModule.DllRegisterServer();
+        return hr;
+}
+
+// DllUnregisterServer - Removes entries from the system registry.
+_Use_decl_annotations_
+STDAPI DllUnregisterServer(void)
+{
+        HRESULT hr = _AtlModule.DllUnregisterServer();
+        return hr;
+}
+
+// DllInstall - Adds/Removes entries to the system registry per user per machine.
+STDAPI DllInstall(BOOL bInstall, _In_opt_  LPCWSTR pszCmdLine)
+{
+        HRESULT hr = E_FAIL;
+        static const wchar_t szUserSwitch[] = L"user";
+
+        if (pszCmdLine != nullptr)
+        {
+                if (_wcsnicmp(pszCmdLine, szUserSwitch, _countof(szUserSwitch)) == 0)
+                {
+                        ATL::AtlSetPerUserRegistration(true);
+                }
+        }
+
+        if (bInstall)
+        {
+                hr = DllRegisterServer();
+                if (FAILED(hr))
+                {
+                        DllUnregisterServer();
+                }
+        }
+        else
+        {
+                hr = DllUnregisterServer();
+        }
+
+        return hr;
+}
+
+