libhook 0.1.0

Hook system calls.
Documentation 
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

use base_coroutine::coroutine::UserFunc;
use base_coroutine::{EventLoop, JoinHandle};
use std::os::raw::c_void;

#[no_mangle]
pub extern "C" fn init_hook() {
    //啥都不做，只是为了保证hook的函数能够被重定向到
    //主要为了防止有的程序压根不调用coroutine_crate的情况
}

///创建协程
#[no_mangle]
pub extern "C" fn coroutine_crate(
    f: UserFunc<&'static mut c_void, (), &'static mut c_void>,
    param: &'static mut c_void,
    stack_size: usize,
) -> JoinHandle {
    match EventLoop::submit(f, param, stack_size) {
        Ok(handle) => handle,
        Err(_) => {
            //we had handled NULL in JoinHandle
            #[allow(invalid_value)]
            JoinHandle(unsafe { std::mem::transmute(0usize) })
        }
    }
}

///等待协程完成
#[no_mangle]
pub extern "C" fn coroutine_join(handle: JoinHandle) -> libc::c_long {
    match handle.join() {
        Ok(ptr) => ptr as libc::c_long,
        Err(_) => -1,
    }
}

///等待协程完成
#[no_mangle]
pub extern "C" fn coroutine_timeout_join(handle: &JoinHandle, ns_time: u64) -> libc::c_long {
    match handle.timeout_join(std::time::Duration::from_nanos(ns_time)) {
        Ok(ptr) => ptr as libc::c_long,
        Err(_) => -1,
    }
}

///轮询协程
#[no_mangle]
pub extern "C" fn try_timed_schedule(ns_time: u64) -> libc::c_int {
    let timeout_time = timer_utils::add_timeout_time(ns_time);
    match EventLoop::round_robin_timeout_schedule(timeout_time) {
        Ok(_) => 0,
        Err(_) => -1,
    }
}

#[no_mangle]
pub extern "C" fn timed_schedule(ns_time: u64) -> libc::c_int {
    let timeout_time = timer_utils::add_timeout_time(ns_time);
    match EventLoop::round_robin_timed_schedule(timeout_time) {
        Ok(_) => 0,
        Err(_) => -1,
    }
}