1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
/*****************************************************************************
# #
# KVMD - The main PiKVM daemon. #
# #
# Copyright (C) 2018-2023 Maxim Devaev <[email protected]> #
# #
# This program is free software: you can redistribute it and/or modify #
# it under the terms of the GNU General Public License as published by #
# the Free Software Foundation, either version 3 of the License, or #
# (at your option) any later version. #
# #
# This program is distributed in the hope that it will be useful, #
# but WITHOUT ANY WARRANTY; without even the implied warranty of #
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
# GNU General Public License for more details. #
# #
# You should have received a copy of the GNU General Public License #
# along with this program. If not, see <https://www.gnu.org/licenses/>. #
# #
*****************************************************************************/
#include "pico/stdlib.h"
#include "hardware/gpio.h"
#include "hardware/watchdog.h"
#include "ph_types.h"
#include "ph_tools.h"
#include "ph_outputs.h"
#include "ph_usb.h"
#include "ph_spi.h"
#include "ph_proto.h"
#include "ph_cmds.h"
#include "ph_debug.h"
static bool _reset_required = false;
static u8 _handle_request(const u8 *data) { // 8 bytes
// FIXME: See kvmd/kvmd#80
// Should input buffer be cleared in this case?
if (data[0] == PH_PROTO_MAGIC && ph_crc16(data, 6) == ph_merge8_u16(data[6], data[7])) {
# define HANDLE(x_handler, x_reset) { \
x_handler(data + 2); \
if (x_reset) { _reset_required = true; } \
return PH_PROTO_PONG_OK; \
}
switch (data[1]) {
case PH_PROTO_CMD_PING: return PH_PROTO_PONG_OK;
case PH_PROTO_CMD_SET_KBD: HANDLE(ph_cmd_set_kbd, true);
case PH_PROTO_CMD_SET_MOUSE: HANDLE(ph_cmd_set_mouse, true);
case PH_PROTO_CMD_SET_CONNECTED: return PH_PROTO_PONG_OK; // Arduino AUM
case PH_PROTO_CMD_CLEAR_HID: HANDLE(ph_cmd_send_clear, false);
case PH_PROTO_CMD_KBD_KEY: HANDLE(ph_cmd_kbd_send_key, false);
case PH_PROTO_CMD_MOUSE_BUTTON: HANDLE(ph_cmd_mouse_send_button, false);
case PH_PROTO_CMD_MOUSE_ABS: HANDLE(ph_cmd_mouse_send_abs, false);
case PH_PROTO_CMD_MOUSE_REL: HANDLE(ph_cmd_mouse_send_rel, false);
case PH_PROTO_CMD_MOUSE_WHEEL: HANDLE(ph_cmd_mouse_send_wheel, false);
case PH_PROTO_CMD_REPEAT: return 0;
}
# undef HANDLE
return PH_PROTO_RESP_INVALID_ERROR;
}
return PH_PROTO_RESP_CRC_ERROR;
}
static void _send_response(u8 code) {
static u8 prev_code = PH_PROTO_RESP_NONE;
if (code == 0) {
code = prev_code; // Repeat the last code
} else {
prev_code = code;
}
u8 resp[8] = {0};
resp[0] = PH_PROTO_MAGIC_RESP;
if (code & PH_PROTO_PONG_OK) {
resp[1] = PH_PROTO_PONG_OK;
if (_reset_required) {
resp[1] |= PH_PROTO_PONG_RESET_REQUIRED;
}
resp[2] = PH_PROTO_OUT1_DYNAMIC;
resp[1] |= ph_cmd_get_offlines();
resp[1] |= ph_cmd_kbd_get_leds();
resp[2] |= ph_g_outputs_active;
resp[3] |= ph_g_outputs_avail;
} else {
resp[1] = code;
}
ph_split16(ph_crc16(resp, 6), &resp[6], &resp[7]);
ph_spi_write(resp);
if (_reset_required) {
watchdog_reboot(0, 0, 100); // Даем немного времени чтобы отправить ответ, а потом ребутимся
}
}
static void _spi_data_handler(const u8 *data) {
_send_response(_handle_request(data));
}
int main(void) {
ph_debug_init(false); // No UART
ph_outputs_init();
ph_usb_init();
ph_spi_init(_spi_data_handler);
while (true) {
ph_usb_task();
if (!_reset_required) {
ph_spi_task();
ph_debug_act_pulse(50);
}
}
return 0;
}
|
