1 files changed, 13 insertions, 13 deletions

diff --git a/rpkid/rpki/xml_utils.py b/rpkid/rpki/xml_utils.py
index 566c9a50..803429e7 100644
--- a/rpkid/rpki/xml_utils.py
+++ b/rpkid/rpki/xml_utils.py
@@ -220,30 +220,30 @@ class data_elt(base_elt):
     """Overridable hook."""
     pass
 
-  def serve_pre_save_hook(self, q_pdu, r_pdu, cb):
+  def serve_pre_save_hook(self, q_pdu, r_pdu, cb, eb):
     """Overridable hook."""
     cb()
 
-  def serve_post_save_hook(self, q_pdu, r_pdu, cb):
+  def serve_post_save_hook(self, q_pdu, r_pdu, cb, eb):
     """Overridable hook."""
     cb()
 
-  def serve_create(self, r_msg, cb):
+  def serve_create(self, r_msg, cb, eb):
     """Handle a create action."""
     r_pdu = self.make_reply()
 
     def one():
       self.sql_store()
       setattr(r_pdu, self.sql_template.index, getattr(self, self.sql_template.index))
-      self.serve_post_save_hook(self, r_pdu, two)
+      self.serve_post_save_hook(self, r_pdu, two, eb)
 
     def two():
       r_msg.append(r_pdu)
       cb()
 
-    self.serve_pre_save_hook(self, r_pdu, one)
+    self.serve_pre_save_hook(self, r_pdu, one, eb)
 
-  def serve_set(self, r_msg, cb):
+  def serve_set(self, r_msg, cb, eb):
     """Handle a set action."""
     db_pdu = self.serve_fetch_one()
     r_pdu = self.make_reply()
@@ -255,36 +255,36 @@ class data_elt(base_elt):
 
     def one():
       db_pdu.sql_store()
-      db_pdu.serve_post_save_hook(self, r_pdu, two)
+      db_pdu.serve_post_save_hook(self, r_pdu, two, eb)
 
     def two():
       r_msg.append(r_pdu)
       cb()
 
-    db_pdu.serve_pre_save_hook(self, r_pdu, one)
+    db_pdu.serve_pre_save_hook(self, r_pdu, one, eb)
 
-  def serve_get(self, r_msg, cb):
+  def serve_get(self, r_msg, cb, eb):
     """Handle a get action."""
     r_pdu = self.serve_fetch_one()
     self.make_reply(r_pdu)
     r_msg.append(r_pdu)
     cb()
 
-  def serve_list(self, r_msg, cb):
+  def serve_list(self, r_msg, cb, eb):
     """Handle a list action for non-self objects."""
     for r_pdu in self.serve_fetch_all():
       self.make_reply(r_pdu)
       r_msg.append(r_pdu)
     cb()
 
-  def serve_destroy(self, r_msg, cb):
+  def serve_destroy(self, r_msg, cb, eb):
     """Handle a destroy action."""
     db_pdu = self.serve_fetch_one()
     db_pdu.sql_delete()
     r_msg.append(self.make_reply())
     cb()
 
-  def serve_dispatch(self, r_msg, cb):
+  def serve_dispatch(self, r_msg, cb, eb):
     """Action dispatch handler."""
     dispatch = { "create"  : self.serve_create,
                  "set"     : self.serve_set,
@@ -293,7 +293,7 @@ class data_elt(base_elt):
                  "destroy" : self.serve_destroy }
     if self.action not in dispatch:
       raise rpki.exceptions.BadQuery, "Unexpected query: action %s" % self.action
-    dispatch[self.action](r_msg, cb)
+    dispatch[self.action](r_msg, cb, eb)
   
   def unimplemented_control(self, *controls):
     """Uniform handling for unimplemented control operations."""