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

来源：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.     }