File: fuzz_tr.rs

package info (click to toggle)
rust-coreutils 0.7.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 505,620 kB
  • sloc: ansic: 103,594; asm: 28,570; sh: 8,910; python: 5,581; makefile: 472; cpp: 97; javascript: 72
file content (72 lines) | stat: -rw-r--r-- 2,083 bytes parent folder | download
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
// This file is part of the uutils coreutils package.
//
// For the full copyright and license information, please view the LICENSE
// file that was distributed with this source code.

#![no_main]
use libfuzzer_sys::fuzz_target;
use std::ffi::OsString;
use uu_tr::uumain;

use rand::Rng;

use uufuzz::{
    CommandResult, compare_result, generate_and_run_uumain, generate_random_string, run_gnu_cmd,
};
static CMD_PATH: &str = "tr";

fn generate_tr_args() -> Vec<String> {
    let mut rng = rand::rng();
    let mut args = Vec::new();

    // Translate, squeeze, and/or delete characters
    let opts = ["-c", "-d", "-s", "-t"];
    for opt in &opts {
        if rng.random_bool(0.25) {
            args.push(opt.to_string());
        }
    }

    // Generating STRING1 and optionally STRING2
    let string1 = generate_random_string(rng.random_range(1..=20));
    args.push(string1);
    if rng.random_bool(0.7) {
        // Higher chance to add STRING2 for translation
        let string2 = generate_random_string(rng.random_range(1..=20));
        args.push(string2);
    }

    args
}

fuzz_target!(|_data: &[u8]| {
    let tr_args = generate_tr_args();
    let mut args = vec![OsString::from("tr")];
    args.extend(tr_args.iter().map(OsString::from));

    let input_chars = generate_random_string(100);

    let rust_result = generate_and_run_uumain(&args, uumain, Some(&input_chars));
    let gnu_result = match run_gnu_cmd(CMD_PATH, &args[1..], false, Some(&input_chars)) {
        Ok(result) => result,
        Err(error_result) => {
            eprintln!("Failed to run GNU command:");
            eprintln!("Stderr: {}", error_result.stderr);
            eprintln!("Exit Code: {}", error_result.exit_code);
            CommandResult {
                stdout: String::new(),
                stderr: error_result.stderr,
                exit_code: error_result.exit_code,
            }
        }
    };

    compare_result(
        "tr",
        &format!("{:?}", &args[1..]),
        Some(&input_chars),
        &rust_result,
        &gnu_result,
        false,
    );
});