Plugged some memory leakage

1 files changed, 11 insertions, 0 deletions

diff --git a/libvncserver/corre.c b/libvncserver/corre.c
index 75f1211..bb07c77 100755
--- a/libvncserver/corre.c
+++ b/libvncserver/corre.c
@@ -50,6 +50,17 @@ static uint32_t getBgColour(char *data, int size, int bpp);
 static rfbBool rfbSendSmallRectEncodingCoRRE(rfbClientPtr cl, int x, int y,
                                           int w, int h);
 
+void rfbCoRRECleanup(rfbScreenInfoPtr screen)
+{
+  if (rreBeforeBufSize) {
+    free(rreBeforeBuf);
+    rreBeforeBufSize=0;
+  }
+  if (rreAfterBufSize) {
+    free(rreAfterBuf);
+    rreAfterBufSize=0;
+  }
+}
 
 /*
  * rfbSendRectEncodingCoRRE - send an arbitrary size rectangle using CoRRE