You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
339 lines
9.4 KiB
339 lines
9.4 KiB
/*
|
|
* downstream_statemachine.c
|
|
*
|
|
* Created on: 2/08/2015
|
|
* Author: Robert Fisk
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
|
|
#include "downstream_statemachine.h"
|
|
#include "downstream_interface_def.h"
|
|
#include "downstream_spi.h"
|
|
#include "downstream_msc.h"
|
|
#include "downstream_hid.h"
|
|
#include "usbh_core.h"
|
|
#include "usbh_msc.h"
|
|
#include "usbh_hid.h"
|
|
#include "led.h"
|
|
#include "build_config.h"
|
|
|
|
|
|
DownstreamStateTypeDef DownstreamState = STATE_DEVICE_NOT_READY;
|
|
InterfaceCommandClassTypeDef ConfiguredDeviceClass = COMMAND_CLASS_INTERFACE;
|
|
uint8_t NotifyDisconnectReply = 0;
|
|
|
|
|
|
static void Downstream_PacketProcessor_Interface(DownstreamPacketTypeDef* receivedPacket);
|
|
static void Downstream_PacketProcessor_Interface_ReplyNotifyDevice(DownstreamPacketTypeDef* replyPacket);
|
|
static void Downstream_PacketProcessor_NotifyDisconnectReply(DownstreamPacketTypeDef* packetToSend);
|
|
|
|
|
|
#ifdef CONFIG_USB_ID_ENABLED
|
|
|
|
//usb device id def
|
|
typedef struct usb_id{
|
|
uint16_t vendor;
|
|
uint16_t product;
|
|
}usb_device_id;
|
|
|
|
#define USB_ID_COUNT 2
|
|
|
|
usb_device_id list[USB_ID_COUNT]={
|
|
{0x046d,0xc077},//usb vendor,usb product, add your device to array and edit USB_ID_COUNT
|
|
// {0x046d,0xc0ff} //bad device test
|
|
{0x12c9,0x1017} //new mouse Fury
|
|
};
|
|
#endif
|
|
|
|
void Downstream_InitStateMachine(void)
|
|
{
|
|
if ((DownstreamState != STATE_DEVICE_NOT_READY) ||
|
|
(ConfiguredDeviceClass != COMMAND_CLASS_INTERFACE))
|
|
{
|
|
DOWNSTREAM_STATEMACHINE_FREAKOUT;
|
|
return;
|
|
}
|
|
|
|
Downstream_InitSPI();
|
|
|
|
//Prepare to receive our first packet from Upstream!
|
|
Downstream_ReceivePacket(Downstream_PacketProcessor);
|
|
}
|
|
|
|
|
|
|
|
void Downstream_PacketProcessor(DownstreamPacketTypeDef* receivedPacket)
|
|
{
|
|
if (DownstreamState >= STATE_ERROR)
|
|
{
|
|
Downstream_ReleasePacket(receivedPacket);
|
|
return;
|
|
}
|
|
|
|
//Device not connected yet
|
|
if (DownstreamState != STATE_ACTIVE)
|
|
{
|
|
if (receivedPacket->CommandClass == COMMAND_CLASS_INTERFACE)
|
|
{
|
|
Downstream_PacketProcessor_Interface(receivedPacket);
|
|
}
|
|
else
|
|
{
|
|
//If we get a class-specific message when our device is disconnected,
|
|
//we need to tell Upstream of the fact (and not freak out).
|
|
Downstream_PacketProcessor_NotifyDisconnectReply(receivedPacket);
|
|
}
|
|
return;
|
|
}
|
|
|
|
//Device is connected: expect only class-specific messages,
|
|
//and only to our currently active device class.
|
|
if ((ConfiguredDeviceClass == COMMAND_CLASS_INTERFACE) ||
|
|
(ConfiguredDeviceClass != receivedPacket->CommandClass))
|
|
{
|
|
DOWNSTREAM_STATEMACHINE_FREAKOUT;
|
|
return;
|
|
}
|
|
|
|
switch (ConfiguredDeviceClass)
|
|
{
|
|
|
|
#ifdef CONFIG_MASS_STORAGE_ENABLED
|
|
case COMMAND_CLASS_MASS_STORAGE:
|
|
Downstream_MSC_PacketProcessor(receivedPacket);
|
|
break;
|
|
#endif
|
|
#ifdef CONFIG_MOUSE_ENABLED
|
|
case COMMAND_CLASS_HID_MOUSE:
|
|
Downstream_HID_PacketProcessor(receivedPacket);
|
|
break;
|
|
#endif
|
|
#ifdef CONFIG_KEYBOARD_ENABLED
|
|
case COMMAND_CLASS_HID_KEYBOARD:
|
|
Downstream_HID_PacketProcessor(receivedPacket);
|
|
break;
|
|
#endif
|
|
|
|
//Add other classes here...
|
|
|
|
default:
|
|
DOWNSTREAM_STATEMACHINE_FREAKOUT;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
//Used by downstream class interfaces, and SPI interface
|
|
void Downstream_PacketProcessor_FreakOut(void)
|
|
{
|
|
DOWNSTREAM_STATEMACHINE_FREAKOUT;
|
|
}
|
|
|
|
|
|
|
|
void Downstream_PacketProcessor_Interface(DownstreamPacketTypeDef* receivedPacket)
|
|
{
|
|
switch (receivedPacket->Command)
|
|
{
|
|
case COMMAND_INTERFACE_ECHO:
|
|
Downstream_TransmitPacket(receivedPacket);
|
|
Downstream_ReceivePacket(Downstream_PacketProcessor);
|
|
break;
|
|
|
|
case COMMAND_INTERFACE_NOTIFY_DEVICE:
|
|
if (DownstreamState == STATE_DEVICE_READY)
|
|
{
|
|
Downstream_PacketProcessor_Interface_ReplyNotifyDevice(receivedPacket);
|
|
return;
|
|
}
|
|
|
|
if (DownstreamState == STATE_DEVICE_NOT_READY)
|
|
{
|
|
DownstreamState = STATE_WAIT_DEVICE_READY;
|
|
Downstream_ReleasePacket(receivedPacket);
|
|
return;
|
|
}
|
|
DOWNSTREAM_STATEMACHINE_FREAKOUT;
|
|
break;
|
|
|
|
|
|
default:
|
|
DOWNSTREAM_STATEMACHINE_FREAKOUT;
|
|
}
|
|
}
|
|
|
|
|
|
void Downstream_PacketProcessor_Interface_ReplyNotifyDevice(DownstreamPacketTypeDef* replyPacket)
|
|
{
|
|
replyPacket->Length16 = DOWNSTREAM_PACKET_HEADER_LEN_16 + 1;
|
|
replyPacket->CommandClass = COMMAND_CLASS_INTERFACE;
|
|
replyPacket->Command = COMMAND_INTERFACE_NOTIFY_DEVICE;
|
|
replyPacket->Data[0] = ConfiguredDeviceClass;
|
|
|
|
if (Downstream_TransmitPacket(replyPacket) == HAL_OK)
|
|
{
|
|
DownstreamState = STATE_ACTIVE;
|
|
Downstream_ReceivePacket(Downstream_PacketProcessor);
|
|
}
|
|
}
|
|
|
|
|
|
void Downstream_PacketProcessor_GenericErrorReply(DownstreamPacketTypeDef* replyPacket)
|
|
{
|
|
replyPacket->Length16 = DOWNSTREAM_PACKET_HEADER_LEN_16;
|
|
replyPacket->CommandClass = COMMAND_CLASS_ERROR;
|
|
replyPacket->Command = COMMAND_ERROR_GENERIC;
|
|
|
|
Downstream_TransmitPacket(replyPacket);
|
|
Downstream_ReceivePacket(Downstream_PacketProcessor);
|
|
NotifyDisconnectReply = 0;
|
|
}
|
|
|
|
|
|
void Downstream_PacketProcessor_ClassReply(DownstreamPacketTypeDef* replyPacket)
|
|
{
|
|
Downstream_TransmitPacket(replyPacket);
|
|
Downstream_ReceivePacket(Downstream_PacketProcessor);
|
|
NotifyDisconnectReply = 0;
|
|
}
|
|
|
|
|
|
void Downstream_PacketProcessor_NotifyDisconnectReplyRequired(void)
|
|
{
|
|
NotifyDisconnectReply = 1;
|
|
}
|
|
|
|
|
|
void Downstream_PacketProcessor_CheckNotifyDisconnectReply(void)
|
|
{
|
|
if (NotifyDisconnectReply == 2)
|
|
{
|
|
Downstream_GetFreePacket(Downstream_PacketProcessor_NotifyDisconnectReply);
|
|
}
|
|
}
|
|
|
|
|
|
void Downstream_PacketProcessor_NotifyDisconnectReply(DownstreamPacketTypeDef* packetToSend)
|
|
{
|
|
packetToSend->Length16 = DOWNSTREAM_PACKET_HEADER_LEN_16;
|
|
packetToSend->CommandClass = COMMAND_CLASS_ERROR;
|
|
packetToSend->Command = COMMAND_ERROR_DEVICE_DISCONNECTED;
|
|
Downstream_PacketProcessor_ClassReply(packetToSend);
|
|
}
|
|
|
|
|
|
//This callback receives various event ids from the host stack,
|
|
//either at INT_PRIORITY_OTG_FS or from main().
|
|
void Downstream_HostUserCallback(USBH_HandleTypeDef *phost, uint8_t id)
|
|
{
|
|
InterfaceCommandClassTypeDef newActiveClass = COMMAND_CLASS_INTERFACE;
|
|
|
|
if (DownstreamState >= STATE_ERROR)
|
|
{
|
|
return;
|
|
}
|
|
|
|
//Called from USB interrupt.
|
|
//Simple function shouldn't need to worry about preempting anything important.
|
|
if (id == HOST_USER_DISCONNECTION)
|
|
{
|
|
DownstreamState = STATE_DEVICE_NOT_READY;
|
|
if (NotifyDisconnectReply == 1)
|
|
{
|
|
NotifyDisconnectReply = 2; //Request a 'device disconnected' reply when we get back to main()
|
|
}
|
|
return;
|
|
}
|
|
|
|
|
|
//Called from main()
|
|
if (id == HOST_USER_CLASS_ACTIVE)
|
|
{
|
|
#ifdef CONFIG_USB_ID_ENABLED
|
|
//check vendor product id
|
|
for(int i=0;i<USB_ID_COUNT;i++){
|
|
if(list[i].vendor==phost->device.DevDesc.idVendor && list[i].product==phost->device.DevDesc.idProduct){
|
|
break;
|
|
}else{
|
|
if(i==(USB_ID_COUNT-1)){//last usb device in list
|
|
DOWNSTREAM_STATEMACHINE_FREAKOUT;
|
|
}
|
|
}
|
|
}
|
|
#endif
|
|
|
|
switch (phost->pActiveClass->ClassCode)
|
|
{
|
|
#ifdef CONFIG_MASS_STORAGE_ENABLED
|
|
case USB_MSC_CLASS:
|
|
newActiveClass = Downstream_MSC_ApproveConnectedDevice();
|
|
break;
|
|
#endif
|
|
#if defined (CONFIG_KEYBOARD_ENABLED) || defined (CONFIG_MOUSE_ENABLED)
|
|
case USB_HID_CLASS:
|
|
newActiveClass = Downstream_HID_ApproveConnectedDevice();
|
|
break;
|
|
#endif
|
|
|
|
//Add other classes here...
|
|
|
|
|
|
}
|
|
|
|
//Unsupported device classes will cause a slow fault flash.
|
|
//This is distinct from the fast freakout flash caused by internal errors or attacks.
|
|
//We consider supported classes that fail their approval checks to also be unsupported devices.
|
|
if (newActiveClass == COMMAND_CLASS_INTERFACE)
|
|
{
|
|
USB_Host_Disconnect();
|
|
LED_SetState(LED_STATUS_FLASH_UNSUPPORTED);
|
|
DownstreamState = STATE_ERROR;
|
|
return;
|
|
}
|
|
|
|
//If we already configured a device class, we cannot change to a different one without rebooting.
|
|
//This blocks 'hidden device' BadUSB attacks.
|
|
if ((ConfiguredDeviceClass != COMMAND_CLASS_INTERFACE) &&
|
|
(ConfiguredDeviceClass != newActiveClass))
|
|
{
|
|
DOWNSTREAM_STATEMACHINE_FREAKOUT;
|
|
return;
|
|
}
|
|
ConfiguredDeviceClass = newActiveClass;
|
|
|
|
if (DownstreamState == STATE_WAIT_DEVICE_READY)
|
|
{
|
|
Downstream_GetFreePacket(Downstream_PacketProcessor_Interface_ReplyNotifyDevice);
|
|
return;
|
|
}
|
|
|
|
if (DownstreamState == STATE_DEVICE_NOT_READY)
|
|
{
|
|
DownstreamState = STATE_DEVICE_READY;
|
|
return;
|
|
}
|
|
|
|
DOWNSTREAM_STATEMACHINE_FREAKOUT;
|
|
return;
|
|
}
|
|
|
|
|
|
//Called from main():
|
|
if ((id == HOST_USER_CLASS_FAILED) ||
|
|
(id == HOST_USER_UNRECOVERED_ERROR)) //Probably due to a crappy device that won't enumerate!
|
|
{
|
|
//Unsupported device classes will cause a slow fault flash.
|
|
//This is distinct from the fast freakout flash caused by internal errors or attacks.
|
|
USB_Host_Disconnect();
|
|
LED_SetState(LED_STATUS_FLASH_UNSUPPORTED);
|
|
DownstreamState = STATE_ERROR;
|
|
return;
|
|
}
|
|
|
|
}
|
|
|