blob: 80ec7ffdd3bac5feb605f2fb7d3d78f014dccc54 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
package com.lumaserv.bgp;
import com.lumaserv.bgp.protocol.BGPPacket;
import com.lumaserv.bgp.protocol.message.BGPOpen;
import lombok.Getter;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;
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);
}
public BGPServer(int port) throws IOException {
serverSocket = new ServerSocket(port);
}
private static boolean checkEqual(byte[] a, byte[] b) {
if(a == b)
return true;
if(a == null || b == null)
return false;
if(a.length != b.length)
return false;
for(int i=0; i<a.length; i++) {
if(a[i] != b[i])
return false;
}
return true;
}
public void run() {
while (true) {
try {
Socket socket = serverSocket.accept();
System.out.println("Accept");
BGPSessionConfiguration config = sessionConfigurations.stream()
.filter(c -> socket.getInetAddress().equals(c.getRemoteAddr()))
.findFirst()
.orElse(null);
System.out.println("Config: " + config);
if(config == null) {
System.out.println("Peer not found:" + socket.getInetAddress());
socket.close();
continue;
}
BGPFsm fsm = new BGPFsm();
BGPSession session = new BGPSession(socket, config, fsm);
fsm.setSession(session);
sessions.add(session);
System.out.println("automaticStartPassive");
fsm.getCurrentState().automaticStartPassive();
// System.out.println("tcpConnectionConfirmed in " + fsm.getCurrentState());
// fsm.getCurrentState().tcpConnectionConfirmed();
} catch (IOException ex) {
ex.printStackTrace();
}
}
}
public boolean connect(BGPSessionConfiguration config, String host) throws IOException {
return connect(config, host, 179);
}
public boolean connect(BGPSessionConfiguration config, String host, int port) throws IOException {
try {
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) {
ex.printStackTrace();
throw(ex);
}
}
public void shutdown() {
for(BGPSession session : sessions) {
session.automaticStop();
}
}
}
|
