TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

io: restrict ReadArray max array size

Browse source
This commit is contained in:
Evgenii Stratonikov 2019-11-15 22:33:28 +03:00
parent 2679d3fa35
commit f65545023d
2 changed files with 29 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
pkg/io/binaryrw_test.go
View file

@ -266,6 +266,17 @@ func TestBinReader_ReadArray(t *testing.T) {
require.NoError(t, r.Err)
require.Equal(t, elems, arrVal)
r = NewBinReaderFromBuf(data)
arrVal = []testSerializable{}
r.ReadArray(&arrVal, 3)
require.NoError(t, r.Err)
require.Equal(t, elems, arrVal)
r = NewBinReaderFromBuf(data)
arrVal = []testSerializable{}
r.ReadArray(&arrVal, 2)
require.Error(t, r.Err)
r = NewBinReaderFromBuf([]byte{0})
r.ReadArray(&arrVal)
require.NoError(t, r.Err)