我正在尝试用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;
}
答案 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++11
或g++ -std=c++0x -pthread
(适用于较旧的GCC)。