void IRC_Server::receive_datagram(Datagram datagram, Host & host) { std::smatch result; std::string datagram_id = host.get_name(); std::string message = datagram.get_message(); if (std::regex_search(message, result, std::regex("^CONNECT$")) || std::regex_search(message, result, std::regex("^USER [\\w\\d]+$")) || std::regex_search(message, result, std::regex("^QUIT$")) ) { datagram_id = datagram_id + "." + std::to_string(host.get_count_and_add()); std::string source_ip = host.ip_; int source_port = IRC_SERVER_PORT; std::string destination_ip = datagram.get_source_ip(); int destination_port = datagram.get_source_port(); std::string content = ":" + host.get_name() + " " + message; host.add_to_send_datagram_queue(Datagram(source_ip, destination_ip, new UDP_Segment(source_port, destination_port, content), datagram_id)); } }
void IRC_Client::process_command(std::string command, Host & host) { std::smatch result; std::string datagram_id = host.get_name(); if (std::regex_search(command, result, std::regex("^CONNECT ([\\w\\d]+)\\s+(\\d+)$"))) { // precisa de DNS datagram_id = datagram_id + "." + std::to_string(host.get_count_and_add()); std::string content = std::string(result[1]) + " A IN"; host.add_to_send_datagram_queue(Datagram(host.ip_, host.dns_server_ip_, new UDP_Segment(DNS_CLIENT_PORT, DNS_SERVER_PORT, content), datagram_id)); commands_waiting_for_dns.push_back(command); } else if (std::regex_search(command, result, std::regex("^CONNECT ([\\d\\.]+)\\s+(\\d+)$"))) { // não precisa de DNS datagram_id = datagram_id + "." + std::to_string(host.get_count_and_add()); std::string content = "CONNECT"; std::string destination_ip = result[1]; int destination_port = stoi(result[2]); std::string source_ip = host.ip_; int source_port = IRC_CLIENT_PORT; host.add_to_send_datagram_queue(Datagram(source_ip, destination_ip, new UDP_Segment(source_port, destination_port, content), datagram_id)); } else if (std::regex_search(command, result, std::regex("^USER ([\\w\\d]+)$")) || std::regex_search(command, result, std::regex("^QUIT$"))) { if (!connected_server_ip.empty() && connected_server_port != -1) { datagram_id = datagram_id + "." + std::to_string(host.get_count_and_add()); std::string source_ip = host.ip_; int source_port = IRC_CLIENT_PORT; std::string destination_ip = connected_server_ip; int destination_port = connected_server_port; host.add_to_send_datagram_queue(Datagram(source_ip, destination_ip, new UDP_Segment(source_port, destination_port, command), datagram_id)); } else printf("Cliente IRC não conectado a nenhum servidor\n"); } else printf("Comando desconhecido no %s\n", host.get_name().c_str()); }
void IRC_Client::receive_datagram(Datagram datagram, Host & host) { int port = datagram.get_destination_port(); if (port == DNS_CLIENT_PORT) { std::string dns_response = datagram.get_message(); std::smatch result; std::regex_search(dns_response, result, std::regex("^(\\w+) A IN ([\\d\\.]+)$")); std::string response_hostname = result[1]; std::string response_ip = result[2]; for (std::string cmd : commands_waiting_for_dns) { std::smatch result2; std::regex_search(cmd, result2, std::regex("^CONNECT ([\\w\\d]+)\\s+(\\d+)$")); if (result2[1] == response_hostname) { std::string new_command = std::string("CONNECT ") + response_ip + " " + std::string(result2[2]); process_command(new_command, host); } } } else if (port == IRC_CLIENT_PORT) { std::string irc_response = datagram.get_message(); std::smatch result; if (std::regex_search(irc_response, result, std::regex("^:([\\w\\d]+) CONNECT$"))) { connected_server_ip = datagram.get_source_ip(); connected_server_port = datagram.get_source_port(); } else if (!std::regex_search(irc_response, result, std::regex("^:([\\w\\d]+) USER .+$")) && !std::regex_search(irc_response, result, std::regex("^:([\\w\\d]+) QUIT$"))) printf("Datagrama desconhecido no %s\n", host.get_name().c_str()); } }