CFSet内存泄漏问题：不知如何/何处调用CFRelease()

嗨，我完全懂你这种明明知道要释放Core Foundation对象但卡在哪一步的困惑！我之前在处理IOKit USB相关代码时也碰到过一模一样的CFSet内存泄漏问题，刚好可以帮你理清思路。

问题回顾 #

你在处理USB设备的代码中遭遇了内存泄漏，已经定位到泄漏源是CFSet。你清楚需要调用CFRelease()来释放CF对象，但USBDeviceCount()函数需要先获取设备数量再返回，这让你不确定该在哪个节点正确释放这个CFSet。

存在内存泄漏的原代码 #

#include <IOKit/usb/IOUSBLib.h>

int USBDeviceCount() {
    CFMutableDictionaryRef matchingDict = IOServiceMatching(kIOUSBDeviceClassName);
    CFSetRef deviceSet = IOServiceGetMatchingServices(kIOMasterPortDefault, matchingDict, NULL);
    int count = CFSetGetCount(deviceSet);
    // 未释放deviceSet，导致内存泄漏
    return count;
}

这段代码的问题很明确：IOServiceGetMatchingServices会创建并返回一个CFSet对象，按照Core Foundation的内存管理规则，我们拥有这个对象的所有权，必须手动调用CFRelease()释放它。你可能担心在return前释放会影响返回的count值——其实完全不用，因为CFSetGetCount已经把数值读取到了count变量里，释放CFSet不会改变这个变量的值。

修复后的解决方案代码 #

#include <IOKit/usb/IOUSBLib.h>

int USBDeviceCount() {
    CFMutableDictionaryRef matchingDict = IOServiceMatching(kIOUSBDeviceClassName);
    CFSetRef deviceSet = IOServiceGetMatchingServices(kIOMasterPortDefault, matchingDict, NULL);
    
    int count = 0;
    if (deviceSet != NULL) { // 先判断对象有效性，避免空指针崩溃
        count = CFSetGetCount(deviceSet);
        CFRelease(deviceSet); // 获取到count后立即释放CFSet，归还所有权
    }
    
    return count;
}

关键修复点说明 #

• 增加空指针判断：IOServiceGetMatchingServices有可能返回NULL（比如没有匹配到USB设备时），先判断deviceSet是否有效，能避免后续调用CFSetGetCount或CFRelease时触发崩溃。
• 及时释放CFSet：在成功获取设备数量count之后，立刻调用CFRelease(deviceSet)——此时我们已经拿到了需要的数值，后续不再依赖这个CFSet对象，释放操作完全安全，不会影响返回的count值。
• 遵循CF内存管理规则：所有通过带有Create、Copy、Get（这里的IOServiceGetMatchingServices属于创建对象的场景）前缀函数获取的CF对象，我们都拥有其所有权，必须手动调用CFRelease()来释放，否则就会造成内存泄漏。

内容的提问来源于stack exchange，提问作者William Gustafson