mirror of https://github.com/status-im/nim-eth.git
70 lines
2.7 KiB
Nim
70 lines
2.7 KiB
Nim
#
|
|
# Ethereum P2P
|
|
# (c) Copyright 2018
|
|
# Status Research & Development GmbH
|
|
#
|
|
# Licensed under either of
|
|
# Apache License, version 2.0, (LICENSE-APACHEv2)
|
|
# MIT license (LICENSE-MIT)
|
|
|
|
import
|
|
std/[sequtils, options, unittest, times],
|
|
../../eth/p2p/rlpx_protocols/whisper_protocol as whisper
|
|
|
|
suite "Whisper envelope validation":
|
|
test "should validate and allow envelope according to config":
|
|
let ttl = 1'u32
|
|
let topic = [byte 1, 2, 3, 4]
|
|
let config = WhisperConfig(powRequirement: 0, bloom: topic.topicBloom(),
|
|
isLightNode: false, maxMsgSize: defaultMaxMsgSize)
|
|
|
|
let env = Envelope(expiry:epochTime().uint32 + ttl, ttl: ttl, topic: topic,
|
|
data: repeat(byte 9, 256), nonce: 0)
|
|
check env.valid()
|
|
|
|
let msg = initMessage(env)
|
|
check msg.allowed(config)
|
|
|
|
test "should invalidate envelope due to ttl 0":
|
|
let ttl = 0'u32
|
|
let topic = [byte 1, 2, 3, 4]
|
|
let config = WhisperConfig(powRequirement: 0, bloom: topic.topicBloom(),
|
|
isLightNode: false, maxMsgSize: defaultMaxMsgSize)
|
|
|
|
let env = Envelope(expiry:epochTime().uint32 + ttl, ttl: ttl, topic: topic,
|
|
data: repeat(byte 9, 256), nonce: 0)
|
|
check env.valid() == false
|
|
|
|
test "should invalidate envelope due to expired":
|
|
let ttl = 1'u32
|
|
let topic = [byte 1, 2, 3, 4]
|
|
let config = WhisperConfig(powRequirement: 0, bloom: topic.topicBloom(),
|
|
isLightNode: false, maxMsgSize: defaultMaxMsgSize)
|
|
|
|
let env = Envelope(expiry:epochTime().uint32, ttl: ttl, topic: topic,
|
|
data: repeat(byte 9, 256), nonce: 0)
|
|
check env.valid() == false
|
|
|
|
test "should invalidate envelope due to in the future":
|
|
let ttl = 1'u32
|
|
let topic = [byte 1, 2, 3, 4]
|
|
let config = WhisperConfig(powRequirement: 0, bloom: topic.topicBloom(),
|
|
isLightNode: false, maxMsgSize: defaultMaxMsgSize)
|
|
|
|
# there is currently a 2 second tolerance, hence the + 3
|
|
let env = Envelope(expiry:epochTime().uint32 + ttl + 3, ttl: ttl, topic: topic,
|
|
data: repeat(byte 9, 256), nonce: 0)
|
|
check env.valid() == false
|
|
|
|
test "should not allow envelope due to bloom filter":
|
|
let topic = [byte 1, 2, 3, 4]
|
|
let wrongTopic = [byte 9, 8, 7, 6]
|
|
let config = WhisperConfig(powRequirement: 0, bloom: wrongTopic.topicBloom(),
|
|
isLightNode: false, maxMsgSize: defaultMaxMsgSize)
|
|
|
|
let env = Envelope(expiry:100000 , ttl: 30, topic: topic,
|
|
data: repeat(byte 9, 256), nonce: 0)
|
|
|
|
let msg = initMessage(env)
|
|
check msg.allowed(config) == false
|