util: use safe variant or xml_marshaller

Also change places where xml_marshaller were used directly to always use
the wrapper from utils (except in tests for simplicity)

master/product/Vifib/Tool/SlapTool.py

@@ -49,7 +49,7 @@ try:
     ComputerPartition as SlapComputerPartition,
     SoftwareInstance,
     SoftwareRelease)
-  from slapos.util import dict2xml, xml2dict, calculate_dict_hash
+  from slapos.util import dict2xml, xml2dict, calculate_dict_hash, loads, dumps
 except ImportError:
   # Do no prevent instance from starting
   # if libs are not installed
@@ -71,9 +71,12 @@ except ImportError:
     raise ImportError
   def calculate_dict_hash(dictionary):
     raise ImportError
+  def loads(*args):
+    raise ImportError
+  def dumps(*args):
+    raise ImportError
 
 from zLOG import LOG, INFO
-import xml_marshaller
 import StringIO
 import pkg_resources
 from Products.Vifib.Conduit import VifibConduit
@@ -185,7 +188,7 @@ class SlapTool(BaseTool):
                     portal_type="Computer Partition"):
       slap_computer._computer_partition_list.append(
           self._getSlapPartitionByPackingList(_assertACI(computer_partition.getObject())))
-    return xml_marshaller.xml_marshaller.dumps(slap_computer)
+    return dumps(slap_computer)
 
   def _fillComputerInformationCache(self, computer_id, user):
     key = '%s_%s' % (computer_id, user)
@@ -278,7 +281,7 @@ class SlapTool(BaseTool):
     for computer_partition in computer_partition_list:
       slap_computer._computer_partition_list.append(
           self._getSlapPartitionByPackingList(_assertACI(computer_partition.getObject())))
-    return xml_marshaller.xml_marshaller.dumps(slap_computer)
+    return dumps(slap_computer)
 
   @UnrestrictedMethod
   def _getHostingSubscriptionIpList(self, computer_id, computer_partition_id):
@@ -287,7 +290,7 @@ class SlapTool(BaseTool):
     
     if software_instance is None or \
                         software_instance.getSlapState() == 'destroy_requested':
-        return xml_marshaller.xml_marshaller.dumps([])
+      return dumps([])
     # Search hosting subscription
     hosting = software_instance.getSpecialiseValue()
     while hosting and hosting.getPortalType() != "Hosting Subscription":
@@ -305,7 +308,7 @@ class SlapTool(BaseTool):
               internet_protocol_address.getIpAddress().decode("UTF-8"))
         )
     
-    return xml_marshaller.xml_marshaller.dumps(ip_address_list)
+    return dumps(ip_address_list)
 
   security.declareProtected(Permissions.AccessContentsInformation,
     'getFullComputerInformation')
@@ -366,7 +369,7 @@ class SlapTool(BaseTool):
       key=software_instance.getSslKey(),
       certificate=software_instance.getSslCertificate()
     )
-    result = xml_marshaller.xml_marshaller.dumps(certificate_dict)
+    result = dumps(certificate_dict)
     # Cache with revalidation
     self.REQUEST.response.setStatus(200)
     self.REQUEST.response.setHeader('Cache-Control',
@@ -458,7 +461,7 @@ class SlapTool(BaseTool):
       reference=software_product_reference,
       validation_state='published')
     if len(software_product_list) is 0:
-      return xml_marshaller.xml_marshaller.dumps([])
+      return dumps([])
     if len(software_product_list) > 1:
       raise NotImplementedError('Several Software Product with the same title.')
     software_release_list = \
@@ -477,7 +480,7 @@ class SlapTool(BaseTool):
         key=sortkey,
         reverse=True,
     )
-    return xml_marshaller.xml_marshaller.dumps(
+    return dumps(
       [software_release.getUrlString()
         for software_release in software_release_list
           if software_release.getValidationState() in \
@@ -555,7 +558,7 @@ class SlapTool(BaseTool):
     person = portal.portal_membership.getAuthenticatedMember().getUserValue()
     person.requestComputer(computer_title=computer_title)
     computer = Computer(self.REQUEST.get('computer_reference').decode("UTF-8"))
-    return xml_marshaller.xml_marshaller.dumps(computer)
+    return dumps(computer)
 
   security.declareProtected(Permissions.AccessContentsInformation,
     'requestComputer')
@@ -733,7 +736,7 @@ class SlapTool(BaseTool):
     'loadComputerConfigurationFromXML')
   def loadComputerConfigurationFromXML(self, xml):
     "Load the given xml as configuration for the computer object"
-    computer_dict = xml_marshaller.xml_marshaller.loads(xml)
+    computer_dict = loads(xml)
     computer = self._getComputerDocument(computer_dict['reference'])
     computer.Computer_updateFromDict(computer_dict)
     return 'Content properly posted.'
@@ -759,7 +762,7 @@ class SlapTool(BaseTool):
      'certificate': self.REQUEST.get('computer_certificate').decode("UTF-8"),
      'key': self.REQUEST.get('computer_key').decode("UTF-8")
      }
-    return xml_marshaller.xml_marshaller.dumps(result)
+    return dumps(result)
 
   security.declareProtected(Permissions.AccessContentsInformation,
     'generateComputerCertificate')
@@ -851,7 +854,7 @@ class SlapTool(BaseTool):
             slave_instance_dict.pop("xml")))
       slap_partition._parameter_dict.update(parameter_dict)
 
-    result = xml_marshaller.xml_marshaller.dumps(slap_partition)
+    result = dumps(slap_partition)
 
     # Keep in cache server for 7 days
     self.REQUEST.response.setStatus(200)
@@ -1154,7 +1157,7 @@ class SlapTool(BaseTool):
                                     'REMOTE_USER')
     self.REQUEST.response.setHeader('Last-Modified', last_modified)
     self.REQUEST.response.setHeader('Content-Type', 'text/xml; charset=utf-8')
-    self.REQUEST.response.setBody(xml_marshaller.xml_marshaller.dumps(d))
+    self.REQUEST.response.setBody(dumps(d))
     return self.REQUEST.response
 
   @convertToREST
@@ -1227,8 +1230,7 @@ class SlapTool(BaseTool):
         computer_id,
         computer_partition_id,
         slave_reference)
-    connection_xml = dict2xml(xml_marshaller.xml_marshaller.loads(
-                                              connection_xml))
+    connection_xml = dict2xml(loads(connection_xml))
     reference = software_instance.getReference()
     if self._getLastData(reference) != connection_xml:
       software_instance.updateConnection(
@@ -1253,20 +1255,19 @@ class SlapTool(BaseTool):
     In any other case returns not important data and HTTP code is 403 Forbidden
     """
     if state:
-      state = xml_marshaller.xml_marshaller.loads(state)
+      state = loads(state)
     if state is None:
       state = 'started'
     if shared_xml is not _MARKER:
-      shared = xml_marshaller.xml_marshaller.loads(shared_xml)
+      shared = loads(shared_xml)
     else:
       shared = False
     if partition_parameter_xml:
-      partition_parameter_kw = xml_marshaller.xml_marshaller.loads(
-                                              partition_parameter_xml)
+      partition_parameter_kw = loads(partition_parameter_xml)
     else:
       partition_parameter_kw = dict()
     if filter_xml:
-      filter_kw = xml_marshaller.xml_marshaller.loads(filter_xml)
+      filter_kw = loads(filter_xml)
       if software_type == 'pull-backup' and not 'retention_delay' in filter_kw:
         filter_kw['retention_delay'] = 7.0
     else:
@@ -1372,7 +1373,7 @@ class SlapTool(BaseTool):
         software_instance._filter_dict = filter_xml
         software_instance._requested_state = state
         software_instance._instance_guid = instance_guid
-        return xml_marshaller.xml_marshaller.dumps(software_instance)
+        return dumps(software_instance)
 
   @UnrestrictedMethod
   def _updateComputerPartitionRelatedInstanceList(self, computer_id,
@@ -1392,8 +1393,7 @@ class SlapTool(BaseTool):
 
     cache_reference = '%s-PREDLIST' % software_instance_document.getReference()
     if self._getLastData(cache_reference) != instance_reference_xml:
-      instance_reference_list = xml_marshaller.xml_marshaller.loads(
-                                                        instance_reference_xml)
+      instance_reference_list = loads(instance_reference_xml)
 
       current_predecessor_list = software_instance_document.getPredecessorValueList(
                             portal_type=['Software Instance', 'Slave Instance'])

slapos/slap/standalone.py

@@ -61,6 +61,8 @@ from .interface.slap import IRequester
 from ..grid.slapgrid import SLAPGRID_PROMISE_FAIL
 
 from .slap import slap
+from ..util import dumps
+
 from ..grid.svcbackend import getSupervisorRPC
 
 
@@ -534,7 +536,7 @@ class StandaloneSlapOS(object):
                   **locals()))
 
     self.computer.updateConfiguration(
-        xml_marshaller.xml_marshaller.dumps({
+        dumps({
             'address': ipv4_address,
             'netmask': '255.255.255.255',
             'partition_list': partition_list,

slapos/tests/test_util.py

@@ -217,6 +217,20 @@ class TestUtil(unittest.TestCase):
       slapos.util.dict2xml(self.xml2dict1_dict)
     )
 
+  def test_dumps_loads(self):
+    simple_object = {"ok": [True]}
+    self.assertEqual(simple_object, slapos.util.loads(slapos.util.dumps(simple_object)))
+
+    self.assertRaises(
+        Exception,
+        slapos.util.loads,
+            b'<marshal><object id="i2" module="nasty" class="klass">'
+            b'<tuple></tuple><dictionary id="i3"/></object></marshal>')
+
+    class Nasty(object):
+      pass
+    self.assertRaises(Exception, slapos.util.dumps, Nasty())
+
 
 if __name__ == '__main__':
   unittest.main()

slapos/util.py

@@ -33,7 +33,7 @@ import socket
 import struct
 import subprocess
 import sqlite3
-from xml_marshaller.xml_marshaller import dumps, loads
+from xml_marshaller.xml_marshaller import Marshaller, Unmarshaller
 from lxml import etree
 import six
 from six.moves.urllib import parse
@@ -49,6 +49,36 @@ except NameError:  # make pylint happy on python2...
 
 
 
+_ALLOWED_CLASS_SET = frozenset((
+    ('slapos.slap.slap', 'Computer'),
+    ('slapos.slap.slap', 'ComputerPartition'),
+    ('slapos.slap.slap', 'SoftwareRelease'),
+    ('slapos.slap.slap', 'SoftwareInstance'),
+))
+
+
+class SafeXMLMarshaller(Marshaller):
+  def m_instance(self, value, kw):
+    cls = value.__class__
+    if (cls.__module__, cls.__name__) in _ALLOWED_CLASS_SET:
+      return super(SafeXMLMarshaller, self).m_instance(value, kw)
+    raise RuntimeError("Refusing to marshall {}.{}".format(
+        cls.__module__, cls.__name__))
+
+
+dumps = SafeXMLMarshaller().dumps
+
+
+class SafeXMLUnmrshaller(Unmarshaller, object):
+  def find_class(self, module, name):
+    if (module, name) in _ALLOWED_CLASS_SET:
+      return super(SafeXMLUnmrshaller, self).find_class(module, name)
+    raise RuntimeError("Refusing to unmarshall {}.{}".format(module, name))
+
+
+loads = SafeXMLUnmrshaller().loads
+
+
 def mkdir_p(path, mode=0o700):
     """\
     Creates a directory and its parents, if needed.