Pooling.hpp

   1 /* Cycript - Remote Execution Server and Disassembler
   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 <minimal/stdlib.h>
  47
  48 _finline void *operator new(size_t size, apr_pool_t *pool) {
  49     return apr_palloc(pool, size);
  50 }
  51
  52 _finline void *operator new [](size_t size, apr_pool_t *pool) {
  53     return apr_palloc(pool, size);
  54 }
  55
  56 class CYPool {
  57   private:
  58     apr_pool_t *pool_;
  59
  60   public:
  61     CYPool() {
  62         _aprcall(apr_pool_create(&pool_, NULL));
  63     }
  64
  65     ~CYPool() {
  66         apr_pool_destroy(pool_);
  67     }
  68
  69     void Clear() {
  70         apr_pool_clear(pool_);
  71     }
  72
  73     operator apr_pool_t *() const {
  74         return pool_;
  75     }
  76
  77     char *operator ()(const char *data) const {
  78         return apr_pstrdup(pool_, data);
  79     }
  80
  81     char *operator ()(const char *data, size_t size) const {
  82         return apr_pstrndup(pool_, data, size);
  83     }
  84 };
  85
  86 struct CYData {
  87     apr_pool_t *pool_;
  88
  89     virtual ~CYData() {
  90     }
  91
  92     static void *operator new(size_t size, apr_pool_t *pool) {
  93         void *data(apr_palloc(pool, size));
  94         reinterpret_cast<CYData *>(data)->pool_ = pool;
  95         return data;
  96     }
  97
  98     static void *operator new(size_t size) {
  99         apr_pool_t *pool;
 100         _aprcall(apr_pool_create(&pool, NULL));
 101         return operator new(size, pool);
 102     }
 103
 104     static void operator delete(void *data) {
 105         apr_pool_destroy(reinterpret_cast<CYData *>(data)->pool_);
 106     }
 107
 108 };
 109
 110 #endif/*CYPOOLING_HPP*/