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() {
  65         _aprcall(apr_pool_create(&pool_, NULL));
  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
  92     virtual ~CYData() {
  93     }
  94
  95     static void *operator new(size_t size, apr_pool_t *pool) {
  96         void *data(apr_palloc(pool, size));
  97         reinterpret_cast<CYData *>(data)->pool_ = pool;
  98         return data;
  99     }
 100
 101     static void *operator new(size_t size) {
 102         apr_pool_t *pool;
 103         _aprcall(apr_pool_create(&pool, NULL));
 104         return operator new(size, pool);
 105     }
 106
 107     static void operator delete(void *data) {
 108         apr_pool_destroy(reinterpret_cast<CYData *>(data)->pool_);
 109     }
 110
 111 };
 112
 113 template <typename Type_>
 114 struct CYPoolAllocator {
 115     apr_pool_t *pool_;
 116
 117     typedef Type_ value_type;
 118     typedef value_type *pointer;
 119     typedef const value_type *const_pointer;
 120     typedef value_type &reference;
 121     typedef const value_type &const_reference;
 122     typedef std::size_t size_type;
 123     typedef std::ptrdiff_t difference_type;
 124
 125     CYPoolAllocator() :
 126         pool_(NULL)
 127     {
 128     }
 129
 130     template <typename Right_>
 131     CYPoolAllocator(const CYPoolAllocator<Right_> &rhs) :
 132         pool_(rhs.pool_)
 133     {
 134     }
 135
 136     pointer allocate(size_type size, const void *hint = 0) {
 137         return reinterpret_cast<pointer>(apr_palloc(pool_, size));
 138     }
 139
 140     void deallocate(pointer data, size_type size) {
 141     }
 142
 143     void construct(pointer address, const Type_ &rhs) {
 144         new(address) Type_(rhs);
 145     }
 146
 147     void destroy(pointer address) {
 148         address->~Type_();
 149     }
 150
 151     template <typename Right_>
 152     inline bool operator==(const CYPoolAllocator<Right_> &rhs) {
 153         return pool_ == rhs.pool_;
 154     }
 155
 156     template <typename Right_>
 157     inline bool operator!=(const CYPoolAllocator<Right_> &rhs) {
 158         return !operator==(rhs);
 159     }
 160
 161     template <typename Right_>
 162     struct rebind {
 163         typedef CYPoolAllocator<Right_> other;
 164     };
 165 };
 166
 167 #endif/*CYPOOLING_HPP*/