Skip to content

Commit

Permalink
Auto merge of #123433 - GnomedDev:remove-threadname-alloc, r=joboet
Browse files Browse the repository at this point in the history
Remove rt::init allocation for thread name

This removes one of the allocations in a `fn main() {}` program.
  • Loading branch information
bors committed Apr 6, 2024
2 parents 11853ec + 0989416 commit 30840c5
Show file tree
Hide file tree
Showing 2 changed files with 27 additions and 6 deletions.
4 changes: 1 addition & 3 deletions library/std/src/rt.rs
Expand Up @@ -16,8 +16,6 @@
#![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, panic_count};
pub use core::panicking::{panic_display, panic_fmt};
Expand Down Expand Up @@ -96,7 +94,7 @@ unsafe fn init(argc: isize, argv: *const *const u8, sigpipe: u8) {
sys::init(argc, argv, sigpipe);

// Set up the current thread to give it the right name.
let thread = Thread::new(Some(rtunwrap!(Ok, CString::new("main"))));
let thread = Thread::new_main();
thread::set_current(thread);
}
}
Expand Down
29 changes: 26 additions & 3 deletions library/std/src/thread/mod.rs
Expand Up @@ -1245,9 +1245,16 @@ impl ThreadId {
// Thread
////////////////////////////////////////////////////////////////////////////////

/// The internal representation of a `Thread`'s name.
enum ThreadName {
Main,
Other(CString),
Unnamed,
}

/// The internal representation of a `Thread` handle
struct Inner {
name: Option<CString>, // Guaranteed to be UTF-8
name: ThreadName, // Guaranteed to be UTF-8
id: ThreadId,
parker: Parker,
}
Expand Down Expand Up @@ -1284,8 +1291,20 @@ 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<CString>) -> Thread {
if let Some(name) = name {
Self::new_inner(ThreadName::Other(name))
} else {
Self::new_inner(ThreadName::Unnamed)
}
}

// Used in runtime to construct main thread
pub(crate) fn new_main() -> Thread {
Self::new_inner(ThreadName::Main)
}

fn new_inner(name: ThreadName) -> Thread {
// We have to use `unsafe` here to construct the `Parker` in-place,
// which is required for the UNIX implementation.
//
Expand Down Expand Up @@ -1412,7 +1431,11 @@ impl Thread {
}

fn cname(&self) -> Option<&CStr> {
self.inner.name.as_deref()
match &self.inner.name {
ThreadName::Main => Some(c"main"),
ThreadName::Other(other) => Some(&other),
ThreadName::Unnamed => None,
}
}
}

Expand Down

0 comments on commit 30840c5

Please sign in to comment.