libreoffice-online/loolwsd/IoUtil.hpp
Tor Lillqvist bfcf9756f5 Don't assert on PING frames that we send
UnitPrefork got what I assume is one of those PING frames that
ChildProcess::isAlive() sends before the actual reply when it sent the
"unit-memdump" message, and did not like it. Uncommenting the line
that outputs the "memory stats" message it expects showed:

Got memory stats 'PING'

Followed by:

Assertion `tokens.count() == 2' failed.

Fix by factoring out the handling of PING frames, PONG frames, and the
pseudo-PONG frames that we send ourselves in reply to PING frames into
a new function IoUtil::receiveFrame(). Use that in a couple of
places. (Probably should use in many more places.)

Getting past this then leads to later cppunit tests again being run,
and their failures then again showing up...
2016-05-03 09:50:38 +03:00

75 lines
2.4 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/.
*/
#ifndef INCLUDED_IOUTIL_HPP
#define INCLUDED_IOUTIL_HPP
#include <functional>
#include <string>
#include <memory>
#include <sys/poll.h>
#include <Poco/Net/WebSocket.h>
#include <Poco/Logger.h>
namespace IoUtil
{
// Wrapper for WebSocket::receiveFrame() that handles PING frames (by replying with a
// "pseudo-PONG" frame, see protocol.txt) and PONG frames. Also our "pseudo-PONG" frames are
// ignored.
// Should we also factor out the handling of non-final and continuation frames into this?
int receiveFrame(Poco::Net::WebSocket& socket, void* buffer, int length, int& flags);
/// Synchronously process WebSocket requests and dispatch to handler.
//. Handler returns false to end.
void SocketProcessor(const std::shared_ptr<Poco::Net::WebSocket>& ws,
const std::function<bool(const std::vector<char>&)>& handler,
const std::function<void()>& closeFrame,
const std::function<bool()>& stopPredicate);
/// Call WebSocket::shutdown() ignoring Poco::IOException.
void shutdownWebSocket(const std::shared_ptr<Poco::Net::WebSocket>& ws);
ssize_t writeFIFO(int pipe, const char* buffer, ssize_t size);
inline
ssize_t writeFIFO(int pipe, const std::string& message)
{
return writeFIFO(pipe, message.c_str(), message.size());
}
ssize_t readFIFO(int pipe, char* buffer, ssize_t size);
class PipeReader
{
public:
PipeReader(const std::string& name, const int pipe) :
_name(name),
_pipe(pipe)
{
}
const std::string& getName() const { return _name; }
/// Reads a single line from the pipe.
/// Returns 0 for timeout, <0 for error, and >0 on success.
/// On success, line will contain the read message.
int readLine(std::string& line,
const std::function<bool()>& stopPredicate);
private:
const std::string _name;
const int _pipe;
std::string _data;
};
}
#endif
/* vim:set shiftwidth=4 softtabstop=4 expandtab: */