merge dev
This commit is contained in:
commit
3e367248b6
164
src/DosAtk.cpp
164
src/DosAtk.cpp
|
|
@ -55,7 +55,6 @@ std::string log_msg; // Сообщение,
|
|||
std::string fin_msg; // Сообщение, которое будет выведено в консоль пользователю
|
||||
std::string msg; // Сообщение, которое будет отправлено в телеграм
|
||||
int status; // Статус работы программы
|
||||
// int log_status;
|
||||
|
||||
/*
|
||||
██████╗ ██████╗██╗ ██████╗ ██████╗ ██████╗ ██████╗███████╗██████╗ ██╗ ██╗██████╗ ███████╗███████╗
|
||||
|
|
@ -83,10 +82,10 @@ int my_check_params()
|
|||
|
||||
//Объявление
|
||||
std::chrono::milliseconds ms; // Количество миллисекунд от целой секунды времени начала выполнения программы
|
||||
std::time_t now_time_t; // текущее время
|
||||
std::time_t now_time_t; // Текущее время
|
||||
int opt; // Прочитанный параметр
|
||||
const char* short_options; // Сокращения для параметров
|
||||
struct option long_options[10]; // Структура, описывающая пеобходимые программе параметры
|
||||
struct option long_options[10]; // Структура, описывающая пеобходимые программе параметры
|
||||
int i; // Счётчик для цикла
|
||||
|
||||
// Инициализация
|
||||
|
|
@ -180,7 +179,91 @@ int my_check_params()
|
|||
return -101;
|
||||
}
|
||||
|
||||
void my_diag()
|
||||
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)
|
||||
{
|
||||
// Данная функция вызывается в случае ошибки на каком-то этапе и на основании поступившего кода,
|
||||
// формирует сообщение с описанием произошедшей ошибки
|
||||
|
|
@ -215,15 +298,6 @@ void my_diag()
|
|||
case -600: // Неполные данные для Telegram-уведомлений
|
||||
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"); // debug
|
||||
|
|
@ -646,69 +720,6 @@ int my_tcp_syn() {
|
|||
return 0;
|
||||
}
|
||||
|
||||
int my_udp() {
|
||||
// Выполняет UDP портовое сканирование well-known портов
|
||||
int status;
|
||||
int sockfd; // Дескриптор сокета
|
||||
struct sockaddr_in target_addr; // Адрес цели
|
||||
static int port_idx; // Текущий индекс порта
|
||||
static const int ports[]; // Список портов
|
||||
static const int ports_total;
|
||||
int curr_port; // Текущий порт
|
||||
const char dummy_data[]; // Данные для отправки
|
||||
ssize_t send_result; // Результат отправки
|
||||
|
||||
// Инициализация структуры адреса
|
||||
memset(&target_addr, 0, sizeof(target_addr));
|
||||
curr_port = ports[port_idx];
|
||||
dummy_data[] = "SCAN";
|
||||
target_addr.sin_family = AF_INET;
|
||||
target_addr.sin_port = htons(curr_port);
|
||||
ports_total = sizeof(ports)/sizeof(ports[0])
|
||||
sockfd = -1;
|
||||
port_idx = 0;
|
||||
ports = { // Список портов
|
||||
53, 67, 68, 69, 123, 161, 162, 389, 443, 500, 514, 520, 1900, 4500
|
||||
};
|
||||
|
||||
// Преобразование IP
|
||||
if (inet_pton(AF_INET, ip, &target_addr.sin_addr) <= 0) {
|
||||
n_fail_requests++;
|
||||
status = -501; // Код ошибки: неверный IP
|
||||
if (sockfd != -1) close(sockfd);
|
||||
return status;
|
||||
}
|
||||
|
||||
// Создание сокета
|
||||
sockfd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
||||
if (sockfd < 0) {
|
||||
n_fail_requests++;
|
||||
status = -502; // Ошибка создания сокета
|
||||
if (sockfd != -1) close(sockfd);
|
||||
return status;
|
||||
}
|
||||
|
||||
// Отправка данных
|
||||
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;
|
||||
}
|
||||
|
||||
// Callback для записи ответа от сервера
|
||||
size_t WriteCallback(void* contents, size_t size, size_t nmemb, std::string* output) {
|
||||
size_t total_size = size * nmemb;
|
||||
|
|
@ -826,7 +837,6 @@ int main(int arg_ctr, char **arg_ptr)
|
|||
|
||||
status = my_check_params(); // Проверяем параметры командной строки
|
||||
|
||||
|
||||
switch (status) // Обрабатываем результат проверки параметров
|
||||
{
|
||||
case 1: // Режим сканирования портов (UDP)
|
||||
|
|
|
|||
Loading…
Reference in New Issue