use crate::compiler::prelude::*;
use std::net::Ipv4Addr;
fn ip_aton(value: &Value) -> Resolved {
let ip: Ipv4Addr = value
.try_bytes_utf8_lossy()?
.parse()
.map_err(|err| format!("unable to parse IPv4 address: {err}"))?;
Ok(u32::from(ip).into())
}
#[derive(Clone, Copy, Debug)]
pub struct IpAton;
impl Function for IpAton {
fn identifier(&self) -> &'static str {
"ip_aton"
}
fn usage(&self) -> &'static str {
indoc! {"
Converts IPv4 address in numbers-and-dots notation into network-order
bytes represented as an integer.
This behavior mimics [inet_aton](https://linux.die.net/man/3/inet_aton).
"}
}
fn category(&self) -> &'static str {
Category::Ip.as_ref()
}
fn internal_failure_reasons(&self) -> &'static [&'static str] {
&["`value` is not a valid IPv4 address."]
}
fn return_kind(&self) -> u16 {
kind::INTEGER
}
fn parameters(&self) -> &'static [Parameter] {
const PARAMETERS: &[Parameter] = &[Parameter::required(
"value",
kind::BYTES,
"The IP address to convert to binary.",
)];
PARAMETERS
}
fn examples(&self) -> &'static [Example] {
&[example! {
title: "IPv4 to integer",
source: r#"ip_aton!("1.2.3.4")"#,
result: Ok("16909060"),
}]
}
fn compile(
&self,
_state: &state::TypeState,
_ctx: &mut FunctionCompileContext,
arguments: ArgumentList,
) -> Compiled {
let value = arguments.required("value");
Ok(IpAtonFn { value }.as_expr())
}
}
#[derive(Debug, Clone)]
struct IpAtonFn {
value: Box<dyn Expression>,
}
impl FunctionExpression for IpAtonFn {
fn resolve(&self, ctx: &mut Context) -> Resolved {
let value = self.value.resolve(ctx)?;
ip_aton(&value)
}
fn type_def(&self, _: &state::TypeState) -> TypeDef {
TypeDef::integer().fallible()
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::value;
test_function![
ip_aton => IpAton;
invalid {
args: func_args![value: "i am not an ipaddress"],
want: Err("unable to parse IPv4 address: invalid IPv4 address syntax"),
tdef: TypeDef::integer().fallible(),
}
valid {
args: func_args![value: "1.2.3.4"],
want: Ok(value!(16_909_060)),
tdef: TypeDef::integer().fallible(),
}
];
}