网络编程作业 客户端下载和上传

下载功能代码

#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<stdlib.h>
#include<netinet/in.h>
#include <arpa/inet.h>
#include<string.h>
#include<unistd.h>
//打印错误新的宏函数
#define ERR_MSG(msg) do{\
			fprintf(stderr,"line: %d ",__LINE__);\
			perror(msg);}while(0);
int main(int argc, const char *argv[])
{
	
	if(argc<3)
	{
		fprintf(stderr,"请输入 IP 和 文件名\n");
		return -1;
	}

//创建套接字
	int sfd=socket(AF_INET,SOCK_DGRAM,0);
	if(sfd==-1)
	{
		ERR_MSG("socket");
		return -1;
	}

//绑定客户端自身的地址信息结构体 ---> 非必须绑定 

//填充服务器的IP地址以及端口号 -->因为客户端要主动发送数据包给服务器
struct sockaddr_in sin;
sin.sin_family     =AF_INET;
sin.sin_port       =htons(69);
sin.sin_addr.s_addr=inet_addr(argv[1]);



//读写请求协议
char buf[1000]="";


//发送读写请求协议
short int *pa=(short int *)buf;
*pa=htons(1);
char *pb=buf+2;
strcpy(pb,argv[2]);
char *pc=pb+strlen(pb);
char *pd=pc+1;
strcpy(pd,"octet");
int size=4+strlen(pb)+strlen("octet");


if(sendto(sfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
{
	ERR_MSG("sendto");
	return -1;
}


//存储接收到的数据包来自哪里
struct sockaddr_in rcv;
socklen_t addrlen=sizeof(rcv);
ssize_t res=0;


//打开要储存下载目标的文件,以写的方式
FILE *fp=fopen(argv[2],"w+");
if(fp==NULL)
{
	ERR_MSG("fopen");
	return -1;
}


//接收文件
short int i=0;
char down[516]="";  //数据包
char *down1=down;  //把数据和数据包分离
down1=down1+4;
while(1)
{
	bzero(down,sizeof(down));
	res=recvfrom(sfd,down,sizeof(down),0,(struct sockaddr*)&rcv,&addrlen);
	if(res<0)
	{
		ERR_MSG("recvfrom");
		return -1;
	}
	//写入数据
	fwrite(down1,res-4,1,fp);
	
	//返回应答
	i++;
	char r[4];
	short int* pa=(short int*)r;
	*pa=htons(4);
	short int* pb=pa+1;
	*pb=htons(i);
	if(sendto(sfd,r,4,0,(struct sockaddr*)&rcv,sizeof(rcv))<0)
	{
		ERR_MSG("sendto");
		return -1;
	}

	//判断文件是否传输完
	if(res<516)
	{
		printf("文件下载完成\n");
		break;
	}
}

close(sfd);
//关闭储存下载目标的文件
fclose(fp);
	return 0;
}

上传功能代码

#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<stdlib.h>
#include<netinet/in.h>
#include <arpa/inet.h>
#include<string.h>
#include<unistd.h>
#include <sys/stat.h>
#include <fcntl.h>
#include<errno.h>
//打印错误新的宏函数
#define ERR_MSG(msg) do{\
			fprintf(stderr,"line: %d ",__LINE__);\
			perror(msg);}while(0);
int main(int argc, const char *argv[])
{
	
	if(argc<3)
	{
		fprintf(stderr,"请输入 IP 和 文件名\n");
		return -1;
	}

//创建套接字
	int sfd=socket(AF_INET,SOCK_DGRAM,0);
	if(sfd==-1)
	{
		ERR_MSG("socket");
		return -1;
	}

//绑定客户端自身的地址信息结构体 ---> 非必须绑定 

//填充服务器的IP地址以及端口号 -->因为客户端要主动发送数据包给服务器
struct sockaddr_in sin;
sin.sin_family     =AF_INET;
sin.sin_port       =htons(69);
sin.sin_addr.s_addr=inet_addr(argv[1]);



//读写请求协议
char buf[1000]="";


//发送写请求协议
short int *pa=(short int *)buf;
*pa=htons(2);
char *pb=buf+2;
strcpy(pb,argv[2]);
char *pc=pb+strlen(pb);
char *pd=pc+1;
strcpy(pd,"octet");
int size=4+strlen(pb)+strlen("octet");


if(sendto(sfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
{
	ERR_MSG("sendto");
	return -1;
}


//存储接收到的数据包来自哪里
struct sockaddr_in rcv;
socklen_t addrlen=sizeof(rcv);


//接收同意
char x[500];  //接收应答
if(recvfrom(sfd,x,sizeof(x),0,(struct sockaddr*)&rcv,&addrlen)<0)
{
	ERR_MSG("recvfrom");
	return -1;
}

//打开要上传目标的文件,以读的方式
umask(0);
int fd=open(argv[2],O_RDONLY);
if(fd==-1)
{
	ERR_MSG("open");
	return -1;
}
printf("open success\n");
//返回值

//发送文件
short int i=0;
char down[516]="";  //数据包
char *down1=down;  //把数据和数据包分离
down1=down1+4;
ssize_t res=0;      //查看数据大小
while(1)
{
	bzero(down,sizeof(down));
	//准备数据包
	short int* pa=(short int*)down; //操作码
	*pa=htons(3);
	short int* pb=pa+1;             //块编号
	*pb=htons(++i);
	res=read(fd,down1,512);         //数据
	if(res<0)
	{
		ERR_MSG("write");
		return -1;
	}

	//发送数据包
	if(sendto(sfd,down,res+4,0,(struct sockaddr*)&rcv,sizeof(rcv))<0)
	{
		ERR_MSG("sendto");
		return -1;
	}
//	printf("sendto success\n");

	//接收应答
	if(recvfrom(sfd,x,sizeof(x),0,(struct sockaddr*)&rcv,&addrlen)<0)
	{
		ERR_MSG("recvfrom");
		return -1;
	}
//	printf("recvfrom success\n");

	//判断文件是否上传完
	if(res<512)
	{
		printf("文件上传完成\n");
		break;
	}
}

close(sfd);
//关闭储存下载目标的文件
close(fd);
	return 0;
}