网络编程——基于UDP的TFTP文件传输

一、基于UDP的TFTP文件传输

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<sys/stat.h>
#include<netinet/in.h>
#include<errno.h>
#include<fcntl.h>
#define ERR_MSG(msg)do{\
	perror(msg);\
	printf("%d\n",__LINE__);\
}while(0)
#define PORT 69
#define N      516

int do_download(int sfd,struct sockaddr_in sin);
int do_upload(int sfd,struct sockaddr_in sin);

int main(int argc, const char *argv[])
{
	if(argc<2)
	{
		printf("请输入IP\n");
		return -1;
	}
	//创建报式套接字
	int sfd=socket(AF_INET,SOCK_DGRAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	//服务器的地址信息
	struct sockaddr_in sin;
	sin.sin_family  =AF_INET;
	sin.sin_port    =htons(PORT);
	sin.sin_addr.s_addr   =inet_addr(argv[1]);


	char chose;
	while(1)
	{
		system("clear");
		printf("*******************\n");
		printf("*********1 上传****\n");
		printf("*********2 下载****\n");
		printf("*********3 退出****\n");
		scanf("%c",&chose);
		while(getchar()!=10);

		switch(chose)
		{
		case '1': 
			//下载功能
			do_download(sfd,sin);
			break;
		case'2':
			//上传功能
			do_upload(sfd,sin);
			break;
		case'3':
			//退出
			goto END;
			break;
		default:
			printf("输出错误\n");
		}
		printf("输入任意键清屏------>");
		while(getchar()!=10);

	}
END:
	close(sfd);
	return 0;
}
//下载功能
int do_download(int sfd,struct sockaddr_in sin)
{
	char filename[20]="";
	printf("请输入要下载的文件\n");
	fgets(filename,20,stdin);
	filename[strlen(filename)-1]=0;

	//发送下载请求
	char buf[N]="";
	int size=sprintf(buf,"%c%c%s%c%s%c",0,2,filename,0,"ocect",0);

	if(sendto(sfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	int flag=0;
	int fd=-1;
	//循环接受发送应答包
	socklen_t addrlen=sizeof(sin);
	ssize_t recv_len;
	unsigned short num=1;
	while(1)
	{
		bzero(buf,N);
		recv_len=recvfrom(sfd,buf,N,0,(struct sockaddr*)&sin,&addrlen);
		if(recv_len<0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		if(buf[1]==3)
		{
			if(flag==0)
			{
				fd=open(filename,O_WRONLY|O_CREAT|O_TRUNC,0664);
				if(fd<0)
				{
					ERR_MSG("open");
					return -1;
				}
				flag=1;

			}
		}
		if(htons(num)==*(unsigned short*)(buf+2))
		{
			if(write(fd,buf+4,recv_len-4)<0)
			{
				printf("fd:%d recv_len:%ld\n",fd,recv_len);
				ERR_MSG("write");
				return -1;
			}
			buf[1]=4;
			if(sendto(sfd,buf,4,0,(struct sockaddr*)&sin,sizeof(sin))<0)
			{
				ERR_MSG("sendto");
				return -1;
			}
			if(recv_len<512+2+2)
			{
				printf("文件下载完毕\n");
				break;
			}
			num++;
		}
		if(buf[1]==5)
		{
			//打印错误信息
			printf("error %s\n",buf+4);
			break;
		}

	}
	close(sfd);
	return 0;
}
//上传功能
int do_upload(int sfd,struct sockaddr_in sin)
{
	char filename[20]="";
	printf("请输入要上传的文件名>>>");
	fgets(filename,20,stdin);
	filename[strlen(filename)-1]=0;
	//判断文件是否存在
	int fd=open(filename,O_RDONLY);
	if(fd<0)
	{
		if(errno==ENOENT)
		{
			printf("输入的文件不存在\n");
			return -1;
		}
		else
		{
			ERR_MSG("open");
			return -1;
		}
	}

	//发送请求包
	//上传协议
	char buf[N]="";
	int size=sprintf(buf,"%c%c%s%c%s%c",0,2,filename,0,"ocect",0);
	if(sendto(sfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("sendto");
		return -1;
	}

	//循环接受发送数据包
	ssize_t recv_len;
	unsigned short num=0;
	socklen_t addrlen=sizeof(sin);
	while(1)
	{	bzero(buf,N);
		recv_len=recvfrom(sfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);
		if(recv_len<0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		//操作码的范围是1-5，因为是网络字节序
		//所以有效存储码，存储在高位buf[1]位置
		if(buf[1]==4)
		{
			if(num=ntohs(*(unsigned short*)(buf+2)))
			{
				buf[1]=3;
				num++;
				*(unsigned short*)(buf+2)=htons(num);
				//读取数据
				//发送数据
				int res=read(sfd,buf+4,N-4);
				if(res<0)
				{
					ERR_MSG("read");
					return -1;
				}
				if(res==0)
				{
					printf("文件上传完毕\n");
				}
				//发送数据包
				if(sendto(sfd,buf,sizeof(buf+4),0,(struct sockaddr*)&sin,sizeof(sin))<0)
				{
					ERR_MSG("sendto");
					return -1;
				}
			}
			else
			{
				printf("文件上传失败\n");
				break;
			}
		}
		if(buf[1]==5)
		{
			printf("error %s\n",buf+4);
			break;
		}
	}
	return 0;

}