diff --git a/tools/mix/mix_relay_dht.nim b/tools/mix/mix_relay_dht.nim index 0f20f933..0721b117 100644 --- a/tools/mix/mix_relay_dht.nim +++ b/tools/mix/mix_relay_dht.nim @@ -431,11 +431,6 @@ proc runWithDhtProxy( config = discoveryConfig, ) - let maxReplyBytes = getMaxMessageSizeForCodec(DhtProxyCodec, 0).valueOr: - raise - newException(ValueError, "DhtProxyCodec does not fit Sphinx payload: " & error) - mixProto.registerDestReadBehavior(DhtProxyCodec, readLp(maxReplyBytes)) - let proxyProto = DhtProxyProtocol.new(dht, maxInFlight = conf.maxInFlight) try: @@ -573,6 +568,11 @@ proc run(conf: Conf) {.async: (raises: [CatchableError]).} = let mixProto = MixProtocol.new(nodeInfo, switch) + let maxReplyBytes = getMaxMessageSizeForCodec(DhtProxyCodec, 0).valueOr: + raise + newException(ValueError, "DhtProxyCodec does not fit Sphinx payload: " & error) + mixProto.registerDestReadBehavior(DhtProxyCodec, readLp(maxReplyBytes)) + if conf.noDhtProxy: await runRelayOnly(conf, switch, mixProto, peerId, tcpAddr) else: