blob: 6d333d559c19db8a9eac6a26ec01eaf73096babb [file] [log] [blame]
wdenkfe8c2802002-11-03 00:38:21 +00001/*
Luca Ceresoli2f094132011-05-14 05:49:56 +00002 * Copyright 1994, 1995, 2000 Neil Russell.
3 * (See License)
4 * Copyright 2000, 2001 DENX Software Engineering, Wolfgang Denk, wd@denx.de
Luca Ceresolie59e3562011-05-17 00:03:39 +00005 * Copyright 2011 Comelit Group SpA,
6 * Luca Ceresoli <luca.ceresoli@comelit.it>
wdenkfe8c2802002-11-03 00:38:21 +00007 */
8
9#include <common.h>
10#include <command.h>
11#include <net.h>
12#include "tftp.h"
13#include "bootp.h"
Joe Hershberger13dfe942012-05-15 08:59:12 +000014#ifdef CONFIG_SYS_DIRECT_FLASH_TFTP
15#include <flash.h>
16#endif
wdenkfe8c2802002-11-03 00:38:21 +000017
Luca Ceresoli2f094132011-05-14 05:49:56 +000018/* Well known TFTP port # */
19#define WELL_KNOWN_PORT 69
20/* Millisecs to timeout for lost pkt */
21#define TIMEOUT 5000UL
wdenkfe8c2802002-11-03 00:38:21 +000022#ifndef CONFIG_NET_RETRY_COUNT
Luca Ceresoli2f094132011-05-14 05:49:56 +000023/* # of timeouts before giving up */
24# define TIMEOUT_COUNT 10
wdenkfe8c2802002-11-03 00:38:21 +000025#else
26# define TIMEOUT_COUNT (CONFIG_NET_RETRY_COUNT * 2)
27#endif
Luca Ceresoli2f094132011-05-14 05:49:56 +000028/* Number of "loading" hashes per line (for checking the image size) */
29#define HASHES_PER_LINE 65
wdenkfe8c2802002-11-03 00:38:21 +000030
31/*
32 * TFTP operations.
33 */
34#define TFTP_RRQ 1
35#define TFTP_WRQ 2
36#define TFTP_DATA 3
37#define TFTP_ACK 4
38#define TFTP_ERROR 5
wdenkfbe4b5c2003-10-06 21:55:32 +000039#define TFTP_OACK 6
wdenkfe8c2802002-11-03 00:38:21 +000040
Bartlomiej Siekae83cc062008-10-01 15:26:29 +020041static ulong TftpTimeoutMSecs = TIMEOUT;
42static int TftpTimeoutCountMax = TIMEOUT_COUNT;
Simon Glass85b19802012-10-11 13:57:36 +000043static ulong time_start; /* Record time we started tftp */
Bartlomiej Siekae83cc062008-10-01 15:26:29 +020044
45/*
46 * These globals govern the timeout behavior when attempting a connection to a
47 * TFTP server. TftpRRQTimeoutMSecs specifies the number of milliseconds to
48 * wait for the server to respond to initial connection. Second global,
49 * TftpRRQTimeoutCountMax, gives the number of such connection retries.
50 * TftpRRQTimeoutCountMax must be non-negative and TftpRRQTimeoutMSecs must be
51 * positive. The globals are meant to be set (and restored) by code needing
52 * non-standard timeout behavior when initiating a TFTP transfer.
53 */
54ulong TftpRRQTimeoutMSecs = TIMEOUT;
55int TftpRRQTimeoutCountMax = TIMEOUT_COUNT;
56
Remy Bohmeraafda382009-10-28 22:13:40 +010057enum {
58 TFTP_ERR_UNDEFINED = 0,
59 TFTP_ERR_FILE_NOT_FOUND = 1,
60 TFTP_ERR_ACCESS_DENIED = 2,
61 TFTP_ERR_DISK_FULL = 3,
62 TFTP_ERR_UNEXPECTED_OPCODE = 4,
63 TFTP_ERR_UNKNOWN_TRANSFER_ID = 5,
64 TFTP_ERR_FILE_ALREADY_EXISTS = 6,
65};
66
Luca Ceresoli20478ce2011-05-17 00:03:37 +000067static IPaddr_t TftpRemoteIP;
Luca Ceresoli2f094132011-05-14 05:49:56 +000068/* The UDP port at their end */
Luca Ceresoli20478ce2011-05-17 00:03:37 +000069static int TftpRemotePort;
Luca Ceresoli2f094132011-05-14 05:49:56 +000070/* The UDP port at our end */
71static int TftpOurPort;
wdenkfe8c2802002-11-03 00:38:21 +000072static int TftpTimeoutCount;
Luca Ceresoli2f094132011-05-14 05:49:56 +000073/* packet sequence number */
74static ulong TftpBlock;
75/* last packet sequence number received */
76static ulong TftpLastBlock;
77/* count of sequence number wraparounds */
78static ulong TftpBlockWrap;
79/* memory offset due to wrapping */
80static ulong TftpBlockWrapOffset;
wdenkfe8c2802002-11-03 00:38:21 +000081static int TftpState;
Robin Getz4fccb812009-08-20 10:50:20 -040082#ifdef CONFIG_TFTP_TSIZE
Luca Ceresoli2f094132011-05-14 05:49:56 +000083/* The file size reported by the server */
84static int TftpTsize;
85/* The number of hashes we printed */
86static short TftpNumchars;
Robin Getz4fccb812009-08-20 10:50:20 -040087#endif
Simon Glass1fb7cd42011-10-24 18:00:07 +000088#ifdef CONFIG_CMD_TFTPPUT
89static int TftpWriting; /* 1 if writing, else 0 */
90static int TftpFinalBlock; /* 1 if we have sent the last block */
91#else
92#define TftpWriting 0
93#endif
wdenk3f85ce22004-02-23 16:11:30 +000094
Luca Ceresolie3fb0ab2011-05-17 00:03:38 +000095#define STATE_SEND_RRQ 1
wdenkfe8c2802002-11-03 00:38:21 +000096#define STATE_DATA 2
97#define STATE_TOO_LARGE 3
98#define STATE_BAD_MAGIC 4
wdenkfbe4b5c2003-10-06 21:55:32 +000099#define STATE_OACK 5
Luca Ceresolie59e3562011-05-17 00:03:39 +0000100#define STATE_RECV_WRQ 6
Simon Glass1fb7cd42011-10-24 18:00:07 +0000101#define STATE_SEND_WRQ 7
wdenkfe8c2802002-11-03 00:38:21 +0000102
Luca Ceresoli2f094132011-05-14 05:49:56 +0000103/* default TFTP block size */
104#define TFTP_BLOCK_SIZE 512
105/* sequence number is 16 bit */
106#define TFTP_SEQUENCE_SIZE ((ulong)(1<<16))
wdenk3f85ce22004-02-23 16:11:30 +0000107
wdenkfe8c2802002-11-03 00:38:21 +0000108#define DEFAULT_NAME_LEN (8 + 4 + 1)
109static char default_filename[DEFAULT_NAME_LEN];
Jean-Christophe PLAGNIOL-VILLARDa93907c2008-01-18 01:14:03 +0100110
111#ifndef CONFIG_TFTP_FILE_NAME_MAX_LEN
112#define MAX_LEN 128
113#else
114#define MAX_LEN CONFIG_TFTP_FILE_NAME_MAX_LEN
115#endif
116
117static char tftp_filename[MAX_LEN];
wdenkfe8c2802002-11-03 00:38:21 +0000118
Wolfgang Denk85eb5ca2007-08-14 09:47:27 +0200119/* 512 is poor choice for ethernet, MTU is typically 1500.
120 * Minus eth.hdrs thats 1468. Can get 2x better throughput with
David Updegraff53a5c422007-06-11 10:41:07 -0500121 * almost-MTU block sizes. At least try... fall back to 512 if need be.
Alessandro Rubini89ba81d2009-08-07 13:59:06 +0200122 * (but those using CONFIG_IP_DEFRAG may want to set a larger block in cfg file)
David Updegraff53a5c422007-06-11 10:41:07 -0500123 */
Alessandro Rubini89ba81d2009-08-07 13:59:06 +0200124#ifdef CONFIG_TFTP_BLOCKSIZE
125#define TFTP_MTU_BLOCKSIZE CONFIG_TFTP_BLOCKSIZE
126#else
David Updegraff53a5c422007-06-11 10:41:07 -0500127#define TFTP_MTU_BLOCKSIZE 1468
Alessandro Rubini89ba81d2009-08-07 13:59:06 +0200128#endif
129
Luca Ceresolic718b142011-05-14 05:49:57 +0000130static unsigned short TftpBlkSize = TFTP_BLOCK_SIZE;
131static unsigned short TftpBlkSizeOption = TFTP_MTU_BLOCKSIZE;
David Updegraff53a5c422007-06-11 10:41:07 -0500132
133#ifdef CONFIG_MCAST_TFTP
134#include <malloc.h>
135#define MTFTP_BITMAPSIZE 0x1000
136static unsigned *Bitmap;
Luca Ceresolic718b142011-05-14 05:49:57 +0000137static int PrevBitmapHole, Mapsize = MTFTP_BITMAPSIZE;
Luca Ceresoli9bb0a1b2011-05-14 05:50:03 +0000138static uchar ProhibitMcast, MasterClient;
139static uchar Multicast;
David Updegraff53a5c422007-06-11 10:41:07 -0500140static int Mcast_port;
141static ulong TftpEndingBlock; /* can get 'last' block before done..*/
142
Luca Ceresolic718b142011-05-14 05:49:57 +0000143static void parse_multicast_oack(char *pkt, int len);
David Updegraff53a5c422007-06-11 10:41:07 -0500144
145static void
146mcast_cleanup(void)
147{
Luca Ceresoli6d2231e2011-05-14 05:50:01 +0000148 if (Mcast_addr)
149 eth_mcast_join(Mcast_addr, 0);
150 if (Bitmap)
151 free(Bitmap);
Luca Ceresolic718b142011-05-14 05:49:57 +0000152 Bitmap = NULL;
David Updegraff53a5c422007-06-11 10:41:07 -0500153 Mcast_addr = Multicast = Mcast_port = 0;
154 TftpEndingBlock = -1;
155}
156
157#endif /* CONFIG_MCAST_TFTP */
158
Joe Hershberger13dfe942012-05-15 08:59:12 +0000159static inline void
Jayachandran Chandrasekharan Nairbc46dfa2012-07-10 11:48:54 +0530160store_block(int block, uchar *src, unsigned len)
wdenkfe8c2802002-11-03 00:38:21 +0000161{
David Updegraff53a5c422007-06-11 10:41:07 -0500162 ulong offset = block * TftpBlkSize + TftpBlockWrapOffset;
wdenk3f85ce22004-02-23 16:11:30 +0000163 ulong newsize = offset + len;
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200164#ifdef CONFIG_SYS_DIRECT_FLASH_TFTP
wdenkfe8c2802002-11-03 00:38:21 +0000165 int i, rc = 0;
166
Luca Ceresolic718b142011-05-14 05:49:57 +0000167 for (i = 0; i < CONFIG_SYS_MAX_FLASH_BANKS; i++) {
wdenkfe8c2802002-11-03 00:38:21 +0000168 /* start address in flash? */
Jochen Friedrichbe1b0d22008-09-02 11:24:59 +0200169 if (flash_info[i].flash_id == FLASH_UNKNOWN)
170 continue;
wdenkfe8c2802002-11-03 00:38:21 +0000171 if (load_addr + offset >= flash_info[i].start[0]) {
172 rc = 1;
173 break;
174 }
175 }
176
177 if (rc) { /* Flash is destination for this packet */
Luca Ceresolic718b142011-05-14 05:49:57 +0000178 rc = flash_write((char *)src, (ulong)(load_addr+offset), len);
wdenkfe8c2802002-11-03 00:38:21 +0000179 if (rc) {
Luca Ceresolic718b142011-05-14 05:49:57 +0000180 flash_perror(rc);
Joe Hershberger22f6e992012-05-23 07:59:14 +0000181 net_set_state(NETLOOP_FAIL);
wdenkfe8c2802002-11-03 00:38:21 +0000182 return;
183 }
Joe Hershberger13dfe942012-05-15 08:59:12 +0000184 } else
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200185#endif /* CONFIG_SYS_DIRECT_FLASH_TFTP */
wdenkfe8c2802002-11-03 00:38:21 +0000186 {
187 (void)memcpy((void *)(load_addr + offset), src, len);
188 }
David Updegraff53a5c422007-06-11 10:41:07 -0500189#ifdef CONFIG_MCAST_TFTP
190 if (Multicast)
191 ext2_set_bit(block, Bitmap);
192#endif
wdenkfe8c2802002-11-03 00:38:21 +0000193
194 if (NetBootFileXferSize < newsize)
195 NetBootFileXferSize = newsize;
196}
197
Simon Glasse4cde2f2011-10-24 18:00:04 +0000198/* Clear our state ready for a new transfer */
Simon Glass165099e2011-10-27 06:24:28 +0000199static void new_transfer(void)
Simon Glasse4cde2f2011-10-24 18:00:04 +0000200{
201 TftpLastBlock = 0;
202 TftpBlockWrap = 0;
203 TftpBlockWrapOffset = 0;
204#ifdef CONFIG_CMD_TFTPPUT
205 TftpFinalBlock = 0;
206#endif
207}
208
Simon Glass1fb7cd42011-10-24 18:00:07 +0000209#ifdef CONFIG_CMD_TFTPPUT
210/**
211 * Load the next block from memory to be sent over tftp.
212 *
213 * @param block Block number to send
214 * @param dst Destination buffer for data
215 * @param len Number of bytes in block (this one and every other)
216 * @return number of bytes loaded
217 */
218static int load_block(unsigned block, uchar *dst, unsigned len)
219{
220 /* We may want to get the final block from the previous set */
221 ulong offset = ((int)block - 1) * len + TftpBlockWrapOffset;
222 ulong tosend = len;
223
224 tosend = min(NetBootFileXferSize - offset, tosend);
225 (void)memcpy(dst, (void *)(save_addr + offset), tosend);
226 debug("%s: block=%d, offset=%ld, len=%d, tosend=%ld\n", __func__,
227 block, offset, len, tosend);
228 return tosend;
229}
230#endif
231
Luca Ceresolic718b142011-05-14 05:49:57 +0000232static void TftpSend(void);
233static void TftpTimeout(void);
wdenkfe8c2802002-11-03 00:38:21 +0000234
235/**********************************************************************/
236
Simon Glassf5329bb2011-10-24 18:00:03 +0000237static void show_block_marker(void)
238{
239#ifdef CONFIG_TFTP_TSIZE
240 if (TftpTsize) {
241 ulong pos = TftpBlock * TftpBlkSize + TftpBlockWrapOffset;
242
243 while (TftpNumchars < pos * 50 / TftpTsize) {
244 putc('#');
245 TftpNumchars++;
246 }
Simon Glass1fb7cd42011-10-24 18:00:07 +0000247 } else
Simon Glassf5329bb2011-10-24 18:00:03 +0000248#endif
Simon Glass1fb7cd42011-10-24 18:00:07 +0000249 {
Simon Glassf5329bb2011-10-24 18:00:03 +0000250 if (((TftpBlock - 1) % 10) == 0)
251 putc('#');
252 else if ((TftpBlock % (10 * HASHES_PER_LINE)) == 0)
253 puts("\n\t ");
254 }
255}
256
Simon Glasse4cde2f2011-10-24 18:00:04 +0000257/**
258 * restart the current transfer due to an error
259 *
260 * @param msg Message to print for user
261 */
262static void restart(const char *msg)
263{
264 printf("\n%s; starting again\n", msg);
265#ifdef CONFIG_MCAST_TFTP
266 mcast_cleanup();
267#endif
268 NetStartAgain();
269}
270
271/*
272 * Check if the block number has wrapped, and update progress
273 *
274 * TODO: The egregious use of global variables in this file should be tidied.
275 */
276static void update_block_number(void)
277{
278 /*
279 * RFC1350 specifies that the first data packet will
280 * have sequence number 1. If we receive a sequence
281 * number of 0 this means that there was a wrap
282 * around of the (16 bit) counter.
283 */
284 if (TftpBlock == 0) {
285 TftpBlockWrap++;
286 TftpBlockWrapOffset += TftpBlkSize * TFTP_SEQUENCE_SIZE;
287 TftpTimeoutCount = 0; /* we've done well, reset thhe timeout */
288 } else {
289 show_block_marker();
290 }
291}
292
Simon Glassf5329bb2011-10-24 18:00:03 +0000293/* The TFTP get or put is complete */
294static void tftp_complete(void)
295{
296#ifdef CONFIG_TFTP_TSIZE
297 /* Print hash marks for the last packet received */
298 while (TftpTsize && TftpNumchars < 49) {
299 putc('#');
300 TftpNumchars++;
301 }
302#endif
Simon Glass85b19802012-10-11 13:57:36 +0000303 time_start = get_timer(time_start);
304 if (time_start > 0) {
305 puts("\n\t "); /* Line up with "Loading: " */
306 print_size(NetBootFileXferSize /
307 time_start * 1000, "/s");
308 }
Simon Glassf5329bb2011-10-24 18:00:03 +0000309 puts("\ndone\n");
Joe Hershberger22f6e992012-05-23 07:59:14 +0000310 net_set_state(NETLOOP_SUCCESS);
Simon Glassf5329bb2011-10-24 18:00:03 +0000311}
312
wdenkfe8c2802002-11-03 00:38:21 +0000313static void
Luca Ceresolic718b142011-05-14 05:49:57 +0000314TftpSend(void)
wdenkfe8c2802002-11-03 00:38:21 +0000315{
Simon Glass1fb7cd42011-10-24 18:00:07 +0000316 uchar *pkt;
Joe Hershbergerdb288a92012-05-15 08:59:04 +0000317 uchar *xp;
318 int len = 0;
319 ushort *s;
wdenkfe8c2802002-11-03 00:38:21 +0000320
Wolfgang Denk85eb5ca2007-08-14 09:47:27 +0200321#ifdef CONFIG_MCAST_TFTP
David Updegraff53a5c422007-06-11 10:41:07 -0500322 /* Multicast TFTP.. non-MasterClients do not ACK data. */
Wolfgang Denk85eb5ca2007-08-14 09:47:27 +0200323 if (Multicast
324 && (TftpState == STATE_DATA)
325 && (MasterClient == 0))
David Updegraff53a5c422007-06-11 10:41:07 -0500326 return;
327#endif
wdenkfe8c2802002-11-03 00:38:21 +0000328 /*
329 * We will always be sending some sort of packet, so
330 * cobble together the packet headers now.
331 */
Joe Hershberger594c26f2012-05-23 07:58:04 +0000332 pkt = NetTxPacket + NetEthHdrSize() + IP_UDP_HDR_SIZE;
wdenkfe8c2802002-11-03 00:38:21 +0000333
334 switch (TftpState) {
Luca Ceresolie3fb0ab2011-05-17 00:03:38 +0000335 case STATE_SEND_RRQ:
Simon Glass1fb7cd42011-10-24 18:00:07 +0000336 case STATE_SEND_WRQ:
wdenkfe8c2802002-11-03 00:38:21 +0000337 xp = pkt;
Wolfgang Denk7bc5ee02005-08-26 01:36:03 +0200338 s = (ushort *)pkt;
Simon Glass8c6914f2011-10-27 06:24:29 +0000339#ifdef CONFIG_CMD_TFTPPUT
Simon Glass1fb7cd42011-10-24 18:00:07 +0000340 *s++ = htons(TftpState == STATE_SEND_RRQ ? TFTP_RRQ :
341 TFTP_WRQ);
Simon Glass8c6914f2011-10-27 06:24:29 +0000342#else
343 *s++ = htons(TFTP_RRQ);
344#endif
Wolfgang Denk7bc5ee02005-08-26 01:36:03 +0200345 pkt = (uchar *)s;
Luca Ceresolic718b142011-05-14 05:49:57 +0000346 strcpy((char *)pkt, tftp_filename);
wdenkfe8c2802002-11-03 00:38:21 +0000347 pkt += strlen(tftp_filename) + 1;
Luca Ceresolic718b142011-05-14 05:49:57 +0000348 strcpy((char *)pkt, "octet");
wdenkfe8c2802002-11-03 00:38:21 +0000349 pkt += 5 /*strlen("octet")*/ + 1;
Luca Ceresolic718b142011-05-14 05:49:57 +0000350 strcpy((char *)pkt, "timeout");
wdenkfbe4b5c2003-10-06 21:55:32 +0000351 pkt += 7 /*strlen("timeout")*/ + 1;
Wolfgang Denkc96f86e2010-01-17 23:55:53 +0100352 sprintf((char *)pkt, "%lu", TftpTimeoutMSecs / 1000);
Robin Getz0ebf04c2009-07-23 03:01:03 -0400353 debug("send option \"timeout %s\"\n", (char *)pkt);
wdenkfbe4b5c2003-10-06 21:55:32 +0000354 pkt += strlen((char *)pkt) + 1;
Robin Getz4fccb812009-08-20 10:50:20 -0400355#ifdef CONFIG_TFTP_TSIZE
Simon Glass1fb7cd42011-10-24 18:00:07 +0000356 pkt += sprintf((char *)pkt, "tsize%c%lu%c",
357 0, NetBootFileXferSize, 0);
Robin Getz4fccb812009-08-20 10:50:20 -0400358#endif
David Updegraff53a5c422007-06-11 10:41:07 -0500359 /* try for more effic. blk size */
Luca Ceresolic718b142011-05-14 05:49:57 +0000360 pkt += sprintf((char *)pkt, "blksize%c%d%c",
361 0, TftpBlkSizeOption, 0);
Wolfgang Denk85eb5ca2007-08-14 09:47:27 +0200362#ifdef CONFIG_MCAST_TFTP
David Updegraff53a5c422007-06-11 10:41:07 -0500363 /* Check all preconditions before even trying the option */
Joe Hershberger13dfe942012-05-15 08:59:12 +0000364 if (!ProhibitMcast) {
365 Bitmap = malloc(Mapsize);
366 if (Bitmap && eth_get_dev()->mcast) {
367 free(Bitmap);
368 Bitmap = NULL;
369 pkt += sprintf((char *)pkt, "multicast%c%c",
370 0, 0);
371 }
David Updegraff53a5c422007-06-11 10:41:07 -0500372 }
373#endif /* CONFIG_MCAST_TFTP */
wdenkfe8c2802002-11-03 00:38:21 +0000374 len = pkt - xp;
375 break;
376
wdenkfbe4b5c2003-10-06 21:55:32 +0000377 case STATE_OACK:
David Updegraff53a5c422007-06-11 10:41:07 -0500378#ifdef CONFIG_MCAST_TFTP
379 /* My turn! Start at where I need blocks I missed.*/
380 if (Multicast)
Luca Ceresolic718b142011-05-14 05:49:57 +0000381 TftpBlock = ext2_find_next_zero_bit(Bitmap,
382 (Mapsize*8), 0);
David Updegraff53a5c422007-06-11 10:41:07 -0500383 /*..falling..*/
384#endif
Luca Ceresolie59e3562011-05-17 00:03:39 +0000385
386 case STATE_RECV_WRQ:
David Updegraff53a5c422007-06-11 10:41:07 -0500387 case STATE_DATA:
wdenkfe8c2802002-11-03 00:38:21 +0000388 xp = pkt;
Wolfgang Denk7bc5ee02005-08-26 01:36:03 +0200389 s = (ushort *)pkt;
Simon Glass1fb7cd42011-10-24 18:00:07 +0000390 s[0] = htons(TFTP_ACK);
391 s[1] = htons(TftpBlock);
392 pkt = (uchar *)(s + 2);
393#ifdef CONFIG_CMD_TFTPPUT
394 if (TftpWriting) {
395 int toload = TftpBlkSize;
396 int loaded = load_block(TftpBlock, pkt, toload);
397
398 s[0] = htons(TFTP_DATA);
399 pkt += loaded;
400 TftpFinalBlock = (loaded < toload);
401 }
402#endif
wdenkfe8c2802002-11-03 00:38:21 +0000403 len = pkt - xp;
404 break;
405
406 case STATE_TOO_LARGE:
407 xp = pkt;
Wolfgang Denk7bc5ee02005-08-26 01:36:03 +0200408 s = (ushort *)pkt;
409 *s++ = htons(TFTP_ERROR);
Simon Glass1fb7cd42011-10-24 18:00:07 +0000410 *s++ = htons(3);
411
Wolfgang Denk7bc5ee02005-08-26 01:36:03 +0200412 pkt = (uchar *)s;
Luca Ceresolic718b142011-05-14 05:49:57 +0000413 strcpy((char *)pkt, "File too large");
wdenkfe8c2802002-11-03 00:38:21 +0000414 pkt += 14 /*strlen("File too large")*/ + 1;
415 len = pkt - xp;
416 break;
417
418 case STATE_BAD_MAGIC:
419 xp = pkt;
Wolfgang Denk7bc5ee02005-08-26 01:36:03 +0200420 s = (ushort *)pkt;
421 *s++ = htons(TFTP_ERROR);
422 *s++ = htons(2);
423 pkt = (uchar *)s;
Luca Ceresolic718b142011-05-14 05:49:57 +0000424 strcpy((char *)pkt, "File has bad magic");
wdenkfe8c2802002-11-03 00:38:21 +0000425 pkt += 18 /*strlen("File has bad magic")*/ + 1;
426 len = pkt - xp;
427 break;
428 }
429
Luca Ceresoli20478ce2011-05-17 00:03:37 +0000430 NetSendUDPPacket(NetServerEther, TftpRemoteIP, TftpRemotePort,
Luca Ceresoli2f094132011-05-14 05:49:56 +0000431 TftpOurPort, len);
wdenkfe8c2802002-11-03 00:38:21 +0000432}
433
Simon Glass39bccd22011-10-26 14:18:38 +0000434#ifdef CONFIG_CMD_TFTPPUT
Simon Glass1fb7cd42011-10-24 18:00:07 +0000435static void icmp_handler(unsigned type, unsigned code, unsigned dest,
436 IPaddr_t sip, unsigned src, uchar *pkt, unsigned len)
437{
438 if (type == ICMP_NOT_REACH && code == ICMP_NOT_REACH_PORT) {
439 /* Oh dear the other end has gone away */
440 restart("TFTP server died");
441 }
442}
Simon Glass39bccd22011-10-26 14:18:38 +0000443#endif
Simon Glass1fb7cd42011-10-24 18:00:07 +0000444
wdenkfe8c2802002-11-03 00:38:21 +0000445static void
Luca Ceresoli03eb1292011-04-18 06:19:50 +0000446TftpHandler(uchar *pkt, unsigned dest, IPaddr_t sip, unsigned src,
447 unsigned len)
wdenkfe8c2802002-11-03 00:38:21 +0000448{
Kim Phillips61fdd4f2013-01-16 18:09:19 -0600449 __be16 proto;
450 __be16 *s;
Wolfgang Denkff13ac82007-08-30 14:42:15 +0200451 int i;
wdenkfe8c2802002-11-03 00:38:21 +0000452
453 if (dest != TftpOurPort) {
David Updegraff53a5c422007-06-11 10:41:07 -0500454#ifdef CONFIG_MCAST_TFTP
Wolfgang Denk85eb5ca2007-08-14 09:47:27 +0200455 if (Multicast
David Updegraff53a5c422007-06-11 10:41:07 -0500456 && (!Mcast_port || (dest != Mcast_port)))
457#endif
Luca Ceresoli0bdd8ac2011-05-14 05:50:02 +0000458 return;
wdenkfe8c2802002-11-03 00:38:21 +0000459 }
Luca Ceresolie59e3562011-05-17 00:03:39 +0000460 if (TftpState != STATE_SEND_RRQ && src != TftpRemotePort &&
Simon Glass1fb7cd42011-10-24 18:00:07 +0000461 TftpState != STATE_RECV_WRQ && TftpState != STATE_SEND_WRQ)
wdenkfe8c2802002-11-03 00:38:21 +0000462 return;
wdenkfe8c2802002-11-03 00:38:21 +0000463
Luca Ceresoli7bc325a2011-05-14 05:50:00 +0000464 if (len < 2)
wdenkfe8c2802002-11-03 00:38:21 +0000465 return;
wdenkfe8c2802002-11-03 00:38:21 +0000466 len -= 2;
467 /* warning: don't use increment (++) in ntohs() macros!! */
Kim Phillips61fdd4f2013-01-16 18:09:19 -0600468 s = (__be16 *)pkt;
Wolfgang Denk7bc5ee02005-08-26 01:36:03 +0200469 proto = *s++;
470 pkt = (uchar *)s;
wdenkfe8c2802002-11-03 00:38:21 +0000471 switch (ntohs(proto)) {
472
473 case TFTP_RRQ:
wdenkfe8c2802002-11-03 00:38:21 +0000474 break;
Simon Glass1fb7cd42011-10-24 18:00:07 +0000475
476 case TFTP_ACK:
477#ifdef CONFIG_CMD_TFTPPUT
478 if (TftpWriting) {
479 if (TftpFinalBlock) {
480 tftp_complete();
481 } else {
482 /*
483 * Move to the next block. We want our block
484 * count to wrap just like the other end!
485 */
486 int block = ntohs(*s);
487 int ack_ok = (TftpBlock == block);
488
489 TftpBlock = (unsigned short)(block + 1);
490 update_block_number();
491 if (ack_ok)
492 TftpSend(); /* Send next data block */
493 }
494 }
495#endif
496 break;
497
wdenkfe8c2802002-11-03 00:38:21 +0000498 default:
499 break;
500
Luca Ceresolie59e3562011-05-17 00:03:39 +0000501#ifdef CONFIG_CMD_TFTPSRV
502 case TFTP_WRQ:
503 debug("Got WRQ\n");
504 TftpRemoteIP = sip;
505 TftpRemotePort = src;
506 TftpOurPort = 1024 + (get_timer(0) % 3072);
Simon Glasse4cde2f2011-10-24 18:00:04 +0000507 new_transfer();
Luca Ceresolie59e3562011-05-17 00:03:39 +0000508 TftpSend(); /* Send ACK(0) */
509 break;
510#endif
511
wdenkfbe4b5c2003-10-06 21:55:32 +0000512 case TFTP_OACK:
Wolfgang Denkd3717082009-08-10 09:59:10 +0200513 debug("Got OACK: %s %s\n",
514 pkt,
515 pkt + strlen((char *)pkt) + 1);
wdenkfbe4b5c2003-10-06 21:55:32 +0000516 TftpState = STATE_OACK;
Luca Ceresoli20478ce2011-05-17 00:03:37 +0000517 TftpRemotePort = src;
Wolfgang Denk60174742007-08-31 10:01:51 +0200518 /*
519 * Check for 'blksize' option.
520 * Careful: "i" is signed, "len" is unsigned, thus
521 * something like "len-8" may give a *huge* number
522 */
Luca Ceresolic718b142011-05-14 05:49:57 +0000523 for (i = 0; i+8 < len; i++) {
Luca Ceresoli2e320252011-05-14 05:49:58 +0000524 if (strcmp((char *)pkt+i, "blksize") == 0) {
Wolfgang Denkff13ac82007-08-30 14:42:15 +0200525 TftpBlkSize = (unsigned short)
Luca Ceresoli2e320252011-05-14 05:49:58 +0000526 simple_strtoul((char *)pkt+i+8, NULL,
Luca Ceresolic718b142011-05-14 05:49:57 +0000527 10);
Robin Getz0ebf04c2009-07-23 03:01:03 -0400528 debug("Blocksize ack: %s, %d\n",
Luca Ceresoli2e320252011-05-14 05:49:58 +0000529 (char *)pkt+i+8, TftpBlkSize);
Wolfgang Denkff13ac82007-08-30 14:42:15 +0200530 }
Robin Getz4fccb812009-08-20 10:50:20 -0400531#ifdef CONFIG_TFTP_TSIZE
Luca Ceresoli2e320252011-05-14 05:49:58 +0000532 if (strcmp((char *)pkt+i, "tsize") == 0) {
533 TftpTsize = simple_strtoul((char *)pkt+i+6,
Luca Ceresoli2f094132011-05-14 05:49:56 +0000534 NULL, 10);
Robin Getz4fccb812009-08-20 10:50:20 -0400535 debug("size = %s, %d\n",
Luca Ceresoli2e320252011-05-14 05:49:58 +0000536 (char *)pkt+i+6, TftpTsize);
Robin Getz4fccb812009-08-20 10:50:20 -0400537 }
538#endif
David Updegraff53a5c422007-06-11 10:41:07 -0500539 }
540#ifdef CONFIG_MCAST_TFTP
Luca Ceresolic718b142011-05-14 05:49:57 +0000541 parse_multicast_oack((char *)pkt, len-1);
Wolfgang Denk85eb5ca2007-08-14 09:47:27 +0200542 if ((Multicast) && (!MasterClient))
David Updegraff53a5c422007-06-11 10:41:07 -0500543 TftpState = STATE_DATA; /* passive.. */
544 else
545#endif
Simon Glass1fb7cd42011-10-24 18:00:07 +0000546#ifdef CONFIG_CMD_TFTPPUT
547 if (TftpWriting) {
548 /* Get ready to send the first block */
549 TftpState = STATE_DATA;
550 TftpBlock++;
551 }
552#endif
553 TftpSend(); /* Send ACK or first data block */
wdenkfbe4b5c2003-10-06 21:55:32 +0000554 break;
wdenkfe8c2802002-11-03 00:38:21 +0000555 case TFTP_DATA:
556 if (len < 2)
557 return;
558 len -= 2;
Kim Phillips61fdd4f2013-01-16 18:09:19 -0600559 TftpBlock = ntohs(*(__be16 *)pkt);
wdenk3f85ce22004-02-23 16:11:30 +0000560
Simon Glasse4cde2f2011-10-24 18:00:04 +0000561 update_block_number();
wdenkfe8c2802002-11-03 00:38:21 +0000562
Luca Ceresolie3fb0ab2011-05-17 00:03:38 +0000563 if (TftpState == STATE_SEND_RRQ)
Robin Getz0ebf04c2009-07-23 03:01:03 -0400564 debug("Server did not acknowledge timeout option!\n");
wdenkfbe4b5c2003-10-06 21:55:32 +0000565
Luca Ceresolie59e3562011-05-17 00:03:39 +0000566 if (TftpState == STATE_SEND_RRQ || TftpState == STATE_OACK ||
567 TftpState == STATE_RECV_WRQ) {
wdenk3f85ce22004-02-23 16:11:30 +0000568 /* first block received */
wdenkfe8c2802002-11-03 00:38:21 +0000569 TftpState = STATE_DATA;
Luca Ceresoli20478ce2011-05-17 00:03:37 +0000570 TftpRemotePort = src;
Simon Glasse4cde2f2011-10-24 18:00:04 +0000571 new_transfer();
wdenkfe8c2802002-11-03 00:38:21 +0000572
David Updegraff53a5c422007-06-11 10:41:07 -0500573#ifdef CONFIG_MCAST_TFTP
574 if (Multicast) { /* start!=1 common if mcast */
575 TftpLastBlock = TftpBlock - 1;
576 } else
577#endif
wdenkfe8c2802002-11-03 00:38:21 +0000578 if (TftpBlock != 1) { /* Assertion */
Luca Ceresolic718b142011-05-14 05:49:57 +0000579 printf("\nTFTP error: "
580 "First block is not block 1 (%ld)\n"
581 "Starting again\n\n",
wdenkfe8c2802002-11-03 00:38:21 +0000582 TftpBlock);
Luca Ceresolic718b142011-05-14 05:49:57 +0000583 NetStartAgain();
wdenkfe8c2802002-11-03 00:38:21 +0000584 break;
585 }
586 }
587
588 if (TftpBlock == TftpLastBlock) {
589 /*
590 * Same block again; ignore it.
591 */
592 break;
593 }
594
595 TftpLastBlock = TftpBlock;
Bartlomiej Siekae83cc062008-10-01 15:26:29 +0200596 TftpTimeoutCountMax = TIMEOUT_COUNT;
Luca Ceresolic718b142011-05-14 05:49:57 +0000597 NetSetTimeout(TftpTimeoutMSecs, TftpTimeout);
wdenkfe8c2802002-11-03 00:38:21 +0000598
Luca Ceresolic718b142011-05-14 05:49:57 +0000599 store_block(TftpBlock - 1, pkt + 2, len);
wdenkfe8c2802002-11-03 00:38:21 +0000600
601 /*
Luca Ceresoli4d69e982011-05-17 00:03:41 +0000602 * Acknowledge the block just received, which will prompt
Luca Ceresoli20478ce2011-05-17 00:03:37 +0000603 * the remote for the next one.
wdenkfe8c2802002-11-03 00:38:21 +0000604 */
David Updegraff53a5c422007-06-11 10:41:07 -0500605#ifdef CONFIG_MCAST_TFTP
Wolfgang Denk85eb5ca2007-08-14 09:47:27 +0200606 /* if I am the MasterClient, actively calculate what my next
607 * needed block is; else I'm passive; not ACKING
Jean-Christophe PLAGNIOL-VILLARDa93907c2008-01-18 01:14:03 +0100608 */
David Updegraff53a5c422007-06-11 10:41:07 -0500609 if (Multicast) {
610 if (len < TftpBlkSize) {
611 TftpEndingBlock = TftpBlock;
612 } else if (MasterClient) {
Wolfgang Denk85eb5ca2007-08-14 09:47:27 +0200613 TftpBlock = PrevBitmapHole =
David Updegraff53a5c422007-06-11 10:41:07 -0500614 ext2_find_next_zero_bit(
615 Bitmap,
616 (Mapsize*8),
617 PrevBitmapHole);
618 if (TftpBlock > ((Mapsize*8) - 1)) {
Luca Ceresolic718b142011-05-14 05:49:57 +0000619 printf("tftpfile too big\n");
David Updegraff53a5c422007-06-11 10:41:07 -0500620 /* try to double it and retry */
Luca Ceresolic718b142011-05-14 05:49:57 +0000621 Mapsize <<= 1;
David Updegraff53a5c422007-06-11 10:41:07 -0500622 mcast_cleanup();
Luca Ceresolic718b142011-05-14 05:49:57 +0000623 NetStartAgain();
David Updegraff53a5c422007-06-11 10:41:07 -0500624 return;
625 }
626 TftpLastBlock = TftpBlock;
627 }
628 }
629#endif
Luca Ceresolic718b142011-05-14 05:49:57 +0000630 TftpSend();
wdenkfe8c2802002-11-03 00:38:21 +0000631
David Updegraff53a5c422007-06-11 10:41:07 -0500632#ifdef CONFIG_MCAST_TFTP
633 if (Multicast) {
634 if (MasterClient && (TftpBlock >= TftpEndingBlock)) {
Luca Ceresolic718b142011-05-14 05:49:57 +0000635 puts("\nMulticast tftp done\n");
David Updegraff53a5c422007-06-11 10:41:07 -0500636 mcast_cleanup();
Joe Hershberger22f6e992012-05-23 07:59:14 +0000637 net_set_state(NETLOOP_SUCCESS);
Wolfgang Denk85eb5ca2007-08-14 09:47:27 +0200638 }
Joe Hershberger13dfe942012-05-15 08:59:12 +0000639 } else
David Updegraff53a5c422007-06-11 10:41:07 -0500640#endif
Simon Glassf5329bb2011-10-24 18:00:03 +0000641 if (len < TftpBlkSize)
642 tftp_complete();
wdenkfe8c2802002-11-03 00:38:21 +0000643 break;
644
645 case TFTP_ERROR:
Luca Ceresolic718b142011-05-14 05:49:57 +0000646 printf("\nTFTP error: '%s' (%d)\n",
Kim Phillips61fdd4f2013-01-16 18:09:19 -0600647 pkt + 2, ntohs(*(__be16 *)pkt));
Remy Bohmeraafda382009-10-28 22:13:40 +0100648
Kim Phillips61fdd4f2013-01-16 18:09:19 -0600649 switch (ntohs(*(__be16 *)pkt)) {
Remy Bohmeraafda382009-10-28 22:13:40 +0100650 case TFTP_ERR_FILE_NOT_FOUND:
651 case TFTP_ERR_ACCESS_DENIED:
652 puts("Not retrying...\n");
653 eth_halt();
Joe Hershberger22f6e992012-05-23 07:59:14 +0000654 net_set_state(NETLOOP_FAIL);
Remy Bohmeraafda382009-10-28 22:13:40 +0100655 break;
656 case TFTP_ERR_UNDEFINED:
657 case TFTP_ERR_DISK_FULL:
658 case TFTP_ERR_UNEXPECTED_OPCODE:
659 case TFTP_ERR_UNKNOWN_TRANSFER_ID:
660 case TFTP_ERR_FILE_ALREADY_EXISTS:
661 default:
662 puts("Starting again\n\n");
David Updegraff53a5c422007-06-11 10:41:07 -0500663#ifdef CONFIG_MCAST_TFTP
Remy Bohmeraafda382009-10-28 22:13:40 +0100664 mcast_cleanup();
David Updegraff53a5c422007-06-11 10:41:07 -0500665#endif
Remy Bohmeraafda382009-10-28 22:13:40 +0100666 NetStartAgain();
667 break;
668 }
wdenkfe8c2802002-11-03 00:38:21 +0000669 break;
670 }
671}
672
673
674static void
Luca Ceresolic718b142011-05-14 05:49:57 +0000675TftpTimeout(void)
wdenkfe8c2802002-11-03 00:38:21 +0000676{
Bartlomiej Siekae83cc062008-10-01 15:26:29 +0200677 if (++TftpTimeoutCount > TftpTimeoutCountMax) {
Simon Glasse4cde2f2011-10-24 18:00:04 +0000678 restart("Retry count exceeded");
wdenkfe8c2802002-11-03 00:38:21 +0000679 } else {
Luca Ceresolic718b142011-05-14 05:49:57 +0000680 puts("T ");
681 NetSetTimeout(TftpTimeoutMSecs, TftpTimeout);
Luca Ceresolie59e3562011-05-17 00:03:39 +0000682 if (TftpState != STATE_RECV_WRQ)
683 TftpSend();
wdenkfe8c2802002-11-03 00:38:21 +0000684 }
685}
686
687
Simon Glass58f317d2011-10-24 18:00:05 +0000688void TftpStart(enum proto_t protocol)
wdenkfe8c2802002-11-03 00:38:21 +0000689{
Wolfgang Denkecb0ccd2005-09-24 22:37:32 +0200690 char *ep; /* Environment pointer */
Alessandro Rubini89ba81d2009-08-07 13:59:06 +0200691
Wolfgang Denkc96f86e2010-01-17 23:55:53 +0100692 /*
693 * Allow the user to choose TFTP blocksize and timeout.
694 * TFTP protocol has a minimal timeout of 1 second.
695 */
Luca Ceresoli2cb53602011-05-14 05:49:59 +0000696 ep = getenv("tftpblocksize");
697 if (ep != NULL)
Alessandro Rubini89ba81d2009-08-07 13:59:06 +0200698 TftpBlkSizeOption = simple_strtol(ep, NULL, 10);
Wolfgang Denkc96f86e2010-01-17 23:55:53 +0100699
Luca Ceresoli2cb53602011-05-14 05:49:59 +0000700 ep = getenv("tftptimeout");
701 if (ep != NULL)
Wolfgang Denkc96f86e2010-01-17 23:55:53 +0100702 TftpTimeoutMSecs = simple_strtol(ep, NULL, 10);
703
704 if (TftpTimeoutMSecs < 1000) {
705 printf("TFTP timeout (%ld ms) too low, "
706 "set minimum = 1000 ms\n",
707 TftpTimeoutMSecs);
708 TftpTimeoutMSecs = 1000;
709 }
710
711 debug("TFTP blocksize = %i, timeout = %ld ms\n",
712 TftpBlkSizeOption, TftpTimeoutMSecs);
Wolfgang Denkecb0ccd2005-09-24 22:37:32 +0200713
Luca Ceresoli20478ce2011-05-17 00:03:37 +0000714 TftpRemoteIP = NetServerIP;
wdenkfe8c2802002-11-03 00:38:21 +0000715 if (BootFile[0] == '\0') {
Matthias Weisserea45cb02011-12-03 03:29:44 +0000716 sprintf(default_filename, "%02X%02X%02X%02X.img",
Wolfgang Denkc43352c2005-08-04 01:09:44 +0200717 NetOurIP & 0xFF,
718 (NetOurIP >> 8) & 0xFF,
719 (NetOurIP >> 16) & 0xFF,
Luca Ceresolic718b142011-05-14 05:49:57 +0000720 (NetOurIP >> 24) & 0xFF);
Jean-Christophe PLAGNIOL-VILLARDa93907c2008-01-18 01:14:03 +0100721
722 strncpy(tftp_filename, default_filename, MAX_LEN);
723 tftp_filename[MAX_LEN-1] = 0;
wdenkfe8c2802002-11-03 00:38:21 +0000724
Luca Ceresolic718b142011-05-14 05:49:57 +0000725 printf("*** Warning: no boot file name; using '%s'\n",
wdenkfe8c2802002-11-03 00:38:21 +0000726 tftp_filename);
727 } else {
Luca Ceresolic718b142011-05-14 05:49:57 +0000728 char *p = strchr(BootFile, ':');
Jean-Christophe PLAGNIOL-VILLARDa93907c2008-01-18 01:14:03 +0100729
730 if (p == NULL) {
731 strncpy(tftp_filename, BootFile, MAX_LEN);
732 tftp_filename[MAX_LEN-1] = 0;
733 } else {
Luca Ceresoli20478ce2011-05-17 00:03:37 +0000734 TftpRemoteIP = string_to_ip(BootFile);
Peter Tyser6a86bb62008-12-01 16:29:38 -0600735 strncpy(tftp_filename, p + 1, MAX_LEN);
Jean-Christophe PLAGNIOL-VILLARDa93907c2008-01-18 01:14:03 +0100736 tftp_filename[MAX_LEN-1] = 0;
737 }
wdenkfe8c2802002-11-03 00:38:21 +0000738 }
739
Luca Ceresolic718b142011-05-14 05:49:57 +0000740 printf("Using %s device\n", eth_get_name());
Simon Glass1fb7cd42011-10-24 18:00:07 +0000741 printf("TFTP %s server %pI4; our IP address is %pI4",
Simon Glass8c6914f2011-10-27 06:24:29 +0000742#ifdef CONFIG_CMD_TFTPPUT
743 protocol == TFTPPUT ? "to" : "from",
744#else
745 "from",
746#endif
747 &TftpRemoteIP, &NetOurIP);
wdenkfe8c2802002-11-03 00:38:21 +0000748
749 /* Check if we need to send across this subnet */
750 if (NetOurGatewayIP && NetOurSubnetMask) {
Luca Ceresoli0bdd8ac2011-05-14 05:50:02 +0000751 IPaddr_t OurNet = NetOurIP & NetOurSubnetMask;
Luca Ceresoli20478ce2011-05-17 00:03:37 +0000752 IPaddr_t RemoteNet = TftpRemoteIP & NetOurSubnetMask;
wdenkfe8c2802002-11-03 00:38:21 +0000753
Luca Ceresoli20478ce2011-05-17 00:03:37 +0000754 if (OurNet != RemoteNet)
Luca Ceresoli0bdd8ac2011-05-14 05:50:02 +0000755 printf("; sending through gateway %pI4",
756 &NetOurGatewayIP);
wdenkfe8c2802002-11-03 00:38:21 +0000757 }
Luca Ceresolic718b142011-05-14 05:49:57 +0000758 putc('\n');
wdenkfe8c2802002-11-03 00:38:21 +0000759
Luca Ceresolic718b142011-05-14 05:49:57 +0000760 printf("Filename '%s'.", tftp_filename);
wdenkfe8c2802002-11-03 00:38:21 +0000761
762 if (NetBootFileSize) {
Luca Ceresolic718b142011-05-14 05:49:57 +0000763 printf(" Size is 0x%x Bytes = ", NetBootFileSize<<9);
764 print_size(NetBootFileSize<<9, "");
wdenkfe8c2802002-11-03 00:38:21 +0000765 }
766
Luca Ceresolic718b142011-05-14 05:49:57 +0000767 putc('\n');
Simon Glass1fb7cd42011-10-24 18:00:07 +0000768#ifdef CONFIG_CMD_TFTPPUT
769 TftpWriting = (protocol == TFTPPUT);
770 if (TftpWriting) {
771 printf("Save address: 0x%lx\n", save_addr);
772 printf("Save size: 0x%lx\n", save_size);
773 NetBootFileXferSize = save_size;
774 puts("Saving: *\b");
775 TftpState = STATE_SEND_WRQ;
776 new_transfer();
777 } else
778#endif
779 {
780 printf("Load address: 0x%lx\n", load_addr);
781 puts("Loading: *\b");
782 TftpState = STATE_SEND_RRQ;
783 }
wdenkfe8c2802002-11-03 00:38:21 +0000784
Simon Glass85b19802012-10-11 13:57:36 +0000785 time_start = get_timer(0);
Bartlomiej Siekae83cc062008-10-01 15:26:29 +0200786 TftpTimeoutCountMax = TftpRRQTimeoutCountMax;
787
Luca Ceresolic718b142011-05-14 05:49:57 +0000788 NetSetTimeout(TftpTimeoutMSecs, TftpTimeout);
Joe Hershbergerece223b2012-05-23 07:59:15 +0000789 net_set_udp_handler(TftpHandler);
Simon Glass39bccd22011-10-26 14:18:38 +0000790#ifdef CONFIG_CMD_TFTPPUT
Simon Glass1fb7cd42011-10-24 18:00:07 +0000791 net_set_icmp_handler(icmp_handler);
Simon Glass39bccd22011-10-26 14:18:38 +0000792#endif
Luca Ceresoli20478ce2011-05-17 00:03:37 +0000793 TftpRemotePort = WELL_KNOWN_PORT;
wdenkfe8c2802002-11-03 00:38:21 +0000794 TftpTimeoutCount = 0;
Wolfgang Denkecb0ccd2005-09-24 22:37:32 +0200795 /* Use a pseudo-random port unless a specific port is set */
wdenkfe8c2802002-11-03 00:38:21 +0000796 TftpOurPort = 1024 + (get_timer(0) % 3072);
David Updegraff53a5c422007-06-11 10:41:07 -0500797
Wolfgang Denkecb0ccd2005-09-24 22:37:32 +0200798#ifdef CONFIG_TFTP_PORT
Luca Ceresoli2cb53602011-05-14 05:49:59 +0000799 ep = getenv("tftpdstp");
Luca Ceresoli7bc325a2011-05-14 05:50:00 +0000800 if (ep != NULL)
Luca Ceresoli20478ce2011-05-17 00:03:37 +0000801 TftpRemotePort = simple_strtol(ep, NULL, 10);
Luca Ceresoli2cb53602011-05-14 05:49:59 +0000802 ep = getenv("tftpsrcp");
Luca Ceresoli7bc325a2011-05-14 05:50:00 +0000803 if (ep != NULL)
Luca Ceresolic718b142011-05-14 05:49:57 +0000804 TftpOurPort = simple_strtol(ep, NULL, 10);
Wolfgang Denkecb0ccd2005-09-24 22:37:32 +0200805#endif
wdenkfbe4b5c2003-10-06 21:55:32 +0000806 TftpBlock = 0;
wdenkfe8c2802002-11-03 00:38:21 +0000807
wdenk73a8b272003-06-05 19:27:42 +0000808 /* zero out server ether in case the server ip has changed */
809 memset(NetServerEther, 0, 6);
David Updegraff53a5c422007-06-11 10:41:07 -0500810 /* Revert TftpBlkSize to dflt */
811 TftpBlkSize = TFTP_BLOCK_SIZE;
812#ifdef CONFIG_MCAST_TFTP
Jean-Christophe PLAGNIOL-VILLARDa93907c2008-01-18 01:14:03 +0100813 mcast_cleanup();
David Updegraff53a5c422007-06-11 10:41:07 -0500814#endif
Robin Getz4fccb812009-08-20 10:50:20 -0400815#ifdef CONFIG_TFTP_TSIZE
816 TftpTsize = 0;
817 TftpNumchars = 0;
818#endif
wdenk73a8b272003-06-05 19:27:42 +0000819
Luca Ceresolic718b142011-05-14 05:49:57 +0000820 TftpSend();
wdenkfe8c2802002-11-03 00:38:21 +0000821}
822
Luca Ceresolie59e3562011-05-17 00:03:39 +0000823#ifdef CONFIG_CMD_TFTPSRV
824void
825TftpStartServer(void)
826{
827 tftp_filename[0] = 0;
828
Luca Ceresolie59e3562011-05-17 00:03:39 +0000829 printf("Using %s device\n", eth_get_name());
Luca Ceresolie59e3562011-05-17 00:03:39 +0000830 printf("Listening for TFTP transfer on %pI4\n", &NetOurIP);
831 printf("Load address: 0x%lx\n", load_addr);
832
833 puts("Loading: *\b");
834
835 TftpTimeoutCountMax = TIMEOUT_COUNT;
836 TftpTimeoutCount = 0;
837 TftpTimeoutMSecs = TIMEOUT;
838 NetSetTimeout(TftpTimeoutMSecs, TftpTimeout);
839
840 /* Revert TftpBlkSize to dflt */
841 TftpBlkSize = TFTP_BLOCK_SIZE;
842 TftpBlock = 0;
843 TftpOurPort = WELL_KNOWN_PORT;
844
845#ifdef CONFIG_TFTP_TSIZE
846 TftpTsize = 0;
847 TftpNumchars = 0;
848#endif
849
850 TftpState = STATE_RECV_WRQ;
Joe Hershbergerece223b2012-05-23 07:59:15 +0000851 net_set_udp_handler(TftpHandler);
Luca Ceresolie59e3562011-05-17 00:03:39 +0000852}
853#endif /* CONFIG_CMD_TFTPSRV */
854
David Updegraff53a5c422007-06-11 10:41:07 -0500855#ifdef CONFIG_MCAST_TFTP
856/* Credits: atftp project.
857 */
858
859/* pick up BcastAddr, Port, and whether I am [now] the master-client. *
860 * Frame:
861 * +-------+-----------+---+-------~~-------+---+
862 * | opc | multicast | 0 | addr, port, mc | 0 |
863 * +-------+-----------+---+-------~~-------+---+
864 * The multicast addr/port becomes what I listen to, and if 'mc' is '1' then
865 * I am the new master-client so must send ACKs to DataBlocks. If I am not
866 * master-client, I'm a passive client, gathering what DataBlocks I may and
867 * making note of which ones I got in my bitmask.
868 * In theory, I never go from master->passive..
869 * .. this comes in with pkt already pointing just past opc
870 */
871static void parse_multicast_oack(char *pkt, int len)
872{
Luca Ceresolic718b142011-05-14 05:49:57 +0000873 int i;
874 IPaddr_t addr;
875 char *mc_adr, *port, *mc;
David Updegraff53a5c422007-06-11 10:41:07 -0500876
Luca Ceresolic718b142011-05-14 05:49:57 +0000877 mc_adr = port = mc = NULL;
David Updegraff53a5c422007-06-11 10:41:07 -0500878 /* march along looking for 'multicast\0', which has to start at least
879 * 14 bytes back from the end.
880 */
Luca Ceresolic718b142011-05-14 05:49:57 +0000881 for (i = 0; i < len-14; i++)
882 if (strcmp(pkt+i, "multicast") == 0)
David Updegraff53a5c422007-06-11 10:41:07 -0500883 break;
884 if (i >= (len-14)) /* non-Multicast OACK, ign. */
885 return;
Wolfgang Denk85eb5ca2007-08-14 09:47:27 +0200886
Luca Ceresolic718b142011-05-14 05:49:57 +0000887 i += 10; /* strlen multicast */
David Updegraff53a5c422007-06-11 10:41:07 -0500888 mc_adr = pkt+i;
Luca Ceresolic718b142011-05-14 05:49:57 +0000889 for (; i < len; i++) {
David Updegraff53a5c422007-06-11 10:41:07 -0500890 if (*(pkt+i) == ',') {
891 *(pkt+i) = '\0';
892 if (port) {
893 mc = pkt+i+1;
894 break;
895 } else {
896 port = pkt+i+1;
897 }
898 }
899 }
Luca Ceresoli6d2231e2011-05-14 05:50:01 +0000900 if (!port || !mc_adr || !mc)
901 return;
David Updegraff53a5c422007-06-11 10:41:07 -0500902 if (Multicast && MasterClient) {
Luca Ceresolic718b142011-05-14 05:49:57 +0000903 printf("I got a OACK as master Client, WRONG!\n");
David Updegraff53a5c422007-06-11 10:41:07 -0500904 return;
905 }
906 /* ..I now accept packets destined for this MCAST addr, port */
907 if (!Multicast) {
908 if (Bitmap) {
Luca Ceresolic718b142011-05-14 05:49:57 +0000909 printf("Internal failure! no mcast.\n");
David Updegraff53a5c422007-06-11 10:41:07 -0500910 free(Bitmap);
Luca Ceresolic718b142011-05-14 05:49:57 +0000911 Bitmap = NULL;
912 ProhibitMcast = 1;
David Updegraff53a5c422007-06-11 10:41:07 -0500913 return ;
914 }
915 /* I malloc instead of pre-declare; so that if the file ends
Wolfgang Denk85eb5ca2007-08-14 09:47:27 +0200916 * up being too big for this bitmap I can retry
917 */
Luca Ceresoli2cb53602011-05-14 05:49:59 +0000918 Bitmap = malloc(Mapsize);
919 if (!Bitmap) {
Luca Ceresolic718b142011-05-14 05:49:57 +0000920 printf("No Bitmap, no multicast. Sorry.\n");
921 ProhibitMcast = 1;
David Updegraff53a5c422007-06-11 10:41:07 -0500922 return;
923 }
Luca Ceresolic718b142011-05-14 05:49:57 +0000924 memset(Bitmap, 0, Mapsize);
David Updegraff53a5c422007-06-11 10:41:07 -0500925 PrevBitmapHole = 0;
926 Multicast = 1;
927 }
928 addr = string_to_ip(mc_adr);
929 if (Mcast_addr != addr) {
930 if (Mcast_addr)
931 eth_mcast_join(Mcast_addr, 0);
Luca Ceresoli2cb53602011-05-14 05:49:59 +0000932 Mcast_addr = addr;
933 if (eth_mcast_join(Mcast_addr, 1)) {
Luca Ceresolic718b142011-05-14 05:49:57 +0000934 printf("Fail to set mcast, revert to TFTP\n");
935 ProhibitMcast = 1;
David Updegraff53a5c422007-06-11 10:41:07 -0500936 mcast_cleanup();
937 NetStartAgain();
938 }
939 }
Luca Ceresolic718b142011-05-14 05:49:57 +0000940 MasterClient = (unsigned char)simple_strtoul((char *)mc, NULL, 10);
941 Mcast_port = (unsigned short)simple_strtoul(port, NULL, 10);
942 printf("Multicast: %s:%d [%d]\n", mc_adr, Mcast_port, MasterClient);
David Updegraff53a5c422007-06-11 10:41:07 -0500943 return;
944}
945
946#endif /* Multicast TFTP */