Skip to content

Commit

Permalink
Prevent automatic conversion between int and float when unmarshaling (#…
Browse files Browse the repository at this point in the history
…390)

Fixes #389
Co-authored-by: Thomas Pelletier <pelletier.thomas@gmail.com>
  • Loading branch information
x-hgg-x authored May 4, 2020
1 parent 3478219 commit 71a8bd4
Show file tree
Hide file tree
Showing 2 changed files with 40 additions and 3 deletions.
6 changes: 3 additions & 3 deletions marshal.go
Original file line number Diff line number Diff line change
Expand Up @@ -985,7 +985,7 @@ func (d *Decoder) valueFromToml(mtype reflect.Type, tval interface{}, mval1 *ref
}
return reflect.ValueOf(d), nil
}
if !val.Type().ConvertibleTo(mtype) {
if !val.Type().ConvertibleTo(mtype) || val.Kind() == reflect.Float64 {
return reflect.ValueOf(nil), fmt.Errorf("Can't convert %v(%T) to %v", tval, tval, mtype.String())
}
if reflect.Indirect(reflect.New(mtype)).OverflowInt(val.Convert(reflect.TypeOf(int64(0))).Int()) {
Expand All @@ -995,7 +995,7 @@ func (d *Decoder) valueFromToml(mtype reflect.Type, tval interface{}, mval1 *ref
return val.Convert(mtype), nil
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
val := reflect.ValueOf(tval)
if !val.Type().ConvertibleTo(mtype) {
if !val.Type().ConvertibleTo(mtype) || val.Kind() == reflect.Float64 {
return reflect.ValueOf(nil), fmt.Errorf("Can't convert %v(%T) to %v", tval, tval, mtype.String())
}

Expand All @@ -1009,7 +1009,7 @@ func (d *Decoder) valueFromToml(mtype reflect.Type, tval interface{}, mval1 *ref
return val.Convert(mtype), nil
case reflect.Float32, reflect.Float64:
val := reflect.ValueOf(tval)
if !val.Type().ConvertibleTo(mtype) {
if !val.Type().ConvertibleTo(mtype) || val.Kind() == reflect.Int64 {
return reflect.ValueOf(nil), fmt.Errorf("Can't convert %v(%T) to %v", tval, tval, mtype.String())
}
if reflect.Indirect(reflect.New(mtype)).OverflowFloat(val.Convert(reflect.TypeOf(float64(0))).Float()) {
Expand Down
37 changes: 37 additions & 0 deletions marshal_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -2023,6 +2023,43 @@ func TestUnmarshalCamelCaseKey(t *testing.T) {
}
}


func TestUnmarshalNegativeUint(t *testing.T) {
type check struct{ U uint }

tree, _ := Load("u = -1")
err := tree.Unmarshal(&check{})
if err.Error() != "(1, 1): -1(int64) is negative so does not fit in uint" {
t.Error("expect err:(1, 1): -1(int64) is negative so does not fit in uint but got:", err)
}
}

func TestUnmarshalCheckConversionFloatInt(t *testing.T) {
type conversionCheck struct {
U uint
I int
F float64
}

treeU, _ := Load("u = 1e300")
treeI, _ := Load("i = 1e300")
treeF, _ := Load("f = 9223372036854775806")

errU := treeU.Unmarshal(&conversionCheck{})
errI := treeI.Unmarshal(&conversionCheck{})
errF := treeF.Unmarshal(&conversionCheck{})

if errU.Error() != "(1, 1): Can't convert 1e+300(float64) to uint" {
t.Error("expect err:(1, 1): Can't convert 1e+300(float64) to uint but got:", errU)
}
if errI.Error() != "(1, 1): Can't convert 1e+300(float64) to int" {
t.Error("expect err:(1, 1): Can't convert 1e+300(float64) to int but got:", errI)
}
if errF.Error() != "(1, 1): Can't convert 9223372036854775806(int64) to float64" {
t.Error("expect err:(1, 1): Can't convert 9223372036854775806(int64) to float64 but got:", errF)
}
}

func TestUnmarshalOverflow(t *testing.T) {
type overflow struct {
U8 uint8
Expand Down

0 comments on commit 71a8bd4

Please sign in to comment.