xmlparser/document.go

Rama master · réplica de lectura del repositorio original.

package xmlparser

import (
	"io"
)

type Document struct {
	Node
}

func NewDocument(r io.Reader) (*Document, error) {
	doc := &Document{
		// Attributes for document are always empty, but they are exposed,
		// so we need to initialize them to avoid nil pointer dereference.
		Attrs: NewAttributes(),
	}

	if err := doc.readFrom(r); err != nil {
		return nil, err
	}

	return doc, nil
}

// ReplaceEntities replaces XML entities in the document
// according to the entity declarations in the DOCTYPE.
// It modifies the document in place.
//
// Entities are replaced only once to avoid attacks like the
// "Billion Laughs" XML entity expansion attack.
func (doc *Document) ReplaceEntities() error {
	if em := ParseEntityMap(doc.Children); em != nil {
		return doc.replaceEntities(em)
	}
	return nil
}