From 3e6325d8436d61cbbe2eb95ec0c016efafc66a68 Mon Sep 17 00:00:00 2001
From: Ayon Ghosh <ayonghosh88@gmail.com>
Date: Sun, 8 May 2016 12:25:35 +0530
Subject: [PATCH] Adding external event handler for WebSocket connection events
 (open, close, error)

---
 include/rfb.js | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/include/rfb.js b/include/rfb.js
index 3e9344bb..682dc679 100644
--- a/include/rfb.js
+++ b/include/rfb.js
@@ -146,6 +146,7 @@ var RFB;
             'onFBResize': function () { },          // onFBResize(rfb, width, height): frame buffer resized
             'onDesktopName': function () { },       // onDesktopName(rfb, name): desktop name received
             'onXvpInit': function () { },           // onXvpInit(version): XVP extensions active for this connection
+            'onWsConnEvent': function () { },       // onWsConnEvent(eventType, event): a WebSocket connection event occurred
         });
 
         // main setup
@@ -188,6 +189,8 @@ var RFB;
             } else {
                 this._fail("Got unexpected WebSocket connection");
             }
+
+            this._onWsConnEvent('open');
         }.bind(this));
         this._sock.on('close', function (e) {
             Util.Warn("WebSocket on-close event");
@@ -209,9 +212,12 @@ var RFB;
                 this._fail("Server disconnected" + msg);
             }
             this._sock.off('close');
+
+            this._onWsConnEvent('close', e);
         }.bind(this));
         this._sock.on('error', function (e) {
             Util.Warn("WebSocket on-error event");
+            this.onWsConnEvent('error', e);
         });
 
         this._init_vars();
@@ -1258,6 +1264,7 @@ var RFB;
         ['onFBResize', 'rw', 'func'],           // onFBResize(rfb, width, height): frame buffer resized
         ['onDesktopName', 'rw', 'func'],        // onDesktopName(rfb, name): desktop name received
         ['onXvpInit', 'rw', 'func'],            // onXvpInit(version): XVP extensions active for this connection
+        ['onWsConnEvent', 'rw', 'func'],        // onWsConnEvent(eventType, event): A WebSocket connection event occurred
     ]);
 
     RFB.prototype.set_local_cursor = function (cursor) {