社区首页 >专栏 >C/C++ 实现枚举网上邻居信息

C/C++ 实现枚举网上邻居信息

王瑞

发布于 2023-12-06 06:32:57

37910

代码可运行

文章被收录于专栏：Visual C++ 编程技术实践Visual C++ 编程技术实践

运行总次数：0

代码可运行

在Windows系统中，通过网络邻居可以方便地查看本地网络中的共享资源和计算机。通过使用Windows API中的一些网络相关函数，我们可以实现枚举网络邻居信息的功能，获取连接到本地网络的其他计算机的相关信息。本文将介绍一个简单的C++程序，使用Windows API枚举网络邻居信息，并获取对端名称、本机名称、主机名称以及主机IP等信息。

引言

首先，我们需要了解一下几个与网络邻居信息有关的Windows API函数：

NETRESOURCE 结构体是 Windows 网络 API 中的一个结构体，用于描述网络资源的信息。它在网络资源的枚举和连接等操作中经常被使用。

结构体定义

typedef struct _NETRESOURCE {
  DWORD         dwScope;
  DWORD         dwType;
  DWORD         dwDisplayType;
  DWORD         dwUsage;
  LPTSTR        lpLocalName;
  LPTSTR        lpRemoteName;
  LPTSTR        lpComment;
  LPTSTR        lpProvider;
} NETRESOURCE, *LPNETRESOURCE;

结构体成员

dwScope：指定资源的可见性。可能的取值为：
- RESOURCE_CONNECTED：表示已连接的资源。
- RESOURCE_CONTEXT：表示会话特定的资源。
dwType：指定资源的类型。可能的取值为：
- RESOURCETYPE_ANY：表示任何类型的资源。
- RESOURCETYPE_DISK：表示磁盘资源。
- RESOURCETYPE_PRINT：表示打印机资源。
dwDisplayType：指定资源的显示类型。可能的取值为：
- RESOURCEDISPLAYTYPE_GENERIC：表示通用资源。
- RESOURCEDISPLAYTYPE_DOMAIN：表示域。
dwUsage：指定资源的使用类型。可能的取值为：
- RESOURCEUSAGE_CONNECTABLE：表示资源可以连接。
- RESOURCEUSAGE_CONTAINER：表示资源是一个容器，可以包含其他资源。
lpLocalName：指定本地驱动器或打印机的本地名称。
lpRemoteName：指定远程共享资源的网络名称。
lpComment：指定资源的注释。
lpProvider：指定提供资源的网络提供程序的名称。

WNetOpenEnum 是 Windows 网络功能中的一个API函数，WNetOpenEnum 通过指定的参数打开一个用于列举网络资源的枚举句柄。这个句柄可以在后续的操作中用于获取网络资源的枚举信息。

函数原型

DWORD WNetOpenEnum(
  DWORD         dwScope,
  DWORD         dwType,
  DWORD         dwUsage,
  LPNETRESOURCE lpNetResource,
  LPHANDLE      lphEnum
);

参数说明

dwScope：指定枚举的范围，可以是 RESOURCE_CONNECTED、RESOURCE_GLOBALNET、RESOURCE_REMEMBERED、RESOURCE_CONTEXT 中的一个。
dwType：指定枚举资源的类型，可以是 RESOURCETYPE_DISK、RESOURCETYPE_PRINT、RESOURCETYPE_ANY 中的一个。
dwUsage：指定枚举资源的使用类型，可以是 RESOURCEUSAGE_CONNECTABLE、RESOURCEUSAGE_CONTAINER、RESOURCEUSAGE_NOLOCALDEVICE、RESOURCEUSAGE_SIBLING 中的一个。
lpNetResource：指向 NETRESOURCE 结构的指针，用于指定枚举的起始位置。如果为 NULL，则从所有资源的最顶层开始枚举。
lphEnum：指向句柄的指针，用于接收枚举句柄。

返回值

如果函数调用成功，返回 NO_ERROR，否则返回一个错误代码，可以通过 GetLastError 函数获取详细信息。

WNetEnumResource 是 Windows 网络功能中的 API 函数之一，用于通过枚举句柄获取网络资源的详细信息，包括资源的类型、用途、本地名、远程名等信息。

函数原型

DWORD WNetEnumResource(
  HANDLE  hEnum,
  LPDWORD lpcCount,
  LPVOID  lpBuffer,
  LPDWORD lpBufferSize
);

参数说明

hEnum：枚举句柄，通过 WNetOpenEnum 函数获取的句柄。
lpcCount：指向一个变量的指针，该变量用于接收枚举的资源数目。
lpBuffer：指向一个缓冲区的指针，用于接收枚举的资源信息。
lpBufferSize：指向一个变量的指针，该变量用于指定或接收缓冲区的大小。

返回值

如果函数调用成功，返回 NO_ERROR，否则返回一个错误代码，可以通过 GetLastError 函数获取详细信息。

WNetCloseEnum 函数用于关闭由 WNetOpenEnum 函数打开的网络资源的枚举句柄。在使用 WNetOpenEnum 函数枚举网络资源后，当不再需要使用枚举句柄时，应该通过调用 WNetCloseEnum 函数来释放资源，避免内存泄漏。

函数定义

DWORD WNetCloseEnum(
  HANDLE hEnum
);

参数

hEnum：网络资源的枚举句柄，该句柄是由 WNetOpenEnum 函数返回的。

返回值

如果函数调用成功，返回 NO_ERROR，否则返回错误代码，可以通过 GetLastError 函数获取详细的错误信息。

代码实现

以下是一个简单的C++程序，使用上述API函数实现了枚举网络邻居信息的功能。该程序通过遍历枚举得到的网络资源信息，获取对端名称、本机名称、主机名称以及主机IP等信息，并输出到控制台。

#include <iostream>
#include <Windows.h>
#include <winnetwk.h>

#pragma comment(lib,"Mpr.lib")
#pragma comment(lib,"Ws2_32.lib")

// 枚举网络邻居
BOOL EnumNetResource()
{
  NETRESOURCE  *NetResource = NULL;
  HANDLE  hEnum;
  unsigned  int  i;
  char szHostName[MAX_PATH] = { 0 };
  hostent *host = NULL;
  char *lpszIP = NULL;

  // 通过WSAStartup函数完成对Winsock服务的初始化
  WSADATA wsaData = { 0 };
  WSAStartup(MAKEWORD(2, 2), &wsaData);

  // 指定枚举范围, 获取枚举句柄
  WNetOpenEnum(RESOURCE_CONTEXT, NULL, NULL, NULL, &hEnum);
  if (hEnum)
  {
    DWORD  Count = 0xFFFFFFFF;
    DWORD  BufferSize = 2048;
    BYTE *pBuffer = new  BYTE[2048];

    // 根据设置的枚举返回, 获取枚举信息
    WNetEnumResource(hEnum, &Count, pBuffer, &BufferSize);
    NetResource = (NETRESOURCE*)pBuffer;

    for (i = 0; i < BufferSize / sizeof(NETRESOURCE); i++, NetResource++)
    {
      // 判断资源类型是否是所有资源 以及 判断资源使用类型是否是容器资源
      if (NetResource->dwUsage == RESOURCEUSAGE_CONTAINER  && NetResource->dwType == RESOURCETYPE_ANY)
      {
        if (NetResource->lpRemoteName)
        {
          // 获取远程主机名
          RtlZeroMemory(szHostName, MAX_PATH);
          lstrcpy(szHostName, (char *)((DWORD64)NetResource->lpRemoteName + 2));

          // 根据主机名获取IP地址信息
          host = gethostbyname(szHostName);
          if (host == NULL)
          {
            std::cout << "错误代码" << std::endl;
            continue;
          }

          // 将Ipv4网络地址转换为Internet标准点分十进制格式的ASCII字符串
          lpszIP = inet_ntoa(*(in_addr *)host->h_addr_list[0]);
          std::cout << "对端名称: " << NetResource->lpRemoteName
            << " 本机名称: " << NetResource->lpLocalName
            << " 主机名称: " << szHostName
            << " 主机IP: " << lpszIP
            << std::endl;
        }
      }
    }

    // 释放内存并关闭句柄
    delete[]pBuffer;
    pBuffer = NULL;
    WNetCloseEnum(hEnum);
  }
  return TRUE;
}

int main(int argc, char *argv[])
{
  EnumNetResource();

  system("pause");
  return 0;
}