Question

在C ++中执行此代码时，出现错误malloc(): memory corruption。基本上，我打开一个内核文件，并使用大小为struct stat st的malloc。我猜这是造成问题的原因。

该代码加载了内核模块（I2C），并且实际上正在加载。但是我想我没有使用malloc()，应该使用。谢谢。

#define _GNU_SOURCE
#include <fcntl.h>
#include <stdio.h>
#include <sys/stat.h>
#include <sys/syscall.h>
#include <sys/types.h>
#include <unistd.h>
#include <stdlib.h>

#include <gtest/gtest.h>
#include <gmock/gmock.h>

#define init_module(mod, len, opts) syscall(__NR_init_module, mod, len, opts)
#define delete_module(name, flags) syscall(__NR_delete_module, name, flags)
class I2CKernelModule : public testing::Test {
public:
    I2CKernelModule() {
    }
};
TEST_F(I2CKernelModule, TestAddAndRemoveKernelModule) {
    char *params;
    int fd;
    size_t image_size;
    struct stat st;
    void *image;

    // command: sudo insmod /root/i2c-tests/i2c-stub.ko chip_addr=0x20
    params = "chip_addr=0x20";
    fd = open("/root/i2c-tests/i2c-stub.ko", O_RDONLY);
    fstat(fd, &st);
    image_size = st.st_size;
    image = malloc(image_size);
    read(fd, image, image_size);
    close(fd);
    if (init_module(image, image_size, params) != 0) {
        perror("init_module");
        GTEST_FAIL();
    }
    free(image);
    GTEST_SUCCESS_("Kernel module loaded.");

    /*
    // sudo rmmod i2c_stub
    if (delete_module("i2c_stub", O_NONBLOCK) != 0) {
        perror("delete_module");
        GTEST_FAIL();
    }
    GTEST_SUCCESS_("Kernel module unloaded.");
    */
}

Answer 1

检查所有函数的返回值是否有错误。如果文件未打开，统计信息失败或malloc失败，则列出的代码将失败。检查read返回的字节数也是一个好主意。

尝试加载内核模块时，如何解决C ++中的malloc（）错误？

1 个答案: