【TFTP】客户端下载文件代码

// 客户端下载文件
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
int main(int argc, char const *argv[])
{
    // 1.创建套接字
    int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sockfd < 0)
    {
        perror("socket");
        return -1;
    }
    // 2.创建文件用来保存下载文件的内容
    int fd = open("a.txt", O_RDWR | O_CREAT, 0777);
    // 3.指定服务器地址
    struct sockaddr_in tftp_server_addr;
    tftp_server_addr.sin_family = AF_INET;
    tftp_server_addr.sin_port = htons(69);
    tftp_server_addr.sin_addr.s_addr = inet_addr("10.9.42.228");
    // 4.组请求+发送请求
    char buf[128] = "";
    int buflen = 0;
    short id = 0;
    buflen = sprintf(buf, "%c%c%s%c%s%c", 0, 1, "a.txt", 0, "netascii", 0);
    sendto(sockfd, buf, buflen, 0, (struct sockaddr *)&tftp_server_addr, sizeof(tftp_server_addr));
    // 5.循环收取数据
    char recvbuf[1024] = "";
    struct sockaddr_in tmp_server_addr;
    socklen_t tmp_len = sizeof(tmp_server_addr);
    while (1)
    {
        // 5.1清空数据内容以及地址结构
        bzero(recvbuf, sizeof(recvbuf));
        bzero(&tmp_server_addr, tmp_len);
        // 5.2接收数据
        buflen = recvfrom(sockfd, recvbuf, sizeof(recvbuf), 0, (struct sockaddr *)&tmp_server_addr, &tmp_len);
        // 5.3判断
        if (recvbuf[1] == 5)
        {
            printf("差错消息：%s\n", recvbuf + 4);
            break;
        }
        else
        {
            recvbuf[1] = 4;
            sendto(sockfd, recvbuf, 4, 0, (struct sockaddr *)&tmp_server_addr, tmp_len);
            // 将内容保存文件
            write(fd, recvbuf + 4, buflen - 4);
            // 提取数据包的编号
            id = ntohs(*((unsigned short *)(recvbuf + 2)));
            printf("收到%d数据包 数据长度%d\n", id, buflen - 4);
            // 如果数据小于512，数据包长度小于516，传输结束
            if (buflen < 516)
                break;
        }
    }
    // 6.关闭套接字
    close(sockfd);
    close(fd);
    return 0;
}