Add a check for valid payloads on lobbyuserlistupdate

etternagame · Dec 2, 2018 · 7d0af34 · 7d0af34
1 parent 54372aa
commit 7d0af34
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/NetworkSyncManager.cpp b/src/NetworkSyncManager.cpp
@@ -944,13 +944,17 @@ ETTProtocol::Update(NetworkSyncManager* n, float fDeltaTime)
 				} break;
 				case ettps_lobbyuserlistupdate: {
 					auto& vec = NSMAN->lobbyuserlist;
-					auto newUsers = payload->at("on");
-					for (auto& user : newUsers) {
-						NSMAN->lobbyuserlist.insert(user.get<string>());
+					if (payload->find("on") != payload->end()) {
+						auto newUsers = payload->at("on");
+						for (auto& user : newUsers) {
+							NSMAN->lobbyuserlist.insert(user.get<string>());
+						}
 					}
-					auto removedUsers = payload->at("off");
-					for (auto& user : removedUsers) {
-						NSMAN->lobbyuserlist.erase(user.get<string>());
+					if (payload->find("off") != payload->end()) {
+						auto removedUsers = payload->at("off");
+						for (auto& user : removedUsers) {
+							NSMAN->lobbyuserlist.erase(user.get<string>());
+						}
 					}
 					SCREENMAN->SendMessageToTopScreen(SM_UsersUpdate);
 				} break;