/*
 * Copyright 2011, Rene Gollent, rene@gollent.com.
 * Distributed under the terms of the MIT License.
 */
#ifndef TEAM_MEMORY_BLOCK_MANAGER_H
#define TEAM_MEMORY_BLOCK_MANAGER_H


#include <Locker.h>
#include <Referenceable.h>
#include <util/DoublyLinkedList.h>
#include <util/OpenHashTable.h>

#include "Types.h"


struct MemoryBlockHashDefinition;
class TeamMemoryBlock;


class TeamMemoryBlockManager
{
public:
								TeamMemoryBlockManager();
								~TeamMemoryBlockManager();

		status_t				Init();

		TeamMemoryBlock*		GetMemoryBlock(target_addr_t address);

private:
		struct Key;
		struct MemoryBlockEntry;
		struct MemoryBlockHashDefinition;
		typedef BOpenHashTable<MemoryBlockHashDefinition> MemoryBlockTable;
		typedef DoublyLinkedList<TeamMemoryBlock> DeadBlockTable;

private:
		void					_Cleanup();
		void					_MarkDeadBlock(target_addr_t address);
		void					_RemoveBlock(target_addr_t address);

private:
		friend class TeamMemoryBlockOwner;

private:
		BLocker					fLock;
		MemoryBlockTable*		fActiveBlocks;
		DeadBlockTable*			fDeadBlocks;
};


class TeamMemoryBlockOwner
{
public:
								TeamMemoryBlockOwner(
									TeamMemoryBlockManager* manager);
								~TeamMemoryBlockOwner();

		void					RemoveBlock(TeamMemoryBlock* block);

private:
	TeamMemoryBlockManager* 	fBlockManager;
};


#endif // TEAM_MEMORY_BLOCK_MANAGER_H