华清远见作业第二十九天——网络编程（第四天）

思维导图：

?基于UDP的TFTP文件传输

代码：

#include <a.h>
void menu();
int download(int cfd, struct sockaddr_in sin);
int up(int cfd, struct sockaddr_in sin);
#define SER_PORT 69
#define SER_IP "192.168.125.4"
int main(int argc, const char *argv[])
{
	//1创建用于通信的套接字文件描述符
	int cfd=socket(AF_INET,SOCK_DGRAM,0);
	if(cfd==-1)
	{
		perror("111socket error:");
		return -1;
	}
	//2绑定
	//3填充服务器的地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(SER_PORT);
	sin.sin_addr.s_addr=inet_addr(SER_IP);
	//选择语句
	int choose;
	while(1)
	{
		menu();
		printf("请输入相关操作:");
		scanf("%d",&choose);
		getchar();
		switch(choose)
		{
			case 1:
      			  {
				//下载功能
				download(cfd,sin);
				break;
			  }
			case 2:
      			  {
				//上传功能
				up(cfd,sin);
				break;
			  }
			case 3:
      			  {
				//退出功能
				//download(cfd,sin);
				goto END;
				break;
			  }
		}
	}
END: close(cfd);
	return 0;
	
}
//下载
int download(int cfd, struct sockaddr_in sin)
{

	char downloadname[128]="";
	printf("请输入你要下载的文件：");
	fgets(downloadname,120,stdin);
	downloadname[strlen(downloadname)-1]=0;
	
	//向服务器发送下载请求
	//向服务器发送下载请求
	char buf[512+2+2]=""; //数据包
	//组装请求数据
	short *p1=(short *)buf;
	*p1=htons(1);
	
	char *p2=buf+2;
	strcpy(p2,downloadname);  // 文件名字
	
	char *p3=p2+strlen(p2)+1;
	strcpy(p3,"octet");  //文件传输模式

	int len=4+strlen(p2)+strlen(p3); //要发送的长度
	//向服务器发送请求
	sendto(cfd,buf,len,0,(struct sockaddr*)&sin,sizeof(sin));
	//循环接收发送应答包
	ssize_t recvlen; //存放客户端发来消息函数的返回值（读取数据的个数）
	int fd=-1;//文件描述符号
	unsigned short num=1; //每一块的数据编号，初始值为1　//为啥是无符号型的我也不知道，有符号的就是错的那个机械臂也是必须要无符号的，真是服了
	socklen_t addrlen = sizeof(sin);
	int flag=0;  //防止被写入的文件重复打开
	while(1)
	{	
		//清空数据包
		bzero(buf,sizeof(buf));
		//读取服务器发回来的信息（读取数据的个数）
		recvlen=recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);
		if(recvlen==-1)
		{
			printf("发送失败\n");
			perror("发送error:");
			return -1;
		}
		if(3==buf[1])   //前两位是3为数据包
		{	
			if(0 == flag)     //防止文件重复打开
			{
				//创建并打开文件，准备存储
				fd = open(downloadname, O_WRONLY|O_CREAT|O_TRUNC, 0664);  //文件名字
				if(fd==-1)
				{
					perror("open error");
					return -1;
				}
				flag=1;
			}
			//判断数据包编号是否是自己想要的
			if(htons(num)==*(unsigned short*)(buf+2))  //强转为short并取值与num的网络字节序比
			{
				//判断成功后开始从数据包中提取数据写入到刚刚创建的文件中
				int geshu=write(fd,buf+4,recvlen-4);//减去前面的操作码和块编号
				if(geshu==-1)
				{
					printf("写入错误\n");
					perror("写入错误:");
					return -1;
				}
				//回复ack包
				//ack的全部内容和数据包的前四位内容一样
				//修改操作码
				buf[1]=4;
				//向服务器发送ack
				geshu=sendto(cfd,buf,4,0,(struct sockaddr*)&sin,sizeof(sin));
				if(geshu==-1)
				{
					printf("ack发送错误\n");
					return -1;
				}
				//判断是否传输结束
				if(recvlen<512+2+2) //数据包要加上操作位和标志位
				{
					printf("文件传输完成了\n");
					
					break;
				}
				num++;  //每一个块编号加1
			}
		}
		else if(5==buf[1]) // 前两位是5表示内容错误，错误包
		{
			//错误
			printf("错误信息为：%s\n",buf+4);
			break;
		}
		
			
	}

	
	return 0;
}
//上传功能
int up(int cfd, struct sockaddr_in sin)
{
	char upname[128]="";
	printf("请输入你要上传的文件：");
	fgets(upname,120,stdin);
	upname[strlen(upname)-1]=0;
	//判断文件是否存在以只读的形式打开
	int fd=open(upname,O_RDONLY);
	if(fd==-1)
	{
		printf("文件不存在，或者打开错误\n");
		return -1;
	}
	printf("打开成功\n");
	//核心上传给服务器
	char buf[512+2+2]=""; //数据包
	//组装请求数据
	short *p1=(short *)buf;
	*p1=htons(2); //请求写入
	
	char *p2=buf+2;
	strcpy(p2,upname);  // 文件名字
	
	char *p3=p2+strlen(p2)+1;
	strcpy(p3,"octet");  //文件传输模式

	int len=4+strlen(p2)+strlen(p3); //要发送的长度
	//向服务器发送请求
	sendto(cfd,buf,len,0,(struct sockaddr*)&sin,sizeof(sin));
	//上传的模块客户端循环向服务器发送数据
	ssize_t recvlen; //存放从服务器的返回值（读取数据的个数）
	unsigned short num=0; //每一块的数据编号，初始值为0　
	socklen_t addrlen = sizeof(sin);
	//循环向服务器发送信息	
	while(1)
	{
		//清空数据包
		bzero(buf,sizeof(buf));
		//读取服务器发回来的信息（读取数据的个数）
		recvlen=recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);
		if(recvlen==-1)
		{
			printf("发送失败\n");
			perror("发送error:");
			return -1;
		}
		if(4==buf[1])	//第一次先返回ack
		{
			printf("1111111111\n");
			//判断块编号
			if(htons(num)==*(unsigned short*)(buf+2))
			{
				buf[1]=3;
				//如果是则块编号加1
				num++;
				*(unsigned short*)(buf+2)=htons(num);//在转换为主机字节排序
				//从客户端上面开始读取文件，通read函数读取fd，存到buf的数据位
				//du_num来统计读取的个数,如果读取的时候du_num=0则代表读取完成了
				int du_num;
				du_num=read(fd,buf+4,512);
				if(du_num==-1)
				{
					perror("读取错误:");
					return -1;
				}
				else if(du_num==0)
				{
					printf("上传完成了\n");
					break;
				}
				//向服务器发送数据包
				sendto(cfd,buf,du_num+4,0,(struct sockaddr*)&sin,sizeof(sin));
				printf("2222\n");
			}
			else
			{
				printf("错误了\n");
				break;
			}
		}
		else if(5==buf[1]) // 前两位是5表示内容错误，错误包
		{
			//错误
			printf("错误信息为：%s\n",buf+4);
			break;
		}		
	}
	return 0;
}



//目录
void menu()
{
        printf("******************\n");
        printf("*     1.下载     *\n");
        printf("*     2.上传     *\n");
        printf("*     3.退出     *\n");
        printf("******************\n");	
}

运行效果：

?