mirror of
				https://github.com/mfulz/qmk_firmware.git
				synced 2025-11-04 07:12:33 +01:00 
			
		
		
		
	Add guard to the HID Host Class driver SetProtocol command, to ensure that the device supports boot protocol mode before issuing the request.
		
			
				
	
	
		
			332 lines
		
	
	
		
			8.4 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			332 lines
		
	
	
		
			8.4 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
             LUFA Library
 | 
						|
     Copyright (C) Dean Camera, 2009.
 | 
						|
              
 | 
						|
  dean [at] fourwalledcubicle [dot] com
 | 
						|
      www.fourwalledcubicle.com
 | 
						|
*/
 | 
						|
 | 
						|
/*
 | 
						|
  Copyright 2009  Dean Camera (dean [at] fourwalledcubicle [dot] com)
 | 
						|
 | 
						|
  Permission to use, copy, modify, and distribute this software
 | 
						|
  and its documentation for any purpose and without fee is hereby
 | 
						|
  granted, provided that the above copyright notice appear in all
 | 
						|
  copies and that both that the copyright notice and this
 | 
						|
  permission notice and warranty disclaimer appear in supporting
 | 
						|
  documentation, and that the name of the author not be used in
 | 
						|
  advertising or publicity pertaining to distribution of the
 | 
						|
  software without specific, written prior permission.
 | 
						|
 | 
						|
  The author disclaim all warranties with regard to this
 | 
						|
  software, including all implied warranties of merchantability
 | 
						|
  and fitness.  In no event shall the author be liable for any
 | 
						|
  special, indirect or consequential damages or any damages
 | 
						|
  whatsoever resulting from loss of use, data or profits, whether
 | 
						|
  in an action of contract, negligence or other tortious action,
 | 
						|
  arising out of or in connection with the use or performance of
 | 
						|
  this software.
 | 
						|
*/
 | 
						|
 | 
						|
#include "../HighLevel/USBMode.h"
 | 
						|
 | 
						|
#if defined(USB_CAN_BE_HOST)
 | 
						|
 | 
						|
#define  INCLUDE_FROM_HOST_C
 | 
						|
#include "Host.h"
 | 
						|
 | 
						|
void USB_Host_ProcessNextHostState(void)
 | 
						|
{
 | 
						|
	uint8_t ErrorCode    = HOST_ENUMERROR_NoError;
 | 
						|
	uint8_t SubErrorCode = HOST_ENUMERROR_NoError;
 | 
						|
 | 
						|
	static uint16_t WaitMSRemaining;
 | 
						|
	static uint8_t  PostWaitState;
 | 
						|
 | 
						|
	switch (USB_HostState)
 | 
						|
	{
 | 
						|
		case HOST_STATE_WaitForDevice:
 | 
						|
			if (WaitMSRemaining)
 | 
						|
			{
 | 
						|
				if ((SubErrorCode = USB_Host_WaitMS(1)) != HOST_WAITERROR_Successful)
 | 
						|
				{
 | 
						|
					USB_HostState = PostWaitState;
 | 
						|
					ErrorCode     = HOST_ENUMERROR_WaitStage;
 | 
						|
					break;
 | 
						|
				}
 | 
						|
				
 | 
						|
				if (!(--WaitMSRemaining))
 | 
						|
				  USB_HostState = PostWaitState;
 | 
						|
			}
 | 
						|
		
 | 
						|
			break;
 | 
						|
		case HOST_STATE_Powered:
 | 
						|
			WaitMSRemaining = HOST_DEVICE_SETTLE_DELAY_MS;
 | 
						|
		
 | 
						|
			USB_HostState = HOST_STATE_Powered_WaitForDeviceSettle;
 | 
						|
			break;
 | 
						|
		case HOST_STATE_Powered_WaitForDeviceSettle:
 | 
						|
			#if HOST_DEVICE_SETTLE_DELAY_MS > 0
 | 
						|
			_delay_ms(1);
 | 
						|
 | 
						|
			if (!(WaitMSRemaining--))
 | 
						|
			{
 | 
						|
				USB_Host_VBUS_Manual_Off();
 | 
						|
 | 
						|
				USB_OTGPAD_On();
 | 
						|
				USB_Host_VBUS_Auto_Enable();
 | 
						|
				USB_Host_VBUS_Auto_On();
 | 
						|
				
 | 
						|
				USB_HostState = HOST_STATE_Powered_WaitForConnect;
 | 
						|
			}
 | 
						|
			#else
 | 
						|
			USB_HostState = HOST_STATE_Powered_WaitForConnect;			
 | 
						|
			#endif
 | 
						|
			
 | 
						|
			break;
 | 
						|
		case HOST_STATE_Powered_WaitForConnect:		
 | 
						|
			if (USB_INT_HasOccurred(USB_INT_DCONNI))
 | 
						|
			{	
 | 
						|
				USB_INT_Clear(USB_INT_DCONNI);
 | 
						|
				USB_INT_Clear(USB_INT_DDISCI);
 | 
						|
 | 
						|
				USB_INT_Clear(USB_INT_VBERRI);
 | 
						|
				USB_INT_Enable(USB_INT_VBERRI);
 | 
						|
					
 | 
						|
				USB_Host_ResumeBus();
 | 
						|
				Pipe_ClearPipes();
 | 
						|
				
 | 
						|
				HOST_TASK_NONBLOCK_WAIT(100, HOST_STATE_Powered_DoReset);
 | 
						|
			}
 | 
						|
 | 
						|
			break;
 | 
						|
		case HOST_STATE_Powered_DoReset:
 | 
						|
			USB_Host_ResetDevice();
 | 
						|
 | 
						|
			HOST_TASK_NONBLOCK_WAIT(200, HOST_STATE_Powered_ConfigPipe);
 | 
						|
			break;
 | 
						|
		case HOST_STATE_Powered_ConfigPipe:
 | 
						|
			Pipe_ConfigurePipe(PIPE_CONTROLPIPE, EP_TYPE_CONTROL,
 | 
						|
							   PIPE_TOKEN_SETUP, ENDPOINT_CONTROLEP,
 | 
						|
							   PIPE_CONTROLPIPE_DEFAULT_SIZE, PIPE_BANK_SINGLE);		
 | 
						|
		
 | 
						|
			if (!(Pipe_IsConfigured()))
 | 
						|
			{
 | 
						|
				ErrorCode    = HOST_ENUMERROR_PipeConfigError;
 | 
						|
				SubErrorCode = 0;
 | 
						|
				break;
 | 
						|
			}
 | 
						|
 | 
						|
			USB_HostState = HOST_STATE_Default;
 | 
						|
			break;
 | 
						|
		case HOST_STATE_Default:
 | 
						|
			USB_ControlRequest = (USB_Request_Header_t)
 | 
						|
				{
 | 
						|
					.bmRequestType = (REQDIR_DEVICETOHOST | REQTYPE_STANDARD | REQREC_DEVICE),
 | 
						|
					.bRequest      = REQ_GetDescriptor,
 | 
						|
					.wValue        = (DTYPE_Device << 8),
 | 
						|
					.wIndex        = 0,
 | 
						|
					.wLength       = 8,
 | 
						|
				};
 | 
						|
 | 
						|
			uint8_t DataBuffer[8];
 | 
						|
 | 
						|
			if ((SubErrorCode = USB_Host_SendControlRequest(DataBuffer)) != HOST_SENDCONTROL_Successful)
 | 
						|
			{
 | 
						|
				ErrorCode = HOST_ENUMERROR_ControlError;
 | 
						|
				break;
 | 
						|
			}
 | 
						|
 | 
						|
			USB_ControlPipeSize = DataBuffer[offsetof(USB_Descriptor_Device_t, Endpoint0Size)];
 | 
						|
	
 | 
						|
			USB_Host_ResetDevice();
 | 
						|
			
 | 
						|
			HOST_TASK_NONBLOCK_WAIT(200, HOST_STATE_Default_PostReset);
 | 
						|
			break;
 | 
						|
		case HOST_STATE_Default_PostReset:
 | 
						|
			Pipe_DisablePipe();
 | 
						|
			Pipe_DeallocateMemory();		
 | 
						|
			Pipe_ResetPipe(PIPE_CONTROLPIPE);
 | 
						|
			
 | 
						|
			Pipe_ConfigurePipe(PIPE_CONTROLPIPE, EP_TYPE_CONTROL,
 | 
						|
			                   PIPE_TOKEN_SETUP, ENDPOINT_CONTROLEP,
 | 
						|
			                   USB_ControlPipeSize, PIPE_BANK_SINGLE);
 | 
						|
 | 
						|
			if (!(Pipe_IsConfigured()))
 | 
						|
			{
 | 
						|
				ErrorCode    = HOST_ENUMERROR_PipeConfigError;
 | 
						|
				SubErrorCode = 0;
 | 
						|
				break;
 | 
						|
			}
 | 
						|
 | 
						|
			USB_ControlRequest = (USB_Request_Header_t)
 | 
						|
				{
 | 
						|
					.bmRequestType = (REQDIR_HOSTTODEVICE | REQTYPE_STANDARD | REQREC_DEVICE),
 | 
						|
					.bRequest      = REQ_SetAddress,
 | 
						|
					.wValue        = USB_HOST_DEVICEADDRESS,
 | 
						|
					.wIndex        = 0,
 | 
						|
					.wLength       = 0,
 | 
						|
				};
 | 
						|
 | 
						|
			if ((SubErrorCode = USB_Host_SendControlRequest(NULL)) != HOST_SENDCONTROL_Successful)
 | 
						|
			{
 | 
						|
				ErrorCode = HOST_ENUMERROR_ControlError;
 | 
						|
				break;
 | 
						|
			}
 | 
						|
 | 
						|
			HOST_TASK_NONBLOCK_WAIT(100, HOST_STATE_Default_PostAddressSet);
 | 
						|
			break;
 | 
						|
		case HOST_STATE_Default_PostAddressSet:
 | 
						|
			USB_Host_SetDeviceAddress(USB_HOST_DEVICEADDRESS);
 | 
						|
 | 
						|
			EVENT_USB_Host_DeviceEnumerationComplete();
 | 
						|
			USB_HostState = HOST_STATE_Addressed;
 | 
						|
			break;
 | 
						|
	}
 | 
						|
 | 
						|
	if ((ErrorCode != HOST_ENUMERROR_NoError) && (USB_HostState != HOST_STATE_Unattached))
 | 
						|
	{
 | 
						|
		EVENT_USB_Host_DeviceEnumerationFailed(ErrorCode, SubErrorCode);
 | 
						|
 | 
						|
		USB_Host_VBUS_Auto_Off();
 | 
						|
 | 
						|
		EVENT_USB_Host_DeviceUnattached();
 | 
						|
 | 
						|
		USB_ResetInterface();
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
uint8_t USB_Host_WaitMS(uint8_t MS)
 | 
						|
{
 | 
						|
	bool    BusSuspended = USB_Host_IsBusSuspended();
 | 
						|
	uint8_t ErrorCode    = HOST_WAITERROR_Successful;
 | 
						|
	
 | 
						|
	USB_Host_ResumeBus();
 | 
						|
 | 
						|
	while (MS)
 | 
						|
	{
 | 
						|
		if (USB_INT_HasOccurred(USB_INT_HSOFI))
 | 
						|
		{
 | 
						|
			USB_INT_Clear(USB_INT_HSOFI);
 | 
						|
			MS--;
 | 
						|
		}
 | 
						|
					
 | 
						|
		if ((USB_HostState == HOST_STATE_Unattached) || (USB_CurrentMode == USB_MODE_DEVICE))
 | 
						|
		{
 | 
						|
			ErrorCode = HOST_WAITERROR_DeviceDisconnect;
 | 
						|
			
 | 
						|
			break;
 | 
						|
		}
 | 
						|
 | 
						|
		if (Pipe_IsError() == true)
 | 
						|
		{
 | 
						|
			Pipe_ClearError();
 | 
						|
			ErrorCode = HOST_WAITERROR_PipeError;
 | 
						|
			
 | 
						|
			break;
 | 
						|
		}
 | 
						|
		
 | 
						|
		if (Pipe_IsStalled() == true)
 | 
						|
		{
 | 
						|
			Pipe_ClearStall();
 | 
						|
			ErrorCode = HOST_WAITERROR_SetupStalled;
 | 
						|
			
 | 
						|
			break;			
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (BusSuspended)
 | 
						|
	  USB_Host_SuspendBus();
 | 
						|
 | 
						|
	return ErrorCode;
 | 
						|
}
 | 
						|
 | 
						|
static void USB_Host_ResetDevice(void)
 | 
						|
{
 | 
						|
	bool BusSuspended = USB_Host_IsBusSuspended();
 | 
						|
 | 
						|
	USB_INT_Disable(USB_INT_DDISCI);
 | 
						|
	
 | 
						|
	USB_Host_ResetBus();
 | 
						|
	while (!(USB_Host_IsBusResetComplete()));
 | 
						|
 | 
						|
	USB_Host_ResumeBus();
 | 
						|
 | 
						|
	USB_INT_Clear(USB_INT_HSOFI);
 | 
						|
 | 
						|
	for (uint8_t MSRem = 10; MSRem != 0; MSRem--)
 | 
						|
	{
 | 
						|
		/* Workaround for powerless-pull-up devices. After a USB bus reset,
 | 
						|
		   all disconnection interrupts are suppressed while a USB frame is
 | 
						|
		   looked for - if it is found within 10ms, the device is still
 | 
						|
		   present.                                                        */
 | 
						|
 | 
						|
		if (USB_INT_HasOccurred(USB_INT_HSOFI))
 | 
						|
		{
 | 
						|
			USB_INT_Clear(USB_INT_HSOFI);
 | 
						|
			USB_INT_Clear(USB_INT_DDISCI);
 | 
						|
			break;
 | 
						|
		}
 | 
						|
		
 | 
						|
		_delay_ms(1);
 | 
						|
	}
 | 
						|
 | 
						|
	if (BusSuspended)
 | 
						|
	  USB_Host_SuspendBus();
 | 
						|
 | 
						|
	USB_INT_Enable(USB_INT_DDISCI);
 | 
						|
}
 | 
						|
 | 
						|
uint8_t USB_Host_SetDeviceConfiguration(const uint8_t ConfigNumber)
 | 
						|
{
 | 
						|
	USB_ControlRequest = (USB_Request_Header_t)
 | 
						|
		{
 | 
						|
			.bmRequestType = (REQDIR_HOSTTODEVICE | REQTYPE_STANDARD | REQREC_DEVICE),
 | 
						|
			.bRequest      = REQ_SetConfiguration,
 | 
						|
			.wValue        = ConfigNumber,
 | 
						|
			.wIndex        = 0,
 | 
						|
			.wLength       = 0,
 | 
						|
		};
 | 
						|
 | 
						|
	Pipe_SelectPipe(PIPE_CONTROLPIPE);
 | 
						|
	
 | 
						|
	return USB_Host_SendControlRequest(NULL);
 | 
						|
}
 | 
						|
 | 
						|
uint8_t USB_Host_GetDeviceDescriptor(void* const DeviceDescriptorPtr)
 | 
						|
{
 | 
						|
	USB_ControlRequest = (USB_Request_Header_t)
 | 
						|
		{
 | 
						|
			bmRequestType: (REQDIR_DEVICETOHOST | REQTYPE_STANDARD | REQREC_DEVICE),
 | 
						|
			bRequest:      REQ_GetDescriptor,
 | 
						|
			wValue:        (DTYPE_Device << 8),
 | 
						|
			wIndex:        0,
 | 
						|
			wLength:       sizeof(USB_Descriptor_Device_t),
 | 
						|
		};
 | 
						|
 | 
						|
	Pipe_SelectPipe(PIPE_CONTROLPIPE);
 | 
						|
	
 | 
						|
	return USB_Host_SendControlRequest(DeviceDescriptorPtr);
 | 
						|
}
 | 
						|
 | 
						|
uint8_t USB_Host_ClearPipeStall(uint8_t EndpointNum)
 | 
						|
{
 | 
						|
	if (Pipe_GetPipeToken() == PIPE_TOKEN_IN)
 | 
						|
	  EndpointNum |= (1 << 7);
 | 
						|
 | 
						|
	USB_ControlRequest = (USB_Request_Header_t)
 | 
						|
		{
 | 
						|
			.bmRequestType = (REQDIR_HOSTTODEVICE | REQTYPE_STANDARD | REQREC_ENDPOINT),
 | 
						|
			.bRequest      = REQ_ClearFeature,
 | 
						|
			.wValue        = FEATURE_ENDPOINT_HALT,
 | 
						|
			.wIndex        = EndpointNum,
 | 
						|
			.wLength       = 0,
 | 
						|
		};
 | 
						|
 | 
						|
	Pipe_SelectPipe(PIPE_CONTROLPIPE);
 | 
						|
	
 | 
						|
	return USB_Host_SendControlRequest(NULL);
 | 
						|
}
 | 
						|
 | 
						|
#endif
 |