Socket编程中检测端口是否被占用

来源:互联网 发布:php https post请求 编辑:程序博客网 时间:2024/05/28 14:57


来源:Socket编程中检测端口是否被占用


   一般检测一个端口是否被占用的方法是看bind是否成功,其实在Windows中有两个API可以获取到当前系统端口的占用情况(GetTcpTable/GetUdpTable),利用这两个函数就可检测端口占用情况。
但在较复杂系统环境下,端口占用表可能随时改变,可能存在两个以上进程使用上述方法检测到同一空闲端口,在这种情况下,可使用Bind方法进行端口选择。
一、通过GetTcpTable/GetUdpTable方法获取端口占用情况。
[cpp] view plain copy
  1. #include <tcpmib.h>  
  2. #include <IPHlpApi.h>  
  3. //依赖lib库 Iphlpapi.lib Ws2_32.lib  
  4.   
  5. //获取Tcp端口状态  
  6. BOOL GetTcpPortState(ULONG nPort, ULONG *nStateID)  
  7. {  
  8.     MIB_TCPTABLE TcpTable[100];  
  9.     DWORD nSize = sizeof(TcpTable);  
  10.     if(NO_ERROR == GetTcpTable(&TcpTable[0],&nSize,TRUE))  
  11.     {  
  12.         DWORD nCount = TcpTable[0].dwNumEntries;  
  13.         if (nCount > 0)  
  14.         {  
  15.             for(DWORD i=0;i<nCount;i++)  
  16.             {  
  17.                 MIB_TCPROW TcpRow = TcpTable[0].table[i];  
  18.                 DWORD temp1 = TcpRow.dwLocalPort;  
  19.                 int temp2 = temp1 / 256 + (temp1 % 256) * 256;  
  20.                 if(temp2 == nPort)  
  21.                 {  
  22.                     *nStateID = TcpRow.dwState;  
  23.                     return TRUE;  
  24.                 }  
  25.             }  
  26.         }  
  27.         return FALSE;  
  28.     }  
  29.     return FALSE;  
  30. }  
  31.   
  32. //获取Udp端口状态  
  33. BOOL GetUdpPortState(ULONG nPort, ULONG *nStateID)  
  34. {  
  35.     MIB_UDPTABLE UdpTable[100];  
  36.     DWORD nSize = sizeof(UdpTable);  
  37.     if(NO_ERROR == GetUdpTable(&UdpTable[0],&nSize,TRUE))  
  38.     {  
  39.         DWORD nCount = UdpTable[0].dwNumEntries;  
  40.         if (nCount > 0)  
  41.         {  
  42.             for(DWORD i=0;i<nCount;i++)  
  43.             {  
  44.                 MIB_UDPROW TcpRow = UdpTable[0].table[i];  
  45.                 DWORD temp1 = TcpRow.dwLocalPort;  
  46.                 int temp2 = temp1 / 256 + (temp1 % 256) * 256;  
  47.                 if(temp2 == nPort)  
  48.                 {  
  49.                     return TRUE;  
  50.                 }  
  51.             }  
  52.         }  
  53.         return FALSE;  
  54.     }  
  55.     return FALSE;  
  56. }  
二、通过bind函数,判断端口是否被占用。
1、原始socket编程时,就使用bind函数判断。
[cpp] view plain copy
  1.  unsigned short usPort = 9090;  
  2.  SOCKET s = socket(AF_INET,SOCK_STREAM,IPPROTO_IP);  
  3. sockaddr_in addr;  
  4. addr.sin_family = AF_INET;  
  5. addr.sin_port = htons(usPort);  
  6. addr.sin_addr.s_addr = htonl(INADDR_ANY);  
  7. bind(s,(LPSOCKADDR)&addr,sizeof(addr));  
  8. if(WSAGetLastError()==WSAEADDRINUSE)  
  9. {  
  10.     //端口已被占用  
  11. }  
2、使用boost asio库时,如何判断端口被占用,使用new socket或bind失败时来判断。
[cpp] view plain copy
  1. boost::asio::io_service ioService;  
  2. unsigned short usPort = 9090;  
  3. udp::socket * pSockUdp = NULL;  
  4. bool bSockUdpError = false;  
  5. try{  
  6.         pSockUdp = new udp::socket(ioService,udp::endpoint(udp::v4(),usPort));  
  7.         bSockUdpError = false;  
  8.     }  
  9.     catch(...)  
  10.     {  
  11.         bSockUdpError = true;  
  12.     }  
  13. //释放  
  14.     if(!bSockUdpError && pSockUdp != NULL)  
  15.     {  
  16.         delete pSockUdp;  
  17.         pSockUdp = NULL;  
  18.     }  



原创粉丝点击