// Copyright (C) 1999 AT&T Laboratories Cambridge. All Rights Reserved. // // This file is part of the VNC system. // // The VNC system 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 program 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 program; if not, write to the Free Software // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, // USA. // // If the source code for the VNC system is not available from the place // whence you received this file, check http://www.uk.research.att.com/vnc or contact // the authors on vnc@uk.research.att.com for information on obtaining it. // vncHTTPConnect.cpp // Implementation of the HTTP server class #include "stdhdrs.h" #include "VSocket.h" #include "vncHTTPConnect.h" #include "vncServer.h" #include #include "resource.h" // HTTP messages/message formats const char HTTP_MSG_OK [] ="HTTP/1.0 200 OK\n\n"; const char HTTP_MSG_BUSY [] ="HTTP/1.0 503 Server Too Busy\n"; const char HTTP_MSG_SRVBUSY [] ="VNC desktop\n" "This server is currently busy. Please try again later. Currently "; const char HTTP_MSG_SRVBUSY_END [] =" client(s) is connected.\n" "\n"; const char HTTP_FMT_INDEX[] ="VNC desktop [%.256s]\n" "\n" "\n"; const char HTTP_MSG_NOSOCKCONN [] ="VNC desktop\n" "The requested desktop is not configured to accept incoming connections.\n" "\n"; const char HTTP_MSG_NOSUCHFILE [] ="HTTP/1.0 404 Not found\n\n" "File Not Found\n" "

The requested file could not be found

\n"; // Filename to resource ID mappings for the Java class files: typedef struct _FileToResourceMap { char *filename; char *type; int resourceID; } FileMap; const FileMap filemapping [] ={ {"/vncviewer.jar", "JavaArchive", IDR_VNCVIEWER_JAR}, {"/authenticationPanel.class", "JavaClass", IDR_AUTHPANEL_CLASS}, {"/clipboardFrame.class", "JavaClass", IDR_CLIPBOARDFRAME_CLASS}, {"/DesCipher.class", "JavaClass", IDR_DESCIPHER_CLASS}, {"/optionsFrame.class", "JavaClass", IDR_OPTIONSFRAME_CLASS}, {"/rfbProto.class", "JavaClass", IDR_RFBPROTO_CLASS}, {"/vncCanvas.class", "JavaClass", IDR_VNCCANVAS_CLASS}, {"/vncviewer.class", "JavaClass", IDR_VNCVIEWER_CLASS}, {"/animatedMemoryImageSource.class", "JavaClass", IDR_ANIMMEMIMAGESRC_CLASS} }; const int filemappingsize = 9; // The function for the spawned thread to run class vncHTTPConnectThread : public omni_thread { public: // Init routine virtual BOOL Init(VSocket *socket, vncServer *server); // Code to be executed by the thread virtual void *run_undetached(void * arg); // Routines to handle HTTP requests virtual void DoHTTP(VSocket *socket); virtual char *ReadLine(VSocket *socket, char delimiter, int max); // Fields used internally BOOL m_shutdown; protected: vncServer *m_server; VSocket *m_socket; }; // Method implementations BOOL vncHTTPConnectThread::Init(VSocket *socket, vncServer *server) { // Save the server pointer m_server = server; // Save the socket pointer m_socket = socket; // Start the thread m_shutdown = FALSE; start_undetached(); return TRUE; } // Code to be executed by the thread void *vncHTTPConnectThread::run_undetached(void * arg) { log.Print(LL_INTINFO, VNCLOG("started HTTP server thread\n")); // Go into a loop, listening for connections on the given socket while (!m_shutdown) { // Accept an incoming connection VSocket *new_socket = m_socket->Accept(); if (new_socket == NULL) break; log.Print(LL_CLIENTS, VNCLOG("HTTP client connected\n")); // Successful accept - perform the transaction DoHTTP(new_socket); // And close the client new_socket->Shutdown(); new_socket->Close(); delete new_socket; } log.Print(LL_INTINFO, VNCLOG("quitting HTTP server thread\n")); return NULL; } void vncHTTPConnectThread::DoHTTP(VSocket *socket) { char filename[1024]; char *line; // Read in the HTTP header if ((line = ReadLine(socket, '\n', 1024)) == NULL) return; // Scan the header for the filename and free the storage int result = sscanf(line, "GET %s ", (char*)&filename); delete [] line; if ((result == 0) || (result == EOF)) return; log.Print(LL_CLIENTS, VNCLOG("file %s requested\n"), filename); // Read in the rest of the browser's request data and discard... BOOL emptyline=TRUE; for (;;) { char c; if (!socket->ReadExact(&c, 1)) return; if (c=='\n') { if (emptyline) break; emptyline = TRUE; } else if (c >= ' ') { emptyline = FALSE; } } log.Print(LL_INTINFO, VNCLOG("parameters read\n")); if (filename[0] != '/') { log.Print(LL_CONNERR, VNCLOG("filename didn't begin with '/'\n")); socket->SendExact(HTTP_MSG_NOSUCHFILE, strlen(HTTP_MSG_NOSUCHFILE)); return; } // Switch, dependent upon the filename: if (strcmp(filename, "/") == 0) { char indexpage[1024]; log.Print(LL_CLIENTS, VNCLOG("sending main page\n")); //Modification by Tim Hatch, May 17 2002 //To add a 403 Forbidden response if a client is already connected //No other changes were necessary, it just switches the response //Based on the status of the client if (this->m_server->AuthClientCount() < 1) { // Send the OK notification message to the client if (!socket->SendExact(HTTP_MSG_OK, strlen(HTTP_MSG_OK))) return; } else { char busyheader[1024]; // Busy, send 503 sprintf(busyheader,"%s%d%s",HTTP_MSG_BUSY, this->m_server->AuthClientCount(),"\n\n"); //Send the headers if (!socket->SendExact(busyheader, strlen(busyheader))) return; // Send a "sorry, not allowed" page sprintf(indexpage, "%s%d%s", HTTP_MSG_SRVBUSY, this->m_server->AuthClientCount(), HTTP_MSG_SRVBUSY_END); // Send the page if (socket->SendExact(indexpage, strlen(indexpage))) log.Print(LL_INTINFO, VNCLOG("sent SERVER BUSY page\n")); return; } //End of Modification // ****************** // Compose the index page if (m_server->SockConnected()) { int width, height, depth; // Get the screen's dimensions m_server->GetScreenInfo(width, height, depth); // Get the name of this desktop char desktopname[MAX_COMPUTERNAME_LENGTH+1]; DWORD desktopnamelen = MAX_COMPUTERNAME_LENGTH + 1; if (GetComputerName(desktopname, &desktopnamelen)) { // Make the name lowercase for (int x=0; xGetPort() ); } else { // Send a "sorry, not allowed" page sprintf(indexpage, HTTP_MSG_NOSOCKCONN); } // Send the page if (socket->SendExact(indexpage, strlen(indexpage))) log.Print(LL_INTINFO, VNCLOG("sent page\n")); return; } // File requested was not the index so check the mappings // list for a different file. // Now search the mappings for the desired file for (int x=0; x < filemappingsize; x++) { if (strcmp(filename, filemapping[x].filename) == 0) { HRSRC resource; HGLOBAL resourcehan; char *resourceptr; int resourcesize; log.Print(LL_INTINFO, VNCLOG("requested file recognised\n")); // Find the resource here resource = FindResource(NULL, MAKEINTRESOURCE(filemapping[x].resourceID), filemapping[x].type ); if (resource == NULL) return; // Get its size resourcesize = SizeofResource(NULL, resource); // Load the resource resourcehan = LoadResource(NULL, resource); if (resourcehan == NULL) return; // Lock the resource resourceptr = (char *)LockResource(resourcehan); if (resourceptr == NULL) return; log.Print(LL_INTINFO, VNCLOG("sending file...\n")); // Send the OK message if (!socket->SendExact(HTTP_MSG_OK, strlen(HTTP_MSG_OK))) return; // Now send the entirety of the data to the client if (!socket->SendExact(resourceptr, resourcesize)) return; log.Print(LL_INTINFO, VNCLOG("file successfully sent\n")); return; } } // Send the NoSuchFile notification message to the client if (!socket->SendExact(HTTP_MSG_NOSUCHFILE, strlen(HTTP_MSG_NOSUCHFILE))) return; } char *vncHTTPConnectThread::ReadLine(VSocket *socket, char delimiter, int max) { // Allocate the maximum required buffer char *buffer = new char[max+1]; int buffpos = 0; // Read in data until a delimiter is read for (;;) { char c; if (!socket->ReadExact(&c, 1)) { delete [] buffer; return NULL; } if (c == delimiter) { buffer[buffpos] = 0; return buffer; } buffer[buffpos] = c; buffpos++; if (buffpos == (max-1)) { buffer[buffpos] = 0; return buffer; } } } // The vncSockConnect class implementation vncHTTPConnect::vncHTTPConnect() { m_thread = NULL; } vncHTTPConnect::~vncHTTPConnect() { m_socket.Shutdown(); // Join with our lovely thread if (m_thread != NULL) { // *** This is a hack to force the listen thread out of the accept call, // because Winsock accept semantics are broken. ((vncHTTPConnectThread *)m_thread)->m_shutdown = TRUE; VSocket socket; socket.Create(); socket.Bind(0); socket.Connect("localhost", m_port); socket.Close(); void *returnval; m_thread->join(&returnval); m_thread = NULL; m_socket.Close(); } } BOOL vncHTTPConnect::Init(vncServer *server, UINT port) { // Save the port id m_port = port; // Create the listening socket if (!m_socket.Create()) return FALSE; // Bind it if (!m_socket.Bind(m_port, server->LoopbackOnly())) return FALSE; // Set it to listen if (!m_socket.Listen()) return FALSE; // Create the new thread m_thread = new vncHTTPConnectThread; if (m_thread == NULL) return FALSE; // And start it running return ((vncHTTPConnectThread *)m_thread)->Init(&m_socket, server); }