device_composition.hpp

#pragma once
 
#include <cstddef>
#include <cstdint>
#include <initializer_list>
 
#include "device_class.hpp"
#include "libxr_type.hpp"
#include "usb/core/bos.hpp"
#include "usb/core/desc_cfg.hpp"
#include "usb/core/desc_str.hpp"
#include "usb/core/ep_pool.hpp"
 
namespace LibXR::USB
{
class DeviceComposition
{
  struct ConfigItems
  {
    ConfigDescriptorItem** items = nullptr;  
    size_t item_num = 0;                     
  };
 
 public:
  DeviceComposition(
      EndpointPool& endpoint_pool,
      const std::initializer_list<const DescriptorStrings::LanguagePack*>& lang_list,
      const std::initializer_list<const std::initializer_list<ConfigDescriptorItem*>>&
          configs,
      ConstRawData uid = {nullptr, 0}, uint8_t bmAttributes = CFG_BUS_POWERED,
      uint8_t bMaxPower = 50);
 
  DeviceComposition(const DeviceComposition&) = delete;
  DeviceComposition& operator=(const DeviceComposition&) = delete;
  DeviceComposition(DeviceComposition&&) = delete;
  DeviceComposition& operator=(DeviceComposition&&) = delete;
 
  void Init(bool in_isr);
 
  void Deinit(bool in_isr);
 
  ErrorCode SwitchConfig(size_t index, bool in_isr);
 
  ErrorCode BuildConfigDescriptor();
 
  [[nodiscard]] RawData GetConfigDescriptor() const;
 
  [[nodiscard]] ConstRawData GetBosDescriptor();
 
  ErrorCode ProcessBosVendorRequest(bool in_isr, const SetupPacket* setup,
                                    BosVendorResult& result);
 
  ErrorCode GetStringDescriptor(uint8_t string_index, uint16_t lang, ConstRawData& data);
 
  [[nodiscard]] bool IsComposite() const;
 
  ErrorCode TryOverrideDeviceDescriptor(DeviceDescriptor& descriptor);
 
  [[nodiscard]] size_t GetConfigNum() const;
 
  [[nodiscard]] size_t GetCurrentConfig() const;
 
  [[nodiscard]] uint16_t GetDeviceStatus() const;
 
  [[nodiscard]] DeviceClass* FindClassByInterfaceNumber(size_t index) const;
 
  [[nodiscard]] DeviceClass* FindClassByEndpointAddress(uint8_t addr) const;
 
 private:
  [[nodiscard]] const ConfigItems& CurrentConfigItems() const;
 
  void BindEndpoints(bool in_isr);
 
  void UnbindEndpoints(bool in_isr);
 
  void RebuildBosCache();
 
  void RegisterInterfaceStrings();
 
  ErrorCode GenerateInterfaceString(uint8_t string_index, ConstRawData& data);
 
  bool configured_ =
      false;  
  bool ep_assigned_ = false;  
 
  EndpointPool& endpoint_pool_;  
  uint8_t current_cfg_ = 0;      
  uint8_t i_configuration_ = 0;  
  uint8_t bm_attributes_ = CFG_BUS_POWERED;  
 
  const bool composite_ = false;     
  const size_t config_num_ = 0;      
  ConfigItems* items_ = nullptr;     
  DeviceClass** classes_ = nullptr;  
  size_t class_count_ = 0;           
 
  DescriptorStrings strings_;  
  const char** interface_strings_ =
      nullptr;  
  size_t interface_string_count_ =
      0;  
  RawData interface_string_buffer_{
      nullptr, 0};  
  BosManager bos_;  
  ConfigDescriptor config_desc_;  
};
 
}  // namespace LibXR::USB