diff options
author | ethersphere <thesw@rm.eth> | 2018-06-20 20:06:27 +0800 |
---|---|---|
committer | ethersphere <thesw@rm.eth> | 2018-06-22 03:10:31 +0800 |
commit | e187711c6545487d4cac3701f0f506bb536234e2 (patch) | |
tree | d2f6150f70b84b36e49a449082aeda267b4b9046 /swarm/network/bitvector | |
parent | 574378edb50c907b532946a1d4654dbd6701b20a (diff) | |
download | go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.tar go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.tar.gz go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.tar.bz2 go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.tar.lz go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.tar.xz go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.tar.zst go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.zip |
swarm: network rewrite merge
Diffstat (limited to 'swarm/network/bitvector')
-rw-r--r-- | swarm/network/bitvector/bitvector.go | 66 | ||||
-rw-r--r-- | swarm/network/bitvector/bitvector_test.go | 104 |
2 files changed, 170 insertions, 0 deletions
diff --git a/swarm/network/bitvector/bitvector.go b/swarm/network/bitvector/bitvector.go new file mode 100644 index 000000000..edc7c50cb --- /dev/null +++ b/swarm/network/bitvector/bitvector.go @@ -0,0 +1,66 @@ +// Copyright 2018 The go-ethereum Authors +// This file is part of the go-ethereum library. +// +// The go-ethereum library is free software: you can redistribute it and/or modify +// it under the terms of the GNU Lesser General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// The go-ethereum library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public License +// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>. + +package bitvector + +import ( + "errors" +) + +var errInvalidLength = errors.New("invalid length") + +type BitVector struct { + len int + b []byte +} + +func New(l int) (bv *BitVector, err error) { + return NewFromBytes(make([]byte, l/8+1), l) +} + +func NewFromBytes(b []byte, l int) (bv *BitVector, err error) { + if l <= 0 { + return nil, errInvalidLength + } + if len(b)*8 < l { + return nil, errInvalidLength + } + return &BitVector{ + len: l, + b: b, + }, nil +} + +func (bv *BitVector) Get(i int) bool { + bi := i / 8 + return bv.b[bi]&(0x1<<uint(i%8)) != 0 +} + +func (bv *BitVector) Set(i int, v bool) { + bi := i / 8 + cv := bv.Get(i) + if cv != v { + bv.b[bi] ^= 0x1 << uint8(i%8) + } +} + +func (bv *BitVector) Bytes() []byte { + return bv.b +} + +func (bv *BitVector) Length() int { + return bv.len +} diff --git a/swarm/network/bitvector/bitvector_test.go b/swarm/network/bitvector/bitvector_test.go new file mode 100644 index 000000000..4e6260038 --- /dev/null +++ b/swarm/network/bitvector/bitvector_test.go @@ -0,0 +1,104 @@ +// Copyright 2018 The go-ethereum Authors +// This file is part of the go-ethereum library. +// +// The go-ethereum library is free software: you can redistribute it and/or modify +// it under the terms of the GNU Lesser General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// The go-ethereum library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public License +// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>. + +package bitvector + +import "testing" + +func TestBitvectorNew(t *testing.T) { + _, err := New(0) + if err != errInvalidLength { + t.Errorf("expected err %v, got %v", errInvalidLength, err) + } + + _, err = NewFromBytes(nil, 0) + if err != errInvalidLength { + t.Errorf("expected err %v, got %v", errInvalidLength, err) + } + + _, err = NewFromBytes([]byte{0}, 9) + if err != errInvalidLength { + t.Errorf("expected err %v, got %v", errInvalidLength, err) + } + + _, err = NewFromBytes(make([]byte, 8), 8) + if err != nil { + t.Error(err) + } +} + +func TestBitvectorGetSet(t *testing.T) { + for _, length := range []int{ + 1, + 2, + 4, + 8, + 9, + 15, + 16, + } { + bv, err := New(length) + if err != nil { + t.Errorf("error for length %v: %v", length, err) + } + + for i := 0; i < length; i++ { + if bv.Get(i) { + t.Errorf("expected false for element on index %v", i) + } + } + + func() { + defer func() { + if err := recover(); err == nil { + t.Errorf("expecting panic") + } + }() + bv.Get(length + 8) + }() + + for i := 0; i < length; i++ { + bv.Set(i, true) + for j := 0; j < length; j++ { + if j == i { + if !bv.Get(j) { + t.Errorf("element on index %v is not set to true", i) + } + } else { + if bv.Get(j) { + t.Errorf("element on index %v is not false", i) + } + } + } + + bv.Set(i, false) + + if bv.Get(i) { + t.Errorf("element on index %v is not set to false", i) + } + } + } +} + +func TestBitvectorNewFromBytesGet(t *testing.T) { + bv, err := NewFromBytes([]byte{8}, 8) + if err != nil { + t.Error(err) + } + if !bv.Get(3) { + t.Fatalf("element 3 is not set to true: state %08b", bv.b[0]) + } +} |