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
use comms_service::CommsResult;
use failure::bail;
use std::net::UdpSocket;
use std::sync::{Arc, Mutex};
pub struct LocalComms {
pub socket: UdpSocket,
pub gateway_ip: String,
pub gateway_port: u16,
}
impl LocalComms {
pub fn new(
listening_ip: &str,
listening_port: u16,
gateway_ip: &str,
gateway_port: u16,
) -> CommsResult<Self> {
let socket = UdpSocket::bind((listening_ip, listening_port))?;
Ok(LocalComms {
socket,
gateway_ip: gateway_ip.to_owned(),
gateway_port,
})
}
pub fn read(&self) -> CommsResult<Vec<u8>> {
let mut buf = [0; 4096];
let (size, _) = self.socket.recv_from(&mut buf)?;
Ok(buf[0..size].to_vec())
}
pub fn write(&self, data: &[u8]) -> CommsResult<()> {
self.socket
.send_to(data, (self.gateway_ip.as_str(), self.gateway_port))?;
Ok(())
}
}
pub fn read(socket: &Arc<Mutex<LocalComms>>) -> CommsResult<Vec<u8>> {
if let Ok(socket) = socket.lock() {
socket.read()
} else {
bail!("Failed to lock socket")
}
}
pub fn write(socket: &Arc<Mutex<LocalComms>>, data: &[u8]) -> CommsResult<()> {
if let Ok(socket) = socket.lock() {
socket.write(data)
} else {
bail!("Failed to lock socket")
}
}