wacore 0.6.0

Core WhatsApp protocol implementation without runtime dependencies
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
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

use wacore_binary::builder::NodeBuilder;
use wacore_binary::marshal::{marshal, unmarshal_ref};

#[test]
fn test_simple_node_roundtrip_with_ref() {
    let original_node = NodeBuilder::new("test").build();

    // marshal() adds a leading flag byte (0 for uncompressed)
    let marshaled_with_flag = marshal(&original_node).expect("Marshal failed");

    // unmarshal_ref() expects the data *without* the flag byte.
    // It returns a borrowed `NodeRef`.
    let unmarshaled_ref = unmarshal_ref(&marshaled_with_flag[1..]).expect("unmarshal_ref failed");

    // Convert the borrowed `NodeRef` back to an owned `Node` for comparison.
    assert_eq!(original_node, unmarshaled_ref.to_owned());
}

#[test]
fn test_node_with_attributes_and_content_with_ref() {
    let original_node = NodeBuilder::new("iq")
        .attrs([("key1", "value1"), ("type", "get")])
        .bytes(b"hello world".to_vec())
        .build();

    let marshaled_with_flag = marshal(&original_node).expect("Marshal failed");
    let unmarshaled_ref = unmarshal_ref(&marshaled_with_flag[1..]).expect("unmarshal_ref failed");

    assert_eq!(original_node, unmarshaled_ref.to_owned());
}

#[test]
fn test_attr_parser_ref_zero_copy_access() {
    let original_node = NodeBuilder::new("iq")
        .attrs([("xmlns", "test"), ("type", "result"), ("id", "123")])
        .build();

    let marshaled_with_flag = marshal(&original_node).expect("Marshal failed");
    let node_ref = unmarshal_ref(&marshaled_with_flag[1..]).expect("unmarshal_ref failed");

    let mut parser = node_ref.attrs();
    assert_eq!(parser.optional_string("xmlns").as_deref(), Some("test"));
    assert_eq!(parser.optional_string("type").as_deref(), Some("result"));
    assert!(parser.ok());
    parser
        .finish()
        .expect("Expected parser to finish without errors");

    let mut parser_with_error = node_ref.attrs();
    assert!(!parser_with_error.bool("missing"));
    assert!(parser_with_error.finish().is_err());
}

#[test]
fn test_node_with_children_with_ref() {
    let child1 = NodeBuilder::new("child1").build();
    let child2 = NodeBuilder::new("child2").attr("id", "123").build();

    let parent_node = NodeBuilder::new("parent")
        .children([child1, child2])
        .build();

    let marshaled_with_flag = marshal(&parent_node).expect("Marshal failed");
    let unmarshaled_ref = unmarshal_ref(&marshaled_with_flag[1..]).expect("unmarshal_ref failed");

    assert_eq!(parent_node, unmarshaled_ref.to_owned());
}

#[test]
fn test_unmarshal_ref_known_good_data() {
    // This is the raw binary data for a <success> node, as the decoder expects it (NO flag byte).
    let success_node_binary_no_flag =
        hex::decode("f80f4c1aff0517520218905cee043dfc0366726376f7012aff88100000000000001f4becb43cff051752020041ece6fc2c643559623557784c2b6b35554b5148564936627546524751456a30475a413565767a3862365632786b64773d").expect("test data should be valid");

    // Call unmarshal_ref and inspect the borrowed view directly.
    let node_ref =
        unmarshal_ref(&success_node_binary_no_flag).expect("Should unmarshal_ref known good data");

    // Assert on the borrowed data without creating an owned Node.
    assert_eq!(node_ref.tag, "success");
    assert_eq!(
        node_ref
            .get_attr("location")
            .expect("test data should be valid")
            .as_str(),
        "frc"
    );
    assert_eq!(
        node_ref
            .get_attr("props")
            .expect("test data should be valid")
            .as_str(),
        "27"
    );
}

#[test]
fn test_unmarshal_ref_leftover_data_error() {
    let node_to_marshal = NodeBuilder::new("test").build();

    // Construct the data that unmarshal_ref will receive.
    let marshaled_with_flag = marshal(&node_to_marshal).expect("Marshal failed");

    // Remove the flag and append junk data.
    let mut raw_node_data = marshaled_with_flag[1..].to_vec();
    raw_node_data.extend_from_slice(&[1, 2, 3]);

    // Now, call unmarshal_ref with the malformed raw data.
    let result = unmarshal_ref(&raw_node_data);
    assert!(result.is_err());

    if let Err(e) = result {
        match e {
            wacore_binary::BinaryError::LeftoverData(n) => assert_eq!(n, 3),
            _ => panic!("Expected LeftoverData error, got {e:?}"),
        }
    }
}