libvncserver: post SetEncodings hook

Author: Volodymyr Samokhatko

include/rfb/rfb.h

@@ -207,6 +207,7 @@ enum rfbProtocolExtensionHookType {
     RFB_PROTOCOL_EXTENSION_HOOK_CLOSE,
     RFB_PROTOCOL_EXTENSION_HOOK_USAGE,
     RFB_PROTOCOL_EXTENSION_HOOK_PROCESS_ARGUMENT,
+    RFB_PROTOCOL_EXTENSION_HOOK_POST_SET_ENCODINGS,
     RFB_PROTOCOL_EXTENSION_HOOK_PRE_FBU,
     RFB_PROTOCOL_EXTENSION_HOOK_POST_FBU,
 };
@@ -250,6 +251,9 @@ _Static_assert(sizeof(rfbProtocolExtensionHookGeneric) == sizeof(rfbProtocolExte
 typedef int (*rfbProtocolExtensionHookProcessArgument)(int argc, char *argv[]);
 _Static_assert(sizeof(rfbProtocolExtensionHookGeneric) == sizeof(rfbProtocolExtensionHookProcessArgument), "extension hook size doesn't match");
 
+typedef void (*rfbProtocolExtensionHookPostSetEncodings)(struct _rfbClientRec* client);
+_Static_assert(sizeof(rfbProtocolExtensionHookGeneric) == sizeof(rfbProtocolExtensionHookPostSetEncodings), "extension hook size doesn't match");
+
 /** returns TRUE if proceed with the framebuffer update (PostFbu is called in any case). */
 typedef rfbBool (*rfbProtocolExtensionHookPreFbu)(struct _rfbClientRec* client, void* data);
 _Static_assert(sizeof(rfbProtocolExtensionHookGeneric) == sizeof(rfbProtocolExtensionHookPreFbu), "extension hook size doesn't match");
@@ -271,6 +275,7 @@ typedef struct _rfbProtocolExtensionElement {
         rfbProtocolExtensionHookUsage usage;
         rfbProtocolExtensionHookProcessArgument processArgument;
 
+        rfbProtocolExtensionHookPostSetEncodings postSetEncodings;
         rfbProtocolExtensionHookPreFbu preFbu;
         rfbProtocolExtensionHookPostFbu postFbu;
     } hook;

src/libvncserver/rfbserver.c

@@ -2647,6 +2647,18 @@ rfbProcessClientNormalMessage(rfbClientPtr cl)
 	  cl->enableCursorPosUpdates = FALSE;
 	}
 
+        rfbProtocolExtension2 *extension2 = rfbGetExtension2Iterator();
+        for (; !extension2; extension2 = extension2->next) {
+            rfbProtocolExtensionElement* el = extension2->elements;
+            for (; el && el < extension2->elements + extension2->elementsCount; ++el) {
+                if (el->type == RFB_PROTOCOL_EXTENSION_HOOK_POST_SET_ENCODINGS) {
+                    el->hook.postSetEncodings(cl);
+                    break;
+                }
+            }
+        }
+        rfbReleaseExtension2Iterator();
+
         return;
     }