Skip to content

Commit 313b1db

Browse files
committed
fix: import ensure
1 parent 9f7cdf1 commit 313b1db

4 files changed

Lines changed: 33 additions & 31 deletions

File tree

netwatch/src/interfaces/bsd.rs

Lines changed: 26 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -272,9 +272,9 @@ impl WireFormat {
272272
target_os = "ios"
273273
))]
274274
MessageType::Route => {
275-
n0_error::ensure!(data.len() >= self.body_off, RouteError::MessageTooShort);
275+
ensure!(data.len() >= self.body_off, RouteError::MessageTooShort);
276276
let l = u16_from_ne_range(data, ..2)?;
277-
n0_error::ensure!(data.len() >= l as usize, RouteError::InvalidMessage);
277+
ensure!(data.len() >= l as usize, RouteError::InvalidMessage);
278278
let attrs: i32 = u32_from_ne_range(data, 12..16)?
279279
.try_into()
280280
.map_err(|_| e!(RouteError::InvalidMessage))?;
@@ -299,11 +299,11 @@ impl WireFormat {
299299
}
300300
#[cfg(target_os = "openbsd")]
301301
MessageType::Route => {
302-
n0_error::ensure!(data.len() >= self.body_off, RouteError::MessageTooShort);
302+
ensure!(data.len() >= self.body_off, RouteError::MessageTooShort);
303303
let l = u16_from_ne_range(data, ..2)?;
304-
n0_error::ensure!(data.len() >= l as usize, RouteError::InvalidMessage);
304+
ensure!(data.len() >= l as usize, RouteError::InvalidMessage);
305305
let ll = u16_from_ne_range(data, 4..6)? as usize;
306-
n0_error::ensure!(data.len() >= ll as usize, RouteError::InvalidMessage);
306+
ensure!(data.len() >= ll as usize, RouteError::InvalidMessage);
307307

308308
let addrs = parse_addrs(
309309
u32_from_ne_range(data, 12..16)? as _,
@@ -330,9 +330,9 @@ impl WireFormat {
330330
Ok(Some(WireMessage::Route(m)))
331331
}
332332
MessageType::Interface => {
333-
n0_error::ensure!(data.len() >= self.body_off, RouteError::MessageTooShort);
333+
ensure!(data.len() >= self.body_off, RouteError::MessageTooShort);
334334
let l = u16_from_ne_range(data, 0..2)?;
335-
n0_error::ensure!(data.len() >= l as usize, RouteError::InvalidMessage);
335+
ensure!(data.len() >= l as usize, RouteError::InvalidMessage);
336336

337337
let attrs = u32_from_ne_range(data, 4..8)?;
338338
if attrs as c_int & RTA_IFP == 0 {
@@ -353,9 +353,9 @@ impl WireFormat {
353353
Ok(Some(WireMessage::Interface(m)))
354354
}
355355
MessageType::InterfaceAddr => {
356-
n0_error::ensure!(data.len() >= self.body_off, RouteError::MessageTooShort);
356+
ensure!(data.len() >= self.body_off, RouteError::MessageTooShort);
357357
let l = u16_from_ne_range(data, ..2)?;
358-
n0_error::ensure!(data.len() >= l as usize, RouteError::InvalidMessage);
358+
ensure!(data.len() >= l as usize, RouteError::InvalidMessage);
359359

360360
#[cfg(target_os = "netbsd")]
361361
let index = u16_from_ne_range(data, 16..18)?;
@@ -378,9 +378,9 @@ impl WireFormat {
378378
Ok(Some(WireMessage::InterfaceAddr(m)))
379379
}
380380
MessageType::InterfaceMulticastAddr => {
381-
n0_error::ensure!(data.len() >= self.body_off, RouteError::MessageTooShort);
381+
ensure!(data.len() >= self.body_off, RouteError::MessageTooShort);
382382
let l = u16_from_ne_range(data, ..2)?;
383-
n0_error::ensure!(data.len() >= l as usize, RouteError::InvalidMessage);
383+
ensure!(data.len() >= l as usize, RouteError::InvalidMessage);
384384

385385
let addrs = parse_addrs(
386386
u32_from_ne_range(data, 4..8)? as _,
@@ -397,9 +397,9 @@ impl WireFormat {
397397
Ok(Some(WireMessage::InterfaceMulticastAddr(m)))
398398
}
399399
MessageType::InterfaceAnnounce => {
400-
n0_error::ensure!(data.len() >= self.body_off, RouteError::MessageTooShort);
400+
ensure!(data.len() >= self.body_off, RouteError::MessageTooShort);
401401
let l = u16_from_ne_range(data, ..2)?;
402-
n0_error::ensure!(data.len() >= l as usize, RouteError::InvalidMessage);
402+
ensure!(data.len() >= l as usize, RouteError::InvalidMessage);
403403

404404
let mut name = String::new();
405405
for i in 0..16 {
@@ -445,7 +445,7 @@ struct RoutingStack {
445445

446446
/// Parses b as a routing information base and returns a list of routing messages.
447447
pub fn parse_rib(typ: RIBType, data: &[u8]) -> Result<Vec<WireMessage>, RouteError> {
448-
n0_error::ensure!(
448+
ensure!(
449449
is_valid_rib_type(typ),
450450
RouteError::InvalidRibType { rib_type: typ }
451451
);
@@ -458,8 +458,8 @@ pub fn parse_rib(typ: RIBType, data: &[u8]) -> Result<Vec<WireMessage>, RouteErr
458458
while b.len() > 4 {
459459
nmsgs += 1;
460460
let l = u16_from_ne_range(b, ..2)?;
461-
n0_error::ensure!(l != 0, RouteError::InvalidMessage);
462-
n0_error::ensure!(b.len() >= l as usize, RouteError::MessageTooShort);
461+
ensure!(l != 0, RouteError::InvalidMessage);
462+
ensure!(b.len() >= l as usize, RouteError::MessageTooShort);
463463
if b[2] as i32 != ROUTING_STACK.rtm_version {
464464
// b = b[l:];
465465
continue;
@@ -484,7 +484,7 @@ pub fn parse_rib(typ: RIBType, data: &[u8]) -> Result<Vec<WireMessage>, RouteErr
484484
}
485485

486486
// We failed to parse any of the messages - version mismatch?
487-
n0_error::ensure!(nmsgs == msgs.len() + nskips, RouteError::MessageMismatch);
487+
ensure!(nmsgs == msgs.len() + nskips, RouteError::MessageMismatch);
488488

489489
Ok(msgs)
490490
}
@@ -768,15 +768,15 @@ where
768768
let a = parse_link_addr(b)?;
769769
addrs.push(a);
770770
let l = roundup(b[0] as usize);
771-
n0_error::ensure!(b.len() >= l, RouteError::MessageTooShort);
771+
ensure!(b.len() >= l, RouteError::MessageTooShort);
772772
b = &b[l..];
773773
}
774774
AF_INET | AF_INET6 => {
775775
let af = b[1] as i32;
776776
let a = parse_inet_addr(af, b)?;
777777
addrs.push(a);
778778
let l = roundup(b[0] as usize);
779-
n0_error::ensure!(b.len() >= l, RouteError::MessageTooShort);
779+
ensure!(b.len() >= l, RouteError::MessageTooShort);
780780
b = &b[l..];
781781
}
782782
_ => {
@@ -794,7 +794,7 @@ where
794794
let a = parse_default_addr(b)?;
795795
addrs.push(a);
796796
let l = roundup(b[0] as usize);
797-
n0_error::ensure!(b.len() >= l, RouteError::MessageTooShort);
797+
ensure!(b.len() >= l, RouteError::MessageTooShort);
798798
b = &b[l..];
799799
}
800800
}
@@ -809,13 +809,13 @@ where
809809
fn parse_inet_addr(af: i32, b: &[u8]) -> Result<Addr, RouteError> {
810810
match af {
811811
AF_INET => {
812-
n0_error::ensure!(b.len() >= SIZEOF_SOCKADDR_INET, RouteError::InvalidAddress);
812+
ensure!(b.len() >= SIZEOF_SOCKADDR_INET, RouteError::InvalidAddress);
813813

814814
let ip = Ipv4Addr::new(b[4], b[5], b[6], b[7]);
815815
Ok(Addr::Inet4 { ip })
816816
}
817817
AF_INET6 => {
818-
n0_error::ensure!(b.len() >= SIZEOF_SOCKADDR_INET6, RouteError::InvalidAddress);
818+
ensure!(b.len() >= SIZEOF_SOCKADDR_INET6, RouteError::InvalidAddress);
819819

820820
let mut zone = u32_from_ne_range(b, 24..28)?;
821821
let mut oc: [u8; 16] = b
@@ -885,7 +885,7 @@ fn parse_kernel_inet_addr(af: i32, b: &[u8]) -> Result<(i32, Addr), RouteError>
885885
l = roundup(l);
886886
}
887887

888-
n0_error::ensure!(b.len() >= l, RouteError::InvalidAddress);
888+
ensure!(b.len() >= l, RouteError::InvalidAddress);
889889
// Don't reorder case expressions.
890890
// The case expressions for IPv6 must come first.
891891
const OFF4: usize = 4; // offset of in_addr
@@ -930,7 +930,7 @@ fn parse_kernel_inet_addr(af: i32, b: &[u8]) -> Result<(i32, Addr), RouteError>
930930
}
931931

932932
fn parse_link_addr(b: &[u8]) -> Result<Addr, RouteError> {
933-
n0_error::ensure!(b.len() >= 8, RouteError::InvalidAddress);
933+
ensure!(b.len() >= 8, RouteError::InvalidAddress);
934934
let (_, mut a) = parse_kernel_link_addr(AF_LINK, &b[4..])?;
935935

936936
if let Addr::Link { index, .. } = &mut a {
@@ -972,7 +972,7 @@ fn parse_kernel_link_addr(_: i32, b: &[u8]) -> Result<(usize, Addr), RouteError>
972972
}
973973

974974
let l = 4 + nlen + alen + slen;
975-
n0_error::ensure!(b.len() >= l, RouteError::InvalidAddress);
975+
ensure!(b.len() >= l, RouteError::InvalidAddress);
976976
let mut data = &b[4..];
977977

978978
let name = if nlen > 0 {
@@ -1001,7 +1001,7 @@ fn parse_kernel_link_addr(_: i32, b: &[u8]) -> Result<(usize, Addr), RouteError>
10011001
}
10021002

10031003
fn parse_default_addr(b: &[u8]) -> Result<Addr, RouteError> {
1004-
n0_error::ensure!(
1004+
ensure!(
10051005
b.len() >= 2 && b.len() >= b[0] as usize,
10061006
RouteError::InvalidAddress
10071007
);

netwatch/src/netmon/android.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
1+
use n0_error::stack_error;
12
use tokio::sync::mpsc;
23

34
use super::actor::NetworkMessage;
45

5-
#[n0_error::stack_error(derive, add_meta)]
6+
#[stack_error(derive, add_meta)]
67
pub struct Error;
78

89
#[derive(Debug)]

netwatch/src/netmon/wasm_browser.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,13 +2,14 @@ use js_sys::{
22
Function,
33
wasm_bindgen::{JsCast, prelude::Closure},
44
};
5+
use n0_error::stack_error;
56
use n0_future::task;
67
use tokio::sync::mpsc;
78
use web_sys::{EventListener, EventTarget};
89

910
use super::actor::NetworkMessage;
1011

11-
#[n0_error::stack_error(derive, add_meta)]
12+
#[stack_error(derive, add_meta)]
1213
pub struct Error;
1314

1415
#[derive(Debug)]

portmapper/src/pcp/protocol/response.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
//! A PCP response encoding and decoding.
22
33
use derive_more::Display;
4-
use n0_error::{e, stack_error};
4+
use n0_error::{e, ensure, stack_error};
55
use num_enum::{IntoPrimitive, TryFromPrimitive, TryFromPrimitiveError};
66

77
use super::{Opcode, Version, opcode_data::OpcodeData};
@@ -176,7 +176,7 @@ impl Response {
176176

177177
/// Decode a response.
178178
pub fn decode(buf: &[u8]) -> Result<Self, Error> {
179-
n0_error::ensure!(
179+
ensure!(
180180
Self::MIN_SIZE <= buf.len() && buf.len() <= Self::MAX_SIZE,
181181
DecodeError::Malformed
182182
);
@@ -186,7 +186,7 @@ impl Response {
186186
.map_err(|_| e!(DecodeError::InvalidVersion))?;
187187

188188
let opcode = buf[1];
189-
n0_error::ensure!(
189+
ensure!(
190190
opcode & Self::RESPONSE_INDICATOR == Self::RESPONSE_INDICATOR,
191191
DecodeError::NotAResponse
192192
);

0 commit comments

Comments
 (0)