update agent
This commit is contained in:
23
src/main.rs
23
src/main.rs
@@ -1,5 +1,5 @@
|
||||
use libc::{
|
||||
c_char, c_int, c_void, mmap, perror, MAP_FAILED, MAP_PRIVATE, MAP_SHARED, PROT_READ, PROT_WRITE,
|
||||
c_int, c_void, mmap, MAP_FAILED, MAP_SHARED, PROT_READ, PROT_WRITE,
|
||||
};
|
||||
use std::fs::File;
|
||||
use std::os::fd::AsRawFd;
|
||||
@@ -9,7 +9,7 @@ const SHARED_MEMORY_SIZE: usize = 1024;
|
||||
|
||||
#[link(name = "agent")]
|
||||
extern "C" {
|
||||
fn agent_start(shared_memory: *mut u8, sync_memory: *mut u8, private_mem: *mut u8);
|
||||
fn agent_start(shared_memory: *mut u8, sync_memory: *mut u8, private_mem_fd: u64);
|
||||
}
|
||||
|
||||
fn main() {
|
||||
@@ -51,21 +51,12 @@ fn main() {
|
||||
.read(true)
|
||||
.write(true)
|
||||
.open(&args[2])
|
||||
.expect("Cannot open KEY_FILE");
|
||||
|
||||
let private_mem = unsafe {
|
||||
mmap(
|
||||
ptr::null_mut() as *mut c_void,
|
||||
32 * 100,
|
||||
PROT_READ | PROT_WRITE,
|
||||
MAP_SHARED,
|
||||
private_file.as_raw_fd(),
|
||||
0,
|
||||
)
|
||||
} as *mut u8;
|
||||
assert_ne!(private_mem, MAP_FAILED as *mut u8);
|
||||
.expect("Cannot open KEY_FILE")
|
||||
.as_raw_fd()
|
||||
.try_into()
|
||||
.unwrap();
|
||||
|
||||
unsafe {
|
||||
agent_start(shared_memory, sync_memory, private_mem);
|
||||
agent_start(shared_memory, sync_memory, private_file);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user