6d22f2d3e32ff3b746030728316e46a3858243a7
[bgpcep.git] / bgp / parser-api / src / main / java / org / opendaylight / protocol / bgp / parser / AsNumberUtil.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;
9
10 import java.util.List;
11 import org.eclipse.jdt.annotation.NonNull;
12 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
13 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.Open;
14 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.BgpParameters;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.bgp.parameters.OptionalCapabilities;
16 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
17 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerNode;
18 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
19 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNodes;
20
21 public final class AsNumberUtil {
22
23     private AsNumberUtil() {
24         throw new UnsupportedOperationException();
25     }
26
27     /**
28      * Looks for As4Byte Capability in capabilities and extracts AS number.
29      *
30      * @param open remote BGP open message
31      * @return AsNumber
32      */
33     public static @NonNull AsNumber advertizedAsNumber(final Open open) {
34         // Look for AS4 capability very defensively
35         final List<BgpParameters> params = open.getBgpParameters();
36         if (params != null) {
37             for (final BgpParameters p : params) {
38                 for (final OptionalCapabilities oc : p.getOptionalCapabilities()) {
39                     if (oc.getCParameters() != null && oc.getCParameters().getAs4BytesCapability() != null) {
40                         return oc.getCParameters().getAs4BytesCapability().getAsNumber();
41                     }
42                 }
43             }
44         }
45         // Fallback to whatever is in the header
46         return new AsNumber(open.getMyAsNumber().longValue());
47     }
48
49     /**
50      * Extract AS from Container Node.
51      *
52      * @param dtc route container
53      * @param nid as node identifier
54      * @return as number
55      */
56     public static AsNumber extractAS(final DataContainerNode<?> dtc, final NodeIdentifier nid) {
57         final NormalizedNode<?, ?> as = NormalizedNodes.findNode(dtc, nid).orElse(null);
58         if (as != null) {
59             return new AsNumber((Long) as.getValue());
60         }
61         return null;
62     }
63 }