FSM: Check if the FSM is defined for transitions and state checks

1 files changed, 8 insertions, 0 deletions

diff --git a/src/fsm.go b/src/fsm.go
index 7eb1ca1..2e778d9 100644
--- a/src/fsm.go
+++ b/src/fsm.go
@@ -93,6 +93,10 @@ type FSM struct {
 }
 
 func (eduvpn *VPNState) HasTransition(check FSMStateID) bool {
+	// No fsm
+	if eduvpn.FSM == nil {
+		return false
+	}
 	for _, transition_state := range eduvpn.FSM.States[eduvpn.FSM.Current] {
 		if transition_state.To == check {
 			return true
@@ -103,6 +107,10 @@ func (eduvpn *VPNState) HasTransition(check FSMStateID) bool {
 }
 
 func (eduvpn *VPNState) InState(check FSMStateID) bool {
+	// No fsm
+	if eduvpn.FSM == nil {
+		return false
+	}
 	return check == eduvpn.FSM.Current
 }