网络编程：5/21

思维导图

工程代码（半成品代码，真的解决不了，直接chatgpt都下载不了文件）

#include <myhead.h>
#define BUFFER_SIZE 516
#define SER_PORT 69
#define SER_IP "192.168.125.90"
#define CLI_IP "192.168.80.128"
#define DATA_SIZE 512
char buf[516]="";
int size;

void send_download_req(int cfd,struct sockaddr_in sin){
	short* p1=(short*)buf;
	*p1=htons(1);

	char* p2=buf+2;
	strcpy(p2,"5.png");

	char* p3=p2+strlen(p2)+1;
	strcpy(p3,"octet");

	size=2+strlen(p2)+2+strlen(p3);
	sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin));
}

void send_ack(int cfd, const struct sockaddr_in *server_addr, uint16_t block_number) {
    char ack_buff[4];
    ack_buff[0] = 0;
    ack_buff[1] = 4;
    ack_buff[2] = block_number >> 8;
    ack_buff[3] = block_number & 0xFF;
    if (sendto(cfd, ack_buff, 4, 0, (struct sockaddr *)server_addr, sizeof(*server_addr)) != 4) {
        perror("sendto");
		return -1;
    }
}

void receive_file(int sockfd, const struct sockaddr_in *server_addr, const char *filename) {
    FILE *file = fopen(filename, "wb");
    if (!file) {
        perror("fopen");
		return -1;
    }

    char buffer[BUFFER_SIZE];
    struct sockaddr_in from_addr;
    socklen_t from_len = sizeof(from_addr);
    uint16_t expected_block = 1;

    while (1) {
        int recv_len = recvfrom(sockfd, buffer, BUFFER_SIZE, 0, (struct sockaddr *)&from_addr, &from_len);
        if (recv_len < 0) {
            perror("recvfrom");
			return -1;
        }

        uint16_t opcode = buffer[1];
        uint16_t block_number = (buffer[2] << 8) | buffer[3];

        if (opcode == 3 && block_number == expected_block) {
            int data_len = recv_len - 4;
            if (fwrite(buffer + 4, 1, data_len, file) != data_len) {
                perror("fwrite");
				return -1;
            }
            send_ack(sockfd, server_addr, block_number);
            expected_block++;
            if (data_len < DATA_SIZE) {
                break; // Last packet
            }
        } else if (opcode == 5) {
            fprintf(stderr, "Error: %s\n", buffer + 4);
            break;
        }
    }

    fclose(file);
}

int main(int argc, const char *argv[])
{
	const char* server_ip=argv[1];

	int cfd=socket(AF_INET,SOCK_DGRAM,0);
	if(cfd==-1){
		perror("socket error");
		return -1;
	}
	printf("cfd=%d\n",cfd);
	
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(SER_PORT);
	if (inet_pton(AF_INET, server_ip, &sin.sin_addr) <= 0) {
    	perror("inet_pton error");
		return -1;
	}

	/*//首先需要先发送一个请求连接包，在循环中接受服务器发送来的包
	//数据包的缓冲区
	send_download_req(cfd,sin);//向tftp服务器发送下载请求包
	int i=1;
	int fd=open("./5.png",O_WRONLY | O_CREAT,0666);
	while(i<10){
		recvfrom(cfd,buf,sizeof(buf),0,NULL,NULL);
		short* p01=buf;
		short* p02=buf+2;
		*p01=htons(4);
		*p02=htons(i++);
		sendto(cfd,buf,4,0,(struct sockaddr*)&sin,sizeof(sin));
	}*/

	
	send_download_req(cfd,sin);
	receive_file(cfd,&sin,"./5.png");

	close(cfd);
	return 0;
}