Pooling.hpp

   1 /* Cycript - Inlining/Optimizing JavaScript Compiler
   2  * Copyright (C) 2009  Jay Freeman (saurik)
   3 */
   4
   5 /* Modified BSD License {{{ */
   6 /*
   7  *        Redistribution and use in source and binary
   8  * forms, with or without modification, are permitted
   9  * provided that the following conditions are met:
  10  *
  11  * 1. Redistributions of source code must retain the
  12  *    above copyright notice, this list of conditions
  13  *    and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the
  15  *    above copyright notice, this list of conditions
  16  *    and the following disclaimer in the documentation
  17  *    and/or other materials provided with the
  18  *    distribution.
  19  * 3. The name of the author may not be used to endorse
  20  *    or promote products derived from this software
  21  *    without specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
  24  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
  25  * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  26  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27  * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
  28  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  29  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  30  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  31  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  32  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  33  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
  34  * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  35  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  36  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  37 */
  38 /* }}} */
  39
  40 #ifndef CYPOOLING_HPP
  41 #define CYPOOLING_HPP
  42
  43 #include <apr_pools.h>
  44 #include <apr_strings.h>
  45
  46 #include "Exception.hpp"
  47 #include "Standard.hpp"
  48
  49 #include <cstdlib>
  50
  51 _finline void *operator new(size_t size, apr_pool_t *pool) {
  52     return apr_palloc(pool, size);
  53 }
  54
  55 _finline void *operator new [](size_t size, apr_pool_t *pool) {
  56     return apr_palloc(pool, size);
  57 }
  58
  59 class CYPool {
  60   private:
  61     apr_pool_t *pool_;
  62
  63   public:
  64     CYPool(apr_pool_t *pool = NULL) {
  65         _aprcall(apr_pool_create(&pool_, pool));
  66     }
  67
  68     ~CYPool() {
  69         apr_pool_destroy(pool_);
  70     }
  71
  72     void Clear() {
  73         apr_pool_clear(pool_);
  74     }
  75
  76     operator apr_pool_t *() const {
  77         return pool_;
  78     }
  79
  80     char *operator ()(const char *data) const {
  81         return apr_pstrdup(pool_, data);
  82     }
  83
  84     char *operator ()(const char *data, size_t size) const {
  85         return apr_pstrndup(pool_, data, size);
  86     }
  87 };
  88
  89 struct CYData {
  90     apr_pool_t *pool_;
  91     unsigned count_;
  92
  93     CYData() :
  94         count_(1)
  95     {
  96     }
  97
  98     virtual ~CYData() {
  99     }
 100
 101     static void *operator new(size_t size, apr_pool_t *pool) {
 102         void *data(apr_palloc(pool, size));
 103         reinterpret_cast<CYData *>(data)->pool_ = pool;
 104         return data;
 105     }
 106
 107     static void *operator new(size_t size) {
 108         apr_pool_t *pool;
 109         _aprcall(apr_pool_create(&pool, NULL));
 110         return operator new(size, pool);
 111     }
 112
 113     static void operator delete(void *data) {
 114         apr_pool_destroy(reinterpret_cast<CYData *>(data)->pool_);
 115     }
 116 };
 117
 118 template <typename Type_>
 119 struct CYPoolAllocator {
 120     apr_pool_t *pool_;
 121
 122     typedef Type_ value_type;
 123     typedef value_type *pointer;
 124     typedef const value_type *const_pointer;
 125     typedef value_type &reference;
 126     typedef const value_type &const_reference;
 127     typedef std::size_t size_type;
 128     typedef std::ptrdiff_t difference_type;
 129
 130     CYPoolAllocator() :
 131         pool_(NULL)
 132     {
 133     }
 134
 135     template <typename Right_>
 136     CYPoolAllocator(const CYPoolAllocator<Right_> &rhs) :
 137         pool_(rhs.pool_)
 138     {
 139     }
 140
 141     pointer allocate(size_type size, const void *hint = 0) {
 142         return reinterpret_cast<pointer>(apr_palloc(pool_, size));
 143     }
 144
 145     void deallocate(pointer data, size_type size) {
 146     }
 147
 148     void construct(pointer address, const Type_ &rhs) {
 149         new(address) Type_(rhs);
 150     }
 151
 152     void destroy(pointer address) {
 153         address->~Type_();
 154     }
 155
 156     template <typename Right_>
 157     inline bool operator==(const CYPoolAllocator<Right_> &rhs) {
 158         return pool_ == rhs.pool_;
 159     }
 160
 161     template <typename Right_>
 162     inline bool operator!=(const CYPoolAllocator<Right_> &rhs) {
 163         return !operator==(rhs);
 164     }
 165
 166     template <typename Right_>
 167     struct rebind {
 168         typedef CYPoolAllocator<Right_> other;
 169     };
 170 };
 171
 172 #endif/*CYPOOLING_HPP*/