www.pudn.com > uc_tftp.rar > tftp.c
/******************************************************************************
* Filename : tftp.c *
* Program : loader. *
* Copyright : Copyright (C) 2001, Young-Su, Ahn. *
* Author : Young-Su, Ahn *
* Description : Use Tftp. *
* Make tftp packet and transmit it. *
* Handling received Tftp packet. *
* Created at : Wed Mar 13 2001. *
* Based on : pptboot-0.5.3, CS8900A device driver. *
* Modified by : *
* Modified at : *
******************************************************************************/
/* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*/
#include "main.h"
#define TFTP_PORT 69 // Well known TFTP port #
#define TFTP_TIMEOUT 5 // Seconds to tftpTimeout for a lost pkt
#define TIMEOUT_COUNT 10 // # of tftpTimeouts before giving up
// TFTP operations.
#define TFTP_RRQ 1
#define TFTP_WRQ 2
#define TFTP_DATA 3
#define TFTP_ACK 4
#define TFTP_ERROR 5
#define TFTP_NONE 0
#define TFTP_STATE_RRQ 1
#define TFTP_STATE_DATA 2
#define TFTP_STATE_TOO_LARGE 3
#define TFTP_STATE_BAD_MAGIC 4
#define TFTP_STATE_SUCCESS 5
#define TFTP_STATE_FAILURE 6
static short tftpState = 0;
static short tftpBlock = 0;
static short tftpLastBlock = 0;
static U32 loadAddr = 0;
short tftpPort = 69;
short tftpHostPort=0;
short tftpClientPort = 7777;
static char requestFile[128];
static U32 tftpTimeout;
// Prototypes.
//bool TftpStart(char *filename);
bool TftpStart(int itype);
void TftpTx(char *txBuf);
/**********************************************************************/
// 역할 : tftp [filename] [addr]의 인자를 tftp를 실행하기 위한 함수와
// 연결.
// 매개 :
// 반환 : 성공하면 true, 실패하면 false.
// 주의 :
bool DoTftp(int itype)
{
// tftp로 받은 file data를 기록할 주소를 생성.
if (itype==0)
{
loadAddr = (U32)KERNEL_DRAM_BASE;
}
else if(itype==1)
{
loadAddr = (U32)RAMDISK_DRAM_BASE;
}
else
{
loadAddr = (U32)LOADER_DRAM_DOWN_BASE;
}
// IP가 없으면 bootp를 실행하여 IP를 할당 받음.
if (clientIP==0){
Serial_Printf("\tNo IP. Bootp start...\n\r");
if (DoBootp()==false)
{
Serial_Printf("\tBootp failed. try again.\n\r");
return false;
}
}
return TftpStart(itype);
} // DoTftp.
// 역할 : tftp를 실행할 환경 출력.
// 매개 :
// 반환 : 성공하면 true, 실패하면 false.
// 주의 :
//bool TftpStart(char *filename)
bool TftpStart(int itype)
{
char *txPktBuf, *rxPktBuf;
char *kfd="zImage";
char *rfd="ramdisk.gz";
char *bfd="kloader.bin";
txPktBuf = (char *)PktBuf;
rxPktBuf = (char *)PktBuf;
Serial_Printf("\ni-type = %d\n",itype);
// tftp 실행 환경 출력.
Serial_Printf("TFTP Start...\n\r");
Serial_Printf("\tHost (server) IP : ");
PrintIPAddr(hostIP);
Serial_Printf("\n\r");
Serial_Printf("\tClient (target) IP : ");
PrintIPAddr(clientIP);
Serial_Printf("\n\r");
if(itype == 0) StrCpy(requestFile, kfd);
else if(itype == 1) StrCpy(requestFile, rfd);
else StrCpy(requestFile, bfd);
Serial_Printf("\tLoading Filename : %s",requestFile);
Serial_Printf("\n\r");
Serial_Printf("\tSave Address : 0x%x",loadAddr);
Serial_Printf("\n\r");
Serial_Printf("\n\r\rLoading start...\n\r");
// tftp 시작 환경.
protocol = PROT_TFTP;
tftpState = TFTP_STATE_RRQ;
tftpHostPort = 0;
// tftp request packet 전송.
TftpTx(txPktBuf);
// tftpTimeout은 이곳에서 정하며, tftp data packet을 받을 때마다 갱신된다.
// 시간이 초과하면 tftp는 실패로 간주하고 중단한다.
tftpTimeout = GetTime() + TFTP_TIMEOUT * HZ;
// tftp가 성공하거나 실패할 때까지 tftp data pakcet을 수신.
while (tftpState!=TFTP_STATE_SUCCESS && tftpState!=TFTP_STATE_FAILURE){
RxPacket(rxPktBuf);
// 시간 초과면 실패하고 중단.
if (GetTime()>tftpTimeout){
Serial_Printf("\n\r\tTftp is failed. Try again.\n\r\n\r");
tftpState=TFTP_STATE_FAILURE;
}
}
// 다음 tftp 실행을 위해 환경을 되돌림.
protocol = NOPROTOCOL;
tftpState = TFTP_NONE;
tftpBlock = 0;
tftpHostPort = 0;
return true;
} // TftpStart.
// 역할 : tftp가 필요로 하는 packet을 만들어 전송..
// 매개 :
// 반환 : 성공하면 true, 실패하면 false.
// 주의 :
void TftpTx(char *txPktBuf)
{
char *pktPtr = (char *)(txPktBuf+ETHER_HDR_SIZE+IP_HDR_SIZE+UDP_HDR_SIZE);
U16 hostPort = 0;
int len = 0;
// Serial_Printf(" ",tftpState);
// MemSet(txPktBuf, 0, MAX_PKT_SIZE);
MemSet(txPktBuf-(ETHER_HDR_SIZE+IP_HDR_SIZE+UDP_HDR_SIZE), 0, MAX_PKT_SIZE);
// make tftp header.
switch (tftpState){
// tftp request packet을 만듬.
case TFTP_STATE_RRQ:
*((U16 *)pktPtr)++ = SWAP16(TFTP_RRQ);
StrCpy((char *)pktPtr, requestFile);
pktPtr += StrLen(requestFile) + 1;
StrCpy((char *)pktPtr, "octet");
pktPtr += 5 + 1;
len = pktPtr - txPktBuf;
hostPort = tftpPort;
break;
// 수신한 data packet에 대한 acknowledge packet을 만듬.
case TFTP_STATE_DATA:
*((U16 *)pktPtr)++ = SWAP16(TFTP_ACK);
*((U16 *)pktPtr)++ = SWAP16(tftpBlock);
len = pktPtr - txPktBuf;
hostPort = tftpHostPort;
break;
// error packet을 만듬.
case TFTP_STATE_TOO_LARGE:
*((U16 *)pktPtr)++ = SWAP16(TFTP_ERROR);
*((U16 *)pktPtr)++ = SWAP16(3);
StrCpy((char *)pktPtr, "File too large");
pktPtr += 14 + 1;
len = pktPtr - txPktBuf;
hostPort = tftpHostPort;
break;
// error packet을 만듬.
case TFTP_STATE_BAD_MAGIC:
*((U16 *)pktPtr)++ = SWAP16(TFTP_ERROR);
*((U16 *)pktPtr)++ = SWAP16(2);
StrCpy((char *)pktPtr, "File has bad magic");
pktPtr += 18 + 1;
len = pktPtr - txPktBuf;
hostPort = tftpHostPort;
break;
default :
return;
}
// Ethernet, IP, UDP Header를 만듬.
SetUdpHeader((char *)(txPktBuf+ETHER_HDR_SIZE+IP_HDR_SIZE), hostPort, tftpClientPort, len);
SetIPHeader((char *)(txPktBuf+ETHER_HDR_SIZE), clientIP, hostIP, UDP_HDR_SIZE+len);
SetEtherHeader(txPktBuf, (char *)hostEther, PROT_IP);
// tftp packet 전송.
TxPacket(txPktBuf, ETHER_HDR_SIZE+IP_HDR_SIZE+UDP_HDR_SIZE+len);
return;
} // TftpTx.
// 역할 : tftp packet을 받아 처리.
// 매개 :
// 반환 : 성공하면 true, 실패하면 false.
// 주의 :
bool TftpRx(char *rxPktBuf, int len){
char *pktPtr = rxPktBuf;
short value;
U32 rlen=0; // tftp로 전송받은 data의 길이(임시 buffer).
// handling received data.
value = SWAP16(*(short *)pktPtr); // packet type.
// Serial_Printf(" ",value);
pktPtr+=2;
len-=2;
switch (value){
case TFTP_RRQ :
case TFTP_WRQ :
case TFTP_ACK :
default:
break;
// tftp data packet.
case TFTP_DATA :
// error. check data index.
if (len < 2)
{
Serial_Printf("\nLength err ");
return false;
}
//Serial_Printf("\nBlockNo - 0x%x",*(U16 *)pktPtr);
tftpBlock = SWAP16(*(U16 *)pktPtr); // tftp data packet에서 block 번호.
pktPtr += 2;
len -= 2;
// tftp receive 진행 표시.
if (((tftpBlock)%128)==0){ // each 64 kbyte.
//ClearLine();
rlen = (long)tftpBlock * 512;
Serial_Printf("\n0x%08lx bytes received.", rlen);
}
// request에 대한 첫번째 data packet일때 필요한 정보 설정.
if (tftpState == TFTP_STATE_RRQ)
{
tftpState = TFTP_STATE_DATA;
tftpLastBlock = 0;
if (tftpBlock != 1){
tftpState = TFTP_STATE_FAILURE;
return false;
}
}
// 이전의 packet을 다시 받았을 때는 무시.
// Serial_Printf(" ", tftpBlock);
if (tftpBlock <= tftpLastBlock) break;
tftpLastBlock = tftpBlock;
// 수신한 tftp packet의 data 부분을 지정한 memory 영역에 기록. block은 512 byte 단위.
MemCpy((char *)(loadAddr+(tftpBlock-1)*512), (char *)pktPtr, len);
// TftpStart()에서 check하는 tftpTimeout 시간을 갱신.
tftpTimeout = GetTime() + TFTP_TIMEOUT * HZ;
// Acknowledge를 보냄. 이것을 받은 후 server는 다음 block의 data를 가진 packet을 전송.
TftpTx(rxPktBuf);
// 수신한 tftp packet이 data의 마지막 block일 때.
if (len < 512){
// We received the whole thing. Try to run it.
//ClearLine();
rlen = ((long)tftpBlock-1) * 512 + len;
Serial_Printf("\n0x%08lx (%ld) bytes received.", rlen, rlen);
Serial_Printf("\nftp done.\n\n");
tftpState = TFTP_STATE_SUCCESS;
}
break;
// error 처리.
case TFTP_ERROR :
// tftp RRQ or tftp ack pakcet has some error.
Serial_Printf("\n\r\rTFTP error: '");
Serial_Printf(pktPtr+2);
Serial_Printf("' (");
OutputDec(SWAP16(*(U16 *)pktPtr));
Serial_Printf(")\n\r");
tftpState = TFTP_STATE_FAILURE;
break;
}
return true;
} // TftpRx.