-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-
-import com.google.common.base.Preconditions;
-
-public final class BGPActivator implements BGPExtensionProviderActivator {
- private static final Logger logger = LoggerFactory.getLogger(BGPActivator.class);
- private List<AutoCloseable> registrations;
-
- @Override
- public synchronized void start(final BGPExtensionProviderContext context) {
- Preconditions.checkState(this.registrations == null);
- final List<AutoCloseable> regs = new ArrayList<>();
-
- final AddressFamilyRegistry afiReg = context.getAddressFamilyRegistry();
- regs.add(context.registerAddressFamily(Ipv4AddressFamily.class, 1));
- regs.add(context.registerAddressFamily(Ipv6AddressFamily.class, 2));
-
- final SubsequentAddressFamilyRegistry safiReg = context.getSubsequentAddressFamilyRegistry();
- regs.add(context.registerSubsequentAddressFamily(UnicastSubsequentAddressFamily.class, 1));
- regs.add(context.registerSubsequentAddressFamily(MplsLabeledVpnSubsequentAddressFamily.class, 128));
-
- final NlriRegistry nlriReg = context.getNlriRegistry();
- regs.add(context.registerNlriParser(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class, new Ipv4NlriParser()));
- regs.add(context.registerNlriParser(Ipv6AddressFamily.class, UnicastSubsequentAddressFamily.class, new Ipv6NlriParser()));
-
- final AttributeRegistry attrReg = context.getAttributeRegistry();
- regs.add(context.registerAttributeParser(OriginAttributeParser.TYPE, new OriginAttributeParser()));
- regs.add(context.registerAttributeParser(AsPathAttributeParser.TYPE, new AsPathAttributeParser()));
- regs.add(context.registerAttributeParser(NextHopAttributeParser.TYPE, new NextHopAttributeParser()));
- regs.add(context.registerAttributeParser(MultiExitDiscriminatorAttributeParser.TYPE, new MultiExitDiscriminatorAttributeParser()));
- regs.add(context.registerAttributeParser(LocalPreferenceAttributeParser.TYPE, new LocalPreferenceAttributeParser()));
- regs.add(context.registerAttributeParser(AtomicAggregateAttributeParser.TYPE, new AtomicAggregateAttributeParser()));
- regs.add(context.registerAttributeParser(AggregatorAttributeParser.TYPE, new AggregatorAttributeParser()));
- regs.add(context.registerAttributeParser(CommunitiesAttributeParser.TYPE, new CommunitiesAttributeParser()));
- regs.add(context.registerAttributeParser(OriginatorIdAttributeParser.TYPE, new OriginatorIdAttributeParser()));
- regs.add(context.registerAttributeParser(ClusterIdAttributeParser.TYPE, new ClusterIdAttributeParser()));
- regs.add(context.registerAttributeParser(MPReachAttributeParser.TYPE, new MPReachAttributeParser(nlriReg)));
- regs.add(context.registerAttributeParser(MPUnreachAttributeParser.TYPE, new MPUnreachAttributeParser(nlriReg)));
- regs.add(context.registerAttributeParser(ExtendedCommunitiesAttributeParser.TYPE, new ExtendedCommunitiesAttributeParser()));
- regs.add(context.registerAttributeParser(AS4AggregatorAttributeParser.TYPE, new AS4AggregatorAttributeParser()));
- regs.add(context.registerAttributeParser(AS4PathAttributeParser.TYPE, new AS4PathAttributeParser()));
-
- final CapabilityRegistry capReg = context.getCapabilityRegistry();
- final MultiProtocolCapabilityHandler multi = new MultiProtocolCapabilityHandler(afiReg, safiReg);
- regs.add(context.registerCapabilityParser(MultiProtocolCapabilityHandler.CODE, multi));
- regs.add(context.registerCapabilitySerializer(CMultiprotocol.class, multi));
-
- final As4CapabilityHandler as4 = new As4CapabilityHandler();
- regs.add(context.registerCapabilityParser(As4CapabilityHandler.CODE, as4));
- regs.add(context.registerCapabilitySerializer(CAs4Bytes.class, as4));
-
- final GracefulCapabilityHandler grace = new GracefulCapabilityHandler(afiReg, safiReg);
- regs.add(context.registerCapabilitySerializer(CGracefulRestart.class, grace));
- regs.add(context.registerCapabilityParser(GracefulCapabilityHandler.CODE, grace));
-
- final ParameterRegistry paramReg = context.getParameterRegistry();
- final CapabilityParameterParser cpp = new CapabilityParameterParser(capReg);
- regs.add(context.registerParameterParser(CapabilityParameterParser.TYPE, cpp));
- regs.add(context.registerParameterSerializer(BgpParameters.class, cpp));
-
- final BGPOpenMessageParser omp = new BGPOpenMessageParser(paramReg);
- regs.add(context.registerMessageParser(BGPOpenMessageParser.TYPE, omp));
- regs.add(context.registerMessageSerializer(Open.class, omp));
-
- final BGPUpdateMessageParser ump = new BGPUpdateMessageParser(attrReg);
- regs.add(context.registerMessageParser(BGPUpdateMessageParser.TYPE, ump));
- // Serialization of Update message is not supported
-
- final BGPNotificationMessageParser nmp = new BGPNotificationMessageParser();
- regs.add(context.registerMessageParser(BGPNotificationMessageParser.TYPE, nmp));
- regs.add(context.registerMessageSerializer(Notify.class, nmp));
-
- final BGPKeepAliveMessageParser kamp = new BGPKeepAliveMessageParser();
- regs.add(context.registerMessageParser(BGPKeepAliveMessageParser.TYPE, kamp));
- regs.add(context.registerMessageSerializer(Keepalive.class, kamp));
-
- this.registrations = regs;
- }
-
- @Override
- public synchronized void stop() {
- Preconditions.checkState(this.registrations != null);
-
- for (final AutoCloseable r : this.registrations) {
- try {
- r.close();
- } catch (final Exception e) {
- logger.warn("Failed to close registration", e);
- }
- }
-
- this.registrations = null;
- }
+
+public final class BGPActivator extends AbstractBGPExtensionProviderActivator {
+
+ private static final int IPV4_AFI = 1;
+ private static final int IPV6_AFI = 2;
+
+ private static final int UNICAST_SAFI = 1;
+ private static final int VPN_SAFI = 128;
+
+ @Override
+ protected List<AutoCloseable> startImpl(final BGPExtensionProviderContext context) {
+ final List<AutoCloseable> regs = new ArrayList<>();
+
+ regs.add(context.registerNlriSerializer(AdvertizedRoutes.class, new AdvertizedRoutesSerializer()));
+ regs.add(context.registerNlriSerializer(WithdrawnRoutes.class, new WithdrawnRoutesSerializer()));
+
+ regs.add(context.registerAddressFamily(Ipv4AddressFamily.class, IPV4_AFI));
+ regs.add(context.registerAddressFamily(Ipv6AddressFamily.class, IPV6_AFI));
+
+ regs.add(context.registerSubsequentAddressFamily(UnicastSubsequentAddressFamily.class, UNICAST_SAFI));
+ regs.add(context.registerSubsequentAddressFamily(MplsLabeledVpnSubsequentAddressFamily.class, VPN_SAFI));
+
+ regs.add(context.registerNlriParser(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class, new Ipv4NlriParser()));
+ regs.add(context.registerNlriParser(Ipv6AddressFamily.class, UnicastSubsequentAddressFamily.class, new Ipv6NlriParser()));
+
+ this.registerCapabilityParsers(regs, context);
+ this.registerAttributeParsers(regs, context);
+ this.registerMessageParsers(regs, context);
+ return regs;
+ }
+
+ private void registerCapabilityParsers(final List<AutoCloseable> regs, final BGPExtensionProviderContext context) {
+ final AddressFamilyRegistry afiReg = context.getAddressFamilyRegistry();
+ final SubsequentAddressFamilyRegistry safiReg = context.getSubsequentAddressFamilyRegistry();
+
+ final MultiProtocolCapabilityHandler multi = new MultiProtocolCapabilityHandler(afiReg, safiReg);
+ regs.add(context.registerCapabilityParser(MultiProtocolCapabilityHandler.CODE, multi));
+ regs.add(context.registerCapabilitySerializer(MultiprotocolCapability.class, multi));
+
+ final As4CapabilityHandler as4 = new As4CapabilityHandler();
+ regs.add(context.registerCapabilityParser(As4CapabilityHandler.CODE, as4));
+ regs.add(context.registerCapabilitySerializer(As4BytesCapability.class, as4));
+
+ final GracefulCapabilityHandler grace = new GracefulCapabilityHandler(afiReg, safiReg);
+ regs.add(context.registerCapabilitySerializer(GracefulRestartCapability.class, grace));
+ regs.add(context.registerCapabilityParser(GracefulCapabilityHandler.CODE, grace));
+
+ final CapabilityParameterParser cpp = new CapabilityParameterParser(context.getCapabilityRegistry());
+ regs.add(context.registerParameterParser(CapabilityParameterParser.TYPE, cpp));
+ regs.add(context.registerParameterSerializer(BgpParameters.class, cpp));
+ }
+
+ private void registerAttributeParsers(final List<AutoCloseable> regs, final BGPExtensionProviderContext context) {
+ final OriginAttributeParser originAttributeParser = new OriginAttributeParser();
+ regs.add(context.registerAttributeSerializer(Origin.class, originAttributeParser));
+ regs.add(context.registerAttributeParser(OriginAttributeParser.TYPE, originAttributeParser));
+
+ final AigpAttributeParser aigpAttributeParser = new AigpAttributeParser();
+ regs.add(context.registerAttributeSerializer(Aigp.class, aigpAttributeParser));
+ regs.add(context.registerAttributeParser(AigpAttributeParser.TYPE, aigpAttributeParser));
+
+ final AsPathAttributeParser asPathAttributeParser = new AsPathAttributeParser(context.getReferenceCache());
+ regs.add(context.registerAttributeSerializer(AsPath.class, asPathAttributeParser));
+ regs.add(context.registerAttributeParser(AsPathAttributeParser.TYPE, asPathAttributeParser));
+
+ final NextHopAttributeParser nextHopAttributeParser = new NextHopAttributeParser();
+ regs.add(context.registerAttributeSerializer(NextHop.class, nextHopAttributeParser));
+ regs.add(context.registerAttributeParser(NextHopAttributeParser.TYPE, nextHopAttributeParser));
+
+ final MultiExitDiscriminatorAttributeParser multiExitDiscriminatorAttributeParser = new MultiExitDiscriminatorAttributeParser();
+ regs.add(context.registerAttributeSerializer(MultiExitDisc.class, multiExitDiscriminatorAttributeParser));
+ regs.add(context.registerAttributeParser(MultiExitDiscriminatorAttributeParser.TYPE, multiExitDiscriminatorAttributeParser));
+
+ final LocalPreferenceAttributeParser localPreferenceAttributeParser = new LocalPreferenceAttributeParser();
+ regs.add(context.registerAttributeSerializer(LocalPref.class, localPreferenceAttributeParser));
+ regs.add(context.registerAttributeParser(LocalPreferenceAttributeParser.TYPE, localPreferenceAttributeParser));
+
+ final AtomicAggregateAttributeParser atomicAggregateAttributeParser = new AtomicAggregateAttributeParser();
+ regs.add(context.registerAttributeSerializer(AtomicAggregate.class, atomicAggregateAttributeParser));
+ regs.add(context.registerAttributeParser(AtomicAggregateAttributeParser.TYPE, atomicAggregateAttributeParser));
+
+ final AggregatorAttributeParser as4AggregatorAttributeParser = new AggregatorAttributeParser(context.getReferenceCache());
+ regs.add(context.registerAttributeSerializer(Aggregator.class, as4AggregatorAttributeParser));
+ regs.add(context.registerAttributeParser(AggregatorAttributeParser.TYPE, as4AggregatorAttributeParser));
+
+ final CommunitiesAttributeParser communitiesAttributeParser = new CommunitiesAttributeParser(context.getReferenceCache());
+ regs.add(context.registerAttributeSerializer(Communities.class, communitiesAttributeParser));
+ regs.add(context.registerAttributeParser(CommunitiesAttributeParser.TYPE, communitiesAttributeParser));
+
+ final OriginatorIdAttributeParser originatorIdAttributeParser = new OriginatorIdAttributeParser();
+ regs.add(context.registerAttributeSerializer(OriginatorId.class, originatorIdAttributeParser));
+ regs.add(context.registerAttributeParser(OriginatorIdAttributeParser.TYPE, originatorIdAttributeParser));
+
+ final ClusterIdAttributeParser clusterIdAttributeParser = new ClusterIdAttributeParser();
+ regs.add(context.registerAttributeSerializer(ClusterId.class, clusterIdAttributeParser));
+ regs.add(context.registerAttributeParser(ClusterIdAttributeParser.TYPE, clusterIdAttributeParser));
+
+ final NlriRegistry nlriReg = context.getNlriRegistry();
+
+ final MPReachAttributeParser mpReachAttributeParser = new MPReachAttributeParser(nlriReg);
+ regs.add(context.registerAttributeSerializer(MpReachNlri.class, mpReachAttributeParser));
+ regs.add(context.registerAttributeParser(MPReachAttributeParser.TYPE, mpReachAttributeParser));
+
+ final MPUnreachAttributeParser mpUnreachAttributeParser = new MPUnreachAttributeParser(nlriReg);
+ regs.add(context.registerAttributeSerializer(MpUnreachNlri.class, mpUnreachAttributeParser));
+ regs.add(context.registerAttributeParser(MPUnreachAttributeParser.TYPE, mpUnreachAttributeParser));
+
+ final ExtendedCommunitiesAttributeParser extendedCommunitiesAttributeParser = new ExtendedCommunitiesAttributeParser(context.getReferenceCache());
+ regs.add(context.registerAttributeSerializer(ExtendedCommunities.class, extendedCommunitiesAttributeParser));
+ regs.add(context.registerAttributeParser(ExtendedCommunitiesAttributeParser.TYPE, extendedCommunitiesAttributeParser));
+
+ regs.add(context.registerAttributeParser(AS4AggregatorAttributeParser.TYPE, new AS4AggregatorAttributeParser()));
+ regs.add(context.registerAttributeParser(AS4PathAttributeParser.TYPE, new AS4PathAttributeParser()));
+ }
+
+ private void registerMessageParsers(final List<AutoCloseable> regs, final BGPExtensionProviderContext context) {
+ final BGPOpenMessageParser omp = new BGPOpenMessageParser(context.getParameterRegistry());
+ regs.add(context.registerMessageParser(BGPOpenMessageParser.TYPE, omp));
+ regs.add(context.registerMessageSerializer(Open.class, omp));
+
+ final BGPUpdateMessageParser ump = new BGPUpdateMessageParser(context.getAttributeRegistry());
+ regs.add(context.registerMessageParser(BGPUpdateMessageParser.TYPE, ump));
+ regs.add(context.registerMessageSerializer(Update.class, ump));
+
+ final BGPNotificationMessageParser nmp = new BGPNotificationMessageParser();
+ regs.add(context.registerMessageParser(BGPNotificationMessageParser.TYPE, nmp));
+ regs.add(context.registerMessageSerializer(Notify.class, nmp));
+
+ final BGPKeepAliveMessageParser kamp = new BGPKeepAliveMessageParser();
+ regs.add(context.registerMessageParser(BGPKeepAliveMessageParser.TYPE, kamp));
+ regs.add(context.registerMessageSerializer(Keepalive.class, kamp));
+ }