Add profiling support to yamltest; fix profiling support in daemons.

Tweak yamltest to force one ROA request per prefix regardless of how
the user expressed it; we may want to revisit this eventually, but
it's more useful this way for now.

svn path=/trunk/; revision=4609

1 files changed, 6 insertions, 1 deletions

diff --git a/rpkid/rpki/rpkid.py b/rpkid/rpki/rpkid.py
index 8c275688..b923b1a3 100644
--- a/rpkid/rpki/rpkid.py
+++ b/rpkid/rpki/rpkid.py
@@ -102,7 +102,12 @@ class main(object):
 
     if self.profile:
       import cProfile
-      cProfile.run("self.main()", self.profile)
+      prof = cProfile.Profile()
+      try:
+        prof.runcall(self.main)
+      finally:
+        prof.dump_stats(self.profile)
+        rpki.log.info("Dumped profile data to %s" % self.profile)
     else:
       self.main()