stm32_usb.cpp

#include "stm32_usb.hpp"
 
#if defined(HAL_PCD_MODULE_ENABLED) && !defined(LIBXR_SYSTEM_ThreadX)
 
using namespace LibXR;
 
STM32VirtualUART *STM32VirtualUART::map[1];
 
int8_t libxr_stm32_virtual_uart_init(void)
{
  STM32VirtualUART *uart = STM32VirtualUART::map[0];
  USBD_CDC_SetTxBuffer(STM32VirtualUART::map[0]->usb_handle_,
                       uart->tx_buffer_.ActiveBuffer(), 0);
  USBD_CDC_SetRxBuffer(STM32VirtualUART::map[0]->usb_handle_,
                       uart->rx_buffer_.ActiveBuffer());
  return (USBD_OK);
}
 
int8_t libxr_stm32_virtual_uart_deinit(void) { return (USBD_OK); }
 
int8_t libxr_stm32_virtual_uart_control(uint8_t cmd, uint8_t *pbuf, uint16_t len)
{
  UNUSED(cmd);
  UNUSED(pbuf);
  UNUSED(len);
  return (USBD_OK);
}
 
int8_t libxr_stm32_virtual_uart_receive(uint8_t *pbuf, uint32_t *Len)
{
  STM32VirtualUART *uart = STM32VirtualUART::map[0];
 
  uint8_t *buffer = nullptr;
 
  if (pbuf == uart->rx_buffer_.ActiveBuffer())
  {
    buffer = uart->rx_buffer_.PendingBuffer();
  }
  else
  {
    buffer = uart->rx_buffer_.ActiveBuffer();
  }
 
  USBD_CDC_SetRxBuffer(STM32VirtualUART::map[0]->usb_handle_, buffer);
 
  USBD_CDC_ReceivePacket(STM32VirtualUART::map[0]->usb_handle_);
 
#if __DCACHE_PRESENT
  SCB_InvalidateDCache_by_Addr(pbuf, *Len);
#endif
 
  uart->read_port_->queue_data_->PushBatch(pbuf, *Len);
  uart->read_port_->ProcessPendingReads(true);
 
  return (USBD_OK);
}
 
int8_t libxr_stm32_virtual_uart_transmit(uint8_t *pbuf, uint32_t *Len, uint8_t epnum)
{
  UNUSED(epnum);
  UNUSED(pbuf);
  UNUSED(Len);
 
  STM32VirtualUART *uart = STM32VirtualUART::map[0];
 
  WriteInfoBlock &current_info = uart->write_info_active_;
 
  if (!uart->tx_buffer_.HasPending())
  {
    return USBD_OK;
  }
 
  if (uart->write_port_->queue_info_->Pop(current_info) != ErrorCode::OK)
  {
    ASSERT(false);
    return USBD_OK;
  }
 
  uart->tx_buffer_.Switch();
 
#if defined(STM32F1)
  uart->write_size_ = current_info.data.size_;
  uart->writing_ = true;
#endif
 
  USBD_CDC_SetTxBuffer(uart->usb_handle_, uart->tx_buffer_.ActiveBuffer(),
                       current_info.data.size_);
#if __DCACHE_PRESENT
  SCB_CleanDCache_by_Addr(reinterpret_cast<uint32_t *>(uart->tx_buffer_.ActiveBuffer()),
                          *Len);
#endif
 
  uart->write_port_->write_size_ = current_info.data.size_;
 
  auto ans = USBD_CDC_TransmitPacket(uart->usb_handle_);
 
  current_info.op.UpdateStatus(true, ans == USBD_OK ? ErrorCode::OK : ErrorCode::BUSY);
 
  WriteInfoBlock next_info;
 
  if (uart->write_port_->queue_info_->Peek(next_info) != ErrorCode::OK)
  {
    return USBD_OK;
  }
 
  if (uart->write_port_->queue_data_->PopBatch(uart->tx_buffer_.PendingBuffer(),
                                               next_info.data.size_) != ErrorCode::OK)
  {
    ASSERT(false);
    return USBD_OK;
  }
 
  uart->tx_buffer_.EnablePending();
 
  return (USBD_OK);
}
 
#if defined(STM32F1)
extern "C" void STM32_USB_ISR_Handler_F1(void)
{
  if (STM32VirtualUART::map[0] == nullptr)
  {
    return;
  }
 
  auto p_data_class = reinterpret_cast<USBD_CDC_HandleTypeDef *>(
      STM32VirtualUART::map[0]->usb_handle_->pClassData);
 
  if (p_data_class == nullptr)
  {
    return;
  }
 
  static uint32_t last_tx_state = 0;
 
  if (last_tx_state != 0 && p_data_class->TxState == 0 &&
      STM32VirtualUART::map[0]->writing_)
  {
    STM32VirtualUART::map[0]->writing_ = false;
    libxr_stm32_virtual_uart_transmit(STM32VirtualUART::map[0]->tx_buffer_.ActiveBuffer(),
                                      &STM32VirtualUART::map[0]->write_size_, 0);
  }
 
  last_tx_state = p_data_class->TxState;
}
#endif
 
ErrorCode STM32VirtualUART::WriteFun(WritePort &port)
{
  STM32VirtualUART *uart = CONTAINER_OF(&port, STM32VirtualUART, _write_port);
  auto p_data_class =
      reinterpret_cast<USBD_CDC_HandleTypeDef *>(uart->usb_handle_->pClassData);
 
  WriteInfoBlock info;
 
  if (p_data_class == nullptr)
  {
    port.queue_info_->Pop(info);
    port.queue_data_->PopBatch(uart->tx_buffer_.PendingBuffer(), info.data.size_);
    port.Finish(false, ErrorCode::INIT_ERR, info, 0);
    return ErrorCode::INIT_ERR;
  }
 
  if (uart->tx_buffer_.HasPending())
  {
    return ErrorCode::FULL;
  }
 
  if (port.queue_info_->Peek(info) != ErrorCode::OK)
  {
    return ErrorCode::EMPTY;
  }
 
  if (port.queue_data_->PopBatch(uart->tx_buffer_.PendingBuffer(), info.data.size_) !=
      ErrorCode::OK)
  {
    ASSERT(false);
    return ErrorCode::EMPTY;
  }
 
  uart->tx_buffer_.EnablePending();
 
#if defined(STM32F1)
  if (!uart->writing_ && uart->tx_buffer_.HasPending())
#else
  if (p_data_class->TxState == 0)
#endif
  {
    uart->tx_buffer_.Switch();
    port.queue_info_->Pop(uart->write_info_active_);
 
#if defined(STM32F1)
    uart->write_size_ = info.data.size_;
    uart->writing_ = true;
#endif
 
    USBD_CDC_SetTxBuffer(uart->usb_handle_, uart->tx_buffer_.ActiveBuffer(),
                         info.data.size_);
#if __DCACHE_PRESENT
    SCB_CleanDCache_by_Addr(reinterpret_cast<uint32_t *>(uart->tx_buffer_.ActiveBuffer()),
                            info.data.size_);
#endif
 
    port.write_size_ = info.data.size_;
 
    auto ans = USBD_CDC_TransmitPacket(uart->usb_handle_);
 
    info.op.UpdateStatus(true, ans == USBD_OK ? ErrorCode::OK : ErrorCode::BUSY);
 
    return ErrorCode::FAILED;
  }
 
  return ErrorCode::FAILED;
}
 
ErrorCode STM32VirtualUART::ReadFun(ReadPort &port)
{
  UNUSED(port);
  return ErrorCode::EMPTY;
}
 
STM32VirtualUART::STM32VirtualUART(USBD_HandleTypeDef &usb_handle, uint8_t *tx_buffer,
                                   uint8_t *rx_buffer, uint32_t tx_queue_size)
    : UART(&_read_port, &_write_port),
      usb_handle_(&usb_handle),
      tx_buffer_(RawData(tx_buffer, APP_TX_DATA_SIZE)),
      rx_buffer_(RawData(rx_buffer, APP_RX_DATA_SIZE)),
      _write_port(tx_queue_size, APP_TX_DATA_SIZE),
      _read_port(APP_RX_DATA_SIZE)
{
  map[0] = this;
 
  static USBD_CDC_ItfTypeDef usbd_cdc_itf = Apply<USBD_CDC_ItfTypeDef>();
 
  USBD_CDC_RegisterInterface(usb_handle_, &usbd_cdc_itf);
 
  _write_port = WriteFun;
  _read_port = ReadFun;
 
  USBD_CDC_ReceivePacket(usb_handle_);
}
 
ErrorCode STM32VirtualUART::SetConfig(UART::Configuration config)
{
  UNUSED(config);
  return ErrorCode::OK;
}
 
#endif