[国嵌攻略][068][tftp网络协议实现]
IP协议结构
UDP协议结构
TFTP协议结构
TFTP端口
读写请求端口： 69
其他请求端口：1024~65535
主程序
/********************************************************************
*名称:menu
*参数:
* none
*返回:
* none
*功能:菜单命令
*********************************************************************/
void menu(){
int num = 0; //输入选项
//显示菜单
printf("\n");
printf("**********************************************************************\n");
printf("* Bootloader *\n");
printf("*[1]Get MAC Address. *\n");
printf("*[2]Download Kernel. *\n");
printf("*[3] *\n");
printf("**********************************************************************\n");
printf("Please Select: ");
//选择菜单
scanf("%d", &num);
switch(num){
case 1: //获取物理地址
send_arp_req();
isDisplay = 0; //关闭显示菜单，等待获取物理地址
break;
case 2: //下载内核文件
send_tftp_read_req();
isDisplay = 0; //关闭显示菜单，等待下载内核文件
break;
default:
break;
}
}
网络头文件
/********************************************************************
*名称:net.h
*作者:D
*时间:2015.11.28
*功能:网络协议头文件
*********************************************************************/
/********************************************************************
*宏定义
*********************************************************************/
//MAC协议
#define ETH_IP 0x0800 //IP
#define ETH_ARP 0x0806 //ARP
#define ETH_RARP 0x0805 //RARP
//ARP协议
#define ARP_ETH 0x0001 //Ethernet
#define ARP_REQ 0x0001 //Request
#define ARP_ACK 0x0002 //Acknowledge
//IP协议
#define IP_VHL 0x45 //version:4, header length:20byte
#define IP_TOS 0x00 //type of service
#define IP_ID 0x0000 //identifiier
#define IP_OFF 0x4000 //flags and fragment offset
#define IP_TTL 0xFF //time of live
#define IP_TCP 0x06 //TCP
#define IP_UDP 0x11 //UDP
#define IP_SUM 0x0000 //clear checksum
#define IP_LEN 20 //header length
//UDP协议
#define UDP_TFTP_SRC 48915 //源端口
#define UDP_TFTP_DST 69 //目的端口
#define UDP_SUM 0 //假校验和
//TFTP协议
#define TFTP_RRQ 0x0001 //Read Request
#define TFTP_WRQ 0x0002 //Write Request
#define TFTP_DAT 0x0003 //File Data
#define TFTP_ACK 0x0004 //Data Acknowledge
#define TFTP_ERR 0x0005 //Error
#define TFTP_DONWLOAD 0x31000000 //TFTP下载地址
/*----------------------------------------分割线----------------------------------------*/
#define EH_ADR_LEN 6 //物理地址长度
#define IP_ADR_LEN 4 //协议地址长度
#define ETH_PKT_MIN 64 //以太网帧最小长度
#define ETH_PKT_MAX 1518 //以太网帧最大长度
#define ETH_DAT_LEN (ETH_PKT_MAX - sizeof(EHHDR)) //MAC数据部分长度
#define IP_DAT_LEN (ETH_DAT_LEN - sizeof(IPHDR)) //IP数据部分长度
#define UDP_DAT_LEN (IP_DAT_LEN - sizeof(UDPHDR)) //UDP数据部分长度
#define TFTP_REQ_LEN (UDP_DAT_LEN - sizeof(TFTPHDR)) //TFTP请求数据部分长度
#define TFTP_ACK_LEN (TFTP_REQ_LEN - sizeof(unsigned short)) //TFTP响应数据部分长度
#define TFTP_DAT_LEN 512 //TFTP数据响应最大长度
/*----------------------------------------分割线----------------------------------------*/
#define HTONS(n) ( (((n)&0xFF00)>>8) | (((n)&0x00FF)<<8) ) //把unsigned short类型的主机序转换到网络序
#define NTOHS(n) ( (((n)&0xFF00)>>8) | (((n)&0x00FF)<<8) ) //把unsigned short类型的网络序转换到主机序
/********************************************************************
*类型定义
*********************************************************************/
//MAC头部
typedef struct ehhdr
{
unsigned char eh_dst[6]; //destination ethernet addrress
unsigned char eh_src[6]; //source ethernet addresss
unsigned short eh_type; //ethernet packet type
}EHHDR;
//MAC帧
typedef struct ehPacket
{
EHHDR ehhdr; //MAC头部
unsigned char data[ETH_DAT_LEN]; //MAC数据
}EHPACKET;
/*----------------------------------------分割线----------------------------------------*/
//ARP头部
typedef struct arphdr
{
unsigned short arp_hrd; //format of hardware address
unsigned short arp_pro; //format of protocol address
unsigned char arp_hln; //length of hardware address
unsigned char arp_pln; //length of protocol address
unsigned short arp_op; //ARP/RARP operation
unsigned char arp_sha[6]; //sender hardware address
unsigned char arp_spa[4]; //sender protocol address
unsigned char arp_tha[6]; //target hardware address
unsigned char arp_tpa[4]; //target protocol address
}ARPHDR;
//ARP报文包
typedef struct arpPacket
{
EHHDR ehhdr; //MAC头部
ARPHDR arphdr; //ARP头部
}ARPPACKET;
/*----------------------------------------分割线----------------------------------------*/
//IP头部
typedef struct iphdr
{
unsigned char ip_vhl; //version and header length
unsigned char ip_tos; //type of service
unsigned short ip_len; //total length
unsigned short ip_id; //identifiier
unsigned short ip_off; //flags and fragment offset
unsigned char ip_ttl; //time of live
unsigned char ip_pro; //protocol
unsigned short ip_sum; //header checksum
unsigned char ip_src[IP_ADR_LEN]; //source address
unsigned char ip_dst[IP_ADR_LEN]; //destination address
}IPHDR;
//IP报文包
typedef struct ipPacket
{
EHHDR ehhdr; //MAC头部
IPHDR iphdr; //IP头部
unsigned char data[IP_DAT_LEN]; //IP数据
}IPPACKET;
/*----------------------------------------分割线----------------------------------------*/
//UDP头部
typedef struct udphdr
{
unsigned short udp_sport; //source port
unsigned short udp_dport; //destination port
unsigned short udp_len; //length
unsigned short udp_sum; //checksum
}UDPHDR;
//UDP报文包
typedef struct udpPacket
{
EHHDR ehhdr; //MAC头部
IPHDR iphdr; //IP头部
UDPHDR udphdr; //UDP头部
unsigned char data[UDP_DAT_LEN]; //UDP数据
}UDPPACKET;
/*----------------------------------------分割线----------------------------------------*/
//TFTP头部
typedef struct tftphdr
{
unsigned short tftp_op; //opcode
}TFTPHDR;
//TFTP请求包
typedef struct tftpReqPacket
{
EHHDR ehhdr; //MAC头部
IPHDR iphdr; //IP头部
UDPHDR udphdr; //UDP头部
TFTPHDR tftphdr; //TFTP头部
unsigned char data[TFTP_REQ_LEN]; //TFTP数据
}TFTPREQPACKET;
//TFTP响应包
typedef struct tftpAckPacket
{
EHHDR ehhdr; //MAC头部
IPHDR iphdr; //IP头部
UDPHDR udphdr; //UDP头部
TFTPHDR tftphdr; //TFTP头部
unsigned short blocknum; //TFTP块号
unsigned char data[TFTP_ACK_LEN]; //TFTP数据
}TFTPACKPACKET;
网卡中断
/********************************************************************
*名称:dm9000_irq
*参数:
* none
*返回:
* none
*功能:网卡中断服务
*********************************************************************/
void dm9000_irq(){
int state; //接收状态
EHPACKET *eh_packet; //MAC帧
//接收网卡数据
state = rx_dm9000(packet);
//处理网卡数据
if(state){ //如果接收成功，那么处理数据
//转换成MAC帧
eh_packet = (EHPACKET *)packet;
//提供网络接口层服务
switch(NTOHS(eh_packet->ehhdr.eh_type)){ //判断网络层协议
case ETH_IP: //IP协议
rece_ip_pro(eh_packet);
break;
case ETH_ARP: //ARP协议
rece_arp_pro(eh_packet);
break;
default:
break;
}
}
//清除外部中断请求
EINTPEND |= (1<<7); //EINT8:cleard
//清除中断源请求
SRCPND |= (1<<4); //EINT4_7:cleard
//清除中断请求
INTPND |= (1<<4); //EINT4_7:cleard
}
ARP协议
/********************************************************************
*名称:arp.c
*作者:D
*时间:2015.11.26
*功能:ARP协议
*********************************************************************/
/********************************************************************
*头文件
*********************************************************************/
#include "net.h"
/********************************************************************
*全局变量声明
*********************************************************************/
//外部变量，定义在DM9000.c中
extern unsigned char eh_src[EH_ADR_LEN]; //物理源地址
extern unsigned char eh_dst[EH_ADR_LEN]; //物理目的地址
extern unsigned char ip_src[IP_ADR_LEN]; //协议源地址
extern unsigned char ip_dst[IP_ADR_LEN]; //协议目的地址
extern int isDisplay; //是否显示菜单标志
/********************************************************************
*函数原型声明
*********************************************************************/
void send_arp_req();
void rece_arp_pro(ARPPACKET *arp_packet);
void send_arp_ack(ARPPACKET *arp_packet);
void rece_arp_ack(ARPPACKET *arp_packet);
int create_arp_packet(ARPPACKET *arp_packet, unsigned short arp_op);
int get_arp_dst_adr(ARPPACKET *arp_packet);
void get_arp_src_adr(ARPPACKET *arp_packet);
void put_arp_src_adr();
/********************************************************************
*名称:send_arp_req
*参数:
* none
*返回:
* none
*功能:发送ARP请求包
*********************************************************************/
void send_arp_req(){
int length = 0; //MAC帧长度
ARPPACKET arp_req_packet; //ARP请求包
//创建ARP请求包
length = create_arp_packet(&arp_req_packet, ARP_REQ);
//发送ARP请求包
tx_dm9000(&arp_req_packet, length);
}
/********************************************************************
*名称:rece_arp_pro
*参数:
* arp_packet ARP报文包
*返回:
* none
*功能:接收ARP协议
*********************************************************************/
void rece_arp_pro(ARPPACKET *arp_packet){
//提取目的地址
if( !get_arp_dst_adr(arp_packet) ){ //如果IP目的地址不匹配，那么返回
return ;
}
//提供网络层服务
switch(NTOHS(arp_packet->arphdr.arp_op)){ //判断ARP操作码
case ARP_REQ: //ARP请求
send_arp_ack(arp_packet);
break;
case ARP_ACK: //ARP响应
rece_arp_ack(arp_packet);
break;
default:
break;
}
}
/********************************************************************
*名称:send_arp_ack
*参数:
* arp_packet ARP报文包
*返回:
* none
*功能:发送ARP响应包
*********************************************************************/
void send_arp_ack(ARPPACKET *arp_packet){
int length = 0; //MAC帧长度
ARPPACKET arp_ack_packet; //ARP响应包
//提取源地址
get_arp_src_adr(arp_packet);
//创建ARP响应包
length = create_arp_packet(&arp_ack_packet, ARP_ACK);
//发送ARP响应包
tx_dm9000(&arp_ack_packet, length);
}
/********************************************************************
*名称:rece_arp_ack
*参数:
* arp_packet ARP报文包
*返回:
* none
*功能:接收ARP响应包
*********************************************************************/
void rece_arp_ack(ARPPACKET *arp_packet){
//提取源地址
get_arp_src_adr(arp_packet);
//打印源地址
put_arp_src_adr();
//打开显示菜单
isDisplay = 1;
}
/*----------------------------------------分割线----------------------------------------*/
/********************************************************************
*名称:create_arp_packet
*参数:
* arp_packet ARP报文包
* arp_op ARP操作码
*返回:
* length MAC帧长度
*功能:创建ARP报文包
*********************************************************************/
int create_arp_packet(ARPPACKET *arp_packet, unsigned short arp_op){
int length = 0;
//填充ARP头部
length = length + sizeof(ARPHDR); //ARP报文长度
arp_packet->arphdr.arp_hrd = HTONS(ARP_ETH); //硬件类型
arp_packet->arphdr.arp_pro = HTONS(ETH_IP); //协议类型
arp_packet->arphdr.arp_hln = EH_ADR_LEN; //MAC地址长度
arp_packet->arphdr.arp_pln = IP_ADR_LEN; //IP地址长度
arp_packet->arphdr.arp_op = HTONS(arp_op); //操作类型
memcpy(arp_packet->arphdr.arp_sha, eh_src, EH_ADR_LEN); //MAC源地址
memcpy(arp_packet->arphdr.arp_spa, ip_src, IP_ADR_LEN); //IP源地址
memcpy(arp_packet->arphdr.arp_tha, eh_dst, EH_ADR_LEN); //MAC目的地址
memcpy(arp_packet->arphdr.arp_tpa, ip_dst, IP_ADR_LEN); //IP目的地址
//填充MAC头部
length = length + sizeof(EHHDR); //MAC帧长度
memcpy(arp_packet->ehhdr.eh_dst, eh_dst, EH_ADR_LEN); //MAC目的地址
memcpy(arp_packet->ehhdr.eh_src, eh_src, EH_ADR_LEN); //MAC源地址
arp_packet->ehhdr.eh_type = HTONS(ETH_ARP); //MAC帧类型
return length;
}
/********************************************************************
*名称:get_arp_dst_adr
*参数:
* arp_packet ARP报文包
*返回:
* return 1 匹配成功
* 0 匹配失败
*功能:提取目的地址
*********************************************************************/
int get_arp_dst_adr(ARPPACKET *arp_packet){
unsigned char arp_ip_dst[IP_ADR_LEN]; //IP目的地址
//提取IP目的地址
memcpy(arp_ip_dst, arp_packet->arphdr.arp_tpa, IP_ADR_LEN);
//判断IP目的地址
if(memcmp(arp_ip_dst, ip_src, IP_ADR_LEN) != 0){ //如果目的地址不等于该主机IP地址，那么匹配失败
return 0;
}
return 1; //匹配成功
}
/********************************************************************
*名称:get_arp_src_adr
*参数:
* arp_packet ARP报文包
*返回:
* none
*功能:提取源地址
*********************************************************************/
void get_arp_src_adr(ARPPACKET *arp_packet){
memcpy(eh_dst, arp_packet->arphdr.arp_sha, EH_ADR_LEN); //提取MAC源地址，并写到目的地址
memcpy(ip_dst, arp_packet->arphdr.arp_spa, IP_ADR_LEN); //提取IP源地址，并写到目的地址
}
/********************************************************************
*名称:put_arp_src_adr
*参数:
* none
*返回:
* none
*功能:打印源地址
*********************************************************************/
void put_arp_src_adr(){
int i;
printf("\nIP Address : ");
for(i = 0; i < IP_ADR_LEN; i++){
printf("%d.", ip_dst[i]); //打印IP源地址
}
printf("\b \n");
printf("MAC Address : ");
for(i = 0; i < EH_ADR_LEN; i++){
printf("%02X:", eh_dst[i]); //打印MAC源地址
}
printf("\b \n");
}
IP协议
/********************************************************************
*名称:ip.c
*作者:D
*时间:2015.11.28
*功能:IP协议
*********************************************************************/
/********************************************************************
*头文件
*********************************************************************/
#include "net.h"
/********************************************************************
*全局变量声明
*********************************************************************/
//外部变量，定义在DM9000.c中
extern unsigned char ip_src[IP_ADR_LEN]; //IP源地址
/********************************************************************
*函数原型声明
*********************************************************************/
void rece_ip_pro(IPPACKET *ip_packet);
int get_ip_dst_adr(IPPACKET *ip_packet);
/********************************************************************
*名称:rece_ip_pro
*参数:
* ip_packet IP报文包
*返回:
* none
*功能:接收IP协议
*********************************************************************/
void rece_ip_pro(IPPACKET *ip_packet){
//提取目的地址
if( !get_ip_dst_adr(ip_packet) ){ //如果IP目的地址不匹配，那么返回
return ;
}
//提供网络层服务
switch(ip_packet->iphdr.ip_pro){ //判断运输层协议
case IP_UDP: //UDP协议
rece_udp_pro(ip_packet);
break;
default:
break;
}
}
/********************************************************************
*名称:get_ip_dst_adr
*参数:
* ip_packet IP报文包
*返回:
* return 1 匹配成功
* 0 匹配失败
*功能:提取目的地址
*********************************************************************/
int get_ip_dst_adr(IPPACKET *ip_packet){
unsigned char ip_dst_adr[IP_ADR_LEN]; //IP目的地址
//提取IP目的地址
memcpy(ip_dst_adr, ip_packet->iphdr.ip_dst, IP_ADR_LEN);
//判断IP目的地址
if(memcmp(ip_dst_adr, ip_src, IP_ADR_LEN) != 0){ //如果目的地址不等于该主机IP地址，那么匹配失败
return 0;
}
return 1; //匹配成功
}
UDP协议
/********************************************************************
*名称:udp.c
*作者:D
*时间:2015.11.30
*功能:UDP协议
*********************************************************************/
/********************************************************************
*头文件
*********************************************************************/
#include "net.h"
/********************************************************************
*函数原型声明
*********************************************************************/
void rece_udp_pro(UDPPACKET *udp_packet);
/********************************************************************
*名称:rece_udp_pro
*参数:
* udp_packet UDP报文包
*返回:
* none
*功能:接收UDP协议
*********************************************************************/
void rece_udp_pro(UDPPACKET *udp_packet){
//提供运输层服务
switch(NTOHS(udp_packet->udphdr.udp_dport)){ //判断应用层端口
case UDP_TFTP_SRC: //TFTP协议源端口
rece_tftp_pro(udp_packet);
break;
default:
break;
}
}
TFTP协议
/********************************************************************
*名称:tftp.c
*作者:D
*时间:2015.11.29
*功能:TFTP协议
*********************************************************************/
/********************************************************************
*头文件
*********************************************************************/
#include "net.h"
/********************************************************************
*全局变量声明
*********************************************************************/
//外部变量，定义在DM9000.c中
extern unsigned char eh_src[EH_ADR_LEN]; //物理源地址
extern unsigned char eh_dst[EH_ADR_LEN]; //物理目的地址
extern unsigned char ip_src[IP_ADR_LEN]; //协议源地址
extern unsigned char ip_dst[IP_ADR_LEN]; //协议目的地址
extern int isDisplay; //是否显示菜单标志
unsigned short reqblknum; //TFTP请求块号
unsigned char *tftp_download; //TFTP下载地址
/********************************************************************
*函数原型声明
*********************************************************************/
void send_tftp_read_req();
void rece_tftp_pro(TFTPACKPACKET *tftp_packet);
void rece_tftp_data_ack(TFTPACKPACKET *tftp_packet);
int get_tftp_data_ack(TFTPACKPACKET *tftp_packet, unsigned short *blocknum, unsigned short *udpdport);
int create_tftp_req_packet(TFTPREQPACKET *tftp_packet, unsigned short tftp_op, const char *filename, const char *mode);
int create_tftp_ack_packet(TFTPACKPACKET *tftp_packet, unsigned short tftp_op, unsigned short blocknum, unsigned short udpdport);
unsigned short checksum(unsigned short* iphdr, int size);
/********************************************************************
*名称:send_tftp_read_req
*参数:
* none
*返回:
* none
*功能:发送TFTP读请求包
*********************************************************************/
void send_tftp_read_req(){
int length = 0; //MAC帧长度
TFTPREQPACKET tftp_req_packet; //TFTP请求包
//设置TFTP请求块号
reqblknum = 1;
//设置TFTP下载地址
tftp_download = (unsigned char *)TFTP_DONWLOAD;
//创建TFTP读请求包
length = create_tftp_req_packet(&tftp_req_packet, TFTP_RRQ, "boot.bin", "octet");
//发送TFTP读请求包
tx_dm9000(&tftp_req_packet, length);
}
/********************************************************************
*名称:rece_tftp_pro
*参数:
* rece_tftp_pro TFTP响应包
*返回:
* none
*功能:接收TFTP协议
*********************************************************************/
void rece_tftp_pro(TFTPACKPACKET *tftp_packet){
//提供应用层服务
switch(NTOHS(tftp_packet->tftphdr.tftp_op)){ //判断TFTP操作码
case TFTP_DAT: //TFTP文件数据
rece_tftp_data_ack(tftp_packet);
break;
default:
break;
}
}
/********************************************************************
*名称:rece_tftp_data_ack
*参数:
* tftp_packet TFTP响应包
*返回:
* none
*功能:接收TFTP数据响应包
*********************************************************************/
void rece_tftp_data_ack(TFTPACKPACKET *tftp_packet){
int length = 0; //MAC帧长度
unsigned short blocknum = 0; //TFTP块号
unsigned short udpdport = 0; //UDP目的端口
TFTPACKPACKET tftp_ack_packet; //TFTP响应包
//提取数据响应包
if( !get_tftp_data_ack(tftp_packet, &blocknum, &udpdport) ){ //如果提取失败，那么返回
return ;
}
//创建TFTP响应包
length = create_tftp_ack_packet(&tftp_ack_packet, TFTP_ACK, blocknum, udpdport);
//发送TFTP响应包
tx_dm9000(&tftp_ack_packet, length);
}
/*----------------------------------------分割线----------------------------------------*/
/********************************************************************
*名称:get_tftp_data_ack
*参数:
* tftp_packet TFTP响应包
* blocknum TFTP块号
* udpdport UDP目的端口
*返回:
* return 1 提取成功
* 0 提取失败
*功能:提取TFTP数据响应包
*********************************************************************/
int get_tftp_data_ack(TFTPACKPACKET *tftp_packet, unsigned short *blocknum, unsigned short *udpdport){
int i;
int length = 0; //TFTP数据长度
//提取TFTP响应块号
*blocknum = NTOHS(tftp_packet->blocknum);
if(*blocknum != reqblknum){ //如果TFTP块号不等于请求块号，那么返回
return 0;
}
//增加TFTP请求块号
reqblknum++;
//提取UDP目的端口
*udpdport = NTOHS(tftp_packet->udphdr.udp_sport);
//计算TFTP数据长度
length = NTOHS(tftp_packet->udphdr.udp_len); //提取UDP报文长度
length = length - sizeof(UDPHDR); //计算TFTP报文长度
length = length - sizeof(TFTPHDR); //计算TFTP数据长度
length = length - sizeof(unsigned short);
//提取TFTP响应数据
for(i = 0; i < length; i++){
*(tftp_download++) = tftp_packet->data[i]; //下载到指定内存地址
}
//判断最后数据响应
if(length < TFTP_DAT_LEN){ //如果数据长度小于最大长度，那么发出最后响应包
printf("\nTFTP Donwload Success!\n");
isDisplay = 1; //打开显示菜单
}
return 1;
}
/********************************************************************
*名称:create_tftp_req_packet
*参数:
* tftp_packet TFTP请求包
* tftp_op TFTP操作码
* filename 请求文件名
* mode 请求模式
*返回:
* length MAC帧长度
*功能:创建TFTP请求包
*********************************************************************/
int create_tftp_req_packet(TFTPREQPACKET *tftp_packet, unsigned short tftp_op, const char *filename, const char *mode){
int i;
int length = 0;
unsigned char data[TFTP_REQ_LEN]; //TFTP数据
unsigned short *iphdr; //IP起始地址
unsigned short chksum; //IP头校验和
//填充TFTP数据
for(i = 0; i < strlen(filename); i++){ //填充请求文件名
data[length++] = filename[i];
}
data[length++] = '\0'; //结束标志
for(i = 0; i < strlen(mode); i++){ //填充请求模式
data[length++] = mode[i];
}
data[length++] = '\0'; //结束标志
memcpy(tftp_packet->data, data, length); //填充TFTP数据
//填充TFTP头部
length = length + sizeof(TFTPHDR); //TFTP报文长度
tftp_packet->tftphdr.tftp_op = HTONS(tftp_op); //填充操作码
//填充UDP头部
length =length + sizeof(UDPHDR); //UDP报文长度
tftp_packet->udphdr.udp_sport = HTONS(UDP_TFTP_SRC); //UDP源端口
tftp_packet->udphdr.udp_dport = HTONS(UDP_TFTP_DST); //UDP目的端口
tftp_packet->udphdr.udp_len = HTONS(length); //UDP报文长度
tftp_packet->udphdr.udp_sum = HTONS(UDP_SUM); //UDP头假校验和
//填充IP头部
length = length + sizeof(IPHDR); //IP报文长度
tftp_packet->iphdr.ip_vhl = IP_VHL; //版本号：IPv4，协议头长度：20字节
tftp_packet->iphdr.ip_tos = IP_TOS; //普通服务类型
tftp_packet->iphdr.ip_len = HTONS(length); //IP报文长度
tftp_packet->iphdr.ip_id = HTONS(IP_ID); //标识符
tftp_packet->iphdr.ip_off = HTONS(IP_OFF); //标记
tftp_packet->iphdr.ip_ttl = IP_TTL; //生存时间
tftp_packet->iphdr.ip_pro = IP_UDP; //UDP协议
memcpy(tftp_packet->iphdr.ip_src, ip_src, IP_ADR_LEN); //IP源地址
memcpy(tftp_packet->iphdr.ip_dst, ip_dst, IP_ADR_LEN); //IP目的地址
tftp_packet->iphdr.ip_sum = HTONS(IP_SUM); //清零校验和
iphdr = (unsigned short *)&(tftp_packet->iphdr); //获取IP指针
chksum = checksum(iphdr, IP_LEN); //计算校验和
tftp_packet->iphdr.ip_sum = chksum; //头部校验和，注意不需要网络序转换，计算的本来就是网络序的校验和
//填充MAC头部
length = length + sizeof(EHHDR); //MAC帧长度
memcpy(tftp_packet->ehhdr.eh_dst, eh_dst, EH_ADR_LEN); //MAC目的地址
memcpy(tftp_packet->ehhdr.eh_src, eh_src, EH_ADR_LEN); //MAC源地址
tftp_packet->ehhdr.eh_type = HTONS(ETH_IP); //MAC帧类型
return length;
}
/********************************************************************
*名称:create_tftp_ack_packet
*参数:
* tftp_packet TFTP响应包
* tftp_op TFTP操作码
* blocknum TFTP块号
* udpdport UDP目的端口
*返回:
* length MAC帧长度
*功能:创建TFTP响应包
*********************************************************************/
int create_tftp_ack_packet(TFTPACKPACKET *tftp_packet, unsigned short tftp_op, unsigned short blocknum, unsigned short udpdport){
int i;
int length = 0; //数据长度
unsigned short *iphdr; //IP起始地址
unsigned short chksum; //IP头校验和
//填充TFTP块号
length = length + sizeof(unsigned short); //TFTP数据长度
tftp_packet->blocknum = HTONS(blocknum); //填充TFTP块号
//填充TFTP头部
length = length + sizeof(TFTPHDR); //TFTP报文长度
tftp_packet->tftphdr.tftp_op = HTONS(tftp_op); //填充操作码
//填充UDP头部
length =length + sizeof(UDPHDR); //UDP报文长度
tftp_packet->udphdr.udp_sport = HTONS(UDP_TFTP_SRC); //UDP源端口
tftp_packet->udphdr.udp_dport = HTONS(udpdport); //UDP目的端口
tftp_packet->udphdr.udp_len = HTONS(length); //UDP数据包长度
tftp_packet->udphdr.udp_sum = HTONS(UDP_SUM); //UDP头假校验和
//填充IP头部
length = length + sizeof(IPHDR); //IP报文长度
tftp_packet->iphdr.ip_vhl = IP_VHL; //版本号：IPv4，协议头长度：20字节
tftp_packet->iphdr.ip_tos = IP_TOS; //普通服务类型
tftp_packet->iphdr.ip_len = HTONS(length); //IP报文长度
tftp_packet->iphdr.ip_id = HTONS(IP_ID); //标识符
tftp_packet->iphdr.ip_off = HTONS(IP_OFF); //标记
tftp_packet->iphdr.ip_ttl = IP_TTL; //生存时间
tftp_packet->iphdr.ip_pro = IP_UDP; //UDP协议
memcpy(tftp_packet->iphdr.ip_src, ip_src, IP_ADR_LEN); //IP源地址
memcpy(tftp_packet->iphdr.ip_dst, ip_dst, IP_ADR_LEN); //IP目的地址
tftp_packet->iphdr.ip_sum = HTONS(IP_SUM); //清零校验和
iphdr = (unsigned short *)&(tftp_packet->iphdr); //获取IP地址
chksum = checksum(iphdr, IP_LEN); //计算校验和
tftp_packet->iphdr.ip_sum = chksum; //头部校验和，注意不需要网络序转换，计算的本来就是网络序的校验和
//填充以太网帧头部
length = length + sizeof(EHHDR); //MAC帧长度
memcpy(tftp_packet->ehhdr.eh_dst, eh_dst, EH_ADR_LEN); //MAC目的地址
memcpy(tftp_packet->ehhdr.eh_src, eh_src, EH_ADR_LEN); //MAC源地址
tftp_packet->ehhdr.eh_type = HTONS(ETH_IP); //MAC帧类型
return length;
}
/********************************************************************
*名称:checksum
*参数:
* iphdr IP起始地址
* size IP头部长度
*返回:
* chksum IP头校验和
*功能:计算IP头校验和
*********************************************************************/
unsigned short checksum(unsigned short* iphdr, int size){
unsigned long chksum = 0;
while(size > 1){ //16位二进制求和
chksum += *iphdr++;
size -= sizeof(unsigned short);
}
if(size == 1){ //如果头部长度为奇数，那么再加上最后8位
chksum += *((unsigned char*)iphdr);
}
chksum = (chksum>>16) + (chksum&0xFFFF); //加上进位
chksum += (chksum>>16); //加上进位产生的进位
return (unsigned short)(~chksum); //返回反码
}
