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