]>
Commit | Line | Data |
---|---|---|
fe8c2806 | 1 | /* |
2f09413f LC |
2 | * Copyright 1994, 1995, 2000 Neil Russell. |
3 | * (See License) | |
4 | * Copyright 2000, 2001 DENX Software Engineering, Wolfgang Denk, wd@denx.de | |
fe8c2806 WD |
5 | */ |
6 | ||
7 | #include <common.h> | |
8 | #include <command.h> | |
9 | #include <net.h> | |
10 | #include "tftp.h" | |
11 | #include "bootp.h" | |
12 | ||
2f09413f LC |
13 | /* Well known TFTP port # */ |
14 | #define WELL_KNOWN_PORT 69 | |
15 | /* Millisecs to timeout for lost pkt */ | |
16 | #define TIMEOUT 5000UL | |
fe8c2806 | 17 | #ifndef CONFIG_NET_RETRY_COUNT |
2f09413f LC |
18 | /* # of timeouts before giving up */ |
19 | # define TIMEOUT_COUNT 10 | |
fe8c2806 WD |
20 | #else |
21 | # define TIMEOUT_COUNT (CONFIG_NET_RETRY_COUNT * 2) | |
22 | #endif | |
2f09413f LC |
23 | /* Number of "loading" hashes per line (for checking the image size) */ |
24 | #define HASHES_PER_LINE 65 | |
fe8c2806 WD |
25 | |
26 | /* | |
27 | * TFTP operations. | |
28 | */ | |
29 | #define TFTP_RRQ 1 | |
30 | #define TFTP_WRQ 2 | |
31 | #define TFTP_DATA 3 | |
32 | #define TFTP_ACK 4 | |
33 | #define TFTP_ERROR 5 | |
fbe4b5cb | 34 | #define TFTP_OACK 6 |
fe8c2806 | 35 | |
e83cc063 BS |
36 | static ulong TftpTimeoutMSecs = TIMEOUT; |
37 | static int TftpTimeoutCountMax = TIMEOUT_COUNT; | |
38 | ||
39 | /* | |
40 | * These globals govern the timeout behavior when attempting a connection to a | |
41 | * TFTP server. TftpRRQTimeoutMSecs specifies the number of milliseconds to | |
42 | * wait for the server to respond to initial connection. Second global, | |
43 | * TftpRRQTimeoutCountMax, gives the number of such connection retries. | |
44 | * TftpRRQTimeoutCountMax must be non-negative and TftpRRQTimeoutMSecs must be | |
45 | * positive. The globals are meant to be set (and restored) by code needing | |
46 | * non-standard timeout behavior when initiating a TFTP transfer. | |
47 | */ | |
48 | ulong TftpRRQTimeoutMSecs = TIMEOUT; | |
49 | int TftpRRQTimeoutCountMax = TIMEOUT_COUNT; | |
50 | ||
aafda38f RB |
51 | enum { |
52 | TFTP_ERR_UNDEFINED = 0, | |
53 | TFTP_ERR_FILE_NOT_FOUND = 1, | |
54 | TFTP_ERR_ACCESS_DENIED = 2, | |
55 | TFTP_ERR_DISK_FULL = 3, | |
56 | TFTP_ERR_UNEXPECTED_OPCODE = 4, | |
57 | TFTP_ERR_UNKNOWN_TRANSFER_ID = 5, | |
58 | TFTP_ERR_FILE_ALREADY_EXISTS = 6, | |
59 | }; | |
60 | ||
a93907c4 | 61 | static IPaddr_t TftpServerIP; |
2f09413f LC |
62 | /* The UDP port at their end */ |
63 | static int TftpServerPort; | |
64 | /* The UDP port at our end */ | |
65 | static int TftpOurPort; | |
fe8c2806 | 66 | static int TftpTimeoutCount; |
2f09413f LC |
67 | /* packet sequence number */ |
68 | static ulong TftpBlock; | |
69 | /* last packet sequence number received */ | |
70 | static ulong TftpLastBlock; | |
71 | /* count of sequence number wraparounds */ | |
72 | static ulong TftpBlockWrap; | |
73 | /* memory offset due to wrapping */ | |
74 | static ulong TftpBlockWrapOffset; | |
fe8c2806 | 75 | static int TftpState; |
4fccb818 | 76 | #ifdef CONFIG_TFTP_TSIZE |
2f09413f LC |
77 | /* The file size reported by the server */ |
78 | static int TftpTsize; | |
79 | /* The number of hashes we printed */ | |
80 | static short TftpNumchars; | |
4fccb818 | 81 | #endif |
3f85ce27 | 82 | |
fe8c2806 WD |
83 | #define STATE_RRQ 1 |
84 | #define STATE_DATA 2 | |
85 | #define STATE_TOO_LARGE 3 | |
86 | #define STATE_BAD_MAGIC 4 | |
fbe4b5cb | 87 | #define STATE_OACK 5 |
fe8c2806 | 88 | |
2f09413f LC |
89 | /* default TFTP block size */ |
90 | #define TFTP_BLOCK_SIZE 512 | |
91 | /* sequence number is 16 bit */ | |
92 | #define TFTP_SEQUENCE_SIZE ((ulong)(1<<16)) | |
3f85ce27 | 93 | |
fe8c2806 WD |
94 | #define DEFAULT_NAME_LEN (8 + 4 + 1) |
95 | static char default_filename[DEFAULT_NAME_LEN]; | |
a93907c4 JCPV |
96 | |
97 | #ifndef CONFIG_TFTP_FILE_NAME_MAX_LEN | |
98 | #define MAX_LEN 128 | |
99 | #else | |
100 | #define MAX_LEN CONFIG_TFTP_FILE_NAME_MAX_LEN | |
101 | #endif | |
102 | ||
103 | static char tftp_filename[MAX_LEN]; | |
fe8c2806 | 104 | |
6d0f6bcf | 105 | #ifdef CONFIG_SYS_DIRECT_FLASH_TFTP |
e6f2e902 | 106 | extern flash_info_t flash_info[]; |
fe8c2806 WD |
107 | #endif |
108 | ||
85eb5caf WD |
109 | /* 512 is poor choice for ethernet, MTU is typically 1500. |
110 | * Minus eth.hdrs thats 1468. Can get 2x better throughput with | |
53a5c424 | 111 | * almost-MTU block sizes. At least try... fall back to 512 if need be. |
89ba81d1 | 112 | * (but those using CONFIG_IP_DEFRAG may want to set a larger block in cfg file) |
53a5c424 | 113 | */ |
89ba81d1 AR |
114 | #ifdef CONFIG_TFTP_BLOCKSIZE |
115 | #define TFTP_MTU_BLOCKSIZE CONFIG_TFTP_BLOCKSIZE | |
116 | #else | |
53a5c424 | 117 | #define TFTP_MTU_BLOCKSIZE 1468 |
89ba81d1 AR |
118 | #endif |
119 | ||
c718b143 LC |
120 | static unsigned short TftpBlkSize = TFTP_BLOCK_SIZE; |
121 | static unsigned short TftpBlkSizeOption = TFTP_MTU_BLOCKSIZE; | |
53a5c424 DU |
122 | |
123 | #ifdef CONFIG_MCAST_TFTP | |
124 | #include <malloc.h> | |
125 | #define MTFTP_BITMAPSIZE 0x1000 | |
126 | static unsigned *Bitmap; | |
c718b143 LC |
127 | static int PrevBitmapHole, Mapsize = MTFTP_BITMAPSIZE; |
128 | static uchar ProhibitMcast = 0, MasterClient = 0; | |
129 | static uchar Multicast = 0; | |
53a5c424 DU |
130 | extern IPaddr_t Mcast_addr; |
131 | static int Mcast_port; | |
132 | static ulong TftpEndingBlock; /* can get 'last' block before done..*/ | |
133 | ||
c718b143 | 134 | static void parse_multicast_oack(char *pkt, int len); |
53a5c424 DU |
135 | |
136 | static void | |
137 | mcast_cleanup(void) | |
138 | { | |
139 | if (Mcast_addr) eth_mcast_join(Mcast_addr, 0); | |
140 | if (Bitmap) free(Bitmap); | |
c718b143 | 141 | Bitmap = NULL; |
53a5c424 DU |
142 | Mcast_addr = Multicast = Mcast_port = 0; |
143 | TftpEndingBlock = -1; | |
144 | } | |
145 | ||
146 | #endif /* CONFIG_MCAST_TFTP */ | |
147 | ||
fe8c2806 | 148 | static __inline__ void |
c718b143 | 149 | store_block(unsigned block, uchar * src, unsigned len) |
fe8c2806 | 150 | { |
53a5c424 | 151 | ulong offset = block * TftpBlkSize + TftpBlockWrapOffset; |
3f85ce27 | 152 | ulong newsize = offset + len; |
6d0f6bcf | 153 | #ifdef CONFIG_SYS_DIRECT_FLASH_TFTP |
fe8c2806 WD |
154 | int i, rc = 0; |
155 | ||
c718b143 | 156 | for (i = 0; i < CONFIG_SYS_MAX_FLASH_BANKS; i++) { |
fe8c2806 | 157 | /* start address in flash? */ |
be1b0d27 JF |
158 | if (flash_info[i].flash_id == FLASH_UNKNOWN) |
159 | continue; | |
fe8c2806 WD |
160 | if (load_addr + offset >= flash_info[i].start[0]) { |
161 | rc = 1; | |
162 | break; | |
163 | } | |
164 | } | |
165 | ||
166 | if (rc) { /* Flash is destination for this packet */ | |
c718b143 | 167 | rc = flash_write((char *)src, (ulong)(load_addr+offset), len); |
fe8c2806 | 168 | if (rc) { |
c718b143 | 169 | flash_perror(rc); |
fe8c2806 WD |
170 | NetState = NETLOOP_FAIL; |
171 | return; | |
172 | } | |
173 | } | |
174 | else | |
6d0f6bcf | 175 | #endif /* CONFIG_SYS_DIRECT_FLASH_TFTP */ |
fe8c2806 WD |
176 | { |
177 | (void)memcpy((void *)(load_addr + offset), src, len); | |
178 | } | |
53a5c424 DU |
179 | #ifdef CONFIG_MCAST_TFTP |
180 | if (Multicast) | |
181 | ext2_set_bit(block, Bitmap); | |
182 | #endif | |
fe8c2806 WD |
183 | |
184 | if (NetBootFileXferSize < newsize) | |
185 | NetBootFileXferSize = newsize; | |
186 | } | |
187 | ||
c718b143 LC |
188 | static void TftpSend(void); |
189 | static void TftpTimeout(void); | |
fe8c2806 WD |
190 | |
191 | /**********************************************************************/ | |
192 | ||
193 | static void | |
c718b143 | 194 | TftpSend(void) |
fe8c2806 WD |
195 | { |
196 | volatile uchar * pkt; | |
197 | volatile uchar * xp; | |
198 | int len = 0; | |
7bc5ee07 | 199 | volatile ushort *s; |
fe8c2806 | 200 | |
85eb5caf | 201 | #ifdef CONFIG_MCAST_TFTP |
53a5c424 | 202 | /* Multicast TFTP.. non-MasterClients do not ACK data. */ |
85eb5caf WD |
203 | if (Multicast |
204 | && (TftpState == STATE_DATA) | |
205 | && (MasterClient == 0)) | |
53a5c424 DU |
206 | return; |
207 | #endif | |
fe8c2806 WD |
208 | /* |
209 | * We will always be sending some sort of packet, so | |
210 | * cobble together the packet headers now. | |
211 | */ | |
a3d991bd | 212 | pkt = NetTxPacket + NetEthHdrSize() + IP_HDR_SIZE; |
fe8c2806 WD |
213 | |
214 | switch (TftpState) { | |
215 | ||
216 | case STATE_RRQ: | |
217 | xp = pkt; | |
7bc5ee07 WD |
218 | s = (ushort *)pkt; |
219 | *s++ = htons(TFTP_RRQ); | |
220 | pkt = (uchar *)s; | |
c718b143 | 221 | strcpy((char *)pkt, tftp_filename); |
fe8c2806 | 222 | pkt += strlen(tftp_filename) + 1; |
c718b143 | 223 | strcpy((char *)pkt, "octet"); |
fe8c2806 | 224 | pkt += 5 /*strlen("octet")*/ + 1; |
c718b143 | 225 | strcpy((char *)pkt, "timeout"); |
fbe4b5cb | 226 | pkt += 7 /*strlen("timeout")*/ + 1; |
c96f86ee | 227 | sprintf((char *)pkt, "%lu", TftpTimeoutMSecs / 1000); |
0ebf04c6 | 228 | debug("send option \"timeout %s\"\n", (char *)pkt); |
fbe4b5cb | 229 | pkt += strlen((char *)pkt) + 1; |
4fccb818 RG |
230 | #ifdef CONFIG_TFTP_TSIZE |
231 | memcpy((char *)pkt, "tsize\0000\0", 8); | |
232 | pkt += 8; | |
233 | #endif | |
53a5c424 | 234 | /* try for more effic. blk size */ |
c718b143 LC |
235 | pkt += sprintf((char *)pkt, "blksize%c%d%c", |
236 | 0, TftpBlkSizeOption, 0); | |
85eb5caf | 237 | #ifdef CONFIG_MCAST_TFTP |
53a5c424 | 238 | /* Check all preconditions before even trying the option */ |
85eb5caf | 239 | if (!ProhibitMcast |
c718b143 | 240 | && (Bitmap = malloc(Mapsize)) |
53a5c424 DU |
241 | && eth_get_dev()->mcast) { |
242 | free(Bitmap); | |
c718b143 LC |
243 | Bitmap = NULL; |
244 | pkt += sprintf((char *)pkt, "multicast%c%c", 0, 0); | |
53a5c424 DU |
245 | } |
246 | #endif /* CONFIG_MCAST_TFTP */ | |
fe8c2806 WD |
247 | len = pkt - xp; |
248 | break; | |
249 | ||
fbe4b5cb | 250 | case STATE_OACK: |
53a5c424 DU |
251 | #ifdef CONFIG_MCAST_TFTP |
252 | /* My turn! Start at where I need blocks I missed.*/ | |
253 | if (Multicast) | |
c718b143 LC |
254 | TftpBlock = ext2_find_next_zero_bit(Bitmap, |
255 | (Mapsize*8), 0); | |
53a5c424 DU |
256 | /*..falling..*/ |
257 | #endif | |
258 | case STATE_DATA: | |
fe8c2806 | 259 | xp = pkt; |
7bc5ee07 WD |
260 | s = (ushort *)pkt; |
261 | *s++ = htons(TFTP_ACK); | |
262 | *s++ = htons(TftpBlock); | |
263 | pkt = (uchar *)s; | |
fe8c2806 WD |
264 | len = pkt - xp; |
265 | break; | |
266 | ||
267 | case STATE_TOO_LARGE: | |
268 | xp = pkt; | |
7bc5ee07 WD |
269 | s = (ushort *)pkt; |
270 | *s++ = htons(TFTP_ERROR); | |
271 | *s++ = htons(3); | |
272 | pkt = (uchar *)s; | |
c718b143 | 273 | strcpy((char *)pkt, "File too large"); |
fe8c2806 WD |
274 | pkt += 14 /*strlen("File too large")*/ + 1; |
275 | len = pkt - xp; | |
276 | break; | |
277 | ||
278 | case STATE_BAD_MAGIC: | |
279 | xp = pkt; | |
7bc5ee07 WD |
280 | s = (ushort *)pkt; |
281 | *s++ = htons(TFTP_ERROR); | |
282 | *s++ = htons(2); | |
283 | pkt = (uchar *)s; | |
c718b143 | 284 | strcpy((char *)pkt, "File has bad magic"); |
fe8c2806 WD |
285 | pkt += 18 /*strlen("File has bad magic")*/ + 1; |
286 | len = pkt - xp; | |
287 | break; | |
288 | } | |
289 | ||
2f09413f LC |
290 | NetSendUDPPacket(NetServerEther, TftpServerIP, TftpServerPort, |
291 | TftpOurPort, len); | |
fe8c2806 WD |
292 | } |
293 | ||
294 | ||
295 | static void | |
03eb129f LC |
296 | TftpHandler(uchar *pkt, unsigned dest, IPaddr_t sip, unsigned src, |
297 | unsigned len) | |
fe8c2806 WD |
298 | { |
299 | ushort proto; | |
7bc5ee07 | 300 | ushort *s; |
ff13ac8c | 301 | int i; |
fe8c2806 WD |
302 | |
303 | if (dest != TftpOurPort) { | |
53a5c424 | 304 | #ifdef CONFIG_MCAST_TFTP |
85eb5caf | 305 | if (Multicast |
53a5c424 DU |
306 | && (!Mcast_port || (dest != Mcast_port))) |
307 | #endif | |
fe8c2806 WD |
308 | return; |
309 | } | |
310 | if (TftpState != STATE_RRQ && src != TftpServerPort) { | |
311 | return; | |
312 | } | |
313 | ||
314 | if (len < 2) { | |
315 | return; | |
316 | } | |
317 | len -= 2; | |
318 | /* warning: don't use increment (++) in ntohs() macros!! */ | |
7bc5ee07 WD |
319 | s = (ushort *)pkt; |
320 | proto = *s++; | |
321 | pkt = (uchar *)s; | |
fe8c2806 WD |
322 | switch (ntohs(proto)) { |
323 | ||
324 | case TFTP_RRQ: | |
325 | case TFTP_WRQ: | |
326 | case TFTP_ACK: | |
327 | break; | |
328 | default: | |
329 | break; | |
330 | ||
fbe4b5cb | 331 | case TFTP_OACK: |
d371708a WD |
332 | debug("Got OACK: %s %s\n", |
333 | pkt, | |
334 | pkt + strlen((char *)pkt) + 1); | |
fbe4b5cb WD |
335 | TftpState = STATE_OACK; |
336 | TftpServerPort = src; | |
60174746 WD |
337 | /* |
338 | * Check for 'blksize' option. | |
339 | * Careful: "i" is signed, "len" is unsigned, thus | |
340 | * something like "len-8" may give a *huge* number | |
341 | */ | |
c718b143 LC |
342 | for (i = 0; i+8 < len; i++) { |
343 | if (strcmp((char*)pkt+i, "blksize") == 0) { | |
ff13ac8c | 344 | TftpBlkSize = (unsigned short) |
c718b143 LC |
345 | simple_strtoul((char*)pkt+i+8, NULL, |
346 | 10); | |
0ebf04c6 | 347 | debug("Blocksize ack: %s, %d\n", |
c718b143 | 348 | (char*)pkt+i+8, TftpBlkSize); |
ff13ac8c | 349 | } |
4fccb818 | 350 | #ifdef CONFIG_TFTP_TSIZE |
c718b143 | 351 | if (strcmp((char*)pkt+i, "tsize") == 0) { |
2f09413f LC |
352 | TftpTsize = simple_strtoul((char*)pkt+i+6, |
353 | NULL, 10); | |
4fccb818 RG |
354 | debug("size = %s, %d\n", |
355 | (char*)pkt+i+6, TftpTsize); | |
356 | } | |
357 | #endif | |
53a5c424 DU |
358 | } |
359 | #ifdef CONFIG_MCAST_TFTP | |
c718b143 | 360 | parse_multicast_oack((char *)pkt, len-1); |
85eb5caf | 361 | if ((Multicast) && (!MasterClient)) |
53a5c424 DU |
362 | TftpState = STATE_DATA; /* passive.. */ |
363 | else | |
364 | #endif | |
c718b143 | 365 | TftpSend(); /* Send ACK */ |
fbe4b5cb | 366 | break; |
fe8c2806 WD |
367 | case TFTP_DATA: |
368 | if (len < 2) | |
369 | return; | |
370 | len -= 2; | |
371 | TftpBlock = ntohs(*(ushort *)pkt); | |
3f85ce27 WD |
372 | |
373 | /* | |
cd0a9de6 WD |
374 | * RFC1350 specifies that the first data packet will |
375 | * have sequence number 1. If we receive a sequence | |
376 | * number of 0 this means that there was a wrap | |
377 | * around of the (16 bit) counter. | |
3f85ce27 WD |
378 | */ |
379 | if (TftpBlock == 0) { | |
380 | TftpBlockWrap++; | |
2f09413f LC |
381 | TftpBlockWrapOffset += |
382 | TftpBlkSize * TFTP_SEQUENCE_SIZE; | |
c718b143 | 383 | printf("\n\t %lu MB received\n\t ", |
2f09413f | 384 | TftpBlockWrapOffset>>20); |
4fccb818 RG |
385 | } |
386 | #ifdef CONFIG_TFTP_TSIZE | |
387 | else if (TftpTsize) { | |
2f09413f LC |
388 | while (TftpNumchars < |
389 | NetBootFileXferSize * 50 / TftpTsize) { | |
4fccb818 RG |
390 | putc('#'); |
391 | TftpNumchars++; | |
392 | } | |
393 | } | |
394 | #endif | |
395 | else { | |
3f85ce27 | 396 | if (((TftpBlock - 1) % 10) == 0) { |
c718b143 | 397 | putc('#'); |
3f85ce27 | 398 | } else if ((TftpBlock % (10 * HASHES_PER_LINE)) == 0) { |
c718b143 | 399 | puts("\n\t "); |
3f85ce27 | 400 | } |
fe8c2806 WD |
401 | } |
402 | ||
0ebf04c6 RG |
403 | if (TftpState == STATE_RRQ) |
404 | debug("Server did not acknowledge timeout option!\n"); | |
fbe4b5cb WD |
405 | |
406 | if (TftpState == STATE_RRQ || TftpState == STATE_OACK) { | |
3f85ce27 | 407 | /* first block received */ |
fe8c2806 WD |
408 | TftpState = STATE_DATA; |
409 | TftpServerPort = src; | |
410 | TftpLastBlock = 0; | |
3f85ce27 WD |
411 | TftpBlockWrap = 0; |
412 | TftpBlockWrapOffset = 0; | |
fe8c2806 | 413 | |
53a5c424 DU |
414 | #ifdef CONFIG_MCAST_TFTP |
415 | if (Multicast) { /* start!=1 common if mcast */ | |
416 | TftpLastBlock = TftpBlock - 1; | |
417 | } else | |
418 | #endif | |
fe8c2806 | 419 | if (TftpBlock != 1) { /* Assertion */ |
c718b143 LC |
420 | printf("\nTFTP error: " |
421 | "First block is not block 1 (%ld)\n" | |
422 | "Starting again\n\n", | |
fe8c2806 | 423 | TftpBlock); |
c718b143 | 424 | NetStartAgain(); |
fe8c2806 WD |
425 | break; |
426 | } | |
427 | } | |
428 | ||
429 | if (TftpBlock == TftpLastBlock) { | |
430 | /* | |
431 | * Same block again; ignore it. | |
432 | */ | |
433 | break; | |
434 | } | |
435 | ||
436 | TftpLastBlock = TftpBlock; | |
e83cc063 | 437 | TftpTimeoutCountMax = TIMEOUT_COUNT; |
c718b143 | 438 | NetSetTimeout(TftpTimeoutMSecs, TftpTimeout); |
fe8c2806 | 439 | |
c718b143 | 440 | store_block(TftpBlock - 1, pkt + 2, len); |
fe8c2806 WD |
441 | |
442 | /* | |
443 | * Acknoledge the block just received, which will prompt | |
444 | * the server for the next one. | |
445 | */ | |
53a5c424 | 446 | #ifdef CONFIG_MCAST_TFTP |
85eb5caf WD |
447 | /* if I am the MasterClient, actively calculate what my next |
448 | * needed block is; else I'm passive; not ACKING | |
a93907c4 | 449 | */ |
53a5c424 DU |
450 | if (Multicast) { |
451 | if (len < TftpBlkSize) { | |
452 | TftpEndingBlock = TftpBlock; | |
453 | } else if (MasterClient) { | |
85eb5caf | 454 | TftpBlock = PrevBitmapHole = |
53a5c424 DU |
455 | ext2_find_next_zero_bit( |
456 | Bitmap, | |
457 | (Mapsize*8), | |
458 | PrevBitmapHole); | |
459 | if (TftpBlock > ((Mapsize*8) - 1)) { | |
c718b143 | 460 | printf("tftpfile too big\n"); |
53a5c424 | 461 | /* try to double it and retry */ |
c718b143 | 462 | Mapsize <<= 1; |
53a5c424 | 463 | mcast_cleanup(); |
c718b143 | 464 | NetStartAgain(); |
53a5c424 DU |
465 | return; |
466 | } | |
467 | TftpLastBlock = TftpBlock; | |
468 | } | |
469 | } | |
470 | #endif | |
c718b143 | 471 | TftpSend(); |
fe8c2806 | 472 | |
53a5c424 DU |
473 | #ifdef CONFIG_MCAST_TFTP |
474 | if (Multicast) { | |
475 | if (MasterClient && (TftpBlock >= TftpEndingBlock)) { | |
c718b143 | 476 | puts("\nMulticast tftp done\n"); |
53a5c424 DU |
477 | mcast_cleanup(); |
478 | NetState = NETLOOP_SUCCESS; | |
85eb5caf | 479 | } |
53a5c424 DU |
480 | } |
481 | else | |
482 | #endif | |
483 | if (len < TftpBlkSize) { | |
fe8c2806 WD |
484 | /* |
485 | * We received the whole thing. Try to | |
486 | * run it. | |
487 | */ | |
4fccb818 | 488 | #ifdef CONFIG_TFTP_TSIZE |
2f09413f | 489 | /* Print hash marks for the last packet received */ |
4fccb818 RG |
490 | while (TftpTsize && TftpNumchars < 49) { |
491 | putc('#'); | |
492 | TftpNumchars++; | |
493 | } | |
494 | #endif | |
c718b143 | 495 | puts("\ndone\n"); |
fe8c2806 WD |
496 | NetState = NETLOOP_SUCCESS; |
497 | } | |
498 | break; | |
499 | ||
500 | case TFTP_ERROR: | |
c718b143 LC |
501 | printf("\nTFTP error: '%s' (%d)\n", |
502 | pkt + 2, ntohs(*(ushort *)pkt)); | |
aafda38f RB |
503 | |
504 | switch (ntohs(*(ushort *)pkt)) { | |
505 | case TFTP_ERR_FILE_NOT_FOUND: | |
506 | case TFTP_ERR_ACCESS_DENIED: | |
507 | puts("Not retrying...\n"); | |
508 | eth_halt(); | |
509 | NetState = NETLOOP_FAIL; | |
510 | break; | |
511 | case TFTP_ERR_UNDEFINED: | |
512 | case TFTP_ERR_DISK_FULL: | |
513 | case TFTP_ERR_UNEXPECTED_OPCODE: | |
514 | case TFTP_ERR_UNKNOWN_TRANSFER_ID: | |
515 | case TFTP_ERR_FILE_ALREADY_EXISTS: | |
516 | default: | |
517 | puts("Starting again\n\n"); | |
53a5c424 | 518 | #ifdef CONFIG_MCAST_TFTP |
aafda38f | 519 | mcast_cleanup(); |
53a5c424 | 520 | #endif |
aafda38f RB |
521 | NetStartAgain(); |
522 | break; | |
523 | } | |
fe8c2806 WD |
524 | break; |
525 | } | |
526 | } | |
527 | ||
528 | ||
529 | static void | |
c718b143 | 530 | TftpTimeout(void) |
fe8c2806 | 531 | { |
e83cc063 | 532 | if (++TftpTimeoutCount > TftpTimeoutCountMax) { |
c718b143 | 533 | puts("\nRetry count exceeded; starting again\n"); |
53a5c424 DU |
534 | #ifdef CONFIG_MCAST_TFTP |
535 | mcast_cleanup(); | |
536 | #endif | |
c718b143 | 537 | NetStartAgain(); |
fe8c2806 | 538 | } else { |
c718b143 LC |
539 | puts("T "); |
540 | NetSetTimeout(TftpTimeoutMSecs, TftpTimeout); | |
541 | TftpSend(); | |
fe8c2806 WD |
542 | } |
543 | } | |
544 | ||
545 | ||
546 | void | |
c718b143 | 547 | TftpStart(void) |
fe8c2806 | 548 | { |
ecb0ccd9 | 549 | char *ep; /* Environment pointer */ |
89ba81d1 | 550 | |
c96f86ee WD |
551 | /* |
552 | * Allow the user to choose TFTP blocksize and timeout. | |
553 | * TFTP protocol has a minimal timeout of 1 second. | |
554 | */ | |
89ba81d1 AR |
555 | if ((ep = getenv("tftpblocksize")) != NULL) |
556 | TftpBlkSizeOption = simple_strtol(ep, NULL, 10); | |
c96f86ee WD |
557 | |
558 | if ((ep = getenv("tftptimeout")) != NULL) | |
559 | TftpTimeoutMSecs = simple_strtol(ep, NULL, 10); | |
560 | ||
561 | if (TftpTimeoutMSecs < 1000) { | |
562 | printf("TFTP timeout (%ld ms) too low, " | |
563 | "set minimum = 1000 ms\n", | |
564 | TftpTimeoutMSecs); | |
565 | TftpTimeoutMSecs = 1000; | |
566 | } | |
567 | ||
568 | debug("TFTP blocksize = %i, timeout = %ld ms\n", | |
569 | TftpBlkSizeOption, TftpTimeoutMSecs); | |
ecb0ccd9 | 570 | |
a93907c4 | 571 | TftpServerIP = NetServerIP; |
fe8c2806 | 572 | if (BootFile[0] == '\0') { |
fe8c2806 | 573 | sprintf(default_filename, "%02lX%02lX%02lX%02lX.img", |
c43352cc WD |
574 | NetOurIP & 0xFF, |
575 | (NetOurIP >> 8) & 0xFF, | |
576 | (NetOurIP >> 16) & 0xFF, | |
c718b143 | 577 | (NetOurIP >> 24) & 0xFF); |
a93907c4 JCPV |
578 | |
579 | strncpy(tftp_filename, default_filename, MAX_LEN); | |
580 | tftp_filename[MAX_LEN-1] = 0; | |
fe8c2806 | 581 | |
c718b143 | 582 | printf("*** Warning: no boot file name; using '%s'\n", |
fe8c2806 WD |
583 | tftp_filename); |
584 | } else { | |
c718b143 | 585 | char *p = strchr(BootFile, ':'); |
a93907c4 JCPV |
586 | |
587 | if (p == NULL) { | |
588 | strncpy(tftp_filename, BootFile, MAX_LEN); | |
589 | tftp_filename[MAX_LEN-1] = 0; | |
590 | } else { | |
c718b143 | 591 | TftpServerIP = string_to_ip(BootFile); |
6a86bb6c | 592 | strncpy(tftp_filename, p + 1, MAX_LEN); |
a93907c4 JCPV |
593 | tftp_filename[MAX_LEN-1] = 0; |
594 | } | |
fe8c2806 WD |
595 | } |
596 | ||
5bb226e8 | 597 | #if defined(CONFIG_NET_MULTI) |
c718b143 | 598 | printf("Using %s device\n", eth_get_name()); |
5bb226e8 | 599 | #endif |
b6446b67 MF |
600 | printf("TFTP from server %pI4" |
601 | "; our IP address is %pI4", &TftpServerIP, &NetOurIP); | |
fe8c2806 WD |
602 | |
603 | /* Check if we need to send across this subnet */ | |
604 | if (NetOurGatewayIP && NetOurSubnetMask) { | |
a93907c4 JCPV |
605 | IPaddr_t OurNet = NetOurIP & NetOurSubnetMask; |
606 | IPaddr_t ServerNet = TftpServerIP & NetOurSubnetMask; | |
fe8c2806 | 607 | |
b6446b67 MF |
608 | if (OurNet != ServerNet) |
609 | printf("; sending through gateway %pI4", &NetOurGatewayIP); | |
fe8c2806 | 610 | } |
c718b143 | 611 | putc('\n'); |
fe8c2806 | 612 | |
c718b143 | 613 | printf("Filename '%s'.", tftp_filename); |
fe8c2806 WD |
614 | |
615 | if (NetBootFileSize) { | |
c718b143 LC |
616 | printf(" Size is 0x%x Bytes = ", NetBootFileSize<<9); |
617 | print_size(NetBootFileSize<<9, ""); | |
fe8c2806 WD |
618 | } |
619 | ||
c718b143 | 620 | putc('\n'); |
fe8c2806 | 621 | |
c718b143 | 622 | printf("Load address: 0x%lx\n", load_addr); |
fe8c2806 | 623 | |
c718b143 | 624 | puts("Loading: *\b"); |
fe8c2806 | 625 | |
e83cc063 BS |
626 | TftpTimeoutCountMax = TftpRRQTimeoutCountMax; |
627 | ||
c718b143 LC |
628 | NetSetTimeout(TftpTimeoutMSecs, TftpTimeout); |
629 | NetSetHandler(TftpHandler); | |
fe8c2806 WD |
630 | |
631 | TftpServerPort = WELL_KNOWN_PORT; | |
632 | TftpTimeoutCount = 0; | |
633 | TftpState = STATE_RRQ; | |
ecb0ccd9 | 634 | /* Use a pseudo-random port unless a specific port is set */ |
fe8c2806 | 635 | TftpOurPort = 1024 + (get_timer(0) % 3072); |
53a5c424 | 636 | |
ecb0ccd9 | 637 | #ifdef CONFIG_TFTP_PORT |
28cb9375 WD |
638 | if ((ep = getenv("tftpdstp")) != NULL) { |
639 | TftpServerPort = simple_strtol(ep, NULL, 10); | |
640 | } | |
641 | if ((ep = getenv("tftpsrcp")) != NULL) { | |
c718b143 | 642 | TftpOurPort = simple_strtol(ep, NULL, 10); |
ecb0ccd9 WD |
643 | } |
644 | #endif | |
fbe4b5cb | 645 | TftpBlock = 0; |
fe8c2806 | 646 | |
73a8b27c WD |
647 | /* zero out server ether in case the server ip has changed */ |
648 | memset(NetServerEther, 0, 6); | |
53a5c424 DU |
649 | /* Revert TftpBlkSize to dflt */ |
650 | TftpBlkSize = TFTP_BLOCK_SIZE; | |
651 | #ifdef CONFIG_MCAST_TFTP | |
a93907c4 | 652 | mcast_cleanup(); |
53a5c424 | 653 | #endif |
4fccb818 RG |
654 | #ifdef CONFIG_TFTP_TSIZE |
655 | TftpTsize = 0; | |
656 | TftpNumchars = 0; | |
657 | #endif | |
73a8b27c | 658 | |
c718b143 | 659 | TftpSend(); |
fe8c2806 WD |
660 | } |
661 | ||
53a5c424 DU |
662 | #ifdef CONFIG_MCAST_TFTP |
663 | /* Credits: atftp project. | |
664 | */ | |
665 | ||
666 | /* pick up BcastAddr, Port, and whether I am [now] the master-client. * | |
667 | * Frame: | |
668 | * +-------+-----------+---+-------~~-------+---+ | |
669 | * | opc | multicast | 0 | addr, port, mc | 0 | | |
670 | * +-------+-----------+---+-------~~-------+---+ | |
671 | * The multicast addr/port becomes what I listen to, and if 'mc' is '1' then | |
672 | * I am the new master-client so must send ACKs to DataBlocks. If I am not | |
673 | * master-client, I'm a passive client, gathering what DataBlocks I may and | |
674 | * making note of which ones I got in my bitmask. | |
675 | * In theory, I never go from master->passive.. | |
676 | * .. this comes in with pkt already pointing just past opc | |
677 | */ | |
678 | static void parse_multicast_oack(char *pkt, int len) | |
679 | { | |
c718b143 LC |
680 | int i; |
681 | IPaddr_t addr; | |
682 | char *mc_adr, *port, *mc; | |
53a5c424 | 683 | |
c718b143 | 684 | mc_adr = port = mc = NULL; |
53a5c424 DU |
685 | /* march along looking for 'multicast\0', which has to start at least |
686 | * 14 bytes back from the end. | |
687 | */ | |
c718b143 LC |
688 | for (i = 0; i < len-14; i++) |
689 | if (strcmp(pkt+i, "multicast") == 0) | |
53a5c424 DU |
690 | break; |
691 | if (i >= (len-14)) /* non-Multicast OACK, ign. */ | |
692 | return; | |
85eb5caf | 693 | |
c718b143 | 694 | i += 10; /* strlen multicast */ |
53a5c424 | 695 | mc_adr = pkt+i; |
c718b143 | 696 | for (; i < len; i++) { |
53a5c424 DU |
697 | if (*(pkt+i) == ',') { |
698 | *(pkt+i) = '\0'; | |
699 | if (port) { | |
700 | mc = pkt+i+1; | |
701 | break; | |
702 | } else { | |
703 | port = pkt+i+1; | |
704 | } | |
705 | } | |
706 | } | |
c718b143 | 707 | if (!port || !mc_adr || !mc) return; |
53a5c424 | 708 | if (Multicast && MasterClient) { |
c718b143 | 709 | printf("I got a OACK as master Client, WRONG!\n"); |
53a5c424 DU |
710 | return; |
711 | } | |
712 | /* ..I now accept packets destined for this MCAST addr, port */ | |
713 | if (!Multicast) { | |
714 | if (Bitmap) { | |
c718b143 | 715 | printf("Internal failure! no mcast.\n"); |
53a5c424 | 716 | free(Bitmap); |
c718b143 LC |
717 | Bitmap = NULL; |
718 | ProhibitMcast = 1; | |
53a5c424 DU |
719 | return ; |
720 | } | |
721 | /* I malloc instead of pre-declare; so that if the file ends | |
85eb5caf WD |
722 | * up being too big for this bitmap I can retry |
723 | */ | |
c718b143 LC |
724 | if (!(Bitmap = malloc(Mapsize))) { |
725 | printf("No Bitmap, no multicast. Sorry.\n"); | |
726 | ProhibitMcast = 1; | |
53a5c424 DU |
727 | return; |
728 | } | |
c718b143 | 729 | memset(Bitmap, 0, Mapsize); |
53a5c424 DU |
730 | PrevBitmapHole = 0; |
731 | Multicast = 1; | |
732 | } | |
733 | addr = string_to_ip(mc_adr); | |
734 | if (Mcast_addr != addr) { | |
735 | if (Mcast_addr) | |
736 | eth_mcast_join(Mcast_addr, 0); | |
c718b143 LC |
737 | if (eth_mcast_join(Mcast_addr = addr, 1)) { |
738 | printf("Fail to set mcast, revert to TFTP\n"); | |
739 | ProhibitMcast = 1; | |
53a5c424 DU |
740 | mcast_cleanup(); |
741 | NetStartAgain(); | |
742 | } | |
743 | } | |
c718b143 LC |
744 | MasterClient = (unsigned char)simple_strtoul((char *)mc, NULL, 10); |
745 | Mcast_port = (unsigned short)simple_strtoul(port, NULL, 10); | |
746 | printf("Multicast: %s:%d [%d]\n", mc_adr, Mcast_port, MasterClient); | |
53a5c424 DU |
747 | return; |
748 | } | |
749 | ||
750 | #endif /* Multicast TFTP */ |