config: ensure ZMQ options when read from the config file are not equal

This commit is contained in:
Wilmer Paulino 2018-08-27 22:20:16 -07:00
parent 12aadd4978
commit 58ab6c1912
No known key found for this signature in database
GPG Key ID: 6DF57B9F9514972F

@ -1076,6 +1076,17 @@ func parseRPCParams(cConfig *chainConfig, nodeConfig interface{}, net chainCode,
} }
case *bitcoindConfig: case *bitcoindConfig:
// Ensure that if the ZMQ options are set, that they are not
// equal.
if conf.ZMQPubRawBlock != "" && conf.ZMQPubRawTx != "" {
err := checkZMQOptions(
conf.ZMQPubRawBlock, conf.ZMQPubRawTx,
)
if err != nil {
return err
}
}
// If all of RPCUser, RPCPass, ZMQBlockHost, and ZMQTxHost are // If all of RPCUser, RPCPass, ZMQBlockHost, and ZMQTxHost are
// set, we assume those parameters are good to use. // set, we assume those parameters are good to use.
if conf.RPCUser != "" && conf.RPCPass != "" && if conf.RPCUser != "" && conf.RPCPass != "" &&
@ -1095,8 +1106,8 @@ func parseRPCParams(cConfig *chainConfig, nodeConfig interface{}, net chainCode,
confFile = "litecoin" confFile = "litecoin"
} }
// If only one or two of the parameters are set, we assume the // If not all of the parameters are set, we'll assume the user
// user did that unintentionally. // did this unintentionally.
if conf.RPCUser != "" || conf.RPCPass != "" || if conf.RPCUser != "" || conf.RPCPass != "" ||
conf.ZMQPubRawBlock != "" || conf.ZMQPubRawTx != "" { conf.ZMQPubRawBlock != "" || conf.ZMQPubRawTx != "" {
@ -1237,9 +1248,8 @@ func extractBitcoindRPCParams(bitcoindConfigPath string) (string, string, string
} }
zmqBlockHost := string(zmqBlockHostSubmatches[1]) zmqBlockHost := string(zmqBlockHostSubmatches[1])
zmqTxHost := string(zmqTxHostSubmatches[1]) zmqTxHost := string(zmqTxHostSubmatches[1])
if zmqBlockHost == zmqTxHost { if err := checkZMQOptions(zmqBlockHost, zmqTxHost); err != nil {
return "", "", "", "", errors.New("zmqpubrawblock and " + return "", "", "", "", err
"zmqpubrawtx must be different")
} }
// Next, we'll try to find an auth cookie. We need to detect the chain // Next, we'll try to find an auth cookie. We need to detect the chain
@ -1303,6 +1313,17 @@ func extractBitcoindRPCParams(bitcoindConfigPath string) (string, string, string
zmqBlockHost, zmqTxHost, nil zmqBlockHost, zmqTxHost, nil
} }
// checkZMQOptions ensures that the provided addresses to use as the hosts for
// ZMQ rawblock and rawtx notifications are different.
func checkZMQOptions(zmqBlockHost, zmqTxHost string) error {
if zmqBlockHost == zmqTxHost {
return errors.New("zmqpubrawblock and zmqpubrawtx must be set" +
"to different addresses")
}
return nil
}
// normalizeNetwork returns the common name of a network type used to create // normalizeNetwork returns the common name of a network type used to create
// file paths. This allows differently versioned networks to use the same path. // file paths. This allows differently versioned networks to use the same path.
func normalizeNetwork(network string) string { func normalizeNetwork(network string) string {