mirror of
https://github.com/status-im/nim-dagger.git
synced 2025-02-28 14:20:36 +00:00
a609baea26
* adding basic retry functionality * avoid duplicate requests and batch them * fix cancelling blocks * properly resolve blocks * minor cleanup - use `self` * avoid useless asyncSpawn * track retries * limit max inflight and set libp2p maxIncomingStreams * cleanup * add basic yield in readLoop * use tuple instead of object * cleanup imports and logs * increase defaults * wip * fix prefetch batching * cleanup * decrease timeouts to speedup tests * remove outdated test * add retry tests * should track retries * remove useless test * use correct block address (index was off by 1) * remove duplicate noop proc * add BlockHandle type * Use BlockHandle type * add fetchLocal to control batching from local store * add format target * revert deps * adjust quotaMaxBytes * cleanup imports and logs * revert deps * cleanup blocks on cancelled * terminate erasure and prefetch jobs on stream end * split storing and retrieving data into separate tests * track `b.discoveryLoop` future * misc * remove useless check
68 lines
1.7 KiB
Nim
68 lines
1.7 KiB
Nim
{.push raises: [].}
|
|
|
|
import
|
|
std/[tables, hashes], pkg/results, pkg/stew/shims/net as stewNet, chronos, chronicles
|
|
|
|
import pkg/libp2p
|
|
|
|
type NatStrategy* = enum
|
|
NatAny
|
|
NatUpnp
|
|
NatPmp
|
|
NatNone
|
|
|
|
type IpLimits* = object
|
|
limit*: uint
|
|
ips: Table[IpAddress, uint]
|
|
|
|
func hash*(ip: IpAddress): Hash =
|
|
case ip.family
|
|
of IpAddressFamily.IPv6:
|
|
hash(ip.address_v6)
|
|
of IpAddressFamily.IPv4:
|
|
hash(ip.address_v4)
|
|
|
|
func inc*(ipLimits: var IpLimits, ip: IpAddress): bool =
|
|
let val = ipLimits.ips.getOrDefault(ip, 0)
|
|
if val < ipLimits.limit:
|
|
ipLimits.ips[ip] = val + 1
|
|
true
|
|
else:
|
|
false
|
|
|
|
func dec*(ipLimits: var IpLimits, ip: IpAddress) =
|
|
let val = ipLimits.ips.getOrDefault(ip, 0)
|
|
if val == 1:
|
|
ipLimits.ips.del(ip)
|
|
elif val > 1:
|
|
ipLimits.ips[ip] = val - 1
|
|
|
|
func isGlobalUnicast*(address: TransportAddress): bool =
|
|
if address.isGlobal() and address.isUnicast(): true else: false
|
|
|
|
func isGlobalUnicast*(address: IpAddress): bool =
|
|
let a = initTAddress(address, Port(0))
|
|
a.isGlobalUnicast()
|
|
|
|
proc getRouteIpv4*(): Result[IpAddress, cstring] =
|
|
# Avoiding Exception with initTAddress and can't make it work with static.
|
|
# Note: `publicAddress` is only used an "example" IP to find the best route,
|
|
# no data is send over the network to this IP!
|
|
let
|
|
publicAddress = TransportAddress(
|
|
family: AddressFamily.IPv4, address_v4: [1'u8, 1, 1, 1], port: Port(0)
|
|
)
|
|
route = getBestRoute(publicAddress)
|
|
|
|
if route.source.isUnspecified():
|
|
err("No best ipv4 route found")
|
|
else:
|
|
let ip =
|
|
try:
|
|
route.source.address()
|
|
except ValueError as e:
|
|
# This should not occur really.
|
|
error "Address conversion error", exception = e.name, msg = e.msg
|
|
return err("Invalid IP address")
|
|
ok(ip)
|