tftpmain.c

优龙YLP270开发板 光盘自带的BIOS和实验例程源码 强烈推荐

#include <stdio.h>
#include "skbuff.h"
#include "eth.h"
#include "arp.h"
#include "ip.h"
#include "udp.h"
#include "utils.h"
//#include "../../inc/board.h"
#define	ESC_KEY		0x1b
#define	ENTER_KEY	0x0d
#define	BACK_KEY	0x08
#ifdef	TFTP_DOWNLOAD_SUPPORT
#define	getch	getchar
#define	putch	putchar

extern UINT downloadAddress, downloadFileSize;
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"

void tftp_main(unsigned int addr, unsigned int give_ip)
{
	unsigned char eth_addr[ETH_ALEN];
	unsigned char *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;
		
	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;
	
	while ((getkey()!=ESC_KEY)&&!TftpLoadEnd) {		
		net_handle();
		if(TftpPutBegin) {
			puts("Starting the TFTP download...\n");
			TftpPutBegin = 0;
		}
		if(TftpPutMark) {
			putch('.');
			TftpPutMark = 0;
		}
	}
}

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