mirror of
https://github.com/kasmtech/KasmVNC.git
synced 2024-11-25 17:44:01 +01:00
336 lines
10 KiB
C++
336 lines
10 KiB
C++
/* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved.
|
|
* Copyright 2012-2016 Pierre Ossman for Cendio AB
|
|
*
|
|
* 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.
|
|
*/
|
|
//
|
|
// VNC server configuration utility
|
|
//
|
|
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <sys/time.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
|
|
#include <signal.h>
|
|
#include <X11/X.h>
|
|
#include <X11/Xlib.h>
|
|
#include <X11/Xatom.h>
|
|
#include <X11/Xutil.h>
|
|
#include <X11/keysym.h>
|
|
#include "vncExt.h"
|
|
#include <rdr/Exception.h>
|
|
#include <rfb/Configuration.h>
|
|
#include <rfb/Logger_stdio.h>
|
|
#include <rfb/LogWriter.h>
|
|
#include "TXWindow.h"
|
|
#include "TXCheckbox.h"
|
|
#include "TXLabel.h"
|
|
#include "QueryConnectDialog.h"
|
|
|
|
using namespace rfb;
|
|
|
|
LogWriter vlog("vncconfig");
|
|
|
|
StringParameter displayname("display", "The X display", "");
|
|
BoolParameter noWindow("nowin", "Don't display a window", 0);
|
|
BoolParameter iconic("iconic", "Start with window iconified", 0);
|
|
|
|
#define ACCEPT_CUT_TEXT "AcceptCutText"
|
|
#define SEND_CUT_TEXT "SendCutText"
|
|
|
|
#define SET_PRIMARY "SetPrimary"
|
|
#define SEND_PRIMARY "SendPrimary"
|
|
|
|
char* programName = 0;
|
|
Display* dpy;
|
|
int vncExtEventBase, vncExtErrorBase;
|
|
|
|
static bool getBoolParam(Display* dpy, const char* param) {
|
|
char* data;
|
|
int len;
|
|
if (XVncExtGetParam(dpy, param, &data, &len)) {
|
|
if (strcmp(data,"1") == 0) return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
class VncConfigWindow : public TXWindow, public TXEventHandler,
|
|
public TXDeleteWindowCallback,
|
|
public TXCheckboxCallback,
|
|
public QueryResultCallback {
|
|
public:
|
|
VncConfigWindow(Display* dpy)
|
|
: TXWindow(dpy, 300, 100),
|
|
acceptClipboard(dpy, "Accept clipboard from viewers", this, false, this),
|
|
setPrimaryCB(dpy, "Also set primary selection", this, false, this),
|
|
sendClipboard(dpy, "Send clipboard to viewers", this, false, this),
|
|
sendPrimaryCB(dpy, "Send primary selection to viewers", this,false,this),
|
|
queryConnectDialog(0)
|
|
{
|
|
int y = yPad;
|
|
acceptClipboard.move(xPad, y);
|
|
acceptClipboard.checked(getBoolParam(dpy, ACCEPT_CUT_TEXT));
|
|
y += acceptClipboard.height();
|
|
setPrimaryCB.move(xPad + 10, y);
|
|
setPrimaryCB.checked(getBoolParam(dpy, SET_PRIMARY));
|
|
setPrimaryCB.disabled(!acceptClipboard.checked());
|
|
y += setPrimaryCB.height();
|
|
sendClipboard.move(xPad, y);
|
|
sendClipboard.checked(getBoolParam(dpy, SEND_CUT_TEXT));
|
|
y += sendClipboard.height();
|
|
sendPrimaryCB.move(xPad + 10, y);
|
|
sendPrimaryCB.checked(getBoolParam(dpy, SEND_PRIMARY));
|
|
sendPrimaryCB.disabled(!sendClipboard.checked());
|
|
y += sendPrimaryCB.height();
|
|
setEventHandler(this);
|
|
toplevel("VNC config", this, 0, 0, 0, iconic);
|
|
XVncExtSelectInput(dpy, win(), VncExtQueryConnectMask);
|
|
}
|
|
|
|
// handleEvent()
|
|
|
|
virtual void handleEvent(TXWindow* w, XEvent* ev) {
|
|
if (ev->type == vncExtEventBase + VncExtQueryConnectNotify) {
|
|
vlog.debug("query connection event");
|
|
if (queryConnectDialog)
|
|
delete queryConnectDialog;
|
|
queryConnectDialog = 0;
|
|
char* qcAddress;
|
|
char* qcUser;
|
|
int qcTimeout;
|
|
if (XVncExtGetQueryConnect(dpy, &qcAddress, &qcUser,
|
|
&qcTimeout, &queryConnectId)) {
|
|
if (qcTimeout)
|
|
queryConnectDialog = new QueryConnectDialog(dpy, qcAddress,
|
|
qcUser, qcTimeout,
|
|
this);
|
|
if (queryConnectDialog)
|
|
queryConnectDialog->map();
|
|
XFree(qcAddress);
|
|
XFree(qcUser);
|
|
}
|
|
}
|
|
}
|
|
|
|
// TXDeleteWindowCallback method
|
|
virtual void deleteWindow(TXWindow* w) {
|
|
exit(1);
|
|
}
|
|
|
|
// TXCheckboxCallback method
|
|
virtual void checkboxSelect(TXCheckbox* checkbox) {
|
|
if (checkbox == &acceptClipboard) {
|
|
XVncExtSetParam(dpy, (acceptClipboard.checked()
|
|
? ACCEPT_CUT_TEXT "=1" : ACCEPT_CUT_TEXT "=0"));
|
|
setPrimaryCB.disabled(!acceptClipboard.checked());
|
|
} else if (checkbox == &sendClipboard) {
|
|
XVncExtSetParam(dpy, (sendClipboard.checked()
|
|
? SEND_CUT_TEXT "=1" : SEND_CUT_TEXT "=0"));
|
|
sendPrimaryCB.disabled(!sendClipboard.checked());
|
|
} else if (checkbox == &setPrimaryCB) {
|
|
XVncExtSetParam(dpy, (setPrimaryCB.checked()
|
|
? SET_PRIMARY "=1" : SET_PRIMARY "=0"));
|
|
} else if (checkbox == &sendPrimaryCB) {
|
|
XVncExtSetParam(dpy, (sendPrimaryCB.checked()
|
|
? SEND_PRIMARY "=1" : SEND_PRIMARY "=0"));
|
|
}
|
|
}
|
|
|
|
// QueryResultCallback interface
|
|
virtual void queryApproved() {
|
|
XVncExtApproveConnect(dpy, queryConnectId, 1);
|
|
}
|
|
virtual void queryRejected() {
|
|
XVncExtApproveConnect(dpy, queryConnectId, 0);
|
|
}
|
|
|
|
private:
|
|
TXCheckbox acceptClipboard, setPrimaryCB;
|
|
TXCheckbox sendClipboard, sendPrimaryCB;
|
|
|
|
QueryConnectDialog* queryConnectDialog;
|
|
void* queryConnectId;
|
|
};
|
|
|
|
static void usage()
|
|
{
|
|
fprintf(stderr,"usage: %s [parameters]\n",
|
|
programName);
|
|
fprintf(stderr," %s [parameters] -connect <host>[:<port>]\n",
|
|
programName);
|
|
fprintf(stderr," %s [parameters] -disconnect\n", programName);
|
|
fprintf(stderr," %s [parameters] [-set] <Xvnc-param>=<value> ...\n",
|
|
programName);
|
|
fprintf(stderr," %s [parameters] -list\n", programName);
|
|
fprintf(stderr," %s [parameters] -get <param>\n", programName);
|
|
fprintf(stderr," %s [parameters] -desc <param>\n",programName);
|
|
fprintf(stderr,"\n"
|
|
"Parameters can be turned on with -<param> or off with -<param>=0\n"
|
|
"Parameters which take a value can be specified as "
|
|
"-<param> <value>\n"
|
|
"Other valid forms are <param>=<value> -<param>=<value> "
|
|
"--<param>=<value>\n"
|
|
"Parameter names are case-insensitive. The parameters are:\n\n");
|
|
Configuration::listParams(79, 14);
|
|
exit(1);
|
|
}
|
|
|
|
void removeArgs(int* argc, char** argv, int first, int n)
|
|
{
|
|
if (first + n > *argc) return;
|
|
for (int i = first + n; i < *argc; i++)
|
|
argv[i-n] = argv[i];
|
|
*argc -= n;
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
programName = argv[0];
|
|
rfb::initStdIOLoggers();
|
|
rfb::LogWriter::setLogParams("*:stderr:30");
|
|
|
|
// Process vncconfig's own parameters first, then we process the
|
|
// other arguments when we have the X display.
|
|
int i;
|
|
for (i = 1; i < argc; i++) {
|
|
if (Configuration::setParam(argv[i]))
|
|
continue;
|
|
|
|
if (argv[i][0] == '-' && i+1 < argc &&
|
|
Configuration::setParam(&argv[i][1], argv[i+1])) {
|
|
i++;
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
|
|
CharArray displaynameStr(displayname.getData());
|
|
if (!(dpy = XOpenDisplay(displaynameStr.buf))) {
|
|
fprintf(stderr,"%s: unable to open display \"%s\"\n",
|
|
programName, XDisplayName(displaynameStr.buf));
|
|
exit(1);
|
|
}
|
|
|
|
if (!XVncExtQueryExtension(dpy, &vncExtEventBase, &vncExtErrorBase)) {
|
|
fprintf(stderr,"No VNC extension on display %s\n",
|
|
XDisplayName(displaynameStr.buf));
|
|
exit(1);
|
|
}
|
|
|
|
if (i < argc) {
|
|
for (; i < argc; i++) {
|
|
if (strcmp(argv[i], "-connect") == 0) {
|
|
i++;
|
|
if (i >= argc) usage();
|
|
if (!XVncExtConnect(dpy, argv[i])) {
|
|
fprintf(stderr,"connecting to %s failed\n",argv[i]);
|
|
}
|
|
} else if (strcmp(argv[i], "-disconnect") == 0) {
|
|
if (!XVncExtConnect(dpy, "")) {
|
|
fprintf(stderr,"disconnecting all clients failed\n");
|
|
}
|
|
} else if (strcmp(argv[i], "-get") == 0) {
|
|
i++;
|
|
if (i >= argc) usage();
|
|
char* data;
|
|
int len;
|
|
if (XVncExtGetParam(dpy, argv[i], &data, &len)) {
|
|
printf("%.*s\n",len,data);
|
|
} else {
|
|
fprintf(stderr,"getting param %s failed\n",argv[i]);
|
|
}
|
|
XFree(data);
|
|
} else if (strcmp(argv[i], "-desc") == 0) {
|
|
i++;
|
|
if (i >= argc) usage();
|
|
char* desc = XVncExtGetParamDesc(dpy, argv[i]);
|
|
if (desc) {
|
|
printf("%s\n",desc);
|
|
} else {
|
|
fprintf(stderr,"getting description for param %s failed\n",argv[i]);
|
|
}
|
|
XFree(desc);
|
|
} else if (strcmp(argv[i], "-list") == 0) {
|
|
int nParams;
|
|
char** list = XVncExtListParams(dpy, &nParams);
|
|
for (int i = 0; i < nParams; i++) {
|
|
printf("%s\n",list[i]);
|
|
}
|
|
XVncExtFreeParamList(list);
|
|
} else if (strcmp(argv[i], "-set") == 0) {
|
|
i++;
|
|
if (i >= argc) usage();
|
|
if (!XVncExtSetParam(dpy, argv[i])) {
|
|
fprintf(stderr,"setting param %s failed\n",argv[i]);
|
|
}
|
|
} else if (XVncExtSetParam(dpy, argv[i])) {
|
|
fprintf(stderr,"set parameter %s\n",argv[i]);
|
|
} else {
|
|
usage();
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
try {
|
|
TXWindow::init(dpy,"Vncconfig");
|
|
|
|
VncConfigWindow w(dpy);
|
|
if (!noWindow) w.map();
|
|
|
|
while (true) {
|
|
struct timeval tv;
|
|
struct timeval* tvp = 0;
|
|
|
|
// Process any incoming X events
|
|
TXWindow::handleXEvents(dpy);
|
|
|
|
// Process expired timers and get the time until the next one
|
|
int timeoutMs = Timer::checkTimeouts();
|
|
if (timeoutMs) {
|
|
tv.tv_sec = timeoutMs / 1000;
|
|
tv.tv_usec = (timeoutMs % 1000) * 1000;
|
|
tvp = &tv;
|
|
}
|
|
|
|
// If there are X requests pending then poll, don't wait!
|
|
if (XPending(dpy)) {
|
|
tv.tv_usec = tv.tv_sec = 0;
|
|
tvp = &tv;
|
|
}
|
|
|
|
// Wait for X events, VNC traffic, or the next timer expiry
|
|
fd_set rfds;
|
|
FD_ZERO(&rfds);
|
|
FD_SET(ConnectionNumber(dpy), &rfds);
|
|
int n = select(FD_SETSIZE, &rfds, 0, 0, tvp);
|
|
if (n < 0) throw rdr::SystemException("select",errno);
|
|
}
|
|
|
|
XCloseDisplay(dpy);
|
|
|
|
} catch (rdr::Exception &e) {
|
|
vlog.error("%s", e.str());
|
|
}
|
|
|
|
return 0;
|
|
}
|