]>
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 | |
e59e3562 LC |
5 | * Copyright 2011 Comelit Group SpA, |
6 | * Luca Ceresoli <luca.ceresoli@comelit.it> | |
fe8c2806 WD |
7 | */ |
8 | ||
9 | #include <common.h> | |
10 | #include <command.h> | |
55d5fd9a | 11 | #include <mapmem.h> |
fe8c2806 WD |
12 | #include <net.h> |
13 | #include "tftp.h" | |
14 | #include "bootp.h" | |
13dfe943 JH |
15 | #ifdef CONFIG_SYS_DIRECT_FLASH_TFTP |
16 | #include <flash.h> | |
17 | #endif | |
fe8c2806 | 18 | |
2f09413f LC |
19 | /* Well known TFTP port # */ |
20 | #define WELL_KNOWN_PORT 69 | |
21 | /* Millisecs to timeout for lost pkt */ | |
22 | #define TIMEOUT 5000UL | |
fe8c2806 | 23 | #ifndef CONFIG_NET_RETRY_COUNT |
2f09413f LC |
24 | /* # of timeouts before giving up */ |
25 | # define TIMEOUT_COUNT 10 | |
fe8c2806 WD |
26 | #else |
27 | # define TIMEOUT_COUNT (CONFIG_NET_RETRY_COUNT * 2) | |
28 | #endif | |
2f09413f LC |
29 | /* Number of "loading" hashes per line (for checking the image size) */ |
30 | #define HASHES_PER_LINE 65 | |
fe8c2806 WD |
31 | |
32 | /* | |
33 | * TFTP operations. | |
34 | */ | |
35 | #define TFTP_RRQ 1 | |
36 | #define TFTP_WRQ 2 | |
37 | #define TFTP_DATA 3 | |
38 | #define TFTP_ACK 4 | |
39 | #define TFTP_ERROR 5 | |
fbe4b5cb | 40 | #define TFTP_OACK 6 |
fe8c2806 | 41 | |
8885c5fe JH |
42 | static ulong timeout_ms = TIMEOUT; |
43 | static int timeout_count_max = TIMEOUT_COUNT; | |
85b19802 | 44 | static ulong time_start; /* Record time we started tftp */ |
e83cc063 BS |
45 | |
46 | /* | |
47 | * These globals govern the timeout behavior when attempting a connection to a | |
8885c5fe | 48 | * TFTP server. tftp_timeout_ms specifies the number of milliseconds to |
e83cc063 | 49 | * wait for the server to respond to initial connection. Second global, |
8885c5fe JH |
50 | * tftp_timeout_count_max, gives the number of such connection retries. |
51 | * tftp_timeout_count_max must be non-negative and tftp_timeout_ms must be | |
e83cc063 BS |
52 | * positive. The globals are meant to be set (and restored) by code needing |
53 | * non-standard timeout behavior when initiating a TFTP transfer. | |
54 | */ | |
8885c5fe JH |
55 | ulong tftp_timeout_ms = TIMEOUT; |
56 | int tftp_timeout_count_max = TIMEOUT_COUNT; | |
e83cc063 | 57 | |
aafda38f RB |
58 | enum { |
59 | TFTP_ERR_UNDEFINED = 0, | |
60 | TFTP_ERR_FILE_NOT_FOUND = 1, | |
61 | TFTP_ERR_ACCESS_DENIED = 2, | |
62 | TFTP_ERR_DISK_FULL = 3, | |
63 | TFTP_ERR_UNEXPECTED_OPCODE = 4, | |
64 | TFTP_ERR_UNKNOWN_TRANSFER_ID = 5, | |
65 | TFTP_ERR_FILE_ALREADY_EXISTS = 6, | |
66 | }; | |
67 | ||
049a95a7 | 68 | static struct in_addr tftp_remote_ip; |
2f09413f | 69 | /* The UDP port at their end */ |
8885c5fe | 70 | static int tftp_remote_port; |
2f09413f | 71 | /* The UDP port at our end */ |
8885c5fe JH |
72 | static int tftp_our_port; |
73 | static int timeout_count; | |
2f09413f | 74 | /* packet sequence number */ |
8885c5fe | 75 | static ulong tftp_cur_block; |
2f09413f | 76 | /* last packet sequence number received */ |
8885c5fe | 77 | static ulong tftp_prev_block; |
2f09413f | 78 | /* count of sequence number wraparounds */ |
8885c5fe | 79 | static ulong tftp_block_wrap; |
2f09413f | 80 | /* memory offset due to wrapping */ |
8885c5fe JH |
81 | static ulong tftp_block_wrap_offset; |
82 | static int tftp_state; | |
4fccb818 | 83 | #ifdef CONFIG_TFTP_TSIZE |
2f09413f | 84 | /* The file size reported by the server */ |
8885c5fe | 85 | static int tftp_tsize; |
2f09413f | 86 | /* The number of hashes we printed */ |
8885c5fe | 87 | static short tftp_tsize_num_hash; |
4fccb818 | 88 | #endif |
1fb7cd49 | 89 | #ifdef CONFIG_CMD_TFTPPUT |
8885c5fe JH |
90 | /* 1 if writing, else 0 */ |
91 | static int tftp_put_active; | |
92 | /* 1 if we have sent the last block */ | |
93 | static int tftp_put_final_block_sent; | |
1fb7cd49 | 94 | #else |
8885c5fe | 95 | #define tftp_put_active 0 |
1fb7cd49 | 96 | #endif |
3f85ce27 | 97 | |
e3fb0abe | 98 | #define STATE_SEND_RRQ 1 |
fe8c2806 WD |
99 | #define STATE_DATA 2 |
100 | #define STATE_TOO_LARGE 3 | |
101 | #define STATE_BAD_MAGIC 4 | |
fbe4b5cb | 102 | #define STATE_OACK 5 |
e59e3562 | 103 | #define STATE_RECV_WRQ 6 |
1fb7cd49 | 104 | #define STATE_SEND_WRQ 7 |
fe8c2806 | 105 | |
2f09413f LC |
106 | /* default TFTP block size */ |
107 | #define TFTP_BLOCK_SIZE 512 | |
108 | /* sequence number is 16 bit */ | |
109 | #define TFTP_SEQUENCE_SIZE ((ulong)(1<<16)) | |
3f85ce27 | 110 | |
fe8c2806 WD |
111 | #define DEFAULT_NAME_LEN (8 + 4 + 1) |
112 | static char default_filename[DEFAULT_NAME_LEN]; | |
a93907c4 JCPV |
113 | |
114 | #ifndef CONFIG_TFTP_FILE_NAME_MAX_LEN | |
115 | #define MAX_LEN 128 | |
116 | #else | |
117 | #define MAX_LEN CONFIG_TFTP_FILE_NAME_MAX_LEN | |
118 | #endif | |
119 | ||
120 | static char tftp_filename[MAX_LEN]; | |
fe8c2806 | 121 | |
85eb5caf WD |
122 | /* 512 is poor choice for ethernet, MTU is typically 1500. |
123 | * Minus eth.hdrs thats 1468. Can get 2x better throughput with | |
53a5c424 | 124 | * almost-MTU block sizes. At least try... fall back to 512 if need be. |
89ba81d1 | 125 | * (but those using CONFIG_IP_DEFRAG may want to set a larger block in cfg file) |
53a5c424 | 126 | */ |
89ba81d1 AR |
127 | #ifdef CONFIG_TFTP_BLOCKSIZE |
128 | #define TFTP_MTU_BLOCKSIZE CONFIG_TFTP_BLOCKSIZE | |
129 | #else | |
53a5c424 | 130 | #define TFTP_MTU_BLOCKSIZE 1468 |
89ba81d1 AR |
131 | #endif |
132 | ||
8885c5fe JH |
133 | static unsigned short tftp_block_size = TFTP_BLOCK_SIZE; |
134 | static unsigned short tftp_block_size_option = TFTP_MTU_BLOCKSIZE; | |
53a5c424 DU |
135 | |
136 | #ifdef CONFIG_MCAST_TFTP | |
137 | #include <malloc.h> | |
138 | #define MTFTP_BITMAPSIZE 0x1000 | |
8885c5fe JH |
139 | static unsigned *tftp_mcast_bitmap; |
140 | static int tftp_mcast_prev_hole; | |
141 | static int tftp_mcast_bitmap_size = MTFTP_BITMAPSIZE; | |
142 | static int tftp_mcast_disabled; | |
143 | static int tftp_mcast_master_client; | |
144 | static int tftp_mcast_active; | |
145 | static int tftp_mcast_port; | |
146 | /* can get 'last' block before done..*/ | |
147 | static ulong tftp_mcast_ending_block; | |
53a5c424 | 148 | |
c718b143 | 149 | static void parse_multicast_oack(char *pkt, int len); |
53a5c424 | 150 | |
8885c5fe | 151 | static void mcast_cleanup(void) |
53a5c424 | 152 | { |
049a95a7 JH |
153 | if (net_mcast_addr) |
154 | eth_mcast_join(net_mcast_addr, 0); | |
8885c5fe JH |
155 | if (tftp_mcast_bitmap) |
156 | free(tftp_mcast_bitmap); | |
157 | tftp_mcast_bitmap = NULL; | |
049a95a7 | 158 | net_mcast_addr.s_addr = 0; |
8885c5fe JH |
159 | tftp_mcast_active = 0; |
160 | tftp_mcast_port = 0; | |
161 | tftp_mcast_ending_block = -1; | |
53a5c424 DU |
162 | } |
163 | ||
164 | #endif /* CONFIG_MCAST_TFTP */ | |
165 | ||
8885c5fe | 166 | static inline void store_block(int block, uchar *src, unsigned len) |
fe8c2806 | 167 | { |
8885c5fe | 168 | ulong offset = block * tftp_block_size + tftp_block_wrap_offset; |
3f85ce27 | 169 | ulong newsize = offset + len; |
6d0f6bcf | 170 | #ifdef CONFIG_SYS_DIRECT_FLASH_TFTP |
fe8c2806 WD |
171 | int i, rc = 0; |
172 | ||
c718b143 | 173 | for (i = 0; i < CONFIG_SYS_MAX_FLASH_BANKS; i++) { |
fe8c2806 | 174 | /* start address in flash? */ |
be1b0d27 JF |
175 | if (flash_info[i].flash_id == FLASH_UNKNOWN) |
176 | continue; | |
fe8c2806 WD |
177 | if (load_addr + offset >= flash_info[i].start[0]) { |
178 | rc = 1; | |
179 | break; | |
180 | } | |
181 | } | |
182 | ||
183 | if (rc) { /* Flash is destination for this packet */ | |
c718b143 | 184 | rc = flash_write((char *)src, (ulong)(load_addr+offset), len); |
fe8c2806 | 185 | if (rc) { |
c718b143 | 186 | flash_perror(rc); |
22f6e99d | 187 | net_set_state(NETLOOP_FAIL); |
fe8c2806 WD |
188 | return; |
189 | } | |
13dfe943 | 190 | } else |
6d0f6bcf | 191 | #endif /* CONFIG_SYS_DIRECT_FLASH_TFTP */ |
fe8c2806 | 192 | { |
55d5fd9a JH |
193 | void *ptr = map_sysmem(load_addr + offset, len); |
194 | ||
195 | memcpy(ptr, src, len); | |
196 | unmap_sysmem(ptr); | |
fe8c2806 | 197 | } |
53a5c424 | 198 | #ifdef CONFIG_MCAST_TFTP |
8885c5fe JH |
199 | if (tftp_mcast_active) |
200 | ext2_set_bit(block, tftp_mcast_bitmap); | |
53a5c424 | 201 | #endif |
fe8c2806 | 202 | |
1411157d JH |
203 | if (net_boot_file_size < newsize) |
204 | net_boot_file_size = newsize; | |
fe8c2806 WD |
205 | } |
206 | ||
e4cde2f7 | 207 | /* Clear our state ready for a new transfer */ |
165099e7 | 208 | static void new_transfer(void) |
e4cde2f7 | 209 | { |
8885c5fe JH |
210 | tftp_prev_block = 0; |
211 | tftp_block_wrap = 0; | |
212 | tftp_block_wrap_offset = 0; | |
e4cde2f7 | 213 | #ifdef CONFIG_CMD_TFTPPUT |
8885c5fe | 214 | tftp_put_final_block_sent = 0; |
e4cde2f7 SG |
215 | #endif |
216 | } | |
217 | ||
1fb7cd49 SG |
218 | #ifdef CONFIG_CMD_TFTPPUT |
219 | /** | |
220 | * Load the next block from memory to be sent over tftp. | |
221 | * | |
222 | * @param block Block number to send | |
223 | * @param dst Destination buffer for data | |
224 | * @param len Number of bytes in block (this one and every other) | |
225 | * @return number of bytes loaded | |
226 | */ | |
227 | static int load_block(unsigned block, uchar *dst, unsigned len) | |
228 | { | |
229 | /* We may want to get the final block from the previous set */ | |
8885c5fe | 230 | ulong offset = ((int)block - 1) * len + tftp_block_wrap_offset; |
1fb7cd49 SG |
231 | ulong tosend = len; |
232 | ||
1411157d | 233 | tosend = min(net_boot_file_size - offset, tosend); |
1fb7cd49 SG |
234 | (void)memcpy(dst, (void *)(save_addr + offset), tosend); |
235 | debug("%s: block=%d, offset=%ld, len=%d, tosend=%ld\n", __func__, | |
8885c5fe | 236 | block, offset, len, tosend); |
1fb7cd49 SG |
237 | return tosend; |
238 | } | |
239 | #endif | |
240 | ||
8885c5fe JH |
241 | static void tftp_send(void); |
242 | static void tftp_timeout_handler(void); | |
fe8c2806 WD |
243 | |
244 | /**********************************************************************/ | |
245 | ||
f5329bbc SG |
246 | static void show_block_marker(void) |
247 | { | |
248 | #ifdef CONFIG_TFTP_TSIZE | |
8885c5fe JH |
249 | if (tftp_tsize) { |
250 | ulong pos = tftp_cur_block * tftp_block_size + | |
251 | tftp_block_wrap_offset; | |
f5329bbc | 252 | |
8885c5fe | 253 | while (tftp_tsize_num_hash < pos * 50 / tftp_tsize) { |
f5329bbc | 254 | putc('#'); |
8885c5fe | 255 | tftp_tsize_num_hash++; |
f5329bbc | 256 | } |
1fb7cd49 | 257 | } else |
f5329bbc | 258 | #endif |
1fb7cd49 | 259 | { |
8885c5fe | 260 | if (((tftp_cur_block - 1) % 10) == 0) |
f5329bbc | 261 | putc('#'); |
8885c5fe | 262 | else if ((tftp_cur_block % (10 * HASHES_PER_LINE)) == 0) |
f5329bbc SG |
263 | puts("\n\t "); |
264 | } | |
265 | } | |
266 | ||
e4cde2f7 SG |
267 | /** |
268 | * restart the current transfer due to an error | |
269 | * | |
270 | * @param msg Message to print for user | |
271 | */ | |
272 | static void restart(const char *msg) | |
273 | { | |
274 | printf("\n%s; starting again\n", msg); | |
275 | #ifdef CONFIG_MCAST_TFTP | |
276 | mcast_cleanup(); | |
277 | #endif | |
bc0571fc | 278 | net_start_again(); |
e4cde2f7 SG |
279 | } |
280 | ||
281 | /* | |
282 | * Check if the block number has wrapped, and update progress | |
283 | * | |
284 | * TODO: The egregious use of global variables in this file should be tidied. | |
285 | */ | |
286 | static void update_block_number(void) | |
287 | { | |
288 | /* | |
289 | * RFC1350 specifies that the first data packet will | |
290 | * have sequence number 1. If we receive a sequence | |
291 | * number of 0 this means that there was a wrap | |
292 | * around of the (16 bit) counter. | |
293 | */ | |
8885c5fe JH |
294 | if (tftp_cur_block == 0 && tftp_prev_block != 0) { |
295 | tftp_block_wrap++; | |
296 | tftp_block_wrap_offset += tftp_block_size * TFTP_SEQUENCE_SIZE; | |
297 | timeout_count = 0; /* we've done well, reset the timeout */ | |
e4cde2f7 SG |
298 | } else { |
299 | show_block_marker(); | |
300 | } | |
301 | } | |
302 | ||
f5329bbc SG |
303 | /* The TFTP get or put is complete */ |
304 | static void tftp_complete(void) | |
305 | { | |
306 | #ifdef CONFIG_TFTP_TSIZE | |
307 | /* Print hash marks for the last packet received */ | |
8885c5fe | 308 | while (tftp_tsize && tftp_tsize_num_hash < 49) { |
f5329bbc | 309 | putc('#'); |
8885c5fe | 310 | tftp_tsize_num_hash++; |
f5329bbc | 311 | } |
8104f546 | 312 | puts(" "); |
8885c5fe | 313 | print_size(tftp_tsize, ""); |
f5329bbc | 314 | #endif |
85b19802 SG |
315 | time_start = get_timer(time_start); |
316 | if (time_start > 0) { | |
317 | puts("\n\t "); /* Line up with "Loading: " */ | |
1411157d | 318 | print_size(net_boot_file_size / |
85b19802 SG |
319 | time_start * 1000, "/s"); |
320 | } | |
f5329bbc | 321 | puts("\ndone\n"); |
22f6e99d | 322 | net_set_state(NETLOOP_SUCCESS); |
f5329bbc SG |
323 | } |
324 | ||
8885c5fe | 325 | static void tftp_send(void) |
fe8c2806 | 326 | { |
1fb7cd49 | 327 | uchar *pkt; |
db288a96 JH |
328 | uchar *xp; |
329 | int len = 0; | |
330 | ushort *s; | |
fe8c2806 | 331 | |
85eb5caf | 332 | #ifdef CONFIG_MCAST_TFTP |
53a5c424 | 333 | /* Multicast TFTP.. non-MasterClients do not ACK data. */ |
8885c5fe JH |
334 | if (tftp_mcast_active && tftp_state == STATE_DATA && |
335 | tftp_mcast_master_client == 0) | |
53a5c424 DU |
336 | return; |
337 | #endif | |
fe8c2806 WD |
338 | /* |
339 | * We will always be sending some sort of packet, so | |
340 | * cobble together the packet headers now. | |
341 | */ | |
1203fcce | 342 | pkt = net_tx_packet + net_eth_hdr_size() + IP_UDP_HDR_SIZE; |
fe8c2806 | 343 | |
8885c5fe | 344 | switch (tftp_state) { |
e3fb0abe | 345 | case STATE_SEND_RRQ: |
1fb7cd49 | 346 | case STATE_SEND_WRQ: |
fe8c2806 | 347 | xp = pkt; |
7bc5ee07 | 348 | s = (ushort *)pkt; |
8c6914f1 | 349 | #ifdef CONFIG_CMD_TFTPPUT |
8885c5fe | 350 | *s++ = htons(tftp_state == STATE_SEND_RRQ ? TFTP_RRQ : |
1fb7cd49 | 351 | TFTP_WRQ); |
8c6914f1 SG |
352 | #else |
353 | *s++ = htons(TFTP_RRQ); | |
354 | #endif | |
7bc5ee07 | 355 | pkt = (uchar *)s; |
c718b143 | 356 | strcpy((char *)pkt, tftp_filename); |
fe8c2806 | 357 | pkt += strlen(tftp_filename) + 1; |
c718b143 | 358 | strcpy((char *)pkt, "octet"); |
fe8c2806 | 359 | pkt += 5 /*strlen("octet")*/ + 1; |
c718b143 | 360 | strcpy((char *)pkt, "timeout"); |
fbe4b5cb | 361 | pkt += 7 /*strlen("timeout")*/ + 1; |
8885c5fe | 362 | sprintf((char *)pkt, "%lu", timeout_ms / 1000); |
0ebf04c6 | 363 | debug("send option \"timeout %s\"\n", (char *)pkt); |
fbe4b5cb | 364 | pkt += strlen((char *)pkt) + 1; |
4fccb818 | 365 | #ifdef CONFIG_TFTP_TSIZE |
1411157d JH |
366 | pkt += sprintf((char *)pkt, "tsize%c%u%c", |
367 | 0, net_boot_file_size, 0); | |
4fccb818 | 368 | #endif |
53a5c424 | 369 | /* try for more effic. blk size */ |
c718b143 | 370 | pkt += sprintf((char *)pkt, "blksize%c%d%c", |
8885c5fe | 371 | 0, tftp_block_size_option, 0); |
85eb5caf | 372 | #ifdef CONFIG_MCAST_TFTP |
53a5c424 | 373 | /* Check all preconditions before even trying the option */ |
8885c5fe JH |
374 | if (!tftp_mcast_disabled) { |
375 | tftp_mcast_bitmap = malloc(tftp_mcast_bitmap_size); | |
376 | if (tftp_mcast_bitmap && eth_get_dev()->mcast) { | |
377 | free(tftp_mcast_bitmap); | |
378 | tftp_mcast_bitmap = NULL; | |
13dfe943 JH |
379 | pkt += sprintf((char *)pkt, "multicast%c%c", |
380 | 0, 0); | |
381 | } | |
53a5c424 DU |
382 | } |
383 | #endif /* CONFIG_MCAST_TFTP */ | |
fe8c2806 WD |
384 | len = pkt - xp; |
385 | break; | |
386 | ||
fbe4b5cb | 387 | case STATE_OACK: |
53a5c424 | 388 | #ifdef CONFIG_MCAST_TFTP |
8885c5fe JH |
389 | /* My turn! Start at where I need blocks I missed. */ |
390 | if (tftp_mcast_active) | |
391 | tftp_cur_block = ext2_find_next_zero_bit( | |
392 | tftp_mcast_bitmap, | |
393 | tftp_mcast_bitmap_size * 8, 0); | |
394 | /* fall through */ | |
53a5c424 | 395 | #endif |
e59e3562 LC |
396 | |
397 | case STATE_RECV_WRQ: | |
53a5c424 | 398 | case STATE_DATA: |
fe8c2806 | 399 | xp = pkt; |
7bc5ee07 | 400 | s = (ushort *)pkt; |
1fb7cd49 | 401 | s[0] = htons(TFTP_ACK); |
8885c5fe | 402 | s[1] = htons(tftp_cur_block); |
1fb7cd49 SG |
403 | pkt = (uchar *)(s + 2); |
404 | #ifdef CONFIG_CMD_TFTPPUT | |
8885c5fe JH |
405 | if (tftp_put_active) { |
406 | int toload = tftp_block_size; | |
407 | int loaded = load_block(tftp_cur_block, pkt, toload); | |
1fb7cd49 SG |
408 | |
409 | s[0] = htons(TFTP_DATA); | |
410 | pkt += loaded; | |
8885c5fe | 411 | tftp_put_final_block_sent = (loaded < toload); |
1fb7cd49 SG |
412 | } |
413 | #endif | |
fe8c2806 WD |
414 | len = pkt - xp; |
415 | break; | |
416 | ||
417 | case STATE_TOO_LARGE: | |
418 | xp = pkt; | |
7bc5ee07 WD |
419 | s = (ushort *)pkt; |
420 | *s++ = htons(TFTP_ERROR); | |
1fb7cd49 SG |
421 | *s++ = htons(3); |
422 | ||
7bc5ee07 | 423 | pkt = (uchar *)s; |
c718b143 | 424 | strcpy((char *)pkt, "File too large"); |
fe8c2806 WD |
425 | pkt += 14 /*strlen("File too large")*/ + 1; |
426 | len = pkt - xp; | |
427 | break; | |
428 | ||
429 | case STATE_BAD_MAGIC: | |
430 | xp = pkt; | |
7bc5ee07 WD |
431 | s = (ushort *)pkt; |
432 | *s++ = htons(TFTP_ERROR); | |
433 | *s++ = htons(2); | |
434 | pkt = (uchar *)s; | |
c718b143 | 435 | strcpy((char *)pkt, "File has bad magic"); |
fe8c2806 WD |
436 | pkt += 18 /*strlen("File has bad magic")*/ + 1; |
437 | len = pkt - xp; | |
438 | break; | |
439 | } | |
440 | ||
8885c5fe JH |
441 | net_send_udp_packet(net_server_ethaddr, tftp_remote_ip, |
442 | tftp_remote_port, tftp_our_port, len); | |
fe8c2806 WD |
443 | } |
444 | ||
39bccd21 | 445 | #ifdef CONFIG_CMD_TFTPPUT |
1fb7cd49 | 446 | static void icmp_handler(unsigned type, unsigned code, unsigned dest, |
049a95a7 JH |
447 | struct in_addr sip, unsigned src, uchar *pkt, |
448 | unsigned len) | |
1fb7cd49 SG |
449 | { |
450 | if (type == ICMP_NOT_REACH && code == ICMP_NOT_REACH_PORT) { | |
451 | /* Oh dear the other end has gone away */ | |
452 | restart("TFTP server died"); | |
453 | } | |
454 | } | |
39bccd21 | 455 | #endif |
1fb7cd49 | 456 | |
049a95a7 JH |
457 | static void tftp_handler(uchar *pkt, unsigned dest, struct in_addr sip, |
458 | unsigned src, unsigned len) | |
fe8c2806 | 459 | { |
61fdd4f7 KP |
460 | __be16 proto; |
461 | __be16 *s; | |
ff13ac8c | 462 | int i; |
fe8c2806 | 463 | |
8885c5fe | 464 | if (dest != tftp_our_port) { |
53a5c424 | 465 | #ifdef CONFIG_MCAST_TFTP |
8885c5fe JH |
466 | if (tftp_mcast_active && |
467 | (!tftp_mcast_port || dest != tftp_mcast_port)) | |
53a5c424 | 468 | #endif |
0bdd8acc | 469 | return; |
fe8c2806 | 470 | } |
8885c5fe JH |
471 | if (tftp_state != STATE_SEND_RRQ && src != tftp_remote_port && |
472 | tftp_state != STATE_RECV_WRQ && tftp_state != STATE_SEND_WRQ) | |
fe8c2806 | 473 | return; |
fe8c2806 | 474 | |
7bc325a1 | 475 | if (len < 2) |
fe8c2806 | 476 | return; |
fe8c2806 WD |
477 | len -= 2; |
478 | /* warning: don't use increment (++) in ntohs() macros!! */ | |
61fdd4f7 | 479 | s = (__be16 *)pkt; |
7bc5ee07 WD |
480 | proto = *s++; |
481 | pkt = (uchar *)s; | |
fe8c2806 | 482 | switch (ntohs(proto)) { |
fe8c2806 | 483 | case TFTP_RRQ: |
1fb7cd49 SG |
484 | break; |
485 | ||
fe8c2806 | 486 | case TFTP_ACK: |
1fb7cd49 | 487 | #ifdef CONFIG_CMD_TFTPPUT |
8885c5fe JH |
488 | if (tftp_put_active) { |
489 | if (tftp_put_final_block_sent) { | |
1fb7cd49 SG |
490 | tftp_complete(); |
491 | } else { | |
492 | /* | |
493 | * Move to the next block. We want our block | |
494 | * count to wrap just like the other end! | |
495 | */ | |
496 | int block = ntohs(*s); | |
8885c5fe | 497 | int ack_ok = (tftp_cur_block == block); |
1fb7cd49 | 498 | |
8885c5fe | 499 | tftp_cur_block = (unsigned short)(block + 1); |
1fb7cd49 SG |
500 | update_block_number(); |
501 | if (ack_ok) | |
8885c5fe | 502 | tftp_send(); /* Send next data block */ |
1fb7cd49 SG |
503 | } |
504 | } | |
505 | #endif | |
fe8c2806 | 506 | break; |
1fb7cd49 | 507 | |
fe8c2806 WD |
508 | default: |
509 | break; | |
510 | ||
e59e3562 LC |
511 | #ifdef CONFIG_CMD_TFTPSRV |
512 | case TFTP_WRQ: | |
513 | debug("Got WRQ\n"); | |
049a95a7 | 514 | tftp_remote_ip = sip; |
8885c5fe JH |
515 | tftp_remote_port = src; |
516 | tftp_our_port = 1024 + (get_timer(0) % 3072); | |
e4cde2f7 | 517 | new_transfer(); |
8885c5fe | 518 | tftp_send(); /* Send ACK(0) */ |
e59e3562 LC |
519 | break; |
520 | #endif | |
521 | ||
fbe4b5cb | 522 | case TFTP_OACK: |
d371708a | 523 | debug("Got OACK: %s %s\n", |
8885c5fe JH |
524 | pkt, pkt + strlen((char *)pkt) + 1); |
525 | tftp_state = STATE_OACK; | |
526 | tftp_remote_port = src; | |
60174746 WD |
527 | /* |
528 | * Check for 'blksize' option. | |
529 | * Careful: "i" is signed, "len" is unsigned, thus | |
530 | * something like "len-8" may give a *huge* number | |
531 | */ | |
c718b143 | 532 | for (i = 0; i+8 < len; i++) { |
8885c5fe JH |
533 | if (strcmp((char *)pkt + i, "blksize") == 0) { |
534 | tftp_block_size = (unsigned short) | |
535 | simple_strtoul((char *)pkt + i + 8, | |
536 | NULL, 10); | |
0ebf04c6 | 537 | debug("Blocksize ack: %s, %d\n", |
8885c5fe | 538 | (char *)pkt + i + 8, tftp_block_size); |
ff13ac8c | 539 | } |
4fccb818 | 540 | #ifdef CONFIG_TFTP_TSIZE |
2e320257 | 541 | if (strcmp((char *)pkt+i, "tsize") == 0) { |
8885c5fe | 542 | tftp_tsize = simple_strtoul((char *)pkt + i + 6, |
2f09413f | 543 | NULL, 10); |
4fccb818 | 544 | debug("size = %s, %d\n", |
8885c5fe | 545 | (char *)pkt + i + 6, tftp_tsize); |
4fccb818 RG |
546 | } |
547 | #endif | |
53a5c424 DU |
548 | } |
549 | #ifdef CONFIG_MCAST_TFTP | |
8885c5fe JH |
550 | parse_multicast_oack((char *)pkt, len - 1); |
551 | if ((tftp_mcast_active) && (!tftp_mcast_master_client)) | |
552 | tftp_state = STATE_DATA; /* passive.. */ | |
53a5c424 DU |
553 | else |
554 | #endif | |
1fb7cd49 | 555 | #ifdef CONFIG_CMD_TFTPPUT |
8885c5fe | 556 | if (tftp_put_active) { |
1fb7cd49 | 557 | /* Get ready to send the first block */ |
8885c5fe JH |
558 | tftp_state = STATE_DATA; |
559 | tftp_cur_block++; | |
1fb7cd49 SG |
560 | } |
561 | #endif | |
8885c5fe | 562 | tftp_send(); /* Send ACK or first data block */ |
fbe4b5cb | 563 | break; |
fe8c2806 WD |
564 | case TFTP_DATA: |
565 | if (len < 2) | |
566 | return; | |
567 | len -= 2; | |
8885c5fe | 568 | tftp_cur_block = ntohs(*(__be16 *)pkt); |
3f85ce27 | 569 | |
e4cde2f7 | 570 | update_block_number(); |
fe8c2806 | 571 | |
8885c5fe | 572 | if (tftp_state == STATE_SEND_RRQ) |
0ebf04c6 | 573 | debug("Server did not acknowledge timeout option!\n"); |
fbe4b5cb | 574 | |
8885c5fe JH |
575 | if (tftp_state == STATE_SEND_RRQ || tftp_state == STATE_OACK || |
576 | tftp_state == STATE_RECV_WRQ) { | |
3f85ce27 | 577 | /* first block received */ |
8885c5fe JH |
578 | tftp_state = STATE_DATA; |
579 | tftp_remote_port = src; | |
e4cde2f7 | 580 | new_transfer(); |
fe8c2806 | 581 | |
53a5c424 | 582 | #ifdef CONFIG_MCAST_TFTP |
8885c5fe JH |
583 | if (tftp_mcast_active) { /* start!=1 common if mcast */ |
584 | tftp_prev_block = tftp_cur_block - 1; | |
53a5c424 DU |
585 | } else |
586 | #endif | |
8885c5fe JH |
587 | if (tftp_cur_block != 1) { /* Assertion */ |
588 | puts("\nTFTP error: "); | |
589 | printf("First block is not block 1 (%ld)\n", | |
590 | tftp_cur_block); | |
591 | puts("Starting again\n\n"); | |
bc0571fc | 592 | net_start_again(); |
fe8c2806 WD |
593 | break; |
594 | } | |
595 | } | |
596 | ||
8885c5fe JH |
597 | if (tftp_cur_block == tftp_prev_block) { |
598 | /* Same block again; ignore it. */ | |
fe8c2806 WD |
599 | break; |
600 | } | |
601 | ||
8885c5fe JH |
602 | tftp_prev_block = tftp_cur_block; |
603 | timeout_count_max = TIMEOUT_COUNT; | |
bc0571fc | 604 | net_set_timeout_handler(timeout_ms, tftp_timeout_handler); |
fe8c2806 | 605 | |
8885c5fe | 606 | store_block(tftp_cur_block - 1, pkt + 2, len); |
fe8c2806 WD |
607 | |
608 | /* | |
4d69e98c | 609 | * Acknowledge the block just received, which will prompt |
20478cea | 610 | * the remote for the next one. |
fe8c2806 | 611 | */ |
53a5c424 | 612 | #ifdef CONFIG_MCAST_TFTP |
85eb5caf WD |
613 | /* if I am the MasterClient, actively calculate what my next |
614 | * needed block is; else I'm passive; not ACKING | |
a93907c4 | 615 | */ |
8885c5fe JH |
616 | if (tftp_mcast_active) { |
617 | if (len < tftp_block_size) { | |
618 | tftp_mcast_ending_block = tftp_cur_block; | |
619 | } else if (tftp_mcast_master_client) { | |
620 | tftp_mcast_prev_hole = ext2_find_next_zero_bit( | |
621 | tftp_mcast_bitmap, | |
622 | tftp_mcast_bitmap_size * 8, | |
623 | tftp_mcast_prev_hole); | |
624 | tftp_cur_block = tftp_mcast_prev_hole; | |
625 | if (tftp_cur_block > | |
626 | ((tftp_mcast_bitmap_size * 8) - 1)) { | |
627 | debug("tftpfile too big\n"); | |
53a5c424 | 628 | /* try to double it and retry */ |
8885c5fe | 629 | tftp_mcast_bitmap_size <<= 1; |
53a5c424 | 630 | mcast_cleanup(); |
bc0571fc | 631 | net_start_again(); |
53a5c424 DU |
632 | return; |
633 | } | |
8885c5fe | 634 | tftp_prev_block = tftp_cur_block; |
53a5c424 DU |
635 | } |
636 | } | |
637 | #endif | |
8885c5fe | 638 | tftp_send(); |
fe8c2806 | 639 | |
53a5c424 | 640 | #ifdef CONFIG_MCAST_TFTP |
8885c5fe JH |
641 | if (tftp_mcast_active) { |
642 | if (tftp_mcast_master_client && | |
643 | (tftp_cur_block >= tftp_mcast_ending_block)) { | |
c718b143 | 644 | puts("\nMulticast tftp done\n"); |
53a5c424 | 645 | mcast_cleanup(); |
22f6e99d | 646 | net_set_state(NETLOOP_SUCCESS); |
85eb5caf | 647 | } |
13dfe943 | 648 | } else |
53a5c424 | 649 | #endif |
8885c5fe | 650 | if (len < tftp_block_size) |
f5329bbc | 651 | tftp_complete(); |
fe8c2806 WD |
652 | break; |
653 | ||
654 | case TFTP_ERROR: | |
c718b143 | 655 | printf("\nTFTP error: '%s' (%d)\n", |
61fdd4f7 | 656 | pkt + 2, ntohs(*(__be16 *)pkt)); |
aafda38f | 657 | |
61fdd4f7 | 658 | switch (ntohs(*(__be16 *)pkt)) { |
aafda38f RB |
659 | case TFTP_ERR_FILE_NOT_FOUND: |
660 | case TFTP_ERR_ACCESS_DENIED: | |
661 | puts("Not retrying...\n"); | |
662 | eth_halt(); | |
22f6e99d | 663 | net_set_state(NETLOOP_FAIL); |
aafda38f RB |
664 | break; |
665 | case TFTP_ERR_UNDEFINED: | |
666 | case TFTP_ERR_DISK_FULL: | |
667 | case TFTP_ERR_UNEXPECTED_OPCODE: | |
668 | case TFTP_ERR_UNKNOWN_TRANSFER_ID: | |
669 | case TFTP_ERR_FILE_ALREADY_EXISTS: | |
670 | default: | |
671 | puts("Starting again\n\n"); | |
53a5c424 | 672 | #ifdef CONFIG_MCAST_TFTP |
aafda38f | 673 | mcast_cleanup(); |
53a5c424 | 674 | #endif |
bc0571fc | 675 | net_start_again(); |
aafda38f RB |
676 | break; |
677 | } | |
fe8c2806 WD |
678 | break; |
679 | } | |
680 | } | |
681 | ||
682 | ||
8885c5fe | 683 | static void tftp_timeout_handler(void) |
fe8c2806 | 684 | { |
8885c5fe | 685 | if (++timeout_count > timeout_count_max) { |
e4cde2f7 | 686 | restart("Retry count exceeded"); |
fe8c2806 | 687 | } else { |
c718b143 | 688 | puts("T "); |
bc0571fc | 689 | net_set_timeout_handler(timeout_ms, tftp_timeout_handler); |
8885c5fe JH |
690 | if (tftp_state != STATE_RECV_WRQ) |
691 | tftp_send(); | |
fe8c2806 WD |
692 | } |
693 | } | |
694 | ||
695 | ||
8885c5fe | 696 | void tftp_start(enum proto_t protocol) |
fe8c2806 | 697 | { |
ecb0ccd9 | 698 | char *ep; /* Environment pointer */ |
89ba81d1 | 699 | |
c96f86ee WD |
700 | /* |
701 | * Allow the user to choose TFTP blocksize and timeout. | |
702 | * TFTP protocol has a minimal timeout of 1 second. | |
703 | */ | |
2cb53608 LC |
704 | ep = getenv("tftpblocksize"); |
705 | if (ep != NULL) | |
8885c5fe | 706 | tftp_block_size_option = simple_strtol(ep, NULL, 10); |
c96f86ee | 707 | |
2cb53608 LC |
708 | ep = getenv("tftptimeout"); |
709 | if (ep != NULL) | |
8885c5fe | 710 | timeout_ms = simple_strtol(ep, NULL, 10); |
c96f86ee | 711 | |
8885c5fe JH |
712 | if (timeout_ms < 1000) { |
713 | printf("TFTP timeout (%ld ms) too low, set min = 1000 ms\n", | |
714 | timeout_ms); | |
715 | timeout_ms = 1000; | |
c96f86ee WD |
716 | } |
717 | ||
718 | debug("TFTP blocksize = %i, timeout = %ld ms\n", | |
8885c5fe | 719 | tftp_block_size_option, timeout_ms); |
ecb0ccd9 | 720 | |
049a95a7 | 721 | tftp_remote_ip = net_server_ip; |
1411157d | 722 | if (net_boot_file_name[0] == '\0') { |
ea45cb0a | 723 | sprintf(default_filename, "%02X%02X%02X%02X.img", |
049a95a7 JH |
724 | net_ip.s_addr & 0xFF, |
725 | (net_ip.s_addr >> 8) & 0xFF, | |
726 | (net_ip.s_addr >> 16) & 0xFF, | |
727 | (net_ip.s_addr >> 24) & 0xFF); | |
a93907c4 JCPV |
728 | |
729 | strncpy(tftp_filename, default_filename, MAX_LEN); | |
8885c5fe | 730 | tftp_filename[MAX_LEN - 1] = 0; |
fe8c2806 | 731 | |
c718b143 | 732 | printf("*** Warning: no boot file name; using '%s'\n", |
8885c5fe | 733 | tftp_filename); |
fe8c2806 | 734 | } else { |
1411157d | 735 | char *p = strchr(net_boot_file_name, ':'); |
a93907c4 JCPV |
736 | |
737 | if (p == NULL) { | |
1411157d | 738 | strncpy(tftp_filename, net_boot_file_name, MAX_LEN); |
8885c5fe | 739 | tftp_filename[MAX_LEN - 1] = 0; |
a93907c4 | 740 | } else { |
1411157d | 741 | tftp_remote_ip = string_to_ip(net_boot_file_name); |
6a86bb6c | 742 | strncpy(tftp_filename, p + 1, MAX_LEN); |
8885c5fe | 743 | tftp_filename[MAX_LEN - 1] = 0; |
a93907c4 | 744 | } |
fe8c2806 WD |
745 | } |
746 | ||
c718b143 | 747 | printf("Using %s device\n", eth_get_name()); |
1fb7cd49 | 748 | printf("TFTP %s server %pI4; our IP address is %pI4", |
8c6914f1 SG |
749 | #ifdef CONFIG_CMD_TFTPPUT |
750 | protocol == TFTPPUT ? "to" : "from", | |
751 | #else | |
8885c5fe | 752 | "from", |
8c6914f1 | 753 | #endif |
8885c5fe | 754 | &tftp_remote_ip, &net_ip); |
fe8c2806 WD |
755 | |
756 | /* Check if we need to send across this subnet */ | |
049a95a7 JH |
757 | if (net_gateway.s_addr && net_netmask.s_addr) { |
758 | struct in_addr our_net; | |
759 | struct in_addr remote_net; | |
760 | ||
761 | our_net.s_addr = net_ip.s_addr & net_netmask.s_addr; | |
762 | remote_net.s_addr = tftp_remote_ip.s_addr & net_netmask.s_addr; | |
763 | if (our_net.s_addr != remote_net.s_addr) | |
764 | printf("; sending through gateway %pI4", &net_gateway); | |
fe8c2806 | 765 | } |
c718b143 | 766 | putc('\n'); |
fe8c2806 | 767 | |
c718b143 | 768 | printf("Filename '%s'.", tftp_filename); |
fe8c2806 | 769 | |
1411157d JH |
770 | if (net_boot_file_expected_size_in_blocks) { |
771 | printf(" Size is 0x%x Bytes = ", | |
772 | net_boot_file_expected_size_in_blocks << 9); | |
773 | print_size(net_boot_file_expected_size_in_blocks << 9, ""); | |
fe8c2806 WD |
774 | } |
775 | ||
c718b143 | 776 | putc('\n'); |
1fb7cd49 | 777 | #ifdef CONFIG_CMD_TFTPPUT |
8885c5fe JH |
778 | tftp_put_active = (protocol == TFTPPUT); |
779 | if (tftp_put_active) { | |
1fb7cd49 SG |
780 | printf("Save address: 0x%lx\n", save_addr); |
781 | printf("Save size: 0x%lx\n", save_size); | |
1411157d | 782 | net_boot_file_size = save_size; |
1fb7cd49 | 783 | puts("Saving: *\b"); |
8885c5fe | 784 | tftp_state = STATE_SEND_WRQ; |
1fb7cd49 SG |
785 | new_transfer(); |
786 | } else | |
787 | #endif | |
788 | { | |
789 | printf("Load address: 0x%lx\n", load_addr); | |
790 | puts("Loading: *\b"); | |
8885c5fe | 791 | tftp_state = STATE_SEND_RRQ; |
1fb7cd49 | 792 | } |
fe8c2806 | 793 | |
85b19802 | 794 | time_start = get_timer(0); |
8885c5fe | 795 | timeout_count_max = tftp_timeout_count_max; |
e83cc063 | 796 | |
bc0571fc | 797 | net_set_timeout_handler(timeout_ms, tftp_timeout_handler); |
049a95a7 | 798 | net_set_udp_handler(tftp_handler); |
39bccd21 | 799 | #ifdef CONFIG_CMD_TFTPPUT |
1fb7cd49 | 800 | net_set_icmp_handler(icmp_handler); |
39bccd21 | 801 | #endif |
8885c5fe JH |
802 | tftp_remote_port = WELL_KNOWN_PORT; |
803 | timeout_count = 0; | |
ecb0ccd9 | 804 | /* Use a pseudo-random port unless a specific port is set */ |
8885c5fe | 805 | tftp_our_port = 1024 + (get_timer(0) % 3072); |
53a5c424 | 806 | |
ecb0ccd9 | 807 | #ifdef CONFIG_TFTP_PORT |
2cb53608 | 808 | ep = getenv("tftpdstp"); |
7bc325a1 | 809 | if (ep != NULL) |
8885c5fe | 810 | tftp_remote_port = simple_strtol(ep, NULL, 10); |
2cb53608 | 811 | ep = getenv("tftpsrcp"); |
7bc325a1 | 812 | if (ep != NULL) |
8885c5fe | 813 | tftp_our_port = simple_strtol(ep, NULL, 10); |
ecb0ccd9 | 814 | #endif |
8885c5fe | 815 | tftp_cur_block = 0; |
fe8c2806 | 816 | |
73a8b27c | 817 | /* zero out server ether in case the server ip has changed */ |
0adb5b76 | 818 | memset(net_server_ethaddr, 0, 6); |
8885c5fe JH |
819 | /* Revert tftp_block_size to dflt */ |
820 | tftp_block_size = TFTP_BLOCK_SIZE; | |
53a5c424 | 821 | #ifdef CONFIG_MCAST_TFTP |
a93907c4 | 822 | mcast_cleanup(); |
53a5c424 | 823 | #endif |
4fccb818 | 824 | #ifdef CONFIG_TFTP_TSIZE |
8885c5fe JH |
825 | tftp_tsize = 0; |
826 | tftp_tsize_num_hash = 0; | |
4fccb818 | 827 | #endif |
73a8b27c | 828 | |
8885c5fe | 829 | tftp_send(); |
fe8c2806 WD |
830 | } |
831 | ||
e59e3562 | 832 | #ifdef CONFIG_CMD_TFTPSRV |
8885c5fe | 833 | void tftp_start_server(void) |
e59e3562 LC |
834 | { |
835 | tftp_filename[0] = 0; | |
836 | ||
e59e3562 | 837 | printf("Using %s device\n", eth_get_name()); |
049a95a7 | 838 | printf("Listening for TFTP transfer on %pI4\n", &net_ip); |
e59e3562 LC |
839 | printf("Load address: 0x%lx\n", load_addr); |
840 | ||
841 | puts("Loading: *\b"); | |
842 | ||
8885c5fe JH |
843 | timeout_count_max = TIMEOUT_COUNT; |
844 | timeout_count = 0; | |
845 | timeout_ms = TIMEOUT; | |
bc0571fc | 846 | net_set_timeout_handler(timeout_ms, tftp_timeout_handler); |
e59e3562 | 847 | |
8885c5fe JH |
848 | /* Revert tftp_block_size to dflt */ |
849 | tftp_block_size = TFTP_BLOCK_SIZE; | |
850 | tftp_cur_block = 0; | |
851 | tftp_our_port = WELL_KNOWN_PORT; | |
e59e3562 LC |
852 | |
853 | #ifdef CONFIG_TFTP_TSIZE | |
8885c5fe JH |
854 | tftp_tsize = 0; |
855 | tftp_tsize_num_hash = 0; | |
e59e3562 LC |
856 | #endif |
857 | ||
8885c5fe | 858 | tftp_state = STATE_RECV_WRQ; |
049a95a7 | 859 | net_set_udp_handler(tftp_handler); |
8e52533d AR |
860 | |
861 | /* zero out server ether in case the server ip has changed */ | |
0adb5b76 | 862 | memset(net_server_ethaddr, 0, 6); |
e59e3562 LC |
863 | } |
864 | #endif /* CONFIG_CMD_TFTPSRV */ | |
865 | ||
53a5c424 | 866 | #ifdef CONFIG_MCAST_TFTP |
8885c5fe JH |
867 | /* |
868 | * Credits: atftp project. | |
53a5c424 DU |
869 | */ |
870 | ||
8885c5fe JH |
871 | /* |
872 | * Pick up BcastAddr, Port, and whether I am [now] the master-client. | |
53a5c424 DU |
873 | * Frame: |
874 | * +-------+-----------+---+-------~~-------+---+ | |
875 | * | opc | multicast | 0 | addr, port, mc | 0 | | |
876 | * +-------+-----------+---+-------~~-------+---+ | |
877 | * The multicast addr/port becomes what I listen to, and if 'mc' is '1' then | |
878 | * I am the new master-client so must send ACKs to DataBlocks. If I am not | |
879 | * master-client, I'm a passive client, gathering what DataBlocks I may and | |
880 | * making note of which ones I got in my bitmask. | |
881 | * In theory, I never go from master->passive.. | |
882 | * .. this comes in with pkt already pointing just past opc | |
883 | */ | |
884 | static void parse_multicast_oack(char *pkt, int len) | |
885 | { | |
c718b143 | 886 | int i; |
049a95a7 | 887 | struct in_addr addr; |
8885c5fe JH |
888 | char *mc_adr; |
889 | char *port; | |
890 | char *mc; | |
53a5c424 | 891 | |
8885c5fe JH |
892 | mc_adr = NULL; |
893 | port = NULL; | |
894 | mc = NULL; | |
53a5c424 DU |
895 | /* march along looking for 'multicast\0', which has to start at least |
896 | * 14 bytes back from the end. | |
897 | */ | |
8885c5fe JH |
898 | for (i = 0; i < len - 14; i++) |
899 | if (strcmp(pkt + i, "multicast") == 0) | |
53a5c424 | 900 | break; |
8885c5fe | 901 | if (i >= (len - 14)) /* non-Multicast OACK, ign. */ |
53a5c424 | 902 | return; |
85eb5caf | 903 | |
c718b143 | 904 | i += 10; /* strlen multicast */ |
8885c5fe | 905 | mc_adr = pkt + i; |
c718b143 | 906 | for (; i < len; i++) { |
8885c5fe JH |
907 | if (*(pkt + i) == ',') { |
908 | *(pkt + i) = '\0'; | |
53a5c424 | 909 | if (port) { |
8885c5fe | 910 | mc = pkt + i + 1; |
53a5c424 DU |
911 | break; |
912 | } else { | |
8885c5fe | 913 | port = pkt + i + 1; |
53a5c424 DU |
914 | } |
915 | } | |
916 | } | |
6d2231e8 LC |
917 | if (!port || !mc_adr || !mc) |
918 | return; | |
8885c5fe | 919 | if (tftp_mcast_active && tftp_mcast_master_client) { |
c718b143 | 920 | printf("I got a OACK as master Client, WRONG!\n"); |
53a5c424 DU |
921 | return; |
922 | } | |
923 | /* ..I now accept packets destined for this MCAST addr, port */ | |
8885c5fe JH |
924 | if (!tftp_mcast_active) { |
925 | if (tftp_mcast_bitmap) { | |
c718b143 | 926 | printf("Internal failure! no mcast.\n"); |
8885c5fe JH |
927 | free(tftp_mcast_bitmap); |
928 | tftp_mcast_bitmap = NULL; | |
929 | tftp_mcast_disabled = 1; | |
930 | return; | |
53a5c424 DU |
931 | } |
932 | /* I malloc instead of pre-declare; so that if the file ends | |
85eb5caf WD |
933 | * up being too big for this bitmap I can retry |
934 | */ | |
8885c5fe JH |
935 | tftp_mcast_bitmap = malloc(tftp_mcast_bitmap_size); |
936 | if (!tftp_mcast_bitmap) { | |
937 | printf("No bitmap, no multicast. Sorry.\n"); | |
938 | tftp_mcast_disabled = 1; | |
53a5c424 DU |
939 | return; |
940 | } | |
8885c5fe JH |
941 | memset(tftp_mcast_bitmap, 0, tftp_mcast_bitmap_size); |
942 | tftp_mcast_prev_hole = 0; | |
943 | tftp_mcast_active = 1; | |
53a5c424 DU |
944 | } |
945 | addr = string_to_ip(mc_adr); | |
049a95a7 JH |
946 | if (net_mcast_addr.s_addr != addr.s_addr) { |
947 | if (net_mcast_addr.s_addr) | |
948 | eth_mcast_join(net_mcast_addr, 0); | |
949 | net_mcast_addr = addr; | |
950 | if (eth_mcast_join(net_mcast_addr, 1)) { | |
c718b143 | 951 | printf("Fail to set mcast, revert to TFTP\n"); |
8885c5fe | 952 | tftp_mcast_disabled = 1; |
53a5c424 | 953 | mcast_cleanup(); |
bc0571fc | 954 | net_start_again(); |
53a5c424 DU |
955 | } |
956 | } | |
8885c5fe JH |
957 | tftp_mcast_master_client = simple_strtoul((char *)mc, NULL, 10); |
958 | tftp_mcast_port = (unsigned short)simple_strtoul(port, NULL, 10); | |
959 | printf("Multicast: %s:%d [%d]\n", mc_adr, tftp_mcast_port, | |
960 | tftp_mcast_master_client); | |
53a5c424 DU |
961 | return; |
962 | } | |
963 | ||
964 | #endif /* Multicast TFTP */ |