Skip to content

Commit c913bd2

Browse files
authored
Group unsafe BorrowedBuffer initialization (python#22)
1 parent c66a76e commit c913bd2

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

Modules/_base64/src/lib.rs

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -85,12 +85,15 @@ struct BorrowedBuffer {
8585
impl BorrowedBuffer {
8686
fn from_object(obj: &PyObject) -> Result<Self, ()> {
8787
let mut view = MaybeUninit::<Py_buffer>::uninit();
88-
if unsafe { PyObject_GetBuffer(obj.as_raw(), view.as_mut_ptr(), PYBUF_SIMPLE) } != 0 {
89-
return Err(());
90-
}
91-
Ok(Self {
92-
view: unsafe { view.assume_init() },
93-
})
88+
let buffer = unsafe {
89+
if PyObject_GetBuffer(obj.as_raw(), view.as_mut_ptr(), PYBUF_SIMPLE) != 0 {
90+
return Err(());
91+
}
92+
Self {
93+
view: view.assume_init(),
94+
}
95+
};
96+
Ok(buffer)
9497
}
9598

9699
fn len(&self) -> Py_ssize_t {

0 commit comments

Comments
 (0)