use llvm_sys::core::{
LLVMCreateMemoryBufferWithContentsOfFile, LLVMCreateMemoryBufferWithMemoryRange,
LLVMCreateMemoryBufferWithMemoryRangeCopy, LLVMCreateMemoryBufferWithSTDIN, LLVMDisposeMemoryBuffer,
LLVMGetBufferSize, LLVMGetBufferStart,
};
use llvm_sys::object::LLVMCreateObjectFile;
use llvm_sys::prelude::LLVMMemoryBufferRef;
use crate::object_file::ObjectFile;
use crate::support::{to_c_str, LLVMString};
use std::mem::{forget, MaybeUninit};
use std::path::Path;
use std::ptr;
use std::slice;
#[derive(Debug)]
pub struct MemoryBuffer {
pub(crate) memory_buffer: LLVMMemoryBufferRef,
}
impl MemoryBuffer {
pub unsafe fn new(memory_buffer: LLVMMemoryBufferRef) -> Self {
assert!(!memory_buffer.is_null());
MemoryBuffer { memory_buffer }
}
pub fn as_mut_ptr(&self) -> LLVMMemoryBufferRef {
self.memory_buffer
}
pub fn create_from_file(path: &Path) -> Result<Self, LLVMString> {
let path = to_c_str(path.to_str().expect("Did not find a valid Unicode path string"));
let mut memory_buffer = ptr::null_mut();
let mut err_string = MaybeUninit::uninit();
let return_code = unsafe {
LLVMCreateMemoryBufferWithContentsOfFile(
path.as_ptr() as *const ::libc::c_char,
&mut memory_buffer,
err_string.as_mut_ptr(),
)
};
if return_code == 1 {
unsafe {
return Err(LLVMString::new(err_string.assume_init()));
}
}
unsafe { Ok(MemoryBuffer::new(memory_buffer)) }
}
pub fn create_from_stdin() -> Result<Self, LLVMString> {
let mut memory_buffer = ptr::null_mut();
let mut err_string = MaybeUninit::uninit();
let return_code = unsafe { LLVMCreateMemoryBufferWithSTDIN(&mut memory_buffer, err_string.as_mut_ptr()) };
if return_code == 1 {
unsafe {
return Err(LLVMString::new(err_string.assume_init()));
}
}
unsafe { Ok(MemoryBuffer::new(memory_buffer)) }
}
pub fn create_from_memory_range(input: &[u8], name: &str) -> Self {
let name_c_string = to_c_str(name);
let memory_buffer = unsafe {
LLVMCreateMemoryBufferWithMemoryRange(
input.as_ptr() as *const ::libc::c_char,
input.len(),
name_c_string.as_ptr(),
false as i32,
)
};
unsafe { MemoryBuffer::new(memory_buffer) }
}
pub fn create_from_memory_range_copy(input: &[u8], name: &str) -> Self {
let name_c_string = to_c_str(name);
let memory_buffer = unsafe {
LLVMCreateMemoryBufferWithMemoryRangeCopy(
input.as_ptr() as *const ::libc::c_char,
input.len(),
name_c_string.as_ptr(),
)
};
unsafe { MemoryBuffer::new(memory_buffer) }
}
pub fn as_slice(&self) -> &[u8] {
unsafe {
let start = LLVMGetBufferStart(self.memory_buffer);
slice::from_raw_parts(start as *const _, self.get_size())
}
}
pub fn get_size(&self) -> usize {
unsafe { LLVMGetBufferSize(self.memory_buffer) }
}
pub fn create_object_file(self) -> Result<ObjectFile, ()> {
let object_file = unsafe { LLVMCreateObjectFile(self.memory_buffer) };
forget(self);
if object_file.is_null() {
return Err(());
}
unsafe { Ok(ObjectFile::new(object_file)) }
}
}
impl Drop for MemoryBuffer {
fn drop(&mut self) {
unsafe {
LLVMDisposeMemoryBuffer(self.memory_buffer);
}
}
}