bgp/parser-spi/src/main/java/org/opendaylight/protocol/bgp/parser/spi/pojo/SimpleExtendedCommunityRegistry.java

   1 /*
   2  * Copyright (c) 2015 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.pojo;
   9
  10 import static org.opendaylight.protocol.bgp.parser.spi.extended.community.ExtendedCommunityUtil.isTransitive;
  11 import static org.opendaylight.protocol.util.Values.UNSIGNED_BYTE_MAX_VALUE;
  12
  13 import com.google.common.base.Preconditions;
  14 import com.google.common.primitives.Shorts;
  15 import io.netty.buffer.ByteBuf;
  16 import org.opendaylight.protocol.bgp.parser.BGPDocumentedException;
  17 import org.opendaylight.protocol.bgp.parser.BGPParsingException;
  18 import org.opendaylight.protocol.bgp.parser.spi.extended.community.ExtendedCommunityParser;
  19 import org.opendaylight.protocol.bgp.parser.spi.extended.community.ExtendedCommunityRegistry;
  20 import org.opendaylight.protocol.bgp.parser.spi.extended.community.ExtendedCommunitySerializer;
  21 import org.opendaylight.protocol.concepts.HandlerRegistry;
  22 import org.opendaylight.protocol.util.ByteBufWriteUtil;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.ExtendedCommunities;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.ExtendedCommunitiesBuilder;
  25 import org.opendaylight.yangtools.yang.binding.DataContainer;
  26 import org.slf4j.Logger;
  27 import org.slf4j.LoggerFactory;
  28
  29 final class SimpleExtendedCommunityRegistry implements ExtendedCommunityRegistry {
  30
  31     private static final Logger LOG = LoggerFactory.getLogger(SimpleExtendedCommunityRegistry.class);
  32
  33     private static final int EXTENDED_COMMUNITY_LENGTH = 6;
  34
  35     private final HandlerRegistry<DataContainer, ExtendedCommunityParser, ExtendedCommunitySerializer> handlers =
  36             new HandlerRegistry<>();
  37
  38     private static int createKey(final int type, final int subtype) {
  39         return type << Byte.SIZE | subtype;
  40     }
  41
  42     @Override
  43     public ExtendedCommunities parseExtendedCommunity(final ByteBuf buffer)
  44             throws BGPDocumentedException, BGPParsingException {
  45         final short type = buffer.readUnsignedByte();
  46         final short subtype = buffer.readUnsignedByte();
  47         final ExtendedCommunityParser parser = this.handlers.getParser(createKey(type, subtype));
  48         if (parser == null) {
  49             buffer.skipBytes(EXTENDED_COMMUNITY_LENGTH);
  50             LOG.info("Skipping unknown extended-community type/sub-type {}/{}.", type, subtype);
  51             return null;
  52         }
  53         return new ExtendedCommunitiesBuilder()
  54             .setTransitive(isTransitive(type))
  55             .setExtendedCommunity(parser.parseExtendedCommunity(buffer))
  56             .build();
  57     }
  58
  59     @Override
  60     public void serializeExtendedCommunity(final ExtendedCommunities extendedCommunity, final ByteBuf byteAggregator) {
  61         final ExtendedCommunitySerializer serializer = this.handlers.getSerializer(extendedCommunity
  62             .getExtendedCommunity().getImplementedInterface());
  63         if (serializer == null) {
  64             return;
  65         }
  66         ByteBufWriteUtil.writeUnsignedByte(Shorts.checkedCast(serializer.getType(extendedCommunity.isTransitive())),
  67             byteAggregator);
  68         ByteBufWriteUtil.writeUnsignedByte(Shorts.checkedCast(serializer.getSubType()), byteAggregator);
  69         serializer.serializeExtendedCommunity(extendedCommunity.getExtendedCommunity(), byteAggregator);
  70     }
  71
  72     synchronized AutoCloseable registerExtendedCommunitySerializer(
  73             final Class<? extends org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329
  74                     .extended.community.ExtendedCommunity> extendedCommunityClass,
  75             final ExtendedCommunitySerializer serializer) {
  76         return this.handlers.registerSerializer(extendedCommunityClass, serializer);
  77     }
  78
  79     synchronized AutoCloseable registerExtendedCommunityParser(final int type, final int subtype,
  80             final ExtendedCommunityParser parser) {
  81         checkTypes(type, subtype);
  82         return this.handlers.registerParser(createKey(type, subtype), parser);
  83     }
  84
  85     private static void checkTypes(final int type, final int subtype) {
  86         Preconditions.checkArgument(type >= 0 && type <= UNSIGNED_BYTE_MAX_VALUE, "Illegal extended-community type %s",
  87                 type);
  88         Preconditions.checkArgument(subtype >= 0 && subtype <= UNSIGNED_BYTE_MAX_VALUE,
  89                 "Illegal extended-community sub-type %s", subtype);
  90     }
  91
  92 }