Initial commit
This commit is contained in:
matt
335
unix/vncconfig/vncconfig.cxx
Normal file
335
unix/vncconfig/vncconfig.cxx
Normal file
@@ -0,0 +1,335 @@
|
||||
/* 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;
|
||||
}
|
||||
Reference in New Issue
Block a user