f05cea2b13
Re-write Server of proxied requests to avoid:
coolwsd: ./net/HttpRequest.hpp:945: bool http::Response::writeData(Buffer &) const:
Assertion `get("Server") == http::getServerString() &&
"Server Agent is always set in http::Response ctor"' failed.
Change-Id: I6076dbc355b0f42a6367271893c9732195c8eb2a
Signed-off-by: Michael Meeks <michael.meeks@collabora.com>
97 lines
3.6 KiB
C++
97 lines
3.6 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4; fill-column: 100 -*- */
|
|
/*
|
|
* Copyright the Collabora Online contributors.
|
|
*
|
|
* SPDX-License-Identifier: MPL-2.0
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
#include <config.h>
|
|
|
|
#include <Poco/URI.h>
|
|
|
|
#include <COOLWSD.hpp>
|
|
#include "ProxyRequestHandler.hpp"
|
|
#include <net/HttpRequest.hpp>
|
|
#include <net/HttpHelper.hpp>
|
|
|
|
std::unordered_map<std::string, std::shared_ptr<http::Response>> ProxyRequestHandler::CacheFileHash;
|
|
std::chrono::system_clock::time_point ProxyRequestHandler::MaxAge;
|
|
|
|
void ProxyRequestHandler::handleRequest(const std::string& relPath,
|
|
const std::shared_ptr<StreamSocket>& socket,
|
|
const std::string& serverUri)
|
|
{
|
|
|
|
Poco::URI uriProxy(serverUri);
|
|
constexpr const auto zero = std::chrono::system_clock::time_point();
|
|
const auto timeNow = std::chrono::system_clock::now();
|
|
|
|
if (MaxAge > zero && timeNow > MaxAge)
|
|
{
|
|
CacheFileHash.clear();
|
|
MaxAge = zero;
|
|
}
|
|
|
|
const auto cacheEntry = CacheFileHash.find(relPath);
|
|
if (cacheEntry != CacheFileHash.end())
|
|
{
|
|
socket->sendAndShutdown(*cacheEntry->second);
|
|
return;
|
|
}
|
|
|
|
uriProxy.setPath(relPath);
|
|
auto sessionProxy = http::Session::create(uriProxy.getHost(),
|
|
http::Session::Protocol::HttpSsl,
|
|
uriProxy.getPort());
|
|
sessionProxy->setTimeout(std::chrono::seconds(10));
|
|
http::Request requestProxy(uriProxy.getPathAndQuery());
|
|
http::Session::FinishedCallback proxyCallback =
|
|
[socket, zero](const std::shared_ptr<http::Session>& httpSession)
|
|
{
|
|
try
|
|
{
|
|
const auto callbackNow = std::chrono::system_clock::now();
|
|
std::shared_ptr<http::Response> httpResponse = httpSession->response();
|
|
if (httpResponse->statusLine().statusCode() == http::StatusCode::OK)
|
|
{
|
|
if (MaxAge == zero)
|
|
{
|
|
MaxAge = callbackNow + std::chrono::hours(10);
|
|
}
|
|
|
|
CacheFileHash[httpSession->getUrl()] = httpResponse;
|
|
|
|
// We're proxying, we take responsibility.
|
|
httpResponse->set("Server", http::getServerString());
|
|
|
|
socket->sendAndShutdown(*httpResponse);
|
|
}
|
|
else
|
|
{
|
|
HttpHelper::sendErrorAndShutdown(http::StatusCode::BadRequest, socket);
|
|
}
|
|
}
|
|
catch(std::exception& exc)
|
|
{
|
|
LOG_ERR("ProxyCallback: " << exc.what());
|
|
HttpHelper::sendErrorAndShutdown(http::StatusCode::BadRequest, socket);
|
|
}
|
|
catch(...)
|
|
{
|
|
LOG_ERR("ProxyCallback: Unknown exception");
|
|
HttpHelper::sendErrorAndShutdown(http::StatusCode::BadRequest, socket);
|
|
}
|
|
};
|
|
|
|
sessionProxy->setFinishedHandler(std::move(proxyCallback));
|
|
if (!sessionProxy->asyncRequest(requestProxy, *COOLWSD::getWebServerPoll()))
|
|
{
|
|
HttpHelper::sendErrorAndShutdown(http::StatusCode::BadRequest, socket);
|
|
}
|
|
}
|
|
|
|
/* vim:set shiftwidth=4 softtabstop=4 expandtab: */
|