CsStaticRoutes.py

#!/usr/bin/python
# -- coding: utf-8 --
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

import logging
from . import CsHelper
from .CsDatabag import CsDataBag
from .CsRoute import CsRoute


class CsStaticRoutes(CsDataBag):

    def process(self):
        logging.debug("Processing CsStaticRoutes file ==> %s" % self.dbag)
        for item in self.dbag:
            if item == "id":
                continue
            self.__update(self.dbag[item])



    def __update(self, route):
        network = route['network']
        gateway = route['gateway']

        if route['revoke']:
            # Delete from main table
            command = "ip route del %s via %s" % (network, gateway)
            CsHelper.execute(command)

            # Delete from PBR table if applicable
            device = CsHelper.find_device_for_gateway(self.config, gateway)
            if device:
                cs_route = CsRoute()
                table_name = cs_route.get_tablename(device)
                command = "ip route del %s via %s table %s" % (network, gateway, table_name)
                CsHelper.execute(command)
                logging.info("Deleted static route %s via %s from PBR table %s" % (network, gateway, table_name))
        else:
            # Add to main table (existing logic)
            command = "ip route show | grep %s | awk '{print $1, $3}'" % network
            result = CsHelper.execute(command)
            if not result:
                route_command = "ip route add %s via %s" % (network, gateway)
                CsHelper.execute(route_command)
                logging.info("Added static route %s via %s to main table" % (network, gateway))

            # Add to PBR table if applicable
            device = CsHelper.find_device_for_gateway(self.config, gateway)
            if device:
                cs_route = CsRoute()
                table_name = cs_route.get_tablename(device)
                # Check if route already exists in the PBR table
                check_command = "ip route show table %s | grep %s | awk '{print $1, $3}'" % (table_name, network)
                result = CsHelper.execute(check_command)
                if not result:
                    # Add route to the interface-specific table
                    route_command = "ip route add %s via %s dev %s table %s" % (network, gateway, device, table_name)
                    CsHelper.execute(route_command)
                    logging.info("Added static route %s via %s to PBR table %s" % (network, gateway, table_name))
            else:
                logging.info("Static route %s via %s added to main table only (no matching interface found for PBR table)" % (network, gateway))