admin/SimpleRemoter
1
0
Fork 0
mirror of https://github.com/yuanyuanxiang/SimpleRemoter.git synced 2026-01-22 07:14:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improve: Reduce new / delete memory frequency in IOCPServer

Browse Source
This commit is contained in:
yuanyuanxiang
2025-11-30 21:20:04 +01:00
parent 9423eb42fc
commit 857bdd34bb
10 changed files with 350 additions and 186 deletions
Download Patch File Download Diff File Expand all files Collapse all files

165
server/2015Remote/Buffer.cpp
View File

@@ -1,15 +1,18 @@
#include "StdAfx.h"
#include "StdAfx.h"
#include "Buffer.h"
#include <math.h>
#define U_PAGE_ALIGNMENT 3
#define F_PAGE_ALIGNMENT 3.0
// 增大页面对齐大小，减少重新分配次数 (4KB对齐)
#define U_PAGE_ALIGNMENT 4096
#define F_PAGE_ALIGNMENT 4096.0
// 压缩阈值：当已读取数据超过此比例时才进行压缩
#define COMPACT_THRESHOLD 0.5
CBuffer::CBuffer(void)
{
m_ulMaxLength = 0;
m_ulReadOffset = 0;
m_Ptr = m_Base = NULL;
@@ -27,6 +30,7 @@ CBuffer::~CBuffer(void)
m_Base = m_Ptr = NULL;
m_ulMaxLength = 0;
m_ulReadOffset = 0;
}
@@ -34,53 +38,76 @@ ULONG CBuffer::RemoveCompletedBuffer(ULONG ulLength)
{
EnterCriticalSection(&m_cs);
if (ulLength > m_ulMaxLength) { //<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ij<EFBFBD><C4B3>ȱ<EFBFBD><C8B1>ڴ<EFBFBD><DAB4>ij<EFBFBD><C4B3>Ȼ<EFBFBD><C8BB><EFBFBD>
ULONG dataLen = m_Ptr - m_Base;
if (ulLength > m_ulMaxLength) { //请求长度比内存总长度还大
LeaveCriticalSection(&m_cs);
return 0;
}
if (ulLength > (m_Ptr - m_Base)) { //<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ij<EFBFBD><C4B3><EFBFBD> <20><><EFBFBD><EFBFBD>Ч<EFBFBD><D0A7><EFBFBD><EFBFBD><EFBFBD>ݳ<EFBFBD><DDB3>Ȼ<EFBFBD><C8BB><EFBFBD>
ulLength = m_Ptr - m_Base;
if (ulLength > dataLen) { //请求长度比有效数据长度还大
ulLength = dataLen;
}
if (ulLength) {
MoveMemory(m_Base,m_Base+ulLength, m_ulMaxLength - ulLength);
// 使用延迟移动策略：只更新读取偏移，不立即移动数据
m_ulReadOffset += ulLength;
m_Ptr -= ulLength;
// 当已读取数据超过阈值时才进行压缩
if (m_ulReadOffset > m_ulMaxLength * COMPACT_THRESHOLD) {
CompactBuffer();
}
}
DeAllocateBuffer(m_Ptr - m_Base);
LeaveCriticalSection(&m_cs);
return ulLength;
}
// 压缩缓冲区，移除已读取的数据
VOID CBuffer::CompactBuffer()
{
// 此函数应在持有锁的情况下调用
if (m_ulReadOffset > 0 && m_Base) {
ULONG remainingData = (m_Ptr - m_Base) - m_ulReadOffset;
if (remainingData > 0) {
MoveMemory(m_Base, m_Base + m_ulReadOffset, remainingData);
}
m_Ptr = m_Base + remainingData;
m_ulReadOffset = 0;
// 尝试缩减缓冲区
DeAllocateBuffer(remainingData);
}
}
ULONG CBuffer::ReadBuffer(PBYTE Buffer, ULONG ulLength)
{
EnterCriticalSection(&m_cs);
if (ulLength > m_ulMaxLength) {
LeaveCriticalSection(&m_cs);
return 0;
}
// 计算有效数据长度（考虑读取偏移）
ULONG effectiveDataLen = (m_Ptr - m_Base) - m_ulReadOffset;
if (ulLength > (m_Ptr - m_Base)) {
ulLength = m_Ptr - m_Base;
if (ulLength > effectiveDataLen) {
ulLength = effectiveDataLen;
}
if (ulLength) {
CopyMemory(Buffer,m_Base,ulLength);
// 从当前读取位置拷贝数据
CopyMemory(Buffer, m_Base + m_ulReadOffset, ulLength);
MoveMemory(m_Base,m_Base+ulLength, m_ulMaxLength - ulLength);
m_Ptr -= ulLength;
// 更新读取偏移而不是移动数据
m_ulReadOffset += ulLength;
// 当已读取数据超过阈值时才进行压缩
if (m_ulReadOffset > m_ulMaxLength * COMPACT_THRESHOLD) {
CompactBuffer();
}
}
DeAllocateBuffer(m_Ptr - m_Base);
LeaveCriticalSection(&m_cs);
return ulLength;
}
// ˽<EFBFBD><EFBFBD>: <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
// 私有: 缩减缓存
ULONG CBuffer::DeAllocateBuffer(ULONG ulLength)
{
if (ulLength < (m_Ptr - m_Base))
@@ -93,7 +120,7 @@ ULONG CBuffer::DeAllocateBuffer(ULONG ulLength)
}
PBYTE NewBase = (PBYTE) VirtualAlloc(NULL,ulNewMaxLength,MEM_COMMIT,PAGE_READWRITE);
ULONG ulv1 = m_Ptr - m_Base; //<EFBFBD><EFBFBD>ԭ<EFBFBD><EFBFBD><EFBFBD>ڴ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ч<EFBFBD><EFBFBD><EFBFBD><EFBFBD>
ULONG ulv1 = m_Ptr - m_Base; //从原来内存中的有效数据
CopyMemory(NewBase,m_Base,ulv1);
VirtualFree(m_Base,0,MEM_RELEASE);
@@ -124,7 +151,7 @@ BOOL CBuffer::WriteBuffer(PBYTE Buffer, ULONG ulLength)
return TRUE;
}
// ˽<EFBFBD><EFBFBD>: <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
// 私有: 扩展缓存
ULONG CBuffer::ReAllocateBuffer(ULONG ulLength)
{
if (ulLength < m_ulMaxLength)
@@ -137,7 +164,7 @@ ULONG CBuffer::ReAllocateBuffer(ULONG ulLength)
}
ULONG ulv1 = m_Ptr - m_Base; //ԭ<EFBFBD>ȵ<EFBFBD><EFBFBD><EFBFBD>Ч<EFBFBD><EFBFBD><EFBFBD>ݳ<EFBFBD><EFBFBD><EFBFBD>
ULONG ulv1 = m_Ptr - m_Base; //原先的有效数据长度
CopyMemory(NewBase,m_Base,ulv1);
@@ -156,19 +183,21 @@ VOID CBuffer::ClearBuffer()
{
EnterCriticalSection(&m_cs);
m_Ptr = m_Base;
m_ulReadOffset = 0; // 重置读取偏移
DeAllocateBuffer(1024);
LeaveCriticalSection(&m_cs);
}
ULONG CBuffer::GetBufferLength() // <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ч<EFBFBD><EFBFBD><EFBFBD>ݳ<EFBFBD><EFBFBD><EFBFBD>
ULONG CBuffer::GetBufferLength() // 返回有效数据长度
{
EnterCriticalSection(&m_cs);
if (m_Base == NULL) {
LeaveCriticalSection(&m_cs);
return 0;
}
ULONG len = m_Ptr - m_Base;
// 有效数据长度需要减去已读取的偏移量
ULONG len = (m_Ptr - m_Base) - m_ulReadOffset;
LeaveCriticalSection(&m_cs);
return len;
@@ -180,67 +209,111 @@ std::string CBuffer::Skip(ULONG ulPos)
return "";
EnterCriticalSection(&m_cs);
std::string ret(m_Base, m_Base + ulPos);
MoveMemory(m_Base, m_Base + ulPos, m_ulMaxLength - ulPos);
m_Ptr -= ulPos;
// 从当前读取位置开始跳过
std::string ret((char*)(m_Base + m_ulReadOffset), (char*)(m_Base + m_ulReadOffset + ulPos));
// 使用延迟移动策略
m_ulReadOffset += ulPos;
// 当已读取数据超过阈值时才进行压缩
if (m_ulReadOffset > m_ulMaxLength * COMPACT_THRESHOLD) {
CompactBuffer();
}
LeaveCriticalSection(&m_cs);
return ret;
}
// <EFBFBD>˺<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ƕ<EFBFBD><EFBFBD>̰߳<EFBFBD>ȫ<EFBFBD><EFBFBD>. ֻ<><D6BB>Զ<EFBFBD><D4B6><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʹ<EFBFBD><CAB9><EFBFBD><EFBFBD>.
// 此函数是多线程安全的. 只能远程调用使用它.
LPBYTE CBuffer::GetBuffer(ULONG ulPos)
{
EnterCriticalSection(&m_cs);
if (m_Base==NULL || ulPos >= (m_Ptr - m_Base)) {
// 计算有效数据长度
ULONG effectiveDataLen = (m_Ptr - m_Base) - m_ulReadOffset;
if (m_Base == NULL || ulPos >= effectiveDataLen) {
LeaveCriticalSection(&m_cs);
return NULL;
}
LPBYTE result = m_Base + ulPos;
// 返回相对于当前读取位置的指针
LPBYTE result = m_Base + m_ulReadOffset + ulPos;
LeaveCriticalSection(&m_cs);
return result;
}
// <EFBFBD>˺<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ƕ<EFBFBD><EFBFBD>̰߳<EFBFBD>ȫ<EFBFBD><EFBFBD>. <20><>ȡ<EFBFBD><C8A1><EFBFBD>棬<EFBFBD>õ<EFBFBD>Buffer<EFBFBD><EFBFBD><EFBFBD><EFBFBD>.
// 此函数是多线程安全的. 获取缓存，得到Buffer对象.
Buffer CBuffer::GetMyBuffer(ULONG ulPos)
{
EnterCriticalSection(&m_cs);
ULONG len = m_Ptr - m_Base;
if (m_Base == NULL || ulPos >= len) {
ULONG effectiveDataLen = (m_Ptr - m_Base) - m_ulReadOffset;
if (m_Base == NULL || ulPos >= effectiveDataLen) {
LeaveCriticalSection(&m_cs);
return Buffer();
}
Buffer result = Buffer(m_Base+ulPos, len - ulPos);
Buffer result = Buffer(m_Base + m_ulReadOffset + ulPos, effectiveDataLen - ulPos);
LeaveCriticalSection(&m_cs);
return result;
}
// <EFBFBD>˺<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ƕ<EFBFBD><EFBFBD>̰߳<EFBFBD>ȫ<EFBFBD><EFBFBD>. <20><>ȡ<EFBFBD><C8A1><EFBFBD><EFBFBD>ָ<EFBFBD><D6B8>λ<EFBFBD>ô<EFBFBD><C3B4><EFBFBD><EFBFBD><EFBFBD>ֵ.
// 此函数是多线程安全的. 获取缓存指定位置处的字节值.
BYTE CBuffer::GetBYTE(ULONG ulPos)
{
EnterCriticalSection(&m_cs);
if (m_Base == NULL || ulPos >= (m_Ptr - m_Base)) {
ULONG effectiveDataLen = (m_Ptr - m_Base) - m_ulReadOffset;
if (m_Base == NULL || ulPos >= effectiveDataLen) {
LeaveCriticalSection(&m_cs);
return NULL;
return 0;
}
BYTE p = *(m_Base + ulPos);
BYTE p = *(m_Base + m_ulReadOffset + ulPos);
LeaveCriticalSection(&m_cs);
return p;
}
// <EFBFBD>˺<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ƕ<EFBFBD><EFBFBD>̰߳<EFBFBD>ȫ<EFBFBD><EFBFBD>. <20><><EFBFBD><EFBFBD><EFBFBD>濽<EFBFBD><E6BFBD><EFBFBD><EFBFBD>Ŀ<EFBFBD><C4BF><EFBFBD>ڴ<EFBFBD><DAB4><EFBFBD>.
// 此函数是多线程安全的. 将缓存拷贝到目标内存中.
BOOL CBuffer::CopyBuffer(PVOID pDst, ULONG nLen, ULONG ulPos)
{
EnterCriticalSection(&m_cs);
ULONG len = m_Ptr - m_Base;
if (m_Base == NULL || len - ulPos < nLen) {
ULONG effectiveDataLen = (m_Ptr - m_Base) - m_ulReadOffset;
if (m_Base == NULL || effectiveDataLen - ulPos < nLen) {
LeaveCriticalSection(&m_cs);
return FALSE;
}
memcpy(pDst, m_Base+ulPos, nLen);
memcpy(pDst, m_Base + m_ulReadOffset + ulPos, nLen);
LeaveCriticalSection(&m_cs);
return TRUE;
}
}
// 获取可直接写入的缓冲区指针，用于零拷贝接收
LPBYTE CBuffer::GetWriteBuffer(ULONG requiredSize, ULONG& availableSize)
{
EnterCriticalSection(&m_cs);
// 先压缩缓冲区以获得更多空间
if (m_ulReadOffset > 0) {
CompactBuffer();
}
// 确保有足够空间
ULONG currentDataLen = m_Ptr - m_Base;
if (ReAllocateBuffer(currentDataLen + requiredSize) == (ULONG)-1) {
LeaveCriticalSection(&m_cs);
availableSize = 0;
return NULL;
}
availableSize = m_ulMaxLength - currentDataLen;
LPBYTE result = m_Ptr;
LeaveCriticalSection(&m_cs);
return result;
}
// 确认写入完成，更新内部指针
VOID CBuffer::CommitWrite(ULONG writtenSize)
{
EnterCriticalSection(&m_cs);
m_Ptr += writtenSize;
LeaveCriticalSection(&m_cs);
}