blob: 9d3370f60a6cbba5155129c092041198057a6619 [file] [log] [blame]
wdenkfe8c2802002-11-03 00:38:21 +00001/*
Luca Ceresolic42fc272011-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 Ceresolib640ed32011-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 */
wdenkfe8c2802002-11-03 00:38:21 +00008#include <common.h>
9#include <command.h>
Simon Glass1ab16922022-07-31 12:28:48 -060010#include <display_options.h>
Alexander Graf94c4b992016-05-06 21:01:01 +020011#include <efi_loader.h>
Simon Glass0af6e2d2019-08-01 09:46:52 -060012#include <env.h>
Simon Glass85f13782019-12-28 10:45:03 -070013#include <image.h>
Simon Glass2dc9c342020-05-10 11:40:01 -060014#include <lmb.h>
Simon Glass0f2af882020-05-10 11:40:05 -060015#include <log.h>
Joe Hershbergered191852015-03-22 17:09:08 -050016#include <mapmem.h>
wdenkfe8c2802002-11-03 00:38:21 +000017#include <net.h>
Simon Glass3ba929a2020-10-30 21:38:53 -060018#include <asm/global_data.h>
Lukasz Majewski21185922015-08-24 00:21:43 +020019#include <net/tftp.h>
wdenkfe8c2802002-11-03 00:38:21 +000020#include "bootp.h"
21
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +010022DECLARE_GLOBAL_DATA_PTR;
23
Luca Ceresolic42fc272011-05-14 05:49:56 +000024/* Well known TFTP port # */
25#define WELL_KNOWN_PORT 69
26/* Millisecs to timeout for lost pkt */
Bin Menge67eaa82015-08-27 22:25:51 -070027#define TIMEOUT 5000UL
Luca Ceresolic42fc272011-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
wdenk7182b0f2003-10-06 21:55:32 +000039#define TFTP_OACK 6
wdenkfe8c2802002-11-03 00:38:21 +000040
Joe Hershberger64701592015-04-08 01:41:07 -050041static ulong timeout_ms = TIMEOUT;
Tom Rini26011a32022-03-11 09:12:02 -050042static int timeout_count_max = (CONFIG_NET_RETRY_COUNT * 2);
Simon Glasscab9a8d2012-10-11 13:57:36 +000043static ulong time_start; /* Record time we started tftp */
Bartlomiej Sieka371cde72008-10-01 15:26:29 +020044
45/*
46 * These globals govern the timeout behavior when attempting a connection to a
Joe Hershberger64701592015-04-08 01:41:07 -050047 * TFTP server. tftp_timeout_ms specifies the number of milliseconds to
Bartlomiej Sieka371cde72008-10-01 15:26:29 +020048 * wait for the server to respond to initial connection. Second global,
Joe Hershberger64701592015-04-08 01:41:07 -050049 * tftp_timeout_count_max, gives the number of such connection retries.
50 * tftp_timeout_count_max must be non-negative and tftp_timeout_ms must be
Bartlomiej Sieka371cde72008-10-01 15:26:29 +020051 * positive. The globals are meant to be set (and restored) by code needing
52 * non-standard timeout behavior when initiating a TFTP transfer.
53 */
Joe Hershberger64701592015-04-08 01:41:07 -050054ulong tftp_timeout_ms = TIMEOUT;
Tom Rini26011a32022-03-11 09:12:02 -050055int tftp_timeout_count_max = (CONFIG_NET_RETRY_COUNT * 2);
Bartlomiej Sieka371cde72008-10-01 15:26:29 +020056
Remy Bohmerb46293f2009-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,
Ravik Hasijae387e9a2020-05-18 21:35:43 -070065 TFTP_ERR_OPTION_NEGOTIATION = 8,
Remy Bohmerb46293f2009-10-28 22:13:40 +010066};
67
Joe Hershberger5874dec2015-04-08 01:41:01 -050068static struct in_addr tftp_remote_ip;
Luca Ceresolic42fc272011-05-14 05:49:56 +000069/* The UDP port at their end */
Joe Hershberger64701592015-04-08 01:41:07 -050070static int tftp_remote_port;
Luca Ceresolic42fc272011-05-14 05:49:56 +000071/* The UDP port at our end */
Joe Hershberger64701592015-04-08 01:41:07 -050072static int tftp_our_port;
73static int timeout_count;
Luca Ceresolic42fc272011-05-14 05:49:56 +000074/* packet sequence number */
Joe Hershberger64701592015-04-08 01:41:07 -050075static ulong tftp_cur_block;
Luca Ceresolic42fc272011-05-14 05:49:56 +000076/* last packet sequence number received */
Joe Hershberger64701592015-04-08 01:41:07 -050077static ulong tftp_prev_block;
Luca Ceresolic42fc272011-05-14 05:49:56 +000078/* count of sequence number wraparounds */
Joe Hershberger64701592015-04-08 01:41:07 -050079static ulong tftp_block_wrap;
Luca Ceresolic42fc272011-05-14 05:49:56 +000080/* memory offset due to wrapping */
Joe Hershberger64701592015-04-08 01:41:07 -050081static ulong tftp_block_wrap_offset;
82static int tftp_state;
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +010083static ulong tftp_load_addr;
84#ifdef CONFIG_LMB
85static ulong tftp_load_size;
86#endif
Robin Getzb3b00ed2009-08-20 10:50:20 -040087#ifdef CONFIG_TFTP_TSIZE
Luca Ceresolic42fc272011-05-14 05:49:56 +000088/* The file size reported by the server */
Joe Hershberger64701592015-04-08 01:41:07 -050089static int tftp_tsize;
Luca Ceresolic42fc272011-05-14 05:49:56 +000090/* The number of hashes we printed */
Joe Hershberger64701592015-04-08 01:41:07 -050091static short tftp_tsize_num_hash;
Robin Getzb3b00ed2009-08-20 10:50:20 -040092#endif
Ramon Fried6e9aa542020-07-18 23:31:46 +030093/* The window size negotiated */
94static ushort tftp_windowsize;
95/* Next block to send ack to */
96static ushort tftp_next_ack;
97/* Last nack block we send */
98static ushort tftp_last_nack;
Simon Glass6a398d22011-10-24 18:00:07 +000099#ifdef CONFIG_CMD_TFTPPUT
Joe Hershberger64701592015-04-08 01:41:07 -0500100/* 1 if writing, else 0 */
101static int tftp_put_active;
102/* 1 if we have sent the last block */
103static int tftp_put_final_block_sent;
Simon Glass6a398d22011-10-24 18:00:07 +0000104#else
Joe Hershberger64701592015-04-08 01:41:07 -0500105#define tftp_put_active 0
Simon Glass6a398d22011-10-24 18:00:07 +0000106#endif
wdenkef893942004-02-23 16:11:30 +0000107
Luca Ceresoli562dfe52011-05-17 00:03:38 +0000108#define STATE_SEND_RRQ 1
wdenkfe8c2802002-11-03 00:38:21 +0000109#define STATE_DATA 2
110#define STATE_TOO_LARGE 3
111#define STATE_BAD_MAGIC 4
wdenk7182b0f2003-10-06 21:55:32 +0000112#define STATE_OACK 5
Luca Ceresolib640ed32011-05-17 00:03:39 +0000113#define STATE_RECV_WRQ 6
Simon Glass6a398d22011-10-24 18:00:07 +0000114#define STATE_SEND_WRQ 7
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700115#define STATE_INVALID_OPTION 8
wdenkfe8c2802002-11-03 00:38:21 +0000116
Luca Ceresolic42fc272011-05-14 05:49:56 +0000117/* default TFTP block size */
118#define TFTP_BLOCK_SIZE 512
119/* sequence number is 16 bit */
120#define TFTP_SEQUENCE_SIZE ((ulong)(1<<16))
wdenkef893942004-02-23 16:11:30 +0000121
wdenkfe8c2802002-11-03 00:38:21 +0000122#define DEFAULT_NAME_LEN (8 + 4 + 1)
123static char default_filename[DEFAULT_NAME_LEN];
Jean-Christophe PLAGNIOL-VILLARD58555412008-01-18 01:14:03 +0100124
125#ifndef CONFIG_TFTP_FILE_NAME_MAX_LEN
126#define MAX_LEN 128
127#else
128#define MAX_LEN CONFIG_TFTP_FILE_NAME_MAX_LEN
129#endif
130
131static char tftp_filename[MAX_LEN];
wdenkfe8c2802002-11-03 00:38:21 +0000132
Wolfgang Denk627f5c32007-08-14 09:47:27 +0200133/* 512 is poor choice for ethernet, MTU is typically 1500.
134 * Minus eth.hdrs thats 1468. Can get 2x better throughput with
David Updegraff7280da72007-06-11 10:41:07 -0500135 * almost-MTU block sizes. At least try... fall back to 512 if need be.
Alessandro Rubini8832f6e2009-08-07 13:59:06 +0200136 * (but those using CONFIG_IP_DEFRAG may want to set a larger block in cfg file)
David Updegraff7280da72007-06-11 10:41:07 -0500137 */
Ramon Fried6e9aa542020-07-18 23:31:46 +0300138
139/* When windowsize is defined to 1,
140 * tftp behaves the same way as it was
141 * never declared
142 */
143#ifdef CONFIG_TFTP_WINDOWSIZE
144#define TFTP_WINDOWSIZE CONFIG_TFTP_WINDOWSIZE
145#else
146#define TFTP_WINDOWSIZE 1
147#endif
Alessandro Rubini8832f6e2009-08-07 13:59:06 +0200148
Joe Hershberger64701592015-04-08 01:41:07 -0500149static unsigned short tftp_block_size = TFTP_BLOCK_SIZE;
Patrick Delaunay6c3a4472020-04-22 14:18:26 +0200150static unsigned short tftp_block_size_option = CONFIG_TFTP_BLOCKSIZE;
Ramon Fried6e9aa542020-07-18 23:31:46 +0300151static unsigned short tftp_window_size_option = TFTP_WINDOWSIZE;
David Updegraff7280da72007-06-11 10:41:07 -0500152
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +0100153static inline int store_block(int block, uchar *src, unsigned int len)
wdenkfe8c2802002-11-03 00:38:21 +0000154{
Ley Foon Tan81bb3242020-08-25 10:26:36 +0800155 ulong offset = block * tftp_block_size + tftp_block_wrap_offset -
156 tftp_block_size;
wdenkef893942004-02-23 16:11:30 +0000157 ulong newsize = offset + len;
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +0100158 ulong store_addr = tftp_load_addr + offset;
Tom Rinidb295e82022-07-23 13:04:54 -0400159 void *ptr;
Joe Hershbergered191852015-03-22 17:09:08 -0500160
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +0100161#ifdef CONFIG_LMB
Tom Rinidb295e82022-07-23 13:04:54 -0400162 ulong end_addr = tftp_load_addr + tftp_load_size;
Bin Mengaf549632019-11-15 22:20:13 -0800163
Tom Rinidb295e82022-07-23 13:04:54 -0400164 if (!end_addr)
165 end_addr = ULONG_MAX;
Bin Mengaf549632019-11-15 22:20:13 -0800166
Tom Rinidb295e82022-07-23 13:04:54 -0400167 if (store_addr < tftp_load_addr ||
168 store_addr + len > end_addr) {
169 puts("\nTFTP error: ");
170 puts("trying to overwrite reserved memory...\n");
171 return -1;
wdenkfe8c2802002-11-03 00:38:21 +0000172 }
Tom Rinidb295e82022-07-23 13:04:54 -0400173#endif
174 ptr = map_sysmem(store_addr, len);
175 memcpy(ptr, src, len);
176 unmap_sysmem(ptr);
wdenkfe8c2802002-11-03 00:38:21 +0000177
Joe Hershberger290c8992015-04-08 01:41:02 -0500178 if (net_boot_file_size < newsize)
179 net_boot_file_size = newsize;
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +0100180
181 return 0;
wdenkfe8c2802002-11-03 00:38:21 +0000182}
183
Simon Glassbd8a0e02011-10-24 18:00:04 +0000184/* Clear our state ready for a new transfer */
Simon Glass84313322011-10-27 06:24:28 +0000185static void new_transfer(void)
Simon Glassbd8a0e02011-10-24 18:00:04 +0000186{
Joe Hershberger64701592015-04-08 01:41:07 -0500187 tftp_prev_block = 0;
188 tftp_block_wrap = 0;
189 tftp_block_wrap_offset = 0;
Simon Glassbd8a0e02011-10-24 18:00:04 +0000190#ifdef CONFIG_CMD_TFTPPUT
Joe Hershberger64701592015-04-08 01:41:07 -0500191 tftp_put_final_block_sent = 0;
Simon Glassbd8a0e02011-10-24 18:00:04 +0000192#endif
193}
Simon Glass6a398d22011-10-24 18:00:07 +0000194
195#ifdef CONFIG_CMD_TFTPPUT
196/**
197 * Load the next block from memory to be sent over tftp.
198 *
199 * @param block Block number to send
200 * @param dst Destination buffer for data
201 * @param len Number of bytes in block (this one and every other)
Heinrich Schuchardt47b4c022022-01-19 18:05:50 +0100202 * Return: number of bytes loaded
Simon Glass6a398d22011-10-24 18:00:07 +0000203 */
204static int load_block(unsigned block, uchar *dst, unsigned len)
205{
206 /* We may want to get the final block from the previous set */
Ley Foon Tan5babe562020-08-25 10:26:37 +0800207 ulong offset = block * tftp_block_size + tftp_block_wrap_offset -
208 tftp_block_size;
Simon Glass6a398d22011-10-24 18:00:07 +0000209 ulong tosend = len;
210
Joe Hershberger290c8992015-04-08 01:41:02 -0500211 tosend = min(net_boot_file_size - offset, tosend);
Simon Glass892265d2019-12-28 10:45:02 -0700212 (void)memcpy(dst, (void *)(image_save_addr + offset), tosend);
Heinrich Schuchardtc81ab832020-02-22 08:43:40 +0100213 debug("%s: block=%u, offset=%lu, len=%u, tosend=%lu\n", __func__,
Joe Hershberger64701592015-04-08 01:41:07 -0500214 block, offset, len, tosend);
Simon Glass6a398d22011-10-24 18:00:07 +0000215 return tosend;
216}
217#endif
Simon Glassbd8a0e02011-10-24 18:00:04 +0000218
Joe Hershberger64701592015-04-08 01:41:07 -0500219static void tftp_send(void);
220static void tftp_timeout_handler(void);
wdenkfe8c2802002-11-03 00:38:21 +0000221
222/**********************************************************************/
223
Simon Glassb437aad2011-10-24 18:00:03 +0000224static void show_block_marker(void)
225{
Ravik Hasijad42686e2020-05-07 14:55:32 -0700226 ulong pos;
227
Simon Glassb437aad2011-10-24 18:00:03 +0000228#ifdef CONFIG_TFTP_TSIZE
Joe Hershberger64701592015-04-08 01:41:07 -0500229 if (tftp_tsize) {
Ravik Hasijad42686e2020-05-07 14:55:32 -0700230 pos = tftp_cur_block * tftp_block_size +
Joe Hershberger64701592015-04-08 01:41:07 -0500231 tftp_block_wrap_offset;
Max Krummenacherdd081cc2015-08-05 17:17:05 +0200232 if (pos > tftp_tsize)
233 pos = tftp_tsize;
Simon Glassb437aad2011-10-24 18:00:03 +0000234
Joe Hershberger64701592015-04-08 01:41:07 -0500235 while (tftp_tsize_num_hash < pos * 50 / tftp_tsize) {
Simon Glassb437aad2011-10-24 18:00:03 +0000236 putc('#');
Joe Hershberger64701592015-04-08 01:41:07 -0500237 tftp_tsize_num_hash++;
Simon Glassb437aad2011-10-24 18:00:03 +0000238 }
Simon Glass6a398d22011-10-24 18:00:07 +0000239 } else
Simon Glassb437aad2011-10-24 18:00:03 +0000240#endif
Simon Glass6a398d22011-10-24 18:00:07 +0000241 {
Ravik Hasijad42686e2020-05-07 14:55:32 -0700242 pos = (tftp_cur_block - 1) +
243 (tftp_block_wrap * TFTP_SEQUENCE_SIZE);
244 if ((pos % 10) == 0)
Simon Glassb437aad2011-10-24 18:00:03 +0000245 putc('#');
Ravik Hasijad42686e2020-05-07 14:55:32 -0700246 else if (((pos + 1) % (10 * HASHES_PER_LINE)) == 0)
Simon Glassb437aad2011-10-24 18:00:03 +0000247 puts("\n\t ");
248 }
249}
250
Simon Glassbd8a0e02011-10-24 18:00:04 +0000251/**
252 * restart the current transfer due to an error
253 *
254 * @param msg Message to print for user
255 */
256static void restart(const char *msg)
257{
258 printf("\n%s; starting again\n", msg);
Joe Hershbergerc80b41b02015-04-08 01:41:21 -0500259 net_start_again();
Simon Glassbd8a0e02011-10-24 18:00:04 +0000260}
261
262/*
263 * Check if the block number has wrapped, and update progress
264 *
265 * TODO: The egregious use of global variables in this file should be tidied.
266 */
267static void update_block_number(void)
268{
269 /*
270 * RFC1350 specifies that the first data packet will
271 * have sequence number 1. If we receive a sequence
272 * number of 0 this means that there was a wrap
273 * around of the (16 bit) counter.
274 */
Joe Hershberger64701592015-04-08 01:41:07 -0500275 if (tftp_cur_block == 0 && tftp_prev_block != 0) {
276 tftp_block_wrap++;
277 tftp_block_wrap_offset += tftp_block_size * TFTP_SEQUENCE_SIZE;
278 timeout_count = 0; /* we've done well, reset the timeout */
Simon Glassbd8a0e02011-10-24 18:00:04 +0000279 }
Ravik Hasijad42686e2020-05-07 14:55:32 -0700280 show_block_marker();
Simon Glassbd8a0e02011-10-24 18:00:04 +0000281}
282
Simon Glassb437aad2011-10-24 18:00:03 +0000283/* The TFTP get or put is complete */
284static void tftp_complete(void)
285{
286#ifdef CONFIG_TFTP_TSIZE
287 /* Print hash marks for the last packet received */
Joe Hershberger64701592015-04-08 01:41:07 -0500288 while (tftp_tsize && tftp_tsize_num_hash < 49) {
Simon Glassb437aad2011-10-24 18:00:03 +0000289 putc('#');
Joe Hershberger64701592015-04-08 01:41:07 -0500290 tftp_tsize_num_hash++;
Simon Glassb437aad2011-10-24 18:00:03 +0000291 }
Simon Glassc8f0a5c2014-10-10 07:30:21 -0600292 puts(" ");
Joe Hershberger64701592015-04-08 01:41:07 -0500293 print_size(tftp_tsize, "");
Simon Glassb437aad2011-10-24 18:00:03 +0000294#endif
Simon Glasscab9a8d2012-10-11 13:57:36 +0000295 time_start = get_timer(time_start);
296 if (time_start > 0) {
297 puts("\n\t "); /* Line up with "Loading: " */
Joe Hershberger290c8992015-04-08 01:41:02 -0500298 print_size(net_boot_file_size /
Simon Glasscab9a8d2012-10-11 13:57:36 +0000299 time_start * 1000, "/s");
300 }
Simon Glassb437aad2011-10-24 18:00:03 +0000301 puts("\ndone\n");
Heinrich Schuchardt6b821592021-01-12 12:46:24 +0100302 if (IS_ENABLED(CONFIG_CMD_BOOTEFI)) {
303 if (!tftp_put_active)
304 efi_set_bootdev("Net", "", tftp_filename,
305 map_sysmem(tftp_load_addr, 0),
306 net_boot_file_size);
307 }
Joe Hershbergerd4bb76a2012-05-23 07:59:14 +0000308 net_set_state(NETLOOP_SUCCESS);
Simon Glassb437aad2011-10-24 18:00:03 +0000309}
310
Joe Hershberger64701592015-04-08 01:41:07 -0500311static void tftp_send(void)
wdenkfe8c2802002-11-03 00:38:21 +0000312{
Simon Glass6a398d22011-10-24 18:00:07 +0000313 uchar *pkt;
Joe Hershberger4b7747e2012-05-15 08:59:04 +0000314 uchar *xp;
315 int len = 0;
316 ushort *s;
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700317 bool err_pkt = false;
wdenkfe8c2802002-11-03 00:38:21 +0000318
319 /*
320 * We will always be sending some sort of packet, so
321 * cobble together the packet headers now.
322 */
Joe Hershbergera8ca4f62015-04-08 01:41:05 -0500323 pkt = net_tx_packet + net_eth_hdr_size() + IP_UDP_HDR_SIZE;
wdenkfe8c2802002-11-03 00:38:21 +0000324
Joe Hershberger64701592015-04-08 01:41:07 -0500325 switch (tftp_state) {
Luca Ceresoli562dfe52011-05-17 00:03:38 +0000326 case STATE_SEND_RRQ:
Simon Glass6a398d22011-10-24 18:00:07 +0000327 case STATE_SEND_WRQ:
wdenkfe8c2802002-11-03 00:38:21 +0000328 xp = pkt;
Wolfgang Denk3135c542005-08-26 01:36:03 +0200329 s = (ushort *)pkt;
Simon Glass4abd0e22011-10-27 06:24:29 +0000330#ifdef CONFIG_CMD_TFTPPUT
Joe Hershberger64701592015-04-08 01:41:07 -0500331 *s++ = htons(tftp_state == STATE_SEND_RRQ ? TFTP_RRQ :
Simon Glass6a398d22011-10-24 18:00:07 +0000332 TFTP_WRQ);
Simon Glass4abd0e22011-10-27 06:24:29 +0000333#else
334 *s++ = htons(TFTP_RRQ);
335#endif
Wolfgang Denk3135c542005-08-26 01:36:03 +0200336 pkt = (uchar *)s;
Luca Ceresolif40538a2011-05-14 05:49:57 +0000337 strcpy((char *)pkt, tftp_filename);
wdenkfe8c2802002-11-03 00:38:21 +0000338 pkt += strlen(tftp_filename) + 1;
Luca Ceresolif40538a2011-05-14 05:49:57 +0000339 strcpy((char *)pkt, "octet");
wdenkfe8c2802002-11-03 00:38:21 +0000340 pkt += 5 /*strlen("octet")*/ + 1;
Luca Ceresolif40538a2011-05-14 05:49:57 +0000341 strcpy((char *)pkt, "timeout");
wdenk7182b0f2003-10-06 21:55:32 +0000342 pkt += 7 /*strlen("timeout")*/ + 1;
Joe Hershberger64701592015-04-08 01:41:07 -0500343 sprintf((char *)pkt, "%lu", timeout_ms / 1000);
Robin Getz9e0a4d62009-07-23 03:01:03 -0400344 debug("send option \"timeout %s\"\n", (char *)pkt);
wdenk7182b0f2003-10-06 21:55:32 +0000345 pkt += strlen((char *)pkt) + 1;
Robin Getzb3b00ed2009-08-20 10:50:20 -0400346#ifdef CONFIG_TFTP_TSIZE
Joe Hershberger290c8992015-04-08 01:41:02 -0500347 pkt += sprintf((char *)pkt, "tsize%c%u%c",
348 0, net_boot_file_size, 0);
Robin Getzb3b00ed2009-08-20 10:50:20 -0400349#endif
David Updegraff7280da72007-06-11 10:41:07 -0500350 /* try for more effic. blk size */
Luca Ceresolif40538a2011-05-14 05:49:57 +0000351 pkt += sprintf((char *)pkt, "blksize%c%d%c",
Joe Hershberger64701592015-04-08 01:41:07 -0500352 0, tftp_block_size_option, 0);
Ramon Fried6e9aa542020-07-18 23:31:46 +0300353
354 /* try for more effic. window size.
355 * Implemented only for tftp get.
356 * Don't bother sending if it's 1
357 */
358 if (tftp_state == STATE_SEND_RRQ && tftp_window_size_option > 1)
359 pkt += sprintf((char *)pkt, "windowsize%c%d%c",
360 0, tftp_window_size_option, 0);
wdenkfe8c2802002-11-03 00:38:21 +0000361 len = pkt - xp;
362 break;
363
wdenk7182b0f2003-10-06 21:55:32 +0000364 case STATE_OACK:
Luca Ceresolib640ed32011-05-17 00:03:39 +0000365
366 case STATE_RECV_WRQ:
David Updegraff7280da72007-06-11 10:41:07 -0500367 case STATE_DATA:
wdenkfe8c2802002-11-03 00:38:21 +0000368 xp = pkt;
Wolfgang Denk3135c542005-08-26 01:36:03 +0200369 s = (ushort *)pkt;
Simon Glass6a398d22011-10-24 18:00:07 +0000370 s[0] = htons(TFTP_ACK);
Joe Hershberger64701592015-04-08 01:41:07 -0500371 s[1] = htons(tftp_cur_block);
Simon Glass6a398d22011-10-24 18:00:07 +0000372 pkt = (uchar *)(s + 2);
373#ifdef CONFIG_CMD_TFTPPUT
Joe Hershberger64701592015-04-08 01:41:07 -0500374 if (tftp_put_active) {
375 int toload = tftp_block_size;
376 int loaded = load_block(tftp_cur_block, pkt, toload);
Simon Glass6a398d22011-10-24 18:00:07 +0000377
378 s[0] = htons(TFTP_DATA);
379 pkt += loaded;
Joe Hershberger64701592015-04-08 01:41:07 -0500380 tftp_put_final_block_sent = (loaded < toload);
Simon Glass6a398d22011-10-24 18:00:07 +0000381 }
382#endif
wdenkfe8c2802002-11-03 00:38:21 +0000383 len = pkt - xp;
384 break;
385
386 case STATE_TOO_LARGE:
387 xp = pkt;
Wolfgang Denk3135c542005-08-26 01:36:03 +0200388 s = (ushort *)pkt;
389 *s++ = htons(TFTP_ERROR);
Simon Glass6a398d22011-10-24 18:00:07 +0000390 *s++ = htons(3);
391
Wolfgang Denk3135c542005-08-26 01:36:03 +0200392 pkt = (uchar *)s;
Luca Ceresolif40538a2011-05-14 05:49:57 +0000393 strcpy((char *)pkt, "File too large");
wdenkfe8c2802002-11-03 00:38:21 +0000394 pkt += 14 /*strlen("File too large")*/ + 1;
395 len = pkt - xp;
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700396 err_pkt = true;
wdenkfe8c2802002-11-03 00:38:21 +0000397 break;
398
399 case STATE_BAD_MAGIC:
400 xp = pkt;
Wolfgang Denk3135c542005-08-26 01:36:03 +0200401 s = (ushort *)pkt;
402 *s++ = htons(TFTP_ERROR);
403 *s++ = htons(2);
404 pkt = (uchar *)s;
Luca Ceresolif40538a2011-05-14 05:49:57 +0000405 strcpy((char *)pkt, "File has bad magic");
wdenkfe8c2802002-11-03 00:38:21 +0000406 pkt += 18 /*strlen("File has bad magic")*/ + 1;
407 len = pkt - xp;
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700408 err_pkt = true;
409 break;
410
411 case STATE_INVALID_OPTION:
412 xp = pkt;
413 s = (ushort *)pkt;
414 *s++ = htons(TFTP_ERROR);
415 *s++ = htons(TFTP_ERR_OPTION_NEGOTIATION);
416 pkt = (uchar *)s;
417 strcpy((char *)pkt, "Option Negotiation Failed");
418 /* strlen("Option Negotiation Failed") + NULL*/
419 pkt += 25 + 1;
420 len = pkt - xp;
421 err_pkt = true;
wdenkfe8c2802002-11-03 00:38:21 +0000422 break;
423 }
424
Joe Hershberger64701592015-04-08 01:41:07 -0500425 net_send_udp_packet(net_server_ethaddr, tftp_remote_ip,
426 tftp_remote_port, tftp_our_port, len);
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700427
428 if (err_pkt)
429 net_set_state(NETLOOP_FAIL);
wdenkfe8c2802002-11-03 00:38:21 +0000430}
431
Simon Glass2928cd82011-10-26 14:18:38 +0000432#ifdef CONFIG_CMD_TFTPPUT
Simon Glass6a398d22011-10-24 18:00:07 +0000433static void icmp_handler(unsigned type, unsigned code, unsigned dest,
Joe Hershberger5874dec2015-04-08 01:41:01 -0500434 struct in_addr sip, unsigned src, uchar *pkt,
435 unsigned len)
Simon Glass6a398d22011-10-24 18:00:07 +0000436{
437 if (type == ICMP_NOT_REACH && code == ICMP_NOT_REACH_PORT) {
438 /* Oh dear the other end has gone away */
439 restart("TFTP server died");
440 }
441}
Simon Glass2928cd82011-10-26 14:18:38 +0000442#endif
wdenkfe8c2802002-11-03 00:38:21 +0000443
Joe Hershberger5874dec2015-04-08 01:41:01 -0500444static void tftp_handler(uchar *pkt, unsigned dest, struct in_addr sip,
445 unsigned src, unsigned len)
wdenkfe8c2802002-11-03 00:38:21 +0000446{
Kim Phillips878dad02013-01-16 18:09:19 -0600447 __be16 proto;
448 __be16 *s;
Wolfgang Denk83e8e472007-08-30 14:42:15 +0200449 int i;
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700450 u16 timeout_val_rcvd;
wdenkfe8c2802002-11-03 00:38:21 +0000451
Joe Hershberger64701592015-04-08 01:41:07 -0500452 if (dest != tftp_our_port) {
Luca Ceresoli09b18232011-05-14 05:50:02 +0000453 return;
wdenkfe8c2802002-11-03 00:38:21 +0000454 }
Joe Hershberger64701592015-04-08 01:41:07 -0500455 if (tftp_state != STATE_SEND_RRQ && src != tftp_remote_port &&
456 tftp_state != STATE_RECV_WRQ && tftp_state != STATE_SEND_WRQ)
wdenkfe8c2802002-11-03 00:38:21 +0000457 return;
wdenkfe8c2802002-11-03 00:38:21 +0000458
Luca Ceresoli35dfb182011-05-14 05:50:00 +0000459 if (len < 2)
wdenkfe8c2802002-11-03 00:38:21 +0000460 return;
wdenkfe8c2802002-11-03 00:38:21 +0000461 len -= 2;
462 /* warning: don't use increment (++) in ntohs() macros!! */
Kim Phillips878dad02013-01-16 18:09:19 -0600463 s = (__be16 *)pkt;
Wolfgang Denk3135c542005-08-26 01:36:03 +0200464 proto = *s++;
465 pkt = (uchar *)s;
wdenkfe8c2802002-11-03 00:38:21 +0000466 switch (ntohs(proto)) {
wdenkfe8c2802002-11-03 00:38:21 +0000467 case TFTP_RRQ:
Simon Glass6a398d22011-10-24 18:00:07 +0000468 break;
469
wdenkfe8c2802002-11-03 00:38:21 +0000470 case TFTP_ACK:
Simon Glass6a398d22011-10-24 18:00:07 +0000471#ifdef CONFIG_CMD_TFTPPUT
Joe Hershberger64701592015-04-08 01:41:07 -0500472 if (tftp_put_active) {
473 if (tftp_put_final_block_sent) {
Simon Glass6a398d22011-10-24 18:00:07 +0000474 tftp_complete();
475 } else {
476 /*
477 * Move to the next block. We want our block
478 * count to wrap just like the other end!
479 */
480 int block = ntohs(*s);
Joe Hershberger64701592015-04-08 01:41:07 -0500481 int ack_ok = (tftp_cur_block == block);
Simon Glass6a398d22011-10-24 18:00:07 +0000482
Ley Foon Tan8de767a2020-08-25 10:26:35 +0800483 tftp_prev_block = tftp_cur_block;
Joe Hershberger64701592015-04-08 01:41:07 -0500484 tftp_cur_block = (unsigned short)(block + 1);
Simon Glass6a398d22011-10-24 18:00:07 +0000485 update_block_number();
486 if (ack_ok)
Joe Hershberger64701592015-04-08 01:41:07 -0500487 tftp_send(); /* Send next data block */
Simon Glass6a398d22011-10-24 18:00:07 +0000488 }
489 }
490#endif
wdenkfe8c2802002-11-03 00:38:21 +0000491 break;
Simon Glass6a398d22011-10-24 18:00:07 +0000492
wdenkfe8c2802002-11-03 00:38:21 +0000493 default:
494 break;
495
Luca Ceresolib640ed32011-05-17 00:03:39 +0000496#ifdef CONFIG_CMD_TFTPSRV
497 case TFTP_WRQ:
498 debug("Got WRQ\n");
Joe Hershberger5874dec2015-04-08 01:41:01 -0500499 tftp_remote_ip = sip;
Joe Hershberger64701592015-04-08 01:41:07 -0500500 tftp_remote_port = src;
501 tftp_our_port = 1024 + (get_timer(0) % 3072);
Simon Glassbd8a0e02011-10-24 18:00:04 +0000502 new_transfer();
Joe Hershberger64701592015-04-08 01:41:07 -0500503 tftp_send(); /* Send ACK(0) */
Luca Ceresolib640ed32011-05-17 00:03:39 +0000504 break;
505#endif
506
wdenk7182b0f2003-10-06 21:55:32 +0000507 case TFTP_OACK:
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700508 debug("Got OACK: ");
509 for (i = 0; i < len; i++) {
510 if (pkt[i] == '\0')
511 debug(" ");
512 else
513 debug("%c", pkt[i]);
514 }
515 debug("\n");
Joe Hershberger64701592015-04-08 01:41:07 -0500516 tftp_state = STATE_OACK;
517 tftp_remote_port = src;
Wolfgang Denk8b70f342007-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 Ceresolif40538a2011-05-14 05:49:57 +0000523 for (i = 0; i+8 < len; i++) {
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700524 if (strcasecmp((char *)pkt + i, "blksize") == 0) {
Joe Hershberger64701592015-04-08 01:41:07 -0500525 tftp_block_size = (unsigned short)
Simon Glassff9b9032021-07-24 09:03:30 -0600526 dectoul((char *)pkt + i + 8, NULL);
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700527 debug("Blocksize oack: %s, %d\n",
Joe Hershberger64701592015-04-08 01:41:07 -0500528 (char *)pkt + i + 8, tftp_block_size);
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700529 if (tftp_block_size > tftp_block_size_option) {
530 printf("Invalid blk size(=%d)\n",
531 tftp_block_size);
532 tftp_state = STATE_INVALID_OPTION;
533 }
534 }
535 if (strcasecmp((char *)pkt + i, "timeout") == 0) {
536 timeout_val_rcvd = (unsigned short)
Simon Glassff9b9032021-07-24 09:03:30 -0600537 dectoul((char *)pkt + i + 8, NULL);
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700538 debug("Timeout oack: %s, %d\n",
539 (char *)pkt + i + 8, timeout_val_rcvd);
540 if (timeout_val_rcvd != (timeout_ms / 1000)) {
541 printf("Invalid timeout val(=%d s)\n",
542 timeout_val_rcvd);
543 tftp_state = STATE_INVALID_OPTION;
544 }
Wolfgang Denk83e8e472007-08-30 14:42:15 +0200545 }
Robin Getzb3b00ed2009-08-20 10:50:20 -0400546#ifdef CONFIG_TFTP_TSIZE
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700547 if (strcasecmp((char *)pkt + i, "tsize") == 0) {
Simon Glassff9b9032021-07-24 09:03:30 -0600548 tftp_tsize = dectoul((char *)pkt + i + 6,
549 NULL);
Robin Getzb3b00ed2009-08-20 10:50:20 -0400550 debug("size = %s, %d\n",
Joe Hershberger64701592015-04-08 01:41:07 -0500551 (char *)pkt + i + 6, tftp_tsize);
Robin Getzb3b00ed2009-08-20 10:50:20 -0400552 }
553#endif
Ramon Fried6e9aa542020-07-18 23:31:46 +0300554 if (strcasecmp((char *)pkt + i, "windowsize") == 0) {
555 tftp_windowsize =
Simon Glassff9b9032021-07-24 09:03:30 -0600556 dectoul((char *)pkt + i + 11, NULL);
Ramon Fried6e9aa542020-07-18 23:31:46 +0300557 debug("windowsize = %s, %d\n",
558 (char *)pkt + i + 11, tftp_windowsize);
559 }
David Updegraff7280da72007-06-11 10:41:07 -0500560 }
Ramon Fried6e9aa542020-07-18 23:31:46 +0300561
562 tftp_next_ack = tftp_windowsize;
563
Simon Glass6a398d22011-10-24 18:00:07 +0000564#ifdef CONFIG_CMD_TFTPPUT
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700565 if (tftp_put_active && tftp_state == STATE_OACK) {
Simon Glass6a398d22011-10-24 18:00:07 +0000566 /* Get ready to send the first block */
Joe Hershberger64701592015-04-08 01:41:07 -0500567 tftp_state = STATE_DATA;
568 tftp_cur_block++;
Simon Glass6a398d22011-10-24 18:00:07 +0000569 }
570#endif
Joe Hershberger64701592015-04-08 01:41:07 -0500571 tftp_send(); /* Send ACK or first data block */
wdenk7182b0f2003-10-06 21:55:32 +0000572 break;
wdenkfe8c2802002-11-03 00:38:21 +0000573 case TFTP_DATA:
574 if (len < 2)
575 return;
576 len -= 2;
Ramon Fried6e9aa542020-07-18 23:31:46 +0300577
578 if (ntohs(*(__be16 *)pkt) != (ushort)(tftp_cur_block + 1)) {
579 debug("Received unexpected block: %d, expected: %d\n",
580 ntohs(*(__be16 *)pkt),
581 (ushort)(tftp_cur_block + 1));
582 /*
583 * If one packet is dropped most likely
584 * all other buffers in the window
585 * that will arrive will cause a sending NACK.
586 * This just overwellms the server, let's just send one.
587 */
588 if (tftp_last_nack != tftp_cur_block) {
589 tftp_send();
590 tftp_last_nack = tftp_cur_block;
591 tftp_next_ack = (ushort)(tftp_cur_block +
592 tftp_windowsize);
593 }
594 break;
595 }
596
597 tftp_cur_block++;
598 tftp_cur_block %= TFTP_SEQUENCE_SIZE;
wdenkef893942004-02-23 16:11:30 +0000599
Harm Berntsend097b1a2020-11-27 21:45:56 +0000600 if (tftp_state == STATE_SEND_RRQ) {
Ravik Hasijae387e9a2020-05-18 21:35:43 -0700601 debug("Server did not acknowledge any options!\n");
Harm Berntsend097b1a2020-11-27 21:45:56 +0000602 tftp_next_ack = tftp_windowsize;
603 }
wdenk7182b0f2003-10-06 21:55:32 +0000604
Joe Hershberger64701592015-04-08 01:41:07 -0500605 if (tftp_state == STATE_SEND_RRQ || tftp_state == STATE_OACK ||
606 tftp_state == STATE_RECV_WRQ) {
wdenkef893942004-02-23 16:11:30 +0000607 /* first block received */
Joe Hershberger64701592015-04-08 01:41:07 -0500608 tftp_state = STATE_DATA;
609 tftp_remote_port = src;
Simon Glassbd8a0e02011-10-24 18:00:04 +0000610 new_transfer();
wdenkfe8c2802002-11-03 00:38:21 +0000611
Joe Hershberger64701592015-04-08 01:41:07 -0500612 if (tftp_cur_block != 1) { /* Assertion */
613 puts("\nTFTP error: ");
614 printf("First block is not block 1 (%ld)\n",
615 tftp_cur_block);
616 puts("Starting again\n\n");
Joe Hershbergerc80b41b02015-04-08 01:41:21 -0500617 net_start_again();
wdenkfe8c2802002-11-03 00:38:21 +0000618 break;
619 }
620 }
621
Joe Hershberger64701592015-04-08 01:41:07 -0500622 if (tftp_cur_block == tftp_prev_block) {
623 /* Same block again; ignore it. */
wdenkfe8c2802002-11-03 00:38:21 +0000624 break;
625 }
626
Ravik Hasijad42686e2020-05-07 14:55:32 -0700627 update_block_number();
Joe Hershberger64701592015-04-08 01:41:07 -0500628 tftp_prev_block = tftp_cur_block;
Albert ARIBAUD \(3ADEV\)83006852015-10-12 00:02:57 +0200629 timeout_count_max = tftp_timeout_count_max;
Joe Hershbergerc80b41b02015-04-08 01:41:21 -0500630 net_set_timeout_handler(timeout_ms, tftp_timeout_handler);
wdenkfe8c2802002-11-03 00:38:21 +0000631
Ley Foon Tan81bb3242020-08-25 10:26:36 +0800632 if (store_block(tftp_cur_block, pkt + 2, len)) {
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +0100633 eth_halt();
634 net_set_state(NETLOOP_FAIL);
635 break;
636 }
wdenkfe8c2802002-11-03 00:38:21 +0000637
Ramon Friedb891c792021-02-03 21:28:59 +0200638 if (len < tftp_block_size) {
639 tftp_send();
640 tftp_complete();
641 break;
642 }
643
wdenkfe8c2802002-11-03 00:38:21 +0000644 /*
Luca Ceresoli2f3dc0a2011-05-17 00:03:41 +0000645 * Acknowledge the block just received, which will prompt
Luca Ceresoli5079c762011-05-17 00:03:37 +0000646 * the remote for the next one.
wdenkfe8c2802002-11-03 00:38:21 +0000647 */
Ramon Fried6e9aa542020-07-18 23:31:46 +0300648 if (tftp_cur_block == tftp_next_ack) {
649 tftp_send();
650 tftp_next_ack += tftp_windowsize;
651 }
wdenkfe8c2802002-11-03 00:38:21 +0000652 break;
653
654 case TFTP_ERROR:
Luca Ceresolif40538a2011-05-14 05:49:57 +0000655 printf("\nTFTP error: '%s' (%d)\n",
Kim Phillips878dad02013-01-16 18:09:19 -0600656 pkt + 2, ntohs(*(__be16 *)pkt));
Remy Bohmerb46293f2009-10-28 22:13:40 +0100657
Kim Phillips878dad02013-01-16 18:09:19 -0600658 switch (ntohs(*(__be16 *)pkt)) {
Remy Bohmerb46293f2009-10-28 22:13:40 +0100659 case TFTP_ERR_FILE_NOT_FOUND:
660 case TFTP_ERR_ACCESS_DENIED:
661 puts("Not retrying...\n");
662 eth_halt();
Joe Hershbergerd4bb76a2012-05-23 07:59:14 +0000663 net_set_state(NETLOOP_FAIL);
Remy Bohmerb46293f2009-10-28 22:13:40 +0100664 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");
Joe Hershbergerc80b41b02015-04-08 01:41:21 -0500672 net_start_again();
Remy Bohmerb46293f2009-10-28 22:13:40 +0100673 break;
674 }
wdenkfe8c2802002-11-03 00:38:21 +0000675 break;
676 }
677}
678
679
Joe Hershberger64701592015-04-08 01:41:07 -0500680static void tftp_timeout_handler(void)
wdenkfe8c2802002-11-03 00:38:21 +0000681{
Joe Hershberger64701592015-04-08 01:41:07 -0500682 if (++timeout_count > timeout_count_max) {
Simon Glassbd8a0e02011-10-24 18:00:04 +0000683 restart("Retry count exceeded");
wdenkfe8c2802002-11-03 00:38:21 +0000684 } else {
Luca Ceresolif40538a2011-05-14 05:49:57 +0000685 puts("T ");
Joe Hershbergerc80b41b02015-04-08 01:41:21 -0500686 net_set_timeout_handler(timeout_ms, tftp_timeout_handler);
Joe Hershberger64701592015-04-08 01:41:07 -0500687 if (tftp_state != STATE_RECV_WRQ)
688 tftp_send();
wdenkfe8c2802002-11-03 00:38:21 +0000689 }
690}
691
Simon Glass892265d2019-12-28 10:45:02 -0700692/* Initialize tftp_load_addr and tftp_load_size from image_load_addr and lmb */
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +0100693static int tftp_init_load_addr(void)
694{
695#ifdef CONFIG_LMB
696 struct lmb lmb;
697 phys_size_t max_size;
698
Simon Goldschmidt8890e7d2019-01-26 22:13:04 +0100699 lmb_init_and_reserve(&lmb, gd->bd, (void *)gd->fdt_blob);
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +0100700
Simon Glass892265d2019-12-28 10:45:02 -0700701 max_size = lmb_get_free_size(&lmb, image_load_addr);
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +0100702 if (!max_size)
703 return -1;
704
705 tftp_load_size = max_size;
706#endif
Simon Glass892265d2019-12-28 10:45:02 -0700707 tftp_load_addr = image_load_addr;
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +0100708 return 0;
709}
wdenkfe8c2802002-11-03 00:38:21 +0000710
Joe Hershberger64701592015-04-08 01:41:07 -0500711void tftp_start(enum proto_t protocol)
wdenkfe8c2802002-11-03 00:38:21 +0000712{
Rasmus Villemoes82ff5b02022-10-14 19:43:41 +0200713 __maybe_unused char *ep; /* Environment pointer */
714 if (IS_ENABLED(CONFIG_NET_TFTP_VARS)) {
Alessandro Rubini8832f6e2009-08-07 13:59:06 +0200715
Rasmus Villemoes82ff5b02022-10-14 19:43:41 +0200716 /*
717 * Allow the user to choose TFTP blocksize and timeout.
718 * TFTP protocol has a minimal timeout of 1 second.
719 */
Albert ARIBAUD \(3ADEV\)83006852015-10-12 00:02:57 +0200720
Rasmus Villemoes82ff5b02022-10-14 19:43:41 +0200721 ep = env_get("tftpblocksize");
722 if (ep != NULL)
723 tftp_block_size_option = simple_strtol(ep, NULL, 10);
Wolfgang Denkb233bd72010-01-17 23:55:53 +0100724
Rasmus Villemoes82ff5b02022-10-14 19:43:41 +0200725 ep = env_get("tftpwindowsize");
726 if (ep != NULL)
727 tftp_window_size_option = simple_strtol(ep, NULL, 10);
Ramon Fried6e9aa542020-07-18 23:31:46 +0300728
Rasmus Villemoes82ff5b02022-10-14 19:43:41 +0200729 ep = env_get("tftptimeout");
730 if (ep != NULL)
731 timeout_ms = simple_strtol(ep, NULL, 10);
Wolfgang Denkb233bd72010-01-17 23:55:53 +0100732
Rasmus Villemoes82ff5b02022-10-14 19:43:41 +0200733 if (timeout_ms < 1000) {
734 printf("TFTP timeout (%ld ms) too low, set min = 1000 ms\n",
735 timeout_ms);
736 timeout_ms = 1000;
737 }
Wolfgang Denkb233bd72010-01-17 23:55:53 +0100738
Rasmus Villemoes82ff5b02022-10-14 19:43:41 +0200739 ep = env_get("tftptimeoutcountmax");
740 if (ep != NULL)
741 tftp_timeout_count_max = simple_strtol(ep, NULL, 10);
Albert ARIBAUD \(3ADEV\)83006852015-10-12 00:02:57 +0200742
Rasmus Villemoes82ff5b02022-10-14 19:43:41 +0200743 if (tftp_timeout_count_max < 0) {
744 printf("TFTP timeout count max (%d ms) negative, set to 0\n",
745 tftp_timeout_count_max);
746 tftp_timeout_count_max = 0;
747 }
Albert ARIBAUD \(3ADEV\)83006852015-10-12 00:02:57 +0200748 }
Albert ARIBAUD \(3ADEV\)83006852015-10-12 00:02:57 +0200749
Ramon Fried6e9aa542020-07-18 23:31:46 +0300750 debug("TFTP blocksize = %i, TFTP windowsize = %d timeout = %ld ms\n",
751 tftp_block_size_option, tftp_window_size_option, timeout_ms);
Wolfgang Denke3cfce52005-09-24 22:37:32 +0200752
Joe Hershberger5874dec2015-04-08 01:41:01 -0500753 tftp_remote_ip = net_server_ip;
Joe Hershberger9cb7b022018-07-03 19:36:43 -0500754 if (!net_parse_bootfile(&tftp_remote_ip, tftp_filename, MAX_LEN)) {
Matthias Weisser6cc63262011-12-03 03:29:44 +0000755 sprintf(default_filename, "%02X%02X%02X%02X.img",
Joe Hershberger5874dec2015-04-08 01:41:01 -0500756 net_ip.s_addr & 0xFF,
757 (net_ip.s_addr >> 8) & 0xFF,
758 (net_ip.s_addr >> 16) & 0xFF,
759 (net_ip.s_addr >> 24) & 0xFF);
Jean-Christophe PLAGNIOL-VILLARD58555412008-01-18 01:14:03 +0100760
Vladimir Zapolskiy04cc2902017-06-28 22:56:07 +0300761 strncpy(tftp_filename, default_filename, DEFAULT_NAME_LEN);
762 tftp_filename[DEFAULT_NAME_LEN - 1] = 0;
wdenkfe8c2802002-11-03 00:38:21 +0000763
Luca Ceresolif40538a2011-05-14 05:49:57 +0000764 printf("*** Warning: no boot file name; using '%s'\n",
Joe Hershberger64701592015-04-08 01:41:07 -0500765 tftp_filename);
wdenkfe8c2802002-11-03 00:38:21 +0000766 }
767
Luca Ceresolif40538a2011-05-14 05:49:57 +0000768 printf("Using %s device\n", eth_get_name());
Simon Glass6a398d22011-10-24 18:00:07 +0000769 printf("TFTP %s server %pI4; our IP address is %pI4",
Simon Glass4abd0e22011-10-27 06:24:29 +0000770#ifdef CONFIG_CMD_TFTPPUT
771 protocol == TFTPPUT ? "to" : "from",
772#else
Joe Hershberger64701592015-04-08 01:41:07 -0500773 "from",
Simon Glass4abd0e22011-10-27 06:24:29 +0000774#endif
Joe Hershberger64701592015-04-08 01:41:07 -0500775 &tftp_remote_ip, &net_ip);
wdenkfe8c2802002-11-03 00:38:21 +0000776
777 /* Check if we need to send across this subnet */
Joe Hershberger5874dec2015-04-08 01:41:01 -0500778 if (net_gateway.s_addr && net_netmask.s_addr) {
779 struct in_addr our_net;
780 struct in_addr remote_net;
wdenkfe8c2802002-11-03 00:38:21 +0000781
Joe Hershberger5874dec2015-04-08 01:41:01 -0500782 our_net.s_addr = net_ip.s_addr & net_netmask.s_addr;
783 remote_net.s_addr = tftp_remote_ip.s_addr & net_netmask.s_addr;
784 if (our_net.s_addr != remote_net.s_addr)
785 printf("; sending through gateway %pI4", &net_gateway);
wdenkfe8c2802002-11-03 00:38:21 +0000786 }
Luca Ceresolif40538a2011-05-14 05:49:57 +0000787 putc('\n');
wdenkfe8c2802002-11-03 00:38:21 +0000788
Luca Ceresolif40538a2011-05-14 05:49:57 +0000789 printf("Filename '%s'.", tftp_filename);
wdenkfe8c2802002-11-03 00:38:21 +0000790
Joe Hershberger290c8992015-04-08 01:41:02 -0500791 if (net_boot_file_expected_size_in_blocks) {
792 printf(" Size is 0x%x Bytes = ",
793 net_boot_file_expected_size_in_blocks << 9);
794 print_size(net_boot_file_expected_size_in_blocks << 9, "");
wdenkfe8c2802002-11-03 00:38:21 +0000795 }
796
Luca Ceresolif40538a2011-05-14 05:49:57 +0000797 putc('\n');
Simon Glass6a398d22011-10-24 18:00:07 +0000798#ifdef CONFIG_CMD_TFTPPUT
Joe Hershberger64701592015-04-08 01:41:07 -0500799 tftp_put_active = (protocol == TFTPPUT);
800 if (tftp_put_active) {
Simon Glass892265d2019-12-28 10:45:02 -0700801 printf("Save address: 0x%lx\n", image_save_addr);
802 printf("Save size: 0x%lx\n", image_save_size);
803 net_boot_file_size = image_save_size;
Simon Glass6a398d22011-10-24 18:00:07 +0000804 puts("Saving: *\b");
Joe Hershberger64701592015-04-08 01:41:07 -0500805 tftp_state = STATE_SEND_WRQ;
Simon Glass6a398d22011-10-24 18:00:07 +0000806 new_transfer();
807 } else
808#endif
809 {
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +0100810 if (tftp_init_load_addr()) {
811 eth_halt();
812 net_set_state(NETLOOP_FAIL);
813 puts("\nTFTP error: ");
814 puts("trying to overwrite reserved memory...\n");
815 return;
816 }
817 printf("Load address: 0x%lx\n", tftp_load_addr);
Simon Glass6a398d22011-10-24 18:00:07 +0000818 puts("Loading: *\b");
Joe Hershberger64701592015-04-08 01:41:07 -0500819 tftp_state = STATE_SEND_RRQ;
Simon Glass6a398d22011-10-24 18:00:07 +0000820 }
wdenkfe8c2802002-11-03 00:38:21 +0000821
Simon Glasscab9a8d2012-10-11 13:57:36 +0000822 time_start = get_timer(0);
Joe Hershberger64701592015-04-08 01:41:07 -0500823 timeout_count_max = tftp_timeout_count_max;
Bartlomiej Sieka371cde72008-10-01 15:26:29 +0200824
Joe Hershbergerc80b41b02015-04-08 01:41:21 -0500825 net_set_timeout_handler(timeout_ms, tftp_timeout_handler);
Joe Hershberger5874dec2015-04-08 01:41:01 -0500826 net_set_udp_handler(tftp_handler);
Simon Glass2928cd82011-10-26 14:18:38 +0000827#ifdef CONFIG_CMD_TFTPPUT
Simon Glass6a398d22011-10-24 18:00:07 +0000828 net_set_icmp_handler(icmp_handler);
Simon Glass2928cd82011-10-26 14:18:38 +0000829#endif
Joe Hershberger64701592015-04-08 01:41:07 -0500830 tftp_remote_port = WELL_KNOWN_PORT;
831 timeout_count = 0;
Wolfgang Denke3cfce52005-09-24 22:37:32 +0200832 /* Use a pseudo-random port unless a specific port is set */
Joe Hershberger64701592015-04-08 01:41:07 -0500833 tftp_our_port = 1024 + (get_timer(0) % 3072);
David Updegraff7280da72007-06-11 10:41:07 -0500834
Wolfgang Denke3cfce52005-09-24 22:37:32 +0200835#ifdef CONFIG_TFTP_PORT
Simon Glass64b723f2017-08-03 12:22:12 -0600836 ep = env_get("tftpdstp");
Luca Ceresoli35dfb182011-05-14 05:50:00 +0000837 if (ep != NULL)
Joe Hershberger64701592015-04-08 01:41:07 -0500838 tftp_remote_port = simple_strtol(ep, NULL, 10);
Simon Glass64b723f2017-08-03 12:22:12 -0600839 ep = env_get("tftpsrcp");
Luca Ceresoli35dfb182011-05-14 05:50:00 +0000840 if (ep != NULL)
Joe Hershberger64701592015-04-08 01:41:07 -0500841 tftp_our_port = simple_strtol(ep, NULL, 10);
Wolfgang Denke3cfce52005-09-24 22:37:32 +0200842#endif
Joe Hershberger64701592015-04-08 01:41:07 -0500843 tftp_cur_block = 0;
Ramon Fried6e9aa542020-07-18 23:31:46 +0300844 tftp_windowsize = 1;
845 tftp_last_nack = 0;
wdenke6466f62003-06-05 19:27:42 +0000846 /* zero out server ether in case the server ip has changed */
Joe Hershberger8ecdbed2015-04-08 01:41:04 -0500847 memset(net_server_ethaddr, 0, 6);
Joe Hershberger64701592015-04-08 01:41:07 -0500848 /* Revert tftp_block_size to dflt */
849 tftp_block_size = TFTP_BLOCK_SIZE;
Robin Getzb3b00ed2009-08-20 10:50:20 -0400850#ifdef CONFIG_TFTP_TSIZE
Joe Hershberger64701592015-04-08 01:41:07 -0500851 tftp_tsize = 0;
852 tftp_tsize_num_hash = 0;
Robin Getzb3b00ed2009-08-20 10:50:20 -0400853#endif
wdenke6466f62003-06-05 19:27:42 +0000854
Joe Hershberger64701592015-04-08 01:41:07 -0500855 tftp_send();
wdenkfe8c2802002-11-03 00:38:21 +0000856}
Luca Ceresolib640ed32011-05-17 00:03:39 +0000857
858#ifdef CONFIG_CMD_TFTPSRV
Joe Hershberger64701592015-04-08 01:41:07 -0500859void tftp_start_server(void)
Luca Ceresolib640ed32011-05-17 00:03:39 +0000860{
861 tftp_filename[0] = 0;
862
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +0100863 if (tftp_init_load_addr()) {
864 eth_halt();
865 net_set_state(NETLOOP_FAIL);
866 puts("\nTFTP error: trying to overwrite reserved memory...\n");
867 return;
868 }
Luca Ceresolib640ed32011-05-17 00:03:39 +0000869 printf("Using %s device\n", eth_get_name());
Joe Hershberger5874dec2015-04-08 01:41:01 -0500870 printf("Listening for TFTP transfer on %pI4\n", &net_ip);
Simon Goldschmidta96cf8f2019-01-14 22:38:22 +0100871 printf("Load address: 0x%lx\n", tftp_load_addr);
Luca Ceresolib640ed32011-05-17 00:03:39 +0000872
873 puts("Loading: *\b");
874
Albert ARIBAUD \(3ADEV\)83006852015-10-12 00:02:57 +0200875 timeout_count_max = tftp_timeout_count_max;
Joe Hershberger64701592015-04-08 01:41:07 -0500876 timeout_count = 0;
877 timeout_ms = TIMEOUT;
Joe Hershbergerc80b41b02015-04-08 01:41:21 -0500878 net_set_timeout_handler(timeout_ms, tftp_timeout_handler);
Luca Ceresolib640ed32011-05-17 00:03:39 +0000879
Joe Hershberger64701592015-04-08 01:41:07 -0500880 /* Revert tftp_block_size to dflt */
881 tftp_block_size = TFTP_BLOCK_SIZE;
882 tftp_cur_block = 0;
883 tftp_our_port = WELL_KNOWN_PORT;
Arjan Minzinga Zijlstrafc134232022-03-31 08:03:16 +0000884 tftp_windowsize = 1;
885 tftp_next_ack = tftp_windowsize;
Luca Ceresolib640ed32011-05-17 00:03:39 +0000886
887#ifdef CONFIG_TFTP_TSIZE
Joe Hershberger64701592015-04-08 01:41:07 -0500888 tftp_tsize = 0;
889 tftp_tsize_num_hash = 0;
Luca Ceresolib640ed32011-05-17 00:03:39 +0000890#endif
891
Joe Hershberger64701592015-04-08 01:41:07 -0500892 tftp_state = STATE_RECV_WRQ;
Joe Hershberger5874dec2015-04-08 01:41:01 -0500893 net_set_udp_handler(tftp_handler);
Andrew Ruder91393c52013-10-22 19:10:28 -0500894
895 /* zero out server ether in case the server ip has changed */
Joe Hershberger8ecdbed2015-04-08 01:41:04 -0500896 memset(net_server_ethaddr, 0, 6);
Luca Ceresolib640ed32011-05-17 00:03:39 +0000897}
898#endif /* CONFIG_CMD_TFTPSRV */