Return to pool_storage.h CVS log

Up to [parser3project] / parser3 / src / targets / cgi

Annotation of parser3/src/targets/cgi/pool_storage.h, revision 1.13

1.1       parser      1: /** @file
1.7       paf         2:        Parser: CGI: pool storage class decl.
1.1       parser      3: 
1.13    ! paf         4:        Copyright (c) 2001, 2003 ArtLebedev Group (http://www.artlebedev.com)
1.9       paf         5:        Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
1.1       parser      6: */
                      7: 
                      8: #ifndef PA_POOL_STORAGE_H
                      9: #define PA_POOL_STORAGE_H
1.11      paf        10: 
1.13    ! paf        11: static const char* IDENT_POOL_STORAGE_H="$Date: 2002/08/01 11:41:21 $";
1.1       parser     12: 
                     13: #include "pa_config_includes.h"
                     14: 
                     15: /**
                     16:        Dumb pool allocations accounter
                     17:        
                     18:        @todo implement at least simple suballocations
                     19: */
1.4       parser     20: template<class T> class List {
                     21: public:
                     22:        List(int preallocate) : 
                     23:                items((T *)::malloc(preallocate*sizeof(T))),
                     24:                allocated(0),
                     25:                used(0) {
                     26:                if(items) // successfully preallocated?
                     27:                        allocated=preallocate;
                     28:        }
1.2       parser     29: 
1.4       parser     30:        ~List() {
1.6       paf        31:                if(items)
                     32:                    ::free(items);
                     33:                items=0;
                     34:                used=0;
1.4       parser     35:        }
1.1       parser     36: 
1.4       parser     37:        bool add(T item) {
                     38:                if(full())
                     39:                        if(!expand())
                     40:                                return false;
                     41: 
                     42:                items[used++]=item;
                     43:                return true;
                     44:        }
                     45: 
                     46:        void for_each_reverse(void (*callback)(T& info)) {
                     47:                size_t top;
                     48: 
                     49:                for(top=used; top; )
                     50:                        callback(items[--top]);
                     51:        }
                     52: 
1.6       paf        53: size_t size() { return used; }
1.4       parser     54: private:
                     55: 
                     56:        bool full() { return used==allocated; }
                     57:        bool expand() {
                     58:                size_t new_allocated=allocated*3/2;
                     59:                T *new_items=(T *)::realloc(items, new_allocated*sizeof(T));
                     60:                if(new_items) {
                     61:                        items=new_items;
                     62:                        allocated=new_allocated;
1.1       parser     63:                        return true;
1.4       parser     64:                } else
                     65:                        return false;                   
                     66:        }
1.1       parser     67: 
1.4       parser     68: private:
1.1       parser     69: 
1.4       parser     70:        T *items;
                     71:        size_t allocated;
                     72:        size_t used;
1.1       parser     73: 
1.4       parser     74: };
1.1       parser     75: 
1.2       parser     76: class Pool_storage {
1.6       paf        77: 
1.1       parser     78:        struct Cleanup_struct {
                     79:                void (*cleanup) (void *);
                     80:                void *data;
                     81:        };
                     82: 
                     83: 
                     84: public:
                     85: 
                     86:        Pool_storage() : 
1.6       paf        87:                cleanups(100),
                     88:                allocations(10*0x400) {
                     89:        }
                     90: 
                     91:        void *malloc(size_t size) { 
                     92: //fprintf(stderr, "malloc: %d\n", size);
                     93:                void *result=::malloc(size);
                     94:                if(result && !allocations.add(result)) {
                     95:                        ::free(result); result=0;
                     96:                }
                     97:                return result;
                     98:        }
                     99:        void *calloc(size_t size) { 
                    100:                void *result=::calloc(size, 1);
                    101:                if(result && !allocations.add(result)) {
                    102:                        ::free(result); result=0;
                    103:                }
                    104:                return result;
1.1       parser    105:        }
                    106: 
                    107:        bool register_cleanup(void (*cleanup) (void *), void *data) {
                    108:                Cleanup_struct item={cleanup, data};
                    109:                return cleanups.add(item)!=0;
                    110:        }
                    111: 
1.4       parser    112:        static void cleanup(Cleanup_struct& item) {
                    113:                item.cleanup(item.data);
                    114:        }
                    115: 
1.6       paf       116:        static void free(void *& item) {
                    117:                ::free(item);  item=0;
                    118:        }
                    119: 
1.1       parser    120:        ~Pool_storage() {
1.10      paf       121:        //__asm__("int3");
                    122: //             _asm int 3;
1.6       paf       123: //fprintf(stderr, "cleanups: %d\n", cleanups.size());
1.1       parser    124:                // cleanups first, because they use some object's memory pointers
1.4       parser    125:                cleanups.for_each_reverse(cleanup);
1.6       paf       126:                
                    127: //fprintf(stderr, "allocs: %d\n", allocations.size());
                    128:                // allocations
                    129:                allocations.for_each_reverse(free);
1.1       parser    130:        }
                    131: 
                    132: private:
                    133: 
                    134:        List<Cleanup_struct> cleanups;
1.6       paf       135:        List<void *> allocations;
1.1       parser    136: 
                    137: };
                    138: 
                    139: 
                    140: #endif