1 files changed, 5 insertions, 2 deletions
diff --git a/src/wireguard/router/types.rs b/src/wireguard/router/types.rs
index 9f769fe..194f0d4 100644
--- a/src/wireguard/router/types.rs
+++ b/src/wireguard/router/types.rs
@@ -1,5 +1,8 @@
 use std::error::Error;
 use std::fmt;
+use std::sync::Arc;
+
+use super::KeyPair;
 
 pub trait Opaque: Send + Sync + 'static {}
 
@@ -23,8 +26,8 @@ impl<T, F> KeyCallback<T> for F where F: Fn(&T) -> () + Sync + Send + 'static {}
 
 pub trait Callbacks: Send + Sync + 'static {
     type Opaque: Opaque;
-    fn send(opaque: &Self::Opaque, size: usize, sent: bool);
-    fn recv(opaque: &Self::Opaque, size: usize, sent: bool);
+    fn send(opaque: &Self::Opaque, size: usize, sent: bool, keypair: &Arc<KeyPair>, counter: u64);
+    fn recv(opaque: &Self::Opaque, size: usize, sent: bool, keypair: &Arc<KeyPair>);
     fn need_key(opaque: &Self::Opaque);
     fn key_confirmed(opaque: &Self::Opaque);
 }