1 files changed, 87 insertions, 0 deletions
diff --git a/src/detection/gpu/gpu_apple.m b/src/detection/gpu/gpu_apple.m
new file mode 100644
index 0000000..e206bdc
--- /dev/null
+++ b/src/detection/gpu/gpu_apple.m
@@ -0,0 +1,87 @@
+#include "gpu.h"
+
+#import <Metal/MTLDevice.h>
+#import <IOKit/kext/KextManager.h>
+
+#ifndef MAC_OS_VERSION_26_0
+    #define MTLGPUFamilyMetal4 ((MTLGPUFamily) 5002)
+#endif
+#ifndef MAC_OS_VERSION_13_0
+    #define MTLGPUFamilyMetal3 ((MTLGPUFamily) 5001)
+#endif
+#ifndef MAC_OS_X_VERSION_10_15
+    #define MTLFeatureSet_macOS_GPUFamily1_v4 ((MTLFeatureSet) 10004)
+    #define MTLFeatureSet_macOS_GPUFamily2_v1 ((MTLFeatureSet) 10005)
+#endif
+
+const char* ffGpuDetectDriverVersion(FFlist* gpus)
+{
+    if (@available(macOS 10.7, *))
+    {
+        NSMutableArray* arr = NSMutableArray.new;
+        FF_LIST_FOR_EACH(FFGPUResult, x, *gpus)
+            [arr addObject:@(x->driver.chars)];
+
+        NSDictionary* dict = CFBridgingRelease(KextManagerCopyLoadedKextInfo((__bridge CFArrayRef)arr, (__bridge CFArrayRef)@[@"CFBundleVersion"]));
+        FF_LIST_FOR_EACH(FFGPUResult, x, *gpus)
+        {
+            NSString* version = dict[@(x->driver.chars)][@"CFBundleVersion"];
+            if (version)
+            {
+                ffStrbufAppendC(&x->driver, ' ');
+                ffStrbufAppendS(&x->driver, version.UTF8String);
+            }
+        }
+        return NULL;
+    }
+    return "Unsupported macOS version";
+}
+
+const char* ffGpuDetectMetal(FFlist* gpus)
+{
+    if (@available(macOS 10.13, *))
+    {
+        for (id<MTLDevice> device in MTLCopyAllDevices())
+        {
+            FFGPUResult* gpu = NULL;
+            FF_LIST_FOR_EACH(FFGPUResult, x, *gpus)
+            {
+                if (x->deviceId == device.registryID)
+                {
+                    gpu = x;
+                    break;
+                }
+            }
+            if (!gpu) continue;
+
+            #ifndef MAC_OS_X_VERSION_10_15
+            if ([device supportsFeatureSet:MTLFeatureSet_macOS_GPUFamily2_v1])
+                ffStrbufSetStatic(&gpu->platformApi, "Metal Feature Set 2");
+            else if ([device supportsFeatureSet:MTLFeatureSet_macOS_GPUFamily1_v1])
+                ffStrbufSetStatic(&gpu->platformApi, "Metal Feature Set 1");
+            #else // MAC_OS_X_VERSION_10_15
+            #pragma clang diagnostic push
+            #pragma clang diagnostic ignored "-Wunguarded-availability-new"
+            if ([device supportsFamily:MTLGPUFamilyMetal4])
+                ffStrbufSetStatic(&gpu->platformApi, "Metal 4");
+            else if ([device supportsFamily:MTLGPUFamilyMetal3])
+                ffStrbufSetStatic(&gpu->platformApi, "Metal 3");
+            #pragma clang diagnostic pop
+            else if ([device supportsFamily:MTLGPUFamilyCommon3])
+                ffStrbufSetStatic(&gpu->platformApi, "Metal Common 3");
+            else if ([device supportsFamily:MTLGPUFamilyCommon2])
+                ffStrbufSetStatic(&gpu->platformApi, "Metal Common 2");
+            else if ([device supportsFamily:MTLGPUFamilyCommon1])
+                ffStrbufSetStatic(&gpu->platformApi, "Metal Common 1");
+
+            gpu->type = device.hasUnifiedMemory ? FF_GPU_TYPE_INTEGRATED : FF_GPU_TYPE_DISCRETE;
+            gpu->index = (uint32_t) device.locationNumber;
+
+            if (device.hasUnifiedMemory && device.recommendedMaxWorkingSetSize > 0)
+                gpu->shared.total = device.recommendedMaxWorkingSetSize;
+            #endif
+        }
+        return NULL;
+    }
+    return "Metal API is not supported by this macOS version";
+}