server/fsm: update fsm state after updating related information

Before updating fsm state to establish, we have to set up fsm.peerInfo, etc.

Signed-off-by: FUJITA Tomonori <[email protected]>

Author: fujita

pkg/server/fsm.go

@@ -539,7 +539,6 @@ func (fsm *fsm) stateChange(nextState bgp.FSMState, reason *fsmStateReason) {
 		slog.String("new", nextState.String()),
 		slog.String("reason", reason.String()))
 
-	fsm.state.Store(nextState)
 	switch nextState {
 	case bgp.BGP_FSM_ESTABLISHED:
 		remoteTCP := fsm.conn.RemoteAddr().(*net.TCPAddr)
@@ -1988,6 +1987,7 @@ func (h *fsmHandler) loop(ctx context.Context, wg *sync.WaitGroup) {
 		}
 
 		h.callback(msg)
+		fsm.state.Store(nextState)
 		oldState = nextState
 	}