diff --git a/src/DosAtk.cpp b/src/DosAtk.cpp index a9a631f..de1d33b 100644 --- a/src/DosAtk.cpp +++ b/src/DosAtk.cpp @@ -55,7 +55,6 @@ std::string log_msg; // Сообщение, std::string fin_msg; // Сообщение, которое будет выведено в консоль пользователю std::string msg; // Сообщение, которое будет отправлено в телеграм int status; // Статус работы программы -// int log_status; /* ██████╗ ██████╗██╗ ██████╗ ██████╗ ██████╗ ██████╗███████╗██████╗ ██╗ ██╗██████╗ ███████╗███████╗ @@ -83,10 +82,10 @@ int my_check_params() //Объявление std::chrono::milliseconds ms; // Количество миллисекунд от целой секунды времени начала выполнения программы - std::time_t now_time_t; // текущее время + std::time_t now_time_t; // Текущее время int opt; // Прочитанный параметр const char* short_options; // Сокращения для параметров - struct option long_options[10]; // Структура, описывающая пеобходимые программе параметры + struct option long_options[10]; // Структура, описывающая пеобходимые программе параметры int i; // Счётчик для цикла // Инициализация @@ -180,7 +179,91 @@ int my_check_params() return -101; } -void my_diag() +int my_udp() { + int sockfd; + int port_idx; + int curr_port; + int ports_total; + int ports[14]; + char dummy_data[5]; + + sockfd = -1; // Дескриптор сокета + struct sockaddr_in target_addr; // Адрес цели + port_idx = 0; // Текущий индекс порта + + // Выполняет UDP портовое сканирование well-known портов + + ports[0] = 53; // DNS + ports[1] = 67; // DHCP (сервер) + ports[2] = 68; // DHCP (клиент) + ports[3] = 69; // TFTP + ports[4] = 123; // NTP + ports[5] = 161; // SNMP + ports[6] = 162; // SNMP Trap + ports[7] = 389; // LDAP + ports[8] = 443; // HTTPS + ports[9] = 500; // IPSec + ports[10] = 514; // Syslog + ports[11] = 520; // RIP + ports[12] = 1900; // SSDP + ports[13] = 4500; // NAT-T + + ports_total = sizeof(ports)/sizeof(ports[0]); + curr_port = ports[port_idx]; // Текущий порт + + + // Данные для отправки + // Поэлементная инициализация + dummy_data[0] = 'S'; + dummy_data[1] = 'C'; + dummy_data[2] = 'A'; + dummy_data[3] = 'N'; + dummy_data[4] = '\0'; // Обязательный терминатор строки + + ssize_t send_result; // Результат отправки + + // Инициализация структуры адреса + memset(&target_addr, 0, sizeof(target_addr)); + target_addr.sin_family = AF_INET; + target_addr.sin_port = htons(curr_port); + + // Преобразование IP + if (inet_pton(AF_INET, global_ip, &target_addr.sin_addr) <= 0) { + n_fail_requests++; + status = -501; // Код ошибки: неверный IP + if (sockfd != -1) close(sockfd); + } + + // Создание сокета + sockfd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); + if (sockfd < 0) { + n_fail_requests++; + status = -502; // Ошибка создания сокета + if (sockfd != -1) close(sockfd); + } + + // Отправка данных + send_result = sendto(sockfd, dummy_data, sizeof(dummy_data), 0, + (struct sockaddr*)&target_addr, sizeof(target_addr)); + if (send_result < 0) { + n_fail_requests++; + status = -503; // Ошибка отправки + } else { + n_ok_requests++; + status = 0; // Успешная отправка + } + + // Проверка общего числа запросов + if (n_ok_requests + n_fail_requests >= 1000) { + status = 2; // Условие завершения + } + + // Переход к следующему порту + port_idx = (port_idx + 1) % ports_total; + return status; +} + +void my_diag(int status) { // Данная функция вызывается в случае ошибки на каком-то этапе и на основании поступившего кода, // формирует сообщение с описанием произошедшей ошибки @@ -215,15 +298,6 @@ void my_diag() case -600: // Неполные данные для Telegram-уведомлений printf("Error: To use telegram integration both telegram_id and telegram_token have to be provided!\n.--help for info\n"); break; - case -501: - printf("Error: Invalid target IP address\n"); - break; - case -502: - printf("Error: Failed to create UDP socket\n"); - break; - case -503: - printf("Error: UDP packet send failed\n"); - break; } printf("end my_diag\n"); // debug @@ -646,69 +720,6 @@ int my_tcp_syn() { return 0; } -int my_udp() { - // Выполняет UDP портовое сканирование well-known портов - int status; - int sockfd; // Дескриптор сокета - struct sockaddr_in target_addr; // Адрес цели - static int port_idx; // Текущий индекс порта - static const int ports[]; // Список портов - static const int ports_total; - int curr_port; // Текущий порт - const char dummy_data[]; // Данные для отправки - ssize_t send_result; // Результат отправки - - // Инициализация структуры адреса - memset(&target_addr, 0, sizeof(target_addr)); - curr_port = ports[port_idx]; - dummy_data[] = "SCAN"; - target_addr.sin_family = AF_INET; - target_addr.sin_port = htons(curr_port); - ports_total = sizeof(ports)/sizeof(ports[0]) - sockfd = -1; - port_idx = 0; - ports = { // Список портов - 53, 67, 68, 69, 123, 161, 162, 389, 443, 500, 514, 520, 1900, 4500 - }; - - // Преобразование IP - if (inet_pton(AF_INET, ip, &target_addr.sin_addr) <= 0) { - n_fail_requests++; - status = -501; // Код ошибки: неверный IP - if (sockfd != -1) close(sockfd); - return status; - } - - // Создание сокета - sockfd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); - if (sockfd < 0) { - n_fail_requests++; - status = -502; // Ошибка создания сокета - if (sockfd != -1) close(sockfd); - return status; - } - - // Отправка данных - send_result = sendto(sockfd, dummy_data, sizeof(dummy_data), 0, - (struct sockaddr*)&target_addr, sizeof(target_addr)); - if (send_result < 0) { - n_fail_requests++; - status = -503; // Ошибка отправки - } else { - n_ok_requests++; - status = 0; // Успешная отправка - } - - // Проверка общего числа запросов - if (n_ok_requests + n_fail_requests >= 1000) { - status = 2; // Условие завершения - } - - // Переход к следующему порту - port_idx = (port_idx + 1) % ports_total; - return status; -} - // Callback для записи ответа от сервера size_t WriteCallback(void* contents, size_t size, size_t nmemb, std::string* output) { size_t total_size = size * nmemb; @@ -826,7 +837,6 @@ int main(int arg_ctr, char **arg_ptr) status = my_check_params(); // Проверяем параметры командной строки - switch (status) // Обрабатываем результат проверки параметров { case 1: // Режим сканирования портов (UDP)