Skip to content

Commit

Permalink
wip commit
Browse files Browse the repository at this point in the history 
Signed-off-by: Rashesh Padia <[email protected]>
Change-Id: I848f853be8c14d3367f8501c603114a3e23a617b
  • Loading branch information
@Rash419
Rash419 committed Sep 3, 2024
1 parent e7966c8 commit 067661d
Show file tree
Hide file tree
Showing 5 changed files with 174 additions and 14 deletions.
55 changes: 52 additions & 3 deletions wsd/Admin.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,10 +9,14 @@
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
*/

#include <Poco/JSON/Array.h>
#include <Poco/JSON/Object.h>
#include <config.h>

#include <chrono>
#include <cstddef>
#include <iomanip>
#include <string>
#include <sys/poll.h>
#include <unistd.h>

Expand All @@ -23,6 +27,7 @@
#include "Admin.hpp"
#include "AdminModel.hpp"
#include "Auth.hpp"
#include "HttpRequest.hpp"
#include <Common.hpp>
#include <Log.hpp>
#include <Protocol.hpp>
Expand All @@ -41,7 +46,6 @@

using namespace COOLProtocol;

using Poco::Net::HTTPResponse;
using Poco::Util::Application;

const int Admin::MinStatsIntervalMs = 50;
Expand Down Expand Up @@ -416,6 +420,10 @@ void AdminSocketHandler::handleMessage(const std::vector<char> &payload)
{
_admin->setCloseMonitorFlag();
}
else if (tokens.equals(0, "rollingupdate") && tokens.size() > 1)
{
_admin->setRollingUpdateInfo(tokens[1]);
}
}

AdminSocketHandler::AdminSocketHandler(Admin* adminManager,
Expand Down Expand Up @@ -496,8 +504,7 @@ bool AdminSocketHandler::handleInitialRequest(
return true;
}

HTTPResponse response;
response.setStatusAndReason(HTTPResponse::HTTP_BAD_REQUEST);
http::Response response(http::StatusCode::BadRequest);
response.setContentLength(0);
LOG_INF_S("Admin::handleInitialRequest bad request");
socket->send(response);
Expand Down Expand Up @@ -1293,6 +1300,48 @@ void Admin::deleteMonitorSocket(const std::string& uriWithoutParam)
}
}

void Admin::setRollingUpdateInfo(const std::string& jsonString)
{
Poco::JSON::Object::Ptr object;
if (JsonUtil::parseJSON(jsonString, object))
{
bool status = JsonUtil::getJSONValue<bool>(object, "inprogress");
setRollingUpdateStatus(status);
Poco::JSON::Array::Ptr infoArray = object->getArray("serverinfo");
if (!infoArray.isNull())
{
for(size_t i=0; i < infoArray->size(); i++)
{
if (!infoArray->isObject(i))
{
return;
}
const auto serverInfoObject = infoArray->getObject(i);
const std::string gitHash = JsonUtil::getJSONValue<std::string>(serverInfoObject , "gitHash");
const std::string serverId = JsonUtil::getJSONValue<std::string>(serverInfoObject, "serverId");
const std::string routeToken = JsonUtil::getJSONValue<std::string>(serverInfoObject, "routeToken");
_rollingUpdateInfo.try_emplace(gitHash, RollingUpdateServerInfo(gitHash, serverId, routeToken));
}
}
}
}

std::string Admin::getBuddyServer(const std::string& gitHash)
{
LOG_DBG("Getting routeToken for gitHash[" << gitHash << ']');
for (auto iterator : _rollingUpdateInfo)
{
LOG_DBG("gitHash[" << iterator.first << "] routeToken[" << iterator.second.getRouteToken()
<< "] serverId[" << iterator.second.getRouteToken() << ']');
}
auto iterator = _rollingUpdateInfo.find(gitHash);
if (iterator != _rollingUpdateInfo.end())
{
return iterator->second.getRouteToken();
}
return std::string();
}

void Admin::stop()
{
joinThread();
Expand Down
34 changes: 34 additions & 0 deletions wsd/Admin.hpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,8 @@

#include "net/WebSocketHandler.hpp"
#include "COOLWSD.hpp"
#include <string>
#include <vector>

class Admin;

Expand Down Expand Up @@ -186,6 +188,14 @@ class Admin : public SocketPoll

void setCloseMonitorFlag() { _closeMonitor = true; }

void setRollingUpdateInfo(const std::string& jsonString);

void setRollingUpdateStatus(bool status) { _rollingUpdateStatus = status; }

bool getRollingUpdateStatus() { return _rollingUpdateStatus; }

std::string getBuddyServer(const std::string& gitHash);

private:
/// Notify Forkit of changed settings.
void notifyForkit();
Expand Down Expand Up @@ -253,6 +263,30 @@ class Admin : public SocketPoll
std::map<std::string, std::shared_ptr<MonitorSocketHandler>> _monitorSockets;

std::atomic<bool> _closeMonitor = false;

class RollingUpdateServerInfo
{
public:
std::string getGitHash() { return _gitHash; }
std::string getServerId() { return _serverId; }
std::string getRouteToken() { return _routeToken; }

RollingUpdateServerInfo(const std::string& gitHash, const std::string& serverId,
const std::string& routeToken)
: _gitHash(gitHash)
, _serverId(serverId)
, _routeToken(routeToken)
{
}

private:
std::string _gitHash;
std::string _serverId;
std::string _routeToken;
};

std::map<std::string, RollingUpdateServerInfo> _rollingUpdateInfo;
std::atomic<bool> _rollingUpdateStatus;
};

/* vim:set shiftwidth=4 softtabstop=4 expandtab: */
86 changes: 79 additions & 7 deletions wsd/COOLWSD.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4321,21 +4321,39 @@ class ClientRequestDispatcher final : public SimpleSocketHandler
{
// Unit testing, nothing to do here
}
else if (requestDetails.equals(RequestDetails::Field::Type, "browser") || requestDetails.equals(RequestDetails::Field::Type, "wopi"))
else if (requestDetails.equals(RequestDetails::Field::Type, "browser") ||
requestDetails.equals(RequestDetails::Field::Type, "wopi"))
{

std::string protocol = "http";
if (socket->sniffSSL())
protocol = "https";

Poco::URI requestUri(protocol + "://" + request.getHost() + request.getURI());
const std::string& path = requestUri.getPath();
bool versionMismatch = false;
if (path.find("browser/" COOLWSD_VERSION_HASH "/") == std::string::npos &&
path.find("admin/") == std::string::npos)
{
LOG_DBG("Client - server version mismatch, proxy request to different server "
"Expected: " COOLWSD_VERSION_HASH
"; Actual URI path with version hash: "
<< path);
versionMismatch = true;
}

// File server
assert(socket && "Must have a valid socket");
constexpr auto ProxyRemote = "/remote/";
constexpr auto ProxyRemoteLen = sizeof(ProxyRemote) - 1;
constexpr auto ProxyRemoteStatic = "/remote/static/";
const auto uri = requestDetails.getURI();
const auto pos = uri.find(ProxyRemoteStatic);
const auto pos = path.find(ProxyRemoteStatic);
if (pos != std::string::npos)
{
if (Util::endsWith(uri, "lokit-extra-img.svg"))
if (Util::endsWith(path, "lokit-extra-img.svg"))
{
ProxyRequestHandler::handleRequest(
uri.substr(pos + ProxyRemoteLen), socket,
path.substr(pos + ProxyRemoteLen), socket,
ProxyRequestHandler::getProxyRatingServer());
}
#if ENABLE_FEATURE_LOCK
Expand All @@ -4347,12 +4365,66 @@ class ClientRequestDispatcher final : public SimpleSocketHandler
{
const std::string& serverUri =
unlockImageUri.getScheme() + "://" + unlockImageUri.getAuthority();
ProxyRequestHandler::handleRequest(uri.substr(pos + sizeof("/remote/static") - 1),
socket, serverUri);
ProxyRequestHandler::handleRequest(
path.substr(pos + sizeof("/remote/static") - 1), socket, serverUri);
}
}
#endif
}
else if (COOLWSD::IndirectionServerEnabled && versionMismatch &&
Admin::instance().getRollingUpdateStatus())
{
std::string searchString = "/browser/";
size_t startHashPos = path.find(searchString);
if (startHashPos != std::string::npos)
{
startHashPos += searchString.length();
size_t endHashPos = path.find('/', startHashPos);

std::string gitHash;
if (endHashPos != std::string::npos)
{
gitHash = path.substr(startHashPos, endHashPos - startHashPos);
}
else
{
gitHash = path.substr(startHashPos);
}

std::string hash(gitHash);
hash.resize(std::min(8, (int)hash.length()));
std::string routeToken = Admin::instance().getBuddyServer(hash);
if (!routeToken.empty())
{
Poco::URI::QueryParameters params = requestUri.getQueryParameters();
const auto routeTokenIt =
std::find_if(params.begin(), params.end(),
[](const std::pair<std::string, std::string>& element)
{ return element.first == "RouteToken"; });
if (routeTokenIt == params.end())
{
LOG_DBG("Adding routeToken[" << routeToken
<< "] as a parameter to requestUri["
<< requestUri.toString() << ']');

requestUri.addQueryParameter("RouteToken", routeToken);
}
else
{
LOG_DBG("Updating routeToken[" << routeToken
<< "] parameter in requestUri["
<< requestUri.toString() << ']');

routeTokenIt->second = routeToken;
requestUri.setQueryParameters(params);
}
}

ProxyRequestHandler::handleRequest(requestUri.getPathAndQuery(), socket,
requestUri.getScheme() + "://" +
requestUri.getAuthority());
}
}
else
{
COOLWSD::FileRequestHandler->handleRequest(request, requestDetails, message, socket);
Expand Down
11 changes: 7 additions & 4 deletions wsd/ProxyRequestHandler.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -43,10 +43,13 @@ void ProxyRequestHandler::handleRequest(const std::string& relPath,
return;
}

uriProxy.setPath(relPath);
auto sessionProxy = http::Session::create(uriProxy.getHost(),
http::Session::Protocol::HttpSsl,
uriProxy.getPort());
uriProxy.setPathEtc(relPath);
LOG_DBG("uriProxy[" << uriProxy.getPathAndQuery() << ']');

auto protocol = uriProxy.getScheme() == "https" ? http::Session::Protocol::HttpSsl
: http::Session::Protocol::HttpUnencrypted;

auto sessionProxy = http::Session::create(uriProxy.getHost(), protocol, uriProxy.getPort());
sessionProxy->setTimeout(std::chrono::seconds(10));
http::Request requestProxy(uriProxy.getPathAndQuery());
http::Session::FinishedCallback proxyCallback =
Expand Down
2 changes: 2 additions & 0 deletions wsd/ProxyRequestHandler.hpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@

#pragma once

#include <Poco/URI.h>
#include <string>
#include "Socket.hpp"

Expand All @@ -20,6 +21,7 @@ class ProxyRequestHandler
static void handleRequest(const std::string& relPath,
const std::shared_ptr<StreamSocket>& socket,
const std::string& serverUri);

static std::string getProxyRatingServer() { return ProxyRatingServer; }

private:
Expand Down

0 comments on commit 067661d

Please sign in to comment.