mirror of https://github.com/milvus-io/milvus.git
some SDKs doesn't check the JSON data validation, add this in server. pr: #28602 Signed-off-by: yah01 <yah2er0ne@outlook.com>pull/28770/head
parent
c4be1ddf44
commit
55fda49104
|
@ -1,6 +1,7 @@
|
||||||
package proxy
|
package proxy
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"math"
|
"math"
|
||||||
"reflect"
|
"reflect"
|
||||||
|
@ -341,6 +342,18 @@ func (v *validateUtil) checkJSONFieldData(field *schemapb.FieldData, fieldSchema
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var jsonMap map[string]interface{}
|
||||||
|
for _, data := range jsonArray {
|
||||||
|
err := json.Unmarshal(data, &jsonMap)
|
||||||
|
if err != nil {
|
||||||
|
log.Warn("insert invalid JSON data",
|
||||||
|
zap.ByteString("data", data),
|
||||||
|
zap.Error(err),
|
||||||
|
)
|
||||||
|
return merr.WrapErrIoFailedReason(err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -3209,4 +3209,27 @@ func Test_validateUtil_checkJSONData(t *testing.T) {
|
||||||
err := v.checkJSONFieldData(data, f)
|
err := v.checkJSONFieldData(data, f)
|
||||||
assert.Error(t, err)
|
assert.Error(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
t.Run("invalid_JSON_data", func(t *testing.T) {
|
||||||
|
v := newValidateUtil(withOverflowCheck(), withMaxLenCheck())
|
||||||
|
jsonData := "hello"
|
||||||
|
f := &schemapb.FieldSchema{
|
||||||
|
DataType: schemapb.DataType_JSON,
|
||||||
|
}
|
||||||
|
data := &schemapb.FieldData{
|
||||||
|
FieldName: "json",
|
||||||
|
Field: &schemapb.FieldData_Scalars{
|
||||||
|
Scalars: &schemapb.ScalarField{
|
||||||
|
Data: &schemapb.ScalarField_JsonData{
|
||||||
|
JsonData: &schemapb.JSONArray{
|
||||||
|
Data: [][]byte{[]byte(jsonData)},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
err := v.checkJSONFieldData(data, f)
|
||||||
|
assert.Error(t, err)
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue