blob: aaa73fa9eac35d7325a9493d4a9408e74d197324 (
plain)
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
|
use binaryninja::headless::Session;
use binaryninja::secrets_provider::{CoreSecretsProvider, SecretsProvider};
#[test]
fn list_secrets_provider() {
let _session = Session::new().expect("Failed to initialize session");
let providers = CoreSecretsProvider::all();
assert!(providers.len() > 0);
let providers_again = CoreSecretsProvider::all();
assert_eq!(providers.len(), providers_again.len());
}
struct MySecretsProvider {}
impl SecretsProvider for MySecretsProvider {
fn has_data(&mut self, key: &str) -> bool {
key == "my_key"
}
fn get_data(&mut self, key: &str) -> Option<String> {
if key == "my_key" {
Some("my_value".to_string())
} else {
None
}
}
fn store_data(&mut self, _key: &str, _data: &str) -> bool {
false
}
fn delete_data(&mut self, _key: &str) -> bool {
false
}
}
#[test]
fn custom_secrets_provider() {
let _session = Session::new().expect("Failed to initialize session");
let my_provider = CoreSecretsProvider::new("MySecretsProvider", MySecretsProvider {});
assert!(my_provider.has_data("my_key"));
assert!(!my_provider.has_data("not_my_key"));
assert_eq!(my_provider.get_data("my_key").as_str(), "my_value");
}
|
