extra: use a mutex to wrap linenoise calls and make them threadsafe.
Fixes #3921.
This commit is contained in:
parent
5bb8aefed6
commit
21ce41d42d
3 changed files with 62 additions and 28 deletions
|
|
@ -8,13 +8,10 @@
|
|||
// option. This file may not be copied, modified, or distributed
|
||||
// except according to those terms.
|
||||
|
||||
// FIXME #3921. This is unsafe because linenoise uses global mutable
|
||||
// state without mutexes.
|
||||
|
||||
use std::c_str::ToCStr;
|
||||
use std::libc::{c_char, c_int};
|
||||
use std::local_data;
|
||||
use std::str;
|
||||
use std::{local_data, str, rt};
|
||||
use std::unstable::finally::Finally;
|
||||
|
||||
#[cfg(stage0)]
|
||||
pub mod rustrt {
|
||||
|
|
@ -28,6 +25,9 @@ pub mod rustrt {
|
|||
fn linenoiseHistoryLoad(file: *c_char) -> c_int;
|
||||
fn linenoiseSetCompletionCallback(callback: *u8);
|
||||
fn linenoiseAddCompletion(completions: *(), line: *c_char);
|
||||
|
||||
fn rust_take_linenoise_lock();
|
||||
fn rust_drop_linenoise_lock();
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -42,38 +42,53 @@ pub mod rustrt {
|
|||
externfn!(fn linenoiseHistoryLoad(file: *c_char) -> c_int)
|
||||
externfn!(fn linenoiseSetCompletionCallback(callback: extern "C" fn(*i8, *())))
|
||||
externfn!(fn linenoiseAddCompletion(completions: *(), line: *c_char))
|
||||
|
||||
externfn!(fn rust_take_linenoise_lock())
|
||||
externfn!(fn rust_drop_linenoise_lock())
|
||||
}
|
||||
|
||||
macro_rules! locked {
|
||||
($expr:expr) => {
|
||||
unsafe {
|
||||
// FIXME #9105: can't use a static mutex in pure Rust yet.
|
||||
rustrt::rust_take_linenoise_lock();
|
||||
let x = $expr;
|
||||
rustrt::rust_drop_linenoise_lock();
|
||||
x
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Add a line to history
|
||||
pub unsafe fn add_history(line: &str) -> bool {
|
||||
pub fn add_history(line: &str) -> bool {
|
||||
do line.with_c_str |buf| {
|
||||
rustrt::linenoiseHistoryAdd(buf) == 1 as c_int
|
||||
(locked!(rustrt::linenoiseHistoryAdd(buf))) == 1 as c_int
|
||||
}
|
||||
}
|
||||
|
||||
/// Set the maximum amount of lines stored
|
||||
pub unsafe fn set_history_max_len(len: int) -> bool {
|
||||
rustrt::linenoiseHistorySetMaxLen(len as c_int) == 1 as c_int
|
||||
pub fn set_history_max_len(len: int) -> bool {
|
||||
(locked!(rustrt::linenoiseHistorySetMaxLen(len as c_int))) == 1 as c_int
|
||||
}
|
||||
|
||||
/// Save line history to a file
|
||||
pub unsafe fn save_history(file: &str) -> bool {
|
||||
pub fn save_history(file: &str) -> bool {
|
||||
do file.with_c_str |buf| {
|
||||
rustrt::linenoiseHistorySave(buf) == 1 as c_int
|
||||
(locked!(rustrt::linenoiseHistorySave(buf))) == 1 as c_int
|
||||
}
|
||||
}
|
||||
|
||||
/// Load line history from a file
|
||||
pub unsafe fn load_history(file: &str) -> bool {
|
||||
pub fn load_history(file: &str) -> bool {
|
||||
do file.with_c_str |buf| {
|
||||
rustrt::linenoiseHistoryLoad(buf) == 1 as c_int
|
||||
(locked!(rustrt::linenoiseHistoryLoad(buf))) == 1 as c_int
|
||||
}
|
||||
}
|
||||
|
||||
/// Print out a prompt and then wait for input and return it
|
||||
pub unsafe fn read(prompt: &str) -> Option<~str> {
|
||||
pub fn read(prompt: &str) -> Option<~str> {
|
||||
do prompt.with_c_str |buf| {
|
||||
let line = rustrt::linenoise(buf);
|
||||
let line = locked!(rustrt::linenoise(buf));
|
||||
|
||||
if line.is_null() { None }
|
||||
else { Some(str::raw::from_c_str(line)) }
|
||||
|
|
@ -84,8 +99,13 @@ pub type CompletionCb = @fn(~str, @fn(~str));
|
|||
|
||||
static complete_key: local_data::Key<@CompletionCb> = &local_data::Key;
|
||||
|
||||
/// Bind to the main completion callback
|
||||
pub unsafe fn complete(cb: CompletionCb) {
|
||||
/// Bind to the main completion callback.
|
||||
///
|
||||
/// The completion callback should not call any `extra::rl` functions
|
||||
/// other than the closure that it receives as its second
|
||||
/// argument. Calling such a function will deadlock on the mutex used
|
||||
/// to ensure that the calls are thread-safe.
|
||||
pub fn complete(cb: CompletionCb) {
|
||||
local_data::set(complete_key, @cb);
|
||||
|
||||
extern fn callback(line: *c_char, completions: *()) {
|
||||
|
|
@ -95,6 +115,10 @@ pub unsafe fn complete(cb: CompletionCb) {
|
|||
unsafe {
|
||||
do cb(str::raw::from_c_str(line)) |suggestion| {
|
||||
do suggestion.with_c_str |buf| {
|
||||
// This isn't locked, because `callback` gets
|
||||
// called inside `rustrt::linenoise`, which
|
||||
// *is* already inside the mutex, so
|
||||
// re-locking would be a deadlock.
|
||||
rustrt::linenoiseAddCompletion(completions, buf);
|
||||
}
|
||||
}
|
||||
|
|
@ -102,5 +126,5 @@ pub unsafe fn complete(cb: CompletionCb) {
|
|||
}
|
||||
}
|
||||
|
||||
rustrt::linenoiseSetCompletionCallback(callback);
|
||||
locked!(rustrt::linenoiseSetCompletionCallback(callback));
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue