#include<myhead.h>
#define SER_IP "192.168.117.36"
#define SER_PORT 69
//客户端
void menu()
{
printf("*************************\n");
printf("*\t 1.下载 \t*\n");
printf("*\t 2.上传 \t*\n");
printf("*\t 3.退出 \t*\n");
printf("*************************\n");
}
int rev()
{
//下载
//1、创建用于通信的套接字文件描述符
int cfd = socket(AF_INET, SOCK_DGRAM, 0);
if(cfd == -1)
{
perror("socket error");
return -1;
}
printf("cfd = %d\n", cfd); //
//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); //ip地址
//3、收发数据
//校验收到的块编号
char filename[128]="";
printf("请输入要下载的文件名>>>");
scanf("%s",filename);
//下载请求
char buf[516] = "";
int size = sprintf(buf, "%c%c%s%c%s%c", 0, 1, filename, 0, "octet", 0);
if(sendto(cfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin))<0)
{
perror("sendto error");
return -1;
}
int flag = 0;
int fd ;
//循环接收发送应答包
ssize_t recv_len;
unsigned short num = 1;
socklen_t addrlen = sizeof(sin);
while(1)
{
bzero(buf, 516);
recv_len = recvfrom(cfd, buf, 516, 0, (struct sockaddr*)&sin, &addrlen);
if(recv_len < 0)
{
perror("recvfrom error");
return -1;
}
if(3 == buf[1]) //如果是数据包
{
if(0 == flag) //防止文件重复打开
{
//存储从服务器下载下来的文件,接收数据成功后再创建数据;
//创建并打开filename文件
fd = open(filename, O_WRONLY|O_CREAT|O_TRUNC, 0664);
if(fd < 0)
{
perror("open error");
return -1;
}
flag = 1;
}
//判断当前的快编号,是否与我期望的快编号一致
if(htons(num) == *(unsigned short*)(buf+2)) //防止数据包重复到达
{
//提取数据,写入到文件中
if(write(fd, buf+4, recv_len-4) < 0)
{
printf("fd:%d recv_len=%ld\n", fd, recv_len);
perror("write error");
break;
}
//回复ack包
//由于数据包前四个字节与ack包除了操作码不一致外,其余全部一致。
//所以我们直接将数据包的操作码修改成4即可,发送数据包的前四个字节。
buf[1] = 4;
if(sendto(cfd, buf, 4, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
{
perror("sendto error");
}
//判断数据包的大小是否小于 512+2+2;
if(recv_len < 512+2+2)
{
printf("-----文件下载完毕!!-----\n");
break;
}
num++;
}
}else if(5 == buf[1]) //错误包
{
//打印错误信息
printf("-----ERROR:%s-----\n", buf+4);
break;
}
}
//5、关闭套接字
close(cfd);
}
int snd()
{
//上传
//1、创建用于通信的套接字文件描述符
int cfd = socket(AF_INET, SOCK_DGRAM, 0);
if(cfd == -1)
{
perror("socket error");
return -1;
}
printf("cfd = %d\n", cfd); //3
//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); //ip地址
char filename[128] = "";
printf("请输入要上传的文件名>>>");
scanf("%s",filename);
//判断该文件是否存在
int fd = open(filename, O_RDONLY);
if(fd < 0)
{
perror("open error");
return -1;
}
//发送上传请求
//上传协议
char buf[516] = "";
int size = sprintf(buf, "%c%c%s%c%s%c", 0, 2, filename, 0, "octet", 0);
if(sendto(cfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin))<0)
{
perror("sendto error");
return -1;
}
//循环接收发送数据包
int recv_len;
unsigned short num = 0;
socklen_t addrlen = sizeof(sin);
while(1)
{
bzero(buf, 516);
recv_len = recvfrom(cfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, &addrlen);
if(recv_len < 0)
{
perror("recvfrom error");
return -1;
}
//操作码的范围是1-5,因为是网络字节序
if(4 == buf[1])
{
//判断当前数据包的编号是否等于应答包的编号
if(num == ntohs(*(unsigned short*)(buf+2)))
{
//修改操作码为数据包
buf[1] = 3;
//填充块编号
num++;
*(unsigned short*)(buf+2) = htons(num);
//读取数据
//发送数据
int res = read(fd, buf+4, 516-4);
if(res < 0)
{
perror("read error");
return -1;
}
else if(0 == res)
{
printf("-----文件上传完毕-----\n");
break;
}
//发送数据包
//发送的数据包大小为,读取到的字节数(res)+操作码(2byte)+快编号(2byte)
if(sendto(cfd, buf, res+4, 0, (struct sockaddr*)&sin, sizeof(sin)) <0)
{
perror("sendto error");
return -1;
}
}else
{
printf("-----文件上传失败,请检查网络环境-----\n");
break;
}
}else if(5 == buf[1])
{
printf("-----ERROR:%s-----\n", buf+4);
break;
}
}
//5、关闭套接字
close(cfd);
}
int main(int argc, const char *argv[])
{
int choice = 0;
while (1)
{
menu();
printf("请输入操作:");
scanf("%d",&choice);
switch (choice)
{
case 1: rev();
break;
case 2: snd();
break;
case 3: exit(0);
break;
default:printf("输入错误,请重新输入\n");
break;
}
getchar();
getchar();
system("clear");
}
return 0;
}
3.15作业
最新推荐文章于 2024-09-23 15:47:10 发布
文章描述了一个使用UDP协议实现的客户端程序,支持文件下载和上传功能,通过套接字进行通信,包括创建套接字、连接服务器、数据传输和错误处理等步骤。
摘要由CSDN通过智能技术生成