1use cfg_if::cfg_if;
2use libc::c_int;
3use std::marker::PhantomData;
4use std::ptr;
5
6use crate::cvt_p;
7use crate::error::ErrorStack;
8use crate::util;
9
10pub struct MemBioSlice<'a>(*mut ffi::BIO, PhantomData<&'a [u8]>);
11
12impl Drop for MemBioSlice<'_> {
13 fn drop(&mut self) {
14 unsafe {
15 ffi::BIO_free_all(self.0);
16 }
17 }
18}
19
20impl<'a> MemBioSlice<'a> {
21 pub fn new(buf: &'a [u8]) -> Result<MemBioSlice<'a>, ErrorStack> {
22 ffi::init();
23
24 assert!(buf.len() <= c_int::MAX as usize);
25 let bio: *mut BIO = unsafe {
26 cvt_p(BIO_new_mem_buf(
27 buf.as_ptr() as *const _,
28 buf.len() as crate::SLenType,
29 ))?
30 };
31
32 Ok(MemBioSlice(bio, PhantomData))
33 }
34
35 pub fn as_ptr(&self) -> *mut ffi::BIO {
36 self.0
37 }
38}
39
40pub struct MemBio(*mut ffi::BIO);
41
42impl Drop for MemBio {
43 fn drop(&mut self) {
44 unsafe {
45 ffi::BIO_free_all(self.0);
46 }
47 }
48}
49
50impl MemBio {
51 pub fn new() -> Result<MemBio, ErrorStack> {
52 ffi::init();
53
54 let bio = unsafe { cvt_p(ffi::BIO_new(ffi::BIO_s_mem()))? };
55 Ok(MemBio(bio))
56 }
57
58 pub fn as_ptr(&self) -> *mut ffi::BIO {
59 self.0
60 }
61
62 pub fn get_buf(&self) -> &[u8] {
63 unsafe {
64 let mut ptr = ptr::null_mut();
65 let len = ffi::BIO_get_mem_data(self.0, &mut ptr);
66 util::from_raw_parts(ptr as *const _ as *const _, len as usize)
67 }
68 }
69
70 #[cfg(not(any(boringssl, awslc)))]
71 pub unsafe fn from_ptr(bio: *mut ffi::BIO) -> MemBio {
72 MemBio(bio)
73 }
74}
75
76cfg_if! {
77 if #[cfg(any(ossl102, boringssl, awslc))] {
78 use ffi::BIO_new_mem_buf;
79 } else {
80 #[allow(bad_style)]
81 unsafe fn BIO_new_mem_buf(buf: *const ::libc::c_void, len: ::libc::c_int) -> *mut ffi::BIO {
82 ffi::BIO_new_mem_buf(buf as *mut _, len)
83 }
84 }
85}
86
87#[cfg(test)]
88mod tests {
89 use super::MemBio;
90
91 #[test]
92 fn test_mem_bio_get_buf_empty() {
93 let b = MemBio::new().unwrap();
94 assert_eq!(b.get_buf(), &[]);
95 }
96}
97