Files
adcs_api
cbor_protocol
channel_protocol
clyde_3g_eps_api
clyde_3g_eps_service
comms_service
db_test
eps_api
example_rust_c_service
example_rust_service
extern_lib
file_protocol
file_service
iobc_supervisor_service
isis_ants
isis_ants_api
isis_ants_service
isis_imtq_api
isis_iobc_supervisor
kubos_app
kubos_app_service
kubos_build_helper
kubos_file_client
kubos_service
kubos_shell_client
kubos_system
kubos_telemetry_db
large_download
large_upload
local_comms_service
mai400
mai400_api
mai400_service
monitor_service
novatel_oem6_api
novatel_oem6_service
nsl_duplex_d2
nsl_duplex_d2_comms_service
obc_hs
radio_api
rust_i2c
rust_mission_app
rust_uart
scheduler_service
serial_comms_service
shell_protocol
shell_service
telemetry_service
uart_comms_client
udp_client
utils
 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
//
// Copyright (C) 2018 Kubos Corporation
//
// Licensed under the Apache License, Version 2.0 (the "License")
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// Contributed by: William Greer (wgreer184@gmail.com) and Sam Justice (sam.justice1@gmail.com)
//

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(())
    }
}

// Function to allow reading from a UDP socket.
pub fn read(socket: &Arc<Mutex<LocalComms>>) -> CommsResult<Vec<u8>> {
    if let Ok(socket) = socket.lock() {
        socket.read()
    } else {
        bail!("Failed to lock socket")
    }
}

// Function to allow writing over a UDP 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")
    }
}