Struct Encoder

Source

pub struct Encoder<W: Write> { /* private fields */ }

Expand description

An encoder for MS-NRBF binary streams.

Implementations§

Source §

impl<W: Write> Encoder<W>

Source

pub fn new(writer: W) -> Self

Creates a new encoder from a writer.

Examples found in repository ?

examples/round_trip.rs (line 66)

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}

Source

pub fn encode(&mut self, record: &Record) -> Result<()>

Encodes a record and writes it to the stream.

Examples found in repository ?

examples/round_trip.rs (line 69)

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}