ostinato/client/streamconfigdialog.cpp

#include <QHostAddress>

#include "streamconfigdialog.h"
#include "stream.h"
#include "abstractprotocol.h"
#include "protocollistiterator.h"

#include "modeltest.h"

// FIXME(HI) - remove
#include "../common/protocolmanager.h"
extern ProtocolManager OstProtocolManager;

int StreamConfigDialog::lastTopLevelTabIndex = 0;
int StreamConfigDialog::lastProtoTabIndex = 0;

StreamConfigDialog::StreamConfigDialog(Port &port, uint streamIndex,
	QWidget *parent) : QDialog (parent), mPort(port)
{
	OstProto::Stream s;
	mCurrentStreamIndex = streamIndex;
	mpStream = new Stream;
	mPort.streamByIndex(mCurrentStreamIndex)->protoDataCopyInto(s);
	mpStream->protoDataCopyFrom(s);
	_iter = mpStream->createProtocolListIterator();

	setupUi(this);
	setupUiExtra();

	connect(bgFrameType, SIGNAL(buttonClicked(int)), 
		this, SLOT(updateContents()));
	connect(bgL3Proto, SIGNAL(buttonClicked(int)),
		this, SLOT(updateContents()));
	connect(bgL4Proto, SIGNAL(buttonClicked(int)),
		this, SLOT(updateContents()));
	
	//! \todo causes a crash!
#if 0	
	connect(lePktLen, SIGNAL(textEdited(QString)),
		this, SLOT(updateContents()));
#endif

	// Time to play match the signals and slots!

	// Enable VLAN Choices only if FT = Eth2 or SNAP
	connect(rbFtNone, SIGNAL(toggled(bool)), gbVlan, SLOT(setDisabled(bool)));

	connect(rbFtNone, SIGNAL(toggled(bool)), rbL3None, SLOT(setChecked(bool)));

	// Enable/Disable L3 Protocol Choices for FT None
	connect(rbFtNone, SIGNAL(toggled(bool)), rbL3None, SLOT(setEnabled(bool)));
	connect(rbFtNone, SIGNAL(toggled(bool)), rbL3Ipv4, SLOT(setDisabled(bool)));
	connect(rbFtNone, SIGNAL(toggled(bool)), rbL3Arp, SLOT(setDisabled(bool)));

	// Enable/Disable L3 Protocol Choices for FT Ethernet2
	connect(rbFtEthernet2, SIGNAL(toggled(bool)), rbL3None, SLOT(setEnabled(bool)));
	connect(rbFtEthernet2, SIGNAL(toggled(bool)), rbL3Ipv4, SLOT(setEnabled(bool)));
	connect(rbFtEthernet2, SIGNAL(toggled(bool)), rbL3Arp, SLOT(setEnabled(bool)));

	// Force L3 = None if FT = 802.3 Raw
	connect(rbFt802Dot3Raw, SIGNAL(toggled(bool)), rbL3None, SLOT(setChecked(bool)));

	// Enable/Disable L3 Protocol Choices for FT 802Dot3Raw
	connect(rbFt802Dot3Raw, SIGNAL(toggled(bool)), rbL3None, SLOT(setEnabled(bool)));
	connect(rbFt802Dot3Raw, SIGNAL(toggled(bool)), rbL3Ipv4, SLOT(setDisabled(bool)));
	connect(rbFt802Dot3Raw, SIGNAL(toggled(bool)), rbL3Arp, SLOT(setDisabled(bool)));

	// Force L3 = None if FT = 802.3 LLC (to ensure a valid L3 is selected)
	connect(rbFt802Dot3Llc, SIGNAL(toggled(bool)), rbL3None, SLOT(setChecked(bool)));

	// Enable/Disable L3 Protocol Choices for FT 802Dot3Llc
	connect(rbFt802Dot3Llc, SIGNAL(toggled(bool)), rbL3None, SLOT(setEnabled(bool)));
	connect(rbFt802Dot3Llc, SIGNAL(toggled(bool)), rbL3Ipv4, SLOT(setEnabled(bool)));
	connect(rbFt802Dot3Llc, SIGNAL(toggled(bool)), rbL3Arp, SLOT(setDisabled(bool)));

	// Enable/Disable L3 Protocol Choices for FT 802.3 LLC SNAP
	connect(rbFtLlcSnap, SIGNAL(toggled(bool)), rbL3None, SLOT(setEnabled(bool)));
	connect(rbFtLlcSnap, SIGNAL(toggled(bool)), rbL3Ipv4, SLOT(setEnabled(bool)));
	connect(rbFtLlcSnap, SIGNAL(toggled(bool)), rbL3Arp, SLOT(setEnabled(bool)));

	// Enable/Disable L4 Protocol Choices for L3 Protocol None
	connect(rbL3None, SIGNAL(toggled(bool)), rbL4None, SLOT(setEnabled(bool)));
	connect(rbL3None, SIGNAL(toggled(bool)), rbL4Icmp, SLOT(setDisabled(bool)));
	connect(rbL3None, SIGNAL(toggled(bool)), rbL4Igmp, SLOT(setDisabled(bool)));
	connect(rbL3None, SIGNAL(toggled(bool)), rbL4Tcp, SLOT(setDisabled(bool)));
	connect(rbL3None, SIGNAL(toggled(bool)), rbL4Udp, SLOT(setDisabled(bool)));

	// Force L4 Protocol = None if L3 Protocol is set to None
	connect(rbL3None, SIGNAL(toggled(bool)), rbL4None, SLOT(setChecked(bool)));

	// Enable/Disable L4 Protocol Choices for L3 Protocol IPv4
	connect(rbL3Ipv4, SIGNAL(toggled(bool)), rbL4None, SLOT(setEnabled(bool)));
	connect(rbL3Ipv4, SIGNAL(toggled(bool)), rbL4Icmp, SLOT(setEnabled(bool)));
	connect(rbL3Ipv4, SIGNAL(toggled(bool)), rbL4Igmp, SLOT(setEnabled(bool)));
	connect(rbL3Ipv4, SIGNAL(toggled(bool)), rbL4Tcp, SLOT(setEnabled(bool)));
	connect(rbL3Ipv4, SIGNAL(toggled(bool)), rbL4Udp, SLOT(setEnabled(bool)));

	// Enable/Disable L4 Protocol Choices for L3 Protocol ARP
	connect(rbL3Arp, SIGNAL(toggled(bool)), rbL4None, SLOT(setEnabled(bool)));
	connect(rbL3Arp, SIGNAL(toggled(bool)), rbL4Icmp, SLOT(setDisabled(bool)));
	connect(rbL3Arp, SIGNAL(toggled(bool)), rbL4Igmp, SLOT(setDisabled(bool)));
	connect(rbL3Arp, SIGNAL(toggled(bool)), rbL4Tcp, SLOT(setDisabled(bool)));
	connect(rbL3Arp, SIGNAL(toggled(bool)), rbL4Udp, SLOT(setDisabled(bool)));

	// Force L4 Protocol = None if L3 Protocol is set to ARP
	connect(rbL3Arp, SIGNAL(toggled(bool)), rbL4None, SLOT(setChecked(bool)));

	mpAvailableProtocolsModel = new QStringListModel(
		OstProtocolManager.protocolDatabase(), this);
	lvAllProtocols->setModel(mpAvailableProtocolsModel);

	LoadCurrentStream();
	mpPacketModel = new PacketModel(this);
	tvPacketTree->setModel(mpPacketModel);
	mpPacketModelTester = new ModelTest(mpPacketModel);
	tvPacketTree->header()->hide();
	vwPacketDump->setModel(mpPacketModel);
	vwPacketDump->setSelectionModel(tvPacketTree->selectionModel());

	// TODO(MED):

	//! \todo Implement then enable these protocols - SVLAN, ARP, ICMP, IGMP
	cbSVlan->setHidden(true);
	rbL3Arp->setHidden(true);
	rbL3Ipv6->setHidden(true);
	rbL4Icmp->setHidden(true);
	rbL4Igmp->setHidden(true);
	//! \todo Enable navigation of streams
	pbPrev->setDisabled(true);
	pbNext->setDisabled(true);
	//! \todo Support Goto Stream Id
	leStreamId->setDisabled(true);
	disconnect(rbActionGotoStream, SIGNAL(toggled(bool)), leStreamId, SLOT(setEnabled(bool)));
	//! \todo Support Continuous Mode
	rbModeContinuous->setDisabled(true);

	// Finally, restore the saved last selected tab for the various tab widgets
	twTopLevel->setCurrentIndex(lastTopLevelTabIndex);
	if (twProto->isTabEnabled(lastProtoTabIndex))
		twProto->setCurrentIndex(lastProtoTabIndex);
}

void StreamConfigDialog::setupUiExtra()
{
	QRegExp reHex2B("[0-9,a-f,A-F]{1,4}");
	QRegExp reHex4B("[0-9,a-f,A-F]{1,8}");
	QRegExp reMac("([0-9,a-f,A-F]{2,2}[:-]){5,5}[0-9,a-f,A-F]{2,2}");

#if 0 // MPS - temp mask
	// Add the Payload widget to the dialog
	{
		QGridLayout		*layout;

		layout = static_cast<QGridLayout*>(twTopLevel->widget(0)->layout());

		layout->addWidget(mpStream->protocol(52)->configWidget(), 0, 1);
		qDebug("setupUi wgt = %p", mpStream->protocol(52)->configWidget());
	}
#endif

	// ---- Setup default stuff that cannot be done in designer ----
	gbVlan->setDisabled(true);

	bgFrameType = new QButtonGroup();
	foreach(QRadioButton *btn, gbFrameType->findChildren<QRadioButton*>())
		bgFrameType->addButton(btn);

	bgL3Proto = new QButtonGroup();
	foreach(QRadioButton *btn, gbL3Proto->findChildren<QRadioButton*>())
		bgL3Proto->addButton(btn);

	bgL4Proto = new QButtonGroup();
	foreach(QRadioButton *btn, gbL4Proto->findChildren<QRadioButton*>())
		bgL4Proto->addButton(btn);

	//twProto->setTabEnabled(2, FALSE);
	//twProto->setTabEnabled(3, FALSE);

	/*
	** Setup Validators
	*/	
	// Meta Data
	//! \todo - doesn't seem to work -  range validator needs a spinbox?
	//lePktLen->setValidator(new QIntValidator(MIN_PKT_LEN, MAX_PKT_LEN, this));
	
	// L2 Ethernet
#if 0 // Proto FW
	leDstMac->setValidator(new QRegExpValidator(reMac, this));
	leSrcMac->setValidator(new QRegExpValidator(reMac, this));
	leDstMacCount->setValidator(new QIntValidator(1, MAX_MAC_ITER_COUNT, this));
	leSrcMacCount->setValidator(new QIntValidator(1, MAX_MAC_ITER_COUNT, this));
	leCvlanTpid->setValidator(new QRegExpValidator(reHex2B, this));
	leSvlanTpid->setValidator(new QRegExpValidator(reHex2B, this));
	//leEtherType->setValidator(new QRegExpValidator(reHex2B, this));
#endif

	/*
	** Setup Connections
	*/
	connect(rbSendPackets, SIGNAL(toggled(bool)), 
		this, SLOT(update_NumPacketsAndNumBursts()));
	connect(rbSendBursts, SIGNAL(toggled(bool)), 
		this, SLOT(update_NumPacketsAndNumBursts()));
	connect(rbModeFixed, SIGNAL(toggled(bool)), 
		this, SLOT(update_NumPacketsAndNumBursts()));
	connect(rbModeContinuous, SIGNAL(toggled(bool)), 
		this, SLOT(update_NumPacketsAndNumBursts()));

}

StreamConfigDialog::~StreamConfigDialog()
{
	delete mpPacketModelTester;
	delete mpPacketModel;

#if 0 // MPS - temp mask
	// Remove payload data widget so that it is not deleted when this object
	// is destroyed
	{
		QLayout *layout = twTopLevel->widget(0)->layout();
		if (layout)
		{
			layout->removeWidget(mpStream->protocol(52)->configWidget());
			mpStream->protocol(52)->configWidget()->setParent(0);
		}
	}
#endif

	// Remove any existing widget on the L2-L4 Tabs lest they are deleted
	// when this object is destoryed
	for (int i = 1; i <= 3; i++)
	{
		QLayout *layout = twProto->widget(i)->layout();
		if (layout)
		{
			QLayoutItem *child;
			while ((child = layout->takeAt(0)) != 0)
			{
				Q_ASSERT(child->widget() != 0);
				// Don't delete the child widget - reparent it
				child->widget()->setParent(0);
			} 
			delete layout;
		}
	}

	delete bgFrameType;
	delete bgL3Proto;
	delete bgL4Proto;

	delete _iter;
	delete mpStream;
}

void StreamConfigDialog::updateSelectedProtocols()
{
#define CHKINS(p)	\
{ \
	if (_iter->hasNext() && (_iter->peekNext()->protocolNumber() == OstProto::Protocol::k##p##FieldNumber))  \
		_iter->next(); \
	else \
		_iter->insert(OstProtocolManager.createProtocol(OstProto::Protocol::k##p##FieldNumber, mpStream)); \
}

	_iter->toFront();
#if 1
	qDebug("Before Update");
	while (_iter->hasNext())
	{
		AbstractProtocol* p;
	   
		p = _iter->next();
		qDebug("%p:[%d]", p, p->protocolNumber());
	}

	_iter->toFront();
#endif

	// Mac
	CHKINS(Mac)

	if (cbCVlan->isEnabled() && cbCVlan->isChecked())
		CHKINS(Vlan);

	if (rbFtEthernet2->isChecked())
		CHKINS(Eth2)
	else if (rbFt802Dot3Raw->isChecked())
		CHKINS(Dot3)
	else if (rbFt802Dot3Llc->isChecked())
	{
		CHKINS(Dot3);
		CHKINS(Llc);
	}
	else if (rbFtLlcSnap->isChecked())
	{
		CHKINS(Dot3);
		CHKINS(Llc);
		CHKINS(Snap);
	}

	if (rbL3Ipv4->isChecked())
		CHKINS(Ip4)
	else if (rbL3Arp->isChecked())
		CHKINS(Arp)

	if (rbL4Tcp->isChecked())
		CHKINS(Tcp)
	else if (rbL4Udp->isChecked())
		CHKINS(Udp)
	else if (rbL4Icmp->isChecked())
		CHKINS(Icmp)
	else if (rbL4Igmp->isChecked())
		CHKINS(Igmp)

	// Payload
	CHKINS(Payload)

	// Remove all protocols, if any, beyond payload
	while (_iter->hasNext())
	{
		_iter->next();
		_iter->remove();
	}


#if 1
	qDebug("After Update");
	_iter->toFront();
	while (_iter->hasNext())
	{
		AbstractProtocol* p;
	   
		p = _iter->next();
		qDebug("%p:[%d]", p, p->protocolNumber());
	}
#endif

#undef CHKINS

}

void StreamConfigDialog::updateContents()
{
	StoreCurrentStream();
	LoadCurrentStream();
}

void StreamConfigDialog::on_cmbPktLenMode_currentIndexChanged(QString mode)
{
	if (mode == "Fixed")
	{
		lePktLen->setEnabled(true);
		lePktLenMin->setDisabled(true);
		lePktLenMax->setDisabled(true);
	}
	else if (mode == "Increment")
	{
		lePktLen->setDisabled(true);
		lePktLenMin->setEnabled(true);
		lePktLenMax->setEnabled(true);
	}
	else if (mode == "Decrement")
	{
		lePktLen->setDisabled(true);
		lePktLenMin->setEnabled(true);
		lePktLenMax->setEnabled(true);
	}
	else if (mode == "Random")
	{
		lePktLen->setDisabled(true);
		lePktLenMin->setEnabled(true);
		lePktLenMax->setEnabled(true);
	}
	else
	{
		qWarning("Unhandled/Unknown PktLenMode = %s", mode.toAscii().data());
	}
}

void StreamConfigDialog::on_pbPrev_clicked()
{
#if 0
	StoreCurrentStream(currStreamIdx);
	currStreamIdx--;
	LoadCurrentStream(currStreamIdx);

	pbPrev->setDisabled((currStreamIdx == 0));
	pbNext->setDisabled((currStreamIdx == 2));
#endif
}

void StreamConfigDialog::on_pbNext_clicked()
{
#if 0
	StoreCurrentStream(currStreamIdx);
	currStreamIdx++;
	LoadCurrentStream(currStreamIdx);

	pbPrev->setDisabled((currStreamIdx == 0));
	pbNext->setDisabled((currStreamIdx == 2));
#endif
}

void StreamConfigDialog::on_twTopLevel_currentChanged(int index)
{
	// We only process the "Packet View" tab
	if (index != 2)
		return;

	updateContents();
	mpPacketModel->setSelectedProtocols(*_iter);
}

void StreamConfigDialog::on_twProto_currentChanged(int index)
{
#if 0 // MPS - temp mask
	QLayout *layout;
	QList<QWidget*> wl;

	// We need to process only indices 1-3 i.e. the L2, L3 and L4 tabs
	if ((index < 1) || (index > 3))
		return;

	// Remove any existing widget on the activated tab
	layout = twProto->widget(index)->layout();
	if (layout)
	{
		QLayoutItem *child;
		while ((child = layout->takeAt(0)) != 0)
		{
			Q_ASSERT(child->widget() != 0);
			// Don't delete the child widget - reparent it
			child->widget()->setParent(0);
		} 
		delete layout;
	}

	// FIXME: protocol id hardcodings
	switch(index)
	{
		case 1: // L2
			wl.append(mpStream->protocol("mac")->configWidget());
			if (cbCVlan->isEnabled() && cbCVlan->isChecked())
				wl.append(mpStream->protocol("vlan")->configWidget());
			if (rbFtEthernet2->isChecked())
				wl.append(mpStream->protocol("eth2")->configWidget());
			else if (rbFt802Dot3Raw->isChecked())
				wl.append(mpStream->protocol("dot3")->configWidget());
			else if (rbFt802Dot3Llc->isChecked())
			{
				wl.append(mpStream->protocol("dot3")->configWidget());
				wl.append(mpStream->protocol("llc")->configWidget());
			}
			else if (rbFtLlcSnap->isChecked())
			{
				wl.append(mpStream->protocol("dot3")->configWidget());
				wl.append(mpStream->protocol("llc")->configWidget());
				wl.append(mpStream->protocol("snap")->configWidget());
			}

			{
				int i, r = 0, c = 0;
				QGridLayout *layout = new QGridLayout;

				Q_ASSERT(wl.size() > 0);

				// We use a 2 column layout for the L2 Tab

				layout->addWidget(wl.at(0), r, c, 1, -1);
				r++;
				for (i=1; i < wl.size(); i++)
				{
					layout->addWidget(wl.at(i), r, c);
					if ((i % 2) == 0)
						r++;	
					c = (c+1) % 2;

				}
				if ((i % 2) == 0)
					r++;	
				layout->setRowStretch(r, 10);

				twProto->widget(index)->setLayout(layout);
			}
			break;
		case 2: // L3
			if (rbL3Ipv4->isChecked())
				wl.append(mpStream->protocol("ip4")->configWidget());

			if (wl.size()) 
			{
				QVBoxLayout *layout = new QVBoxLayout;

				for (int i=0; i < wl.size(); i++)
					layout->addWidget(wl.at(i));

				twProto->widget(index)->setLayout(layout);
			}
			break;
		case 3: // L4
			if (rbL4Tcp->isChecked())
				wl.append(mpStream->protocol("tcp")->configWidget());
			else if (rbL4Udp->isChecked())
				wl.append(mpStream->protocol("udp")->configWidget());

			if (wl.size()) 
			{
				QVBoxLayout *layout = new QVBoxLayout;

				for (int i=0; i < wl.size(); i++)
					layout->addWidget(wl.at(i));

				twProto->widget(index)->setLayout(layout);
			}
			break;
	}
#else
	int selTab;

	qDebug("In %s (tab index = %d)", __FUNCTION__, index);
	// We need to process only index 4 i.e. the Protocol Data tab
	if (index != 4)
		return;

	// Hide the ToolBox before modifying it - otherwise we have a crash !!!
	tbProtocolData->hide();

	selTab = tbProtocolData->currentIndex();

	// Remove any existing widget on the activated tab
	while (tbProtocolData->count() > 0)
	{
		QWidget* w = tbProtocolData->widget(0);
		tbProtocolData->removeItem(0);
		w->setParent(0);
	}

	_iter->toFront();
	while (_iter->hasNext())
	{
		AbstractProtocol* p = _iter->next();
		tbProtocolData->addItem(p->configWidget(), p->name());
	}

	if (selTab < tbProtocolData->count())
		tbProtocolData->setCurrentIndex(selTab);

	tbProtocolData->show();
#endif
}

void StreamConfigDialog::update_NumPacketsAndNumBursts()
{
	if (rbSendPackets->isChecked() && rbModeFixed->isChecked())
		leNumPackets->setEnabled(true);
	else
		leNumPackets->setEnabled(false);

	if (rbSendBursts->isChecked() && rbModeFixed->isChecked())
		leNumBursts->setEnabled(true);
	else
		leNumBursts->setEnabled(false);
}

#if 0
void StreamConfigDialog::on_lePattern_editingFinished()
{
	ulong	num = 0;
	bool	isOk;
	QString	str;

	num = lePattern->text().remove(QChar(' ')).toULong(&isOk, 16);
	qDebug("editfinished (%s | %x)\n", lePattern->text().toAscii().data(), num);
	lePattern->setText(uintToHexStr(num, str, 4));
	qDebug("editfinished (%s | %x)\n", lePattern->text().toAscii().data(), num);
}
#endif

void StreamConfigDialog::LoadCurrentStream()
{
	QString	str;

	qDebug("loading mpStream %p", mpStream);

	// Meta Data
	{
		cmbPktLenMode->setCurrentIndex(mpStream->lenMode());
		lePktLen->setText(str.setNum(mpStream->frameLen()));
		lePktLenMin->setText(str.setNum(mpStream->frameLenMin()));
		lePktLenMax->setText(str.setNum(mpStream->frameLenMax()));
	}

#if 0 // MPS - temp mask
	// Protocols
	{
		int i;

		qDebug("Selected Protocols.size() = %d", mSelectedProtocols.size());
		mSelectedProtocols = mpStream->frameProtocol();
		qDebug("Selected Protocols.size() = %d", mSelectedProtocols.size());
		for (i = 0; i < mSelectedProtocols.size(); i++)
			qDebug("%d: %d", i, mSelectedProtocols.at(i));

		Q_ASSERT(mSelectedProtocols.size() >= 2); // Mac + Payload: mandatory

		i = 0;
		Q_ASSERT(mSelectedProtocols.at(i) == 51); // Mac
		i++;

		// VLAN
		if (mSelectedProtocols.at(i) == 126) // VLAN
		{
			cbCVlan->setChecked(true);
			i++;
		}

		if (mSelectedProtocols.at(i) == 52) // Payload
		{
			i++;
			goto _proto_parse_done;
		}
		else if (mSelectedProtocols.at(i) == 121) // Eth2
		{
			rbFtEthernet2->setChecked(true);
			i++;
		}
		else if (mSelectedProtocols.at(i) == 122) // 802.3 RAW
		{
			if ((mSelectedProtocols.size() > (i+1)) &&
					(mSelectedProtocols.at(i+1) == 123)) // 802.3 LLC
			{
				if ((mSelectedProtocols.size() > (i+2)) && 
						(mSelectedProtocols.at(i+2) == 124)) // SNAP
				{
					rbFtLlcSnap->setChecked(true);
					i+=3;
				}
				else
				{
					rbFt802Dot3Llc->setChecked(true);
					i+=2;
				}
			}
			else
			{
				rbFt802Dot3Raw->setChecked(true);
				i++;
			}
		}
		else
			rbFtNone->setChecked(true);


		// L3
		if (mSelectedProtocols.at(i) == 52) // Payload
		{
			i++;
			goto _proto_parse_done;
		}
		else if (mSelectedProtocols.at(i) == 130) // IP4
		{
			rbL3Ipv4->setChecked(true);
			i++;
		}
	   	else if (mSelectedProtocols.at(i) == 131) // ARP
		{
			rbL3Arp->setChecked(true);
			i++;
		}
		else
			rbL3None->setChecked(true);

		if (i == mSelectedProtocols.size())
			goto _proto_parse_done;

		// L4
		if (mSelectedProtocols.at(i) == 52) // Payload
		{
			i++;
			goto _proto_parse_done;
		}
		else if (mSelectedProtocols.at(i) == 140) // TCP
		{
			rbL4Tcp->setChecked(true);
			i++;
		}
		else if (mSelectedProtocols.at(i) == 141) // UDP
		{
			rbL4Udp->setChecked(true);
			i++;
		}
		else if (mSelectedProtocols.at(i) == 142) // ICMP
		{
			rbL4Icmp->setChecked(true);
			i++;
		}
		else if (mSelectedProtocols.at(i) == 143) // IGMP
		{
			rbL4Igmp->setChecked(true);
			i++;
		}
		else
			rbL4None->setChecked(true);

		Q_ASSERT(mSelectedProtocols.at(i) == 52); // Payload
		i++;

_proto_parse_done:
		Q_ASSERT(i == mSelectedProtocols.size());
		mpStream->loadProtocolWidgets();
	}
#else
	// Protocols
	{
		qDebug("Loading - current list");
		_iter->toFront();
		while(_iter->hasNext())
		{
			AbstractProtocol* p = _iter->next();
			qDebug("%p -- %d", p, p->protocolNumber());
		}
		_iter->toFront();

#define CHK(p) (_iter->hasNext() && _iter->peekNext()->protocolNumber() == p)

		Q_ASSERT(CHK(51)); // Mac
		_iter->next();

		// VLAN
		if (CHK(126)) // VLAN
		{
			cbCVlan->setChecked(true);
			_iter->next();
		}

		if (CHK(52)) // Payload
		{
			_iter->next();
			goto _proto_parse_done;
		}
		else if (CHK(121)) // Eth2
		{
			rbFtEthernet2->setChecked(true);
			_iter->next();
		}
		else if (CHK(122)) // 802.3 RAW
		{
			_iter->next();
			if (CHK(123)) // 802.3 LLC
			{
				_iter->next();
				if (CHK(124)) // SNAP
				{
					rbFtLlcSnap->setChecked(true);
					_iter->next();
				}
				else
				{
					rbFt802Dot3Llc->setChecked(true);
				}
			}
			else
			{
				rbFt802Dot3Raw->setChecked(true);
			}
		}
		else
			rbFtNone->setChecked(true);


		// L3
		if (CHK(52)) // Payload
		{
			_iter->next();	
			goto _proto_parse_done;
		}
		else if (CHK(130)) // IP4
		{
			rbL3Ipv4->setChecked(true);
			_iter->next();
		}
	   	else if (CHK(131)) // ARP
		{
			rbL3Arp->setChecked(true);
			_iter->next();
		}
		else
			rbL3None->setChecked(true);

		if (!_iter->hasNext())
			goto _proto_parse_done;

		// L4
		if (CHK(52)) // Payload
		{
			_iter->next();
			goto _proto_parse_done;
		}
		else if (CHK(140)) // TCP
		{
			rbL4Tcp->setChecked(true);
			_iter->next();
		}
		else if (CHK(141)) // UDP
		{
			rbL4Udp->setChecked(true);
			_iter->next();
		}
		else if (CHK(142)) // ICMP
		{
			rbL4Icmp->setChecked(true);
			_iter->next();
		}
		else if (CHK(143)) // IGMP
		{
			rbL4Igmp->setChecked(true);
			_iter->next();
		}
		else
			rbL4None->setChecked(true);

		Q_ASSERT(CHK(52)); // Payload
		_iter->next();

_proto_parse_done:
		Q_ASSERT(!_iter->hasNext());
		mpStream->loadProtocolWidgets();

#undef CHK
	}


#endif

	// Stream Control
	{
		switch (mpStream->sendUnit())
		{
		case Stream::e_su_packets:
			rbSendPackets->setChecked(true);
			break;
		case Stream::e_su_bursts:
			rbSendBursts->setChecked(true);
			break;
		default:
			qWarning("Unhandled sendUnit = %d\n", mpStream->sendUnit());
		}

		switch (mpStream->sendMode())
		{
		case Stream::e_sm_fixed:
			rbModeFixed->setChecked(true);
			break;
		case Stream::e_sm_continuous:
			rbModeContinuous->setChecked(true);
			break;
		default:
			qWarning("Unhandled sendMode = %d\n", mpStream->sendMode());
		}

		switch(mpStream->nextWhat())
		{
		case Stream::e_nw_stop:
			rbActionStop->setChecked(true);
			break;
		case Stream::e_nw_goto_next:
			rbActionGotoNext->setChecked(true);
			break;
		case Stream::e_nw_goto_id:
			rbActionGotoStream->setChecked(true);
			break;
		default:
			qWarning("Unhandled nextAction = %d\n", mpStream->nextWhat());
		}

		leNumPackets->setText(QString().setNum(mpStream->numPackets()));
		leNumBursts->setText(QString().setNum(mpStream->numBursts()));
		lePacketsPerBurst->setText(QString().setNum(mpStream->burstSize()));
		lePacketsPerSec->setText(QString().setNum(mpStream->packetRate()));
		leBurstsPerSec->setText(QString().setNum(mpStream->burstRate()));
		// TODO(MED): Change this when we support goto to specific stream
		leStreamId->setText(QString("0"));
	}
	qDebug("loading stream done");
}

void StreamConfigDialog::StoreCurrentStream()
{
	QString	str;
	bool	isOk;
	Stream	*pStream = mpStream;

	qDebug("storing pStream %p", pStream);

	// Meta Data
	pStream->setLenMode((Stream::FrameLengthMode) cmbPktLenMode->currentIndex());
	pStream->setFrameLen(lePktLen->text().toULong(&isOk));
	pStream->setFrameLenMin(lePktLenMin->text().toULong(&isOk));
	pStream->setFrameLenMax(lePktLenMax->text().toULong(&isOk));

	// Protocols
	{
		updateSelectedProtocols();
		//pStream->setFrameProtocol(mSelectedProtocols);
		pStream->storeProtocolWidgets();
	}

	// Stream Control
	{
		if (rbSendPackets->isChecked())
			pStream->setSendUnit(Stream::e_su_packets);
		if (rbSendBursts->isChecked())
			pStream->setSendUnit(Stream::e_su_bursts);

		if (rbModeFixed->isChecked())
			pStream->setSendMode(Stream::e_sm_fixed);
		if (rbModeContinuous->isChecked())
			pStream->setSendMode(Stream::e_sm_continuous);

		if (rbActionStop->isChecked())
			pStream->setNextWhat(Stream::e_nw_stop);
		if (rbActionGotoNext->isChecked())
			pStream->setNextWhat(Stream::e_nw_goto_next);
		if (rbActionGotoStream->isChecked())
			pStream->setNextWhat(Stream::e_nw_goto_id);

		pStream->setNumPackets(leNumPackets->text().toULong(&isOk));
		pStream->setNumBursts(leNumBursts->text().toULong(&isOk));
		pStream->setBurstSize(lePacketsPerBurst->text().toULong(&isOk));
		pStream->setPacketRate(lePacketsPerSec->text().toULong(&isOk));
		pStream->setBurstRate(leBurstsPerSec->text().toULong(&isOk));
	}
}

void StreamConfigDialog::on_pbOk_clicked()
{
	OstProto::Stream	s;

	// Store dialog contents into stream
	//updateSelectedProtocols();
	StoreCurrentStream();
	mpStream->protoDataCopyInto(s);
	mPort.streamByIndex(mCurrentStreamIndex)->protoDataCopyFrom(s);;
	qDebug("stream stored");

	lastTopLevelTabIndex = twTopLevel->currentIndex();
	lastProtoTabIndex = twProto->currentIndex();
}