网络编程day4

1.思维导图

2.2.基于UDP的TFTP文件传输的客户端

#include <myhead.h>
#define SER_PORT 69
#define SER_IP "192.168.122.25"
int upload(char* name)
{
	//以只读的形式打开文件
	int fd=-1;
	if((fd=open(name,O_RDONLY))==-1)
	{
		perror("open error");
		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(SER_PORT);
	sin.sin_addr.s_addr = inet_addr(SER_IP);
	socklen_t socklen = sizeof(sin);
	//向服务器发送上传请求
	short str1=0;
	short str2=0;
	char buf[516]="";
	//组装请求数据
	short *p1 = (short *)buf;
	*p1 =htons(2);
	char *p2 = buf+2;
	strcpy(p2,name);
	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));
	while(1)
	{
 
		bzero(buf,sizeof(buf));
		//接收ACK包
		int res1=recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&socklen);
		//发送数据包
		short *p1 = (short *)buf;
		*p1 =htons(3);
		short *p2 = (short *)(buf+2);
		str1++;
		*p2=htons(str1);
		char *p3 =buf+4;
		int res=read(fd,p3,sizeof(buf)-4);
		//printf("%d\n",res);
		if(res<512)
		{
			sendto(cfd,buf,res+4,0,(struct sockaddr*)&sin,sizeof(sin));
			break;
		}
		sendto(cfd,buf,516,0,(struct sockaddr*)&sin,sizeof(sin));
		//printf("%d\n",res);
	}
	//关闭套接字
	close(cfd);
	//关闭文件指针
	close(fd);
	return 0;
 
 
}
int download(char* name)
{
	//以写的形式打开文件
	int fd=-1;
	if((fd=open(name,O_WRONLY|O_CREAT|O_TRUNC,0664))==-1)
	{
		perror("open error");
		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(SER_PORT);
	sin.sin_addr.s_addr = inet_addr(SER_IP);
	socklen_t socklen = sizeof(sin);
	//向服务器发送下载请求
	short str=0;
	char buf[516]="";
	//组装请求数据
	short *p1 = (short *)buf;
	*p1 =htons(1);
	char *p2 = buf+2;
	strcpy(p2,name);
	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));
	while(1)
	{
		bzero(buf,sizeof(buf));
		//向后移动指针，读取数据
		lseek(cfd,4,SEEK_SET);
		int res=recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&socklen);
		char *sbuf=buf+4;
		write(fd,sbuf,res-4);
	//	printf("%d\n",res);
		//判断是否拷贝完成
		if(res<516)
		{
			break;
		}
		bzero(buf,sizeof(buf));
		short *p1 = (short *)buf;
		*p1 =htons(4);
		short *p2 = (short *)(buf+2);
		str++;
		*p2=htons(str);
	//	printf("%d\n",str);
		len =4;
		//发送ACK,继续拷贝
		sendto(cfd,buf,len,0,(struct sockaddr*)&sin,sizeof(sin));
	}
	//关闭套接字
	close(cfd);
	//关闭文件指针
	close(fd);
	return 0;
 
}
 
int main(int argc, const char *argv[])
{
	//创建菜单目录
	while(1)
	{
		printf("********选择对文件功能*********\t\n");
		printf("*************1.上传*********\t\n");
		printf("*************2.下载****************\t\n");
		printf("*************3.退出****************\t\n");
		int news;
		printf("请输入你要执行的功能\n");
		scanf("%d",&news);
		getchar();
		char name[128]="";
		char *file =name;
		switch(news)
		{
		case 1:
			printf("请输入要上传的文件名\n");
			fgets(name,sizeof(name),stdin);
			name[strlen(name)-1]=0;
			upload(file);
			break;
		case 2:
			printf("请输入要下载的文件名\n");
			fgets(name,sizeof(name),stdin);
			name[strlen(name)-1]=0;
			download(file);
			break;
		case 3:
			goto END;
		default:
			printf("输入内容不在选项内，请重新输入\n");
		}
		printf("请按任意键清屏！！！\n");
		getchar();
		system("clear");
	}
END:
	return 0;
}