fix possible null reference exceptions

This commit is contained in:
Matthias Mailänder
2015-05-31 15:55:28 +02:00
parent a12e6a191b
commit cd5c0eca79

View File

@@ -17,7 +17,8 @@ namespace OpenRA.Mods.Common.AI
public void Update(Squad squad) public void Update(Squad squad)
{ {
currentState.Tick(squad); if (currentState != null)
currentState.Tick(squad);
} }
public void ChangeState(Squad squad, IState newState, bool rememberPrevious) public void ChangeState(Squad squad, IState newState, bool rememberPrevious)
@@ -31,7 +32,8 @@ namespace OpenRA.Mods.Common.AI
if (newState != null) if (newState != null)
currentState = newState; currentState = newState;
currentState.Activate(squad); if (currentState != null)
currentState.Activate(squad);
} }
public void RevertToPreviousState(Squad squad, bool saveCurrentState) public void RevertToPreviousState(Squad squad, bool saveCurrentState)