check VP8LBitWriterInit return

Change-Id: I460906281598f5792bd75a25b14b449c8daaff8c
This commit is contained in:
James Zern 2012-07-31 12:11:40 -07:00
parent cbfa9eecf4
commit 183cba83a7
2 changed files with 6 additions and 3 deletions

View File

@ -84,8 +84,8 @@ static int EncodeLossless(const uint8_t* const data, int width, int height,
config.quality = 10.f + 15.f * effort_level; config.quality = 10.f + 15.f * effort_level;
if (config.quality > 100.f) config.quality = 100.f; if (config.quality > 100.f) config.quality = 100.f;
VP8LBitWriterInit(&tmp_bw, (width * height) >> 3); ok = VP8LBitWriterInit(&tmp_bw, (width * height) >> 3);
ok = (VP8LEncodeStream(&config, &picture, &tmp_bw) == VP8_ENC_OK); ok = ok && (VP8LEncodeStream(&config, &picture, &tmp_bw) == VP8_ENC_OK);
WebPPictureFree(&picture); WebPPictureFree(&picture);
if (ok) { if (ok) {
const uint8_t* const data = VP8LBitWriterFinish(&tmp_bw); const uint8_t* const data = VP8LBitWriterFinish(&tmp_bw);

View File

@ -1057,7 +1057,10 @@ int VP8LEncodeImage(const WebPConfig* const config,
width = picture->width; width = picture->width;
height = picture->height; height = picture->height;
VP8LBitWriterInit(&bw, (width * height) >> 1); if (!VP8LBitWriterInit(&bw, (width * height) >> 1)) {
err = VP8_ENC_ERROR_OUT_OF_MEMORY;
goto Error;
}
if (!WebPReportProgress(picture, 1, &percent)) { if (!WebPReportProgress(picture, 1, &percent)) {
UserAbort: UserAbort: