151 lines
3.9 KiB
Rust
151 lines
3.9 KiB
Rust
// Copyright 2013-2014 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution and at
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
//! A wrapper around LLVM's archive (.a) code
|
|
|
|
use ArchiveRef;
|
|
|
|
use std::ffi::CString;
|
|
use std::marker;
|
|
use std::path::Path;
|
|
use std::slice;
|
|
use std::str;
|
|
|
|
pub struct ArchiveRO {
|
|
ptr: ArchiveRef,
|
|
}
|
|
|
|
unsafe impl Send for ArchiveRO {}
|
|
|
|
pub struct Iter<'a> {
|
|
archive: &'a ArchiveRO,
|
|
ptr: ::ArchiveIteratorRef,
|
|
}
|
|
|
|
pub struct Child<'a> {
|
|
ptr: ::ArchiveChildRef,
|
|
_data: marker::PhantomData<&'a ArchiveRO>,
|
|
}
|
|
|
|
impl ArchiveRO {
|
|
/// Opens a static archive for read-only purposes. This is more optimized
|
|
/// than the `open` method because it uses LLVM's internal `Archive` class
|
|
/// rather than shelling out to `ar` for everything.
|
|
///
|
|
/// If this archive is used with a mutable method, then an error will be
|
|
/// raised.
|
|
pub fn open(dst: &Path) -> Result<ArchiveRO, String> {
|
|
return unsafe {
|
|
let s = path2cstr(dst);
|
|
let ar = ::LLVMRustOpenArchive(s.as_ptr());
|
|
if ar.is_null() {
|
|
Err(::last_error().unwrap_or("failed to open archive".to_string()))
|
|
} else {
|
|
Ok(ArchiveRO { ptr: ar })
|
|
}
|
|
};
|
|
|
|
#[cfg(unix)]
|
|
fn path2cstr(p: &Path) -> CString {
|
|
use std::os::unix::prelude::*;
|
|
use std::ffi::OsStr;
|
|
let p: &OsStr = p.as_ref();
|
|
CString::new(p.as_bytes()).unwrap()
|
|
}
|
|
#[cfg(windows)]
|
|
fn path2cstr(p: &Path) -> CString {
|
|
CString::new(p.to_str().unwrap()).unwrap()
|
|
}
|
|
}
|
|
|
|
pub fn raw(&self) -> ArchiveRef {
|
|
self.ptr
|
|
}
|
|
|
|
pub fn iter(&self) -> Iter {
|
|
unsafe {
|
|
Iter {
|
|
ptr: ::LLVMRustArchiveIteratorNew(self.ptr),
|
|
archive: self,
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Drop for ArchiveRO {
|
|
fn drop(&mut self) {
|
|
unsafe {
|
|
::LLVMRustDestroyArchive(self.ptr);
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'a> Iterator for Iter<'a> {
|
|
type Item = Result<Child<'a>, String>;
|
|
|
|
fn next(&mut self) -> Option<Result<Child<'a>, String>> {
|
|
let ptr = unsafe { ::LLVMRustArchiveIteratorNext(self.ptr) };
|
|
if ptr.is_null() {
|
|
::last_error().map(Err)
|
|
} else {
|
|
Some(Ok(Child {
|
|
ptr,
|
|
_data: marker::PhantomData,
|
|
}))
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'a> Drop for Iter<'a> {
|
|
fn drop(&mut self) {
|
|
unsafe {
|
|
::LLVMRustArchiveIteratorFree(self.ptr);
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'a> Child<'a> {
|
|
pub fn name(&self) -> Option<&'a str> {
|
|
unsafe {
|
|
let mut name_len = 0;
|
|
let name_ptr = ::LLVMRustArchiveChildName(self.ptr, &mut name_len);
|
|
if name_ptr.is_null() {
|
|
None
|
|
} else {
|
|
let name = slice::from_raw_parts(name_ptr as *const u8, name_len as usize);
|
|
str::from_utf8(name).ok().map(|s| s.trim())
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn data(&self) -> &'a [u8] {
|
|
unsafe {
|
|
let mut data_len = 0;
|
|
let data_ptr = ::LLVMRustArchiveChildData(self.ptr, &mut data_len);
|
|
if data_ptr.is_null() {
|
|
panic!("failed to read data from archive child");
|
|
}
|
|
slice::from_raw_parts(data_ptr as *const u8, data_len as usize)
|
|
}
|
|
}
|
|
|
|
pub fn raw(&self) -> ::ArchiveChildRef {
|
|
self.ptr
|
|
}
|
|
}
|
|
|
|
impl<'a> Drop for Child<'a> {
|
|
fn drop(&mut self) {
|
|
unsafe {
|
|
::LLVMRustArchiveChildFree(self.ptr);
|
|
}
|
|
}
|
|
}
|