Files
@ 7175e8b0e1e8
Branch filter:
Location: CSY/reowolf/src/ffi/pseudo_socket_api.rs - annotation
7175e8b0e1e8
8.0 KiB
application/rls-services+xml
restricted pseudo-socket API to connection-oriented API
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 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 | 4db4ea6f0eb3 6c04a99de862 3801521d486d 700221108e9f 700221108e9f 700221108e9f 700221108e9f 700221108e9f 700221108e9f 700221108e9f bdcf814bde8c bdcf814bde8c bdcf814bde8c 6c04a99de862 6c04a99de862 bdcf814bde8c 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e bdcf814bde8c bdcf814bde8c 14c1a07d148a 14c1a07d148a bdcf814bde8c bdcf814bde8c fdc328afed95 3028e66928d3 3028e66928d3 3028e66928d3 3028e66928d3 3028e66928d3 c8994576abce c8994576abce fdc328afed95 3801521d486d 700221108e9f 3801521d486d 700221108e9f bdcf814bde8c bdcf814bde8c bdcf814bde8c bdcf814bde8c bdcf814bde8c bdcf814bde8c bdcf814bde8c bdcf814bde8c bdcf814bde8c 6c04a99de862 bdcf814bde8c bdcf814bde8c bdcf814bde8c bdcf814bde8c bdcf814bde8c bdcf814bde8c bdcf814bde8c bdcf814bde8c bdcf814bde8c 6c04a99de862 bdcf814bde8c bdcf814bde8c bdcf814bde8c bdcf814bde8c 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e bdcf814bde8c 1f3007fd929e bdcf814bde8c 7b9df91324ad 700221108e9f 3801521d486d 1f3007fd929e bdcf814bde8c 1f3007fd929e 14c1a07d148a bdcf814bde8c bdcf814bde8c bdcf814bde8c 6c04a99de862 bdcf814bde8c 3801521d486d 1f3007fd929e 14c1a07d148a 1db6ec21a437 4db4ea6f0eb3 bdcf814bde8c 4db4ea6f0eb3 bdcf814bde8c bdcf814bde8c bdcf814bde8c fdc328afed95 3801521d486d fdc328afed95 fdc328afed95 fdc328afed95 fdc328afed95 3801521d486d 1f3007fd929e 14c1a07d148a 1f3007fd929e 14c1a07d148a 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 700221108e9f bdcf814bde8c bdcf814bde8c 6c04a99de862 6c04a99de862 fdc328afed95 fdc328afed95 bdcf814bde8c fdc328afed95 fdc328afed95 fdc328afed95 fdc328afed95 bdcf814bde8c 3801521d486d 1f3007fd929e 14c1a07d148a 1f3007fd929e 14c1a07d148a 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 3801521d486d 3801521d486d bdcf814bde8c bdcf814bde8c 6c04a99de862 6c04a99de862 6c04a99de862 6c04a99de862 7b9df91324ad 7b9df91324ad 6c04a99de862 3801521d486d 1f3007fd929e 14c1a07d148a 1f3007fd929e 14c1a07d148a 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 3801521d486d 3801521d486d b747c570a11c b747c570a11c 3801521d486d b747c570a11c b747c570a11c b747c570a11c b747c570a11c b747c570a11c 3801521d486d 3801521d486d 7175e8b0e1e8 14c1a07d148a 7175e8b0e1e8 14c1a07d148a 3028e66928d3 3028e66928d3 3801521d486d 3028e66928d3 3801521d486d 3801521d486d 48cedf26fcaa 48cedf26fcaa 48cedf26fcaa 48cedf26fcaa 48cedf26fcaa 48cedf26fcaa 3801521d486d 48cedf26fcaa 48cedf26fcaa b747c570a11c b747c570a11c b747c570a11c b747c570a11c b747c570a11c b747c570a11c b747c570a11c b747c570a11c b747c570a11c 3801521d486d 3801521d486d b747c570a11c 3801521d486d 3801521d486d 7175e8b0e1e8 14c1a07d148a 1f3007fd929e 14c1a07d148a 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 1f3007fd929e 6c04a99de862 bdcf814bde8c | use super::*;
use libc::{sockaddr, socklen_t};
use std::{
collections::HashMap,
ffi::c_void,
net::{Ipv4Addr, SocketAddr, SocketAddrV4},
os::raw::c_int,
sync::RwLock,
};
///////////////////////////////////////////////////////////////////
struct FdAllocator {
next: Option<c_int>,
freed: Vec<c_int>,
}
enum ConnectorComplex {
Setup {
local: Option<SocketAddr>,
peer: Option<SocketAddr>,
},
Communication {
connector: Connector,
putter: PortId,
getter: PortId,
},
}
#[derive(Default)]
struct CcMap {
fd_to_cc: HashMap<c_int, RwLock<ConnectorComplex>>,
fd_allocator: FdAllocator,
}
///////////////////////////////////////////////////////////////////
unsafe fn payload_from_raw(bytes_ptr: *const c_void, bytes_len: usize) -> Payload {
let bytes_ptr = std::mem::transmute(bytes_ptr);
let bytes = &*slice_from_raw_parts(bytes_ptr, bytes_len);
Payload::from(bytes)
}
unsafe fn addr_from_raw(addr: *const sockaddr, addr_len: socklen_t) -> Option<SocketAddr> {
os_socketaddr::OsSocketAddr::from_raw_parts(addr as _, addr_len as usize).into_addr()
}
fn dummy_peer_addr() -> SocketAddr {
// SocketAddrV4::new isn't a constant-time func
SocketAddr::V4(SocketAddrV4::new(Ipv4Addr::new(127, 0, 0, 0), 8000))
}
impl Default for FdAllocator {
fn default() -> Self {
Self {
next: Some(0), // positive values used only
freed: vec![],
}
}
}
impl FdAllocator {
fn alloc(&mut self) -> c_int {
if let Some(fd) = self.freed.pop() {
return fd;
}
if let Some(fd) = self.next {
self.next = fd.checked_add(1);
return fd;
}
panic!("No more Connector FDs to allocate!")
}
fn free(&mut self, fd: c_int) {
self.freed.push(fd);
}
}
lazy_static::lazy_static! {
static ref LOCK_POISONED: RwLock<CcMap> = Default::default();
}
impl ConnectorComplex {
fn try_become_connected(&mut self) {
match self {
ConnectorComplex::Setup { Some(local), Some(peer) } => {
// setup complete
let connector = Connector::new(crate::DummyLogger, TRIVIAL_PD.clone(), Connector::random_id());
let [putter, getter] = connector.new_udp_mediator_component(local, peer).unwrap();
*self = ConnectorComplex::Communication { connector, putter, getter }
}
_ => {} // setup incomplete
}
}
}
/////////////////////////////////
#[no_mangle]
pub extern "C" fn rw_socket(_domain: c_int, _type: c_int) -> c_int {
// ignoring domain and type
// get writer lock
let mut w = if let Ok(w) = LOCK_POISONED.write() { w } else { return CCMLP };
let fd = w.fd_allocator.alloc();
let cc = ConnectorComplex::Setup { local: None, peer: None };
w.fd_to_cc.insert(fd, RwLock::new(cc));
fd
}
#[no_mangle]
pub extern "C" fn rw_close(fd: c_int, _how: c_int) -> c_int {
// ignoring HOW
// get writer lock
let mut w = if let Ok(w) = LOCK_POISONED.write() { w } else { return CCMLP };
if w.fd_to_cc.remove(&fd).is_some() {
w.fd_allocator.free(fd);
ERR_OK
} else {
CLOSE_FAIL
}
}
#[no_mangle]
pub unsafe extern "C" fn rw_bind(fd: c_int, addr: *const sockaddr, addr_len: socklen_t) -> c_int {
// assuming _domain is AF_INET and _type is SOCK_DGRAM
let addr = match addr_from_raw(addr, addr_len) {
Some(addr) => addr,
_ => return BAD_SOCKADDR,
};
// get outer reader, inner writer locks
let r = if let Ok(r) = LOCK_POISONED.read() { r } else { return CCMLP };
let cc = if let Some(cc) = r.fd_to_cc.get(&fd) { cc } else { return BAD_FD };
let mut cc = if let Ok(cc) = cc.write() { cc } else { return CCMLP };
let cc: &mut ConnectorComplex = &mut cc;
match cc {
ConnectorComplex::Communication { ..} => WRONG_STATE,
ConnectorComplex::Setup { local, peer } => {
ConnectorComplex::Setup { local, .. } => {
*local = Some(addr);
cc.try_become_connected();
ERR_OK
}
}
}
#[no_mangle]
pub unsafe extern "C" fn rw_connect(
fd: c_int,
addr: *const sockaddr,
addr_len: socklen_t,
) -> c_int {
let addr = match addr_from_raw(addr, addr_len) {
Some(addr) => addr,
_ => return BAD_SOCKADDR,
};
// assuming _domain is AF_INET and _type is SOCK_DGRAM
// get outer reader, inner writer locks
let r = if let Ok(r) = LOCK_POISONED.read() { r } else { return CCMLP };
let cc = if let Some(cc) = r.fd_to_cc.get(&fd) { cc } else { return BAD_FD };
let mut cc = if let Ok(cc) = cc.write() { cc } else { return CCMLP };
let cc: &mut ConnectorComplex = &mut cc;
match cc {
ConnectorComplex::Communication { .. } => WRONG_STATE,
ConnectorComplex::Setup { peer, .. } => {
*peer = Some(addr);
cc.try_become_connected();
ERR_OK
}
}
}
#[no_mangle]
pub unsafe extern "C" fn rw_send(
fd: c_int,
bytes_ptr: *const c_void,
bytes_len: usize,
_flags: c_int,
) -> isize {
// ignoring flags
// get outer reader, inner writer locks
let r = if let Ok(r) = LOCK_POISONED.read() { r } else { return CCMLP as isize };
let cc = if let Some(cc) = r.fd_to_cc.get(&fd) { cc } else { return BAD_FD as isize };
let mut cc = if let Ok(cc) = cc.write() { cc } else { return CCMLP as isize };
let cc: &mut ConnectorComplex = &mut cc;
match cc {
ConnectorComplex::Setup { .. } => WRONG_STATE as isize,
ConnectorComplex::Communication { connector, putter, .. } => {
let payload = payload_from_raw(bytes_ptr, bytes_len);
connector.put(*putter, payload).unwrap();
connector.sync(None).unwrap();
bytes_len
}
}
}
#[no_mangle]
pub unsafe extern "C" fn rw_recv(
fd: c_int,
bytes_ptr: *mut c_void,
bytes_len: usize,
_flags: c_int,
) -> isize {
// ignoring flags
// get outer reader, inner writer locks
let r = if let Ok(r) = CC_MAP.read() { r } else { return LOCK_POISONED as isize };
let cc = if let Some(cc) = r.fd_to_cc.get(&fd) { cc } else { return BAD_FD as isize };
let mut cc = if let Ok(cc) = cc.write() { cc } else { return LOCK_POISONED as isize };
let cc: &mut ConnectorComplex = &mut cc;
if let Some(ConnectorBound { connector, getter, .. }) = &mut cc.connector_bound {
connector.get(*getter).unwrap();
// this call BLOCKS until it succeeds, and its got no reason to fail
connector.sync(None).unwrap();
// copy from gotten to caller's buffer (truncating if necessary)
let slice = connector.gotten(*getter).unwrap().as_slice();
if !bytes_ptr.is_null() {
let cpy_msg_bytes = slice.len().min(bytes_len);
std::ptr::copy_nonoverlapping(slice.as_ptr(), bytes_ptr as *mut u8, cpy_msg_bytes);
}
// return number of bytes sent
slice.len() as isize
} else {
// not bound!
WRONG_STATE as isize
}
}
#[no_mangle]
pub unsafe extern "C" fn rw_recvfrom(
fd: c_int,
bytes_ptr: *mut c_void,
bytes_len: usize,
_flags: c_int,
addr: *mut sockaddr,
addr_len: *mut socklen_t,
) -> isize {
// ignoring flags
// get outer reader, inner writer locks
let r = if let Ok(r) = CC_MAP.read() { r } else { return LOCK_POISONED as isize };
let cc = if let Some(cc) = r.fd_to_cc.get(&fd) { cc } else { return BAD_FD as isize };
let mut cc = if let Ok(cc) = cc.write() { cc } else { return CCMLP as isize };
let cc: &mut ConnectorComplex = &mut cc;
match cc {
ConnectorComplex::Setup { .. } => WRONG_STATE as isize,
ConnectorComplex::Communication { connector, getter, .. } => {
connector.get(*getter).unwrap();
connector.sync(None).unwrap();
let slice = connector.gotten(*getter).unwrap().as_slice();
let cpy_msg_bytes = slice.len().min(bytes_len);
std::ptr::copy_nonoverlapping(slice.as_ptr(), bytes_ptr as *mut u8, cpy_msg_bytes);
cpy_msg_bytes
}
}
}
|