hide user profile avatar if obs is running

2020-07-12 22:44:33 +02:00
parent 515657be19
commit 9739d8d9c3
4 changed files with 69 additions and 1 deletions
@@ -0,0 +1,54 @@
+#include "StreamerMode.hpp"
+
+#ifdef USEWINSDK
+#    include <Windows.h>
+
+#    include <VersionHelpers.h>
+#    include <WtsApi32.h>
+#    pragma comment(lib, "Wtsapi32.lib")
+#endif
+
+namespace chatterino {
+
+const QStringList &broadcastingBinaries()
+{
+#ifdef USEWINSDK
+    static QStringList bins = {"obs.exe", "obs64.exe"};
+#else
+    static QStringList bins = {};
+#endif
+    return bins;
+}
+
+bool isInStreamerMode()
+{
+#ifdef USEWINSDK
+    if (IsWindowsVistaOrGreater())
+    {
+        WTS_PROCESS_INFO *pWPIs = nullptr;
+        DWORD dwProcCount = 0;
+
+        if (WTSEnumerateProcesses(WTS_CURRENT_SERVER_HANDLE, NULL, 1, &pWPIs,
+                                  &dwProcCount))
+        {
+            //Go through all processes retrieved
+            for (DWORD i = 0; i < dwProcCount; i++)
+            {
+                QString processName = QString::fromUtf16(
+                    reinterpret_cast<char16_t *>(pWPIs[i].pProcessName));
+                qDebug() << processName;
+
+                if (broadcastingBinaries().contains(processName))
+                    return true;
+            }
+        }
+
+        if (pWPIs)
+            WTSFreeMemory(pWPIs);
+    }
+#endif
+
+    return false;
+}
+
+}  // namespace chatterino