beta 0.3

.gitignore

@@ -1 +0,0 @@
-.vscode

build.sh

@@ -1,2 +0,0 @@
-g++ src/DosAtk.cpp -o DosAtk -lcurl
-./DosAtk "$@"

src/DosAtk.cpp

@@ -1,328 +0,0 @@
-#include <iostream>
-#include <stdlib.h>
-#include <string>
-#include <chrono>
-#include <ctime>
-#include <iomanip>
-#include <getopt.h>
-#include <cctype>
-#include <curl/curl.h>
-#include <iostream>
-#include <string>
-
-// ====== DCL ====== //
-
-std::string attack_type;
-std::string domain;
-std::string ip;
-std::string port;
-std::string log_file;
-std::string 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 my_check_params(int argc, char **argv)
-{
-    std::string debug_msg;
-    debug_msg = "";
-    for (int i = 0; i < argc; i++) {
-        debug_msg += argv[i];
-        debug_msg += " ";
-    }
-    printf("begin my_check_params (argc: %i, argv: %s)\n", argc, debug_msg.c_str());
-
-    int status;
-    int opt;
-    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}
-    };
-
-    while ((opt = getopt_long(argc, argv, short_options, long_options, NULL)) != -1) {
-        switch (opt) {
-        case 'a':
-            attack_type = optarg;
-            break;
-        case 'd':
-            domain = optarg;
-            break;
-        case 'i':
-            ip = optarg;
-            break;
-        case 'p':
-            port = optarg;
-            break;
-        case 'l':
-            log_file = optarg;
-            break;
-        case 't':
-            telegram_id = optarg;
-            break;
-        case 'b':
-            telegram_token = optarg;
-            break;
-        case 'h':
-            status = 0;
-            break;
-        case '?':
-            status = -101;
-            break;
-        }
-    }
-
-    if (status != 0 && status != -101)
-    {
-        if (attack_type != "flood" && attack_type != "scan") {
-            status = -1;
-        }
-        else if (attack_type == "scan" && domain.empty() && ip.empty()) {
-            status = -10;
-        }
-        else if (attack_type == "flood" && domain.empty() && ip.empty()) {
-            status = -20;
-        }
-        else if ((!telegram_id.empty() && telegram_token.empty()) || (telegram_id.empty() && !telegram_token.empty())) {
-            status = -600;
-        }
-        else if (attack_type == "scan") {
-            status = 1;
-        }
-        else if (attack_type == "flood") {
-            status = 2;
-        }
-    }
-
-    printf("end my_check_params status: %i\n", status);
-    return status;
-}
-
-void my_diag(int status)
-{
-    printf("begin my_diag (status: %i)\n", status);
-    switch (status)
-    {
-    case 0:
-        printf("Usage: ./DosAtk [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");
-        break;
-    case -1:
-        printf("Error: Invalid attack type!\n--help for more info\n");
-        break;
-    case -10:
-        printf("Error: Missing required parameters for port scanning!\n--help for more info\n");
-        break;
-    case -20:
-        printf("Error: Missing required parameters for tcp syn dos attack!\n--help for more info\n");
-        break;
-    case -101:
-        printf("Error: Unknown option!\n.--help for info\n");
-        break;
-    case -600:
-        printf("Error: To use telegram integration both telegram_id and telegram_token have to be provided!\n.--help for info\n");
-        break;
-    }
-    printf("end my_diag\n");
-}
-
-void my_msg()
-{
-    printf("begin my_msg()\n");
-    printf("%s\n", msg.c_str());
-    printf("end my_msg");
-}
-
-int my_log()
-{
-    return 0;
-}
-
-void my_fin()
-{
-    auto end_timestamp = std::chrono::system_clock::now();
-    auto end_time_t = std::chrono::system_clock::to_time_t(end_timestamp);
-    auto end_ms = std::chrono::duration_cast<std::chrono::milliseconds>(end_timestamp.time_since_epoch()) % 1000;
-
-    auto duration = end_timestamp - start_timestamp;
-    auto hours = std::chrono::duration_cast<std::chrono::hours>(duration);
-    auto 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));
-    auto milliseconds = std::chrono::duration_cast<std::chrono::milliseconds>(duration % std::chrono::seconds(1));
-
-    std::cout << "Worked for ";
-    if (duration < std::chrono::minutes(2)) {
-        double 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;
-
-    std::cout << "Sent " << (n_ok_requests + n_fail_requests) << " requests (" 
-              << n_ok_requests << " ok, " << n_fail_requests << " failed)" << std::endl;
-
-    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::exit(0);
-}
-
-int my_dns()
-{
-    return 0;
-}
-
-int my_tcp_syn()
-{
-    return 2;
-}
-
-int my_udp()
-{
-    return 2;
-}
-
-int main(int argc, char **argv)
-{
-    int check_params_status;
-    int log_status;
-    int dns_status;
-    int udp_status;
-    int tcp_syn_status;
-
-    // ====== Тело программы ====== //
-    n_ok_requests = 0;
-    n_fail_requests = 0;
-    start_timestamp = std::chrono::system_clock::now();
-
-    time_t now_time_t = std::chrono::system_clock::to_time_t(start_timestamp);
-
-    std::chrono::milliseconds 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, 
-        std::localtime(&now_time_t)->tm_mon + 1, 
-        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());
-
-    check_params_status = my_check_params(argc, argv);
-    switch (check_params_status)
-    {
-    case 1:
-        dns_status = my_dns();
-        if (dns_status == 0)
-        {
-            while (udp_status = my_udp())
-            {
-                if (udp_status == 2)
-                {
-                    break;
-                }
-                else if (udp_status < 0)
-                {
-                    my_diag(udp_status);
-                    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(check_params_status);
-            log_status = my_log();
-            if (log_status == 0){
-                my_fin();
-            }
-            else if (log_status == 1)
-            {
-                my_msg();
-                my_fin();
-            }
-        }
-        break;
-    case 2:
-        dns_status = my_dns();
-        if (dns_status == 0)
-        {
-            while (tcp_syn_status = my_tcp_syn())
-            {
-                if (tcp_syn_status == 2)
-                {
-                    break;
-                }
-                else if (tcp_syn_status < 0)
-                {
-                    my_diag(tcp_syn_status);
-                    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(check_params_status);
-            log_status = my_log();
-            if (log_status == 0)
-            {
-                my_fin();
-            }
-            else if (log_status == 1)
-            {
-                my_msg();
-                my_fin();
-            }
-        }
-        break;
-    default:
-        my_diag(check_params_status);
-        log_status = my_log();
-        if (log_status == 0)
-        {
-            my_fin();
-        }
-        else if (log_status == 1)
-        {
-            my_msg();
-            my_fin();
-        }
-        break;
-    }
-    return 0;
-}

src/Makefile

@@ -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

src/main.cpp

@@ -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;
+}

src/my_check_params.cpp

@@ -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;
+}

src/my_check_params.hpp

@@ -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);