Collect host hardware info on mac

Author: briantu

pkg/inventory/hardware/collector_darwin.go

@@ -0,0 +1,67 @@
+// Unless explicitly stated otherwise all files in this repository are licensed
+// under the Apache License Version 2.0.
+// This product includes software developed at Datadog (https://www.datadoghq.com/).
+// Copyright 2016-present Datadog, Inc.
+
+//go:build darwin
+
+package hardware
+
+/*
+#cgo CFLAGS: -x objective-c -fobjc-arc
+#cgo LDFLAGS: -framework Foundation -framework IOKit
+
+#include <stdlib.h>
+#include "hosthardware_darwin.h"
+*/
+import "C"
+import (
+	"strings"
+	"unsafe"
+)
+
+func collect() (*SystemHardwareInfo, error) {
+	cInfo := C.getDeviceInfo()
+	defer C.free(unsafe.Pointer(cInfo.modelIdentifier))
+	defer C.free(unsafe.Pointer(cInfo.modelNumber))
+	defer C.free(unsafe.Pointer(cInfo.productName))
+	defer C.free(unsafe.Pointer(cInfo.serialNumber))
+
+	return &SystemHardwareInfo{
+		Manufacturer: "Apple Inc.",
+		Name:         C.GoString(cInfo.productName),
+		Identifier:   C.GoString(cInfo.modelIdentifier),
+		ModelNumber:  C.GoString(cInfo.modelNumber),
+		SerialNumber: C.GoString(cInfo.serialNumber),
+		ChassisType:  getChassisType(C.GoString(cInfo.productName), C.GoString(cInfo.modelIdentifier)),
+	}, nil
+}
+
+func getChassisType(productName string, modelIdentifier string) string {
+	lowerName := strings.ToLower(productName)
+	lowerModel := strings.ToLower(modelIdentifier)
+
+	// Check for virtual machines first
+	// VMware VMs have modelIdentifier like "VMware7,1"
+	// Apple Silicon VMs have modelIdentifier like "VirtualMac2,1" and productName "Apple Virtual Machine 1"
+	// Parallels VMs have "Parallels" in the modelIdentifier
+	if strings.Contains(lowerModel, "vmware") ||
+		strings.Contains(lowerModel, "virtual") ||
+		strings.Contains(lowerModel, "parallels") ||
+		strings.Contains(lowerName, "virtual") {
+		return "Virtual Machine"
+	}
+
+	if strings.HasPrefix(lowerName, "macbook") {
+		return "Laptop"
+	}
+
+	if strings.HasPrefix(lowerName, "imac") ||
+		strings.HasPrefix(lowerName, "mac mini") ||
+		strings.HasPrefix(lowerName, "mac pro") ||
+		strings.HasPrefix(lowerName, "mac studio") {
+		return "Desktop"
+	}
+
+	return "Other"
+}

pkg/inventory/hardware/collector_nix.go

@@ -3,7 +3,7 @@
 // This product includes software developed at Datadog (https://www.datadoghq.com/).
 // Copyright 2016-present Datadog, Inc.
 
-//go:build !windows
+//go:build !windows && !darwin
 
 package hardware
 

pkg/inventory/hardware/hosthardware_darwin.m

@@ -0,0 +1,64 @@
+#import <Foundation/Foundation.h>
+#import <IOKit/IOKitLib.h>
+#import "hosthardware_darwin.h"
+#import <stdlib.h>
+#import <string.h>
+
+static NSDictionary *getServiceProperties(CFMutableDictionaryRef matching) {
+    io_service_t service = IOServiceGetMatchingService(kIOMainPortDefault, matching);
+    if (!service) return nil;
+
+    CFMutableDictionaryRef props = NULL;
+    if (IORegistryEntryCreateCFProperties(service, &props, kCFAllocatorDefault, 0) != KERN_SUCCESS) {
+        IOObjectRelease(service);
+        return nil;
+    }
+
+    IOObjectRelease(service);
+    return (__bridge_transfer NSDictionary *)props;
+}
+
+static char *copyStringProperty(NSDictionary *props, NSString *key) {
+    id value = props[key];
+    if (!value) return NULL;
+    
+    if ([value isKindOfClass:[NSString class]]) {
+        return strdup([value UTF8String]);
+    } else if ([value isKindOfClass:[NSData class]]) {
+        const char *bytes = (const char *)[value bytes];
+        return strndup(bytes, strnlen(bytes, [value length]));
+    }
+    return NULL;
+}
+
+DeviceInfo getDeviceInfo(void) {
+    @autoreleasepool {
+        DeviceInfo info = {0};
+
+        NSDictionary *platform = getServiceProperties(IOServiceMatching("IOPlatformExpertDevice"));
+        if (platform) {
+            info.found = true;
+            info.modelIdentifier = copyStringProperty(platform, @"model");
+            info.serialNumber = copyStringProperty(platform, @"IOPlatformSerialNumber");
+
+            char *modelNum = copyStringProperty(platform, @"model-number");
+            char *region = copyStringProperty(platform, @"region-info");
+            if (modelNum && region) {
+                asprintf(&info.modelNumber, "%s%s", modelNum, region);
+                free(modelNum);
+                free(region);
+            } else {
+                info.modelNumber = modelNum;
+                free(region);
+            }
+        }
+
+        NSDictionary *product = getServiceProperties(IOServiceNameMatching("product"));
+        if (product) {
+            info.productName = copyStringProperty(product, @"product-name");
+        }
+
+        return info;
+    }
+}
+

pkg/inventory/hardware/hosthwardware_darwin.h

@@ -0,0 +1,16 @@
+#ifndef HOSTHARDWARE_DARWIN_H
+#define HOSTHARDWARE_DARWIN_H
+
+#include <stdbool.h>
+
+typedef struct {
+    bool found;
+    char *modelIdentifier;
+    char *modelNumber;
+    char *productName;
+    char *serialNumber;
+} DeviceInfo;
+
+DeviceInfo getDeviceInfo(void);
+
+#endif