MIME type is now required when calling appcast.sh.

[cycript.git] / Pooling.hpp

diff --git a/Pooling.hpp b/Pooling.hpp
index 24c7cc25af044776623638f1dd9efc36650c1c84..1c155fee5318a7f3a0f4082dcee16aeb1ec29399 100644 (file)
--- a/Pooling.hpp
+++ b/Pooling.hpp
@@ -1,21 +1,21 @@
-/* Cycript - Optimizing JavaScript Compiler/Runtime
- * Copyright (C) 2009-2013  Jay Freeman (saurik)
+/* Cycript - The Truly Universal Scripting Language
+ * Copyright (C) 2009-2016  Jay Freeman (saurik)

 */
 
 */
 

-/* GNU General Public License, Version 3 {{{ */
+/* GNU Affero 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
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+
+ * This program 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
  * 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/>.
+ * GNU Affero General Public License for more details.
+
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.

 **/
 /* }}} */
 
 **/
 /* }}} */
 


@@ -35,6 +35,14 @@
 #include "Local.hpp"
 #include "Standard.hpp"
 
 #include "Local.hpp"
 #include "Standard.hpp"
 

+// XXX: std::aligned_storage and alignof
+static const size_t CYAlignment(sizeof(void *));
+
+template <typename Type_>
+static void CYAlign(Type_ &data, size_t size) {
+    data = reinterpret_cast<Type_>((reinterpret_cast<uintptr_t>(data) + (size - 1)) & ~static_cast<uintptr_t>(size - 1));
+}
+

 class CYPool;
 _finline void *operator new(size_t size, CYPool &pool);
 _finline void *operator new [](size_t size, CYPool &pool);
 class CYPool;
 _finline void *operator new(size_t size, CYPool &pool);
 _finline void *operator new [](size_t size, CYPool &pool);


@@ -43,6 +51,7 @@ class CYPool {
   private:
     uint8_t *data_;
     size_t size_;
   private:
     uint8_t *data_;
     size_t size_;

+    size_t next_;

 
     struct Cleaner {
         Cleaner *next_;
 
     struct Cleaner {
         Cleaner *next_;


@@ -57,17 +66,18 @@ class CYPool {
         }
     } *cleaner_;
 
         }
     } *cleaner_;
 

-    static _finline size_t align(size_t size) {
-        // XXX: alignment is more complex than this
-        return (size + 7) & ~0x3;
+    template <typename Type_>
+    static void delete_(void *data) {
+        reinterpret_cast<Type_ *>(data)->~Type_();

     }
 
     CYPool(const CYPool &);
 
   public:
     }
 
     CYPool(const CYPool &);
 
   public:

-    CYPool() :
+    CYPool(size_t next = 64) :

         data_(NULL),
         size_(0),
         data_(NULL),
         size_(0),

+        next_(next),

         cleaner_(NULL)
     {
     }
         cleaner_(NULL)
     {
     }


@@ -81,41 +91,56 @@ class CYPool {
     }
 
     template <typename Type_>
     }
 
     template <typename Type_>

-    Type_ *malloc(size_t size) {
-        size = align(size);
-
-        if (size > size_) {
-            // XXX: is this an optimal malloc size?
-            size_ = std::max<size_t>(size, size + align(sizeof(Cleaner)));
+    Type_ *malloc(size_t size, size_t alignment = CYAlignment) {
+        uint8_t *end(data_);
+        CYAlign(end, alignment);
+        end += size;
+
+        if (size_t(end - data_) > size_) {
+            size_t need(sizeof(Cleaner));
+            CYAlign(need, alignment);
+            need += size;
+            size_ = std::max<size_t>(next_, need);
+            next_ *= 2;

             data_ = reinterpret_cast<uint8_t *>(::malloc(size_));
             atexit(free, data_);
             _assert(size <= size_);
         }
 
             data_ = reinterpret_cast<uint8_t *>(::malloc(size_));
             atexit(free, data_);
             _assert(size <= size_);
         }
 

-        void *data(data_);
-        data_ += size;
-        size_ -= size;
+        uint8_t *data(data_);
+        CYAlign(data, alignment);
+        end = data + size;
+        size_ -= end - data_;
+        data_ = end;

         return reinterpret_cast<Type_ *>(data);
     }
 
         return reinterpret_cast<Type_ *>(data);
     }
 

+    template <typename Type_>
+    Type_ *calloc(size_t count, size_t size, size_t alignment = CYAlignment) {
+        Type_ *data(malloc<Type_>(count * size, alignment));
+        memset(data, 0, count * size);
+        return data;
+    }
+

     char *strdup(const char *data) {
         if (data == NULL)
             return NULL;
     char *strdup(const char *data) {
         if (data == NULL)
             return NULL;

-        return reinterpret_cast<char *>(memdup(data, strlen(data) + 1));
+        return memdup(data, strlen(data) + 1, 1);

     }
 
     }
 

-    void *memdup(const void *data, size_t size) {
-        void *copy(malloc<void>(size));
+    template <typename Type_>
+    Type_ *memdup(const Type_ *data, size_t size, size_t alignment = CYAlignment) {
+        Type_ *copy(malloc<Type_>(size, alignment));

         memcpy(copy, data, size);
         return copy;
     }
 
         memcpy(copy, data, size);
         return copy;
     }
 

-    char *strndup(const char *data, size_t size) const {
+    char *strndup(const char *data, size_t size) {

         return strmemdup(data, strnlen(data, size));
     }
 
         return strmemdup(data, strnlen(data, size));
     }
 

-    char *strmemdup(const char *data, size_t size) const {
-        char *copy(new char[size + 1]);
+    char *strmemdup(const char *data, size_t size) {
+        char *copy(malloc<char>(size + 1, 1));

         memcpy(copy, data, size);
         copy[size] = '\0';
         return copy;
         memcpy(copy, data, size);
         copy[size] = '\0';
         return copy;


@@ -134,7 +159,7 @@ class CYPool {
             va_end(args);
         }
 
             va_end(args);
         }
 

-        char *copy(malloc<char>(size + 1)); {
+        char *copy(malloc<char>(size + 1, 1)); {

             va_list args;
             va_start(args, data);
 
             va_list args;
             va_start(args, data);
 


@@ -182,6 +207,13 @@ class CYPool {
     }
 
     void atexit(void (*code)(void *), void *data = NULL);
     }
 
     void atexit(void (*code)(void *), void *data = NULL);

+
+    template <typename Type_>
+    Type_ &object() {
+        Type_ *value(new(*this) Type_());
+        atexit(&delete_<Type_>, value);
+        return *value;
+    }

 };
 
 _finline void *operator new(size_t size, CYPool &pool) {
 };
 
 _finline void *operator new(size_t size, CYPool &pool) {


@@ -203,6 +235,7 @@ struct CYData {
     CYData() :
         count_(1)
     {
     CYData() :
         count_(1)
     {

+        _assert(pool_ != NULL);

     }
 
     CYData(CYPool &pool) :
     }
 
     CYData(CYPool &pool) :


@@ -300,4 +333,7 @@ class CYLocalPool :
 #define $pool \
     (*CYLocal<CYPool>::Get())
 
 #define $pool \
     (*CYLocal<CYPool>::Get())
 

+template <>
+::pthread_key_t CYLocal<CYPool>::key_;
+

 #endif/*CYCRIPT_POOLING_HPP*/
 #endif/*CYCRIPT_POOLING_HPP*/