网络编程作业2

代码：

#include <myhead.h>


int do_download(int cfd,struct sockaddr_in sin)
{
	//定义字符数组
	char buf[516]="";
	char filename[40]="";
	//得到文件名
	printf("please enter filename:");
	scanf("%s",filename);
	getchar();
	short *p1=(short *)buf;
	//将下载操作码转化为网络字节序
	*p1=htons(1);
	char *p2=buf+2;
	//将文件名放入buf中
	strcpy(p2,filename);
	char *p3=p2+strlen(p2)+1;
	//将模式放入buf中
	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;
	}
	//以只写的形式打开文件
	int fd=-1;
	if((fd=open(filename,O_WRONLY|O_CREAT|O_TRUNC,0664))==-1){
		perror("open error");
		return -1;
	}
	socklen_t socklen=sizeof(sin);
	while(1){
		bzero(buf,sizeof(buf));
		//接收服务器发来的数据包
		if(recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr *)&sin,&socklen)==-1){
			perror("sendto error");
			return -1;
		}
		int len=strlen(buf+4);
		//判断数据包中数据的大小,小于512则结束
		//将数据包中的数据放入文件中
		write(fd,buf+4,len);
		//向服务器发送ACK
		short *pb=(short *)buf;
		*pb=htons(4);
		if(sendto(cfd,buf,4,0,(struct sockaddr *)&sin,sizeof(sin))==-1){
			perror("sendto error");
			return -1;
		}
		if(len<512){
			close(cfd);
			close(fd);
			break;
		}
	}
	return 0;
}


int do_upload(int cfd,struct sockaddr_in sin)
{
	//定义字符数组
	char buf[516]="";
	char filename[40]="";
	//得到文件名
	printf("please enter filename:");
	scanf("%s",filename);
	getchar();
	short *p1=(short *)buf;
	//将上传操作码转化为网络字节序
	*p1=htons(2);
	char *p2=buf+2;
	//将文件名放入buf中
	strcpy(p2,filename);
	char *p3=p2+strlen(p2)+1;
	//将模式放入buf中
	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;
	}
	//以只读的形式打开文件
	int fd=-1;
	if((fd=open(filename,O_RDONLY))==-1){
		perror("open error");
		return -1;
	}
	socklen_t socklen=sizeof(sin);
	int n=1;
	while(1){
		bzero(buf,sizeof(buf));
		//接收服务器发来的ACK
		if(recvfrom(cfd,buf,4,0,(struct sockaddr *)&sin,&socklen)==-1){
			perror("sendto error");
			return -1;
		}
		short *pb=(short *)buf;
		*pb=htons(3);
		buf[3]=n;
		//从文件中取数据
		int res=read(fd,buf+4,512);
		//向服务器发送数据包
		if(sendto(cfd,buf,516,0,(struct sockaddr *)&sin,sizeof(sin))==-1){
			perror("sendto error");
			return -1;
		}
		//判断文件中的数据是否读完
		n++;
		if(res<512){
				close(cfd);
				close(fd);
				break;
		}
	}
	return 0;
}


int main(int argc, const char *argv[])
{
	//判断终端执行输入是否正确
	if(argc!=2){
		printf("input error\n");
		return -1;
	}
	//创建套接字
	int cfd=socket(AF_INET,SOCK_DGRAM,0);
	if(cfd==-1){
		perror("socket error");
		return -1;
	}
	//填充服务器地址结构体
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(69);
	sin.sin_addr.s_addr=inet_addr(argv[1]);
	int muex;
	while(1){
		system("clear");
		printf("\t\t========1.下载=========\n");
		printf("\t\t========2.上传=========\n");
		printf("\t\t========3.清空=========\n");
		printf("please enter muex:");
		scanf("%d",&muex);
		switch(muex){
		case 1:{
				   do_download(cfd,sin);
				   printf("下载成功\n");
				   break;
			   }
		case 2:{
				   do_upload(cfd,sin);
				   printf("上传成功\n");
				   break;
			   }
		case 3:goto EXIT;
		default:printf("enter error");
		}
		printf("输入任意建,按回车清空:");
		while(getchar()!='\n');
	}
	close(cfd);
EXIT:
	return 0;
}

运行结果：