add protocols

MARTNSObject.h

@@ -2,6 +2,7 @@
 #import <Foundation/Foundation.h>
 
 
+@class RTProtocol;
 @class RTIvar;
 @class RTProperty;
 @class RTMethod;
@@ -20,6 +21,8 @@
 + (Class)rt_setSuperclass: (Class)newSuperclass;
 + (size_t)rt_instanceSize;
 
++ (NSArray *)rt_protocols;
+
 + (NSArray *)rt_methods;
 + (RTMethod *)rt_methodForSelector: (SEL)sel;
 

MARTNSObject.m

@@ -3,6 +3,7 @@
 
 #import <objc/runtime.h>
 
+#import "RTProtocol.h"
 #import "RTIvar.h"
 #import "RTProperty.h"
 #import "RTMethod.h"
@@ -80,6 +81,19 @@ + (size_t)rt_instanceSize
     return class_getInstanceSize(self);
 }
 
++ (NSArray *)rt_protocols
+{
+    unsigned int count;
+    Protocol **protocols = class_copyProtocolList(self, &count);
+
+    NSMutableArray *array = [NSMutableArray array];
+    for(unsigned i = 0; i < count; i++)
+        [array addObject: [RTProtocol protocolWithObjCProtocol: protocols[i]]];
+
+    free(protocols);
+    return array;
+}
+
 + (NSArray *)rt_methods
 {
     unsigned int count;

Makefile

@@ -1,7 +1,7 @@
 all: tests
 
 tests:
-	gcc -framework Foundation --std=c99 main.m MARTNSObject.m RTMethod.m RTIvar.m RTProperty.m RTUnregisteredClass.m
+	gcc -framework Foundation --std=c99 main.m MARTNSObject.m RTProtocol.m RTMethod.m RTIvar.m RTProperty.m RTUnregisteredClass.m
 
 clean:
 	rm -f a.out

RTProtocol.h

@@ -0,0 +1,23 @@
+
+#import <Foundation/Foundation.h>
+#import <objc/runtime.h>
+
+
+@interface RTProtocol : NSObject
+{
+}
+
++ (NSArray *)allProtocols;
+
++ (id)protocolWithObjCProtocol: (Protocol *)protocol;
++ (id)protocolWithName: (NSString *)name;
+
+- (id)initWithObjCProtocol: (Protocol *)protocol;
+- (id)initWithName: (NSString *)name;
+
+- (Protocol *)objCProtocol;
+- (NSString *)name;
+- (NSArray *)incorporatedProtocols;
+- (NSArray *)methodsRequired: (BOOL)isRequiredMethod instance: (BOOL)isInstanceMethod;
+
+@end

RTProtocol.m

@@ -0,0 +1,122 @@
+
+#import "RTProtocol.h"
+#import "RTMethod.h"
+
+
+@interface _RTObjCProtocol : RTProtocol
+{
+    Protocol *_protocol;
+}
+@end
+
+@implementation _RTObjCProtocol
+
+- (id)initWithObjCProtocol: (Protocol *)protocol
+{
+    if((self = [self init]))
+    {
+        _protocol = protocol;
+    }
+    return self;
+}
+
+- (Protocol *)objCProtocol
+{
+    return _protocol;
+}
+
+@end
+
+@implementation RTProtocol
+
++ (NSArray *)allProtocols
+{
+    unsigned int count;
+    Protocol **protocols = objc_copyProtocolList(&count);
+
+    NSMutableArray *array = [NSMutableArray array];
+    for(unsigned i = 0; i < count; i++)
+        [array addObject: [[[self alloc] initWithObjCProtocol: protocols[i]] autorelease]];
+
+    free(protocols);
+    return array;
+}
+
++ (id)protocolWithObjCProtocol: (Protocol *)protocol
+{
+    return [[[self alloc] initWithObjCProtocol: protocol] autorelease];
+}
+
++ (id)protocolWithName: (NSString *)name
+{
+    return [[[self alloc] initWithName: name] autorelease];
+}
+
+- (id)initWithObjCProtocol: (Protocol *)protocol
+{
+    [self release];
+    return [[_RTObjCProtocol alloc] initWithObjCProtocol: protocol];
+}
+
+- (id)initWithName: (NSString *)name
+{
+    return [self initWithObjCProtocol:objc_getProtocol([name cStringUsingEncoding:[NSString defaultCStringEncoding]])];
+}
+
+- (NSString *)description
+{
+    return [NSString stringWithFormat: @"<%@ %p: %@>", [self class], self, [self name]];
+}
+
+- (BOOL)isEqual: (id)other
+{
+    return [other isKindOfClass: [RTProtocol class]] &&
+           [[self objCProtocol] isEqual: [other objCProtocol]];
+}
+
+- (NSUInteger)hash
+{
+    return [[self objCProtocol] hash];
+}
+
+- (Protocol *)objCProtocol
+{
+    [self doesNotRecognizeSelector: _cmd];
+    return nil;
+}
+
+- (NSString *)name
+{
+    return [NSString stringWithUTF8String: protocol_getName([self objCProtocol])];
+}
+
+- (NSArray *)incorporatedProtocols
+{
+    unsigned int count;
+    Protocol **protocols = protocol_copyProtocolList([self objCProtocol], &count);
+
+    NSMutableArray *array = [NSMutableArray array];
+    for(unsigned i = 0; i < count; i++)
+        [array addObject: [RTProtocol protocolWithObjCProtocol: protocols[i]]];
+
+    free(protocols);
+    return array;
+}
+
+- (NSArray *)methodsRequired: (BOOL)isRequiredMethod instance: (BOOL)isInstanceMethod
+{
+    unsigned int count;
+    struct objc_method_description *methods = protocol_copyMethodDescriptionList([self objCProtocol], isRequiredMethod, isInstanceMethod, &count);
+
+    NSMutableArray *array = [NSMutableArray array];
+    for(unsigned i = 0; i < count; i++)
+    {
+        NSString *signature = [NSString stringWithCString: methods[i].types encoding: [NSString defaultCStringEncoding]];
+        [array addObject: [RTMethod methodWithSelector: methods[i].name implementation: NULL signature: signature]];
+    }
+
+    free(methods);
+    return array;
+}
+
+@end

RTUnregisteredClass.h

@@ -2,6 +2,7 @@
 #import <Foundation/Foundation.h>
 
 
+@class RTProtocol;
 @class RTIvar;
 @class RTMethod;
 
@@ -14,6 +15,7 @@
 
 - (id)initWithClass: (Class)c;
 
+- (void)addProtocol: (RTProtocol *)protocol;
 - (void)addIvar: (RTIvar *)ivar;
 - (void)addMethod: (RTMethod *)method;
 

RTUnregisteredClass.m

@@ -1,6 +1,7 @@
 
 #import "RTUnregisteredClass.h"
 
+#import "RTProtocol.h"
 #import "RTIvar.h"
 #import "RTMethod.h"
 
@@ -21,6 +22,11 @@ - (id)initWithClass: (Class)c
     return self;
 }
 
+- (void)addProtocol: (RTProtocol *)protocol
+{
+    class_addProtocol(_class, [protocol objCProtocol]);
+}
+
 - (void)addIvar: (RTIvar *)ivar
 {
     const char *typeStr = [[ivar typeEncoding] UTF8String];

main.m

@@ -1,6 +1,7 @@
-// gcc -framework Foundation --std=c99 main.m MARTNSObject.m RTMethod.m RTIvar.m RTProperty.m RTUnregisteredClass.m
+// gcc -framework Foundation --std=c99 main.m MARTNSObject.m RTProtocol.m RTMethod.m RTIvar.m RTProperty.m RTUnregisteredClass.m
 
 #import "MARTNSObject.h"
+#import "RTProtocol.h"
 #import "RTIvar.h"
 #import "RTProperty.h"
 #import "RTMethod.h"
@@ -154,16 +155,44 @@ static void TestSetMethod(void)
     [obj release];
 }
 
-@interface SampleClass : NSObject
+@protocol SampleProtocol <NSObject>
++ (void)requiredClassMethod;
+- (void)requiredInstanceMethod;
+@optional
++ (void)optionalClassMethod;
+- (void)optionalInstanceMethod;
+@end
+@interface SampleClass : NSObject <SampleProtocol>
 {
     id someIvar;
 }
 @property (assign, getter=customGetter) id someProperty;
 @end
 @implementation SampleClass
 @synthesize someProperty=someIvar;
++ (void)requiredClassMethod
+{
+    return;
+}
+- (void)requiredInstanceMethod
+{
+    return;
+}
 @end
 
+static void TestProtocolQuery(void)
+{
+    NSArray *protocols = [SampleClass rt_protocols];
+    TEST_ASSERT([[protocols valueForKey: @"name"] containsObject: @"SampleProtocol"]);
+
+    Protocol *protocol = [RTProtocol protocolWithObjCProtocol: NSProtocolFromString(@"SampleProtocol")];
+    TEST_ASSERT([[protocol incorporatedProtocols] containsObject: [RTProtocol protocolWithObjCProtocol: NSProtocolFromString(@"NSObject")]]);
+    TEST_ASSERT([[[protocol methodsRequired: YES instance: YES] valueForKey: @"selectorName"] containsObject: @"requiredInstanceMethod"]);
+    TEST_ASSERT([[[protocol methodsRequired: YES instance: NO] valueForKey: @"selectorName"] containsObject: @"requiredClassMethod"]);
+    TEST_ASSERT([[[protocol methodsRequired: NO instance: YES] valueForKey: @"selectorName"] containsObject: @"optionalInstanceMethod"]);
+    TEST_ASSERT([[[protocol methodsRequired: NO instance: NO] valueForKey: @"selectorName"] containsObject: @"optionalClassMethod"]);
+}
+
 static void TestIvarQuery(void)
 {
     NSArray *ivars = [SampleClass rt_ivars];
@@ -281,6 +310,7 @@ int main(int argc, char **argv)
             TEST(TestAddMethod);
             TEST(TestMethodFetching);
             TEST(TestSetMethod);
+            TEST(TestProtocolQuery);
             TEST(TestIvarQuery);
             TEST(TestPropertyQuery);
             TEST(TestIvarAdd);