fastfetch/src/util/apple/cf_helpers.c

#include "cf_helpers.h"

const char* ffCfStrGetString(CFStringRef str, FFstrbuf* result)
{
    uint32_t length = (uint32_t)CFStringGetLength(str);
    //CFString stores UTF16 characters, therefore may require larger buffer to convert to UTF8 string
    ffStrbufEnsureFree(result, length * 2);
    if(!CFStringGetCString(str, result->chars, result->allocated, kCFStringEncodingUTF8))
    {
        ffStrbufEnsureFree(result, length * 4);
        if(!CFStringGetCString(str, result->chars, result->allocated, kCFStringEncodingUTF8))
            return "CFStringGetCString() failed";
    }
    // CFStringGetCString ensures the buffer is NUL terminated
    // https://developer.apple.com/documentation/corefoundation/1542721-cfstringgetcstring
    result->length = (uint32_t) strnlen(result->chars, (uint32_t)result->allocated);
    return NULL;
}

const char* ffCfDictGetString(CFDictionaryRef dict, CFStringRef key, FFstrbuf* result)
{
    CFTypeRef cf = (CFTypeRef)CFDictionaryGetValue(dict, key);
    if(cf == NULL)
        return "CFDictionaryGetValue() failed";

    if(CFGetTypeID(cf) == CFStringGetTypeID())
    {
        return ffCfStrGetString((CFStringRef)cf, result);
    }
    else if(CFGetTypeID(cf) == CFDataGetTypeID())
    {
        CFDataRef cfData = (CFDataRef)cf;
        uint32_t length = (uint32_t)CFDataGetLength(cfData);
        ffStrbufEnsureFree(result, length + 1);
        CFDataGetBytes(cfData, CFRangeMake(0, length), (uint8_t*)result->chars);
        result->length = (uint32_t)strnlen(result->chars, length);
        result->chars[result->length] = '\0';
        return NULL;
    }

    return "TypeID is neither 'CFString' nor 'CFData'";
}

const char* ffCfDictGetBool(CFDictionaryRef dict, CFStringRef key, bool* result)
{
    CFBooleanRef cf = (CFBooleanRef)CFDictionaryGetValue(dict, key);
    if(cf == NULL || CFGetTypeID(cf) != CFBooleanGetTypeID())
        return "TypeID is not 'CFBoolean'";

    *result = CFBooleanGetValue(cf);
    return NULL;
}

const char* ffCfDictGetInt(CFDictionaryRef dict, CFStringRef key, int* result)
{
    CFNumberRef cf = (CFNumberRef)CFDictionaryGetValue(dict, key);
    if (cf == NULL || CFGetTypeID(cf) != CFNumberGetTypeID())
        return "TypeID is not 'CFNumber'";

    if(!CFNumberGetValue(cf, kCFNumberSInt32Type, result))
        return "Number type is not SInt32";
    return NULL;
}

const char* ffCfDictGetDict(CFDictionaryRef dict, CFStringRef key, CFDictionaryRef* result)
{
    CFDictionaryRef cf = (CFDictionaryRef)CFDictionaryGetValue(dict, key);
    if (cf == NULL || CFGetTypeID(cf) != CFDictionaryGetTypeID())
        return "TypeID is not 'CFDictionary'";

    *result = cf;
    return NULL;
}