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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
use bytes_kman::TBytes;
use crate::common::packets::{Packets, Search, SearchResponse, SearchType};
use super::{ClientStage, RelayServer};
impl RelayServer {
pub(crate) fn on_search(&mut self, index: usize, search: Search) {
let session;
if let Some(client) = self.clients.get(index) {
session = client.session;
} else {
return;
}
let mut adresses = Vec::new();
for client in self.clients.iter() {
if let ClientStage::Registered(rclient) = &client.stage {
let mut valid = true;
match &search.name {
SearchType::Fuzzy(name) => {
if !rclient.name.contains(name) {
break;
}
valid = false;
}
SearchType::Exact(name) => {
if rclient.name != *name {
valid = false;
}
}
SearchType::None => {}
};
match &search.client {
SearchType::Fuzzy(name) => {
if !rclient.client.contains(name) {
break;
}
valid = false;
}
SearchType::Exact(name) => {
if rclient.client != *name {
valid = false;
}
}
SearchType::None => {}
};
match &search.other {
SearchType::Fuzzy(name) => {
let mut finds = 0usize;
for by in &rclient.other {
if let Some(b) = name.get(finds) {
if *by == *b {
finds += 1;
} else {
finds = 0;
}
if finds == rclient.other.len() {
break;
}
} else {
break;
}
}
if name.len() < finds {
valid = false
}
}
SearchType::Exact(name) => {
if rclient.other != *name {
valid = false;
}
}
SearchType::None => {}
};
if valid {
adresses.push(rclient.adress.clone());
}
}
}
let pak = Packets::SearchResponse(SearchResponse { session, adresses });
let mut bytes = pak.to_bytes();
bytes.reverse();
let _ = self.clients.get_mut(index).unwrap().conn.send(&bytes);
}
}