services: add new fetch

Signed-off-by: Ekaterina Pavlova <[email protected]>

cli/server/server.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"os"
 	"os/signal"
+	"path/filepath"
 	"syscall"
 	"time"
 
@@ -213,6 +214,80 @@ func dumpDB(ctx *cli.Context) error {
 	return nil
 }
 
+func dumpBin(ctx *cli.Context) error {
+	if err := cmdargs.EnsureNone(ctx); err != nil {
+		return err
+	}
+	cfg, err := options.GetConfigFromContext(ctx)
+	if err != nil {
+		return cli.NewExitError(err, 1)
+	}
+	log, _, logCloser, err := options.HandleLoggingParams(ctx.Bool("debug"), cfg.ApplicationConfiguration)
+	if err != nil {
+		return cli.NewExitError(err, 1)
+	}
+	if logCloser != nil {
+		defer func() { _ = logCloser() }()
+	}
+	count := uint32(ctx.Uint("count"))
+	start := uint32(ctx.Uint("start"))
+
+	chain, prometheus, pprof, err := initBCWithMetrics(cfg, log)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		pprof.ShutDown()
+		prometheus.ShutDown()
+		chain.Close()
+	}()
+
+	chainCount := chain.BlockHeight() + 1
+	if start+count > chainCount {
+		return cli.NewExitError(fmt.Errorf("chain is not that high (%d) to dump %d blocks starting from %d", chainCount-1, count, start), 1)
+	}
+	if count == 0 {
+		count = chainCount - start
+	}
+
+	testDir := "./test/"
+	if _, err := os.Stat(testDir); os.IsNotExist(err) {
+		if err := os.MkdirAll(testDir, 0755); err != nil {
+			return cli.NewExitError(fmt.Sprintf("failed to create directory %s: %v", testDir, err), 1)
+		}
+	}
+
+	for i := start; i < start+count; i++ {
+		bh := chain.GetHeaderHash(i)
+		blk, err := chain.GetBlock(bh)
+		if err != nil {
+			return cli.NewExitError(fmt.Sprintf("failed to get block %d: %v", i, err), 1)
+		}
+		filePath := filepath.Join(testDir, fmt.Sprintf("block-%d.bin", i))
+		if err := saveBlockToFile(blk, filePath); err != nil {
+			return cli.NewExitError(fmt.Sprintf("failed to save block %d to file: %v", i, err), 1)
+		}
+	}
+
+	return nil
+}
+
+func saveBlockToFile(blk *block.Block, filePath string) error {
+	file, err := os.Create(filePath)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	writer := io.NewBinWriterFromIO(file)
+	blk.EncodeBinary(writer)
+	if writer.Err != nil {
+		return writer.Err
+	}
+
+	return nil
+}
+
 func restoreDB(ctx *cli.Context) error {
 	if err := cmdargs.EnsureNone(ctx); err != nil {
 		return err

pkg/network/server.go

@@ -295,6 +295,11 @@ func (s *Server) Start() {
 	go s.relayBlocksLoop()
 	go s.bQueue.Run()
 	go s.bSyncQueue.Run()
+	//if s.chain.neofs.IsEnabled() {
+	//	go s.neofs.Run()
+	//	wait until end
+	//}
+	// tr.Accept() has PutBlock inside so here should be neofs block querying and then check tpcpeers transport.
 	for _, tr := range s.transports {
 		go tr.Accept()
 	}
@@ -785,6 +790,7 @@ func (s *Server) handlePing(p Peer, ping *payload.Ping) error {
 	return p.EnqueueP2PMessage(NewMessage(CMDPong, payload.NewPing(s.chain.BlockHeight(), s.id)))
 }
 
+// here requestBlocksOrHeaders is called
 func (s *Server) requestBlocksOrHeaders(p Peer) error {
 	if s.stateSync.NeedHeaders() {
 		if s.chain.HeaderHeight() < p.LastBlockIndex() {
@@ -813,6 +819,7 @@ func (s *Server) requestBlocksOrHeaders(p Peer) error {
 	return nil
 }
 
+// here
 // requestHeaders sends a CMDGetHeaders message to the peer to sync up in headers.
 func (s *Server) requestHeaders(p Peer) error {
 	pl := getRequestBlocksPayload(p, s.chain.HeaderHeight(), &s.lastRequestedHeader)