C ++端口扫描程序

时间:2013-12-17 14:19:51

标签: c++ sockets

我正在尝试用c ++编写端口扫描程序,这样我就可以从网络中某些已打开某个端口的设备上获取IP地址。我实现了超时,因为我正在测试网络中的每个IP地址,如果有时我没有得到响应,它会自动关闭连接。

如果我把这个超时时间设置为大约30秒,它就会检测到所有关闭的设备,如果我放了一个更大的值,它会挂起并且永远不会完成。

#include <sys/socket.h>
#include <sys/time.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <netdb.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <string>

using namespace std;    

static bool port_is_open(string ip, int port){

    struct sockaddr_in address;  /* the libc network address data structure */
    short int sock = -1;         /* file descriptor for the network socket */
    fd_set fdset;
    struct timeval tv;

    address.sin_family = AF_INET;
    address.sin_addr.s_addr = inet_addr(ip.c_str()); /* assign the address */
    address.sin_port = htons(port);

    /* translate int2port num */
    sock = socket(AF_INET, SOCK_STREAM, 0);
    fcntl(sock, F_SETFL, O_NONBLOCK);

    connect(sock, (struct sockaddr *)&address, sizeof(address));

    FD_ZERO(&fdset);
    FD_SET(sock, &fdset);
    tv.tv_sec = 0;             /* timeout */
    tv.tv_usec = 50;

    if (select(sock + 1, NULL, &fdset, NULL, &tv) == 1)
    {
        int so_error;
        socklen_t len = sizeof so_error;

        getsockopt(sock, SOL_SOCKET, SO_ERROR, &so_error, &len);

        if (so_error == 0){
            close(sock);
            return true;
        }else{
            close(sock);
            return false;
        }
    }        
    return false;
}


int main(int argc, char **argv){    

    int i=1;        
    int port = 22;        
    while (i<255) {            
        string ip = "10.0.60.";                        
        std::string host = std::to_string(i);
        ip.append(host);            
        if (port_is_open(ip, port)){                
            printf("%s:%d is open\n", ip.c_str(), port);                
        }            
        i++;
    }           
    return 0;        
}

1 个答案:

答案 0 :(得分:2)

你可以将你的逻辑包装成异步调用,并以合理的超时并行启动(例如10秒,因为30us在标准条件下没有意义)。线程会使你的程序加速大约255次,在最坏的情况下,它会在超时发生后完成:

...
#include <iostream>
#include <thread>
#include <vector>
#include <sstream>
...

void task(std::string ip, int port){
    if (port_is_open(ip, port))
        cout << ip << ":" << port << " is open\n";
}

int main(int argc, char **argv){        
    const std::string ip_prefix = "10.0.60.";
    const int port = 22;
    std::vector<std::thread *> tasks;

    for (int i=0; i<255; i++){      
        std::ostringstream ip;
        ip << ip_prefix << i;
        tasks.push_back(new std::thread(task, ip.str(), port));
    }
    for (int i=0; i<255; i++){
        tasks[i]->join();
        delete tasks[i];
    }
    return 0;
}

您可能希望像下面这样编译它:g++ -std=c++11g++ -std=c++0x -pthread(适用于较旧的GCC)。