ort-openrouter-cli 0.4.7

Open Router CLI
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
102
103
104
105
106
107
108
109
110
111
112
113
114
115

//! ort: Open Router CLI
//! https://github.com/grahamking/ort
//!
//! MIT License
//! Copyright (c) 2025 Graham King

use core::ffi::{c_int, c_void};
use core::mem::size_of;
use core::net::{Ipv4Addr, SocketAddrV4};

use crate::{ErrorKind, OrtResult, Read, Write, ort_error, syscall, utils};

pub struct TcpSocket {
    fd: i32,
}

impl TcpSocket {
    pub fn new() -> OrtResult<Self> {
        let fd = syscall::socket(
            syscall::AF_INET,
            syscall::SOCK_STREAM | syscall::SOCK_CLOEXEC,
            0,
        );
        if fd == -1 {
            return Err(ort_error(ErrorKind::SocketCreateFailed, ""));
        }
        set_tcp_fastopen(fd);
        Ok(TcpSocket { fd })
    }

    pub fn connect(&self, addr: &SocketAddrV4) -> OrtResult<()> {
        let c_addr = socket_addr_v4_to_c(addr);
        let len = size_of::<syscall::sockaddr_in>() as syscall::socklen_t;
        let res = syscall::connect(
            self.fd,
            &c_addr as *const _ as *const syscall::sockaddr,
            len,
        );
        if res == -1 {
            return Err(ort_error(ErrorKind::SocketConnectFailed, ""));
        }
        Ok(())
    }
}

impl super::AsFd for TcpSocket {
    fn as_fd(&self) -> i32 {
        self.fd
    }
}

impl Read for TcpSocket {
    fn read(&mut self, buf: &mut [u8]) -> OrtResult<usize> {
        let bytes_read = syscall::read(self.fd, buf.as_mut_ptr() as *mut c_void, buf.len());
        if bytes_read < 0 {
            if bytes_read == syscall::EAGAIN {
                return Err(ort_error(ErrorKind::WouldBlock, ""));
            }
            // see /usr/include/asm-generic/errno.h to translate the codes
            let err_code = utils::num_to_string(-bytes_read);
            utils::print_string(c"socket read err: ", &err_code);
            Err(ort_error(ErrorKind::SocketReadFailed, "syscall read error"))
        } else {
            Ok(bytes_read as usize)
        }
    }
}

impl Write for TcpSocket {
    fn write(&mut self, buf: &[u8]) -> OrtResult<usize> {
        let bytes_written = syscall::write(self.fd, buf.as_ptr() as *const c_void, buf.len());
        if bytes_written < 0 {
            // see /usr/include/asm-generic/errno.h to translate the codes
            let err_code = utils::num_to_string(-bytes_written);
            utils::print_string(c"socket write err: ", &err_code);
            Err(ort_error(
                ErrorKind::SocketWriteFailed,
                "syscall write error",
            ))
        } else {
            Ok(bytes_written as usize)
        }
    }

    fn flush(&mut self) -> OrtResult<()> {
        Ok(())
    }
}

fn set_tcp_fastopen(fd: i32) {
    let optval: c_int = 1; // Enable
    syscall::setsockopt(
        fd,
        syscall::IPPROTO_TCP,
        syscall::TCP_FASTOPEN_CONNECT,
        &optval as *const _ as *const core::ffi::c_void,
        size_of::<i32>() as u32,
    );
}

fn socket_addr_v4_to_c(addr: &SocketAddrV4) -> syscall::sockaddr_in {
    syscall::sockaddr_in {
        sin_family: syscall::AF_INET as syscall::sa_family_t,
        sin_port: addr.port().to_be(),
        sin_addr: ip_v4_addr_to_c(addr.ip()),
        ..unsafe { core::mem::zeroed() }
    }
}
fn ip_v4_addr_to_c(addr: &Ipv4Addr) -> syscall::in_addr {
    // `s_addr` is stored as BE on all machines and the array is in BE order.
    // So the native endian conversion method is used so that it's never swapped.
    syscall::in_addr {
        s_addr: u32::from_ne_bytes(addr.octets()),
    }
}