bgpd: Show RPKI prefixes filtered by ASN

```
spine1-debian-9# show rpki
  as-number         Lookup by ASN in prefix table
  cache-connection  Show to which RPKI Cache Servers we have a connection
  cache-server      SHOW configured cache server
  prefix            Lookup IP prefix and optionally ASN in prefix table
  prefix-table      Show validated prefixes which were received from RPKI Cache
spine1-debian-9# show rpki as-number 47583
2.57.88.0                                   23 -  23        47583
2.57.91.0                                   24 -  24        47583
2.57.90.0                                   24 -  24        47583
5.183.8.0                                   22 -  24        47583
31.170.164.0                                23 -  23        47583
31.170.165.0                                24 -  24        47583
31.170.164.0                                24 -  24        47583
31.170.166.0                                23 -  23        47583
31.170.160.0                                22 -  22        47583
31.220.16.0                                 24 -  24        47583
...
```

Signed-off-by: Donatas Abraitis <donatas.abraitis@gmail.com>

diff --git a/bgpd/bgp_rpki.c b/bgpd/bgp_rpki.c
index 2ca0c7b96d84050312a5f7d24fa5ca3d28cddf2e..ee1c49666b9d4794ff5dd8b34f02417fd2a4e0a2 100644 (file)
--- a/bgpd/bgp_rpki.c
+++ b/bgpd/bgp_rpki.c
@@ -94,6 +94,7 @@ enum return_values { SUCCESS = 0, ERROR = -1 };
 struct rpki_for_each_record_arg {
        struct vty *vty;
        unsigned int *prefix_amount;
+       as_t as;
 };
 
 static int start(void);
@@ -273,6 +274,17 @@ static void print_record(const struct pfx_record *record, struct vty *vty)
                record->max_len, record->asn);
 }
 
+static void print_record_by_asn(const struct pfx_record *record, void *data)
+{
+       struct rpki_for_each_record_arg *arg = data;
+       struct vty *vty = arg->vty;
+
+       if (record->asn == arg->as) {
+               (*arg->prefix_amount)++;
+               print_record(record, vty);
+       }
+}
+
 static void print_record_cb(const struct pfx_record *record, void *data)
 {
        struct rpki_for_each_record_arg *arg = data;
@@ -621,6 +633,36 @@ static struct rtr_mgr_group *get_connected_group(void)
        return rtr_mgr_get_first_group(rtr_config);
 }
 
+static void print_prefix_table_by_asn(struct vty *vty, as_t as)
+{
+       unsigned int number_of_ipv4_prefixes = 0;
+       unsigned int number_of_ipv6_prefixes = 0;
+       struct rtr_mgr_group *group = get_connected_group();
+       struct rpki_for_each_record_arg arg;
+
+       arg.vty = vty;
+       arg.as = as;
+
+       if (!group) {
+               vty_out(vty, "Cannot find a connected group.\n");
+               return;
+       }
+
+       struct pfx_table *pfx_table = group->sockets[0]->pfx_table;
+
+       vty_out(vty, "RPKI/RTR prefix table\n");
+       vty_out(vty, "%-40s %s  %s\n", "Prefix", "Prefix Length", "Origin-AS");
+
+       arg.prefix_amount = &number_of_ipv4_prefixes;
+       pfx_table_for_each_ipv4_record(pfx_table, print_record_by_asn, &arg);
+
+       arg.prefix_amount = &number_of_ipv6_prefixes;
+       pfx_table_for_each_ipv6_record(pfx_table, print_record_by_asn, &arg);
+
+       vty_out(vty, "Number of IPv4 Prefixes: %u\n", number_of_ipv4_prefixes);
+       vty_out(vty, "Number of IPv6 Prefixes: %u\n", number_of_ipv6_prefixes);
+}
+
 static void print_prefix_table(struct vty *vty)
 {
        struct rpki_for_each_record_arg arg;
@@ -1190,6 +1232,21 @@ DEFUN (show_rpki_prefix_table,
        return CMD_SUCCESS;
 }
 
+DEFPY(show_rpki_as_number, show_rpki_as_number_cmd,
+      "show rpki as-number (1-4294967295)$by_asn",
+      SHOW_STR RPKI_OUTPUT_STRING
+      "Lookup by ASN in prefix table\n"
+      "AS Number\n")
+{
+       if (!is_synchronized()) {
+               vty_out(vty, "No Connection to RPKI cache server.\n");
+               return CMD_WARNING;
+       }
+
+       print_prefix_table_by_asn(vty, by_asn);
+       return CMD_SUCCESS;
+}
+
 DEFPY (show_rpki_prefix,
        show_rpki_prefix_cmd,
        "show rpki prefix <A.B.C.D/M|X:X::X:X/M> [(1-4294967295)$asn]",
@@ -1523,6 +1580,7 @@ static void install_cli_commands(void)
        install_element(VIEW_NODE, &show_rpki_cache_connection_cmd);
        install_element(VIEW_NODE, &show_rpki_cache_server_cmd);
        install_element(VIEW_NODE, &show_rpki_prefix_cmd);
+       install_element(VIEW_NODE, &show_rpki_as_number_cmd);
 
        /* Install debug commands */
        install_element(CONFIG_NODE, &debug_rpki_cmd);