Skip to main content

pyo3_ffi/cpython/
bytearrayobject.rs

1#[cfg(Py_GIL_DISABLED)]
2use crate::cpython::pyatomic::_Py_atomic_load_ssize_relaxed;
3use crate::object::*;
4use crate::pyport::Py_ssize_t;
5use crate::PyByteArray_Check;
6#[cfg(not(any(PyPy, GraalPy)))]
7use core::ffi::c_char;
8
9#[cfg(not(any(PyPy, GraalPy, Py_LIMITED_API)))]
10#[repr(C)]
11pub struct PyByteArrayObject {
12    pub ob_base: PyVarObject,
13    pub ob_alloc: Py_ssize_t,
14    pub ob_bytes: *mut c_char,
15    pub ob_start: *mut c_char,
16    pub ob_exports: Py_ssize_t,
17    #[cfg(Py_3_15)]
18    pub ob_bytes_object: *mut PyObject,
19}
20
21#[cfg(any(PyPy, GraalPy))]
22opaque_struct!(pub PyByteArrayObject);
23
24#[inline]
25pub(crate) unsafe fn _PyByteArray_CAST(op: *mut PyObject) -> *mut PyByteArrayObject {
26    debug_assert_eq!(PyByteArray_Check(op), 1);
27    op.cast()
28}
29
30#[inline]
31#[cfg(not(any(PyPy, GraalPy)))]
32pub unsafe fn PyByteArray_AS_STRING(op: *mut PyObject) -> *mut c_char {
33    (*_PyByteArray_CAST(op)).ob_start
34}
35
36#[inline]
37pub unsafe fn PyByteArray_GET_SIZE(op: *mut PyObject) -> Py_ssize_t {
38    let byte_array = _PyByteArray_CAST(op);
39    #[cfg(Py_GIL_DISABLED)]
40    {
41        _Py_atomic_load_ssize_relaxed(&raw const (*_PyVarObject_CAST(byte_array.cast())).ob_size)
42    }
43    #[cfg(not(Py_GIL_DISABLED))]
44    {
45        Py_SIZE(byte_array.cast())
46    }
47}
⚠️ Internal Docs ⚠️ Not Public API 👉 Official Docs Here