diff options
Diffstat (limited to 'src/main/java/com/lumaserv/bgp/protocol/message')
| -rw-r--r-- | src/main/java/com/lumaserv/bgp/protocol/message/BGPOpen.java | 66 | ||||
| -rw-r--r-- | src/main/java/com/lumaserv/bgp/protocol/message/BGPUpdate.java | 62 |
2 files changed, 81 insertions, 47 deletions
diff --git a/src/main/java/com/lumaserv/bgp/protocol/message/BGPOpen.java b/src/main/java/com/lumaserv/bgp/protocol/message/BGPOpen.java index e83b751..37b791b 100644 --- a/src/main/java/com/lumaserv/bgp/protocol/message/BGPOpen.java +++ b/src/main/java/com/lumaserv/bgp/protocol/message/BGPOpen.java @@ -1,52 +1,54 @@ package com.lumaserv.bgp.protocol.message; +import com.lumaserv.bgp.protocol.DataBuilder; + import lombok.Getter; import lombok.NoArgsConstructor; import lombok.Setter; +import java.nio.ByteBuffer; + @Getter @Setter @NoArgsConstructor -public class BGPOpen { +public class BGPOpen implements DataBuilder { byte version; int asn; int holdTime; byte[] identifier; - public BGPOpen(byte[] message) { - version = message[0]; - asn = ((message[1] & 0xFF) << 8) | (message[2] & 0xFF); - holdTime = ((message[3] & 0xFF) << 8) | (message[4] & 0xFF); + public BGPOpen(ByteBuffer message) { +System.out.println("BGPOpen pos:" + message.position()); + version = message.get(); + asn = message.getShort() & 0xFFFF; + holdTime = message.getShort() & 0xFFFF; identifier = new byte[4]; +System.out.println("BGPOpen asn:" + asn); + message.get(identifier); } - public byte[] build() { - byte[] message = new byte[5 + identifier.length + 1 + 2 + 2 * 6]; - message[0] = version; - message[1] = (byte) (asn >> 8); - message[2] = (byte) (asn & 0xFF); - message[3] = (byte) (holdTime >> 8); - message[4] = (byte) (holdTime & 0xFF); - System.arraycopy(identifier, 0, message, 5, identifier.length); - message[5 + identifier.length + 0] = 14; // Length (0E) - message[5 + identifier.length + 1] = 2; // Parameter Type: capability (02) - message[5 + identifier.length + 2] = 12; // Length (0C) - - message[5 + identifier.length + 3] = 1; // Type: multiprotocol (01) - message[5 + identifier.length + 4] = 4; // Length (04) - message[5 + identifier.length + 5] = 0; // IPv6 (0002) - message[5 + identifier.length + 6] = 2; // - message[5 + identifier.length + 7] = 0; // Reserved - message[5 + identifier.length + 8] = 1; // Unicast (01) - - message[5 + identifier.length + 9] = 1; // Type: multiprotocol (01) - message[5 + identifier.length +10] = 4; // Length (04) - message[5 + identifier.length +11] = 0; // IPv4 (0001) - message[5 + identifier.length +12] = 1; // - message[5 + identifier.length +13] = 0; // Reserved - message[5 + identifier.length +14] = 1; // Unicast (01) - return message; + public void build(ByteBuffer b) { + b.put(version); + b.put((byte) (asn >> 8)); + b.put((byte) (asn & 0xFF)); + b.put((byte) (holdTime >> 8)); + b.put((byte) (holdTime & 0xFF)); + b.put(identifier); + b.put((byte)14); // Length (0E) + b.put((byte)2); // Parameter Type: capability (02) + b.put((byte)12); // Length (0C) + + b.put((byte)1); // Type: multiprotocol (01) + b.put((byte)4); // Length (04) + b.putShort((short)2); // IPv6 (0002) + b.put((byte)0); // Reserved + b.put((byte)1); // Unicast (01) + + b.put((byte)1); // Type: multiprotocol (01) + b.put((byte)4); // Length (04) + b.putShort((short)1); // IPv4 (0001) + b.put((byte)0); // Reserved + b.put((byte)1); // Unicast (01) } - } diff --git a/src/main/java/com/lumaserv/bgp/protocol/message/BGPUpdate.java b/src/main/java/com/lumaserv/bgp/protocol/message/BGPUpdate.java index fa23432..ad6c39b 100644 --- a/src/main/java/com/lumaserv/bgp/protocol/message/BGPUpdate.java +++ b/src/main/java/com/lumaserv/bgp/protocol/message/BGPUpdate.java @@ -1,63 +1,95 @@ package com.lumaserv.bgp.protocol.message; +import com.lumaserv.bgp.protocol.DataBuilder; import com.lumaserv.bgp.protocol.IPPrefix; import com.lumaserv.bgp.protocol.attribute.PathAttribute; import lombok.Getter; +import lombok.NoArgsConstructor; import lombok.Setter; +import java.nio.ByteBuffer; import java.util.ArrayList; import java.util.List; +@NoArgsConstructor @Getter @Setter -public class BGPUpdate { +public class BGPUpdate implements DataBuilder { List<IPPrefix> withdrawnPrefixes = new ArrayList<>(); List<PathAttribute> attributes = new ArrayList<>(); List<IPPrefix> prefixes = new ArrayList<>(); - public BGPUpdate(byte[] message) { - int routesLength = ((message[0] & 0xFF) << 8) | (message[1] & 0xFF); + public BGPUpdate(ByteBuffer message) { + int routesLength = message.getShort() & 0xFFFF; int offset = 2; int offsetOffset = 2; while ((offset - offsetOffset) < routesLength) { IPPrefix prefix = new IPPrefix() - .setLength(message[offset]) + .setLength(message.get()) .setAddress(new byte[4]); offset++; int addressLen = (int) Math.ceil(prefix.getLength() / 8d); - System.arraycopy(message, offset, prefix.getAddress(), 0, addressLen); + message.get(prefix.getAddress(), 0, addressLen); offset += addressLen; withdrawnPrefixes.add(prefix); } - int attributesLength = ((message[offset] & 0xFF) << 8) | (message[offset + 1] & 0xFF); + int attributesLength = message.getShort() & 0xFFFF; offset += 2; offsetOffset = offset; while ((offset - offsetOffset) < attributesLength) { - byte flags = message[offset]; - byte typeCode = message[offset + 1]; - int length = message[offset + 2] & 0xFF; + byte flags = message.get(); + byte typeCode = message.get(); + int length = message.get() & 0xFF; offset += 3; if((flags & 0b0001_0000) > 0) { length <<= 8; - length |= message[offset] & 0xFF; + length |= message.get() & 0xFF; offset++; } byte[] data = new byte[length]; - System.arraycopy(message, offset, data, 0, length); + message.get(data); offset += length; - attributes.add(PathAttribute.from(typeCode, data)); + PathAttribute attr = PathAttribute.from(typeCode, data); + attributes.add(attr); } - while (offset < message.length) { + while (message.hasRemaining()) { IPPrefix prefix = new IPPrefix() - .setLength(message[offset]) + .setLength(message.get()) .setAddress(new byte[4]); offset++; int addressLen = (int) Math.ceil(prefix.getLength() / 8d); - System.arraycopy(message, offset, prefix.getAddress(), 0, addressLen); + message.get(prefix.getAddress(), 0, addressLen); offset += addressLen; prefixes.add(prefix); } } + public void build(ByteBuffer message) { + int first = message.position(); + message.putShort((short)0); // Withdrawn + message.putShort((short)0); // Attributes + // TODO + int pos = message.position(); + for (PathAttribute attr: attributes) { +// FIXME add header + byte flags = 0b0100_0000; // TODO + int start = message.position(); + message.put(flags); + message.put(attr.getTypeCode()); + message.put((byte)0); // Dummy length + attr.build(message); + int length = message.position() - (start + 3); + if (length > 255) + throw new RuntimeException("attribute length > 255"); + message.put(start + 2, (byte)(length & 0xFF)); + } + int attributesLength = message.position() - pos; + message.putShort(first + 2, (short)attributesLength); + for (IPPrefix prefix : prefixes) { + message.put((byte)prefix.getLength()); + int addressLen = (int) Math.ceil(prefix.getLength() / 8d); + message.put(prefix.getAddress(), 0, addressLen); + } + } } |