/* Cycript - Optimizing JavaScript Compiler/Runtime
- * Copyright (C) 2009-2010 Jay Freeman (saurik)
+ * Copyright (C) 2009-2012 Jay Freeman (saurik)
*/
/* GNU Lesser General Public License, Version 3 {{{ */
#include <cmath>
#include <map>
+#include <dlfcn.h>
+
#define CYObjectiveTry_(context) { \
JSContextRef context_(context); \
try
#define protocol_getName(protocol) [(protocol) name]
#endif
+static void (*$objc_setAssociatedObject)(id object, void *key, id value, objc_AssociationPolicy policy);
+static id (*$objc_getAssociatedObject)(id object, void *key);
+static void (*$objc_removeAssociatedObjects)(id object);
+
JSValueRef CYSendMessage(apr_pool_t *pool, JSContextRef context, id self, Class super, SEL _cmd, size_t count, const JSValueRef arguments[], bool initialize, JSValueRef *exception);
/* Objective-C Pool Release {{{ */
}
static JSClassRef Instance_;
+
static JSClassRef ArrayInstance_;
+static JSClassRef FunctionInstance_;
static JSClassRef ObjectInstance_;
static JSClassRef StringInstance_;
-static JSClassRef *Instances_[] = {
- &Instance_,
- &ArrayInstance_,
- &ObjectInstance_,
- &StringInstance_,
-};
-
static JSClassRef Internal_;
static JSClassRef Message_;
static JSClassRef Messages_;
#endif
static Class NSArray_;
+static Class NSBlock_;
static Class NSDictionary_;
static Class NSString_;
static Class Object_;
static Type_privateData *Object_type;
static Type_privateData *Selector_type;
-static bool CYValueIsObjectOfClassInstance(JSContextRef context, JSValueRef value) {
- for (size_t i(0); i != sizeof(Instances_) / sizeof(Instances_[0]); ++i)
- if (JSValueIsObjectOfClass(context, value, *Instances_[i]))
- return true;
- return false;
-}
-
Type_privateData *Instance::GetType() const {
return Object_type;
}
if (self == NSArray_)
prototype = CYGetCachedObject(context, CYJSString("ArrayInstance_prototype"));
+ else if (self == NSBlock_)
+ prototype = CYGetCachedObject(context, CYJSString("FunctionInstance_prototype"));
else if (self == NSDictionary_)
prototype = CYGetCachedObject(context, CYJSString("ObjectInstance_prototype"));
else if (self == NSString_)
@interface NSObject (Cycript)
-- (JSValueRef) cy$JSValueInContext:(JSContextRef)context;
+- (JSValueRef) cy$valueOfInContext:(JSContextRef)context;
- (JSType) cy$JSType;
- (NSObject *) cy$toJSON:(NSString *)key;
@protocol Cycript
- (id) cy$box;
-- (JSValueRef) cy$JSValueInContext:(JSContextRef)context;
+- (JSValueRef) cy$valueOfInContext:(JSContextRef)context;
@end
NSString *CYCastNSCYON(id value) {
}
NSObject *CYCastNSObject(apr_pool_t *pool, JSContextRef context, JSObjectRef object) {
- if (!CYValueIsObjectOfClassInstance(context, object))
+ if (!JSValueIsObjectOfClass(context, object, Instance_))
return CYCastNSObject_(pool, context, object);
else {
Instance *internal(reinterpret_cast<Instance *>(JSObjectGetPrivate(object)));
@end
/* }}} */
+/* Bridge: NSBlock {{{ */
+#ifdef __APPLE__
+@interface NSBlock
+- (void) invoke;
+@end
+#endif
+/* }}} */
/* Bridge: NSBoolNumber {{{ */
#ifndef __APPLE__
@implementation NSBoolNumber (Cycript)
return [self boolValue] ? @"@true" : @"@false";
}
-- (JSValueRef) cy$JSValueInContext:(JSContextRef)context { CYObjectiveTry_(context) {
+- (JSValueRef) cy$valueOfInContext:(JSContextRef)context { CYObjectiveTry_(context) {
return CYCastJSValue(context, (bool) [self boolValue]);
} CYObjectiveCatch }
return [self cy$JSType] != kJSTypeBoolean ? [NSString stringWithFormat:@"@%@", self] : [self boolValue] ? @"@true" : @"@false";
}
-- (JSValueRef) cy$JSValueInContext:(JSContextRef)context { CYObjectiveTry_(context) {
+- (JSValueRef) cy$valueOfInContext:(JSContextRef)context { CYObjectiveTry_(context) {
return [self cy$JSType] != kJSTypeBoolean ? CYCastJSValue(context, [self doubleValue]) : CYCastJSValue(context, [self boolValue]);
} CYObjectiveCatch }
return self;
}
-- (JSValueRef) cy$JSValueInContext:(JSContextRef)context { CYObjectiveTry_(context) {
+- (JSValueRef) cy$valueOfInContext:(JSContextRef)context { CYObjectiveTry_(context) {
return NULL;
} CYObjectiveCatch }
return false;
}
-- (JSValueRef) cy$JSValueInContext:(JSContextRef)context { CYObjectiveTry_(context) {
+- (JSValueRef) cy$valueOfInContext:(JSContextRef)context { CYObjectiveTry_(context) {
return CYCastJSValue(context, CYJSString(context, self));
} CYObjectiveCatch }
return @"undefined";
}
-- (JSValueRef) cy$JSValueInContext:(JSContextRef)context { CYObjectiveTry_(context) {
+- (JSValueRef) cy$valueOfInContext:(JSContextRef)context { CYObjectiveTry_(context) {
return CYJSUndefined(context);
} CYObjectiveCatch }
@end
-// XXX: use objc_getAssociatedObject and objc_setAssociatedObject on 10.6
-struct CYInternal :
- CYData
-{
+// XXX: inherit from or replace with CYJSObject
+@interface CYInternal : NSObject {
+ JSContextRef context_;
JSObjectRef object_;
+}
- CYInternal() :
- object_(NULL)
- {
- }
+@end
- ~CYInternal() {
- // XXX: delete object_? ;(
- }
+@implementation CYInternal
- static CYInternal *Get(id self) {
- CYInternal *internal(NULL);
- if (object_getInstanceVariable(self, "cy$internal_", reinterpret_cast<void **>(&internal)) == NULL) {
- // XXX: do something epic? ;P
- }
+- (void) dealloc {
+ JSValueUnprotect(context_, object_);
+ //XXX:JSGlobalContextRelease(context_);
+ [super dealloc];
+}
- return internal;
- }
+- (id) initInContext:(JSContextRef)context {
+ if ((self = [super init]) != nil) {
+ context_ = CYGetJSContext(context);
+ //XXX:JSGlobalContextRetain(context_);
+ } return self;
+}
- static CYInternal *Set(id self) {
- CYInternal *internal(NULL);
- if (objc_ivar *ivar = object_getInstanceVariable(self, "cy$internal_", reinterpret_cast<void **>(&internal))) {
- if (internal == NULL) {
- internal = new CYInternal();
- object_setIvar(self, ivar, reinterpret_cast<id>(internal));
- }
- } else {
- // XXX: do something epic? ;P
- }
+- (bool) hasProperty:(JSStringRef)name inContext:(JSContextRef)context {
+ if (object_ == NULL)
+ return false;
- return internal;
- }
+ return JSObjectHasProperty(context, object_, name);
+}
- bool HasProperty(JSContextRef context, JSStringRef name) {
- if (object_ == NULL)
- return false;
- return JSObjectHasProperty(context, object_, name);
+- (JSValueRef) getProperty:(JSStringRef)name inContext:(JSContextRef)context {
+ if (object_ == NULL)
+ return NULL;
+
+ return CYGetProperty(context, object_, name);
+}
+
+- (void) setProperty:(JSStringRef)name toValue:(JSValueRef)value inContext:(JSContextRef)context {
+ @synchronized (self) {
+ if (object_ == NULL) {
+ object_ = JSObjectMake(context, NULL, NULL);
+ JSValueProtect(context, object_);
+ }
}
- JSValueRef GetProperty(JSContextRef context, JSStringRef name) {
- if (object_ == NULL)
- return NULL;
- return CYGetProperty(context, object_, name);
+ CYSetProperty(context, object_, name, value);
+}
+
++ (CYInternal *) get:(id)object {
+ if ($objc_getAssociatedObject == NULL)
+ return nil;
+
+ @synchronized (object) {
+ if (CYInternal *internal = $objc_getAssociatedObject(object, @selector(cy$internal)))
+ return internal;
}
- void SetProperty(JSContextRef context, JSStringRef name, JSValueRef value) {
- if (object_ == NULL)
- object_ = JSObjectMake(context, NULL, NULL);
- CYSetProperty(context, object_, name, value);
+ return nil;
+}
+
++ (CYInternal *) set:(id)object inContext:(JSContextRef)context {
+ if ($objc_getAssociatedObject == NULL)
+ return nil;
+
+ @synchronized (object) {
+ if (CYInternal *internal = $objc_getAssociatedObject(object, @selector(cy$internal)))
+ return internal;
+
+ if ($objc_setAssociatedObject == NULL)
+ return nil;
+
+ CYInternal *internal([[[CYInternal alloc] initInContext:context] autorelease]);
+ objc_setAssociatedObject(object, @selector(cy$internal), internal, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
+ return internal;
}
-};
+
+ return nil;
+}
+
+@end
static JSObjectRef CYMakeSelector(JSContextRef context, SEL sel) {
Selector_privateData *internal(new Selector_privateData(sel));
CYPool pool;
NSString *name(CYCastNSString(pool, context, property));
- if (CYInternal *internal = CYInternal::Get(self))
- if (internal->HasProperty(context, property))
+ if (CYInternal *internal = [CYInternal get:self])
+ if ([internal hasProperty:property inContext:context])
return true;
Class _class(object_getClass(self));
CYPool pool;
NSString *name(CYCastNSString(pool, context, property));
- if (CYInternal *internal = CYInternal::Get(self))
- if (JSValueRef value = internal->GetProperty(context, property))
+ if (CYInternal *internal = [CYInternal get:self])
+ if (JSValueRef value = [internal getProperty:property inContext:context])
return value;
CYPoolTry {
CYSendMessage(pool, context, self, NULL, sel, 1, arguments, false, exception);
}
- if (CYInternal *internal = CYInternal::Set(self)) {
- internal->SetProperty(context, property, value);
+ if (CYInternal *internal = [CYInternal set:self inContext:context]) {
+ [internal setProperty:property toValue:value inContext:context];
return true;
}
return value;
} CYCatch }
+static JSValueRef Instance_callAsFunction(JSContextRef context, JSObjectRef object, JSObjectRef _this, size_t count, const JSValueRef arguments[], JSValueRef *exception) { CYTry {
+ Instance *internal(reinterpret_cast<Instance *>(JSObjectGetPrivate(object)));
+ id self(internal->GetValue());
+
+ if (![self isKindOfClass:NSBlock_])
+ CYThrow("non-NSBlock object is not a function");
+
+ struct BlockDescriptor1 {
+ unsigned long int reserved;
+ unsigned long int size;
+ };
+
+ struct BlockDescriptor2 {
+ void (*copy_helper)(void *dst, void *src);
+ void (*dispose_helper)(void *src);
+ };
+
+ struct BlockDescriptor3 {
+ const char *signature;
+ const char *layout;
+ };
+
+ struct BlockLiteral {
+ Class isa;
+ int flags;
+ int reserved;
+ void (*invoke)(void *, ...);
+ void *descriptor;
+ } *literal = reinterpret_cast<BlockLiteral *>(self);
+
+ enum {
+ BLOCK_DEALLOCATING = 0x0001,
+ BLOCK_REFCOUNT_MASK = 0xfffe,
+ BLOCK_NEEDS_FREE = 1 << 24,
+ BLOCK_HAS_COPY_DISPOSE = 1 << 25,
+ BLOCK_HAS_CTOR = 1 << 26,
+ BLOCK_IS_GC = 1 << 27,
+ BLOCK_IS_GLOBAL = 1 << 28,
+ BLOCK_HAS_STRET = 1 << 29,
+ BLOCK_HAS_SIGNATURE = 1 << 30,
+ };
+
+ if ((literal->flags & BLOCK_HAS_SIGNATURE) != 0) {
+ uint8_t *descriptor(reinterpret_cast<uint8_t *>(literal->descriptor));
+ descriptor += sizeof(BlockDescriptor1);
+ if ((literal->flags & BLOCK_HAS_COPY_DISPOSE) != 0)
+ descriptor += sizeof(BlockDescriptor2);
+ BlockDescriptor3 *descriptor3(reinterpret_cast<BlockDescriptor3 *>(descriptor));
+
+ if (const char *type = descriptor3->signature) {
+ CYPool pool;
+
+ void *setup[1];
+ setup[0] = &self;
+
+ sig::Signature signature;
+ sig::Parse(pool, &signature, type, &Structor_);
+
+ ffi_cif cif;
+ sig::sig_ffi_cif(pool, &sig::ObjectiveC, &signature, &cif);
+
+ void (*function)() = reinterpret_cast<void (*)()>(literal->invoke);
+ return CYCallFunction(pool, context, 1, setup, count, arguments, false, exception, &signature, &cif, function);
+ }
+ }
+
+ if (count != 0)
+ CYThrow("NSBlock without signature field passed arguments");
+
+ CYPoolTry {
+ [self invoke];
+ } CYPoolCatch(NULL);
+
+ return NULL;
+} CYCatch }
+
static bool Instance_hasInstance(JSContextRef context, JSObjectRef constructor, JSValueRef instance, JSValueRef *exception) { CYTry {
Instance *internal(reinterpret_cast<Instance *>(JSObjectGetPrivate((JSObjectRef) constructor)));
Class _class(internal->GetValue());
if (!CYIsClass(_class))
return false;
- if (CYValueIsObjectOfClassInstance(context, instance)) {
+ if (JSValueIsObjectOfClass(context, instance, Instance_)) {
Instance *linternal(reinterpret_cast<Instance *>(JSObjectGetPrivate((JSObjectRef) instance)));
// XXX: this isn't always safe
return [linternal->GetValue() isKindOfClass:_class];
self = internal->GetValue();
_class = internal->class_;;
uninitialized = false;
- } else if (CYValueIsObjectOfClassInstance(context, arguments[0])) {
+ } else if (JSValueIsObjectOfClass(context, arguments[0], Instance_)) {
Instance *internal(reinterpret_cast<Instance *>(JSObjectGetPrivate((JSObjectRef) arguments[0])));
self = internal->GetValue();
_class = nil;
return CYSendMessage(pool, context, self, _class, _cmd, count - 2, arguments + 2, uninitialized, exception);
} CYCatch }
-/* Hook: objc_registerClassPair {{{ */
-#if defined(__APPLE__) && defined(__arm__) && 0
-// XXX: replace this with associated objects
-
-MSHook(void, CYDealloc, id self, SEL sel) {
- CYInternal *internal;
- object_getInstanceVariable(self, "cy$internal_", reinterpret_cast<void **>(&internal));
- if (internal != NULL)
- delete internal;
- _CYDealloc(self, sel);
-}
-
-MSHook(void, objc_registerClassPair, Class _class) {
- Class super(class_getSuperclass(_class));
- if (super == NULL || class_getInstanceVariable(super, "cy$internal_") == NULL) {
- class_addIvar(_class, "cy$internal_", sizeof(CYInternal *), log2(sizeof(CYInternal *)), "^{CYInternal}");
- MSHookMessage(_class, @selector(dealloc), MSHake(CYDealloc));
- }
-
- _objc_registerClassPair(_class);
-}
-
-static JSValueRef objc_registerClassPair_(JSContextRef context, JSObjectRef object, JSObjectRef _this, size_t count, const JSValueRef arguments[], JSValueRef *exception) { CYTry {
- if (count != 1)
- throw CYJSError(context, "incorrect number of arguments to objc_registerClassPair");
- CYPool pool;
- NSObject *value(CYCastNSObject(pool, context, arguments[0]));
- if (value == NULL || !CYIsClass(value))
- throw CYJSError(context, "incorrect number of arguments to objc_registerClassPair");
- Class _class((Class) value);
- $objc_registerClassPair(_class);
- return CYJSUndefined(context);
-} CYCatch }
-#endif
-/* }}} */
-
static JSValueRef Selector_callAsFunction(JSContextRef context, JSObjectRef object, JSObjectRef _this, size_t count, const JSValueRef arguments[], JSValueRef *exception) {
JSValueRef setup[count + 2];
setup[0] = _this;
}
static JSValueRef Instance_callAsFunction_toCYON(JSContextRef context, JSObjectRef object, JSObjectRef _this, size_t count, const JSValueRef arguments[], JSValueRef *exception) { CYTry {
- if (!CYValueIsObjectOfClassInstance(context, _this))
+ if (!JSValueIsObjectOfClass(context, _this, Instance_))
return NULL;
Instance *internal(reinterpret_cast<Instance *>(JSObjectGetPrivate(_this)));
} CYCatch }
static JSValueRef Instance_callAsFunction_toJSON(JSContextRef context, JSObjectRef object, JSObjectRef _this, size_t count, const JSValueRef arguments[], JSValueRef *exception) { CYTry {
- if (!CYValueIsObjectOfClassInstance(context, _this))
+ if (!JSValueIsObjectOfClass(context, _this, Instance_))
return NULL;
Instance *internal(reinterpret_cast<Instance *>(JSObjectGetPrivate(_this)));
} CYCatch return /*XXX*/ NULL; }
static JSValueRef Instance_callAsFunction_valueOf(JSContextRef context, JSObjectRef object, JSObjectRef _this, size_t count, const JSValueRef arguments[], JSValueRef *exception) { CYTry {
- if (!CYValueIsObjectOfClassInstance(context, _this))
+ if (!JSValueIsObjectOfClass(context, _this, Instance_))
return NULL;
Instance *internal(reinterpret_cast<Instance *>(JSObjectGetPrivate(_this)));
id value(internal->GetValue());
- if (![value respondsToSelector:@selector(cy$JSValueInContext:)])
+ if (![value respondsToSelector:@selector(cy$valueOfInContext:)])
return _this;
- if (JSValueRef result = [value cy$JSValueInContext:context])
+ if (JSValueRef result = [value cy$valueOfInContext:context])
return result;
return _this;
} CYCatch return /*XXX*/ NULL; }
static JSValueRef Instance_callAsFunction_toPointer(JSContextRef context, JSObjectRef object, JSObjectRef _this, size_t count, const JSValueRef arguments[], JSValueRef *exception) { CYTry {
- if (!CYValueIsObjectOfClassInstance(context, _this))
+ if (!JSValueIsObjectOfClass(context, _this, Instance_))
return NULL;
Instance *internal(reinterpret_cast<Instance *>(JSObjectGetPrivate(_this)));
} CYCatch return /*XXX*/ NULL; }
static JSValueRef Instance_callAsFunction_toString(JSContextRef context, JSObjectRef object, JSObjectRef _this, size_t count, const JSValueRef arguments[], JSValueRef *exception) { CYTry {
- if (!CYValueIsObjectOfClassInstance(context, _this))
+ if (!JSValueIsObjectOfClass(context, _this, Instance_))
return NULL;
Instance *internal(reinterpret_cast<Instance *>(JSObjectGetPrivate(_this)));
};
void CYObjectiveC_Initialize() { /*XXX*/ JSContextRef context(NULL); CYPoolTry {
+ $objc_setAssociatedObject = reinterpret_cast<void (*)(id, void *, id value, objc_AssociationPolicy)>(dlsym(RTLD_DEFAULT, "objc_setAssociatedObject"));
+ $objc_getAssociatedObject = reinterpret_cast<id (*)(id, void *)>(dlsym(RTLD_DEFAULT, "objc_getAssociatedObject"));
+ $objc_removeAssociatedObjects = reinterpret_cast<void (*)(id)>(dlsym(RTLD_DEFAULT, "objc_removeAssociatedObjects"));
+
apr_pool_t *pool(CYGetGlobalPool());
Object_type = new(pool) Type_privateData("@");
#ifdef __APPLE__
NSCFBoolean_ = objc_getClass("NSCFBoolean");
+ if (NSCFBoolean_ == nil)
+ NSCFBoolean_ = objc_getClass("__NSCFBoolean");
+
NSCFType_ = objc_getClass("NSCFType");
NSGenericDeallocHandler_ = objc_getClass("__NSGenericDeallocHandler");
NSMessageBuilder_ = objc_getClass("NSMessageBuilder");
#endif
NSArray_ = objc_getClass("NSArray");
+ NSBlock_ = objc_getClass("NSBlock");
NSDictionary_ = objc_getClass("NSDictionary");
NSString_ = objc_getClass("NSString");
Object_ = objc_getClass("Object");
definition.deleteProperty = &Instance_deleteProperty;
definition.getPropertyNames = &Instance_getPropertyNames;
definition.callAsConstructor = &Instance_callAsConstructor;
+ definition.callAsFunction = &Instance_callAsFunction;
definition.hasInstance = &Instance_hasInstance;
definition.finalize = &CYFinalize;
Instance_ = JSClassCreate(&definition);
definition.className = "ArrayInstance";
ArrayInstance_ = JSClassCreate(&definition);
+ definition.className = "FunctionInstance";
+ FunctionInstance_ = JSClassCreate(&definition);
+
definition.className = "ObjectInstance";
ObjectInstance_ = JSClassCreate(&definition);
definition.getPropertyNames = &ObjectiveC_Protocols_getPropertyNames;
ObjectiveC_Protocols_ = JSClassCreate(&definition);
-#if defined(__APPLE__) && defined(__arm__) && 0
- MSHookFunction(&objc_registerClassPair, MSHake(objc_registerClassPair));
-#endif
-
#ifdef __APPLE__
class_addMethod(NSCFType_, @selector(cy$toJSON:), reinterpret_cast<IMP>(&NSCFType$cy$toJSON), "@12@0:4@8");
#endif
JSObjectRef Array_prototype(CYGetCachedObject(context, CYJSString("Array_prototype")));
JSObjectSetPrototype(context, ArrayInstance_prototype, Array_prototype);
+ JSObjectRef FunctionInstance(JSObjectMakeConstructor(context, FunctionInstance_, NULL));
+ JSObjectRef FunctionInstance_prototype(CYCastJSObject(context, CYGetProperty(context, FunctionInstance, prototype_s)));
+ CYSetProperty(context, cy, CYJSString("FunctionInstance_prototype"), FunctionInstance_prototype);
+ JSObjectRef Function_prototype(CYGetCachedObject(context, CYJSString("Function_prototype")));
+ JSObjectSetPrototype(context, FunctionInstance_prototype, Function_prototype);
+
JSObjectRef ObjectInstance(JSObjectMakeConstructor(context, ObjectInstance_, NULL));
JSObjectRef ObjectInstance_prototype(CYCastJSObject(context, CYGetProperty(context, ObjectInstance, prototype_s)));
CYSetProperty(context, cy, CYJSString("ObjectInstance_prototype"), ObjectInstance_prototype);
CYSetProperty(context, all, CYJSString("objc_msgSend"), &$objc_msgSend, kJSPropertyAttributeDontEnum);
- JSObjectRef Function_prototype(CYGetCachedObject(context, CYJSString("Function_prototype")));
JSObjectSetPrototype(context, CYCastJSObject(context, CYGetProperty(context, Message, prototype_s)), Function_prototype);
JSObjectSetPrototype(context, CYCastJSObject(context, CYGetProperty(context, Selector, prototype_s)), Function_prototype);
} CYPoolCatch() }
projects / cycript.git / blobdiff