Compare commits
18 Commits
7e033b7779
...
master
| Author | SHA1 | Date | |
|---|---|---|---|
| a8cfd61491 | |||
| 14ca6d651b | |||
| ff8d7914a9 | |||
| 6a71c2863d | |||
| c8ebbb2946 | |||
| 998bafc53a | |||
| 6a0df95eeb | |||
| 9f7d1aae4f | |||
| 55b3404674 | |||
| 7aa5a91181 | |||
| bfcd91e61a | |||
| 7edc2e21da | |||
| a5f79f5eac | |||
| cf4914d939 | |||
| 854d6eab36 | |||
| 24554e9d23 | |||
| fe09b689c8 | |||
| 628f031f26 |
@@ -31,7 +31,7 @@ add_executable(${EXECUTABLE} "")
|
||||
# Set compiler/linker options
|
||||
target_compile_definitions(${EXECUTABLE} PRIVATE
|
||||
${COMPILER_DEFINES}
|
||||
INCPHUB_LOCAL_NIL_ADDR=1
|
||||
LIBSHUNT_LOCAL_NIL_ADDR=1
|
||||
)
|
||||
target_compile_options(${EXECUTABLE} PRIVATE
|
||||
-Wall
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
add_subdirectory(tasks)
|
||||
add_subdirectory(incphub)
|
||||
add_subdirectory(libshunt)
|
||||
add_subdirectory(nil)
|
||||
add_subdirectory(ntl)
|
||||
|
||||
|
||||
@@ -1,19 +0,0 @@
|
||||
# INCPHUB
|
||||
|
||||
## FreeRTOS requirements
|
||||
|
||||
- INCLUDE_xTaskGetCurrentTaskHandle set to 1
|
||||
- INCLUDE_vTaskSuspend set to 1
|
||||
- configUSE_TASK_NOTIFICATIONS set to 1
|
||||
- configTASK_NOTIFICATION_ARRAY_ENTRIES set to >= 2 (0 is for stream/message buffers, 1(configurable) is used by incphub)
|
||||
- configUSE_MUTEXES set to 1
|
||||
|
||||
## Configuration
|
||||
|
||||
Set as a macro in compile flags if wishing to override the defaults
|
||||
|
||||
- INCPHUB_TASKINDEX_MAILBOX, defaults to 1 (avoid using 0)
|
||||
|
||||
## Setup
|
||||
|
||||
- incphub_task_preinit MUST be run before any calls to the client, likely even before the scheduler is started
|
||||
@@ -1,29 +0,0 @@
|
||||
// incphub.h - Public definitions common to incphub and incphub client
|
||||
|
||||
#ifndef INCPHUB_H_
|
||||
#define INCPHUB_H_
|
||||
|
||||
#include "ntl/ntl.h"
|
||||
#include "nil/nil.h"
|
||||
|
||||
/// @brief Success
|
||||
#define INCPHUB_OK (0)
|
||||
/// @brief Unspecified error
|
||||
#define INCPHUB_ERR (1)
|
||||
/// @brief Invalid argument
|
||||
#define INCPHUB_EINVAL (2)
|
||||
/// @brief Resource or entry not found
|
||||
#define INCPHUB_ENOTFOUND (3)
|
||||
/// @brief Resource in use
|
||||
#define INCPHUB_EBUSY (4)
|
||||
/// @brief Resource temporarily unavailable
|
||||
#define INCPHUB_EAGAIN (5)
|
||||
/// @brief Could not allocate memory
|
||||
#define INCPHUB_ENOMEM (6)
|
||||
|
||||
typedef struct {
|
||||
nil_addr_t addr;
|
||||
ntl_port_t port;
|
||||
} incphub_addr_t;
|
||||
|
||||
#endif
|
||||
@@ -1,53 +0,0 @@
|
||||
// incphub_client.h - Public interface of incphub client side
|
||||
|
||||
#ifndef INCPHUB_CLIENT_H_
|
||||
#define INCPHUB_CLIENT_H_
|
||||
|
||||
#include <stddef.h>
|
||||
|
||||
#include "incphub/incphub.h"
|
||||
|
||||
#include "nil/nil.h"
|
||||
#include "ntl/ntl.h"
|
||||
|
||||
|
||||
|
||||
typedef struct incphub_cli_t incphub_cli_t;
|
||||
|
||||
|
||||
|
||||
/// @brief Initializes an incphub client
|
||||
/// @param port The port number to assign to the caller
|
||||
/// @return INCPHUB_OK on success, an error otherwise
|
||||
int incphub_init_client(incphub_cli_t **cli, ntl_port_t port);
|
||||
|
||||
/// @brief Attempts to get a free message buffer from the central storage space
|
||||
/// @param cli Current client handle
|
||||
/// @param buffer_ptr Location to return the buffer
|
||||
/// @param length The minimum size, in bytes, of the buffer get
|
||||
/// @return INCPHUB_OK on success, an error otherwise
|
||||
int incphub_get_msg_buffer(incphub_cli_t *cli, uint8_t **buffer_ptr, size_t length);
|
||||
|
||||
/// @brief Releases a buffer previously acquired with incphub_get_msg_buffer
|
||||
/// @param cli Current client handle
|
||||
/// @param buffer The buffer to release
|
||||
/// @return INCPHUB_OK on success, an error otherwise
|
||||
int incphub_release_msg_buffer(incphub_cli_t *cli, uint8_t *buffer);
|
||||
|
||||
/// @brief Attempts to send a message to the local interface
|
||||
/// @param cli Current client handle
|
||||
/// @param buffer The message buffer
|
||||
/// @param length The message length
|
||||
/// @param dst_port The destination NTL port
|
||||
/// @return INCPHUB_OK on success, an error otherwise
|
||||
int incphub_send_local(incphub_cli_t *cli, uint8_t *buffer, size_t length, ntl_port_t dst_port);
|
||||
|
||||
/// @brief Awaits the receival of a message
|
||||
/// @param cli Current client handle
|
||||
/// @param buffer_ptr Location to return the buffer
|
||||
/// @param length_ptr Location to return the length
|
||||
/// @param source_ptr Location to return the message sender
|
||||
/// @return INCPHUB_OK on success, an error otherwise
|
||||
int incphub_await_msg(incphub_cli_t *cli, uint8_t **buffer_ptr, size_t *length_ptr, incphub_addr_t *source_ptr);
|
||||
|
||||
#endif
|
||||
@@ -1,17 +0,0 @@
|
||||
#ifndef INCPHUB_TASK_H_
|
||||
#define INCPHUB_TASK_H_
|
||||
|
||||
#include "FreeRTOS.h"
|
||||
#include "task.h"
|
||||
#include "FreeRTOSConfig.h"
|
||||
|
||||
#define INCPHUB_TASK_STACK_DEPTH configMINIMAL_STACK_SIZE
|
||||
#define INCPHUB_TASK_PRIORITY (configMAX_PRIORITIES - 1)
|
||||
|
||||
extern StackType_t _incphub_task_stack[INCPHUB_TASK_STACK_DEPTH];
|
||||
extern StaticTask_t _incphub_task;
|
||||
|
||||
void incphub_task_preinit();
|
||||
void incphub_task_main(void *params);
|
||||
|
||||
#endif
|
||||
@@ -1,101 +0,0 @@
|
||||
#include <stdbool.h>
|
||||
|
||||
#include "incphub/incphub_client.h"
|
||||
#include "incphub/incphub_task.h"
|
||||
#include "incphub/incphub.h"
|
||||
#include "incphub_config.h"
|
||||
#include "incphub_shared.h"
|
||||
|
||||
#include "ntl/ntl.h"
|
||||
#include "nil/nil.h"
|
||||
|
||||
#include "message_buffer.h"
|
||||
|
||||
|
||||
|
||||
int incphub_init_client(incphub_cli_t **cli, ntl_port_t port)
|
||||
{
|
||||
return register_client(cli, port, xTaskGetCurrentTaskHandle());
|
||||
}
|
||||
|
||||
int incphub_get_msg_buffer(incphub_cli_t *cli, uint8_t **buffer_ptr, size_t length)
|
||||
{
|
||||
if (cli == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
if (buffer_ptr == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
|
||||
// Request extra space for NIL/NTL headers
|
||||
size_t extended_len = sizeof(nil_header_t) + sizeof(ntl_header_t) + length;
|
||||
uint8_t *buf;
|
||||
|
||||
int status = get_msg_buffer(&buf, extended_len);
|
||||
if (status != INCPHUB_OK)
|
||||
return status;
|
||||
|
||||
// Reserve space for NIL/NTL headers
|
||||
buf += sizeof(nil_header_t) + sizeof(ntl_header_t);
|
||||
|
||||
// Return buffer
|
||||
*buffer_ptr = buf;
|
||||
|
||||
return INCPHUB_OK;
|
||||
}
|
||||
|
||||
int incphub_release_msg_buffer(incphub_cli_t *cli, uint8_t *buffer)
|
||||
{
|
||||
if (cli == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
if (buffer == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
|
||||
// cli requested for interface consistency only (for now)
|
||||
(void)cli;
|
||||
|
||||
// Unwind buffer pointer
|
||||
uint8_t *unwound = buffer - sizeof(nil_header_t) + sizeof(ntl_header_t);
|
||||
|
||||
// Release correct buffer pointer
|
||||
return release_msg_buffer(buffer);
|
||||
}
|
||||
|
||||
int incphub_send_local(incphub_cli_t *cli, uint8_t *buffer, size_t length, ntl_port_t dst_port)
|
||||
{
|
||||
if (cli == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
if (buffer == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
if (dst_port == NTL_PORT_EMPTY)
|
||||
return INCPHUB_EINVAL;
|
||||
|
||||
// Unwind buffer pointer
|
||||
ntl_packet_t *ntl_packet = (ntl_packet_t*)(buffer - sizeof(ntl_header_t));
|
||||
nil_packet_t *nil_packet = (nil_packet_t*)(ntl_packet - sizeof(ntl_header_t));
|
||||
size_t extended_len = sizeof(nil_header_t) + sizeof(ntl_header_t) + length;
|
||||
|
||||
// Populate NTL header
|
||||
ntl_build_header(&ntl_packet->header, dst_port, cli->port, length);
|
||||
|
||||
// Populate NIL header
|
||||
nil_build_header(&nil_packet->header, NIL_ADDR_LOCALHOST, INCPHUB_LOCAL_NIL_ADDR, length + sizeof(ntl_header_t));
|
||||
|
||||
// Pass to ingest
|
||||
return enqueue_message_ingest((uint8_t *)nil_packet, extended_len);
|
||||
}
|
||||
|
||||
int incphub_await_msg(incphub_cli_t *cli, uint8_t **buffer_ptr, size_t *length_ptr, incphub_addr_t *source_ptr)
|
||||
{
|
||||
if (cli == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
if (buffer_ptr == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
if (length_ptr == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
if (source_ptr == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
|
||||
// cli requested for interface consistency only (for now)
|
||||
(void)cli;
|
||||
|
||||
return await_message_dispatch(buffer_ptr, length_ptr, source_ptr);
|
||||
}
|
||||
@@ -1,30 +0,0 @@
|
||||
// incphub_config.h - Default configurations for incphub
|
||||
|
||||
#ifndef INCPHUB_CONFIG_H_
|
||||
#define INCPHUB_CONFIG_H_
|
||||
|
||||
#ifndef INCPHUB_TASKINDEX_MAILBOX
|
||||
/// @brief Index of direct-to-task notification for mailbox to use
|
||||
#define INCPHUB_TASKINDEX_MAILBOX 1
|
||||
#endif
|
||||
|
||||
#ifndef INCPHUB_MAX_CLIENTS
|
||||
/// @brief Number of client slots to allocate
|
||||
#define INCPHUB_MAX_CLIENTS 16
|
||||
#endif
|
||||
|
||||
#ifndef INCPHUB_MSG_BUFFER_SIZE
|
||||
/// @brief Message buffer size in bytes. Must allow space for NIL and NTL headers
|
||||
#define INCPHUB_MSG_BUFFER_SIZE 32
|
||||
#endif
|
||||
|
||||
#ifndef INCPHUB_MSG_BUFFER_COUNT
|
||||
/// @brief Number of message buffers to allocate
|
||||
#define INCPHUB_MSG_BUFFER_COUNT 8
|
||||
#endif
|
||||
|
||||
#ifndef INCPHUB_LOCAL_NIL_ADDR
|
||||
#error "Local NIL address must be specified with INCPHUB_LOCAL_NIL_ADDR"
|
||||
#endif
|
||||
|
||||
#endif
|
||||
@@ -1,242 +0,0 @@
|
||||
#include "incphub_shared.h"
|
||||
|
||||
#include <string.h>
|
||||
#include <stdbool.h>
|
||||
|
||||
#include "FreeRTOS.h"
|
||||
#include "semphr.h"
|
||||
#include "task.h"
|
||||
|
||||
#include "incphub/incphub.h"
|
||||
#include "incphub_config.h"
|
||||
|
||||
#include "nil/nil.h"
|
||||
|
||||
|
||||
// === Private variables ===
|
||||
|
||||
// Global lock
|
||||
static SemaphoreHandle_t _incphub_lock;
|
||||
static StaticSemaphore_t _incphub_lock_store;
|
||||
|
||||
// Clients
|
||||
static incphub_cli_t _clients[INCPHUB_MAX_CLIENTS];
|
||||
|
||||
// Message buffers
|
||||
// INCPHUB_MSG_BUFFER_COUNT buffers of INCPHUB_MSG_BUFFER_SIZE bytes
|
||||
static uint8_t _msg_buffers[INCPHUB_MSG_BUFFER_COUNT][INCPHUB_MSG_BUFFER_SIZE];
|
||||
static bool _msg_buffer_used[INCPHUB_MSG_BUFFER_COUNT];
|
||||
|
||||
|
||||
|
||||
// === Private function declarations ===
|
||||
static bool global_lock(TickType_t max_delay);
|
||||
static void global_unlock();
|
||||
/// @remark Returns NULL if not found
|
||||
static incphub_cli_t *find_empty_client();
|
||||
/// @remark Returns NULL if not found
|
||||
static incphub_cli_t *find_client_with_port(ntl_port_t port);
|
||||
/// @remark Returns NULL if not found
|
||||
static uint8_t *find_unused_buffer();
|
||||
/// @remark Returns -1 if not found
|
||||
static int find_index_of_buffer(uint8_t *buffer);
|
||||
|
||||
|
||||
|
||||
// === Public functions ===
|
||||
int internal_init()
|
||||
{
|
||||
_incphub_lock = xSemaphoreCreateMutexStatic(&_incphub_lock_store);
|
||||
memset(_clients, 0, sizeof(_clients));
|
||||
memset(_msg_buffer_used, 0, sizeof(_msg_buffer_used));
|
||||
return INCPHUB_OK;
|
||||
}
|
||||
|
||||
int register_client(incphub_cli_t **cli_ptr, ntl_port_t port, TaskHandle_t task)
|
||||
{
|
||||
if (port == NTL_PORT_EMPTY)
|
||||
return INCPHUB_EINVAL;
|
||||
|
||||
if (cli_ptr == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
|
||||
// == Enter ==
|
||||
if (!global_lock(portMAX_DELAY))
|
||||
return INCPHUB_EAGAIN;
|
||||
|
||||
// Prevent double binding of ports
|
||||
if (find_client_with_port(port) != NULL)
|
||||
{
|
||||
global_unlock();
|
||||
return INCPHUB_EBUSY;
|
||||
}
|
||||
|
||||
incphub_cli_t *client = find_empty_client();
|
||||
if (client == NULL)
|
||||
{
|
||||
global_unlock();
|
||||
return INCPHUB_EBUSY;
|
||||
}
|
||||
|
||||
// Populate client data
|
||||
client->port = port;
|
||||
client->task = task;
|
||||
|
||||
// Save to passed pointer
|
||||
*cli_ptr = client;
|
||||
|
||||
global_unlock();
|
||||
return INCPHUB_OK;
|
||||
}
|
||||
|
||||
int get_msg_buffer(uint8_t **buffer_ptr, size_t len)
|
||||
{
|
||||
if (buffer_ptr == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
|
||||
// == Enter ==
|
||||
if (!global_lock(portMAX_DELAY))
|
||||
return INCPHUB_EAGAIN;
|
||||
|
||||
//TODO: Allow buffer to be given by another layer (below NIL/NTL)
|
||||
uint8_t *buf = find_unused_buffer();
|
||||
if (buf == NULL)
|
||||
{
|
||||
global_unlock();
|
||||
return INCPHUB_ENOMEM;
|
||||
}
|
||||
|
||||
// Return buffer
|
||||
*buffer_ptr = buf;
|
||||
|
||||
global_unlock();
|
||||
return INCPHUB_OK;
|
||||
}
|
||||
|
||||
int release_msg_buffer(uint8_t *buffer)
|
||||
{
|
||||
if (buffer == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
|
||||
// == Enter ==
|
||||
if (!global_lock(portMAX_DELAY))
|
||||
return INCPHUB_EAGAIN;
|
||||
|
||||
int index = find_index_of_buffer(buffer);
|
||||
|
||||
// Handle invalid buffers
|
||||
if (index == -1)
|
||||
{
|
||||
global_unlock();
|
||||
return INCPHUB_EINVAL;
|
||||
}
|
||||
|
||||
// Handle not-in-use buffers
|
||||
if (_msg_buffer_used[index] == false)
|
||||
{
|
||||
global_unlock();
|
||||
return INCPHUB_EINVAL;
|
||||
}
|
||||
|
||||
// Mark buffer as free
|
||||
_msg_buffer_used[index] = false;
|
||||
|
||||
global_unlock();
|
||||
return INCPHUB_OK;
|
||||
}
|
||||
|
||||
int get_client_with_port(ntl_port_t port, incphub_cli_t **client_ptr)
|
||||
{
|
||||
if (port == NTL_PORT_EMPTY)
|
||||
return INCPHUB_EINVAL;
|
||||
|
||||
if (client_ptr == NULL)
|
||||
return INCPHUB_EINVAL;
|
||||
|
||||
// == Enter ==
|
||||
if (!global_lock(portMAX_DELAY))
|
||||
return INCPHUB_EAGAIN;
|
||||
|
||||
// Map port to client
|
||||
incphub_cli_t *client = find_client_with_port(port);
|
||||
if (client == NULL)
|
||||
{
|
||||
global_unlock();
|
||||
return INCPHUB_ENOTFOUND;
|
||||
}
|
||||
|
||||
// Return client
|
||||
*client_ptr = client;
|
||||
|
||||
global_unlock();
|
||||
return INCPHUB_OK;
|
||||
}
|
||||
|
||||
int enqueue_message_ingest(uint8_t *buffer, size_t length)
|
||||
{
|
||||
(void)buffer;
|
||||
(void)length;
|
||||
//TODO: Implement
|
||||
return INCPHUB_ERR;
|
||||
}
|
||||
|
||||
int dequeue_message_ingest(uint8_t **buffer_ptr, size_t *length_ptr)
|
||||
{
|
||||
(void)buffer_ptr;
|
||||
(void)length_ptr;
|
||||
//TODO: Implement
|
||||
return INCPHUB_ERR;
|
||||
}
|
||||
|
||||
|
||||
|
||||
// === Private function definitions ===
|
||||
static bool global_lock(TickType_t max_delay)
|
||||
{
|
||||
return xSemaphoreTake(_incphub_lock, max_delay) == pdTRUE;
|
||||
}
|
||||
|
||||
static void global_unlock()
|
||||
{
|
||||
xSemaphoreGive(_incphub_lock);
|
||||
}
|
||||
|
||||
static incphub_cli_t *find_empty_client()
|
||||
{
|
||||
return find_client_with_port(NTL_PORT_EMPTY);
|
||||
}
|
||||
|
||||
static incphub_cli_t *find_client_with_port(ntl_port_t port)
|
||||
{
|
||||
for (int i = 0; i < INCPHUB_MAX_CLIENTS; ++i)
|
||||
{
|
||||
incphub_cli_t *client = &_clients[i];
|
||||
|
||||
if (client->port == port)
|
||||
return client;
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static uint8_t *find_unused_buffer()
|
||||
{
|
||||
for (int i = 0; i < INCPHUB_MSG_BUFFER_COUNT; ++i)
|
||||
{
|
||||
if (!_msg_buffer_used[i])
|
||||
return _msg_buffers[i];
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static int find_index_of_buffer(uint8_t *buffer)
|
||||
{
|
||||
for (int i = 0; i < INCPHUB_MSG_BUFFER_COUNT; ++i)
|
||||
{
|
||||
if (buffer = _msg_buffers[i])
|
||||
return i;
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
@@ -1,76 +0,0 @@
|
||||
// incphub_sharedl.h - Internal mechanisms shared by incphub and incphub_client
|
||||
|
||||
#ifndef INCPHUB_SHARED_H_
|
||||
#define INCPHUB_SHARED_H_
|
||||
|
||||
#include "FreeRTOS.h"
|
||||
#include "task.h"
|
||||
#include "queue.h"
|
||||
|
||||
#include "nil/nil.h"
|
||||
#include "ntl/ntl.h"
|
||||
|
||||
#include "incphub/incphub.h"
|
||||
|
||||
|
||||
|
||||
struct incphub_cli_t {
|
||||
/// @remark If set to NTL_PORT_EMPTY denotes an empty client
|
||||
ntl_port_t port;
|
||||
TaskHandle_t task;
|
||||
};
|
||||
|
||||
// Queue<incphub_msg_t>
|
||||
extern QueueHandle_t incphub_ingest_queue;
|
||||
|
||||
#ifndef INCPHUB_CLIENT_H_
|
||||
typedef struct incphub_cli_t incphub_cli_t;
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
/// @brief Initializes the internal state of incphub
|
||||
/// @return INCPHUB_OK on success, an error otherwise
|
||||
int internal_init();
|
||||
|
||||
/// @brief Registers a task as an incphub client
|
||||
/// @param cli_ptr Location to return client
|
||||
/// @param port The port to register to
|
||||
/// @param task The task to register as the handler
|
||||
/// @return INCPHUB_OK on success, an error otherwise
|
||||
int register_client(incphub_cli_t **cli_ptr, ntl_port_t port, TaskHandle_t task);
|
||||
|
||||
/// @brief Attempts to get a free message buffer from the central storage space
|
||||
/// @param buffer_ptr Location to return the buffer
|
||||
/// @param len The minimum size, in bytes, of the buffer get
|
||||
/// @return INCPHUB_OK on success, an error otherwise
|
||||
int get_msg_buffer(uint8_t **buffer_ptr, size_t len);
|
||||
|
||||
/// @brief Releases a buffer previously acquired with get_msg_buffer
|
||||
/// @param buffer The buffer to release
|
||||
/// @return INCPHUB_OK on success, an error otherwise
|
||||
int release_msg_buffer(uint8_t *buffer);
|
||||
|
||||
/// @brief Attempts to get the client assigned to the given port
|
||||
/// @param port The port to look for
|
||||
/// @param client_ptr Location to return the client
|
||||
/// @return INCPHUB_OK on success, an error otherwise
|
||||
int get_client_with_port(ntl_port_t port, incphub_cli_t **client_ptr);
|
||||
|
||||
/// @brief Enqueues a message for ingest by the incphub task
|
||||
/// @param buffer The buffer to enqueue
|
||||
/// @param length The message length
|
||||
/// @return INCPHUB_OK on success, an error otherwise
|
||||
int enqueue_message_ingest(uint8_t *buffer, size_t length);
|
||||
|
||||
/// @brief Dequeues a message for ingest
|
||||
/// @param buffer_ptr Location to return the buffer
|
||||
/// @param length_ptr Location to store the message length
|
||||
/// @return INCPHUB_OK on success, an error otherwise
|
||||
int dequeue_message_ingest(uint8_t **buffer_ptr, size_t *length_ptr);
|
||||
|
||||
int do_message_dispatch(incphub_cli_t *cli, uint8_t *buffer, size_t length, incphub_addr_t source);
|
||||
|
||||
int await_message_dispatch(uint8_t **buffer_ptr, size_t *length_ptr, incphub_addr_t *source_ptr);
|
||||
|
||||
#endif
|
||||
@@ -1,159 +0,0 @@
|
||||
#include "incphub/incphub_task.h"
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
#include "FreeRTOS.h"
|
||||
#include "queue.h"
|
||||
|
||||
#include "incphub/incphub.h"
|
||||
#include "incphub_config.h"
|
||||
#include "incphub_shared.h"
|
||||
|
||||
#include "ntl/ntl.h"
|
||||
#include "nil/nil.h"
|
||||
|
||||
|
||||
|
||||
// === Static task space ===
|
||||
StackType_t _incphub_task_stack[INCPHUB_TASK_STACK_DEPTH];
|
||||
StaticTask_t _incphub_task;
|
||||
|
||||
|
||||
|
||||
// === Public variables ===
|
||||
|
||||
// Queue<nil_packet_t*> Packets awaiting to be accepted by incphub
|
||||
QueueHandle_t incphub_ingest_queue;
|
||||
|
||||
|
||||
|
||||
// === Private variables ===
|
||||
#define INGEST_QUEUE_LENGTH (16)
|
||||
#define INGEST_QUEUE_ITEM_SIZE (sizeof(ntl_packet_t *))
|
||||
static uint8_t _ingest_queue_store[INGEST_QUEUE_LENGTH * INGEST_QUEUE_ITEM_SIZE];
|
||||
static StaticQueue_t _ingest_queue_static;
|
||||
|
||||
|
||||
|
||||
// === Private functions ===
|
||||
static nil_packet_t *extract_nil_packet(uint8_t *buffer, size_t length);
|
||||
static ntl_packet_t *extract_ntl_packet(nil_packet_t *nil);
|
||||
static void handle_nil_packet(nil_packet_t *nil);
|
||||
static void handle_ntl_packet(ntl_packet_t *ntl, nil_addr_t src_addr);
|
||||
|
||||
|
||||
|
||||
// === Public functions ===
|
||||
void incphub_task_preinit()
|
||||
{
|
||||
internal_init();
|
||||
incphub_ingest_queue =
|
||||
xQueueCreateStatic(INGEST_QUEUE_LENGTH, INGEST_QUEUE_ITEM_SIZE, _ingest_queue_store, &_ingest_queue_static);
|
||||
}
|
||||
|
||||
void incphub_task_main(void *params)
|
||||
{
|
||||
(void)params;
|
||||
|
||||
while (1)
|
||||
{
|
||||
// Dequeue message
|
||||
uint8_t *buffer;
|
||||
size_t length;
|
||||
dequeue_message_ingest(&buffer, &length);
|
||||
|
||||
// Get NIL packet
|
||||
nil_packet_t *nil = extract_nil_packet(buffer, length);
|
||||
if (nil == NULL)
|
||||
{
|
||||
//TODO: Error
|
||||
continue;
|
||||
}
|
||||
|
||||
// Process or route
|
||||
handle_nil_packet(nil);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
// === Private functions ===
|
||||
static nil_packet_t *extract_nil_packet(uint8_t *buffer, size_t length)
|
||||
{
|
||||
// Ensure message is big enough to hold a packet
|
||||
if (length < sizeof(nil_packet_t))
|
||||
{
|
||||
//TODO: Error
|
||||
return NULL;
|
||||
}
|
||||
|
||||
nil_packet_t *nil = (nil_packet_t *)buffer;
|
||||
// Ensure length field is not going to cause buffer overflow
|
||||
if (length < sizeof(nil_packet_t) + nil->header.len )
|
||||
{
|
||||
//TODO: Error
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return nil;
|
||||
}
|
||||
|
||||
static ntl_packet_t *extract_ntl_packet(nil_packet_t *nil)
|
||||
{
|
||||
// Ensure message is big enough to hold a packet
|
||||
if (nil->header.len < sizeof(ntl_packet_t))
|
||||
{
|
||||
//TODO: Error
|
||||
return NULL;
|
||||
}
|
||||
|
||||
ntl_packet_t *ntl = (ntl_packet_t *)nil->payload;
|
||||
// Ensure length field is not going to cause buffer overflow
|
||||
if (nil->header.len < sizeof(ntl_packet_t) + ntl->header.len)
|
||||
{
|
||||
//TODO: Error
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return ntl;
|
||||
}
|
||||
|
||||
static void handle_nil_packet(nil_packet_t *nil)
|
||||
{
|
||||
if (nil->header.dst_addr == INCPHUB_LOCAL_NIL_ADDR)
|
||||
{
|
||||
ntl_packet_t *ntl = extract_ntl_packet(nil);
|
||||
if (ntl == NULL)
|
||||
{
|
||||
//TODO: Error
|
||||
return;
|
||||
}
|
||||
|
||||
handle_ntl_packet(ntl, nil->header.src_addr);
|
||||
return;
|
||||
}
|
||||
|
||||
//TODO: Route to external interface
|
||||
return;
|
||||
}
|
||||
|
||||
static void handle_ntl_packet(ntl_packet_t *ntl, nil_addr_t src_addr)
|
||||
{
|
||||
ntl_port_t port = ntl_get_dst(&ntl->header);
|
||||
|
||||
incphub_cli_t *client;
|
||||
if (get_client_with_port(port, &client) != INCPHUB_OK)
|
||||
{
|
||||
//TODO: Error
|
||||
return;
|
||||
}
|
||||
|
||||
incphub_addr_t sender = { .addr = src_addr, .port = ntl->header.src_port };
|
||||
if (do_message_dispatch(client, ntl->payload, ntl->header.len, sender) != INCPHUB_OK)
|
||||
{
|
||||
//TODO: Error
|
||||
return;
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
@@ -1,7 +1,7 @@
|
||||
target_sources(${EXECUTABLE} PRIVATE
|
||||
src/incphub_client.c
|
||||
src/incphub_task.c
|
||||
src/incphub_shared.c
|
||||
src/shunt_client.c
|
||||
src/shunt_task.c
|
||||
src/shunt_shared.c
|
||||
)
|
||||
|
||||
target_include_directories(${EXECUTABLE} PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/include)
|
||||
37
app/src/libshunt/README.md
Normal file
37
app/src/libshunt/README.md
Normal file
@@ -0,0 +1,37 @@
|
||||
# LibShunt
|
||||
|
||||
Shunts messages, like trains
|
||||
|
||||
## FreeRTOS requirements
|
||||
|
||||
- INCLUDE_xTaskGetCurrentTaskHandle set to 1
|
||||
- INCLUDE_vTaskSuspend set to 1
|
||||
- configUSE_TASK_NOTIFICATIONS set to 1
|
||||
- configTASK_NOTIFICATION_ARRAY_ENTRIES set to >= 2 (0 is for stream/message buffers, 1(configurable) is used by libshunt)
|
||||
- configUSE_MUTEXES set to 1
|
||||
|
||||
## Configuration
|
||||
|
||||
Set as a macro in compile flags if wishing to override the defaults
|
||||
|
||||
- LIBSHUNT_TASKINDEX_MAILBOX, defaults to 1 (avoid using 0)
|
||||
|
||||
## Setup
|
||||
|
||||
- libshunt_task_preinit MUST be run before any calls to the client, likely even before the scheduler is started
|
||||
|
||||
## IPC flow
|
||||
|
||||
Ingest (accepting of messages into libshunt) is done via queue (many to one)
|
||||
|
||||
Dispatch (internal routing to local task) is done via direct-to-task notifications (see LIBSHUNT_TASKINDEX_MAILBOX) by passing the message buffer pointer
|
||||
|
||||
## Buffer ownership
|
||||
|
||||
Buffers are allocated with shunt_get_msg_buffer, at which point they belong to the requesting task.
|
||||
|
||||
When passed to libshunt_send_XXX functions, they become owned by libshunt
|
||||
|
||||
When dispatched internally, the receiving task becomes the owner and must invoke shunt_release_msg_buffer when done
|
||||
|
||||
When routed externally, after calling the transmitter function, libshunt releases the buffer
|
||||
31
app/src/libshunt/include/libshunt/libshunt.h
Normal file
31
app/src/libshunt/include/libshunt/libshunt.h
Normal file
@@ -0,0 +1,31 @@
|
||||
// libshunt.h - Public definitions common to all libshunt components
|
||||
|
||||
#ifndef LIBSHUNT_H_
|
||||
#define LIBSHUNT_H_
|
||||
|
||||
#include "ntl/ntl.h"
|
||||
#include "nil/nil.h"
|
||||
|
||||
/// @brief Success
|
||||
#define LIBSHUNT_OK (0)
|
||||
/// @brief Unspecified error
|
||||
#define LIBSHUNT_ERR (1)
|
||||
/// @brief Invalid argument
|
||||
#define LIBSHUNT_EINVAL (2)
|
||||
/// @brief Resource or entry not found
|
||||
#define LIBSHUNT_ENOTFOUND (3)
|
||||
/// @brief Resource in use
|
||||
#define LIBSHUNT_EBUSY (4)
|
||||
/// @brief Resource temporarily unavailable
|
||||
#define LIBSHUNT_EAGAIN (5)
|
||||
/// @brief Could not allocate memory
|
||||
#define LIBSHUNT_ENOMEM (6)
|
||||
/// @brief Packet failed integrity checks
|
||||
#define LIBSHUNT_BAD_PACKET (7)
|
||||
|
||||
typedef struct {
|
||||
nil_addr_t addr;
|
||||
ntl_port_t port;
|
||||
} shunt_addr_t;
|
||||
|
||||
#endif
|
||||
57
app/src/libshunt/include/libshunt/shunt_client.h
Normal file
57
app/src/libshunt/include/libshunt/shunt_client.h
Normal file
@@ -0,0 +1,57 @@
|
||||
// shunt_client.h - Public interface of libshunt client side
|
||||
|
||||
#ifndef LIBSHUNT_CLIENT_H_
|
||||
#define LIBSHUNT_CLIENT_H_
|
||||
|
||||
#include <stddef.h>
|
||||
|
||||
#include "FreeRTOS.h"
|
||||
|
||||
#include "libshunt/libshunt.h"
|
||||
|
||||
#include "nil/nil.h"
|
||||
#include "ntl/ntl.h"
|
||||
|
||||
|
||||
|
||||
typedef struct shunt_cli_t shunt_cli_t;
|
||||
|
||||
|
||||
|
||||
/// @brief Initializes an libshunt client
|
||||
/// @param port The port number to assign to the caller
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
int shunt_init_client(shunt_cli_t **cli, ntl_port_t port);
|
||||
|
||||
/// @brief Attempts to get a free message buffer from the central storage space
|
||||
/// @param cli Current client handle
|
||||
/// @param buffer_ptr Location to return the buffer
|
||||
/// @param length The minimum size, in bytes, of the buffer get
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
int shunt_get_msg_buffer(shunt_cli_t *cli, uint8_t **buffer_ptr, size_t length);
|
||||
|
||||
/// @brief Releases a buffer received from shunt_await_msg
|
||||
/// @param cli Current client handle
|
||||
/// @param buffer The buffer to release
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
int shunt_release_msg_buffer(shunt_cli_t *cli, uint8_t *buffer);
|
||||
|
||||
/// @brief Attempts to send a message
|
||||
/// @param cli Current client handle
|
||||
/// @param buffer The message buffer (gets consumed)
|
||||
/// @param length The message length
|
||||
/// @param dst_addr The destination address
|
||||
/// @param max_delay The maximum time to wait for a resources, in ticks
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
int shunt_send_msg(shunt_cli_t *cli, uint8_t *buffer, size_t length, shunt_addr_t dst, TickType_t max_delay);
|
||||
|
||||
/// @brief Awaits the receival of a message
|
||||
/// @param cli Current client handle
|
||||
/// @param buffer_ptr Location to return the buffer
|
||||
/// @param length_ptr Location to return the length
|
||||
/// @param source_ptr Location to return the message sender
|
||||
/// @param max_delay The maximum time to wait for a message, in ticks
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
int shunt_await_msg(shunt_cli_t *cli, uint8_t **buffer_ptr, size_t *length_ptr, shunt_addr_t *source_ptr, TickType_t max_delay);
|
||||
|
||||
#endif
|
||||
19
app/src/libshunt/include/libshunt/shunt_task.h
Normal file
19
app/src/libshunt/include/libshunt/shunt_task.h
Normal file
@@ -0,0 +1,19 @@
|
||||
// shunt_task.h - Public interface to launch the libshunt FreeRTOS task
|
||||
|
||||
#ifndef LIBSHUNT_TASK_H_
|
||||
#define LIBSHUNT_TASK_H_
|
||||
|
||||
#include "FreeRTOS.h"
|
||||
#include "task.h"
|
||||
#include "FreeRTOSConfig.h"
|
||||
|
||||
#define LIBSHUNT_TASK_STACK_DEPTH configMINIMAL_STACK_SIZE
|
||||
#define LIBSHUNT_TASK_PRIORITY (configMAX_PRIORITIES - 1)
|
||||
|
||||
extern StackType_t _libshunt_task_stack[LIBSHUNT_TASK_STACK_DEPTH];
|
||||
extern StaticTask_t _libshunt_task;
|
||||
|
||||
void libshunt_task_preinit();
|
||||
void libshunt_task_main(void *params);
|
||||
|
||||
#endif
|
||||
129
app/src/libshunt/src/shunt_client.c
Normal file
129
app/src/libshunt/src/shunt_client.c
Normal file
@@ -0,0 +1,129 @@
|
||||
#include <stdbool.h>
|
||||
|
||||
#include "libshunt/shunt_client.h"
|
||||
#include "libshunt/shunt_task.h"
|
||||
#include "libshunt/libshunt.h"
|
||||
#include "shunt_config.h"
|
||||
#include "shunt_shared.h"
|
||||
|
||||
#include "ntl/ntl.h"
|
||||
#include "nil/nil.h"
|
||||
|
||||
#include "message_buffer.h"
|
||||
|
||||
|
||||
|
||||
int shunt_init_client(shunt_cli_t **cli, ntl_port_t port)
|
||||
{
|
||||
return register_client(cli, port, xTaskGetCurrentTaskHandle());
|
||||
}
|
||||
|
||||
int shunt_get_msg_buffer(shunt_cli_t *cli, uint8_t **buffer_ptr, size_t length)
|
||||
{
|
||||
if (cli == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
if (buffer_ptr == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
|
||||
// Request extra space for NIL/NTL headers
|
||||
size_t extended_len = sizeof(nil_header_t) + sizeof(ntl_header_t) + length;
|
||||
uint8_t *buf;
|
||||
|
||||
int status = get_msg_buffer(&buf, extended_len);
|
||||
if (status != LIBSHUNT_OK)
|
||||
return status;
|
||||
|
||||
// Reserve space for NIL/NTL headers
|
||||
buf += sizeof(nil_header_t) + sizeof(ntl_header_t);
|
||||
|
||||
// Return buffer
|
||||
*buffer_ptr = buf;
|
||||
|
||||
return LIBSHUNT_OK;
|
||||
}
|
||||
|
||||
int shunt_release_msg_buffer(shunt_cli_t *cli, uint8_t *buffer)
|
||||
{
|
||||
if (cli == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
if (buffer == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
|
||||
// cli requested for interface consistency only (for now)
|
||||
(void)cli;
|
||||
|
||||
// Unwind buffer pointer
|
||||
uint8_t *unwound = buffer - sizeof(nil_header_t) + sizeof(ntl_header_t);
|
||||
|
||||
// Release correct buffer pointer
|
||||
return release_msg_buffer(unwound);
|
||||
}
|
||||
|
||||
int shunt_send_msg(shunt_cli_t *cli, uint8_t *buffer, size_t length, shunt_addr_t dst, TickType_t max_delay)
|
||||
{
|
||||
if (cli == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
if (buffer == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
if (dst.port == NTL_PORT_EMPTY)
|
||||
return LIBSHUNT_EINVAL;
|
||||
|
||||
// Unwind buffer pointer
|
||||
ntl_packet_t *ntl_packet = (ntl_packet_t*)((uint8_t*)buffer - sizeof(ntl_header_t));
|
||||
nil_packet_t *nil_packet = (nil_packet_t*)((uint8_t*)ntl_packet - sizeof(ntl_header_t));
|
||||
|
||||
// Populate NTL header
|
||||
ntl_build_header(&ntl_packet->header, dst.port, cli->port, length);
|
||||
|
||||
// Populate NIL header
|
||||
nil_build_header(&nil_packet->header, dst.addr, LIBSHUNT_LOCAL_NIL_ADDR, length + sizeof(ntl_header_t));
|
||||
|
||||
// Pass to ingest
|
||||
return enqueue_message_ingest((uint8_t *)nil_packet, max_delay);
|
||||
}
|
||||
|
||||
int shunt_await_msg(shunt_cli_t *cli, uint8_t **buffer_ptr, size_t *length_ptr, shunt_addr_t *source_ptr, TickType_t max_delay)
|
||||
{
|
||||
if (cli == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
if (buffer_ptr == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
if (length_ptr == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
if (source_ptr == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
|
||||
// cli requested for interface consistency only (for now)
|
||||
(void)cli;
|
||||
|
||||
uint8_t *buffer;
|
||||
int err = await_message_dispatch(&buffer, max_delay);
|
||||
if (err != LIBSHUNT_OK)
|
||||
{
|
||||
return err;
|
||||
}
|
||||
|
||||
// Extract header information
|
||||
nil_packet_t *nil = extract_nil_packet(buffer, LIBSHUNT_MSG_BUFFER_SIZE);
|
||||
if (nil == NULL)
|
||||
{
|
||||
return LIBSHUNT_BAD_PACKET;
|
||||
}
|
||||
ntl_packet_t *ntl = extract_ntl_packet(nil);
|
||||
if (nil == NULL)
|
||||
{
|
||||
return LIBSHUNT_BAD_PACKET;
|
||||
}
|
||||
|
||||
// Build source spec
|
||||
shunt_addr_t source = { 0 };
|
||||
source.addr = nil->header.src_addr;
|
||||
source.port = ntl->header.src_port;
|
||||
|
||||
// Return values
|
||||
*buffer_ptr = ntl->payload;
|
||||
*length_ptr = ntl->header.len;
|
||||
*source_ptr = source;
|
||||
|
||||
return LIBSHUNT_OK;
|
||||
}
|
||||
30
app/src/libshunt/src/shunt_config.h
Normal file
30
app/src/libshunt/src/shunt_config.h
Normal file
@@ -0,0 +1,30 @@
|
||||
// shunt_config.h - Default configurations for libshunt
|
||||
|
||||
#ifndef LIBSHUNT_CONFIG_H_
|
||||
#define LIBSHUNT_CONFIG_H_
|
||||
|
||||
#ifndef LIBSHUNT_TASKINDEX_MAILBOX
|
||||
/// @brief Index of direct-to-task notification for mailbox to use
|
||||
#define LIBSHUNT_TASKINDEX_MAILBOX 1
|
||||
#endif
|
||||
|
||||
#ifndef LIBSHUNT_MAX_CLIENTS
|
||||
/// @brief Number of client slots to allocate
|
||||
#define LIBSHUNT_MAX_CLIENTS 8
|
||||
#endif
|
||||
|
||||
#ifndef LIBSHUNT_MSG_BUFFER_SIZE
|
||||
/// @brief Message buffer size in bytes. Must allow space for NIL and NTL headers
|
||||
#define LIBSHUNT_MSG_BUFFER_SIZE 32
|
||||
#endif
|
||||
|
||||
#ifndef LIBSHUNT_MSG_BUFFER_COUNT
|
||||
/// @brief Number of message buffers to allocate
|
||||
#define LIBSHUNT_MSG_BUFFER_COUNT 8
|
||||
#endif
|
||||
|
||||
#ifndef LIBSHUNT_LOCAL_NIL_ADDR
|
||||
#error "Local NIL address must be specified with LIBSHUNT_LOCAL_NIL_ADDR"
|
||||
#endif
|
||||
|
||||
#endif
|
||||
311
app/src/libshunt/src/shunt_shared.c
Normal file
311
app/src/libshunt/src/shunt_shared.c
Normal file
@@ -0,0 +1,311 @@
|
||||
#include "shunt_shared.h"
|
||||
|
||||
#include <string.h>
|
||||
#include <stdbool.h>
|
||||
#include <stdint.h>
|
||||
|
||||
#include "FreeRTOS.h"
|
||||
#include "semphr.h"
|
||||
#include "task.h"
|
||||
|
||||
#include "libshunt/libshunt.h"
|
||||
#include "shunt_config.h"
|
||||
|
||||
#include "nil/nil.h"
|
||||
|
||||
|
||||
|
||||
#if UINTPTR_MAX > UINT32_MAX
|
||||
#error "Pointers of 32-bits or less are required for libshunt mailbox to function correctly."
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
// === Private variables ===
|
||||
|
||||
// Global lock
|
||||
static SemaphoreHandle_t _libshunt_lock;
|
||||
static StaticSemaphore_t _libshunt_lock_store;
|
||||
|
||||
// Clients
|
||||
static shunt_cli_t _clients[LIBSHUNT_MAX_CLIENTS];
|
||||
|
||||
// Message buffers
|
||||
// LIBSHUNT_MSG_BUFFER_COUNT buffers of LIBSHUNT_MSG_BUFFER_SIZE bytes
|
||||
static uint8_t _msg_buffers[LIBSHUNT_MSG_BUFFER_COUNT][LIBSHUNT_MSG_BUFFER_SIZE];
|
||||
static bool _msg_buffer_used[LIBSHUNT_MSG_BUFFER_COUNT];
|
||||
|
||||
|
||||
|
||||
// === Private function declarations ===
|
||||
static bool global_lock(TickType_t max_delay);
|
||||
static void global_unlock();
|
||||
/// @remark Returns NULL if not found
|
||||
static shunt_cli_t *find_empty_client();
|
||||
/// @remark Returns NULL if not found
|
||||
static shunt_cli_t *find_client_with_port(ntl_port_t port);
|
||||
/// @remark Returns NULL if not found
|
||||
static uint8_t *find_unused_buffer();
|
||||
/// @remark Returns -1 if not found
|
||||
static int find_index_of_buffer(uint8_t *buffer);
|
||||
|
||||
|
||||
|
||||
// === Public functions ===
|
||||
int internal_init()
|
||||
{
|
||||
_libshunt_lock = xSemaphoreCreateMutexStatic(&_libshunt_lock_store);
|
||||
memset(_clients, 0, sizeof(_clients));
|
||||
memset(_msg_buffer_used, 0, sizeof(_msg_buffer_used));
|
||||
return LIBSHUNT_OK;
|
||||
}
|
||||
|
||||
int register_client(shunt_cli_t **cli_ptr, ntl_port_t port, TaskHandle_t task)
|
||||
{
|
||||
if (port == NTL_PORT_EMPTY)
|
||||
return LIBSHUNT_EINVAL;
|
||||
if (cli_ptr == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
|
||||
// == Enter ==
|
||||
if (!global_lock(portMAX_DELAY))
|
||||
return LIBSHUNT_EAGAIN;
|
||||
|
||||
// Prevent double binding of ports
|
||||
if (find_client_with_port(port) != NULL)
|
||||
{
|
||||
global_unlock();
|
||||
return LIBSHUNT_EBUSY;
|
||||
}
|
||||
|
||||
shunt_cli_t *client = find_empty_client();
|
||||
if (client == NULL)
|
||||
{
|
||||
global_unlock();
|
||||
return LIBSHUNT_EBUSY;
|
||||
}
|
||||
|
||||
// Populate client data
|
||||
client->port = port;
|
||||
client->task = task;
|
||||
|
||||
// Save to passed pointer
|
||||
*cli_ptr = client;
|
||||
|
||||
global_unlock();
|
||||
return LIBSHUNT_OK;
|
||||
}
|
||||
|
||||
int get_msg_buffer(uint8_t **buffer_ptr, size_t len)
|
||||
{
|
||||
if (buffer_ptr == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
if (len > LIBSHUNT_MSG_BUFFER_SIZE)
|
||||
return LIBSHUNT_ENOMEM;
|
||||
|
||||
// == Enter ==
|
||||
if (!global_lock(portMAX_DELAY))
|
||||
return LIBSHUNT_EAGAIN;
|
||||
|
||||
//TODO: Allow buffer to be given by another layer (below NIL/NTL)
|
||||
uint8_t *buf = find_unused_buffer();
|
||||
if (buf == NULL)
|
||||
{
|
||||
global_unlock();
|
||||
return LIBSHUNT_ENOMEM;
|
||||
}
|
||||
|
||||
// Return buffer
|
||||
*buffer_ptr = buf;
|
||||
|
||||
global_unlock();
|
||||
return LIBSHUNT_OK;
|
||||
}
|
||||
|
||||
int release_msg_buffer(uint8_t *buffer)
|
||||
{
|
||||
if (buffer == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
|
||||
// == Enter ==
|
||||
if (!global_lock(portMAX_DELAY))
|
||||
return LIBSHUNT_EAGAIN;
|
||||
|
||||
int index = find_index_of_buffer(buffer);
|
||||
|
||||
// Handle invalid buffers
|
||||
if (index == -1)
|
||||
{
|
||||
global_unlock();
|
||||
return LIBSHUNT_EINVAL;
|
||||
}
|
||||
|
||||
// Handle not-in-use buffers
|
||||
if (_msg_buffer_used[index] == false)
|
||||
{
|
||||
global_unlock();
|
||||
return LIBSHUNT_EINVAL;
|
||||
}
|
||||
|
||||
// Mark buffer as free
|
||||
_msg_buffer_used[index] = false;
|
||||
|
||||
global_unlock();
|
||||
return LIBSHUNT_OK;
|
||||
}
|
||||
|
||||
int get_client_with_port(ntl_port_t port, shunt_cli_t **client_ptr)
|
||||
{
|
||||
if (port == NTL_PORT_EMPTY)
|
||||
return LIBSHUNT_EINVAL;
|
||||
if (client_ptr == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
|
||||
// == Enter ==
|
||||
if (!global_lock(portMAX_DELAY))
|
||||
return LIBSHUNT_EAGAIN;
|
||||
|
||||
// Map port to client
|
||||
shunt_cli_t *client = find_client_with_port(port);
|
||||
if (client == NULL)
|
||||
{
|
||||
global_unlock();
|
||||
return LIBSHUNT_ENOTFOUND;
|
||||
}
|
||||
|
||||
// Return client
|
||||
*client_ptr = client;
|
||||
|
||||
global_unlock();
|
||||
return LIBSHUNT_OK;
|
||||
}
|
||||
|
||||
int enqueue_message_ingest(uint8_t *buffer, TickType_t max_delay)
|
||||
{
|
||||
//NOTE: Does not need global lock
|
||||
|
||||
if (buffer == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
|
||||
// Try to queue message
|
||||
if (xQueueSendToBack(libshunt_ingest_queue, &buffer, max_delay) != pdPASS)
|
||||
{
|
||||
return LIBSHUNT_EAGAIN;
|
||||
}
|
||||
|
||||
return LIBSHUNT_OK;
|
||||
}
|
||||
|
||||
int dequeue_message_ingest(uint8_t **buffer_ptr, TickType_t max_delay)
|
||||
{
|
||||
//NOTE: Does not need global lock
|
||||
|
||||
if (buffer_ptr == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
|
||||
uint8_t *buffer;
|
||||
|
||||
// Try to dequeue message
|
||||
if (xQueueReceive(libshunt_ingest_queue, &buffer, max_delay) == pdFALSE)
|
||||
{
|
||||
return LIBSHUNT_EAGAIN;
|
||||
}
|
||||
|
||||
// Return buffer
|
||||
*buffer_ptr = buffer;
|
||||
|
||||
return LIBSHUNT_OK;
|
||||
}
|
||||
|
||||
int do_message_dispatch(shunt_cli_t *cli, uint8_t *buffer)
|
||||
{
|
||||
//NOTE: Does not need global lock
|
||||
|
||||
if (cli == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
if (buffer == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
|
||||
// Try to notify
|
||||
if (xTaskNotifyIndexed(cli->task, LIBSHUNT_TASKINDEX_MAILBOX, (uint32_t)buffer, eSetValueWithoutOverwrite) == pdFALSE)
|
||||
{
|
||||
// Failed because previous message has not yet been processed
|
||||
return LIBSHUNT_EAGAIN;
|
||||
}
|
||||
|
||||
return LIBSHUNT_OK;
|
||||
}
|
||||
|
||||
int await_message_dispatch(uint8_t **buffer_ptr, TickType_t max_delay)
|
||||
{
|
||||
//NOTE: Does not need global lock
|
||||
|
||||
if (buffer_ptr == NULL)
|
||||
return LIBSHUNT_EINVAL;
|
||||
|
||||
// Get raw msg buffer
|
||||
uint8_t *buffer;
|
||||
if (xTaskNotifyWaitIndexed(LIBSHUNT_TASKINDEX_MAILBOX, 0, 0, (uint32_t*)&buffer, max_delay) == pdFALSE)
|
||||
{
|
||||
// Nothing pending
|
||||
return LIBSHUNT_EAGAIN;
|
||||
}
|
||||
|
||||
// Return buffer
|
||||
*buffer_ptr = buffer;
|
||||
|
||||
return LIBSHUNT_OK;
|
||||
}
|
||||
|
||||
|
||||
|
||||
// === Private function definitions ===
|
||||
static bool global_lock(TickType_t max_delay)
|
||||
{
|
||||
return xSemaphoreTake(_libshunt_lock, max_delay) == pdTRUE;
|
||||
}
|
||||
|
||||
static void global_unlock()
|
||||
{
|
||||
xSemaphoreGive(_libshunt_lock);
|
||||
}
|
||||
|
||||
static shunt_cli_t *find_empty_client()
|
||||
{
|
||||
return find_client_with_port(NTL_PORT_EMPTY);
|
||||
}
|
||||
|
||||
static shunt_cli_t *find_client_with_port(ntl_port_t port)
|
||||
{
|
||||
for (int i = 0; i < LIBSHUNT_MAX_CLIENTS; ++i)
|
||||
{
|
||||
shunt_cli_t *client = &_clients[i];
|
||||
|
||||
if (client->port == port)
|
||||
return client;
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static uint8_t *find_unused_buffer()
|
||||
{
|
||||
for (int i = 0; i < LIBSHUNT_MSG_BUFFER_COUNT; ++i)
|
||||
{
|
||||
if (!_msg_buffer_used[i])
|
||||
return _msg_buffers[i];
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static int find_index_of_buffer(uint8_t *buffer)
|
||||
{
|
||||
for (int i = 0; i < LIBSHUNT_MSG_BUFFER_COUNT; ++i)
|
||||
{
|
||||
if (buffer == _msg_buffers[i])
|
||||
return i;
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
129
app/src/libshunt/src/shunt_shared.h
Normal file
129
app/src/libshunt/src/shunt_shared.h
Normal file
@@ -0,0 +1,129 @@
|
||||
// shunt_sharedl.h - Internal mechanisms shared by libshunt components
|
||||
|
||||
#ifndef LIBSHUNT_SHARED_H_
|
||||
#define LIBSHUNT_SHARED_H_
|
||||
|
||||
#include "FreeRTOS.h"
|
||||
#include "task.h"
|
||||
#include "queue.h"
|
||||
|
||||
#include "nil/nil.h"
|
||||
#include "ntl/ntl.h"
|
||||
|
||||
#include "libshunt/libshunt.h"
|
||||
|
||||
|
||||
|
||||
struct shunt_cli_t {
|
||||
/// @remark If set to NTL_PORT_EMPTY denotes an empty client
|
||||
ntl_port_t port;
|
||||
TaskHandle_t task;
|
||||
};
|
||||
|
||||
// Queue<uint8_t*> (queue of buffers)
|
||||
extern QueueHandle_t libshunt_ingest_queue;
|
||||
|
||||
#ifndef LIBSHUNT_CLIENT_H_
|
||||
typedef struct shunt_cli_t shunt_cli_t;
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
/// @brief Initializes the internal state of libshunt
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
int internal_init();
|
||||
|
||||
/// @brief Registers a task as an libshunt client
|
||||
/// @param cli_ptr Location to return client
|
||||
/// @param port The port to register to
|
||||
/// @param task The task to register as the handler
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
int register_client(shunt_cli_t **cli_ptr, ntl_port_t port, TaskHandle_t task);
|
||||
|
||||
/// @brief Attempts to get a free message buffer from the central storage space
|
||||
/// @param buffer_ptr Location to return the buffer
|
||||
/// @param len The minimum size, in bytes, of the buffer get
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
int get_msg_buffer(uint8_t **buffer_ptr, size_t len);
|
||||
|
||||
/// @brief Releases a buffer previously acquired with get_msg_buffer
|
||||
/// @param buffer The buffer to release
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
int release_msg_buffer(uint8_t *buffer);
|
||||
|
||||
/// @brief Attempts to get the client assigned to the given port
|
||||
/// @param port The port to look for
|
||||
/// @param client_ptr Location to return the client
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
int get_client_with_port(ntl_port_t port, shunt_cli_t **client_ptr);
|
||||
|
||||
/// @brief Enqueues a message for ingest by the libshunt task
|
||||
/// @param buffer The buffer to enqueue
|
||||
/// @param max_delay The maximum time to wait for resources, in ticks
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
int enqueue_message_ingest(uint8_t *buffer, TickType_t max_delay);
|
||||
|
||||
/// @brief Dequeues a message for ingest
|
||||
/// @param buffer_ptr Location to return the buffer
|
||||
/// @param max_delay The maximum time to wait for a message, in ticks
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
int dequeue_message_ingest(uint8_t **buffer_ptr, TickType_t max_delay);
|
||||
|
||||
/// @brief Sends a message for dispatch to the corresponding client
|
||||
/// @param cli The client to send the message to
|
||||
/// @param buffer The message buffer to send
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
int do_message_dispatch(shunt_cli_t *cli, uint8_t *buffer);
|
||||
|
||||
/// @brief Blocks until a message is received for dispatch, or until a given timeout
|
||||
/// @param buffer_ptr Location to return the message buffer
|
||||
/// @param max_delay The maximum time to wait for a message, in ticks
|
||||
/// @return LIBSHUNT_OK on success, an error otherwise
|
||||
/// @remark Buffers obatined from this function must be release with release_msg_buffer once no longer in use
|
||||
int await_message_dispatch(uint8_t **buffer_ptr, TickType_t max_delay);
|
||||
|
||||
/// @brief Extracts a NIL packet from a buffer
|
||||
/// @param buffer The byte buffer to read from
|
||||
/// @param length The maximum length to read from
|
||||
/// @return Pointer to the NIL packet in the buffer or NULL if the packet is invalid
|
||||
static inline nil_packet_t *extract_nil_packet(uint8_t *buffer, size_t length)
|
||||
{
|
||||
// Ensure message is big enough to hold a packet
|
||||
if (length < sizeof(nil_packet_t))
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
nil_packet_t *nil = (nil_packet_t *)buffer;
|
||||
// Ensure length field is not going to cause buffer overflow
|
||||
if (length < sizeof(nil_packet_t) + nil->header.len )
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return nil;
|
||||
}
|
||||
|
||||
/// @brief Extracts an NTL packet from a buffer
|
||||
/// @param buffer The byte buffer to read from
|
||||
/// @param length The maximum length to read from
|
||||
/// @return Pointer to the NTL packet in the buffer or NULL if the packet is invalid
|
||||
static inline ntl_packet_t *extract_ntl_packet(nil_packet_t *nil)
|
||||
{
|
||||
// Ensure message is big enough to hold a packet
|
||||
if (nil->header.len < sizeof(ntl_packet_t))
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
ntl_packet_t *ntl = (ntl_packet_t *)nil->payload;
|
||||
// Ensure length field is not going to cause buffer overflow
|
||||
if (nil->header.len < sizeof(ntl_packet_t) + ntl->header.len)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return ntl;
|
||||
}
|
||||
|
||||
#endif
|
||||
129
app/src/libshunt/src/shunt_task.c
Normal file
129
app/src/libshunt/src/shunt_task.c
Normal file
@@ -0,0 +1,129 @@
|
||||
#include "libshunt/shunt_task.h"
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
#include "FreeRTOS.h"
|
||||
#include "queue.h"
|
||||
|
||||
#include "libshunt/libshunt.h"
|
||||
#include "shunt_config.h"
|
||||
#include "shunt_shared.h"
|
||||
|
||||
#include "ntl/ntl.h"
|
||||
#include "nil/nil.h"
|
||||
|
||||
|
||||
|
||||
// === Static task space ===
|
||||
StackType_t _libshunt_task_stack[LIBSHUNT_TASK_STACK_DEPTH];
|
||||
StaticTask_t _libshunt_task;
|
||||
|
||||
|
||||
|
||||
// === Public variables ===
|
||||
|
||||
// Queue<nil_packet_t*> Packets awaiting to be accepted by libshunt
|
||||
QueueHandle_t libshunt_ingest_queue;
|
||||
|
||||
|
||||
|
||||
// === Private variables ===
|
||||
#define INGEST_QUEUE_LENGTH (16)
|
||||
#define INGEST_QUEUE_ITEM_SIZE (sizeof(uint8_t*))
|
||||
static uint8_t _ingest_queue_store[INGEST_QUEUE_LENGTH * INGEST_QUEUE_ITEM_SIZE];
|
||||
static StaticQueue_t _ingest_queue_static;
|
||||
|
||||
|
||||
|
||||
// === Private functions ===
|
||||
static int handle_nil_packet(nil_packet_t *nil);
|
||||
static int handle_ntl_packet(nil_packet_t *nil, ntl_packet_t *ntl);
|
||||
|
||||
|
||||
|
||||
// === Public functions ===
|
||||
void libshunt_task_preinit()
|
||||
{
|
||||
internal_init();
|
||||
libshunt_ingest_queue =
|
||||
xQueueCreateStatic(INGEST_QUEUE_LENGTH, INGEST_QUEUE_ITEM_SIZE, _ingest_queue_store, &_ingest_queue_static);
|
||||
}
|
||||
|
||||
void libshunt_task_main(void *params)
|
||||
{
|
||||
(void)params;
|
||||
int err;
|
||||
|
||||
while (1)
|
||||
{
|
||||
// Dequeue message
|
||||
uint8_t *buffer;
|
||||
err = dequeue_message_ingest(&buffer, portMAX_DELAY);
|
||||
|
||||
// Get NIL packet
|
||||
nil_packet_t *nil = extract_nil_packet(buffer, LIBSHUNT_MSG_BUFFER_SIZE);
|
||||
if (nil == NULL)
|
||||
{
|
||||
//TODO: Error
|
||||
continue;
|
||||
}
|
||||
|
||||
// Process or route
|
||||
err = handle_nil_packet(nil);
|
||||
if (err == LIBSHUNT_EAGAIN)
|
||||
{
|
||||
// Failed but only temporarily
|
||||
// Could be target task not ready
|
||||
// Attempt requeue to allow retries
|
||||
enqueue_message_ingest(buffer, 0);
|
||||
|
||||
// If failed to requeue (no space), discard message
|
||||
// This avoids clogging the system due to a single unresponsive task
|
||||
continue;
|
||||
}
|
||||
else if (err != LIBSHUNT_OK)
|
||||
{
|
||||
// Failed for some other reason
|
||||
// Do not retry
|
||||
continue;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
// === Private functions ===
|
||||
static int handle_nil_packet(nil_packet_t *nil)
|
||||
{
|
||||
if (nil->header.dst_addr == LIBSHUNT_LOCAL_NIL_ADDR
|
||||
|| nil->header.dst_addr == NIL_ADDR_LOCALHOST)
|
||||
{
|
||||
ntl_packet_t *ntl = extract_ntl_packet(nil);
|
||||
if (ntl == NULL)
|
||||
{
|
||||
//TODO: Error
|
||||
return LIBSHUNT_BAD_PACKET;
|
||||
}
|
||||
|
||||
return handle_ntl_packet(nil, ntl);
|
||||
}
|
||||
|
||||
//TODO: Route to external interface
|
||||
return LIBSHUNT_ERR;
|
||||
}
|
||||
|
||||
static int handle_ntl_packet(nil_packet_t *nil, ntl_packet_t *ntl)
|
||||
{
|
||||
int err;
|
||||
ntl_port_t port = ntl_get_dst(&ntl->header);
|
||||
|
||||
shunt_cli_t *client;
|
||||
err = get_client_with_port(port, &client);
|
||||
if (err != LIBSHUNT_OK)
|
||||
{
|
||||
//TODO: Error
|
||||
return err;
|
||||
}
|
||||
|
||||
return do_message_dispatch(client, (uint8_t*)nil);
|
||||
}
|
||||
@@ -4,16 +4,20 @@
|
||||
#include "task.h"
|
||||
|
||||
#include "tasks/heartbeat_task.h"
|
||||
#include "incphub/incphub_task.h"
|
||||
#include "tasks/sensor_task.h"
|
||||
#include "tasks/requester_task.h"
|
||||
#include "libshunt/shunt_task.h"
|
||||
|
||||
int my_main()
|
||||
{
|
||||
// Init
|
||||
incphub_task_preinit();
|
||||
libshunt_task_preinit();
|
||||
|
||||
// Create tasks
|
||||
xTaskCreateStatic(heartbeat_task_main, "heartbeat", HEARTBEAT_TASK_STACK_DEPTH, NULL, HEARTBEAT_TASK_PRIORITY, _heartbeat_task_stack, &_heartbeat_task);
|
||||
xTaskCreateStatic(incphub_task_main, "incphub", INCPHUB_TASK_STACK_DEPTH, NULL, INCPHUB_TASK_PRIORITY, _incphub_task_stack, &_incphub_task);
|
||||
xTaskCreateStatic(sensor_task_main, "sensor", SENSOR_TASK_STACK_DEPTH, NULL, SENSOR_TASK_PRIORITY, _sensor_task_stack, &_sensor_task);
|
||||
xTaskCreateStatic(requester_task_main, "requester", REQUESTER_TASK_STACK_DEPTH, NULL, REQUESTER_TASK_PRIORITY, _requester_task_stack, &_requester_task);
|
||||
xTaskCreateStatic(libshunt_task_main, "shunt", LIBSHUNT_TASK_STACK_DEPTH, NULL, LIBSHUNT_TASK_PRIORITY, _libshunt_task_stack, &_libshunt_task);
|
||||
|
||||
// Run
|
||||
vTaskStartScheduler();
|
||||
|
||||
@@ -12,7 +12,7 @@ typedef uint32_t ntl_chk_t;
|
||||
|
||||
|
||||
|
||||
// === Special Port Values ===
|
||||
// === Special Address Values ===
|
||||
|
||||
#define NIL_ADDR_LOCALHOST ((nil_addr_t)0)
|
||||
|
||||
|
||||
@@ -1,4 +1,6 @@
|
||||
target_sources(${EXECUTABLE} PRIVATE
|
||||
tasks.c
|
||||
heartbeat_task.c
|
||||
sensor_task.c
|
||||
requester_task.c
|
||||
)
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
#ifndef _HEARTBEAT_TASK_H_
|
||||
#define _HEARTBEAT_TASK_H_
|
||||
#ifndef HEARTBEAT_TASK_H_
|
||||
#define HEARTBEAT_TASK_H_
|
||||
|
||||
#include "FreeRTOS.h"
|
||||
#include "task.h"
|
||||
|
||||
73
app/src/tasks/requester_task.c
Normal file
73
app/src/tasks/requester_task.c
Normal file
@@ -0,0 +1,73 @@
|
||||
#include "requester_task.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "stm32f4xx_hal.h"
|
||||
#include "stm32f4xx_hal_uart.h"
|
||||
#include "usart.h"
|
||||
|
||||
#include "libshunt/shunt_client.h"
|
||||
|
||||
|
||||
|
||||
StackType_t _requester_task_stack[REQUESTER_TASK_STACK_DEPTH];
|
||||
StaticTask_t _requester_task;
|
||||
|
||||
|
||||
#define RANDOM_COUNT 4
|
||||
uint32_t random_values[RANDOM_COUNT] = { 411, 69, 420, 0xdeadbeef };
|
||||
|
||||
void requester_task_main(void *params)
|
||||
{
|
||||
(void)params;
|
||||
|
||||
TickType_t last_wake = xTaskGetTickCount();
|
||||
int random_index = 0;
|
||||
|
||||
uint8_t *buffer;
|
||||
size_t length;
|
||||
shunt_addr_t source;
|
||||
|
||||
// Init libshunt client
|
||||
shunt_cli_t *cli;
|
||||
shunt_init_client(&cli, 70);
|
||||
|
||||
while (1)
|
||||
{
|
||||
// Avoid spamming the sensor task
|
||||
xTaskDelayUntil(&last_wake, 50);
|
||||
|
||||
// Generate request
|
||||
uint32_t request = random_values[random_index++];
|
||||
random_index %= RANDOM_COUNT;
|
||||
|
||||
// Get buffer
|
||||
shunt_get_msg_buffer(cli, &buffer, sizeof(uint32_t));
|
||||
|
||||
// Populate buffer
|
||||
*(uint32_t*)buffer = request;
|
||||
|
||||
// Send to sendor task
|
||||
shunt_addr_t destination = { .addr = NIL_ADDR_LOCALHOST, .port = 69 };
|
||||
shunt_send_msg(cli, buffer, sizeof(uint32_t), destination, portMAX_DELAY);
|
||||
|
||||
// Await reply (assumes no other task will send to this one)
|
||||
shunt_await_msg(cli, &buffer, &length, &source, portMAX_DELAY);
|
||||
|
||||
// Ignore messages of different size
|
||||
if (length != sizeof(uint32_t))
|
||||
continue;
|
||||
|
||||
// Read reply message
|
||||
uint32_t num = *(uint32_t*)buffer;
|
||||
|
||||
// Release buffer
|
||||
shunt_release_msg_buffer(cli, buffer);
|
||||
|
||||
// Profit (send to UART to ensure it worked)
|
||||
char strbuf[32];
|
||||
snprintf(strbuf, sizeof(strbuf), "%lu\r\n", num);
|
||||
HAL_UART_Transmit(&huart2, (uint8_t*)strbuf, strlen(strbuf), 9999);
|
||||
}
|
||||
}
|
||||
16
app/src/tasks/requester_task.h
Normal file
16
app/src/tasks/requester_task.h
Normal file
@@ -0,0 +1,16 @@
|
||||
#ifndef REQUSTER_TASK_H_
|
||||
#define REQUSTER_TASK_H_
|
||||
|
||||
#include "FreeRTOS.h"
|
||||
#include "task.h"
|
||||
#include "FreeRTOSConfig.h"
|
||||
|
||||
#define REQUESTER_TASK_STACK_DEPTH 256
|
||||
#define REQUESTER_TASK_PRIORITY (configMAX_PRIORITIES - 1)
|
||||
|
||||
extern StackType_t _requester_task_stack[REQUESTER_TASK_STACK_DEPTH];
|
||||
extern StaticTask_t _requester_task;
|
||||
|
||||
void requester_task_main(void *params);
|
||||
|
||||
#endif
|
||||
55
app/src/tasks/sensor_task.c
Normal file
55
app/src/tasks/sensor_task.c
Normal file
@@ -0,0 +1,55 @@
|
||||
#include "sensor_task.h"
|
||||
|
||||
#include "stm32f4xx_hal.h"
|
||||
#include "stm32f4xx_hal_gpio.h"
|
||||
#include "gpio.h"
|
||||
|
||||
#include "libshunt/shunt_client.h"
|
||||
|
||||
|
||||
|
||||
StackType_t _sensor_task_stack[SENSOR_TASK_STACK_DEPTH];
|
||||
StaticTask_t _sensor_task;
|
||||
|
||||
|
||||
|
||||
void sensor_task_main(void *params)
|
||||
{
|
||||
(void)params;
|
||||
|
||||
uint8_t *buffer;
|
||||
size_t length;
|
||||
shunt_addr_t source;
|
||||
|
||||
// Init libshunt client
|
||||
shunt_cli_t *cli;
|
||||
shunt_init_client(&cli, 69);
|
||||
|
||||
while (1)
|
||||
{
|
||||
// Get a message (assume it is a uint32_t)
|
||||
shunt_await_msg(cli, &buffer, &length, &source, portMAX_DELAY);
|
||||
|
||||
// Ignore messages of different size
|
||||
if (length != sizeof(uint32_t))
|
||||
continue;
|
||||
|
||||
// Read incoming message
|
||||
uint32_t num = *(uint32_t*)buffer;
|
||||
|
||||
// Release buffer once read
|
||||
shunt_release_msg_buffer(cli, buffer);
|
||||
|
||||
// Process message
|
||||
num++;
|
||||
|
||||
// Request buffer for reply
|
||||
shunt_get_msg_buffer(cli, &buffer, sizeof(uint32_t));
|
||||
|
||||
// Prepare reply message
|
||||
*(uint32_t*)buffer = num;
|
||||
|
||||
// Send reply
|
||||
shunt_send_msg(cli, buffer, sizeof(uint32_t), source, portMAX_DELAY);
|
||||
}
|
||||
}
|
||||
16
app/src/tasks/sensor_task.h
Normal file
16
app/src/tasks/sensor_task.h
Normal file
@@ -0,0 +1,16 @@
|
||||
#ifndef SENSOR_TASK_H_
|
||||
#define SENSOR_TASK_H_
|
||||
|
||||
#include "FreeRTOS.h"
|
||||
#include "task.h"
|
||||
#include "FreeRTOSConfig.h"
|
||||
|
||||
#define SENSOR_TASK_STACK_DEPTH configMINIMAL_STACK_SIZE
|
||||
#define SENSOR_TASK_PRIORITY (configMAX_PRIORITIES - 1)
|
||||
|
||||
extern StackType_t _sensor_task_stack[SENSOR_TASK_STACK_DEPTH];
|
||||
extern StaticTask_t _sensor_task;
|
||||
|
||||
void sensor_task_main(void *params);
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user