darling-objc4/test/runtime.m
2022-03-31 21:15:07 -07:00

235 lines
9.0 KiB
Objective-C

/*
TEST_BUILD_OUTPUT
.*runtime.m:\d+:\d+: warning: null passed to a callee that requires a non-null argument \[-Wnonnull\](\n.* note: expanded from macro 'testassert')?
.*runtime.m:\d+:\d+: warning: null passed to a callee that requires a non-null argument \[-Wnonnull\](\n.* note: expanded from macro 'testassert')?
.*runtime.m:\d+:\d+: warning: null passed to a callee that requires a non-null argument \[-Wnonnull\](\n.* note: expanded from macro 'testassert')?
END
TEST_RUN_OUTPUT
objc\[\d+\]: class `SwiftV1Class\' not linked into application
objc\[\d+\]: class `DoesNotExist\' not linked into application
OK: runtime.m
OR
confused by Foundation
OK: runtime.m
END
*/
#include "test.h"
#include "testroot.i"
#include <string.h>
#include <dlfcn.h>
#include <mach-o/ldsyms.h>
#include <objc/objc-runtime.h>
#if __has_feature(objc_arc)
int main()
{
// provoke the same nullability warnings as the real test
objc_getClass(nil);
objc_getClass(nil);
objc_getClass(nil);
testwarn("rdar://11368528 confused by Foundation");
fprintf(stderr, "confused by Foundation\n");
succeed(__FILE__);
}
#else
@interface Sub : TestRoot @end
@implementation Sub @end
#define SwiftV1MangledName "_TtC6Module12SwiftV1Class"
#define SwiftV1MangledName2 "_TtC2Sw13SwiftV1Class2"
#define SwiftV1MangledName3 "_TtCs13SwiftV1Class3"
#define SwiftV1MangledName4 "_TtC6Swiftt13SwiftV1Class4"
__attribute__((objc_runtime_name(SwiftV1MangledName)))
@interface SwiftV1Class : TestRoot @end
@implementation SwiftV1Class @end
__attribute__((objc_runtime_name(SwiftV1MangledName2)))
@interface SwiftV1Class2 : TestRoot @end
@implementation SwiftV1Class2 @end
__attribute__((objc_runtime_name(SwiftV1MangledName3)))
@interface SwiftV1Class3 : TestRoot @end
@implementation SwiftV1Class3 @end
__attribute__((objc_runtime_name(SwiftV1MangledName4)))
@interface SwiftV1Class4 : TestRoot @end
@implementation SwiftV1Class4 @end
int main()
{
Class list[100];
Class *list2;
unsigned int count, count0, count2;
unsigned int i;
int foundTestRoot;
int foundSub;
int foundSwiftV1;
int foundSwiftV1class2;
int foundSwiftV1class3;
int foundSwiftV1class4;
const char **names;
const char **namesFromHeader;
Dl_info info;
[TestRoot class];
// This shouldn't touch any classes.
dladdr(&_mh_execute_header, &info);
names = objc_copyClassNamesForImage(info.dli_fname, &count);
testassert(names);
testassert(count == 6);
testassert(names[count] == NULL);
foundTestRoot = 0;
foundSub = 0;
foundSwiftV1 = 0;
foundSwiftV1class2 = 0;
foundSwiftV1class3 = 0;
foundSwiftV1class4 = 0;
for (i = 0; i < count; i++) {
if (0 == strcmp(names[i], "TestRoot")) foundTestRoot++;
if (0 == strcmp(names[i], "Sub")) foundSub++;
if (0 == strcmp(names[i], "Module.SwiftV1Class")) foundSwiftV1++;
if (0 == strcmp(names[i], "Sw.SwiftV1Class2")) foundSwiftV1class2++;
if (0 == strcmp(names[i], "Swift.SwiftV1Class3")) foundSwiftV1class3++;
if (0 == strcmp(names[i], "Swiftt.SwiftV1Class4")) foundSwiftV1class4++;
}
testassert(foundTestRoot == 1);
testassert(foundSub == 1);
testassert(foundSwiftV1 == 1);
testassert(foundSwiftV1class2 == 1);
testassert(foundSwiftV1class3 == 1);
testassert(foundSwiftV1class4 == 1);
// Getting the names using the header should give us the same list.
namesFromHeader = objc_copyClassNamesForImage(info.dli_fname, &count0);
testassert(namesFromHeader);
testassert(count == count0);
for (i = 0; i < count; i++) {
testassert(!strcmp(names[i], namesFromHeader[i]));
}
// class Sub hasn't been touched - make sure it's in the class list too
count0 = objc_getClassList(NULL, 0);
testassert(count0 >= 2 && count0 < 100);
list[count0-1] = NULL;
count = objc_getClassList(list, count0-1);
testassert(list[count0-1] == NULL);
testassert(count == count0);
// So that demangling works, fake what would have happened with Swift
// and force the "Swift" bit on the class
((uintptr_t *)objc_getClass(SwiftV1MangledName))[4] |= 2;
((uintptr_t *)objc_getClass(SwiftV1MangledName2))[4] |= 2;
((uintptr_t *)objc_getClass(SwiftV1MangledName3))[4] |= 2;
((uintptr_t *)objc_getClass(SwiftV1MangledName4))[4] |= 2;
count = objc_getClassList(list, count0);
testassert(count == count0);
for (i = 0; i < count; i++) {
testprintf("%s\n", class_getName(list[i]));
}
foundTestRoot = 0;
foundSub = 0;
foundSwiftV1 = 0;
foundSwiftV1class2 = 0;
foundSwiftV1class3 = 0;
foundSwiftV1class4 = 0;
for (i = 0; i < count; i++) {
if (0 == strcmp(class_getName(list[i]), "TestRoot")) foundTestRoot++;
if (0 == strcmp(class_getName(list[i]), "Sub")) foundSub++;
if (0 == strcmp(class_getName(list[i]), "Module.SwiftV1Class")) foundSwiftV1++;
if (0 == strcmp(class_getName(list[i]), "Sw.SwiftV1Class2")) foundSwiftV1class2++;
if (0 == strcmp(class_getName(list[i]), "Swift.SwiftV1Class3")) foundSwiftV1class3++;
if (0 == strcmp(class_getName(list[i]), "Swiftt.SwiftV1Class4")) foundSwiftV1class4++;
// list should be non-meta classes only
testassert(!class_isMetaClass(list[i]));
}
testassert(foundTestRoot == 1);
testassert(foundSub == 1);
testassert(foundSwiftV1 == 1);
testassert(foundSwiftV1class2 == 1);
testassert(foundSwiftV1class3 == 1);
testassert(foundSwiftV1class4 == 1);
// fixme check class handler
testassert(objc_getClass("TestRoot") == [TestRoot class]);
testassert(objc_getClass("Module.SwiftV1Class") == [SwiftV1Class class]);
testassert(objc_getClass(SwiftV1MangledName) == [SwiftV1Class class]);
testassert(objc_getClass("Sw.SwiftV1Class2") == [SwiftV1Class2 class]);
testassert(objc_getClass(SwiftV1MangledName2) == [SwiftV1Class2 class]);
testassert(objc_getClass("Swift.SwiftV1Class3") == [SwiftV1Class3 class]);
testassert(objc_getClass(SwiftV1MangledName3) == [SwiftV1Class3 class]);
testassert(objc_getClass("Swiftt.SwiftV1Class4") == [SwiftV1Class4 class]);
testassert(objc_getClass(SwiftV1MangledName4) == [SwiftV1Class4 class]);
testassert(objc_getClass("SwiftV1Class") == nil);
testassert(objc_getClass("DoesNotExist") == nil);
testassert(objc_getClass(NULL) == nil);
testassert(objc_getMetaClass("TestRoot") == object_getClass([TestRoot class]));
testassert(objc_getMetaClass("Module.SwiftV1Class") == object_getClass([SwiftV1Class class]));
testassert(objc_getMetaClass(SwiftV1MangledName) == object_getClass([SwiftV1Class class]));
testassert(objc_getMetaClass("SwiftV1Class") == nil);
testassert(objc_getMetaClass("DoesNotExist") == nil);
testassert(objc_getMetaClass(NULL) == nil);
// fixme check class no handler
testassert(objc_lookUpClass("TestRoot") == [TestRoot class]);
testassert(objc_lookUpClass("Module.SwiftV1Class") == [SwiftV1Class class]);
testassert(objc_lookUpClass(SwiftV1MangledName) == [SwiftV1Class class]);
testassert(objc_lookUpClass("SwiftV1Class") == nil);
testassert(objc_lookUpClass("DoesNotExist") == nil);
testassert(objc_lookUpClass(NULL) == nil);
testassert(! object_isClass(nil));
testassert(! object_isClass([TestRoot new]));
testassert(object_isClass([TestRoot class]));
testassert(object_isClass(object_getClass([TestRoot class])));
testassert(object_isClass([Sub class]));
testassert(object_isClass(object_getClass([Sub class])));
testassert(object_isClass([SwiftV1Class class]));
testassert(object_isClass(object_getClass([SwiftV1Class class])));
list2 = objc_copyClassList(&count2);
testassert(count2 == count);
testassert(list2);
testassert(malloc_size(list2) >= (1+count2) * sizeof(Class));
for (i = 0; i < count; i++) {
testassert(list[i] == list2[i]);
}
testassert(list2[count] == NULL);
free(list2);
free(objc_copyClassList(NULL));
// Make sure metaclasses get demangled too.
testassert(strcmp(class_getName([TestRoot class]), class_getName(object_getClass([TestRoot class]))) == 0);
testassert(strcmp(class_getName([Sub class]), class_getName(object_getClass([Sub class]))) == 0);
testassert(strcmp(class_getName([SwiftV1Class class]), class_getName(object_getClass([SwiftV1Class class]))) == 0);
testassert(strcmp(class_getName([SwiftV1Class2 class]), class_getName(object_getClass([SwiftV1Class2 class]))) == 0);
testassert(strcmp(class_getName([SwiftV1Class3 class]), class_getName(object_getClass([SwiftV1Class3 class]))) == 0);
testassert(strcmp(class_getName([SwiftV1Class4 class]), class_getName(object_getClass([SwiftV1Class4 class]))) == 0);
testassert(!_class_isSwift([TestRoot class]));
testassert(!_class_isSwift([Sub class]));
testassert(_class_isSwift([SwiftV1Class class]));
testassert(_class_isSwift([SwiftV1Class2 class]));
testassert(_class_isSwift([SwiftV1Class3 class]));
testassert(_class_isSwift([SwiftV1Class4 class]));
succeed(__FILE__);
}
#endif