+/*\r
+ LUFA Library\r
+ Copyright (C) Dean Camera, 2009.\r
+ \r
+ dean [at] fourwalledcubicle [dot] com\r
+ www.fourwalledcubicle.com\r
+*/\r
+\r
+/*\r
+ Copyright 2009 Dean Camera (dean [at] fourwalledcubicle [dot] com)\r
+\r
+ Permission to use, copy, modify, and distribute this software\r
+ and its documentation for any purpose and without fee is hereby\r
+ granted, provided that the above copyright notice appear in all\r
+ copies and that both that the copyright notice and this\r
+ permission notice and warranty disclaimer appear in supporting\r
+ documentation, and that the name of the author not be used in\r
+ advertising or publicity pertaining to distribution of the\r
+ software without specific, written prior permission.\r
+\r
+ The author disclaim all warranties with regard to this\r
+ software, including all implied warranties of merchantability\r
+ and fitness. In no event shall the author be liable for any\r
+ special, indirect or consequential damages or any damages\r
+ whatsoever resulting from loss of use, data or profits, whether\r
+ in an action of contract, negligence or other tortious action,\r
+ arising out of or in connection with the use or performance of\r
+ this software.\r
+*/\r
+\r
+#if defined(ENABLE_XPROG_PROTOCOL)\r
+\r
+/** \file\r
+ *\r
+ * PDI Protocol handler, to process V2 Protocol wrapped PDI commands used in Atmel programmer devices.\r
+ */\r
+\r
+#define INCLUDE_FROM_XPROG_C\r
+#include "PDIProtocol.h"\r
+\r
+uint32_t XPROG_Param_NVMBase;\r
+uint32_t XPROG_Param_EEPageSize;\r
+\r
+/** Handler for the CMD_XPROG_SETMODE command, which sets the programmer-to-target protocol used for PDI\r
+ * XMEGA programming (either PDI or JTAG). Only PDI programming is supported.\r
+ */\r
+void PDIProtocol_XPROG_SetMode(void)\r
+{\r
+ struct\r
+ {\r
+ uint8_t Protocol;\r
+ } SetMode_XPROG_Params;\r
+ \r
+ Endpoint_Read_Stream_LE(&SetMode_XPROG_Params, sizeof(SetMode_XPROG_Params));\r
+\r
+ Endpoint_ClearOUT();\r
+ Endpoint_SetEndpointDirection(ENDPOINT_DIR_IN);\r
+ \r
+ Endpoint_Write_Byte(CMD_XPROG_SETMODE);\r
+ Endpoint_Write_Byte(SetMode_XPROG_Params.Protocol ? STATUS_CMD_FAILED : STATUS_CMD_OK);\r
+ Endpoint_ClearIN(); \r
+}\r
+\r
+void PDIProtocol_XPROG_Command(void)\r
+{\r
+ uint8_t XPROGCommand = Endpoint_Read_Byte();\r
+\r
+ switch (XPROGCommand)\r
+ {\r
+ case XPRG_CMD_ENTER_PROGMODE:\r
+ PDIProtocol_EnterXPROGMode();\r
+ break;\r
+ case XPRG_CMD_LEAVE_PROGMODE:\r
+ PDIProtocol_LeaveXPROGMode();\r
+ break;\r
+ case XPRG_CMD_ERASE:\r
+ PDIProtocol_EraseChip();\r
+ break;\r
+ case XPRG_CMD_WRITE_MEM:\r
+ PDIProtocol_WriteMemory();\r
+ break;\r
+ case XPRG_CMD_READ_MEM:\r
+ PDIProtocol_ReadMemory();\r
+ break;\r
+ case XPRG_CMD_CRC:\r
+ PDIProtocol_ReadCRC();\r
+ break;\r
+ case XPRG_CMD_SET_PARAM:\r
+ PDIProtocol_SetParam();\r
+ break;\r
+ }\r
+}\r
+\r
+static void PDIProtocol_EraseChip(void)\r
+{\r
+ uint8_t ReturnStatus = XPRG_ERR_OK;\r
+\r
+ struct\r
+ {\r
+ uint8_t MemoryType;\r
+ uint32_t Address;\r
+ } Erase_XPROG_Params;\r
+\r
+ Endpoint_Read_Stream_LE(&Erase_XPROG_Params, sizeof(Erase_XPROG_Params));\r
+\r
+ Endpoint_ClearOUT();\r
+ Endpoint_SetEndpointDirection(ENDPOINT_DIR_IN);\r
+ \r
+ // TODO: Send erase command here via PDI protocol\r
+ \r
+ Endpoint_Write_Byte(CMD_XPROG);\r
+ Endpoint_Write_Byte(XPRG_CMD_ERASE);\r
+ Endpoint_Write_Byte(ReturnStatus);\r
+ Endpoint_ClearIN(); \r
+}\r
+\r
+static void PDIProtocol_WriteMemory(void)\r
+{\r
+ uint8_t ReturnStatus = XPRG_ERR_OK;\r
+\r
+ struct\r
+ {\r
+ uint8_t MemoryType;\r
+ uint32_t Address;\r
+ uint16_t Length;\r
+ uint8_t ProgData[256];\r
+ } WriteMemory_XPROG_Params;\r
+ \r
+ Endpoint_Read_Stream_LE(&WriteMemory_XPROG_Params, (sizeof(WriteMemory_XPROG_Params) -\r
+ sizeof(WriteMemory_XPROG_Params).ProgData));\r
+ WriteMemory_XPROG_Params.Address = SwapEndian_32(WriteMemory_XPROG_Params.Address);\r
+ WriteMemory_XPROG_Params.Length = SwapEndian_16(WriteMemory_XPROG_Params.Length);\r
+ Endpoint_Read_Stream_LE(&WriteMemory_XPROG_Params.ProgData, WriteMemory_XPROG_Params.Length);\r
+\r
+ Endpoint_ClearOUT();\r
+ Endpoint_SetEndpointDirection(ENDPOINT_DIR_IN);\r
+ \r
+ // TODO: Send program command here via PDI protocol\r
+ \r
+ Endpoint_Write_Byte(CMD_XPROG);\r
+ Endpoint_Write_Byte(XPRG_CMD_READ_MEM);\r
+ Endpoint_Write_Byte(ReturnStatus); \r
+ Endpoint_ClearIN();\r
+}\r
+\r
+static void PDIProtocol_ReadMemory(void)\r
+{\r
+ uint8_t ReturnStatus = XPRG_ERR_OK;\r
+\r
+ struct\r
+ {\r
+ uint8_t MemoryType;\r
+ uint32_t Address;\r
+ uint16_t Length;\r
+ } ReadMemory_XPROG_Params;\r
+ \r
+ Endpoint_Read_Stream_LE(&ReadMemory_XPROG_Params, sizeof(ReadMemory_XPROG_Params));\r
+ ReadMemory_XPROG_Params.Address = SwapEndian_32(ReadMemory_XPROG_Params.Address);\r
+ ReadMemory_XPROG_Params.Length = SwapEndian_16(ReadMemory_XPROG_Params.Length);\r
+\r
+ Endpoint_ClearOUT();\r
+ Endpoint_SetEndpointDirection(ENDPOINT_DIR_IN);\r
+ \r
+ // TODO: Send read command here via PDI protocol\r
+ \r
+ Endpoint_Write_Byte(CMD_XPROG);\r
+ Endpoint_Write_Byte(XPRG_CMD_READ_MEM);\r
+ Endpoint_Write_Byte(ReturnStatus);\r
+\r
+ // START TEMP\r
+ uint8_t ProgData[256];\r
+ for (uint16_t i = 0; i < ReadMemory_XPROG_Params.Length; i++)\r
+ ProgData[i] = i;\r
+ Endpoint_Write_Stream_LE(ProgData, ReadMemory_XPROG_Params.Length);\r
+\r
+ if (!Endpoint_IsReadWriteAllowed())\r
+ {\r
+ Endpoint_ClearIN(); \r
+ while(!(Endpoint_IsReadWriteAllowed()));\r
+ }\r
+ // END TEMP\r
+ \r
+ Endpoint_ClearIN();\r
+}\r
+\r
+static void PDIProtocol_ReadCRC(void)\r
+{\r
+ uint8_t ReturnStatus = XPRG_ERR_OK;\r
+\r
+ uint8_t CRCType = Endpoint_Read_Byte();\r
+ \r
+ Endpoint_ClearOUT();\r
+ Endpoint_SetEndpointDirection(ENDPOINT_DIR_IN);\r
+ \r
+ uint32_t MemoryCRC = 0;\r
+ \r
+ // TODO: Read device CRC for desired memory via PDI protocol\r
+ \r
+ Endpoint_Write_Byte(CMD_XPROG);\r
+ Endpoint_Write_Byte(XPRG_CMD_CRC);\r
+ Endpoint_Write_Byte(ReturnStatus);\r
+ \r
+ if (ReturnStatus == XPRG_ERR_OK)\r
+ {\r
+ Endpoint_Write_Byte(MemoryCRC >> 16);\r
+ Endpoint_Write_Word_LE(MemoryCRC & 0xFFFF); \r
+ }\r
+ \r
+ Endpoint_ClearIN(); \r
+}\r
+\r
+static void PDIProtocol_EnterXPROGMode(void)\r
+{\r
+ uint8_t ReturnStatus = XPRG_ERR_OK;\r
+\r
+ Endpoint_ClearOUT();\r
+ Endpoint_SetEndpointDirection(ENDPOINT_DIR_IN);\r
+ \r
+ // TODO: Enter device programming mode here via PDI protocol\r
+ \r
+ Endpoint_Write_Byte(CMD_XPROG);\r
+ Endpoint_Write_Byte(XPRG_CMD_ENTER_PROGMODE);\r
+ Endpoint_Write_Byte(ReturnStatus);\r
+ Endpoint_ClearIN();\r
+}\r
+\r
+static void PDIProtocol_LeaveXPROGMode(void)\r
+{\r
+ Endpoint_ClearOUT();\r
+ Endpoint_SetEndpointDirection(ENDPOINT_DIR_IN);\r
+ \r
+ // TODO: Exit device programming mode here via PDI protocol\r
+\r
+ Endpoint_Write_Byte(CMD_XPROG);\r
+ Endpoint_Write_Byte(XPRG_CMD_LEAVE_PROGMODE);\r
+ Endpoint_Write_Byte(XPRG_ERR_OK);\r
+ Endpoint_ClearIN();\r
+}\r
+\r
+static void PDIProtocol_SetParam(void)\r
+{\r
+ uint8_t ReturnStatus = XPRG_ERR_OK;\r
+\r
+ uint8_t XPROGParam = Endpoint_Read_Byte();\r
+ \r
+ if (XPROGParam == XPRG_PARAM_NVMBASE)\r
+ XPROG_Param_NVMBase = Endpoint_Read_DWord_LE();\r
+ else if (XPROGParam == XPRG_PARAM_EEPPAGESIZE)\r
+ XPROG_Param_EEPageSize = Endpoint_Read_Word_LE();\r
+ else\r
+ ReturnStatus = XPRG_ERR_FAILED;\r
+ \r
+ Endpoint_ClearOUT();\r
+ Endpoint_SetEndpointDirection(ENDPOINT_DIR_IN);\r
+ \r
+ Endpoint_Write_Byte(CMD_XPROG);\r
+ Endpoint_Write_Byte(XPRG_CMD_SET_PARAM);\r
+ Endpoint_Write_Byte(ReturnStatus);\r
+ Endpoint_ClearIN();\r
+}\r
+\r
+#endif\r
projects / pub / USBasp.git / commitdiff