probe-rs-tools 0.29.1

A collection of on chip debugging tools to communicate with microchips.
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

use std::io::Write;
use std::net::{SocketAddr, TcpStream};

#[derive(Debug)]
pub struct TcpPublisher {
    pub address: SocketAddr,
    pub socket: Option<TcpStream>,
}

impl TcpPublisher {
    pub fn new(address: SocketAddr) -> Self {
        Self {
            address,
            socket: None,
        }
    }

    pub fn send(&mut self, bytes: &[u8]) {
        if self.socket.is_none() {
            // Try to connect if there is no socket
            if let Ok(stream) = TcpStream::connect(self.address) {
                self.socket = Some(stream);
            }
        }

        if let Some(socket) = self.socket.as_mut() {
            if socket.write_all(bytes).is_err() {
                // Discard socket on error. Try reconnect next time.
                self.socket = None;
            }
        }
    }
}