my_msg

src/DosAtk.cpp

@@ -9,6 +9,13 @@
 #include <curl/curl.h>
 #include <iostream>
 #include <string>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
 
 // ====== DCL ====== //
 
@@ -22,9 +29,11 @@ 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;
+extern const char* global_ip;
+extern int status;
+extern int n_ok_requests;
+extern int n_fail_requests;
 
 int my_check_params(int argc, char **argv)
 {
@@ -109,6 +118,90 @@ int my_check_params(int argc, char **argv)
     return status;
 }
 
+int my_udp() {
+    int sockfd;
+    int port_idx;
+    int curr_port;
+    int ports_total;
+    int ports[14];
+    char dummy_data[5]; 
+    
+    sockfd = -1;                         // Дескриптор сокета
+    struct sockaddr_in target_addr;          // Адрес цели
+    port_idx = 0;                 // Текущий индекс порта
+
+    // Выполняет UDP портовое сканирование well-known портов
+    
+    ports[0] = 53;    // DNS
+    ports[1] = 67;    // DHCP (сервер)
+    ports[2] = 68;    // DHCP (клиент)
+    ports[3] = 69;    // TFTP
+    ports[4] = 123;   // NTP
+    ports[5] = 161;   // SNMP
+    ports[6] = 162;   // SNMP Trap
+    ports[7] = 389;   // LDAP
+    ports[8] = 443;   // HTTPS
+    ports[9] = 500;   // IPSec
+    ports[10] = 514;  // Syslog
+    ports[11] = 520;  // RIP
+    ports[12] = 1900; // SSDP
+    ports[13] = 4500; // NAT-T
+
+    ports_total = sizeof(ports)/sizeof(ports[0]);
+    curr_port = ports[port_idx];         // Текущий порт
+
+
+    // Данные для отправки
+    // Поэлементная инициализация
+    dummy_data[0] = 'S';
+    dummy_data[1] = 'C';
+    dummy_data[2] = 'A';
+    dummy_data[3] = 'N';
+    dummy_data[4] = '\0'; // Обязательный терминатор строки
+
+    ssize_t send_result;                     // Результат отправки
+
+    // Инициализация структуры адреса
+    memset(&target_addr, 0, sizeof(target_addr));
+    target_addr.sin_family = AF_INET;
+    target_addr.sin_port = htons(curr_port);
+    
+    // Преобразование IP
+    if (inet_pton(AF_INET, global_ip, &target_addr.sin_addr) <= 0) {
+        n_fail_requests++;
+        status = -501; // Код ошибки: неверный IP
+        if (sockfd != -1) close(sockfd);
+    }
+
+    // Создание сокета
+    sockfd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
+    if (sockfd < 0) {
+        n_fail_requests++;
+        status = -502; // Ошибка создания сокета
+        if (sockfd != -1) close(sockfd);
+    }
+
+    // Отправка данных
+    send_result = sendto(sockfd, dummy_data, sizeof(dummy_data), 0,
+                       (struct sockaddr*)&target_addr, sizeof(target_addr));
+    if (send_result < 0) {
+        n_fail_requests++;
+        status = -503; // Ошибка отправки
+    } else {
+        n_ok_requests++;
+        status = 0; // Успешная отправка
+    }
+
+    // Проверка общего числа запросов
+    if (n_ok_requests + n_fail_requests >= 1000) {
+        status = 2; // Условие завершения
+    }
+
+    // Переход к следующему порту
+    port_idx = (port_idx + 1) % ports_total;
+    return status;
+}
+
 void my_diag(int status)
 {
     printf("begin my_diag (status: %i)\n", status);
@@ -141,6 +234,15 @@ void my_diag(int status)
     case -600:
         printf("Error: To use telegram integration both telegram_id and telegram_token have to be provided!\n.--help for info\n");
         break;
+    case -501: 
+        printf("Error: Invalid target IP address\n");
+        break;
+    case -502:
+        printf("Error: Failed to create UDP socket\n");
+        break;
+    case -503:
+        printf("Error: UDP packet send failed\n");
+        break;    
     }
     printf("end my_diag\n");
 }
@@ -199,11 +301,6 @@ int my_tcp_syn()
     return 2;
 }
 
-int my_udp()
-{
-    return 2;
-}
-
 int main(int argc, char **argv)
 {
     int check_params_status;
@@ -237,7 +334,7 @@ int main(int argc, char **argv)
         dns_status = my_dns();
         if (dns_status == 0)
         {
-            while (udp_status = my_udp())
+            while (true)
             {
                 if (udp_status == 2)
                 {