查找所有单例的可可类？

Question

Cocoa中有几个系统类是单例的，比如UIApplication、NSNotificationCenter。现在，我想找到所有单例类，有什么建议可以让我快速找到它们吗？

我正在处理一个庞大的代码库，我需要将系统单例对象与自定义单例对象分开。

Answer 1

Objective-C 运行时骇客！好玩！

现在，在我继续之前，我将提出免责声明，我绝不建议在实际运输代码中添加这样的内容，如果您这样做，这完全不是我的错。不过，出于教育目的，这可能很有趣/有趣。

这不会是一门精确的科学，因为语言本身没有任何“单例”的实际概念。基本上，我们只是在寻找具有带有某些赠品前缀的类方法的 Objective-C 类。如果我们找到其中一个，我们很可能有一个单身人士。

考虑到这一点：

#import <Foundation/Foundation.h>
#import <objc/runtime.h>

static BOOL ClassIsSingleton(Class class) {
    unsigned int methodCount = 0;
    Method *methods = class_copyMethodList(object_getClass(class), &methodCount);

    @try {
        for (unsigned int i = 0; i < methodCount; i++) {
            Method eachMethod = methods[i];

            // only consider class methods with no arguments
            if (method_getNumberOfArguments(eachMethod) != 2) {
                continue;
            }

            char *returnType = method_copyReturnType(eachMethod);

            @try {
                // only consider class methods that return objects
                if (strcmp(returnType, @encode(id)) != 0) {
                    continue;
                }
            }
            @finally {
                free(returnType);
            }

            NSString *name = NSStringFromSelector(method_getName(methods[i]));

            // look for class methods with telltale prefixes
            if ([name hasPrefix:@"shared"]) {
                return YES;
            } else if ([name hasPrefix:@"standard"]) {
                return YES;
            } else if ([name hasPrefix:@"default"]) {
                return YES;
            } else if ([name hasPrefix:@"main"]) {
                return YES;
            } // feel free to add any additional prefixes here that I may have neglected
        }
    }
    @finally {
        free(methods);
    }

    return NO;
}

int main(int argc, const char * argv[]) {
    @autoreleasepool {
        NSMutableArray *singletons = [NSMutableArray new];

        int classCount = objc_getClassList(NULL, 0);

        Class *classes = (Class *)malloc(classCount * sizeof(Class));

        @try {
            classCount = objc_getClassList(classes, classCount);

            for (int i = 0; i < classCount; i++) {
                Class eachClass = classes[i];

                if (ClassIsSingleton(eachClass)) {
                    [singletons addObject:NSStringFromClass(eachClass)];
                }
            }
        }
        @finally {
            free(classes);
        }

        NSLog(@"Singletons: %@", singletons);
    }
    return 0;
}