SIGINT 上的 C++ 程序内存泄漏

Question

我有一个程序，用作套接字客户端，这里是代码

#include <stdio.h>
#include <signal.h>
#include <iostream>
#include <stdlib.h>
#include <string.h>
#include <string>
#include <sys/socket.h>
#include <netdb.h>

using namespace std;

void signalHandler(const int signal) {
    cout << "SIGINT handled" << endl;
    exit(EXIT_SUCCESS);
}

void error(const char *msg) {
    perror(msg);
    exit(0);
}

const string readStr(int descriptor) {
    int n = 0;
    string cmd = "";

    char buffer[256];
    memset(buffer, 0, sizeof(buffer));

    while ((n = read(descriptor, buffer, 255)) != 0) {
        if (n < 0) {
            error("Error reading string");
        }

        // full string - just copy
        if (n == 255) {
            cmd += buffer;
            memset(buffer, 0, sizeof(buffer));
        }
        else {
            cmd += buffer;
            break;
        }
    }

    return cmd;
}

int main(int argc, char** argv) {
    signal(SIGINT, signalHandler);

    int fd, port = (argc >= 3 ? atoi(argv[2]) : 11212), n;
    fd = socket(AF_INET, SOCK_STREAM, 0);
    if (fd < 0) {
        error("ERROR opening socket");
    }

    struct hostent *server = (argc >= 2 ? gethostbyname(argv[1]) : gethostbyname("localhost"));
    if (server == NULL) {
        error("ERROR no such host");
    }

    struct sockaddr_in addr;
    memset(&addr, 0, sizeof(addr));

    addr.sin_family = AF_INET;
    bcopy(server->h_addr, &addr.sin_addr.s_addr, server->h_length);
    addr.sin_port = htons(port);

    if (connect(fd, (struct sockaddr*)&addr, sizeof(addr)) < 0) {
        error("ERROR connecting");
    }

    string tmp;
    while (1) {
        cout << "Please enter the message: \n< " << flush;
        tmp = readStr(0);
        n = send(fd, tmp.c_str(), tmp.size(), 0);
        if (n < 0) {
            perror("ERROR writing to socket");
        }

        tmp = readStr(fd);
        cout << "> " << tmp << endl;
    }
    close(fd);

    return EXIT_SUCCESS;
}

当我将 SIGNINT 发送到进程 (CTRL + C) 时，valgrind 说：

^CSIGINT 已处理 ==3798== ==3798== 堆摘要： ==3798== 在退出时使用：1 个块中的 39 个字节 ==3798== 总堆使用量：45 分配，44 释放，4,778 字节分配 ==3798== ==3798== 1 个块中的 39 个字节可能在丢失记录 1 of 1 中丢失 ==3798== at 0x402471C: operator new(unsigned int) (vg_replace_malloc.c:255) ==3798== by 0x40DBB64: std::string::_Rep::_S_create(unsigned int, 无符号整数，std::allocator const&) (在 /usr/lib/libstdc++.so.6.0.15) ==3798== 由 0x1BFF403: ??? ==3798== ==3798== 泄漏摘要： ==3798== 肯定丢失：0 个块中的 0 个字节 ==3798== 间接丢失：0 个块中的 0 个字节 ==3798== 可能丢失：1 个块中的 39 个字节 ==3798== 仍然可访问：0 个块中的 0 个字节 ==3798== 抑制：0 个块中的 0 个字节 ==3798== ==3798== 对于检测到和抑制的错误计数，重新运行：-v ==3798== 错误摘要：1 个上下文中的 1 个错误（抑制：9 个中的 20 个）

如果程序正确退出，例如如果将while(1) 替换为int i = 0; while(i++ <= 2)，则不会发生内存泄漏

那么有人可以帮我解决这个问题吗？

Answer 1

当您从信号处理程序调用std::exit 时，堆栈不会展开，并且不会破坏main 中的std::string 对象tmp 等局部变量。

由于tmp 在程序终止之前没有被销毁，它分配的内存永远不会被释放，因此泄漏。