Fix LeafNode conversions
[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.common.Uint32;
17 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
18 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerNode;
19 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
20 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNodes;
21
22 public final class AsNumberUtil {
23
24     private AsNumberUtil() {
25         throw new UnsupportedOperationException();
26     }
27
28     /**
29      * Looks for As4Byte Capability in capabilities and extracts AS number.
30      *
31      * @param open remote BGP open message
32      * @return AsNumber
33      */
34     public static @NonNull AsNumber advertizedAsNumber(final Open open) {
35         // Look for AS4 capability very defensively
36         final List<BgpParameters> params = open.getBgpParameters();
37         if (params != null) {
38             for (final BgpParameters p : params) {
39                 for (final OptionalCapabilities oc : p.getOptionalCapabilities()) {
40                     if (oc.getCParameters() != null && oc.getCParameters().getAs4BytesCapability() != null) {
41                         return oc.getCParameters().getAs4BytesCapability().getAsNumber();
42                     }
43                 }
44             }
45         }
46         // Fallback to whatever is in the header
47         return new AsNumber(open.getMyAsNumber().longValue());
48     }
49
50     /**
51      * Extract AS from Container Node.
52      *
53      * @param dtc route container
54      * @param nid as node identifier
55      * @return as number
56      */
57     public static AsNumber extractAS(final DataContainerNode<?> dtc, final NodeIdentifier nid) {
58         final NormalizedNode<?, ?> as = NormalizedNodes.findNode(dtc, nid).orElse(null);
59         if (as != null) {
60             return new AsNumber((Uint32) as.getValue());
61         }
62         return null;
63     }
64 }