1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
// Copyright 2021-2022 Vector 35 Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
use std::env;
use std::ffi::CString;
use std::path::PathBuf;
#[cfg(not(target_os = "windows"))]
fn binja_path() -> PathBuf {
use std::ffi::{CStr, OsStr};
use std::mem;
use std::os::raw;
use std::os::unix::ffi::OsStrExt;
#[repr(C)]
struct DlInfo {
dli_fname: *const raw::c_char,
dli_fbase: *mut raw::c_void,
dli_sname: *const raw::c_char,
dli_saddr: *mut raw::c_void,
}
if let Ok(p) = env::var("BINJA_DIR") {
return PathBuf::from(p);
}
extern "C" {
fn dladdr(addr: *mut raw::c_void, info: *mut DlInfo) -> raw::c_int;
}
unsafe {
let mut info: DlInfo = mem::zeroed();
if dladdr(BNSetBundledPluginDirectory as *mut _, &mut info) == 0 {
panic!("Failed to find libbinaryninjacore path!");
}
if info.dli_fname.is_null() {
panic!("Failed to find libbinaryninjacore path!");
}
let path = CStr::from_ptr(info.dli_fname);
let path = OsStr::from_bytes(path.to_bytes());
let mut path = PathBuf::from(path);
path.pop();
path
}
}
#[cfg(target_os = "windows")]
fn binja_path() -> PathBuf {
PathBuf::from(env::var("PROGRAMFILES").unwrap()).join("Vector35\\BinaryNinja\\")
}
use binaryninjacore_sys::{BNInitPlugins, BNInitRepoPlugins, BNSetBundledPluginDirectory};
/// Loads plugins, core architecture, platform, etc.
///
/// ⚠️ Important! Must be called at the beginning of scripts. Plugins do not need to call this. ⚠️
pub fn init() {
unsafe {
let path = binja_path().join("plugins").into_os_string();
let path = CString::new(path.into_string().unwrap()).unwrap();
BNSetBundledPluginDirectory(path.as_ptr());
BNInitPlugins(true);
BNInitRepoPlugins();
}
}
/// Unloads plugins, stops all worker threads, and closes open logs
///
/// ⚠️ Important! Must be called at the end of scripts. ⚠️
pub fn shutdown() {
unsafe { binaryninjacore_sys::BNShutdown() };
}
/// Prelued-postlued helper function (calls [`init`] and [`shutdown`] for you)
/// ```rust
/// fn main() {
/// binaryninja::headless::script_helper(|| {
/// binaryninja::open_view("/bin/cat")
/// .expect("Couldn't open `/bin/cat`")
/// .iter()
/// .for_each(|func| println!(" `{}`", func.symbol().full_name()));
/// });
/// }
/// ```
pub fn script_helper(func: fn()) {
init();
func();
shutdown();
}
|
