esp_usb_dev.hpp

#pragma once
 
#include <cstddef>
#include <cstdint>
#include <initializer_list>
 
#include "esp_def.hpp"
#include "esp_intr_alloc.h"
#include "libxr_type.hpp"
#include "usb/core/ep_pool.hpp"
#include "usb/device/dev_core.hpp"
 
#if SOC_USB_OTG_SUPPORTED && defined(CONFIG_IDF_TARGET_ESP32S3) && \
    CONFIG_IDF_TARGET_ESP32S3
 
namespace LibXR
{
 
class ESP32USBEndpoint;
 
class ESP32USBDevice : public USB::EndpointPool, public USB::DeviceCore
{
 public:
  struct EPConfig
  {
    enum class DirectionHint : int8_t
    {
      BothDirections =
          -1,       
      OutOnly = 0,  
      InOnly = 1,   
    };
 
    RawData buffer;  
    DirectionHint direction_hint =
        DirectionHint::BothDirections;  
 
    EPConfig() = delete;
 
    explicit EPConfig(RawData buffer) : buffer(buffer) {}
 
    EPConfig(RawData buffer, bool is_in)
        : buffer(buffer),
          direction_hint(is_in ? DirectionHint::InOnly : DirectionHint::OutOnly)
    {
    }
  };
 
  ESP32USBDevice(
      const std::initializer_list<EPConfig> ep_cfgs,
      USB::DeviceDescriptor::PacketSize0 packet_size, uint16_t vid, uint16_t pid,
      uint16_t bcd,
      const std::initializer_list<const USB::DescriptorStrings::LanguagePack*> lang_list,
      const std::initializer_list<const std::initializer_list<USB::ConfigDescriptorItem*>>
          configs,
      ConstRawData uid = {nullptr, 0});
 
  void Init(bool in_isr) override;
 
  void Deinit(bool in_isr) override;
 
  ErrorCode SetAddress(uint8_t address, USB::DeviceCore::Context context) override;
 
  void Start(bool in_isr) override;
 
  void Stop(bool in_isr) override;
 
 private:
  friend class ESP32USBEndpoint;
 
  static constexpr uint8_t ENDPOINT_COUNT = 7;
  static constexpr uint8_t IN_ENDPOINT_LIMIT = 5;
  static constexpr uint32_t INTERRUPT_DISPATCH_GUARD = 64U;
  static constexpr size_t SETUP_PACKET_BYTES = 8U;
  static constexpr size_t SETUP_DMA_BUFFER_BYTES = 64U;
 
  struct ControlState
  {
    bool setup_direction_out =
        false;  
  };
 
  struct EndpointMap
  {
    USB::Endpoint* in[ENDPOINT_COUNT] =
        {};  
    USB::Endpoint* out[ENDPOINT_COUNT] =
        {};  
  };
 
  struct FifoState
  {
    uint16_t depth_words =
        0U;  
    uint16_t rx_words =
        0U;  
    uint16_t tx_next_words =
        0U;  
    uint16_t tx_words[ENDPOINT_COUNT] =
        {};  
    bool tx_bound[ENDPOINT_COUNT] = {};  
    uint8_t allocated_in = 0U;  
  };
 
  struct RuntimeState
  {
    intr_handle_t intr_handle =
        nullptr;  
    void* phy_handle = nullptr;  
    bool phy_ready = false;  
    bool irq_ready =
        false;  
    bool started = false;  
    bool rom_usb_cleaned = false;  
  };
 
  static void IRAM_ATTR IsrEntry(void* arg);
 
  bool EnsurePhyReady();
 
  bool EnsureInterruptReady();
 
  void EnsureRomUsbCleaned();
 
  void InitializeCore();
 
  void ClearTxFifoRegisters();
 
  void FlushFifos();
 
  void ResetFifoState();
 
  void ResetDeviceState();
 
  void ResetEndpointHardwareState();
 
  void ReloadSetupPacketCount();
 
  void HandleInterrupt();
 
  void HandleBusReset(bool in_isr);
 
  void HandleEndpointInterrupt(bool in_isr, bool in_dir);
 
  void HandleRxFifoLevel();
 
  void ResetControlState();
 
  void UpdateSetupState(const uint8_t* setup);
 
  bool LastSetupDirectionOut() const { return control_.setup_direction_out; }
 
  bool DmaEnabled() const { return true; }
 
  bool AllocateTxFifo(uint8_t ep_num, uint16_t packet_size, bool is_bulk,
                      uint16_t& fifo_words);
 
  bool EnsureRxFifo(uint16_t packet_size);
 
  EndpointMap endpoint_map_ = {};  
  FifoState fifo_state_ =
      {};  
  RuntimeState runtime_ =
      {};  
  alignas(SETUP_DMA_BUFFER_BYTES) uint8_t setup_packet_[SETUP_DMA_BUFFER_BYTES] =
      {};  
  ControlState control_ = {};  
};
 
}  // namespace LibXR
 
#endif