diff options
| -rw-r--r-- | config/bgp_configs.go | 164 | ||||
| -rw-r--r-- | tools/pyang_plugins/README.rst | 3 | ||||
| -rw-r--r-- | tools/pyang_plugins/bgpyang2golang.py | 27 |
3 files changed, 193 insertions, 1 deletions
diff --git a/config/bgp_configs.go b/config/bgp_configs.go index 2178344b..9d6c27fc 100644 --- a/config/bgp_configs.go +++ b/config/bgp_configs.go @@ -89,6 +89,86 @@ const ( DEFAULT_POLICY_TYPE_REJECT_ROUTE ) +// typedef for typedef bgp-pol:bgp-set-community-option-type +type BgpSetCommunityOptionType int + +const ( + BGP_SET_COMMUNITY_OPTION_TYPE_ADD = iota + BGP_SET_COMMUNITY_OPTION_TYPE_REMOVE + BGP_SET_COMMUNITY_OPTION_TYPE_REPLACE + BGP_SET_COMMUNITY_OPTION_TYPE_NULL +) + +// typedef for typedef bgp-pol:bgp-community-regexp-type +type BgpCommunityRegexpType string + +// typedef for typedef bgp-pol:bgp-set-med-type +type BgpSetMedType string + +// typedef for typedef bgp-pol:bgp-ext-community-type +type BgpExtCommunityType string + +// typedef for typedef bgp-pol:bgp-as-path-prepend-repeat +type BgpAsPathPrependRepeat uint8 + +// typedef for typedef bgp-pol:bgp-std-community-type +type BgpStdCommunityType string + +// typedef for typedef bgp-pol:bgp-next-hop-type +type BgpNextHopType string + +// typedef for typedef bgp-pol:bgp-well-known-community-type +type BgpWellKnownCommunityType int + +const ( + BGP_WELL_KNOWN_COMMUNITY_TYPE_INTERNET = iota + BGP_WELL_KNOWN_COMMUNITY_TYPE_NO_EXPORT + BGP_WELL_KNOWN_COMMUNITY_TYPE_NO_ADVERTISE + BGP_WELL_KNOWN_COMMUNITY_TYPE_NO_EXPORT_SUBCONFED +) + +//struct for container bgp-pol:set-ext-community +type SetExtCommunity struct { + // original -> bgp-pol:communities + //original type is list of union + Communities []string + // original -> bgp-pol:options + Options BgpSetCommunityOptionType +} + +//struct for container bgp-pol:set-community +type SetCommunity struct { + // original -> bgp-pol:communities + //original type is list of union + Communities []string + // original -> bgp-pol:options + Options BgpSetCommunityOptionType +} + +//struct for container bgp-pol:set-as-path-prepend +type SetAsPathPrepend struct { + // original -> bgp-pol:repeat-n + RepeatN uint8 +} + +//struct for container bgp-pol:bgp-actions +type BgpActions struct { + // original -> bgp-pol:set-as-path-prepend + SetAsPathPrepend SetAsPathPrepend + // original -> bgp-pol:set-community + SetCommunity SetCommunity + // original -> bgp-pol:set-ext-community + SetExtCommunity SetExtCommunity + // original -> bgp-pol:set-route-origin + SetRouteOrigin BgpOriginAttrType + // original -> bgp-pol:set-local-pref + SetLocalPref uint32 + // original -> bgp-pol:set-next-hop + SetNextHop BgpNextHopType + // original -> bgp-pol:set-med + SetMed BgpSetMedType +} + //struct for container rpol:igp-actions type IgpActions struct { // original -> rpol:set-tag @@ -105,6 +185,50 @@ type Actions struct { RejectRoute bool // original -> rpol:igp-actions IgpActions IgpActions + // original -> bgp-pol:bgp-actions + BgpActions BgpActions +} + +//struct for container bgp-pol:as-path-length +type AsPathLength struct { + // original -> ptypes:operator + Operator string + // original -> ptypes:value + Value uint32 +} + +//struct for container bgp-pol:community-count +type CommunityCount struct { + // original -> ptypes:operator + Operator string + // original -> ptypes:value + Value uint32 +} + +//struct for container bgp-pol:bgp-conditions +type BgpConditions struct { + // original -> bgp-pol:match-community-set + MatchCommunitySet string + // original -> bgp-pol:match-ext-community-set + MatchExtCommunitySet string + // original -> bgp-pol:match-as-path-set + MatchAsPathSet string + // original -> bgp-pol:med-eq + MedEq uint32 + // original -> bgp-pol:origin-eq + OriginEq BgpOriginAttrType + // original -> bgp-pol:next-hop-in + //original type is list of inet:ip-address + NextHopIn []net.IP + // original -> bgp-pol:local-pref-eq + LocalPrefEq uint32 + // original -> bgp-pol:community-count + CommunityCount CommunityCount + // original -> bgp-pol:as-path-length + AsPathLength AsPathLength + // original -> bgp-pol:route-type + //bgp-pol:route-type's original type is enumeration + RouteType string } //struct for container rpol:igp-conditions @@ -127,6 +251,8 @@ type Conditions struct { InstallProtocolEq InstallProtocolType // original -> rpol:igp-conditions IgpConditions IgpConditions + // original -> bgp-pol:bgp-conditions + BgpConditions BgpConditions } //struct for container rpol:statement @@ -147,6 +273,42 @@ type PolicyDefinition struct { StatementList []Statement } +//struct for container bgp-pol:as-path-set +type AsPathSet struct { + // original -> bgp-pol:as-path-set-name + AsPathSetName string + // original -> bgp-pol:as-path-set-members + AsPathSetMembers []string +} + +//struct for container bgp-pol:ext-community-set +type ExtCommunitySet struct { + // original -> bgp-pol:ext-community-set-name + ExtCommunitySetName string + // original -> bgp-pol:ext-community-members + //original type is list of union + ExtCommunityMembers []string +} + +//struct for container bgp-pol:community-set +type CommunitySet struct { + // original -> bgp-pol:community-set-name + CommunitySetName string + // original -> bgp-pol:community-members + //original type is list of union + CommunityMembers []string +} + +//struct for container bgp-pol:bgp-defined-sets +type BgpDefinedSets struct { + // original -> bgp-pol:community-set + CommunitySetList []CommunitySet + // original -> bgp-pol:ext-community-set + ExtCommunitySetList []ExtCommunitySet + // original -> bgp-pol:as-path-set + AsPathSetList []AsPathSet +} + //struct for container rpol:neighbor-info type NeighborInfo struct { // original -> rpol:address @@ -187,6 +349,8 @@ type DefinedSets struct { PrefixSetList []PrefixSet // original -> rpol:neighbor-set NeighborSetList []NeighborSet + // original -> bgp-pol:bgp-defined-sets + BgpDefinedSets BgpDefinedSets } //struct for container rpol:routing-policy diff --git a/tools/pyang_plugins/README.rst b/tools/pyang_plugins/README.rst index 7fcb3018..4d45904e 100644 --- a/tools/pyang_plugins/README.rst +++ b/tools/pyang_plugins/README.rst @@ -18,4 +18,5 @@ How to use $ source ./env.sh $ PYTHONPATH=. ./bin/pyang --plugindir $GOBGP_PATH/tools/pyang_plugins \ -p $YANG_DIR/bgp -p $YANG_DIR/policy \ - -f golang $YANG_DIR/bgp/bgp.yang |gofmt > $GOBGP_PATH/config/bgp_configs.go + -f golang $YANG_DIR/bgp/bgp.yang \ + --augment $YANG_DIR/bgp/bgp-policy.yang |gofmt > $GOBGP_PATH/config/bgp_configs.go diff --git a/tools/pyang_plugins/bgpyang2golang.py b/tools/pyang_plugins/bgpyang2golang.py index c61d9c85..765e2f12 100644 --- a/tools/pyang_plugins/bgpyang2golang.py +++ b/tools/pyang_plugins/bgpyang2golang.py @@ -14,6 +14,7 @@ # limitations under the License. +import optparse import StringIO import sys from pyang import plugin @@ -46,6 +47,19 @@ class GolangPlugin(plugin.PyangPlugin): def add_output_format(self, fmts): fmts['golang'] = self + + + def add_opts(self, optparser): + optlist = [ + optparse.make_option("--augment", + dest="augment", + help="Yang file which has augment statements"), + ] + g = optparser.add_option_group("GolangPlugin specific options") + g.add_options(optlist) + + + def emit(self, ctx, modules, fd): ctx.golang_identity_map = {} @@ -57,6 +71,19 @@ class GolangPlugin(plugin.PyangPlugin): ctx.module_deps = [] check_module_deps(ctx, modules[0]) + + # load augment module + if ctx.opts.augment: + aug_mod_path = ctx.opts.augment + try: + fd = open(aug_mod_path) + text = fd.read() + except IOError as ex: + sys.stderr.write("error %s: %s\n" % (aug_mod_path, str(ex))) + sys.exit(1) + aug_mod = ctx.add_module(aug_mod_path, text) + check_module_deps(ctx, aug_mod) + # visit yang statements visit_modules(ctx) # emit bgp_configs |