Rewrite normalizeDocument

Author: tvdijen

src/XML/DOMDocumentFactory.php

@@ -16,7 +16,6 @@
 use function restore_error_handler;
 use function set_error_handler;
 use function sprintf;
-use function strpos;
 
 /**
  * @package simplesamlphp/xml-common
@@ -170,51 +169,37 @@ public static function create(string $encoding = 'UTF-8'): Dom\XMLDocument
      */
     public static function normalizeDocument(Dom\XMLDocument $doc): Dom\XMLDocument
     {
-        // Get the root element
         $root = $doc->documentElement;
-        if ($root === null) {
+        if (!$root) {
             return $doc;
         }
 
         $xpath = XPath::getXPath($doc);
-        $xmlnsAttributes = [];
-
-        // Collect namespace declarations needed for prefixed elements in the document
-        foreach ($xpath->query('//*[namespace::*]') as $node) {
-            if ($node instanceof Dom\Element) {
-                $name = 'xmlns:' . $node->prefix;
-                // Both prefix and namespaceURI NULL equals the default xmlns:xml namespace
-                if ($node->prefix !== null && $node->namespaceURI !== null) {
-                    $xmlnsAttributes[$name] = $node->namespaceURI;
-                }
-            }
-        }
 
-        // If no xmlns attributes found, return early with debug info
-        if (empty($xmlnsAttributes)) {
-            return $doc;
+        // 1. Collect all xmlns* attributes from the entire document
+        $xmlnsAttributes = [];
+        foreach ($xpath->query('//@*[starts-with(name(), "xmlns")]') as $attr) {
+            $name = $attr->nodeName;
+            $value = $attr->nodeValue;
+            $xmlnsAttributes[$name] = $value;
         }
 
-        // Remove xmlns attributes from all elements (proper XMLNS namespace removal)
-        foreach ($xpath->query('//*[namespace::*]') as $node) {
-            if (!$node instanceof Dom\Element) {
-                continue;
-            }
-
-            foreach ($node->attributes as $attr) {
-                if ($attr->namespaceURI === C::NS_XMLNS) {
-                    $node->removeAttributeNS(C::NS_XMLNS, $attr->localName);
-                    continue;
+        // 2. Remove all xmlns* attributes from every element
+        foreach ($xpath->query('//*') as $element) {
+            if ($element instanceof Dom\Element) {
+                $toRemove = [];
+                foreach ($element->attributes as $attr) {
+                    if (str_starts_with($attr->nodeName, 'xmlns')) {
+                        $toRemove[] = $attr->nodeName;
+                    }
                 }
-
-                if (strpos($attr->nodeName, 'xmlns') === 0 || $attr->nodeName === 'xmlns') {
-                    // Fallback for implementations that still expose xmlns attrs without namespaceURI
-                    $node->removeAttribute($attr->nodeName);
+                foreach ($toRemove as $name) {
+                    $element->removeAttribute($name);
                 }
             }
         }
 
-        // Add all collected xmlns attributes to the root element
+        // 3. Re-attach all collected declarations to the root (correctly)
         foreach ($xmlnsAttributes as $name => $value) {
             $root->setAttributeNS(C::NS_XMLNS, $name, $value);
         }

src/XML/SerializableElementTrait.php

@@ -31,23 +31,18 @@ trait SerializableElementTrait
     public function __toString(): string
     {
         $xml = $this->toXML();
-
-        /** @var \Dom\XMLDocument $ownerDocument */
-        $ownerDocument = $xml->ownerDocument;
-        $xmlString = $ownerDocument->saveXml($ownerDocument->documentElement);
-
-        $doc = DOMDocumentFactory::fromString($xmlString);
-        $doc->formatOutput = $this->formatOutput;
+        $doc = $xml->ownerDocument;
 
         if (static::getNormalization() === true) {
-            /** @var \Dom\XMLDocument $normalized */
             $normalized = DOMDocumentFactory::normalizeDocument($doc);
-            $normalized->normalize();
-            return $normalized->saveXml($normalized->firstChild);
+            $normalized->formatOutput = $this->formatOutput;
+            return $normalized->saveXML($normalized->documentElement);
         }
 
-        $doc->normalize();
-        return $doc->saveXml($doc->firstChild);
+        // Non-normalized path
+        $doc->formatOutput = $this->formatOutput;
+        $doc->normalizeDocument();
+        return $doc->saveXML($doc->documentElement);
     }