1 files changed, 12 insertions, 3 deletions

diff --git a/pacman-c++/mainwidget.cpp b/pacman-c++/mainwidget.cpp
index b69e0c0..6c4012e 100644
--- a/pacman-c++/mainwidget.cpp
+++ b/pacman-c++/mainwidget.cpp
@@ -149,7 +149,8 @@ void MainWidget::tick()
 
     //qDebug() << "read str " << dataStr.length();
     ProtoBuf::MapUpdate packet;
-    packet.ParseFromString(dataStr);
+    bool worked = packet.ParseFromString(dataStr);
+    Q_ASSERT(worked);
     Transmission::map_t map = Util::createUninitialisedMap();
     Q_ASSERT(packet.field_size() == (int) (Constants::map_size.width * Constants::map_size.height));
     int i = 0;
@@ -229,8 +230,14 @@ void MainWidget::playerScoreClicked()
 
 Color::Color MainWidget::connectToServer()
 {
+  // check command line arguments for server port
+  const QStringList &args = QCoreApplication::arguments();
+  QString srv = args.value(1, "127.0.0.1");
+  qDebug() << "srv: " << srv;
+
+  // connect to server
   m_socket = new QTcpSocket(this);
-  m_socket->connectToHost("127.0.0.1", Constants::port);
+  m_socket->connectToHost(srv, Constants::port);
   bool worked =  m_socket->waitForConnected(Constants::connection_timeout);
   if (worked)
   {
@@ -239,10 +246,12 @@ Color::Color MainWidget::connectToServer()
     worked = m_socket->waitForReadyRead();
     if (worked)
     {
+      // receive color
       std::string data;
       Util::QByteArrayToStdString(m_socket->readAll(), data);
       ProtoBuf::WhoAmI packet;
-      packet.ParseFromString(data);
+      Q_ASSERT(worked);
+      bool worked = packet.ParseFromString(data);
       return static_cast<Color::Color>(packet.color() & Transmission::color_mask);
     }
   }