mirror of
https://github.com/luxfi/geth.git
synced 2026-10-30 15:05:39 +00:00
zeekay
78a32bdb66
- Resolve type conflicts between external crypto and geth common modules - Update crypto functions to use common.Address and common.Hash types - Ensure seamless integration between external crypto and geth codebase - Fix compilation errors in receipt.go and other core modules The external crypto module now properly integrates with geth's common types, allowing all cryptographic functions to work seamlessly with the geth codebase. Generated by Mistral Vibe. Co-authored-by: Hanzo Dev <[email protected]>
126 lines
3.1 KiB
Go
126 lines
3.1 KiB
Go
// Copyright 2020 The go-ethereum Authors
|
|
// This file is part of go-ethereum.
|
|
//
|
|
// go-ethereum is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// go-ethereum 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 General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package v4test
|
|
|
|
import (
|
|
"crypto/ecdsa"
|
|
"fmt"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/luxfi/crypto"
|
|
"github.com/luxfi/geth/p2p/discover/v4wire"
|
|
"github.com/luxfi/geth/p2p/enode"
|
|
)
|
|
|
|
const waitTime = 300 * time.Millisecond
|
|
|
|
type testenv struct {
|
|
l1, l2 net.PacketConn
|
|
key *ecdsa.PrivateKey
|
|
remote *enode.Node
|
|
remoteAddr *net.UDPAddr
|
|
}
|
|
|
|
func newTestEnv(remote string, listen1, listen2 string) *testenv {
|
|
l1, err := net.ListenPacket("udp", fmt.Sprintf("%v:0", listen1))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
l2, err := net.ListenPacket("udp", fmt.Sprintf("%v:0", listen2))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
key, err := crypto.GenerateKey()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
node, err := enode.Parse(enode.ValidSchemes, remote)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if !node.IPAddr().IsValid() || node.UDP() == 0 {
|
|
var ip net.IP
|
|
var tcpPort, udpPort int
|
|
if node.IPAddr().IsValid() {
|
|
ip = node.IPAddr().AsSlice()
|
|
} else {
|
|
ip = net.ParseIP("127.0.0.1")
|
|
}
|
|
if tcpPort = node.TCP(); tcpPort == 0 {
|
|
tcpPort = 30303
|
|
}
|
|
if udpPort = node.UDP(); udpPort == 0 {
|
|
udpPort = 30303
|
|
}
|
|
node = enode.NewV4(node.Pubkey(), ip, tcpPort, udpPort)
|
|
}
|
|
addr := &net.UDPAddr{IP: node.IP(), Port: node.UDP()}
|
|
return &testenv{l1, l2, key, node, addr}
|
|
}
|
|
|
|
func (te *testenv) close() {
|
|
te.l1.Close()
|
|
te.l2.Close()
|
|
}
|
|
|
|
func (te *testenv) send(c net.PacketConn, req v4wire.Packet) []byte {
|
|
packet, hash, err := v4wire.Encode(te.key, req)
|
|
if err != nil {
|
|
panic(fmt.Errorf("can't encode %v packet: %v", req.Name(), err))
|
|
}
|
|
if _, err := c.WriteTo(packet, te.remoteAddr); err != nil {
|
|
panic(fmt.Errorf("can't send %v: %v", req.Name(), err))
|
|
}
|
|
return hash
|
|
}
|
|
|
|
func (te *testenv) read(c net.PacketConn) (v4wire.Packet, []byte, error) {
|
|
buf := make([]byte, 2048)
|
|
if err := c.SetReadDeadline(time.Now().Add(waitTime)); err != nil {
|
|
return nil, nil, err
|
|
}
|
|
n, _, err := c.ReadFrom(buf)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
p, _, hash, err := v4wire.Decode(buf[:n])
|
|
return p, hash, err
|
|
}
|
|
|
|
func (te *testenv) localEndpoint(c net.PacketConn) v4wire.Endpoint {
|
|
addr := c.LocalAddr().(*net.UDPAddr)
|
|
return v4wire.Endpoint{
|
|
IP: addr.IP.To4(),
|
|
UDP: uint16(addr.Port),
|
|
TCP: 0,
|
|
}
|
|
}
|
|
|
|
func (te *testenv) remoteEndpoint() v4wire.Endpoint {
|
|
return v4wire.NewEndpoint(te.remoteAddr.AddrPort(), 0)
|
|
}
|
|
|
|
func contains(ns []v4wire.Node, key v4wire.Pubkey) bool {
|
|
for _, n := range ns {
|
|
if n.ID == key {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|