[4b06930] | 1 | /* Name: main.c
|
---|
| 2 | * Project: USBaspLoader
|
---|
| 3 | * Author: Christian Starkjohann
|
---|
| 4 | * Creation Date: 2007-12-08
|
---|
| 5 | * Tabsize: 4
|
---|
| 6 | * Copyright: (c) 2007 by OBJECTIVE DEVELOPMENT Software GmbH
|
---|
| 7 | * License: GNU GPL v2 (see License.txt)
|
---|
| 8 | * This Revision: $Id: main.c 730 2009-03-20 09:05:11Z cs $
|
---|
| 9 | */
|
---|
| 10 |
|
---|
| 11 | #include <avr/io.h>
|
---|
| 12 | #include <avr/interrupt.h>
|
---|
| 13 | #include <avr/pgmspace.h>
|
---|
| 14 | #include <avr/wdt.h>
|
---|
| 15 | #include <avr/boot.h>
|
---|
| 16 | #include <avr/eeprom.h>
|
---|
| 17 | #include <util/delay.h>
|
---|
| 18 | #include <string.h>
|
---|
| 19 |
|
---|
| 20 | static void leaveBootloader() __attribute__((__noreturn__));
|
---|
| 21 |
|
---|
| 22 | #include "bootloaderconfig.h"
|
---|
| 23 | #include "usbdrv/usbdrv.c"
|
---|
| 24 |
|
---|
| 25 | /* ------------------------------------------------------------------------ */
|
---|
| 26 |
|
---|
| 27 | /* Request constants used by USBasp */
|
---|
| 28 | #define USBASP_FUNC_CONNECT 1
|
---|
| 29 | #define USBASP_FUNC_DISCONNECT 2
|
---|
| 30 | #define USBASP_FUNC_TRANSMIT 3
|
---|
| 31 | #define USBASP_FUNC_READFLASH 4
|
---|
| 32 | #define USBASP_FUNC_ENABLEPROG 5
|
---|
| 33 | #define USBASP_FUNC_WRITEFLASH 6
|
---|
| 34 | #define USBASP_FUNC_READEEPROM 7
|
---|
| 35 | #define USBASP_FUNC_WRITEEEPROM 8
|
---|
| 36 | #define USBASP_FUNC_SETLONGADDRESS 9
|
---|
| 37 |
|
---|
| 38 | /* ------------------------------------------------------------------------ */
|
---|
| 39 |
|
---|
| 40 | #ifndef ulong
|
---|
| 41 | # define ulong unsigned long
|
---|
| 42 | #endif
|
---|
| 43 | #ifndef uint
|
---|
| 44 | # define uint unsigned int
|
---|
| 45 | #endif
|
---|
| 46 |
|
---|
| 47 | /* defaults if not in config file: */
|
---|
| 48 | #ifndef HAVE_EEPROM_PAGED_ACCESS
|
---|
| 49 | # define HAVE_EEPROM_PAGED_ACCESS 0
|
---|
| 50 | #endif
|
---|
| 51 | #ifndef HAVE_EEPROM_BYTE_ACCESS
|
---|
| 52 | # define HAVE_EEPROM_BYTE_ACCESS 0
|
---|
| 53 | #endif
|
---|
| 54 | #ifndef BOOTLOADER_CAN_EXIT
|
---|
| 55 | # define BOOTLOADER_CAN_EXIT 0
|
---|
| 56 | #endif
|
---|
| 57 |
|
---|
| 58 | /* allow compatibility with avrusbboot's bootloaderconfig.h: */
|
---|
| 59 | #ifdef BOOTLOADER_INIT
|
---|
| 60 | # define bootLoaderInit() BOOTLOADER_INIT
|
---|
| 61 | # define bootLoaderExit()
|
---|
| 62 | #endif
|
---|
| 63 | #ifdef BOOTLOADER_CONDITION
|
---|
| 64 | # define bootLoaderCondition() BOOTLOADER_CONDITION
|
---|
| 65 | #endif
|
---|
| 66 |
|
---|
| 67 | /* device compatibility: */
|
---|
| 68 | #ifndef GICR /* ATMega*8 don't have GICR, use MCUCR instead */
|
---|
| 69 | # define GICR MCUCR
|
---|
| 70 | #endif
|
---|
| 71 |
|
---|
| 72 | /* ------------------------------------------------------------------------ */
|
---|
| 73 |
|
---|
| 74 | #if (FLASHEND) > 0xffff /* we need long addressing */
|
---|
| 75 | # define CURRENT_ADDRESS currentAddress.l
|
---|
| 76 | # define addr_t ulong
|
---|
| 77 | #else
|
---|
| 78 | # define CURRENT_ADDRESS currentAddress.w[0]
|
---|
| 79 | # define addr_t uint
|
---|
| 80 | #endif
|
---|
| 81 |
|
---|
| 82 | typedef union longConverter{
|
---|
| 83 | addr_t l;
|
---|
| 84 | uint w[sizeof(addr_t)/2];
|
---|
| 85 | uchar b[sizeof(addr_t)];
|
---|
| 86 | }longConverter_t;
|
---|
| 87 |
|
---|
| 88 | static uchar requestBootLoaderExit;
|
---|
| 89 | static longConverter_t currentAddress; /* in bytes */
|
---|
| 90 | static uchar bytesRemaining;
|
---|
| 91 | static uchar isLastPage;
|
---|
| 92 | #if HAVE_EEPROM_PAGED_ACCESS
|
---|
| 93 | static uchar currentRequest;
|
---|
| 94 | #else
|
---|
| 95 | static const uchar currentRequest = 0;
|
---|
| 96 | #endif
|
---|
| 97 |
|
---|
| 98 | static const uchar signatureBytes[4] = {
|
---|
| 99 | #ifdef SIGNATURE_BYTES
|
---|
| 100 | SIGNATURE_BYTES
|
---|
| 101 | #elif defined (__AVR_ATmega8__) || defined (__AVR_ATmega8HVA__)
|
---|
| 102 | 0x1e, 0x93, 0x07, 0
|
---|
| 103 | #elif defined (__AVR_ATmega48__) || defined (__AVR_ATmega48P__)
|
---|
| 104 | 0x1e, 0x92, 0x05, 0
|
---|
| 105 | #elif defined (__AVR_ATmega88__) || defined (__AVR_ATmega88P__)
|
---|
| 106 | 0x1e, 0x93, 0x0a, 0
|
---|
| 107 | #elif defined (__AVR_ATmega168__) || defined (__AVR_ATmega168P__)
|
---|
| 108 | 0x1e, 0x94, 0x06, 0
|
---|
| 109 | #elif defined (__AVR_ATmega328P__)
|
---|
| 110 | 0x1e, 0x95, 0x0f, 0
|
---|
| 111 | #else
|
---|
| 112 | # error "Device signature is not known, please edit main.c!"
|
---|
| 113 | #endif
|
---|
| 114 | };
|
---|
| 115 |
|
---|
| 116 | /* ------------------------------------------------------------------------ */
|
---|
| 117 |
|
---|
| 118 | static void (*nullVector)(void) __attribute__((__noreturn__));
|
---|
| 119 |
|
---|
| 120 | static void leaveBootloader()
|
---|
| 121 | {
|
---|
| 122 | #define OBDEBUG 1DBG1(0x01, 0, 0);
|
---|
| 123 | bootLoaderExit();
|
---|
| 124 | cli();
|
---|
| 125 | USB_INTR_ENABLE = 0;
|
---|
| 126 | USB_INTR_CFG = 0; /* also reset config bits */
|
---|
| 127 | GICR = (1 << IVCE); /* enable change of interrupt vectors */
|
---|
| 128 | GICR = (0 << IVSEL); /* move interrupts to application flash section */
|
---|
| 129 | /* We must go through a global function pointer variable instead of writing
|
---|
| 130 | * ((void (*)(void))0)();
|
---|
| 131 | * because the compiler optimizes a constant 0 to "rcall 0" which is not
|
---|
| 132 | * handled correctly by the assembler.
|
---|
| 133 | */
|
---|
| 134 | nullVector();
|
---|
| 135 | }
|
---|
| 136 |
|
---|
| 137 | /* ------------------------------------------------------------------------ */
|
---|
| 138 |
|
---|
| 139 | uchar usbFunctionSetup(uchar data[8])
|
---|
| 140 | {
|
---|
| 141 | usbRequest_t *rq = (void *)data;
|
---|
| 142 | uchar len = 0;
|
---|
| 143 | static uchar replyBuffer[4];
|
---|
| 144 |
|
---|
| 145 | usbMsgPtr = replyBuffer;
|
---|
| 146 | if(rq->bRequest == USBASP_FUNC_TRANSMIT){ /* emulate parts of ISP protocol */
|
---|
| 147 | uchar rval = 0;
|
---|
| 148 | usbWord_t address;
|
---|
| 149 | address.bytes[1] = rq->wValue.bytes[1];
|
---|
| 150 | address.bytes[0] = rq->wIndex.bytes[0];
|
---|
| 151 | if(rq->wValue.bytes[0] == 0x30){ /* read signature */
|
---|
| 152 | rval = rq->wIndex.bytes[0] & 3;
|
---|
| 153 | rval = signatureBytes[rval];
|
---|
| 154 | #if HAVE_EEPROM_BYTE_ACCESS
|
---|
| 155 | }else if(rq->wValue.bytes[0] == 0xa0){ /* read EEPROM byte */
|
---|
| 156 | rval = eeprom_read_byte((void *)address.word);
|
---|
| 157 | }else if(rq->wValue.bytes[0] == 0xc0){ /* write EEPROM byte */
|
---|
| 158 | eeprom_write_byte((void *)address.word, rq->wIndex.bytes[1]);
|
---|
| 159 | #endif
|
---|
| 160 | #if HAVE_CHIP_ERASE
|
---|
| 161 | }else if(rq->wValue.bytes[0] == 0xac && rq->wValue.bytes[1] == 0x80){ /* chip erase */
|
---|
| 162 | addr_t addr;
|
---|
| 163 | for(addr = 0; addr < FLASHEND + 1 - 2048; addr += SPM_PAGESIZE) {
|
---|
| 164 | /* wait and erase page */
|
---|
| 165 | //DBG1(0x33, 0, 0);
|
---|
| 166 | # ifndef NO_FLASH_WRITE
|
---|
| 167 | boot_spm_busy_wait();
|
---|
| 168 | cli();
|
---|
| 169 | boot_page_erase(addr);
|
---|
| 170 | sei();
|
---|
| 171 | # endif
|
---|
| 172 | }
|
---|
| 173 | #endif
|
---|
| 174 | }else{
|
---|
| 175 | /* ignore all others, return default value == 0 */
|
---|
| 176 | }
|
---|
| 177 | replyBuffer[3] = rval;
|
---|
| 178 | len = 4;
|
---|
| 179 | }else if(rq->bRequest == USBASP_FUNC_ENABLEPROG){
|
---|
| 180 | /* replyBuffer[0] = 0; is never touched and thus always 0 which means success */
|
---|
| 181 | len = 1;
|
---|
| 182 | }else if(rq->bRequest >= USBASP_FUNC_READFLASH && rq->bRequest <= USBASP_FUNC_SETLONGADDRESS){
|
---|
| 183 | currentAddress.w[0] = rq->wValue.word;
|
---|
| 184 | if(rq->bRequest == USBASP_FUNC_SETLONGADDRESS){
|
---|
| 185 | #if (FLASHEND) > 0xffff
|
---|
| 186 | currentAddress.w[1] = rq->wIndex.word;
|
---|
| 187 | #endif
|
---|
| 188 | }else{
|
---|
| 189 | bytesRemaining = rq->wLength.bytes[0];
|
---|
| 190 | /* if(rq->bRequest == USBASP_FUNC_WRITEFLASH) only evaluated during writeFlash anyway */
|
---|
| 191 | isLastPage = rq->wIndex.bytes[1] & 0x02;
|
---|
| 192 | #if HAVE_EEPROM_PAGED_ACCESS
|
---|
| 193 | currentRequest = rq->bRequest;
|
---|
| 194 | #endif
|
---|
| 195 | len = 0xff; /* hand over to usbFunctionRead() / usbFunctionWrite() */
|
---|
| 196 | }
|
---|
| 197 | #if BOOTLOADER_CAN_EXIT
|
---|
| 198 | }else if(rq->bRequest == USBASP_FUNC_DISCONNECT){
|
---|
| 199 | requestBootLoaderExit = 1; /* allow proper shutdown/close of connection */
|
---|
| 200 | #endif
|
---|
| 201 | }else{
|
---|
| 202 | /* ignore: USBASP_FUNC_CONNECT */
|
---|
| 203 | }
|
---|
| 204 | return len;
|
---|
| 205 | }
|
---|
| 206 |
|
---|
| 207 | uchar usbFunctionWrite(uchar *data, uchar len)
|
---|
| 208 | {
|
---|
| 209 | uchar isLast;
|
---|
| 210 |
|
---|
| 211 | //DBG1(0x31, (void *)¤tAddress.l, 4);
|
---|
| 212 | if(len > bytesRemaining)
|
---|
| 213 | len = bytesRemaining;
|
---|
| 214 | bytesRemaining -= len;
|
---|
| 215 | isLast = bytesRemaining == 0;
|
---|
| 216 | if(currentRequest >= USBASP_FUNC_READEEPROM){
|
---|
| 217 | uchar i;
|
---|
| 218 | for(i = 0; i < len; i++){
|
---|
| 219 | eeprom_write_byte((void *)(currentAddress.w[0]++), *data++);
|
---|
| 220 | }
|
---|
| 221 | }else{
|
---|
| 222 | uchar i;
|
---|
| 223 | for(i = 0; i < len;){
|
---|
| 224 | #if !HAVE_CHIP_ERASE
|
---|
| 225 | if((currentAddress.w[0] & (SPM_PAGESIZE - 1)) == 0){ /* if page start: erase */
|
---|
| 226 | //DBG1(0x33, 0, 0);
|
---|
| 227 | # ifndef NO_FLASH_WRITE
|
---|
| 228 | cli();
|
---|
| 229 | boot_page_erase(CURRENT_ADDRESS); /* erase page */
|
---|
| 230 | sei();
|
---|
| 231 | boot_spm_busy_wait(); /* wait until page is erased */
|
---|
| 232 | # endif
|
---|
| 233 | }
|
---|
| 234 | #endif
|
---|
| 235 | i += 2;
|
---|
| 236 | //DBG1(0x32, 0, 0);
|
---|
| 237 | cli();
|
---|
| 238 | boot_page_fill(CURRENT_ADDRESS, *(short *)data);
|
---|
| 239 | sei();
|
---|
| 240 | CURRENT_ADDRESS += 2;
|
---|
| 241 | data += 2;
|
---|
| 242 | /* write page when we cross page boundary or we have the last partial page */
|
---|
| 243 | if((currentAddress.w[0] & (SPM_PAGESIZE - 1)) == 0 || (isLast && i >= len && isLastPage)){
|
---|
| 244 | //DBG1(0x34, 0, 0);
|
---|
| 245 | #ifndef NO_FLASH_WRITE
|
---|
| 246 | cli();
|
---|
| 247 | boot_page_write(CURRENT_ADDRESS - 2);
|
---|
| 248 | sei();
|
---|
| 249 | boot_spm_busy_wait();
|
---|
| 250 | cli();
|
---|
| 251 | boot_rww_enable();
|
---|
| 252 | sei();
|
---|
| 253 | #endif
|
---|
| 254 | }
|
---|
| 255 | }
|
---|
| 256 | //DBG1(0x35, (void *)¤tAddress.l, 4);
|
---|
| 257 | }
|
---|
| 258 | return isLast;
|
---|
| 259 | }
|
---|
| 260 |
|
---|
| 261 | uchar usbFunctionRead(uchar *data, uchar len)
|
---|
| 262 | {
|
---|
| 263 | uchar i;
|
---|
| 264 |
|
---|
| 265 | if(len > bytesRemaining)
|
---|
| 266 | len = bytesRemaining;
|
---|
| 267 | bytesRemaining -= len;
|
---|
| 268 | for(i = 0; i < len; i++){
|
---|
| 269 | if(currentRequest >= USBASP_FUNC_READEEPROM){
|
---|
| 270 | *data = eeprom_read_byte((void *)currentAddress.w[0]);
|
---|
| 271 | }else{
|
---|
| 272 | *data = pgm_read_byte((void *)CURRENT_ADDRESS);
|
---|
| 273 | }
|
---|
| 274 | data++;
|
---|
| 275 | CURRENT_ADDRESS++;
|
---|
| 276 | }
|
---|
| 277 | return len;
|
---|
| 278 | }
|
---|
| 279 |
|
---|
| 280 | /* ------------------------------------------------------------------------ */
|
---|
| 281 |
|
---|
| 282 | static void initForUsbConnectivity(void)
|
---|
| 283 | {
|
---|
| 284 | uchar i = 0;
|
---|
| 285 |
|
---|
| 286 | usbInit();
|
---|
| 287 | /* enforce USB re-enumerate: */
|
---|
| 288 | #if 0
|
---|
| 289 | usbDeviceDisconnect(); /* do this while interrupts are disabled */
|
---|
| 290 | while(--i){ /* fake USB disconnect for > 250 ms */
|
---|
| 291 | wdt_reset();
|
---|
| 292 | _delay_ms(1);
|
---|
| 293 | }
|
---|
| 294 | #endif
|
---|
| 295 | usbDeviceConnect();
|
---|
| 296 | sei();
|
---|
| 297 | }
|
---|
| 298 |
|
---|
| 299 | int main(void)
|
---|
| 300 | {
|
---|
| 301 | /* initialize */
|
---|
| 302 | wdt_disable(); /* main app may have enabled watchdog */
|
---|
| 303 | bootLoaderInit();
|
---|
| 304 | odDebugInit();
|
---|
| 305 | DBG1(0x00, 0, 0);
|
---|
| 306 | #ifndef NO_FLASH_WRITE
|
---|
| 307 | GICR = (1 << IVCE); /* enable change of interrupt vectors */
|
---|
| 308 | GICR = (1 << IVSEL); /* move interrupts to boot flash section */
|
---|
| 309 | #endif
|
---|
| 310 | if(bootLoaderCondition()){
|
---|
| 311 | #if BOOTLOADER_CAN_EXIT
|
---|
| 312 | uchar i = 0, j = 0;
|
---|
| 313 | #endif
|
---|
| 314 | initForUsbConnectivity();
|
---|
| 315 | do{
|
---|
| 316 | usbPoll();
|
---|
| 317 | #if BOOTLOADER_CAN_EXIT
|
---|
| 318 | if(requestBootLoaderExit){
|
---|
| 319 | if(--i == 0){
|
---|
| 320 | if(--j == 0)
|
---|
| 321 | break;
|
---|
| 322 | }
|
---|
| 323 | }
|
---|
| 324 | #endif
|
---|
| 325 | }while(bootLoaderCondition()); /* main event loop */
|
---|
| 326 | }
|
---|
| 327 | leaveBootloader();
|
---|
| 328 | return 0;
|
---|
| 329 | }
|
---|
| 330 |
|
---|
| 331 | /* ------------------------------------------------------------------------ */
|
---|