0

我编写了一个简单的端口扫描器代码，通过建立套接字连接来查看主机是否可用。逻辑是检查连接是否建立成功，然后主机可用，否则如果我们得到错误代码为WSACONNREFUSED那么主机就存在了。套接字连接是一个阻塞连接，如果该IP地址上没有主机，则程序将被阻塞，直到发生超时。我在互联网上阅读了几个关于如何使用非阻塞套接字的示例，但他们使用读写来检查我们是否可以写入服务器端或从服务器端读取以建立连接。为此应该有一个要接受的服务器代码。有没有办法使用非阻塞套接字修改端口扫描器的现有代码，因为我希望本应用程序运行得更快一点，任何有价值的提示都会很有用。

目前的代码如下。

#ifndef UNICODE
#define UNICODE
#endif

#define WIN32_LEAN_AND_MEAN

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

// Need to link with Ws2_32.lib
#pragma comment(lib, "ws2_32.lib")

int port[]={80,139};

int wmain()
{

  // Initialize Winsock
    WSADATA wsaData;
    int i=0,flag=0;
    char ip[20];
    int iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
    if (iResult != NO_ERROR) {
    wprintf(L"WSAStartup function failed with error: %d\n", iResult);
    return 1;
   }
    SOCKET ConnectSocket;
    sockaddr_in clientService;

// Create a SOCKET for connecting to server

    ConnectSocket = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
    if (ConnectSocket == INVALID_SOCKET) {
    wprintf(L"socket function failed with error: %ld\n", WSAGetLastError());
    WSACleanup();
    return 1;
}

// The sockaddr_in structure specifies the address family,
// IP address, and port of the server to be connected to.

printf("\n Enter the Ip Address : ");
scanf("%s",ip);

clientService.sin_family = AF_INET;
clientService.sin_addr.S_un.S_addr = inet_addr(ip);


for(i=0;i<2;i++)
{
    clientService.sin_port = htons((unsigned short)port[i]);
    iResult = connect(ConnectSocket, (SOCKADDR *) & clientService, sizeof (clientService));

    if((iResult==0)||((iResult=WSAGetLastError())==WSAECONNREFUSED))
   {
     printf(" %d ",iResult);
     printf("\n Port Number : %d",port[i]);
     printf("\n Machine Found  ");
     flag=1;
     break;
   }
}

if(flag==0)
{
    printf("\n Machine not found ");
}

iResult = closesocket(ConnectSocket);
if (iResult == SOCKET_ERROR) {
    wprintf(L"closesocket function failed with  machine %d error: %ld\n",i, WSAGetLastError());
    WSACleanup();
    return 1;
}

WSACleanup();
return 0;
}
4

1 回答 1

2

在执行非阻塞后connect(),，连接完成时套接字将变为可写。此时，您应该检查SO_LASTERROR套接字以查看连接是否成功完成。

于 2013-08-13T07:47:07.163 回答