SetRibPolicyExample.cpp

/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#include <gflags/gflags.h>

#include <folly/io/async/EventBase.h>
#include <openr/common/NetworkUtil.h>
#include <openr/common/OpenrClient.h>
#include <openr/if/gen-cpp2/OpenrCtrlCppAsyncClient.h>
#include <openr/if/gen-cpp2/OpenrCtrl_types.h>

DEFINE_string(host, "::1", "Host to talk to");

DEFINE_int32(ttl_secs, 300, "Number of seconds the policy is alive for");
DEFINE_int32(default_weight, 1, "Weight for nexthops with no area");
DEFINE_int32(area0_weight, 2, "Area0 (default area in Open/R) weight");
DEFINE_string(
    neighbor_weight,
    "",
    "Comma separated list of neighborName:weight (ex: fsw001.p001.f01.atn6:10)");
DEFINE_string(prefixes, "", "Comma separated list of prefixes to apply policy");
DEFINE_string(tags, "", "Comma separated list of tags to apply policy");

using namespace openr;

int
main(int argc, char* argv[]) {
  gflags::ParseCommandLineFlags(&argc, &argv, true);
  google::InitGoogleLogging(argv[0]);
  google::InstallFailureSignalHandler();

  // Create list of prefixes for match
  std::vector<std::string> prefixStrs;
  folly::split(",", FLAGS_prefixes, prefixStrs, true);
  std::vector<thrift::IpPrefix> prefixes;
  for (auto const& prefixStr : prefixStrs) {
    LOG(INFO) << "Prefix - " << prefixStr;
    prefixes.emplace_back(toIpPrefix(prefixStr));
  }

  // Create list of tags for match
  std::vector<std::string> tags;
  folly::split(",", FLAGS_tags, tags, true);

  // Action weight
  thrift::RibRouteActionWeight actionWeight;
  actionWeight.default_weight() = FLAGS_default_weight;
  actionWeight.area_to_weight()->emplace(
      Constants::kDefaultArea.toString(), FLAGS_area0_weight);
  // Parse neighbor->weight map and insert into the policy statement
  std::vector<std::string> neighborWeights;
  folly::split(",", FLAGS_neighbor_weight, neighborWeights, true);
  for (auto const& neighborWeight : neighborWeights) {
    std::vector<std::string> neighborWeightSplit;
    folly::split(":", neighborWeight, neighborWeightSplit, true);
    CHECK_EQ(2, neighborWeightSplit.size());
    int32_t weight = 0;
    try {
      weight = folly::to<int32_t>(neighborWeightSplit.at(1));
    } catch (std::exception& e) {
      LOG(ERROR) << "Failed to convert string to weight int32, "
                 << folly::exceptionStr(e);
      return -1;
    }
    LOG(INFO) << "Neighbor: " << neighborWeightSplit.at(0)
              << " -> weight: " << weight;
    actionWeight.neighbor_to_weight()->emplace(
        neighborWeightSplit.at(0), weight);
  }

  // Create PolicyStatement
  thrift::RibPolicyStatement policyStatement;
  policyStatement.matcher()->prefixes() = prefixes;
  policyStatement.matcher()->tags() = tags;
  policyStatement.action()->set_weight() = actionWeight;

  // Create RibPolicy
  thrift::RibPolicy policy;
  policy.statements()->emplace_back(policyStatement);
  policy.ttl_secs() = FLAGS_ttl_secs;

  // Create OpenrClient and set policy
  LOG(INFO) << "Creating connection to host " << FLAGS_host;
  folly::EventBase evb;
  auto client =
      getOpenrCtrlPlainTextClient<openr::thrift::OpenrCtrlCppAsyncClient>(
          evb, folly::IPAddress(FLAGS_host));
  client->sync_setRibPolicy(policy);
  LOG(INFO) << "Done setting policy";

  // Done setting policy (any exception will be thrown or printed)
  return 0;
}