diff options
Diffstat (limited to 'src/main/java')
4 files changed, 90 insertions, 3 deletions
diff --git a/src/main/java/com/lumaserv/bgp/BGPFsm.java b/src/main/java/com/lumaserv/bgp/BGPFsm.java index a2143bd..7929d36 100644 --- a/src/main/java/com/lumaserv/bgp/BGPFsm.java +++ b/src/main/java/com/lumaserv/bgp/BGPFsm.java @@ -244,6 +244,36 @@ public class BGPFsm { } class OpenSent extends State { + protected void stop(boolean isAutomatic) { + try { + BGPNotification notification = new BGPNotification() + .setMajorErrorCode(BGPNotification.Error.CEASE.getCode()); + session.sendNotification(notification); + } catch (IOException ex) { + } + setConnectRetryTimer(0); + // - releases all the BGP resources, + session.dropConnection(); + if (isAutomatic) { + connectRetryCounter++; + // - (optionally) performs peer oscillation damping if the + // DampPeerOscillations attribute is set to TRUE, and + } else { + connectRetryCounter = 0; + } + setState(IDLE); + } + + @Override + void manualStop() { + stop(false); + } + + @Override + void automaticStop() { + stop(true); + } + @Override void openMsg(BGPOpen msg) { System.out.println("openMsg:" + msg); @@ -311,6 +341,37 @@ public class BGPFsm { } class Established extends State { + protected void stop(boolean isAutomatic) { + try { + BGPNotification notification = new BGPNotification() + .setMajorErrorCode(BGPNotification.Error.CEASE.getCode()); + session.sendNotification(notification); + } catch (IOException ex) { + } + setConnectRetryTimer(0); + // - deletes all routes associated with this connection, + // - releases all BGP resources, + session.dropConnection(); + if (isAutomatic) { + connectRetryCounter++; + // - (optionally in automaticStop) performs peer oscillation damping if the + // DampPeerOscillations attribute is set to TRUE, and + } else { + connectRetryCounter = 0; + } + setState(IDLE); + } + + @Override + void manualStop() { + stop(false); + } + + @Override + void automaticStop() { + stop(true); + } + @Override void keepAliveMsg() { setHoldTimer(holdTime); diff --git a/src/main/java/com/lumaserv/bgp/BGPServer.java b/src/main/java/com/lumaserv/bgp/BGPServer.java index 695ef96..80ec7ff 100644 --- a/src/main/java/com/lumaserv/bgp/BGPServer.java +++ b/src/main/java/com/lumaserv/bgp/BGPServer.java @@ -15,6 +15,7 @@ public class BGPServer implements Runnable { final ServerSocket serverSocket; @Getter final List<BGPSessionConfiguration> sessionConfigurations = new ArrayList<>(); + final List<BGPSession> sessions = new ArrayList<>(); public BGPServer() throws IOException { this(179); @@ -58,6 +59,7 @@ public class BGPServer implements Runnable { BGPFsm fsm = new BGPFsm(); BGPSession session = new BGPSession(socket, config, fsm); fsm.setSession(session); + sessions.add(session); System.out.println("automaticStartPassive"); fsm.getCurrentState().automaticStartPassive(); @@ -78,6 +80,7 @@ public class BGPServer implements Runnable { BGPFsm fsm = new BGPFsm(); BGPSession session = new BGPSession(config, fsm, host, port); fsm.setSession(session); + sessions.add(session); fsm.getCurrentState().automaticStart(); return true; } catch (IOException ex) { @@ -85,4 +88,10 @@ public class BGPServer implements Runnable { throw(ex); } } + + public void shutdown() { + for(BGPSession session : sessions) { + session.automaticStop(); + } + } } diff --git a/src/main/java/com/lumaserv/bgp/BGPSession.java b/src/main/java/com/lumaserv/bgp/BGPSession.java index 21a5564..c6a8cd2 100644 --- a/src/main/java/com/lumaserv/bgp/BGPSession.java +++ b/src/main/java/com/lumaserv/bgp/BGPSession.java @@ -47,6 +47,10 @@ public class BGPSession implements Runnable { this.port = port; } + public void automaticStop() { + fsm.getCurrentState().automaticStop(); + } + public void keepAlive() { try { outputStream.write(new BGPPacket().setType(BGPPacket.Type.KEEPALIVE).setMessage(new byte[0]).build()); @@ -94,6 +98,10 @@ public class BGPSession implements Runnable { thread.start(); } + void sendNotification(BGPNotification notification) throws IOException { + outputStream.write(new BGPPacket().setType(BGPPacket.Type.NOTIFICATION).setMessage(notification.build()).build()); + } + void sendOpen(BGPOpen request) throws IOException { outputStream.write(new BGPPacket().setType(BGPPacket.Type.OPEN).setMessage(request.build()).build()); System.out.println("Sent open"); @@ -167,9 +175,10 @@ System.out.println("Sent open"); } System.out.println("Closed!"); } catch (IOException ex) { - if (!closed) + if (!closed) { fsm.getCurrentState().tcpConnectionFails(); - ex.printStackTrace(); + ex.printStackTrace(); + } closed = true; configuration.getListener().onClose(this); } diff --git a/src/main/java/com/lumaserv/bgp/protocol/message/BGPNotification.java b/src/main/java/com/lumaserv/bgp/protocol/message/BGPNotification.java index 19272e5..94e3fae 100644 --- a/src/main/java/com/lumaserv/bgp/protocol/message/BGPNotification.java +++ b/src/main/java/com/lumaserv/bgp/protocol/message/BGPNotification.java @@ -1,10 +1,11 @@ package com.lumaserv.bgp.protocol.message; -import lombok.AllArgsConstructor; +import lombok.NoArgsConstructor; import lombok.Getter; import lombok.Setter; @Getter +@NoArgsConstructor @Setter public class BGPNotification { @@ -49,6 +50,13 @@ public class BGPNotification { this.minorErrorCode = message[1]; } + public byte[] build() { + byte[] message = new byte[2]; + message[0] = majorErrorCode; + message[1] = minorErrorCode; + return message; + } + @Override public String toString() { return "" + majorErrorCode + ":" + minorErrorCode; |