raiden-state-machine 0.1.0

Raiden Network implementation in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

#![warn(clippy::missing_docs_in_private_items)]

use std::collections::HashMap;

use raiden_primitives::types::{
	Address,
	ChannelIdentifier,
	TokenNetworkAddress,
};

use crate::types::{
	ChannelState,
	RouteState,
};

/// Filter out routes
pub fn prune_route_table(
	route_states: Vec<RouteState>,
	selected_route: RouteState,
	our_address: Address,
) -> Vec<RouteState> {
	route_states
		.iter()
		.filter(|route_state| {
			route_state.hop_after(our_address) == selected_route.hop_after(our_address)
		})
		.map(|route_state| RouteState {
			route: route_state.route[1..].to_vec(),
			..route_state.clone()
		})
		.collect()
}

/// Filter routes which are not blacklisted.
pub fn filter_acceptable_routes(
	route_states: Vec<RouteState>,
	blacklisted_channel_ids: Vec<ChannelIdentifier>,
	addresses_to_channels: HashMap<(TokenNetworkAddress, Address), &ChannelState>,
	token_network_address: TokenNetworkAddress,
	our_address: Address,
) -> Vec<RouteState> {
	let mut acceptable_routes = vec![];
	for route in route_states {
		let next_hop = match route.hop_after(our_address) {
			Some(next_hop) => next_hop,
			None => continue,
		};
		let channel = match addresses_to_channels.get(&(token_network_address, next_hop)) {
			Some(channel) => channel,
			None => continue,
		};
		if !blacklisted_channel_ids.contains(&channel.canonical_identifier.channel_identifier) {
			acceptable_routes.push(route);
		}
	}
	acceptable_routes
}