From 9995cb70a998a7818cfa9588c21297cc23589751 Mon Sep 17 00:00:00 2001
From: Robert Osfield <robert@openscenegraph.com>
Date: Wed, 26 Oct 2011 12:29:38 +0000
Subject: [PATCH] Added password support into osgvnc example and vnc plugin

---
 examples/osgvnc/osgvnc.cpp             | 14 ++++++++
 src/osgPlugins/vnc/ReaderWriterVNC.cpp | 47 ++++++++++++++++++++++++--
 2 files changed, 58 insertions(+), 3 deletions(-)

diff --git a/examples/osgvnc/osgvnc.cpp b/examples/osgvnc/osgvnc.cpp
index c02794338..a286b59b0 100644
--- a/examples/osgvnc/osgvnc.cpp
+++ b/examples/osgvnc/osgvnc.cpp
@@ -1,5 +1,6 @@
 #include <osgWidget/VncClient>
 
+#include <osgDB/Registry>
 #include <osgViewer/Viewer>
 #include <osgViewer/ViewerEventHandlers>
 
@@ -46,10 +47,23 @@ int main(int argc,char** argv)
 
     osg::ref_ptr<osg::Group> group = new osg::Group;
 
+    std::string password;
+    while(arguments.read("--password",password))
+    {
+    }
+
     for(int i=1; i<arguments.argc(); ++i)
     {
         if (!arguments.isOption(i))
         {
+            std::string hostname = arguments[i];
+
+            if (!password.empty())
+            {
+                if (!osgDB::Registry::instance()->getAuthenticationMap()) osgDB::Registry::instance()->setAuthenticationMap(new osgDB::AuthenticationMap);
+                osgDB::Registry::instance()->getAuthenticationMap()->addAuthenticationDetails(hostname, new osgDB::AuthenticationDetails("", password));
+            }
+
             osg::ref_ptr<osgWidget::VncClient> vncClient = new osgWidget::VncClient;
             if (vncClient->connect(arguments[i], hints))
             {            
diff --git a/src/osgPlugins/vnc/ReaderWriterVNC.cpp b/src/osgPlugins/vnc/ReaderWriterVNC.cpp
index 55cf4eabd..581d20bd1 100644
--- a/src/osgPlugins/vnc/ReaderWriterVNC.cpp
+++ b/src/osgPlugins/vnc/ReaderWriterVNC.cpp
@@ -49,6 +49,11 @@ class LibVncImage : public osgWidget::VncImage
         
         static void updateImage(rfbClient* client,int x,int y,int w,int h);
         
+        static void passwordCheck(rfbClient* client,const char* encryptedPassWord,int len);
+        static char* getPassword(rfbClient* client);
+
+        std::string                 _username;
+        std::string                 _password;
         double                      _timeOfLastUpdate;
         double                      _timeOfLastRender;
 
@@ -137,7 +142,6 @@ LibVncImage::LibVncImage():
     // setPixelBufferObject(new osg::PixelBufferObject(this);
 
     _inactiveBlock = new osg::RefBlock;
-
 }
 
 LibVncImage::~LibVncImage()
@@ -197,6 +201,18 @@ static rfbBool rfbInitConnection(rfbClient* client)
   return TRUE;
 }
 
+void LibVncImage::passwordCheck(rfbClient* client,const char* encryptedPassWord,int len)
+{
+    OSG_NOTICE<<"LibVncImage::passwordCheck"<<std::endl;
+}
+
+char* LibVncImage::getPassword(rfbClient* client)
+{
+    LibVncImage* image = (LibVncImage*)(rfbClientGetClientData(client, 0));
+    OSG_NOTICE<<"LibVncImage::getPassword "<<image->_password<<std::endl;
+    return strdup(image->_password.c_str());
+}
+
 
 bool LibVncImage::connect(const std::string& hostname)
 {
@@ -211,6 +227,9 @@ bool LibVncImage::connect(const std::string& hostname)
     _client->HandleKeyboardLedState = 0;
     _client->HandleTextChat = 0;
 
+    // provide the password if we have one assigned
+    if (!_password.empty())  _client->GetPassword = getPassword;
+
     rfbClientSetClientData(_client, 0, this);
     
     _client->serverHost = strdup(hostname.c_str());
@@ -267,7 +286,13 @@ rfbBool LibVncImage::resizeImage(rfbClient* client)
     OSG_NOTICE<<"resize "<<width<<", "<<height<<", "<<depth<<" image = "<<image<<std::endl;
     PrintPixelFormat(&(client->format));
 
+#ifdef __APPLE__
+    // feedback is that Mac's have an endian swap even though the PixelFormat results see under OSX are identical.
+    bool swap = true;
+#else
     bool swap = client->format.redShift!=0;
+#endif
+
     GLenum gl_pixelFormat = swap ? GL_BGRA : GL_RGBA;
 
     image->allocateImage(width, height, 1, gl_pixelFormat, GL_UNSIGNED_BYTE);
@@ -348,10 +373,26 @@ class ReaderWriterVNC : public osgDB::ReaderWriter
             OSG_NOTICE<<"Hostname = "<<hostname<<std::endl;
 
             osg::ref_ptr<LibVncImage> image = new LibVncImage;
-            image->setDataVariance(osg::Object::DYNAMIC);
-            
+            image->setDataVariance(osg::Object::DYNAMIC);           
             image->setOrigin(osg::Image::TOP_LEFT);
 
+            const osgDB::AuthenticationMap* authenticationMap = (options && options->getAuthenticationMap()) ?
+                    options->getAuthenticationMap() :
+                    osgDB::Registry::instance()->getAuthenticationMap();
+
+            const osgDB::AuthenticationDetails* details = authenticationMap ?
+                authenticationMap->getAuthenticationDetails(hostname) :
+                0;
+
+            // configure authentication if required.
+            if (details)
+            {
+                OSG_NOTICE<<"Passing in password = "<<details->password<<std::endl;
+
+                image->_username = details->username;
+                image->_password = details->password;
+            }
+
             if (!image->connect(hostname))
             {
                 return "Could not connect to "+hostname;