Add {GET,SET}_USB_BUFFER_UINT* and use them wherever possible. Get rid of SetUsb*() functions.

This commit is contained in:
László Monda
2017-11-05 22:09:34 +01:00
parent 58778feb5b
commit a8bde4e007
16 changed files with 64 additions and 64 deletions

View File

@@ -12,11 +12,13 @@ void UsbCommand_ApplyConfig(void)
ParserRunDry = true; ParserRunDry = true;
StagingUserConfigBuffer.offset = 0; StagingUserConfigBuffer.offset = 0;
GenericHidOutBuffer[0] = ParseConfig(&StagingUserConfigBuffer); uint8_t parseConfigStatus = ParseConfig(&StagingUserConfigBuffer);
*(uint16_t*)(GenericHidOutBuffer+1) = StagingUserConfigBuffer.offset;
GenericHidOutBuffer[3] = 0;
if (GenericHidOutBuffer[0] != UsbStatusCode_Success) { SET_USB_BUFFER_UINT8(0, parseConfigStatus);
SET_USB_BUFFER_UINT16(1, StagingUserConfigBuffer.offset);
SET_USB_BUFFER_UINT8(3, 0);
if (parseConfigStatus != UsbStatusCode_Success) {
return; return;
} }
@@ -33,11 +35,13 @@ void UsbCommand_ApplyConfig(void)
ParserRunDry = false; ParserRunDry = false;
ValidatedUserConfigBuffer.offset = 0; ValidatedUserConfigBuffer.offset = 0;
GenericHidOutBuffer[0] = ParseConfig(&ValidatedUserConfigBuffer); parseConfigStatus = ParseConfig(&ValidatedUserConfigBuffer);
*(uint16_t*)(GenericHidOutBuffer+1) = ValidatedUserConfigBuffer.offset;
GenericHidOutBuffer[3] = 1;
if (GenericHidOutBuffer[0] != UsbStatusCode_Success) { SET_USB_BUFFER_UINT8(0, parseConfigStatus);
SET_USB_BUFFER_UINT16(1, ValidatedUserConfigBuffer.offset);
SET_USB_BUFFER_UINT8(3, 1);
if (parseConfigStatus != UsbStatusCode_Success) {
return; return;
} }

View File

@@ -4,5 +4,5 @@
void UsbCommand_GetAdcValue(void) void UsbCommand_GetAdcValue(void)
{ {
*(uint32_t*)(GenericHidOutBuffer+1) = ADC_Measure(); SET_USB_BUFFER_UINT32(1, ADC_Measure());
} }

View File

@@ -7,9 +7,9 @@
void UsbCommand_GetKeyboardState(void) void UsbCommand_GetKeyboardState(void)
{ {
GenericHidOutBuffer[1] = IsEepromBusy; SET_USB_BUFFER_UINT8(1, IsEepromBusy);
GenericHidOutBuffer[2] = MERGE_SENSOR_IS_MERGED; SET_USB_BUFFER_UINT8(2, MERGE_SENSOR_IS_MERGED);
GenericHidOutBuffer[3] = UhkModuleStates[UhkModuleDriverId_LeftKeyboardHalf].moduleId; SET_USB_BUFFER_UINT8(3, UhkModuleStates[UhkModuleDriverId_LeftKeyboardHalf].moduleId);
GenericHidOutBuffer[4] = UhkModuleStates[UhkModuleDriverId_LeftAddon].moduleId; SET_USB_BUFFER_UINT8(4, UhkModuleStates[UhkModuleDriverId_LeftAddon].moduleId);
GenericHidOutBuffer[5] = UhkModuleStates[UhkModuleDriverId_RightAddon].moduleId; SET_USB_BUFFER_UINT8(5, UhkModuleStates[UhkModuleDriverId_RightAddon].moduleId);
} }

View File

@@ -5,29 +5,29 @@
void UsbCommand_GetProperty(void) void UsbCommand_GetProperty(void)
{ {
uint8_t propertyId = GenericHidInBuffer[1]; uint8_t propertyId = GET_USB_BUFFER_UINT8(1);
switch (propertyId) { switch (propertyId) {
case SystemPropertyId_UsbProtocolVersion: case SystemPropertyId_UsbProtocolVersion:
SetUsbResponseByte(SYSTEM_PROPERTY_USB_PROTOCOL_VERSION); SET_USB_BUFFER_UINT8(1, SYSTEM_PROPERTY_USB_PROTOCOL_VERSION);
break; break;
case SystemPropertyId_BridgeProtocolVersion: case SystemPropertyId_BridgeProtocolVersion:
SetUsbResponseByte(SYSTEM_PROPERTY_BRIDGE_PROTOCOL_VERSION); SET_USB_BUFFER_UINT8(1, SYSTEM_PROPERTY_BRIDGE_PROTOCOL_VERSION);
break; break;
case SystemPropertyId_DataModelVersion: case SystemPropertyId_DataModelVersion:
SetUsbResponseByte(SYSTEM_PROPERTY_DATA_MODEL_VERSION); SET_USB_BUFFER_UINT8(1, SYSTEM_PROPERTY_DATA_MODEL_VERSION);
break; break;
case SystemPropertyId_FirmwareVersion: case SystemPropertyId_FirmwareVersion:
SetUsbResponseByte(SYSTEM_PROPERTY_FIRMWARE_VERSION); SET_USB_BUFFER_UINT8(1, SYSTEM_PROPERTY_FIRMWARE_VERSION);
break; break;
case SystemPropertyId_HardwareConfigSize: case SystemPropertyId_HardwareConfigSize:
SetUsbResponseWord(HARDWARE_CONFIG_SIZE); SET_USB_BUFFER_UINT16(1, HARDWARE_CONFIG_SIZE);
break; break;
case SystemPropertyId_UserConfigSize: case SystemPropertyId_UserConfigSize:
SetUsbResponseWord(USER_CONFIG_SIZE); SET_USB_BUFFER_UINT16(1, USER_CONFIG_SIZE);
break; break;
default: default:
SetUsbStatusCode(1); SET_USB_BUFFER_UINT8(0, UsbStatusCode_GetProperty_InvalidProperty);
break; break;
} }
} }

View File

@@ -19,6 +19,10 @@
SystemPropertyId_UserConfigSize = 5, SystemPropertyId_UserConfigSize = 5,
} system_property_t; } system_property_t;
typedef enum {
UsbStatusCode_GetProperty_InvalidProperty = 2,
} usb_status_code_get_property_t;
// Functions: // Functions:
void UsbCommand_GetProperty(void); void UsbCommand_GetProperty(void);

View File

@@ -4,10 +4,10 @@
void UsbCommand_JumpToSlaveBootloader(void) void UsbCommand_JumpToSlaveBootloader(void)
{ {
uint8_t uhkModuleDriverId = GenericHidInBuffer[1]; uint8_t uhkModuleDriverId = GET_USB_BUFFER_UINT8(1);
if (uhkModuleDriverId >= UHK_MODULE_MAX_COUNT) { if (uhkModuleDriverId >= UHK_MODULE_MAX_COUNT) {
SetUsbStatusCode(UsbStatusCode_JumpToSlaveBootloader_InvalidModuleDriverId); SET_USB_BUFFER_UINT8(0, UsbStatusCode_JumpToSlaveBootloader_InvalidModuleDriverId);
return; return;
} }

View File

@@ -9,6 +9,6 @@
typedef enum { typedef enum {
UsbStatusCode_JumpToSlaveBootloader_InvalidModuleDriverId = 2, UsbStatusCode_JumpToSlaveBootloader_InvalidModuleDriverId = 2,
} config_transfer_response_jump_to_slave_bootloader_t; } usb_status_code_jump_to_slave_bootloader_t;
#endif #endif

View File

@@ -5,7 +5,7 @@
void UsbCommand_LaunchEepromTransferLegacy(void) void UsbCommand_LaunchEepromTransferLegacy(void)
{ {
uint8_t legacyEepromTransferId = GenericHidInBuffer[1]; uint8_t legacyEepromTransferId = GET_USB_BUFFER_UINT8(1);
switch (legacyEepromTransferId) { switch (legacyEepromTransferId) {
case 0: case 0:
EEPROM_LaunchTransfer(EepromOperation_Read, ConfigBufferId_HardwareConfig, NULL); EEPROM_LaunchTransfer(EepromOperation_Read, ConfigBufferId_HardwareConfig, NULL);

View File

@@ -5,11 +5,11 @@
void UsbCommand_ReadConfig(bool isHardware) void UsbCommand_ReadConfig(bool isHardware)
{ {
uint8_t length = GenericHidInBuffer[1]; uint8_t length = GET_USB_BUFFER_UINT8(1);
uint16_t offset = *(uint16_t*)(GenericHidInBuffer+2); uint16_t offset = GET_USB_BUFFER_UINT16(2);
if (length > USB_GENERIC_HID_OUT_BUFFER_LENGTH-1) { if (length > USB_GENERIC_HID_OUT_BUFFER_LENGTH-1) {
SetUsbStatusCode(UsbStatusCode_TransferConfig_LengthTooLarge); SET_USB_BUFFER_UINT8(0, UsbStatusCode_TransferConfig_LengthTooLarge);
return; return;
} }
@@ -17,7 +17,7 @@ void UsbCommand_ReadConfig(bool isHardware)
uint16_t bufferLength = isHardware ? HARDWARE_CONFIG_SIZE : USER_CONFIG_SIZE; uint16_t bufferLength = isHardware ? HARDWARE_CONFIG_SIZE : USER_CONFIG_SIZE;
if (offset + length > bufferLength) { if (offset + length > bufferLength) {
SetUsbStatusCode(UsbStatusCode_TransferConfig_BufferOutOfBounds); SET_USB_BUFFER_UINT8(0, UsbStatusCode_TransferConfig_BufferOutOfBounds);
return; return;
} }

View File

@@ -6,7 +6,7 @@
void UsbCommand_Reenumerate(void) void UsbCommand_Reenumerate(void)
{ {
Wormhole.magicNumber = WORMHOLE_MAGIC_NUMBER; Wormhole.magicNumber = WORMHOLE_MAGIC_NUMBER;
Wormhole.enumerationMode = GenericHidInBuffer[1]; Wormhole.enumerationMode = GET_USB_BUFFER_UINT8(1);
Wormhole.timeoutMs = *((uint32_t*)(GenericHidInBuffer+2)); Wormhole.timeoutMs = GET_USB_BUFFER_UINT32(2);
NVIC_SystemReset(); NVIC_SystemReset();
} }

View File

@@ -4,6 +4,6 @@
void UsbCommand_SendKbootCommand(void) void UsbCommand_SendKbootCommand(void)
{ {
KbootDriverState.phase = 0; KbootDriverState.phase = 0;
KbootDriverState.i2cAddress = GenericHidInBuffer[2]; KbootDriverState.i2cAddress = GET_USB_BUFFER_UINT8(2);
KbootDriverState.commandType = GenericHidInBuffer[1]; // Command should be set last. KbootDriverState.commandType = GET_USB_BUFFER_UINT8(1); // Command should be set last.
} }

View File

@@ -4,7 +4,7 @@
void UsbCommand_SetLedPwmBrightness(void) void UsbCommand_SetLedPwmBrightness(void)
{ {
uint8_t brightnessPercent = GenericHidInBuffer[1]; uint8_t brightnessPercent = GET_USB_BUFFER_UINT8(1);
LedPwm_SetBrightness(brightnessPercent); LedPwm_SetBrightness(brightnessPercent);
UhkModuleStates[UhkModuleDriverId_LeftKeyboardHalf].sourceVars.ledPwmBrightness = brightnessPercent; UhkModuleStates[UhkModuleDriverId_LeftKeyboardHalf].sourceVars.ledPwmBrightness = brightnessPercent;
} }

View File

@@ -5,7 +5,7 @@
void UsbCommand_SetTestLed(void) void UsbCommand_SetTestLed(void)
{ {
uint8_t ledState = GenericHidInBuffer[1]; uint8_t ledState = GET_USB_BUFFER_UINT8(1);
TEST_LED_SET(ledState); TEST_LED_SET(ledState);
UhkModuleStates[UhkModuleDriverId_LeftKeyboardHalf].sourceVars.isTestLedOn = ledState; UhkModuleStates[UhkModuleDriverId_LeftKeyboardHalf].sourceVars.isTestLedOn = ledState;
} }

View File

@@ -5,11 +5,11 @@
void UsbCommand_WriteConfig(bool isHardware) void UsbCommand_WriteConfig(bool isHardware)
{ {
uint8_t length = GenericHidInBuffer[1]; uint8_t length = GET_USB_BUFFER_UINT8(1);
uint16_t offset = *((uint16_t*)(GenericHidInBuffer+1+1)); uint16_t offset = GET_USB_BUFFER_UINT16(2);
if (length > USB_GENERIC_HID_OUT_BUFFER_LENGTH-1-1-2) { if (length > USB_GENERIC_HID_OUT_BUFFER_LENGTH-4) {
SetUsbStatusCode(UsbStatusCode_TransferConfig_LengthTooLarge); SET_USB_BUFFER_UINT8(0, UsbStatusCode_TransferConfig_LengthTooLarge);
return; return;
} }
@@ -17,9 +17,9 @@ void UsbCommand_WriteConfig(bool isHardware)
uint16_t bufferLength = isHardware ? HARDWARE_CONFIG_SIZE : USER_CONFIG_SIZE; uint16_t bufferLength = isHardware ? HARDWARE_CONFIG_SIZE : USER_CONFIG_SIZE;
if (offset + length > bufferLength) { if (offset + length > bufferLength) {
SetUsbStatusCode(UsbStatusCode_TransferConfig_BufferOutOfBounds); SET_USB_BUFFER_UINT8(0, UsbStatusCode_TransferConfig_BufferOutOfBounds);
return; return;
} }
memcpy(buffer+offset, GenericHidInBuffer+1+1+2, length); memcpy(buffer+offset, GenericHidInBuffer+4, length);
} }

View File

@@ -13,25 +13,10 @@
#include "usb_commands/usb_command_jump_to_slave_bootloader.h" #include "usb_commands/usb_command_jump_to_slave_bootloader.h"
#include "usb_commands/usb_command_send_kboot_command.h" #include "usb_commands/usb_command_send_kboot_command.h"
void SetUsbStatusCode(uint8_t error)
{
GenericHidOutBuffer[0] = error;
}
void SetUsbResponseByte(uint8_t response)
{
GenericHidOutBuffer[1] = response;
}
void SetUsbResponseWord(uint16_t response)
{
*((uint16_t*)(GenericHidOutBuffer+1)) = response;
}
void UsbProtocolHandler(void) void UsbProtocolHandler(void)
{ {
bzero(GenericHidOutBuffer, USB_GENERIC_HID_OUT_BUFFER_LENGTH); bzero(GenericHidOutBuffer, USB_GENERIC_HID_OUT_BUFFER_LENGTH);
uint8_t command = GenericHidInBuffer[0]; uint8_t command = GET_USB_BUFFER_UINT8(0);
switch (command) { switch (command) {
case UsbCommandId_GetProperty: case UsbCommandId_GetProperty:
UsbCommand_GetProperty(); UsbCommand_GetProperty();
@@ -79,7 +64,7 @@ void UsbProtocolHandler(void)
UsbCommand_SendKbootCommand(); UsbCommand_SendKbootCommand();
break; break;
default: default:
SetUsbStatusCode(UsbStatusCode_InvalidCommand); SET_USB_BUFFER_UINT8(0, UsbStatusCode_InvalidCommand);
break; break;
} }
} }

View File

@@ -6,6 +6,16 @@
#include "fsl_common.h" #include "fsl_common.h"
#include "usb_interfaces/usb_interface_generic_hid.h" #include "usb_interfaces/usb_interface_generic_hid.h"
// Macros:
#define GET_USB_BUFFER_UINT8(offset) (*(uint8_t*)(GenericHidInBuffer+offset))
#define GET_USB_BUFFER_UINT16(offset) (*(uint16_t*)(GenericHidInBuffer+offset))
#define GET_USB_BUFFER_UINT32(offset) (*(uint32_t*)(GenericHidInBuffer+offset))
#define SET_USB_BUFFER_UINT8(offset, value) (*(uint8_t*)(GenericHidOutBuffer+(offset)) = (value))
#define SET_USB_BUFFER_UINT16(offset, value) (*(uint16_t*)(GenericHidOutBuffer+(offset)) = (value))
#define SET_USB_BUFFER_UINT32(offset, value) (*(uint32_t*)(GenericHidOutBuffer+(offset)) = (value))
// Typedefs: // Typedefs:
typedef enum { typedef enum {
@@ -29,18 +39,15 @@
typedef enum { typedef enum {
UsbStatusCode_Success = 0, UsbStatusCode_Success = 0,
UsbStatusCode_InvalidCommand = 1, UsbStatusCode_InvalidCommand = 1,
} usb_response_t; } usb_status_code_general_t;
typedef enum { typedef enum {
UsbStatusCode_TransferConfig_LengthTooLarge = 2, UsbStatusCode_TransferConfig_LengthTooLarge = 2,
UsbStatusCode_TransferConfig_BufferOutOfBounds = 3, UsbStatusCode_TransferConfig_BufferOutOfBounds = 3,
} config_transfer_response_t; } usb_status_code_transfer_config_t;
// Functions: // Functions:
void UsbProtocolHandler(void); void UsbProtocolHandler(void);
void SetUsbStatusCode(uint8_t status);
void SetUsbResponseByte(uint8_t response);
void SetUsbResponseWord(uint16_t response);
#endif #endif