diff options
author | Yonit Halperin <yhalperi@redhat.com> | 2012-05-10 12:26:01 +0300 |
---|---|---|
committer | Alon Levy <alevy@redhat.com> | 2012-05-14 10:58:01 +0300 |
commit | a3c4dc8f36e65a49a7c5ccf549c83af90a0d2047 (patch) | |
tree | 8c0a8d058db14e2caebb0eb5ff32d9af088244ab | |
parent | 63e3456e12f85a3c5d23e9d00d9dbd77232ee876 (diff) | |
download | spice-a3c4dc8f36e65a49a7c5ccf549c83af90a0d2047.tar.gz spice-a3c4dc8f36e65a49a7c5ccf549c83af90a0d2047.tar.xz spice-a3c4dc8f36e65a49a7c5ccf549c83af90a0d2047.zip |
server/mjpeg_encoder: Fix memory leak for the inital output buffer given for each frame
(cherry picked from commit fecc1e3d4d3d93ade71a318ad655e85974638162)
-rw-r--r-- | server/mjpeg_encoder.c | 11 | ||||
-rw-r--r-- | server/mjpeg_encoder.h | 5 |
2 files changed, 8 insertions, 8 deletions
diff --git a/server/mjpeg_encoder.c b/server/mjpeg_encoder.c index 6b68549c..74062f5a 100644 --- a/server/mjpeg_encoder.c +++ b/server/mjpeg_encoder.c @@ -105,7 +105,6 @@ typedef struct { unsigned char ** outbuffer; /* target buffer */ size_t * outsize; - unsigned char * newbuffer; /* newly allocated buffer */ uint8_t * buffer; /* start of buffer */ size_t bufsize; } mem_destination_mgr; @@ -129,9 +128,7 @@ static boolean empty_mem_output_buffer(j_compress_ptr cinfo) memcpy(nextbuffer, dest->buffer, dest->bufsize); - free(dest->newbuffer); - - dest->newbuffer = nextbuffer; + free(dest->buffer); dest->pub.next_output_byte = nextbuffer + dest->bufsize; dest->pub.free_in_buffer = dest->bufsize; @@ -184,12 +181,10 @@ spice_jpeg_mem_dest(j_compress_ptr cinfo, dest->pub.term_destination = term_mem_destination; dest->outbuffer = outbuffer; dest->outsize = outsize; - dest->newbuffer = NULL; - if (*outbuffer == NULL || *outsize == 0) { /* Allocate initial buffer */ - dest->newbuffer = *outbuffer = malloc(OUTPUT_BUF_SIZE); - if (dest->newbuffer == NULL) + *outbuffer = malloc(OUTPUT_BUF_SIZE); + if (*outbuffer == NULL) ERREXIT1(cinfo, JERR_OUT_OF_MEMORY, 10); *outsize = OUTPUT_BUF_SIZE; } diff --git a/server/mjpeg_encoder.h b/server/mjpeg_encoder.h index c43827f9..62ef207b 100644 --- a/server/mjpeg_encoder.h +++ b/server/mjpeg_encoder.h @@ -27,6 +27,11 @@ MJpegEncoder *mjpeg_encoder_new(int width, int height); void mjpeg_encoder_destroy(MJpegEncoder *encoder); uint8_t mjpeg_encoder_get_bytes_per_pixel(MJpegEncoder *encoder); + +/* + * *dest must be either NULL or allocated by malloc, since it might be freed + * during the encoding, if its size is too small. + */ int mjpeg_encoder_start_frame(MJpegEncoder *encoder, SpiceBitmapFmt format, uint8_t **dest, size_t *dest_len); int mjpeg_encoder_encode_scanline(MJpegEncoder *encoder, uint8_t *src_pixels, |