mirror of
https://github.com/Theldus/alertik.git
synced 2024-11-07 08:34:44 +01:00
Setup proper static events with env vars too
This commit is contained in:
parent
17c7f129e3
commit
770e95dd04
10
alertik.c
10
alertik.c
@ -49,13 +49,11 @@ int main(void)
|
||||
pthread_t handler;
|
||||
int fd;
|
||||
|
||||
/* TODO: remove setup_notifiers()..
|
||||
* think about env vars for the static events too, like enable/disable
|
||||
*/
|
||||
|
||||
log_init();
|
||||
setup_notifiers();
|
||||
init_environment_events();
|
||||
|
||||
if (!init_static_events() && !init_environment_events())
|
||||
panic("No event was configured, please configure at least one\n"
|
||||
"before proceeding!\n");
|
||||
|
||||
log_msg(
|
||||
"Alertik (" GIT_HASH ") (built at " __DATE__ " " __TIME__ ")\n");
|
||||
|
13
env_events.c
13
env_events.c
@ -147,8 +147,13 @@ int process_environment_event(struct log_event *ev)
|
||||
return handled;
|
||||
}
|
||||
|
||||
|
||||
/**/
|
||||
/**
|
||||
* @brief Initialize environment variables events.
|
||||
*
|
||||
* @return Returns 0 if there is no environment event,
|
||||
* 1 if there is at least one _and_ is successfully
|
||||
* configured.
|
||||
*/
|
||||
int init_environment_events(void)
|
||||
{
|
||||
char *tmp;
|
||||
@ -163,7 +168,7 @@ int init_environment_events(void)
|
||||
panic("Environment ENV_EVENTS exceeds the maximum supported (%d/%d)\n",
|
||||
num_env_events, MAX_ENV_EVENTS);
|
||||
|
||||
log_msg("%d environment events found, registering...\n");
|
||||
log_msg("%d environment event(s) found, registering...\n");
|
||||
for (int i = 0; i < num_env_events; i++) {
|
||||
/* EVENTn_MATCH_TYPE. */
|
||||
env_events[i].ev_match_type = get_event_idx(i, "MATCH_TYPE",
|
||||
@ -193,5 +198,5 @@ int init_environment_events(void)
|
||||
/* Try to setup notifier if not yet. */
|
||||
notifiers[env_events[i].ev_notifier_idx].setup();
|
||||
}
|
||||
return (0);
|
||||
return 1;
|
||||
}
|
||||
|
106
events.c
106
events.c
@ -3,12 +3,17 @@
|
||||
* This is free and unencumbered software released into the public domain.
|
||||
*/
|
||||
|
||||
#include <ctype.h>
|
||||
#include <errno.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include "events.h"
|
||||
#include "alertik.h"
|
||||
#include "notifiers.h"
|
||||
#include "log.h"
|
||||
|
||||
void handle_wifi_login_attempts(struct log_event *ev);
|
||||
static void handle_wifi_login_attempts(struct log_event *, int);
|
||||
|
||||
/* Handlers. */
|
||||
struct ev_handler handlers[NUM_EVENTS] = {
|
||||
@ -16,11 +21,36 @@ struct ev_handler handlers[NUM_EVENTS] = {
|
||||
{
|
||||
.str = "unicast key exchange timeout",
|
||||
.hnd = handle_wifi_login_attempts,
|
||||
.evnt_type = EVNT_SUBSTR
|
||||
.evnt_type = EVNT_SUBSTR,
|
||||
.enabled = 0,
|
||||
.evnt_notifier_idx = NOTIFY_IDX_TELE
|
||||
},
|
||||
/* Add new handlers here. */
|
||||
};
|
||||
|
||||
/**/
|
||||
static char *get_event_str(long ev_num, char *str)
|
||||
{
|
||||
char *env;
|
||||
char ev[64] = {0};
|
||||
snprintf(ev, sizeof ev - 1, "STATIC_EVENT%ld_%s", ev_num, str);
|
||||
if (!(env = getenv(ev)))
|
||||
panic("Unable to find event for %s\n", ev);
|
||||
return env;
|
||||
}
|
||||
|
||||
/**/
|
||||
static int
|
||||
get_event_idx(long ev_num, char *str, const char *const *str_list, int size)
|
||||
{
|
||||
char *env = get_event_str(ev_num, str);
|
||||
for (int i = 0; i < size; i++) {
|
||||
if (!strcmp(env, str_list[i]))
|
||||
return i;
|
||||
}
|
||||
panic("String parameter (%s) invalid for %s\n", env, str);
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Given an event, checks if it belongs to one of the
|
||||
* registered events and then, handle it.
|
||||
@ -36,13 +66,77 @@ int process_static_event(struct log_event *ev)
|
||||
|
||||
for (i = 0, handled = 0; i < NUM_EVENTS; i++) {
|
||||
if (strstr(ev->msg, handlers[i].str)) {
|
||||
handlers[i].hnd(ev);
|
||||
handlers[i].hnd(ev, i);
|
||||
handled += 1;
|
||||
}
|
||||
}
|
||||
return handled;
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Initialize static events.
|
||||
*
|
||||
* @return Returns 0 if there is no static event, and 1 if
|
||||
* there is at least one _and_ is successfully configured.
|
||||
*/
|
||||
int init_static_events(void)
|
||||
{
|
||||
char *ptr, *end;
|
||||
long ev;
|
||||
|
||||
/* Check for: STATIC_EVENTS_ENABLED=0,3,5,2... */
|
||||
ptr = getenv("STATIC_EVENTS_ENABLED");
|
||||
if (!ptr || ptr[0] == '\0') {
|
||||
log_msg("Static events not detected, disabling...\n");
|
||||
return (0);
|
||||
}
|
||||
|
||||
end = ptr;
|
||||
errno = 0;
|
||||
|
||||
do
|
||||
{
|
||||
ev = strtol(end, &end, 10);
|
||||
if (errno != 0 || ((ptr == end) && ev == 0))
|
||||
panic("Unable to parse STATIC_EVENTS_ENABLED, aborting...\n");
|
||||
|
||||
/* Skip whitespaces. */
|
||||
while (*end != '\0' && isspace(*end))
|
||||
end++;
|
||||
|
||||
/* Check if ev number is sane. */
|
||||
if (ev < 0 || ev >= NUM_EVENTS)
|
||||
panic("Event (%ld) is not valid!, should be between 0-%d\n",
|
||||
ev, NUM_EVENTS - 1);
|
||||
|
||||
/* Try to retrieve & initialize notifier for the event. */
|
||||
handlers[ev].evnt_notifier_idx =
|
||||
get_event_idx(ev, "NOTIFIER", notifiers_str, NUM_NOTIFIERS);
|
||||
handlers[ev].enabled = 1;
|
||||
|
||||
if (*end != ',' && *end != '\0')
|
||||
panic("Wrong event number in STATIC_EVENTS_ENABLED, aborting...\n");
|
||||
|
||||
} while (*end++ != '\0');
|
||||
|
||||
|
||||
log_msg("Static events summary:\n");
|
||||
for (int i = 0; i < NUM_EVENTS; i++) {
|
||||
if (!handlers[i].enabled)
|
||||
continue;
|
||||
|
||||
printf(
|
||||
"STATIC_EVENT%d : enabled\n"
|
||||
"STATIC_EVENT%d_NOTIFIER: %s\n\n",
|
||||
i, i, notifiers_str[handlers[i].evnt_notifier_idx]
|
||||
);
|
||||
|
||||
/* Try to setup notifier if not yet. */
|
||||
notifiers[handlers[i].evnt_notifier_idx].setup();
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
|
||||
///////////////////////////// FAILED LOGIN ATTEMPTS ///////////////////////////
|
||||
static int
|
||||
@ -79,12 +173,13 @@ parse_login_attempt_msg(const char *msg, char *wifi_iface, char *mac_addr)
|
||||
return (0);
|
||||
}
|
||||
|
||||
void handle_wifi_login_attempts(struct log_event *ev)
|
||||
static void handle_wifi_login_attempts(struct log_event *ev, int idx_env)
|
||||
{
|
||||
char time_str[32] = {0};
|
||||
char mac_addr[32] = {0};
|
||||
char wifi_iface[32] = {0};
|
||||
char notification_message[2048] = {0};
|
||||
int notif_idx;
|
||||
|
||||
log_msg("> Login attempt detected!\n");
|
||||
|
||||
@ -104,7 +199,8 @@ void handle_wifi_login_attempts(struct log_event *ev)
|
||||
|
||||
log_msg("> Retrieved info, MAC: (%s), Interface: (%s)\n", mac_addr, wifi_iface);
|
||||
|
||||
if (send_telegram_notification(notification_message) < 0) {
|
||||
notif_idx = handlers[idx_env].evnt_notifier_idx;
|
||||
if (notifiers[notif_idx].send_notification(notification_message) < 0) {
|
||||
log_msg("unable to send the notification!\n");
|
||||
return;
|
||||
}
|
||||
|
9
events.h
9
events.h
@ -21,12 +21,15 @@
|
||||
};
|
||||
|
||||
struct ev_handler {
|
||||
const char *str;
|
||||
void(*hnd)(struct log_event *);
|
||||
int evnt_type;
|
||||
const char *str; /* Substr or regex to match. */
|
||||
void(*hnd)(struct log_event *, int); /* Event handler. */
|
||||
int evnt_type; /* Whether substr or regex. */
|
||||
int evnt_notifier_idx; /* Telegram, Discord... */
|
||||
int enabled; /* Whether if handler enabled or not. */
|
||||
};
|
||||
|
||||
extern struct ev_handler handlers[NUM_EVENTS];
|
||||
extern int process_static_event(struct log_event *ev);
|
||||
extern int init_static_events(void);
|
||||
|
||||
#endif /* EVENTS_H */
|
||||
|
@ -132,10 +132,3 @@ struct notifier notifiers[] = {
|
||||
.send_notification = send_telegram_notification
|
||||
}
|
||||
};
|
||||
|
||||
/* Global setup. */
|
||||
void setup_notifiers(void)
|
||||
{
|
||||
for (int i = 0; i < NUM_NOTIFIERS; i++)
|
||||
notifiers[i].setup();
|
||||
}
|
||||
|
@ -11,6 +11,11 @@
|
||||
// #define VALIDATE_CERTS
|
||||
// #define DISABLE_NOTIFICATIONS
|
||||
|
||||
/*
|
||||
* Notifier indexes.
|
||||
*/
|
||||
#define NOTIFY_IDX_TELE 0
|
||||
|
||||
#define CURL_USER_AGENT "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 " \
|
||||
"(KHTML, like Gecko) Chrome/125.0.0.0 Safari/537.36"
|
||||
|
||||
@ -34,7 +39,6 @@
|
||||
};
|
||||
|
||||
extern struct notifier notifiers[NUM_NOTIFIERS];
|
||||
extern void setup_notifiers(void);
|
||||
extern int is_within_notify_threshold(void);
|
||||
extern void update_notify_last_sent(void);
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user