Don't crash if ivar_getTypeEncoding returns NULL.

[cycript.git] / Internal.hpp

diff --git a/Internal.hpp b/Internal.hpp
index eaf7150e4d7edc537892a256e0d375423f368058..cf357086deaad9f5c2c875f5c02a8244dd9a8cfc 100644 (file)
--- a/Internal.hpp
+++ b/Internal.hpp
@@ -1,15 +1,44 @@
+/* Cycript - Optimizing JavaScript Compiler/Runtime
+ * Copyright (C) 2009-2013  Jay Freeman (saurik)
+*/
+
+/* GNU General Public License, Version 3 {{{ */
+/*
+ * Cycript is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published
+ * by the Free Software Foundation, either version 3 of the License,
+ * or (at your option) any later version.
+ *
+ * Cycript is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Cycript.  If not, see <http://www.gnu.org/licenses/>.
+**/
+/* }}} */
+
 #ifndef CYCRIPT_INTERNAL_HPP
 #define CYCRIPT_INTERNAL_HPP
 
-#include "Pooling.hpp"
+#include <sig/parse.hpp>
+#include <sig/ffi_type.hpp>
 
 #include <JavaScriptCore/JSBase.h>
+#include <JavaScriptCore/JSContextRef.h>
+#include <JavaScriptCore/JSObjectRef.h>
 #include <JavaScriptCore/JSValueRef.h>
 
-#include <sig/parse.hpp>
-#include <sig/ffi_type.hpp>
+#include "Pooling.hpp"
+
+JSGlobalContextRef CYGetJSContext(JSContextRef context);
+void Structor_(CYPool &pool, sig::Type *&type);
+
+JSObjectRef CYMakeType(JSContextRef context, const char *type);
+JSObjectRef CYMakeType(JSContextRef context, sig::Type *type);
 
-void Structor_(apr_pool_t *pool, const char *name, const char *types, sig::Type *&type);
+extern JSClassRef Functor_;
 
 struct Type_privateData :
     CYData
@@ -20,38 +49,43 @@ struct Type_privateData :
     sig::Type *type_;
 
     void Set(sig::Type *type) {
-        type_ = new(pool_) sig::Type;
-        sig::Copy(pool_, *type_, *type);
+        type_ = new(*pool_) sig::Type;
+        sig::Copy(*pool_, *type_, *type);
     }
 
-    Type_privateData(apr_pool_t *pool, const char *type) :
+    Type_privateData(CYPool &pool, const char *type) :
+        CYData(pool),
         ffi_(NULL)
     {
-        if (pool != NULL)
-            pool_ = pool;
+        sig::Signature signature;
+        sig::Parse(*pool_, &signature, type, &Structor_);
+        type_ = signature.elements[0].type;
+    }
 
+    Type_privateData(const char *type) :
+        ffi_(NULL)
+    {
         sig::Signature signature;
-        sig::Parse(pool_, &signature, type, &Structor_);
+        sig::Parse(*pool_, &signature, type, &Structor_);
         type_ = signature.elements[0].type;
     }
 
     Type_privateData(sig::Type *type) :
         ffi_(NULL)
     {
-        if (type != NULL)
-            Set(type);
+        // XXX: just in case I messed up migrating
+        _assert(type != NULL);
+        Set(type);
     }
 
     Type_privateData(sig::Type *type, ffi_type *ffi) {
-        ffi_ = new(pool_) ffi_type;
-        sig::Copy(pool_, *ffi_, *ffi);
+        ffi_ = new(*pool_) ffi_type;
+        sig::Copy(*pool_, *ffi_, *ffi);
         Set(type);
     }
 
     ffi_type *GetFFI() {
         if (ffi_ == NULL) {
-            ffi_ = new(pool_) ffi_type;
-
             sig::Element element;
             element.name = NULL;
             element.type = type_;
@@ -62,7 +96,9 @@ struct Type_privateData :
             signature.count = 1;
 
             ffi_cif cif;
-            sig::sig_ffi_cif(pool_, &sig::ObjectiveC, &signature, &cif);
+            sig::sig_ffi_cif(*pool_, &sig::ObjectiveC, &signature, &cif);
+
+            ffi_ = new(*pool_) ffi_type;
             *ffi_ = *cif.rtype;
         }
 
@@ -97,15 +133,16 @@ struct CYOwned :
     CYValue
 {
   private:
-    JSContextRef context_;
+    JSGlobalContextRef context_;
     JSObjectRef owner_;
 
   public:
     CYOwned(void *value, JSContextRef context, JSObjectRef owner) :
         CYValue(value),
-        context_(context),
+        context_(CYGetJSContext(context)),
         owner_(owner)
     {
+        //XXX:JSGlobalContextRetain(context_);
         if (owner_ != NULL)
             JSValueProtect(context_, owner_);
     }
@@ -113,6 +150,7 @@ struct CYOwned :
     virtual ~CYOwned() {
         if (owner_ != NULL)
             JSValueUnprotect(context_, owner_);
+        //XXX:JSGlobalContextRelease(context_);
     }
 
     JSObjectRef GetOwner() const {
@@ -120,4 +158,63 @@ struct CYOwned :
     }
 };
 
+namespace cy {
+struct Functor :
+    CYValue
+{
+  private:
+    void set() {
+        sig::sig_ffi_cif(*pool_, &sig::ObjectiveC, &signature_, &cif_);
+    }
+
+  public:
+    sig::Signature signature_;
+    ffi_cif cif_;
+
+    Functor(const sig::Signature &signature, void (*value)()) :
+        CYValue(reinterpret_cast<void *>(value))
+    {
+        sig::Copy(*pool_, signature_, signature);
+        set();
+    }
+
+    Functor(const char *encoding, void (*value)()) :
+        CYValue(reinterpret_cast<void *>(value))
+    {
+        sig::Parse(*pool_, &signature_, encoding, &Structor_);
+        set();
+    }
+
+    void (*GetValue() const)() {
+        return reinterpret_cast<void (*)()>(value_);
+    }
+
+    static JSStaticFunction const * const StaticFunctions;
+    static JSStaticValue const * const StaticValues;
+}; }
+
+struct Closure_privateData :
+    cy::Functor
+{
+    JSGlobalContextRef context_;
+    JSObjectRef function_;
+
+    Closure_privateData(JSContextRef context, JSObjectRef function, const sig::Signature &signature) :
+        cy::Functor(signature, NULL),
+        context_(CYGetJSContext(context)),
+        function_(function)
+    {
+        //XXX:JSGlobalContextRetain(context_);
+        JSValueProtect(context_, function_);
+    }
+
+    virtual ~Closure_privateData() {
+        JSValueUnprotect(context_, function_);
+        //XXX:JSGlobalContextRelease(context_);
+    }
+};
+
+Closure_privateData *CYMakeFunctor_(JSContextRef context, JSObjectRef function, const sig::Signature &signature, void (*callback)(ffi_cif *, void *, void **, void *));
+void CYExecuteClosure(ffi_cif *cif, void *result, void **arguments, void *arg, JSValueRef (*adapter)(JSContextRef, size_t, JSValueRef[], JSObjectRef));
+
 #endif/*CYCRIPT_INTERNAL_HPP*/