From 878d50a7ccba207b1b4bca66cd88fff9ed974c25 Mon Sep 17 00:00:00 2001
From: Delweng <delweng@gmail.com>
Date: Sat, 30 Jul 2022 00:28:14 +0800
Subject: [PATCH] cm/puppeth: fix crash when of ethstats specifier doesn't
 contain `:` (#25405)

Signed-off-by: Delweng <delweng@gmail.com>
---
 cmd/puppeth/module.go          | 9 +++++++++
 cmd/puppeth/module_explorer.go | 2 +-
 cmd/puppeth/module_faucet.go   | 2 +-
 cmd/puppeth/module_node.go     | 2 +-
 4 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/cmd/puppeth/module.go b/cmd/puppeth/module.go
index b6a029a01a48a..771ae38058bc9 100644
--- a/cmd/puppeth/module.go
+++ b/cmd/puppeth/module.go
@@ -150,3 +150,12 @@ func checkPort(host string, port int) error {
 	conn.Close()
 	return nil
 }
+
+// getEthName gets the Ethereum Name from ethstats
+func getEthName(s string) string {
+	n := strings.Index(s, ":")
+	if n >= 0 {
+		return s[:n]
+	}
+	return s
+}
diff --git a/cmd/puppeth/module_explorer.go b/cmd/puppeth/module_explorer.go
index 1165f70fcf516..3812f9fdb9637 100644
--- a/cmd/puppeth/module_explorer.go
+++ b/cmd/puppeth/module_explorer.go
@@ -104,7 +104,7 @@ func deployExplorer(client *sshClient, network string, bootnodes []string, confi
 		"Datadir":     config.node.datadir,
 		"DBDir":       config.dbdir,
 		"EthPort":     config.node.port,
-		"EthName":     config.node.ethstats[:strings.Index(config.node.ethstats, ":")],
+		"EthName":     getEthName(config.node.ethstats),
 		"WebPort":     config.port,
 		"Transformer": transformer,
 	})
diff --git a/cmd/puppeth/module_faucet.go b/cmd/puppeth/module_faucet.go
index 88cb80ae4c42c..a4f6e65694df4 100644
--- a/cmd/puppeth/module_faucet.go
+++ b/cmd/puppeth/module_faucet.go
@@ -116,7 +116,7 @@ func deployFaucet(client *sshClient, network string, bootnodes []string, config
 		"VHost":         config.host,
 		"ApiPort":       config.port,
 		"EthPort":       config.node.port,
-		"EthName":       config.node.ethstats[:strings.Index(config.node.ethstats, ":")],
+		"EthName":       getEthName(config.node.ethstats),
 		"CaptchaToken":  config.captchaToken,
 		"CaptchaSecret": config.captchaSecret,
 		"FaucetAmount":  config.amount,
diff --git a/cmd/puppeth/module_node.go b/cmd/puppeth/module_node.go
index 3ea96870d4f5d..b8aa30db39a8d 100644
--- a/cmd/puppeth/module_node.go
+++ b/cmd/puppeth/module_node.go
@@ -123,7 +123,7 @@ func deployNode(client *sshClient, network string, bootnodes []string, config *n
 		"TotalPeers": config.peersTotal,
 		"Light":      config.peersLight > 0,
 		"LightPeers": config.peersLight,
-		"Ethstats":   config.ethstats[:strings.Index(config.ethstats, ":")],
+		"Ethstats":   getEthName(config.ethstats),
 		"Etherbase":  config.etherbase,
 		"GasTarget":  config.gasTarget,
 		"GasLimit":   config.gasLimit,