2014-04-07 86 views
0

我一直在尝试编写一个为您创建视频的小应用程序,并且您可以发送自己的图像。为了对视频进行编码,我使用了媒体基础库。使用SinkWriter(媒体基础)时发生内存泄漏C++

一切工作正常。我可以创建一个非常短的视频,但是一旦我尝试创建一个大视频,我的应用程序就会开始占用大量内存,并最终导致内存和崩溃。内存在1.5 GB的范围内。

我已经做了一些调试发现,这种方法泄漏内存。尝试大量删除和可用内存组合后,我无法找出问题所在。你能帮我一下吗?这是添加框架的方法。

HRESULT VideoCompressor::addFrame(BYTE* bitmapBytes) 
{ 
    const long imageBufferWidth = VIDEO_WIDTH * 4; 
    const DWORD bufferLength = imageBufferWidth * VIDEO_HEIGHT; 
    IMFSample *sample = NULL; 
    BYTE *dataBuffer = NULL; 
    IMFMediaBuffer *sampleBuffer = NULL; 
    DWORD maxLength; 
    //std::shared_ptr<BYTE> *dataBuffer = NULL; 

    if (!SUCCEEDED(MFCreateMemoryBuffer(bufferLength, &sampleBuffer))) 
     return E_FAIL; 

    sampleBuffer->GetMaxLength(&maxLength); 
    if (!SUCCEEDED(sampleBuffer->Lock(&dataBuffer, NULL, NULL))) 
     return E_FAIL; 

    if (MFCopyImage(dataBuffer, imageBufferWidth, 
     (BYTE *) bitmapBytes, imageBufferWidth, 
     imageBufferWidth, VIDEO_HEIGHT) 
     != S_OK) 
     return E_FAIL; 

    sampleBuffer->Unlock(); 
    sampleBuffer->SetCurrentLength(bufferLength); 

    if (!SUCCEEDED(MFCreateSample(&sample))) 
     return E_FAIL; 

    if (!(SUCCEEDED(sample->AddBuffer(sampleBuffer)))) 
     return E_FAIL; 


    if(!(SUCCEEDED(sample->SetSampleTime(startTime)))) 
     return E_FAIL; 

    if(!(SUCCEEDED(sample->SetSampleDuration(durationTime)))) 
     return E_FAIL; 

    if(!(SUCCEEDED(writer->WriteSample(/*streamIndex*/ 0, sample)))) 
     return E_FAIL; 

    startTime = startTime + durationTime; 

    //sample->Release(); 
    //sampleBuffer->Release(); 

    SafeRelease(&sample); 
    SafeRelease(&sampleBuffer); 

    return S_OK; 
} 

SafeRelease的实施是

template <class T> void SafeRelease(T **_object) 
{ 
    if (*_object) 
    { 
     (*_object)->Release(); 
     *_object = NULL; 
    } 
} 

我怀疑它与BYTE做*的DataBuffer。我尝试使用delete []关键字删除它,但它没有任何用处。

+1

如果你的代码是通过整个函数执行并返回S_OK,那么我没有看到它,我不相信它在这个函数中。但是,您应该修正事实,即您有多个退出点而不调用SafeRelease(..)。 – Jeff

+0

非常非常天真的我......我在这个问题上花了太多时间,我看不到那显而易见的......你非常杰夫,它现在工作,没有泄漏。 – virusrocks

+0

听起来不错。不要忘记标记为已回答(复选标记)。 – Jeff

回答

3

回答@jeff

如果你的代码是通过整个函数执行并返回S_OK,那么我没有看到它,我不相信这是在这个函数。但是,您应该修正事实,即您有多个退出点而不调用SafeRelease(..)。