Go调用Objective-C引发内存快速增长问题排查与修复请求

环境配置 #

• MacOS 12.6.8 Apple M1 Pro
• Go版本go1.20.7 darwin/arm64

问题描述 #

调用GetWindowImage函数时出现内存快速膨胀现象，无法定位问题出在Objective-C还是C层面，请求诊断并修复。

原代码 #

package main

/*
#cgo CFLAGS: -x objective-c
#cgo LDFLAGS: -framework Cocoa -framework ApplicationServices

#import <Cocoa/Cocoa.h>
#import <ApplicationServices/ApplicationServices.h>

CGWindowID GetWindowNumber(const char *title) {
    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
    NSString *windowTitle = [NSString stringWithUTF8String:title];
    CFArrayRef windowList = CGWindowListCopyWindowInfo(kCGWindowListOptionOnScreenOnly, kCGNullWindowID);
    CGWindowID windowNumber = 0;
    for (NSDictionary *entry in (__bridge NSArray *)windowList) {
        NSString *winTitle = [entry objectForKey:(id)kCGWindowName];
        // NSLog(@"Window Title: %@", winTitle);
        if ([winTitle isEqualToString:windowTitle]) {
            windowNumber = (CGWindowID)[[entry objectForKey:(id)kCGWindowNumber] integerValue];
            break;
        }
    }
    CFRelease(windowList);
    [pool drain];
    return windowNumber;
}


CFDataRef CaptureWindowImageData(CGWindowID windowNumber) {
    CGWindowID windowList[] = {windowNumber};
    CFArrayRef windowArray = CFArrayCreate(NULL, (const void **)windowList, 1, NULL);
    CGImageRef image = CGWindowListCreateImageFromArray(CGRectNull, windowArray, kCGWindowImageDefault);
    NSImage *nsImage = [[NSImage alloc] initWithCGImage:image size:NSZeroSize];
    CFRelease(windowArray);
    CGImageRelease(image);

    NSBitmapImageRep *rep = [NSBitmapImageRep imageRepWithData:[nsImage TIFFRepresentation]];
    NSDictionary *props = [NSDictionary dictionaryWithObject:[NSNumber numberWithFloat:1.0] forKey:NSImageCompressionFactor];
    CFDataRef imageData = (CFDataRef)[rep representationUsingType:NSBitmapImageFileTypePNG properties:props];

    [nsImage release];
    [rep release];

    return imageData;
}
*/
import "C"
import (
    "bytes"
    "fmt"
    "image"
    "image/png"
    "unsafe"
)

func GetWindowNumber(name string) uint {
    cTitle := C.CString(name)
    defer C.free(unsafe.Pointer(cTitle))
    return uint(C.GetWindowNumber(cTitle))
}

func GetWindowImage(windowNumber uint) (image.Image, error) {
    imageData := C.CaptureWindowImageData(C.uint(windowNumber))
    if imageData == C.CFDataRef(unsafe.Pointer(nil)) {
        return nil, fmt.Errorf("failed to get window image data for %d", windowNumber)
    }
    defer C.CFRelease(C.CFTypeRef(imageData))
    dataLength := C.CFDataGetLength(imageData)
    dataPtr := C.CFDataGetBytePtr(imageData)
    data := C.GoBytes(unsafe.Pointer(dataPtr), C.int(dataLength))
    return png.Decode(bytes.NewReader(data))
}

func main() {
    winId := GetWindowNumber("Clock")
    fmt.Println(winId)
    for i := 0; i < 100; i++ {
        _, err := GetWindowImage(winId)
        if err != nil {
            panic(err)
        }
    }
}

诊断分析 #

内存膨胀的根源在于CaptureWindowImageData函数的Objective-C代码中存在未处理的自动释放对象：

• 函数内调用[nsImage TIFFRepresentation]返回的NSData、[NSDictionary dictionaryWithObject:...])这类通过便利构造器创建的对象，默认属于自动释放对象，会被加入当前线程的自动释放池等待回收。
• 该函数没有创建NSAutoreleasePool，导致这些自动释放对象无法及时被销毁，每次调用都会积累内存，最终引发快速膨胀。
• 虽然手动释放了nsImage和rep，但遗漏了自动释放池的管理，这是Objective-C内存管理的关键环节。

修复后的代码 #

package main

/*
#cgo CFLAGS: -x objective-c
#cgo LDFLAGS: -framework Cocoa -framework ApplicationServices

#import <Cocoa/Cocoa.h>
#import <ApplicationServices/ApplicationServices.h>

CGWindowID GetWindowNumber(const char *title) {
    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
    NSString *windowTitle = [NSString stringWithUTF8String:title];
    CFArrayRef windowList = CGWindowListCopyWindowInfo(kCGWindowListOptionOnScreenOnly, kCGNullWindowID);
    CGWindowID windowNumber = 0;
    for (NSDictionary *entry in (__bridge NSArray *)windowList) {
        NSString *winTitle = [entry objectForKey:(id)kCGWindowName];
        // NSLog(@"Window Title: %@", winTitle);
        if ([winTitle isEqualToString:windowTitle]) {
            windowNumber = (CGWindowID)[[entry objectForKey:(id)kCGWindowNumber] integerValue];
            break;
        }
    }
    CFRelease(windowList);
    [pool drain];
    return windowNumber;
}


CFDataRef CaptureWindowImageData(CGWindowID windowNumber) {
    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init]; // 添加自动释放池
    CGWindowID windowList[] = {windowNumber};
    CFArrayRef windowArray = CFArrayCreate(NULL, (const void **)windowList, 1, NULL);
    CGImageRef image = CGWindowListCreateImageFromArray(CGRectNull, windowArray, kCGWindowImageDefault);
    NSImage *nsImage = [[NSImage alloc] initWithCGImage:image size:NSZeroSize];
    CFRelease(windowArray);
    CGImageRelease(image);

    NSBitmapImageRep *rep = [NSBitmapImageRep imageRepWithData:[nsImage TIFFRepresentation]];
    NSDictionary *props = [NSDictionary dictionaryWithObject:[NSNumber numberWithFloat:1.0] forKey:NSImageCompressionFactor];
    CFDataRef imageData = (CFDataRef)[rep representationUsingType:NSBitmapImageFileTypePNG properties:props];
    // 桥接时需要retain，确保返回后CFRelease能正确释放
    CFRetain(imageData);

    [nsImage release];
    [rep release];
    [pool drain]; // 释放自动释放池，回收所有自动释放对象

    return imageData;
}
*/
import "C"
import (
    "bytes"
    "fmt"
    "image"
    "image/png"
    "unsafe"
)

func GetWindowNumber(name string) uint {
    cTitle := C.CString(name)
    defer C.free(unsafe.Pointer(cTitle))
    return uint(C.GetWindowNumber(cTitle))
}

func GetWindowImage(windowNumber uint) (image.Image, error) {
    imageData := C.CaptureWindowImageData(C.uint(windowNumber))
    if imageData == C.CFDataRef(unsafe.Pointer(nil)) {
        return nil, fmt.Errorf("failed to get window image data for %d", windowNumber)
    }
    defer C.CFRelease(C.CFTypeRef(imageData))
    dataLength := C.CFDataGetLength(imageData)
    dataPtr := C.CFDataGetBytePtr(imageData)
    data := C.GoBytes(unsafe.Pointer(dataPtr), C.int(dataLength))
    return png.Decode(bytes.NewReader(data))
}

func main() {
    winId := GetWindowNumber("Clock")
    fmt.Println(winId)
    for i := 0; i < 100; i++ {
        _, err := GetWindowImage(winId)
        if err != nil {
            panic(err)
        }
    }
}

关键修复点说明 #

1. 添加自动释放池：在CaptureWindowImageData函数开头创建NSAutoreleasePool，结尾调用[pool drain]，确保所有自动释放对象在函数结束时被及时回收。
2. CFDataRef的Retain处理：由于[rep representationUsingType:...])返回的NSData是自动释放对象，桥接为CFDataRef后，需要调用CFRetain增加引用计数，确保在Go侧调用CFRelease时能正确释放该对象，避免提前被自动释放池销毁。

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