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::interleaved::{from_interleaved, to_interleaved};
20use nrbf_parser::records::Record;
21use std::env;
22use std::fs::File;
23use std::io::{BufReader, BufWriter, Read};
24
25fn main() -> Result<(), Box<dyn std::error::Error>> {
26    let args: Vec<String> = env::args().collect();
27    if args.len() < 2 {
28        eprintln!("Usage: {} <nrbf_file>", args[0]);
29        std::process::exit(1);
30    }
31
32    let input_path = &args[1];
33    println!("Reading original file: {}", input_path);
34    let file = File::open(input_path)?;
35    let mut original_data = Vec::new();
36    File::open(input_path)?.read_to_end(&mut original_data)?;
37
38    let reader = BufReader::new(file);
39    let mut decoder = Decoder::new(reader);
40
41    let mut records = Vec::new();
42    while let Some(record) = decoder.decode_next()? {
43        let is_end = matches!(record, Record::MessageEnd);
44        records.push(record);
45        if is_end {
46            break;
47        }
48    }
49    println!("Parsed {} records.", records.len());
50
51    // Serialize to JSON
52    let json = serde_json::to_string_pretty(&records)?;
53    let json_path = "output.json";
54    std::fs::write(json_path, &json)?;
55    println!("Saved records to {}", json_path);
56
57    // Serialize to Interleaved JSON
58    let interleaved_json = to_interleaved(&records);
59    let interleaved_json_str = serde_json::to_string_pretty(&interleaved_json)?;
60    let interleaved_path = "interleaved.json";
61    std::fs::write(interleaved_path, &interleaved_json_str)?;
62    println!("Saved interleaved records to {}", interleaved_path);
63
64    // Deserialize from JSON
65    let deserialized_records: Vec<Record> = serde_json::from_str(&json)?;
66    println!(
67        "Deserialized {} records from JSON.",
68        deserialized_records.len()
69    );
70
71    // Encode back to binary
72    let output_path = "reconstructed.meta";
73    let out_file = File::create(output_path)?;
74    let mut encoder = Encoder::new(BufWriter::new(out_file));
75
76    for record in &deserialized_records {
77        encoder.encode(record)?;
78    }
79    // Ensure everything is flushed
80    drop(encoder);
81    println!("Reconstructed binary saved to {}", output_path);
82
83    // Interleaved reconstruction check
84    println!("--- Interleaved Round Trip Check ---");
85    let interleaved_reconstructed_records = from_interleaved(interleaved_json);
86    println!(
87        "Deserialized {} records from Interleaved JSON.",
88        interleaved_reconstructed_records.len()
89    );
90
91    let interleaved_output_path = "reconstructed_interleaved.meta";
92    let int_out_file = File::create(interleaved_output_path)?;
93    let mut int_encoder = Encoder::new(BufWriter::new(int_out_file));
94
95    for record in &interleaved_reconstructed_records {
96        int_encoder.encode(record)?;
97    }
98    drop(int_encoder);
99
100    let mut int_reconstructed_data = Vec::new();
101    File::open(interleaved_output_path)?.read_to_end(&mut int_reconstructed_data)?;
102
103    if original_data == int_reconstructed_data {
104        println!("SUCCESS: Interleaved reconstructed binary is identical to original!");
105    } else {
106        println!("FAILURE: Interleaved reconstructed binary differs from original.");
107        println!(
108            "Original size: {}, Reconstructed size: {}",
109            original_data.len(),
110            int_reconstructed_data.len()
111        );
112        // Find first difference
113        let min_len = std::cmp::min(original_data.len(), int_reconstructed_data.len());
114        for i in 0..min_len {
115            if original_data[i] != int_reconstructed_data[i] {
116                println!(
117                    "First difference at offset 0x{:x}: original 0x{:02x}, reconstructed 0x{:02x}",
118                    i, original_data[i], int_reconstructed_data[i]
119                );
120                break;
121            }
122        }
123    }
124
125    Ok(())
126}
round_trip/round_trip.rs

round_trip/
round_trip.rs
