net: Add wget application
This commit adds a simple wget command that can download files
from http server.
The command syntax is
wget ${loadaddr} <path of the file from server>
Signed-off-by: Duncan Hare <DuncanCHare@yahoo.com>
Signed-off-by: Ying-Chun Liu (PaulLiu) <paul.liu@linaro.org>
Reviewed-by: Simon Glass <sjg@chromium.org>
Cc: Christian Gmeiner <christian.gmeiner@gmail.com>
Cc: Joe Hershberger <joe.hershberger@ni.com>
Cc: Michal Simek <michal.simek@xilinx.com>
Cc: Ramon Fried <rfried.dev@gmail.com>
Reviewed-by: Ramon Fried <rfried.dev@gmail.com>
This commit is contained in:
Ying-Chun Liu (PaulLiu)
Tom Rini
parent
a3bf193bf4
commit
cfbae48219
@ -1812,6 +1812,13 @@ config SYS_DISABLE_AUTOLOAD
|
|||||||
is complete. Enable this option to disable this behavior and instead
|
is complete. Enable this option to disable this behavior and instead
|
||||||
require files to be loaded over the network by subsequent commands.
|
require files to be loaded over the network by subsequent commands.
|
||||||
|
|
||||||
|
config CMD_WGET
|
||||||
|
bool "wget"
|
||||||
|
select TCP
|
||||||
|
help
|
||||||
|
wget is a simple command to download kernel, or other files,
|
||||||
|
from a http server over TCP.
|
||||||
|
|
||||||
config CMD_MII
|
config CMD_MII
|
||||||
bool "mii"
|
bool "mii"
|
||||||
imply CMD_MDIO
|
imply CMD_MDIO
|
||||||
|
|||||||
13
cmd/net.c
13
cmd/net.c
@ -125,6 +125,19 @@ U_BOOT_CMD(
|
|||||||
);
|
);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#if defined(CONFIG_CMD_WGET)
|
||||||
|
static int do_wget(struct cmd_tbl *cmdtp, int flag, int argc, char * const argv[])
|
||||||
|
{
|
||||||
|
return netboot_common(WGET, cmdtp, argc, argv);
|
||||||
|
}
|
||||||
|
|
||||||
|
U_BOOT_CMD(
|
||||||
|
wget, 3, 1, do_wget,
|
||||||
|
"boot image via network using HTTP protocol",
|
||||||
|
"[loadAddress] [[hostIPaddr:]path and image name]"
|
||||||
|
);
|
||||||
|
#endif
|
||||||
|
|
||||||
static void netboot_update_env(void)
|
static void netboot_update_env(void)
|
||||||
{
|
{
|
||||||
char tmp[22];
|
char tmp[22];
|
||||||
|
|||||||
@ -561,7 +561,7 @@ extern int net_restart_wrap; /* Tried all network devices */
|
|||||||
|
|
||||||
enum proto_t {
|
enum proto_t {
|
||||||
BOOTP, RARP, ARP, TFTPGET, DHCP, PING, DNS, NFS, CDP, NETCONS, SNTP,
|
BOOTP, RARP, ARP, TFTPGET, DHCP, PING, DNS, NFS, CDP, NETCONS, SNTP,
|
||||||
TFTPSRV, TFTPPUT, LINKLOCAL, FASTBOOT, WOL, UDP, NCSI
|
TFTPSRV, TFTPPUT, LINKLOCAL, FASTBOOT, WOL, UDP, NCSI, WGET
|
||||||
};
|
};
|
||||||
|
|
||||||
extern char net_boot_file_name[1024];/* Boot File name */
|
extern char net_boot_file_name[1024];/* Boot File name */
|
||||||
|
|||||||
22
include/net/wget.h
Normal file
22
include/net/wget.h
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
/* SPDX-License-Identifier: GPL-2.0 */
|
||||||
|
/*
|
||||||
|
* Duncan Hare Copyright 2017
|
||||||
|
*/
|
||||||
|
|
||||||
|
/**
|
||||||
|
* wget_start() - begin wget
|
||||||
|
*/
|
||||||
|
void wget_start(void);
|
||||||
|
|
||||||
|
enum wget_state {
|
||||||
|
WGET_CLOSED,
|
||||||
|
WGET_CONNECTING,
|
||||||
|
WGET_CONNECTED,
|
||||||
|
WGET_TRANSFERRING,
|
||||||
|
WGET_TRANSFERRED
|
||||||
|
};
|
||||||
|
|
||||||
|
#define DEBUG_WGET 0 /* Set to 1 for debug messages */
|
||||||
|
#define SERVER_PORT 80
|
||||||
|
#define WGET_RETRY_COUNT 30
|
||||||
|
#define WGET_TIMEOUT 2000UL
|
||||||
@ -31,6 +31,7 @@ obj-$(CONFIG_UDP_FUNCTION_FASTBOOT) += fastboot.o
|
|||||||
obj-$(CONFIG_CMD_WOL) += wol.o
|
obj-$(CONFIG_CMD_WOL) += wol.o
|
||||||
obj-$(CONFIG_PROT_UDP) += udp.o
|
obj-$(CONFIG_PROT_UDP) += udp.o
|
||||||
obj-$(CONFIG_PROT_TCP) += tcp.o
|
obj-$(CONFIG_PROT_TCP) += tcp.o
|
||||||
|
obj-$(CONFIG_CMD_WGET) += wget.o
|
||||||
|
|
||||||
# Disable this warning as it is triggered by:
|
# Disable this warning as it is triggered by:
|
||||||
# sprintf(buf, index ? "foo%d" : "foo", index)
|
# sprintf(buf, index ? "foo%d" : "foo", index)
|
||||||
|
|||||||
@ -118,6 +118,7 @@
|
|||||||
#include "wol.h"
|
#include "wol.h"
|
||||||
#endif
|
#endif
|
||||||
#include <net/tcp.h>
|
#include <net/tcp.h>
|
||||||
|
#include <net/wget.h>
|
||||||
|
|
||||||
/** BOOTP EXTENTIONS **/
|
/** BOOTP EXTENTIONS **/
|
||||||
|
|
||||||
@ -517,6 +518,11 @@ restart:
|
|||||||
nfs_start();
|
nfs_start();
|
||||||
break;
|
break;
|
||||||
#endif
|
#endif
|
||||||
|
#if defined(CONFIG_CMD_WGET)
|
||||||
|
case WGET:
|
||||||
|
wget_start();
|
||||||
|
break;
|
||||||
|
#endif
|
||||||
#if defined(CONFIG_CMD_CDP)
|
#if defined(CONFIG_CMD_CDP)
|
||||||
case CDP:
|
case CDP:
|
||||||
cdp_start();
|
cdp_start();
|
||||||
|
|||||||
438
net/wget.c
Normal file
438
net/wget.c
Normal file
@ -0,0 +1,438 @@
|
|||||||
|
// SPDX-License-Identifier: GPL-2.0
|
||||||
|
/*
|
||||||
|
* WGET/HTTP support driver based on U-BOOT's nfs.c
|
||||||
|
* Copyright Duncan Hare <dh@synoia.com> 2017
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <command.h>
|
||||||
|
#include <common.h>
|
||||||
|
#include <env.h>
|
||||||
|
#include <image.h>
|
||||||
|
#include <mapmem.h>
|
||||||
|
#include <net.h>
|
||||||
|
#include <net/tcp.h>
|
||||||
|
#include <net/wget.h>
|
||||||
|
|
||||||
|
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.
|
||||||
|
*/
|
||||||
|
static struct pkt_qd pkt_q[PKTBUFSRX / 4];
|
||||||
|
static int pkt_q_idx;
|
||||||
|
static unsigned long content_length;
|
||||||
|
static unsigned int packets;
|
||||||
|
|
||||||
|
static unsigned int initial_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 */
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 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 newsize = offset + len;
|
||||||
|
uchar *ptr;
|
||||||
|
|
||||||
|
ptr = map_sysmem(image_load_addr + offset, 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_ack_num + len;
|
||||||
|
unsigned int tcp_seq_num = retry_tcp_seq_num;
|
||||||
|
uchar *ptr, *offset;
|
||||||
|
|
||||||
|
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_ack_num,
|
||||||
|
unsigned int tcp_seq_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,
|
||||||
|
struct in_addr action_and_state,
|
||||||
|
unsigned int tcp_ack_num, unsigned int len)
|
||||||
|
{
|
||||||
|
u8 action = action_and_state.s_addr;
|
||||||
|
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++;
|
||||||
|
} 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;
|
||||||
|
|
||||||
|
if (strstr((char *)pkt, http_ok) == 0) {
|
||||||
|
debug_cond(DEBUG_WGET,
|
||||||
|
"wget: Connected Bad Xfer\n");
|
||||||
|
initial_data_seq_num = tcp_seq_num + hlen;
|
||||||
|
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);
|
||||||
|
initial_data_seq_num = tcp_seq_num + 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)
|
||||||
|
store_block(pkt + hlen, 0, len - hlen);
|
||||||
|
|
||||||
|
debug_cond(DEBUG_WGET,
|
||||||
|
"wget: Connected Pkt %p hlen %x\n",
|
||||||
|
pkt, hlen);
|
||||||
|
|
||||||
|
for (i = 0; i < pkt_q_idx; i++) {
|
||||||
|
ptr1 = map_sysmem(
|
||||||
|
(phys_addr_t)(pkt_q[i].pkt),
|
||||||
|
pkt_q[i].len);
|
||||||
|
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);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
wget_send(action, tcp_seq_num, tcp_ack_num, len);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* wget_handler() - handler of wget
|
||||||
|
* @pkt: the pointer to the payload
|
||||||
|
* @tcp_seq_num: tcp sequence number
|
||||||
|
* @action_and_state: TCP state
|
||||||
|
* @tcp_ack_num: tcp acknowledge number
|
||||||
|
* @len: length of the payload
|
||||||
|
*
|
||||||
|
* 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, unsigned int tcp_seq_num,
|
||||||
|
struct in_addr action_and_state,
|
||||||
|
unsigned int tcp_ack_num, unsigned int len)
|
||||||
|
{
|
||||||
|
enum tcp_state wget_tcp_state = tcp_get_tcp_state();
|
||||||
|
u8 action = action_and_state.s_addr;
|
||||||
|
|
||||||
|
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=%x, Ack=%x\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=%x, 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_and_state,
|
||||||
|
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 (tcp_seq_num >= initial_data_seq_num &&
|
||||||
|
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);
|
||||||
|
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_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);
|
||||||
|
|
||||||
|
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);
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user