Improve validation of uploaded items and stored items
This reverts commit 4533f76df9.
			
			
This commit is contained in:
		@@ -818,21 +818,27 @@ class Application:
 | 
				
			|||||||
                if not write_whole_collection and len(items) != 1:
 | 
					                if not write_whole_collection and len(items) != 1:
 | 
				
			||||||
                    raise RuntimeError(
 | 
					                    raise RuntimeError(
 | 
				
			||||||
                        "Content contains %d components" % len(items))
 | 
					                        "Content contains %d components" % len(items))
 | 
				
			||||||
 | 
					                if write_whole_collection or not parent_item.get_meta("tag"):
 | 
				
			||||||
 | 
					                    content_type = environ.get("CONTENT_TYPE",
 | 
				
			||||||
 | 
					                                               "").split(";")[0]
 | 
				
			||||||
 | 
					                    tags = {value: key
 | 
				
			||||||
 | 
					                            for key, value in xmlutils.MIMETYPES.items()}
 | 
				
			||||||
 | 
					                    tag = tags.get(content_type)
 | 
				
			||||||
 | 
					                    if items and items[0].name == "VCALENDAR":
 | 
				
			||||||
 | 
					                        tag = "VCALENDAR"
 | 
				
			||||||
 | 
					                    elif items and items[0].name in ("VCARD", "VLIST"):
 | 
				
			||||||
 | 
					                        tag = "VADDRESSBOOK"
 | 
				
			||||||
 | 
					                else:
 | 
				
			||||||
 | 
					                    tag = parent_item.get_meta("tag")
 | 
				
			||||||
                for i in items:
 | 
					                for i in items:
 | 
				
			||||||
                    storage.check_and_sanitize_item(
 | 
					                    storage.check_and_sanitize_item(
 | 
				
			||||||
                        i, is_collection=write_whole_collection, uid=item.uid
 | 
					                        i, is_collection=write_whole_collection, uid=item.uid
 | 
				
			||||||
                        if not write_whole_collection and item else None)
 | 
					                        if not write_whole_collection and item else None,
 | 
				
			||||||
 | 
					                        tag=tag)
 | 
				
			||||||
            except Exception as e:
 | 
					            except Exception as e:
 | 
				
			||||||
                self.logger.warning(
 | 
					                self.logger.warning(
 | 
				
			||||||
                    "Bad PUT request on %r: %s", path, e, exc_info=True)
 | 
					                    "Bad PUT request on %r: %s", path, e, exc_info=True)
 | 
				
			||||||
                return BAD_REQUEST
 | 
					                return BAD_REQUEST
 | 
				
			||||||
            content_type = environ.get("CONTENT_TYPE", "").split(";")[0]
 | 
					 | 
				
			||||||
            tags = {value: key for key, value in xmlutils.MIMETYPES.items()}
 | 
					 | 
				
			||||||
            tag = tags.get(content_type)
 | 
					 | 
				
			||||||
            if items and items[0].name == "VCALENDAR":
 | 
					 | 
				
			||||||
                tag = "VCALENDAR"
 | 
					 | 
				
			||||||
            elif items and items[0].name == "VCARD":
 | 
					 | 
				
			||||||
                tag = "VADDRESSBOOK"
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
            if write_whole_collection:
 | 
					            if write_whole_collection:
 | 
				
			||||||
                try:
 | 
					                try:
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -116,7 +116,8 @@ def load(configuration, logger):
 | 
				
			|||||||
    return CollectionCopy
 | 
					    return CollectionCopy
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def check_and_sanitize_item(vobject_item, is_collection=False, uid=None):
 | 
					def check_and_sanitize_item(vobject_item, is_collection=False, uid=None,
 | 
				
			||||||
 | 
					                            tag=None):
 | 
				
			||||||
    """Check vobject items for common errors and add missing UIDs.
 | 
					    """Check vobject items for common errors and add missing UIDs.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    ``multiple`` indicates that the vobject_item contains unrelated components.
 | 
					    ``multiple`` indicates that the vobject_item contains unrelated components.
 | 
				
			||||||
@@ -124,7 +125,9 @@ def check_and_sanitize_item(vobject_item, is_collection=False, uid=None):
 | 
				
			|||||||
    If ``uid`` is not set, the UID is generated randomly.
 | 
					    If ``uid`` is not set, the UID is generated randomly.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    """
 | 
					    """
 | 
				
			||||||
    if vobject_item.name == "VCALENDAR":
 | 
					    if tag and tag not in ("VCALENDAR", "VADDRESSBOOK"):
 | 
				
			||||||
 | 
					        raise ValueError("Unsupported collection tag: %r" % tag)
 | 
				
			||||||
 | 
					    if vobject_item.name == "VCALENDAR" and tag == "VCALENDAR":
 | 
				
			||||||
        component_name = None
 | 
					        component_name = None
 | 
				
			||||||
        object_uid = None
 | 
					        object_uid = None
 | 
				
			||||||
        object_uid_set = False
 | 
					        object_uid_set = False
 | 
				
			||||||
@@ -160,18 +163,19 @@ def check_and_sanitize_item(vobject_item, is_collection=False, uid=None):
 | 
				
			|||||||
            except Exception as e:
 | 
					            except Exception as e:
 | 
				
			||||||
                raise ValueError("invalid recurrence rules in %s" %
 | 
					                raise ValueError("invalid recurrence rules in %s" %
 | 
				
			||||||
                                 component.name) from e
 | 
					                                 component.name) from e
 | 
				
			||||||
    elif vobject_item.name == "VCARD":
 | 
					    elif vobject_item.name == "VCARD" and tag == "VADDRESSBOOK":
 | 
				
			||||||
        # https://tools.ietf.org/html/rfc6352#section-5.1
 | 
					        # https://tools.ietf.org/html/rfc6352#section-5.1
 | 
				
			||||||
        object_uid = get_uid(vobject_item)
 | 
					        object_uid = get_uid(vobject_item)
 | 
				
			||||||
        if object_uid is None:
 | 
					        if object_uid is None:
 | 
				
			||||||
            vobject_item.add("UID").value = uid or random_uuid4()
 | 
					            vobject_item.add("UID").value = uid or random_uuid4()
 | 
				
			||||||
        elif not object_uid:
 | 
					        elif not object_uid:
 | 
				
			||||||
            vobject_item.uid.value = uid or random_uuid4()
 | 
					            vobject_item.uid.value = uid or random_uuid4()
 | 
				
			||||||
    elif vobject_item.name == "VLIST":
 | 
					    elif vobject_item.name == "VLIST" and tag == "VADDRESSBOOK":
 | 
				
			||||||
        # Custom format used by SOGo Connector to store lists of contacts
 | 
					        # Custom format used by SOGo Connector to store lists of contacts
 | 
				
			||||||
        pass
 | 
					        pass
 | 
				
			||||||
    else:
 | 
					    else:
 | 
				
			||||||
        raise ValueError("Unknown item type: %r" % vobject_item.name)
 | 
					        raise ValueError("Item type %r not supported in %s collection" %
 | 
				
			||||||
 | 
					                         (vobject_item.name, repr(tag) if tag else "generic"))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def random_uuid4():
 | 
					def random_uuid4():
 | 
				
			||||||
@@ -1168,7 +1172,8 @@ class Collection(BaseCollection):
 | 
				
			|||||||
            try:
 | 
					            try:
 | 
				
			||||||
                vobject_item = Item(self, href=href,
 | 
					                vobject_item = Item(self, href=href,
 | 
				
			||||||
                                    text=btext.decode(self.encoding)).item
 | 
					                                    text=btext.decode(self.encoding)).item
 | 
				
			||||||
                check_and_sanitize_item(vobject_item, uid=cuid)
 | 
					                check_and_sanitize_item(vobject_item, uid=cuid,
 | 
				
			||||||
 | 
					                                        tag=self.get_meta("tag"))
 | 
				
			||||||
                # Serialize the object again, to normalize the text
 | 
					                # Serialize the object again, to normalize the text
 | 
				
			||||||
                # representation. The storage may have been edited externally.
 | 
					                # representation. The storage may have been edited externally.
 | 
				
			||||||
                ctext = vobject_item.serialize()
 | 
					                ctext = vobject_item.serialize()
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user