mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-07-30 00:23:46 +00:00
0d2b1d0630
* bad block tracing * Add debug env framework * add implementation of custom block tracing * add few comments * use tar name fn * wip: add unit tests for block tracing * complete test for block tracer * fix: close grpc server * refactor cli tests * fix: change condition for parsing args * Fix port binding for test server * add helper for creating and closing mock server for tests * consume mock server in tests * fixes due to geth merge * fix: handle port selection for http server * update help and markdown for debug command * update docs * update debug synopsis * fix: use chunked encoder to handle large data over grpc * fix prints * lint * rm unused function, rename fn to TraceBorBlock Co-authored-by: Ferran Borreguero <ferranbt@protonmail.com>
72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"math/rand"
|
|
"net"
|
|
"os"
|
|
"sync/atomic"
|
|
"time"
|
|
)
|
|
|
|
var maxPortCheck int32 = 100
|
|
|
|
// findAvailablePort returns the next available port starting from `from`
|
|
func findAvailablePort(from int32, count int32) (int32, error) {
|
|
if count == maxPortCheck {
|
|
return 0, fmt.Errorf("no available port found")
|
|
}
|
|
port := atomic.AddInt32(&from, 1)
|
|
addr := fmt.Sprintf("localhost:%d", port)
|
|
lis, err := net.Listen("tcp", addr)
|
|
count++
|
|
if err == nil {
|
|
lis.Close()
|
|
return port, nil
|
|
} else {
|
|
return findAvailablePort(from, count)
|
|
}
|
|
}
|
|
|
|
func CreateMockServer(config *Config) (*Server, error) {
|
|
if config == nil {
|
|
config = DefaultConfig()
|
|
}
|
|
|
|
// find available port for grpc server
|
|
rand.Seed(time.Now().UnixNano())
|
|
var from int32 = 60000 // the min port to start checking from
|
|
var to int32 = 61000 // the max port to start checking from
|
|
port, err := findAvailablePort(rand.Int31n(to-from+1)+from, 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// grpc port
|
|
config.GRPC.Addr = fmt.Sprintf(":%d", port)
|
|
|
|
// datadir
|
|
datadir, _ := ioutil.TempDir("/tmp", "bor-cli-test")
|
|
config.DataDir = datadir
|
|
|
|
// find available port for http server
|
|
from = 8545
|
|
to = 9545
|
|
port, err = findAvailablePort(rand.Int31n(to-from+1)+from, 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
config.JsonRPC.Http.Port = uint64(port)
|
|
|
|
// start the server
|
|
return NewServer(config)
|
|
}
|
|
|
|
func CloseMockServer(server *Server) {
|
|
// remove the contents of temp data dir
|
|
os.RemoveAll(server.config.DataDir)
|
|
|
|
// close the server
|
|
server.Stop()
|
|
}
|