mirror of
https://github.com/go-i2p/go-i2p-bt.git
synced 2025-07-01 20:20:04 -04:00
104 lines
2.5 KiB
Go
104 lines
2.5 KiB
Go
// Copyright 2020 xgfone, 2023 idk
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package peerprotocol
|
|
|
|
import (
|
|
"bytes"
|
|
"log"
|
|
"testing"
|
|
|
|
"github.com/go-i2p/sam3"
|
|
)
|
|
|
|
func TestCompactIP(t *testing.T) {
|
|
ipv4 := CompactIP([]byte{1, 2, 3, 4})
|
|
b, err := ipv4.MarshalBencode()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
log.Println("IPv4 Test", ipv4.String(), len(ipv4))
|
|
|
|
var ip CompactIP
|
|
if err = ip.UnmarshalBencode(b); err != nil {
|
|
t.Error(err, ip)
|
|
} else if ip.String() != "1.2.3.4" {
|
|
t.Error(ip.String(), ",", ip)
|
|
}
|
|
}
|
|
|
|
func TestCompactIP6(t *testing.T) {
|
|
ipv6 := CompactIP([]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16})
|
|
b, err := ipv6.MarshalBencode()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
log.Println("IPv6 Test", ipv6.String(), len(ipv6))
|
|
|
|
var ip CompactIP
|
|
if err = ip.UnmarshalBencode(b); err != nil {
|
|
t.Error(err)
|
|
} else if ip.String() != "102:304:506:708:90a:b0c:d0e:f10" {
|
|
t.Error(ip.String(), ",", ip)
|
|
}
|
|
}
|
|
|
|
func TestCompactI2P(t *testing.T) {
|
|
sam, err := sam3.NewSAM("127.0.0.1:7656")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer sam.Close()
|
|
i2pkeys, err := sam.NewKeys()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
dh := i2pkeys.Address.DestHash()
|
|
i2p := CompactIP(dh[:])
|
|
b, err := i2p.MarshalBencode()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
log.Println("I2P Test", i2p.String(), len(b))
|
|
|
|
var ip CompactIP
|
|
if err = ip.UnmarshalBencode(b); err != nil {
|
|
t.Error(err)
|
|
} else if ip.String() != dh.String() {
|
|
t.Error(ip, dh)
|
|
}
|
|
}
|
|
|
|
func TestUtMetadataExtendedMsg(t *testing.T) {
|
|
buf := new(bytes.Buffer)
|
|
data := []byte{0x31, 0x32, 0x33, 0x34, 0x35}
|
|
m1 := UtMetadataExtendedMsg{MsgType: 1, Piece: 2, TotalSize: 1024, Data: data}
|
|
if err := m1.EncodeToPayload(buf); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
msg := Message{Type: MTypeExtended, ExtendedPayload: buf.Bytes()}
|
|
m2, err := msg.UtMetadataExtendedMsg()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
} else if m2.MsgType != 1 || m2.Piece != 2 || m2.TotalSize != 1024 {
|
|
t.Error(m2)
|
|
} else if !bytes.Equal(m2.Data, data) {
|
|
t.Fail()
|
|
}
|
|
}
|