/************************************************************
 * 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;
/********************************************************************************************************
 * 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;
	
    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;
		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);
		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)
		{ 
			Uart_Printf("TFTP transfer finished, and receive %d bytes.\n", (tftp_ack-1)*MAXTFTP+tftp_len);
		}
		memcpy(buf + data_len, (unsigned char *)tftpd->data, tftp_len);
		data_len += tftp_len;
	}
	desnode.ip = READ_PACKED(tftpd->i.sip);
	desnode.port = tftpd->u.sport;
	ArpFindCache(desnode.ip, desnode.mac);
	txlen = MakeTftpAck(efp, &locnode, &desnode, TFTP_OP_ACK, tftp_ack);
	DEBUGF(TFTP_DEBUG,("Put TFTP ACK\n"));
    PutEthernet(efp, txlen);              //Send packet

}