Skip to content
This repository has been archived by the owner on Jun 28, 2022. It is now read-only.

[PROF-4780] Breaking change: Change FFI File struct to contain a Buffer instead of a ByteSlice #33

Merged
merged 5 commits into from
Mar 11, 2022
Merged
Show file tree
Hide file tree
Changes from 4 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
88 changes: 58 additions & 30 deletions ddprof-ffi/src/exporter.rs
Original file line number Diff line number Diff line change
Expand Up @@ -121,19 +121,6 @@ pub unsafe extern "C" fn buffer_reset(buffer: *mut Buffer) {
}
}

/// Used to convert "external" data (e.g. not originated in libddprof) into a libddprof buffer, for
/// instance, when you want to attach JSON data (such as code_provenance.json or metrics.json) to a
/// profile.
///
/// The resulting buffer should be treated by the caller as opaque data that can only be manipulated
/// and accessed by libddprof.
#[no_mangle]
#[must_use]
pub unsafe extern "C" fn ddprof_ffi_Buffer_from_byte_slice(buffer: ByteSlice) -> Box<Buffer> {
let buffer_slice: &[u8] = buffer.into();
Box::new(Buffer::from_vec(Vec::from(buffer_slice)))
}

/// Destroys the Exporter.
#[export_name = "ddprof_ffi_Exporter_delete"]
pub extern "C" fn exporter_delete(exporter: Option<Box<Exporter>>) {
Expand All @@ -155,7 +142,7 @@ pub enum EndpointV3<'a> {
#[repr(C)]
pub struct File<'a> {
name: ByteSlice<'a>,
file: Option<NonNull<Buffer>>,
file: ByteSlice<'a>,
}

/// This type only exists to workaround a bug in cbindgen; may be removed in the
Expand Down Expand Up @@ -290,7 +277,7 @@ unsafe fn try_into_vec_files<'a>(slice: Slice<'a, File>) -> Option<Vec<ddprof_ex

for file in slice.into_slice().iter() {
let name = file.name.try_into().ok()?;
let bytes: &[u8] = file.file.as_ref()?.as_ref().as_slice();
let bytes: &[u8] = file.file.into_slice();
vec.push(ddprof_exporter::File { name, bytes });
}
Some(vec)
Expand Down Expand Up @@ -372,6 +359,16 @@ mod test {
use crate::exporter::*;
use crate::Slice;

fn family() -> ByteSlice<'static> {
ByteSlice::new("native".as_ptr(), "native".len())
}
fn base_url() -> &'static str {
"https://localhost:1337"
}
fn endpoint() -> ByteSlice<'static> {
ByteSlice::new(base_url().as_ptr(), base_url().len())
}

#[test]
fn exporter_new_and_delete() {
let exporter = exporter_new();
Expand All @@ -391,17 +388,16 @@ mod test {

#[test]
fn profile_exporter_v3_new_and_delete() {
let family = ByteSlice::new("native".as_ptr(), "native".len());

let tags = [Tag {
name: ByteSlice::new("host".as_ptr(), "host".len()),
value: ByteSlice::new("localhost".as_ptr(), "localhost".len()),
}];

let base_url = "https://localhost:1337";
let endpoint = endpoint_agent(ByteSlice::new(base_url.as_ptr(), base_url.len()));

let result = profile_exporter_new(family, Slice::new(tags.as_ptr(), tags.len()), endpoint);
let result = profile_exporter_new(
family(),
Slice::new(tags.as_ptr(), tags.len()),
endpoint_agent(endpoint()),
);

match result {
NewProfileExporterV3Result::Ok(exporter) => unsafe {
Expand All @@ -415,17 +411,49 @@ mod test {
}

#[test]
fn byte_slice_to_buffer_conversion() {
let raw: &mut [u8] = &mut [104, 101, 108, 108, 111]; // "hello"
let slice = Slice::new(raw.as_ptr(), raw.len());
fn profile_exporter_v3_build() {
let exporter_result =
profile_exporter_new(family(), Slice::default(), endpoint_agent(endpoint()));

let exporter = match exporter_result {
NewProfileExporterV3Result::Ok(exporter) => unsafe {
Some(NonNull::new_unchecked(exporter))
},
NewProfileExporterV3Result::Err(message) => {
std::mem::drop(message);
panic!("Should not occur!")
}
};

let converted: Box<Buffer> = unsafe { ddprof_ffi_Buffer_from_byte_slice(slice) };
let converted_contents: &[u8] = unsafe { (*converted).as_slice() };
let files = [File {
name: ByteSlice::new("foo.pprof".as_ptr(), "foo.pprof".len()),
file: ByteSlice::new("dummy contents".as_ptr(), "dummy contents".len()),
}];

let start = Timespec {
seconds: 12,
nanoseconds: 34,
};
let finish = Timespec {
seconds: 56,
nanoseconds: 78,
};
let timeout_milliseconds = 90;

let maybe_request = unsafe {
profile_exporter_build(
exporter,
start,
finish,
Slice::new(files.as_ptr(), files.len()),
timeout_milliseconds,
)
};

raw.reverse(); // mutate the original buffer, to validate that a copy is really being made
assert!(maybe_request.is_some());

assert_eq!(b"hello", converted_contents);
assert_eq!(b"olleh", raw);
std::mem::drop(converted);
// TODO: Currently, we're only testing that a request was built (building did not fail), but we have no
// coverage for the request actually being correct.
// It'd be nice to actually perform the request, capture its contents, and assert that they are as expected
}
}
1 change: 1 addition & 0 deletions examples/ffi/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ cmake_minimum_required(VERSION 3.19)
find_package(DDProf)

add_executable(exporter exporter.cpp)
target_compile_features(exporter PRIVATE cxx_std_11)
target_link_libraries(exporter DDProf::FFI)
add_executable(profiles profiles.c)
target_link_libraries(profiles DDProf::FFI)
10 changes: 2 additions & 8 deletions examples/ffi/exporter.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -120,15 +120,9 @@ int main(int argc, char* argv[])

Holder<ddprof_ffi_ProfileExporterV3> exporter{exporter_new_result.ok};

ddprof_ffi_Buffer profile_buffer = {
.ptr = encoded_profile->buffer.ptr,
.len = encoded_profile->buffer.len,
.capacity = encoded_profile->buffer.capacity,
};

ddprof_ffi_File files_[] = {{
.name = to_byteslice("auto.pprof"),
.file = &profile_buffer,
.file = (ddprof_ffi_ByteSlice) {.ptr = encoded_profile->buffer.ptr, .len = encoded_profile->buffer.len},
ivoanjo marked this conversation as resolved.
Show resolved Hide resolved
}};

ddprof_ffi_Slice_file files = {
Expand All @@ -147,4 +141,4 @@ int main(int argc, char* argv[])
{
printf("Response code: %d\n", send_result.http_response.code);
}
}
}