blob: b8f934a1d87650a400e2a12fb41ef35d2743be5e (
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
|
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<>();
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();
BGPPacket packet = BGPPacket.read(socket.getInputStream());
if(packet.getType() != BGPPacket.Type.OPEN)
continue;
BGPOpen request = new BGPOpen(packet.getMessage());
BGPSessionConfiguration config = sessionConfigurations.stream()
.filter(c -> c.getRemoteAs() == request.getAsn())
.findFirst()
.orElse(null);
if(config == null)
continue;
BGPSession session = new BGPSession(socket, config);
config.getListener().onOpen(session);
BGPOpen response = new BGPOpen()
.setAsn(config.getLocalAs())
.setHoldTime(request.getHoldTime())
.setVersion(request.getVersion())
.setIdentifier(config.getLocalIdentifier());
try {
socket.getOutputStream().write(new BGPPacket().setType(BGPPacket.Type.OPEN).setMessage(response.build()).build());
} catch (IOException e) {
e.printStackTrace();
}
session.keepAlive();
new Thread(session).start();
} catch (IOException ex) {
ex.printStackTrace();
}
}
}
}
|
