kossLAN/wry
1
0
Fork 0
forked from wry/wry
Code Pull requests Activity

all: mark env::set_var/remove_var unsafe

Browse source
This commit is contained in:
Julian Orth 2024-10-20 18:43:51 +02:00
parent 73a6e0d5c2
commit a0984a1205
2 changed files with 18 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/cli/run_privileged.rs
View file

@ -20,7 +20,9 @@ pub fn main(global: GlobalArgs, args: RunPrivilegedArgs) {
let mut path = PathBuf::from(xrd);
path.push(&wd);
if path.exists() {
std::env::set_var(WAYLAND_DISPLAY, &wd);
unsafe {
std::env::set_var(WAYLAND_DISPLAY, &wd);
}
}
}
let mut argv = UstrPtr::new();

24
src/forker.rs
View file

@ -332,9 +332,11 @@ struct Forker {
impl Forker {
fn handle(ppid: c::pid_t, socket: OwnedFd) -> ! {
env::set_var("XDG_SESSION_TYPE", "wayland");
env::remove_var(DISPLAY);
env::remove_var(WAYLAND_DISPLAY);
unsafe {
env::set_var("XDG_SESSION_TYPE", "wayland");
env::remove_var(DISPLAY);
env::remove_var(WAYLAND_DISPLAY);
}
set_process_name("the ol' forker");
setup_deathsig(ppid);
reset_signals();
@ -409,9 +411,11 @@ impl Forker {
fn handle_set_env(self: &Rc<Self>, var: &[u8], val: Option<Vec<u8>>) {
let var = OsStr::from_bytes(var);
match val {
Some(val) => env::set_var(var, OsStr::from_bytes(&val)),
_ => env::remove_var(var),
unsafe {
match val {
Some(val) => env::set_var(var, OsStr::from_bytes(&val)),
_ => env::remove_var(var),
}
}
}
@ -513,9 +517,11 @@ impl Forker {
c::signal(c::SIGCHLD, c::SIG_DFL);
}
for (key, val) in env {
match val {
None => env::remove_var(&key),
Some(val) => env::set_var(&key, &val),
unsafe {
match val {
None => env::remove_var(&key),
Some(val) => env::set_var(&key, &val),
}
}
}
let prog = prog.into_ustr();