2016-03-31 01:48:34 -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/.
|
|
|
|
*/
|
|
|
|
|
loolwsd: include cleanup and organization
A source file (.cpp) must include its own header first.
This insures that the header is self-contained and
doesn't depend on arbitrary (and accidental) includes
before it to compile.
Furthermore, system headers should go next, followed by
C then C++ headers, then libraries (Poco, etc) and, finally,
project headers come last.
This makes sure that headers and included in the same dependency
order to avoid side-effects. For example, Poco should never rely on
anything from our project in the same way that a C header should
never rely on anything in C++, Poco, or project headers.
Also, includes ought to be sorted where possible, to improve
readability and avoid accidental duplicates (of which there
were a few).
Change-Id: I62cc1343e4a091d69195e37ed659dba20cfcb1ef
Reviewed-on: https://gerrit.libreoffice.org/25262
Reviewed-by: Ashod Nakashian <ashnakash@gmail.com>
Tested-by: Ashod Nakashian <ashnakash@gmail.com>
2016-05-21 09:23:07 -05:00
|
|
|
#include "Storage.hpp"
|
2016-04-08 06:49:33 -05:00
|
|
|
#include "config.h"
|
|
|
|
|
2016-03-31 01:48:34 -05:00
|
|
|
#include <cassert>
|
|
|
|
#include <fstream>
|
loolwsd: include cleanup and organization
A source file (.cpp) must include its own header first.
This insures that the header is self-contained and
doesn't depend on arbitrary (and accidental) includes
before it to compile.
Furthermore, system headers should go next, followed by
C then C++ headers, then libraries (Poco, etc) and, finally,
project headers come last.
This makes sure that headers and included in the same dependency
order to avoid side-effects. For example, Poco should never rely on
anything from our project in the same way that a C header should
never rely on anything in C++, Poco, or project headers.
Also, includes ought to be sorted where possible, to improve
readability and avoid accidental duplicates (of which there
were a few).
Change-Id: I62cc1343e4a091d69195e37ed659dba20cfcb1ef
Reviewed-on: https://gerrit.libreoffice.org/25262
Reviewed-by: Ashod Nakashian <ashnakash@gmail.com>
Tested-by: Ashod Nakashian <ashnakash@gmail.com>
2016-05-21 09:23:07 -05:00
|
|
|
#include <string>
|
2016-03-31 01:48:34 -05:00
|
|
|
|
loolwsd: include cleanup and organization
A source file (.cpp) must include its own header first.
This insures that the header is self-contained and
doesn't depend on arbitrary (and accidental) includes
before it to compile.
Furthermore, system headers should go next, followed by
C then C++ headers, then libraries (Poco, etc) and, finally,
project headers come last.
This makes sure that headers and included in the same dependency
order to avoid side-effects. For example, Poco should never rely on
anything from our project in the same way that a C header should
never rely on anything in C++, Poco, or project headers.
Also, includes ought to be sorted where possible, to improve
readability and avoid accidental duplicates (of which there
were a few).
Change-Id: I62cc1343e4a091d69195e37ed659dba20cfcb1ef
Reviewed-on: https://gerrit.libreoffice.org/25262
Reviewed-by: Ashod Nakashian <ashnakash@gmail.com>
Tested-by: Ashod Nakashian <ashnakash@gmail.com>
2016-05-21 09:23:07 -05:00
|
|
|
#include <Poco/JSON/Object.h>
|
|
|
|
#include <Poco/JSON/Parser.h>
|
|
|
|
#include <Poco/Net/DNS.h>
|
2016-04-08 04:24:52 -05:00
|
|
|
#include <Poco/Net/HTTPClientSession.h>
|
loolwsd: include cleanup and organization
A source file (.cpp) must include its own header first.
This insures that the header is self-contained and
doesn't depend on arbitrary (and accidental) includes
before it to compile.
Furthermore, system headers should go next, followed by
C then C++ headers, then libraries (Poco, etc) and, finally,
project headers come last.
This makes sure that headers and included in the same dependency
order to avoid side-effects. For example, Poco should never rely on
anything from our project in the same way that a C header should
never rely on anything in C++, Poco, or project headers.
Also, includes ought to be sorted where possible, to improve
readability and avoid accidental duplicates (of which there
were a few).
Change-Id: I62cc1343e4a091d69195e37ed659dba20cfcb1ef
Reviewed-on: https://gerrit.libreoffice.org/25262
Reviewed-by: Ashod Nakashian <ashnakash@gmail.com>
Tested-by: Ashod Nakashian <ashnakash@gmail.com>
2016-05-21 09:23:07 -05:00
|
|
|
#include <Poco/Net/HTTPRequest.h>
|
|
|
|
#include <Poco/Net/HTTPResponse.h>
|
2016-03-31 01:48:34 -05:00
|
|
|
#include <Poco/Net/HTTPSClientSession.h>
|
2016-04-19 03:17:50 -05:00
|
|
|
#include <Poco/Net/NetworkInterface.h>
|
2016-03-31 01:48:34 -05:00
|
|
|
#include <Poco/Net/SSLManager.h>
|
|
|
|
#include <Poco/StreamCopier.h>
|
|
|
|
|
|
|
|
#include "Auth.hpp"
|
2016-04-16 11:55:56 -05:00
|
|
|
#include "Common.hpp"
|
|
|
|
#include "Exceptions.hpp"
|
2016-07-18 06:45:36 -05:00
|
|
|
#include "LOOLWSD.hpp"
|
loolwsd: include cleanup and organization
A source file (.cpp) must include its own header first.
This insures that the header is self-contained and
doesn't depend on arbitrary (and accidental) includes
before it to compile.
Furthermore, system headers should go next, followed by
C then C++ headers, then libraries (Poco, etc) and, finally,
project headers come last.
This makes sure that headers and included in the same dependency
order to avoid side-effects. For example, Poco should never rely on
anything from our project in the same way that a C header should
never rely on anything in C++, Poco, or project headers.
Also, includes ought to be sorted where possible, to improve
readability and avoid accidental duplicates (of which there
were a few).
Change-Id: I62cc1343e4a091d69195e37ed659dba20cfcb1ef
Reviewed-on: https://gerrit.libreoffice.org/25262
Reviewed-by: Ashod Nakashian <ashnakash@gmail.com>
Tested-by: Ashod Nakashian <ashnakash@gmail.com>
2016-05-21 09:23:07 -05:00
|
|
|
#include "Log.hpp"
|
2016-04-07 15:59:27 -05:00
|
|
|
#include "Unit.hpp"
|
loolwsd: include cleanup and organization
A source file (.cpp) must include its own header first.
This insures that the header is self-contained and
doesn't depend on arbitrary (and accidental) includes
before it to compile.
Furthermore, system headers should go next, followed by
C then C++ headers, then libraries (Poco, etc) and, finally,
project headers come last.
This makes sure that headers and included in the same dependency
order to avoid side-effects. For example, Poco should never rely on
anything from our project in the same way that a C header should
never rely on anything in C++, Poco, or project headers.
Also, includes ought to be sorted where possible, to improve
readability and avoid accidental duplicates (of which there
were a few).
Change-Id: I62cc1343e4a091d69195e37ed659dba20cfcb1ef
Reviewed-on: https://gerrit.libreoffice.org/25262
Reviewed-by: Ashod Nakashian <ashnakash@gmail.com>
Tested-by: Ashod Nakashian <ashnakash@gmail.com>
2016-05-21 09:23:07 -05:00
|
|
|
#include "Util.hpp"
|
2016-03-31 01:48:34 -05:00
|
|
|
|
|
|
|
///////////////////
|
|
|
|
// StorageBase Impl
|
|
|
|
///////////////////
|
2016-06-07 02:18:49 -05:00
|
|
|
bool StorageBase::FilesystemEnabled;
|
|
|
|
bool StorageBase::WopiEnabled;
|
|
|
|
Util::RegexListMatcher StorageBase::WopiHosts;
|
2016-04-16 07:13:59 -05:00
|
|
|
|
2016-03-31 01:48:34 -05:00
|
|
|
std::string StorageBase::getLocalRootPath() const
|
|
|
|
{
|
|
|
|
auto localPath = _jailPath;
|
|
|
|
if (localPath[0] == '/')
|
|
|
|
{
|
|
|
|
// Remove the leading /
|
|
|
|
localPath.erase(0, 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
// /chroot/jailId/user/doc/childId
|
|
|
|
const auto rootPath = Poco::Path(_localStorePath, localPath);
|
|
|
|
Poco::File(rootPath).createDirectories();
|
|
|
|
|
|
|
|
return rootPath.toString();
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t StorageBase::getFileSize(const std::string& filename)
|
|
|
|
{
|
|
|
|
return std::ifstream(filename, std::ifstream::ate | std::ifstream::binary).tellg();
|
|
|
|
}
|
|
|
|
|
2016-04-16 07:13:59 -05:00
|
|
|
void StorageBase::initialize()
|
|
|
|
{
|
|
|
|
const auto& app = Poco::Util::Application::instance();
|
2016-06-07 02:18:49 -05:00
|
|
|
FilesystemEnabled = app.config().getBool("storage.filesystem[@allow]", false);
|
2016-04-16 07:13:59 -05:00
|
|
|
|
|
|
|
// Parse the WOPI settings.
|
2016-06-07 02:18:49 -05:00
|
|
|
WopiHosts.clear();
|
|
|
|
WopiEnabled = app.config().getBool("storage.wopi[@allow]", false);
|
|
|
|
if (WopiEnabled)
|
2016-04-16 07:13:59 -05:00
|
|
|
{
|
|
|
|
for (size_t i = 0; ; ++i)
|
|
|
|
{
|
|
|
|
const std::string path = "storage.wopi.host[" + std::to_string(i) + "]";
|
2016-04-16 17:12:10 -05:00
|
|
|
const auto host = app.config().getString(path, "");
|
|
|
|
if (!host.empty())
|
2016-04-16 07:13:59 -05:00
|
|
|
{
|
2016-04-16 17:12:10 -05:00
|
|
|
if (app.config().getBool(path + "[@allow]", false))
|
2016-04-16 07:13:59 -05:00
|
|
|
{
|
|
|
|
Log::info("Adding trusted WOPI host: [" + host + "].");
|
2016-06-07 02:18:49 -05:00
|
|
|
WopiHosts.allow(host);
|
2016-04-16 17:12:10 -05:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2016-04-18 20:13:17 -05:00
|
|
|
Log::info("Adding blocked WOPI host: [" + host + "].");
|
2016-06-07 02:18:49 -05:00
|
|
|
WopiHosts.deny(host);
|
2016-04-16 07:13:59 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else if (!app.config().has(path))
|
|
|
|
{
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-04-19 03:17:50 -05:00
|
|
|
bool isLocalhost(const std::string& targetHost)
|
|
|
|
{
|
2016-04-19 08:32:22 -05:00
|
|
|
std::string targetAddress;
|
|
|
|
try
|
|
|
|
{
|
|
|
|
targetAddress = Poco::Net::DNS::resolveOne(targetHost).toString();
|
|
|
|
}
|
|
|
|
catch (const Poco::Exception& exc)
|
|
|
|
{
|
|
|
|
Log::warn("Poco::Net::DNS::resolveOne(\"" + targetHost + "\") failed: " + exc.displayText());
|
|
|
|
try
|
|
|
|
{
|
|
|
|
targetAddress = Poco::Net::IPAddress(targetHost).toString();
|
|
|
|
}
|
|
|
|
catch (const Poco::Exception& exc1)
|
|
|
|
{
|
|
|
|
Log::warn("Poco::Net::IPAddress(\"" + targetHost + "\") failed: " + exc1.displayText());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-04-19 03:17:50 -05:00
|
|
|
Poco::Net::NetworkInterface::NetworkInterfaceList list = Poco::Net::NetworkInterface::list(true,true);
|
2016-04-28 01:43:19 -05:00
|
|
|
for (auto& netif : list)
|
2016-04-19 03:17:50 -05:00
|
|
|
{
|
|
|
|
std::string address = netif.address().toString();
|
2016-04-26 02:34:32 -05:00
|
|
|
address = address.substr(0, address.find('%', 0));
|
2016-04-19 03:17:50 -05:00
|
|
|
if (address == targetAddress)
|
2016-04-19 08:32:22 -05:00
|
|
|
{
|
|
|
|
Log::info("WOPI host is on the same host as the WOPI client: \"" + targetAddress + "\". Connection is allowed.");
|
2016-04-19 03:17:50 -05:00
|
|
|
return true;
|
2016-04-19 08:32:22 -05:00
|
|
|
}
|
2016-04-19 03:17:50 -05:00
|
|
|
}
|
2016-04-19 08:32:22 -05:00
|
|
|
Log::info("WOPI host is not on the same host as the WOPI client: \"" + targetAddress + "\". Connection is not allowed.");
|
2016-04-19 03:17:50 -05:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2016-04-07 15:59:27 -05:00
|
|
|
std::unique_ptr<StorageBase> StorageBase::create(const std::string& jailRoot, const std::string& jailPath, const Poco::URI& uri)
|
|
|
|
{
|
|
|
|
std::unique_ptr<StorageBase> storage;
|
|
|
|
|
2016-04-09 11:30:48 -05:00
|
|
|
if (UnitWSD::get().createStorage(jailRoot, jailPath, uri, storage))
|
2016-04-16 07:13:59 -05:00
|
|
|
{
|
|
|
|
Log::info("Storage load hooked.");
|
|
|
|
}
|
2016-04-07 15:59:27 -05:00
|
|
|
else if (uri.isRelative() || uri.getScheme() == "file")
|
|
|
|
{
|
2016-04-16 11:55:56 -05:00
|
|
|
Log::info("Public URI [" + uri.toString() + "] is a file.");
|
2016-06-07 02:18:49 -05:00
|
|
|
if (FilesystemEnabled)
|
2016-04-07 15:59:27 -05:00
|
|
|
{
|
2016-04-16 11:55:56 -05:00
|
|
|
return std::unique_ptr<StorageBase>(new LocalStorage(jailRoot, jailPath, uri.getPath()));
|
2016-04-07 15:59:27 -05:00
|
|
|
}
|
|
|
|
|
2016-07-06 07:31:53 -05:00
|
|
|
Log::error("Local Storage is disabled by default. Enable in the config file or on the command-line to enable.");
|
2016-04-07 15:59:27 -05:00
|
|
|
}
|
2016-06-07 02:18:49 -05:00
|
|
|
else if (WopiEnabled)
|
2016-04-07 15:59:27 -05:00
|
|
|
{
|
2016-04-16 11:55:56 -05:00
|
|
|
Log::info("Public URI [" + uri.toString() + "] considered WOPI.");
|
2016-04-26 02:34:32 -05:00
|
|
|
const auto& targetHost = uri.getHost();
|
2016-06-07 02:18:49 -05:00
|
|
|
if (WopiHosts.match(targetHost) || isLocalhost(targetHost))
|
2016-04-16 11:55:56 -05:00
|
|
|
{
|
2016-04-18 20:13:17 -05:00
|
|
|
return std::unique_ptr<StorageBase>(new WopiStorage(jailRoot, jailPath, uri.toString()));
|
2016-04-16 11:55:56 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
Log::error("No acceptable WOPI hosts found matching the target host [" + targetHost + "] in config.");
|
2016-04-16 07:13:59 -05:00
|
|
|
}
|
|
|
|
|
2016-04-16 11:55:56 -05:00
|
|
|
throw BadRequestException("No Storage configured or invalid URI.");
|
2016-04-07 15:59:27 -05:00
|
|
|
}
|
|
|
|
|
2016-03-31 01:48:34 -05:00
|
|
|
////////////////////
|
|
|
|
// LocalStorage Impl
|
|
|
|
/////////////////////
|
|
|
|
StorageBase::FileInfo LocalStorage::getFileInfo(const Poco::URI& uri)
|
|
|
|
{
|
2016-04-17 10:05:39 -05:00
|
|
|
const auto path = Poco::Path(uri.getPath());
|
|
|
|
Log::debug("Getting info for local uri [" + uri.toString() + "], path [" + path.toString() + "].");
|
2016-04-26 02:34:32 -05:00
|
|
|
const auto& filename = path.getFileName();
|
2016-04-17 10:05:39 -05:00
|
|
|
const auto file = Poco::File(path);
|
|
|
|
const auto lastModified = file.getLastModified();
|
|
|
|
const auto size = file.getSize();
|
2016-03-31 01:48:34 -05:00
|
|
|
return FileInfo({filename, lastModified, size});
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string LocalStorage::loadStorageFileToLocal()
|
|
|
|
{
|
|
|
|
const auto rootPath = getLocalRootPath();
|
|
|
|
|
|
|
|
// /chroot/jailId/user/doc/childId/file.ext
|
|
|
|
const auto filename = Poco::Path(_uri).getFileName();
|
|
|
|
_jailedFilePath = Poco::Path(rootPath, filename).toString();
|
|
|
|
|
|
|
|
Log::info("Public URI [" + _uri +
|
|
|
|
"] jailed to [" + _jailedFilePath + "].");
|
|
|
|
|
|
|
|
const auto publicFilePath = _uri;
|
|
|
|
Log::info("Linking " + publicFilePath + " to " + _jailedFilePath);
|
|
|
|
if (!Poco::File(_jailedFilePath).exists() && link(publicFilePath.c_str(), _jailedFilePath.c_str()) == -1)
|
|
|
|
{
|
|
|
|
// Failed
|
|
|
|
Log::warn("link(\"" + publicFilePath + "\", \"" + _jailedFilePath + "\") failed. Will copy.");
|
|
|
|
}
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
|
|
|
// Fallback to copying.
|
|
|
|
if (!Poco::File(_jailedFilePath).exists())
|
|
|
|
{
|
|
|
|
Log::info("Copying " + publicFilePath + " to " + _jailedFilePath);
|
|
|
|
Poco::File(publicFilePath).copyTo(_jailedFilePath);
|
|
|
|
_isCopy = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
catch (const Poco::Exception& exc)
|
|
|
|
{
|
|
|
|
Log::error("copyTo(\"" + publicFilePath + "\", \"" + _jailedFilePath + "\") failed: " + exc.displayText());
|
|
|
|
throw;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Now return the jailed path.
|
|
|
|
return Poco::Path(_jailPath, filename).toString();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool LocalStorage::saveLocalFileToStorage()
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
// Copy the file back.
|
|
|
|
if (_isCopy && Poco::File(_jailedFilePath).exists())
|
|
|
|
{
|
|
|
|
Log::info("Copying " + _jailedFilePath + " to " + _uri);
|
|
|
|
Poco::File(_jailedFilePath).copyTo(_uri);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
catch (const Poco::Exception& exc)
|
|
|
|
{
|
|
|
|
Log::error("copyTo(\"" + _jailedFilePath + "\", \"" + _uri + "\") failed: " + exc.displayText());
|
|
|
|
throw;
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2016-07-18 06:45:36 -05:00
|
|
|
namespace {
|
|
|
|
|
|
|
|
static inline
|
|
|
|
Poco::Net::HTTPClientSession* lcl_getHTTPClientSession(const Poco::URI& uri)
|
|
|
|
{
|
|
|
|
return (LOOLWSD::isSSLEnabled()) ? new Poco::Net::HTTPSClientSession(uri.getHost(), uri.getPort(), Poco::Net::SSLManager::instance().defaultClientContext())
|
|
|
|
: new Poco::Net::HTTPClientSession(uri.getHost(), uri.getPort());
|
|
|
|
}
|
|
|
|
|
|
|
|
} // anonymous namespace
|
|
|
|
|
2016-03-31 01:48:34 -05:00
|
|
|
///////////////////
|
|
|
|
// WopiStorage Impl
|
|
|
|
///////////////////
|
|
|
|
StorageBase::FileInfo WopiStorage::getFileInfo(const Poco::URI& uri)
|
|
|
|
{
|
|
|
|
Log::debug("Getting info for wopi uri [" + uri.toString() + "].");
|
|
|
|
|
2016-07-18 06:45:36 -05:00
|
|
|
std::unique_ptr<Poco::Net::HTTPClientSession> psession(lcl_getHTTPClientSession(uri));
|
|
|
|
|
2016-04-15 01:49:35 -05:00
|
|
|
Poco::Net::HTTPRequest request(Poco::Net::HTTPRequest::HTTP_GET, uri.getPathAndQuery(), Poco::Net::HTTPMessage::HTTP_1_1);
|
2016-03-31 01:48:34 -05:00
|
|
|
request.set("User-Agent", "LOOLWSD WOPI Agent");
|
2016-07-18 06:45:36 -05:00
|
|
|
psession->sendRequest(request);
|
2016-03-31 01:48:34 -05:00
|
|
|
|
|
|
|
Poco::Net::HTTPResponse response;
|
2016-07-18 06:45:36 -05:00
|
|
|
std::istream& rs = psession->receiveResponse(response);
|
2016-03-31 01:48:34 -05:00
|
|
|
|
|
|
|
auto logger = Log::trace();
|
|
|
|
logger << "WOPI::CheckFileInfo header for URI [" << uri.toString() << "]:\n";
|
|
|
|
for (auto& pair : response)
|
|
|
|
{
|
|
|
|
logger << '\t' + pair.first + ": " + pair.second << " / ";
|
|
|
|
}
|
|
|
|
|
|
|
|
logger << Log::end;
|
|
|
|
|
|
|
|
// Parse the response.
|
|
|
|
std::string filename;
|
|
|
|
size_t size = 0;
|
|
|
|
std::string resMsg;
|
|
|
|
Poco::StreamCopier::copyToString(rs, resMsg);
|
|
|
|
Log::debug("WOPI::CheckFileInfo returned: " + resMsg);
|
2016-04-15 01:49:35 -05:00
|
|
|
const auto index = resMsg.find_first_of('{');
|
2016-03-31 01:48:34 -05:00
|
|
|
if (index != std::string::npos)
|
|
|
|
{
|
|
|
|
const std::string stringJSON = resMsg.substr(index);
|
|
|
|
Poco::JSON::Parser parser;
|
|
|
|
const auto result = parser.parse(stringJSON);
|
2016-04-15 01:49:35 -05:00
|
|
|
const auto& object = result.extract<Poco::JSON::Object::Ptr>();
|
2016-03-31 01:48:34 -05:00
|
|
|
filename = object->get("BaseFileName").toString();
|
|
|
|
size = std::stoul (object->get("Size").toString(), nullptr, 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
// WOPI doesn't support file last modified time.
|
|
|
|
return FileInfo({filename, Poco::Timestamp(), size});
|
|
|
|
}
|
|
|
|
|
|
|
|
/// uri format: http://server/<...>/wopi*/files/<id>/content
|
|
|
|
std::string WopiStorage::loadStorageFileToLocal()
|
|
|
|
{
|
|
|
|
Log::info("Downloading URI [" + _uri + "].");
|
|
|
|
|
|
|
|
_fileInfo = getFileInfo(Poco::URI(_uri));
|
2016-04-19 02:10:07 -05:00
|
|
|
if (_fileInfo._size == 0 && _fileInfo._filename.empty())
|
2016-03-31 01:48:34 -05:00
|
|
|
{
|
|
|
|
//TODO: Should throw a more appropriate exception.
|
|
|
|
throw std::runtime_error("Failed to load file from storage.");
|
|
|
|
}
|
|
|
|
|
|
|
|
// WOPI URI to download files ends in '/contents'.
|
|
|
|
// Add it here to get the payload instead of file info.
|
|
|
|
Poco::URI uriObject(_uri);
|
|
|
|
const auto url = uriObject.getPath() + "/contents?" + uriObject.getQuery();
|
|
|
|
Log::debug("Wopi requesting: " + url);
|
|
|
|
|
2016-07-18 06:45:36 -05:00
|
|
|
std::unique_ptr<Poco::Net::HTTPClientSession> psession(lcl_getHTTPClientSession(uriObject));
|
|
|
|
|
2016-03-31 01:48:34 -05:00
|
|
|
Poco::Net::HTTPRequest request(Poco::Net::HTTPRequest::HTTP_GET, url, Poco::Net::HTTPMessage::HTTP_1_1);
|
|
|
|
request.set("User-Agent", "LOOLWSD WOPI Agent");
|
2016-07-18 06:45:36 -05:00
|
|
|
psession->sendRequest(request);
|
2016-03-31 01:48:34 -05:00
|
|
|
|
|
|
|
Poco::Net::HTTPResponse response;
|
2016-07-18 06:45:36 -05:00
|
|
|
std::istream& rs = psession->receiveResponse(response);
|
2016-03-31 01:48:34 -05:00
|
|
|
|
|
|
|
auto logger = Log::trace();
|
|
|
|
logger << "WOPI::GetFile header for URI [" << _uri << "]:\n";
|
|
|
|
for (auto& pair : response)
|
|
|
|
{
|
|
|
|
logger << '\t' + pair.first + ": " + pair.second << " / ";
|
|
|
|
}
|
|
|
|
|
|
|
|
logger << Log::end;
|
|
|
|
|
2016-04-19 02:10:07 -05:00
|
|
|
_jailedFilePath = Poco::Path(getLocalRootPath(), _fileInfo._filename).toString();
|
2016-03-31 01:48:34 -05:00
|
|
|
std::ofstream ofs(_jailedFilePath);
|
|
|
|
std::copy(std::istreambuf_iterator<char>(rs),
|
|
|
|
std::istreambuf_iterator<char>(),
|
|
|
|
std::ostreambuf_iterator<char>(ofs));
|
|
|
|
const auto size = getFileSize(_jailedFilePath);
|
|
|
|
|
|
|
|
Log::info() << "WOPI::GetFile downloaded " << size << " bytes from [" << _uri
|
|
|
|
<< "] -> " << _jailedFilePath << ": "
|
|
|
|
<< response.getStatus() << " " << response.getReason() << Log::end;
|
|
|
|
|
|
|
|
// Now return the jailed path.
|
2016-04-19 02:10:07 -05:00
|
|
|
return Poco::Path(_jailPath, _fileInfo._filename).toString();
|
2016-03-31 01:48:34 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
bool WopiStorage::saveLocalFileToStorage()
|
|
|
|
{
|
|
|
|
Log::info("Uploading URI [" + _uri + "] from [" + _jailedFilePath + "].");
|
|
|
|
const auto size = getFileSize(_jailedFilePath);
|
|
|
|
|
|
|
|
Poco::URI uriObject(_uri);
|
|
|
|
const auto url = uriObject.getPath() + "/contents?" + uriObject.getQuery();
|
|
|
|
Log::debug("Wopi posting: " + url);
|
|
|
|
|
2016-07-18 06:45:36 -05:00
|
|
|
std::unique_ptr<Poco::Net::HTTPClientSession> psession(lcl_getHTTPClientSession(uriObject));
|
|
|
|
|
2016-03-31 01:48:34 -05:00
|
|
|
Poco::Net::HTTPRequest request(Poco::Net::HTTPRequest::HTTP_POST, url, Poco::Net::HTTPMessage::HTTP_1_1);
|
|
|
|
request.set("X-WOPIOverride", "PUT");
|
|
|
|
request.setContentType("application/octet-stream");
|
|
|
|
request.setContentLength(size);
|
|
|
|
|
2016-07-18 06:45:36 -05:00
|
|
|
std::ostream& os = psession->sendRequest(request);
|
2016-03-31 01:48:34 -05:00
|
|
|
std::ifstream ifs(_jailedFilePath);
|
|
|
|
Poco::StreamCopier::copyStream(ifs, os);
|
|
|
|
|
|
|
|
Poco::Net::HTTPResponse response;
|
2016-07-18 06:45:36 -05:00
|
|
|
std::istream& rs = psession->receiveResponse(response);
|
2016-03-31 01:48:34 -05:00
|
|
|
std::ostringstream oss;
|
|
|
|
Poco::StreamCopier::copyStream(rs, oss);
|
|
|
|
|
|
|
|
Log::info("WOPI::PutFile response: " + oss.str());
|
|
|
|
const auto success = (response.getStatus() == Poco::Net::HTTPResponse::HTTP_OK);
|
|
|
|
Log::info() << "WOPI::PutFile uploaded " << size << " bytes from [" << _jailedFilePath << "]:"
|
2016-06-28 01:41:10 -05:00
|
|
|
<< "] -> [" << url << "]: "
|
2016-03-31 01:48:34 -05:00
|
|
|
<< response.getStatus() << " " << response.getReason() << Log::end;
|
|
|
|
|
|
|
|
return success;
|
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////
|
|
|
|
// WebDAVStorage Impl
|
|
|
|
///////////////////////
|
|
|
|
StorageBase::FileInfo WebDAVStorage::getFileInfo(const Poco::URI& uri)
|
|
|
|
{
|
|
|
|
Log::debug("Getting info for webdav uri [" + uri.toString() + "].");
|
|
|
|
(void)uri;
|
|
|
|
assert(!"Not Implemented!");
|
|
|
|
return FileInfo({"bazinga", Poco::Timestamp(), 0});
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string WebDAVStorage::loadStorageFileToLocal()
|
|
|
|
{
|
|
|
|
// TODO: implement webdav GET.
|
|
|
|
return _uri;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool WebDAVStorage::saveLocalFileToStorage()
|
|
|
|
{
|
|
|
|
// TODO: implement webdav PUT.
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* vim:set shiftwidth=4 softtabstop=4 expandtab: */
|