Implement initial multicast receive functionality in PVS daemon

1 files changed, 134 insertions, 0 deletions

diff --git a/src/net/pvsIncomingMulticastTransfer.cpp b/src/net/pvsIncomingMulticastTransfer.cpp
new file mode 100644
index 0000000..01507a9
--- /dev/null
+++ b/src/net/pvsIncomingMulticastTransfer.cpp
@@ -0,0 +1,134 @@
+/*
+# Copyright (c) 2009 - OpenSLX Project, Computer Center University of Freiburg
+#
+# This program is free software distributed under the GPL version 2.
+# See http://openslx.org/COPYING
+#
+# If you have any feedback please consult http://openslx.org/feedback and
+# send your suggestions, praise, or complaints to feedback@openslx.org
+#
+# General information about OpenSLX can be found at http://openslx.org/
+# -----------------------------------------------------------------------------
+# src/net/pcsIncomingMulticastTransfer.h
+#    - wrap McastReceiver functionality in PVS daemon
+# -----------------------------------------------------------------------------
+*/
+
+#include <QDir>
+#include <QTemporaryFile>
+#include <QTimer>
+
+#include "pvsIncomingMulticastTransfer.h"
+#include <src/net/mcast/McastReceiver.h>
+
+PVSIncomingMulticastTransfer::PVSIncomingMulticastTransfer(QString const& sender, qulonglong transferID, qulonglong size,
+        ushort port, McastConfiguration const* configTemplate, QObject* parent) :
+    QObject(parent),
+    _sender(sender),
+    _transferID(transferID),
+    _bytes(0),
+    _size(size),
+    _port(port),
+    _temporaryFile(new QTemporaryFile(QFileInfo(QDir::homePath(), "incoming.mcastft.XXXXXX").absolutePath(), this)),
+    _finalFile(0),
+    _receiver(0),
+    _config(configTemplate ?
+            new McastConfiguration(*configTemplate) :
+            new McastConfiguration()),
+    _progressTimer(new QTimer(this))
+{
+    _config->multicastUDPPortBase(port);
+    _config->multicastDPort(port);
+    _config->multicastSPort(port);
+
+    connect(_progressTimer, SIGNAL(timeout()), SLOT(updateProgress()));
+}
+
+PVSIncomingMulticastTransfer::~PVSIncomingMulticastTransfer()
+{
+    // TODO Auto-generated destructor stub
+}
+
+bool PVSIncomingMulticastTransfer::start()
+{
+    QFile *dest = _finalFile ? _finalFile : _temporaryFile;
+    _receiver = new McastReceiver(dest, new McastConfiguration(*_config), this);
+    connect(_receiver, SIGNAL(finished(int)), SLOT(receiverFinished(int)));
+    connect(_receiver, SIGNAL(progress(quint64)), SLOT(receiverProgress(quint64)));
+
+    if (!_receiver->start())
+    {
+        emit retry(_sender, _transferID);
+        return false;
+    }
+    else
+    {
+        _progressTimer->start(333);
+        return true;
+    }
+}
+
+void PVSIncomingMulticastTransfer::abort()
+{
+    delete _receiver;
+    _receiver = 0;
+
+    delete _progressTimer;
+    _progressTimer = 0;
+
+    if (_temporaryFile)
+    {
+        _temporaryFile->remove();
+    }
+    delete _temporaryFile;
+
+    if (_finalFile)
+    {
+        _finalFile->remove();
+    }
+    delete _finalFile;
+}
+
+void PVSIncomingMulticastTransfer::updatePort(ushort port)
+{
+    _config->multicastUDPPortBase(port);
+    _config->multicastSPort(port);
+    _config->multicastDPort(port);
+}
+
+void PVSIncomingMulticastTransfer::receiverProgressed(quint64 bytes)
+{
+    _bytes = bytes;
+}
+
+void PVSIncomingMulticastTransfer::receiverFinished(int how)
+{
+    switch(how)
+    {
+    case McastReceiver::RES_OK:
+        emit finished(_transferID);
+        break;
+    case McastReceiver::RES_ABORTED:
+        emit failed(_transferID, tr("Aborted"));
+        break;
+    case McastReceiver::RES_MD5_MISMATCH:
+    case McastReceiver::RES_CHECKSUM_MISMATCH:
+        emit failed(_transferID, tr("Unrecoverable data corruption"));
+        break;
+    case McastReceiver::RES_CONNECTION_RESET:
+        emit failed(_transferID, tr("Connection was reset"));
+        break;
+    case McastReceiver::RES_OFFSET_MISMATCH:
+        emit failed(_transferID, tr("Unrecoverable data loss. Try a lower transfer rate"));
+        break;
+    }
+}
+
+void PVSIncomingMulticastTransfer::updateProgress()
+{
+    if (!_started)
+    {
+        emit started(_transferID);
+    }
+    emit progress(_transferID, _bytes, _size);
+}