OpenE2K/rust
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

rt: Move receive code into rust_port

Browse Source
This commit is contained in:
Brian Anderson 2012-03-02 00:55:27 -08:00
parent 77295c56c5
commit e08f46db68
3 changed files with 24 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/rt/rust_builtin.cpp
View File

@ -544,29 +544,7 @@ rust_task_yield(rust_task *task, bool *killed) {
extern "C" CDECL void
port_recv(uintptr_t *dptr, rust_port *port, uintptr_t *yield) {
*yield = false;
rust_task *task = rust_task_thread::get_task();
{
scoped_lock with(port->lock);
LOG(task, comm, "port: 0x%" PRIxPTR ", dptr: 0x%" PRIxPTR
", size: 0x%" PRIxPTR,
(uintptr_t) port, (uintptr_t) dptr, port->unit_sz);
if (port->receive(dptr)) {
return;
}
// No data was buffered on any incoming channel, so block this task on
// the port. Remember the rendezvous location so that any sender task
// can write to it before waking up this task.
LOG(task, comm, "<=== waiting for rendezvous data ===");
task->rendezvous_ptr = dptr;
task->block(port, "waiting for rendezvous data");
}
*yield = true;
return;
port->receive(dptr, yield);
}
extern "C" CDECL void

26
src/rt/rust_port.cpp
View File

@ -63,14 +63,32 @@ void rust_port::send(void *sptr) {
}
}
bool rust_port::receive(void *dptr) {
I(task->thread, lock.lock_held_by_current_thread());
void rust_port::receive(void *dptr, uintptr_t *yield) {
I(task->thread, !lock.lock_held_by_current_thread());
LOG(task, comm, "port: 0x%" PRIxPTR ", dptr: 0x%" PRIxPTR
", size: 0x%" PRIxPTR,
(uintptr_t) this, (uintptr_t) dptr, unit_sz);
scoped_lock with(lock);
*yield = false;
if (buffer.is_empty() == false) {
buffer.dequeue(dptr);
LOG(task, comm, "<=== read data ===");
return true;
return;
}
return false;
// No data was buffered on any incoming channel, so block this task on
// the port. Remember the rendezvous location so that any sender task
// can write to it before waking up this task.
LOG(task, comm, "<=== waiting for rendezvous data ===");
task->rendezvous_ptr = (uintptr_t*) dptr;
task->block(this, "waiting for rendezvous data");
*yield = true;
}
size_t rust_port::size() {

2
src/rt/rust_port.h
View File

@ -20,7 +20,7 @@ public:
~rust_port();
void log_state();
void send(void *sptr);
bool receive(void *dptr);
void receive(void *dptr, uintptr_t *yield);
size_t size();
void detach();
};