#include "StdAfx.h"
#include "Buffer.h"

#include <math.h>


#define U_PAGE_ALIGNMENT   3
#define F_PAGE_ALIGNMENT 3.0
CBuffer::CBuffer(void)
{
	m_ulMaxLength = 0;

	m_Ptr = m_Base = NULL;

	InitializeCriticalSection(&m_cs);
}

CBuffer::~CBuffer(void)
{
	if (m_Base)
	{
		VirtualFree(m_Base, 0, MEM_RELEASE);
		m_Base = NULL;
	}

	DeleteCriticalSection(&m_cs);

	m_Base = m_Ptr = NULL;
	m_ulMaxLength = 0;
}


ULONG CBuffer::RemoveComletedBuffer(ULONG ulLength)
{
	if (ulLength >GetBufferMaxLength())   //如果传进的长度比内存的长度还大
	{
		return 0;
	}
	if (ulLength >GetBufferLength())  //如果传进的长度 比有效的数据长度还大
	{
		ulLength = GetBufferLength();
	}

	if (ulLength)
	{
		MoveMemory(m_Base,m_Base+ulLength,GetBufferMaxLength() - ulLength);   //数组前移  [Shinexxxx??]

		m_Ptr -= ulLength;
	}

	DeAllocateBuffer(GetBufferLength());   

	return ulLength;
}

ULONG CBuffer::ReadBuffer(PBYTE Buffer, ULONG ulLength)
{
	EnterCriticalSection(&m_cs);

	if (ulLength > GetBufferMaxLength())
	{
		LeaveCriticalSection(&m_cs);
		return 0;
	}
	if (ulLength > GetBufferLength())
	{
		ulLength = GetBufferLength();
	}

	if (ulLength)
	{
		CopyMemory(Buffer,m_Base,ulLength);  

		MoveMemory(m_Base,m_Base+ulLength,GetBufferMaxLength() - ulLength);
		m_Ptr -= ulLength;
	}

	DeAllocateBuffer(GetBufferLength());   

	LeaveCriticalSection(&m_cs);
	return ulLength;
}

ULONG CBuffer::DeAllocateBuffer(ULONG ulLength)        
{
	if (ulLength < GetBufferLength())     
		return 0;

	ULONG ulNewMaxLength = (ULONG)ceil(ulLength / F_PAGE_ALIGNMENT) * U_PAGE_ALIGNMENT;  

	if (GetBufferMaxLength() <= ulNewMaxLength) 
	{
		return 0;
	}
	PBYTE NewBase = (PBYTE) VirtualAlloc(NULL,ulNewMaxLength,MEM_COMMIT,PAGE_READWRITE);

	ULONG ulv1 = GetBufferLength();  //算原先内存的有效长度
	CopyMemory(NewBase,m_Base,ulv1);

	VirtualFree(m_Base,0,MEM_RELEASE);

	m_Base = NewBase;

	m_Ptr = m_Base + ulv1;

	m_ulMaxLength = ulNewMaxLength;

	return m_ulMaxLength;
}


BOOL CBuffer::WriteBuffer(PBYTE Buffer, ULONG ulLength)
{
	EnterCriticalSection(&m_cs);

	if (ReAllocateBuffer(ulLength + GetBufferLength()) == -1)//10 +1   1024
	{
		LeaveCriticalSection(&m_cs);
		return false;
	}

	CopyMemory(m_Ptr,Buffer,ulLength);//Hello 5

	m_Ptr+=ulLength;
	LeaveCriticalSection(&m_cs);
	return TRUE;
}

ULONG CBuffer::ReAllocateBuffer(ULONG ulLength)
{
	if (ulLength < GetBufferMaxLength())   
		return 0;

	ULONG  ulNewMaxLength = (ULONG)ceil(ulLength / F_PAGE_ALIGNMENT) * U_PAGE_ALIGNMENT;  
	PBYTE  NewBase  = (PBYTE) VirtualAlloc(NULL,ulNewMaxLength,MEM_COMMIT,PAGE_READWRITE);
	if (NewBase == NULL)
	{
		return -1; 
	}

	ULONG ulv1 = GetBufferLength();   //原先的有效数据长度  
	CopyMemory(NewBase,m_Base,ulv1);

	if (m_Base)
	{
		VirtualFree(m_Base,0,MEM_RELEASE);
	}
	m_Base = NewBase;
	m_Ptr = m_Base + ulv1; //1024

	m_ulMaxLength = ulNewMaxLength;  //2048

	return m_ulMaxLength;
}

VOID CBuffer::ClearBuffer()
{
	EnterCriticalSection(&m_cs);
	m_Ptr = m_Base;

	DeAllocateBuffer(1024);  
	LeaveCriticalSection(&m_cs);
}

ULONG CBuffer::GetBufferLength() //获得有效数据长度
{
	if (m_Base == NULL)
		return 0;

	return (ULONG)m_Ptr - (ULONG)m_Base;
}


ULONG CBuffer::GetBufferMaxLength()   
{
	return m_ulMaxLength;
}

PBYTE CBuffer::GetBuffer(ULONG ulPos)
{
	if (m_Base==NULL)
	{
		return NULL;
	}
	if (ulPos>=GetBufferLength())
	{
		return NULL;
	}
	return m_Base+ulPos;
}