181 lines
4.0 KiB
C
181 lines
4.0 KiB
C
#pragma once
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#include <stdbool.h>
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
#include <stdarg.h>
|
|
#include <inttypes.h>
|
|
#include <stdio.h>
|
|
|
|
#include "sdkconfig.h"
|
|
#include "esp_log.h"
|
|
|
|
/* >>> CRITIQUE <<< */
|
|
#include "c2.pb.h" /* c2_Command, c2_AgentMsgType */
|
|
|
|
/* ============================================================
|
|
* GLOBAL DEFINES
|
|
* ============================================================ */
|
|
|
|
#define MAX_ARGS 10
|
|
#define MAX_RESPONSE_SIZE 1024
|
|
|
|
/* ============================================================
|
|
* LOG HELPERS
|
|
* ============================================================ */
|
|
#ifdef CONFIG_LOG_COLORS
|
|
#define ESPILON_LOG_PURPLE "\033[0;35m"
|
|
#define ESPILON_LOG_RESET "\033[0m"
|
|
#else
|
|
#define ESPILON_LOG_PURPLE ""
|
|
#define ESPILON_LOG_RESET ""
|
|
#endif
|
|
|
|
static inline void espilon_log_purple(
|
|
const char *tag,
|
|
const char *fmt,
|
|
...
|
|
) {
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
|
|
printf(ESPILON_LOG_PURPLE "I (%" PRIu32 ") %s: ",
|
|
(uint32_t)esp_log_timestamp(), tag);
|
|
vprintf(fmt, args);
|
|
printf(ESPILON_LOG_RESET "\n");
|
|
|
|
va_end(args);
|
|
}
|
|
|
|
#define ESPILON_LOGI_PURPLE(tag, fmt, ...) \
|
|
espilon_log_purple(tag, fmt, ##__VA_ARGS__)
|
|
|
|
/* Socket TCP global */
|
|
extern int sock;
|
|
|
|
/* ============================================================
|
|
* COM INIT
|
|
* ============================================================ */
|
|
|
|
bool com_init(void);
|
|
|
|
/* ============================================================
|
|
* CRYPTO API
|
|
* ============================================================ */
|
|
|
|
/*
|
|
* ChaCha20 encrypt/decrypt
|
|
* Retourne un buffer malloc()'d → free() obligatoire
|
|
*/
|
|
unsigned char *chacha_cd(const unsigned char *data, size_t data_len);
|
|
|
|
/* Base64 helpers */
|
|
char *base64_decode(const char *input, size_t *output_len);
|
|
char *base64_encode(const unsigned char *input, size_t input_len);
|
|
|
|
/* C2 decode + decrypt + protobuf + exec */
|
|
bool c2_decode_and_exec(const char *frame);
|
|
/* ============================================================
|
|
* ESP → C2 Messaging API
|
|
* ============================================================ */
|
|
|
|
bool agent_send(
|
|
c2_AgentMsgType type,
|
|
const char *source,
|
|
const char *request_id,
|
|
const void *data,
|
|
size_t len,
|
|
bool eof
|
|
);
|
|
|
|
/* Helpers globaux */
|
|
bool msg_info(
|
|
const char *src,
|
|
const char *msg,
|
|
const char *req
|
|
);
|
|
|
|
bool msg_error(
|
|
const char *src,
|
|
const char *msg,
|
|
const char *req
|
|
);
|
|
|
|
bool msg_data(
|
|
const char *src,
|
|
const void *data,
|
|
size_t len,
|
|
bool eof,
|
|
const char *req
|
|
);
|
|
|
|
/* ============================================================
|
|
* DEVICE
|
|
* ============================================================ */
|
|
|
|
bool device_id_matches(
|
|
const char *local_id,
|
|
const char *target_id
|
|
);
|
|
|
|
/* ============================================================
|
|
* CORE PROCESSING (C2 → ESP)
|
|
* ============================================================ */
|
|
|
|
void process_command(
|
|
const c2_Command *cmd
|
|
);
|
|
|
|
/*
|
|
* Compat legacy optionnel
|
|
*/
|
|
void process_command_from_buffer(
|
|
uint8_t *buffer,
|
|
size_t len
|
|
);
|
|
|
|
/* ============================================================
|
|
* WIFI
|
|
* ============================================================ */
|
|
#ifdef CONFIG_NETWORK_WIFI
|
|
void wifi_init(void);
|
|
void tcp_client_task(void *pvParameters);
|
|
#endif
|
|
|
|
/* ============================================================
|
|
* GPRS
|
|
* ============================================================ */
|
|
|
|
#ifdef CONFIG_NETWORK_GPRS
|
|
#define BUFF_SIZE 1024
|
|
#define UART_NUM UART_NUM_1
|
|
#define TXD_PIN 27
|
|
#define RXD_PIN 26
|
|
#define PWR_KEY 4
|
|
#define PWR_EN 23
|
|
#define RESET 5
|
|
#define LED_GPIO 13
|
|
|
|
void setup_uart(void);
|
|
void setup_modem(void);
|
|
|
|
bool connect_gprs(void);
|
|
bool connect_tcp(void);
|
|
|
|
bool gprs_send(const void *buf, size_t len);
|
|
void gprs_rx_poll(void);
|
|
void close_tcp_connection(void);
|
|
|
|
void gprs_client_task(void *pvParameters);
|
|
void send_at_command(const char *cmd);
|
|
#endif
|
|
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|