From b01b65b2d3da124ea55d21e6e31f5fcf72cd80fb Mon Sep 17 00:00:00 2001
From: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>
Date: Tue, 6 Oct 2015 23:28:42 +0900
Subject: [PATCH] server: fix TERM signal handling

gobgpd should die after sending a notification all the peers in
the established state.

Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>
---
 server/server.go | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/server/server.go b/server/server.go
index 82193d00..ff4b292f 100644
--- a/server/server.go
+++ b/server/server.go
@@ -106,6 +106,7 @@ type BgpServer struct {
 	roaClient      *roaClient
 	bmpClient      *bmpClient
 	bmpConnCh      chan *bmpConn
+	shutdown       bool
 }
 
 func NewBgpServer(port int) *BgpServer {
@@ -785,6 +786,18 @@ func (server *BgpServer) handleFSMMessage(peer *Peer, e *fsmMsg, incoming chan *
 				msgs = append(msgs, newSenderMsg(peer, table.CreateUpdateMsgFromPaths(pathList)))
 			}
 		} else {
+			if server.shutdown && nextState == bgp.BGP_FSM_IDLE {
+				die := true
+				for _, p := range server.neighborMap {
+					if p.fsm.state != bgp.BGP_FSM_IDLE {
+						die = false
+						break
+					}
+				}
+				if die {
+					os.Exit(0)
+				}
+			}
 			peer.conf.Timers.TimersState.Downtime = time.Now().Unix()
 		}
 		// clear counter
@@ -876,6 +889,7 @@ func (server *BgpServer) PeerUpdate(peer config.Neighbor) {
 }
 
 func (server *BgpServer) Shutdown() {
+	server.shutdown = true
 	for _, p := range server.neighborMap {
 		p.fsm.adminStateCh <- ADMIN_STATE_DOWN
 	}