5.17作业

//TFTPcli
#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 BUFFER_SIZE 516
#define ERR_MSG(msg) do{\
    printf("line: %d\n", __LINE__); \
    perror(msg);\
}while(0)                                                                                          

#define PORT 69   
#define IP "192.168.10.128" //ifconfig查看本机IP


int main(int argc, const char *argv[])
{
    //创建报式套接字
    int cfd = socket(AF_INET, SOCK_DGRAM, 0);
    if(cfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("报式套接字创建成功 cfd=%d __%d__\n", cfd, __LINE__);
	//填充服务器自身的地址信息结构体
    //给下面的sendto函数使用
    struct sockaddr_in sin;
    sin.sin_family      = AF_INET;
    sin.sin_port        = htons(PORT);
    sin.sin_addr.s_addr = inet_addr(IP);

	//组下载请求协议
    char buf[BUFFER_SIZE] = "";
	short* ptr1=(short*)buf;
	*ptr1=htons(1);//主机字节序转网络字节序\操作码1，下载（RD）\操作码2，上传（WR）
		
	char* ptr2=(char*)(ptr1+1);//char* ptr2=buf+2;
	strcpy(ptr2,"5.png");//要下载的文件名

	char* ptr3=ptr2+strlen(ptr2);//char* ptr3=buf+2+strlen(ptr2);
	*ptr3=0;

	char* ptr4=ptr3+1;
	strcpy(ptr4,"octet");//模式

	int size=2+strlen(ptr2)+1+strlen(ptr4)+1;

	//发送数据给服务器，所以需要填服务器的地址信息结构体；
	if(sendto(cfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	printf("发送成功\n");

	int fd=open("./copy.png",O_WRONLY|O_CREAT|O_TRUNC,0664);
	//接收来自服务器的数据包
	char buf1[BUFFER_SIZE]="";
	int block_id=1;//块编号
	while(1)
	{
		struct sockaddr_in recvaddr;
		socklen_t recvlen=sizeof(recvaddr);
	//	int read_bytes=0;
		int bytes_received=recvfrom(cfd,buf1,BUFFER_SIZE,0,(struct sockaddr*)&recvaddr,&recvlen); 
		if(bytes_received<0)
		{
			ERR_MSG("recvfrom");
		}
		//获取操作码
		short opcode=ntohs(*(short*)buf1);
		switch(opcode)
		{
		case 3:{//读取数据包
				   short data_block_id=ntohs(*(short*)(buf1+2));
				   if(data_block_id!=block_id){
					   break;
				   }
				   int data_len=bytes_received-4;
				   write(fd,buf1+4,data_len);

				   //ACK
				   char buf_ack[4]="";
				   *(short*)buf_ack=htons(4);
				   *(short*)(buf_ack+2)=htons(block_id);//块编号

				   //发送ACK给服务器
				   if(sendto(cfd, buf_ack, 4, 0, (struct sockaddr*)&recvaddr, sizeof(recvaddr)) < 0)
				   {
					   ERR_MSG("sendto");
					   return -1;
				   }
				   block_id++;
				   printf("ACK发送成功\n");
				   break;
			   }

		case 5:{//读取ERROR
				   short err_code=ntohs(*(short*)(buf1+2));//差错码
				  // char err_note[]=*(char*)(buf1+4);//差错信息
				   printf("err_code=%d:%s\n",err_code,buf1+4);
				   break;
			   }
		default://忽略其他
			   break;
		}
		if(bytes_received<BUFFER_SIZE)
		{
			printf("读完\n");
			break;
		}
	}

	//关闭套接字文件描述符
	close(cfd);
	//关闭拷贝文件描述符
	close(fd);
	return 0;
}