Drop _desiredProjectionSource. Dont send a stopVncClient if it has not been started before.

1 files changed, 58 insertions, 64 deletions

diff --git a/src/server/mainwindow/mainwindow.cpp b/src/server/mainwindow/mainwindow.cpp
index 97b4528..4c0a497 100644
--- a/src/server/mainwindow/mainwindow.cpp
+++ b/src/server/mainwindow/mainwindow.cpp
@@ -450,13 +450,14 @@ void MainWindow::broadcast(Client *from)
 		_streamingSource->stopVncServer();
 	_streamingSource = from;
 
-	// Server must not listen to itself
-	_streamingSource->setDesiredProjectionSource(0);
+	// Server must not listen to anybody
+	_streamingSource->stopVncClient();
+
 
 	// Set desired projection source on all clients, if not "from" or oflline
 	for (QList<ConnectionFrame*>::iterator it(_clientFrames.begin()); it != _clientFrames.end(); ++it)
 		if ((*it)->client() != NULL && (*it)->client() != _streamingSource)
-			(*it)->client()->setDesiredProjectionSource(_streamingSource->id());
+			_watchers.insert((*it)->client()->id(), (*it)->client());
 
 	if (_streamingSource->isActiveVncServer())
 		// From is already active
@@ -475,6 +476,9 @@ void MainWindow::broadcast(Client *from)
  */
 void MainWindow::multicast(Client *from, Client *to, bool blockOthers)
 {
+	if (_broadcast)
+		_watchers.clear();
+
 	_broadcast = false;
 	_lockOthers = blockOthers;
 
@@ -491,9 +495,9 @@ void MainWindow::multicast(Client *from, Client *to, bool blockOthers)
 	if (to->isActiveVncServer())
 		to->stopVncServer();
 
-	from->setDesiredProjectionSource(0);
+	from->stopVncClient();
 	to->setBroadcastSource(false);
-	to->setDesiredProjectionSource(from->id());
+	_watchers.insert(to->id(), to);
 
 	if (blockOthers) {
 		// Set lock others
@@ -624,7 +628,7 @@ void MainWindow::onButtonStopProjection()
 		Client *c = (**it).client();
 		if (c == NULL)
 			continue;
-		c->setDesiredProjectionSource(0);
+		c->stopVncClient();
 		c->setBroadcastSource(false);
 	}
 
@@ -840,7 +844,7 @@ void MainWindow::onClientAuthenticated(Client* client)
 		client->lockScreen(true);
 
 	if (_broadcast){
-		client->setDesiredProjectionSource(_streamingSource->id());
+		_watchers.insert(client->id(), client);
 		client->startVncClient(_streamingSource);
 	} else if (_lockOthers)
 		client->lockScreen(true);
@@ -858,44 +862,51 @@ void MainWindow::onVncServerStateChange(Client* client)
 		return;
 	if (client->isActiveVncServer())
 	{
-		// VNC Server started on some client - start projection on all clients interested in that client's screen
-		for (QList<ConnectionFrame*>::iterator it(_clientFrames.begin()); it != _clientFrames.end(); ++it)
+		if (_broadcast)
 		{
-			Client *c = (**it).client();
-			if (c == NULL || c->id() == client->id())
-				continue; // Offline or self
-			if (c->currentProjectionSource() == client->id())
-				continue; // Already watching this client
-			if (c->desiredProjectionSource() != client->id() && !client->isBroadcastSource())
-				continue; // Not interested
-			c->startVncClient(client);
+			for (QList<ConnectionFrame*>::iterator it(_clientFrames.begin()); it != _clientFrames.end(); ++it)
+			{
+				if ((*it)->client() != NULL)
+				{
+					// Unlock all online clients
+					(*it)->client()->lockScreen(false);
+
+					// Start VNCclients on all online clients but source
+					if ((*it)->client() != client)
+						(*it)->client()->startVncClient(client);
+				}
+			}
+		}
+		else  // !_broadcast
+		{
+			for (QList<ConnectionFrame*>::iterator it(_clientFrames.begin()); it != _clientFrames.end(); ++it) {
+				if ( (*it)->client() != NULL) {
+					if (_watchers.contains((*it)->client()->id()))
+					{
+						// Unlock destination and connect VNCclient
+						(*it)->client()->lockScreen(false);
+						(*it)->client()->startVncClient(client);
+					}
+					else if ((*it)->client()->id() == client->id() )
+					{
+						// Unlock source
+						(*it)->client()->lockScreen(false);
+					}
+					else
+					{
+						// Lock others and stop their clients
+						(*it)->client()->lockScreen(_lockOthers);
+						(*it)->client()->stopVncClient();
+					}
+				}
+			}
 		}
 	}
 	else
 	{
 		// VNC server stopped on some client or failed to start - reset local pending projection information
-		// If at least one other client seemed to be waiting for this client's screen, pop up a message
-		// on the console
-		bool wasInterested = client->isBroadcastSource();
-		//qDebug() << "On VNC stop:" << client->ip() << "was projection source:" << wasInterested;
-		client->setBroadcastSource(false);
-		for (QList<ConnectionFrame*>::iterator it(_clientFrames.begin()); it != _clientFrames.end(); ++it)
-		{
-			Client *c = (**it).client();
-			if (c == NULL)
-				continue;
-			if (c->desiredProjectionSource() == client->id())
-			{
-				wasInterested = true;
-				c->setDesiredProjectionSource(0);
-			}
-		}
-		if (wasInterested && QDateTime::currentMSecsSinceEpoch() < _buttonBlockTime)
-		{
-			QMessageBox::information(this,
-				tr("Projection Error"),
-				tr("Could not send screen contents of %1 to other clients: VNC Startup failed.").arg(client->name())
-			);
+		foreach (Client *c, _watchers) {
+			c->stopVncClient();
 		}
 	}
 }
@@ -910,32 +921,15 @@ void MainWindow::onVncClientStateChange(Client* client, int lastProjectionSource
 {
 	if (!isValidClient(client)) // Check here because this slot is connected queued
 		return;
-	// If last source is not known or not existent, we cannot do anything meaningful here
-	if (lastProjectionSource == 0)
-		return;
-	// Client is not a vnc client anymore. Check if the VNC server it was (supposedly) connected
-	// to is still running, and kill it if there are no more clients connected to it.
-	// (client->currentProjectionSource() will still contain the id of the server it was connected to during this event)
-	// 1. check if vnc server is still running
-	// 2. check if there are any other clients "using" that server
-	bool inUse = false;
-	Client *server = NULL;
-	for (QList<ConnectionFrame*>::iterator it(_clientFrames.begin()); it != _clientFrames.end(); ++it)
-	{
-		Client *c = (**it).client();
-		if (c == NULL)
-			continue;
-		if (c->id() == lastProjectionSource)
-			server = c;
-		if (c != client && c->currentProjectionSource() == lastProjectionSource)
-		{
-			inUse = true;
-			break;
-		}
+
+	// VNC Client stopped -> remove from watchers
+	if (!client->isActiveVncClient()){
+		_watchers.remove(client->id());
+
+		// If noboody is watching the multicast stop VNC server
+		if (_watchers.isEmpty() && !_broadcast)
+			_streamingSource->stopVncServer();
 	}
-	// 3. kill server if applicable
-	if (!inUse && server != NULL)
-		server->stopVncServer();
 }
 
 /***************************************************************************//**