Changeset 86300 in vbox for trunk/src/VBox/Runtime/r3/darwin/krnlmod-darwin.cpp

Timestamp:

Sep 26, 2020 9:59:44 AM (4 years ago)

Author:

vboxsync

Message:

Runtime: Add API to load kernel modules by name and add simple implementation for macOS, bugref:9836

File:

• trunk/src/VBox/Runtime/r3/darwin/krnlmod-darwin.cpp (modified) (5 diffs)

trunk/src/VBox/Runtime/r3/darwin/krnlmod-darwin.cpp

@@ -43 +43 @@
 #include <CoreFoundation/CoreFoundation.h>
 #include <IOKit/IOKitLib.h>
+#include <libkern/OSReturn.h>
 
 
@@ -49 +50 @@
 /** OSKextCopyLoadedKextInfo in IOKit. */
 typedef CFDictionaryRef (* PFNOSKEXTCOPYLOADEDKEXTINFO)(CFArrayRef, CFArrayRef);
+/** KextManagerLoadKextWithURL in IOKit. */
+typedef OSReturn (* PFNKEXTMANAGERLOADKEXTWITHURL)(CFURLRef, CFArrayRef);
+/** KextManagerLoadKextWithIdentifier in IOKit */
+typedef OSReturn (* PFNKEXTMANAGERLOADKEXTWITHIDENTIFIER)(CFStringRef, CFArrayRef);
+/** KextManagerUnloadKextWithIdentifier in IOKit */
+typedef OSReturn (* PFNKEXTMANAGERUNLOADKEXTWITHIDENTIFIER)(CFStringRef);
 
 #ifndef kOSBundleRetainCountKey
@@ -84 +91 @@
 *   Global Variables                                                                                                             *
 *********************************************************************************************************************************/
-static RTONCE                       g_GetIoKitApisOnce = RTONCE_INITIALIZER;
-static PFNOSKEXTCOPYLOADEDKEXTINFO  g_pfnOSKextCopyLoadedKextInfo = NULL;
+static RTONCE                                 g_GetIoKitApisOnce = RTONCE_INITIALIZER;
+static PFNOSKEXTCOPYLOADEDKEXTINFO            g_pfnOSKextCopyLoadedKextInfo = NULL;
+static PFNKEXTMANAGERLOADKEXTWITHURL          g_pfnKextManagerLoadKextWithUrl = NULL;
+static PFNKEXTMANAGERLOADKEXTWITHIDENTIFIER   g_pfnKextManagerLoadKextWithIdentifier = NULL;
+static PFNKEXTMANAGERUNLOADKEXTWITHIDENTIFIER g_pfnKextManagerUnloadKextWithIdentifier = NULL;
 
 /** Do-once callback for setting g_pfnOSKextCopyLoadedKextInfo.   */
@@ -94 +104 @@
     int rc = RTLdrLoadEx("/System/Library/Frameworks/IOKit.framework/Versions/Current/IOKit", &hMod, RTLDRLOAD_FLAGS_NO_SUFFIX, NULL);
     if (RT_SUCCESS(rc))
+    {
         RTLdrGetSymbol(hMod, "OSKextCopyLoadedKextInfo",  (void **)&g_pfnOSKextCopyLoadedKextInfo);
+        RTLdrGetSymbol(hMod, "KextManagerLoadKextWithURL",  (void **)&g_pfnKextManagerLoadKextWithUrl);
+        RTLdrGetSymbol(hMod, "KextManagerLoadKextWithIdentifier",  (void **)&g_pfnKextManagerLoadKextWithIdentifier);
+        RTLdrGetSymbol(hMod, "KextManagerUnloadKextWithIdentifier",  (void **)&g_pfnKextManagerUnloadKextWithIdentifier);
+    }
 
     RT_NOREF(pvUser);
@@ -390 +405 @@
     return VERR_NOT_IMPLEMENTED;
 }
+
+
+RTDECL(int) RTKrnlModLoadByName(const char *pszName)
+{
+    AssertPtrReturn(pszName, VERR_INVALID_PARAMETER);
+
+    RTOnce(&g_GetIoKitApisOnce, rtKrnlModDarwinResolveIoKitApis, NULL);
+    if (!g_pfnKextManagerLoadKextWithIdentifier) return VERR_NOT_SUPPORTED;
+
+    int rc = VINF_SUCCESS;
+    CFStringRef hKextName = CFStringCreateWithCString(kCFAllocatorDefault, pszName, kCFStringEncodingUTF8);
+    if (hKextName)
+    {
+        OSReturn rcOsx = g_pfnKextManagerLoadKextWithIdentifier(hKextName, NULL /*dependencyKextAndFolderURLs*/);
+        if (rcOsx != kOSReturnSuccess)
+            rc = VERR_NOT_SUPPORTED; /** @todo Convert OSReturn values. */
+        CFRelease(hKextName);
+    }
+    else
+        rc = VERR_NO_MEMORY;
+
+    return rc;
+}
+
+
+RTDECL(int) RTKrnlModLoadByPath(const char *pszPath)
+{
+    AssertPtrReturn(pszPath, VERR_INVALID_PARAMETER);
+
+    return VERR_NOT_SUPPORTED;
+}
+
+
+RTDECL(int) RTKrnlModUnloadByName(const char *pszName)
+{
+    AssertPtrReturn(pszName, VERR_INVALID_PARAMETER);
+
+    RTOnce(&g_GetIoKitApisOnce, rtKrnlModDarwinResolveIoKitApis, NULL);
+    if (!g_pfnKextManagerUnloadKextWithIdentifier) return VERR_NOT_SUPPORTED;
+
+    int rc = VINF_SUCCESS;
+    CFStringRef hKextName = CFStringCreateWithCString(kCFAllocatorDefault, pszName, kCFStringEncodingUTF8);
+    if (hKextName)
+    {
+        OSReturn rcOsx = g_pfnKextManagerUnloadKextWithIdentifier(hKextName);
+        if (rcOsx != kOSReturnSuccess)
+            rc = VERR_NOT_SUPPORTED; /** @todo Convert OSReturn values. */
+        CFRelease(hKextName);
+    }
+    else
+        rc = VERR_NO_MEMORY;
+
+    return rc;
+}