基于tftp服务器的客户端下载、上传

 实现过程：

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

#define ERR_MSG(msg) do{\
	fprintf(stderr, "line:%d\n", __LINE__);\
	perror(msg);\
}while(0)

#define IP "192.168.8.177"   //windowsIP
#define PORT 69           //1024~49151

int download(int sfd, struct sockaddr_in sin);
int upload(int sfd, struct sockaddr_in sin);

int main(int argc, const char *argv[])
{
	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}


	//绑定客户端自身的地址信息结构体-->非必须绑定，建议不写


	//填充服务器自身的地址信息结构体，AF_INET: man 7  IP
	//供于下方的sendto使用，因为sendto必须要指明发给谁
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	char input=0;
	while(1)
	{
		//system("clear");

		printf("-----------------------\n");
		printf("------1.下载文件-------\n");
		printf("------2.上传文件-------\n");
		printf("------3.退出-----------\n");
		printf("-----------------------\n");
		printf("请输入选项>>>");
		scanf("%c",&input);
		while(getchar()!=10);

		switch(input)
		{
		case '1': 	
			download(sfd, sin);
			break;
		case '2':
			upload(sfd,sin);
			break;
		case '3':
			//关闭套接字
			close(sfd);
			return 0;

		}

		//printf("输入任意字符清屏>>>");
		//while(getchar()!=10);
	}


}
int upload(int sfd, struct sockaddr_in sin)
{
	char buf[516] = "";
	ssize_t res = 0;

	char filename[20] = "";
	printf("请输入要上传的文件名>>>");
	scanf("%s", filename);
	while(getchar()!=10);

	//组合上传协议
	int size = sprintf(buf, "%c%c%s%c%s%c", 0, 2, filename, 0, "octet", 0);

	//发送上传协议
	if(sendto(sfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}

	//打开一个文件，用于存储下载到的数据,设置标记符判断文件是否存在

	int fd=-1;
	fd = open(filename, O_RDONLY);
	if(fd < 0)
	{
		ERR_MSG("open");
		return -1;
	}
	socklen_t addrlen = sizeof(sin);

	unsigned short opcode = 0;     //操作码
	unsigned short num = 0; 	   //块编号
	unsigned short numcode = 0;    //块编号效验码判断数据包是否重复发送或漏发
	while(1)
	{
		bzero(buf, sizeof(buf));
		//接收数据包
		recvfrom(sfd, buf, 4, 0, (struct sockaddr*)&sin, &addrlen);
		if(res < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		opcode=ntohs(*(short*)buf);
		num=ntohs(*(short*)(buf+2));

		//效验ACK回复正确，错误不再发送
		if(num==numcode++&&4==opcode)
		{
			//组数据包
			*(short*)buf=htons(3);
			*(short*)(buf+2)=htons(numcode);
            res=read(fd, buf+4, 512); 
			if(res< 0)
			{
				ERR_MSG("read");
				break;
			}

			//发送数据包
			if(sendto(sfd, buf,res+4, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
			{
				ERR_MSG("sendto");
				break;
			}

			//判断数据是否＜512 
			if(res < 512)
			{
				printf("文件 %s 上传完毕\n", filename);
				break;
			}
		}
	}

	close(fd);
	return 0;
}



int download(int sfd, struct sockaddr_in sin)
{
	char buf[516] = "";
	ssize_t res = 0;

	char filename[20] = "";
	printf("请输入要下载的文件名>>>");
	scanf("%s", filename);
	while(getchar()!=10);

	//组合下载协议
	int size = sprintf(buf, "%c%c%s%c%s%c", 0, 1, filename, 0, "octet", 0);
	//发送下载协议
	if(sendto(sfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}

	//打开一个文件，用于存储下载到的数据,设置标记符判断文件是否存在
	int fd=-1, flag = 0;
    socklen_t addrlen = sizeof(sin);
	
	unsigned short opcode = 0;     //操作码
	unsigned short num = 1; 	   //块编号
	unsigned short numcode = 1;    //块编号效验码判断数据包是否重复发送或漏发
	while(1)
	{
		bzero(buf, sizeof(buf));
		//接收数据包
		res = recvfrom(sfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, &addrlen);
		if(res < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		opcode=ntohs(*(short*)buf);
		num=ntohs(*(short*)(buf+2));

		//判断操作码进行下一步操作，3接收数据包，5打印错误信息
		if(3 == opcode) 
		{	
			printf("操作码:%d 块编号:%d\n",opcode,num);
			if(0 == flag)
			{
				fd = open(filename, O_WRONLY|O_CREAT|O_TRUNC, 0664);
				if(fd < 0)
				{
					ERR_MSG("open");
					return -1;
				}
				flag = 1;
			}

			//UDP是无连接的不可靠的，数据包可能会重复接收到的通信;
			//可以在本地记录每次收到的包的编号，
			//如果本地记录的包的编号与数据包发送回来的快编号不一致，则不做处理

			if(numcode == num)  //核对块编号
			{
				//将数据保存到文件中，
				//最后一次下载到的数据肯定小于512，所以不能直接写512
				if(write(fd, buf+4, res-4) < 0)
				{
					ERR_MSG("write");
					break;
				}

				//回复ACK
				buf[1] = 4;  //将操作码置4
				if(sendto(sfd, buf, 4, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
				{
					ERR_MSG("sendto");
					break;
				}

				//判断数据是否＜512 
				if(res-4 < 512)
				{
					printf("文件 %s 下载完毕\n", filename);
					break;
				}

				numcode++;
			}
		}
		else if(5 == opcode) 	//错误包
		{
			printf("ERROR: 差错码:%d 差错信息:%s\n", num, buf+4);
			break;
		}
	}

	close(fd);
	return 0;
}

 实现效果：