killpoco: reduce scope and number of various includes.
Change-Id: Ic3eb409fbb11cc665f0f3a55bb7a4e59cbd28f8b Signed-off-by: Michael Meeks <michael.meeks@collabora.com>pull/8479/head
parent
5ed9ade9da
commit
0b91bacdc8
|
@ -59,7 +59,6 @@
|
|||
#include <Poco/Exception.h>
|
||||
#include <Poco/JSON/Object.h>
|
||||
#include <Poco/JSON/Parser.h>
|
||||
#include <Poco/Net/Socket.h>
|
||||
#include <Poco/URI.h>
|
||||
|
||||
#include "ChildSession.hpp"
|
||||
|
|
|
@ -34,6 +34,7 @@
|
|||
#include <Poco/MemoryStream.h>
|
||||
#include <Poco/Net/HTTPRequest.h>
|
||||
#include <Poco/Net/HTTPResponse.h>
|
||||
#include <Poco/Net/WebSocket.h> // computeAccept
|
||||
#include <Poco/URI.h>
|
||||
#if ENABLE_SSL
|
||||
#include <Poco/Net/X509Certificate.h>
|
||||
|
@ -1315,6 +1316,30 @@ bool StreamSocket::sniffSSL() const
|
|||
|
||||
#endif // !MOBILEAPP
|
||||
|
||||
namespace {
|
||||
/// To make the protected 'computeAccept' accessible.
|
||||
class PublicComputeAccept final : public Poco::Net::WebSocket
|
||||
{
|
||||
public:
|
||||
static std::string doComputeAccept(const std::string &key)
|
||||
{
|
||||
return computeAccept(key);
|
||||
}
|
||||
|
||||
static std::string generateKey() { return createKey(); }
|
||||
};
|
||||
}
|
||||
|
||||
std::string WebSocketHandler::computeAccept(const std::string &key)
|
||||
{
|
||||
return PublicComputeAccept::doComputeAccept(key);
|
||||
}
|
||||
|
||||
std::string WebSocketHandler::generateKey()
|
||||
{
|
||||
return PublicComputeAccept::generateKey();
|
||||
}
|
||||
|
||||
// Required by Android and iOS apps.
|
||||
namespace http
|
||||
{
|
||||
|
|
|
@ -19,10 +19,7 @@
|
|||
#include "Socket.hpp"
|
||||
#include <net/HttpRequest.hpp>
|
||||
|
||||
#include <Poco/MemoryStream.h>
|
||||
#include <Poco/Net/HTTPRequest.h>
|
||||
#include <Poco/Net/HTTPResponse.h>
|
||||
#include <Poco/Net/WebSocket.h>
|
||||
|
||||
#include <chrono>
|
||||
#include <memory>
|
||||
|
@ -79,7 +76,7 @@ public:
|
|||
, _pingTimeUs(0)
|
||||
, _isMasking(isClient && isMasking)
|
||||
, _inFragmentBlock(false)
|
||||
, _key(isClient ? PublicComputeAccept::generateKey() : std::string())
|
||||
, _key(isClient ? generateKey() : std::string())
|
||||
, _lastFlags(0)
|
||||
,
|
||||
#endif
|
||||
|
@ -931,17 +928,8 @@ protected:
|
|||
/// Implementation of the ProtocolHandlerInterface.
|
||||
void dumpState(std::ostream& os) const override;
|
||||
|
||||
/// To make the protected 'computeAccept' accessible.
|
||||
class PublicComputeAccept final : public Poco::Net::WebSocket
|
||||
{
|
||||
public:
|
||||
static std::string doComputeAccept(const std::string &key)
|
||||
{
|
||||
return computeAccept(key);
|
||||
}
|
||||
|
||||
static std::string generateKey() { return createKey(); }
|
||||
};
|
||||
static std::string generateKey();
|
||||
static std::string computeAccept(const std::string &key);
|
||||
|
||||
/// Upgrade the http(s) connection to a websocket.
|
||||
template <typename T>
|
||||
|
@ -969,7 +957,7 @@ protected:
|
|||
http::Response httpResponse(http::StatusCode::SwitchingProtocols, socket->getFD());
|
||||
httpResponse.set("Upgrade", "websocket");
|
||||
httpResponse.set("Connection", "Upgrade");
|
||||
httpResponse.set("Sec-WebSocket-Accept", PublicComputeAccept::doComputeAccept(wsKey));
|
||||
httpResponse.set("Sec-WebSocket-Accept", computeAccept(wsKey));
|
||||
LOG_TRC("Sending WS Upgrade response: " << httpResponse.header().toString());
|
||||
socket->send(httpResponse);
|
||||
#else
|
||||
|
@ -997,8 +985,7 @@ protected:
|
|||
if (response.statusLine().statusCode() == http::StatusCode::SwitchingProtocols &&
|
||||
Util::iequal(response.get("Upgrade"), "websocket") &&
|
||||
Util::iequal(response.get("Connection", ""), "Upgrade") &&
|
||||
response.get("Sec-WebSocket-Accept", "") ==
|
||||
PublicComputeAccept::doComputeAccept(_key))
|
||||
response.get("Sec-WebSocket-Accept", "") == computeAccept(_key))
|
||||
{
|
||||
LOG_TRC("Accepted incoming websocket response");
|
||||
setWebSocket(socket);
|
||||
|
|
|
@ -14,6 +14,7 @@
|
|||
#include <helpers.hpp>
|
||||
#include <Poco/Util/Application.h>
|
||||
#include <Poco/Net/StreamSocket.h>
|
||||
#include <Poco/Net/SecureStreamSocket.h>
|
||||
#include <Poco/Net/StringPartSource.h>
|
||||
#include <Poco/Net/HTMLForm.h>
|
||||
#include <Poco/Net/HTTPRequest.h>
|
||||
|
@ -115,11 +116,22 @@ public:
|
|||
return true;
|
||||
}
|
||||
|
||||
inline std::shared_ptr<Poco::Net::StreamSocket> createRawSocket()
|
||||
{
|
||||
return
|
||||
#if ENABLE_SSL
|
||||
std::make_shared<Poco::Net::SecureStreamSocket>
|
||||
#else
|
||||
std::make_shared<Poco::Net::StreamSocket>
|
||||
#endif
|
||||
(Poco::Net::SocketAddress("127.0.0.1", ClientPortNumber));
|
||||
}
|
||||
|
||||
void testChunks()
|
||||
{
|
||||
LOG_TST("testChunks");
|
||||
|
||||
std::shared_ptr<Poco::Net::StreamSocket> socket = helpers::createRawSocket();
|
||||
std::shared_ptr<Poco::Net::StreamSocket> socket = createRawSocket();
|
||||
|
||||
writeString(
|
||||
socket,
|
||||
|
|
|
@ -17,7 +17,6 @@
|
|||
#include <WopiTestServer.hpp>
|
||||
|
||||
#include <wsd/DocumentBroker.hpp>
|
||||
#include <wsd/Admin.hpp>
|
||||
|
||||
#include <Poco/Net/HTTPRequest.h>
|
||||
|
||||
|
|
|
@ -24,9 +24,6 @@
|
|||
#include <Poco/Net/HTTPResponse.h>
|
||||
#include <Poco/Net/HTTPSClientSession.h>
|
||||
#include <Poco/Net/NetException.h>
|
||||
#include <Poco/Net/StreamSocket.h>
|
||||
#include <Poco/Net/SecureStreamSocket.h>
|
||||
#include <Poco/Net/Socket.h>
|
||||
#include <Poco/Path.h>
|
||||
#include <Poco/URI.h>
|
||||
|
||||
|
@ -302,17 +299,6 @@ pocoGet(bool secure, const std::string& host, const int port, const std::string&
|
|||
return pocoGet(uri);
|
||||
}
|
||||
|
||||
inline std::shared_ptr<Poco::Net::StreamSocket> createRawSocket()
|
||||
{
|
||||
return
|
||||
#if ENABLE_SSL
|
||||
std::make_shared<Poco::Net::SecureStreamSocket>
|
||||
#else
|
||||
std::make_shared<Poco::Net::StreamSocket>
|
||||
#endif
|
||||
(Poco::Net::SocketAddress("127.0.0.1", ClientPortNumber));
|
||||
}
|
||||
|
||||
// Sets read / write timeout for the given file descriptor.
|
||||
inline void setSocketTimeOut(int socketFD, int timeMS)
|
||||
{
|
||||
|
|
Loading…
Reference in New Issue