tftp上传下载功能实现

#include <stdio.h> 
#include <sys/socket.h>
#include <sys/ipc.h> 
#include <sys/types.h>
#include <netinet/in.h> 
#include <arpa/inet.h> 
#include <string.h> 
#include <unistd.h> 
#include <fcntl.h>
#include<sys/stat.h>
#define ERR_MSG(msg) do{\
	fprintf(stderr, "__%d__:", __LINE__);\
	perror(msg);\
}while(0) 
#define PORT 69 
#define IP "192.168.31.206" 

//下载
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[]) 
{
	//创建报式套接字 
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{ 
		ERR_MSG("socket");
		return -1; 
	}
	//绑定发送方自身的IP和端口 --->非必须绑定

	//因为要发送给接收方，所以要先填充接收方的地址信息结构体 
	struct sockaddr_in sin; 
	sin.sin_family = AF_INET; 
	sin.sin_port = htons(PORT); 
	//接收端的端口号
	sin.sin_addr.s_addr = inet_addr(IP); 
	//
	char c = 0; 
	while(1) { 
		printf("****************\n");
		printf("******1.下载*********\n");
		printf("******2.上传*********\n");
		printf("******3.退出*********\n");
		printf("****************\n");
		printf("请输入(1,2,3)>>>");
		c = getchar();
		while(getchar()!= 10);
		//循环接收字符，直到\n取值后！=10为假，退出循环
		switch(c) {
		case '1':
			//下载
			do_download(sfd,sin);
			break; 
		case '2':
			//上传
			do_upload(sfd,sin); 
			break;
		case '3':
			goto END;
		default:
			printf("输入错误，请重新输入\n");
		}
	} 
END:
	close(sfd);
	return 0;
}
int do_download(int sfd,struct sockaddr_in sin)
{
	//buf数组存放数据包（操作码，文件名+0+模式+0）
	char buf[516]="";
	char* ptr =buf;
	unsigned short* ptr1=(unsigned short*)ptr;
	*ptr1=htons(1);
	char *ptr2=ptr+2;
	//定义指针存放要下载的文件名
	char* filename=NULL;
	scanf("%s",filename);
	//	printf("filename=%s",*filename);
	strcpy(ptr2,filename);
	char* ptr3=ptr2+strlen(ptr2);
	*ptr3=0;
	char* ptr4=ptr3+1;
	strcpy(ptr4,"octet");
	int size=2+strlen(ptr2)+1+strlen(ptr4)+1;

	//发送下载请求
	if(sendto(sfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1; 
	}
	printf("sendto success\n");
	//打开文件

	int fp=open("./download",O_WRONLY|O_CREAT|O_TRUNC,0664);
	if(fp<0)
	{
		ERR_MSG("open");
	}
	struct sockaddr_in rec;
	socklen_t len = sizeof(rec);
	while(1)
	{
		//循环接收服务器返回的数据包
		bzero(buf,sizeof(buf));
		ssize_t res=recvfrom(sfd,buf,sizeof(buf),0,(struct sockaddr*)&rec,&len);
		//分析是数据包还是错误包
		//如果是数据包，将数据包提取出来，数据在buf+4
		char *p1=buf+4;
		char *p2=buf+2;
		unsigned short* p3=(unsigned short*)p2;
		char *buf1=buf;
		unsigned short* buf2=(unsigned short*)buf1;
		if(3==ntohs(*buf2))
		{
			if(write(fp,p1,res-4)<0)
			{
				ERR_MSG("write");
				return -1;
			}


			//ACK包，回复给服务器
			char str[20]="";
			char* str1=str;
			unsigned short* str2=(unsigned short*)str1;
			*str2=htons(4);
			char* str3 = str2+2;
			unsigned short* str4=(unsigned short*)str3;
			*str4=*p3;
			if(sendto(sfd,str,4,0,(struct sockaddr*)&rec,sizeof(rec))<0)
			{
				ERR_MSG("sendto");
				return -1;
			}

			//直到数据内容的长度小于512字节
			if(res<516)
			{
				break;
			}
		}

		//错误包则打印错误信息
		else
		{
			printf("%s\n",p1);
		}
	}
	return 0;
}

int do_upload(int sfd, struct sockaddr_in sin)
{
	char arr[1024] = ""; 	//初始化字符串
	//组下载请求包
	char* brr = arr;
 
	unsigned short* brr1 = (unsigned short*)brr;
	*brr1 = htons(2);  
	char* brr2 = arr+2;
	char buf5[20]="";
	//终端获取文件名
	printf("请输入文件名>>");
	fgets(buf5, sizeof(buf5),stdin);
	buf5[strlen(buf5)-1]=0;
	strcpy(brr2, buf5);
 
	char* brr3 = brr2+strlen(brr2);
	*brr3 = 0;
 
	char* brr4 = brr3+1;
	strcpy(brr4, "octet");
 
	int size = 2+strlen(brr2)+1+strlen(brr4)+1;
 
	//发送上传请求包
	if(sendto(sfd, arr, size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	bzero(arr, sizeof(arr));
	struct sockaddr_in rec;
	socklen_t size4 =  sizeof(rec);
	if(recvfrom(sfd, arr, sizeof(arr), 0,(struct sockaddr*)&rec,&size4)<0)
	{
		ERR_MSG("recvfrom");
		return -1;
	}
	printf("发送上传请求包成功\n");
 
	int mp = open("./fp.png", O_RDONLY);
	ssize_t ze;
	char drr[512]="";
 
	//组上传请求包
	//发送上传请求
		int num=1;
 
	    char arr8[128]="";
	while(1)
	{
		//组建发送包
		bzero(arr, sizeof(arr));
		char* p=arr;
		unsigned short* p1 = (unsigned short*)p;
		*p1 = htons(3);    
		unsigned short* p2 = p1+1;
		*p2 = htons(num);     
		ze = read(mp,drr, sizeof(drr));
		if(ze<0)
		{
			ERR_MSG("read");
			return -1;
		}
		else if(0==ze)
		{
			printf("读取完毕\n");
			break;
		}
		char* p3=arr+4;
		strcpy(p3, drr);       //文件内容
		int size3=4+strlen(p3);
		if(sendto(sfd, arr, ze+4, 0, (struct sockaddr*)&rec, sizeof(rec)) < 0)
		{
			ERR_MSG("sendto");
			return -1;
		}
	//	接收ACK应答包
		if(recvfrom(sfd, arr8, sizeof(arr8), 0,(struct sockaddr*)&rec, &size4)<0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
 
		//如果发送数据包不成功， 接收打印错误报信息
		char* str6 =arr8;
		unsigned short* str7 =(unsigned short*)str6;
		if(*str7==5)
		{
			printf("%s\n", arr8+4);
			
		}
		num ++;
 
	}
	return 0;
}