网络编程 传输层 UDP 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>
                                                    
#define ERR_MSG(msg) do{\
    fprintf(stderr, "line: %d ", __LINE__);\
    perror(msg);\
}while(0)

int do_upload(int sfd, struct sockaddr_in sin);
int do_download(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;
	}
	
	//绑定客户端自身的IP和端口--->非必须绑定，所以可以绑定，也可以不绑定
	//如果不绑定，则系统会自动绑定一个端口号到客户端的套接字上
	//所以以下这段绑定代码，可以选择写，也可以选择不写
	
	//填充客户端的IP和端口供于bind函数绑定
	struct sockaddr_in cin; 
	cin.sin_family 		= AF_INET;
	cin.sin_port 		= htons(6666); 			//客户端绑定到6666端口号上
	cin.sin_addr.s_addr = inet_addr("192.168.125.129"); 	//客户端运行后所在环境的IP 

	if(bind(sfd, (struct sockaddr*)&cin, sizeof(cin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");

	//填充服务器的IP和端口, 供于下面sendto函数的时候发送给服务器使用;
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET; 		//必须填这个
	sin.sin_port 		= htons(69); 	//服务器绑定的端口号，69
	sin.sin_addr.s_addr = inet_addr("192.168.8.239");//服务器绑定的IP，填本机tftp服务器运行环境的I



	//选择是上传还是下载，或者退出
	char c = 0;
	while(1)
	{
		printf("------------------------------\n");
		printf("-----------1. 上传------------\n");
		printf("-----------2. 下载------------\n");
		printf("-----------3. 退出------------\n");
		printf("------------------------------\n");
		printf("请输入>>>");	
		c = getchar();
		while(getchar()!=10);

		switch(c)
		{
		case '1': 		
			do_upload(sfd, sin);
			break;
		case '2':
			do_download(sfd, sin);
			break;
		case '3':
			goto END;
			break;
		default:
			printf("输入错误，请重新输入!\n");

		}

	}


END:
	//关闭文件描述符
	close(sfd);

	return 0;
}

int do_upload(int sfd, struct sockaddr_in sin)
{
	char name[20] = "";
	printf("请输入：");
	fgets(name, 20, stdin);
	name[strlen(name)-1] = 0;

	int fd = open(name, O_RDONLY); // 只读打开准备上传的文件
	if(fd < 0)
	{
		ERR_MSG("poen");
		return -1;
	}

	char buf[516] = "";
	int size = sprintf(buf, "%c%c%s%c%s%c", 0,2,name,0,"octet",0); // 组请求包 发送给服务器

	if(sendto(sfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");	
		return -1;
	}

	int recv_len; // 成功收到的字节数
	unsigned short num = 0; // 校验数据包顺序
	socklen_t addrlen = sizeof(sin); // 存放 临时地址信息结构体大小
	while(1)
	{
		bzero(buf, sizeof(buf));
		recv_len = recvfrom(sfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, &addrlen);
		if(recv_len < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}

		if(4 == buf[1])
		{
			if(num == ntohs(*(unsigned short*)(buf+2))) // 第一次的对应请求的ACK（num == 0）
			{
				buf[1] = 3;

				// 填充块编号
				num++;
				*(unsigned short*)(buf+2) = htons(num);

				// 读数据
				int res = read(fd, buf+4, 512);
				if(res < 0)
				{
					ERR_MSG("read");
					return -1;
				}
				else if(0 == res)
				{
					printf("文件上传完毕\n");
					break;
				}

				// 发送数据包（上传）
				if(sendto(sfd, buf, res+4, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
				{
					ERR_MSG("sendto");
					return -1;
				}
			}

			else
			{
				printf("块编号对应不上\n");
				break;
			}
		}

		if(5 == buf[1])
		{
			printf("ERROR:%s\n", buf+4);
			break;
		}
	}
	return 0;
}

int do_download(int sfd, struct sockaddr_in sin)
{
	//组下载请求包，发送给服务器，服务器在69号端口等待请求
	char buf[520] = "";

	unsigned short* pa = (unsigned short*)buf;
	*pa = htons(1); 		//代表下载操作码

	printf("请输入文件名>>>");
	char name[20] = "";
	scanf("%s", name);
	char* pb = buf+2;
	strcpy(pb, name);

	char* pd = pb+strlen(pb)+1;
	strcpy(pd, "octet");

	int size = 2+strlen(pb)+7;
	//发送协议
	if(sendto(sfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	printf("下载请求发送成功\n");

	socklen_t addrlen = sizeof(sin);
	ssize_t res = 0;
	unsigned short num = 0;
	while(1)
	{
		bzero(buf, sizeof(buf));
		//循环接收数据包，地址信息结构体必须接，因为后续我们要将ack发给临时端口
		res = recvfrom(sfd, buf, 516, 0, (struct sockaddr*)&sin, &addrlen);
		if(res < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		
	//	printf("%d %d\n", *(unsigned short*)buf, ntohs(*(unsigned short*)buf));
	//	printf("%d %d\n", buf[0], buf[1]);

		//由于发送回来的操作码都是网络字节序。所有有效的操作码都存储在buf[1]的位置
		//只要判断buf[1]是3还是5即可
		if(3 == buf[1]) 		//数据包
		{
			//由于UDP可能丢包，或者重复到达，所以需要在每次收到数据包的时候
			//判断一下这个包是不是我需要的，通过快编号判断
			if(num+1 == ntohs(*(unsigned short*)(buf+2)))	
			{
				//处理数据，数据首地址在buf+4的位置上
				printf("%s", buf+4); 		//修改成写入到文件中
				fflush(stdout);

				//回复ACK
				buf[1] = 4;
				if(sendto(sfd, buf, 4, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
				{
					ERR_MSG("sendto");
					return -1;
				}

				//如果数据小于512个字节，则传输结束
				if(res-2-2<512)
				{
					printf("数据传输结束\n");
					break;
				}

				num++;
			}
		}
		else if(5 == buf[1]) 	//错误包
		{
			printf("ERROR:%s\n", buf+4);
			return -2;
		}

	}
	return 0;
}

测试结果：

上传:

  下载：（打印到终端）

 服务端日志：