diff options
Diffstat (limited to 'src/batch.rs')
-rw-r--r-- | src/batch.rs | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/batch.rs b/src/batch.rs index d885813..b5c88b8 100644 --- a/src/batch.rs +++ b/src/batch.rs @@ -2,11 +2,11 @@ use libc; use thiserror::Error; +use crate::error::QueryError; use crate::nlmsg::{NfNetlinkObject, NfNetlinkWriter}; use crate::sys::NFNL_SUBSYS_NFTABLES; use crate::{MsgType, ProtocolFamily}; -use crate::query::Error; use nix::sys::socket::{ self, AddressFamily, MsgFlags, NetlinkAddr, SockAddr, SockFlag, SockProtocol, SockType, }; @@ -88,7 +88,7 @@ impl Batch { *self.buf } - pub fn send(mut self) -> Result<(), Error> { + pub fn send(self) -> Result<(), QueryError> { use crate::query::{recv_and_process, socket_close_wrapper}; let sock = socket::socket( @@ -97,7 +97,7 @@ impl Batch { SockFlag::empty(), SockProtocol::NetlinkNetFilter, ) - .map_err(Error::NetlinkOpenError)?; + .map_err(QueryError::NetlinkOpenError)?; let max_seq = self.seq - 1; @@ -110,7 +110,7 @@ impl Batch { let mut sent = 0; while sent != to_send.len() { sent += socket::send(sock, &to_send[sent..], MsgFlags::empty()) - .map_err(Error::NetlinkSendError)?; + .map_err(QueryError::NetlinkSendError)?; } Ok(socket_close_wrapper(sock, move |sock| { |