From d91c0b6c6a607e2c318818c1aeb1ac756bd0492d Mon Sep 17 00:00:00 2001
From: Rob Austein <sra@hactrn.net>
Date: Wed, 16 Jul 2014 16:30:57 +0000
Subject: Snapshot before reworking pubd SQL.

svn path=/branches/tk705/; revision=5897
---
 rpki/pubd.py | 65 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 63 insertions(+), 2 deletions(-)

(limited to 'rpki/pubd.py')

diff --git a/rpki/pubd.py b/rpki/pubd.py
index 0ee4d38c..8f5b2605 100644
--- a/rpki/pubd.py
+++ b/rpki/pubd.py
@@ -110,6 +110,7 @@ class main(object):
     self.publication_multimodule = self.cfg.getboolean("publication-multimodule", False)
 
     self.rrdp_expiration_interval = rpki.sundial.timedelta.parse(self.cfg.get("rrdp-expiration-interval", "6h"))
+    self.rrdp_uri_base = self.cfg.get("rrdp-uri-base")
     self.rrdp_publication_base = self.cfg.get("rrdp-publication-base", "rrdp-publication/")
 
     self.session = session_obj.fetch(self)
@@ -227,7 +228,7 @@ class session_obj(rpki.sql.sql_persistent):
   def new_snapshot(self):
     return snapshot_obj.create(self)
 
-  def add_snapshot(self, new_snapshot):
+  def activate_snapshot(self, new_snapshot):
     now = rpki.sundial.now()
     old_snapshot = self.current_snapshot
     if old_snapshot is not None:
@@ -242,6 +243,66 @@ class session_obj(rpki.sql.sql_persistent):
                                                  (self.session_id, rpki.sundial.now())):
       snapshot.sql_delete()
 
+  def write_notification(self):
+    """
+    Write current notification file to disk.
+    """
+
+    serial = self.current_shapshot.serial
+    fn = "%s/notification.xml" % self.uuid
+
+    xml = Element(rrdp_namespace + "notification",
+                  version = rrdp_version,
+                  session_id = uuid,
+                  serial = serial)
+
+    SubElement(xml, rrdp_namespace + "snapshot",
+               uri = "%s/%s/snapshot/%d.xml" % (self.rrdp_uri_base, self.uuid, serial),
+               hash = um_where_do_we_store_this)
+
+    for delta in some_sql_query_here():
+      SubElement(xml, rrdp_namespace + "delta",
+                 from = delta.from_serial,
+                 to = delta.to_serial,
+                 uri = delta.uri,
+                 hash = delta.hash)
+
+    rpki.relaxng.rrdp.assertValid(xml)
+    tn = os.path.join(self.rrdp_publication_base, fn + ".%s.tmp" % os.getpid())
+    if not os.path.isdir(os.path.dirname(tn)):
+      os.makedirs(os.path.dirname(tn))
+    ElementTree(xml).write(tn)
+    os.rename(tn, os.path.join(self.rrdp_publication_base, fn))
+
+
+  def write_snapshot(self):
+    """
+    Write current RRDP snapshot to disk.
+    """
+
+    serial = self.current_shapshot.serial
+    fn = "%s/snapshot/%d.xml" % (self.uuid, serial)
+
+    if os.path.exists(os.path.join(self.rrdp_publication_base, fn)):
+      logger.warning("Snapshot %s already exists, this is suprising, not regenerating")
+      return
+
+    xml = Element(rrdp_namespace + "snapshot", version = rrdp_version, session_id = uuid, serial = serial)
+
+    for obj in object_obj.sql_fetch_where(self.gctx, "session_id = %s AND withdrawn_snapshot_id IS NULL",
+                                          (self.session_id,)):
+      se = SubElement(xml, rrdp_namespace + "publish", uri = obj.uri)
+      se.text = "\n" + obj.get_Base64()
+      se.tail = "\n"
+
+    rpki.relaxng.rrdp.assertValid(xml)
+
+    tn = os.path.join(self.rrdp_publication_base, fn + ".%s.tmp" % os.getpid())
+    if not os.path.isdir(os.path.dirname(tn)):
+      os.makedirs(os.path.dirname(tn))
+    ElementTree(xml).write(tn)
+    os.rename(tn, os.path.join(self.rrdp_publication_base, fn))
+
 
 class snapshot_obj(rpki.sql.sql_persistent):
   """
@@ -341,7 +402,7 @@ class object_obj(rpki.sql.sql_persistent):
     self.gctx = snapshot.gctx
     self.uri = uri
     self.payload = obj
-    self.hash = rpki.x509.sha256(obj.get_Base64()).encode("hex")
+    self.hash = rpki.x509.sha256(obj.get_DER()).encode("hex")
     logger.debug("Computed hash %s of %r", self.hash, obj)
     self.published_snapshot_id = snapshot.snapshot_id
     self.withdrawn_snapshot_id = None
-- 
cgit v1.2.3