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

本文主要是介绍华清远见作业第二十九天——网络编程（第四天），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

思维导图：

基于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;//向服务器发送ackgeshu=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;//如果是则块编号加1num++;*(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");	
}