Streaming clients can connect via Websockets

2025-05-31 18:06:15 +02:00 · 2020-07-02 22:26:53 +02:00 · 2020-07-02 22:26:53 +02:00 · 01ce9a60c0
commit 01ce9a60c0
parent 5723791f8a
19 changed files with 789 additions and 356 deletions
--- a/server/stream_session_ws.cpp
+++ b/server/stream_session_ws.cpp
@ -0,0 +1,125 @@
+/***
+    This file is part of snapcast
+    Copyright (C) 2014-2020  Johannes Pohl
+
+    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 3 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, see <http://www.gnu.org/licenses/>.
+***/
+
+#include "stream_session_ws.hpp"
+#include "common/aixlog.hpp"
+#include "message/pcm_chunk.hpp"
+#include <iostream>
+
+using namespace std;
+
+static constexpr auto LOG_TAG = "StreamSessionWS";
+
+
+StreamSessionWebsocket::StreamSessionWebsocket(boost::asio::io_context& ioc, MessageReceiver* receiver, websocket::stream<beast::tcp_stream>&& socket)
+    : StreamSession(ioc, receiver), ws_(std::move(socket))
+{
+    LOG(DEBUG, LOG_TAG) << "StreamSessionWS\n";
+}
+
+
+StreamSessionWebsocket::~StreamSessionWebsocket()
+{
+    LOG(DEBUG, LOG_TAG) << "~StreamSessionWS\n";
+    stop();
+}
+
+
+void StreamSessionWebsocket::start()
+{
+    // Read a message
+    LOG(DEBUG, LOG_TAG) << "start\n";
+    ws_.binary(true);
+    do_read_ws();
+}
+
+
+void StreamSessionWebsocket::stop()
+{
+    if (ws_.is_open())
+    {
+        boost::beast::error_code ec;
+        ws_.close(beast::websocket::close_code::normal, ec);
+        if (ec)
+            LOG(ERROR, LOG_TAG) << "Error in socket close: " << ec.message() << "\n";
+    }
+}
+
+
+std::string StreamSessionWebsocket::getIP()
+{
+    try
+    {
+        return ws_.next_layer().socket().remote_endpoint().address().to_string();
+    }
+    catch (...)
+    {
+        return "0.0.0.0";
+    }
+}
+
+
+void StreamSessionWebsocket::sendAsync(const shared_const_buffer& buffer, const WriteHandler& handler)
+{
+    LOG(TRACE, LOG_TAG) << "sendAsync: " << buffer.message().type << "\n";
+    ws_.async_write(buffer, boost::asio::bind_executor(strand_, [ this, self = shared_from_this(), buffer,
+                                                                  handler ](boost::system::error_code ec, std::size_t length) { handler(ec, length); }));
+}
+
+
+void StreamSessionWebsocket::do_read_ws()
+{
+    // Read a message into our buffer
+    ws_.async_read(buffer_, boost::asio::bind_executor(strand_, [ this, self = shared_from_this() ](beast::error_code ec, std::size_t bytes_transferred) {
+                       on_read_ws(ec, bytes_transferred);
+                   }));
+}
+
+
+void StreamSessionWebsocket::on_read_ws(beast::error_code ec, std::size_t bytes_transferred)
+{
+    LOG(DEBUG, LOG_TAG) << "on_read_ws, ec: " << ec << ", bytes_transferred: " << bytes_transferred << "\n";
+    boost::ignore_unused(bytes_transferred);
+
+    // This indicates that the session was closed
+    if (ec == websocket::error::closed)
+    {
+        messageReceiver_->onDisconnect(this);
+        return;
+    }
+
+    if (ec)
+    {
+        LOG(ERROR) << "ControlSessionWebsocket::on_read_ws error: " << ec.message() << "\n";
+        messageReceiver_->onDisconnect(this);
+        return;
+    }
+
+    auto data = boost::asio::buffer_cast<char*>(buffer_.data());
+    baseMessage_.deserialize(data);
+    LOG(DEBUG, LOG_TAG) << "getNextMessage: " << baseMessage_.type << ", size: " << baseMessage_.size << ", id: " << baseMessage_.id
+                        << ", refers: " << baseMessage_.refersTo << "\n";
+
+    tv t;
+    baseMessage_.received = t;
+    if (messageReceiver_ != nullptr)
+        messageReceiver_->onMessageReceived(this, baseMessage_, data + base_msg_size_);
+
+    buffer_.consume(bytes_transferred);
+    do_read_ws();
+}