common/cmd_net.c - uboot/prism - Git at Google

 /*
  * (C) Copyright 2000
  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
  *
  * See file CREDITS for list of people who contributed to this
  * project.
  *
  * This program is free software; you can redistribute it and/or
  * modify it under the terms of the GNU General Public License as
  * published by the Free Software Foundation; either version 2 of
  * the License, or (at your option) any later version.
  *
  * This program is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  * GNU General Public License for more details.
  *
  * You should have received a copy of the GNU General Public License
  * along with this program; if not, write to the Free Software
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
  * MA 02111-1307 USA
  */

 /*
  * Boot support
  */
 #include <common.h>
 #include <command.h>
 #include <net.h>

 extern int do_bootm (cmd_tbl_t *, int, int, char *[]);

 static int netboot_common (proto_t, cmd_tbl_t *, int , char *[]);

 int do_bootp (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
 {
 	return netboot_common (BOOTP, cmdtp, argc, argv);
 }

 U_BOOT_CMD(
 	bootp,	3,	1,	do_bootp,
 	"boot image via network using BOOTP/TFTP protocol",
 	"[loadAddress] [[hostIPaddr:]bootfilename]"
 );

 int do_tftpb (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
 {
 	return netboot_common (TFTP, cmdtp, argc, argv);
 }

 U_BOOT_CMD(
 	tftpboot,	3,	1,	do_tftpb,
 	"boot image via network using TFTP protocol",
 	"[loadAddress] [[hostIPaddr:]bootfilename]"
 );

 int do_rarpb (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
 {
 	return netboot_common (RARP, cmdtp, argc, argv);
 }

 U_BOOT_CMD(
 	rarpboot,	3,	1,	do_rarpb,
 	"boot image via network using RARP/TFTP protocol",
 	"[loadAddress] [[hostIPaddr:]bootfilename]"
 );

 #if defined(CONFIG_CMD_DHCP)
 int do_dhcp (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
 {
 	return netboot_common(DHCP, cmdtp, argc, argv);
 }

 U_BOOT_CMD(
 	dhcp,	3,	1,	do_dhcp,
 	"boot image via network using DHCP/TFTP protocol",
 	"[loadAddress] [[hostIPaddr:]bootfilename]"
 );
 #endif

 #if defined(CONFIG_CMD_NFS)
 int do_nfs (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
 {
 	return netboot_common(NFS, cmdtp, argc, argv);
 }

 U_BOOT_CMD(
 	nfs,	3,	1,	do_nfs,
 	"boot image via network using NFS protocol",
 	"[loadAddress] [[hostIPaddr:]bootfilename]"
 );
 #endif

 static void netboot_update_env (void)
 {
 	char tmp[22];

 	if (NetOurGatewayIP) {
 		ip_to_string (NetOurGatewayIP, tmp);
 		setenv ("gatewayip", tmp);
 	}

 	if (NetOurSubnetMask) {
 		ip_to_string (NetOurSubnetMask, tmp);
 		setenv ("netmask", tmp);
 	}

 	if (NetOurHostName[0])
 		setenv ("hostname", NetOurHostName);

 	if (NetOurRootPath[0])
 		setenv ("rootpath", NetOurRootPath);

 	if (NetOurIP) {
 		ip_to_string (NetOurIP, tmp);
 		setenv ("ipaddr", tmp);
 	}

 	if (NetServerIP) {
 		ip_to_string (NetServerIP, tmp);
 		setenv ("serverip", tmp);
 	}

 	if (NetOurDNSIP) {
 		ip_to_string (NetOurDNSIP, tmp);
 		setenv ("dnsip", tmp);
 	}
 #if defined(CONFIG_BOOTP_DNS2)
 	if (NetOurDNS2IP) {
 		ip_to_string (NetOurDNS2IP, tmp);
 		setenv ("dnsip2", tmp);
 	}
 #endif
 	if (NetOurNISDomain[0])
 		setenv ("domain", NetOurNISDomain);

 #if defined(CONFIG_CMD_SNTP) \
     && defined(CONFIG_BOOTP_TIMEOFFSET)
 	if (NetTimeOffset) {
 		sprintf (tmp, "%d", NetTimeOffset);
 		setenv ("timeoffset", tmp);
 	}
 #endif
 #if defined(CONFIG_CMD_SNTP) \
     && defined(CONFIG_BOOTP_NTPSERVER)
 	if (NetNtpServerIP) {
 		ip_to_string (NetNtpServerIP, tmp);
 		setenv ("ntpserverip", tmp);
 	}
 #endif
 }

 static int
 netboot_common (proto_t proto, cmd_tbl_t *cmdtp, int argc, char *argv[])
 {
 	char *s;
 	char *end;
 	int   rcode = 0;
 	int   size;
 	ulong addr;

 	/* pre-set load_addr */
 	if ((s = getenv("loadaddr")) != NULL) {
 		load_addr = simple_strtoul(s, NULL, 16);
 	}

 	switch (argc) {
 	case 1:
 		break;

 	case 2:	/*
 		 * Only one arg - accept two forms:
 		 * Just load address, or just boot file name. The latter
 		 * form must be written in a format which can not be
 		 * mis-interpreted as a valid number.
 		 */
 		addr = simple_strtoul(argv[1], &end, 16);
 		if (end == (argv[1] + strlen(argv[1])))
 			load_addr = addr;
 		else
 			copy_filename(BootFile, argv[1], sizeof(BootFile));
 		break;

 	case 3:	load_addr = simple_strtoul(argv[1], NULL, 16);
 		copy_filename (BootFile, argv[2], sizeof(BootFile));

 		break;

 	default: cmd_usage(cmdtp);
 		show_boot_progress (-80);
 		return 1;
 	}

 	show_boot_progress (80);
 	if ((size = NetLoop(proto)) < 0) {
 		show_boot_progress (-81);
 		return 1;
 	}

 	show_boot_progress (81);
 	/* NetLoop ok, update environment */
 	netboot_update_env();

 	/* done if no file was loaded (no errors though) */
 	if (size == 0) {
 		show_boot_progress (-82);
 		return 0;
 	}

 	/* flush cache */
 	flush_cache(load_addr, size);

 	/* Loading ok, check if we should attempt an auto-start */
 	if (((s = getenv("autostart")) != NULL) && (strcmp(s,"yes") == 0)) {
 		char *local_args[2];
 		local_args[0] = argv[0];
 		local_args[1] = NULL;

 		printf ("Automatic boot of image at addr 0x%08lX ...\n",
 			load_addr);
 		show_boot_progress (82);
 		rcode = do_bootm (cmdtp, 0, 1, local_args);
 	}

 #ifdef CONFIG_SOURCE
 	if (((s = getenv("autoscript")) != NULL) && (strcmp(s,"yes") == 0)) {
 		printf ("Running \"source\" command at addr 0x%08lX",
 			load_addr);

 		s = getenv ("autoscript_uname");
 		if (s)
 			printf (":%s ...\n", s);
 		else
 			puts (" ...\n");

 		show_boot_progress (83);
 		rcode = source (load_addr, s);
 	}
 #endif
 	if (rcode < 0)
 		show_boot_progress (-83);
 	else
 		show_boot_progress (84);
 	return rcode;
 }

 #if defined(CONFIG_CMD_PING)
 int do_ping (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
 {
 	if (argc < 2)
 		return -1;

 	NetPingIP = string_to_ip(argv[1]);
 	if (NetPingIP == 0) {
 		cmd_usage(cmdtp);
 		return -1;
 	}

 	if (NetLoop(PING) < 0) {
 		printf("ping failed; host %s is not alive\n", argv[1]);
 		return 1;
 	}

 	printf("host %s is alive\n", argv[1]);

 	return 0;
 }

 U_BOOT_CMD(
 	ping,	2,	1,	do_ping,
 	"send ICMP ECHO_REQUEST to network host",
 	"pingAddress"
 );
 #endif

 #if defined(CONFIG_CMD_CDP)

 static void cdp_update_env(void)
 {
 	char tmp[16];

 	if (CDPApplianceVLAN != htons(-1)) {
 		printf("CDP offered appliance VLAN %d\n", ntohs(CDPApplianceVLAN));
 		VLAN_to_string(CDPApplianceVLAN, tmp);
 		setenv("vlan", tmp);
 		NetOurVLAN = CDPApplianceVLAN;
 	}

 	if (CDPNativeVLAN != htons(-1)) {
 		printf("CDP offered native VLAN %d\n", ntohs(CDPNativeVLAN));
 		VLAN_to_string(CDPNativeVLAN, tmp);
 		setenv("nvlan", tmp);
 		NetOurNativeVLAN = CDPNativeVLAN;
 	}

 }

 int do_cdp (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
 {
 	int r;

 	r = NetLoop(CDP);
 	if (r < 0) {
 		printf("cdp failed; perhaps not a CISCO switch?\n");
 		return 1;
 	}

 	cdp_update_env();

 	return 0;
 }

 U_BOOT_CMD(
 	cdp,	1,	1,	do_cdp,
 	"Perform CDP network configuration",
 );
 #endif

 #if defined(CONFIG_CMD_SNTP)
 int do_sntp (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
 {
 	char *toff;

 	if (argc < 2) {
 		NetNtpServerIP = getenv_IPaddr ("ntpserverip");
 		if (NetNtpServerIP == 0) {
 			printf ("ntpserverip not set\n");
 			return (1);
 		}
 	} else {
 		NetNtpServerIP = string_to_ip(argv[1]);
 		if (NetNtpServerIP == 0) {
 			printf ("Bad NTP server IP address\n");
 			return (1);
 		}
 	}

 	toff = getenv ("timeoffset");
 	if (toff == NULL) NetTimeOffset = 0;
 	else NetTimeOffset = simple_strtol (toff, NULL, 10);

 	if (NetLoop(SNTP) < 0) {
 		printf("SNTP failed: host %s not responding\n", argv[1]);
 		return 1;
 	}

 	return 0;
 }

 U_BOOT_CMD(
 	sntp,	2,	1,	do_sntp,
 	"synchronize RTC via network",
 	"[NTP server IP]\n"
 );
 #endif

 #if defined(CONFIG_CMD_DNS)
 int do_dns(cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
 {
 	if (argc == 1) {
 		cmd_usage(cmdtp);
 		return -1;
 	}

 	/*
 	 * We should check for a valid hostname:
 	 * - Each label must be between 1 and 63 characters long
 	 * - the entire hostname has a maximum of 255 characters
 	 * - only the ASCII letters 'a' through 'z' (case-insensitive),
 	 *   the digits '0' through '9', and the hyphen
 	 * - cannot begin or end with a hyphen
 	 * - no other symbols, punctuation characters, or blank spaces are
 	 *   permitted
 	 * but hey - this is a minimalist implmentation, so only check length
 	 * and let the name server deal with things.
 	 */
 	if (strlen(argv[1]) >= 255) {
 		printf("dns error: hostname too long\n");
 		return 1;
 	}

 	NetDNSResolve = argv[1];

 	if (argc == 3)
 		NetDNSenvvar = argv[2];
 	else
 		NetDNSenvvar = NULL;

 	if (NetLoop(DNS) < 0) {
 		printf("dns lookup of %s failed, check setup\n", argv[1]);
 		return 1;
 	}

 	return 0;
 }

 U_BOOT_CMD(
 	dns,	3,	1,	do_dns,
 	"lookup the IP of a hostname",
 	"hostname [envvar]"
 );

 #endif	/* CONFIG_CMD_DNS */
	/*
	* (C) Copyright 2000
	* Wolfgang Denk, DENX Software Engineering, wd@denx.de.
	*
	* See file CREDITS for list of people who contributed to this
	* project.
	*
	* This program is free software; you can redistribute it and/or
	* modify it under the terms of the GNU General Public License as
	* published by the Free Software Foundation; either version 2 of
	* the License, or (at your option) any later version.
	*
	* This program is distributed in the hope that it will be useful,
	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	* GNU General Public License for more details.
	*
	* You should have received a copy of the GNU General Public License
	* along with this program; if not, write to the Free Software
	* Foundation, Inc., 59 Temple Place, Suite 330, Boston,
	* MA 02111-1307 USA
	*/

	/*
	* Boot support
	*/
	#include <common.h>
	#include <command.h>
	#include <net.h>

	extern int do_bootm (cmd_tbl_t , int, int, char []);

	static int netboot_common (proto_t, cmd_tbl_t , int , char []);

	int do_bootp (cmd_tbl_t cmdtp, int flag, int argc, char argv[])
	{
	return netboot_common (BOOTP, cmdtp, argc, argv);
	}

	U_BOOT_CMD(
	bootp, 3, 1, do_bootp,
	"boot image via network using BOOTP/TFTP protocol",
	"[loadAddress] [[hostIPaddr:]bootfilename]"
	);

	int do_tftpb (cmd_tbl_t cmdtp, int flag, int argc, char argv[])
	{
	return netboot_common (TFTP, cmdtp, argc, argv);
	}

	U_BOOT_CMD(
	tftpboot, 3, 1, do_tftpb,
	"boot image via network using TFTP protocol",
	"[loadAddress] [[hostIPaddr:]bootfilename]"
	);

	int do_rarpb (cmd_tbl_t cmdtp, int flag, int argc, char argv[])
	{
	return netboot_common (RARP, cmdtp, argc, argv);
	}

	U_BOOT_CMD(
	rarpboot, 3, 1, do_rarpb,
	"boot image via network using RARP/TFTP protocol",
	"[loadAddress] [[hostIPaddr:]bootfilename]"
	);

	#if defined(CONFIG_CMD_DHCP)
	int do_dhcp (cmd_tbl_t cmdtp, int flag, int argc, char argv[])
	{
	return netboot_common(DHCP, cmdtp, argc, argv);
	}

	U_BOOT_CMD(
	dhcp, 3, 1, do_dhcp,
	"boot image via network using DHCP/TFTP protocol",
	"[loadAddress] [[hostIPaddr:]bootfilename]"
	);
	#endif

	#if defined(CONFIG_CMD_NFS)
	int do_nfs (cmd_tbl_t cmdtp, int flag, int argc, char argv[])
	{
	return netboot_common(NFS, cmdtp, argc, argv);
	}

	U_BOOT_CMD(
	nfs, 3, 1, do_nfs,
	"boot image via network using NFS protocol",
	"[loadAddress] [[hostIPaddr:]bootfilename]"
	);
	#endif

	static void netboot_update_env (void)
	{
	char tmp[22];

	if (NetOurGatewayIP) {
	ip_to_string (NetOurGatewayIP, tmp);
	setenv ("gatewayip", tmp);
	}

	if (NetOurSubnetMask) {
	ip_to_string (NetOurSubnetMask, tmp);
	setenv ("netmask", tmp);
	}

	if (NetOurHostName[0])
	setenv ("hostname", NetOurHostName);

	if (NetOurRootPath[0])
	setenv ("rootpath", NetOurRootPath);

	if (NetOurIP) {
	ip_to_string (NetOurIP, tmp);
	setenv ("ipaddr", tmp);
	}

	if (NetServerIP) {
	ip_to_string (NetServerIP, tmp);
	setenv ("serverip", tmp);
	}

	if (NetOurDNSIP) {
	ip_to_string (NetOurDNSIP, tmp);
	setenv ("dnsip", tmp);
	}
	#if defined(CONFIG_BOOTP_DNS2)
	if (NetOurDNS2IP) {
	ip_to_string (NetOurDNS2IP, tmp);
	setenv ("dnsip2", tmp);
	}
	#endif
	if (NetOurNISDomain[0])
	setenv ("domain", NetOurNISDomain);

	#if defined(CONFIG_CMD_SNTP) \
	&& defined(CONFIG_BOOTP_TIMEOFFSET)
	if (NetTimeOffset) {
	sprintf (tmp, "%d", NetTimeOffset);
	setenv ("timeoffset", tmp);
	}
	#endif
	#if defined(CONFIG_CMD_SNTP) \
	&& defined(CONFIG_BOOTP_NTPSERVER)
	if (NetNtpServerIP) {
	ip_to_string (NetNtpServerIP, tmp);
	setenv ("ntpserverip", tmp);
	}
	#endif
	}

	static int
	netboot_common (proto_t proto, cmd_tbl_t cmdtp, int argc, char argv[])
	{
	char *s;
	char *end;
	int rcode = 0;
	int size;
	ulong addr;

	/* pre-set load_addr */
	if ((s = getenv("loadaddr")) != NULL) {
	load_addr = simple_strtoul(s, NULL, 16);
	}

	switch (argc) {
	case 1:
	break;

	case 2: /*
	* Only one arg - accept two forms:
	* Just load address, or just boot file name. The latter
	* form must be written in a format which can not be
	* mis-interpreted as a valid number.
	*/
	addr = simple_strtoul(argv[1], &end, 16);
	if (end == (argv[1] + strlen(argv[1])))
	load_addr = addr;
	else
	copy_filename(BootFile, argv[1], sizeof(BootFile));
	break;

	case 3: load_addr = simple_strtoul(argv[1], NULL, 16);
	copy_filename (BootFile, argv[2], sizeof(BootFile));

	break;

	default: cmd_usage(cmdtp);
	show_boot_progress (-80);
	return 1;
	}

	show_boot_progress (80);
	if ((size = NetLoop(proto)) < 0) {
	show_boot_progress (-81);
	return 1;
	}

	show_boot_progress (81);
	/* NetLoop ok, update environment */
	netboot_update_env();

	/* done if no file was loaded (no errors though) */
	if (size == 0) {
	show_boot_progress (-82);
	return 0;
	}

	/* flush cache */
	flush_cache(load_addr, size);

	/* Loading ok, check if we should attempt an auto-start */
	if (((s = getenv("autostart")) != NULL) && (strcmp(s,"yes") == 0)) {
	char *local_args[2];
	local_args[0] = argv[0];
	local_args[1] = NULL;

	printf ("Automatic boot of image at addr 0x%08lX ...\n",
	load_addr);
	show_boot_progress (82);
	rcode = do_bootm (cmdtp, 0, 1, local_args);
	}

	#ifdef CONFIG_SOURCE
	if (((s = getenv("autoscript")) != NULL) && (strcmp(s,"yes") == 0)) {
	printf ("Running \"source\" command at addr 0x%08lX",
	load_addr);

	s = getenv ("autoscript_uname");
	if (s)
	printf (":%s ...\n", s);
	else
	puts (" ...\n");

	show_boot_progress (83);
	rcode = source (load_addr, s);
	}
	#endif
	if (rcode < 0)
	show_boot_progress (-83);
	else
	show_boot_progress (84);
	return rcode;
	}

	#if defined(CONFIG_CMD_PING)
	int do_ping (cmd_tbl_t cmdtp, int flag, int argc, char argv[])
	{
	if (argc < 2)
	return -1;

	NetPingIP = string_to_ip(argv[1]);
	if (NetPingIP == 0) {
	cmd_usage(cmdtp);
	return -1;
	}

	if (NetLoop(PING) < 0) {
	printf("ping failed; host %s is not alive\n", argv[1]);
	return 1;
	}

	printf("host %s is alive\n", argv[1]);

	return 0;
	}

	U_BOOT_CMD(
	ping, 2, 1, do_ping,
	"send ICMP ECHO_REQUEST to network host",
	"pingAddress"
	);
	#endif

	#if defined(CONFIG_CMD_CDP)

	static void cdp_update_env(void)
	{
	char tmp[16];

	if (CDPApplianceVLAN != htons(-1)) {
	printf("CDP offered appliance VLAN %d\n", ntohs(CDPApplianceVLAN));
	VLAN_to_string(CDPApplianceVLAN, tmp);
	setenv("vlan", tmp);
	NetOurVLAN = CDPApplianceVLAN;
	}

	if (CDPNativeVLAN != htons(-1)) {
	printf("CDP offered native VLAN %d\n", ntohs(CDPNativeVLAN));
	VLAN_to_string(CDPNativeVLAN, tmp);
	setenv("nvlan", tmp);
	NetOurNativeVLAN = CDPNativeVLAN;
	}

	}

	int do_cdp (cmd_tbl_t cmdtp, int flag, int argc, char argv[])
	{
	int r;

	r = NetLoop(CDP);
	if (r < 0) {
	printf("cdp failed; perhaps not a CISCO switch?\n");
	return 1;
	}

	cdp_update_env();

	return 0;
	}

	U_BOOT_CMD(
	cdp, 1, 1, do_cdp,
	"Perform CDP network configuration",
	);
	#endif

	#if defined(CONFIG_CMD_SNTP)
	int do_sntp (cmd_tbl_t cmdtp, int flag, int argc, char argv[])
	{
	char *toff;

	if (argc < 2) {
	NetNtpServerIP = getenv_IPaddr ("ntpserverip");
	if (NetNtpServerIP == 0) {
	printf ("ntpserverip not set\n");
	return (1);
	}
	} else {
	NetNtpServerIP = string_to_ip(argv[1]);
	if (NetNtpServerIP == 0) {
	printf ("Bad NTP server IP address\n");
	return (1);
	}
	}

	toff = getenv ("timeoffset");
	if (toff == NULL) NetTimeOffset = 0;
	else NetTimeOffset = simple_strtol (toff, NULL, 10);

	if (NetLoop(SNTP) < 0) {
	printf("SNTP failed: host %s not responding\n", argv[1]);
	return 1;
	}

	return 0;
	}

	U_BOOT_CMD(
	sntp, 2, 1, do_sntp,
	"synchronize RTC via network",
	"[NTP server IP]\n"
	);
	#endif

	#if defined(CONFIG_CMD_DNS)
	int do_dns(cmd_tbl_t cmdtp, int flag, int argc, char argv[])
	{
	if (argc == 1) {
	cmd_usage(cmdtp);
	return -1;
	}

	/*
	* We should check for a valid hostname:
	* - Each label must be between 1 and 63 characters long
	* - the entire hostname has a maximum of 255 characters
	* - only the ASCII letters 'a' through 'z' (case-insensitive),
	* the digits '0' through '9', and the hyphen
	* - cannot begin or end with a hyphen
	* - no other symbols, punctuation characters, or blank spaces are
	* permitted
	* but hey - this is a minimalist implmentation, so only check length
	* and let the name server deal with things.
	*/
	if (strlen(argv[1]) >= 255) {
	printf("dns error: hostname too long\n");
	return 1;
	}

	NetDNSResolve = argv[1];

	if (argc == 3)
	NetDNSenvvar = argv[2];
	else
	NetDNSenvvar = NULL;

	if (NetLoop(DNS) < 0) {
	printf("dns lookup of %s failed, check setup\n", argv[1]);
	return 1;
	}

	return 0;
	}

	U_BOOT_CMD(
	dns, 3, 1, do_dns,
	"lookup the IP of a hostname",
	"hostname [envvar]"
	);

	#endif /* CONFIG_CMD_DNS */