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

最新推荐文章于 2024-08-14 17:39:33 发布

转载最新推荐文章于 2024-08-14 17:39:33 发布 · 830 阅读

本文介绍了一个用于检查指定端口是否被占用，并找出占用端口的进程的方法。该方法通过调用GetExtendedTcpTable函数获取TCP连接表，然后遍历连接表来查找目标端口及其对应的进程ID。

boolCheckPortState( IN unsigned num)
{
    PMIB_TCPTABLE_OWNER_PID pTcpTable;
    pTcpTable = newMIB_TCPTABLE_OWNER_PID;
     
    //获取所需要的内存大小
    DWORDtmpSize = sizeof(MIB_TCPTABLE_OWNER_PID);
    GetExtendedTcpTable( pTcpTable, &tmpSize,false, AF_INET,  TCP_TABLE_OWNER_PID_ALL, 0);
 
    //分配足够大小的内存并获取端口信息
    DWORDdwSize = tmpSize/sizeof(MIB_TCPTABLE_OWNER_PID);
    deletepTcpTable;
    pTcpTable = NULL;
    pTcpTable = newMIB_TCPTABLE_OWNER_PID[dwSize];
    GetExtendedTcpTable( pTcpTable, &tmpSize, true, AF_INET,  TCP_TABLE_OWNER_PID_ALL, 0);
 
    //判断端口是否被占用，并找出占用端口的进程，对于某些system权限的进程需要提权
    for(inti = 0; i < (int) pTcpTable->dwNumEntries; i++) {
        if( num == ntohs( (u_short) pTcpTable->table[i].dwLocalPort ) )
        {
            HANDLEprovileges = NULL;
            LUID Luid;
            //提权操作
            if( !OpenProcessToken( GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES| TOKEN_QUERY, &provileges) )
            {
                longres = GetLastError();
                cout<<"error code "<<res<<endl;
                if(pTcpTable != NULL)
                {
                    delete[]pTcpTable;
                    pTcpTable = NULL;
                }
                returnfalse;
            }
 
            if(!LookupPrivilegeValue(NULL,SE_DEBUG_NAME,&Luid))
            {
                cout<<"LookupPrivilegeValue err!"<<endl;
                if(pTcpTable != NULL)
                {
                    delete[]pTcpTable;
                    pTcpTable = NULL;
                }
                returnfalse;
            }
 
            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(pTcpTable != NULL)
                {
                    delete[]pTcpTable;
                    pTcpTable = NULL;
                }
                returnfalse;
            }
 
            HANDLEhProcess = OpenProcess( PROCESS_ALL_ACCESS, false, pTcpTable->table[i].dwOwningPid);
            if( hProcess == NULL )
            {
                longres = GetLastError();
                cout<<"error code "<<res<<endl;
                if(pTcpTable != NULL)
                {
                    delete[]pTcpTable;
                    pTcpTable = NULL;
                }
                returnfalse;
            }
            wchar_twsProcessName[MAX_PATH + 1] = {0};
            DWORDlen = MAX_PATH;
            if( QueryFullProcessImageName(hProcess, 0, wsProcessName, &len) )
            {
                wcout<<L"Port["<<num<<L"] is occupied "<<L"by process["<<wsProcessName<<L"] PID["
                    <<pTcpTable->table[i].dwOwningPid<<L"]"<<endl;
                CloseHandle(hProcess);
            }
            else
            {
                CloseHandle(hProcess);
                hProcess = NULL;
                if(pTcpTable != NULL)
                {
                    delete[]pTcpTable;
                    pTcpTable = NULL;
                }
                returnfalse;
            }
        }
    }
 
    if(pTcpTable != NULL)
    {
        delete[]pTcpTable;
        pTcpTable = NULL;
    }
 
    returntrue;
}