diff options
| author | Daniel Patterson <dbp@riseup.net> | 2012-10-03 20:32:30 -0400 |
|---|---|---|
| committer | Tim Chevalier <chevalier@alum.wellesley.edu> | 2012-11-03 18:23:43 -0700 |
| commit | 6c79d788439de8446272a81f3aae6cc4d1d4ddd4 (patch) | |
| tree | 4d364a928a5ffba946371d4b19f45a77dae5e81c /src/libfuzzer | |
| parent | 563aa0bf6c932b2150557b335f9a40675639c323 (diff) | |
library-ifying of tools - turning cargo, rustdoc, fuzzer, into libraries, as per #3543
Diffstat (limited to 'src/libfuzzer')
| -rw-r--r-- | src/libfuzzer/ast_match.rs | 30 | ||||
| -rw-r--r-- | src/libfuzzer/cycles.rs | 96 | ||||
| -rw-r--r-- | src/libfuzzer/fuzzer.rc | 32 | ||||
| -rw-r--r-- | src/libfuzzer/fuzzer.rs | 628 | ||||
| -rw-r--r-- | src/libfuzzer/ivec_fuzz.rs | 110 | ||||
| -rw-r--r-- | src/libfuzzer/rand_util.rs | 96 |
6 files changed, 992 insertions, 0 deletions
diff --git a/src/libfuzzer/ast_match.rs b/src/libfuzzer/ast_match.rs new file mode 100644 index 00000000000..611402476f9 --- /dev/null +++ b/src/libfuzzer/ast_match.rs @@ -0,0 +1,30 @@ +use std; +use vec; + +fn vec_equal<T>(v: ~[T], u: ~[T], + element_equality_test: fn@(&&T, &&T) -> bool) -> + bool { + let Lv = vec::len(v); + if Lv != vec::len(u) { return false; } + let i = 0u; + while i < Lv { + if !element_equality_test(v[i], u[i]) { return false; } + i += 1u; + } + return true; +} + +pure fn builtin_equal<T>(&&a: T, &&b: T) -> bool { return a == b; } +pure fn builtin_equal_int(&&a: int, &&b: int) -> bool { return a == b; } + +fn main() { + assert (builtin_equal(5, 5)); + assert (!builtin_equal(5, 4)); + assert (!vec_equal(~[5, 5], ~[5], bind builtin_equal(_, _))); + assert (!vec_equal(~[5, 5], ~[5], builtin_equal_int)); + assert (!vec_equal(~[5, 5], ~[5, 4], builtin_equal_int)); + assert (!vec_equal(~[5, 5], ~[4, 5], builtin_equal_int)); + assert (vec_equal(~[5, 5], ~[5, 5], builtin_equal_int)); + + error!("Pass"); +} diff --git a/src/libfuzzer/cycles.rs b/src/libfuzzer/cycles.rs new file mode 100644 index 00000000000..ec263ead954 --- /dev/null +++ b/src/libfuzzer/cycles.rs @@ -0,0 +1,96 @@ +use std; +use std::rand; +use uint::range; + +// random uint less than n +fn under(r : rand::rng, n : uint) -> uint { + assert n != 0u; r.next() as uint % n +} + +// random choice from a vec +fn choice<T: copy>(r : rand::rng, v : ~[const T]) -> T { + assert vec::len(v) != 0u; v[under(r, vec::len(v))] +} + +// k in n chance of being true +fn likelihood(r : rand::rng, k : uint, n : uint) -> bool { under(r, n) < k } + + +const iters : uint = 1000u; +const vlen : uint = 100u; + +enum maybe_pointy { + none, + p(@pointy) +} + +type pointy = { + mut a : maybe_pointy, + mut b : ~maybe_pointy, + mut c : @maybe_pointy, + + mut f : fn@()->(), + mut g : fn~()->(), + + mut m : ~[maybe_pointy], + mut n : ~[mut maybe_pointy], + mut o : {x : int, y : maybe_pointy} +}; +// To add: objects; traits; anything type-parameterized? + +fn empty_pointy() -> @pointy { + return @{ + mut a : none, + mut b : ~none, + mut c : @none, + + mut f : fn@()->(){}, + mut g : fn~()->(){}, + + mut m : ~[], + mut n : ~[mut], + mut o : {x : 0, y : none} + } +} + +fn nopP(_x : @pointy) { } +fn nop<T>(_x: T) { } + +fn test_cycles(r : rand::rng, k: uint, n: uint) +{ + let v : ~[mut @pointy] = ~[mut]; + + // Create a graph with no edges + range(0u, vlen) {|_i| + v.push(empty_pointy()); + } + + // Fill in the graph with random edges, with density k/n + range(0u, vlen) {|i| + if (likelihood(r, k, n)) { v[i].a = p(choice(r, v)); } + if (likelihood(r, k, n)) { v[i].b = ~p(choice(r, v)); } + if (likelihood(r, k, n)) { v[i].c = @p(choice(r, v)); } + + if (likelihood(r, k, n)) { v[i].f = bind nopP(choice(r, v)); } + //if (false) { v[i].g = bind (fn~(_x: @pointy) { })( + // choice(r, v)); } + // https://github.com/mozilla/rust/issues/1899 + + if (likelihood(r, k, n)) { v[i].m = [p(choice(r, v))]; } + if (likelihood(r, k, n)) { v[i].n.push(mut p(choice(r, v))); } + if (likelihood(r, k, n)) { v[i].o = {x: 0, y: p(choice(r, v))}; } + } + + // Drop refs one at a time + range(0u, vlen) {|i| + v[i] = empty_pointy() + } +} + +fn main() +{ + let r = rand::rng(); + range(0u, iters) {|i| + test_cycles(r, i, iters); + } +} diff --git a/src/libfuzzer/fuzzer.rc b/src/libfuzzer/fuzzer.rc new file mode 100644 index 00000000000..11456ba3e1a --- /dev/null +++ b/src/libfuzzer/fuzzer.rc @@ -0,0 +1,32 @@ +// -*- rust -*- + +#[link(name = "fuzzer", + vers = "0.5", + uuid = "d6418797-2736-4833-bd82-d3c684b7c1b0", + url = "https://github.com/mozilla/rust/tree/master/src/libfuzzer")]; + +#[comment = "The Rust fuzzer library"]; +#[license = "MIT"]; +#[crate_type = "lib"]; +#[no_core]; + +#[legacy_modes]; +#[legacy_exports]; + +#[allow(vecs_implicitly_copyable)]; +#[allow(non_camel_case_types)]; +#[allow(deprecated_mode)]; +#[allow(deprecated_pattern)]; + +extern mod core(vers = "0.5"); +extern mod std(vers = "0.5"); +extern mod syntax(vers = "0.5"); + +use core::*; + +// Local Variables: +// fill-column: 78; +// indent-tabs-mode: nil +// c-basic-offset: 4 +// buffer-file-coding-system: utf-8-unix +// End: diff --git a/src/libfuzzer/fuzzer.rs b/src/libfuzzer/fuzzer.rs new file mode 100644 index 00000000000..a4968382cf4 --- /dev/null +++ b/src/libfuzzer/fuzzer.rs @@ -0,0 +1,628 @@ +#[legacy_exports]; + +use io::WriterUtil; + +use syntax::{ast, ast_util, fold, visit, codemap}; +use syntax::parse; +use syntax::print::pprust; +use syntax::diagnostic; + +enum test_mode { tm_converge, tm_run, } +type context = { mode: test_mode }; // + rng + +impl test_mode : cmp::Eq { + pure fn eq(other: &test_mode) -> bool { + (self as uint) == ((*other) as uint) + } + pure fn ne(other: &test_mode) -> bool { !self.eq(other) } +} + +fn write_file(filename: &Path, content: ~str) { + result::get( + &io::file_writer(filename, ~[io::Create, io::Truncate])) + .write_str(content); +} + +fn contains(haystack: ~str, needle: ~str) -> bool { + str::contains(haystack, needle) +} + +fn find_rust_files(files: &mut ~[Path], path: &Path) { + if path.filetype() == Some(~".rs") && !contains(path.to_str(), ~"utf8") { + // ignoring "utf8" tests because something is broken + files.push(*path); + } else if os::path_is_dir(path) + && !contains(path.to_str(), ~"compile-fail") + && !contains(path.to_str(), ~"build") { + for os::list_dir_path(path).each |p| { + find_rust_files(files, *p); + } + } +} + + +fn common_exprs() -> ~[ast::expr] { + fn dse(e: ast::expr_) -> ast::expr { + { id: 0, callee_id: -1, node: e, span: ast_util::dummy_sp() } + } + + fn dsl(l: ast::lit_) -> ast::lit { + { node: l, span: ast_util::dummy_sp() } + } + + ~[dse(ast::expr_break(option::None)), + dse(ast::expr_again(option::None)), + dse(ast::expr_fail(option::None)), + dse(ast::expr_fail(option::Some( + @dse(ast::expr_lit(@dsl(ast::lit_str(@~"boo"))))))), + dse(ast::expr_ret(option::None)), + dse(ast::expr_lit(@dsl(ast::lit_nil))), + dse(ast::expr_lit(@dsl(ast::lit_bool(false)))), + dse(ast::expr_lit(@dsl(ast::lit_bool(true)))), + dse(ast::expr_unary(ast::box(ast::m_imm), + @dse(ast::expr_lit(@dsl(ast::lit_bool(true)))))), + dse(ast::expr_unary(ast::uniq(ast::m_imm), + @dse(ast::expr_lit(@dsl(ast::lit_bool(true)))))) + ] +} + +pure fn safe_to_steal_expr(e: @ast::expr, tm: test_mode) -> bool { + safe_to_use_expr(*e, tm) +} + +pure fn safe_to_use_expr(e: ast::expr, tm: test_mode) -> bool { + match tm { + tm_converge => { + match e.node { + // If the fuzzer moves a block-ending-in-semicolon into callee + // position, the pretty-printer can't preserve this even by + // parenthesizing!! See email to marijn. + ast::expr_if(*) | ast::expr_block(*) + | ast::expr_match(*) | ast::expr_while(*) => { false } + + // https://github.com/mozilla/rust/issues/929 + ast::expr_cast(*) | ast::expr_assert(*) | + ast::expr_binary(*) | ast::expr_assign(*) | + ast::expr_assign_op(*) => { false } + + ast::expr_fail(option::None) | + ast::expr_ret(option::None) => { false } + + // https://github.com/mozilla/rust/issues/953 + ast::expr_fail(option::Some(_)) => { false } + + // https://github.com/mozilla/rust/issues/928 + //ast::expr_cast(_, _) { false } + + // https://github.com/mozilla/rust/issues/1458 + ast::expr_call(_, _, _) => { false } + + _ => { true } + } + } + tm_run => { true } + } +} + +fn safe_to_steal_ty(t: @ast::Ty, tm: test_mode) -> bool { + // Restrictions happen to be the same. + safe_to_replace_ty(t.node, tm) +} + +// Not type-parameterized: https://github.com/mozilla/rust/issues/898 (FIXED) +fn stash_expr_if(c: fn@(@ast::expr, test_mode)->bool, + es: @mut ~[ast::expr], + e: @ast::expr, + tm: test_mode) { + if c(e, tm) { + *es += ~[*e]; + } else {/* now my indices are wrong :( */ } +} + +fn stash_ty_if(c: fn@(@ast::Ty, test_mode)->bool, + es: @mut ~[ast::Ty], + e: @ast::Ty, + tm: test_mode) { + if c(e, tm) { + es.push(*e); + } else {/* now my indices are wrong :( */ } +} + +type stolen_stuff = {exprs: ~[ast::expr], tys: ~[ast::Ty]}; + +fn steal(crate: ast::crate, tm: test_mode) -> stolen_stuff { + let exprs = @mut ~[]; + let tys = @mut ~[]; + let v = visit::mk_simple_visitor(@{ + visit_expr: |a| stash_expr_if(safe_to_steal_expr, exprs, a, tm), + visit_ty: |a| stash_ty_if(safe_to_steal_ty, tys, a, tm), + .. *visit::default_simple_visitor() + }); + visit::visit_crate(crate, (), v); + {exprs: *exprs, tys: *tys} +} + + +fn safe_to_replace_expr(e: ast::expr_, _tm: test_mode) -> bool { + match e { + // https://github.com/mozilla/rust/issues/652 + ast::expr_if(*) => { false } + ast::expr_block(_) => { false } + + // expr_call is also missing a constraint + ast::expr_fn_block(*) => { false } + + _ => { true } + } +} + +fn safe_to_replace_ty(t: ast::ty_, _tm: test_mode) -> bool { + match t { + ast::ty_infer => { false } // always implicit, always top level + ast::ty_bot => { false } // in source, can only appear + // as the out type of a function + ast::ty_mac(_) => { false } + _ => { true } + } +} + +// Replace the |i|th expr (in fold order) of |crate| with |newexpr|. +fn replace_expr_in_crate(crate: ast::crate, i: uint, + newexpr: ast::expr, tm: test_mode) -> + ast::crate { + let j: @mut uint = @mut 0u; + fn fold_expr_rep(j_: @mut uint, i_: uint, newexpr_: ast::expr_, + original: ast::expr_, fld: fold::ast_fold, + tm_: test_mode) -> + ast::expr_ { + *j_ += 1u; + if i_ + 1u == *j_ && safe_to_replace_expr(original, tm_) { + newexpr_ + } else { + fold::noop_fold_expr(original, fld) + } + } + let afp = @{ + fold_expr: fold::wrap(|a,b| { + fold_expr_rep(j, i, newexpr.node, a, b, tm) + }), + .. *fold::default_ast_fold() + }; + let af = fold::make_fold(afp); + let crate2: @ast::crate = @af.fold_crate(crate); + *crate2 +} + + +// Replace the |i|th ty (in fold order) of |crate| with |newty|. +fn replace_ty_in_crate(crate: ast::crate, i: uint, newty: ast::Ty, + tm: test_mode) -> ast::crate { + let j: @mut uint = @mut 0u; + fn fold_ty_rep(j_: @mut uint, i_: uint, newty_: ast::ty_, + original: ast::ty_, fld: fold::ast_fold, + tm_: test_mode) -> + ast::ty_ { + *j_ += 1u; + if i_ + 1u == *j_ && safe_to_replace_ty(original, tm_) { + newty_ + } else { fold::noop_fold_ty(original, fld) } + } + let afp = @{ + fold_ty: fold::wrap(|a,b| fold_ty_rep(j, i, newty.node, a, b, tm) ), + .. *fold::default_ast_fold() + }; + let af = fold::make_fold(afp); + let crate2: @ast::crate = @af.fold_crate(crate); + *crate2 +} + +fn under(n: uint, it: fn(uint)) { + let mut i: uint = 0u; + while i < n { it(i); i += 1u; } +} + +fn as_str(f: fn@(+x: io::Writer)) -> ~str { + io::with_str_writer(f) +} + +fn check_variants_of_ast(crate: ast::crate, codemap: codemap::CodeMap, + filename: &Path, cx: context) { + let stolen = steal(crate, cx.mode); + let extra_exprs = vec::filter(common_exprs(), + |a| safe_to_use_expr(*a, cx.mode) ); + check_variants_T(crate, codemap, filename, ~"expr", + extra_exprs + stolen.exprs, pprust::expr_to_str, + replace_expr_in_crate, cx); + check_variants_T(crate, codemap, filename, ~"ty", stolen.tys, + pprust::ty_to_str, replace_ty_in_crate, cx); +} + +fn check_variants_T<T: Copy>( + crate: ast::crate, + codemap: codemap::CodeMap, + filename: &Path, + thing_label: ~str, + things: ~[T], + stringifier: fn@(@T, @syntax::parse::token::ident_interner) -> ~str, + replacer: fn@(ast::crate, uint, T, test_mode) -> ast::crate, + cx: context + ) { + error!("%s contains %u %s objects", filename.to_str(), + things.len(), thing_label); + + // Assuming we're not generating any token_trees + let intr = syntax::parse::token::mk_fake_ident_interner(); + + let L = things.len(); + + if L < 100 { + do under(uint::min(L, 20)) |i| { + log(error, ~"Replacing... #" + uint::str(i)); + let fname = str::from_slice(filename.to_str()); + do under(uint::min(L, 30)) |j| { + log(error, ~"With... " + stringifier(@things[j], intr)); + let crate2 = @replacer(crate, i, things[j], cx.mode); + // It would be best to test the *crate* for stability, but + // testing the string for stability is easier and ok for now. + let handler = diagnostic::mk_handler(None); + let str3 = do io::with_str_reader("") |rdr| { + @as_str(|a|pprust::print_crate( + codemap, + intr, + diagnostic::mk_span_handler(handler, codemap), + crate2, + fname, + rdr, a, + pprust::no_ann(), + false)) + }; + match cx.mode { + tm_converge => { + check_roundtrip_convergence(str3, 1u); + } + tm_run => { + let file_label = fmt!("rusttmp/%s_%s_%u_%u", + last_part(filename.to_str()), + thing_label, i, j); + let safe_to_run = !(content_is_dangerous_to_run(*str3) + || has_raw_pointers(*crate2)); + check_whole_compiler(*str3, &Path(file_label), + safe_to_run); + } + } + } + } + } +} + +fn last_part(filename: ~str) -> ~str { + let ix = option::get(str::rfind_char(filename, '/')); + str::slice(filename, ix + 1u, str::len(filename) - 3u) +} + +enum happiness { + passed, + cleanly_rejected(~str), + known_bug(~str), + failed(~str), +} + +// We'd find more bugs if we could take an AST here, but +// - that would find many "false positives" or unimportant bugs +// - that would be tricky, requiring use of tasks or serialization +// or randomness. +// This seems to find plenty of bugs as it is :) +fn check_whole_compiler(code: ~str, suggested_filename_prefix: &Path, + allow_running: bool) { + let filename = &suggested_filename_prefix.with_filetype("rs"); + write_file(filename, code); + + let compile_result = check_compiling(filename); + + let run_result = match (compile_result, allow_running) { + (passed, true) => { check_running(suggested_filename_prefix) } + (h, _) => { h } + }; + + match run_result { + passed | cleanly_rejected(_) | known_bug(_) => { + removeIfExists(suggested_filename_prefix); + removeIfExists(&suggested_filename_prefix.with_filetype("rs")); + removeDirIfExists(&suggested_filename_prefix.with_filetype("dSYM")); + } + failed(s) => { + log(error, ~"check_whole_compiler failure: " + s); + log(error, ~"Saved as: " + filename.to_str()); + } + } +} + +fn removeIfExists(filename: &Path) { + // So sketchy! + assert !contains(filename.to_str(), ~" "); + run::program_output(~"bash", ~[~"-c", ~"rm " + filename.to_str()]); +} + +fn removeDirIfExists(filename: &Path) { + // So sketchy! + assert !contains(filename.to_str(), ~" "); + run::program_output(~"bash", ~[~"-c", ~"rm -r " + filename.to_str()]); +} + +fn check_running(exe_filename: &Path) -> happiness { + let p = run::program_output( + ~"/Users/jruderman/scripts/timed_run_rust_program.py", + ~[exe_filename.to_str()]); + let comb = p.out + ~"\n" + p.err; + if str::len(comb) > 1u { + log(error, ~"comb comb comb: " + comb); + } + + if contains(comb, ~"Assertion failed:") { + failed(~"C++ assertion failure") + } else if contains(comb, ~"leaked memory in rust main loop") { + // might also use exit code 134 + //failed("Leaked") + known_bug(~"https://github.com/mozilla/rust/issues/910") + } else if contains(comb, ~"src/rt/") { + failed(~"Mentioned src/rt/") + } else if contains(comb, ~"malloc") { + failed(~"Mentioned malloc") + } else { + match p.status { + 0 => { passed } + 100 => { cleanly_rejected(~"running: explicit fail") } + 101 | 247 => { cleanly_rejected(~"running: timed out") } + 245 | 246 | 138 | 252 => { + known_bug(~"https://github.com/mozilla/rust/issues/1466") + } + 136 | 248 => { + known_bug( + ~"SIGFPE - https://github.com/mozilla/rust/issues/944") + } + rc => { + failed(~"Rust program ran but exited with status " + + int::str(rc)) + } + } + } +} + +fn check_compiling(filename: &Path) -> happiness { + let p = run::program_output( + ~"/Users/jruderman/code/rust/build/x86_64-apple-darwin/\ + stage1/bin/rustc", + ~[filename.to_str()]); + + //error!("Status: %d", p.status); + if p.status == 0 { + passed + } else if p.err != ~"" { + if contains(p.err, ~"error:") { + cleanly_rejected(~"rejected with span_error") + } else { + log(error, ~"Stderr: " + p.err); + failed(~"Unfamiliar error message") + } + } else if contains(p.out, ~"Assertion") && contains(p.out, ~"failed") { + log(error, ~"Stdout: " + p.out); + failed(~"Looks like an llvm assertion failure") + } else if contains(p.out, ~"internal compiler error unimplemented") { + known_bug(~"Something unimplemented") + } else if contains(p.out, ~"internal compiler error") { + log(error, ~"Stdout: " + p.out); + failed(~"internal compiler error") + + } else { + log(error, p.status); + log(error, ~"!Stdout: " + p.out); + failed(~"What happened?") + } +} + + +fn parse_and_print(code: @~str) -> ~str { + let filename = Path("tmp.rs"); + let sess = parse::new_parse_sess(option::None); + write_file(&filename, *code); + let crate = parse::parse_crate_from_source_str( + filename.to_str(), code, ~[], sess); + do io::with_str_reader(*code) |rdr| { + as_str(|a| + pprust::print_crate( + sess.cm, + // Assuming there are no token_trees + syntax::parse::token::mk_fake_ident_interner(), + sess.span_diagnostic, + crate, + filename.to_str(), + rdr, a, + pprust::no_ann(), + false) ) + } +} + +fn has_raw_pointers(c: ast::crate) -> bool { + let has_rp = @mut false; + fn visit_ty(flag: @mut bool, t: @ast::Ty) { + match t.node { + ast::ty_ptr(_) => { *flag = true; } + _ => { } + } + } + let v = + visit::mk_simple_visitor(@{visit_ty: |a| visit_ty(has_rp, a), + .. *visit::default_simple_visitor()}); + visit::visit_crate(c, (), v); + return *has_rp; +} + +fn content_is_dangerous_to_run(code: ~str) -> bool { + let dangerous_patterns = + ~[~"xfail-test", + ~"import", // espeically fs, run + ~"extern", + ~"unsafe", + ~"log"]; // python --> rust pipe deadlock? + + for dangerous_patterns.each |p| { if contains(code, *p) { return true; } } + return false; +} + +fn content_is_dangerous_to_compile(code: ~str) -> bool { + let dangerous_patterns = + ~[~"xfail-test"]; + + for dangerous_patterns.each |p| { if contains(code, *p) { return true; } } + return false; +} + +fn content_might_not_converge(code: ~str) -> bool { + let confusing_patterns = + ~[~"xfail-test", + ~"xfail-pretty", + ~"self", // crazy rules enforced by parser not typechecker? + ~"spawn", // precedence issues? + ~"bind", // precedence issues? + ~" be ", // don't want to replace its child with a non-call: + // "Non-call expression in tail call" + ~"\n\n\n\n\n" // https://github.com/mozilla/rust/issues/850 + ]; + + for confusing_patterns.each |p| { if contains(code, *p) { return true; } } + return false; +} + +fn file_might_not_converge(filename: &Path) -> bool { + let confusing_files = ~[ + ~"expr-alt.rs", // pretty-printing "(a = b) = c" + // vs "a = b = c" and wrapping + ~"block-arg-in-ternary.rs", // wrapping + ~"move-3-unique.rs", // 0 becomes (0), but both seem reasonable. wtf? + ~"move-3.rs" // 0 becomes (0), but both seem reasonable. wtf? + ]; + + + for confusing_files.each |f| { + if contains(filename.to_str(), *f) { + return true; + } + } + + return false; +} + +fn check_roundtrip_convergence(code: @~str, maxIters: uint) { + + let mut i = 0u; + let mut newv = code; + let mut oldv = code; + + while i < maxIters { + oldv = newv; + if content_might_not_converge(*oldv) { return; } + newv = @parse_and_print(oldv); + if oldv == newv { break; } + i += 1u; + } + + if oldv == newv { + error!("Converged after %u iterations", i); + } else { + error!("Did not converge after %u iterations!", i); + write_file(&Path("round-trip-a.rs"), *oldv); + write_file(&Path("round-trip-b.rs"), *newv); + run::run_program(~"diff", + ~[~"-w", ~"-u", ~"round-trip-a.rs", + ~"round-trip-b.rs"]); + fail ~"Mismatch"; + } +} + +fn check_convergence(files: &[Path]) { + error!("pp convergence tests: %u files", vec::len(files)); + for files.each |file| { + if !file_might_not_converge(file) { + let s = @result::get(&io::read_whole_file_str(file)); + if !content_might_not_converge(*s) { + error!("pp converge: %s", file.to_str()); + // Change from 7u to 2u once + // https://github.com/mozilla/rust/issues/850 is fixed + check_roundtrip_convergence(s, 7u); + } + } + } +} + +fn check_variants(files: &[Path], cx: context) { + for files.each |file| { + if cx.mode == tm_converge && + file_might_not_converge(file) { + error!("Skipping convergence test based on\ + file_might_not_converge"); + loop; + } + + let s = @result::get(&io::read_whole_file_str(file)); + if contains(*s, ~"#") { + loop; // Macros are confusing + } + if cx.mode == tm_converge && content_might_not_converge(*s) { + loop; + } + if cx.mode == tm_run && content_is_dangerous_to_compile(*s) { + loop; + } + + let file_str = file.to_str(); + + log(error, ~"check_variants: " + file_str); + let sess = parse::new_parse_sess(option::None); + let crate = + parse::parse_crate_from_source_str( + file_str, + s, ~[], sess); + io::with_str_reader(*s, |rdr| { + error!("%s", + as_str(|a| pprust::print_crate( + sess.cm, + // Assuming no token_trees + syntax::parse::token::mk_fake_ident_interner(), + sess.span_diagnostic, + crate, + file_str, + rdr, a, + pprust::no_ann(), + false))) + }); + check_variants_of_ast(*crate, sess.cm, file, cx); + } +} + +fn main() { + let args = os::args(); + if vec::len(args) != 2u { + error!("usage: %s <testdir>", args[0]); + return; + } + let mut files = ~[]; + let root = Path(args[1]); + + find_rust_files(&mut files, &root); + error!("== check_convergence =="); + check_convergence(files); + error!("== check_variants: converge =="); + check_variants(files, { mode: tm_converge }); + error!("== check_variants: run =="); + check_variants(files, { mode: tm_run }); + + error!("Fuzzer done"); +} + +// Local Variables: +// mode: rust; +// fill-column: 78; +// indent-tabs-mode: nil +// c-basic-offset: 4 +// buffer-file-coding-system: utf-8-unix +// End: diff --git a/src/libfuzzer/ivec_fuzz.rs b/src/libfuzzer/ivec_fuzz.rs new file mode 100644 index 00000000000..f6133b8f954 --- /dev/null +++ b/src/libfuzzer/ivec_fuzz.rs @@ -0,0 +1,110 @@ +/* + +Idea: provide functions for 'exhaustive' and 'random' modification of vecs. + + two functions, "return all edits" and "return a random edit" = move- + leaning toward this model or two functions, "return the number of + possible edits" and "return edit #n" + +It would be nice if this could be data-driven, so the two functions +could share information: + type vec_modifier = rec(fn (<T> v, uint i) -> ~[T] fun, uint lo, uint di); + const ~[vec_modifier] vec_modifiers = ~[rec(fun=vec_omit, 0u, 1u), ...]/~; +But that gives me "error: internal compiler error unimplemented consts +that's not a plain literal". +https://github.com/graydon/rust/issues/570 + +vec_edits is not an iter because iters might go away. + +*/ + +extern mod std; +use vec::slice; +use vec::len; + +fn vec_omit<T: copy>(v: ~[T], i: uint) -> ~[T] { + slice(v, 0u, i) + slice(v, i + 1u, len(v)) +} +fn vec_dup<T: copy>(v: ~[T], i: uint) -> ~[T] { + slice(v, 0u, i) + [v[i]] + slice(v, i, len(v)) +} +fn vec_swadj<T: copy>(v: ~[T], i: uint) -> ~[T] { + slice(v, 0u, i) + [v[i + 1u], v[i]] + slice(v, i + 2u, len(v)) +} +fn vec_prefix<T: copy>(v: ~[T], i: uint) -> ~[T] { slice(v, 0u, i) } +fn vec_suffix<T: copy>(v: ~[T], i: uint) -> ~[T] { slice(v, i, len(v)) } + +fn vec_poke<T: copy>(v: ~[T], i: uint, x: T) -> ~[T] { + slice(v, 0u, i) + ~[x] + slice(v, i + 1u, len(v)) +} +fn vec_insert<T: copy>(v: ~[T], i: uint, x: T) -> ~[T] { + slice(v, 0u, i) + ~[x] + slice(v, i, len(v)) +} + +// Iterates over 0...length, skipping the specified number on each side. +fn ix(skip_low: uint, skip_high: uint, length: uint, it: block(uint)) { + let i: uint = skip_low; + while i + skip_high <= length { it(i); i += 1u; } +} + +// Returns a bunch of modified versions of v, some of which introduce +// new elements (borrowed from xs). +fn vec_edits<T: copy>(v: ~[T], xs: ~[T]) -> ~[~[T]] { + let edits: ~[~[T]] = ~[]; + let Lv: uint = len(v); + + if Lv != 1u { + // When Lv == 1u, this is redundant with omit. + edits.push(~[]); + } + if Lv >= 3u { + // When Lv == 2u, this is redundant with swap. + edits.push(vec::reversed(v)); + } + ix(0u, 1u, Lv) {|i| edits += ~[vec_omit(v, i)]; } + ix(0u, 1u, Lv) {|i| edits += ~[vec_dup(v, i)]; } + ix(0u, 2u, Lv) {|i| edits += ~[vec_swadj(v, i)]; } + ix(1u, 2u, Lv) {|i| edits += ~[vec_prefix(v, i)]; } + ix(2u, 1u, Lv) {|i| edits += ~[vec_suffix(v, i)]; } + + ix(0u, 1u, len(xs)) {|j| + ix(0u, 1u, Lv) {|i| + edits.push(vec_poke(v, i, xs[j])); + } + ix(0u, 0u, Lv) {|i| + edits.push(vec_insert(v, i, xs[j])); + } + } + + edits +} + +// Would be nice if this were built in: +// https://github.com/graydon/rust/issues/424 +fn vec_to_str(v: ~[int]) -> str { + let i = 0u; + let s = "["; + while i < len(v) { + s += int::str(v[i]); + if i + 1u < len(v) { s += ", "; } + i += 1u; + } + return s + "]"; +} + +fn show_edits(a: ~[int], xs: ~[int]) { + log(error, "=== Edits of " + vec_to_str(a) + " ==="); + let b = vec_edits(a, xs); + ix(0u, 1u, len(b)) {|i| log(error, vec_to_str(b[i])); } +} + +fn demo_edits() { + let xs = ~[7, 8]; + show_edits(~[], xs); + show_edits(~[1], xs); + show_edits(~[1, 2], xs); + show_edits(~[1, 2, 3], xs); + show_edits(~[1, 2, 3, 4], xs); +} + +fn main() { demo_edits(); } diff --git a/src/libfuzzer/rand_util.rs b/src/libfuzzer/rand_util.rs new file mode 100644 index 00000000000..6745805e2d8 --- /dev/null +++ b/src/libfuzzer/rand_util.rs @@ -0,0 +1,96 @@ +extern mod std; +use std::rand; + +// random uint less than n +fn under(r : rand::rng, n : uint) -> uint { + assert n != 0u; r.next() as uint % n +} + +// random choice from a vec +fn choice<T: copy>(r : rand::rng, v : ~[T]) -> T { + assert vec::len(v) != 0u; v[under(r, vec::len(v))] +} + +// 1 in n chance of being true +fn unlikely(r : rand::rng, n : uint) -> bool { under(r, n) == 0u } + +// shuffle a vec in place +fn shuffle<T>(r : rand::rng, &v : ~[mut T]) { + let i = vec::len(v); + while i >= 2u { + // Loop invariant: elements with index >= i have been locked in place. + i -= 1u; + vec::swap(v, i, under(r, i + 1u)); // Lock element i in place. + } +} + +// create a shuffled copy of a vec +fn shuffled<T: copy>(r : rand::rng, v : ~[T]) -> ~[T] { + let w = vec::to_mut(v); + shuffle(r, w); + vec::from_mut(w) // Shouldn't this happen automatically? +} + +// sample from a population without replacement +//fn sample<T>(r : rand::rng, pop : ~[T], k : uint) -> ~[T] { fail } + +// Two ways to make a weighted choice. +// * weighted_choice is O(number of choices) time +// * weighted_vec is O(total weight) space +type weighted<T> = { weight: uint, item: T }; +fn weighted_choice<T: copy>(r : rand::rng, v : ~[weighted<T>]) -> T { + assert vec::len(v) != 0u; + let total = 0u; + for {weight: weight, item: _} in v { + total += weight; + } + assert total >= 0u; + let chosen = under(r, total); + let so_far = 0u; + for {weight: weight, item: item} in v { + so_far += weight; + if so_far > chosen { + return item; + } + } + core::unreachable(); +} + +fn weighted_vec<T: copy>(v : ~[weighted<T>]) -> ~[T] { + let r = ~[]; + for {weight: weight, item: item} in v { + let i = 0u; + while i < weight { + r.push(item); + i += 1u; + } + } + r +} + +fn main() +{ + let r = rand::mk_rng(); + + log(error, under(r, 5u)); + log(error, choice(r, ~[10, 20, 30])); + log(error, if unlikely(r, 5u) { "unlikely" } else { "likely" }); + + let a = ~[mut 1, 2, 3]; + shuffle(r, a); + log(error, a); + + let i = 0u; + let v = ~[ + {weight:1u, item:"low"}, + {weight:8u, item:"middle"}, + {weight:1u, item:"high"} + ]; + let w = weighted_vec(v); + + while i < 1000u { + log(error, "Immed: " + weighted_choice(r, v)); + log(error, "Fast: " + choice(r, w)); + i += 1u; + } +} |
