Bug 3117 - Capability Parameter Parse Issue
[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.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.AsNumber;
12 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Open;
13 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.BgpParameters;
14 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.bgp.parameters.OptionalCapabilities;
15
16 public final class AsNumberUtil {
17
18     private AsNumberUtil() {
19         throw new UnsupportedOperationException();
20     }
21
22     /**
23      * Looks for As4Byte Capability in capabilities and extracts AS number.
24      *
25      * @param open remote BGP open message
26      * @return AsNumber
27      */
28     public static AsNumber advertizedAsNumber(final Open open) {
29         // Look for AS4 capability very defensively
30         final List<BgpParameters> params = open.getBgpParameters();
31         if (params != null) {
32             for (final BgpParameters p : params) {
33                 for (final OptionalCapabilities oc : p.getOptionalCapabilities()) {
34                     if (oc.getCParameters() != null && oc.getCParameters().getAs4BytesCapability() != null) {
35                         return oc.getCParameters().getAs4BytesCapability().getAsNumber();
36                     }
37                 }
38             }
39         }
40         // Fallback to whatever is in the header
41         return new AsNumber(open.getMyAsNumber().longValue());
42     }
43 }