ads 0.4.4

Client for the Beckhoff Automation Device Specification protocol for PLCs
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

//! Test for the UDP client.

// use std::io::{self, Read, Write};
// use std::time::Duration;
use std::net::UdpSocket;

use crate::udp;

const OS_VERSION: &[u8] = b"\0\0\0\0\x05\0\0\0\x08\0\0\0\x09\0\0\0\x02\0\0\0T\0e\0s\0t\0\0\0";

fn udp_replier(sock: UdpSocket) {
    let mut buf = [0; 2048];
    let my_addr = crate::AmsAddr::new(crate::AmsNetId::from([1, 2, 3, 4, 5, 6]), 10000);
    loop {
        let (n, sender) = sock.recv_from(&mut buf).unwrap();
        let mut reply = udp::Message::new(udp::ServiceId::Identify, my_addr);
        if udp::Message::parse(&buf[..n], udp::ServiceId::Identify, false).is_ok() {
            reply.set_service(udp::ServiceId::Identify, true);
            reply.add_str(udp::Tag::ComputerName, "box");
            reply.add_bytes(udp::Tag::OSVersion, OS_VERSION);
            reply.add_bytes(udp::Tag::TCVersion, b"\x04\x01\x07\x00");
        } else if let Ok(msg) = udp::Message::parse(&buf[..n], udp::ServiceId::AddRoute, false) {
            reply.set_service(udp::ServiceId::AddRoute, true);
            let status = msg.get_str(udp::Tag::RouteName) != Some("route");
            reply.add_u32(udp::Tag::Status, status as u32);
        } else {
            panic!("received invalid UDP packet");
        };
        sock.send_to(reply.as_bytes(), sender).unwrap();
    }
}

#[test]
fn test_udp() {
    let serversock = UdpSocket::bind("127.0.0.1:0").unwrap();
    let port = serversock.local_addr().unwrap().port();
    let tgt_netid = crate::AmsNetId::from([1, 2, 3, 4, 5, 6]);

    std::thread::spawn(move || udp_replier(serversock));

    let netid = udp::get_netid(("127.0.0.1", port)).unwrap();
    assert_eq!(netid, tgt_netid);

    let info = udp::get_info(("127.0.0.1", port)).unwrap();
    assert_eq!(info.hostname, "box");
    assert_eq!(info.netid, tgt_netid);
    assert_eq!(info.twincat_version, (4, 1, 7));
    assert_eq!(info.os_version, ("Windows NT", 5, 8, 9, "Test".into()));

    udp::add_route(("127.0.0.1", port), tgt_netid, "a", Some("route"), None, None, false).unwrap();
    assert!(udp::add_route(("127.0.0.1", port), tgt_netid, "a", None, None, None, false).is_err());
}