123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108 |
- // 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 main
- import (
- "fmt"
- "net"
- "github.com/ethereum/go-ethereum/cmd/devp2p/internal/ethtest"
- "github.com/ethereum/go-ethereum/crypto"
- "github.com/ethereum/go-ethereum/internal/utesting"
- "github.com/ethereum/go-ethereum/p2p"
- "github.com/ethereum/go-ethereum/p2p/rlpx"
- "github.com/ethereum/go-ethereum/rlp"
- "gopkg.in/urfave/cli.v1"
- )
- var (
- rlpxCommand = cli.Command{
- Name: "rlpx",
- Usage: "RLPx Commands",
- Subcommands: []cli.Command{
- rlpxPingCommand,
- rlpxEthTestCommand,
- },
- }
- rlpxPingCommand = cli.Command{
- Name: "ping",
- Usage: "ping <node>",
- Action: rlpxPing,
- }
- rlpxEthTestCommand = cli.Command{
- Name: "eth-test",
- Usage: "Runs tests against a node",
- ArgsUsage: "<node> <chain.rlp> <genesis.json>",
- Action: rlpxEthTest,
- Flags: []cli.Flag{
- testPatternFlag,
- testTAPFlag,
- },
- }
- )
- func rlpxPing(ctx *cli.Context) error {
- n := getNodeArg(ctx)
- fd, err := net.Dial("tcp", fmt.Sprintf("%v:%d", n.IP(), n.TCP()))
- if err != nil {
- return err
- }
- conn := rlpx.NewConn(fd, n.Pubkey())
- ourKey, _ := crypto.GenerateKey()
- _, err = conn.Handshake(ourKey)
- if err != nil {
- return err
- }
- code, data, _, err := conn.Read()
- if err != nil {
- return err
- }
- switch code {
- case 0:
- var h ethtest.Hello
- if err := rlp.DecodeBytes(data, &h); err != nil {
- return fmt.Errorf("invalid handshake: %v", err)
- }
- fmt.Printf("%+v\n", h)
- case 1:
- var msg []p2p.DiscReason
- if rlp.DecodeBytes(data, &msg); len(msg) == 0 {
- return fmt.Errorf("invalid disconnect message")
- }
- return fmt.Errorf("received disconnect message: %v", msg[0])
- default:
- return fmt.Errorf("invalid message code %d, expected handshake (code zero)", code)
- }
- return nil
- }
- // rlpxEthTest runs the eth protocol test suite.
- func rlpxEthTest(ctx *cli.Context) error {
- if ctx.NArg() < 3 {
- exit("missing path to chain.rlp as command-line argument")
- }
- suite, err := ethtest.NewSuite(getNodeArg(ctx), ctx.Args()[1], ctx.Args()[2])
- if err != nil {
- exit(err)
- }
- // check if given node supports eth66, and if so, run eth66 protocol tests as well
- is66Failed, _ := utesting.Run(utesting.Test{Name: "Is_66", Fn: suite.Is_66})
- if is66Failed {
- return runTests(ctx, suite.EthTests())
- }
- return runTests(ctx, suite.AllEthTests())
- }
|