Skip to content

Commit 72bf895

Browse files
committed
auto merge of #8854 : huonw/rust/rt-papercuts, r=brson
The only user-facing change is handling non-integer (and zero) `RUST_THREADS` more nicely: ``` $ RUST_THREADS=x rustc # old You've met with a terrible fate, haven't you? fatal runtime error: runtime tls key not initialized Aborted $ RUST_THREADS=x ./x86_64-unknown-linux-gnu/stage2/bin/rustc # new You've met with a terrible fate, haven't you? fatal runtime error: `RUST_THREADS` is `x`, should be a positive integer Aborted ``` The other changes are converting some `for .. in range(x,y)` to `vec::from_fn` or `for .. in x.iter()` as appropriate; and removing a chain of (seemingly) unnecessary pointer casts. (Also, fixes a typo in `extra::test` from #8823.)
2 parents f10e50a + 4fea236 commit 72bf895

File tree

6 files changed

+27
-33
lines changed

6 files changed

+27
-33
lines changed

src/libextra/test.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -745,7 +745,7 @@ fn get_concurrency() -> uint {
745745
let opt_n: Option<uint> = FromStr::from_str(s);
746746
match opt_n {
747747
Some(n) if n > 0 => n,
748-
_ => fail!("RUST_TEST_TASKS is `%s`, should be a non-negative integer.", s)
748+
_ => fail!("RUST_TEST_TASKS is `%s`, should be a positive integer.", s)
749749
}
750750
}
751751
None => {

src/libstd/rt/args.rs

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -55,10 +55,11 @@ pub fn clone() -> Option<~[~str]> {
5555
mod imp {
5656
use libc;
5757
use option::{Option, Some, None};
58-
use iterator::{Iterator, range};
58+
use iterator::Iterator;
5959
use str;
6060
use unstable::finally::Finally;
6161
use util;
62+
use vec;
6263

6364
pub unsafe fn init(argc: int, argv: **u8) {
6465
let args = load_argc_and_argv(argc, argv);
@@ -111,11 +112,9 @@ mod imp {
111112

112113
// Copied from `os`.
113114
unsafe fn load_argc_and_argv(argc: int, argv: **u8) -> ~[~str] {
114-
let mut args = ~[];
115-
for i in range(0u, argc as uint) {
116-
args.push(str::raw::from_c_str(*(argv as **libc::c_char).offset(i as int)));
115+
do vec::from_fn(argc as uint) |i| {
116+
str::raw::from_c_str(*(argv as **libc::c_char).offset(i as int))
117117
}
118-
args
119118
}
120119

121120
#[cfg(stage0)]

src/libstd/rt/local_ptr.rs

Lines changed: 6 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -121,27 +121,20 @@ pub unsafe fn borrow<T>(f: &fn(&mut T)) {
121121
/// For the Scheduler pointer to be aliased
122122
pub unsafe fn unsafe_borrow<T>() -> *mut T {
123123
let key = tls_key();
124-
let mut void_ptr: *mut c_void = tls::get(key);
124+
let void_ptr = tls::get(key);
125125
if void_ptr.is_null() {
126126
rtabort!("thread-local pointer is null. bogus!");
127127
}
128-
let ptr: *mut *mut c_void = &mut void_ptr;
129-
let ptr: *mut ~T = ptr as *mut ~T;
130-
let ptr: *mut T = &mut **ptr;
131-
return ptr;
128+
void_ptr as *mut T
132129
}
133130

134131
pub unsafe fn try_unsafe_borrow<T>() -> Option<*mut T> {
135132
let key = tls_key();
136-
let mut void_ptr: *mut c_void = tls::get(key);
133+
let void_ptr = tls::get(key);
137134
if void_ptr.is_null() {
138-
return None;
139-
}
140-
{
141-
let ptr: *mut *mut c_void = &mut void_ptr;
142-
let ptr: *mut ~T = ptr as *mut ~T;
143-
let ptr: *mut T = &mut **ptr;
144-
return Some(ptr);
135+
None
136+
} else {
137+
Some(void_ptr as *mut T)
145138
}
146139
}
147140

src/libstd/rt/mod.rs

Lines changed: 7 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ Several modules in `core` are clients of `rt`:
5959
use cell::Cell;
6060
use clone::Clone;
6161
use container::Container;
62-
use iterator::{Iterator, range};
62+
use iterator::Iterator;
6363
use option::{Option, None, Some};
6464
use ptr::RawPtr;
6565
use rt::local::Local;
@@ -71,7 +71,8 @@ use rt::work_queue::WorkQueue;
7171
use rt::uv::uvio::UvEventLoop;
7272
use unstable::atomics::{AtomicInt, SeqCst};
7373
use unstable::sync::UnsafeArc;
74-
use vec::{OwnedVector, MutableVector};
74+
use vec;
75+
use vec::{OwnedVector, MutableVector, ImmutableVector};
7576

7677
/// The global (exchange) heap.
7778
pub mod global_heap;
@@ -251,25 +252,21 @@ fn run_(main: ~fn(), use_main_sched: bool) -> int {
251252

252253
// Create a work queue for each scheduler, ntimes. Create an extra
253254
// for the main thread if that flag is set. We won't steal from it.
254-
let mut work_queues = ~[];
255-
for _ in range(0u, nscheds) {
256-
let work_queue: WorkQueue<~Task> = WorkQueue::new();
257-
work_queues.push(work_queue);
258-
}
255+
let work_queues: ~[WorkQueue<~Task>] = vec::from_fn(nscheds, |_| WorkQueue::new());
259256

260257
// The schedulers.
261258
let mut scheds = ~[];
262259
// Handles to the schedulers. When the main task ends these will be
263260
// sent the Shutdown message to terminate the schedulers.
264261
let mut handles = ~[];
265262

266-
for i in range(0u, nscheds) {
263+
for work_queue in work_queues.iter() {
267264
rtdebug!("inserting a regular scheduler");
268265

269266
// Every scheduler is driven by an I/O event loop.
270267
let loop_ = ~UvEventLoop::new();
271268
let mut sched = ~Scheduler::new(loop_,
272-
work_queues[i].clone(),
269+
work_queue.clone(),
273270
work_queues.clone(),
274271
sleepers.clone());
275272
let handle = sched.make_handle();
@@ -358,9 +355,8 @@ fn run_(main: ~fn(), use_main_sched: bool) -> int {
358355
}
359356

360357
// Run each remaining scheduler in a thread.
361-
while !scheds.is_empty() {
358+
for sched in scheds.move_rev_iter() {
362359
rtdebug!("creating regular schedulers");
363-
let sched = scheds.pop();
364360
let sched_cell = Cell::new(sched);
365361
let thread = do Thread::start {
366362
let mut sched = sched_cell.take();

src/libstd/rt/util.rs

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
use container::Container;
1212
use from_str::FromStr;
1313
use libc;
14-
use option::{Some, None};
14+
use option::{Some, None, Option};
1515
use os;
1616
use str::StrSlice;
1717
use unstable::atomics::{AtomicInt, INIT_ATOMIC_INT, SeqCst};
@@ -57,7 +57,13 @@ pub fn limit_thread_creation_due_to_osx_and_valgrind() -> bool {
5757
/// either `RUST_THREADS` or `num_cpus`.
5858
pub fn default_sched_threads() -> uint {
5959
match os::getenv("RUST_THREADS") {
60-
Some(nstr) => FromStr::from_str(nstr).unwrap(),
60+
Some(nstr) => {
61+
let opt_n: Option<uint> = FromStr::from_str(nstr);
62+
match opt_n {
63+
Some(n) if n > 0 => n,
64+
_ => rtabort!("`RUST_THREADS` is `%s`, should be a positive integer", nstr)
65+
}
66+
}
6167
None => {
6268
if limit_thread_creation_due_to_osx_and_valgrind() {
6369
1

src/test/run-fail/test-threads-invalid-value.rs renamed to src/test/run-fail/test-tasks-invalid-value.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
// This checks that RUST_TEST_TASKS not being 1, 2, ... is detected
1212
// properly.
1313

14-
// error-pattern:should be a non-negative integer
14+
// error-pattern:should be a positive integer
1515
// compile-flags: --test
1616
// exec-env:RUST_TEST_TASKS=foo
1717

0 commit comments

Comments
 (0)