思维导图:
向服务器发送数据实现下载
代码:
#include <a.h>
void menu();
int download(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);
switch(choose)
{
case 1:
{
//下载功能
download(cfd,sin);
break;
}
case 2:
{
//上传功能
//download(cfd,sin);
break;
}
case 3:
{
//退出功能
//download(cfd,sin);
goto END;
break;
}
}
}
END:
return 0;
}
//下载
int download(int cfd, struct sockaddr_in sin)
{
char downloadname[128]="";
printf("请输入你要下载的文件:");
scanf("%s",downloadname);
getchar(); //防止垃圾字符
//向服务器发送下载请求
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 //为啥是无符号型的我也不知道,有符号的就是错的那个机械臂也是必须要无符号的,好像是取值范围的问题,原来的short的取值范围太小了,unsigned short的取值范围大一点
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;
}
}
//关闭套接字
close(cfd);
return 0;
}
//目录
void menu()
{
printf("******************\n");
printf("* 1.下载 *\n");
printf("* 2.上传 *\n");
printf("* 3.退出 *\n");
printf("******************\n");
}
运行效果: