/************************************************************
 * File name	: Tftp.c									*
 * By 			: hugang, hgx2000@mail.china.com			*
 ************************************************************/

#include <string.h>
#include <ctype.h>
#include "armnet.h"
#include "../inc/44blib.h"


extern NODE locnode;
unsigned char *buf;
extern unsigned int download_addr;
extern unsigned int download_begin;
extern unsigned int download_end;
extern unsigned int download_len;

/********************************************************************************************************
 * Make a TPTP Request									 												*
 ********************************************************************************************************/
short MakeTftpReq(ETHERFRAME *efp, NODE *sp, NODE *dp, unsigned short op, char *fname, char *mode)
{
    TFTP_REQ *tpr;
    unsigned short len;

    tpr = (TFTP_REQ *)(efp->edata);
    tpr->op = swapw(op);
    len = strlen(strcpy((char *)tpr->data, fname)) + 1;
    len += strlen(strcpy((char *)&tpr->data[len], mode)) + 1;
    return(MakeUdp(efp, sp, dp, (unsigned short)(len+2)));
}


/********************************************************************************************************
 * Make a TFPT ACK										 												*
 ********************************************************************************************************/
short MakeTftpAck(ETHERFRAME *efp, NODE *sp, NODE *dp, unsigned short op, unsigned short block)
{
	TFTP_DATA *tpd;
	
    tpd = (TFTP_DATA *)(efp->edata);
    tpd->op = swapw(op);
    tpd->block = swapw(block);
    return(MakeUdp(efp, sp, dp, 4));
}


/********************************************************************************************************
 * TFPT handler										 													*
 ********************************************************************************************************/

void  TftpRcvPacket(ETHERFRAME *efp)
{	
	TFTP_DATA *tftpd;
	NODE  desnode;
	short txlen;
	unsigned short tftp_len, tftp_ack;
	unsigned char filename[10];
	unsigned char transtype[10];
	unsigned char *filedata;
	unsigned int i, j;
	static int data_len;
	static unsigned short next_block;
	
	if(!download_begin) return;
	
    tftpd = (TFTP_DATA *)(efp->edata);
			
    locnode.port = TFTPORT;
	if(swapw(tftpd->op) == TFTP_OP_WRITE) 
	{
		tftp_ack = 0;
		tftp_len = tftpd->u.len - 10;
		filedata = tftpd->data;
		filedata -= 2;
		for(i=0;i<tftp_len;i++)
		{
			if(filedata[i] != 0)
				filename[i] = filedata[i];
			else
			{
				filename[i] = '\0';
				i++;
				break;
			}
		}
			
		for(j=0;i<tftp_len;i++,j++)
		{
			if(filedata[i] != 0)
				transtype[j] = filedata[i];
			else
			{
				transtype[j] = '\0';
				break;
			}
		}
		buf = (unsigned char *)download_addr;
		data_len = 0;
		next_block = 1;
		DEBUGF(TFTP_DEBUG,("TFTP Write command\n"));
		Uart_Printf("Begin to receive filename: %s, transtype: %s.\n", filename, transtype);
	}
	else if(swapw(tftpd->op) == TFTP_OP_DATA)
	{
		tftp_ack = swapw(tftpd->block);
		if(tftp_ack == next_block)
		{
			tftp_len = tftpd->u.len - 12;
			DEBUGF(TFTP_DEBUG,("TFTP Data command block = %d len = %d\n",tftp_ack, tftp_len));
			if(tftp_len < MAXTFTP)
			{	 
				download_len = (tftp_ack-1)*MAXTFTP+tftp_len;
				Uart_Printf("TFTP transfer finished, and receive %x bytes.\n", download_len);
				download_end = 1;
			}
			memcpy(buf + data_len, (unsigned char *)tftpd->data, tftp_len);
			data_len += tftp_len;
			next_block ++;
		}
		else
		{
			tftp_ack = next_block - 1;
		}
	}
	desnode.ip = READ_PACKED(tftpd->i.sip);
	desnode.port = tftpd->u.sport;
	if(ArpFindCache(desnode.ip, desnode.mac))
	{
		DEBUGF(ARP_DEBUG, ("Fail to find ip %x in arp cache\n", desnode.ip));
		DEBUGF(ARP_DEBUG, ("Send Arp Request to ip %x\n",desnode.ip));
		ArpSendReq(efp, &desnode);
	}
	else
	{
		txlen = MakeTftpAck(efp, &locnode, &desnode, TFTP_OP_ACK, tftp_ack);
		DEBUGF(TFTP_DEBUG,("Put TFTP ACK\n"));
    	PutEthernet(efp, txlen);              //Send packet
	}
}