encoding/xml: embedded reference to substruct causes XML marshaller to panic on encoding

When encoding a xml attribute is zero value (IsValid == false), we need
a `continue` to jump over the attribute. If not, followed marshalAttr
function will panic.

Fixes: #50164
Change-Id: I42e064558e7becfbf47728b14cbf5c7afa1e8798
Reviewed-on: https://go-review.googlesource.com/c/go/+/385514
Reviewed-by: Daniel Martí <mvdan@mvdan.cc>
Trust: Daniel Martí <mvdan@mvdan.cc>
Run-TryBot: Daniel Martí <mvdan@mvdan.cc>
TryBot-Result: Gopher Robot <gobot@golang.org>
Reviewed-by: Ian Lance Taylor <iant@golang.org>
This commit is contained in:
hopehook 2022-02-13 22:03:56 +08:00 committed by Ian Lance Taylor
parent 0261fa616a
commit 851ecea4cc
2 changed files with 37 additions and 1 deletions

View file

@ -512,7 +512,7 @@ func (p *printer) marshalValue(val reflect.Value, finfo *fieldInfo, startTemplat
}
fv := finfo.value(val, dontInitNilPointers)
if finfo.flags&fOmitEmpty != 0 && isEmptyValue(fv) {
if finfo.flags&fOmitEmpty != 0 && (!fv.IsValid() || isEmptyValue(fv)) {
continue
}

View file

@ -2495,3 +2495,39 @@ func TestInvalidXMLName(t *testing.T) {
t.Errorf("error %q does not contain %q", err, want)
}
}
// Issue 50164. Crash on zero value XML attribute.
type LayerOne struct {
XMLName Name `xml:"l1"`
Value *float64 `xml:"value,omitempty"`
*LayerTwo `xml:",omitempty"`
}
type LayerTwo struct {
ValueTwo *int `xml:"value_two,attr,omitempty"`
}
func TestMarshalZeroValue(t *testing.T) {
proofXml := `<l1><value>1.2345</value></l1>`
var l1 LayerOne
err := Unmarshal([]byte(proofXml), &l1)
if err != nil {
t.Fatalf("unmarshal XML error: %v", err)
}
want := float64(1.2345)
got := *l1.Value
if got != want {
t.Fatalf("unexpected unmarshal result, want %f but got %f", want, got)
}
// Marshal again (or Encode again)
// In issue 50164, here `Marshal(l1)` will panic because of the zero value of xml attribute ValueTwo `value_two`.
anotherXML, err := Marshal(l1)
if err != nil {
t.Fatalf("marshal XML error: %v", err)
}
if string(anotherXML) != proofXml {
t.Fatalf("unexpected unmarshal result, want %q but got %q", proofXml, anotherXML)
}
}