diff options
Diffstat (limited to 'src/event')
-rw-r--r-- | src/event/BufferedSocket.cxx | 90 | ||||
-rw-r--r-- | src/event/BufferedSocket.hxx | 29 | ||||
-rw-r--r-- | src/event/FullyBufferedSocket.cxx | 132 | ||||
-rw-r--r-- | src/event/FullyBufferedSocket.hxx | 63 |
4 files changed, 202 insertions, 112 deletions
diff --git a/src/event/BufferedSocket.cxx b/src/event/BufferedSocket.cxx index dec60b5f7..05e703441 100644 --- a/src/event/BufferedSocket.cxx +++ b/src/event/BufferedSocket.cxx @@ -33,26 +33,6 @@ BufferedSocket::~BufferedSocket() } BufferedSocket::ssize_t -BufferedSocket::DirectWrite(const void *data, size_t length) -{ - const auto nbytes = SocketMonitor::Write((const char *)data, length); - if (gcc_unlikely(nbytes < 0)) { - const auto code = GetSocketError(); - if (IsSocketErrorAgain(code)) - return 0; - - Cancel(); - - if (IsSocketErrorClosed(code)) - OnSocketClosed(); - else - OnSocketError(NewSocketError(code)); - } - - return nbytes; -} - -BufferedSocket::ssize_t BufferedSocket::DirectRead(void *data, size_t length) { const auto nbytes = SocketMonitor::Read((char *)data, length); @@ -76,30 +56,6 @@ BufferedSocket::DirectRead(void *data, size_t length) } bool -BufferedSocket::WriteFromBuffer() -{ - assert(IsDefined()); - - size_t length; - const void *data = output.Read(&length); - if (data == nullptr) { - CancelWrite(); - return true; - } - - auto nbytes = DirectWrite(data, length); - if (gcc_unlikely(nbytes <= 0)) - return nbytes == 0; - - output.Consume(nbytes); - - if (output.IsEmpty()) - CancelWrite(); - - return true; -} - -bool BufferedSocket::ReadToBuffer() { assert(IsDefined()); @@ -119,38 +75,6 @@ BufferedSocket::ReadToBuffer() } bool -BufferedSocket::Write(const void *data, size_t length) -{ - assert(IsDefined()); - -#if 0 - /* TODO: disabled because this would add overhead on some callers (the ones that often), but it may be useful */ - - if (output.IsEmpty()) { - /* try to write it directly first */ - const auto nbytes = DirectWrite(data, length); - if (gcc_likely(nbytes > 0)) { - data = (const uint8_t *)data + nbytes; - length -= nbytes; - if (length == 0) - return true; - } else if (nbytes < 0) - return false; - } -#endif - - if (!output.Append(data, length)) { - // TODO - OnSocketError(g_error_new_literal(g_quark_from_static_string("buffered_socket"), - 0, "Output buffer is full")); - return false; - } - - ScheduleWrite(); - return true; -} - -bool BufferedSocket::ResumeInput() { assert(IsDefined()); @@ -220,20 +144,6 @@ BufferedSocket::OnSocketReady(unsigned flags) if (input == nullptr || !fifo_buffer_is_full(input)) ScheduleRead(); - - /* just in case the OnSocketInput() method has added - data to the output buffer: try to send it now - instead of waiting for the next event loop - iteration */ - if (!output.IsEmpty()) - flags |= WRITE; - } - - if (flags & WRITE) { - assert(!output.IsEmpty()); - - if (!WriteFromBuffer()) - return false; } return true; diff --git a/src/event/BufferedSocket.hxx b/src/event/BufferedSocket.hxx index 23fd97d38..86deb8d98 100644 --- a/src/event/BufferedSocket.hxx +++ b/src/event/BufferedSocket.hxx @@ -22,21 +22,19 @@ #include "check.h" #include "SocketMonitor.hxx" -#include "util/PeakBuffer.hxx" #include "gcc.h" struct fifo_buffer; -class EventLoop; -class BufferedSocket : private SocketMonitor { +/** + * A #SocketMonitor specialization that adds an input buffer. + */ +class BufferedSocket : protected SocketMonitor { fifo_buffer *input; - PeakBuffer output; public: - BufferedSocket(int _fd, EventLoop &_loop, - size_t normal_size, size_t peak_size=0) - :SocketMonitor(_fd, _loop), input(nullptr), - output(normal_size, peak_size) { + BufferedSocket(int _fd, EventLoop &_loop) + :SocketMonitor(_fd, _loop), input(nullptr) { ScheduleRead(); } @@ -44,19 +42,12 @@ public: using SocketMonitor::IsDefined; using SocketMonitor::Close; + using SocketMonitor::Write; private: - ssize_t DirectWrite(const void *data, size_t length); ssize_t DirectRead(void *data, size_t length); /** - * Send data from the output buffer to the socket. - * - * @return false if the socket has been closed - */ - bool WriteFromBuffer(); - - /** * Receive data from the socket to the input buffer. * * @return false if the socket has been closed @@ -67,11 +58,6 @@ protected: /** * @return false if the socket has been closed */ - bool Write(const void *data, size_t length); - - /** - * @return false if the socket has been closed - */ bool ResumeInput(); /** @@ -112,7 +98,6 @@ protected: virtual void OnSocketError(GError *error) = 0; virtual void OnSocketClosed() = 0; -private: virtual bool OnSocketReady(unsigned flags) override; }; diff --git a/src/event/FullyBufferedSocket.cxx b/src/event/FullyBufferedSocket.cxx new file mode 100644 index 000000000..a92cb68aa --- /dev/null +++ b/src/event/FullyBufferedSocket.cxx @@ -0,0 +1,132 @@ +/* + * Copyright (C) 2003-2013 The Music Player Daemon Project + * http://www.musicpd.org + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program; if not, write to the Free Software Foundation, Inc., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#include "config.h" +#include "FullyBufferedSocket.hxx" +#include "SocketError.hxx" +#include "util/fifo_buffer.h" + +#include <assert.h> +#include <stdint.h> +#include <string.h> + +#ifndef WIN32 +#include <sys/types.h> +#include <sys/socket.h> +#endif + +FullyBufferedSocket::ssize_t +FullyBufferedSocket::DirectWrite(const void *data, size_t length) +{ + const auto nbytes = SocketMonitor::Write((const char *)data, length); + if (gcc_unlikely(nbytes < 0)) { + const auto code = GetSocketError(); + if (IsSocketErrorAgain(code)) + return 0; + + Cancel(); + + if (IsSocketErrorClosed(code)) + OnSocketClosed(); + else + OnSocketError(NewSocketError(code)); + } + + return nbytes; +} + +bool +FullyBufferedSocket::WriteFromBuffer() +{ + assert(IsDefined()); + + size_t length; + const void *data = output.Read(&length); + if (data == nullptr) { + CancelWrite(); + return true; + } + + auto nbytes = DirectWrite(data, length); + if (gcc_unlikely(nbytes <= 0)) + return nbytes == 0; + + output.Consume(nbytes); + + if (output.IsEmpty()) + CancelWrite(); + + return true; +} + +bool +FullyBufferedSocket::Write(const void *data, size_t length) +{ + assert(IsDefined()); + +#if 0 + /* TODO: disabled because this would add overhead on some callers (the ones that often), but it may be useful */ + + if (output.IsEmpty()) { + /* try to write it directly first */ + const auto nbytes = DirectWrite(data, length); + if (gcc_likely(nbytes > 0)) { + data = (const uint8_t *)data + nbytes; + length -= nbytes; + if (length == 0) + return true; + } else if (nbytes < 0) + return false; + } +#endif + + if (!output.Append(data, length)) { + // TODO + OnSocketError(g_error_new_literal(g_quark_from_static_string("buffered_socket"), + 0, "Output buffer is full")); + return false; + } + + ScheduleWrite(); + return true; +} + +bool +FullyBufferedSocket::OnSocketReady(unsigned flags) +{ + const bool was_empty = output.IsEmpty(); + if (!BufferedSocket::OnSocketReady(flags)) + return false; + + if (was_empty && !output.IsEmpty()) + /* just in case the OnSocketInput() method has added + data to the output buffer: try to send it now + instead of waiting for the next event loop + iteration */ + flags |= WRITE; + + if (flags & WRITE) { + assert(!output.IsEmpty()); + + if (!WriteFromBuffer()) + return false; + } + + return true; +} diff --git a/src/event/FullyBufferedSocket.hxx b/src/event/FullyBufferedSocket.hxx new file mode 100644 index 000000000..c67c2c78d --- /dev/null +++ b/src/event/FullyBufferedSocket.hxx @@ -0,0 +1,63 @@ +/* + * Copyright (C) 2003-2013 The Music Player Daemon Project + * http://www.musicpd.org + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program; if not, write to the Free Software Foundation, Inc., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#ifndef MPD_FULLY_BUFFERED_SOCKET_HXX +#define MPD_FULLY_BUFFERED_SOCKET_HXX + +#include "check.h" +#include "BufferedSocket.hxx" +#include "util/PeakBuffer.hxx" +#include "gcc.h" + +/** + * A #BufferedSocket specialization that adds an output buffer. + */ +class FullyBufferedSocket : protected BufferedSocket { + PeakBuffer output; + +public: + FullyBufferedSocket(int _fd, EventLoop &_loop, + size_t normal_size, size_t peak_size=0) + :BufferedSocket(_fd, _loop), + output(normal_size, peak_size) { + } + + using BufferedSocket::IsDefined; + using BufferedSocket::Close; + +private: + ssize_t DirectWrite(const void *data, size_t length); + + /** + * Send data from the output buffer to the socket. + * + * @return false if the socket has been closed + */ + bool WriteFromBuffer(); + +protected: + /** + * @return false if the socket has been closed + */ + bool Write(const void *data, size_t length); + + virtual bool OnSocketReady(unsigned flags) override; +}; + +#endif |