// Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "mist/usb_config_descriptor.h"

#include <libusb.h>

#include <base/logging.h>
#include <base/strings/stringprintf.h>

#include "mist/usb_device.h"
#include "mist/usb_interface.h"

using base::StringPrintf;
using std::ostream;
using std::string;

namespace mist {

UsbConfigDescriptor::UsbConfigDescriptor(
    const base::WeakPtr<UsbDevice>& device,
    libusb_config_descriptor* config_descriptor,
    bool own_config_descriptor)
    : device_(device),
      config_descriptor_(config_descriptor),
      own_config_descriptor_(own_config_descriptor) {
  CHECK(config_descriptor_);
}

UsbConfigDescriptor::~UsbConfigDescriptor() {
  if (own_config_descriptor_) {
    libusb_free_config_descriptor(config_descriptor_);
    config_descriptor_ = nullptr;
  }
}

uint8_t UsbConfigDescriptor::GetLength() const {
  return config_descriptor_->bLength;
}

uint8_t UsbConfigDescriptor::GetDescriptorType() const {
  return config_descriptor_->bDescriptorType;
}

uint16_t UsbConfigDescriptor::GetTotalLength() const {
  return config_descriptor_->wTotalLength;
}

uint8_t UsbConfigDescriptor::GetNumInterfaces() const {
  return config_descriptor_->bNumInterfaces;
}

uint8_t UsbConfigDescriptor::GetConfigurationValue() const {
  return config_descriptor_->bConfigurationValue;
}

string UsbConfigDescriptor::GetConfigurationDescription() const {
  return device_ ?
      device_->GetStringDescriptorAscii(config_descriptor_->iConfiguration) :
      string();
}

uint8_t UsbConfigDescriptor::GetAttributes() const {
  return config_descriptor_->bmAttributes;
}

uint8_t UsbConfigDescriptor::GetMaxPower() const {
  return config_descriptor_->MaxPower;
}

UsbInterface* UsbConfigDescriptor::GetInterface(uint8_t index) const {
  if (index >= GetNumInterfaces()) {
    LOG(ERROR) << StringPrintf("Invalid interface index %d. "
                               "Must be less than %d.",
                               index, GetNumInterfaces());
    return nullptr;
  }

  return new UsbInterface(device_, &config_descriptor_->interface[index]);
}

string UsbConfigDescriptor::ToString() const {
  return StringPrintf("Configuration (Length=%u, "
                      "DescriptorType=%u, "
                      "TotalLength=%u, "
                      "NumInterfaces=%u, "
                      "ConfigurationValue=%u, "
                      "Configuration='%s', "
                      "Attributes=0x%02x, "
                      "MaxPower=%u)",
                      GetLength(),
                      GetDescriptorType(),
                      GetTotalLength(),
                      GetNumInterfaces(),
                      GetConfigurationValue(),
                      GetConfigurationDescription().c_str(),
                      GetAttributes(),
                      GetMaxPower());
}

}  // namespace mist

ostream& operator<<(ostream& stream,
                    const mist::UsbConfigDescriptor& config_descriptor) {
  stream << config_descriptor.ToString();
  return stream;
}