jpeg 压缩中的内存泄漏。错误还是我的错误？

Question

我写了一个 npm module for capturing webcam input on linux. The captured frame in yuyv 格式转换为 rgb24 并压缩为 jpeg 图像。在 jpeg 压缩中似乎存在内存泄漏。所以内存的使用不断增加。

 Image* rgb24_to_jpeg(Image *img, Image *jpeg) { // img = RGB24
    jpeg_compress_struct cinfo;
    jpeg_error_mgr jerr;
    cinfo.err = jpeg_std_error(&jerr);
    jerr.trace_level = 10;
    jpeg_create_compress(&cinfo);

    unsigned char *imgd = new unsigned char[img->size];
    long unsigned int size = 0;
    jpeg_mem_dest(&cinfo, &imgd, &size);

    cinfo.image_width = img->width;
    cinfo.image_height = img->height;
    cinfo.input_components = 3;
    cinfo.in_color_space = JCS_RGB;
    jpeg_set_defaults(&cinfo);

    jpeg_set_quality(&cinfo, 100, true);
    jpeg_start_compress(&cinfo, true);
    int row_stride = cinfo.image_width * 3;
    JSAMPROW row_pointer[1];
    while (cinfo.next_scanline < cinfo.image_height) {
      row_pointer[0] = &img->data[cinfo.next_scanline * row_stride];
      jpeg_write_scanlines(&cinfo, row_pointer, 1);
    }
    jpeg_finish_compress(&cinfo);
    jpeg_destroy_compress(&cinfo);
//    size += 512; // TODO: actual value to expand jpeg buffer... JPEG header?
    if (jpeg->data == NULL) {
      jpeg->data = (unsigned char *) malloc(size);
    } else {
      jpeg->data = (unsigned char *) realloc(jpeg->data, size);
    }
    memcpy(jpeg->data, imgd, size);
    delete[] imgd;
    jpeg->size = size;
    return jpeg;
  }

rgb24 和 jpeg 缓冲区在每个循环中重新分配。所以看起来泄漏是在 libjpeg 层内部。这是真的还是我只是在代码中的某处犯了错误？

注意：压缩后的图片不能保存为文件，因为数据可能会用于直播。

Answer 1

您以错误的方式使用了 jpeg_mem_dest - 第二个参数是指向 char 的指针，因为它实际上是由库设置的，然后您必须在完成后释放它。现在你用一个指针初始化它，它被覆盖并且你释放了库分配的内存区域但是原始内存区域被泄漏了。

这是你应该如何改变你的功能：

Image* rgb24_to_jpeg(Image *img, Image *jpeg) { // img = RGB24
    jpeg_compress_struct cinfo;
    jpeg_error_mgr jerr;
    cinfo.err = jpeg_std_error(&jerr);
    jerr.trace_level = 10;
    jpeg_create_compress(&cinfo);

    unsigned char *imgd = 0;
    long unsigned int size = 0;
    cinfo.image_width = img->width;
    cinfo.image_height = img->height;
    cinfo.input_components = 3;
    cinfo.in_color_space = JCS_RGB;
    jpeg_set_defaults(&cinfo);
    jpeg_set_quality(&cinfo, 100, true);
    jpeg_mem_dest(&cinfo, &imgd, &size); // imgd will be set by the library
    jpeg_start_compress(&cinfo, true);
    int row_stride = cinfo.image_width * 3;
    JSAMPROW row_pointer[1];
    while (cinfo.next_scanline < cinfo.image_height) {
        row_pointer[0] = &img->data[cinfo.next_scanline * row_stride];
        jpeg_write_scanlines(&cinfo, row_pointer, 1);
    }
    jpeg_finish_compress(&cinfo);
    jpeg_destroy_compress(&cinfo);
    //    size += 512; // TODO: actual value to expand jpeg buffer... JPEG header?
    if (jpeg->data == NULL) {
        jpeg->data = (unsigned char *) malloc(size);
    } else if (jpeg->size != size) {
        jpeg->data = (unsigned char *) realloc(jpeg->data, size);
    }
    memcpy(jpeg->data, imgd, size);
    free(imgd); // dispose of imgd when you are done
    jpeg->size = size;
    return jpeg;
}

此代码段形式 jpeg_mem_dest 解释了内存管理：

  if (*outbuffer == NULL || *outsize == 0) {
    /* Allocate initial buffer */
    dest->newbuffer = *outbuffer = (unsigned char *) malloc(OUTPUT_BUF_SIZE);
    if (dest->newbuffer == NULL)
      ERREXIT1(cinfo, JERR_OUT_OF_MEMORY, 10);
    *outsize = OUTPUT_BUF_SIZE;
  }

因此，如果您传递空指针或大小为零的缓冲区，库将为您执行分配。因此 - 另一种方法也是正确设置大小然后你可以使用最初提供的指针

Answer 2

在我的例子中，我没有用之前的回答解决问题，没有办法释放内存图像指针，唯一的方法是为图像保留足够的内存，这样库就不会保留内存，我可以控制内存并且在我的应用程序的同一个堆上而不是在库的堆上，这是我的例子：

//previous code...
struct jpeg_compress_struct cinfo;

//reserving the enough memory for my image (width * height)
unsigned char* _image = (unsigned char*)malloc(Width * Height);

//putting the reserved size into _imageSize
_imageSize = Width * Height;

//call the function like this:
jpeg_mem_dest(&cinfo, &_image, &_imageSize);
................
//releasing the reserved memory
free(_image);

注意：如果您将 _imageSize = 0，库将假设您没有保留内存，而自己的库会这样做。因此您需要将 _imageSize 中的字节数放入 _image

这样你就可以完全控制保留的内存，你可以在你的软件中随时释放它..

jpeg 压缩中的内存泄漏。错误还是我的错误？

Memory leak in jpeg compression. Bug or my mistake?

c++

libjpeg