src/test/run-pass/rt-sched-1.rs


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

// Tests of the runtime's scheduler interface

use ptr::is_null;

type sched_id = int;
type task_id = *libc::c_void;

type task = *libc::c_void;
type closure = *libc::c_void;

extern mod rustrt {
    #[legacy_exports];
    fn rust_new_sched(num_threads: libc::uintptr_t) -> sched_id;
    fn rust_get_sched_id() -> sched_id;
    fn rust_new_task_in_sched(id: sched_id) -> task_id;
    fn start_task(id: task_id, f: closure);
}

fn main() unsafe {
    let po = comm::Port();
    let ch = comm::Chan(&po);
    let parent_sched_id = rustrt::rust_get_sched_id();
    error!("parent %?", parent_sched_id);
    let num_threads = 1u;
    let new_sched_id = rustrt::rust_new_sched(num_threads);
    error!("new_sched_id %?", new_sched_id);
    let new_task_id = rustrt::rust_new_task_in_sched(new_sched_id);
    assert !new_task_id.is_null();
    let f = fn~() {
        let child_sched_id = rustrt::rust_get_sched_id();
        error!("child_sched_id %?", child_sched_id);
        assert child_sched_id != parent_sched_id;
        assert child_sched_id == new_sched_id;
        comm::send(ch, ());
    };
    let fptr = cast::reinterpret_cast(&ptr::addr_of(&f));
    rustrt::start_task(new_task_id, fptr);
    cast::forget(f);
    comm::recv(po);
}