BUG-113: Split capability handlers out
[bgpcep.git] / bgp / parser-impl / src / main / java / org / opendaylight / protocol / bgp / parser / impl / message / open / MultiProtocolCapabilityHandler.java
1 /*
2  * Copyright (c) 2013 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.impl.message.open;
9
10 import org.opendaylight.protocol.bgp.parser.BGPDocumentedException;
11 import org.opendaylight.protocol.bgp.parser.BGPParsingException;
12 import org.opendaylight.protocol.bgp.parser.impl.ParserUtil;
13 import org.opendaylight.protocol.bgp.parser.spi.CapabilityParser;
14 import org.opendaylight.protocol.bgp.parser.spi.CapabilitySerializer;
15 import org.opendaylight.protocol.bgp.parser.spi.CapabilityUtil;
16 import org.opendaylight.protocol.util.ByteArray;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.open.bgp.parameters.CParameters;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130918.open.bgp.parameters.c.parameters.CMultiprotocol;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130918.open.bgp.parameters.c.parameters.CMultiprotocolBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130918.open.bgp.parameters.c.parameters.c.multiprotocol.MultiprotocolCapabilityBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.AddressFamily;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.SubsequentAddressFamily;
23
24 import com.google.common.primitives.UnsignedBytes;
25
26 final class MultiProtocolCapabilityHandler implements CapabilityParser, CapabilitySerializer {
27         static final int CODE = 1;
28
29         private static final int AFI_SIZE = 2; // bytes
30         private static final int SAFI_SIZE = 1; // bytes
31
32         @Override
33         public CMultiprotocol parseCapability(final byte[] bytes) throws BGPDocumentedException, BGPParsingException {
34                 final Class<? extends AddressFamily> afi = ParserUtil.afiForValue(ByteArray.bytesToInt(ByteArray.subByte(bytes, 0, AFI_SIZE)));
35                 if (afi == null) {
36                         throw new BGPParsingException("Address Family Identifier: '" + ByteArray.bytesToInt(ByteArray.subByte(bytes, 0, AFI_SIZE))
37                                         + "' not supported.");
38                 }
39                 final Class<? extends SubsequentAddressFamily> safi = ParserUtil.safiForValue(ByteArray.bytesToInt(ByteArray.subByte(bytes,
40                                 AFI_SIZE + 1, SAFI_SIZE)));
41                 if (safi == null) {
42                         throw new BGPParsingException("Subsequent Address Family Identifier: '"
43                                         + ByteArray.bytesToInt(ByteArray.subByte(bytes, AFI_SIZE + 1, SAFI_SIZE)) + "' not supported.");
44                 }
45
46                 return new CMultiprotocolBuilder().setMultiprotocolCapability(
47                                 new MultiprotocolCapabilityBuilder().setAfi(afi).setSafi(safi).build()).build();
48         }
49
50         @Override
51         public byte[] serializeCapability(final CParameters capability) {
52                 final CMultiprotocol mp = (CMultiprotocol) capability;
53
54                 final int afival = ParserUtil.valueForAfi(mp.getMultiprotocolCapability().getAfi());
55                 final int safival = ParserUtil.valueForSafi(mp.getMultiprotocolCapability().getSafi());
56
57                 return CapabilityUtil.formatCapability(CODE, new byte[] {
58                                 UnsignedBytes.checkedCast(afival / 256),
59                                 UnsignedBytes.checkedCast(afival % 256),
60                                 0,
61                                 UnsignedBytes.checkedCast(safival)
62                 });
63         }
64 }