vc6编译gettcptable

 

 vc6编译gettcptable 

时间:2009.02.09 19:24:00 

标签: 

微软官方例程

#include <winsock2.h>
#include <ws2tcpip.h>
#include <iphlpapi.h>
#include <stdio.h>

// Need to link with Iphlpapi.lib and Ws2_32.lib


#define MALLOC(x) HeapAlloc(GetProcessHeap(), 0, (x))
#define FREE(x) HeapFree(GetProcessHeap(), 0, (x))
/* Note: could also use malloc() and free() */

int main()
{

    // Declare and initialize variables
    PMIB_TCPTABLE pTcpTable;
    DWORD dwSize = 0;
    DWORD dwRetVal = 0;

    char szLocalAddr[128];
    char szRemoteAddr[128];

    struct in_addr IpAddr;

    int i;

    pTcpTable = (MIB_TCPTABLE *) MALLOC(sizeof (MIB_TCPTABLE));
    if (pTcpTable == NULL) {
        printf("Error allocating memory/n");
        return 1;
    }

    dwSize = sizeof (MIB_TCPTABLE);
// Make an initial call to GetTcpTable to
// get the necessary size into the dwSize variable
    if ((dwRetVal = GetTcpTable(pTcpTable, &dwSize, TRUE)) ==
        ERROR_INSUFFICIENT_BUFFER) {
        FREE(pTcpTable);
        pTcpTable = (MIB_TCPTABLE *) MALLOC(dwSize);
        if (pTcpTable == NULL) {
            printf("Error allocating memory/n");
            return 1;
        }
    }
// Make a second call to GetTcpTable to get
// the actual data we require
    if ((dwRetVal = GetTcpTable(pTcpTable, &dwSize, TRUE)) == NO_ERROR) {
        printf("/tNumber of entries: %d/n", (int) pTcpTable->dwNumEntries);
        for (i = 0; i < (int) pTcpTable->dwNumEntries; i++) {
            IpAddr.S_un.S_addr = (u_long) pTcpTable->table[i].dwLocalAddr;
            strcpy_s(szLocalAddr, sizeof (szLocalAddr), inet_ntoa(IpAddr));
            IpAddr.S_un.S_addr = (u_long) pTcpTable->table[i].dwRemoteAddr;
            strcpy_s(szRemoteAddr, sizeof (szRemoteAddr), inet_ntoa(IpAddr));

            printf("/n/tTCP[%d] State: %ld - ", i,
                   pTcpTable->table[i].dwState);
            switch (pTcpTable->table[i].dwState) {
            case MIB_TCP_STATE_CLOSED:
                printf("CLOSED/n");
                break;
            case MIB_TCP_STATE_LISTEN:
                printf("LISTEN/n");
                break;
            case MIB_TCP_STATE_SYN_SENT:
                printf("SYN-SENT/n");
                break;
            case MIB_TCP_STATE_SYN_RCVD:
                printf("SYN-RECEIVED/n");
                break;
            case MIB_TCP_STATE_ESTAB:
                printf("ESTABLISHED/n");
                break;
            case MIB_TCP_STATE_FIN_WAIT1:
                printf("FIN-WAIT-1/n");
                break;
            case MIB_TCP_STATE_FIN_WAIT2:
                printf("FIN-WAIT-2 /n");
                break;
            case MIB_TCP_STATE_CLOSE_WAIT:
                printf("CLOSE-WAIT/n");
                break;
            case MIB_TCP_STATE_CLOSING:
                printf("CLOSING/n");
                break;
            case MIB_TCP_STATE_LAST_ACK:
                printf("LAST-ACK/n");
                break;
            case MIB_TCP_STATE_TIME_WAIT:
                printf("TIME-WAIT/n");
                break;
            case MIB_TCP_STATE_DELETE_TCB:
                printf("DELETE-TCB/n");
                break;
            default:
                printf("UNKNOWN dwState value/n");
                break;
            }
            printf("/tTCP[%d] Local Addr: %s/n", i, szLocalAddr);
            printf("/tTCP[%d] Local Port: %d /n", i,
                   ntohs((u_short)pTcpTable->table[i].dwLocalPort));
            printf("/tTCP[%d] Remote Addr: %s/n", i, szRemoteAddr);
            printf("/tTCP[%d] Remote Port: %d/n", i,
                   ntohs((u_short)pTcpTable->table[i].dwRemotePort));
        }
    } else {
        printf("/tGetTcpTable failed with %d/n", dwRetVal);
        FREE(pTcpTable);
        return 1;
    }

    return 0;    
}

csdn例程

#include "Iphlpapi.h"
#include <malloc.h>
#include "Winsock2.h"
#include <stdlib.h>

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

static char TcpState[][32] =
{
 "???",
 "CLOSED",
 "LISTENING",
 "SYN_SENT",
 "SEN_RECEIVED",
 "ESTABLISHED",
 "FIN_WAIT",
 "FIN_WAIT2",
 "CLOSE_WAIT",
 "CLOSING",
 "LAST_ACK",
 "TIME_WAIT"
};

DWORD EnumTCPTable()
{
 PMIB_TCPTABLE pTcpTable = NULL;
 DWORD dwSize = 0;
 DWORD dwRetVal = ERROR_SUCCESS;

 struct   in_addr rip;
 struct   in_addr lip;
 char  szrip[32] = {0};
 char  szlip[32] = {0};
 
 //获得pTcpTable所需要的真实长度,dwSize
 if (GetTcpTable(pTcpTable, &dwSize, TRUE) == ERROR_INSUFFICIENT_BUFFER)
 {
  pTcpTable = (MIB_TCPTABLE*) malloc ((UINT) dwSize);
 }
 else
  return dwRetVal;

 printf("Active Connections/n/n");
 printf("  Proto/t%-24s%-24s%s/n","Local Address","Foreign Address","State");

 if ((dwRetVal = GetTcpTable(pTcpTable, &dwSize, TRUE)) == NO_ERROR)
 {
  for (int i = 0; i < (int) pTcpTable->dwNumEntries; i++)
  {
   rip.S_un.S_addr = pTcpTable->table[i].dwRemoteAddr;
   lip.S_un.S_addr = pTcpTable->table[i].dwLocalAddr;
   //监听端口，远程主机端口为0，但函数返回是有值的，不知道它是怎么考虑的
   if (pTcpTable->table[i].dwState == MIB_TCP_STATE_LISTEN)   
    pTcpTable->table[i].dwRemotePort = 0;

   //dwLocalPort，dwRemotePort 是网络字节
   _snprintf(szlip,sizeof(szlip),"%s:%d",inet_ntoa(lip),htons((u_short)pTcpTable->table[i].dwLocalPort));
   _snprintf(szrip,sizeof(szrip),"%s:%d",inet_ntoa(rip),htons((u_short)pTcpTable->table[i].dwRemotePort));
   printf("  TCP/t%-24s%-24s%s/n",szlip,szrip,TcpState[pTcpTable->table[i].dwState]);
  }
 }
 else
 {
  printf("/tCall to GetTcpTable failed./n");
   
  LPVOID lpMsgBuf;
   
  if (FormatMessage(
      FORMAT_MESSAGE_ALLOCATE_BUFFER |
      FORMAT_MESSAGE_FROM_SYSTEM |
      FORMAT_MESSAGE_IGNORE_INSERTS,
      NULL,
      dwRetVal,
      MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
      (LPTSTR) &lpMsgBuf,
      0,
      NULL ))
  {
   printf("/tError: %s", lpMsgBuf);
  }
  LocalFree( lpMsgBuf );
 }
 GlobalFree(pTcpTable);
 return dwRetVal;
}

DWORD DelTCPConnect(const char *lpszLocalAddr,
      const char * lpszRemoteAddr,
      DWORD dwLocalPort,
      DWORD dwRemotePort)
{
 DWORD dwRetVal = ERROR_NOT_FOUND;
 MIB_TCPROW srtTcpRow;
 srtTcpRow.dwLocalAddr = inet_addr(lpszLocalAddr);
 srtTcpRow.dwRemoteAddr = inet_addr(lpszRemoteAddr);
 srtTcpRow.dwLocalPort = htons(dwLocalPort);
 srtTcpRow.dwRemotePort = htons(dwRemotePort);
 srtTcpRow.dwState = MIB_TCP_STATE_DELETE_TCB; //目前为止，settcpEntry只支持该参数
 dwRetVal = SetTcpEntry(&srtTcpRow);
 if (dwRetVal != ERROR_SUCCESS)
 {
  LPVOID lpMsgBuf;
  if (FormatMessage(
      FORMAT_MESSAGE_ALLOCATE_BUFFER |
      FORMAT_MESSAGE_FROM_SYSTEM |
      FORMAT_MESSAGE_IGNORE_INSERTS,
      NULL,
      dwRetVal,
      MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
      (LPTSTR) &lpMsgBuf,
      0,
      NULL ))
  {
   printf("/tError: %s", lpMsgBuf);
  }
  LocalFree( lpMsgBuf );
 }
 return dwRetVal;
}

vc6编译时，产生如下错误：

wspiapi.h(45) : error C2265: '<Unknown>' : reference to a zero-sized array is illegal.

 

编写IP多播程序时,要用到ip_mrep结构,在编译时会遇到如下的错误:

/include/wspiapi.h(45) : error C2265: '<Unknown>' : reference to a zero-sized array is illegal.

这种错误是由于wspiapi.h头文件里缺少#define _WSPIAPI_COUNTOF语句造成的.

 

找到你安装的platformsdk/include目录下的WS2tcpip.h头文件

 

解决办法是在#include <wspiapi.h>之前#define _WSPIAPI_COUNTOF

 

具体代码如下：

#if !defined(_WIN32_WINNT) || (_WIN32_WINNT <= 0x0500)
#define _WSPIAPI_COUNTOF
#include <wspiapi.h>
#endif