网络编程day3

最新推荐文章于 2024-07-31 00:00:00 发布

m0_70569664

最新推荐文章于 2024-07-31 00:00:00 发布

阅读量74

点赞数

文章标签：网络算法

本文链接：https://blog.csdn.net/m0_70569664/article/details/132570185

版权

下载

#include<myhead.h>
#define IP "192.168.115.124"
int do_download(int socfd,struct sockaddr_in cin,FILE *fp);
int main(int argc, const char *argv[])
{
	FILE *fp;
	fp=fopen(argv[1],"w+");
	if(fp==NULL)
	{
		perror("fopen error");
		return -1;
	}
	//创建报式套接字  socket
	int socfd=socket(AF_INET,SOCK_DGRAM,0);
	if(socfd<0)
	{
		perror("socket error");
		return -1;
	}
	printf("socket success,socfd=%d\n",socfd);


	char buf[256]="";
	struct sockaddr_in cin;
	cin.sin_family=AF_INET;
	cin.sin_port=htons(69);
 	cin.sin_addr.s_addr=inet_addr(IP);	
	char choose=0;
	while(1)
	{
		system("clear");
		printf("------------------\n");
		printf("-----1、下载------\n");
		printf("-----2、上传------\n");
		printf("-----3、退出------\n");
		printf("------------------\n");
		printf("请输入>>>");
		scanf("%c",&choose);
		while(getchar()!=10);

		switch(choose)
		{
		case '1':
			do_download(socfd,cin,fp);
			break;
		case '2':

			break;
		case '3':
			goto END;
			break;
		default:
			printf("输入错误，请从重新输入\n");
			break;
		}
		printf("输入任意字符清屏>>>>");
		while(getchar()!=10);
	}

END:
	fclose(fp);
	//关闭套接字，释放资源    close
	close(socfd);
	return 0;
}



int do_download(int socfd,struct sockaddr_in cin,FILE *fp)
{
	char filename[20]="";
	printf("请输入要下载的文件名>>");
	scanf("%s",filename);
	while(getchar()!=10);
	//组下载请求包
	char buf[516];
	unsigned short* ptr1=(unsigned short*)buf;
	*ptr1=htons(1);

	char* ptr2=buf+2;
	strcpy(ptr2,filename);    //组名字

	char*ptr4=ptr2+strlen(filename)+1;
	strcpy(ptr4,"octet");    	//组模式

	int size=2+strlen(ptr2)+1+strlen(ptr4)+1;

	if(sendto(socfd,buf,size,0,(struct sockaddr *)&cin,sizeof(cin))<0)
	{
		perror("sendto 请求 error");
		return -1;
	}
	printf("sendto download request success\n");
	socklen_t addrlen=sizeof(cin);
	//发送下载请求
	while(1)
	{
		bzero(buf,sizeof(buf));
		//接收数据，接收地址信息
		ssize_t a=recvfrom(socfd,buf,sizeof(buf),0,(struct sockaddr *)&cin,&addrlen);
		if(a<0)
		{
			perror("recvfrom error");
			return -1;
		}

		unsigned short* ptr1=(unsigned short*)buf;
		if(fwrite((buf+4),(a-4),1,fp)<0)
		{
			printf("fwrite error\n");
			return -1;
		}
		//组ACK包
		*ptr1=htons(4);
		size=2+2;
		//发送ACK
		if(sendto(socfd,buf,size,0,(struct sockaddr *)&cin,sizeof(cin))<0)
		{
			perror("sendto ACK error");
			return -1;
		}
		printf("sendto download  success\n");
		//判断是否小于512字节，若小于则下载完成
		if((a-4)<512)
		{
			printf("下载完成\n");
			break;
		}
	}
}