mirror of
				https://github.com/mfulz/qmk_firmware.git
				synced 2025-11-03 23:02:34 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			324 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			324 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
             LUFA Library
 | 
						|
     Copyright (C) Dean Camera, 2013.
 | 
						|
 | 
						|
  dean [at] fourwalledcubicle [dot] com
 | 
						|
           www.lufa-lib.org
 | 
						|
*/
 | 
						|
 | 
						|
/*
 | 
						|
  Copyright 2013  Dean Camera (dean [at] fourwalledcubicle [dot] com)
 | 
						|
 | 
						|
  Permission to use, copy, modify, distribute, and sell this
 | 
						|
  software and its documentation for any purpose is hereby granted
 | 
						|
  without fee, 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 disclaims 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.
 | 
						|
*/
 | 
						|
 | 
						|
/** \file
 | 
						|
 *
 | 
						|
 *  Main source file for the HIDReportViewer project. This file contains the main tasks of
 | 
						|
 *  the project and is responsible for the initial application hardware configuration.
 | 
						|
 */
 | 
						|
 | 
						|
#include "HIDReportViewer.h"
 | 
						|
 | 
						|
/** Processed HID report descriptor items structure, containing information on each HID report element */
 | 
						|
static HID_ReportInfo_t HIDReportInfo;
 | 
						|
 | 
						|
/** LUFA HID Class driver interface configuration and state information. This structure is
 | 
						|
 *  passed to all HID Class driver functions, so that multiple instances of the same class
 | 
						|
 *  within a device can be differentiated from one another.
 | 
						|
 */
 | 
						|
USB_ClassInfo_HID_Host_t Device_HID_Interface =
 | 
						|
	{
 | 
						|
		.Config =
 | 
						|
			{
 | 
						|
				.DataINPipe             =
 | 
						|
					{
 | 
						|
						.Address        = (PIPE_DIR_IN  | 1),
 | 
						|
						.Banks          = 1,
 | 
						|
					},
 | 
						|
				.DataOUTPipe            =
 | 
						|
					{
 | 
						|
						.Address        = (PIPE_DIR_OUT | 2),
 | 
						|
						.Banks          = 1,
 | 
						|
					},
 | 
						|
				.HIDInterfaceProtocol   = HID_CSCP_NonBootProtocol,
 | 
						|
				.HIDParserData          = &HIDReportInfo
 | 
						|
			},
 | 
						|
	};
 | 
						|
 | 
						|
 | 
						|
/** Main program entry point. This routine configures the hardware required by the application, then
 | 
						|
 *  enters a loop to run the application tasks in sequence.
 | 
						|
 */
 | 
						|
int main(void)
 | 
						|
{
 | 
						|
	SetupHardware();
 | 
						|
 | 
						|
	puts_P(PSTR(ESC_FG_CYAN "HID Device Report Viewer Running.\r\n" ESC_FG_WHITE));
 | 
						|
 | 
						|
	LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 | 
						|
	GlobalInterruptEnable();
 | 
						|
 | 
						|
	for (;;)
 | 
						|
	{
 | 
						|
		RetrieveDeviceData();
 | 
						|
 | 
						|
		HID_Host_USBTask(&Device_HID_Interface);
 | 
						|
		USB_USBTask();
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** Task to retrieve the HID device information from an attached device, and output
 | 
						|
 *  the relevant data to the serial port for analysis.
 | 
						|
 */
 | 
						|
void RetrieveDeviceData(void)
 | 
						|
{
 | 
						|
	if (USB_CurrentMode != USB_MODE_Host)
 | 
						|
	  return;
 | 
						|
 | 
						|
	LEDs_SetAllLEDs(LEDMASK_USB_BUSY);
 | 
						|
 | 
						|
	OutputReportSizes();
 | 
						|
	OutputParsedReportItems();
 | 
						|
 | 
						|
	LEDs_SetAllLEDs(LEDMASK_USB_READY);
 | 
						|
	USB_Host_SetDeviceConfiguration(0);
 | 
						|
}
 | 
						|
 | 
						|
/** Prints a summary of the device's HID report sizes from the HID parser output to the serial port
 | 
						|
 *  for display to the user.
 | 
						|
 */
 | 
						|
void OutputReportSizes(void)
 | 
						|
{
 | 
						|
	printf_P(PSTR("\r\n\r\nTotal Device Reports: %" PRId8 "\r\n"), HIDReportInfo.TotalDeviceReports);
 | 
						|
 | 
						|
	for (uint8_t ReportIndex = 0; ReportIndex < HIDReportInfo.TotalDeviceReports; ReportIndex++)
 | 
						|
	{
 | 
						|
		const HID_ReportSizeInfo_t* CurrReportIDInfo = &HIDReportInfo.ReportIDSizes[ReportIndex];
 | 
						|
 | 
						|
		uint8_t ReportSizeInBits      = CurrReportIDInfo->ReportSizeBits[HID_REPORT_ITEM_In];
 | 
						|
		uint8_t ReportSizeOutBits     = CurrReportIDInfo->ReportSizeBits[HID_REPORT_ITEM_Out];
 | 
						|
		uint8_t ReportSizeFeatureBits = CurrReportIDInfo->ReportSizeBits[HID_REPORT_ITEM_Feature];
 | 
						|
 | 
						|
		/* Print out the byte sizes of each report within the device */
 | 
						|
		printf_P(PSTR("  + Report ID 0x%02" PRIX8 "\r\n"
 | 
						|
					  "    - Input Data:   %" PRId8 " bits (%" PRId8 " bytes)\r\n"
 | 
						|
					  "    - Output Data:  %" PRId8 " bits (%" PRId8 " bytes)\r\n"
 | 
						|
					  "    - Feature Data: %" PRId8 " bits (%" PRId8 " bytes)\r\n"),
 | 
						|
				 CurrReportIDInfo->ReportID,
 | 
						|
				 ReportSizeInBits,
 | 
						|
				 ((ReportSizeInBits      >> 3) + ((ReportSizeInBits      & 0x07) != 0)),
 | 
						|
				 ReportSizeOutBits,
 | 
						|
				 ((ReportSizeOutBits     >> 3) + ((ReportSizeOutBits     & 0x07) != 0)),
 | 
						|
				 ReportSizeFeatureBits,
 | 
						|
				 ((ReportSizeFeatureBits >> 3) + ((ReportSizeFeatureBits & 0x07) != 0)));
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** Prints a summary of the device's parsed and stored report items along with their attributes
 | 
						|
 *  to the serial port for display to the user.
 | 
						|
 */
 | 
						|
void OutputParsedReportItems(void)
 | 
						|
{
 | 
						|
	printf_P(PSTR("\r\nReport Items (%" PRId8 " in Table):\r\n"), HIDReportInfo.TotalReportItems);
 | 
						|
 | 
						|
	for (uint8_t ItemIndex = 0; ItemIndex < HIDReportInfo.TotalReportItems; ItemIndex++)
 | 
						|
	{
 | 
						|
		const HID_ReportItem_t* RItem = &HIDReportInfo.ReportItems[ItemIndex];
 | 
						|
 | 
						|
		printf_P(PSTR("  + Item %" PRId8 ":\r\n"
 | 
						|
					  "    - Report ID:          0x%02" PRIX8  "\r\n"
 | 
						|
					  "    - Data Direction:     %s\r\n"
 | 
						|
					  "    - Item Flags:         0x%02" PRIX8  "\r\n"
 | 
						|
					  "    - Item Offset (Bits): 0x%02" PRIX8  "\r\n"
 | 
						|
					  "    - Item Size (Bits):   0x%02" PRIX8  "\r\n"
 | 
						|
					  "    - Usage Page:         0x%04" PRIX16 "\r\n"
 | 
						|
					  "    - Usage:              0x%04" PRIX16 "\r\n"
 | 
						|
					  "    - Unit Type:          0x%08" PRIX32 "\r\n"
 | 
						|
					  "    - Unit Exponent:      0x%02" PRIX8  "\r\n"
 | 
						|
					  "    - Logical Minimum:    0x%08" PRIX32 "\r\n"
 | 
						|
					  "    - Logical Maximum:    0x%08" PRIX32 "\r\n"
 | 
						|
					  "    - Physical Minimum:   0x%08" PRIX32 "\r\n"
 | 
						|
					  "    - Physical Maximum:   0x%08" PRIX32 "\r\n"
 | 
						|
					  "    - Collection Path:\r\n"),
 | 
						|
				 ItemIndex,
 | 
						|
				 RItem->ReportID,
 | 
						|
				 ((RItem->ItemType == HID_REPORT_ITEM_In) ? "IN" : ((RItem->ItemType == HID_REPORT_ITEM_Out) ? "OUT" : "FEATURE")),
 | 
						|
				 RItem->ItemFlags,
 | 
						|
				 RItem->BitOffset,
 | 
						|
				 RItem->Attributes.BitSize,
 | 
						|
				 RItem->Attributes.Usage.Page,
 | 
						|
				 RItem->Attributes.Usage.Usage,
 | 
						|
				 RItem->Attributes.Unit.Type,
 | 
						|
				 RItem->Attributes.Unit.Exponent,
 | 
						|
				 RItem->Attributes.Logical.Minimum,
 | 
						|
				 RItem->Attributes.Logical.Maximum,
 | 
						|
				 RItem->Attributes.Physical.Minimum,
 | 
						|
				 RItem->Attributes.Physical.Maximum);
 | 
						|
 | 
						|
		OutputCollectionPath(RItem->CollectionPath);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** Prints the HID Collection path (along with each node's attributes) to the serial port
 | 
						|
 *  for display to the user, from the given starting node to the root node.
 | 
						|
 *
 | 
						|
 *  \param[in] CollectionPath  Starting HID Collection node to print
 | 
						|
 */
 | 
						|
void OutputCollectionPath(const HID_CollectionPath_t* const CollectionPath)
 | 
						|
{
 | 
						|
	const HID_CollectionPath_t* CurrentNode = CollectionPath;
 | 
						|
 | 
						|
	while (CurrentNode != NULL)
 | 
						|
	{
 | 
						|
		printf_P(PSTR("      |\r\n"
 | 
						|
					  "      - Type:  0x%02" PRIX8 "\r\n"
 | 
						|
					  "      - Usage: 0x%02" PRIX8 "\r\n"),
 | 
						|
					  CurrentNode->Type, CurrentNode->Usage);
 | 
						|
 | 
						|
		CurrentNode = CurrentNode->Parent;
 | 
						|
	}
 | 
						|
 | 
						|
	printf_P(PSTR("      |\r\n"
 | 
						|
				  "      END\r\n"));
 | 
						|
}
 | 
						|
 | 
						|
/** Configures the board hardware and chip peripherals for the demo's functionality. */
 | 
						|
void SetupHardware(void)
 | 
						|
{
 | 
						|
	/* Disable watchdog if enabled by bootloader/fuses */
 | 
						|
	MCUSR &= ~(1 << WDRF);
 | 
						|
	wdt_disable();
 | 
						|
 | 
						|
	/* Disable clock division */
 | 
						|
	clock_prescale_set(clock_div_1);
 | 
						|
 | 
						|
	/* Hardware Initialization */
 | 
						|
	Serial_Init(9600, false);
 | 
						|
	LEDs_Init();
 | 
						|
	USB_Init();
 | 
						|
 | 
						|
	/* Create a stdio stream for the serial port for stdin and stdout */
 | 
						|
	Serial_CreateStream(NULL);
 | 
						|
}
 | 
						|
 | 
						|
/** Event handler for the USB_DeviceAttached event. This indicates that a device has been attached to the host, and
 | 
						|
 *  starts the library USB task to begin the enumeration and USB management process.
 | 
						|
 */
 | 
						|
void EVENT_USB_Host_DeviceAttached(void)
 | 
						|
{
 | 
						|
	puts_P(PSTR("Device Attached.\r\n"));
 | 
						|
	LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
 | 
						|
}
 | 
						|
 | 
						|
/** Event handler for the USB_DeviceUnattached event. This indicates that a device has been removed from the host, and
 | 
						|
 *  stops the library USB task management process.
 | 
						|
 */
 | 
						|
void EVENT_USB_Host_DeviceUnattached(void)
 | 
						|
{
 | 
						|
	puts_P(PSTR("\r\nDevice Unattached.\r\n"));
 | 
						|
	LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 | 
						|
}
 | 
						|
 | 
						|
/** Event handler for the USB_DeviceEnumerationComplete event. This indicates that a device has been successfully
 | 
						|
 *  enumerated by the host and is now ready to be used by the application.
 | 
						|
 */
 | 
						|
void EVENT_USB_Host_DeviceEnumerationComplete(void)
 | 
						|
{
 | 
						|
	LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
 | 
						|
 | 
						|
	uint16_t ConfigDescriptorSize;
 | 
						|
	uint8_t  ConfigDescriptorData[512];
 | 
						|
 | 
						|
	if (USB_Host_GetDeviceConfigDescriptor(1, &ConfigDescriptorSize, ConfigDescriptorData,
 | 
						|
	                                       sizeof(ConfigDescriptorData)) != HOST_GETCONFIG_Successful)
 | 
						|
	{
 | 
						|
		puts_P(PSTR("Error Retrieving Configuration Descriptor.\r\n"));
 | 
						|
		LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	if (HID_Host_ConfigurePipes(&Device_HID_Interface,
 | 
						|
	                            ConfigDescriptorSize, ConfigDescriptorData) != HID_ENUMERROR_NoError)
 | 
						|
	{
 | 
						|
		puts_P(PSTR("Attached Device Not a Valid HID Device.\r\n"));
 | 
						|
		LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	if (USB_Host_SetDeviceConfiguration(1) != HOST_SENDCONTROL_Successful)
 | 
						|
	{
 | 
						|
		puts_P(PSTR("Error Setting Device Configuration.\r\n"));
 | 
						|
		LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	if (HID_Host_SetReportProtocol(&Device_HID_Interface) != 0)
 | 
						|
	{
 | 
						|
		puts_P(PSTR("Error Setting Report Protocol Mode.\r\n"));
 | 
						|
		LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 | 
						|
		USB_Host_SetDeviceConfiguration(0);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	puts_P(PSTR("HID Device Enumerated.\r\n"));
 | 
						|
	LEDs_SetAllLEDs(LEDMASK_USB_READY);
 | 
						|
}
 | 
						|
 | 
						|
/** Event handler for the USB_HostError event. This indicates that a hardware error occurred while in host mode. */
 | 
						|
void EVENT_USB_Host_HostError(const uint8_t ErrorCode)
 | 
						|
{
 | 
						|
	USB_Disable();
 | 
						|
 | 
						|
	printf_P(PSTR(ESC_FG_RED "Host Mode Error\r\n"
 | 
						|
	                         " -- Error Code %d\r\n" ESC_FG_WHITE), ErrorCode);
 | 
						|
 | 
						|
	LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 | 
						|
	for(;;);
 | 
						|
}
 | 
						|
 | 
						|
/** Event handler for the USB_DeviceEnumerationFailed event. This indicates that a problem occurred while
 | 
						|
 *  enumerating an attached USB device.
 | 
						|
 */
 | 
						|
void EVENT_USB_Host_DeviceEnumerationFailed(const uint8_t ErrorCode,
 | 
						|
                                            const uint8_t SubErrorCode)
 | 
						|
{
 | 
						|
	printf_P(PSTR(ESC_FG_RED "Dev Enum Error\r\n"
 | 
						|
	                         " -- Error Code %d\r\n"
 | 
						|
	                         " -- Sub Error Code %d\r\n"
 | 
						|
	                         " -- In State %d\r\n" ESC_FG_WHITE), ErrorCode, SubErrorCode, USB_HostState);
 | 
						|
 | 
						|
	LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 | 
						|
}
 | 
						|
 | 
						|
/** Callback for the HID Report Parser. This function is called each time the HID report parser is about to store
 | 
						|
 *  an IN, OUT or FEATURE item into the HIDReportInfo structure. To save on RAM, we are able to filter out items
 | 
						|
 *  we aren't interested in (preventing us from being able to extract them later on, but saving on the RAM they would
 | 
						|
 *  have occupied).
 | 
						|
 *
 | 
						|
 *  \param[in] CurrentItem  Pointer to the item the HID report parser is currently working with
 | 
						|
 *
 | 
						|
 *  \return Boolean true if the item should be stored into the HID report structure, false if it should be discarded
 | 
						|
 */
 | 
						|
bool CALLBACK_HIDParser_FilterHIDReportItem(HID_ReportItem_t* const CurrentItem)
 | 
						|
{
 | 
						|
	return true;
 | 
						|
}
 | 
						|
 |