objc4-437.tar.gz

[apple/objc4.git] / test / cacheflush.m

diff --git a/test/cacheflush.m b/test/cacheflush.m
new file mode 100644 (file)
index 0000000..0de609e
--- /dev/null
+++ b/test/cacheflush.m
@@ -0,0 +1,54 @@
+#include "test.h"
+#include <objc/runtime.h>
+#include <dlfcn.h>
+
+#include "cacheflush.h"
+
+@interface Sub : Super @end
+@implementation Sub @end
+
+
+int main()
+{
+    uintptr_t buf[10];
+    uintptr_t buf2[10];
+    buf[0] = (uintptr_t)[Super class];
+    buf2[0] = (uintptr_t)[Sub class];
+
+    // Fill method cache
+    testassert(1 == [Super classMethod]);
+    testassert(1 == [(Super *)buf instanceMethod]);
+    testassert(1 == [Super classMethod]);
+    testassert(1 == [(Super *)buf instanceMethod]);
+
+    testassert(1 == [Sub classMethod]);
+    testassert(1 == [(Sub *)buf2 instanceMethod]);
+    testassert(1 == [Sub classMethod]);
+    testassert(1 == [(Sub *)buf2 instanceMethod]);
+
+    // Dynamically load a category
+    dlopen("cacheflush2.out", 0);
+
+    // Make sure old cache results are gone
+    testassert(2 == [Super classMethod]);
+    testassert(2 == [(Super *)buf instanceMethod]);
+
+    testassert(2 == [Sub classMethod]);
+    testassert(2 == [(Sub *)buf2 instanceMethod]);
+
+    // Dynamically load another category
+    dlopen("cacheflush3.out", 0);
+
+    // Make sure old cache results are gone
+    testassert(3 == [Super classMethod]);
+    testassert(3 == [(Super *)buf instanceMethod]);
+
+    testassert(3 == [Sub classMethod]);
+    testassert(3 == [(Sub *)buf2 instanceMethod]);
+
+    // fixme test subclasses
+
+    // fixme test objc_flush_caches(), class_addMethod(), class_addMethods()
+
+    succeed(__FILE__);
+}