检测UDP端口占用状态的函数

#include <WinSock2.h>
#include <windows.h>
#include <stdio.h>
#include <tcpmib.h>
#include <iprtrmib.h>
#include <iphlpapi.h>
#include <iostream>

#pragma comment (lib,"Iphlpapi.lib")
#pragma comment (lib,"Ws2_32.lib")
using namespace std;

bool CheckUdpPortState( IN unsigned num);

int main (void)
{
CheckUdpPortState(57016);
return 0;
}

bool CheckUdpPortState(IN unsigned num )
{
PMIB_UDPTABLE_OWNER_PID pUdpTable;
pUdpTable = new MIB_UDPTABLE_OWNER_PID;

//获取所需要的内存大小
DWORD tmpSize = sizeof(MIB_UDPTABLE_OWNER_PID);
GetExtendedUdpTable( pUdpTable, &tmpSize,false , AF_INET,  UDP_TABLE_OWNER_PID, 0);

//分配足够大小的内存并获取端口信息
DWORD dwSize = tmpSize/sizeof(MIB_UDPTABLE_OWNER_PID);
delete pUdpTable;
pUdpTable = NULL;
pUdpTable = new MIB_UDPTABLE_OWNER_PID[dwSize];
GetExtendedUdpTable( pUdpTable, &tmpSize, true, AF_INET,  UDP_TABLE_OWNER_PID, 0);

HANDLE provileges = NULL;
LUID Luid;
//提权操作
if ( !OpenProcessToken( GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES| TOKEN_QUERY, &provileges) )
{
long res = GetLastError();
cout<<"error code "<<res<<endl;
if (pUdpTable != NULL)
{
delete []pUdpTable;
pUdpTable = NULL;
}
return false;
}

if (!LookupPrivilegeValue(NULL,SE_DEBUG_NAME,&Luid))
{
cout<<"LookupPrivilegeValue err!"<<endl;
if (pUdpTable != NULL)
{
delete []pUdpTable;
pUdpTable = NULL;
}
return false;
}

TOKEN_PRIVILEGES tp;
tp.PrivilegeCount=1;
tp.Privileges[0].Attributes=SE_PRIVILEGE_ENABLED;
tp.Privileges[0].Luid=Luid;

if (!AdjustTokenPrivileges(provileges,0,&tp,sizeof(TOKEN_PRIVILEGES),NULL,NULL))
{
cout<<"AdjustTokenPrivileges err!"<<endl;
if (pUdpTable != NULL)
{
delete []pUdpTable;
pUdpTable = NULL;
}
return false;
}

//判断端口是否被占用，并找出占用端口的进程，对于某些system权限的进程需要提权
for (int i = 0; i < (int) pUdpTable->dwNumEntries; i++) {
if ( num == ntohs( (u_short) pUdpTable->table[i].dwLocalPort ) )
{
HANDLE hProcess = OpenProcess( PROCESS_ALL_ACCESS, false, pUdpTable->table[i].dwOwningPid);
if ( hProcess == NULL )
{
long res = GetLastError();
cout<<"error code "<<res<<endl;
if (pUdpTable != NULL)
{
delete []pUdpTable;
pUdpTable = NULL;
}
return false;
}
wchar_t wsProcessName[MAX_PATH + 1] = {0};
DWORD len = MAX_PATH;
if ( QueryFullProcessImageName(hProcess, 0, wsProcessName, &len) )
{
wcout<<L"Port["<<num<<L"] is occupied "<<L"by process["<<wsProcessName<<L"] PID["
<<pUdpTable->table[i].dwOwningPid<<L"]"<<endl;
CloseHandle(hProcess);
}
else
{
CloseHandle(hProcess);
hProcess = NULL;
if (pUdpTable != NULL)
{
delete []pUdpTable;
pUdpTable = NULL;
}
return false;
}
}
}

if (pUdpTable != NULL)
{
delete []pUdpTable;
pUdpTable = NULL;
}

return true;
}