round_trip/
round_trip.rs

1// nrbf-parser - A high-performance MS-NRBF binary parser and encoder.
2// Copyright (C) 2026  driedpampas@proton.me
3//
4// This program is free software: you can redistribute it and/or modify
5// it under the terms of the GNU General Public License as published by
6// the Free Software Foundation, either version 3 of the License, or
7// (at your option) any later version.
8//
9// This program is distributed in the hope that it will be useful,
10// but WITHOUT ANY WARRANTY; without even the implied warranty of
11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12// GNU General Public License for more details.
13//
14// You should have received a copy of the GNU General Public License
15// along with this program.  If not, see <https://www.gnu.org/licenses/>.
16
17use nrbf_parser::Decoder;
18use nrbf_parser::Encoder;
19use nrbf_parser::records::Record;
20use std::env;
21use std::fs::File;
22use std::io::{BufReader, BufWriter, Read};
23
24fn main() -> Result<(), Box<dyn std::error::Error>> {
25    let args: Vec<String> = env::args().collect();
26    if args.len() < 2 {
27        eprintln!("Usage: {} <nrbf_file>", args[0]);
28        std::process::exit(1);
29    }
30
31    let input_path = &args[1];
32    println!("Reading original file: {}", input_path);
33    let file = File::open(input_path)?;
34    let mut original_data = Vec::new();
35    File::open(input_path)?.read_to_end(&mut original_data)?;
36
37    let reader = BufReader::new(file);
38    let mut decoder = Decoder::new(reader);
39
40    let mut records = Vec::new();
41    while let Some(record) = decoder.decode_next()? {
42        let is_end = matches!(record, Record::MessageEnd);
43        records.push(record);
44        if is_end {
45            break;
46        }
47    }
48    println!("Parsed {} records.", records.len());
49
50    // Serialize to JSON
51    let json = serde_json::to_string_pretty(&records)?;
52    let json_path = "output.json";
53    std::fs::write(json_path, &json)?;
54    println!("Saved records to {}", json_path);
55
56    // Deserialize from JSON
57    let deserialized_records: Vec<Record> = serde_json::from_str(&json)?;
58    println!(
59        "Deserialized {} records from JSON.",
60        deserialized_records.len()
61    );
62
63    // Encode back to binary
64    let output_path = "reconstructed.meta";
65    let out_file = File::create(output_path)?;
66    let mut encoder = Encoder::new(BufWriter::new(out_file));
67
68    for record in &deserialized_records {
69        encoder.encode(record)?;
70    }
71    // Ensure everything is flushed
72    drop(encoder);
73    println!("Reconstructed binary saved to {}", output_path);
74
75    // Compare
76    let mut reconstructed_data = Vec::new();
77    File::open(output_path)?.read_to_end(&mut reconstructed_data)?;
78
79    if original_data == reconstructed_data {
80        println!("SUCCESS: Reconstructed binary is identical to original!");
81    } else {
82        println!("FAILURE: Reconstructed binary differs from original.");
83        println!(
84            "Original size: {}, Reconstructed size: {}",
85            original_data.len(),
86            reconstructed_data.len()
87        );
88
89        // Find first difference
90        let min_len = std::cmp::min(original_data.len(), reconstructed_data.len());
91        for i in 0..min_len {
92            if original_data[i] != reconstructed_data[i] {
93                println!(
94                    "First difference at offset 0x{:x}: original 0x{:02x}, reconstructed 0x{:02x}",
95                    i, original_data[i], reconstructed_data[i]
96                );
97                break;
98            }
99        }
100    }
101
102    Ok(())
103}
round_trip/round_trip.rs

round_trip/
round_trip.rs
