Annotation of win32/gnome/gmime-x.x.x/memchunk.c, revision 1.1

1.1     ! paf         1: /* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
        !             2: /*
        !             3:  *  Authors: Jeffrey Stedfast <fejj@ximian.com>
        !             4:  *
        !             5:  *  Copyright 2001 Ximian, Inc. (www.ximian.com)
        !             6:  *
        !             7:  *  This program is free software; you can redistribute it and/or modify
        !             8:  *  it under the terms of the GNU General Public License as published by
        !             9:  *  the Free Software Foundation; either version 2 of the License, or
        !            10:  *  (at your option) any later version.
        !            11:  *
        !            12:  *  This program is distributed in the hope that it will be useful,
        !            13:  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
        !            14:  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
        !            15:  *  GNU General Public License for more details.
        !            16:  *
        !            17:  *  You should have received a copy of the GNU General Public License
        !            18:  *  along with this program; if not, write to the Free Software
        !            19:  *  Foundation, Inc., 59 Temple Street #330, Boston, MA 02111-1307, USA.
        !            20:  *
        !            21:  */
        !            22: 
        !            23: 
        !            24: #ifdef HAVE_CONFIG_H
        !            25: #include <config.h>
        !            26: #endif
        !            27: 
        !            28: #include <stdlib.h>
        !            29: #include <string.h>
        !            30: 
        !            31: #ifdef HAVE_ALLOCA_H
        !            32: #include <alloca.h>
        !            33: #endif
        !            34: 
        !            35: #include "memchunk.h"
        !            36: 
        !            37: 
        !            38: typedef struct _MemChunkFreeNode {
        !            39:        struct _MemChunkFreeNode *next;
        !            40:        unsigned int atoms;
        !            41: } MemChunkFreeNode;
        !            42: 
        !            43: typedef struct _MemChunkNodeInfo {
        !            44:        struct _MemChunkNodeInfo *next;
        !            45:        unsigned char *block;
        !            46:        size_t blocksize;
        !            47:        size_t atoms;
        !            48: } MemChunkNodeInfo;
        !            49: 
        !            50: struct _MemChunk {
        !            51:        size_t atomsize;
        !            52:        size_t atomcount;
        !            53:        size_t blocksize;
        !            54:        GPtrArray *blocks;
        !            55:        gboolean autoclean;
        !            56:        MemChunkFreeNode *free;
        !            57: };
        !            58: 
        !            59: 
        !            60: static int
        !            61: tree_compare (MemChunkNodeInfo *a, MemChunkNodeInfo *b)
        !            62: {
        !            63:        if (a->block > b->block)
        !            64:                return 1;
        !            65:        else if (a->block < b->block)
        !            66:                return -1;
        !            67:        else
        !            68:                return 0;
        !            69: }
        !            70: 
        !            71: static int
        !            72: tree_search (MemChunkNodeInfo *info, unsigned char *mem)
        !            73: {
        !            74:        if (info->block <= mem) {
        !            75:                if (mem < info->block + info->blocksize)
        !            76:                        return 0;
        !            77:                
        !            78:                return 1;
        !            79:        }
        !            80:        
        !            81:        return -1;
        !            82: }
        !            83: 
        !            84: MemChunk *
        !            85: memchunk_new (size_t atomsize, size_t atomcount, gboolean autoclean)
        !            86: {
        !            87:        MemChunk *chunk;
        !            88:        
        !            89:        /* each atom has to be at least the size of a MemChunkFreeNode
        !            90:            for this to work */
        !            91:        atomsize = MAX (atomsize, sizeof (MemChunkFreeNode));
        !            92:        
        !            93:        chunk = g_malloc (sizeof (MemChunk));
        !            94:        chunk->atomsize = atomsize;
        !            95:        chunk->atomcount = atomcount;
        !            96:        chunk->blocksize = atomsize * atomcount;
        !            97:        chunk->autoclean = autoclean;
        !            98:        chunk->blocks = g_ptr_array_new ();
        !            99:        chunk->free = NULL;
        !           100:        
        !           101:        return chunk;
        !           102: }
        !           103: 
        !           104: 
        !           105: void *
        !           106: memchunk_alloc (MemChunk *memchunk)
        !           107: {
        !           108:        MemChunkFreeNode *node;
        !           109:        char *block;
        !           110:        
        !           111:        if (memchunk->free) {
        !           112:                node = memchunk->free;
        !           113:                node->atoms--;
        !           114:                if (node->atoms > 0)
        !           115:                        return (char *) node + (node->atoms * memchunk->atomsize);
        !           116:                
        !           117:                memchunk->free = node->next;
        !           118:                
        !           119:                return (void *) node;
        !           120:        } else {
        !           121:                block = g_malloc (memchunk->blocksize);
        !           122:                g_ptr_array_add (memchunk->blocks, block);
        !           123:                node = (MemChunkFreeNode *) (block + memchunk->atomsize);
        !           124:                node->next = NULL;
        !           125:                node->atoms = memchunk->atomcount - 1;
        !           126:                memchunk->free = node;
        !           127:                
        !           128:                return (void *) block;
        !           129:        }
        !           130: }
        !           131: 
        !           132: 
        !           133: void *
        !           134: memchunk_alloc0 (MemChunk *memchunk)
        !           135: {
        !           136:        void *mem;
        !           137:        
        !           138:        mem = memchunk_alloc (memchunk);
        !           139:        memset (mem, 0, memchunk->atomsize);
        !           140:        
        !           141:        return mem;
        !           142: }
        !           143: 
        !           144: 
        !           145: void
        !           146: memchunk_free (MemChunk *memchunk, void *mem)
        !           147: {
        !           148:        MemChunkFreeNode *node;
        !           149:        
        !           150:        node = (MemChunkFreeNode *) mem;
        !           151:        node->next = memchunk->free;
        !           152:        node->atoms = 1;
        !           153:        memchunk->free = node;
        !           154:        
        !           155:        /* yuck, this'll be slow... */
        !           156:        if (memchunk->autoclean)
        !           157:                memchunk_clean (memchunk);
        !           158: }
        !           159: 
        !           160: 
        !           161: void
        !           162: memchunk_reset (MemChunk *memchunk)
        !           163: {
        !           164:        MemChunkFreeNode *node, *next = NULL;
        !           165:        int i;
        !           166:        
        !           167:        for (i = 0; i < memchunk->blocks->len; i++) {
        !           168:                node = memchunk->blocks->pdata[i];
        !           169:                node->atoms = memchunk->atomcount;
        !           170:                node->next = next;
        !           171:                next = node;
        !           172:        }
        !           173:        
        !           174:        memchunk->free = next;
        !           175: }
        !           176: 
        !           177: void
        !           178: memchunk_clean (MemChunk *memchunk)
        !           179: {
        !           180:        MemChunkNodeInfo *info, *next = NULL;
        !           181:        MemChunkFreeNode *node;
        !           182:        GTree *tree;
        !           183:        int i;
        !           184:        
        !           185:        node = memchunk->free;
        !           186:        
        !           187:        if (memchunk->blocks->len == 0 || node == NULL)
        !           188:                return;
        !           189:        
        !           190:        tree = g_tree_new ((GCompareFunc) tree_compare);
        !           191:        for (i = 0; i < memchunk->blocks->len; i++) {
        !           192:                info = alloca (sizeof (MemChunkNodeInfo));
        !           193:                info->next = next;
        !           194:                info->block = memchunk->blocks->pdata[i];
        !           195:                info->blocksize = memchunk->blocksize;
        !           196:                info->atoms = 0;
        !           197:                
        !           198:                next = info;
        !           199:                
        !           200:                g_tree_insert (tree, info, info);
        !           201:        }
        !           202:        
        !           203:        while (node) {
        !           204:                info = g_tree_search (tree, (GCompareFunc) tree_search, node);
        !           205:                if (info)
        !           206:                        info->atoms += node->atoms;
        !           207:                
        !           208:                node = node->next;
        !           209:        }
        !           210:        
        !           211:        info = next;
        !           212:        while (info) {
        !           213:                if (info->atoms == memchunk->atomcount) {
        !           214:                        MemChunkFreeNode *prev = NULL;
        !           215:                        
        !           216:                        node = memchunk->free;
        !           217:                        while (node) {
        !           218:                                if (tree_search (info, (void *) node) == 0) {
        !           219:                                        /* prune this node from our free-node list */
        !           220:                                        if (prev)
        !           221:                                                prev->next = node->next;
        !           222:                                        else
        !           223:                                                memchunk->free = node->next;
        !           224:                                } else {
        !           225:                                        prev = node;
        !           226:                                }
        !           227:                                
        !           228:                                node = node->next;
        !           229:                        }
        !           230:                        
        !           231:                        /* free the block */
        !           232:                        g_ptr_array_remove_fast (memchunk->blocks, info->block);
        !           233:                        g_free (info->block);
        !           234:                }
        !           235:                
        !           236:                info = info->next;
        !           237:        }
        !           238:        
        !           239:        g_tree_destroy (tree);
        !           240: }
        !           241: 
        !           242: void
        !           243: memchunk_destroy (MemChunk *memchunk)
        !           244: {
        !           245:        int i;
        !           246:        
        !           247:        for (i = 0; i < memchunk->blocks->len; i++)
        !           248:                g_free (memchunk->blocks->pdata[i]);
        !           249:        
        !           250:        g_ptr_array_free (memchunk->blocks, TRUE);
        !           251:        
        !           252:        g_free (memchunk);
        !           253: }

E-mail: