whisper: add light mode check to handshake (#16725)

2018-09-05 11:57:45 +03:00
parent cf33d8b83c
commit 8711e2b636
8 changed files with 126 additions and 18 deletions
--- a/cmd/geth/config.go
+++ b/cmd/geth/config.go
@ -168,6 +168,9 @@ func makeFullNode(ctx *cli.Context) *node.Node {
 		if ctx.GlobalIsSet(utils.WhisperMinPOWFlag.Name) {
 			cfg.Shh.MinimumAcceptedPOW = ctx.Float64(utils.WhisperMinPOWFlag.Name)
 		}
+		if ctx.GlobalIsSet(utils.WhisperRestrictConnectionBetweenLightClientsFlag.Name) {
+			cfg.Shh.RestrictConnectionBetweenLightClients = true
+		}
 		utils.RegisterShhService(stack, &cfg.Shh)
 	}

--- a/cmd/geth/main.go
+++ b/cmd/geth/main.go
@ -151,6 +151,7 @@ var (
 		utils.WhisperEnabledFlag,
 		utils.WhisperMaxMessageSizeFlag,
 		utils.WhisperMinPOWFlag,
+		utils.WhisperRestrictConnectionBetweenLightClientsFlag,
 	}

 	metricsFlags = []cli.Flag{