Implement a DisplayFinishedHook for libvncserver.

If set, this hook gets called just before
rfbSendFrameBufferUpdate() returns.

Signed-off-by: Christian Beier <dontmind@freeshell.org>

2 files changed, 12 insertions, 1 deletions

diff --git a/libvncserver/main.c b/libvncserver/main.c
index b6bd930..6aa82d3 100644
--- a/libvncserver/main.c
+++ b/libvncserver/main.c
@@ -894,6 +894,7 @@ rfbScreenInfoPtr rfbGetScreen(int* argc,char** argv,
    screen->setTranslateFunction = rfbSetTranslateFunction;
    screen->newClientHook = rfbDefaultNewClientHook;
    screen->displayHook = NULL;
+   screen->displayFinishedHook = NULL;
    screen->getKeyboardLedStateHook = NULL;
 
    /* initialize client list and iterator mutex */
diff --git a/libvncserver/rfbserver.c b/libvncserver/rfbserver.c
index 465d767..48fd74a 100644
--- a/libvncserver/rfbserver.c
+++ b/libvncserver/rfbserver.c
@@ -2446,9 +2446,14 @@ rfbSendFramebufferUpdate(rfbClientPtr cl,
       fu->nRects = Swap16IfLE(1);
       cl->ublen = sz_rfbFramebufferUpdateMsg;
       if (!rfbSendNewFBSize(cl, cl->scaledScreen->width, cl->scaledScreen->height)) {
+	if(cl->screen->displayFinishedHook)
+	  cl->screen->displayFinishedHook(cl, FALSE);
         return FALSE;
       }
-      return rfbSendUpdateBuf(cl);
+      result = rfbSendUpdateBuf(cl);
+      if(cl->screen->displayFinishedHook)
+	cl->screen->displayFinishedHook(cl, result);
+      return result;
     }
     
     /*
@@ -2564,6 +2569,8 @@ rfbSendFramebufferUpdate(rfbClientPtr cl,
        !sendSupportedMessages && !sendSupportedEncodings && !sendServerIdentity) {
       sraRgnDestroy(updateRegion);
       UNLOCK(cl->updateMutex);
+      if(cl->screen->displayFinishedHook)
+	cl->screen->displayFinishedHook(cl, TRUE);
       return TRUE;
     }
 
@@ -2841,6 +2848,9 @@ updateFailed:
         sraRgnReleaseIterator(i);
     sraRgnDestroy(updateRegion);
     sraRgnDestroy(updateCopyRegion);
+
+    if(cl->screen->displayFinishedHook)
+      cl->screen->displayFinishedHook(cl, result);
     return result;
 }