mirror of
https://github.com/Keychron/qmk_firmware.git
synced 2024-11-30 12:47:37 +06:00
Add master to slave transport
This commit is contained in:
parent
fe1db50fc3
commit
415ab91c9d
|
@ -23,18 +23,43 @@ SOFTWARE.
|
|||
*/
|
||||
|
||||
#include "protocol/transport.h"
|
||||
#include "protocol/frame_router.h"
|
||||
#include "protocol/triple_buffered_object.h"
|
||||
|
||||
static remote_object_t* remote_objects;
|
||||
static remote_object_t** remote_objects;
|
||||
static uint32_t num_remote_objects;
|
||||
|
||||
void init_transport(remote_object_t* _remote_objects, uint32_t _num_remote_objects) {
|
||||
void init_transport(remote_object_t** _remote_objects, uint32_t _num_remote_objects) {
|
||||
remote_objects = _remote_objects;
|
||||
num_remote_objects = _num_remote_objects;
|
||||
}
|
||||
|
||||
void transport_recv_frame(uint8_t from, uint8_t* data, uint16_t size) {
|
||||
|
||||
uint8_t id = data[size-1];
|
||||
remote_object_t* obj = remote_objects[id];
|
||||
if (obj->object_type == MASTER_TO_ALL_SLAVES) {
|
||||
uint8_t* start = obj->buffer + LOCAL_OBJECT_SIZE(obj->object_size);
|
||||
triple_buffer_object_t* tb = (triple_buffer_object_t*)start;
|
||||
void* ptr = triple_buffer_begin_write_internal(obj->object_size, tb);
|
||||
memcpy(ptr, data, size -1);
|
||||
triple_buffer_end_write_internal(tb);
|
||||
}
|
||||
}
|
||||
|
||||
uint32_t transport_send_frame(uint8_t to, uint8_t* data, uint16_t size) {
|
||||
}
|
||||
|
||||
void update_transport(void) {
|
||||
int i;
|
||||
for(i=0;i<num_remote_objects;i++) {
|
||||
remote_object_t* obj = remote_objects[i];
|
||||
if (obj->object_type == MASTER_TO_ALL_SLAVES) {
|
||||
triple_buffer_object_t* tb = (triple_buffer_object_t*)obj->buffer;
|
||||
uint8_t* ptr = (uint8_t*)triple_buffer_read_internal(obj->object_size + LOCAL_OBJECT_EXTRA, tb);
|
||||
if (ptr) {
|
||||
ptr[obj->object_size] = i;
|
||||
router_send_frame(0xFF, ptr, obj->object_size + 1);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -68,9 +68,21 @@ typedef struct { \
|
|||
} \
|
||||
}; \
|
||||
type* begin_write_##name(void) { \
|
||||
remote_object_t* obj = (remote_object_t*)&remote_object_##name; \
|
||||
triple_buffer_object_t* tb = (triple_buffer_object_t*)obj->buffer; \
|
||||
return (type*)triple_buffer_begin_write_internal(sizeof(type) + LOCAL_OBJECT_EXTRA, tb); \
|
||||
}\
|
||||
void end_write_##name(void) { \
|
||||
remote_object_t* obj = (remote_object_t*)&remote_object_##name; \
|
||||
triple_buffer_object_t* tb = (triple_buffer_object_t*)obj->buffer; \
|
||||
triple_buffer_end_write_internal(tb); \
|
||||
signal_data_written(); \
|
||||
}\
|
||||
type* read_##name(void) { \
|
||||
remote_object_t* obj = (remote_object_t*)&remote_object_##name; \
|
||||
uint8_t* start = obj->buffer + LOCAL_OBJECT_SIZE(obj->object_size);\
|
||||
triple_buffer_object_t* tb = (triple_buffer_object_t*)start; \
|
||||
return triple_buffer_read_internal(obj->object_size, tb); \
|
||||
}
|
||||
|
||||
#define MASTER_TO_SINGLE_SLAVE_OBJECT(name, type) \
|
||||
|
@ -103,11 +115,9 @@ typedef struct { \
|
|||
|
||||
#define REMOTE_OBJECT(name) (remote_object_t*)&remote_object_##name
|
||||
|
||||
void init_transport(remote_object_t* remote_objects, uint32_t num_remote_objects);
|
||||
void init_transport(remote_object_t** remote_objects, uint32_t num_remote_objects);
|
||||
void transport_recv_frame(uint8_t from, uint8_t* data, uint16_t size);
|
||||
uint32_t transport_send_frame(uint8_t to, uint8_t* data, uint16_t size);
|
||||
|
||||
void transport_register_master_remote_object(uint8_t id, void* ptr, uint16_t size);
|
||||
void transport_register_slave_remote_object(uint8_t id, void* ptr, uint16_t size);
|
||||
void update_transport(void);
|
||||
|
||||
#endif
|
||||
|
|
|
@ -25,11 +25,21 @@ SOFTWARE.
|
|||
#include <cgreen/cgreen.h>
|
||||
#include <cgreen/mocks.h>
|
||||
#include "protocol/transport.c"
|
||||
#include "protocol/triple_buffered_object.c"
|
||||
|
||||
void signal_data_written(void) {
|
||||
mock();
|
||||
}
|
||||
|
||||
static uint8_t sent_data[2048];
|
||||
static uint16_t sent_data_size;
|
||||
|
||||
void router_send_frame(uint8_t destination, uint8_t* data, uint16_t size) {
|
||||
mock(destination);
|
||||
memcpy(sent_data + sent_data_size, data, size);
|
||||
sent_data_size += size;
|
||||
}
|
||||
|
||||
typedef struct {
|
||||
uint32_t test;
|
||||
} test_object1_t;
|
||||
|
@ -51,7 +61,8 @@ remote_object_t* test_remote_objects[] = {
|
|||
|
||||
Describe(Transport);
|
||||
BeforeEach(Transport) {
|
||||
init_transport(remote_objects, sizeof(remote_objects) / sizeof(remote_object_t));
|
||||
init_transport(test_remote_objects, sizeof(test_remote_objects) / sizeof(remote_object_t*));
|
||||
sent_data_size = 0;
|
||||
}
|
||||
AfterEach(Transport) {}
|
||||
|
||||
|
@ -66,3 +77,18 @@ Ensure(Transport, write_to_local_signals_an_event) {
|
|||
expect(signal_data_written);
|
||||
end_write_master_to_single_slave(1);
|
||||
}
|
||||
|
||||
Ensure(Transport, writes_from_master_to_all_slaves) {
|
||||
update_transport();
|
||||
test_object1_t* obj = begin_write_master_to_slave();
|
||||
obj->test = 5;
|
||||
expect(signal_data_written);
|
||||
end_write_master_to_slave();
|
||||
expect(router_send_frame,
|
||||
when(destination, is_equal_to(0xFF)));
|
||||
update_transport();
|
||||
transport_recv_frame(0, sent_data, sent_data_size);
|
||||
test_object1_t* obj2 = read_master_to_slave();
|
||||
assert_that(obj2, is_not_equal_to(NULL));
|
||||
assert_that(obj2->test, is_equal_to(5));
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue
Block a user