refactor EncryptedStream::reset

src/lib.rs

@@ -65,7 +65,10 @@ impl EncryptedStream {
         self.socket.shutdown(how)
     }
 
-    pub fn reset(&mut self, next_nonce: [u32; 3]) {
+    pub fn reset(&mut self, next_nonce_bytes: [u8; 12]) {
+        let mut next_nonce = [0u32; 3];
+        u8_to_u32(&mut next_nonce, &next_nonce_bytes);
+
         self.block.reset(next_nonce)
     }
 
@@ -76,10 +79,7 @@ impl EncryptedStream {
             return Err(IO(err));
         }
 
-        let mut nonce = [0u32; 3];
+        self.reset(nonce_bytes);
-        u8_to_u32(&mut nonce, &nonce_bytes);
-
-        self.block.reset(nonce);
 
         let poly_key = self.block.get_bytes();
         self.block.advance();