|
|
#include <iostream>
|
|
|
#include <thread>
|
|
|
#include <vector>
|
|
|
#include <map>
|
|
|
#include <mutex>
|
|
|
#include <atomic>
|
|
|
#include <sys/socket.h>
|
|
|
#include <netinet/in.h>
|
|
|
#include <unistd.h>
|
|
|
#include <arpa/inet.h>
|
|
|
#include <cstring>
|
|
|
|
|
|
#define BUFFER_SIZE 1024
|
|
|
#define HTTP_SERVER_PORT 8080
|
|
|
|
|
|
std::vector<std::thread> threads; // 存储所有连接线程
|
|
|
std::map<int, int> sockets; // 存储套接字
|
|
|
std::mutex sockets_mutex; // 用于同步访问 sockets
|
|
|
std::atomic<bool> exit_flag(false); // 原子退出标志
|
|
|
|
|
|
|
|
|
// 维护连接的函数
|
|
|
void KeepConnection(int sock) {
|
|
|
char buffer[BUFFER_SIZE];
|
|
|
int nbytes;
|
|
|
while (!exit_flag) {
|
|
|
nbytes = recv(sock, buffer, BUFFER_SIZE, 0);
|
|
|
if (nbytes <= 0) {
|
|
|
std::cout << "Gateway disconnected" << std::endl;
|
|
|
break;
|
|
|
}
|
|
|
std::cout << "Received from gateway (Socket " << sock << "): " << buffer << std::endl;
|
|
|
}
|
|
|
|
|
|
{
|
|
|
std::lock_guard<std::mutex> lock(sockets_mutex);
|
|
|
sockets.erase(sock); // 从集合中移除套接字
|
|
|
}
|
|
|
close(sock);
|
|
|
}
|
|
|
// 连接到网关的函数
|
|
|
void ConnectToGateway(const std::string& ip, int port) {
|
|
|
int sock = socket(AF_INET, SOCK_STREAM, 0);
|
|
|
if (sock < 0) {
|
|
|
perror("Could not create socket");
|
|
|
return;
|
|
|
}
|
|
|
|
|
|
struct sockaddr_in server_addr;
|
|
|
server_addr.sin_family = AF_INET;
|
|
|
server_addr.sin_port = htons(port);
|
|
|
if (inet_pton(AF_INET, ip.c_str(), &server_addr.sin_addr) <= 0) {
|
|
|
perror("Invalid address");
|
|
|
close(sock);
|
|
|
return;
|
|
|
}
|
|
|
|
|
|
if (connect(sock, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
|
|
|
perror("Connection to the server failed");
|
|
|
close(sock);
|
|
|
return;
|
|
|
}
|
|
|
|
|
|
{
|
|
|
std::lock_guard<std::mutex> lock(sockets_mutex);
|
|
|
sockets[sock] = 1; // 将套接字添加到集合中
|
|
|
}
|
|
|
|
|
|
std::cout << "Connected to gateway at " << ip << ":" << port << " (Socket " << sock << ")" << std::endl;
|
|
|
|
|
|
// 在新线程中维护连接
|
|
|
threads.emplace_back(KeepConnection, sock);
|
|
|
}
|
|
|
|
|
|
// 处理HTTP请求的函数
|
|
|
void HandleHttpRequest(int client_socket) {
|
|
|
char buffer[BUFFER_SIZE];
|
|
|
int bytes_read = recv(client_socket, buffer, sizeof(buffer), 0);
|
|
|
if (bytes_read <= 0) {
|
|
|
close(client_socket);
|
|
|
return;
|
|
|
}
|
|
|
|
|
|
// 简单解析HTTP请求
|
|
|
std::string http_request(buffer, bytes_read);
|
|
|
std::string ip;
|
|
|
int port;
|
|
|
|
|
|
// 寻找IP和端口号
|
|
|
auto host_pos = http_request.find("Host: ");
|
|
|
if (host_pos != std::string::npos) {
|
|
|
auto start = host_pos + std::string("Host: ").size();
|
|
|
ip = http_request.substr(start, http_request.find("\r\n", start) - start);
|
|
|
}
|
|
|
|
|
|
auto content_length_pos = http_request.find("Content-Length:");
|
|
|
if (content_length_pos != std::string::npos) {
|
|
|
auto start = content_length_pos + std::string("Content-Length: ").size();
|
|
|
int length = std::stoi(http_request.substr(start, http_request.find("\r\n", start) - start));
|
|
|
if (length > 0) {
|
|
|
char* content = new char[length];
|
|
|
bytes_read = recv(client_socket, content, length, 0);
|
|
|
if (bytes_read > 0) {
|
|
|
// 解析JSON内容（假设前端发送的是JSON格式）
|
|
|
// 这里需要添加JSON解析逻辑来提取IP和端口
|
|
|
// 示例：{"ip":"192.168.1.1","port":8080}
|
|
|
// 可以使用第三方库如nlohmann/json来解析JSON
|
|
|
|
|
|
// 假设解析后得到ip和port
|
|
|
// ConnectToGateway(ip, port);
|
|
|
|
|
|
delete[] content;
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// 发送HTTP响应
|
|
|
std::string response = "HTTP/1.1 200 OK\r\nContent-Type: text/plain\r\n\r\n";
|
|
|
send(client_socket, response.c_str(), response.size(), 0);
|
|
|
close(client_socket);
|
|
|
}
|
|
|
|
|
|
// 启动HTTP服务器的函数
|
|
|
void StartHttpServer() {
|
|
|
int http_server_socket = socket(AF_INET, SOCK_STREAM, 0);
|
|
|
if (http_server_socket < 0) {
|
|
|
perror("Could not create HTTP server socket");
|
|
|
return;
|
|
|
}
|
|
|
|
|
|
int opt = 1;
|
|
|
if (setsockopt(http_server_socket, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &opt, sizeof(opt))) {
|
|
|
perror("Setsockopt failed");
|
|
|
close(http_server_socket);
|
|
|
return;
|
|
|
}
|
|
|
|
|
|
struct sockaddr_in server_addr;
|
|
|
server_addr.sin_family = AF_INET;
|
|
|
server_addr.sin_addr.s_addr = INADDR_ANY;
|
|
|
server_addr.sin_port = htons(HTTP_SERVER_PORT);
|
|
|
|
|
|
if (bind(http_server_socket, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
|
|
|
perror("Bind failed");
|
|
|
close(http_server_socket);
|
|
|
return;
|
|
|
}
|
|
|
|
|
|
if (listen(http_server_socket, 5) < 0) {
|
|
|
perror("Listen failed");
|
|
|
close(http_server_socket);
|
|
|
return;
|
|
|
}
|
|
|
|
|
|
std::cout << "HTTP server listening on port " << HTTP_SERVER_PORT << std::endl;
|
|
|
|
|
|
while (!exit_flag) {
|
|
|
int client_socket = accept(http_server_socket, NULL, NULL);
|
|
|
if (client_socket < 0) {
|
|
|
perror("Accept failed");
|
|
|
continue;
|
|
|
}
|
|
|
|
|
|
threads.emplace_back(HandleHttpRequest, client_socket);
|
|
|
}
|
|
|
|
|
|
close(http_server_socket);
|
|
|
}
|
|
|
|
|
|
int main() {
|
|
|
// threads.emplace_back(StartHttpServer);
|
|
|
|
|
|
// std::string command, ip;
|
|
|
// int port;
|
|
|
// bool in_connect_mode = false;
|
|
|
// std::cout << "Enter command (connect/exit): ";
|
|
|
// while (true) {
|
|
|
|
|
|
// if (in_connect_mode)
|
|
|
// {
|
|
|
// std::cout << "continue connect or not ? (y or n) ";
|
|
|
// std::cin >> command;
|
|
|
// if(command=="y"){
|
|
|
// std::cout << "Enter the gateway IP address: ";
|
|
|
// std::cin >> ip;
|
|
|
// std::cout << "Enter the gateway port: ";
|
|
|
// std::cin >> port;
|
|
|
// ConnectToGateway(ip, port);
|
|
|
// }
|
|
|
// else {
|
|
|
// in_connect_mode = false; // 退出连接模式
|
|
|
// std::cout << "Exiting connect mode." << std::endl;
|
|
|
// }
|
|
|
// }
|
|
|
// else{
|
|
|
// std::cout << "Enter command (connect/exit): ";
|
|
|
// std::cin>>command;
|
|
|
// if (command == "connect") {
|
|
|
// if (in_connect_mode) {
|
|
|
// std::cout << "Already in connect mode." << std::endl;
|
|
|
// continue;
|
|
|
// }
|
|
|
// in_connect_mode = true; // 进入连接模式
|
|
|
// std::cout << "Enter the gateway IP address: ";
|
|
|
// std::cin >> ip;
|
|
|
// std::cout << "Enter the gateway port: ";
|
|
|
// std::cin >> port;
|
|
|
// ConnectToGateway(ip, port);
|
|
|
// } else if (command == "exit") {
|
|
|
// exit_flag = true; // 设置退出标志
|
|
|
// std::cout << "Exiting program." << std::endl;
|
|
|
// // 关闭所有套接字
|
|
|
// for (auto& sock_pair : sockets) {
|
|
|
// shutdown(sock_pair.first, SHUT_RDWR); // 关闭套接字的发送和接收
|
|
|
// close(sock_pair.first);
|
|
|
// }
|
|
|
// sockets.clear();
|
|
|
// std::cout << "sockets.clear" << std::endl;
|
|
|
// // 等待所有线程结束
|
|
|
// for (auto& thread : threads) {
|
|
|
// if (thread.joinable()) {
|
|
|
// thread.join();
|
|
|
// }
|
|
|
// }
|
|
|
// threads.clear();
|
|
|
// break; // 退出主循环
|
|
|
// } else {
|
|
|
// std::cout << "Unknown command" << std::endl;
|
|
|
// }
|
|
|
// }
|
|
|
// }
|
|
|
// std::cout<<"out"<<std::endl;
|
|
|
// return 0;
|
|
|
std::thread http_server_thread(StartHttpServer);
|
|
|
|
|
|
std::string command, ip;
|
|
|
int port;
|
|
|
bool in_connect_mode = false;
|
|
|
|
|
|
while (true) {
|
|
|
std::cout << "Enter command (connect/exit): ";
|
|
|
std::cin >> command;
|
|
|
|
|
|
if (command == "connect") {
|
|
|
if (in_connect_mode) {
|
|
|
std::cout << "Already in connect mode." << std::endl;
|
|
|
continue;
|
|
|
}
|
|
|
in_connect_mode = true;
|
|
|
std::cout << "Enter the gateway IP address: ";
|
|
|
std::cin >> ip;
|
|
|
std::cout << "Enter the gateway port: ";
|
|
|
std::cin >> port;
|
|
|
ConnectToGateway(ip, port);
|
|
|
} else if (command == "exit") {
|
|
|
break; // 接收到退出命令，退出主循环
|
|
|
} else {
|
|
|
std::cout << "Unknown command" << std::endl;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// 设置退出标志
|
|
|
exit_flag = true;
|
|
|
|
|
|
// 强制结束HTTP服务器线程
|
|
|
if (http_server_thread.joinable()) {
|
|
|
http_server_thread.join();
|
|
|
}
|
|
|
|
|
|
// 关闭所有套接字
|
|
|
for (auto& sock_pair : sockets) {
|
|
|
shutdown(sock_pair.first, SHUT_RDWR);
|
|
|
close(sock_pair.first);
|
|
|
}
|
|
|
sockets.clear();
|
|
|
|
|
|
// 等待所有工作线程结束
|
|
|
for (auto& thread : threads) {
|
|
|
if (thread.joinable()) {
|
|
|
thread.join();
|
|
|
}
|
|
|
}
|
|
|
threads.clear();
|
|
|
|
|
|
std::cout << "Exited program." << std::endl;
|
|
|
return 0;
|
|
|
|
|
|
} |
