use alloc::{boxed::Box, str::SplitWhitespace, vec::Vec}; use core::{mem, ptr}; use crate::{ c_str::CString, error::ResultExt, header::{ arpa_inet::inet_aton, fcntl::O_RDONLY, netinet_in::in_addr, sys_socket::constants::AF_INET, unistd::SEEK_SET, }, platform::{ rlb::{Line, RawLineBuffer}, types::*, Pal, Sys, }, }; use super::{bytes_to_box_str, hostent}; static mut HOSTDB: c_int = -1; pub static mut HOST_ENTRY: hostent = hostent { h_name: ptr::null_mut(), h_aliases: ptr::null_mut(), h_addrtype: 0, h_length: 0, h_addr_list: ptr::null_mut(), }; pub static mut HOST_NAME: Option> = None; pub static mut HOST_ALIASES: Option>> = None; static mut _HOST_ALIASES: Option> = None; pub static mut HOST_ADDR: Option = None; pub static mut HOST_ADDR_LIST: [*mut c_char; 2] = [ptr::null_mut(); 2]; pub static mut _HOST_ADDR_LIST: [u8; 4] = [0u8; 4]; static mut H_POS: usize = 0; pub static mut HOST_STAYOPEN: c_int = 0; #[no_mangle] pub unsafe extern "C" fn endhostent() { if HOSTDB >= 0 { Sys::close(HOSTDB); } HOSTDB = -1; } #[no_mangle] pub unsafe extern "C" fn sethostent(stayopen: c_int) { HOST_STAYOPEN = stayopen; if HOSTDB < 0 { HOSTDB = Sys::open(c"/etc/hosts".into(), O_RDONLY, 0).or_minus_one_errno() } else { Sys::lseek(HOSTDB, 0, SEEK_SET); } H_POS = 0; } #[no_mangle] pub unsafe extern "C" fn gethostent() -> *mut hostent { if HOSTDB < 0 { HOSTDB = Sys::open(c"/etc/hosts".into(), O_RDONLY, 0).or_minus_one_errno(); } let mut rlb = RawLineBuffer::new(HOSTDB); rlb.seek(H_POS); let mut r: Box = Box::default(); while r.is_empty() || r.split_whitespace().next() == None || r.starts_with('#') { r = match rlb.next() { Line::Some(s) => bytes_to_box_str(s), _ => { if HOST_STAYOPEN == 0 { endhostent(); } return ptr::null_mut(); } }; } rlb.next(); H_POS = rlb.line_pos(); let mut iter: SplitWhitespace = r.split_whitespace(); let addr_vec: Vec = iter.next().unwrap().bytes().chain(Some(b'\0')).collect(); let addr_cstr = addr_vec.as_slice().as_ptr() as *const c_char; let mut addr = mem::MaybeUninit::uninit(); inet_aton(addr_cstr, addr.as_mut_ptr()); let addr = addr.assume_init(); _HOST_ADDR_LIST = addr.s_addr.to_ne_bytes(); HOST_ADDR_LIST = [_HOST_ADDR_LIST.as_mut_ptr() as *mut c_char, ptr::null_mut()]; HOST_ADDR = Some(addr); let host_name = iter.next().unwrap().bytes().chain(Some(b'\0')).collect(); let mut _host_aliases: Vec> = iter .map(|alias| alias.bytes().chain(Some(b'\0')).collect()) .collect(); HOST_ALIASES = Some(_host_aliases); let mut host_aliases: Vec<*mut c_char> = HOST_ALIASES .as_mut() .unwrap() .iter_mut() .map(|x| x.as_mut_ptr() as *mut c_char) .chain([ptr::null_mut(), ptr::null_mut()]) .collect(); HOST_NAME = Some(host_name); HOST_ENTRY = hostent { h_name: HOST_NAME.as_mut().unwrap().as_mut_ptr() as *mut c_char, h_aliases: host_aliases.as_mut_slice().as_mut_ptr(), h_addrtype: AF_INET, h_length: 4, h_addr_list: HOST_ADDR_LIST.as_mut_ptr(), }; _HOST_ALIASES = Some(host_aliases); if HOST_STAYOPEN == 0 { endhostent(); } &mut HOST_ENTRY as *mut hostent }