libreoffice-online/loolwsd/LOOLWSD.cpp
2015-09-08 11:31:49 +03:00

1176 lines
39 KiB
C++

/* -*- 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/.
*
* Parts of this file is covered by:
Boost Software License - Version 1.0 - August 17th, 2003
Permission is hereby granted, free of charge, to any person or organization
obtaining a copy of the software and accompanying documentation covered by
this license (the "Software") to use, reproduce, display, distribute,
execute, and transmit the Software, and to prepare derivative works of the
Software, and to permit third-parties to whom the Software is furnished to
do so, all subject to the following:
The copyright notices in the Software and this entire statement, including
the above license grant, this restriction and the following disclaimer,
must be included in all copies of the Software, in whole or in part, and
all derivative works of the Software, unless such copies or derivative
works are solely in the form of machine-executable object code generated by
a source language processor.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
DEALINGS IN THE SOFTWARE.
*/
#include "config.h"
// This is the main source for the loolwsd program. LOOL uses several loolwsd processes: one main
// parent process that listens on the TCP port and accepts connections from LOOL clients, and a
// number of child processes, each which handles a viewing (editing) session for one document.
#include <errno.h>
#include <pwd.h>
#include <unistd.h>
#ifdef __linux
#include <sys/capability.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/prctl.h>
#endif
#include <ftw.h>
#include <utime.h>
#include <cassert>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <sstream>
#include <mutex>
#define LOK_USE_UNSTABLE_API
#include <LibreOfficeKit/LibreOfficeKitInit.h>
#include <Poco/Exception.h>
#include <Poco/File.h>
#include <Poco/Net/HTTPClientSession.h>
#include <Poco/Net/HTTPRequest.h>
#include <Poco/Net/HTTPRequestHandler.h>
#include <Poco/Net/HTTPRequestHandlerFactory.h>
#include <Poco/Net/HTTPServer.h>
#include <Poco/Net/HTTPServerParams.h>
#include <Poco/Net/HTTPServerParams.h>
#include <Poco/Net/HTTPServerRequest.h>
#include <Poco/Net/HTTPServerResponse.h>
#include <Poco/Net/NetException.h>
#include <Poco/Net/ServerSocket.h>
#include <Poco/Net/SocketAddress.h>
#include <Poco/Net/WebSocket.h>
#include <Poco/Path.h>
#include <Poco/Process.h>
#include <Poco/StringTokenizer.h>
#include <Poco/ThreadPool.h>
#include <Poco/Util/HelpFormatter.h>
#include <Poco/Util/Option.h>
#include <Poco/Util/OptionException.h>
#include <Poco/Util/OptionSet.h>
#include <Poco/Util/ServerApplication.h>
#include <Poco/Mutex.h>
#include <Poco/Net/DialogSocket.h>
#include <Poco/Net/Net.h>
#include <Poco/ThreadLocal.h>
#include <Poco/NamedMutex.h>
#include <Poco/FileStream.h>
#include "LOOLProtocol.hpp"
#include "LOOLSession.hpp"
#include "LOOLWSD.hpp"
#include "tsqueue.h"
#include "Util.hpp"
using namespace LOOLProtocol;
using Poco::Exception;
using Poco::File;
using Poco::IOException;
using Poco::Net::HTTPClientSession;
using Poco::Net::HTTPRequest;
using Poco::Net::HTTPRequestHandler;
using Poco::Net::HTTPRequestHandlerFactory;
using Poco::Net::HTTPResponse;
using Poco::Net::HTTPServer;
using Poco::Net::HTTPServerParams;
using Poco::Net::HTTPServerRequest;
using Poco::Net::HTTPServerResponse;
using Poco::Net::ServerSocket;
using Poco::Net::SocketAddress;
using Poco::Net::WebSocket;
using Poco::Net::WebSocketException;
using Poco::Path;
using Poco::Process;
using Poco::Runnable;
using Poco::StringTokenizer;
using Poco::Thread;
using Poco::ThreadPool;
using Poco::Util::Application;
using Poco::Util::HelpFormatter;
using Poco::Util::IncompatibleOptionsException;
using Poco::Util::MissingOptionException;
using Poco::Util::Option;
using Poco::Util::OptionSet;
using Poco::Util::ServerApplication;
using Poco::Net::DialogSocket;
using Poco::FastMutex;
using Poco::Net::Socket;
using Poco::ThreadLocal;
using Poco::Random;
using Poco::NamedMutex;
class QueueHandler: public Runnable
{
public:
QueueHandler(tsqueue<std::string>& queue):
_queue(queue)
{
}
void setSession(std::shared_ptr<LOOLSession> session)
{
_session = session;
}
void run() override
{
#ifdef __linux
if (prctl(PR_SET_NAME, reinterpret_cast<unsigned long>("queue_handler"), 0, 0, 0) != 0)
std::cout << Util::logPrefix() << "Cannot set thread name :" << strerror(errno) << std::endl;
#endif
while (true)
{
std::string input = _queue.get();
if (input == "eof")
break;
if (!_session->handleInput(input.c_str(), input.size()))
break;
}
}
private:
std::shared_ptr<LOOLSession> _session;
tsqueue<std::string>& _queue;
};
class WebSocketRequestHandler: public HTTPRequestHandler
/// Handle a WebSocket connection.
{
public:
WebSocketRequestHandler()
{
}
void handleRequest(HTTPServerRequest& request, HTTPServerResponse& response) override
{
#ifdef __linux
std::string thread_name;
if (request.serverAddress().port() == LOOLWSD::MASTER_PORT_NUMBER)
thread_name = "prision_socket";
else
thread_name = "client_socket";
if (prctl(PR_SET_NAME, reinterpret_cast<unsigned long>(thread_name.c_str()), 0, 0, 0) != 0)
std::cout << Util::logPrefix() << "Cannot set thread name :" << strerror(errno) << std::endl;
#endif
if(!(request.find("Upgrade") != request.end() && Poco::icompare(request["Upgrade"], "websocket") == 0))
{
response.setStatusAndReason(HTTPResponse::HTTP_BAD_REQUEST);
response.setContentLength(0);
response.send();
return;
}
tsqueue<std::string> queue;
Thread queueHandlerThread;
QueueHandler handler(queue);
try
{
try
{
std::shared_ptr<WebSocket> ws(new WebSocket(request, response));
LOOLSession::Kind kind;
if (request.getURI() == LOOLWSD::CHILD_URI && request.serverAddress().port() == LOOLWSD::MASTER_PORT_NUMBER)
kind = LOOLSession::Kind::ToPrisoner;
else
kind = LOOLSession::Kind::ToClient;
std::shared_ptr<MasterProcessSession> session(new MasterProcessSession(ws, kind));
// For ToClient sessions, we store incoming messages in a queue and have a separate
// thread that handles them. This is so that we can empty the queue when we get a
// "canceltiles" message.
if (kind == LOOLSession::Kind::ToClient)
{
handler.setSession(session);
queueHandlerThread.start(handler);
}
// Loop, receiving WebSocket messages either from the client, or from the child
// process (to be forwarded to the client).
int flags;
int n;
ws->setReceiveTimeout(0);
do
{
char buffer[100000];
n = ws->receiveFrame(buffer, sizeof(buffer), flags);
if (n > 0 && (flags & WebSocket::FRAME_OP_BITMASK) != WebSocket::FRAME_OP_CLOSE)
{
std::string firstLine = getFirstLine(buffer, n);
StringTokenizer tokens(firstLine, " ", StringTokenizer::TOK_IGNORE_EMPTY | StringTokenizer::TOK_TRIM);
if (kind == LOOLSession::Kind::ToClient && firstLine.size() == static_cast<std::string::size_type>(n))
{
// Check if it is a "canceltiles" and in that case remove outstanding
// "tile" messages from the queue.
if (tokens.count() == 1 && tokens[0] == "canceltiles")
{
queue.remove_if([](std::string& x) {
return (x.find("tile ") == 0 && x.find("id=") == std::string::npos);
});
// Also forward the "canceltiles" to the child process, if any
session->handleInput(buffer, n);
}
else if (!queue.alreadyInQueue(firstLine))
{
queue.put(firstLine);
}
}
else
{
// Check if it is a "nextmessage:" and in that case read the large
// follow-up message separately, and handle that only.
int size;
if (tokens.count() == 2 && tokens[0] == "nextmessage:" && getTokenInteger(tokens[1], "size", size) && size > 0)
{
char largeBuffer[size];
n = ws->receiveFrame(largeBuffer, size, flags);
if (n > 0 && (flags & WebSocket::FRAME_OP_BITMASK) != WebSocket::FRAME_OP_CLOSE)
{
if (!session->handleInput(largeBuffer, n))
n = 0;
}
}
else
{
if (!session->handleInput(buffer, n))
n = 0;
}
}
}
}
while (n > 0 && (flags & WebSocket::FRAME_OP_BITMASK) != WebSocket::FRAME_OP_CLOSE);
}
catch (WebSocketException& exc)
{
Application::instance().logger().error(Util::logPrefix() + "WebSocketException: " + exc.message());
switch (exc.code())
{
case WebSocket::WS_ERR_HANDSHAKE_UNSUPPORTED_VERSION:
response.set("Sec-WebSocket-Version", WebSocket::WEBSOCKET_VERSION);
// fallthrough
case WebSocket::WS_ERR_NO_HANDSHAKE:
case WebSocket::WS_ERR_HANDSHAKE_NO_VERSION:
case WebSocket::WS_ERR_HANDSHAKE_NO_KEY:
response.setStatusAndReason(HTTPResponse::HTTP_BAD_REQUEST);
response.setContentLength(0);
response.send();
break;
}
}
}
catch (IOException& exc)
{
Application::instance().logger().error(Util::logPrefix() + "IOException: " + exc.message());
}
queue.clear();
queue.put("eof");
queueHandlerThread.join();
}
};
class RequestHandlerFactory: public HTTPRequestHandlerFactory
{
public:
RequestHandlerFactory()
{
}
HTTPRequestHandler* createRequestHandler(const HTTPServerRequest& request) override
{
std::string line = (Util::logPrefix() + "Request from " +
request.clientAddress().toString() + ": " +
request.getMethod() + " " +
request.getURI() + " " +
request.getVersion());
for (HTTPServerRequest::ConstIterator it = request.begin(); it != request.end(); ++it)
{
line += " / " + it->first + ": " + it->second;
}
Application::instance().logger().information(line);
return new WebSocketRequestHandler();
}
};
class TestOutput : public Runnable
{
public:
TestOutput(WebSocket& ws) :
_ws(ws)
{
}
void run() override
{
int flags;
int n;
_ws.setReceiveTimeout(0);
try
{
do
{
char buffer[100000];
n = _ws.receiveFrame(buffer, sizeof(buffer), flags);
if (n > 0 && (flags & WebSocket::FRAME_OP_BITMASK) != WebSocket::FRAME_OP_CLOSE)
{
std::cout <<
Util::logPrefix() <<
"Client got " << n << " bytes: " << getAbbreviatedMessage(buffer, n) <<
std::endl;
}
}
while (n > 0 && (flags & WebSocket::FRAME_OP_BITMASK) != WebSocket::FRAME_OP_CLOSE);
}
catch (WebSocketException& exc)
{
Application::instance().logger().error(Util::logPrefix() + "WebSocketException: " + exc.message());
_ws.close();
}
}
private:
WebSocket& _ws;
};
class TestInput : public Runnable
{
public:
TestInput(ServerApplication& main, ServerSocket& svs, HTTPServer& srv) :
_main(main),
_svs(svs),
_srv(srv)
{
}
void run() override
{
HTTPClientSession cs("127.0.0.1", _svs.address().port());
HTTPRequest request(HTTPRequest::HTTP_GET, "/ws");
HTTPResponse response;
WebSocket ws(cs, request, response);
Thread thread;
TestOutput output(ws);
thread.start(output);
if (isatty(0))
{
std::cout << std::endl;
std::cout << "Enter LOOL WS requests, one per line. Enter EOF to finish." << std::endl;
}
while (!std::cin.eof())
{
std::string line;
std::getline(std::cin, line);
ws.sendFrame(line.c_str(), line.size());
}
thread.join();
_srv.stopAll();
_main.terminate();
}
private:
ServerApplication& _main;
ServerSocket& _svs;
HTTPServer& _srv;
};
int LOOLWSD::portNumber = DEFAULT_CLIENT_PORT_NUMBER;
int LOOLWSD::timeoutCounter = 0;
std::string LOOLWSD::cache = LOOLWSD_CACHEDIR;
std::string LOOLWSD::sysTemplate;
std::string LOOLWSD::loTemplate;
std::string LOOLWSD::childRoot;
std::string LOOLWSD::loSubPath = "lo";
std::string LOOLWSD::jail;
std::mutex LOOLWSD::_rngMutex;
Random LOOLWSD::_rng;
Poco::NamedMutex LOOLWSD::_namedMutexLOOL("loolwsd");
Poco::SharedMemory LOOLWSD::_sharedForkChild("loolwsd", sizeof(bool), Poco::SharedMemory::AM_WRITE);
int LOOLWSD::_numPreSpawnedChildren = 10;
bool LOOLWSD::doTest = false;
#if ENABLE_DEBUG
bool LOOLWSD::runningAsRoot = false;
int LOOLWSD::uid = 0;
#endif
const std::string LOOLWSD::CHILD_URI = "/loolws/child/";
const std::string LOOLWSD::PIDLOG = "/tmp/loolwsd.pid";
const std::string LOOLWSD::LOKIT_PIDLOG = "/tmp/lokit.pid";
LOOLWSD::LOOLWSD() :
_childId(0)
{
}
LOOLWSD::~LOOLWSD()
{
}
void LOOLWSD::initialize(Application& self)
{
ServerApplication::initialize(self);
}
void LOOLWSD::uninitialize()
{
ServerApplication::uninitialize();
}
void LOOLWSD::defineOptions(OptionSet& options)
{
ServerApplication::defineOptions(options);
options.addOption(Option("help", "", "Display help information on command line arguments.")
.required(false)
.repeatable(false));
options.addOption(Option("port", "", "Port number to listen to (default: " + std::to_string(DEFAULT_CLIENT_PORT_NUMBER) + "),"
" must not be " + std::to_string(MASTER_PORT_NUMBER) + ".")
.required(false)
.repeatable(false)
.argument("port number"));
options.addOption(Option("cache", "", "Path to a directory where to keep the persistent tile cache (default: " + std::string(LOOLWSD_CACHEDIR) + ").")
.required(false)
.repeatable(false)
.argument("directory"));
options.addOption(Option("systemplate", "", "Path to a template tree with shared libraries etc to be used as source for chroot jails for child processes.")
.required(false)
.repeatable(false)
.argument("directory"));
options.addOption(Option("lotemplate", "", "Path to a LibreOffice installation tree to be copied (linked) into the jails for child processes. Should be on the same file system as systemplate.")
.required(false)
.repeatable(false)
.argument("directory"));
options.addOption(Option("childroot", "", "Path to the directory under which the chroot jails for the child processes will be created. Should be on the same file system as systemplate and lotemplate.")
.required(false)
.repeatable(false)
.argument("directory"));
options.addOption(Option("losubpath", "", "Relative path where the LibreOffice installation will be copied inside a jail (default: '" + loSubPath + "').")
.required(false)
.repeatable(false)
.argument("relative path"));
options.addOption(Option("numprespawns", "", "Number of child processes to keep started in advance and waiting for new clients.")
.required(false)
.repeatable(false)
.argument("number"));
options.addOption(Option("test", "", "Interactive testing.")
.required(false)
.repeatable(false));
options.addOption(Option("child", "", "For internal use only.")
.required(false)
.repeatable(false)
.argument("child id"));
options.addOption(Option("jail", "", "For internal use only.")
.required(false)
.repeatable(false)
.argument("directory"));
#if ENABLE_DEBUG
options.addOption(Option("uid", "", "Uid to assume if running under sudo for debugging purposes.")
.required(false)
.repeatable(false)
.argument("uid"));
#endif
}
void LOOLWSD::handleOption(const std::string& name, const std::string& value)
{
ServerApplication::handleOption(name, value);
if (name == "help")
{
displayHelp();
exit(Application::EXIT_OK);
}
else if (name == "port")
portNumber = std::stoi(value);
else if (name == "cache")
cache = value;
else if (name == "systemplate")
sysTemplate = value;
else if (name == "lotemplate")
loTemplate = value;
else if (name == "childroot")
childRoot = value;
else if (name == "losubpath")
loSubPath = value;
else if (name == "numprespawns")
_numPreSpawnedChildren = std::stoi(value);
else if (name == "test")
LOOLWSD::doTest = true;
else if (name == "child")
_childId = std::stoull(value);
else if (name == "jail")
jail = value;
#if ENABLE_DEBUG
else if (name == "uid")
uid = std::stoull(value);
#endif
}
void LOOLWSD::displayHelp()
{
HelpFormatter helpFormatter(options());
helpFormatter.setCommand(commandName());
helpFormatter.setUsage("OPTIONS");
helpFormatter.setHeader("LibreOffice On-Line WebSocket server.");
helpFormatter.format(std::cout);
}
namespace
{
ThreadLocal<std::string> sourceForLinkOrCopy;
ThreadLocal<Path> destinationForLinkOrCopy;
int linkOrCopyFunction(const char *fpath,
const struct stat *sb,
int typeflag,
struct FTW *ftwbuf)
{
if (strcmp(fpath, sourceForLinkOrCopy->c_str()) == 0)
return 0;
assert(fpath[strlen(sourceForLinkOrCopy->c_str())] == '/');
const char *relativeOldPath = fpath + strlen(sourceForLinkOrCopy->c_str()) + 1;
#ifdef __APPLE__
if (strcmp(relativeOldPath, "PkgInfo") == 0)
return 0;
#endif
Path newPath(*destinationForLinkOrCopy, Path(relativeOldPath));
switch (typeflag)
{
case FTW_F:
File(newPath.parent()).createDirectories();
if (link(fpath, newPath.toString().c_str()) == -1)
{
Application::instance().logger().error(Util::logPrefix() +
"link(\"" + fpath + "\",\"" + newPath.toString() + "\") failed: " +
strerror(errno));
exit(1);
}
break;
case FTW_DP:
{
struct stat st;
if (stat(fpath, &st) == -1)
{
Application::instance().logger().error(Util::logPrefix() +
"stat(\"" + fpath + "\") failed: " +
strerror(errno));
return 1;
}
File(newPath).createDirectories();
struct utimbuf ut;
ut.actime = st.st_atime;
ut.modtime = st.st_mtime;
if (utime(newPath.toString().c_str(), &ut) == -1)
{
Application::instance().logger().error(Util::logPrefix() +
"utime(\"" + newPath.toString() + "\", &ut) failed: " +
strerror(errno));
return 1;
}
}
break;
case FTW_DNR:
Application::instance().logger().error(Util::logPrefix() +
"Cannot read directory '" + fpath + "'");
return 1;
case FTW_NS:
Application::instance().logger().error(Util::logPrefix() +
"nftw: stat failed for '" + fpath + "'");
return 1;
case FTW_SLN:
Application::instance().logger().information(Util::logPrefix() +
"nftw: symlink to nonexistent file: '" + fpath + "', ignored");
break;
default:
assert(false);
}
return 0;
}
void linkOrCopy(const std::string& source, const Path& destination)
{
*sourceForLinkOrCopy = source;
if (sourceForLinkOrCopy->back() == '/')
sourceForLinkOrCopy->pop_back();
*destinationForLinkOrCopy = destination;
if (nftw(source.c_str(), linkOrCopyFunction, 10, FTW_DEPTH) == -1)
Application::instance().logger().error(Util::logPrefix() +
"linkOrCopy: nftw() failed for '" + source + "'");
}
void dropCapability(
#ifdef __linux
cap_value_t capability
#endif
)
{
#ifdef __linux
cap_t caps;
cap_value_t cap_list[] = { capability };
caps = cap_get_proc();
if (caps == NULL)
{
Application::instance().logger().error(Util::logPrefix() + "cap_get_proc() failed: " + strerror(errno));
exit(1);
}
if (cap_set_flag(caps, CAP_EFFECTIVE, sizeof(cap_list)/sizeof(cap_list[0]), cap_list, CAP_CLEAR) == -1 ||
cap_set_flag(caps, CAP_PERMITTED, sizeof(cap_list)/sizeof(cap_list[0]), cap_list, CAP_CLEAR) == -1)
{
Application::instance().logger().error(Util::logPrefix() + "cap_set_flag() failed: " + strerror(errno));
exit(1);
}
if (cap_set_proc(caps) == -1)
{
Application::instance().logger().error(std::string("cap_set_proc() failed: ") + strerror(errno));
exit(1);
}
char *capText = cap_to_text(caps, NULL);
Application::instance().logger().information(Util::logPrefix() + "Capabilities now: " + capText);
cap_free(capText);
cap_free(caps);
#endif
// We assume that on non-Linux we don't need to be root to be able to hardlink to files we
// don't own, so drop root.
if (geteuid() == 0 && getuid() != 0)
{
// The program is setuid root. Not normal on Linux where we use setcap, but if this
// needs to run on non-Linux Unixes, setuid root is what it will bneed to be to be able
// to do chroot().
if (setuid(getuid()) != 0) {
Application::instance().logger().error(std::string("setuid() failed: ") + strerror(errno));
}
}
#if ENABLE_DEBUG
if (geteuid() == 0 && getuid() == 0)
{
#ifdef __linux
// Argh, awful hack
if (capability == CAP_FOWNER)
return;
#endif
// Running under sudo, probably because being debugged? Let's drop super-user rights.
LOOLWSD::runningAsRoot = true;
if (LOOLWSD::uid == 0)
{
struct passwd *nobody = getpwnam("nobody");
if (nobody)
LOOLWSD::uid = nobody->pw_uid;
else
LOOLWSD::uid = 65534;
}
if (setuid(LOOLWSD::uid) != 0) {
Application::instance().logger().error(std::string("setuid() failed: ") + strerror(errno));
}
}
#endif
}
}
// Writer, Impress or Calc
void LOOLWSD::componentMain()
{
#ifdef __linux
if (prctl(PR_SET_NAME, reinterpret_cast<unsigned long>("libreofficekit"), 0, 0, 0) != 0)
std::cout << Util::logPrefix() << "Cannot set thread name :" << strerror(errno) << std::endl;
#endif
try
{
_namedMutexLOOL.lock();
#ifdef __APPLE__
LibreOfficeKit *loKit(lok_init_2(("/" + loSubPath + "/Frameworks").c_str(), "file:///user"));
#else
LibreOfficeKit *loKit(lok_init_2(("/" + loSubPath + "/program").c_str(), "file:///user"));
#endif
if (!loKit)
{
logger().fatal(Util::logPrefix() + "LibreOfficeKit initialisation failed");
exit(Application::EXIT_UNAVAILABLE);
}
_namedMutexLOOL.unlock();
// Open websocket connection between the child process and the
// parent. The parent forwards us requests that it can't handle.
HTTPClientSession cs("127.0.0.1", MASTER_PORT_NUMBER);
cs.setTimeout(0);
HTTPRequest request(HTTPRequest::HTTP_GET, LOOLWSD::CHILD_URI);
HTTPResponse response;
std::shared_ptr<WebSocket> ws(new WebSocket(cs, request, response));
std::shared_ptr<ChildProcessSession> session(new ChildProcessSession(ws, loKit));
ws->setReceiveTimeout(0);
std::string hello("child " + std::to_string(_childId) + " " + std::to_string(Process::id()));
session->sendTextFrame(hello);
tsqueue<std::string> queue;
Thread queueHandlerThread;
QueueHandler handler(queue);
handler.setSession(session);
queueHandlerThread.start(handler);
int flags;
int n;
do
{
char buffer[1024];
n = ws->receiveFrame(buffer, sizeof(buffer), flags);
if (n > 0 && (flags & WebSocket::FRAME_OP_BITMASK) != WebSocket::FRAME_OP_CLOSE)
{
std::string firstLine = getFirstLine(buffer, n);
StringTokenizer tokens(firstLine, " ", StringTokenizer::TOK_IGNORE_EMPTY | StringTokenizer::TOK_TRIM);
// The only kind of messages a child process receives are the single-line ones (?)
assert(firstLine.size() == static_cast<std::string::size_type>(n));
// Check if it is a "canceltiles" and in that case remove outstanding
// "tile" messages from the queue.
if (tokens.count() == 1 && tokens[0] == "canceltiles")
{
queue.remove_if([](std::string& x) {
return (x.find("tile ") == 0 && x.find("id=") == std::string::npos);
});
}
else
{
queue.put(firstLine);
}
}
}
while (n > 0 && (flags & WebSocket::FRAME_OP_BITMASK) != WebSocket::FRAME_OP_CLOSE);
queue.clear();
queue.put("eof");
queueHandlerThread.join();
// Destroy LibreOfficeKit
loKit->pClass->destroy(loKit);
}
catch (Exception& exc)
{
logger().log(Util::logPrefix() + "Exception: " + exc.what());
}
catch (std::exception& exc)
{
logger().error(Util::logPrefix() + "Exception: " + exc.what());
}
exit(Application::EXIT_OK);
}
int LOOLWSD::createComponent()
{
int pid;
if ((pid = fork()) == -1)
{
std::cout << "Fork failed." << std::endl;
return Application::EXIT_UNAVAILABLE;
}
if (!pid)
{
componentMain();
}
MasterProcessSession::_childProcesses[pid] = pid;
return Application::EXIT_OK;
}
void LOOLWSD::startupComponent(int nComponents)
{
for (int nCntr = nComponents; nCntr; nCntr--)
{
if (createComponent() < 0)
break;
}
}
void LOOLWSD::desktopMain()
{
#ifdef __linux
if (prctl(PR_SET_NAME, reinterpret_cast<unsigned long>("loolbroker"), 0, 0, 0) != 0)
std::cout << Util::logPrefix() << "Cannot set thread name :" << strerror(errno) << std::endl;
#endif
// Initialization
std::unique_lock<std::mutex> rngLock(_rngMutex);
_childId = (((Poco::UInt64)_rng.next()) << 32) | _rng.next() | 1;
rngLock.unlock();
Path jail = Path::forDirectory(LOOLWSD::childRoot + Path::separator() + std::to_string(_childId));
File(jail).createDirectory();
Path jailLOInstallation(jail, LOOLWSD::loSubPath);
jailLOInstallation.makeDirectory();
File(jailLOInstallation).createDirectory();
// Copy (link) LO installation and other necessary files into it from the template
linkOrCopy(LOOLWSD::sysTemplate, jail);
linkOrCopy(LOOLWSD::loTemplate, jailLOInstallation);
// We need this because sometimes the hostname is not resolved
std::vector<std::string> networkFiles = {"/etc/hosts", "/etc/nsswitch.conf", "/etc/resolv.conf"};
for (std::vector<std::string>::iterator it = networkFiles.begin(); it != networkFiles.end(); ++it)
{
File networkFile(*it);
if (networkFile.exists())
{
networkFile.copyTo(Path(jail, "/etc").toString());
}
}
#ifdef __linux
// Create the urandom and random devices
File(Path(jail, "/dev")).createDirectory();
if (mknod((jail.toString() + "/dev/random").c_str(),
S_IFCHR | S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH,
makedev(1, 8)) != 0)
{
Application::instance().logger().error(Util::logPrefix() +
"mknod(" + jail.toString() + "/dev/random) failed: " +
strerror(errno));
}
if (mknod((jail.toString() + "/dev/urandom").c_str(),
S_IFCHR | S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH,
makedev(1, 9)) != 0)
{
Application::instance().logger().error(Util::logPrefix() +
"mknod(" + jail.toString() + "/dev/urandom) failed: " +
strerror(errno));
}
#endif
Application::instance().logger().information("desktopMain -> chroot(\"" + jail.toString() + "\")");
if (chroot(jail.toString().c_str()) == -1)
{
logger().error("chroot(\"" + jail.toString() + "\") failed: " + strerror(errno));
exit(Application::EXIT_UNAVAILABLE);
}
if (chdir("/") == -1)
{
logger().error(std::string("chdir(\"/\") in jail failed: ") + strerror(errno));
exit(Application::EXIT_UNAVAILABLE);
}
#ifdef __linux
dropCapability(CAP_SYS_CHROOT);
#else
dropCapability();
#endif
if (std::getenv("SLEEPFORDEBUGGER"))
{
std::cout << "Sleeping " << std::getenv("SLEEPFORDEBUGGER") << " seconds, " <<
"attach process " << Process::id() << " in debugger now." << std::endl;
Thread::sleep(std::stoul(std::getenv("SLEEPFORDEBUGGER")) * 1000);
}
startupComponent(_numPreSpawnedChildren);
while (MasterProcessSession::_childProcesses.size() > 0)
{
int status;
pid_t pid = waitpid(-1, &status, WUNTRACED | WNOHANG);
if (pid > 0)
{
if ( MasterProcessSession::_childProcesses.find(pid) != MasterProcessSession::_childProcesses.end() )
{
if ((WIFEXITED(status) || WIFSIGNALED(status) || WTERMSIG(status) ) )
{
std::cout << Util::logPrefix() << "One of our known child processes died :" << std::to_string(pid) << std::endl;
MasterProcessSession::_childProcesses.erase(pid);
}
if ( WCOREDUMP(status) )
std::cout << Util::logPrefix() << "The child produced a core dump." << std::endl;
if ( WIFSTOPPED(status) )
std::cout << Util::logPrefix() << "The child process was stopped by delivery of a signal." << std::endl;
if ( WSTOPSIG(status) )
std::cout << Util::logPrefix() << "The child process was stopped." << std::endl;
if ( WIFCONTINUED(status) )
std::cout << Util::logPrefix() << "The child process was resumed." << std::endl;
}
else
{
std::cout << Util::logPrefix() << "None of our known child processes died :" << std::to_string(pid) << std::endl;
}
}
else if (pid < 0)
std::cout << Util::logPrefix() << "Child error: " << strerror(errno);
if ( _sharedForkChild.begin()[0] > 0 )
{
_namedMutexLOOL.lock();
_sharedForkChild.begin()[0] = _sharedForkChild.begin()[0] - 1;
std::cout << Util::logPrefix() << "Create child session, fork new one" << std::endl;
_namedMutexLOOL.unlock();
if (createComponent() < 0 )
break;
}
++timeoutCounter;
if (timeoutCounter == INTERVAL_PROBES)
{
timeoutCounter = 0;
sleep(MAINTENANCE_INTERVAL);
}
}
// Terminate child processes
for (auto i : MasterProcessSession::_childProcesses)
{
logger().information(Util::logPrefix() + "Requesting child process " + std::to_string(i.first) + " to terminate");
Process::requestTermination(i.first);
}
exit(Application::EXIT_OK);
}
int LOOLWSD::createDesktop()
{
int pid;
if ((pid = fork()) == -1)
{
std::cout << "createDesktop fork failed." << std::endl;
return Application::EXIT_UNAVAILABLE;
}
if (!pid)
{
desktopMain();
}
MasterProcessSession::_childProcesses[pid] = pid;
return Application::EXIT_OK;
}
void LOOLWSD::startupDesktop(int nDesktops)
{
for (int nCntr = nDesktops; nCntr; nCntr--)
{
if (createDesktop() < 0)
break;
}
}
int LOOLWSD::main(const std::vector<std::string>& args)
{
if (access(cache.c_str(), R_OK | W_OK | X_OK) != 0)
{
std::cout << "Unable to access " << cache <<
", please make sure it exists, and has write permission for this user." << std::endl;
return Application::EXIT_UNAVAILABLE;
}
// We use the same option set for both parent and child loolwsd,
// so must check options required in the parent (but not in the
// child) separately now. Also check for options that are
// meaningless for the parent.
if (sysTemplate == "")
throw MissingOptionException("systemplate");
if (loTemplate == "")
throw MissingOptionException("lotemplate");
if (childRoot == "")
throw MissingOptionException("childroot");
if (_childId != 0)
throw IncompatibleOptionsException("child");
if (jail != "")
throw IncompatibleOptionsException("jail");
if (portNumber == MASTER_PORT_NUMBER)
throw IncompatibleOptionsException("port");
if (LOOLWSD::doTest)
_numPreSpawnedChildren = 1;
// log pid information
{
Poco::FileOutputStream filePID(LOOLWSD::PIDLOG);
if (filePID.good())
filePID << Process::id();
}
std::unique_lock<std::mutex> rngLock(_rngMutex);
_childId = (((Poco::UInt64)_rng.next()) << 32) | _rng.next() | 1;
rngLock.unlock();
_namedMutexLOOL.lock();
startupDesktop(1);
#ifdef __linux
dropCapability(CAP_SYS_CHROOT);
#else
dropCapability();
#endif
// Start a server listening on the port for clients
ServerSocket svs(portNumber, _numPreSpawnedChildren*10);
ThreadPool threadPool(_numPreSpawnedChildren*2, _numPreSpawnedChildren*5);
HTTPServer srv(new RequestHandlerFactory(), threadPool, svs, new HTTPServerParams);
srv.start();
// And one on the port for child processes
SocketAddress addr2("127.0.0.1", MASTER_PORT_NUMBER);
ServerSocket svs2(addr2, _numPreSpawnedChildren);
ThreadPool threadPool2(_numPreSpawnedChildren*2, _numPreSpawnedChildren*5);
HTTPServer srv2(new RequestHandlerFactory(), threadPool2, svs2, new HTTPServerParams);
srv2.start();
_namedMutexLOOL.unlock();
while (MasterProcessSession::_childProcesses.size() > 0)
{
int status;
pid_t pid = waitpid(-1, &status, WUNTRACED | WNOHANG);
if (pid > 0)
{
if ( MasterProcessSession::_childProcesses.find(pid) != MasterProcessSession::_childProcesses.end() )
{
if ((WIFEXITED(status) || WIFSIGNALED(status) || WTERMSIG(status) ) )
{
std::cout << Util::logPrefix() << "One of our known child processes died :" << std::to_string(pid) << std::endl;
MasterProcessSession::_childProcesses.erase(pid);
}
if ( WCOREDUMP(status) )
std::cout << Util::logPrefix() << "The child produced a core dump." << std::endl;
if ( WIFSTOPPED(status) )
std::cout << Util::logPrefix() << "The child process was stopped by delivery of a signal." << std::endl;
if ( WSTOPSIG(status) )
std::cout << Util::logPrefix() << "The child process was stopped." << std::endl;
if ( WIFCONTINUED(status) )
std::cout << Util::logPrefix() << "The child process was resumed." << std::endl;
}
else
{
std::cout << Util::logPrefix() << "None of our known child processes died :" << std::to_string(pid) << std::endl;
}
}
else if (pid < 0)
std::cout << Util::logPrefix() << "Child error: " << strerror(errno);
++timeoutCounter;
if (timeoutCounter == INTERVAL_PROBES)
{
timeoutCounter = 0;
sleep(MAINTENANCE_INTERVAL*2);
}
}
// Terminate child processes
for (auto i : MasterProcessSession::_childProcesses)
{
logger().information(Util::logPrefix() + "Requesting child process " + std::to_string(i.first) + " to terminate");
Process::requestTermination(i.first);
}
return Application::EXIT_OK;
}
bool LOOLWSD::childMode() const
{
return _childId != 0;
}
POCO_SERVER_MAIN(LOOLWSD)
/* vim:set shiftwidth=4 softtabstop=4 expandtab: */