123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392 |
- /* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved.
- * Copyright (C) 2004-2008 Constantin Kaplinsky. 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.
- */
-
- // FIXME: Check cases when screen width/height is not a multiply of 32.
- // e.g. 800x600.
-
- #include <strings.h>
- #include <sys/types.h>
- #include <sys/stat.h>
- #include <unistd.h>
- #include <errno.h>
- #include <rfb/Logger_stdio.h>
- #include <rfb/LogWriter.h>
- #include <rfb/VNCServerST.h>
- #include <rfb/Configuration.h>
- #include <rfb/Timer.h>
- #include <network/TcpSocket.h>
- #include <network/UnixSocket.h>
-
- #include <signal.h>
- #include <X11/X.h>
- #include <X11/Xlib.h>
- #include <X11/Xutil.h>
-
- #include <x0vncserver/XDesktop.h>
- #include <x0vncserver/Geometry.h>
- #include <x0vncserver/Image.h>
- #include <x0vncserver/PollingScheduler.h>
-
- extern char buildtime[];
-
- using namespace rfb;
- using namespace network;
-
- static LogWriter vlog("Main");
-
- IntParameter pollingCycle("PollingCycle", "Milliseconds per one polling "
- "cycle; actual interval may be dynamically "
- "adjusted to satisfy MaxProcessorUsage setting", 30);
- IntParameter maxProcessorUsage("MaxProcessorUsage", "Maximum percentage of "
- "CPU time to be consumed", 35);
- StringParameter displayname("display", "The X display", "");
- IntParameter rfbport("rfbport", "TCP port to listen for RFB protocol",5900);
- StringParameter rfbunixpath("rfbunixpath", "Unix socket to listen for RFB protocol", "");
- IntParameter rfbunixmode("rfbunixmode", "Unix socket access mode", 0600);
- StringParameter hostsFile("HostsFile", "File with IP access control rules", "");
-
- //
- // Allow the main loop terminate itself gracefully on receiving a signal.
- //
-
- static bool caughtSignal = false;
-
- static void CleanupSignalHandler(int sig)
- {
- caughtSignal = true;
- }
-
-
- class FileTcpFilter : public TcpFilter
- {
-
- public:
-
- FileTcpFilter(const char *fname)
- : TcpFilter("-"), fileName(NULL), lastModTime(0)
- {
- if (fname != NULL)
- fileName = strdup((char *)fname);
- }
-
- virtual ~FileTcpFilter()
- {
- if (fileName != NULL)
- free(fileName);
- }
-
- virtual bool verifyConnection(Socket* s)
- {
- if (!reloadRules()) {
- vlog.error("Could not read IP filtering rules: rejecting all clients");
- filter.clear();
- filter.push_back(parsePattern("-"));
- return false;
- }
-
- return TcpFilter::verifyConnection(s);
- }
-
- protected:
-
- bool reloadRules()
- {
- if (fileName == NULL)
- return true;
-
- struct stat st;
- if (stat(fileName, &st) != 0)
- return false;
-
- if (st.st_mtime != lastModTime) {
- // Actually reload only if the file was modified
- FILE *fp = fopen(fileName, "r");
- if (fp == NULL)
- return false;
-
- // Remove all the rules from the parent class
- filter.clear();
-
- // Parse the file contents adding rules to the parent class
- char buf[32];
- while (readLine(buf, 32, fp)) {
- if (buf[0] && strchr("+-?", buf[0])) {
- filter.push_back(parsePattern(buf));
- }
- }
-
- fclose(fp);
- lastModTime = st.st_mtime;
- }
- return true;
- }
-
- protected:
-
- char *fileName;
- time_t lastModTime;
-
- private:
-
- //
- // NOTE: we silently truncate long lines in this function.
- //
-
- bool readLine(char *buf, int bufSize, FILE *fp)
- {
- if (fp == NULL || buf == NULL || bufSize == 0)
- return false;
-
- if (fgets(buf, bufSize, fp) == NULL)
- return false;
-
- char *ptr = strchr(buf, '\n');
- if (ptr != NULL) {
- *ptr = '\0'; // remove newline at the end
- } else {
- if (!feof(fp)) {
- int c;
- do { // skip the rest of a long line
- c = getc(fp);
- } while (c != '\n' && c != EOF);
- }
- }
- return true;
- }
-
- };
-
- char* programName;
-
- static void printVersion(FILE *fp)
- {
- fprintf(fp, "TigerVNC Server version %s, built %s\n",
- PACKAGE_VERSION, buildtime);
- }
-
- static void usage()
- {
- printVersion(stderr);
- fprintf(stderr, "\nUsage: %s [<parameters>]\n", programName);
- fprintf(stderr, " %s --version\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);
- }
-
- int main(int argc, char** argv)
- {
- initStdIOLoggers();
- LogWriter::setLogParams("*:stderr:30");
-
- programName = argv[0];
- Display* dpy;
-
- Configuration::enableServerParams();
-
- // Disable configuration parameters which we do not support
- Configuration::removeParam("AcceptSetDesktopSize");
-
- for (int i = 1; i < argc; i++) {
- if (Configuration::setParam(argv[i]))
- continue;
-
- if (argv[i][0] == '-') {
- if (i+1 < argc) {
- if (Configuration::setParam(&argv[i][1], argv[i+1])) {
- i++;
- continue;
- }
- }
- if (strcmp(argv[i], "-v") == 0 ||
- strcmp(argv[i], "-version") == 0 ||
- strcmp(argv[i], "--version") == 0) {
- printVersion(stdout);
- return 0;
- }
- usage();
- }
-
- usage();
- }
-
- CharArray dpyStr(displayname.getData());
- if (!(dpy = XOpenDisplay(dpyStr.buf[0] ? dpyStr.buf : 0))) {
- // FIXME: Why not vlog.error(...)?
- fprintf(stderr,"%s: unable to open display \"%s\"\r\n",
- programName, XDisplayName(dpyStr.buf));
- exit(1);
- }
-
- signal(SIGHUP, CleanupSignalHandler);
- signal(SIGINT, CleanupSignalHandler);
- signal(SIGTERM, CleanupSignalHandler);
-
- std::list<SocketListener*> listeners;
-
- try {
- TXWindow::init(dpy,"x0vncserver");
- Geometry geo(DisplayWidth(dpy, DefaultScreen(dpy)),
- DisplayHeight(dpy, DefaultScreen(dpy)));
- if (geo.getRect().is_empty()) {
- vlog.error("Exiting with error");
- return 1;
- }
- XDesktop desktop(dpy, &geo);
-
- VNCServerST server("x0vncserver", &desktop);
-
- if (rfbunixpath.getValueStr()[0] != '\0') {
- listeners.push_back(new network::UnixListener(rfbunixpath, rfbunixmode));
- vlog.info("Listening on %s (mode %04o)", (const char*)rfbunixpath, (int)rfbunixmode);
- } else {
- createTcpListeners(&listeners, 0, (int)rfbport);
- vlog.info("Listening on port %d", (int)rfbport);
- }
-
- const char *hostsData = hostsFile.getData();
- FileTcpFilter fileTcpFilter(hostsData);
- if (strlen(hostsData) != 0)
- for (std::list<SocketListener*>::iterator i = listeners.begin();
- i != listeners.end();
- i++)
- (*i)->setFilter(&fileTcpFilter);
- delete[] hostsData;
-
- PollingScheduler sched((int)pollingCycle, (int)maxProcessorUsage);
-
- while (!caughtSignal) {
- int wait_ms;
- struct timeval tv;
- fd_set rfds, wfds;
- std::list<Socket*> sockets;
- std::list<Socket*>::iterator i;
-
- // Process any incoming X events
- TXWindow::handleXEvents(dpy);
-
- FD_ZERO(&rfds);
- FD_ZERO(&wfds);
-
- FD_SET(ConnectionNumber(dpy), &rfds);
- for (std::list<SocketListener*>::iterator i = listeners.begin();
- i != listeners.end();
- i++)
- FD_SET((*i)->getFd(), &rfds);
-
- server.getSockets(&sockets);
- int clients_connected = 0;
- for (i = sockets.begin(); i != sockets.end(); i++) {
- if ((*i)->isShutdown()) {
- server.removeSocket(*i);
- delete (*i);
- } else {
- FD_SET((*i)->getFd(), &rfds);
- if ((*i)->outStream().bufferUsage() > 0)
- FD_SET((*i)->getFd(), &wfds);
- clients_connected++;
- }
- }
-
- if (!clients_connected)
- sched.reset();
-
- wait_ms = 0;
-
- if (sched.isRunning()) {
- wait_ms = sched.millisRemaining();
- if (wait_ms > 500) {
- wait_ms = 500;
- }
- }
-
- soonestTimeout(&wait_ms, Timer::checkTimeouts());
-
- tv.tv_sec = wait_ms / 1000;
- tv.tv_usec = (wait_ms % 1000) * 1000;
-
- // Do the wait...
- sched.sleepStarted();
- int n = select(FD_SETSIZE, &rfds, &wfds, 0,
- wait_ms ? &tv : NULL);
- sched.sleepFinished();
-
- if (n < 0) {
- if (errno == EINTR) {
- vlog.debug("Interrupted select() system call");
- continue;
- } else {
- throw rdr::SystemException("select", errno);
- }
- }
-
- // Accept new VNC connections
- for (std::list<SocketListener*>::iterator i = listeners.begin();
- i != listeners.end();
- i++) {
- if (FD_ISSET((*i)->getFd(), &rfds)) {
- Socket* sock = (*i)->accept();
- if (sock) {
- sock->outStream().setBlocking(false);
- server.addSocket(sock);
- } else {
- vlog.status("Client connection rejected");
- }
- }
- }
-
- Timer::checkTimeouts();
-
- // Client list could have been changed.
- server.getSockets(&sockets);
-
- // Nothing more to do if there are no client connections.
- if (sockets.empty())
- continue;
-
- // Process events on existing VNC connections
- for (i = sockets.begin(); i != sockets.end(); i++) {
- if (FD_ISSET((*i)->getFd(), &rfds))
- server.processSocketReadEvent(*i);
- if (FD_ISSET((*i)->getFd(), &wfds))
- server.processSocketWriteEvent(*i);
- }
-
- if (desktop.isRunning() && sched.goodTimeToPoll()) {
- sched.newPass();
- desktop.poll();
- }
- }
-
- } catch (rdr::Exception &e) {
- vlog.error("%s", e.str());
- return 1;
- }
-
- TXWindow::handleXEvents(dpy);
-
- vlog.info("Terminated");
- return 0;
- }
|