Skip to content

Commit

Permalink
Remove an allocation from rt::init
Browse files Browse the repository at this point in the history
Previously the thread name would first be heap allocated and then
re-allocated to add a nul terminator. Now it will be heap allocated only
once with nul terminator added form the start.
  • Loading branch information
bjorn3 committed Sep 16, 2021
1 parent 6f6bb16 commit af7eede
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 8 deletions.
4 changes: 3 additions & 1 deletion library/std/src/rt.rs
Expand Up @@ -16,6 +16,8 @@
#![deny(unsafe_op_in_unsafe_fn)]
#![allow(unused_macros)]

use crate::ffi::CString;

// Re-export some of our utilities which are expected by other crates.
pub use crate::panicking::{begin_panic, begin_panic_fmt, panic_count};

Expand All @@ -38,7 +40,7 @@ unsafe fn init(argc: isize, argv: *const *const u8) {
// created. Note that this isn't necessary in general for new threads,
// but we just do this to name the main thread and to give it correct
// info about the stack bounds.
let thread = Thread::new(Some("main".to_owned()));
let thread = Thread::new(Some(CString::new("main").unwrap()));
thread_info::set(main_guard, thread);
}
}
Expand Down
12 changes: 5 additions & 7 deletions library/std/src/thread/mod.rs
Expand Up @@ -457,7 +457,9 @@ impl Builder {

let stack_size = stack_size.unwrap_or_else(thread::min_stack);

let my_thread = Thread::new(name);
let my_thread = Thread::new(name.map(|name| {
CString::new(name).expect("thread name may not contain interior null bytes")
}));
let their_thread = my_thread.clone();

let my_packet: Arc<UnsafeCell<Option<Result<T>>>> = Arc::new(UnsafeCell::new(None));
Expand Down Expand Up @@ -1073,12 +1075,8 @@ pub struct Thread {
impl Thread {
// Used only internally to construct a thread object without spawning
// Panics if the name contains nuls.
pub(crate) fn new(name: Option<String>) -> Thread {
let cname =
name.map(|n| CString::new(n).expect("thread name may not contain interior null bytes"));
Thread {
inner: Arc::new(Inner { name: cname, id: ThreadId::new(), parker: Parker::new() }),
}
pub(crate) fn new(name: Option<CString>) -> Thread {
Thread { inner: Arc::new(Inner { name, id: ThreadId::new(), parker: Parker::new() }) }
}

/// Atomically makes the handle's token available if it is not already.
Expand Down

0 comments on commit af7eede

Please sign in to comment.