114 lines
2.7 KiB
Rust
114 lines
2.7 KiB
Rust
// This is an attempt to port the iwgetid and iwlib code
|
|
|
|
mod wireless;
|
|
use std::{
|
|
ffi::{
|
|
c_char,
|
|
c_int,
|
|
c_uchar,
|
|
c_void,
|
|
},
|
|
mem::size_of,
|
|
os::fd::RawFd,
|
|
str,
|
|
};
|
|
|
|
use mlua::prelude::*;
|
|
use nix::{
|
|
ioctl_read_bad,
|
|
libc::IF_NAMESIZE,
|
|
sys::socket::{
|
|
socket as open_socket,
|
|
AddressFamily,
|
|
SockFlag,
|
|
SockType,
|
|
},
|
|
unistd::close
|
|
};
|
|
use wireless::{
|
|
IfConf,
|
|
IfConfData,
|
|
IwPoint,
|
|
IwReq,
|
|
IwReqData,
|
|
IwReqName,
|
|
IW_ESSID_MAX_SIZE,
|
|
SIOCGIFCONF,
|
|
SIOCGIWESSID,
|
|
};
|
|
|
|
ioctl_read_bad!(ioctl_get_interfaces, SIOCGIFCONF, IfConf);
|
|
ioctl_read_bad!(ioctl_get_essid, SIOCGIWESSID, IwReq);
|
|
|
|
pub fn get_first_essid(_: &Lua, _: ()) -> LuaResult<String> {
|
|
type Buffer = [c_char; 1024];
|
|
let mut buffer: Buffer = [0; 1024];
|
|
|
|
let mut if_conf = IfConf {
|
|
ifc_len: size_of::<Buffer>() as c_int,
|
|
data: IfConfData {
|
|
ifc_buf: buffer.as_mut_ptr(),
|
|
},
|
|
};
|
|
|
|
let mut if_req;
|
|
let socket = get_first_socket().map_err(LuaError::external)?;
|
|
|
|
unsafe {
|
|
ioctl_get_interfaces(socket, &mut if_conf).map_err(LuaError::external)?;
|
|
if_req = if_conf.data.ifc_req;
|
|
}
|
|
|
|
for _ in 0..if_conf.ifc_len / size_of::<IfConf>() as c_int {
|
|
if let Ok(essid) = get_essid(socket, unsafe { *if_req }.ifr_name) {
|
|
close(socket).map_err(LuaError::external)?;
|
|
return Ok(essid);
|
|
};
|
|
|
|
unsafe { if_req = if_req.add(1) };
|
|
}
|
|
|
|
close(socket).map_err(LuaError::external)?;
|
|
Err(LuaError::external("couldn't find any valid interfaces"))
|
|
}
|
|
|
|
fn get_essid(socket: RawFd, if_name: [c_char; IF_NAMESIZE]) -> LuaResult<String> {
|
|
let mut essid: [c_uchar; (IW_ESSID_MAX_SIZE + 1) as usize] =
|
|
[0; (IW_ESSID_MAX_SIZE + 1) as usize];
|
|
let mut wrq = IwReq {
|
|
ifr_ifrn: IwReqName { ifrn_name: if_name },
|
|
u: IwReqData {
|
|
essid: IwPoint {
|
|
pointer: essid.as_mut_ptr().cast::<c_void>(),
|
|
length: IW_ESSID_MAX_SIZE + 1,
|
|
flags: 0,
|
|
},
|
|
},
|
|
};
|
|
|
|
unsafe {
|
|
ioctl_get_essid(socket, &mut wrq).map_err(LuaError::external)?;
|
|
}
|
|
|
|
Ok(str::from_utf8(essid.as_slice())
|
|
.map_err(LuaError::external)?
|
|
.to_owned())
|
|
}
|
|
|
|
fn get_first_socket() -> LuaResult<RawFd> {
|
|
let families = [
|
|
AddressFamily::Inet,
|
|
AddressFamily::Ipx,
|
|
AddressFamily::Ax25,
|
|
AddressFamily::AppleTalk,
|
|
];
|
|
|
|
for family in families {
|
|
if let Ok(socket) = open_socket(family, SockType::Datagram, SockFlag::empty(), None) {
|
|
return Ok(socket);
|
|
}
|
|
}
|
|
|
|
Err(LuaError::external("couldn't get a valid kernel socket"))
|
|
}
|