作业11.21

1.基于UDP的TFTP文件传输

#include <myhead.h>
int do_download(int cfd,struct sockaddr_in sin)
{
	//存储文件名
	char buf[516]="";
	char filename[40]="";
	printf("输入文件名:");
	scanf("%s",filename);
	getchar();
	//组装请求包
	short *p1=(short*)buf;
	*p1=htons(1);
	char *p2=buf+2;
	strcpy(p2,filename);
	char *p4=p2+strlen(p2)+1;
	strcpy(p4,"octet");
	int size=4+strlen(p2)+strlen(p4);
	//发送下载请求
	if(sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))==-1)
	{
		perror("sendto error");
		return -1;
	}
	printf("请求成功");
	//创建空文件
	int fd=-1;
	if((fd=open("./test.txt",O_WRONLY|O_CREAT|O_TRUNC,0666))==-1)
	{
		perror("open error");
		return -1;
	}
	//接收信息
	while(1)
	{
		bzero(buf,sizeof(buf));
		struct sockaddr_in cin;
		socklen_t socklen=sizeof(cin);
        //接受目标信息
		int rcv=recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&cin,&socklen);
		write(fd,buf+4,rcv-4);
		//传ASK码
		buf[1]=4;
		sendto(cfd,buf,4,0,(struct sockaddr*)&cin,sizeof(cin));
		if(rcv<516)
			break;

	}
	close(fd);
	close(cfd);
	return 0;

}
int do_updata(int cfd,struct sockaddr_in sin)
{
	//存储文件名
	char buf[516]="";
	char filename[40]="";
	printf("输入文件名:");
	scanf("%s",filename);
	getchar();
	//组装请求包
	short *p1=(short*)buf;
	*p1=htons(2);
	char *p2=buf+2;
	strcpy(p2,filename);
	char *p4=p2+strlen(p2)+1;
	strcpy(p4,"octet");
	int size=4+strlen(p2)+strlen(p4);
	//发送上传请求
	if(sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))==-1)
	{
		perror("sendto error");
		return -1;
	}
	printf("请求成功\n");
	//打开要传文件
	int fd=-1;
	if((fd=open(filename,O_RDONLY))==-1)
	{
		perror("open error");
		return -1;
	}
	int i=1;
	//发送文件
	while(1)
	{
		bzero(buf,sizeof(buf));
		struct sockaddr_in cin;
		socklen_t socklen=sizeof(cin);
		buf[1]=4;
		//接收ASK码
		if(recvfrom(cfd,buf,4,0,(struct sockaddr*)&cin,&socklen)<0)
		{
			perror("recvfrom error");
			return -1;
		}
		buf[1]=3;
		buf[3]=i;
		//从文件中读取内容
		int res=read(fd,buf+4,512);
		if(res<0)
		{
			perror("read error");
			return -1;
		}
		//发送到目的地
		sendto(cfd,buf,516,0,(struct sockaddr*)&cin,sizeof(cin));
		i++;
		if(res<512)
		{
			printf("上传结束\n");
			break;
		}
	}
	close(fd);
	close(cfd);
	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 mune=-1;
	while(1)
	{
		system("clear");
		printf("\t\t======1、下载======\n");
		printf("\t\t======2、上传======\n");
		printf("\t\t======0、退出======\n");
		printf("请输入:");
		scanf("%d",&mune);
		getchar();
		switch(mune)
		{
			case 1:
				{
					do_download(cfd,sin);
				}
				break;
			case 2:
				{
					do_updata(cfd,sin);
				}
				break;
			case 0:
				goto POS;
			default:printf("重新输入:\n");
		}
		printf("按回车键清空:");
		while(getchar()!='\n');
	}
POS:
	close(cfd);
	return 0;
}

2.思维导图