tftpmain.c

arm9 嵌入式系统开发与实践光盘

#include "../../inc/config.h"
#include "skbuff.h"
#include "eth.h"
#include "arp.h"
#include "ip.h"
#include "udp.h"
#include "utils.h"
#include "../../inc/board.h"

#ifdef	TFTP_DOWNLOAD_SUPPORT

char TftpLoadEnd;
char TftpPutBegin;
char TftpPutMark;

int net_handle(void)
{
	struct sk_buff *skb;
	struct ethhdr *eth_hdr;		
	
	skb = alloc_skb(ETH_FRAME_LEN);

	if (eth_rcv(skb) != -1) {

		eth_hdr = (struct ethhdr *)(skb->data);					
		skb_pull(skb, ETH_HLEN);

/*	{
		int i;
		char *p = (char *)eth_hdr;
		
		for(i=0; i<32; i++)
			printf("%x,", p[i]);
		printf("\n----%x", eth_hdr->h_proto);
				
	}*/
		if (ntohs(eth_hdr->h_proto) == ETH_P_ARP)
			arp_rcv_packet(skb);

		else if(ntohs(eth_hdr->h_proto) == ETH_P_IP)						
		 	ip_rcv_packet(skb);
	 	
	}

	free_skb(skb);

	return 0;
}

//#include "params.h"

#define	LOCAL_IP_ADDR	((192UL<<24)|(168<<16)|(3<<8)|100)

extern unsigned long download_len;
extern unsigned long download_addr;

static int NetLoadFile(U32 addr, U32 give_ip, U32 a3, U32 a4)
{
	unsigned char eth_addr[ETH_ALEN];	
	unsigned char *s;
	
	give_ip = LOCAL_IP_ADDR;
	s = (unsigned char *)&give_ip;
	
	printf("Mini TFTP Server 1.0 (IP : %d.%d.%d.%d PORT: %d)\n", s[3], s[2], s[1], s[0], TFTP);		
	printf("Type tftp -i %d.%d.%d.%d put filename at the host PC\n", s[3], s[2], s[1], s[0]);
	puts("Press ESC key to exit\n");

	eth_init();
	if(eth_lnk_stat())
		return 0;
		
	eth_get_addr(eth_addr);		
//	arp_init();
	ip_init(give_ip);
	udp_init();
		
	arp_add_entry(eth_addr, give_ip);	

	TftpLoadEnd  = 0;
	TftpPutMark  = 0;
	TftpPutBegin = 0;
	
	download_addr = 0x30008000;
	download_len = 0;
	
	while ((getkey()!=ESC_KEY)&&!TftpLoadEnd) {		
		net_handle();
		if(TftpPutBegin) {
			puts("Starting the TFTP download...\n");
			TftpPutBegin = 0;
		}
		if(TftpPutMark) {
			putch('.');
			TftpPutMark = 0;
		}
	}

	if(TftpLoadEnd) {
		printf("\ndownload 0x%x bytes to 0x%08x\n\n", download_len, download_addr);
//		printf("\nPress any key to continue...\n");
//		getch();
		return download_len;
	}

	return 0;
}

static int NetLoadProg(U32 a1, U32 a2, U32 a3, U32 a4)
{
	if(NetLoadFile(0, 0, 0, 0)) {
		void (*fun)(void) = (void (*)(void))download_addr;		
		CacheDisable();
		//CacheFlush();
		(*fun)();
	}
	return 0;
}

static char title[] = "网口下载文件";
static char tip[]   = "实验网络接口,通过TFTP下载文件来进行实验,启动程序后可在PC端执行\"tftp -i 开发板IP地址 put 文件名\"传送文件到板上,或者直接ping开发板的IP地址,传送结束或按ESC键可以退出";

//NetLoadFileItem 在 prog_entry.c 里被引用
TEST_PROGRAM_ITEM NetLoadFileItem = {
				(TEST_PROGRAM)NetLoadFile,	//入口地址
				title,						//显示名称
				tip, 						//帮助或提示信息,可为NULL
				1};							//使用printf,puts,putch等函数时在LCD上也显示输出字符(串)

static char title1[] = "网口下载运行";
static char tip1[]   = "通过网络接口用TFTP下载目标代码并运行,启动程序后可在PC端执行\"tftp -i 开发板IP地址 put 文件名\"传送文件到板上,传送结束自动运行，按ESC键可以退出";

//NetLoadProgItem 在 prog_entry.c 里被引用
TEST_PROGRAM_ITEM NetLoadProgItem = {
				(TEST_PROGRAM)NetLoadProg,	//入口地址
				title1,						//显示名称
				tip1, 						//帮助或提示信息,可为NULL
				0};							//使用printf,puts,putch等函数时在LCD上也显示输出字符(串)

#else
//void tftp_main(unsigned long addr, int dummy) {}
#endif