完成tftp文件传输客户端

#include <stdio.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <string.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>
#define PORT 69
#define IP "192.168.8.242"
#define ERR_MSG(msg) do{\
	fprintf(stderr,"%d:"__LINE__);\
	perror(msg);\
}while(0)
int download(int sfd,struct sockaddr_in sin);
int main(int argc, const char *argv[])
{
	int sfd=socket(AF_INET,SOCK_DGRAM,0);

	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(PORT);
	sin.sin_addr.s_addr=inet_addr(IP);

    download(sfd,sin);
	upload();
	
	return 0;
}
int download(int sfd,struct sockaddr_in sin){
	char buf[516]={0};
	char filename[10]={0};
	printf("请输入需要下载的文件名\n");
	scanf("%s",filename);
	/*short* p1=(short*)buf;
	 *p1=htons(1);
	 char* p2=buf+2;
	 strcpy(p2,filename);
	 char* p3=p2+strlen(filename);
	 *p3=0;
	 char* p4=p3+1;
	 strcpy(p4,"octet");
	 char* p5=p4+strlen("octet");
	 *p5=0;
	 int size=2+strlen(p2)+1+strlen(p4)+1;*/
	int size=sprintf(buf,"%c%c%s%c%s%c",0,1,filename,0,"octet",0);
	if(sendto(sfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0){
		ERR_MSG("sendto");
	}
	printf("sendto success\n");
	int fd=-1,flag=0;
	//记录数据包编号
	unsigned short num=0;
	socklen_t addrlen=sizeof(cin);
	ssize_t res=0;
	//接收数据
	while(1){
		bzero(buf,sizeof(buf));
		res=recvfrom(sfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);
		if(res<0){
			ERR_MSG("recvfrom");
			return -1;
		}
		if(3==buf[1]){
			if(htons(num+1)==*(unsigned short*)(buf+2)){
				num++;
				if(0==flag){
					fd=open(filename,O_WRONLY|o_CREAT|O_TRUNC,0664);
					if(fd<0){
						ERR_MSG("open");
						return -1;
					}
					flag=1;
				}
			}
			if(write(fd,buf+4,res-4)<0){
				ERR_MSG("write");
				return -1;
			}
			//回复ack
			buf[1]=4;
			if(sendto(sfd,buf,4,0,(struct sockaddr*)&sin,addrlen)<0){
				ERR_MSG("sendto");
				return -1;
			}
			//判断是否下载完毕
			if(res-4<512){
				printf("%s:下载完毕\n",filename);
				break;
			}
		}else if(5==buf[1]){
			fprintf(stderr,"错误码:%d 错误信息:%s\n",\
					ntohs(*(unsigned short*)(buf+2)),buf+4);
			break;
		}
	}
	close(fd);
	return 0;
}