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/irdbd.py b/rpkid/rpki/irdbd.py
index 60f6a549..a0fc0ec1 100644
--- a/rpkid/rpki/irdbd.py
+++ b/rpkid/rpki/irdbd.py
@@ -174,7 +174,12 @@ class main(object):
 
     if profile:
       import cProfile
-      cProfile.run("self.main()", profile)
+      prof = cProfile.Profile()
+      try:
+        prof.runcall(self.main)
+      finally:
+        prof.dump_stats(profile)
+        rpki.log.info("Dumped profile data to %s" % profile)
     else:
       self.main()