[J-u-boot.git] / net / wget.c

// SPDX-License-Identifier: GPL-2.0
/*
 * WGET/HTTP support driver based on U-BOOT's nfs.c
 * Copyright Duncan Hare <[email protected]> 2017
 */

#include <asm/global_data.h>
#include <command.h>
#include <common.h>
#include <display_options.h>
#include <env.h>
#include <image.h>
#include <lmb.h>
#include <mapmem.h>
#include <net.h>
#include <net/tcp.h>
#include <net/wget.h>
#include <stdlib.h>

DECLARE_GLOBAL_DATA_PTR;

/* The default, change with environment variable 'httpdstp' */
#define SERVER_PORT		80

static const char bootfile1[] = "GET ";
static const char bootfile3[] = " HTTP/1.0\r\n\r\n";
static const char http_eom[] = "\r\n\r\n";
static const char http_ok[] = "200";
static const char content_len[] = "Content-Length";
static const char linefeed[] = "\r\n";
static struct in_addr web_server_ip;
static int our_port;
static int wget_timeout_count;

struct pkt_qd {
	uchar *pkt;
	unsigned int tcp_seq_num;
	unsigned int len;
};

/*
 * This is a control structure for out of order packets received.
 * The actual packet bufers are in the kernel space, and are
 * expected to be overwritten by the downloaded image.
 */
#define PKTQ_SZ (PKTBUFSRX / 4)
static struct pkt_qd pkt_q[PKTQ_SZ];
static int pkt_q_idx;
static unsigned long content_length;
static unsigned int packets;

static unsigned int initial_data_seq_num;
static unsigned int next_data_seq_num;

static enum  wget_state current_wget_state;

static char *image_url;
static unsigned int wget_timeout = WGET_TIMEOUT;

static enum net_loop_state wget_loop_state;

/* Timeout retry parameters */
static u8 retry_action;			/* actions for TCP retry */
static unsigned int retry_tcp_ack_num;	/* TCP retry acknowledge number*/
static unsigned int retry_tcp_seq_num;	/* TCP retry sequence number */
static int retry_len;			/* TCP retry length */

static ulong wget_load_size;

/**
 * wget_init_max_size() - initialize maximum load size
 *
 * Return:	0 if success, -1 if fails
 */
static int wget_init_load_size(void)
{
	struct lmb lmb;
	phys_size_t max_size;

	lmb_init_and_reserve(&lmb, gd->bd, (void *)gd->fdt_blob);

	max_size = lmb_get_free_size(&lmb, image_load_addr);
	if (!max_size)
		return -1;

	wget_load_size = max_size;

	return 0;
}

/**
 * store_block() - store block in memory
 * @src: source of data
 * @offset: offset
 * @len: length
 */
static inline int store_block(uchar *src, unsigned int offset, unsigned int len)
{
	ulong store_addr = image_load_addr + offset;
	ulong newsize = offset + len;
	uchar *ptr;

	if (IS_ENABLED(CONFIG_LMB)) {
		ulong end_addr = image_load_addr + wget_load_size;

		if (!end_addr)
			end_addr = ULONG_MAX;

		if (store_addr < image_load_addr ||
		    store_addr + len > end_addr) {
			printf("\nwget error: ");
			printf("trying to overwrite reserved memory...\n");
			return -1;
		}
	}

	ptr = map_sysmem(store_addr, len);
	memcpy(ptr, src, len);
	unmap_sysmem(ptr);

	if (net_boot_file_size < (offset + len))
		net_boot_file_size = newsize;

	return 0;
}

/**
 * wget_send_stored() - wget response dispatcher
 *
 * WARNING, This, and only this, is the place in wget.c where
 * SEQUENCE NUMBERS are swapped between incoming (RX)
 * and outgoing (TX).
 * Procedure wget_handler() is correct for RX traffic.
 */
static void wget_send_stored(void)
{
	u8 action = retry_action;
	int len = retry_len;
	unsigned int tcp_ack_num = retry_tcp_seq_num + (len == 0 ? 1 : len);
	unsigned int tcp_seq_num = retry_tcp_ack_num;
	unsigned int server_port;
	uchar *ptr, *offset;

	server_port = env_get_ulong("httpdstp", 10, SERVER_PORT) & 0xffff;

	switch (current_wget_state) {
	case WGET_CLOSED:
		debug_cond(DEBUG_WGET, "wget: send SYN\n");
		current_wget_state = WGET_CONNECTING;
		net_send_tcp_packet(0, server_port, our_port, action,
				    tcp_seq_num, tcp_ack_num);
		packets = 0;
		break;
	case WGET_CONNECTING:
		pkt_q_idx = 0;
		net_send_tcp_packet(0, server_port, our_port, action,
				    tcp_seq_num, tcp_ack_num);

		ptr = net_tx_packet + net_eth_hdr_size() +
			IP_TCP_HDR_SIZE + TCP_TSOPT_SIZE + 2;
		offset = ptr;

		memcpy(offset, &bootfile1, strlen(bootfile1));
		offset += strlen(bootfile1);

		memcpy(offset, image_url, strlen(image_url));
		offset += strlen(image_url);

		memcpy(offset, &bootfile3, strlen(bootfile3));
		offset += strlen(bootfile3);
		net_send_tcp_packet((offset - ptr), server_port, our_port,
				    TCP_PUSH, tcp_seq_num, tcp_ack_num);
		current_wget_state = WGET_CONNECTED;
		break;
	case WGET_CONNECTED:
	case WGET_TRANSFERRING:
	case WGET_TRANSFERRED:
		net_send_tcp_packet(0, server_port, our_port, action,
				    tcp_seq_num, tcp_ack_num);
		break;
	}
}

static void wget_send(u8 action, unsigned int tcp_seq_num,
		      unsigned int tcp_ack_num, int len)
{
	retry_action = action;
	retry_tcp_ack_num = tcp_ack_num;
	retry_tcp_seq_num = tcp_seq_num;
	retry_len = len;

	wget_send_stored();
}

void wget_fail(char *error_message, unsigned int tcp_seq_num,
	       unsigned int tcp_ack_num, u8 action)
{
	printf("wget: Transfer Fail - %s\n", error_message);
	net_set_timeout_handler(0, NULL);
	wget_send(action, tcp_seq_num, tcp_ack_num, 0);
}

void wget_success(u8 action, unsigned int tcp_seq_num,
		  unsigned int tcp_ack_num, int len, int packets)
{
	printf("Packets received %d, Transfer Successful\n", packets);
	wget_send(action, tcp_seq_num, tcp_ack_num, len);
}

/*
 * Interfaces of U-BOOT
 */
static void wget_timeout_handler(void)
{
	if (++wget_timeout_count > WGET_RETRY_COUNT) {
		puts("\nRetry count exceeded; starting again\n");
		wget_send(TCP_RST, 0, 0, 0);
		net_start_again();
	} else {
		puts("T ");
		net_set_timeout_handler(wget_timeout +
					WGET_TIMEOUT * wget_timeout_count,
					wget_timeout_handler);
		wget_send_stored();
	}
}

#define PKT_QUEUE_OFFSET 0x20000
#define PKT_QUEUE_PACKET_SIZE 0x800

static void wget_connected(uchar *pkt, unsigned int tcp_seq_num,
			   u8 action, unsigned int tcp_ack_num, unsigned int len)
{
	uchar *pkt_in_q;
	char *pos;
	int hlen, i;
	uchar *ptr1;

	pkt[len] = '\0';
	pos = strstr((char *)pkt, http_eom);

	if (!pos) {
		debug_cond(DEBUG_WGET,
			   "wget: Connected, data before Header %p\n", pkt);
		pkt_in_q = (void *)image_load_addr + PKT_QUEUE_OFFSET +
			(pkt_q_idx * PKT_QUEUE_PACKET_SIZE);

		ptr1 = map_sysmem((phys_addr_t)pkt_in_q, len);
		memcpy(ptr1, pkt, len);
		unmap_sysmem(ptr1);

		pkt_q[pkt_q_idx].pkt = pkt_in_q;
		pkt_q[pkt_q_idx].tcp_seq_num = tcp_seq_num;
		pkt_q[pkt_q_idx].len = len;
		pkt_q_idx++;

		if (pkt_q_idx >= PKTQ_SZ) {
			printf("wget: Fatal error, queue overrun!\n");
			net_set_state(NETLOOP_FAIL);

			return;
		}
	} else {
		debug_cond(DEBUG_WGET, "wget: Connected HTTP Header %p\n", pkt);
		/* sizeof(http_eom) - 1 is the string length of (http_eom) */
		hlen = pos - (char *)pkt + sizeof(http_eom) - 1;
		pos = strstr((char *)pkt, linefeed);
		if (pos > 0)
			i = pos - (char *)pkt;
		else
			i = hlen;
		printf("%.*s", i,  pkt);

		current_wget_state = WGET_TRANSFERRING;

		initial_data_seq_num = tcp_seq_num + hlen;
		next_data_seq_num    = tcp_seq_num + len;

		if (strstr((char *)pkt, http_ok) == 0) {
			debug_cond(DEBUG_WGET,
				   "wget: Connected Bad Xfer\n");
			wget_loop_state = NETLOOP_FAIL;
			wget_send(action, tcp_seq_num, tcp_ack_num, len);
		} else {
			debug_cond(DEBUG_WGET,
				   "wget: Connctd pkt %p  hlen %x\n",
				   pkt, hlen);

			pos = strstr((char *)pkt, content_len);
			if (!pos) {
				content_length = -1;
			} else {
				pos += sizeof(content_len) + 2;
				strict_strtoul(pos, 10, &content_length);
				debug_cond(DEBUG_WGET,
					   "wget: Connected Len %lu\n",
					   content_length);
			}

			net_boot_file_size = 0;

			if (len > hlen) {
				if (store_block(pkt + hlen, 0, len - hlen) != 0) {
					wget_loop_state = NETLOOP_FAIL;
					wget_fail("wget: store error\n", tcp_seq_num, tcp_ack_num, action);
					net_set_state(NETLOOP_FAIL);
					return;
				}
			}

			debug_cond(DEBUG_WGET,
				   "wget: Connected Pkt %p hlen %x\n",
				   pkt, hlen);

			for (i = 0; i < pkt_q_idx; i++) {
				int err;

				ptr1 = map_sysmem(
					(phys_addr_t)(pkt_q[i].pkt),
					pkt_q[i].len);
				err = store_block(ptr1,
					  pkt_q[i].tcp_seq_num -
					  initial_data_seq_num,
					  pkt_q[i].len);
				unmap_sysmem(ptr1);
				debug_cond(DEBUG_WGET,
					   "wget: Connctd pkt Q %p len %x\n",
					   pkt_q[i].pkt, pkt_q[i].len);
				if (err) {
					wget_loop_state = NETLOOP_FAIL;
					wget_fail("wget: store error\n", tcp_seq_num, tcp_ack_num, action);
					net_set_state(NETLOOP_FAIL);
					return;
				}
			}
		}
	}
	wget_send(action, tcp_seq_num, tcp_ack_num, len);
}

/**
 * wget_handler() - TCP handler of wget
 * @pkt: pointer to the application packet
 * @dport: destination TCP port
 * @sip: source IP address
 * @sport: source TCP port
 * @tcp_seq_num: TCP sequential number
 * @tcp_ack_num: TCP acknowledgment number
 * @action: TCP action (SYN, ACK, FIN, etc)
 * @len: packet length
 *
 * In the "application push" invocation, the TCP header with all
 * its information is pointed to by the packet pointer.
 */
static void wget_handler(uchar *pkt, u16 dport,
			 struct in_addr sip, u16 sport,
			 u32 tcp_seq_num, u32 tcp_ack_num,
			 u8 action, unsigned int len)
{
	enum tcp_state wget_tcp_state = tcp_get_tcp_state();

	net_set_timeout_handler(wget_timeout, wget_timeout_handler);
	packets++;

	switch (current_wget_state) {
	case WGET_CLOSED:
		debug_cond(DEBUG_WGET, "wget: Handler: Error!, State wrong\n");
		break;
	case WGET_CONNECTING:
		debug_cond(DEBUG_WGET,
			   "wget: Connecting In len=%x, Seq=%u, Ack=%u\n",
			   len, tcp_seq_num, tcp_ack_num);
		if (!len) {
			if (wget_tcp_state == TCP_ESTABLISHED) {
				debug_cond(DEBUG_WGET,
					   "wget: Cting, send, len=%x\n", len);
				wget_send(action, tcp_seq_num, tcp_ack_num,
					  len);
			} else {
				printf("%.*s", len,  pkt);
				wget_fail("wget: Handler Connected Fail\n",
					  tcp_seq_num, tcp_ack_num, action);
			}
		}
		break;
	case WGET_CONNECTED:
		debug_cond(DEBUG_WGET, "wget: Connected seq=%u, len=%x\n",
			   tcp_seq_num, len);
		if (!len) {
			wget_fail("Image not found, no data returned\n",
				  tcp_seq_num, tcp_ack_num, action);
		} else {
			wget_connected(pkt, tcp_seq_num, action, tcp_ack_num, len);
		}
		break;
	case WGET_TRANSFERRING:
		debug_cond(DEBUG_WGET,
			   "wget: Transferring, seq=%x, ack=%x,len=%x\n",
			   tcp_seq_num, tcp_ack_num, len);

		if (next_data_seq_num != tcp_seq_num) {
			debug_cond(DEBUG_WGET, "wget: seq=%x packet was lost\n", next_data_seq_num);
			return;
		}
		next_data_seq_num = tcp_seq_num + len;

		if (store_block(pkt, tcp_seq_num - initial_data_seq_num, len) != 0) {
			wget_fail("wget: store error\n",
				  tcp_seq_num, tcp_ack_num, action);
			net_set_state(NETLOOP_FAIL);
			return;
		}

		switch (wget_tcp_state) {
		case TCP_FIN_WAIT_2:
			wget_send(TCP_ACK, tcp_seq_num, tcp_ack_num, len);
			fallthrough;
		case TCP_SYN_SENT:
		case TCP_SYN_RECEIVED:
		case TCP_CLOSING:
		case TCP_FIN_WAIT_1:
		case TCP_CLOSED:
			net_set_state(NETLOOP_FAIL);
			break;
		case TCP_ESTABLISHED:
			wget_send(TCP_ACK, tcp_seq_num, tcp_ack_num,
				  len);
			wget_loop_state = NETLOOP_SUCCESS;
			break;
		case TCP_CLOSE_WAIT:     /* End of transfer */
			current_wget_state = WGET_TRANSFERRED;
			wget_send(action | TCP_ACK | TCP_FIN,
				  tcp_seq_num, tcp_ack_num, len);
			break;
		}
		break;
	case WGET_TRANSFERRED:
		printf("Packets received %d, Transfer Successful\n", packets);
		net_set_state(wget_loop_state);
		break;
	}
}

#define RANDOM_PORT_START 1024
#define RANDOM_PORT_RANGE 0x4000

/**
 * random_port() - make port a little random (1024-17407)
 *
 * Return: random port number from 1024 to 17407
 *
 * This keeps the math somewhat trivial to compute, and seems to work with
 * all supported protocols/clients/servers
 */
static unsigned int random_port(void)
{
	return RANDOM_PORT_START + (get_timer(0) % RANDOM_PORT_RANGE);
}

#define BLOCKSIZE 512

void wget_start(void)
{
	image_url = strchr(net_boot_file_name, ':');
	if (image_url > 0) {
		web_server_ip = string_to_ip(net_boot_file_name);
		++image_url;
		net_server_ip = web_server_ip;
	} else {
		web_server_ip = net_server_ip;
		image_url = net_boot_file_name;
	}

	debug_cond(DEBUG_WGET,
		   "wget: Transfer HTTP Server %pI4; our IP %pI4\n",
		   &web_server_ip, &net_ip);

	/* Check if we need to send across this subnet */
	if (net_gateway.s_addr && net_netmask.s_addr) {
		struct in_addr our_net;
		struct in_addr server_net;

		our_net.s_addr = net_ip.s_addr & net_netmask.s_addr;
		server_net.s_addr = net_server_ip.s_addr & net_netmask.s_addr;
		if (our_net.s_addr != server_net.s_addr)
			debug_cond(DEBUG_WGET,
				   "wget: sending through gateway %pI4",
				   &net_gateway);
	}
	debug_cond(DEBUG_WGET, "URL '%s'\n", image_url);

	if (net_boot_file_expected_size_in_blocks) {
		debug_cond(DEBUG_WGET, "wget: Size is 0x%x Bytes = ",
			   net_boot_file_expected_size_in_blocks * BLOCKSIZE);
		print_size(net_boot_file_expected_size_in_blocks * BLOCKSIZE,
			   "");
	}
	debug_cond(DEBUG_WGET,
		   "\nwget:Load address: 0x%lx\nLoading: *\b", image_load_addr);

	if (IS_ENABLED(CONFIG_LMB)) {
		if (wget_init_load_size()) {
			printf("\nwget error: ");
			printf("trying to overwrite reserved memory...\n");
			net_set_state(NETLOOP_FAIL);
			return;
		}
	}

	net_set_timeout_handler(wget_timeout, wget_timeout_handler);
	tcp_set_tcp_handler(wget_handler);

	wget_timeout_count = 0;
	current_wget_state = WGET_CLOSED;

	our_port = random_port();

	/*
	 * Zero out server ether to force arp resolution in case
	 * the server ip for the previous u-boot command, for example dns
	 * is not the same as the web server ip.
	 */

	memset(net_server_ethaddr, 0, 6);

	wget_send(TCP_SYN, 0, 0, 0);
}

#if (IS_ENABLED(CONFIG_CMD_DNS))
int wget_with_dns(ulong dst_addr, char *uri)
{
	int ret;
	char *s, *host_name, *file_name, *str_copy;

	/*
	 * Download file using wget.
	 *
	 * U-Boot wget takes the target uri in this format.
	 *  "<http server ip>:<file path>"  e.g.) 192.168.1.1:/sample/test.iso
	 * Need to resolve the http server ip address before starting wget.
	 */
	str_copy = strdup(uri);
	if (!str_copy)
		return -ENOMEM;

	s = str_copy + strlen("http://");
	host_name = strsep(&s, "/");
	if (!s) {
		log_err("Error: invalied uri, no file path\n");
		ret = -EINVAL;
		goto out;
	}
	file_name = s;

	/* TODO: If the given uri has ip address for the http server, skip dns */
	net_dns_resolve = host_name;
	net_dns_env_var = "httpserverip";
	if (net_loop(DNS) < 0) {
		log_err("Error: dns lookup of %s failed, check setup\n", net_dns_resolve);
		ret = -EINVAL;
		goto out;
	}
	s = env_get("httpserverip");
	if (!s) {
		ret = -EINVAL;
		goto out;
	}

	strlcpy(net_boot_file_name, s, sizeof(net_boot_file_name));
	strlcat(net_boot_file_name, ":/", sizeof(net_boot_file_name)); /* append '/' which is removed by strsep() */
	strlcat(net_boot_file_name, file_name, sizeof(net_boot_file_name));
	image_load_addr = dst_addr;
	ret = net_loop(WGET);

out:
	free(str_copy);

	return ret;
}
#endif

/**
 * wget_validate_uri() - validate the uri for wget
 *
 * @uri:	uri string
 *
 * This function follows the current U-Boot wget implementation.
 * scheme: only "http:" is supported
 * authority:
 *   - user information: not supported
 *   - host: supported
 *   - port: not supported(always use the default port)
 *
 * Uri is expected to be correctly percent encoded.
 * This is the minimum check, control codes(0x1-0x19, 0x7F, except '\0')
 * and space character(0x20) are not allowed.
 *
 * TODO: stricter uri conformance check
 *
 * Return:	true on success, false on failure
 */
bool wget_validate_uri(char *uri)
{
	char c;
	bool ret = true;
	char *str_copy, *s, *authority;

	for (c = 0x1; c < 0x21; c++) {
		if (strchr(uri, c)) {
			log_err("invalid character is used\n");
			return false;
		}
	}
	if (strchr(uri, 0x7f)) {
		log_err("invalid character is used\n");
		return false;
	}

	if (strncmp(uri, "http://", 7)) {
		log_err("only http:// is supported\n");
		return false;
	}
	str_copy = strdup(uri);
	if (!str_copy)
		return false;

	s = str_copy + strlen("http://");
	authority = strsep(&s, "/");
	if (!s) {
		log_err("invalid uri, no file path\n");
		ret = false;
		goto out;
	}
	s = strchr(authority, '@');
	if (s) {
		log_err("user information is not supported\n");
		ret = false;
		goto out;
	}
	s = strchr(authority, ':');
	if (s) {
		log_err("user defined port is not supported\n");
		ret = false;
		goto out;
	}

out:
	free(str_copy);

	return ret;
}
Commit	Line	Data
cfbae482 YCLP	1	// SPDX-License-Identifier: GPL-2.0
	2	/*
	3	* WGET/HTTP support driver based on U-BOOT's nfs.c
	4	* Copyright Duncan Hare <[email protected]> 2017
	5	*/
	6
04592adb	7	#include <asm/global_data.h>
cfbae482 YCLP	8	#include <command.h>
cfbae482 YCLP	9	#include <common.h>
fe1489bc	10	#include <display_options.h>
cfbae482 YCLP	11	#include <env.h>
cfbae482 YCLP	12	#include <image.h>
04592adb	13	#include <lmb.h>
cfbae482 YCLP	14	#include <mapmem.h>
	15	#include <net.h>
	16	#include <net/tcp.h>
	17	#include <net/wget.h>
8cf18da1	18	#include <stdlib.h>
cfbae482	19
04592adb MK	20	DECLARE_GLOBAL_DATA_PTR;
04592adb MK	21
4caacb2f MV	22	/* The default, change with environment variable 'httpdstp' */
	23	#define SERVER_PORT 80
	24
cfbae482 YCLP	25	static const char bootfile1[] = "GET ";
	26	static const char bootfile3[] = " HTTP/1.0\r\n\r\n";
	27	static const char http_eom[] = "\r\n\r\n";
	28	static const char http_ok[] = "200";
	29	static const char content_len[] = "Content-Length";
	30	static const char linefeed[] = "\r\n";
	31	static struct in_addr web_server_ip;
	32	static int our_port;
	33	static int wget_timeout_count;
	34
	35	struct pkt_qd {
	36	uchar *pkt;
	37	unsigned int tcp_seq_num;
	38	unsigned int len;
	39	};
	40
	41	/*
	42	* This is a control structure for out of order packets received.
	43	* The actual packet bufers are in the kernel space, and are
	44	* expected to be overwritten by the downloaded image.
	45	*/
a8bd5ec0 RW	46	#define PKTQ_SZ (PKTBUFSRX / 4)
a8bd5ec0 RW	47	static struct pkt_qd pkt_q[PKTQ_SZ];
cfbae482 YCLP	48	static int pkt_q_idx;
	49	static unsigned long content_length;
	50	static unsigned int packets;
	51
	52	static unsigned int initial_data_seq_num;
cab7867c	53	static unsigned int next_data_seq_num;
cfbae482 YCLP	54
	55	static enum wget_state current_wget_state;
	56
	57	static char *image_url;
	58	static unsigned int wget_timeout = WGET_TIMEOUT;
	59
	60	static enum net_loop_state wget_loop_state;
	61
	62	/* Timeout retry parameters */
	63	static u8 retry_action; /* actions for TCP retry */
	64	static unsigned int retry_tcp_ack_num; /* TCP retry acknowledge number*/
	65	static unsigned int retry_tcp_seq_num; /* TCP retry sequence number */
	66	static int retry_len; /* TCP retry length */
	67
04592adb MK	68	static ulong wget_load_size;
	69
	70	/**
	71	* wget_init_max_size() - initialize maximum load size
	72	*
	73	* Return: 0 if success, -1 if fails
	74	*/
	75	static int wget_init_load_size(void)
	76	{
	77	struct lmb lmb;
	78	phys_size_t max_size;
	79
	80	lmb_init_and_reserve(&lmb, gd->bd, (void *)gd->fdt_blob);
	81
	82	max_size = lmb_get_free_size(&lmb, image_load_addr);
	83	if (!max_size)
	84	return -1;
	85
	86	wget_load_size = max_size;
	87
	88	return 0;
	89	}
	90
cfbae482 YCLP	91	/**
	92	* store_block() - store block in memory
	93	* @src: source of data
	94	* @offset: offset
	95	* @len: length
	96	*/
	97	static inline int store_block(uchar *src, unsigned int offset, unsigned int len)
	98	{
04592adb	99	ulong store_addr = image_load_addr + offset;
cfbae482 YCLP	100	ulong newsize = offset + len;
	101	uchar *ptr;
	102
04592adb MK	103	if (IS_ENABLED(CONFIG_LMB)) {
	104	ulong end_addr = image_load_addr + wget_load_size;
	105
	106	if (!end_addr)
	107	end_addr = ULONG_MAX;
	108
	109	if (store_addr < image_load_addr \|\|
	110	store_addr + len > end_addr) {
	111	printf("\nwget error: ");
	112	printf("trying to overwrite reserved memory...\n");
	113	return -1;
	114	}
	115	}
	116
	117	ptr = map_sysmem(store_addr, len);
cfbae482 YCLP	118	memcpy(ptr, src, len);
	119	unmap_sysmem(ptr);
	120
	121	if (net_boot_file_size < (offset + len))
	122	net_boot_file_size = newsize;
	123
	124	return 0;
	125	}
	126
	127	/**
	128	* wget_send_stored() - wget response dispatcher
	129	*
	130	* WARNING, This, and only this, is the place in wget.c where
	131	* SEQUENCE NUMBERS are swapped between incoming (RX)
	132	* and outgoing (TX).
	133	* Procedure wget_handler() is correct for RX traffic.
	134	*/
	135	static void wget_send_stored(void)
	136	{
	137	u8 action = retry_action;
	138	int len = retry_len;
08fb8da3 DM	139	unsigned int tcp_ack_num = retry_tcp_seq_num + (len == 0 ? 1 : len);
08fb8da3 DM	140	unsigned int tcp_seq_num = retry_tcp_ack_num;
4caacb2f	141	unsigned int server_port;
cfbae482 YCLP	142	uchar ptr, offset;
cfbae482 YCLP	143
4caacb2f MV	144	server_port = env_get_ulong("httpdstp", 10, SERVER_PORT) & 0xffff;
4caacb2f MV	145
cfbae482 YCLP	146	switch (current_wget_state) {
	147	case WGET_CLOSED:
	148	debug_cond(DEBUG_WGET, "wget: send SYN\n");
	149	current_wget_state = WGET_CONNECTING;
4caacb2f	150	net_send_tcp_packet(0, server_port, our_port, action,
cfbae482 YCLP	151	tcp_seq_num, tcp_ack_num);
	152	packets = 0;
	153	break;
	154	case WGET_CONNECTING:
	155	pkt_q_idx = 0;
4caacb2f	156	net_send_tcp_packet(0, server_port, our_port, action,
cfbae482 YCLP	157	tcp_seq_num, tcp_ack_num);
	158
	159	ptr = net_tx_packet + net_eth_hdr_size() +
	160	IP_TCP_HDR_SIZE + TCP_TSOPT_SIZE + 2;
	161	offset = ptr;
	162
	163	memcpy(offset, &bootfile1, strlen(bootfile1));
	164	offset += strlen(bootfile1);
	165
	166	memcpy(offset, image_url, strlen(image_url));
	167	offset += strlen(image_url);
	168
	169	memcpy(offset, &bootfile3, strlen(bootfile3));
	170	offset += strlen(bootfile3);
4caacb2f	171	net_send_tcp_packet((offset - ptr), server_port, our_port,
cfbae482 YCLP	172	TCP_PUSH, tcp_seq_num, tcp_ack_num);
	173	current_wget_state = WGET_CONNECTED;
	174	break;
	175	case WGET_CONNECTED:
	176	case WGET_TRANSFERRING:
	177	case WGET_TRANSFERRED:
4caacb2f	178	net_send_tcp_packet(0, server_port, our_port, action,
cfbae482 YCLP	179	tcp_seq_num, tcp_ack_num);
	180	break;
	181	}
	182	}
	183
08fb8da3 DM	184	static void wget_send(u8 action, unsigned int tcp_seq_num,
08fb8da3 DM	185	unsigned int tcp_ack_num, int len)
cfbae482 YCLP	186	{
	187	retry_action = action;
	188	retry_tcp_ack_num = tcp_ack_num;
	189	retry_tcp_seq_num = tcp_seq_num;
	190	retry_len = len;
	191
	192	wget_send_stored();
	193	}
	194
	195	void wget_fail(char *error_message, unsigned int tcp_seq_num,
	196	unsigned int tcp_ack_num, u8 action)
	197	{
	198	printf("wget: Transfer Fail - %s\n", error_message);
	199	net_set_timeout_handler(0, NULL);
	200	wget_send(action, tcp_seq_num, tcp_ack_num, 0);
	201	}
	202
	203	void wget_success(u8 action, unsigned int tcp_seq_num,
	204	unsigned int tcp_ack_num, int len, int packets)
	205	{
	206	printf("Packets received %d, Transfer Successful\n", packets);
	207	wget_send(action, tcp_seq_num, tcp_ack_num, len);
	208	}
	209
	210	/*
	211	* Interfaces of U-BOOT
	212	*/
	213	static void wget_timeout_handler(void)
	214	{
	215	if (++wget_timeout_count > WGET_RETRY_COUNT) {
	216	puts("\nRetry count exceeded; starting again\n");
	217	wget_send(TCP_RST, 0, 0, 0);
	218	net_start_again();
	219	} else {
	220	puts("T ");
	221	net_set_timeout_handler(wget_timeout +
	222	WGET_TIMEOUT * wget_timeout_count,
	223	wget_timeout_handler);
	224	wget_send_stored();
	225	}
	226	}
	227
	228	#define PKT_QUEUE_OFFSET 0x20000
	229	#define PKT_QUEUE_PACKET_SIZE 0x800
	230
	231	static void wget_connected(uchar *pkt, unsigned int tcp_seq_num,
08fb8da3	232	u8 action, unsigned int tcp_ack_num, unsigned int len)
cfbae482	233	{
cfbae482 YCLP	234	uchar *pkt_in_q;
	235	char *pos;
	236	int hlen, i;
	237	uchar *ptr1;
	238
	239	pkt[len] = '\0';
	240	pos = strstr((char *)pkt, http_eom);
	241
	242	if (!pos) {
	243	debug_cond(DEBUG_WGET,
	244	"wget: Connected, data before Header %p\n", pkt);
	245	pkt_in_q = (void *)image_load_addr + PKT_QUEUE_OFFSET +
	246	(pkt_q_idx * PKT_QUEUE_PACKET_SIZE);
	247
	248	ptr1 = map_sysmem((phys_addr_t)pkt_in_q, len);
	249	memcpy(ptr1, pkt, len);
	250	unmap_sysmem(ptr1);
	251
	252	pkt_q[pkt_q_idx].pkt = pkt_in_q;
	253	pkt_q[pkt_q_idx].tcp_seq_num = tcp_seq_num;
	254	pkt_q[pkt_q_idx].len = len;
	255	pkt_q_idx++;
a8bd5ec0 RW	256
	257	if (pkt_q_idx >= PKTQ_SZ) {
	258	printf("wget: Fatal error, queue overrun!\n");
	259	net_set_state(NETLOOP_FAIL);
	260
	261	return;
	262	}
cfbae482 YCLP	263	} else {
	264	debug_cond(DEBUG_WGET, "wget: Connected HTTP Header %p\n", pkt);
	265	/* sizeof(http_eom) - 1 is the string length of (http_eom) */
	266	hlen = pos - (char *)pkt + sizeof(http_eom) - 1;
	267	pos = strstr((char *)pkt, linefeed);
	268	if (pos > 0)
	269	i = pos - (char *)pkt;
	270	else
	271	i = hlen;
	272	printf("%.*s", i, pkt);
	273
	274	current_wget_state = WGET_TRANSFERRING;
	275
cab7867c YS	276	initial_data_seq_num = tcp_seq_num + hlen;
	277	next_data_seq_num = tcp_seq_num + len;
	278
cfbae482 YCLP	279	if (strstr((char *)pkt, http_ok) == 0) {
	280	debug_cond(DEBUG_WGET,
	281	"wget: Connected Bad Xfer\n");
cfbae482 YCLP	282	wget_loop_state = NETLOOP_FAIL;
	283	wget_send(action, tcp_seq_num, tcp_ack_num, len);
	284	} else {
	285	debug_cond(DEBUG_WGET,
	286	"wget: Connctd pkt %p hlen %x\n",
	287	pkt, hlen);
cfbae482 YCLP	288
	289	pos = strstr((char *)pkt, content_len);
	290	if (!pos) {
	291	content_length = -1;
	292	} else {
	293	pos += sizeof(content_len) + 2;
	294	strict_strtoul(pos, 10, &content_length);
	295	debug_cond(DEBUG_WGET,
	296	"wget: Connected Len %lu\n",
	297	content_length);
	298	}
	299
	300	net_boot_file_size = 0;
	301
04592adb MK	302	if (len > hlen) {
	303	if (store_block(pkt + hlen, 0, len - hlen) != 0) {
	304	wget_loop_state = NETLOOP_FAIL;
	305	wget_fail("wget: store error\n", tcp_seq_num, tcp_ack_num, action);
	306	net_set_state(NETLOOP_FAIL);
	307	return;
	308	}
	309	}
cfbae482 YCLP	310
	311	debug_cond(DEBUG_WGET,
	312	"wget: Connected Pkt %p hlen %x\n",
	313	pkt, hlen);
	314
	315	for (i = 0; i < pkt_q_idx; i++) {
04592adb MK	316	int err;
04592adb MK	317
cfbae482 YCLP	318	ptr1 = map_sysmem(
	319	(phys_addr_t)(pkt_q[i].pkt),
	320	pkt_q[i].len);
04592adb MK	321	err = store_block(ptr1,
	322	pkt_q[i].tcp_seq_num -
	323	initial_data_seq_num,
	324	pkt_q[i].len);
cfbae482 YCLP	325	unmap_sysmem(ptr1);
	326	debug_cond(DEBUG_WGET,
	327	"wget: Connctd pkt Q %p len %x\n",
	328	pkt_q[i].pkt, pkt_q[i].len);
04592adb MK	329	if (err) {
	330	wget_loop_state = NETLOOP_FAIL;
	331	wget_fail("wget: store error\n", tcp_seq_num, tcp_ack_num, action);
	332	net_set_state(NETLOOP_FAIL);
	333	return;
	334	}
cfbae482 YCLP	335	}
	336	}
	337	}
	338	wget_send(action, tcp_seq_num, tcp_ack_num, len);
	339	}
	340
	341	/**
08fb8da3 DM	342	* wget_handler() - TCP handler of wget
	343	* @pkt: pointer to the application packet
	344	* @dport: destination TCP port
	345	* @sip: source IP address
	346	* @sport: source TCP port
	347	* @tcp_seq_num: TCP sequential number
	348	* @tcp_ack_num: TCP acknowledgment number
	349	* @action: TCP action (SYN, ACK, FIN, etc)
	350	* @len: packet length
cfbae482 YCLP	351	*
	352	* In the "application push" invocation, the TCP header with all
	353	* its information is pointed to by the packet pointer.
	354	*/
08fb8da3 DM	355	static void wget_handler(uchar *pkt, u16 dport,
	356	struct in_addr sip, u16 sport,
	357	u32 tcp_seq_num, u32 tcp_ack_num,
	358	u8 action, unsigned int len)
cfbae482 YCLP	359	{
cfbae482 YCLP	360	enum tcp_state wget_tcp_state = tcp_get_tcp_state();
cfbae482 YCLP	361
	362	net_set_timeout_handler(wget_timeout, wget_timeout_handler);
	363	packets++;
	364
	365	switch (current_wget_state) {
	366	case WGET_CLOSED:
	367	debug_cond(DEBUG_WGET, "wget: Handler: Error!, State wrong\n");
	368	break;
	369	case WGET_CONNECTING:
	370	debug_cond(DEBUG_WGET,
08fb8da3	371	"wget: Connecting In len=%x, Seq=%u, Ack=%u\n",
cfbae482 YCLP	372	len, tcp_seq_num, tcp_ack_num);
	373	if (!len) {
	374	if (wget_tcp_state == TCP_ESTABLISHED) {
	375	debug_cond(DEBUG_WGET,
	376	"wget: Cting, send, len=%x\n", len);
	377	wget_send(action, tcp_seq_num, tcp_ack_num,
	378	len);
	379	} else {
	380	printf("%.*s", len, pkt);
	381	wget_fail("wget: Handler Connected Fail\n",
	382	tcp_seq_num, tcp_ack_num, action);
	383	}
	384	}
	385	break;
	386	case WGET_CONNECTED:
08fb8da3	387	debug_cond(DEBUG_WGET, "wget: Connected seq=%u, len=%x\n",
cfbae482 YCLP	388	tcp_seq_num, len);
	389	if (!len) {
	390	wget_fail("Image not found, no data returned\n",
	391	tcp_seq_num, tcp_ack_num, action);
	392	} else {
08fb8da3	393	wget_connected(pkt, tcp_seq_num, action, tcp_ack_num, len);
cfbae482 YCLP	394	}
	395	break;
	396	case WGET_TRANSFERRING:
	397	debug_cond(DEBUG_WGET,
	398	"wget: Transferring, seq=%x, ack=%x,len=%x\n",
	399	tcp_seq_num, tcp_ack_num, len);
	400
cab7867c YS	401	if (next_data_seq_num != tcp_seq_num) {
	402	debug_cond(DEBUG_WGET, "wget: seq=%x packet was lost\n", next_data_seq_num);
	403	return;
	404	}
	405	next_data_seq_num = tcp_seq_num + len;
	406
beac9581	407	if (store_block(pkt, tcp_seq_num - initial_data_seq_num, len) != 0) {
cfbae482 YCLP	408	wget_fail("wget: store error\n",
cfbae482 YCLP	409	tcp_seq_num, tcp_ack_num, action);
04592adb	410	net_set_state(NETLOOP_FAIL);
cfbae482 YCLP	411	return;
	412	}
	413
	414	switch (wget_tcp_state) {
	415	case TCP_FIN_WAIT_2:
	416	wget_send(TCP_ACK, tcp_seq_num, tcp_ack_num, len);
	417	fallthrough;
	418	case TCP_SYN_SENT:
08fb8da3	419	case TCP_SYN_RECEIVED:
cfbae482 YCLP	420	case TCP_CLOSING:
	421	case TCP_FIN_WAIT_1:
	422	case TCP_CLOSED:
	423	net_set_state(NETLOOP_FAIL);
	424	break;
	425	case TCP_ESTABLISHED:
	426	wget_send(TCP_ACK, tcp_seq_num, tcp_ack_num,
	427	len);
	428	wget_loop_state = NETLOOP_SUCCESS;
	429	break;
	430	case TCP_CLOSE_WAIT: /* End of transfer */
	431	current_wget_state = WGET_TRANSFERRED;
	432	wget_send(action \| TCP_ACK \| TCP_FIN,
	433	tcp_seq_num, tcp_ack_num, len);
	434	break;
	435	}
	436	break;
	437	case WGET_TRANSFERRED:
	438	printf("Packets received %d, Transfer Successful\n", packets);
	439	net_set_state(wget_loop_state);
	440	break;
	441	}
	442	}
	443
	444	#define RANDOM_PORT_START 1024
	445	#define RANDOM_PORT_RANGE 0x4000
	446
	447	/**
	448	* random_port() - make port a little random (1024-17407)
	449	*
	450	* Return: random port number from 1024 to 17407
	451	*
	452	* This keeps the math somewhat trivial to compute, and seems to work with
	453	* all supported protocols/clients/servers
	454	*/
	455	static unsigned int random_port(void)
	456	{
	457	return RANDOM_PORT_START + (get_timer(0) % RANDOM_PORT_RANGE);
	458	}
	459
	460	#define BLOCKSIZE 512
	461
	462	void wget_start(void)
	463	{
	464	image_url = strchr(net_boot_file_name, ':');
	465	if (image_url > 0) {
	466	web_server_ip = string_to_ip(net_boot_file_name);
	467	++image_url;
	468	net_server_ip = web_server_ip;
	469	} else {
	470	web_server_ip = net_server_ip;
	471	image_url = net_boot_file_name;
	472	}
	473
	474	debug_cond(DEBUG_WGET,
	475	"wget: Transfer HTTP Server %pI4; our IP %pI4\n",
	476	&web_server_ip, &net_ip);
	477
	478	/* Check if we need to send across this subnet */
	479	if (net_gateway.s_addr && net_netmask.s_addr) {
	480	struct in_addr our_net;
	481	struct in_addr server_net;
	482
	483	our_net.s_addr = net_ip.s_addr & net_netmask.s_addr;
484	server_net.s_addr = net_server_ip.s_addr & net_netmask.s_addr;
485	if (our_net.s_addr != server_net.s_addr)
486	debug_cond(DEBUG_WGET,
487	"wget: sending through gateway %pI4",
488	&net_gateway);
489	}
490	debug_cond(DEBUG_WGET, "URL '%s'\n", image_url);
491
492	if (net_boot_file_expected_size_in_blocks) {
493	debug_cond(DEBUG_WGET, "wget: Size is 0x%x Bytes = ",
494	net_boot_file_expected_size_in_blocks * BLOCKSIZE);
495	print_size(net_boot_file_expected_size_in_blocks * BLOCKSIZE,
496	"");
497	}
498	debug_cond(DEBUG_WGET,
499	"\nwget:Load address: 0x%lx\nLoading: *\b", image_load_addr);
500
04592adb MK	501	if (IS_ENABLED(CONFIG_LMB)) {
	502	if (wget_init_load_size()) {
	503	printf("\nwget error: ");
	504	printf("trying to overwrite reserved memory...\n");
	505	net_set_state(NETLOOP_FAIL);
	506	return;
	507	}
	508	}
	509
cfbae482 YCLP	510	net_set_timeout_handler(wget_timeout, wget_timeout_handler);
	511	tcp_set_tcp_handler(wget_handler);
	512
	513	wget_timeout_count = 0;
	514	current_wget_state = WGET_CLOSED;
	515
	516	our_port = random_port();
	517
	518	/*
	519	* Zero out server ether to force arp resolution in case
	520	* the server ip for the previous u-boot command, for example dns
	521	* is not the same as the web server ip.
	522	*/
	523
	524	memset(net_server_ethaddr, 0, 6);
	525
	526	wget_send(TCP_SYN, 0, 0, 0);
	527	}
8cf18da1 MK	528
	529	#if (IS_ENABLED(CONFIG_CMD_DNS))
	530	int wget_with_dns(ulong dst_addr, char *uri)
	531	{
	532	int ret;
	533	char s, host_name, file_name, str_copy;
	534
	535	/*
	536	* Download file using wget.
	537	*
	538	* U-Boot wget takes the target uri in this format.
	539	* "<http server ip>:<file path>" e.g.) 192.168.1.1:/sample/test.iso
	540	* Need to resolve the http server ip address before starting wget.
	541	*/
	542	str_copy = strdup(uri);
	543	if (!str_copy)
	544	return -ENOMEM;
	545
	546	s = str_copy + strlen("http://");
	547	host_name = strsep(&s, "/");
	548	if (!s) {
	549	log_err("Error: invalied uri, no file path\n");
	550	ret = -EINVAL;
	551	goto out;
	552	}
	553	file_name = s;
	554
	555	/* TODO: If the given uri has ip address for the http server, skip dns */
	556	net_dns_resolve = host_name;
	557	net_dns_env_var = "httpserverip";
	558	if (net_loop(DNS) < 0) {
	559	log_err("Error: dns lookup of %s failed, check setup\n", net_dns_resolve);
	560	ret = -EINVAL;
	561	goto out;
	562	}
	563	s = env_get("httpserverip");
	564	if (!s) {
	565	ret = -EINVAL;
	566	goto out;
	567	}
	568
	569	strlcpy(net_boot_file_name, s, sizeof(net_boot_file_name));
	570	strlcat(net_boot_file_name, ":/", sizeof(net_boot_file_name)); /* append '/' which is removed by strsep() */
	571	strlcat(net_boot_file_name, file_name, sizeof(net_boot_file_name));
	572	image_load_addr = dst_addr;
	573	ret = net_loop(WGET);
	574
	575	out:
	576	free(str_copy);
	577
	578	return ret;
	579	}
	580	#endif
f01c961e MK	581
	582	/**
	583	* wget_validate_uri() - validate the uri for wget
	584	*
	585	* @uri: uri string
	586	*
	587	* This function follows the current U-Boot wget implementation.
	588	* scheme: only "http:" is supported
	589	* authority:
	590	* - user information: not supported
	591	* - host: supported
	592	* - port: not supported(always use the default port)
	593	*
	594	* Uri is expected to be correctly percent encoded.
	595	* This is the minimum check, control codes(0x1-0x19, 0x7F, except '\0')
	596	* and space character(0x20) are not allowed.
	597	*
	598	* TODO: stricter uri conformance check
	599	*
	600	* Return: true on success, false on failure
	601	*/
	602	bool wget_validate_uri(char *uri)
	603	{
	604	char c;
	605	bool ret = true;
	606	char str_copy, s, *authority;
	607
	608	for (c = 0x1; c < 0x21; c++) {
	609	if (strchr(uri, c)) {
	610	log_err("invalid character is used\n");
	611	return false;
	612	}
	613	}
	614	if (strchr(uri, 0x7f)) {
	615	log_err("invalid character is used\n");
	616	return false;
	617	}
	618
	619	if (strncmp(uri, "http://", 7)) {
	620	log_err("only http:// is supported\n");
	621	return false;
	622	}
	623	str_copy = strdup(uri);
	624	if (!str_copy)
	625	return false;
	626
	627	s = str_copy + strlen("http://");
	628	authority = strsep(&s, "/");
	629	if (!s) {
	630	log_err("invalid uri, no file path\n");
	631	ret = false;
	632	goto out;
	633	}
	634	s = strchr(authority, '@');
	635	if (s) {
	636	log_err("user information is not supported\n");
	637	ret = false;
	638	goto out;
	639	}
	640	s = strchr(authority, ':');
	641	if (s) {
	642	log_err("user defined port is not supported\n");
	643	ret = false;
	644	goto out;
645	}
646
647	out:
648	free(str_copy);
649
650	return ret;
651	}