void free(void * ptr)
{
char * charPtr = static_cast<char *>(ptr);
Block & prev = prevBlock(charPtr);
Block & next = nextBlock(charPtr);
Block & block = blocks.at(charPtr);
assert(block.free == false);
assert(prev.ptr == NULL || prev.ptr + prev.size == block.ptr);
assert(next.ptr == NULL || block.ptr + block.size == next.ptr);
/* prev is busy and next is busy */
/* just mark current block as free */
if(!prev.free && !next.free)
{
markBlockFree(block);
}
/* prev is free and next is busy */
/* concat prev and curr block into one free block */
else if(prev.free && !next.free)
{
resizeBlock(prev, prev.size + block.size);
removeBlock(block);
}
/* prev is busy and next is free */
/* concat curr and next block into one free block */
else if(!prev.free && next.free)
{
resizeBlock(block, block.size + next.size);
removeBlock(next);
markBlockFree(block);
}
/* concat all three blocks into one block*/
else if(prev.free && next.free)
{
resizeBlock(prev, prev.size + block.size + next.size);
removeBlock(block);
removeBlock(next);
}
#ifdef SELFTEST
selfTest();
#endif
}
void * realloc(void * ptr, size_t size)
{
char * charPtr = static_cast<char *>(ptr);
Block & block = blocks.at(charPtr);
Block & next = nextBlock(charPtr);
/* There is enough place ahead of the current block */
if(block.size == size)
{
}
else if(block.size > size)
{
splitBlock(block, size);
Block & rest = blocks.at(block.ptr + size);
markBlockFree(rest);
if(next.free)
{
resizeBlock(rest, rest.size + next.size);
removeBlock(next);
}
}
else if(next.free && block.size + next.size >= size)
{
splitBlock(next, size-block.size);
removeBlock(next);
resizeBlock(block, size);
}
else
{
void * newPtr = alloc(size);
if(newPtr != NULL)
{
memcpy(newPtr, ptr, block.size);
free(ptr);
ptr = newPtr;
}
}
#ifdef SELFTEST
selfTest();
#endif
return ptr;
}
int main ()
{
testResult_t *result = initializeTestResult();
fileSystem_t *FS = getFS();
FS->superBlock->blockBitmap[0] = 0b11111111;
FS->superBlock->blockBitmap[1] = 0b11111111;
FS->superBlock->blockBitmap[2] = 0b00001101;
assert(findEmptyBlock() == 17, "empty block find test.", result);
assert(FS->activeDir->entryCount == 0, "add entry test begin", result);
activeFile_t * aF = openFileInternal(FS->activeDir, "TestFile");
assert(FS->activeDir->entryCount == 1, "add entry test begin", result);
directoryEntry_t dE = FS->activeDir->entries[0];
fprintf(stdout, "%s, %d, %d\n", dE.name, dE.blockType, dE.startingBlockNum);
fprintf(stdout, "%s, %d\n", aF->name, aF->curContentsPtr);
fileBlock_t *fB;
fB = malloc(sizeof(fileBlock_t));
readBlockFromDisk(17, fB);
fprintf(stdout, "%d\n", fB->blockType);
assert(fB->blockType == 3, "read blockType", result);
fprintf(stdout, "%s\n", fB->name);
assert(findEmptyBlock() == 20, "bitmap affect test.", result);
markBlockFree(17);
assert(findEmptyBlock() == 17, "bitmap affect test.", result);
memset(fB, 0, sizeof(fileBlock_t));
readBlockFromDisk(17, fB);
fprintf(stdout, "%d\n", fB->blockType);
assert(fB->blockType == 0, "free block zero set test.", result);
printTestResult(result);
freeTestResult(result);
return 0;
}
