|
|
- /*
- The MIT License (MIT)
-
- Copyright (c) 2016 Fred Sundvik
-
- Permission is hereby granted, free of charge, to any person obtaining a copy
- of this software and associated documentation files (the "Software"), to deal
- in the Software without restriction, including without limitation the rights
- to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
- copies of the Software, and to permit persons to whom the Software is
- furnished to do so, subject to the following conditions:
-
- The above copyright notice and this permission notice shall be included in all
- copies or substantial portions of the Software.
-
- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
- OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
- SOFTWARE.
- */
- #include "report.h"
- #include "host_driver.h"
- #include "serial_link/system/serial_link.h"
- #include <hal.h>
- #include "serial_link/protocol/byte_stuffer.h"
- #include "serial_link/protocol/transport.h"
- #include "serial_link/protocol/frame_router.h"
- #include "matrix.h"
- #include "sync_timer.h"
- #include <stdbool.h>
- #include "print.h"
- #include "config.h"
-
- #define SYNC_TIMER_OFFSET 2
-
- static event_source_t new_data_event;
- static bool serial_link_connected;
- static bool is_master = false;
-
- static uint8_t keyboard_leds(void);
- static void send_keyboard(report_keyboard_t* report);
- static void send_mouse(report_mouse_t* report);
- static void send_system(uint16_t data);
- static void send_consumer(uint16_t data);
-
- host_driver_t serial_driver = {keyboard_leds, send_keyboard, send_mouse, send_system, send_consumer};
-
- // Define these in your Config.h file
- #ifndef SERIAL_LINK_BAUD
- # error "Serial link baud is not set"
- #endif
-
- #ifndef SERIAL_LINK_THREAD_PRIORITY
- # error "Serial link thread priority not set"
- #endif
-
- static SerialConfig config = {.sc_speed = SERIAL_LINK_BAUD};
-
- //#define DEBUG_LINK_ERRORS
-
- static uint32_t read_from_serial(SerialDriver* driver, uint8_t link) {
- const uint32_t buffer_size = 16;
- uint8_t buffer[buffer_size];
- uint32_t bytes_read = sdAsynchronousRead(driver, buffer, buffer_size);
- uint8_t* current = buffer;
- uint8_t* end = current + bytes_read;
- while (current < end) {
- byte_stuffer_recv_byte(link, *current);
- current++;
- }
- return bytes_read;
- }
-
- static void print_error(char* str, eventflags_t flags, SerialDriver* driver) {
- #ifdef DEBUG_LINK_ERRORS
- if (flags & SD_PARITY_ERROR) {
- print(str);
- print(" Parity error\n");
- }
- if (flags & SD_FRAMING_ERROR) {
- print(str);
- print(" Framing error\n");
- }
- if (flags & SD_OVERRUN_ERROR) {
- print(str);
- uint32_t size = qSpaceI(&(driver->iqueue));
- xprintf(" Overrun error, queue size %d\n", size);
- }
- if (flags & SD_NOISE_ERROR) {
- print(str);
- print(" Noise error\n");
- }
- if (flags & SD_BREAK_DETECTED) {
- print(str);
- print(" Break detected\n");
- }
- #else
- (void)str;
- (void)flags;
- (void)driver;
- #endif
- }
-
- bool is_serial_link_master(void) { return is_master; }
-
- // TODO: Optimize the stack size, this is probably way too big
- static THD_WORKING_AREA(serialThreadStack, 1024);
- static THD_FUNCTION(serialThread, arg) {
- (void)arg;
- event_listener_t new_data_listener;
- event_listener_t sd1_listener;
- event_listener_t sd2_listener;
- chEvtRegister(&new_data_event, &new_data_listener, 0);
- eventflags_t events = CHN_INPUT_AVAILABLE | SD_PARITY_ERROR | SD_FRAMING_ERROR | SD_OVERRUN_ERROR | SD_NOISE_ERROR | SD_BREAK_DETECTED;
- chEvtRegisterMaskWithFlags(chnGetEventSource(&SD1), &sd1_listener, EVENT_MASK(1), events);
- chEvtRegisterMaskWithFlags(chnGetEventSource(&SD2), &sd2_listener, EVENT_MASK(2), events);
- bool need_wait = false;
- while (true) {
- eventflags_t flags1 = 0;
- eventflags_t flags2 = 0;
- if (need_wait) {
- eventmask_t mask = chEvtWaitAnyTimeout(ALL_EVENTS, TIME_MS2I(1000));
- if (mask & EVENT_MASK(1)) {
- flags1 = chEvtGetAndClearFlags(&sd1_listener);
- print_error("DOWNLINK", flags1, &SD1);
- }
- if (mask & EVENT_MASK(2)) {
- flags2 = chEvtGetAndClearFlags(&sd2_listener);
- print_error("UPLINK", flags2, &SD2);
- }
- }
-
- // Always stay as master, even if the USB goes into sleep mode
- is_master |= usbGetDriverStateI(&USBD1) == USB_ACTIVE;
- router_set_master(is_master);
-
- need_wait = true;
- need_wait &= read_from_serial(&SD2, UP_LINK) == 0;
- need_wait &= read_from_serial(&SD1, DOWN_LINK) == 0;
- update_transport();
- }
- }
-
- void send_data(uint8_t link, const uint8_t* data, uint16_t size) {
- if (link == DOWN_LINK) {
- sdWrite(&SD1, data, size);
- } else {
- sdWrite(&SD2, data, size);
- }
- }
-
- static systime_t last_update = 0;
-
- typedef struct {
- matrix_row_t rows[MATRIX_ROWS];
- } matrix_object_t;
-
- static matrix_object_t last_matrix = {};
-
- SLAVE_TO_MASTER_OBJECT(keyboard_matrix, matrix_object_t);
- MASTER_TO_ALL_SLAVES_OBJECT(serial_link_connected, bool);
- #ifndef DISABLE_SYNC_TIMER
- MASTER_TO_ALL_SLAVES_OBJECT(sync_timer, uint32_t);
- #endif
-
- static remote_object_t* remote_objects[] = {
- REMOTE_OBJECT(serial_link_connected),
- REMOTE_OBJECT(keyboard_matrix),
- #ifndef DISABLE_SYNC_TIMER
- REMOTE_OBJECT(sync_timer),
- #endif
- };
-
- void init_serial_link(void) {
- serial_link_connected = false;
- init_serial_link_hal();
- add_remote_objects(remote_objects, sizeof(remote_objects) / sizeof(remote_object_t*));
- init_byte_stuffer();
- sdStart(&SD1, &config);
- sdStart(&SD2, &config);
- chEvtObjectInit(&new_data_event);
- (void)chThdCreateStatic(serialThreadStack, sizeof(serialThreadStack), SERIAL_LINK_THREAD_PRIORITY, serialThread, NULL);
- }
-
- void matrix_set_remote(matrix_row_t* rows, uint8_t index);
-
- void serial_link_update(void) {
- if (read_serial_link_connected()) {
- serial_link_connected = true;
- }
-
- matrix_object_t matrix;
- bool changed = false;
- for (uint8_t i = 0; i < MATRIX_ROWS; i++) {
- matrix.rows[i] = matrix_get_row(i);
- changed |= matrix.rows[i] != last_matrix.rows[i];
- }
-
- systime_t current_time = chVTGetSystemTimeX();
- systime_t delta = current_time - last_update;
- if (changed || delta > TIME_US2I(5000)) {
- last_update = current_time;
- last_matrix = matrix;
- matrix_object_t* m = begin_write_keyboard_matrix();
- for (uint8_t i = 0; i < MATRIX_ROWS; i++) {
- m->rows[i] = matrix.rows[i];
- }
- end_write_keyboard_matrix();
-
- *begin_write_serial_link_connected() = true;
- end_write_serial_link_connected();
-
- #ifndef DISABLE_SYNC_TIMER
- *begin_write_sync_timer() = sync_timer_read32() + SYNC_TIMER_OFFSET;
- end_write_sync_timer();
- #endif
- }
-
- matrix_object_t* m = read_keyboard_matrix(0);
- if (m) {
- matrix_set_remote(m->rows, 0);
- }
-
- #ifndef DISABLE_SYNC_TIMER
- uint32_t* t = read_sync_timer();
- if (t) {
- sync_timer_update(*t);
- }
- #endif
- }
-
- void signal_data_written(void) { chEvtBroadcast(&new_data_event); }
-
- bool is_serial_link_connected(void) { return serial_link_connected; }
-
- host_driver_t* get_serial_link_driver(void) { return &serial_driver; }
-
- // NOTE: The driver does nothing, because the master handles everything
- uint8_t keyboard_leds(void) { return 0; }
-
- void send_keyboard(report_keyboard_t* report) { (void)report; }
-
- void send_mouse(report_mouse_t* report) { (void)report; }
-
- void send_system(uint16_t data) { (void)data; }
-
- void send_consumer(uint16_t data) { (void)data; }
|