Initial commit.

2018-02-27 18:06:05 +01:00
parent 85d9664df7
commit b2b1ba7abe
1393 changed files with 177311 additions and 1 deletions
--- a/ext/libstrawberry-common/core/messagehandler.cpp
+++ b/ext/libstrawberry-common/core/messagehandler.cpp
@@ -0,0 +1,112 @@
+/* This file is part of Strawberry.
+   Copyright 2011, David Sansome <me@davidsansome.com>
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+*/
+
+#include "messagehandler.h"
+#include "core/logging.h"
+
+#include <QAbstractSocket>
+#include <QLocalSocket>
+#include <QDataStream>
+
+_MessageHandlerBase::_MessageHandlerBase(QIODevice *device, QObject *parent)
+    : QObject(parent),
+      device_(nullptr),
+      flush_abstract_socket_(nullptr),
+      flush_local_socket_(nullptr),
+      reading_protobuf_(false),
+      expected_length_(0),
+      is_device_closed_(false) {
+  if (device) {
+    SetDevice(device);
+  }
+}
+
+void _MessageHandlerBase::SetDevice(QIODevice *device) {
+
+  device_ = device;
+
+  buffer_.open(QIODevice::ReadWrite);
+
+  connect(device, SIGNAL(readyRead()), SLOT(DeviceReadyRead()));
+
+  // Yeah I know.
+  if (QAbstractSocket *socket = qobject_cast<QAbstractSocket*>(device)) {
+    flush_abstract_socket_ = &QAbstractSocket::flush;
+    connect(socket, SIGNAL(disconnected()), SLOT(DeviceClosed()));
+  }
+  else if (QLocalSocket* socket = qobject_cast<QLocalSocket*>(device)) {
+    flush_local_socket_ = &QLocalSocket::flush;
+    connect(socket, SIGNAL(disconnected()), SLOT(DeviceClosed()));
+  }
+  else {
+    qFatal("Unsupported device type passed to _MessageHandlerBase");
+  }
+
+}
+
+void _MessageHandlerBase::DeviceReadyRead() {
+
+  while (device_->bytesAvailable()) {
+    if (!reading_protobuf_) {
+      // Read the length of the next message
+      QDataStream s(device_);
+      s >> expected_length_;
+
+      reading_protobuf_ = true;
+    }
+
+    // Read some of the message
+    buffer_.write(device_->read(expected_length_ - buffer_.size()));
+
+    // Did we get everything?
+    if (buffer_.size() == expected_length_) {
+      // Parse the message
+      if (!RawMessageArrived(buffer_.data())) {
+        qLog(Error) << "Malformed protobuf message";
+        device_->close();
+        return;
+      }
+
+      // Clear the buffer
+      buffer_.close();
+      buffer_.setData(QByteArray());
+      buffer_.open(QIODevice::ReadWrite);
+      reading_protobuf_ = false;
+    }
+  }
+
+}
+
+void _MessageHandlerBase::WriteMessage(const QByteArray &data) {
+
+  QDataStream s(device_);
+  s << quint32(data.length());
+  s.writeRawData(data.data(), data.length());
+
+  // Sorry.
+  if (flush_abstract_socket_) {
+    ((static_cast<QAbstractSocket*>(device_))->*(flush_abstract_socket_))();
+  }
+  else if (flush_local_socket_) {
+    ((static_cast<QLocalSocket*>(device_))->*(flush_local_socket_))();
+  }
+}
+
+void _MessageHandlerBase::DeviceClosed() {
+  is_device_closed_ = true;
+  AbortAll();
+}
+