2017-09-26 09:12:58 -05:00
|
|
|
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4; fill-column: 100 -*- */
|
|
|
|
/*
|
|
|
|
* This file is part of the LibreOffice project.
|
|
|
|
*
|
|
|
|
* 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"
|
|
|
|
|
2017-09-27 12:57:22 -05:00
|
|
|
#include "helpers.hpp"
|
2017-09-26 09:12:58 -05:00
|
|
|
#include "Log.hpp"
|
|
|
|
#include "Unit.hpp"
|
|
|
|
#include "UnitHTTP.hpp"
|
2018-02-08 08:24:37 -06:00
|
|
|
|
|
|
|
|
2017-09-26 09:12:58 -05:00
|
|
|
#include <Poco/DateTimeFormat.h>
|
|
|
|
#include <Poco/DateTimeFormatter.h>
|
|
|
|
#include <Poco/JSON/Object.h>
|
2018-02-08 12:51:54 -06:00
|
|
|
#include <Poco/MemoryStream.h>
|
2017-09-26 09:12:58 -05:00
|
|
|
#include <Poco/Net/HTTPRequest.h>
|
|
|
|
#include <Poco/URI.h>
|
2018-02-08 08:24:37 -06:00
|
|
|
#include <Poco/Timestamp.h>
|
2017-09-26 09:12:58 -05:00
|
|
|
|
|
|
|
class WopiTestServer : public UnitWSD
|
|
|
|
{
|
2018-02-08 12:51:54 -06:00
|
|
|
enum class LOOLStatusCode
|
|
|
|
{
|
|
|
|
DocChanged = 1010
|
|
|
|
};
|
|
|
|
|
2018-12-13 01:48:50 -06:00
|
|
|
private:
|
2017-09-27 12:57:22 -05:00
|
|
|
/// The WOPISrc URL.
|
|
|
|
std::string _wopiSrc;
|
|
|
|
|
|
|
|
/// Websocket to communicate.
|
|
|
|
std::unique_ptr<UnitWebSocket> _ws;
|
|
|
|
|
2017-10-20 11:12:05 -05:00
|
|
|
/// Content of the file.
|
|
|
|
std::string _fileContent;
|
|
|
|
|
2018-02-08 08:45:47 -06:00
|
|
|
/// Last modified time of the file
|
2019-09-27 11:56:16 -05:00
|
|
|
std::chrono::system_clock::time_point _fileLastModifiedTime;
|
2018-02-08 08:45:47 -06:00
|
|
|
|
2018-12-13 01:48:50 -06:00
|
|
|
protected:
|
|
|
|
const std::string& getWopiSrc() const { return _wopiSrc; }
|
|
|
|
|
|
|
|
const std::unique_ptr<UnitWebSocket>& getWs() const { return _ws; }
|
|
|
|
|
|
|
|
const std::string& getFileContent() const { return _fileContent; }
|
|
|
|
|
2018-02-08 12:51:54 -06:00
|
|
|
/// Sets the file content to a given value and update the last file modified time
|
|
|
|
void setFileContent(const std::string& fileContent)
|
|
|
|
{
|
|
|
|
_fileContent = fileContent;
|
2019-09-27 11:56:16 -05:00
|
|
|
_fileLastModifiedTime = std::chrono::system_clock::now();
|
2018-02-08 12:51:54 -06:00
|
|
|
}
|
|
|
|
|
2019-09-27 11:56:16 -05:00
|
|
|
const std::chrono::system_clock::time_point& getFileLastModifiedTime() const { return _fileLastModifiedTime; }
|
2019-07-25 00:58:19 -05:00
|
|
|
|
2017-09-26 09:12:58 -05:00
|
|
|
public:
|
2017-10-20 11:12:05 -05:00
|
|
|
WopiTestServer(std::string fileContent = "Hello, world")
|
2018-12-13 01:48:50 -06:00
|
|
|
: _fileContent(std::move(fileContent))
|
2017-09-26 09:12:58 -05:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2018-10-17 02:34:05 -05:00
|
|
|
void initWebsocket(const std::string& wopiName)
|
2017-09-27 12:57:22 -05:00
|
|
|
{
|
|
|
|
Poco::URI wopiURL(helpers::getTestServerURI() + wopiName);
|
|
|
|
|
|
|
|
_wopiSrc = "";
|
|
|
|
Poco::URI::encode(wopiURL.toString(), ":/?", _wopiSrc);
|
|
|
|
|
|
|
|
LOG_INF("Connecting to the fake WOPI server: /lool/" << _wopiSrc << "/ws");
|
|
|
|
|
|
|
|
_ws.reset(new UnitWebSocket("/lool/" + _wopiSrc + "/ws"));
|
|
|
|
assert(_ws.get());
|
|
|
|
}
|
|
|
|
|
2017-10-20 11:12:05 -05:00
|
|
|
virtual void assertCheckFileInfoRequest(const Poco::Net::HTTPRequest& /*request*/)
|
|
|
|
{
|
|
|
|
}
|
2017-09-26 09:12:58 -05:00
|
|
|
|
2017-10-20 11:12:05 -05:00
|
|
|
virtual void assertGetFileRequest(const Poco::Net::HTTPRequest& /*request*/)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual void assertPutFileRequest(const Poco::Net::HTTPRequest& /*request*/)
|
|
|
|
{
|
|
|
|
}
|
2017-09-26 09:12:58 -05:00
|
|
|
|
2017-10-25 07:09:27 -05:00
|
|
|
virtual void assertPutRelativeFileRequest(const Poco::Net::HTTPRequest& /*request*/)
|
2017-10-20 11:12:05 -05:00
|
|
|
{
|
|
|
|
}
|
2017-09-26 09:12:58 -05:00
|
|
|
|
2019-04-30 09:21:44 -05:00
|
|
|
virtual void assertRenameFileRequest(const Poco::Net::HTTPRequest& /*request*/)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2017-09-26 09:12:58 -05:00
|
|
|
protected:
|
|
|
|
/// Here we act as a WOPI server, so that we have a server that responds to
|
|
|
|
/// the wopi requests without additional expensive setup.
|
2018-02-08 12:51:54 -06:00
|
|
|
virtual bool handleHttpRequest(const Poco::Net::HTTPRequest& request, Poco::MemoryInputStream& message, std::shared_ptr<StreamSocket>& socket) override
|
2017-09-26 09:12:58 -05:00
|
|
|
{
|
|
|
|
Poco::URI uriReq(request.getURI());
|
2019-05-15 21:14:29 -05:00
|
|
|
Poco::RegularExpression regInfo("/wopi/files/[0-9]");
|
|
|
|
Poco::RegularExpression regContent("/wopi/files/[0-9]/contents");
|
2017-09-26 09:12:58 -05:00
|
|
|
LOG_INF("Fake wopi host request: " << uriReq.toString());
|
|
|
|
|
|
|
|
// CheckFileInfo
|
2019-05-15 21:14:29 -05:00
|
|
|
if (request.getMethod() == "GET" && regInfo.match(uriReq.getPath()))
|
2017-09-26 09:12:58 -05:00
|
|
|
{
|
|
|
|
LOG_INF("Fake wopi host request, handling CheckFileInfo: " << uriReq.getPath());
|
|
|
|
|
|
|
|
assertCheckFileInfoRequest(request);
|
|
|
|
|
|
|
|
Poco::LocalDateTime now;
|
2019-05-15 21:14:29 -05:00
|
|
|
const std::string fileName(uriReq.getPath() == "/wopi/files/3" ? "he%llo.txt" : "hello.txt");
|
2017-09-26 09:12:58 -05:00
|
|
|
Poco::JSON::Object::Ptr fileInfo = new Poco::JSON::Object();
|
2019-05-15 21:14:29 -05:00
|
|
|
fileInfo->set("BaseFileName", fileName);
|
2017-10-20 11:12:05 -05:00
|
|
|
fileInfo->set("Size", _fileContent.size());
|
2017-09-26 09:12:58 -05:00
|
|
|
fileInfo->set("Version", "1.0");
|
|
|
|
fileInfo->set("OwnerId", "test");
|
|
|
|
fileInfo->set("UserId", "test");
|
|
|
|
fileInfo->set("UserFriendlyName", "test");
|
|
|
|
fileInfo->set("UserCanWrite", "true");
|
|
|
|
fileInfo->set("PostMessageOrigin", "localhost");
|
2019-09-27 11:56:16 -05:00
|
|
|
fileInfo->set("LastModifiedTime", Util::getIso8601FracformatTime(_fileLastModifiedTime));
|
2018-02-01 22:28:20 -06:00
|
|
|
fileInfo->set("EnableOwnerTermination", "true");
|
2017-09-26 09:12:58 -05:00
|
|
|
|
|
|
|
std::ostringstream jsonStream;
|
|
|
|
fileInfo->stringify(jsonStream);
|
|
|
|
std::string responseString = jsonStream.str();
|
|
|
|
|
|
|
|
const std::string mimeType = "application/json; charset=utf-8";
|
|
|
|
|
|
|
|
std::ostringstream oss;
|
|
|
|
oss << "HTTP/1.1 200 OK\r\n"
|
2019-09-27 11:56:16 -05:00
|
|
|
"Last-Modified: " << Util::getHttpTime(_fileLastModifiedTime) << "\r\n"
|
2019-05-20 06:11:38 -05:00
|
|
|
"User-Agent: " WOPI_AGENT_STRING "\r\n"
|
|
|
|
"Content-Length: " << responseString.size() << "\r\n"
|
|
|
|
"Content-Type: " << mimeType << "\r\n"
|
|
|
|
"\r\n"
|
2017-09-26 09:12:58 -05:00
|
|
|
<< responseString;
|
|
|
|
|
|
|
|
socket->send(oss.str());
|
|
|
|
socket->shutdown();
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
// GetFile
|
2019-05-15 21:14:29 -05:00
|
|
|
else if (request.getMethod() == "GET" && regContent.match(uriReq.getPath()))
|
2017-09-26 09:12:58 -05:00
|
|
|
{
|
|
|
|
LOG_INF("Fake wopi host request, handling GetFile: " << uriReq.getPath());
|
|
|
|
|
|
|
|
assertGetFileRequest(request);
|
|
|
|
|
|
|
|
const std::string mimeType = "text/plain; charset=utf-8";
|
|
|
|
|
|
|
|
std::ostringstream oss;
|
|
|
|
oss << "HTTP/1.1 200 OK\r\n"
|
2019-09-27 11:56:16 -05:00
|
|
|
"Last-Modified: " << Util::getHttpTime(_fileLastModifiedTime) << "\r\n"
|
2019-05-20 06:11:38 -05:00
|
|
|
"User-Agent: " WOPI_AGENT_STRING "\r\n"
|
|
|
|
"Content-Length: " << _fileContent.size() << "\r\n"
|
|
|
|
"Content-Type: " << mimeType << "\r\n"
|
|
|
|
"\r\n"
|
2017-10-20 11:12:05 -05:00
|
|
|
<< _fileContent;
|
|
|
|
|
|
|
|
socket->send(oss.str());
|
|
|
|
socket->shutdown();
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
2019-05-15 21:14:29 -05:00
|
|
|
else if (request.getMethod() == "POST" && regInfo.match(uriReq.getPath()))
|
2017-10-20 11:12:05 -05:00
|
|
|
{
|
2017-10-25 07:09:27 -05:00
|
|
|
LOG_INF("Fake wopi host request, handling PutRelativeFile: " << uriReq.getPath());
|
|
|
|
std::string wopiURL = helpers::getTestServerURI() + "/something wopi/files/1?access_token=anything";
|
2019-04-30 09:21:44 -05:00
|
|
|
std::string content;
|
|
|
|
|
|
|
|
if(request.get("X-WOPI-Override") == std::string("PUT_RELATIVE"))
|
|
|
|
{
|
|
|
|
CPPUNIT_ASSERT_EQUAL(std::string("PUT_RELATIVE"), request.get("X-WOPI-Override"));
|
|
|
|
assertPutRelativeFileRequest(request);
|
|
|
|
content = "{ \"Name\":\"hello world%1.pdf\", \"Url\":\"" + wopiURL + "\" }";
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// rename file; response should be the file name without the url and the extension
|
|
|
|
CPPUNIT_ASSERT_EQUAL(std::string("RENAME_FILE"), request.get("X-WOPI-Override"));
|
|
|
|
assertRenameFileRequest(request);
|
|
|
|
content = "{ \"Name\":\"hello\", \"Url\":\"" + wopiURL + "\" }";
|
|
|
|
}
|
2017-10-20 11:12:05 -05:00
|
|
|
|
|
|
|
std::ostringstream oss;
|
|
|
|
oss << "HTTP/1.1 200 OK\r\n"
|
2019-09-27 11:56:16 -05:00
|
|
|
"Last-Modified: " << Util::getHttpTime(_fileLastModifiedTime) << "\r\n"
|
2019-05-20 06:11:38 -05:00
|
|
|
"User-Agent: " WOPI_AGENT_STRING "\r\n"
|
|
|
|
"Content-Length: " << content.size() << "\r\n"
|
|
|
|
"Content-Type: application/json\r\n"
|
|
|
|
"\r\n"
|
2017-10-20 11:12:05 -05:00
|
|
|
<< content;
|
2017-09-26 09:12:58 -05:00
|
|
|
|
|
|
|
socket->send(oss.str());
|
|
|
|
socket->shutdown();
|
|
|
|
|
2017-09-27 07:13:43 -05:00
|
|
|
return true;
|
|
|
|
}
|
2019-05-15 21:14:29 -05:00
|
|
|
else if (request.getMethod() == "POST" && regContent.match(uriReq.getPath()))
|
2017-09-27 07:13:43 -05:00
|
|
|
{
|
|
|
|
LOG_INF("Fake wopi host request, handling PutFile: " << uriReq.getPath());
|
|
|
|
|
2018-02-08 12:51:54 -06:00
|
|
|
std::string wopiTimestamp = request.get("X-LOOL-WOPI-Timestamp");
|
|
|
|
if (!wopiTimestamp.empty())
|
|
|
|
{
|
2018-02-08 13:44:18 -06:00
|
|
|
|
2019-09-27 11:56:16 -05:00
|
|
|
const std::string fileModifiedTime = Util::getIso8601FracformatTime(_fileLastModifiedTime);
|
2018-02-08 12:51:54 -06:00
|
|
|
if (wopiTimestamp != fileModifiedTime)
|
|
|
|
{
|
|
|
|
std::ostringstream oss;
|
|
|
|
oss << "HTTP/1.1 409 Conflict\r\n"
|
2019-05-20 06:11:38 -05:00
|
|
|
"User-Agent: " WOPI_AGENT_STRING "\r\n"
|
|
|
|
"\r\n"
|
|
|
|
"{\"LOOLStatusCode\":" << static_cast<int>(LOOLStatusCode::DocChanged) << "}";
|
2018-02-08 12:51:54 -06:00
|
|
|
|
|
|
|
socket->send(oss.str());
|
|
|
|
socket->shutdown();
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
std::streamsize size = request.getContentLength();
|
|
|
|
char buffer[size];
|
|
|
|
message.read(buffer, size);
|
2019-08-07 02:10:57 -05:00
|
|
|
setFileContent(std::string(buffer, size));
|
2018-02-08 12:51:54 -06:00
|
|
|
|
2017-09-27 07:13:43 -05:00
|
|
|
assertPutFileRequest(request);
|
|
|
|
|
|
|
|
std::ostringstream oss;
|
|
|
|
oss << "HTTP/1.1 200 OK\r\n"
|
2019-05-20 06:11:38 -05:00
|
|
|
"User-Agent: " WOPI_AGENT_STRING "\r\n"
|
|
|
|
"\r\n"
|
2019-09-27 11:56:16 -05:00
|
|
|
"{\"LastModifiedTime\": \"" << Util::getIso8601FracformatTime(_fileLastModifiedTime) << "\" }";
|
2017-09-27 07:13:43 -05:00
|
|
|
|
|
|
|
socket->send(oss.str());
|
|
|
|
socket->shutdown();
|
2017-09-26 09:12:58 -05:00
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
};
|
|
|
|
|
|
|
|
/* vim:set shiftwidth=4 softtabstop=4 expandtab: */
|