tftp.c

Boot code for ADM5120 with serial console for Edimax router.

/*****************************************************************************
;
;   (C) Unpublished Work of ADMtek Incorporated.  All Rights Reserved.
;
;       THIS WORK IS AN UNPUBLISHED WORK AND CONTAINS CONFIDENTIAL,
;       PROPRIETARY AND TRADESECRET INFORMATION OF ADMTEK INCORPORATED.
;       ACCESS TO THIS WORK IS RESTRICTED TO (I) ADMTEK EMPLOYEES WHO HAVE A
;       NEED TO KNOW TO PERFORM TASKS WITHIN THE SCOPE OF THEIR ASSIGNMENTS
;       AND (II) ENTITIES OTHER THAN ADMTEK WHO HAVE ENTERED INTO APPROPRIATE
;       LICENSE AGREEMENTS.  NO PART OF THIS WORK MAY BE USED, PRACTICED,
;       PERFORMED, COPIED, DISTRIBUTED, REVISED, MODIFIED, TRANSLATED,
;       ABBRIDGED, CONDENSED, EXPANDED, COLLECTED, COMPILED, LINKED, RECAST,
;       TRANSFORMED OR ADAPTED WITHOUT THE PRIOR WRITTEN CONSENT OF ADMTEK.
;       ANY USE OR EXPLOITATION OF THIS WORK WITHOUT AUTHORIZATION COULD
;       SUBJECT THE PERPERTRATOR TO CRIMINAL AND CIVIL LIABILITY.
;
;------------------------------------------------------------------------------
;
;    Project : Common Plateform
;    Creator : 
;    File    : Tftp.c
;    Abstract: defines the specific items that loader nedded.
;
;Modification History:
;       By              Date     Ver.   Modification Description
;       --------------- -------- -----  --------------------------------------
;
;*****************************************************************************/
#include <tftp.h>
#include <ctype.h>
#include <skbuff.h>
#include <eth.h>
#include <arp.h>
#include <ip.h>
#include <udp.h>
#include <if_5120.h>

static UINT32 remote_ip;		// TFTP server ip
static unsigned short remote_port;	// TFTP server port
static unsigned short remote_block;	// TFTP server data block number
static int tftpc_read_start;

void print_tftpc_menu(void)
{
	buart_print("\n\rTFTP Client Menu");
	buart_print("\n\r===============================");
	buart_print("\n\r [D]: start download");
	buart_print("\n\r [S]: set parameters");
	buart_print("\n\r [X]: exit");
	buart_print("\n\rEnter your option:");
}

void tftp_client_menu(void)
{
	char key;
	UINT32 ticks;

	while (1) {
		print_tftpc_param();
		print_tftpc_menu();
		key = buart_getchar();
		buart_put(key);
		switch (key) {
			case 'D':
			case 'd':
				tftpc_download();
				break;

			case 'S':
			case 's':
				set_tftpc_param();
				break;

			case 'X':
			case 'x':
				return;

			default:
				break;
		}
	}

}

int tftp_send_ack(int block)
{
	struct tftphdr *tftp_ack;
	struct sk_buff skb;

	skb_headerinit(&skb);
	udp_skb_reserve(&skb);
	tftp_ack = (struct tftphdr *) skb_put(&skb, sizeof(struct tftphdr));
	tftp_ack->th_opcode = htons(ACK);
	tftp_ack->th_block = htons(block);

	udp_send(&skb, remote_ip, TFTP, remote_port);

	return 0;
}

int tftp_send_rrq(UINT32 servip, char *filename)
{
	struct sk_buff skb;
	char *temp;
	unsigned int rrqlen;
	short *opCode;
	int fileNameLen, i;

	skb_headerinit(&skb);
	udp_skb_reserve(&skb);
	temp = (char *) skb.data;

	// operating code
	opCode = (short *) temp;
	*opCode = htons(RRQ);
	temp += 2;

	// remote file name 
	fileNameLen = strlen(filename);
	strncpy(temp, filename, fileNameLen);
	temp += fileNameLen;

	// stuff byte
	*temp++ = 0;

	// operating mode "octet"
	strncpy(temp, "octet", 5);
	temp += 5;

	// stuff byte
	*temp = 0;

	// Opcode + filename + one stuff + "octet" + one stuff, the format of rrq.
	rrqlen = 2 + fileNameLen + 1 + strlen("octet") + 1;
	skb.len = rrqlen;

	remote_ip = servip;
	remote_port = TFTP;
	remote_block = 1;

	udp_send(&skb, remote_ip, TFTP, remote_port);
	return 0;
}

int rcv_imgpkt(char *buf, int *buf_len)
{
	struct sk_buff skb;
	int status;

	skb_headerinit(&skb);
	if (tftp_rcv_packet(&skb) == 1) {
		*buf_len = skb.len;
		memcpy(buf, skb.data, *buf_len);
		if (*buf_len < 512)
			status = TFTP_END;
		else
			status = TFTP_CONTINUE;
	} else {
		*buf_len = 0;
		status = TFTP_NO_DATA;
	}
	return status;
}

int tftp_rcv_packet(struct sk_buff *skb)
{
	struct tftphdr *tftp_hdr;
	int data_flag, len;

	data_flag = 0;
	if (udp_rcv_packet(skb) == 1) {
		tftp_hdr = (struct tftphdr *) skb->data;
		if (ntohs(tftp_hdr->th_opcode) == DATA) {
			if (remote_ip != ip_get_source_ip(skb)) {
				return data_flag;
			}

			remote_port = udp_get_source_port(skb);

			if (remote_block == ntohs(tftp_hdr->th_block)) {
				tftpc_read_start = 1;
				skb_pull(skb, sizeof(struct tftphdr));
				tftp_send_ack(remote_block);
				remote_block++;
				data_flag = 1;
			} else if (remote_block > ntohs(tftp_hdr->th_block))
				tftp_send_ack(ntohs(tftp_hdr->th_block));
			else
				tftp_send_ack(remote_block);
		}
	}
	return data_flag;
}

UINT32 tftpc(char *buf, int buf_size)
{
	UINT32 ticks, servip, total_len;
	char servfile[15];
	int transmit_flag, pkt_len = 0, rrqcount = 0;
	char *working = buf;

	if (get_tftp_param(&servip, servfile) != 0) {
		buart_print("\n\rTFTP Server IP or Filename Error.");
		return;
	}

	transmit_flag = TFTP_START;
	total_len = 0;
	if5120turnon();
	tftp_send_rrq(servip, servfile);
	ticks = UpTime();
	while (1) {
		if (buart_get(0) == 0x1b)	//press ESC to stop.
		{
			total_len = 0;
			break;
		}
		if ((!tftpc_read_start) && ((UpTime() - ticks) > 100)) {
			tftp_send_rrq(servip, servfile);
			ticks = UpTime();
			rrqcount++;
			if (rrqcount == 5)
				break;
		}
		transmit_flag = rcv_imgpkt(working, &pkt_len);
		if (tftpc_read_start) {
			if (transmit_flag == TFTP_CONTINUE)	//Get img packet successfully.
			{
				working += pkt_len;
				total_len += pkt_len;
				if ((total_len % (32 * 1024 * 2) == 0))
					buart_put('.');
				ticks = UpTime();
			} else if (transmit_flag == TFTP_NO_DATA)	//Cannot get img packet.
			{
				if ((UpTime() - ticks) > 5 * 100)	//Timeout.
				{
					total_len = 0;
					//buart_print("\n\rTimeOut..");
					break;
				}
			} else if (transmit_flag == TFTP_END)	//transmit successfully.
			{
				working += pkt_len;
				total_len += pkt_len;
				break;
			}
		}
	}
	tftpc_read_start = 0;
	if5120shutdown();
	return total_len;
}

void tftp_init(void)
{
	remote_ip = 0;
	remote_port = TFTP;
	remote_block = 1;
	tftpc_read_start = 0;

	//Init socket buffer
	skb_init();

	//Init Ethernet protocol.
	eth_init();

	//Init ARP protocol
	arp_init();

	//Init IP protocol
	ip_init();

	//Init UDP protocol
	udp_init();

	//Init Data Receive Queue
	if (InitEthPktQueue() != 0) {
		buart_print("\n\rInit ethernet receive queue error");
		panic();
	}
	//Shutdown Switch Until TFTP Download Start.
	if5120shutdown();
}