overlapped.cpp

Windows防火墙与网络封包截获技术源码

//——————————————————————————————————————
// 重叠操作控制，因为异步重叠操作接收数据时可能不能立即返回从而导致
// 无法立即知道接收数据内容和大小，所以设置会调函数，接收完成时调用。
//

#include "stdafx.h"
#include "overlapped.h"

COverlapped::COverlapped()
{
	InitializeCriticalSection(&m_CriticalSection);
	m_OverlappedRecorder.SetSize(0);
}

BOOL COverlapped::AddOverlapped(
	SOCKET			s,
	LPWSABUF		lpBuffers,
	DWORD			dwBufferCount,
	LPDWORD			lpNumberOfBytesRecvd,
	LPDWORD			lpFlags,
	LPWSAOVERLAPPED lpOverlapped,
	LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine,
	struct sockaddr FAR * lpFrom,
	LPINT			lpFromlen,
	int				FunctionType //0: WSPRecv; 1:WSPRecvFrom
)
{
	EnterCriticalSection(&m_CriticalSection);
	__try
	{
		OVERLAPPED_RECORDER OverlappedRecorder;
		OverlappedRecorder.s					= s;
		OverlappedRecorder.lpBuffers			= lpBuffers;
		OverlappedRecorder.dwBufferCount		= dwBufferCount;
		OverlappedRecorder.lpNumberOfBytesRecvd = lpNumberOfBytesRecvd;
		OverlappedRecorder.lpFlags				= lpFlags;
		OverlappedRecorder.lpOverlapped			= lpOverlapped;
		OverlappedRecorder.lpCompletionRoutine	= lpCompletionRoutine;
		OverlappedRecorder.lpFrom				= lpFrom;
		OverlappedRecorder.lpFromlen			= lpFromlen;
		OverlappedRecorder.FunctionType			= FunctionType;

		int iIndex = FindOverlapped(lpOverlapped);
		if(iIndex < 0)
		{
			if(m_OverlappedRecorder.Add(OverlappedRecorder) < 0)
				return FALSE;
		}
		else
		{
			m_OverlappedRecorder[iIndex] = OverlappedRecorder;
		}
	}
	__finally
	{
		LeaveCriticalSection(&m_CriticalSection);
	}
	return TRUE;
}

int COverlapped::FindOverlapped(LPWSAOVERLAPPED lpOverlapped)
{
	int i = 0, j = m_OverlappedRecorder.GetSize();
	for(i = 0; i < j; i++)
	{
		if(m_OverlappedRecorder[i].lpOverlapped == lpOverlapped)
			return i;
	}
	return -1;
}

BOOL COverlapped::DeleteOverlapped(int iIndex)
{
	try
	{
		EnterCriticalSection(&m_CriticalSection);
		m_OverlappedRecorder.RemoveAt(iIndex);
		LeaveCriticalSection(&m_CriticalSection);
	}
	catch(...)
	{
		return FALSE;
	}
	return TRUE;
}