dpdu-wrapper 0.1.3

High-level Rust wrapper for D-PDU implementations.
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

use dpdu_api_types::PduError;

mod detours;
mod softing;
mod vxdiag;

pub fn wrap_pdu_call<F>(func: &str, mut f: F) -> PduError
where
    F: FnMut() -> PduError,
{
    register_detours();
    register_detour_callbacks();
    do_miscellaneous();

    let result = f();

    if vxdiag::has_device_not_connected() {
        return match func {
            "PDUModuleDisconnect"
            | "PDUGetTimestamp"
            | "PDUIoCtl"
            | "PDUGetVersion"
            | "PDUGetLastError"
            | "PDUCreateComLogicalLink"
            | "PDUDestroyComLogicalLink"
            | "PDUDisconnect"
            | "PDULockResource"
            | "PDUUnlockResource"
            | "PDUGetComParam"
            | "PDUSetComParam"
            | "PDUStartComPrimitive"
            | "PDUCancelComPrimitive"
            | "PDUGetEventItem"
            | "PDURegisterEventCallback"
            | "PDUGetUniqueRespIdTable"
            | "PDUSetUniqueRespIdTable" => PduError::ModuleNotConnected,
            _ => PduError::FctFailed,
        };
    }

    result
}

fn register_detours() {
    detours::msg_box_a::hook_message_box_a();
}

fn register_detour_callbacks() {
    vxdiag::register_callback();
}

fn do_miscellaneous() {
    softing::restore_protocol_dll();
}