Question

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

目前的代码如下。

#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;
}

Answer 1

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

在端口扫描程序中使用非阻塞套接字

1 个答案: