Memory/memoryChunk.cpp at master · ElleryHao/Memory · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
#include "memoryChunk.h"
memChunk::memChunk(int count,int size)
{
    m_size=size;
	m_memList=NULL;
	memBlock* pBlock=NULL;
    for(int i=0; i<count; i++)
    {
        pBlock=createBlock();
		if(!pBlock)
	    {
	       continue;
	    }
		pBlock->data.pNext=m_memList;
		m_memList=pBlock;
		m_count++;
    }
}
memChunk::~memChunk()
{
    memBlock* temp=NULL;
    for(int i=0; i<m_count; i++)
    {
        temp=m_memList;
		m_memList=temp->data.pNext;
		delBlock(temp);
		m_count--;
    }
	m_size=0;
}

memBlock* memChunk::createBlock()
{
    memBlock* pBlock=(memBlock*)::malloc(m_size+sizeof(memHeader));
    if(!pBlock)
    {
        return NULL;
    }
    pBlock->head.pMemChunk=this;
    pBlock->head.size=m_size;
    pBlock->data.pNext=NULL;
}

void* memChunk::malloc()
{
    if(!m_memList)
    {
       m_memList=createBlock();
    }
    memBlock* pBlock=m_memList;
    m_memList=pBlock->data.pNext;
    m_count--;
    return (void *)&pBlock->data.pNext;
}

void memChunk::free(void * pMem)
{
    if(!pMem)
    {
       return;
    }
    memBlock* pBlock=(memBlock*)((char*)pMem-sizeof(memHeader));
    pBlock->head.pMemChunk->free(pBlock);
}

void memChunk::free(memBlock* pBlock)
{
    if(!pBlock)
    {
        return;
    }
    pBlock->data.pNext=m_memList;
    m_memList=pBlock;
    m_count++;
}

void memChunk::delBlock(memBlock * pBlock)
{
   pBlock->head.pMemChunk=NULL;
   pBlock->head.size=0;
   free(pBlock);
}