reqrio 0.3.0-rc2

A lightweight, high-performance, fingerprint-based HTTP request library.
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

use crate::error::HlsResult;
use std::ffi::{c_char, CStr, CString};
use std::ptr::null_mut;
use crate::{json, HlsError, Proxy, ScReq, Url, WebSocket, WebSocketBuilder, WsFrame, WsOpcode};

#[unsafe(no_mangle)]
pub extern "system" fn ws_build() -> *mut WebSocketBuilder<ScReq> {
    Box::into_raw(Box::new(WebSocket::sync_build()))
}

#[unsafe(no_mangle)]
pub extern "system" fn ws_add_header(builder: *mut WebSocketBuilder<ScReq>, name: *const c_char, value: *const c_char) -> i32 {
    || -> HlsResult<i32>{
        let builder = unsafe { builder.as_mut().ok_or(HlsError::NullPointer) }?;
        let name = unsafe { CStr::from_ptr(name) }.to_str()?;
        let value = unsafe { CStr::from_ptr(value) }.to_str()?;
        builder.add_header(name, value)?;
        Ok(0)
    }().unwrap_or(-1)
}

#[unsafe(no_mangle)]
pub extern "system" fn ws_set_proxy(builder: *mut WebSocketBuilder<ScReq>, proxy: *const c_char) -> i32 {
    || -> HlsResult<i32>{
        let builder = unsafe { builder.as_mut().ok_or(HlsError::NullPointer) }?;
        let proxy = unsafe { CStr::from_ptr(proxy) }.to_str()?;
        let url = Url::try_from(proxy)?;
        builder.set_proxy(Proxy::HttpPlain(url));
        Ok(0)
    }().unwrap_or(-1)
}

#[unsafe(no_mangle)]
pub extern "system" fn ws_set_uri(builder: *mut WebSocketBuilder<ScReq>, uri: *const c_char) -> i32 {
    || -> HlsResult<i32>{
        let builder = unsafe { builder.as_mut().ok_or(HlsError::NullPointer) }?;
        let uri = unsafe { CStr::from_ptr(uri) }.to_str()?;
        builder.set_uri(uri)?;
        Ok(0)
    }().unwrap_or(-1)
}

#[unsafe(no_mangle)]
pub extern "system" fn ws_open(builder: *mut WebSocketBuilder<ScReq>, url: *const Url) -> *mut WebSocket {
    || -> HlsResult<*mut WebSocket>{
        let builder = unsafe { Box::from_raw(builder) };
        let url=unsafe{url.as_ref()}.ok_or(HlsError::NullPointer)?;
        let ws = builder.build(url)?;
        Ok(Box::into_raw(Box::new(ws)))
    }().unwrap_or_else(|e| {
        println!("{}", e);
        null_mut()
    })
}

#[unsafe(no_mangle)]
pub extern "system" fn ws_open_raw(url: *const c_char, context: *const c_char) -> *mut WebSocket {
    || -> HlsResult<*mut WebSocket>{
        let url = unsafe { CStr::from_ptr(url) }.to_str()?;
        let context = unsafe { CStr::from_ptr(context) }.to_bytes();
        let ws = WebSocket::open_raw(url, context)?;
        Ok(Box::into_raw(Box::new(ws)))
    }().unwrap_or(null_mut())
}

#[unsafe(no_mangle)]
pub extern "system" fn ws_read(websocket: *mut WebSocket) -> *mut c_char {
    || -> HlsResult<*mut c_char>{
        let websocket = unsafe { websocket.as_mut() }.ok_or(HlsError::NullPointer)?;
        let frame = websocket.read_frame()?;
        let res = json::object! {
            "opcode":*frame.frame_type().op_code() as u8,
            "payload":frame.payload().as_bytes(),
        };
        let res = CString::new(res.dump()).unwrap();
        Ok(res.into_raw())
    }().unwrap_or_else(|e| {
        let res = CString::new(e.to_string()).unwrap();
        res.into_raw()
    })
}

#[unsafe(no_mangle)]
pub extern "system" fn ws_write(websocket: *mut WebSocket, op_code: i32, mask: bool, payload: *const c_char) -> i32 {
    || -> HlsResult<i32> {
        let websocket = unsafe { websocket.as_mut() }.ok_or(HlsError::NullPointer)?;
        let payload = unsafe { CStr::from_ptr(payload) }.to_bytes();
        let opcode = WsOpcode::from_u8(op_code as u8).ok_or("opcode unknow")?;
        let frame = WsFrame::new_frame(opcode, mask, payload);
        websocket.write_frame(frame)?;
        Ok(0)
    }().unwrap_or(-1)
}

#[unsafe(no_mangle)]
pub extern "system" fn ws_close(websocket: *mut WebSocket) {
    println!("{}", websocket.is_null());
    if websocket.is_null() { return; }
    let websocket = unsafe { Box::from_raw(websocket) };
    let _ = websocket.shutdown();
}