Bug 611 - serialization of path attributes

[bgpcep.git] / bgp / parser-impl / src / main / java / org / opendaylight / protocol / bgp / parser / impl / message / update / AggregatorAttributeParser.java

diff --git a/bgp/parser-impl/src/main/java/org/opendaylight/protocol/bgp/parser/impl/message/update/AggregatorAttributeParser.java b/bgp/parser-impl/src/main/java/org/opendaylight/protocol/bgp/parser/impl/message/update/AggregatorAttributeParser.java
index 92dda87efd1e6d72b5961891a4fb92966ef715b0..8d5a430eec6692a34fcc6ba773dfd9d0ea2041a4 100644 (file)
--- a/bgp/parser-impl/src/main/java/org/opendaylight/protocol/bgp/parser/impl/message/update/AggregatorAttributeParser.java
+++ b/bgp/parser-impl/src/main/java/org/opendaylight/protocol/bgp/parser/impl/message/update/AggregatorAttributeParser.java
@@ -1,32 +1,68 @@
+/*
+ * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
 package org.opendaylight.protocol.bgp.parser.impl.message.update;
 
+import com.google.common.base.Preconditions;
+import io.netty.buffer.ByteBuf;
+import io.netty.buffer.Unpooled;
+import org.opendaylight.protocol.bgp.parser.AttributeFlags;
 import org.opendaylight.protocol.bgp.parser.spi.AttributeParser;
-import org.opendaylight.protocol.concepts.IPv4;
+import org.opendaylight.protocol.bgp.parser.spi.AttributeSerializer;
+import org.opendaylight.protocol.concepts.Ipv4Util;
 import org.opendaylight.protocol.util.ByteArray;
+import org.opendaylight.protocol.util.ReferenceCache;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.AsNumber;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.path.attributes.Aggregator;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.path.attributes.AggregatorBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.update.PathAttributesBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.PathAttributes;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.Aggregator;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.AggregatorBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.update.PathAttributesBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.ShortAsNumber;
+import org.opendaylight.yangtools.yang.binding.DataObject;
 
-final class AggregatorAttributeParser implements AttributeParser {
-       static final int TYPE = 7;
+public final class AggregatorAttributeParser implements AttributeParser, AttributeSerializer {
 
-       /**
-        * Parse AGGREGATOR from bytes
-        * 
-        * @param bytes byte array to be parsed
-        * @return {@link Aggregator} BGP Aggregator
-        */
-       private static Aggregator parseAggregator(final byte[] bytes) {
-               final AsNumber asNumber = new AsNumber(ByteArray.bytesToLong(ByteArray.subByte(bytes, 0, AsPathSegmentParser.AS_NUMBER_LENGTH)));
-               final Ipv4Address address = new Ipv4Address(IPv4.FAMILY.addressForBytes(
-                               ByteArray.subByte(bytes, AsPathSegmentParser.AS_NUMBER_LENGTH, 4)).toString());
-               return new AggregatorBuilder().setAsNumber(asNumber).setNetworkAddress(address).build();
-       }
+    public static final int TYPE = 7;
 
-       @Override
-       public void parseAttribute(final byte[] bytes, final PathAttributesBuilder builder) {
-               builder.setAggregator(parseAggregator(bytes));
-       }
-}
\ No newline at end of file
+    private final ReferenceCache refCache;
+
+    public AggregatorAttributeParser(final ReferenceCache refCache) {
+        this.refCache = Preconditions.checkNotNull(refCache);
+    }
+
+    /**
+     * Parse AGGREGATOR from bytes
+     *
+     * @param buffer byte buffer to be parsed
+     * @return {@link Aggregator} BGP Aggregator
+     */
+    @Override
+    public void parseAttribute(final ByteBuf buffer, final PathAttributesBuilder builder) {
+        final AsNumber asNumber = this.refCache.getSharedReference(new AsNumber(buffer.readUnsignedInt()));
+        final Ipv4Address address = Ipv4Util.addressForBytes(ByteArray.readAllBytes(buffer));
+        builder.setAggregator(new AggregatorBuilder().setAsNumber(asNumber).setNetworkAddress(address).build());
+    }
+
+    @Override
+    public void serializeAttribute(DataObject attribute, ByteBuf byteAggregator) {
+        PathAttributes pathAttributes = (PathAttributes) attribute;
+        Aggregator aggregator = pathAttributes.getAggregator();
+        if (aggregator == null) {
+            return;
+        }
+        Preconditions.checkArgument(aggregator.getAsNumber() != null, "Missing AS number that formed the aggregate route (encoded as 2 octets).");
+        ShortAsNumber shortAsNumber = new ShortAsNumber(aggregator.getAsNumber());
+        byteAggregator.writeByte(AttributeFlags.OPTIONAL | AttributeFlags.TRANSITIVE);
+        byteAggregator.writeByte(TYPE);
+        ByteBuf aggregatorBuffer = Unpooled.buffer();
+        aggregatorBuffer.writeInt(shortAsNumber.getValue().intValue());
+        aggregatorBuffer.writeBytes(Ipv4Util.bytesForAddress(aggregator.getNetworkAddress()));
+        byteAggregator.writeByte(aggregatorBuffer.writerIndex());
+        byteAggregator.writeBytes(aggregatorBuffer);
+    }
+}