1 files changed, 85 insertions, 0 deletions
diff --git a/rpki/rtr/main.py b/rpki/rtr/main.py
new file mode 100644
index 00000000..e5b50ef5
--- /dev/null
+++ b/rpki/rtr/main.py
@@ -0,0 +1,85 @@
+# $Id$
+#
+# Copyright (C) 2014  Dragon Research Labs ("DRL")
+# Portions copyright (C) 2009-2013  Internet Systems Consortium ("ISC")
+#
+# Permission to use, copy, modify, and distribute this software for any
+# purpose with or without fee is hereby granted, provided that the above
+# copyright notices and this permission notice appear in all copies.
+#
+# THE SOFTWARE IS PROVIDED "AS IS" AND DRL AND ISC DISCLAIM ALL
+# WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED
+# WARRANTIES OF MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL DRL OR
+# ISC BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
+# DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA
+# OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
+# TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+# PERFORMANCE OF THIS SOFTWARE.
+
+"""
+RPKI-Router protocol implementation.  See RFC 6810 et sequalia in fine
+RFC and Internet-Draft repositories near you.
+"""
+
+import os
+import time
+import logging
+import logging.handlers
+import argparse
+
+from rpki.rtr.server    import argparse_setup as argparse_setup_server
+from rpki.rtr.client    import argparse_setup as argparse_setup_client
+from rpki.rtr.generator import argparse_setup as argparse_setup_generator
+
+class Formatter(logging.Formatter):
+
+  converter = time.gmtime
+
+  def __init__(self, debug, fmt, datefmt):
+    self.debug = debug
+    super(Formatter, self).__init__(fmt, datefmt)
+
+  def format(self, record):
+    if getattr(record, "connection", None) is None:
+      record.connection = ""
+    return super(Formatter, self).format(record)
+
+  def formatException(self, ei):
+    if self.debug:
+      return super(Formatter, self).formatException(ei)
+    else:
+      return str(ei[1])
+
+def main():
+
+  os.environ["TZ"] = "UTC"
+  time.tzset()
+
+  argparser = argparse.ArgumentParser(description = __doc__)
+  argparser.add_argument("--debug", action = "store_true", help = "debugging mode")
+  argparser.add_argument("--log-level", default = logging.DEBUG,
+                         choices = ("debug", "info", "warning", "error", "critical"),
+                         type = lambda s: int(getattr(logging, s.upper())))
+  argparser.add_argument("--log-to",
+                         choices = ("syslog", "stderr"))
+  subparsers = argparser.add_subparsers(title = "Commands", metavar = "", dest = "mode")
+  argparse_setup_server(subparsers)
+  argparse_setup_client(subparsers)
+  argparse_setup_generator(subparsers)
+  args = argparser.parse_args()
+
+  fmt = "rpki-rtr/" + args.mode + "%(connection)s[%(process)d] %(message)s"
+
+  if (args.log_to or args.default_log_to) == "stderr":
+    handler = logging.StreamHandler()
+    fmt = "%(asctime)s " + fmt
+  elif os.path.exists("/dev/log"):
+    handler = logging.handlers.SysLogHandler("/dev/log")
+  else:
+    handler = logging.handlers.SysLogHandler()
+
+  handler.setFormatter(Formatter(args.debug, fmt, "%Y-%m-%dT%H:%M:%SZ"))
+  logging.root.addHandler(handler)
+  logging.root.setLevel(args.log_level)
+
+  return args.func(args)