refactor: all

src/DosAtk.cpp

@@ -1,3 +1,5 @@
+// ====== DCL библиотеки ====== //
+
 #include <iostream>          // Для работы с вводом-выводом (cout, cin и др.)
 #include <stdlib.h>          // Стандартная библиотека C (функции работы с памятью, преобразования типов и др.)
 #include <string>            // Для работы со строками string
@@ -8,25 +10,32 @@
 #include <cctype>            // Функции для работы с символами (isdigit, isalpha и др.)
 #include <curl/curl.h>       // Библиотека libcurl для HTTP-запросов (отправка в Telegram)
 
-// Глобальные переменные
+// ====== DCL глобальные переменные ====== //
+
+// Параметры
+int argc                                                // Количество аргументов при вызове программы
+char **argv                                             // Массив строк с агрументами
 std::string attack_type;                                // Тип атаки: scan или syn
 std::string domain;                                     // Доменное Имя
 std::string ip;                                         // Ip жертвы
-std::string port; // Порт
+std::string port;                                       // Порт для syn атаки
 std::string log_file;                                   // Путь к директории для логов
 std::string telegram_id;                                // Telegram ID для уведомлений
-std::string telegram_token; // Токен бота
+std::string telegram_token;                             // Токен бота для уведомлений
+
+// Статистика
 int n_ok_requests;                                      // Количество успешных запросов
 int n_fail_requests;                                    // Количество не успешных запросов
 std::chrono::system_clock::time_point start_timestamp;  // Начальное время работы
+// Переменные
 std::string log_msg;                                    // Сообщение, которое будет записано в лог-файл
 std::string fin_msg;                                    // Сообщение, которое будет выведено в консоль пользователю
 std::string msg;                                        // Сообщение, которое будет отправлено в телеграм 
 int status;                                             // Статус работы программы
 
-// ====== end of DCL ====== //
+// ====== DCL процедуры ====== //
 
-int my_check_params(int argc, char **argv)
+void my_check_params()
 {
     // Данная процедура записывает в глобальные переменные параметры 
     // (attack_type, domain, ip, port, log_file, telegram_id, telegram_token) проводимой атаки, поступившие при вызове программы
@@ -41,11 +50,18 @@ int my_check_params(int argc, char **argv)
     // -101 - неизвестная опция или потерян аргумент, следует предложить вызвать флаг помощи
     // -600 -  пользователь ввел токен, но не id или наоборот
 
-    int status;
+    //Объявление
-    int opt;
+    std::chrono::milliseconds ms;   // Количество миллисекунд от целой секунды времени начала выполнения программы
-    const char* short_options;
+    int opt;                        // Прочитанный параметр
+    const char* short_options;      // Сокращения для параметров
+    struct option long_options[];   // Структура, описывающая пеобходимые программе параметры
+    int i;                          // Счётчик для цикла
+
+    // Инициализация
+    ms = std::chrono::duration_cast<std::chrono::milliseconds>(start_timestamp.time_since_epoch()) % 1000;
+    opt = -1;
     short_options = "a:d:i:p:l:t:b:h";   
-    const struct option long_options[] = 
+    long_options = 
     {
         {"attack",    required_argument, NULL, 'a'},
         {"domain",    required_argument, NULL, 'd'},
@@ -57,14 +73,19 @@ int my_check_params(int argc, char **argv)
         {"help",      no_argument,       NULL, 'h'},
         {NULL,     0,                 NULL,  0}
     };
+    i = 0;
 
-    std::string debug_msg;
+    printf("begin my_check_params"); // debug
-    debug_msg = "";
+
-    for (int i = 0; i < argc; i++) {
+    // Выводим информацию о времени запуска программы
-        debug_msg += argv[i];
+    printf("Starting DosAtk at %04d-%02d-%02d %02d:%02d:%02d.%03ld\n",  // Шаблон для вывода
-        debug_msg += " ";
+        std::localtime(&now_time_t)->tm_year + 1900,                    // Год (с 1900)
-    }
+        std::localtime(&now_time_t)->tm_mon + 1,                        // Месяц (0-11)
-    printf("begin my_check_params (argc: %i, argv: %s)\n", argc, debug_msg.c_str()); // debug
+        std::localtime(&now_time_t)->tm_mday,                           // День месяца
+        std::localtime(&now_time_t)->tm_hour,                           // Часы
+        std::localtime(&now_time_t)->tm_min,                            // Минуты
+        std::localtime(&now_time_t)->tm_sec,                            // Секунды
+        ms.count());                                                    // Миллисекунды
 
     // Обрабатываем аргументы командной строки с помощью getopt_long
     // Цикл продолжается, пока getopt_long возвращает валидные опции (-1 означает конец опций)
@@ -131,7 +152,6 @@ int my_check_params(int argc, char **argv)
     }
 
     printf("end my_check_params status: %i\n", status); // debug
-    return status;
 }
 
 void my_diag()
@@ -139,8 +159,8 @@ void my_diag()
     // Данная функция вызывается в случае ошибки на каком-то этапе и на основании поступившего кода, 
     // формирует сообщение с описанием произошедшей ошибки
 
-    // Отладочный вывод - начало работы функции (можно отключить в релизной версии)
     printf("begin my_diag, status: %i\n", status); // debug
+
     // Выбор сообщения в зависимости от кода ошибки
     switch (status)
     {
@@ -222,43 +242,39 @@ bool is_numeric(const std::string& s) {
     return true;
 }
 
-int my_msg() {
+void my_msg() {
-    // Инициализация локальных переменных для работы с CURL
 
+    printf("begin my_msg"); // debug
+
+    // Объявление
+    struct curl_slist* headers;     // Заголовки HTTP-запроса
     CURL* curl;                     // Указатель на CURL-объект
     std::string escaped_msg;        // Экранированное сообщение для JSON
-    std::cout << msg << std::endl;  // Вывод сообщения в консоль (для отладки)
     std::string chat_id_field;      // Поле chat_id для JSON
     std::string json_data;          // Итоговый JSON для отправки
-    struct curl_slist* headers;     // Заголовки HTTP-запроса
-    headers = nullptr;              // Инициализация заголовков
     long http_code;                 // HTTP-код ответа
     CURLcode res;                   // Код результата CURL-операции
     
+    // Инициализация
+    headers = curl_slist_append(nullptr, "Content-Type: application/json"); // Установка заголовка Content-Type
+    curl = curl_easy_init(); // Инициализация CURL
+    escaped_msg = escape_json(msg); // Экранирование спецсимволов в сообщении для JSON
+    chat_id_field = is_numeric(telegram_id) ? "\"chat_id\": " + telegram_id : "\"chat_id\": \"" + telegram_id + "\"";
+    json_data = "{" + chat_id_field + ", \"text\": \"" + escaped_msg + "\"}"; // Сборка JSON-запроса
+    http_code = 0;
+    res = nullptr;
+    
     // Проверка наличия обязательных параметров Telegram
-    if (telegram_token.empty() || telegram_id.empty()) {
+    if (telegram_token.empty() || telegram_id.empty())
-        return 0; // Интеграция с Telegram не настроена (отсутствует токен или ID)
+    {
+        status = 0; // Интеграция с Telegram не настроена (отсутствует токен или ID)
     }
 
-    // Инициализация CURL
+    if (!curl)
-    curl = curl_easy_init();
+    {
-    if (!curl) return 6; // Ошибка инициализации CURL
+        status = 6; // Ошибка инициализации CURL
-
-    // Экранирование спецсимволов в сообщении для JSON
-    escaped_msg = escape_json(msg);
-    
-    // Формирование поля chat_id в зависимости от типа ID (число или строка)
-    if (is_numeric(telegram_id)) {
-        chat_id_field = "\"chat_id\": " + telegram_id;  // Числовой ID
-    } else {
-        chat_id_field = "\"chat_id\": \"" + telegram_id + "\"";  // Строковый ID
     }
 
-    // Сборка JSON-запроса
-    json_data = "{" + chat_id_field + ", \"text\": \"" + escaped_msg + "\"}";
-    // Установка заголовка Content-Type
-    headers = curl_slist_append(headers, "Content-Type: application/json");
-
     // Настройка параметров CURL-запроса
     curl_easy_setopt(curl, CURLOPT_URL, ("https://api.telegram.org/bot" + telegram_token + "/sendMessage").c_str()); // URL API Telegram
     curl_easy_setopt(curl, CURLOPT_POST, 1L);                      // Использовать POST-метод
@@ -267,15 +283,17 @@ int my_msg() {
     curl_easy_setopt(curl, CURLOPT_USERAGENT, "libcurl/7.68.0");   // User-Agent
     
     // Игнорирование вывода ответа (пустая функция обратного вызова)
-    curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, [](void*, size_t size, size_t nmemb, void*) -> size_t {
+    curl_easy_setopt(
+        curl, CURLOPT_WRITEFUNCTION,
+        [](void*, size_t size, size_t nmemb, void*) -> size_t {
             return size * nmemb;
-    });
+        }
+    );
 
     // Выполнение HTTP-запроса
     res = curl_easy_perform(curl);
 
     // Получение HTTP-кода ответа
-    http_code = 0;
     curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &http_code);
 
     // Освобождение ресурсов
@@ -283,57 +301,80 @@ int my_msg() {
     curl_easy_cleanup(curl);       // Завершение работы CURL
 
     // Обработка ошибок CURL
-    if (res != CURLE_OK) return 5; // Ошибка выполнения запроса
+    if (res != CURLE_OK)
+    {
+        status =  5; // Ошибка выполнения запроса
+    }
     
     // Обработка HTTP-кодов ответа
-    switch (http_code) {
+    switch (http_code)
-        case 200: return 0; // Успешный запрос
+    {
-        case 401: return 1; // Ошибка авторизации (неверный токен)
+        case 200:
-        case 400: return 2; // Неверный запрос
+            status =  0; // Успешный запрос
-        case 404: return 3; // Ресурс не найден
+        case 401:
-        default:  
+            status =  1; // Ошибка авторизации (неверный токен)
-            if (http_code >= 500) return 4; // Ошибка сервера
+        case 400:
-            return 4;                       // Прочие ошибки
+            status =  2; // Неверный запрос
-    }
+        case 404:
+            status =  3; // Ресурс не найден
+        case 500:  
+            status =  4; // Ошибка сервера
     }
 
-int my_log()
+    printf("end my_msg"); // debug
+}
+
+void my_log()
 {
     // Данная функция записывает в файл лога сообщение
-    return 0;
+    printf("start my_log"); // debug
+    printf("end my_log"); // debug
 }
 
 void my_fin()
 {
-    // Данная функция завершает программу и рассчитывает итоговое время выполнения программы 
+    // Данная процедура завершает программу и рассчитывает итоговое время выполнения программы
     
-    // Фиксируем время окончания работы программы
+    // Объявления
-    auto end_timestamp = std::chrono::system_clock::now();
+    time_t end_time_t;                      // Время завершения выполненя программы
-    // Преобразуем время окончания в time_t и миллисекунды
+    std::chrono::duration duration;         // Длительность выполнения программы
-    auto end_time_t = std::chrono::system_clock::to_time_t(end_timestamp);
+    std::chrono::hours hours;               // Компонента часов времени завершения
-    auto end_ms = std::chrono::duration_cast<std::chrono::milliseconds>(end_timestamp.time_since_epoch()) % 1000;
+    std::chrono::minutes minutes;           // Компонента минут времени завершения
+    std::chrono::seconds seconds;           // Компонента секунд времени завершения
+    std::chrono::milliseconds milliseconds; // Компонента миллисекунд времени завершения
 
-    // Вычисляем продолжительность работы программы
+    // Иницаализация
-    auto duration = end_timestamp - start_timestamp;
+    end_time_t = std::chrono::system_clock::to_time_t(std::chrono::system_clock::now());
-    // Разбиваем продолжительность на компоненты
+    duration = end_timestamp - start_timestamp; // Вычисляем продолжительность работы программы
-    auto hours = std::chrono::duration_cast<std::chrono::hours>(duration);
+    hours = std::chrono::duration_cast<std::chrono::hours>(duration); // Разбиваем продолжительность на компоненты
-    auto minutes = std::chrono::duration_cast<std::chrono::minutes>(duration % std::chrono::hours(1));
+    minutes = std::chrono::duration_cast<std::chrono::minutes>(duration % std::chrono::hours(1)); // Разбиваем продолжительность на компоненты
-    auto seconds = std::chrono::duration_cast<std::chrono::seconds>(duration % std::chrono::minutes(1));
+    seconds = std::chrono::duration_cast<std::chrono::seconds>(duration % std::chrono::minutes(1)); // Разбиваем продолжительность на компоненты
-    auto milliseconds = std::chrono::duration_cast<std::chrono::milliseconds>(duration % std::chrono::seconds(1));
+    milliseconds = std::chrono::duration_cast<std::chrono::milliseconds>(duration % std::chrono::seconds(1));  // Разбиваем продолжительность на компоненты
+    double total_seconds
 
-    // Выводим информацию о времени работы
+    printf("begin my_diag\n"); // debug
-    std::cout << "Worked for ";
+
+    std::cout << "Worked for "; // Выводим информацию о времени работы
     // Для коротких периодов (<2 минут) выводим в секундах с дробной частью
-    if (duration < std::chrono::minutes(2)) {
+    if (duration < std::chrono::minutes(2))
-        double total_seconds = std::chrono::duration<double>(duration).count();
+    {
+        total_seconds = std::chrono::duration<double>(duration).count();
         std::cout << std::fixed << std::setprecision(3) << total_seconds << " seconds";
-    } else {
-        // Для длительных периодов выводим в формате ЧЧ:ММ:СС.мс
-        if (hours.count() > 0) std::cout << hours.count() << "h ";
-        if (minutes.count() > 0) std::cout << minutes.count() << "m ";
-        std::cout << seconds.count() << "s " << milliseconds.count() << "ms";
     }
-    std::cout << std::endl;
+    else
+    {
+        // Для длительных периодов выводим в формате ЧЧ:ММ:СС.мс
+        if (hours.count() > 0) // Если программа работала хотя бы один час
+        {
+            std::cout << hours.count() << "h "; // Выводим часы
+        }
+        if (minutes.count() > 0) // Если программа работала хотя бы одину минуту
+        {
+            std::cout << minutes.count() << "m "; // Вывовим минуты
+        }
+        std::cout << seconds.count() << "s " << milliseconds.count() << "ms"; // Выводим секунды и миллисекунды
+    }
+    std::cout << std::endl; // После осуществления вывода переходим на новую строку
 
     // Выводим статистику по запросам
     std::cout << "Sent " << (n_ok_requests + n_fail_requests) << " requests (" 
@@ -341,165 +382,165 @@ void my_fin()
 
     // Выводим точное время завершения работы
     std::cout << "DosAtk stopped at " << std::put_time(std::localtime(&end_time_t), "%Y-%m-%d %H:%M:%S")
-              << "." << std::setfill('0') << std::setw(3) << end_ms.count() << std::endl;
+              << "." << std::setfill('0') << std::setw(3) << milliseconds.count() << std::endl;
 
-    // Завершаем программу с кодом 0 (успешное завершение)
+    // Завершаем программу с кодом состояния
-    std::exit(0);
+    std::exit(status);
 }
 
-int my_dns()
+void my_dns()
 {
     // Данная процедура сопостовляет доменное имя с IP 
-    return 0;
+    printf("start my_dns"); // debug
+    printf("end my_dns"); // debug
 }
 
-int my_tcp_syn()
+void my_tcp_syn()
 {
     // Данная процедура выполняет TCP SYN Flood атаку
-    return 2;
+    printf("start my_tcp_syn"); // debug
+    printf("end my_tcp_syn"); // debug
 }
 
-int my_udp()
+void my_udp()
 {
     // Данная процедура выполняет UDP Flood (port scanning) атаку
-    return 2;
+    printf("start my_udp"); // debug
+    printf("end my_udp"); // debug
 }
 
-int main(int argc, char **argv)
+int main(int arg_ctr, char **arg_ptr)
 {
-    // Главная функция программы - точка входа
+    // Инициализация глобальных переменных
+    int argc = arg_ctr;
+    char **argv = arg_ptr;
+    std::string attack_type = "";
+    std::string domain = "";
+    std::string ip = "";
+    std::string port = "";
+    std::string log_file = "/var/log/DosAtk";
+    std::string telegram_id = "";
+    std::string telegram_token = "";
+    int n_ok_requests = 0;
+    int n_fail_requests = 0;
+    std::chrono::system_clock::time_point start_timestamp = std::chrono::system_clock::now();
+    std::string log_msg = "";
+    std::string fin_msg = "";
+    std::string msg;
+    int status = 0;
 
-    // Объявление переменных для хранения статусов операций
+    // Тело программы
-    int log_status;          // Статус записи в лог
-    int dns_status;         // Статус DNS-разрешения
-    int udp_status;         // Статус UDP-атаки
-    int tcp_syn_status;     // Статус TCP SYN-атаки
 
-    // ====== Инициализация программы ====== //
+    // Проверяем параметры командной строки
-    n_ok_requests = 0;      // Счетчик успешных запросов
+    my_check_params();
-    n_fail_requests = 0;    // Счетчик неудачных запросов
-    start_timestamp = std::chrono::system_clock::now(); // Засекаем время начала
-    status = 0; // Статус программы
-
-    // Получаем текущее время в различных форматах
-    time_t now_time_t = std::chrono::system_clock::to_time_t(start_timestamp);
-    auto ms = std::chrono::duration_cast<std::chrono::milliseconds>(
-                start_timestamp.time_since_epoch()) % 1000;
-
-    // Выводим информацию о времени запуска программы
-    printf("Starting DosAtk at %04d-%02d-%02d %02d:%02d:%02d.%03ld\n",
-        std::localtime(&now_time_t)->tm_year + 1900,  // Год (с 1900)
-        std::localtime(&now_time_t)->tm_mon + 1,     // Месяц (0-11)
-        std::localtime(&now_time_t)->tm_mday,        // День месяца
-        std::localtime(&now_time_t)->tm_hour,        // Часы
-        std::localtime(&now_time_t)->tm_min,         // Минуты
-        std::localtime(&now_time_t)->tm_sec,         // Секунды
-        ms.count());                                // Миллисекунды
-
-    // ====== Основная логика программы ====== //
-    // 1. Проверяем параметры командной строки
-    status = my_check_params(argc, argv);
     
     // Обрабатываем результат проверки параметров
     switch (status)
     {
     case 1: // Режим сканирования портов (UDP)
         // Пытаемся разрешить DNS (если указано доменное имя)
-        dns_status = my_dns();
+        my_dns();
         
-        if (dns_status == 0) // Если DNS разрешен успешно
+        if (status == 0) // Если DNS разрешен успешно
         {
             // Запускаем цикл UDP-атаки
-            while (udp_status = my_udp())
+            while (true)
             {
-                if (udp_status == 2) // Код завершения атаки
+                my_udp()
+                if (status == 2) // Код завершения атаки
                 {
                     break;
                 }
-                else if (udp_status < 0) // Обработка ошибок
+                else if (status < 0) // Обработка ошибок
                 {
-                    status = udp_status;
                     my_diag(); // Выводим диагностику
-                    log_status = my_log();    // Пытаемся записать в лог
+                    my_log();  // Пытаемся записать в лог
-                    if (log_status == 1)      // Если требуется уведомление
+                    if (status == 1) // Если записать лог не удалось
                     {
                         my_msg(); // Отправляем сообщение
                     }
                 }
             }
             // Завершающие действия после атаки
-            log_status = my_log(); // Логируем завершение
+            my_log(); // Логируем завершение
             my_msg(); // Отправляем финальное сообщение
             my_fin(); // Корректно завершаем программу
         }
-        else if (dns_status == 1) // Ошибка DNS-разрешения
+        else if (status == 1) // Ошибка DNS-разрешения
         {
             my_diag(); // Выводим ошибку
-            log_status = my_log();        // Логируем ошибку
+            my_log(); // Логируем ошибку
             
-            if (log_status == 0) {        // Если не требуется уведомление
+            if (status == 0) // Если лог получилось записать
+            {
                 my_fin(); // Просто завершаем программу
             }
-            else if (log_status == 1)     // Если требуется уведомление
+            else if (log_status == 1) // Если лог не удалось записать
             {
                 my_msg(); // Отправляем сообщение
                 my_fin(); // Завершаем программу
             }
         }
         break;
-
     case 2: // Режим SYN-флуда (TCP)
-        // Аналогичная логика как для UDP-режима
+        // Пытаемся разрешить DNS (если указано доменное имя)
-        dns_status = my_dns();
+        my_dns();
             
-        if (dns_status == 0)
+        if (status == 0) // Если DNS разрешен успешно
         {
-            while (tcp_syn_status = my_tcp_syn())
+            // Запускаем цикл UDP-атаки
+            while (true)
             {
-                if (tcp_syn_status == 2)
+                my_tcp_syn()
+                if (status == 2) // Код завершения атаки
                 {
                     break;
                 }
-                else if (tcp_syn_status < 0)
+                else if (status < 0) // Обработка ошибок
                 {
-                    status = tcp_syn_status;
-                    my_diag();
-                    log_status = my_log();
-                    if (log_status == 1)
-                    {
-                        my_msg();
-                    }
-                }
-            }
-            log_status = my_log();
-            my_msg();
-            my_fin();
-        }
-        else if (dns_status == 1)
-        {
-            my_diag();
-            log_status = my_log();
-            if (log_status == 0)
-            {
-                my_fin();
-            }
-            else if (log_status == 1)
-            {
-                my_msg();
-                my_fin();
-            }
-        }
-        break;
-
-    default: // Некорректные параметры или запрос справки
                     my_diag(); // Выводим диагностику
-        log_status = my_log();        // Логируем событие
+                    my_log();  // Пытаемся записать в лог
+                    if (status == 1) // Если записать лог не удалось
+                    {
+                        my_msg(); // Отправляем сообщение
+                    }
+                }
+            }
+            // Завершающие действия после атаки
+            my_log(); // Логируем завершение
+            my_msg(); // Отправляем финальное сообщение
+            my_fin(); // Корректно завершаем программу
+        }
+        else if (status == 1) // Ошибка DNS-разрешения
+        {
+            my_diag(); // Выводим ошибку
+            my_log(); // Логируем ошибку
             
-        if (log_status == 0)          // Без уведомления
+            if (status == 0) // Если лог получилось записать
+            {
+                my_fin(); // Просто завершаем программу
+            }
+            else if (log_status == 1) // Если лог не удалось записать
+            {
+                my_msg(); // Отправляем сообщение
+                my_fin(); // Завершаем программу
+            }
+        }
+        break;
+    case 0: // Некорректные параметры или запрос справки
+    case -1:
+    case -10:
+    case -20:
+    case -101:
+    case -600:
+        my_diag(); // Выводим диагностику
+        my_log(); // Логируем событие
+        
+        if (status == 0) // Без уведомления
         {
             my_fin(); // Завершаем программу
         }
-        else if (log_status == 1)     // С уведомлением
+        else if (status == 1) // С уведомлением
         {
             my_msg(); // Отправляем сообщение
             my_fin(); // Завершаем программу