2016-05-16 18:26:18 -05:00
|
|
|
// Copyright 2016 The Rust Project Developers. See the COPYRIGHT
|
|
|
|
// file at the top-level directory of this distribution and at
|
|
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
|
|
// option. This file may not be copied, modified, or distributed
|
|
|
|
// except according to those terms.
|
|
|
|
|
2017-10-22 22:01:00 -05:00
|
|
|
// ignore-wasm32-bare compiled with panic=abort by default
|
|
|
|
|
2017-11-28 06:48:23 -06:00
|
|
|
#![feature(generators, generator_trait, untagged_unions, slice_patterns, advanced_slice_patterns)]
|
2016-05-29 14:01:06 -05:00
|
|
|
|
|
|
|
use std::cell::{Cell, RefCell};
|
2017-10-17 07:20:22 -05:00
|
|
|
use std::ops::Generator;
|
2016-05-29 14:01:06 -05:00
|
|
|
use std::panic;
|
|
|
|
use std::usize;
|
|
|
|
|
|
|
|
struct InjectedFailure;
|
2016-05-16 18:26:18 -05:00
|
|
|
|
|
|
|
struct Allocator {
|
|
|
|
data: RefCell<Vec<bool>>,
|
2016-05-29 14:01:06 -05:00
|
|
|
failing_op: usize,
|
|
|
|
cur_ops: Cell<usize>,
|
2016-05-16 18:26:18 -05:00
|
|
|
}
|
|
|
|
|
2016-05-29 14:01:06 -05:00
|
|
|
impl panic::UnwindSafe for Allocator {}
|
|
|
|
impl panic::RefUnwindSafe for Allocator {}
|
|
|
|
|
2016-05-16 18:26:18 -05:00
|
|
|
impl Drop for Allocator {
|
|
|
|
fn drop(&mut self) {
|
|
|
|
let data = self.data.borrow();
|
|
|
|
if data.iter().any(|d| *d) {
|
|
|
|
panic!("missing free: {:?}", data);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Allocator {
|
2016-05-29 14:01:06 -05:00
|
|
|
fn new(failing_op: usize) -> Self {
|
|
|
|
Allocator {
|
|
|
|
failing_op: failing_op,
|
|
|
|
cur_ops: Cell::new(0),
|
|
|
|
data: RefCell::new(vec![])
|
|
|
|
}
|
|
|
|
}
|
2016-05-16 18:26:18 -05:00
|
|
|
fn alloc(&self) -> Ptr {
|
2016-05-29 14:01:06 -05:00
|
|
|
self.cur_ops.set(self.cur_ops.get() + 1);
|
|
|
|
|
|
|
|
if self.cur_ops.get() == self.failing_op {
|
|
|
|
panic!(InjectedFailure);
|
|
|
|
}
|
|
|
|
|
2016-05-16 18:26:18 -05:00
|
|
|
let mut data = self.data.borrow_mut();
|
|
|
|
let addr = data.len();
|
|
|
|
data.push(true);
|
|
|
|
Ptr(addr, self)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
struct Ptr<'a>(usize, &'a Allocator);
|
|
|
|
impl<'a> Drop for Ptr<'a> {
|
|
|
|
fn drop(&mut self) {
|
|
|
|
match self.1.data.borrow_mut()[self.0] {
|
|
|
|
false => {
|
|
|
|
panic!("double free at index {:?}", self.0)
|
|
|
|
}
|
|
|
|
ref mut d => *d = false
|
|
|
|
}
|
2016-05-29 14:01:06 -05:00
|
|
|
|
|
|
|
self.1.cur_ops.set(self.1.cur_ops.get()+1);
|
|
|
|
|
|
|
|
if self.1.cur_ops.get() == self.1.failing_op {
|
|
|
|
panic!(InjectedFailure);
|
|
|
|
}
|
2016-05-16 18:26:18 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn dynamic_init(a: &Allocator, c: bool) {
|
|
|
|
let _x;
|
|
|
|
if c {
|
|
|
|
_x = Some(a.alloc());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-05-29 14:01:06 -05:00
|
|
|
fn dynamic_drop(a: &Allocator, c: bool) {
|
2016-05-16 18:26:18 -05:00
|
|
|
let x = a.alloc();
|
|
|
|
if c {
|
|
|
|
Some(x)
|
|
|
|
} else {
|
|
|
|
None
|
2016-05-29 14:01:06 -05:00
|
|
|
};
|
2016-05-16 18:26:18 -05:00
|
|
|
}
|
|
|
|
|
2017-05-15 07:22:59 -05:00
|
|
|
struct TwoPtrs<'a>(Ptr<'a>, Ptr<'a>);
|
|
|
|
fn struct_dynamic_drop(a: &Allocator, c0: bool, c1: bool, c: bool) {
|
|
|
|
for i in 0..2 {
|
|
|
|
let x;
|
|
|
|
let y;
|
|
|
|
if (c0 && i == 0) || (c1 && i == 1) {
|
|
|
|
x = (a.alloc(), a.alloc(), a.alloc());
|
|
|
|
y = TwoPtrs(a.alloc(), a.alloc());
|
|
|
|
if c {
|
|
|
|
drop(x.1);
|
|
|
|
drop(y.0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-06-27 10:34:07 -05:00
|
|
|
fn field_assignment(a: &Allocator, c0: bool) {
|
|
|
|
let mut x = (TwoPtrs(a.alloc(), a.alloc()), a.alloc());
|
|
|
|
|
|
|
|
x.1 = a.alloc();
|
|
|
|
x.1 = a.alloc();
|
|
|
|
|
|
|
|
let f = (x.0).0;
|
|
|
|
if c0 {
|
|
|
|
(x.0).0 = f;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-05-16 18:26:18 -05:00
|
|
|
fn assignment2(a: &Allocator, c0: bool, c1: bool) {
|
|
|
|
let mut _v = a.alloc();
|
|
|
|
let mut _w = a.alloc();
|
|
|
|
if c0 {
|
|
|
|
drop(_v);
|
|
|
|
}
|
|
|
|
_v = _w;
|
|
|
|
if c1 {
|
|
|
|
_w = a.alloc();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn assignment1(a: &Allocator, c0: bool) {
|
|
|
|
let mut _v = a.alloc();
|
|
|
|
let mut _w = a.alloc();
|
|
|
|
if c0 {
|
|
|
|
drop(_v);
|
|
|
|
}
|
|
|
|
_v = _w;
|
|
|
|
}
|
|
|
|
|
2016-09-08 12:12:53 -05:00
|
|
|
#[allow(unions_with_drop_fields)]
|
|
|
|
union Boxy<T> {
|
|
|
|
a: T,
|
|
|
|
b: T,
|
|
|
|
}
|
|
|
|
|
|
|
|
fn union1(a: &Allocator) {
|
|
|
|
unsafe {
|
|
|
|
let mut u = Boxy { a: a.alloc() };
|
|
|
|
u.b = a.alloc();
|
|
|
|
drop(u.a);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-05-10 17:02:52 -05:00
|
|
|
fn array_simple(a: &Allocator) {
|
|
|
|
let _x = [a.alloc(), a.alloc(), a.alloc(), a.alloc()];
|
|
|
|
}
|
|
|
|
|
|
|
|
fn vec_simple(a: &Allocator) {
|
|
|
|
let _x = vec![a.alloc(), a.alloc(), a.alloc(), a.alloc()];
|
|
|
|
}
|
|
|
|
|
2017-10-17 07:20:22 -05:00
|
|
|
fn generator(a: &Allocator, run_count: usize) {
|
|
|
|
assert!(run_count < 4);
|
|
|
|
|
|
|
|
let mut gen = || {
|
|
|
|
(a.alloc(),
|
|
|
|
yield a.alloc(),
|
|
|
|
a.alloc(),
|
|
|
|
yield a.alloc()
|
|
|
|
);
|
|
|
|
};
|
|
|
|
for _ in 0..run_count {
|
|
|
|
gen.resume();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-10-17 13:37:18 -05:00
|
|
|
fn mixed_drop_and_nondrop(a: &Allocator) {
|
|
|
|
// check that destructor panics handle drop
|
|
|
|
// and non-drop blocks in the same scope correctly.
|
|
|
|
//
|
|
|
|
// Surprisingly enough, this used to not work.
|
|
|
|
let (x, y, z);
|
|
|
|
x = a.alloc();
|
|
|
|
y = 5;
|
|
|
|
z = a.alloc();
|
|
|
|
}
|
|
|
|
|
2017-08-09 14:23:27 -05:00
|
|
|
#[allow(unreachable_code)]
|
|
|
|
fn vec_unreachable(a: &Allocator) {
|
|
|
|
let _x = vec![a.alloc(), a.alloc(), a.alloc(), return];
|
|
|
|
}
|
|
|
|
|
2017-11-28 06:48:23 -06:00
|
|
|
fn slice_pattern_first(a: &Allocator) {
|
|
|
|
let[_x, ..] = [a.alloc(), a.alloc(), a.alloc()];
|
|
|
|
}
|
|
|
|
|
|
|
|
fn slice_pattern_middle(a: &Allocator) {
|
|
|
|
let[_, _x, _] = [a.alloc(), a.alloc(), a.alloc()];
|
|
|
|
}
|
|
|
|
|
|
|
|
fn slice_pattern_two(a: &Allocator) {
|
|
|
|
let[_x, _, _y, _] = [a.alloc(), a.alloc(), a.alloc(), a.alloc()];
|
|
|
|
}
|
|
|
|
|
|
|
|
fn slice_pattern_last(a: &Allocator) {
|
|
|
|
let[.., _y] = [a.alloc(), a.alloc(), a.alloc(), a.alloc()];
|
|
|
|
}
|
|
|
|
|
|
|
|
fn slice_pattern_one_of(a: &Allocator, i: usize) {
|
|
|
|
let array = [a.alloc(), a.alloc(), a.alloc(), a.alloc()];
|
|
|
|
let _x = match i {
|
|
|
|
0 => { let [a, ..] = array; a }
|
|
|
|
1 => { let [_, a, ..] = array; a }
|
|
|
|
2 => { let [_, _, a, _] = array; a }
|
|
|
|
3 => { let [_, _, _, a] = array; a }
|
|
|
|
_ => panic!("unmatched"),
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2016-05-29 14:01:06 -05:00
|
|
|
fn run_test<F>(mut f: F)
|
|
|
|
where F: FnMut(&Allocator)
|
|
|
|
{
|
|
|
|
let first_alloc = Allocator::new(usize::MAX);
|
|
|
|
f(&first_alloc);
|
|
|
|
|
|
|
|
for failing_op in 1..first_alloc.cur_ops.get()+1 {
|
|
|
|
let alloc = Allocator::new(failing_op);
|
|
|
|
let alloc = &alloc;
|
|
|
|
let f = panic::AssertUnwindSafe(&mut f);
|
|
|
|
let result = panic::catch_unwind(move || {
|
|
|
|
f.0(alloc);
|
|
|
|
});
|
|
|
|
match result {
|
|
|
|
Ok(..) => panic!("test executed {} ops but now {}",
|
|
|
|
first_alloc.cur_ops.get(), alloc.cur_ops.get()),
|
|
|
|
Err(e) => {
|
|
|
|
if e.downcast_ref::<InjectedFailure>().is_none() {
|
|
|
|
panic::resume_unwind(e);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2016-05-16 18:26:18 -05:00
|
|
|
|
2016-09-08 12:12:53 -05:00
|
|
|
fn run_test_nopanic<F>(mut f: F)
|
|
|
|
where F: FnMut(&Allocator)
|
|
|
|
{
|
|
|
|
let first_alloc = Allocator::new(usize::MAX);
|
|
|
|
f(&first_alloc);
|
|
|
|
}
|
|
|
|
|
2016-05-16 18:26:18 -05:00
|
|
|
fn main() {
|
2016-05-29 14:01:06 -05:00
|
|
|
run_test(|a| dynamic_init(a, false));
|
|
|
|
run_test(|a| dynamic_init(a, true));
|
|
|
|
run_test(|a| dynamic_drop(a, false));
|
|
|
|
run_test(|a| dynamic_drop(a, true));
|
|
|
|
|
|
|
|
run_test(|a| assignment2(a, false, false));
|
|
|
|
run_test(|a| assignment2(a, false, true));
|
|
|
|
run_test(|a| assignment2(a, true, false));
|
|
|
|
run_test(|a| assignment2(a, true, true));
|
|
|
|
|
|
|
|
run_test(|a| assignment1(a, false));
|
|
|
|
run_test(|a| assignment1(a, true));
|
2016-09-08 12:12:53 -05:00
|
|
|
|
2017-05-10 17:02:52 -05:00
|
|
|
run_test(|a| array_simple(a));
|
|
|
|
run_test(|a| vec_simple(a));
|
2017-08-09 14:23:27 -05:00
|
|
|
run_test(|a| vec_unreachable(a));
|
2017-05-10 17:02:52 -05:00
|
|
|
|
2017-05-15 07:22:59 -05:00
|
|
|
run_test(|a| struct_dynamic_drop(a, false, false, false));
|
|
|
|
run_test(|a| struct_dynamic_drop(a, false, false, true));
|
|
|
|
run_test(|a| struct_dynamic_drop(a, false, true, false));
|
|
|
|
run_test(|a| struct_dynamic_drop(a, false, true, true));
|
|
|
|
run_test(|a| struct_dynamic_drop(a, true, false, false));
|
|
|
|
run_test(|a| struct_dynamic_drop(a, true, false, true));
|
|
|
|
run_test(|a| struct_dynamic_drop(a, true, true, false));
|
|
|
|
run_test(|a| struct_dynamic_drop(a, true, true, true));
|
|
|
|
|
2017-06-27 10:34:07 -05:00
|
|
|
run_test(|a| field_assignment(a, false));
|
|
|
|
run_test(|a| field_assignment(a, true));
|
|
|
|
|
2017-10-17 13:37:18 -05:00
|
|
|
run_test(|a| generator(a, 0));
|
|
|
|
run_test(|a| generator(a, 1));
|
|
|
|
run_test(|a| generator(a, 2));
|
|
|
|
run_test(|a| generator(a, 3));
|
|
|
|
|
|
|
|
run_test(|a| mixed_drop_and_nondrop(a));
|
2017-10-17 07:20:22 -05:00
|
|
|
|
2017-11-28 06:48:23 -06:00
|
|
|
run_test(|a| slice_pattern_first(a));
|
|
|
|
run_test(|a| slice_pattern_middle(a));
|
|
|
|
run_test(|a| slice_pattern_two(a));
|
|
|
|
run_test(|a| slice_pattern_last(a));
|
|
|
|
run_test(|a| slice_pattern_one_of(a, 0));
|
|
|
|
run_test(|a| slice_pattern_one_of(a, 1));
|
|
|
|
run_test(|a| slice_pattern_one_of(a, 2));
|
|
|
|
run_test(|a| slice_pattern_one_of(a, 3));
|
|
|
|
|
2016-09-08 12:12:53 -05:00
|
|
|
run_test_nopanic(|a| union1(a));
|
2016-05-16 18:26:18 -05:00
|
|
|
}
|