You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
548 lines
15 KiB
548 lines
15 KiB
/*
|
|
* Copyright (C) 1999 AT&T Laboratories Cambridge. All Rights Reserved.
|
|
*
|
|
* This is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This software is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this software; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
|
|
* USA.
|
|
*/
|
|
|
|
/*
|
|
* vncviewer.c - the Xt-based VNC viewer.
|
|
*/
|
|
|
|
#ifdef WIN32
|
|
#undef SOCKET
|
|
#include <winsock2.h>
|
|
#endif
|
|
|
|
#ifdef _MSC_VER
|
|
#define strdup _strdup /* Prevent POSIX deprecation warnings */
|
|
#endif
|
|
|
|
#ifdef __STRICT_ANSI__
|
|
#define _BSD_SOURCE
|
|
#define _POSIX_SOURCE
|
|
#endif
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <time.h>
|
|
#include <rfb/rfbclient.h>
|
|
#include "tls.h"
|
|
|
|
static void Dummy(rfbClient* client) {
|
|
}
|
|
static rfbBool DummyPoint(rfbClient* client, int x, int y) {
|
|
return TRUE;
|
|
}
|
|
static void DummyRect(rfbClient* client, int x, int y, int w, int h) {
|
|
}
|
|
|
|
#ifdef WIN32
|
|
static char* NoPassword(rfbClient* client) {
|
|
return strdup("");
|
|
}
|
|
#define close closesocket
|
|
#else
|
|
#include <stdio.h>
|
|
#include <termios.h>
|
|
#endif
|
|
|
|
static char* ReadPassword(rfbClient* client) {
|
|
#ifdef WIN32
|
|
/* FIXME */
|
|
rfbClientErr("ReadPassword on Windows NOT IMPLEMENTED\n");
|
|
return NoPassword(client);
|
|
#else
|
|
int i;
|
|
char* p=malloc(9);
|
|
struct termios save,noecho;
|
|
p[0]=0;
|
|
if(tcgetattr(fileno(stdin),&save)!=0) return p;
|
|
noecho=save; noecho.c_lflag &= ~ECHO;
|
|
if(tcsetattr(fileno(stdin),TCSAFLUSH,&noecho)!=0) return p;
|
|
fprintf(stderr,"Password: ");
|
|
i=0;
|
|
while(1) {
|
|
int c=fgetc(stdin);
|
|
if(c=='\n')
|
|
break;
|
|
if(i<8) {
|
|
p[i]=c;
|
|
i++;
|
|
p[i]=0;
|
|
}
|
|
}
|
|
tcsetattr(fileno(stdin),TCSAFLUSH,&save);
|
|
return p;
|
|
#endif
|
|
}
|
|
static rfbBool MallocFrameBuffer(rfbClient* client) {
|
|
uint64_t allocSize;
|
|
|
|
if(client->frameBuffer)
|
|
free(client->frameBuffer);
|
|
|
|
/* SECURITY: promote 'width' into uint64_t so that the multiplication does not overflow
|
|
'width' and 'height' are 16-bit integers per RFB protocol design
|
|
SIZE_MAX is the maximum value that can fit into size_t
|
|
*/
|
|
allocSize = (uint64_t)client->width * client->height * client->format.bitsPerPixel/8;
|
|
|
|
if (allocSize >= SIZE_MAX) {
|
|
rfbClientErr("CRITICAL: cannot allocate frameBuffer, requested size is too large\n");
|
|
return FALSE;
|
|
}
|
|
|
|
client->frameBuffer=malloc( (size_t)allocSize );
|
|
|
|
if (client->frameBuffer == NULL)
|
|
rfbClientErr("CRITICAL: frameBuffer allocation failed, requested size too large or not enough memory?\n");
|
|
|
|
return client->frameBuffer?TRUE:FALSE;
|
|
}
|
|
|
|
/* messages */
|
|
|
|
static rfbBool CheckRect(rfbClient* client, int x, int y, int w, int h) {
|
|
return x + w <= client->width && y + h <= client->height;
|
|
}
|
|
|
|
static void FillRectangle(rfbClient* client, int x, int y, int w, int h, uint32_t colour) {
|
|
int i,j;
|
|
|
|
if (client->frameBuffer == NULL) {
|
|
return;
|
|
}
|
|
|
|
if (!CheckRect(client, x, y, w, h)) {
|
|
rfbClientLog("Rect out of bounds: %dx%d at (%d, %d)\n", x, y, w, h);
|
|
return;
|
|
}
|
|
|
|
#define FILL_RECT(BPP) \
|
|
for(j=y*client->width;j<(y+h)*client->width;j+=client->width) \
|
|
for(i=x;i<x+w;i++) \
|
|
((uint##BPP##_t*)client->frameBuffer)[j+i]=colour;
|
|
|
|
switch(client->format.bitsPerPixel) {
|
|
case 8: FILL_RECT(8); break;
|
|
case 16: FILL_RECT(16); break;
|
|
case 32: FILL_RECT(32); break;
|
|
default:
|
|
rfbClientLog("Unsupported bitsPerPixel: %d\n",client->format.bitsPerPixel);
|
|
}
|
|
}
|
|
|
|
static void CopyRectangle(rfbClient* client, const uint8_t* buffer, int x, int y, int w, int h) {
|
|
int j;
|
|
|
|
if (client->frameBuffer == NULL) {
|
|
return;
|
|
}
|
|
|
|
if (!CheckRect(client, x, y, w, h)) {
|
|
rfbClientLog("Rect out of bounds: %dx%d at (%d, %d)\n", x, y, w, h);
|
|
return;
|
|
}
|
|
|
|
#define COPY_RECT(BPP) \
|
|
{ \
|
|
int rs = w * BPP / 8, rs2 = client->width * BPP / 8; \
|
|
for (j = ((x * (BPP / 8)) + (y * rs2)); j < (y + h) * rs2; j += rs2) { \
|
|
memcpy(client->frameBuffer + j, buffer, rs); \
|
|
buffer += rs; \
|
|
} \
|
|
}
|
|
|
|
switch(client->format.bitsPerPixel) {
|
|
case 8: COPY_RECT(8); break;
|
|
case 16: COPY_RECT(16); break;
|
|
case 32: COPY_RECT(32); break;
|
|
default:
|
|
rfbClientLog("Unsupported bitsPerPixel: %d\n",client->format.bitsPerPixel);
|
|
}
|
|
}
|
|
|
|
/* TODO: test */
|
|
static void CopyRectangleFromRectangle(rfbClient* client, int src_x, int src_y, int w, int h, int dest_x, int dest_y) {
|
|
int i,j;
|
|
|
|
if (client->frameBuffer == NULL) {
|
|
return;
|
|
}
|
|
|
|
if (!CheckRect(client, src_x, src_y, w, h)) {
|
|
rfbClientLog("Source rect out of bounds: %dx%d at (%d, %d)\n", src_x, src_y, w, h);
|
|
return;
|
|
}
|
|
|
|
if (!CheckRect(client, dest_x, dest_y, w, h)) {
|
|
rfbClientLog("Dest rect out of bounds: %dx%d at (%d, %d)\n", dest_x, dest_y, w, h);
|
|
return;
|
|
}
|
|
|
|
#define COPY_RECT_FROM_RECT(BPP) \
|
|
{ \
|
|
uint##BPP##_t* _buffer=((uint##BPP##_t*)client->frameBuffer)+(src_y-dest_y)*client->width+src_x-dest_x; \
|
|
if (dest_y < src_y) { \
|
|
for(j = dest_y*client->width; j < (dest_y+h)*client->width; j += client->width) { \
|
|
if (dest_x < src_x) { \
|
|
for(i = dest_x; i < dest_x+w; i++) { \
|
|
((uint##BPP##_t*)client->frameBuffer)[j+i]=_buffer[j+i]; \
|
|
} \
|
|
} else { \
|
|
for(i = dest_x+w-1; i >= dest_x; i--) { \
|
|
((uint##BPP##_t*)client->frameBuffer)[j+i]=_buffer[j+i]; \
|
|
} \
|
|
} \
|
|
} \
|
|
} else { \
|
|
for(j = (dest_y+h-1)*client->width; j >= dest_y*client->width; j-=client->width) { \
|
|
if (dest_x < src_x) { \
|
|
for(i = dest_x; i < dest_x+w; i++) { \
|
|
((uint##BPP##_t*)client->frameBuffer)[j+i]=_buffer[j+i]; \
|
|
} \
|
|
} else { \
|
|
for(i = dest_x+w-1; i >= dest_x; i--) { \
|
|
((uint##BPP##_t*)client->frameBuffer)[j+i]=_buffer[j+i]; \
|
|
} \
|
|
} \
|
|
} \
|
|
} \
|
|
}
|
|
|
|
switch(client->format.bitsPerPixel) {
|
|
case 8: COPY_RECT_FROM_RECT(8); break;
|
|
case 16: COPY_RECT_FROM_RECT(16); break;
|
|
case 32: COPY_RECT_FROM_RECT(32); break;
|
|
default:
|
|
rfbClientLog("Unsupported bitsPerPixel: %d\n",client->format.bitsPerPixel);
|
|
}
|
|
}
|
|
|
|
static void initAppData(AppData* data) {
|
|
data->shareDesktop=TRUE;
|
|
data->viewOnly=FALSE;
|
|
data->encodingsString="tight zrle ultra copyrect hextile zlib corre rre raw";
|
|
data->useBGR233=FALSE;
|
|
data->nColours=0;
|
|
data->forceOwnCmap=FALSE;
|
|
data->forceTrueColour=FALSE;
|
|
data->requestedDepth=0;
|
|
data->compressLevel=3;
|
|
data->qualityLevel=5;
|
|
#ifdef LIBVNCSERVER_HAVE_LIBJPEG
|
|
data->enableJPEG=TRUE;
|
|
#else
|
|
data->enableJPEG=FALSE;
|
|
#endif
|
|
data->useRemoteCursor=FALSE;
|
|
}
|
|
|
|
rfbClient* rfbGetClient(int bitsPerSample,int samplesPerPixel,
|
|
int bytesPerPixel) {
|
|
rfbClient* client=(rfbClient*)calloc(sizeof(rfbClient),1);
|
|
if(!client) {
|
|
rfbClientErr("Couldn't allocate client structure!\n");
|
|
return NULL;
|
|
}
|
|
initAppData(&client->appData);
|
|
client->endianTest = 1;
|
|
client->programName="";
|
|
client->serverHost=strdup("");
|
|
client->serverPort=5900;
|
|
|
|
client->destHost = NULL;
|
|
client->destPort = 5900;
|
|
|
|
client->CurrentKeyboardLedState = 0;
|
|
client->HandleKeyboardLedState = (HandleKeyboardLedStateProc)DummyPoint;
|
|
|
|
/* default: use complete frame buffer */
|
|
client->updateRect.x = -1;
|
|
|
|
client->frameBuffer = NULL;
|
|
client->outputWindow = 0;
|
|
|
|
client->format.bitsPerPixel = bytesPerPixel*8;
|
|
client->format.depth = bitsPerSample*samplesPerPixel;
|
|
client->appData.requestedDepth=client->format.depth;
|
|
client->format.bigEndian = *(char *)&client->endianTest?FALSE:TRUE;
|
|
client->format.trueColour = 1;
|
|
|
|
if (client->format.bitsPerPixel == 8) {
|
|
client->format.redMax = 7;
|
|
client->format.greenMax = 7;
|
|
client->format.blueMax = 3;
|
|
client->format.redShift = 0;
|
|
client->format.greenShift = 3;
|
|
client->format.blueShift = 6;
|
|
} else {
|
|
client->format.redMax = (1 << bitsPerSample) - 1;
|
|
client->format.greenMax = (1 << bitsPerSample) - 1;
|
|
client->format.blueMax = (1 << bitsPerSample) - 1;
|
|
if(!client->format.bigEndian) {
|
|
client->format.redShift = 0;
|
|
client->format.greenShift = bitsPerSample;
|
|
client->format.blueShift = bitsPerSample * 2;
|
|
} else {
|
|
if(client->format.bitsPerPixel==8*3) {
|
|
client->format.redShift = bitsPerSample*2;
|
|
client->format.greenShift = bitsPerSample*1;
|
|
client->format.blueShift = 0;
|
|
} else {
|
|
client->format.redShift = bitsPerSample*3;
|
|
client->format.greenShift = bitsPerSample*2;
|
|
client->format.blueShift = bitsPerSample;
|
|
}
|
|
}
|
|
}
|
|
|
|
client->bufoutptr=client->buf;
|
|
client->buffered=0;
|
|
|
|
#ifdef LIBVNCSERVER_HAVE_LIBZ
|
|
client->raw_buffer_size = -1;
|
|
client->decompStreamInited = FALSE;
|
|
|
|
#ifdef LIBVNCSERVER_HAVE_LIBJPEG
|
|
memset(client->zlibStreamActive,0,sizeof(rfbBool)*4);
|
|
#endif
|
|
#endif
|
|
|
|
client->HandleCursorPos = DummyPoint;
|
|
client->SoftCursorLockArea = DummyRect;
|
|
client->SoftCursorUnlockScreen = Dummy;
|
|
client->GotFrameBufferUpdate = DummyRect;
|
|
client->GotCopyRect = CopyRectangleFromRectangle;
|
|
client->GotFillRect = FillRectangle;
|
|
client->GotBitmap = CopyRectangle;
|
|
client->FinishedFrameBufferUpdate = NULL;
|
|
client->GetPassword = ReadPassword;
|
|
client->MallocFrameBuffer = MallocFrameBuffer;
|
|
client->Bell = Dummy;
|
|
client->CurrentKeyboardLedState = 0;
|
|
client->HandleKeyboardLedState = (HandleKeyboardLedStateProc)DummyPoint;
|
|
client->QoS_DSCP = 0;
|
|
|
|
client->authScheme = 0;
|
|
client->subAuthScheme = 0;
|
|
client->GetCredential = NULL;
|
|
client->tlsSession = NULL;
|
|
client->LockWriteToTLS = NULL;
|
|
client->UnlockWriteToTLS = NULL;
|
|
client->sock = -1;
|
|
client->listenSock = -1;
|
|
client->listenAddress = NULL;
|
|
client->listen6Sock = -1;
|
|
client->listen6Address = NULL;
|
|
client->clientAuthSchemes = NULL;
|
|
|
|
#ifdef LIBVNCSERVER_HAVE_SASL
|
|
client->GetSASLMechanism = NULL;
|
|
client->GetUser = NULL;
|
|
client->saslSecret = NULL;
|
|
#endif /* LIBVNCSERVER_HAVE_SASL */
|
|
|
|
return client;
|
|
}
|
|
|
|
static rfbBool rfbInitConnection(rfbClient* client)
|
|
{
|
|
/* Unless we accepted an incoming connection, make a TCP connection to the
|
|
given VNC server */
|
|
|
|
if (!client->listenSpecified) {
|
|
if (!client->serverHost)
|
|
return FALSE;
|
|
if (client->destHost) {
|
|
if (!ConnectToRFBRepeater(client,client->serverHost,client->serverPort,client->destHost,client->destPort))
|
|
return FALSE;
|
|
} else {
|
|
if (!ConnectToRFBServer(client,client->serverHost,client->serverPort))
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
/* Initialise the VNC connection, including reading the password */
|
|
|
|
if (!InitialiseRFBConnection(client))
|
|
return FALSE;
|
|
|
|
client->width=client->si.framebufferWidth;
|
|
client->height=client->si.framebufferHeight;
|
|
if (!client->MallocFrameBuffer(client))
|
|
return FALSE;
|
|
|
|
if (!SetFormatAndEncodings(client))
|
|
return FALSE;
|
|
|
|
if (client->updateRect.x < 0) {
|
|
client->updateRect.x = client->updateRect.y = 0;
|
|
client->updateRect.w = client->width;
|
|
client->updateRect.h = client->height;
|
|
}
|
|
|
|
if (client->appData.scaleSetting>1)
|
|
{
|
|
if (!SendScaleSetting(client, client->appData.scaleSetting))
|
|
return FALSE;
|
|
if (!SendFramebufferUpdateRequest(client,
|
|
client->updateRect.x / client->appData.scaleSetting,
|
|
client->updateRect.y / client->appData.scaleSetting,
|
|
client->updateRect.w / client->appData.scaleSetting,
|
|
client->updateRect.h / client->appData.scaleSetting,
|
|
FALSE))
|
|
return FALSE;
|
|
}
|
|
else
|
|
{
|
|
if (!SendFramebufferUpdateRequest(client,
|
|
client->updateRect.x, client->updateRect.y,
|
|
client->updateRect.w, client->updateRect.h,
|
|
FALSE))
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
rfbBool rfbInitClient(rfbClient* client,int* argc,char** argv) {
|
|
int i,j;
|
|
|
|
if(argv && argc && *argc) {
|
|
if(client->programName==0)
|
|
client->programName=argv[0];
|
|
|
|
for (i = 1; i < *argc; i++) {
|
|
j = i;
|
|
if (strcmp(argv[i], "-listen") == 0) {
|
|
listenForIncomingConnections(client);
|
|
break;
|
|
} else if (strcmp(argv[i], "-listennofork") == 0) {
|
|
listenForIncomingConnectionsNoFork(client, -1);
|
|
break;
|
|
} else if (strcmp(argv[i], "-play") == 0) {
|
|
client->serverPort = -1;
|
|
j++;
|
|
} else if (i+1<*argc && strcmp(argv[i], "-encodings") == 0) {
|
|
client->appData.encodingsString = argv[i+1];
|
|
j+=2;
|
|
} else if (i+1<*argc && strcmp(argv[i], "-compress") == 0) {
|
|
client->appData.compressLevel = atoi(argv[i+1]);
|
|
j+=2;
|
|
} else if (i+1<*argc && strcmp(argv[i], "-quality") == 0) {
|
|
client->appData.qualityLevel = atoi(argv[i+1]);
|
|
j+=2;
|
|
} else if (i+1<*argc && strcmp(argv[i], "-scale") == 0) {
|
|
client->appData.scaleSetting = atoi(argv[i+1]);
|
|
j+=2;
|
|
} else if (i+1<*argc && strcmp(argv[i], "-qosdscp") == 0) {
|
|
client->QoS_DSCP = atoi(argv[i+1]);
|
|
j+=2;
|
|
} else if (i+1<*argc && strcmp(argv[i], "-repeaterdest") == 0) {
|
|
char* colon=strchr(argv[i+1],':');
|
|
|
|
if(client->destHost)
|
|
free(client->destHost);
|
|
client->destPort = 5900;
|
|
|
|
client->destHost = strdup(argv[i+1]);
|
|
if(colon) {
|
|
client->destHost[(int)(colon-argv[i+1])] = '\0';
|
|
client->destPort = atoi(colon+1);
|
|
}
|
|
j+=2;
|
|
} else {
|
|
char* colon=strchr(argv[i],':');
|
|
|
|
if(client->serverHost)
|
|
free(client->serverHost);
|
|
|
|
if(colon) {
|
|
client->serverHost = strdup(argv[i]);
|
|
client->serverHost[(int)(colon-argv[i])] = '\0';
|
|
client->serverPort = atoi(colon+1);
|
|
} else {
|
|
client->serverHost = strdup(argv[i]);
|
|
}
|
|
if(client->serverPort >= 0 && client->serverPort < 5900)
|
|
client->serverPort += 5900;
|
|
}
|
|
/* purge arguments */
|
|
if (j>i) {
|
|
*argc-=j-i;
|
|
memmove(argv+i,argv+j,(*argc-i)*sizeof(char*));
|
|
i--;
|
|
}
|
|
}
|
|
}
|
|
|
|
if(!rfbInitConnection(client)) {
|
|
rfbClientCleanup(client);
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
void rfbClientCleanup(rfbClient* client) {
|
|
#ifdef LIBVNCSERVER_HAVE_LIBZ
|
|
#ifdef LIBVNCSERVER_HAVE_LIBJPEG
|
|
int i;
|
|
|
|
for ( i = 0; i < 4; i++ ) {
|
|
if (client->zlibStreamActive[i] == TRUE ) {
|
|
if (inflateEnd (&client->zlibStream[i]) != Z_OK &&
|
|
client->zlibStream[i].msg != NULL)
|
|
rfbClientLog("inflateEnd: %s\n", client->zlibStream[i].msg);
|
|
}
|
|
}
|
|
|
|
if ( client->decompStreamInited == TRUE ) {
|
|
if (inflateEnd (&client->decompStream) != Z_OK &&
|
|
client->decompStream.msg != NULL)
|
|
rfbClientLog("inflateEnd: %s\n", client->decompStream.msg );
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
FreeTLS(client);
|
|
|
|
while (client->clientData) {
|
|
rfbClientData* next = client->clientData->next;
|
|
free(client->clientData);
|
|
client->clientData = next;
|
|
}
|
|
|
|
if (client->sock >= 0)
|
|
close(client->sock);
|
|
if (client->listenSock >= 0)
|
|
close(client->listenSock);
|
|
free(client->desktopName);
|
|
free(client->serverHost);
|
|
if (client->destHost)
|
|
free(client->destHost);
|
|
if (client->clientAuthSchemes)
|
|
free(client->clientAuthSchemes);
|
|
|
|
#ifdef LIBVNCSERVER_HAVE_SASL
|
|
if (client->saslSecret)
|
|
free(client->saslSecret);
|
|
#endif /* LIBVNCSERVER_HAVE_SASL */
|
|
|
|
free(client);
|
|
}
|