8.10 网络编程——客户端从服务器中下载文件 TFTP协议

利用TFTP协议 客户端从服务器中下载指定文件

过程分析

TFTP通信过程

1. 服务器在69号端口等待客户端的请求
2. 服务器若批准此请求，则使用 临时端口 与客户端进行通信。
3. 每个数据包的编号都有变化（从1开始）
4. 每个数据包都要得到ACK的确认，如果出现超时，则需要重新发送最后的数据包或ACK包
5. 数据长度以512Byte传输的，小于512Byte的数据意味着数据传输结束。

代码示例

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <fcntl.h>

//打印错误新的宏函数
#define ERR_MSG(msg)  do{\
	fprintf(stderr, " __%d__ ", __LINE__);\
	perror(msg);\
}while(0)

int main(int argc, const char *argv[])
{
	if(argc < 2)
	{
		fprintf(stderr, "请输入IP port\n");
		return -1;
	}
	
	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}

	//绑定客户端自身的地址信息结构体 ---> 非必须绑定 
	//填充服务器的IP地址以及端口号 -->因为客户端要主动发送数据包给服务器
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(69);
	sin.sin_addr.s_addr = inet_addr(argv[1]);
	
	struct sockaddr_in rcv_addrmsg; 	//存储接收到的数据包来自哪里
	socklen_t addrlen = sizeof(rcv_addrmsg);
	//发送请求
	char buf[517] = "";
	bzero(buf, sizeof(buf));
	char* ptr = buf;
	short int* pa = (short int*)ptr;
	*pa = htons(1);
	char *pb = ptr+2;
	char filename[128] = "";
	printf("请输入要下载的文件名>>>");
	scanf("%s",filename);
	getchar();
	strcpy(pb,filename);
	char* pd = pb+strlen(filename)+1;
	strcpy(pd,"octet");
	int size = 2+strlen(pb)+1+strlen(pd)+1;
	//将数据包发送给服务器，所以地址信息结构体需要填服务器的信息
	if(sendto(sfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	//printf("sendto success\n");
	
	//打开文件获取文件描述符
	int fd = open(filename,O_WRONLY|O_CREAT|O_TRUNC,0664);
	if(fd<0)
	{
		ERR_MSG("open");
		return -1;
	}
	short int i=0;  							//定义块编号
	while(1)
	{
		//接收数据包
		bzero(buf,sizeof(buf));
		ssize_t res;
		if((res=recvfrom(sfd,buf,sizeof(buf),0,(struct sockaddr*)&rcv_addrmsg,&addrlen)) < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		i++; 									//块编号++
		char *ptr = buf; 						//数据包
		short int* rpa = (short int*)ptr; 		//数据包操作码首地址
		short int* rpb = rpa+1; 				//数据包块编号首地址
		char *rpc = ptr+4; 						//数据包数据首地址

		char *errptr = buf; 					//错误包
		short int* epa = (short int*)errptr; 	//错误包操作码地址
		short int* epb = epa+1; 				//错误包的差错码地址
		char* epc = (char*)(epb+1); 			//错误包差错信息首地址

		char ACK[5] = ""; 						//ACK
		bzero(ACK,sizeof(ACK));
		short int *acka = (short int*)ACK; 		//ACK操作码首地址
		*acka = htons(4);
		short int* ackb = acka+1; 				//ACK块编号首地址
		*ackb = htons(i);
		//判断是否是数据包
		if(3==ntohs(*rpa)&&i==ntohs(*rpb))
		{
			//判断数据包中数据大小是否小于512字节
			if((res-4)<512)
			{
				//写入文件
				write(fd,rpc,res-4);
				//发送ACK验证
				if(sendto(sfd,ACK,4,0,(struct sockaddr*)&rcv_addrmsg,addrlen)<0)
				{
					ERR_MSG("sendto");
					return -1;
				}
				//printf("sendto success 结束\n");
				break;
			}
			else
			{
				write(fd,rpc,512);
				if(sendto(sfd,ACK,4,0,(struct sockaddr*)&rcv_addrmsg,addrlen)<0)
				{
					ERR_MSG("sendto");
					return -1;
				}
				//printf("sendto success 512 \n");
			}
		}
		else
			printf("ERROR:%s\n",epc); 					//打印错误信息
	}
	printf("下载成功!\n");
	//关闭套接字
	close(sfd);
	return 0;
}

测试结果