forked from serafim/dos
Compare commits
3 Commits
dev
...
nekitmilk_
Author | SHA1 | Date |
---|---|---|
|
0bc97e71fe | |
|
2b79163781 | |
|
7493c803fa |
|
@ -0,0 +1,14 @@
|
||||||
|
CXX = g++
|
||||||
|
CXXFLAGS = -Iinclude/
|
||||||
|
|
||||||
|
all: dos
|
||||||
|
|
||||||
|
# Сюда дописывать файлики для компиляции
|
||||||
|
dos:
|
||||||
|
$(CXX) $(CXXFLAGS) ./main.cpp ./my_check_params.cpp -o dos
|
||||||
|
|
||||||
|
rebuild:
|
||||||
|
rm -f ./dos && make dos
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm -f ./dos
|
|
@ -0,0 +1,46 @@
|
||||||
|
#include "my_check_params.hpp"
|
||||||
|
|
||||||
|
Options opts;
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
|
// Компиляция из директории вызвать команду make my_app
|
||||||
|
// Запуск: ./my_app [флаги и аргументы к ним] (начать можно с флага --help)
|
||||||
|
int main(int argc, char *argv[]) {
|
||||||
|
|
||||||
|
int error = my_check_params(argc, argv);
|
||||||
|
if (error > 0)
|
||||||
|
{
|
||||||
|
if (opts.attack_type == "flood") {
|
||||||
|
cout << "type attack: " << opts.attack_type << "\n";
|
||||||
|
cout << "domain: " << opts.domain << "\n";
|
||||||
|
cout << "ip: " << opts.ip << "\n";
|
||||||
|
cout << "port: " << opts.port << "\n";
|
||||||
|
cout << "log file path: " << opts.log_file << "\n";
|
||||||
|
}
|
||||||
|
else if (opts.attack_type == "scan") {
|
||||||
|
cout << "type attack: " << opts.attack_type << "\n";
|
||||||
|
cout << "domain: " << opts.domain << "\n";
|
||||||
|
cout << "ip: " << opts.ip << "\n";
|
||||||
|
cout << "log file path: " << opts.log_file << "\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!opts.telegram_id.empty()) {
|
||||||
|
cout << "telegram_id: " << opts.telegram_id << "\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!opts.telegram_token.empty()) {
|
||||||
|
cout << "telegram_token: " << opts.telegram_token << "\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
cout << "err code:" << error << "\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
|
@ -0,0 +1,117 @@
|
||||||
|
#include "my_check_params.hpp"
|
||||||
|
|
||||||
|
// Гарантируется наличие минимума нужных аргументов для flood и scan
|
||||||
|
// Гарантируется, что после работы парсера мы получим только валидный тип атаки
|
||||||
|
|
||||||
|
// Добавить:
|
||||||
|
// 1. Валидацию IP, port
|
||||||
|
|
||||||
|
// Статус коды:
|
||||||
|
// 2 - Атака флуд, все нужные опции есть
|
||||||
|
// 1 - Атака порт скан, все нужные опции есть
|
||||||
|
// 0 - нужна помощь
|
||||||
|
// -1 - пользователь не ввел тип атаки или ввел неверный тип атаки
|
||||||
|
// -10 - Пользователь выбрал тип атаки порт сканнинг, но не ввел нужные параметры
|
||||||
|
// -20 - Пользователь выбрал тип атаки флуд, но не ввел нужные параметры
|
||||||
|
// -100 - неизвестная ошибка
|
||||||
|
// -101 - неизвестная опция или потерян аргумент, следует предложить вызвать флаг помощи
|
||||||
|
// -600 - пользователь ввел токен, но не id или наоборот
|
||||||
|
// Какие-то еще коды?
|
||||||
|
int my_check_params(int argc, char** argv) {
|
||||||
|
|
||||||
|
int status = -100;
|
||||||
|
// Короткие опции (с двоеточием для параметров)
|
||||||
|
const char* short_options = "a:d:i:p:l:t:b:h";
|
||||||
|
|
||||||
|
// Длинные опции
|
||||||
|
const struct option long_options[] = {
|
||||||
|
{"attack", required_argument, NULL, 'a'},
|
||||||
|
{"domain", required_argument, NULL, 'd'},
|
||||||
|
{"ip", required_argument, NULL, 'i'},
|
||||||
|
{"port", required_argument, NULL, 'p'},
|
||||||
|
{"log", required_argument, NULL, 'l'},
|
||||||
|
{"telegram", required_argument, NULL, 't'},
|
||||||
|
{"token", required_argument, NULL, 'b'},
|
||||||
|
{"help", no_argument, NULL, 'h'},
|
||||||
|
{NULL, 0, NULL, 0}
|
||||||
|
};
|
||||||
|
|
||||||
|
int opt;
|
||||||
|
while ((opt = getopt_long(argc, argv, short_options, long_options, NULL)) != -1) {
|
||||||
|
switch (opt) {
|
||||||
|
case 'a':
|
||||||
|
opts.attack_type = optarg;
|
||||||
|
break;
|
||||||
|
case 'd':
|
||||||
|
opts.domain = optarg;
|
||||||
|
break;
|
||||||
|
case 'i':
|
||||||
|
opts.ip = optarg;
|
||||||
|
break;
|
||||||
|
case 'p':
|
||||||
|
opts.port = optarg;
|
||||||
|
break;
|
||||||
|
case 'l':
|
||||||
|
opts.log_file = optarg;
|
||||||
|
break;
|
||||||
|
case 't':
|
||||||
|
opts.telegram_id = optarg;
|
||||||
|
break;
|
||||||
|
case 'b':
|
||||||
|
opts.telegram_token = optarg;
|
||||||
|
break;
|
||||||
|
case 'h':
|
||||||
|
// std::cout << "Usage: " << argv[0] << " [options]\n"
|
||||||
|
// << "Required:\n"
|
||||||
|
// << " -a, --attack TYPE Type of attack (scan|flood)\n"
|
||||||
|
// << " -d, --domain DOMAIN Target domain\n"
|
||||||
|
// << " -i, --ip IP Target IP\n"
|
||||||
|
// << " -p, --port PORT Port. Required only for flood type!\n"
|
||||||
|
// << "Optional:\n"
|
||||||
|
// << " -l, --log FILE Log file\n"
|
||||||
|
// << " -t, --telegram ID Telegram ID\n"
|
||||||
|
// << " -b, --token TOKEN Telegram bot token\n";
|
||||||
|
status = 0;
|
||||||
|
break;
|
||||||
|
case '?':
|
||||||
|
// std::cerr << "Unknown option!\n.--help for info\n";
|
||||||
|
status = -101;
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
status = -100;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Проверка обязательных параметров
|
||||||
|
|
||||||
|
if (status != 0 && status != -101)
|
||||||
|
{
|
||||||
|
if (opts.attack_type != "flood" && opts.attack_type != "scan") {
|
||||||
|
// std::cerr << "Error: Missing required parameters!\n--help for more info\n";
|
||||||
|
status = -1;
|
||||||
|
}
|
||||||
|
else if (opts.attack_type == "scan" && (opts.domain.empty() || opts.ip.empty())) {
|
||||||
|
// std::cerr << "Error: Missing required parameters!\n--help for more info\n";
|
||||||
|
status = -10;
|
||||||
|
}
|
||||||
|
else if (opts.attack_type == "flood" && (opts.domain.empty() || opts.ip.empty() || opts.port.empty())) {
|
||||||
|
// std::cerr << "Error: Missing required parameters!\n--help for more info\n";
|
||||||
|
status = -20;
|
||||||
|
}
|
||||||
|
else if ((!opts.telegram_id.empty() && opts.telegram_token.empty()) || (opts.telegram_id.empty() && !opts.telegram_token.empty())) {
|
||||||
|
status = -600;
|
||||||
|
}
|
||||||
|
else if (opts.attack_type == "scan" && !opts.domain.empty() && !opts.ip.empty()) {
|
||||||
|
// std::cerr << "Error: Missing required parameters!\n--help for more info\n";
|
||||||
|
status = 1;
|
||||||
|
}
|
||||||
|
else if (opts.attack_type == "flood" && !opts.domain.empty() && !opts.ip.empty() && !opts.port.empty()) {
|
||||||
|
// std::cerr << "Error: Missing required parameters!\n--help for more info\n";
|
||||||
|
status = 2;
|
||||||
|
}
|
||||||
|
// Какие-то еще коды?
|
||||||
|
}
|
||||||
|
|
||||||
|
return status;
|
||||||
|
}
|
|
@ -0,0 +1,22 @@
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <iostream>
|
||||||
|
#include <string>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <getopt.h> // Для struct option
|
||||||
|
|
||||||
|
|
||||||
|
struct Options {
|
||||||
|
std::string attack_type; // Обязательный параметр
|
||||||
|
std::string domain; // Обязательный параметр
|
||||||
|
std::string ip; // Обязательный параметр
|
||||||
|
std::string port; // Обязательный параметр (не обязательный для скана)
|
||||||
|
std::string log_file = "/var/logs/DosAtk"; // Значение по умолчанию, не обязательный
|
||||||
|
std::string telegram_id; // Не обязательный параметр
|
||||||
|
std::string telegram_token; // Не обязательный параметр
|
||||||
|
};
|
||||||
|
|
||||||
|
extern Options opts; // Теперь это глобальная переменная, где она должна определяться?
|
||||||
|
|
||||||
|
// Прототип функции парсинга
|
||||||
|
int my_check_params(int argc, char** argv);
|
Loading…
Reference in New Issue