8.10作业

完成TFTP 下载功能代码

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



int main(int argc, const char *argv[])
{
	if(argc!=3)
	{
		printf("./a.out server_ip file_name\n");
		return 0;
	}

	//创建socket套接字（基于UDP)
	int socket_fd = socket(AF_INET, SOCK_DGRAM,0);
	if(socket_fd<0)
	{
		perror("socket");
		return 0;
	}

	unsigned char cmd[128]="";
	int len = sprintf(cmd,"%c%c%s%c%s%c",0x00,0x01,argv[2],0,"octet",0);

    //将请求cmd发送到服务器69端口
	struct sockaddr_in server;
	bzero(&server,sizeof(server));
	server.sin_family = AF_INET;
	server.sin_port = htons(69);
	inet_pton(AF_INET,argv[1],&server.sin_addr.s_addr);

	sendto(socket_fd,cmd,len,0,(struct sockaddr*)&server,sizeof(server));

	//打开一个本地空文件，将得到的数据写进去
	int fd=open(argv[2],O_WRONLY|O_CREAT,0666);
	if(fd < 0)
	{
		perror("open");
		return 0;
	}

	//不停读取服务器传过去的文件数据
	unsigned short num=0;

	while(1)
	{
		unsigned char buf[1024]="";
		struct sockaddr_in from;
		socklen_t from_len = sizeof(from);
		int len = recvfrom(socket_fd,buf,sizeof(buf),0,(struct sockaddr *)&from, &from_len);

		//判断收到的数据的操作码 （必须是00 03表示文件数据)
		if(buf[1]==0x03)
		{
			//将文件数据写入本地文件中
			//防止写入重复数据
			if((num+1) == ntohs(*(unsigned short *)(buf+2)))
			{
				write(fd,buf+4,len-4);
				num = ntohs(*(unsigned short *)(buf+2));
				printf("recv:%d\n",num);
			}

			//给服务器发送ACK回应
			buf[1]=4;
			sendto(socket_fd,buf,4,0,(struct sockaddr *)&from,sizeof(from));

			if(len<516)
			{
				break;
			}
		}
	}

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

结果