From d19f1bcd9a64f4eaa269b276ee5a71615842b2e7 Mon Sep 17 00:00:00 2001
From: Igor Sikachyna <sikachynairl@gmail.com>
Date: Thu, 3 Jun 2021 13:04:45 +0300
Subject: [PATCH] SYNERGY-1017 - Added a basic IO registry search for secure
 input presence

---
 src/lib/platform/OSXScreen.h  |  1 +
 src/lib/platform/OSXScreen.mm | 73 +++++++++++++++++++++++++++++++++++
 2 files changed, 74 insertions(+)

diff --git a/src/lib/platform/OSXScreen.h b/src/lib/platform/OSXScreen.h
index fcc0e20f8..e9bd90b8d 100644
--- a/src/lib/platform/OSXScreen.h
+++ b/src/lib/platform/OSXScreen.h
@@ -32,6 +32,7 @@
 #include <mach/mach_init.h>
 #include <IOKit/pwr_mgt/IOPMLib.h>
 #include <IOKit/IOMessage.h>
+#include <IOKit/IOKitLib.h>
 
 extern "C" {
     typedef int CGSConnectionID;
diff --git a/src/lib/platform/OSXScreen.mm b/src/lib/platform/OSXScreen.mm
index bd3a69b06..ddeaf4416 100755
--- a/src/lib/platform/OSXScreen.mm
+++ b/src/lib/platform/OSXScreen.mm
@@ -845,9 +845,82 @@ OSXScreen::disable()
 	m_isOnScreen = m_isPrimary;
 }
 
+std::string  CFStringRefToStdString(const CFStringRef pCFStringRef)
+{
+    const char* pCStr = NULL;
+
+    std::string sRet;
+
+    if (pCFStringRef)
+        {
+        pCStr = CFStringGetCStringPtr(pCFStringRef, kCFStringEncodingMacRoman);
+        if(pCStr)
+            sRet.assign(pCStr);
+        }
+    return sRet;
+}
+
+bool isSecureInput() {
+    NXEventHandle 		handle = MACH_PORT_NULL;
+    io_service_t		service = MACH_PORT_NULL, service_p = MACH_PORT_NULL;
+    mach_port_t			masterPort;
+    io_name_t n;
+    char buf[4096];
+    uint32_t s = sizeof(buf);
+
+	kern_return_t kr = IOMasterPort( MACH_PORT_NULL, &masterPort );
+	if(kr != KERN_SUCCESS) return false;
+
+	service = IORegistryEntryFromPath( masterPort, kIOServicePlane ":/" );
+	IORegistryEntryGetParentEntry(service, kIOServicePlane, &service_p);
+	CFTypeRef consoleUsers = IORegistryEntrySearchCFProperty(service_p, kIOServicePlane, CFSTR("IOConsoleUsers"), NULL, kIORegistryIterateParents | kIORegistryIterateRecursively);
+	if(!consoleUsers) return false;
+
+	CFTypeID type = CFGetTypeID(consoleUsers);
+	if(type != CFArrayGetTypeID())
+	{
+		CFRelease(consoleUsers);
+		return false;
+	}
+	CFTypeRef dict = CFArrayGetValueAtIndex((CFArrayRef)consoleUsers, 0);
+	if(!dict)
+	{
+		CFRelease(consoleUsers);
+		return false;
+	}
+	type = CFGetTypeID(dict);
+	if(type != CFDictionaryGetTypeID())
+	{
+		CFRelease(consoleUsers);
+		return false;
+	}
+
+	CFIndex count = CFDictionaryGetCount((CFDictionaryRef)dict);
+	const void * keys [count];
+	const void * values [count];
+	CFDictionaryGetKeysAndValues((CFDictionaryRef)dict, reinterpret_cast<const void **>(keys), reinterpret_cast<const void **>(values));
+	for(CFIndex i = 0; i < count; ++i) {
+		CFIndex keySize = CFStringGetMaximumSizeForEncoding(CFStringGetLength(reinterpret_cast<CFStringRef>(keys[i])), kCFStringEncodingASCII);
+		char key [keySize + 1];
+
+		if(!CFStringGetCString(reinterpret_cast<CFStringRef>(keys[i]), key, keySize + 1, kCFStringEncodingASCII)) {
+			continue;
+		}
+		LOG((CLOG_INFO "registry %s", key));
+	}
+
+
+	CFIndex secureInputCount = CFDictionaryGetCountOfKey((CFDictionaryRef)dict, CFSTR("kCGSSessionSecureInputPID"));
+	CFRelease(consoleUsers);
+	return secureInputCount > 0;
+}
+
 void
 OSXScreen::enter()
 {
+
+	bool secure = isSecureInput();
+
 	showCursor();
 
 	if (m_isPrimary) {