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
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
|
/*
* DiskLoader.cpp
*/
#include "Platform.h"
//#include "USBCore.h"
//extern "C"
void entrypoint(void) __attribute__ ((naked)) __attribute__ ((section (".vectors")));
void entrypoint(void)
{
asm volatile (
"eor r1, r1\n" // Zero register
"out 0x3F, r1\n" // SREG
"ldi r28, 0xFF\n" // Y-register
"ldi r29, 0x0A\n" // Y-register
// "out 0x3E, r29\n" // SPH
// "out 0x3D, r28\n" // SPL
"rjmp main" // Stack is all set up, start the main code
::);
}
u8 _flashbuf[128];
u8 _inSync;
u8 _ok;
volatile u16 _timeout;
void Program(u8 ep, u16 page, u8 count)
{
u8 write = page < 30*1024; // Don't write over firmware please
if (write)
boot_page_erase(page);
USB_Recv_block(ep,_flashbuf,count); // Read while page is erasing
if (!write)
return;
boot_spm_busy_wait(); // Wait until the memory is erased.
count >>= 1;
u16* p = (u16*)page;
u16* b = (u16*)_flashbuf;
for (u8 i = 0; i < count; i++)
boot_page_fill(p++, b[i]);
boot_page_write(page);
boot_spm_busy_wait();
boot_rww_enable ();
}
#define HW_VER 0x02
#define SW_MAJOR 0x01
#define SW_MINOR 0x10
#define STK_OK 0x10
#define STK_INSYNC 0x14 // ' '
#define CRC_EOP 0x20 // 'SPACE'
#define STK_GET_SYNC 0x30 // '0'
#define STK_GET_PARAMETER 0x41 // 'A'
#define STK_SET_DEVICE 0x42 // 'B'
#define STK_SET_DEVICE_EXT 0x45 // 'E'
#define STK_LOAD_ADDRESS 0x55 // 'U'
#define STK_UNIVERSAL 0x56 // 'V'
#define STK_PROG_PAGE 0x64 // 'd'
#define STK_READ_PAGE 0x74 // 't'
#define STK_READ_SIGN 0x75 // 'u'
extern const u8 _readSize[] PROGMEM;
const u8 _readSize[] =
{
STK_GET_PARAMETER, 1,
STK_SET_DEVICE, 20,
STK_SET_DEVICE_EXT, 5,
STK_UNIVERSAL, 4,
STK_LOAD_ADDRESS, 2,
STK_PROG_PAGE, 3,
STK_READ_PAGE, 3,
0,0
};
extern const u8 _consts[] PROGMEM;
const u8 _consts[] =
{
SIGNATURE_0,
SIGNATURE_1,
SIGNATURE_2,
HW_VER, // Hardware version
SW_MAJOR, // Software major version
SW_MINOR, // Software minor version
0x03, // Unknown but seems to be required by avr studio 3.56
0x00, //
};
int getch(void)
{
u16 timeout;
u8 c;
for (timeout = 0; timeout; timeout--)
{
c = USB_Recv(CDC_RX);
if (c != -1)
return c;
}
return -1;
}
int main(void) __attribute__ ((naked));
int main()
{
wdt_disable();
BOARD_INIT();
/* move interrupts to boot section:
* uses inline assembly because the procedure must be completed in four cycles.
* seems to fail if called before disabling WDT and calling BOARD_INIT()
*/
asm volatile (
"ldi r16, 0x01\n" // (1<<IVCE) /* Enable change of interrupt vectors */
"out 0x35, r16\n" // MCUCR
"ldi r16, 0x02\n" // (1<<IVSEL) /* Move interrupts to boot flash section */
"out 0x35, r16\n" // MCUCR
);
TX_LED_OFF();
RX_LED_OFF();
L_LED_OFF();
/* UART setup stuff */
/* end UART setup stuff */
USB.attach();
sei();
_inSync = STK_INSYNC;
_ok = STK_OK;
for (;;)
{
u8* packet = _flashbuf;
u16 address = 0;
for (;;)
{
// while (Serial.available() < 1)
// ;
// u8 cmd = Serial.read();
while (!USB_Available(CDC_RX))
;
u8 cmd = USB_Recv(CDC_RX);
// Read packet contents
u8 len;
const u8* rs = _readSize;
for (;;)
{
u8 c = pgm_read_byte(rs++);
len = pgm_read_byte(rs++);
if (c == cmd || c == 0)
break;
}
_timeout = 0;
// Read params
// USB_Recv(CDC_RX, packet, len);
USB_Recv_block(CDC_RX, packet, len);
// Send a response
u8 send = 0;
const u8* pgm = _consts+7;
if (STK_GET_PARAMETER == cmd)
{
u8 i = packet[0] - 0x80;
if (i > 2)
i = (i==0x18) ? 3 : 4; // 0x80:HW_VER,0x81:SW_MAJOR,0x82:SW_MINOR,0x18:3 or 0
pgm = _consts + i + 3;
send = 1;
}
else if (STK_UNIVERSAL == cmd)
{
if (packet[0] == 0x30)
pgm = _consts + packet[2];
send = 1;
}
else if (STK_READ_SIGN == cmd)
{
pgm = _consts;
send = 3;
}
else if (STK_LOAD_ADDRESS == cmd)
{
address = *((u16*)packet); // word address
address += address;
}
else if (STK_PROG_PAGE == cmd)
{
Program(CDC_RX, address, packet[1]);
}
else if (STK_READ_PAGE == cmd)
{
send = packet[1];
pgm = (const u8*)address;
address += send;
}
// Check sync
// if (Serial.available() > 0 && Serial.read() != ' ')
// break;
// if (USB_Available(CDC_RX) && USB_Recv(CDC_RX) != ' ')
// u8 countdown = 10;
// while (!USB_Available(CDC_RX))
// {
// if (countdown-- == 0)
// break;
// }
// u8 x = USB_Recv(CDC_RX);
// if (x != -1 && x != ' ')
// {
// L_LED_ON();
// break;
// }
// if (getch() != ' ')
// break;
// while (!USB_Available(CDC_RX))
// ;
//
// int x = USB_Recv(CDC_RX);
// if (x == -1)
// {
// UEINTX = 0x6B;
// break;
// }
// else if (x != ' ')
// {
//// UEINTX = 0x6B;
// break;
// }
u16 countdown = 5000;
while (countdown-- > 10 && !USB_Available(CDC_RX))
;
if (USB_Recv(CDC_RX) != ' ')
break;
USB_Send(CDC_TX, &_inSync, 1);
if (send)
USB_Send(CDC_TX|TRANSFER_PGM, pgm, send);
// Send ok
USB_Send(CDC_TX|TRANSFER_RELEASE, &_ok, 1);
if ('Q' == cmd)
break;
}
}
}
|
