tftp12header.h

#pragma once

#include "windows.h"
#include "stdio.h"
#include "stdlib.h"
#include "tftp12Log.h"

/*协商过程报文最大长度*/
#define TFTP12_CONTROL_PACKET_MAX_SIZE	(512)
#define TFTP12_BUFFER_SIZE(desc)		(((TFTP12Description *)desc)->option.blockSize+1)
#define TFTP12_GET_OPCODE(buf)			(htons(*(INT16*)buf))
#define TFTP12_GET_BLOCKNUM(buf)		htons((*(UINT16*)(buf+2)))
#define TFTP12_GET_ERRORCODE(buf)		TFTP12_GET_BLOCKNUM(buf)

#define TFTP12_BLOCKSIZE_MAX		(65464)
#define TFTP12_BLOCKSIZE_MIN		(512)

#define TFTP12_TIMEOUT_MAX			(256)
#define TFTP12_TIMEOUT_MIN			(1)

#define TFTP12_NUMBER_OF_OPTIONS	(3)

#define TFTP12_IO_MEMEORY_MAX		(10*1024*1024) /*10M，IOBuffer用到的最大的缓冲区大小*/
#define TFTP12_IO_BUFFERSIZE(x)		((TFTP12_IO_MEMEORY_MAX/x)*x)

#define UNUSED(d)					d

#define FREE_Z(m)	{\
						if(m!=NULL)		\
						{				\
							free(m);	\
							m=NULL;		\
						}				\
					}				

#define FCLOSE_Z(f)	{\
						if(f!=NULL)		\
						{				\
							fclose(f);	\
							f=NULL;		\
						}				\
					}		

#define SCLOSE_Z(s)	{\
						if(s!=0)		\
						{				\
							closesocket(s);	\
							s=0;		\
						}				\
					}				



enum TFTP12_ReadOrWrite {
	TFTP12_READ=1,
	TFTP12_WRITE,
};

enum TFTP12_TRANS_MODE
{
	TFTP12_NETASCII = 0,
	TFTP12_OCTET,
};

enum TFTP12_OPCODE
{
	TFTP12_OPCODE_READ_REQUEST = 1,
	TFTP12_OPCODE_WRITE_REQUEST,
	TFTP12_OPCODE_DATA,
	TFTP12_OPCODE_ACK,
	TFTP12_OPCODE_ERROR,
	TFTP12_OPCODE_OACK,
};
enum TFTP12_ERROR_CODE
{
	TFTP12_NOT_DEFINED = 0,
	TFTP12_FILE_NOT_FOUND,
	TFTP12_ACCESS_VIOLATION,
	TFTP12_DISK_FULL_OR_ALLOCATION_EXCEEDED,
	TFTP12_ILLEGAL_TFTP_OPERATION,
	TFTP12_UNKNOWN_TRANSFER_ID,
	TFTP12_FILE_ALREADY_EXISTS,
	TFTP12_NO_SUCH_USER,
};

enum TFTP12_SEND_RECV_STAUTS
{
	TFTP12_OK = 0,
	TFTP12_TIMEOUT,
	TFTP12_SELECT_ERROR,
	TFTP12_SEND_FAILED,
	TFTP12_NOCONNECT,
	TFTP12_RECV_A_ERROR_PKT,
};

typedef struct
{
	INT32 blockSize;
	INT32 timeout;
	INT32 tsize;
}TFTP12Option;


typedef struct
{
	INT16 writeOrRead;
	INT8 *filename;	//目的文件名
	FILE *openFile;	//打开的文件
	INT8 *mode;	//netascii/octet/mail
	TFTP12Option option;
	INT32 localPort;	//本地出端口
	INT32 sock;
	INT32 transmitBytes;	//已经接收/发送的字节数
	struct sockaddr_in peerAddr;		//对端地址
	INT8 *recvBuffer;	//接收缓冲区
	INT32 recvBufferSize;/*接收缓冲区的大小，在给recvBuffer赋值的时候给这个变量赋值*/
	INT8 *dataPktBuffer;//在发送数据报文的时候用的缓冲区指针
	INT8 controlPktBuffer[TFTP12_CONTROL_PACKET_MAX_SIZE];/*控制报文的缓冲区，不超过512字节*/
	INT32 maxRetransmit;
}TFTP12Description;


/******************************创建报文**********************************************/
/*创建一个Request报文，返回报文长度*/
INT32 tftp12CreateREQPkt(TFTP12Description *pktDescriptor);

//还未实现
/*创建一个Data报文，返回报文长度*/
//INT32 tftp12CreateDataPkt(TFTP12Description *pktDescriptor);

/*创建一个ACK报文，返回报文长度*/
INT32 tftp12CreateACKPkt(TFTP12Description *pktDescriptor, INT32 blockNum);

/*创建一个OACK报文，返回报文长度*/
INT32 tftp12CreateOACKPkt(TFTP12Description *pktDescriptor);

/*创建一个Error报文，返回报文长度*/
INT32 tftp12CreateERRPkt(TFTP12Description *pktDescriptor, INT16 errorCode, UINT8 *errorMsg);
/****************************************************************************/

/*****************************解析报文***********************************************/
/*解析Request报文，返回报文长度*/
INT32 tftp12ParseREQPkt(TFTP12Description *pktDescriptor);

//还未实现
/*解析一个Data报文，返回报文长度*/
//INT32 tftp12ParseDataPkt(TFTP12Description *pktDescriptor);

/*解析一个ACK报文，返回报文长度*/
INT16 tftp12ParseACKPkt(TFTP12Description *pktDescriptor);

/*解析一个OACK报文，返回报文长度*/
INT32 tftp12ParseOACKPkt(TFTP12Description *pktDescriptor);

/*解析一个Error报文，返回报文长度*/
INT32 tftp12ParseERRPkt(TFTP12Description *pktDescriptor, INT16 errorCode, UINT8 *errorMsg);
/****************************************************************************/

char *tftp12CreateDataPkt(char *buf, INT16 blockNum);

INT16 tftp12ParseDataPkt(char *buf, INT32 blockSize);

INT32 test();