leyoda/cosmic-comp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use to_ne_bytes() instead of unsound Vec::from_raw_parts

Browse source 
`from_raw_parts()` is only sound if the alignment is the same, so it
can't be used to covert to a `Vec<u8>` from a type with a greater
alignment.

This isn't hard to avoid, luckily.
This commit is contained in:
Ian Douglas Scott 2024-10-11 12:00:47 -07:00 • committed by Victoria Brekenfeld
parent 50d6dc3d21
commit 540ed3d170
4 changed files with 43 additions and 78 deletions
Download patch file Download diff file Expand all files Collapse all files

36
src/wayland/protocols/screencopy.rs
View file

@ -139,14 +139,10 @@ impl Session {
let node = Vec::from(dma.node.dev_id().to_ne_bytes());
self.obj.dmabuf_device(node);
for (fmt, modifiers) in &dma.formats {
let mut modifiers = modifiers.clone();
let modifiers: Vec<u8> = {
let ptr = modifiers.as_mut_ptr() as *mut u8;
let len = modifiers.len() * 4;
let cap = modifiers.capacity() * 4;
std::mem::forget(modifiers);
unsafe { Vec::from_raw_parts(ptr, len, cap) }
};
let modifiers = modifiers
.iter()
.flat_map(|modifier| u64::from(*modifier).to_ne_bytes())
.collect::<Vec<u8>>();
self.obj.dmabuf_format(*fmt as u32, modifiers);
}
}
@ -249,14 +245,10 @@ impl CursorSession {
let node = Vec::from(dma.node.dev_id().to_ne_bytes());
session_obj.dmabuf_device(node);
for (fmt, modifiers) in &dma.formats {
let mut modifiers = modifiers.clone();
let modifiers: Vec<u8> = {
let ptr = modifiers.as_mut_ptr() as *mut u8;
let len = modifiers.len() * 4;
let cap = modifiers.capacity() * 4;
std::mem::forget(modifiers);
unsafe { Vec::from_raw_parts(ptr, len, cap) }
};
let modifiers = modifiers
.iter()
.flat_map(|modifier| u64::from(*modifier).to_ne_bytes())
.collect::<Vec<u8>>();
session_obj.dmabuf_format(*fmt as u32, modifiers);
}
}
@ -747,14 +739,10 @@ where
let node = Vec::from(dma.node.dev_id().to_ne_bytes());
session.dmabuf_device(node);
for (fmt, modifiers) in &dma.formats {
let mut modifiers = modifiers.clone();
let modifiers: Vec<u8> = {
let ptr = modifiers.as_mut_ptr() as *mut u8;
let len = modifiers.len() * 4;
let cap = modifiers.capacity() * 4;
std::mem::forget(modifiers);
unsafe { Vec::from_raw_parts(ptr, len, cap) }
};
let modifiers = modifiers
.iter()
.flat_map(|modifier| u64::from(*modifier).to_ne_bytes())
.collect::<Vec<u8>>();
session.dmabuf_format(*fmt as u32, modifiers);
}
}

12
src/wayland/protocols/toplevel_info.rs
View file

@ -509,14 +509,10 @@ where
}
handle_state.states = states.clone();
let states: Vec<u8> = {
let ratio = std::mem::size_of::<States>() / std::mem::size_of::<u8>();
let ptr = states.as_mut_ptr() as *mut u8;
let len = states.len() * ratio;
let cap = states.capacity() * ratio;
std::mem::forget(states);
unsafe { Vec::from_raw_parts(ptr, len, cap) }
};
let states = states
.iter()
.flat_map(|state| (*state as u32).to_ne_bytes())
.collect::<Vec<u8>>();
instance.state(states);
changed = true;
}

16
src/wayland/protocols/toplevel_management.rs
View file

@ -151,15 +151,13 @@ where
data_init: &mut DataInit<'_, D>,
) {
let instance = data_init.init(resource, ());
let capabilities = {
let mut caps = state.toplevel_management_state().capabilities.clone();
let ratio = std::mem::size_of::<ManagementCapabilities>() / std::mem::size_of::<u8>();
let ptr = caps.as_mut_ptr() as *mut u8;
let len = caps.len() * ratio;
let cap = caps.capacity() * ratio;
std::mem::forget(caps);
unsafe { Vec::from_raw_parts(ptr, len, cap) }
};
let capabilities = state
.toplevel_management_state()
.capabilities
.iter()
.flat_map(|cap| (*cap as u32).to_ne_bytes())
.collect::<Vec<u8>>();
instance.capabilities(capabilities);
state.toplevel_management_state().instances.push(instance);
}

57
src/wayland/protocols/workspace.rs
View file

@ -929,15 +929,11 @@ where
}
if handle_state.capabilities != group.capabilities {
let caps: Vec<u8> = {
let mut caps = group.capabilities.clone();
let ratio = std::mem::size_of::<GroupCapabilities>() / std::mem::size_of::<u8>();
let ptr = caps.as_mut_ptr() as *mut u8;
let len = caps.len() * ratio;
let cap = caps.capacity() * ratio;
std::mem::forget(caps);
unsafe { Vec::from_raw_parts(ptr, len, cap) }
};
let caps = group
.capabilities
.iter()
.flat_map(|cap| (*cap as u32).to_ne_bytes())
.collect::<Vec<u8>>();
instance.capabilities(caps);
handle_state.capabilities = group.capabilities.clone();
changed = true;
@ -1005,44 +1001,31 @@ where
changed = true;
}
if handle_state.coordinates != workspace.coordinates {
let coords: Vec<u8> = {
let mut coords = workspace.coordinates.clone();
let ratio = std::mem::size_of::<u32>() / std::mem::size_of::<u8>();
let ptr = coords.as_mut_ptr() as *mut u8;
let len = coords.len() * ratio;
let cap = coords.capacity() * ratio;
std::mem::forget(coords);
unsafe { Vec::from_raw_parts(ptr, len, cap) }
};
let coords = workspace
.coordinates
.iter()
.flat_map(|coord| coord.to_ne_bytes())
.collect::<Vec<u8>>();
instance.coordinates(coords);
handle_state.coordinates = workspace.coordinates.clone();
changed = true;
}
if handle_state.capabilities != workspace.capabilities {
let caps: Vec<u8> = {
let mut caps = workspace.capabilities.clone();
let ratio = std::mem::size_of::<WorkspaceCapabilities>() / std::mem::size_of::<u8>();
let ptr = caps.as_mut_ptr() as *mut u8;
let len = caps.len() * ratio;
let cap = caps.capacity() * ratio;
std::mem::forget(caps);
unsafe { Vec::from_raw_parts(ptr, len, cap) }
};
let caps = workspace
.capabilities
.iter()
.flat_map(|cap| (*cap as u32).to_ne_bytes())
.collect::<Vec<u8>>();
instance.capabilities(caps);
handle_state.capabilities = workspace.capabilities.clone();
changed = true;
}
if handle_state.states != workspace.states {
let states: Vec<u8> = {
let mut states = workspace.states.iter().cloned().collect::<Vec<_>>();
let ratio = std::mem::size_of::<zcosmic_workspace_handle_v1::State>()
/ std::mem::size_of::<u8>();
let ptr = states.as_mut_ptr() as *mut u8;
let len = states.len() * ratio;
let cap = states.capacity() * ratio;
std::mem::forget(states);
unsafe { Vec::from_raw_parts(ptr, len, cap) }
};
let states = workspace
.states
.iter()
.flat_map(|state| (*state as u32).to_ne_bytes())
.collect::<Vec<u8>>();
instance.state(states);
handle_state.states = workspace.states.clone();
changed = true;