使用 malloc 分配缓冲区时客户端服务器应用程序中的问题

Question

我正在从事一个包括服务器-客户端通信的项目。 我们的代码并非一直有效，有时它可以完美运行。 但有时我们要么超时，要么我们的缓冲区不能正常工作。 这就是我们要实施malloc()的原因。 你认为这有帮助吗？ 我们在malloc()之前的代码：

#include <stdio.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <string.h>
#include <stdbool.h>
#include <unistd.h>

#define BUFFERSIZE 1024                                              
#define VERSION "VERSION 3.4\n"

#include "functions.h"

char buffer[BUFFERSIZE];

void resetBuffer(char *buffer) {
    memset((buffer), '\0', strlen(buffer));
}

void receiveAnswer(int sock) {
    resetBuffer(buffer);
    size_t length;
    bool x = true;

    while (x) {
        recv(sock, buffer, sizeof(buffer), 0);
        length = strlen(buffer);
        if (buffer[length-1]  == '\n') {
            x = false;
        }
    }

    if (buffer[0] == '-') {
        printf("Error: %s", buffer);
    } else {
        printf("%s\n ", buffer);
    }
}

void sendResponse(int sock, char *message) {
    resetBuffer(buffer);
    strcpy(buffer, message);

    bool x = true;
    size_t length;

    while(x) {
        send(sock, buffer, strlen(buffer), 0);
        length = strlen(buffer);
        if (buffer[length - 1] == '\n') {
            x = false;
        }
    }

    printf("Client: %s\n", buffer);
}

int performConnection(int sock, char *gameID) {
    receiveAnswer(sock);
    sleep(1);
    receiveAnswer(sock);
    sleep(1);
    sendResponse(sock, VERSION);
    sleep(1);
    receiveAnswer(sock);
    sleep(1);
    sendResponse(sock, gameID);
    sleep(1);
    receiveAnswer(sock);
    sleep(1);
    sendResponse(sock, "PLAYER \n");
    sleep(1);
    receiveAnswer(sock);

    resetBuffer(buffer);

    return 0;
}

我们的malloc()代码不再有效：

#include <stdio.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <string.h>
#include <stdbool.h>
#include <unistd.h>

#define BUFFERSIZE 1024
#define VERSION "VERSION 3.4\n"

#include "functions.h"

char *buffer;

void resetBuffer(char *buffer) {
    memset((buffer), '\0', strlen(buffer));
}

void receiveAnswer(int sock) {
    resetBuffer(buffer);
    size_t length;
    bool x = true;

    while (x) {
        recv(sock, buffer, sizeof(buffer), 0);
        length = strlen(buffer);
        if (buffer[length-1] == '\n') {
            x = false;
        }
    }

    if (buffer[0] == '-') {
        printf("Error: %s", buffer);
    } else {
        printf("%s\n ", buffer);
    }
}

void sendResponse(int sock, char *message) {
    resetBuffer(buffer);
    strcpy(buffer, message);

    bool x = true;
    size_t length;

    while (x) {
        send(sock, buffer, strlen(buffer), 0);
        length = strlen(buffer);
        if (buffer[length - 1] == '\n') {
            x = false;
        }
    }

    printf("Client: %s\n", buffer);
}

int performConnection(int sock, char *gameID) {
    buffer = (char *)malloc(sizeof(char) * BUFFERSIZE);
    receiveAnswer(sock);
    sleep(1);
    receiveAnswer(sock);
    sleep(1);
    sendResponse(sock, VERSION);
    sleep(1);
    receiveAnswer(sock);
    sleep(1);
    sendResponse(sock, gameID);
    sleep(1);
    receiveAnswer(sock);
    sleep(1);
    sendResponse(sock, "PLAYER \n");
    sleep(1);
    receiveAnswer(sock);

    resetBuffer(buffer);
    free(buffer);
    return 0;
}

任何帮助表示赞赏！ 最佳恩诺

Answer 1

resetBuffer(buffer); 失败，因为它尝试对未初始化的数据进行strlen(buffer) 。 这会调用未定义行为(UB)，因为strlen()需要一个指向string的指针。

反而：

// resetBuffer(buffer);
memset(buffer, 0, sizeof(char)* BUFFERSIZE);
// or simply
memset(buffer, 0, BUFFERSIZE);

Answer 2

这个

void resetBuffer(char* buffer){
   memset((buffer), '\0', strlen(buffer));
}

是未定义的行为。 这取决于缓冲区的先前内容（strlen 寻找 0 终止符）

你需要传递一个长度（不是从 strlen 得到的）

Answer 3

你为什么要为 function reset_buffer()而烦恼呢？

例如，每次调用都紧接在strcpy()之前。 后者不关心缓冲区是否已“重置”。 （希望缓冲区足够大以容纳放入其中的内容。）

最后一次调用出现在free()之前。 同样，缓冲区的内容无关紧要。

更令人担心的是buffer[]变成了*buffer ... receive function 中的sizeof(buffer)现在告诉recv有 8 个字节可用（你机器上指针的大小）要填充，而不是“pre-malloc”版本的 1024 字节。

改变：

void receiveAnswer (int sock){
// ...
        recv(sock, buffer, sizeof(buffer), 0);

到

        recv(sock, buffer, BUFFERSIZE, 0);

这不是 MRE，因此这个“修复”虽然正确，但可能不是完整的解决方案。

Answer 4

buffer不一定包含正确的 C 字符串，因此使用strlen计算接收到的字节数是不正确的：改用recv的返回值。 同样，缓冲区清除 function 应该获取长度作为参数或使用BUFFER_SIZE 。

buffer是定义为全局字节数组还是在使用前从堆中分配并不重要，您的代码与实现malloc function 无关。