////////////////////////////////////////////
// protoutils.h文件




#ifndef __PROTOUTILS_H__
#define __PROTOUTILS_H__

#include "nuiouser.h"

//////////////////////////////////////
// 函數(shù)定義

BOOL ProtoStartService();
void ProtoStopService();
HANDLE ProtoOpenControlDevice();


//////////////////////////////////////
// CPROTOAdapters類，枚舉綁定的適配器

#define MAX_ADAPTERS 10

class CPROTOAdapters
{
public:
	BOOL EnumAdapters(HANDLE hControlDevice);
	
	int m_nAdapters;
	LPWSTR m_pwszAdapterName[MAX_ADAPTERS];
	LPWSTR m_pwszSymbolicLink[MAX_ADAPTERS];
protected:
	char m_buffer[MAX_ADAPTERS*256];
};

////////////////////////////////////////
// CAdapter類，管理綁定的適配器

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

	// 打開、關(guān)閉適配器
	BOOL OpenAdapter(LPCWSTR pwszSymbolicLink, BOOL bAsyn = FALSE);
	void CloseAdapter();

	// 設置過濾屬性，如NDIS_PACKET_TYPE_PROMISCUOUS、NDIS_PACKET_TYPE_DIRECTED等
	BOOL SetFilter(ULONG nFilters);

	// 接收、發(fā)送數(shù)據(jù)
	int RecieveData(PVOID pBuffer, int nLen, LPOVERLAPPED lpOverlapped = NULL);
	int SendData(PVOID pBuffer, int nLen, LPOVERLAPPED lpOverlapped = NULL);

	// 重啟下層NIC、管理OID信息
	BOOL ResetAdapter();
	BOOL ProtoRequest(PPROTOCOL_OID_DATA pOidData, BOOL bQuery);

protected:
	HANDLE m_hAdapter;
};


#endif // __PROTOUTILS_H__