go/json-iterator
1
0
Fork 0
mirror of https://github.com/json-iterator/go.git synced 2025-02-07 19:30:06 +02:00
Code Issues Releases Activity

#68 handle float to int safely

Browse Source
This commit is contained in:
Tao Wen 2017-06-20 15:46:22 +08:00
parent a5ae3a2649
commit 8ef0c22f25
2 changed files with 37 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
extra/fuzzy_decoder.go
View File

@ -4,11 +4,28 @@ import (
"github.com/json-iterator/go"
"unsafe"
"encoding/json"
"strings"
)
const MaxUint = ^uint(0)
const MinUint = 0
const MaxInt = int(MaxUint >> 1)
const MinInt = -MaxInt - 1
func RegisterFuzzyDecoders() {
jsoniter.RegisterTypeDecoder("string", &FuzzyStringDecoder{})
jsoniter.RegisterTypeDecoder("int", &FuzzyIntDecoder{})
jsoniter.RegisterTypeDecoder("int", &FuzzyNumberDecoder{func(isFloat bool, ptr unsafe.Pointer, iter *jsoniter.Iterator, errorReporter *jsoniter.Iterator) {
if isFloat {
val := iter.ReadFloat64()
if val > float64(MaxInt) || val < float64(MinInt) {
errorReporter.ReportError("fuzzy decode int", "exceed range")
return
}
*((*int)(ptr)) = int(val)
} else {
*((*int)(ptr)) = iter.ReadInt()
}
}})
}
type FuzzyStringDecoder struct {
@ -28,20 +45,25 @@ func (decoder *FuzzyStringDecoder) Decode(ptr unsafe.Pointer, iter *jsoniter.Ite
}
}
type FuzzyIntDecoder struct {
type FuzzyNumberDecoder struct {
fun func(isFloat bool, ptr unsafe.Pointer, iter *jsoniter.Iterator, errorReporter *jsoniter.Iterator)
}
func (decoder *FuzzyIntDecoder) Decode(ptr unsafe.Pointer, iter *jsoniter.Iterator) {
func (decoder *FuzzyNumberDecoder) Decode(ptr unsafe.Pointer, iter *jsoniter.Iterator) {
valueType := iter.WhatIsNext()
var str string
switch valueType {
case jsoniter.Number:
// use current iterator
var number json.Number
iter.ReadVal(&number)
str = string(number)
case jsoniter.String:
str := iter.ReadString()
iter = iter.Config().BorrowIterator([]byte(str))
defer iter.Config().ReturnIterator(iter)
str = iter.ReadString()
default:
iter.ReportError("FuzzyIntDecoder", "not number or string")
iter.ReportError("FuzzyNumberDecoder", "not number or string")
}
*((*int)(ptr)) = int(iter.ReadFloat64())
newIter := iter.Config().BorrowIterator([]byte(str))
defer iter.Config().ReturnIterator(newIter)
isFloat := strings.IndexByte(str, '.') != -1
decoder.fun(isFloat, ptr, newIter, iter)
}

6
extra/fuzzy_decoder_test.go
View File

@ -51,3 +51,9 @@ func Test_float_to_int(t *testing.T) {
should.Nil(jsoniter.UnmarshalFromString(`1.23`, &val))
should.Equal(1, val)
}
func Test_large_float_to_int(t *testing.T) {
should := require.New(t)
var val int
should.NotNil(jsoniter.UnmarshalFromString(`1234512345123451234512345.0`, &val))
}