Skip to content

Commit

Permalink
fix: #5165
Browse files Browse the repository at this point in the history
  • Loading branch information
ethe committed Oct 7, 2024
1 parent d74bf7c commit 77193ff
Show file tree
Hide file tree
Showing 3 changed files with 30 additions and 11 deletions.
4 changes: 4 additions & 0 deletions bindings/c/include/opendal.h
Original file line number Diff line number Diff line change
Expand Up @@ -99,6 +99,10 @@ typedef struct opendal_bytes {
* The length of the byte array
*/
uintptr_t len;
/**
* The capacity of the byte array
*/
uintptr_t capacity;
} opendal_bytes;

/**
Expand Down
26 changes: 19 additions & 7 deletions bindings/c/src/types.rs
Original file line number Diff line number Diff line change
Expand Up @@ -34,25 +34,37 @@ pub struct opendal_bytes {
pub data: *const u8,
/// The length of the byte array
pub len: usize,
/// The capacity of the byte array
pub capacity: usize,
}

impl opendal_bytes {
/// Construct a [`opendal_bytes`] from the Rust [`Vec`] of bytes
pub(crate) fn new(buf: Buffer) -> Self {
let vec = buf.to_vec();
let data = vec.as_ptr();
let len = vec.len();
std::mem::forget(vec);
Self { data, len }
let mut buf = std::mem::ManuallyDrop::new(vec);
let data = buf.as_mut_ptr();
let len = buf.len();
let capacity = buf.capacity();
Self {
data,
len,
capacity,
}
}

/// \brief Frees the heap memory used by the opendal_bytes
#[no_mangle]
pub unsafe extern "C" fn opendal_bytes_free(ptr: *mut opendal_bytes) {
if !ptr.is_null() {
let data_mut = (*ptr).data as *mut u8;
drop(Vec::from_raw_parts(data_mut, (*ptr).len, (*ptr).len));
drop(Box::from_raw(ptr));
// transmuting `*const u8` to `*mut u8` is undefined behavior in any cases
// however, fields type of `opendal_bytes` is already related to the zig binding
// it should be fixed later
let _ = Vec::from_raw_parts((*ptr).data as *mut u8, (*ptr).len, (*ptr).capacity);
// it is too weird that call `Box::new` outside `opendal_bytes::new` but dealloc it here
// also, boxing `opendal_bytes` might not be necessary
// `data` points to heap, so `opendal_bytes` could be passed as a stack value
let _ = Box::from_raw(ptr);
}
}
}
Expand Down
11 changes: 7 additions & 4 deletions bindings/go/types.go
Original file line number Diff line number Diff line change
Expand Up @@ -257,8 +257,9 @@ type resultStat struct {
type opendalMetadata struct{}

type opendalBytes struct {
data *byte
len uintptr
data *byte
len uintptr
capacity uintptr
}

type opendalError struct {
Expand All @@ -285,15 +286,17 @@ type opendalEntry struct{}
func toOpendalBytes(data []byte) opendalBytes {
var ptr *byte
l := len(data)
c := cap(data)
if l > 0 {
ptr = &data[0]
} else {
var b byte
ptr = &b
}
return opendalBytes{
data: ptr,
len: uintptr(l),
data: ptr,
len: uintptr(l),
capacity: uintptr(c),
}
}

Expand Down

0 comments on commit 77193ff

Please sign in to comment.