Implement multicast transfer protocol.

7 files changed, 574 insertions, 0 deletions

diff --git a/src/net/mcast/trial_programs/CMakeLists.txt b/src/net/mcast/trial_programs/CMakeLists.txt
new file mode 100644
index 0000000..d0f68fa
--- /dev/null
+++ b/src/net/mcast/trial_programs/CMakeLists.txt
@@ -0,0 +1,38 @@
+INCLUDE(${QT_USE_FILE})
+
+QT4_WRAP_CPP(
+    mcastsend_MOC
+    mcastsend.h
+)
+
+QT4_WRAP_CPP(
+    mcastreceive_MOC
+    mcastreceive.h
+)
+
+SET(argparser_SRC
+    McastConfigArgParser.h
+    McastConfigArgParser.cpp
+)
+
+ADD_EXECUTABLE(mcastsend
+    mcastsend.cpp
+    mcastsend.h
+    ${argparser_SRC}
+    ${mcastsend_MOC}
+)
+
+ADD_EXECUTABLE(mcastreceive
+    mcastreceive.cpp
+    mcastreceive.h
+    ${argparser_SRC}
+    ${mcastreceive_MOC}
+)
+
+TARGET_LINK_LIBRARIES(mcastsend
+    pvsmcast
+)
+
+TARGET_LINK_LIBRARIES(mcastreceive
+    pvsmcast
+)
\ No newline at end of file
diff --git a/src/net/mcast/trial_programs/McastConfigArgParser.cpp b/src/net/mcast/trial_programs/McastConfigArgParser.cpp
new file mode 100644
index 0000000..8849544
--- /dev/null
+++ b/src/net/mcast/trial_programs/McastConfigArgParser.cpp
@@ -0,0 +1,151 @@
+/*
+# 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/mcast/trial_programs/McastConfigArgParser.cpp
+#    - Parse common Multicast Configuration CLI arguments
+# -----------------------------------------------------------------------------
+*/
+
+#include <iostream>
+
+#include <QCoreApplication>
+
+#include "McastConfigArgParser.h"
+
+using namespace std;
+
+bool parseMcastConfigArg(QStringList::iterator& i, QStringList::iterator const& end, McastConfiguration* config)
+{
+	QString arg = *i;
+
+	if (arg == "-addr")
+	{
+		i++;
+		if(i == end)
+		{
+			cerr << "Option " << arg.toLatin1().constData() << " is missing argument" << endl;
+			return false;
+		}
+		config->multicastAddress(*i);
+	}
+	else if (arg == "-dport")
+	{
+		i++;
+		if(i == end)
+		{
+			cerr << "Option " << arg.toLatin1().constData() << " is missing argument" << endl;
+			return false;
+		}
+		bool ok;
+		quint16 dport = (quint16)i->toInt(&ok);
+		if (!ok)
+		{
+			cerr << "Error: dport is not an integer" << endl;
+			return false;
+		}
+		config->multicastDPort(dport);
+	}
+	else if (arg == "-sport")
+	{
+		i++;
+		if(i == end)
+		{
+			cerr << "Option " << arg.toLatin1().constData() << " is missing argument" << endl;
+			return false;
+		}
+		bool ok;
+		quint16 sport = (quint16)i->toInt(&ok);
+		if (!ok)
+		{
+			cerr << "Error: sport is not an integer" << endl;
+			return false;
+		}
+		config->multicastSPort(sport);
+	}
+	else if (arg == "-mtu")
+	{
+		i++;
+		if(i == end)
+		{
+			cerr << "Option " << arg.toLatin1().constData() << " is missing argument" << endl;
+			return false;
+		}
+		bool ok;
+		quint16 mtu = (quint16)i->toInt(&ok);
+		if (!ok)
+		{
+			cerr << "Error: MTU is not an integer" << endl;
+			return false;
+		}
+		config->multicastMTU(mtu);
+	}
+	else if (arg == "-rate")
+	{
+		i++;
+		if(i == end)
+		{
+			cerr << "Option " << arg.toLatin1().constData() << " is missing argument" << endl;
+			return false;
+		}
+		bool ok;
+		quint32 rate = i->toInt(&ok);
+		if (!ok)
+		{
+			cerr << "Error: Rate is not an integer" << endl;
+			return false;
+		}
+		config->multicastRate(rate);
+	}
+	else if (arg == "-winsize")
+	{
+		i++;
+		if(i == end)
+		{
+			cerr << "Option " << arg.toLatin1().constData() << " is missing argument" << endl;
+			return false;
+		}
+		bool ok;
+		quint16 winsize = (quint16)i->toInt(&ok);
+		if (!ok)
+		{
+			cerr << "Error: Winsize is not an integer" << endl;
+			return false;
+		}
+		config->multicastWinSize(winsize);
+	}
+	else if (arg == "-udp")
+	{
+		config->multicastUseUDP(true);
+	}
+	else if (arg == "-udp-port")
+	{
+		i++;
+		if(i == end)
+		{
+			cerr << "Option " << arg.toLatin1().constData() << " is missing argument" << endl;
+			return false;
+		}
+		bool ok;
+		quint16 udpport = (quint16)i->toInt(&ok);
+		if (!ok)
+		{
+			cerr << "Error: UDP-Port is not an integer" << endl;
+			return false;
+		}
+		config->multicastUDPPort(udpport);
+	}
+	else
+	{
+		return false;
+	}
+
+	return true;
+}
diff --git a/src/net/mcast/trial_programs/McastConfigArgParser.h b/src/net/mcast/trial_programs/McastConfigArgParser.h
new file mode 100644
index 0000000..4fb18a7
--- /dev/null
+++ b/src/net/mcast/trial_programs/McastConfigArgParser.h
@@ -0,0 +1,26 @@
+/*
+# 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/mcast/trial_programs/McastConfigArgParser.h
+#    - Parse common Multicast Configuration CLI arguments
+# -----------------------------------------------------------------------------
+*/
+
+#ifndef MCASTCONFIGARGPARSER_H_
+#define MCASTCONFIGARGPARSER_H_
+
+#include <QStringList>
+
+#include "../McastConfiguration.h"
+
+bool parseMcastConfigArg(QStringList::iterator& i, QStringList::iterator const& end, McastConfiguration* config);
+
+#endif /* MCASTCONFIGARGPARSER_H_ */
diff --git a/src/net/mcast/trial_programs/mcastreceive.cpp b/src/net/mcast/trial_programs/mcastreceive.cpp
new file mode 100644
index 0000000..48a0f10
--- /dev/null
+++ b/src/net/mcast/trial_programs/mcastreceive.cpp
@@ -0,0 +1,150 @@
+/*
+# 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/mcast/trial_programs/mcastsend.cpp
+#    - Receive a file via the PVS Mcast protocol
+# -----------------------------------------------------------------------------
+*/
+
+#include <iostream>
+
+#include <QCoreApplication>
+#include <QFile>
+#include <QStringList>
+#include <QTimer>
+
+#include "mcastreceive.h"
+#include "McastConfigArgParser.h"
+#include "../McastConfiguration.h"
+#include "../McastReceiver.h"
+
+using namespace std;
+
+int main(int argc, char** argv)
+{
+	QCoreApplication app(argc, argv);
+	McastReceive me;
+
+	QTimer::singleShot(0, &me, SLOT(run()));
+
+	return app.exec();
+}
+
+void McastReceive::run()
+{
+	QStringList args = QCoreApplication::arguments();
+	QStringList::iterator i = args.begin();
+	QStringList::iterator const end = args.end();
+
+	QString filename("");
+
+	McastConfiguration config;
+
+	++i;
+	while (i != end)
+	{
+		QString arg = *i;
+
+		cerr << "Arg: " << arg.toLatin1().constData() << endl;
+
+		if (arg == "-file")
+		{
+			++i;
+			if (i == end)
+			{
+				cerr << "Option " << arg.toLatin1().constData() << " is missing its argument" << endl;
+				QCoreApplication::exit(1);
+				return;
+			}
+			filename = *i;
+		}
+		else if (arg == "-help")
+		{
+			cerr << "Options:" << endl << endl
+					<< " -file <FILE>       Receive to file FILE" << endl
+					<< " -addr <ADDR>       Use ADDR as address specification" << endl
+					<< " -dport <PORT>      Send to port PORT" << endl
+					<< " -sport <PORT>		Send from port PORT" << endl
+					<< " -mtu <BYTES>       Set MTU to BYTES" << endl
+					<< " -rate <BYTES>      Send BYTES per second" << endl
+					<< " -winsize <SECONDS> Set Window Size to SECONDS" << endl
+					<< " -udp               Use UDP encapsulation" << endl
+					<< " -udp-port PORT     Use UDP port PORT" << endl;
+			QCoreApplication::quit();
+			return;
+		}
+		else
+		{
+			if (!parseMcastConfigArg(i, end, &config))
+			{
+				cerr << "Unknown argument: " << arg.toLatin1().constData() << endl;
+				QCoreApplication::exit(1);
+				return;
+			}
+		}
+
+		++i;
+	}
+
+	if (filename == "")
+	{
+		cerr << "No Filename given" << endl;
+		QCoreApplication::exit(1);
+		return;
+	}
+
+	_target = new QFile(filename, this);
+	_target->open(QIODevice::WriteOnly);
+
+	McastReceiver* recv = new McastReceiver(_target, &config, this);
+
+	connect(recv, SIGNAL(finished(int)), this, SLOT(finished(int)));
+
+	QTimer::singleShot(0, recv, SLOT(start()));
+}
+
+void McastReceive::finished(int state)
+{
+	cerr << "finished: ";
+
+	switch(state)
+	{
+	case McastReceiver::RES_OK:
+		cerr << "OK." << endl;
+		break;
+	case McastReceiver::RES_ABORTED:
+		cerr << "Aborted." << endl;
+		goto failed;
+	case McastReceiver::RES_CHECKSUM_MISMATCH:
+		cerr << "Checksum mismatch." << endl;
+		goto failed;
+	case McastReceiver::RES_CONNECTION_RESET:
+		cerr << "Connection reset." << endl;
+		goto failed;
+	case McastReceiver::RES_MD5_MISMATCH:
+		cerr << "MD5 mismatch." << endl;
+		goto failed;
+	case McastReceiver::RES_OFFSET_MISMATCH:
+		cerr << "Offset mismatch. Undetected packet loss?" << endl;
+		goto failed;
+	default:
+		cerr << "Unknown error code!" << endl;
+		goto failed;
+	}
+
+	QCoreApplication::quit();
+	return;
+failed:
+	cerr << "Deleting file." << endl;
+	_target->remove();
+	QCoreApplication::exit(1);
+	return;
+}
diff --git a/src/net/mcast/trial_programs/mcastreceive.h b/src/net/mcast/trial_programs/mcastreceive.h
new file mode 100644
index 0000000..3e72d4c
--- /dev/null
+++ b/src/net/mcast/trial_programs/mcastreceive.h
@@ -0,0 +1,44 @@
+/*
+# 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/mcast/trial_programs/mcastsend.cpp
+#    - Receive a file via the PVS Mcast protocol
+# -----------------------------------------------------------------------------
+*/
+
+#ifndef MCASTRECEIVE_H_
+#define MCASTRECEIVE_H_
+
+#include <QObject>
+
+class QFile;
+class McastReceiver;
+
+class McastReceive : public QObject
+{
+	Q_OBJECT
+public:
+	McastReceive() :
+		QObject(),
+		_receiver(0)
+	{
+	}
+
+public slots:
+	void run();
+	void finished(int state);
+
+private:
+	McastReceiver* _receiver;
+	QFile* _target;
+};
+
+#endif /* MCASTRECEIVE_H_ */
diff --git a/src/net/mcast/trial_programs/mcastsend.cpp b/src/net/mcast/trial_programs/mcastsend.cpp
new file mode 100644
index 0000000..da8ecf4
--- /dev/null
+++ b/src/net/mcast/trial_programs/mcastsend.cpp
@@ -0,0 +1,123 @@
+/*
+# 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/mcast/trial_programs/mcastsend.cpp
+#    - Send a file via the PVS Mcast protocol
+# -----------------------------------------------------------------------------
+*/
+
+#include <iostream>
+
+#include <QCoreApplication>
+#include <QFile>
+#include <QStringList>
+#include <QTimer>
+
+#include <src/net/mcast/McastSender.h>
+#include "mcastsend.h"
+#include "McastConfigArgParser.h"
+#include "../McastConstants.h"
+#include "../McastConfiguration.h"
+
+using namespace std;
+
+int
+main(int argc, char**argv)
+{
+	QCoreApplication app(argc, argv);
+	McastSend me;
+
+	QTimer::singleShot(0, &me, SLOT(run()));
+
+	return app.exec();
+}
+
+void McastSend::run()
+{
+	QStringList args = QCoreApplication::arguments();
+	QStringList::iterator i = args.begin();
+	QStringList::iterator const end = args.end();
+	QString filename("");
+	McastConfiguration config;
+
+	++i;
+	while(i != end)
+	{
+		// parse command line arguments
+
+		QString arg = *i;
+
+		cerr << "Arg: " << arg.toLatin1().constData() << endl;
+
+		if (arg == "-file")
+		{
+			i++;
+			if(i == end)
+			{
+				cerr << "Option " << arg.toLatin1().constData() << " is missing argument" << endl;
+				QCoreApplication::exit(1);
+				return;
+			}
+			filename = *i;
+		}
+		else if (arg == "-help")
+		{
+			cerr << "Options:" << endl << endl
+					<< " -file <FILE>       Send FILE to the listeners" << endl
+					<< " -addr <ADDR>       Use ADDR as address specification" << endl
+					<< " -dport <PORT>      Send to port PORT" << endl
+					<< " -sport <PORT>		Send from port PORT" << endl
+					<< " -mtu <BYTES>       Set MTU to BYTES" << endl
+					<< " -rate <BYTES>      Send BYTES per second" << endl
+					<< " -winsize <SECONDS> Set Window Size to SECONDS" << endl
+					<< " -udp               Use UDP encapsulation" << endl
+					<< " -udp-port PORT     Use UDP port PORT" << endl;
+			QCoreApplication::quit();
+			return;
+		}
+		else
+		{
+			if (!parseMcastConfigArg(i, end, &config))
+			{
+				cerr << "Unknown command line argument: " << arg.toLatin1().constData() << endl;
+				QCoreApplication::exit(1);
+				return;
+			}
+		}
+
+		++i;
+	}
+
+	if(filename == "")
+	{
+		cerr << "No filename given" << endl;
+		QCoreApplication::exit(1);
+		return;
+	}
+
+	// now, do it.
+	QFile* file = new QFile(filename);
+	file->open(QIODevice::ReadOnly);
+
+	McastSender* sender = new McastSender(file, &config, this);
+	file->setParent(sender);
+
+	connect(sender, SIGNAL(finished()), this, SLOT(finished()));
+
+	QTimer::singleShot(0, sender, SLOT(start()));
+}
+
+void McastSend::finished()
+{
+	cerr << "finished." << endl;
+	// QTimer::singleShot(30000, QCoreApplication::instance(), SLOT(quit()));
+	// QCoreApplication::quit();
+}
diff --git a/src/net/mcast/trial_programs/mcastsend.h b/src/net/mcast/trial_programs/mcastsend.h
new file mode 100644
index 0000000..ae15eb4
--- /dev/null
+++ b/src/net/mcast/trial_programs/mcastsend.h
@@ -0,0 +1,42 @@
+/*
+# 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/mcast/trial_programs/mcastsend.cpp
+#    - Send a file via the PVS Mcast protocol
+# -----------------------------------------------------------------------------
+*/
+
+#ifndef MCASTSEND_H_
+#define MCASTSEND_H_
+
+#include <QObject>
+
+#include "../McastSender.h"
+
+class McastSend : public QObject
+{
+	Q_OBJECT
+public:
+	McastSend() :
+		QObject(),
+		_sender(0)
+	{
+	}
+
+public slots:
+	void run();
+	void finished();
+
+private:
+	McastSender* _sender;
+};
+
+#endif /* MCASTSEND_H_ */