134 lines
4.1 KiB
C
134 lines
4.1 KiB
C
#ifndef COMMUNICATION_HANDLER_H
|
|
#define COMMUNICATION_HANDLER_H
|
|
|
|
#include "client_handler.h"
|
|
#include <esp_now.h>
|
|
#include <esp_wifi.h>
|
|
#include <freertos/FreeRTOS.h>
|
|
#include <freertos/queue.h>
|
|
#include <freertos/task.h>
|
|
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <sys/types.h>
|
|
#include "esp_partition.h"
|
|
#include "message_structs.h"
|
|
|
|
#define BROADCAST_INTERVAL_MS 500
|
|
|
|
#define CLIENT_TIMEOUT_MS 5000 // 5 Sekunden Timeout
|
|
#define CHECK_INTERVAL_MS 1000 // Jede Sekunde überprüfen
|
|
|
|
extern uint8_t broadcast_address[ESP_NOW_ETH_ALEN];
|
|
#define IS_BROADCAST_ADDR(addr) (memcmp(addr, broadcast_address, ESP_NOW_ETH_ALEN) == 0)
|
|
|
|
#define MESSAGE_QUEUE_SIZE 10
|
|
|
|
typedef union __attribute__((packed)) {
|
|
OTA_PREPARE_FOR_UPDATE_Payload ota_prepare_for_update_payload;
|
|
OTA_PREPARE_ACKNOWLEDGED_Payload ota_prepare_acknowledged_payload;
|
|
OTA_READY_TO_RECEIVE_Payload ota_ready_to_receive_payload;
|
|
OTA_CHUNK_Payload ota_chunk_payload;
|
|
OTA_REQUEST_BLOCK_STATUS_Payload ota_request_block_status_payload;
|
|
OTA_BLOCK_STATUS_REPORT_Payload ota_block_status_report_payload;
|
|
OTA_COMMIT_BLOCK_Payload ota_commit_block_payload;
|
|
OTA_BLOCK_COMMITTED_Payload ota_block_committed_payload;
|
|
OTA_FINISH_UPDATE_Payload ota_finish_update_payload;
|
|
OTA_UPDATE_STATUS_Payload ota_update_status_payload;
|
|
StatusPayload status_payload;
|
|
ConfigPayload config_payload;
|
|
PingPayload ping_payload;
|
|
BroadCastPayload broadcast_payload;
|
|
RegisterPayload register_payload;
|
|
FirmwarePrepPayload firmware_prep_payload;
|
|
FirmwarePayload firmware_payload;
|
|
} PayloadUnion;
|
|
|
|
typedef struct __attribute__((packed)) {
|
|
uint16_t version; // protcol version
|
|
CommandPages commandPage;
|
|
uint16_t length;
|
|
PayloadUnion payload;
|
|
} BaseMessage;
|
|
|
|
static_assert(sizeof(BaseMessage) <= 255,
|
|
"BaseMessage darf nicht größer als 255 sein");
|
|
|
|
typedef void (*ESP_RegisterFunctionCallback)(
|
|
const esp_now_recv_info_t *esp_now_info, const uint8_t *data, int data_len);
|
|
typedef void (*ESP_RegisterTaskCallback)(
|
|
const esp_now_recv_info_t *esp_now_info, const uint8_t *data, int data_len);
|
|
|
|
struct ESP_RegisterdFunction {
|
|
CommandPages MSGID;
|
|
ESP_RegisterFunctionCallback callback;
|
|
};
|
|
|
|
struct ESP_RegisterdTask {
|
|
CommandPages MSGID;
|
|
ESP_RegisterTaskCallback task;
|
|
};
|
|
|
|
struct ESP_MessageBroker {
|
|
struct ESP_RegisterdFunction FunctionList[64];
|
|
uint8_t num_direct_callbacks;
|
|
struct ESP_RegisterdTask TaskList[64];
|
|
uint8_t num_task_callbacks;
|
|
};
|
|
|
|
typedef struct {
|
|
QueueHandle_t message_queue;
|
|
} ESP_MessageBrokerTaskParams_t;
|
|
|
|
typedef struct {
|
|
esp_now_recv_info_t esp_now_info;
|
|
uint8_t *data;
|
|
int data_len;
|
|
} ESPNOW_MessageInfo;
|
|
|
|
void ESP_InitMessageBroker(QueueHandle_t msg_queue_handle);
|
|
void ESP_RegisterFunction(CommandPages command,
|
|
ESP_RegisterFunctionCallback callback);
|
|
void ESP_RegisterTask(CommandPages command, ESP_RegisterTaskCallback callback);
|
|
void ESP_MessageBrokerTask(void *param);
|
|
|
|
void ESPNOW_RegisterMasterCallbacks();
|
|
void ESPNOW_RegisterSlaveCallbacks();
|
|
void ESPNOW_RegisterOTAMaster();
|
|
void ESPNOW_RegisterOTASlave();
|
|
|
|
int init_com(ClientList *clients, uint8_t wifi_channel);
|
|
int getNextFreeClientId();
|
|
int add_peer(uint8_t *macAddr);
|
|
BaseMessage MessageBuilder(CommandPages commandPage, PayloadUnion payload,
|
|
size_t payload_size);
|
|
|
|
void master_broadcast_task(void *param);
|
|
void master_ping_task(void *param);
|
|
void master_broadcast_ping(void *param);
|
|
void master_receive_callback(const esp_now_recv_info_t *esp_now_info,
|
|
const uint8_t *data, int data_len);
|
|
|
|
void client_receive_callback(const esp_now_recv_info_t *esp_now_info,
|
|
const uint8_t *data, int data_len);
|
|
void client_data_sending_task(void *param);
|
|
void client_send_random_data_task(void *param);
|
|
#include "esp_partition.h"
|
|
|
|
typedef struct {
|
|
const esp_partition_t *update_partition;
|
|
} master_ota_task_params_t;
|
|
|
|
|
|
|
|
extern bool g_ota_in_progress;
|
|
|
|
void client_monitor_task(void *pvParameters);
|
|
|
|
void send_ota_block_chunks(uint8_t client_id, uint16_t block_id);
|
|
|
|
extern const esp_partition_t *ota_update_partition;
|
|
|
|
#endif
|