Compare commits

..

No commits in common. "1a15954507e65229c09d83036b900337b1a042e7" and "4c173756a06ee12e11cca772e7d0816328be8bea" have entirely different histories.

2 changed files with 210 additions and 251 deletions

View File

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

BIN
src/a.out Executable file

Binary file not shown.