檢測UDP端口占用狀態的函數

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;
}