File: fragmentizer.rs

package info (click to toggle)
rust-imap-codec 2.0.0~alpha6-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 3,952 kB
  • sloc: makefile: 2; sh: 1
file content (42 lines) | stat: -rw-r--r-- 1,365 bytes parent folder | download | duplicates (2)
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
use std::io::{Read, stdin};

use imap_codec::{fragmentizer::Fragmentizer, imap_types::utils::escape_byte_string};

fn main() {
    let mut fragmentizer = Fragmentizer::new(1024);

    loop {
        match fragmentizer.progress() {
            Some(fragment_info) => {
                println!(
                    "[!] Fragment: {fragment_info:#?} // b\"{}\"",
                    escape_byte_string(fragmentizer.fragment_bytes(fragment_info))
                );

                if fragmentizer.is_message_complete() {
                    println!(
                        "[!] Complete message: {}",
                        escape_byte_string(fragmentizer.message_bytes())
                    );
                }
            }
            None => {
                println!("[!] Reading stdin (ctrl+d to flush)...");
                let mut buffer = [0; 64];
                let count = stdin().read(&mut buffer).unwrap();
                if count == 0 {
                    println!("[!] Connection closed");
                    break;
                }
                let chunk = &buffer[..count];

                println!(
                    "[!] Enqueueing {} byte(s) (b\"{}\")",
                    count,
                    escape_byte_string(chunk)
                );
                fragmentizer.enqueue_bytes(chunk);
            }
        }
    }
}