Added password support into osgvnc example and vnc plugin

This commit is contained in:
Robert Osfield 2011-10-26 12:29:38 +00:00
parent b9d7989b44
commit 9995cb70a9
2 changed files with 58 additions and 3 deletions

View File

@ -1,5 +1,6 @@
#include <osgWidget/VncClient> #include <osgWidget/VncClient>
#include <osgDB/Registry>
#include <osgViewer/Viewer> #include <osgViewer/Viewer>
#include <osgViewer/ViewerEventHandlers> #include <osgViewer/ViewerEventHandlers>
@ -46,10 +47,23 @@ int main(int argc,char** argv)
osg::ref_ptr<osg::Group> group = new osg::Group; 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) for(int i=1; i<arguments.argc(); ++i)
{ {
if (!arguments.isOption(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; osg::ref_ptr<osgWidget::VncClient> vncClient = new osgWidget::VncClient;
if (vncClient->connect(arguments[i], hints)) if (vncClient->connect(arguments[i], hints))
{ {

View File

@ -49,6 +49,11 @@ class LibVncImage : public osgWidget::VncImage
static void updateImage(rfbClient* client,int x,int y,int w,int h); 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 _timeOfLastUpdate;
double _timeOfLastRender; double _timeOfLastRender;
@ -137,7 +142,6 @@ LibVncImage::LibVncImage():
// setPixelBufferObject(new osg::PixelBufferObject(this); // setPixelBufferObject(new osg::PixelBufferObject(this);
_inactiveBlock = new osg::RefBlock; _inactiveBlock = new osg::RefBlock;
} }
LibVncImage::~LibVncImage() LibVncImage::~LibVncImage()
@ -197,6 +201,18 @@ static rfbBool rfbInitConnection(rfbClient* client)
return TRUE; 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) bool LibVncImage::connect(const std::string& hostname)
{ {
@ -211,6 +227,9 @@ bool LibVncImage::connect(const std::string& hostname)
_client->HandleKeyboardLedState = 0; _client->HandleKeyboardLedState = 0;
_client->HandleTextChat = 0; _client->HandleTextChat = 0;
// provide the password if we have one assigned
if (!_password.empty()) _client->GetPassword = getPassword;
rfbClientSetClientData(_client, 0, this); rfbClientSetClientData(_client, 0, this);
_client->serverHost = strdup(hostname.c_str()); _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; OSG_NOTICE<<"resize "<<width<<", "<<height<<", "<<depth<<" image = "<<image<<std::endl;
PrintPixelFormat(&(client->format)); 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; bool swap = client->format.redShift!=0;
#endif
GLenum gl_pixelFormat = swap ? GL_BGRA : GL_RGBA; GLenum gl_pixelFormat = swap ? GL_BGRA : GL_RGBA;
image->allocateImage(width, height, 1, gl_pixelFormat, GL_UNSIGNED_BYTE); 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_NOTICE<<"Hostname = "<<hostname<<std::endl;
osg::ref_ptr<LibVncImage> image = new LibVncImage; osg::ref_ptr<LibVncImage> image = new LibVncImage;
image->setDataVariance(osg::Object::DYNAMIC); image->setDataVariance(osg::Object::DYNAMIC);
image->setOrigin(osg::Image::TOP_LEFT); 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)) if (!image->connect(hostname))
{ {
return "Could not connect to "+hostname; return "Could not connect to "+hostname;