void handleRequest(HTTPServerRequest& request, HTTPServerResponse& response) { Application& app = Application::instance(); try { std::string proto = request.get("Sec-WebSocket-Protocol", ""); Poco::SharedPtr<Poco::Net::WebSocket> pWebSocket; if (proto == "com.appinf.webtunnel.server/1.0") { response.set("Sec-WebSocket-Protocol", proto); pWebSocket = new Poco::Net::WebSocket(request, response); _portReflector.addServerSocket(pWebSocket, "ac9667bb-6032-4267-af61-9a7aafd40479"); } else if (proto == "com.appinf.webtunnel.client/1.0") { response.set("Sec-WebSocket-Protocol", proto); std::string portStr = request.get("X-WebTunnel-RemotePort", ""); unsigned port; if (!portStr.empty() && Poco::NumberParser::tryParseUnsigned(portStr, port) && port > 0 && port < 65536) { pWebSocket = new Poco::Net::WebSocket(request, response); try { _portReflector.addClientSocket(pWebSocket, "ac9667bb-6032-4267-af61-9a7aafd40479", static_cast<Poco::UInt16>(port)); } catch (Poco::NotFoundException&) { pWebSocket->shutdown(Poco::Net::WebSocket::WS_UNEXPECTED_CONDITION, "No connection to target available"); } } else { pWebSocket = new Poco::Net::WebSocket(request, response); pWebSocket->shutdown(Poco::Net::WebSocket::WS_UNEXPECTED_CONDITION, "Missing or invalid X-WebTunnel-RemotePort header"); } } else { pWebSocket = new Poco::Net::WebSocket(request, response); pWebSocket->shutdown(Poco::Net::WebSocket::WS_PROTOCOL_ERROR); } } catch (WebSocketException& exc) { app.logger().log(exc); switch (exc.code()) { case Poco::Net::WebSocket::WS_ERR_HANDSHAKE_UNSUPPORTED_VERSION: response.set("Sec-WebSocket-Version", WebSocket::WEBSOCKET_VERSION); // fallthrough case Poco::Net::WebSocket::WS_ERR_NO_HANDSHAKE: case Poco::Net::WebSocket::WS_ERR_HANDSHAKE_NO_VERSION: case Poco::Net::WebSocket::WS_ERR_HANDSHAKE_NO_KEY: response.setStatusAndReason(HTTPResponse::HTTP_BAD_REQUEST); response.setContentLength(0); response.send(); break; } } }
void LocalPortForwarder::forward(Poco::Net::StreamSocket& socket) { if (_logger.debug()) { _logger.debug(Poco::format("Local connection accepted, creating forwarding connection to %s, remote port %hu", _remoteURI.toString(), _remotePort)); } try { std::string path(_remoteURI.getPathEtc()); if (path.empty()) path = "/"; Poco::Net::HTTPRequest request(Poco::Net::HTTPRequest::HTTP_POST, path, Poco::Net::HTTPRequest::HTTP_1_1); request.set(SEC_WEBSOCKET_PROTOCOL, WEBTUNNEL_PROTOCOL); request.set(X_WEBTUNNEL_REMOTEPORT, Poco::NumberFormatter::format(_remotePort)); Poco::Net::HTTPResponse response; Poco::SharedPtr<Poco::Net::WebSocket> pWebSocket = _pWebSocketFactory->createWebSocket(_remoteURI, request, response); if (response.get(SEC_WEBSOCKET_PROTOCOL, "") != WEBTUNNEL_PROTOCOL) { _logger.error("The remote host does not support the WebTunnel protocol."); pWebSocket->shutdown(Poco::Net::WebSocket::WS_PROTOCOL_ERROR); pWebSocket->close(); socket.close(); return; } _pDispatcher->addSocket(socket, new StreamSocketToWebSocketForwarder(_pDispatcher, pWebSocket), _localTimeout); _pDispatcher->addSocket(*pWebSocket, new WebSocketToStreamSocketForwarder(_pDispatcher, socket), _remoteTimeout); } catch (Poco::Exception& exc) { _logger.error(Poco::format("Failed to open forwarding connection: %s", exc.displayText())); socket.close(); } }
void WebEventServiceImpl::shutdownImpl(Poco::SharedPtr<Poco::Net::WebSocket> pWS, Poco::UInt16 statusCode, const std::string& statusMessage, bool passive) { //_pContext->logger().notice("Shutting down WebSocket."); try { pWS->shutdown(statusCode, statusMessage); } catch (Poco::Exception& exc) { _pContext->logger().notice("Error while shutting down WebSocket: " + exc.displayText()); passive = true; // don't wait for client confirmation } if (!passive) { if (pWS->poll(Poco::Timespan(WEBEVENT_SHUTDOWN_TIMEOUT, 0), Poco::Net::Socket::SELECT_READ)) { Poco::Buffer<char> buffer(4096); int flags; pWS->receiveFrame(buffer.begin(), buffer.size(), flags); if (flags & Poco::Net::WebSocket::FRAME_OP_CLOSE) { pWS->close(); return; } } _pContext->logger().notice("Client failed to confirm shutdown."); } pWS->close(); }