当前位置：首页 > news >正文

华清远见嵌入式学习——网络编程——作业3

news 2025/8/9 19:21:08

作业要求：基于UDP的TFTP文件传输

代码

下载功能效果图编辑

上传功能效果图

思维导图

模拟面试题和答案（定期更新）

作业要求：基于UDP的TFTP文件传输

完成文件的上传和下载功能

代码

#include<myhead.h>//实现下载功能
int do_download(int cfd, struct sockaddr_in sin)
{//定义变量存储下载请求包char buf[516] = "";//定义变量存储文件名char fileName[40] = "";printf("请输入文件名：");scanf("%s", fileName);getchar();//组装请求包short *p1 = (short *)buf;*p1 = htons(1);                 //写1表明要下载char *p2 = buf+2;           //文件名段strcpy(p2, fileName);char *p3 = p2+strlen(p2)+1;    //模式段strcpy(p3, "octet");int size = 4 + strlen(p2) + strlen(p3);      //要发送的请求包的大小//向服务器发送下载请求if(sendto(cfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin)) == -1){perror("sendto error");return -1;}printf("请求成功\n");socklen_t socklen = sizeof(sin);//打开一文件接受下载文件int fd = open(fileName,O_RDWR|O_TRUNC|O_CREAT,0666);if(fd<0){perror("open error");return -1;}int  res;while(1){//接受数据包bzero(buf,sizeof(buf));res = recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&socklen);if(res<0){perror("recvfrom error");return -1;}if(buf[1]==3){//写入文件if(write(fd,buf+4,res-4)<0){perror("write");return -1;}//构造ACKbuf[1] = 4;//发送ACKif(sendto(cfd,buf,4,0,(struct sockaddr*)&sin,sizeof(sin))<0){perror("sendto");return -1;}//判断数据包中的数据大小if(res-4<512){printf("%s 下载完毕\n",fileName);break;}}else if(buf[1]==5){printf("%d %s\n",ntohs(*(short*)(buf+2)),buf+4);break;}}close(fd);	return 0;
}//实现上传功能
int do_upload(int cfd,struct sockaddr_in sin)
{//定义变量存储上传请求包char buf[516] = "";//定义变量存储文件名char fileName[40] = "";printf("请输入文件名：");scanf("%s", fileName);getchar();//组装请求包short *p1 = (short *)buf;*p1 = htons(2);                 //写2表明要上传char *p2 = buf+2;           //文件名段strcpy(p2, fileName);char *p3 = p2+strlen(p2)+1;    //模式段strcpy(p3, "octet");int size = 4 + strlen(p2) + strlen(p3);      //要发送的请求包的大小//向服务器发送上传请求if(sendto(cfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin)) == -1){perror("sendto error");return -1;}printf("请求成功\n");socklen_t socklen = sizeof(sin);//打开准备上传的文件int fd = open(fileName,O_RDONLY);if(fd<0){perror("open error");return -1;}int res1,res2;int kbh = 1;while(1){//接受数据包bzero(buf,sizeof(buf));res1 = recvfrom(cfd,buf,4,0,(struct sockaddr*)&sin,&socklen);if(res1 < 0){perror("recvfrom error");return -1;}//判断是否接收到服务器发来的ACKif(buf[1]==4){//填充发送的数据包buf[1] = 3;buf[3] = kbh;res2 = read(fd,buf+4,512);if(res2 < 0){perror("read error");return -1;}					//向服务器发送数据包if(sendto(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,sizeof(sin)) == -1){perror("sendto error");return -1;}//发一次块编号加一一次kbh++;//循环退出条件if(res2 < 512){close(fd);printf("文件上传完毕\n");break;}}else if(buf[1]==5){printf("%d %s\n",ntohs(*(short*)(buf+2)),buf+4);break;}}return 0;
}
int main(int argc, const char *argv[])
{if(argc != 2){printf("input error\n");printf("usage:./a.out ip\n");return -1;}//1、创建套接字int cfd = socket(AF_INET, SOCK_DGRAM, 0);if(cfd == -1){perror("socket error");return -1;}//2、填充服务器地址信息结构体struct sockaddr_in sin;sin.sin_family = AF_INET;sin.sin_port = htons(69);sin.sin_addr.s_addr = inet_addr(argv[1]);int menu = -1;while(1){system("clear");                     //清屏printf("\t\t======1、下载=======\n");printf("\t\t======2、上传=======\n");printf("\t\t======0、退出=======\n");printf("请输入功能：");scanf("%d", &menu);getchar();//多分支选择switch(menu){case 1:{do_download(cfd, sin);}break;case 2:{do_upload(cfd,sin);}break;case 0:goto POS;default:printf("输入功能有误，请重新输入\n");}//阻塞printf("输入任意键，按回车清空:");while(getchar() != '\n');}POS://关闭套接字close(cfd);return 0;
}