rpc: check module availability at startup (ethereum#20597)

cmd/XDC/config.go

@@ -120,8 +120,8 @@ func defaultNodeConfig() node.Config {
 	cfg := node.DefaultConfig
 	cfg.Name = clientIdentifier
 	cfg.Version = params.VersionWithCommit(gitCommit)
-	cfg.HTTPModules = append(cfg.HTTPModules, "eth", "shh")
-	cfg.WSModules = append(cfg.WSModules, "eth", "shh")
+	cfg.HTTPModules = append(cfg.HTTPModules, "eth")
+	cfg.WSModules = append(cfg.WSModules, "eth")
 	cfg.IPCPath = "XDC.ipc"
 	return cfg
 }

rpc/endpoints.go

@@ -17,14 +17,40 @@
 package rpc
 
 import (
+	"fmt"
 	"net"
 	"strings"
 
 	"github.com/XinFinOrg/XDPoSChain/log"
 )
 
-// StartHTTPEndpoint starts the HTTP RPC endpoint, configured with cors/vhosts/modules
+// checkModuleAvailability check that all names given in modules are actually
+// available API services.
+func checkModuleAvailability(modules []string, apis []API) error {
+	available := make(map[string]struct{})
+	var availableNames string
+	for i, api := range apis {
+		if _, ok := available[api.Namespace]; !ok {
+			available[api.Namespace] = struct{}{}
+			if i > 0 {
+				availableNames += ", "
+			}
+			availableNames += api.Namespace
+		}
+	}
+	for _, name := range modules {
+		if _, ok := available[name]; !ok {
+			return fmt.Errorf("invalid API %q in whitelist (available: %s)", name, availableNames)
+		}
+	}
+	return nil
+}
+
+// StartHTTPEndpoint starts the HTTP RPC endpoint, configured with cors/vhosts/modules.
 func StartHTTPEndpoint(endpoint string, apis []API, modules []string, cors []string, vhosts []string, timeouts HTTPTimeouts) (net.Listener, *Server, error) {
+	if err := checkModuleAvailability(modules, apis); err != nil {
+		return nil, nil, err
+	}
 	// Generate the whitelist based on the allowed modules
 	whitelist := make(map[string]bool)
 	for _, module := range modules {
@@ -52,8 +78,11 @@ func StartHTTPEndpoint(endpoint string, apis []API, modules []string, cors []str
 	return listener, handler, err
 }
 
-// StartWSEndpoint starts a websocket endpoint
+// StartWSEndpoint starts a websocket endpoint.
 func StartWSEndpoint(endpoint string, apis []API, modules []string, wsOrigins []string, exposeAll bool) (net.Listener, *Server, error) {
+	if err := checkModuleAvailability(modules, apis); err != nil {
+		return nil, nil, err
+	}
 	// Generate the whitelist based on the allowed modules
 	whitelist := make(map[string]bool)
 	for _, module := range modules {