#import <objc/runtime.h>

#import <wtf/HashSet.h>

#import <wtf/Vector.h>

inline bool protocolImplementsProtocol(Protocol *candidate, Protocol *target)

{

unsigned protocolProtocolsCount;

Protocol ** protocolProtocols = protocol_copyProtocolList(candidate, &protocolProtocolsCount);

for (unsigned i = 0; i < protocolProtocolsCount; ++i) {

if (protocol_isEqual(protocolProtocols[i], target)) {

free(protocolProtocols);

return true;

}

}

free(protocolProtocols);

return false;

}

inline void forEachProtocolImplementingProtocol(Class cls, Protocol *target, void (^callback)(Protocol *))

{

ASSERT(cls);

ASSERT(target);

Vector<Protocol *> worklist;

HashSet<void*> visited;

// Initially fill the worklist with the Class's protocols.

unsigned protocolsCount;

Protocol ** protocols = class_copyProtocolList(cls, &protocolsCount);

worklist.append(protocols, protocolsCount);

free(protocols);

while (!worklist.isEmpty()) {

Protocol *protocol = worklist.last();

worklist.removeLast();

// Are we encountering this Protocol for the first time?

if (!visited.add(protocol).isNewEntry)

continue;

// If it implements the protocol, make the callback.

if (protocolImplementsProtocol(protocol, target))

callback(protocol);

// Add incorporated protocols to the worklist.

protocols = protocol_copyProtocolList(protocol, &protocolsCount);

worklist.append(protocols, protocolsCount);

free(protocols);

}

}

inline void forEachMethodInClass(Class cls, void (^callback)(Method))

{

unsigned count;

Method* methods = class_copyMethodList(cls, &count);

for (unsigned i = 0; i < count; ++i)

callback(methods[i]);

free(methods);

}

inline void forEachMethodInProtocol(Protocol *protocol, BOOL isRequiredMethod, BOOL isInstanceMethod, void (^callback)(SEL, const char*))

{

unsigned count;

struct objc_method_description* methods = protocol_copyMethodDescriptionList(protocol, isRequiredMethod, isInstanceMethod, &count);

for (unsigned i = 0; i < count; ++i)

callback(methods[i].name, methods[i].types);

free(methods);

}

inline void forEachPropertyInProtocol(Protocol *protocol, void (^callback)(objc_property_t))

{

unsigned count;

objc_property_t* properties = protocol_copyPropertyList(protocol, &count);

for (unsigned i = 0; i < count; ++i)

callback(properties[i]);

free(properties);

}

template<char open, char close>

void skipPair(const char*& position)

{

size_t count = 1;

do {

char c = *position++;

if (!c)

@throw [NSException exceptionWithName:NSInternalInconsistencyException reason:@"Malformed type encoding" userInfo:nil];

if (c == open)

++count;

else if (c == close)

--count;

} while (count);

}

class StringRange {

WTF_MAKE_NONCOPYABLE(StringRange);

public:

StringRange(const char* begin, const char* end) : m_ptr(strndup(begin, end - begin)) { }

~StringRange() { free(m_ptr); }

operator const char*() const { return m_ptr; }

const char* get() const { return m_ptr; }

private:

char* m_ptr;

};

class StructBuffer {

WTF_MAKE_NONCOPYABLE(StructBuffer);

public:

StructBuffer(const char* encodedType)

{

NSUInteger size, alignment;

NSGetSizeAndAlignment(encodedType, &size, &alignment);

--alignment;

m_allocation = static_cast<char*>(malloc(size + alignment));

m_buffer = reinterpret_cast<char*>((reinterpret_cast<intptr_t>(m_allocation) + alignment) & ~alignment);

}

~StructBuffer() { free(m_allocation); }

operator void*() const { return m_buffer; }

private:

void* m_allocation;

void* m_buffer;

};

template<typename DelegateType>

typename DelegateType::ResultType parseObjCType(const char*& position)

{

ASSERT(*position);

switch (*position++) {

case 'c':

return DelegateType::template typeInteger<char>();

case 'i':

return DelegateType::template typeInteger<int>();

case 's':

return DelegateType::template typeInteger<short>();

case 'l':

return DelegateType::template typeInteger<long>();

case 'q':

return DelegateType::template typeDouble<unsigned long long>();

case 'C':

return DelegateType::template typeInteger<unsigned char>();

case 'I':

return DelegateType::template typeInteger<unsigned>();

case 'S':

return DelegateType::template typeInteger<unsigned short>();

case 'L':

return DelegateType::template typeInteger<unsigned long>();

case 'Q':

return DelegateType::template typeDouble<unsigned long long>();

case 'f':

return DelegateType::template typeDouble<float>();

case 'd':

return DelegateType::template typeDouble<double>();

case 'B':

return DelegateType::typeBool();

case 'v':

return DelegateType::typeVoid();

case '@': { // An object (whether statically typed or typed id)

if (position[0] == '?' && position[1] == '<') {

position += 2;

const char* begin = position;

skipPair<'<','>'>(position);

return DelegateType::typeBlock(begin, position - 1);

}

if (*position == '"') {

const char* begin = position + 1;

const char* protocolPosition = strchr(begin, '<');

const char* endOfType = strchr(begin, '"');

position = endOfType + 1;

// There's no protocol involved in this type, so just handle the class name.

if (!protocolPosition || protocolPosition > endOfType)

return DelegateType::typeOfClass(begin, endOfType);

// We skipped the class name and went straight to the protocol, so this is an id type.

if (begin == protocolPosition)

return DelegateType::typeId();

// We have a class name with a protocol. For now, ignore the protocol.

return DelegateType::typeOfClass(begin, protocolPosition);

}

return DelegateType::typeId();

}

case '{': { // {name=type...} A structure

const char* begin = position - 1;

skipPair<'{','}'>(position);

return DelegateType::typeStruct(begin, position);

}

// NOT supporting C strings, arrays, pointers, unions, bitfields, function pointers.

case '*': // A character string (char *)

case '[': // [array type] An array

case '(': // (name=type...) A union

case 'b': // bnum A bit field of num bits

case '^': // ^type A pointer to type

case '?': // An unknown type (among other things, this code is used for function pointers)

// NOT supporting Objective-C Class, SEL

case '#': // A class object (Class)

case ':': // A method selector (SEL)

default:

return nil;

}

}

extern "C" {

// Forward declare some Objective-C runtime internal methods that are not API.

const char *_protocol_getMethodTypeEncoding(Protocol *, SEL, BOOL isRequiredMethod, BOOL isInstanceMethod);

id objc_initWeak(id *, id);

void objc_destroyWeak(id *);

bool _Block_has_signature(void *);

const char * _Block_signature(void *);

}