redirectionio 2.5.2

Redirection IO Library to handle matching rule, redirect and filtering headers and body.
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

use crate::ffi_helpers::string_to_c_char;
use log::{Metadata, Record};
use std::os::raw::{c_char, c_short, c_void};
use std::sync::Once;

#[allow(non_camel_case_types)]
pub type redirectionio_log_callback = extern "C" fn(*const c_char, *const c_void, c_short);

pub struct CallbackLogger {
    pub callback: Option<redirectionio_log_callback>,
    pub data: Option<&'static c_void>,
}

impl log::Log for CallbackLogger {
    fn enabled(&self, _metadata: &Metadata) -> bool {
        true
    }

    fn log(&self, record: &Record) {
        if self.callback.is_none() {
            return;
        }

        if self.data.is_none() {
            return;
        }

        if self.enabled(record.metadata()) {
            let log_str = format!("{} - {}", record.level(), record.args());
            let cstr = unsafe { string_to_c_char(log_str) };

            (self.callback.unwrap())(cstr, self.data.unwrap(), record.level() as i16);
        }
    }

    fn flush(&self) {}
}

static mut LOGGER: CallbackLogger = CallbackLogger {
    callback: None,
    data: None,
};

static INIT: Once = Once::new();

#[no_mangle]
pub extern "C" fn redirectionio_log_init_stderr() {
    stderrlog::new().init().unwrap();
}

#[no_mangle]
pub unsafe extern "C" fn redirectionio_log_init_with_callback(callback: redirectionio_log_callback, data: &'static c_void) {
    LOGGER.callback = Some(callback);
    LOGGER.data = Some(data);

    INIT.call_once(|| {
        log::set_logger(&LOGGER)
            .map(|()| log::set_max_level(log::LevelFilter::Trace))
            .expect("cannot set logger");
    });
}