mod pipes { import unsafe::{forget, transmute}; enum state { empty, full, blocked, terminated } impl state : cmp::Eq { pure fn eq(&&other: state) -> bool { (self as uint) == (other as uint) } pure fn ne(&&other: state) -> bool { !self.eq(other) } } type packet = { mut state: state, mut blocked_task: Option, mut payload: Option }; fn packet() -> *packet unsafe { let p: *packet = unsafe::transmute(~{ mut state: empty, mut blocked_task: None::, mut payload: None:: }); p } #[abi = "rust-intrinsic"] mod rusti { fn atomic_xchg(_dst: &mut int, _src: int) -> int { fail; } fn atomic_xchg_acq(_dst: &mut int, _src: int) -> int { fail; } fn atomic_xchg_rel(_dst: &mut int, _src: int) -> int { fail; } } // We should consider moving this to core::unsafe, although I // suspect graydon would want us to use void pointers instead. unsafe fn uniquify(+x: *T) -> ~T { unsafe { unsafe::transmute(x) } } fn swap_state_acq(+dst: &mut state, src: state) -> state { unsafe { transmute(rusti::atomic_xchg_acq(transmute(dst), src as int)) } } fn swap_state_rel(+dst: &mut state, src: state) -> state { unsafe { transmute(rusti::atomic_xchg_rel(transmute(dst), src as int)) } } fn send(-p: send_packet, -payload: T) { let p = p.unwrap(); let p = unsafe { uniquify(p) }; assert (*p).payload.is_none(); (*p).payload <- Some(payload); let old_state = swap_state_rel(&mut (*p).state, full); match old_state { empty => { // Yay, fastpath. // The receiver will eventually clean this up. unsafe { forget(p); } } full => { fail ~"duplicate send" } blocked => { // The receiver will eventually clean this up. unsafe { forget(p); } } terminated => { // The receiver will never receive this. Rely on drop_glue // to clean everything up. } } } fn recv(-p: recv_packet) -> Option { let p = p.unwrap(); let p = unsafe { uniquify(p) }; loop { let old_state = swap_state_acq(&mut (*p).state, blocked); match old_state { empty | blocked => { task::yield(); } full => { let mut payload = None; payload <-> (*p).payload; return Some(option::unwrap(payload)) } terminated => { assert old_state == terminated; return None; } } } } fn sender_terminate(p: *packet) { let p = unsafe { uniquify(p) }; match swap_state_rel(&mut (*p).state, terminated) { empty | blocked => { // The receiver will eventually clean up. unsafe { forget(p) } } full => { // This is impossible fail ~"you dun goofed" } terminated => { // I have to clean up, use drop_glue } } } fn receiver_terminate(p: *packet) { let p = unsafe { uniquify(p) }; match swap_state_rel(&mut (*p).state, terminated) { empty => { // the sender will clean up unsafe { forget(p) } } blocked => { // this shouldn't happen. fail ~"terminating a blocked packet" } terminated | full => { // I have to clean up, use drop_glue } } } struct send_packet { let mut p: Option<*packet>; drop { if self.p != None { let mut p = None; p <-> self.p; sender_terminate(option::unwrap(p)) } } fn unwrap() -> *packet { let mut p = None; p <-> self.p; option::unwrap(p) } } fn send_packet(p: *packet) -> send_packet { send_packet { p: Some(p) } } struct recv_packet { let mut p: Option<*packet>; drop { if self.p != None { let mut p = None; p <-> self.p; receiver_terminate(option::unwrap(p)) } } fn unwrap() -> *packet { let mut p = None; p <-> self.p; option::unwrap(p) } } fn recv_packet(p: *packet) -> recv_packet { recv_packet { p: Some(p) } } fn entangle() -> (send_packet, recv_packet) { let p = packet(); (send_packet(p), recv_packet(p)) } } mod pingpong { enum ping = pipes::send_packet; enum pong = pipes::send_packet; fn liberate_ping(-p: ping) -> pipes::send_packet unsafe { let addr : *pipes::send_packet = match p { ping(x) => { unsafe::transmute(ptr::addr_of(x)) } }; let liberated_value <- *addr; unsafe::forget(p); liberated_value } fn liberate_pong(-p: pong) -> pipes::send_packet unsafe { let addr : *pipes::send_packet = match p { pong(x) => { unsafe::transmute(ptr::addr_of(x)) } }; let liberated_value <- *addr; unsafe::forget(p); liberated_value } fn init() -> (client::ping, server::ping) { pipes::entangle() } mod client { type ping = pipes::send_packet; type pong = pipes::recv_packet; fn do_ping(-c: ping) -> pong { let (sp, rp) = pipes::entangle(); pipes::send(c, ping(sp)); rp } fn do_pong(-c: pong) -> (ping, ()) { let packet = pipes::recv(c); if packet.is_none() { fail ~"sender closed the connection" } (liberate_pong(option::unwrap(packet)), ()) } } mod server { type ping = pipes::recv_packet; type pong = pipes::send_packet; fn do_ping(-c: ping) -> (pong, ()) { let packet = pipes::recv(c); if packet.is_none() { fail ~"sender closed the connection" } (liberate_ping(option::unwrap(packet)), ()) } fn do_pong(-c: pong) -> ping { let (sp, rp) = pipes::entangle(); pipes::send(c, pong(sp)); rp } } } fn client(-chan: pingpong::client::ping) { let chan = pingpong::client::do_ping(chan); log(error, ~"Sent ping"); let (chan, _data) = pingpong::client::do_pong(chan); log(error, ~"Received pong"); } fn server(-chan: pingpong::server::ping) { let (chan, _data) = pingpong::server::do_ping(chan); log(error, ~"Received ping"); let chan = pingpong::server::do_pong(chan); log(error, ~"Sent pong"); } fn main() { /* // Commented out because of option::get error let (client_, server_) = pingpong::init(); let client_ = ~mut some(client_); let server_ = ~mut some(server_); task::spawn {|move client_| let mut client__ = none; *client_ <-> client__; client(option::unwrap(client__)); }; task::spawn {|move server_| let mut server_ˊ = none; *server_ <-> server_ˊ; server(option::unwrap(server_ˊ)); }; */ }