getconntable.cpp

VisualC++网络程序设计配套光盘代码第七章

//////////////////////////////////////////////////////////////////////
// GetInterfaceInfo.cpp文件


#include <stdio.h>
#include <windows.h>
#include <Iphlpapi.h>

#pragma comment(lib, "Iphlpapi.lib")
#pragma comment(lib, "WS2_32.lib")


PMIB_TCPTABLE MyGetTcpTable(BOOL bOrder);
PMIB_UDPTABLE MyGetUdpTable(BOOL bOrder);

void MyFreeTcpTable(PMIB_TCPTABLE pTcpTable);
void MyFreeUdpTable(PMIB_UDPTABLE pUdpTable);


int main()
{
	// 打印TCP连接表信息
	PMIB_TCPTABLE pTcpTable = MyGetTcpTable(TRUE);
	if(pTcpTable != NULL)
	{
		char    strState[128];
		struct  in_addr    inadLocal, inadRemote;
		DWORD   dwRemotePort = 0;
		char    szLocalIp[128];
		char    szRemIp[128];

		printf("TCP TABLE\n");
		printf("%20s %10s %20s %10s %s\n", "Loc Addr", "Loc Port", "Rem Addr",
			"Rem Port", "State");
		for(UINT i = 0; i < pTcpTable->dwNumEntries; ++i)
		{
			// 状态
			switch (pTcpTable->table[i].dwState)
			{
			case MIB_TCP_STATE_CLOSED:
				strcpy(strState, "CLOSED");
				break;
			case MIB_TCP_STATE_TIME_WAIT:
				strcpy(strState, "TIME_WAIT");
				break;
			case MIB_TCP_STATE_LAST_ACK:
				strcpy(strState, "LAST_ACK");
				break;
			case MIB_TCP_STATE_CLOSING:
				strcpy(strState, "CLOSING");
				break;
			case MIB_TCP_STATE_CLOSE_WAIT:
				strcpy(strState, "CLOSE_WAIT");
				break;
			case MIB_TCP_STATE_FIN_WAIT1:
				strcpy(strState, "FIN_WAIT1");
				break;
			case MIB_TCP_STATE_ESTAB:
				strcpy(strState, "ESTAB");
				break;
			case MIB_TCP_STATE_SYN_RCVD:
				strcpy(strState, "SYN_RCVD");
				break;
			case MIB_TCP_STATE_SYN_SENT:
				strcpy(strState, "SYN_SENT");
				break;
			case MIB_TCP_STATE_LISTEN:
				strcpy(strState, "LISTEN");
				break;
			case MIB_TCP_STATE_DELETE_TCB:
				strcpy(strState, "DELETE");
				break;
			default:
				printf("Error: unknown state!\n");
				break;
			}
			// 本地地址
			inadLocal.s_addr = pTcpTable->table[i].dwLocalAddr;
			
			// 远程端口
			if (strcmp(strState, "LISTEN") != 0)
			{
				dwRemotePort = pTcpTable->table[i].dwRemotePort;
			}
			else
				dwRemotePort = 0;
			// 远程IP地址
			inadRemote.s_addr = pTcpTable->table[i].dwRemoteAddr;

			strcpy(szLocalIp, inet_ntoa(inadLocal));
			strcpy(szRemIp, inet_ntoa(inadRemote));
			printf("%20s %10u %20s %10u %s\n", 
				szLocalIp,  ntohs((unsigned short)(0x0000FFFF & pTcpTable->table[i].dwLocalPort)),
				szRemIp, ntohs((unsigned short)(0x0000FFFF & dwRemotePort)),
				strState);
		}
		MyFreeTcpTable(pTcpTable);
	}

	printf(" \n\n");

	// 打印UDP监听表信息
	PMIB_UDPTABLE pUdpTable = MyGetUdpTable(TRUE);
	if(pUdpTable != NULL)
	{
		struct in_addr inadLocal;
		printf("UDP TABLE\n");

        printf("%20s %10s\n", "Loc Addr", "Loc Port");
        for (UINT i = 0; i < pUdpTable->dwNumEntries; ++i)
        {
            inadLocal.s_addr = pUdpTable->table[i].dwLocalAddr;
			// 打印出此入口的信息
            printf("%20s %10u \n", 
                inet_ntoa(inadLocal), ntohs((unsigned short)(0x0000FFFF & pUdpTable->table[i].dwLocalPort)));
        }

		MyFreeUdpTable(pUdpTable);
	}
		
	return 0;
}


PMIB_TCPTABLE MyGetTcpTable(BOOL bOrder)
{
	PMIB_TCPTABLE pTcpTable = NULL;
    DWORD dwActualSize = 0;


    // 查询所需缓冲区的大小
    if(::GetTcpTable(pTcpTable, &dwActualSize, bOrder) == ERROR_INSUFFICIENT_BUFFER)
	{
        // 为MIB_TCPTABLE结构申请内存
        pTcpTable = (PMIB_TCPTABLE)::GlobalAlloc(GPTR, dwActualSize);
        // 获取TCP连接表
        if(::GetTcpTable(pTcpTable, &dwActualSize, bOrder) == NO_ERROR)
			return pTcpTable;
		::GlobalFree(pTcpTable);
    }
	return NULL;
}

void MyFreeTcpTable(PMIB_TCPTABLE pTcpTable)
{
	if(pTcpTable != NULL)
		::GlobalFree(pTcpTable);
}


PMIB_UDPTABLE MyGetUdpTable(BOOL bOrder)
{
	PMIB_UDPTABLE pUdpTable = NULL;
    DWORD dwActualSize = 0;

    // 查询所需缓冲区的大小
    if(::GetUdpTable(pUdpTable, &dwActualSize, bOrder) == ERROR_INSUFFICIENT_BUFFER)
	{
        // 为MIB_UDPTABLE结构申请内存
        pUdpTable = (PMIB_UDPTABLE)::GlobalAlloc(GPTR, dwActualSize);
        // 获取UDP监听表
        if(::GetUdpTable(pUdpTable, &dwActualSize, bOrder) == NO_ERROR)
			return pUdpTable;
		::GlobalFree(pUdpTable);
    }
	return NULL;
}

void MyFreeUdpTable(PMIB_UDPTABLE pUdpTable)
{
	if(pUdpTable != NULL)
		::GlobalFree(pUdpTable);
}