notepad-plus-plus
diff --git a/‎src/Common.h‎
Lines changed: 38 additions & 1 deletion b/‎src/Common.h‎
Lines changed: 38 additions & 1 deletion
@@ -18,6 +18,43 @@
 
 #include <string>
 
+class ScopedCOMInit final // never use this in DllMain
+{
+public:
+	ScopedCOMInit() {
+		HRESULT hr = ::CoInitializeEx(nullptr, COINIT_APARTMENTTHREADED); // attempt STA init 1st (older CoInitialize(NULL))
+
+		if (hr == RPC_E_CHANGED_MODE)
+		{
+			hr = ::CoInitializeEx(nullptr, COINIT_MULTITHREADED); // STA init failed, switch to MTA
+		}
+
+		if (SUCCEEDED(hr))
+		{
+			// S_OK or S_FALSE, both needs subsequent CoUninitialize()
+			_bInitialized = true;
+		}
+	}
+
+	~ScopedCOMInit() {
+		if (_bInitialized)
+		{
+			_bInitialized = false;
+			::CoUninitialize();
+		}
+	}
+
+	bool isInitialized() const {
+		return _bInitialized;
+	}
+
+private:
+	bool _bInitialized = false;
+
+	ScopedCOMInit(const ScopedCOMInit&) = delete;
+	ScopedCOMInit& operator=(const ScopedCOMInit&) = delete;
+};
+
 void expandEnv(std::wstring& s);
 std::wstring getDateTimeStrFrom(const std::wstring& dateTimeFormat, const SYSTEMTIME& st);
 void writeLog(const wchar_t* logFileName, const wchar_t* logSuffix, const wchar_t* log2write);
@@ -27,4 +64,4 @@ std::string getFileContentA(const char* file2read);
 std::wstring GetLastErrorAsString(DWORD errorCode);
 std::wstring stringToUpper(std::wstring strToConvert);
 std::wstring stringToLower(std::wstring strToConvert);
-std::wstring stringReplace(std::wstring subject, const std::wstring& search, const std::wstring& replace);
+std::wstring stringReplace(std::wstring subject, const std::wstring& search, const std::wstring& replace);