下载
#include<myhead.h>
#include <sys/types.h>
#include <sys/socket.h>
#include<arpa/inet.h>
#include<netinet/in.h>
#define ERR_MSG(msg)do{\
fprintf(stderr,"line:%d: %s %s",__LINE__,__FILE__,__func__);\
perror(msg);\
}while(0)
#define IP "192.168.2.64"
#define PORT 69
int main(int argc, const char *argv[])
{
//创建报试套接字
int cfd=socket(AF_INET,SOCK_DGRAM,0);
if(cfd<0)
{ ERR_MSG("socket");
return -1;
}
printf("create sock success cfd=%d __%d__\n",cfd,__LINE__);
//填充服务器的地址信息结构体,给bind函数绑定
//真实的地址信息结构体根据地址族制定:AF_INET:man 7 ip
struct sockaddr_in sin;
sin.sin_family =AF_INET; //必须填AF_INET;
sin.sin_port =htons(PORT);//1024~49151 网络字节序
sin.sin_addr.s_addr=inet_addr(IP);//ifconfig出来本机IP
//输入提示信息
int num=0;
char buf[516]="";
char testname[20]="";
testname[strlen(testname)]='\0';
printf("请输入下载的文件\n");
scanf("%s",testname);
//发送下载请求
int size=sprintf(buf,"%c%c%s%c%s%c",0,1,testname,0,"octet",0);
if(sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("send");
return -1;
}
printf("send success __%d__\n",__LINE__);
//打开文件用于存储数据
int fd= open(testname,O_RDWR | O_CREAT | O_APPEND,0664);
// int fd=-1;
struct sockaddr_in cin;
socklen_t addrlen=sizeof(cin);
size_t sec;
while(1)
{
//接收数据包
strcpy(buf,"");
if((sec=recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&cin,&addrlen))<0)
{
ERR_MSG("recvfrom");
return -1;
}
//存储ACK值
char ACK[4];
ACK[0]=0;
ACK[1]=4;
ACK[2]=buf[2];
ACK[3]=buf[3];
//根据ACK判断是数据包还是错误包
if(buf[1]==3)//数据包
{
//创建文件用于存储信息
/* if(fd<0)
{
open(testname,O_RDWR | O_CREAT | O_APPEND,0664);
if(fd<0)
{
ERR_MSG("open failed");
return -1;
}
}
*/
// if(htons(num+1)==*(unsigned short *)(buf+2) )
// {
char *str=buf+4;
write(fd,str,sec-4);
if(sec<516)
{
printf("下载成功\n");
break;
}
//发送ACK
if(sendto(cfd,ACK,sizeof(ACK),0,(struct sockaddr*)&cin,sizeof(cin))<0)
{
ERR_MSG("send");
return -1;
}
// }
}
//错误包
else if(5==buf[1])
{
printf("download failed\n");
break;
}
}
//关闭套接字
close(cfd);
return 0;
}
上传
#include<myhead.h>
#include <sys/types.h>
#include <sys/socket.h>
#include<arpa/inet.h>
#include<netinet/in.h>
#define ERR_MSG(msg)do{\
fprintf(stderr,"line:%d: %s %s",__LINE__,__FILE__,__func__);\
perror(msg);\
}while(0)
#define IP "192.168.2.64"
#define PORT 69
int main(int argc, const char *argv[])
{
//创建报试套接字
int cfd=socket(AF_INET,SOCK_DGRAM,0);
if(cfd<0)
{ ERR_MSG("socket");
return -1;
}
printf("create sock success cfd=%d __%d__\n",cfd,__LINE__);
//填充服务器的地址信息结构体,给bind函数绑定
//真实的地址信息结构体根据地址族制定:AF_INET:man 7 ip
struct sockaddr_in sin;
sin.sin_family =AF_INET; //必须填AF_INET;
sin.sin_port =htons(PORT);//1024~49151 网络字节序
sin.sin_addr.s_addr=inet_addr(IP);//ifconfig出来本机IP
//客户端结构体
struct sockaddr_in cin;
socklen_t addrlen=sizeof(cin);
//输入提示信息
int num=0;
char buf[516]="";
char testname[20]="";
testname[strlen(testname)]=0;
printf("请输入上传的文件\n");
scanf("%s",testname);
//发送上传请求
int size=sprintf(buf,"%c%c%s%c%s%c",0,2,testname,0,"octet",0);
if(sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("send");
return -1;
}
printf("send success __%d__\n",__LINE__);
//打开文件
int fd=open(testname,O_RDONLY);
size_t sec;
while(1)
{
//接收ACK
strcpy(buf,"");
if((recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&cin,&addrlen))<0)
{
ERR_MSG("recvfrom");
return -1;
}
printf("%d",buf[3]);
//存储ACK值
char ACK[4];
ACK[0]=0;
ACK[1]=4;
ACK[2]=buf[2];
ACK[3]=buf[3];
//根据ACK判断是数据包还是错误包
if(buf[1]==4)//发送数据包
{
//打开文件并读取存储数据
// bzero(buf,sizeof(buf));
// short *str1=(short *)(buf+2);
char *str2=buf+4;
buf[1]=3;
buf[3]=num+1;
sec=read(fd,str2,sizeof(buf)-4);
if(sec==0)
{
printf("下载成功\n");
break;
}
//发送数据包
if(sendto(cfd,buf,4+sec,0,(struct sockaddr*)&cin,sizeof(cin))<0)
{
ERR_MSG("send");
return -1;
}
num++;
}
//错误包
else if(5==buf[1])
{
printf("updown failed\n");
break;
}
}
close(fd);
//关闭套接字
close(cfd);
return 0;
}