TFTP服务器代码文档格式.docx

1、 - | Opcode | Block # | #define TFTP_ACK 4 /*Acknowledgment （ACK）*/ /*ERROR packet format 2 bytes 2 bytes string 1 byte - | Opcode | ErrorCode | ErrMsg | 0 | #define TFTP_ERROR 5 /*Error （ERROR）*/ #define TFTP_NETASCII 0 #define TFTP_OCTET 1 #define TFTP_WSTAT_FIRSTACKE 0 #define TFTP_WSTAT_NEXTACK

2、1 #define TFTP_WSTAT_LASTACK 2 #define MAX_RETRY 3 #define TFTP_NOTEND_DATALEN 512+2+2 #ifdef _VC #pragma comment（lib,Wsock32.lib） /*typedef void （* CMDFUNC）（char 256,int pcout）;typedef struct _cmdnum char *cmd; int num; int paramcount; CMDFUNC callback;CMDNUM,*PCMDNUM;typedef struct DATA int socknu

3、m; char *filename; unsigned short remoteport; char *remoteaddr;DATA passdata6;ofstream logfile（log.txt,ios:app）;int makeack（unsigned short num,char *buffer,int size）;void showsysinfo（）;void getfile（DATA *needdata）;void putfile（DATA *needdata）;int makeerror（int errcode,char errstring256,char *buffer,

4、int bufsize）;SOCKET sock6=INVALID_SOCKET,INVALID_SOCKET,INVALID_SOCKET,INVALID_SOCKET,INVALID_SOCKET,INVALID_SOCKET;int filemode =TFTP_OCTET;#define MYPORT 69 void main（int argc,char *argv） WSADATA stWSAData; int ret=0; sockaddr_in addr; char recvbuf1024=0; char sendbuf1024=0; sockaddr_in from; sock

5、addr_in newaddr; int fromlen=0; char filename256; int i,len; DWORD ThreadID6; showsysinfo（）; if（WSAStartup（WSA_VERSION,&stWSAData）!=0） printf（cant start socket rn）; exit（0）; if （sock0=socket（AF_INET,SOCK_DGRAM,0）0） perror（socket exit（1）; addr.sin_family=AF_INET; addr.sin_port=htons（MYPORT）; addr.sin

6、_addr.s_addr=INADDR_ANY; memset（&（addr.sin_zero）,0,8）; if （bind（sock0,（struct sockaddr *）&addr,sizeof（struct sockaddr）bind for （;） fromlen=sizeof（from）; recvfrom（sock0,recvbuf,sizeof（recvbuf）,0,（sockaddr *）&from,&fromlen）; if （recvbuf0=0 & recvbuf1 for （i=0;i+） filenamei=recvbufi+2; if （filenamei=0）

7、 break; for （i=1;i=5;i+） if （socki=INVALID_SOCKET） passdatai.socknum=i; if （socki=socket（AF_INET,SOCK_DGRAM,0） newaddr.sin_family=AF_INET; newaddr.sin_port=0; newaddr.sin_addr.s_addr=INADDR_ANY;（newaddr.sin_zero）,0,8）; if （bind（socki,（struct sockaddr *）&newaddr,sizeof（struct sockaddr）8）; bufferpos=（

8、char）num; return pos;int makedata（int num,char *data,int datasize,char *buffer,int bufsize） bufferpos=TFTP_DATA; memcpy（&bufferpos,data,datasize）; pos=pos+datasize;int makeerror（int errcode,char *errstring,char *buffer,int bufsize） int i; bufferpos=TFTP_ERROR; bufferpos=（char）（errcode bufferpos=（cha

9、r）errcode; for（i=0; bufferpos=errstringi; if （errstringi=void getfile（DATA *needdata） int len=0;/ fd_set fdr; int retry=0;/ struct timeval timeout=5,0; int stat=0; int lastdata=0; long flen=0; FILE *file; from.sin_family=AF_INET; addr.sin_port=htons（needdata-remoteport）; addr.sin_addr.s_addr=inet_ad

10、dr（needdata-remoteaddr）;/ printf（received WRQ form %sn,needdata- if（file=fopen（needdata-filename,rb）!=NULL） fclose（file）;file %s already existnfilename）; len=makeerror（6,file already exist0 sendto（sockneeddata-socknum,sendbuf,len,0,（sockaddr *）&addr,sizeof（addr）; closesocket（sockneeddata-socknum）; s

11、ockneeddata-socknum=INVALID_SOCKET; return; if （file=fopen（needdata-w+b）=NULL） t create file rn len=makeack（lastdata,sendbuf,sizeof（sendbuf）; ret=sendto（sockneeddata- / printf（send ack 0# to %sn while（true） /* FD_ZERO（&fdr）; FD_SET（sockneeddata-socknum,& ret=select（sockneeddata-fdr,NULL,NULL,NULL）;

12、if（SOCKET_ERROR=ret） socket error rn else if （0=ret） if （MAX_RETRY=retry） Time outrn retry+; if （FD_ISSET（sockneeddata-fdr） retry=0; fromlen=sizeof（sockaddr）; ret=recvfrom（sockneeddata-socknum,recvbuf,sizeof（recvbuf）,0,（sockaddr *）& if（0=stat） addr.sin_port=from.sin_port; stat=1; if（TFTP_DATA=recvbu

13、f1） lastdata=（recvbuf2+256）%256）*256+（recvbuf3+256）%256;received data block %d# from %sn,lastdata,needdata-send ack %d# to %sn if（ret logfile received filename from remoteaddrrnendl;recvbuf4,1,512,file）; flen=flen+512;%d byte receivedr,flen）; / void putfile（DATA *needdata） char databuf1024=0; fd_set

14、 fdr; struct timeval timeout=5,0; int stat=TFTP_WSTAT_NEXTACK; int lastack=0; int flen=0; int blocknum=0; size_t rlen=0;received RRQ form %snfile %s not foundrn len=makeerror（1,file not found0 rlen=fread（databuf,1,512,file）; if （rlen for（; FD_ZERO（&fdr,NULL,NULL,&timeout）;socket errorrntime outrn /