add tool rtrclient

mroethke · mroethke · commit 09cac018c4d1 · 2016-09-27T15:27:39.000+02:00
diff --git a/README.rst b/README.rst
@@ -1,3 +1,4 @@
+=============
 rtrlib-python
 =============
 
@@ -9,17 +10,6 @@ Sphinx docs can be found in the docs directory and build with ``make html``.
 
 for usage examples see the tools directory
 
-
-Requirements
-''''''''''''
-- python 2.7 or python 3
-- cffi>=1.4.0
-- enum34
-- six
-
-and a c compiler for install.
-
-
 Features
 --------
 Features supported so far:
@@ -32,4 +22,13 @@ Features supported so far:
 Install Instructions
 --------------------
 rtrlib-python is not on pypi yet, but you can just run ``python setup.py install``.
-I is probably a good idea to use a virtualenv.
+It is probably a good idea to use a virtualenv.
+
+Requirements
+''''''''''''
+- python 2.7 or python 3
+- cffi>=1.4.0
+- enum34
+- six
+
+and a c compiler for building the module.
diff --git a/tools/cli-validator.py b/tools/cli-validator.py
@@ -22,7 +22,6 @@ def __init__(self):
         self.error = False
 
 def connection_status_collback(rtr_mgr_group, group_status, rtr_socket, data):
-    #print(group_status)
     if group_status == ManagerGroupStatus.ERROR:
         data.error = True
 
diff --git a/tools/rtrclient.py b/tools/rtrclient.py
@@ -0,0 +1,68 @@
+#!/usr/bin/env python
+# -*- coding: utf8 -*-
+
+import argparse
+import signal
+
+from rtrlib import RTRManager, register_pfx_update_callback, register_spki_update_callback
+
+def pfx_callback(pfx_record, added):
+    if added:
+        c = '+'
+    else:
+        c = '-'
+
+    print("{sign} {prefix:40} {max:3} - {min:3} {asn:10}".format(
+                    sign=c,
+                    prefix=pfx_record.prefix,
+                    max=pfx_record.max_len,
+                    min=pfx_record.min_len,
+                    asn=pfx_record.asn
+                )
+    )
+
+def spki_callback(spki_table, spki_record, added):
+    if added:
+        c = '+'
+    else:
+        c = '-'
+
+    print("{sign} {asn}".format(sign=c, asn=spki_record.asn))
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("protocol", choices=('tcp', ))
+    parser.add_argument("hostname")
+    parser.add_argument("port", type=int)
+    parser.add_argument(
+                        "-k",
+                        default=False,
+                        action="store_true",
+                        help="Print information about SPKI updates"
+                        )
+    parser.add_argument(
+                        "-p",
+                        default=False,
+                        action="store_true",
+                        help="Print information about PFX updates"
+                        )
+    args = parser.parse_args()
+
+    if args.p:
+        register_pfx_update_callback(pfx_callback)
+
+    if args.k:
+        register_spki_update_callback(spki_callback)
+
+    print("{:40}   {:3}   {:4}".format("Prefix", "Prefix Length", "ASN"))
+    with RTRManager(args.hostname, args.port) as mgr:
+        signal.pause()
+
+
+
+
+
+
+if __name__ == '__main__':
+    main()
