Blame view

atmega16u2/custom/lufa-LUFA-170418/Bootloaders/DFU/BootloaderDFU.h 7.69 KB
8a7dc1f5   adorian   Seance 4
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
  /*
               LUFA Library
       Copyright (C) Dean Camera, 2017.
  
    dean [at] fourwalledcubicle [dot] com
             www.lufa-lib.org
  */
  
  /*
    Copyright 2017  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
   *
   *  Header file for BootloaderDFU.c.
   */
  
  #ifndef _BOOTLOADER_H_
  #define _BOOTLOADER_H_
  
  	/* Includes: */
  		#include <avr/io.h>
  		#include <avr/wdt.h>
  		#include <avr/boot.h>
  		#include <avr/pgmspace.h>
  		#include <avr/eeprom.h>
  		#include <avr/power.h>
  		#include <avr/interrupt.h>
  		#include <util/delay.h>
  		#include <stdbool.h>
  
  		#include "Descriptors.h"
  		#include "BootloaderAPI.h"
  		#include "Config/AppConfig.h"
  
  		#include <LUFA/Drivers/USB/USB.h>
  		#include <LUFA/Drivers/Board/LEDs.h>
  		#include <LUFA/Platform/Platform.h>
  
  	/* Preprocessor Checks: */
  		#if !defined(__OPTIMIZE_SIZE__)
  			#error This bootloader requires that it be optimized for size, not speed, to fit into the target device. Change optimization settings and try again.
  		#endif
  
  	/* Macros: */
  		/** Major bootloader version number. */
  		#define BOOTLOADER_VERSION_MINOR 2
  
  		/** Minor bootloader version number. */
  		#define BOOTLOADER_VERSION_REV   0
  
  		/** Magic bootloader key to unlock forced application start mode. */
  		#define MAGIC_BOOT_KEY           0xDC42
  
  		/** Complete bootloader version number expressed as a packed byte, constructed from the
  		 *  two individual bootloader version macros.
  		 */
  		#define BOOTLOADER_VERSION       ((BOOTLOADER_VERSION_MINOR << 4) | BOOTLOADER_VERSION_REV)
  
  		/** First byte of the bootloader identification bytes, used to identify a device's bootloader. */
  		#define BOOTLOADER_ID_BYTE1      0xDC
  
  		/** Second byte of the bootloader identification bytes, used to identify a device's bootloader. */
  		#define BOOTLOADER_ID_BYTE2      0xFB
  
  		/** Convenience macro, used to determine if the issued command is the given one-byte long command.
  		 *
  		 *  \param[in] dataarr  Command byte array to check against
  		 *  \param[in] cb1      First command byte to check
  		 */
  		#define IS_ONEBYTE_COMMAND(dataarr, cb1)       (dataarr[0] == (cb1))
  
  		/** Convenience macro, used to determine if the issued command is the given two-byte long command.
  		 *
  		 *  \param[in] dataarr  Command byte array to check against
  		 *  \param[in] cb1      First command byte to check
  		 *  \param[in] cb2      Second command byte to check
  		 */
  		#define IS_TWOBYTE_COMMAND(dataarr, cb1, cb2) ((dataarr[0] == (cb1)) && (dataarr[1] == (cb2)))
  
  		/** Length of the DFU file suffix block, appended to the end of each complete memory write command.
  		 *  The DFU file suffix is currently unused (but is designed to give extra file information, such as
  		 *  a CRC of the complete firmware for error checking) and so is discarded.
  		 */
  		#define DFU_FILE_SUFFIX_SIZE     16
  
  		/** Length of the DFU file filler block, appended to the start of each complete memory write command.
  		 *  Filler bytes are added to the start of each complete memory write command, and must be discarded.
  		 */
  		#define DFU_FILLER_BYTES_SIZE    26
  
  		/** DFU class command request to detach from the host. */
  		#define DFU_REQ_DETATCH          0x00
  
  		/** DFU class command request to send data from the host to the bootloader. */
  		#define DFU_REQ_DNLOAD           0x01
  
  		/** DFU class command request to send data from the bootloader to the host. */
  		#define DFU_REQ_UPLOAD           0x02
  
  		/** DFU class command request to get the current DFU status and state from the bootloader. */
  		#define DFU_REQ_GETSTATUS        0x03
  
  		/** DFU class command request to reset the current DFU status and state variables to their defaults. */
  		#define DFU_REQ_CLRSTATUS        0x04
  
  		/** DFU class command request to get the current DFU state of the bootloader. */
  		#define DFU_REQ_GETSTATE         0x05
  
  		/** DFU class command request to abort the current multi-request transfer and return to the dfuIDLE state. */
  		#define DFU_REQ_ABORT            0x06
  
  		/** DFU command to begin programming the device's memory. */
  		#define COMMAND_PROG_START       0x01
  
  		/** DFU command to begin reading the device's memory. */
  		#define COMMAND_DISP_DATA        0x03
  
  		/** DFU command to issue a write command. */
  		#define COMMAND_WRITE            0x04
  
  		/** DFU command to issue a read command. */
  		#define COMMAND_READ             0x05
  
  		/** DFU command to issue a memory base address change command, to set the current 64KB flash page
  		 *  that subsequent flash operations should use. */
  		#define COMMAND_CHANGE_BASE_ADDR 0x06
  
  	/* Type Defines: */
  		/** Type define for a non-returning function pointer to the loaded application. */
  		typedef void (*AppPtr_t)(void) ATTR_NO_RETURN;
  
  		/** Type define for a structure containing a complete DFU command issued by the host. */
  		typedef struct
  		{
  			uint8_t  Command; /**< Single byte command to perform, one of the \c COMMAND_* macro values */
  			uint8_t  Data[5]; /**< Command parameters */
  			uint16_t DataSize; /**< Size of the command parameters */
  		} DFU_Command_t;
  
  	/* Enums: */
  		/** DFU bootloader states. Refer to the DFU class specification for information on each state. */
  		enum DFU_State_t
  		{
  			appIDLE                      = 0,
  			appDETACH                    = 1,
  			dfuIDLE                      = 2,
  			dfuDNLOAD_SYNC               = 3,
  			dfuDNBUSY                    = 4,
  			dfuDNLOAD_IDLE               = 5,
  			dfuMANIFEST_SYNC             = 6,
  			dfuMANIFEST                  = 7,
  			dfuMANIFEST_WAIT_RESET       = 8,
  			dfuUPLOAD_IDLE               = 9,
  			dfuERROR                     = 10
  		};
  
  		/** DFU command status error codes. Refer to the DFU class specification for information on each error code. */
  		enum DFU_Status_t
  		{
  			OK                           = 0,
  			errTARGET                    = 1,
  			errFILE                      = 2,
  			errWRITE                     = 3,
  			errERASE                     = 4,
  			errCHECK_ERASED              = 5,
  			errPROG                      = 6,
  			errVERIFY                    = 7,
  			errADDRESS                   = 8,
  			errNOTDONE                   = 9,
  			errFIRMWARE                  = 10,
  			errVENDOR                    = 11,
  			errUSBR                      = 12,
  			errPOR                       = 13,
  			errUNKNOWN                   = 14,
  			errSTALLEDPKT                = 15
  		};
  
  	/* Function Prototypes: */
  		static void SetupHardware(void);
  		static void ResetHardware(void);
  
  		void EVENT_USB_Device_ControlRequest(void);
  
  		#if defined(INCLUDE_FROM_BOOTLOADER_C)
  			static void DiscardFillerBytes(uint8_t NumberOfBytes);
  			static void ProcessBootloaderCommand(void);
  			static void LoadStartEndAddresses(void);
  			static void ProcessMemProgCommand(void);
  			static void ProcessMemReadCommand(void);
  			static void ProcessWriteCommand(void);
  			static void ProcessReadCommand(void);
  		#endif
  
  		void Application_Jump_Check(void) ATTR_INIT_SECTION(3);
  
  #endif