网络编程day2--tftp协议

TFTP通信过程总结

1. 服务器在69号端口等待客户端的请求

2. 服务器若批准此请求，则使用 临时端口 与客户端进行通信。

3. 每个数据包的编号都有变化（从1开始）

4. 每个数据包都要得到ACK的确认，如果出现超时，则需要重新发送最后的数据包或ACK包

5. 数据长度以512Byte传输的，小于512Byte的数据意味着数据传输结束。

#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<unistd.h>
#include<string.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <fcntl.h>

//打印错误信息的宏函数
#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__",__LINE__);\
	perror(msg);\
}while(0);
int main(int argc, const char *argv[])
{
	if(argc<3)
	{
		fprintf(stderr,"请输入filename\n");
		return -1;
	}
	//创建报式套接字
	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(69);
	sin.sin_addr.s_addr=inet_addr("192.168.31.210");
	//绑定服务器的地址信息结构体-->非必须绑定

	//创建读写请求数据包
	char buf[128]="";
	char* ptr=buf;
	short int* pa=(short int*)ptr;
	*pa=htons(1);
	char* pb=ptr+2;
	char filename[128]="";
	strcpy(filename,argv[1]);
	strcpy(pb,filename);
	char* pc=pb+strlen(pb);
	char* pd=pc+1;
	strcpy(pd,"octet");
	int size=2+strlen(pb)+1+strlen("octet")+1;
	if(sendto(sfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
		{
			ERR_MSG("sendto");
			return -1;
		}
		printf("sendto success\n");
		int fd=open(argv[2],O_WRONLY|O_CREAT|O_TRUNC,0664);
		if(fd<0)
		{
			ERR_MSG("open");
			return -1;
		}
		int rcv_len;
		unsigned short num=1;
	while(1)
	{
		//接收数据包
		unsigned char buf1[1024]="";
		struct sockaddr_in cin;
		socklen_t addrlen = sizeof(cin);
		if((rcv_len=recvfrom(sfd,buf1,sizeof(buf1),0,(struct sockaddr*)&cin,&addrlen))<0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		if(buf1[1]==3)
		{
			if(num==ntohs(*(unsigned short *)(buf1+2)))
			{
				if(write(fd,buf1+4,rcv_len-4)<0)
                {
                      ERR_MSG("write");
                      return -1;      
                }
				num++;
			}
			//发送ACK包给服务器
			buf1[1]=4;
			if(sendto(sfd,buf1,4,0,(struct sockaddr*)&cin,sizeof(cin))<0)
			{
				ERR_MSG("sendto");
				return -1;
			}
			if(rcv_len<516)
				break;
		}
		
	}
	close(fd);
	//关闭套接字
	close(sfd);
	return 0;
}