Make MP capabilities reusable
[bgpcep.git] / bgp / parser-spi / src / main / java / org / opendaylight / protocol / bgp / parser / spi / MultiprotocolCapabilitiesUtil.java
1 /*
2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.protocol.bgp.parser.spi;
9
10 import com.google.common.base.Preconditions;
11 import io.netty.buffer.ByteBuf;
12 import org.opendaylight.protocol.bgp.parser.BGPParsingException;
13 import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
14 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.optional.capabilities.CParameters;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.CParameters1;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.CParameters1Builder;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.RouteRefreshCapabilityBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.AddressFamily;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.SubsequentAddressFamily;
22
23 public final class MultiprotocolCapabilitiesUtil {
24
25     public static final CParameters RR_CAPABILITY = new CParametersBuilder().addAugmentation(CParameters1.class,
26         new CParameters1Builder().setRouteRefreshCapability(new RouteRefreshCapabilityBuilder().build()).build()).build();
27
28     private static final int RESERVED = 1;
29
30     public static void serializeMPAfiSafi(final AddressFamilyRegistry afiReg, final SubsequentAddressFamilyRegistry safiReg,
31         final Class<? extends AddressFamily> afi, final Class<? extends SubsequentAddressFamily> safi, final ByteBuf capBuffer) {
32         final Integer afival = afiReg.numberForClass(afi);
33         Preconditions.checkArgument(afival != null, "Unhandled address family " + afi);
34         capBuffer.writeShort(afival);
35
36         capBuffer.writeZero(RESERVED);
37
38         final Integer safival = safiReg.numberForClass(safi);
39         Preconditions.checkArgument(safival != null, "Unhandled subsequent address family " + safi);
40         capBuffer.writeByte(safival);
41     }
42
43     public static BgpTableType parseMPAfiSafi(final ByteBuf buffer, final AddressFamilyRegistry afiReg, final SubsequentAddressFamilyRegistry safiReg) throws BGPParsingException {
44         final int afiVal = buffer.readUnsignedShort();
45         final Class<? extends AddressFamily> afi = afiReg.classForFamily(afiVal);
46         if (afi == null) {
47             throw new BGPParsingException("Address Family Identifier: '" + afiVal + "' not supported.");
48         }
49         // skip reserved
50         buffer.skipBytes(RESERVED);
51         final int safiVal = buffer.readUnsignedByte();
52         final Class<? extends SubsequentAddressFamily> safi = safiReg.classForFamily(safiVal);
53         if (safi == null) {
54             throw new BGPParsingException("Subsequent Address Family Identifier: '" + safiVal + "' not supported.");
55         }
56         return new BgpTableTypeImpl(afi, safi);
57     }
58 }