usb_rename.cpp

#include "usb_rename.h"

static
bool SendControl(u8 d)
{
  return USB_SendControl(0, &d, 1) == 1;
}

// Send a USB descriptor string. The string is stored in PROGMEM as a
// plain ASCII string but is sent out as UTF-16 with the correct 2-byte
// prefix
// taken from Arduino's USBCore.cpp
static bool USB_SendStringDescriptor(const char*string_P, size_t string_len, uint8_t flags) {
        SendControl(2 + string_len * 2);
        SendControl(3);
        bool pgm = flags & TRANSFER_PGM;
        for(size_t i = 0; i < string_len; i++) {
                bool r = SendControl(pgm ? pgm_read_byte(&string_P[i]) : (u8)string_P[i]);
                r &= SendControl(0); // high byte
                if(!r) {
                        return false;
                }
        }
        return true;
}

USBRename::USBRename(
      const char *product_name,
      const char *manufacturer_name,
      const char *serial_num) :
        manufacturer_name(manufacturer_name),
        product_name(product_name),
        serial_num(serial_num),
        PluggableUSBModule(1, 1, epType) {
    PluggableUSB().plug(this);
}

int USBRename::getInterface(uint8_t* interfaceCount) {
    return 0;
}

int USBRename::getDescriptor(USBSetup& setup)
{
    if(setup.wValueH != USB_STRING_DESCRIPTOR_TYPE) {
        return 0;
    }

    if(setup.wValueL == IMANUFACTURER && manufacturer_name) {
        return USB_SendStringDescriptor(manufacturer_name, strlen(manufacturer_name), 0);
    } else if (setup.wValueL == IPRODUCT && product_name) {
        return USB_SendStringDescriptor(product_name, strlen(product_name), 0);
    } else if (setup.wValueL == ISERIAL && serial_num) {
        return USB_SendStringDescriptor(serial_num, strlen(serial_num), 0);
    }

    return 0;
}

bool USBRename::setup(USBSetup& setup)
{
    return false;
}