目录
作业要求：基于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;
}
//构造ACK
buf[1] = 4;
//发送ACK
if(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;
}
//判断是否接收到服务器发来的ACK
if(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;
}
下载功能效果图
上传功能效果图
思维导图
（从第二章到第五章）
模拟面试题和答案（定期更新）
