X-Git-Url: https://git.saurik.com/cycript.git/blobdiff_plain/9cad30fab188a57c3db8df0912a1691099468e64..06293152b6505062453463ef34c0d90e0c8db17e:/Execute.cpp diff --git a/Execute.cpp b/Execute.cpp index 3135c2c..99dbda8 100644 --- a/Execute.cpp +++ b/Execute.cpp @@ -37,8 +37,6 @@ */ /* }}} */ -#include - #include "Internal.hpp" #include @@ -50,6 +48,7 @@ #include "sig/ffi_type.hpp" #include "Pooling.hpp" +#include "Execute.hpp" #include @@ -68,12 +67,6 @@ #include "JavaScript.hpp" #include "String.hpp" -char *sqlite3_column_pooled(apr_pool_t *pool, sqlite3_stmt *stmt, int n) { - if (const unsigned char *value = sqlite3_column_text(stmt, n)) - return apr_pstrdup(pool, (const char *) value); - else return NULL; -} - struct CYHooks *hooks_; /* JavaScript Properties {{{ */ @@ -172,28 +165,23 @@ JSStringRef push_s; JSStringRef splice_s; JSStringRef toCYON_s; JSStringRef toJSON_s; +JSStringRef toPointer_s; +JSStringRef toString_s; static JSStringRef Result_; -sqlite3 *Bridge_; - void CYFinalize(JSObjectRef object) { - delete reinterpret_cast(JSObjectGetPrivate(object)); + CYData *internal(reinterpret_cast(JSObjectGetPrivate(object))); + if (--internal->count_ == 0) + delete internal; } -struct CStringMapLess : - std::binary_function -{ - _finline bool operator ()(const char *lhs, const char *rhs) const { - return strcmp(lhs, rhs) < 0; - } -}; - void Structor_(apr_pool_t *pool, sig::Type *&type) { if ( type->primitive == sig::pointer_P && type->data.data.type != NULL && type->data.data.type->primitive == sig::struct_P && + type->data.data.type->name != NULL && strcmp(type->data.data.type->name, "_objc_class") == 0 ) { type->primitive = sig::typename_P; @@ -204,49 +192,27 @@ void Structor_(apr_pool_t *pool, sig::Type *&type) { if (type->primitive != sig::struct_P || type->name == NULL) return; - sqlite3_stmt *statement; - - _sqlcall(sqlite3_prepare(Bridge_, - "select " - "\"bridge\".\"mode\", " - "\"bridge\".\"value\" " - "from \"bridge\" " - "where" - " \"bridge\".\"mode\" in (3, 4) and" - " \"bridge\".\"name\" = ?" - " limit 1" - , -1, &statement, NULL)); - - _sqlcall(sqlite3_bind_text(statement, 1, type->name, -1, SQLITE_STATIC)); - - int mode; - const char *value; - - if (_sqlcall(sqlite3_step(statement)) == SQLITE_DONE) { - mode = -1; - value = NULL; - } else { - mode = sqlite3_column_int(statement, 0); - value = sqlite3_column_pooled(pool, statement, 1); - } - - _sqlcall(sqlite3_finalize(statement)); - - switch (mode) { - default: - _assert(false); - case -1: - break; - - case 3: { - sig::Parse(pool, &type->data.signature, value, &Structor_); - } break; - - case 4: { - sig::Signature signature; - sig::Parse(pool, &signature, value, &Structor_); - type = signature.elements[0].type; - } break; + size_t length(strlen(type->name)); + char keyed[length + 2]; + memcpy(keyed + 1, type->name, length + 1); + + static const char *modes = "34"; + for (size_t i(0); i != 2; ++i) { + char mode(modes[i]); + keyed[0] = mode; + + if (CYBridgeEntry *entry = CYBridgeHash(keyed, length + 1)) + switch (mode) { + case '3': + sig::Parse(pool, &type->data.signature, entry->value_, &Structor_); + break; + + case '4': { + sig::Signature signature; + sig::Parse(pool, &signature, entry->value_, &Structor_); + type = signature.elements[0].type; + } break; + } } } @@ -288,7 +254,7 @@ struct Struct_privateData : } }; -typedef std::map TypeMap; +typedef std::map TypeMap; static TypeMap Types_; JSObjectRef CYMakeStruct(JSContextRef context, void *data, sig::Type *type, ffi_type *ffi, JSObjectRef owner) { @@ -519,13 +485,37 @@ static JSValueRef Array_callAsFunction_toCYON(JSContextRef context, JSObjectRef return CYCastJSValue(context, CYJSString(CYUTF8String(value.c_str(), value.size()))); } CYCatch } +static JSValueRef String_callAsFunction_toCYON(JSContextRef context, JSObjectRef object, JSObjectRef _this, size_t count, const JSValueRef arguments[], JSValueRef *exception) { CYTry { + CYPool pool; + std::ostringstream str; + + CYUTF8String string(CYPoolUTF8String(pool, context, CYJSString(context, _this))); + CYStringify(str, string.data, string.size); + + std::string value(str.str()); + return CYCastJSValue(context, CYJSString(CYUTF8String(value.c_str(), value.size()))); +} CYCatch } + JSObjectRef CYMakePointer(JSContextRef context, void *pointer, size_t length, sig::Type *type, ffi_type *ffi, JSObjectRef owner) { Pointer *internal(new Pointer(pointer, context, owner, length, type)); return JSObjectMake(context, Pointer_, internal); } -static JSObjectRef CYMakeFunctor(JSContextRef context, void (*function)(), const char *type) { - cy::Functor *internal(new cy::Functor(type, function)); +static JSObjectRef CYMakeFunctor(JSContextRef context, void (*function)(), const char *type, void **cache = NULL) { + cy::Functor *internal; + + if (cache != NULL && *cache != NULL) { + internal = reinterpret_cast(*cache); + ++internal->count_; + } else { + internal = new cy::Functor(type, function); + + if (cache != NULL) { + *cache = internal; + ++internal->count_; + } + } + return JSObjectMake(context, Functor_, internal); } @@ -537,12 +527,19 @@ void *CYCastPointer_(JSContextRef context, JSValueRef value) { switch (JSValueGetType(context, value)) { case kJSTypeNull: return NULL; - /*case kJSTypeObject: + case kJSTypeObject: { + JSObjectRef object((JSObjectRef) value); if (JSValueIsObjectOfClass(context, value, Pointer_)) { - Pointer *internal(reinterpret_cast(JSObjectGetPrivate((JSObjectRef) value))); + Pointer *internal(reinterpret_cast(JSObjectGetPrivate(object))); return internal->value_; - }*/ - default: + } + JSValueRef toPointer(CYGetProperty(context, object, toPointer_s)); + if (CYIsCallable(context, toPointer)) { + JSValueRef value(CYCallAsFunction(context, (JSObjectRef) toPointer, object, 0, NULL)); + _assert(value != NULL); + return CYCastPointer_(context, value); + } + } default: double number(CYCastDouble(context, value)); if (std::isnan(number)) throw CYJSError(context, "cannot convert value to pointer"); @@ -976,61 +973,42 @@ static JSValueRef All_getProperty(JSContextRef context, JSObjectRef object, JSSt if (JSValueRef value = (*hooks_->RuntimeProperty)(context, name)) return value; - sqlite3_stmt *statement; - - _sqlcall(sqlite3_prepare(Bridge_, - "select " - "\"bridge\".\"mode\", " - "\"bridge\".\"value\" " - "from \"bridge\" " - "where" - " \"bridge\".\"name\" = ?" - " limit 1" - , -1, &statement, NULL)); - - _sqlcall(sqlite3_bind_text(statement, 1, name.data, name.size, SQLITE_STATIC)); - - int mode; - const char *value; - - if (_sqlcall(sqlite3_step(statement)) == SQLITE_DONE) { - mode = -1; - value = NULL; - } else { - mode = sqlite3_column_int(statement, 0); - value = sqlite3_column_pooled(pool, statement, 1); + size_t length(name.size); + char keyed[length + 2]; + memcpy(keyed + 1, name.data, length + 1); + + static const char *modes = "0124"; + for (size_t i(0); i != 4; ++i) { + char mode(modes[i]); + keyed[0] = mode; + + if (CYBridgeEntry *entry = CYBridgeHash(keyed, length + 1)) + switch (mode) { + case '0': + return JSEvaluateScript(CYGetJSContext(context), CYJSString(entry->value_), NULL, NULL, 0, NULL); + + case '1': + if (void (*symbol)() = reinterpret_cast(CYCastSymbol(name.data))) + return CYMakeFunctor(context, symbol, entry->value_, &entry->cache_); + else return NULL; + + case '2': + if (void *symbol = CYCastSymbol(name.data)) { + // XXX: this is horrendously inefficient + sig::Signature signature; + sig::Parse(pool, &signature, entry->value_, &Structor_); + ffi_cif cif; + sig::sig_ffi_cif(pool, &sig::ObjectiveC, &signature, &cif); + return CYFromFFI(context, signature.elements[0].type, cif.rtype, symbol); + } else return NULL; + + // XXX: implement case 3 + case '4': + return CYMakeType(context, entry->value_); + } } - _sqlcall(sqlite3_finalize(statement)); - - switch (mode) { - default: - CYThrow("invalid mode from bridge table: %d", mode); - case -1: - return NULL; - - case 0: - return JSEvaluateScript(CYGetJSContext(context), CYJSString(value), NULL, NULL, 0, NULL); - - case 1: - if (void (*symbol)() = reinterpret_cast(CYCastSymbol(name.data))) - return CYMakeFunctor(context, symbol, value); - else return NULL; - - case 2: - if (void *symbol = CYCastSymbol(name.data)) { - // XXX: this is horrendously inefficient - sig::Signature signature; - sig::Parse(pool, &signature, value, &Structor_); - ffi_cif cif; - sig::sig_ffi_cif(pool, &sig::ObjectiveC, &signature, &cif); - return CYFromFFI(context, signature.elements[0].type, cif.rtype, symbol); - } else return NULL; - - // XXX: implement case 3 - case 4: - return CYMakeType(context, value); - } + return NULL; } CYCatch } static JSObjectRef Pointer_new(JSContextRef context, JSObjectRef object, size_t count, const JSValueRef arguments[], JSValueRef *exception) { CYTry { @@ -1233,7 +1211,7 @@ JSObjectRef CYGetGlobalObject(JSContextRef context) { return JSContextGetGlobalObject(context); } -const char *CYExecute(apr_pool_t *pool, const char *code) { +const char *CYExecute(apr_pool_t *pool, CYUTF8String code) { JSContextRef context(CYGetJSContext()); JSValueRef exception(NULL), result; @@ -1279,10 +1257,14 @@ extern "C" void CydgetSetupContext(JSGlobalContextRef context) { CYSetupContext(context); } +static bool initialized_ = false; + void CYInitializeDynamic() { - CYInitializeStatic(); + if (!initialized_) + initialized_ = true; + else return; - _sqlcall(sqlite3_open("/usr/lib/libcycript.db", &Bridge_)); + CYInitializeStatic(); JSObjectMakeArray$ = reinterpret_cast(dlsym(RTLD_DEFAULT, "JSObjectMakeArray")); @@ -1332,6 +1314,7 @@ void CYInitializeDynamic() { Type_privateData::Class_ = JSClassCreate(&definition); definition = kJSClassDefinitionEmpty; + definition.className = "Global"; //definition.getProperty = &Global_getProperty; Global_ = JSClassCreate(&definition); @@ -1346,6 +1329,8 @@ void CYInitializeDynamic() { splice_s = JSStringCreateWithUTF8CString("splice"); toCYON_s = JSStringCreateWithUTF8CString("toCYON"); toJSON_s = JSStringCreateWithUTF8CString("toJSON"); + toPointer_s = JSStringCreateWithUTF8CString("toPointer"); + toString_s = JSStringCreateWithUTF8CString("toString"); Result_ = JSStringCreateWithUTF8CString("_"); @@ -1433,9 +1418,13 @@ extern "C" void CYSetupContext(JSGlobalContextRef context) { JSObjectRef String(CYCastJSObject(context, CYGetProperty(context, global, CYJSString("String")))); CYSetProperty(context, cy, CYJSString("String"), String); + + JSObjectRef String_prototype(CYCastJSObject(context, CYGetProperty(context, String, prototype_s))); + CYSetProperty(context, cy, CYJSString("String_prototype"), String_prototype); /* }}} */ CYSetProperty(context, Array_prototype, toCYON_s, &Array_callAsFunction_toCYON, kJSPropertyAttributeDontEnum); + CYSetProperty(context, String_prototype, toCYON_s, &String_callAsFunction_toCYON, kJSPropertyAttributeDontEnum); JSObjectRef cycript(JSObjectMake(context, NULL, NULL)); CYSetProperty(context, global, CYJSString("Cycript"), cycript); @@ -1451,23 +1440,25 @@ extern "C" void CYSetupContext(JSGlobalContextRef context) { JSObjectRef all(JSObjectMake(context, All_, NULL)); CYSetProperty(context, cycript, CYJSString("all"), all); - JSObjectRef last(NULL), curr(global); + if (true) { + JSObjectRef last(NULL), curr(global); - goto next; for (JSValueRef next;;) { - if (JSValueIsNull(context, next)) - break; - last = curr; - curr = CYCastJSObject(context, next); - next: - next = JSObjectGetPrototype(context, curr); - } + goto next; for (JSValueRef next;;) { + if (JSValueIsNull(context, next)) + break; + last = curr; + curr = CYCastJSObject(context, next); + next: + next = JSObjectGetPrototype(context, curr); + } - JSObjectSetPrototype(context, last, all); + JSObjectSetPrototype(context, last, all); + } - CYSetProperty(context, global, CYJSString("$cyq"), &$cyq); + CYSetProperty(context, global, CYJSString("$cyq"), &$cyq, kJSPropertyAttributeDontEnum); JSObjectRef System(JSObjectMake(context, NULL, NULL)); - CYSetProperty(context, cy, CYJSString("System"), Function); + CYSetProperty(context, cy, CYJSString("System"), System); CYSetProperty(context, global, CYJSString("system"), System); CYSetProperty(context, System, CYJSString("args"), CYJSNull(context));