#include "general.h"
#include "platform.h"
#include <assert.h>
#include "usb_serial.h"
#include "rtt.h"
#include "rtt_if.h"
static char recv_buf[RTT_DOWN_BUF_SIZE];
static uint32_t recv_head = 0;
static uint32_t recv_tail = 0;
inline static uint32_t recv_bytes_free()
{
if (recv_tail <= recv_head)
return RTT_DOWN_BUF_SIZE - recv_head + recv_tail - 1U;
return recv_tail - recv_head - 1U;
}
inline static bool recv_set_nak()
{
assert(sizeof(recv_buf) > 2U * CDCACM_PACKET_SIZE);
return recv_bytes_free() < 2U * CDCACM_PACKET_SIZE;
}
void rtt_serial_receive_callback(usbd_device *dev, uint8_t ep)
{
(void)dev;
(void)ep;
char usb_buf[CDCACM_PACKET_SIZE];
usbd_ep_nak_set(usbdev, CDCACM_UART_ENDPOINT, 1);
const uint16_t len = usbd_ep_read_packet(usbdev, CDCACM_UART_ENDPOINT, usb_buf, CDCACM_PACKET_SIZE);
if (rtt_flag_skip && len > recv_bytes_free()) {
usbd_ep_nak_set(usbdev, CDCACM_UART_ENDPOINT, 0);
return;
}
for (int i = 0; i < len; i++) {
uint32_t next_recv_head = (recv_head + 1U) % sizeof(recv_buf);
if (next_recv_head == recv_tail)
break;
recv_buf[recv_head] = usb_buf[i];
recv_head = next_recv_head;
}
if (!(rtt_flag_block && recv_set_nak()))
usbd_ep_nak_set(usbdev, CDCACM_UART_ENDPOINT, 0);
}
int32_t rtt_getchar()
{
int retval;
if (recv_head == recv_tail)
return -1;
retval = (uint8_t)recv_buf[recv_tail];
recv_tail = (recv_tail + 1U) % sizeof(recv_buf);
if (!recv_set_nak())
usbd_ep_nak_set(usbdev, CDCACM_UART_ENDPOINT, 0);
return retval;
}
bool rtt_nodata()
{
return recv_head == recv_tail;
}
uint32_t rtt_write(const char *buf, uint32_t len)
{
if (len != 0 && usbdev && usb_get_config() && gdb_serial_get_dtr()) {
for (uint32_t p = 0; p < len; p += CDCACM_PACKET_SIZE) {
uint32_t plen = MIN(CDCACM_PACKET_SIZE, len - p);
while (usbd_ep_write_packet(usbdev, CDCACM_UART_ENDPOINT, buf + p, plen) <= 0)
continue;
}
if (CDCACM_PACKET_SIZE == 64 && (len % CDCACM_PACKET_SIZE) == 0)
usbd_ep_write_packet(usbdev, CDCACM_UART_ENDPOINT, NULL, 0);
}
return len;
}