VC++实现获取所有的TCP与UDP链接(三)

2014-11-23 19:26:24 · 作者: · 浏览: 77
// 为MIB_TCPTABLE结构申请内存
pTcpTable = (PMIB_TCPTABLE)::GlobalAlloc(GPTR, dwActualSize);
// 获取TCP连接表
if(::GetTcpTable(pTcpTable, &dwActualSize, bOrder) == NO_ERROR)
return pTcpTable;
::GlobalFree(pTcpTable);
}
return NULL;
}

void MyFreeTcpTable(PMIB_TCPTABLE pTcpTable)
{
if(pTcpTable != NULL)
::GlobalFree(pTcpTable);
}


PMIB_UDPTABLE MyGetUdpTable(BOOL bOrder)
{
PMIB_UDPTABLE pUdpTable = NULL;
DWORD dwActualSize = 0;

// 查询所需缓冲区的大小
if(::GetUdpTable(pUdpTable, &dwActualSize, bOrder) == ERROR_INSUFFICIENT_BUFFER)
{
// 为MIB_UDPTABLE结构申请内存 www.2cto.com
pUdpTable = (PMIB_UDPTABLE)::GlobalAlloc(GPTR, dwActualSize);
// 获取UDP监听表
if(::GetUdpTable(pUdpTable, &dwActualSize, bOrder) == NO_ERROR)
return pUdpTable;
::GlobalFree(pUdpTable);
}
return NULL;
}

void MyFreeUdpTable(PMIB_UDPTABLE pUdpTable)
{
if(pUdpTable != NULL)
::GlobalFree(pUdpTable);
}