2014-07-06 23:43:22 -05:00
|
|
|
// 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 libc;
|
|
|
|
use ArchiveRef;
|
|
|
|
|
2014-11-25 15:28:35 -06:00
|
|
|
use std::ffi::CString;
|
2015-03-13 03:56:18 -05:00
|
|
|
use std::slice;
|
2015-02-26 23:00:43 -06:00
|
|
|
use std::path::Path;
|
2014-07-06 23:43:22 -05:00
|
|
|
|
|
|
|
pub struct ArchiveRO {
|
|
|
|
ptr: ArchiveRef,
|
|
|
|
}
|
|
|
|
|
|
|
|
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) -> Option<ArchiveRO> {
|
2015-02-26 23:00:43 -06:00
|
|
|
return unsafe {
|
|
|
|
let s = path2cstr(dst);
|
2014-11-25 15:28:35 -06:00
|
|
|
let ar = ::LLVMRustOpenArchive(s.as_ptr());
|
2014-07-06 23:43:22 -05:00
|
|
|
if ar.is_null() {
|
|
|
|
None
|
|
|
|
} else {
|
|
|
|
Some(ArchiveRO { ptr: ar })
|
|
|
|
}
|
2015-02-26 23:00:43 -06:00
|
|
|
};
|
|
|
|
|
|
|
|
#[cfg(unix)]
|
|
|
|
fn path2cstr(p: &Path) -> CString {
|
|
|
|
use std::os::unix::prelude::*;
|
|
|
|
use std::ffi::AsOsStr;
|
|
|
|
CString::new(p.as_os_str().as_bytes()).unwrap()
|
|
|
|
}
|
|
|
|
#[cfg(windows)]
|
|
|
|
fn path2cstr(p: &Path) -> CString {
|
|
|
|
CString::new(p.to_str().unwrap()).unwrap()
|
2014-07-06 23:43:22 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Reads a file in the archive
|
|
|
|
pub fn read<'a>(&'a self, file: &str) -> Option<&'a [u8]> {
|
|
|
|
unsafe {
|
|
|
|
let mut size = 0 as libc::size_t;
|
2015-02-18 00:47:40 -06:00
|
|
|
let file = CString::new(file).unwrap();
|
2014-11-25 15:28:35 -06:00
|
|
|
let ptr = ::LLVMRustArchiveReadSection(self.ptr, file.as_ptr(),
|
|
|
|
&mut size);
|
2014-07-06 23:43:22 -05:00
|
|
|
if ptr.is_null() {
|
|
|
|
None
|
|
|
|
} else {
|
2015-03-13 03:56:18 -05:00
|
|
|
Some(slice::from_raw_parts(ptr as *const u8, size as uint))
|
2014-07-06 23:43:22 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Drop for ArchiveRO {
|
|
|
|
fn drop(&mut self) {
|
|
|
|
unsafe {
|
2014-07-07 19:58:01 -05:00
|
|
|
::LLVMRustDestroyArchive(self.ptr);
|
2014-07-06 23:43:22 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|