@@ -27,18 +27,16 @@ use tokio_io::codec::{Decoder as TokioDecoder, Encoder as TokioEncoder};
27
27
use std:: collections:: HashMap ;
28
28
use std:: str:: FromStr ;
29
29
30
- use aes:: block_cipher:: generic_array:: GenericArray ;
31
- use aes:: Aes128 ;
30
+ use aes:: cipher:: generic_array:: GenericArray ;
31
+ use aes:: cipher:: { NewCipher , StreamCipher } ;
32
+ use aes:: { Aes128 , Aes128Ctr } ;
32
33
use blake2:: { Blake2s , Digest } ;
33
34
use block_modes:: block_padding:: Pkcs7 ;
34
35
use block_modes:: { BlockMode , Ecb } ;
35
36
type Aes128Ecb = Ecb < Aes128 , Pkcs7 > ;
36
37
37
38
use base64:: { decode as b64decode, encode as b64encode} ;
38
39
39
- use aes_ctr:: stream_cipher:: { NewStreamCipher , SyncStreamCipher } ;
40
- use aes_ctr:: Aes128Ctr ;
41
-
42
40
use mles_utils:: * ;
43
41
use std:: time:: Duration ;
44
42
use tokio:: timer:: Interval ;
@@ -457,7 +455,7 @@ fn run_websocket_proxy(
457
455
let duid;
458
456
let dchannel;
459
457
if let Ok ( uid) = b64decode ( uid) {
460
- let cipher = Aes128Ecb :: new_var ( aesecbkey, Default :: default ( ) ) . unwrap ( ) ;
458
+ let cipher = Aes128Ecb :: new_from_slices ( aesecbkey, Default :: default ( ) ) . unwrap ( ) ;
461
459
if let Ok ( uid) = cipher. decrypt_vec ( & uid) {
462
460
duid = uid;
463
461
} else {
@@ -467,7 +465,7 @@ fn run_websocket_proxy(
467
465
return Ok ( ( ) ) ;
468
466
}
469
467
if let Ok ( channel) = b64decode ( channel) {
470
- let cipher = Aes128Ecb :: new_var ( aesecbkey, Default :: default ( ) ) . unwrap ( ) ;
468
+ let cipher = Aes128Ecb :: new_from_slices ( aesecbkey, Default :: default ( ) ) . unwrap ( ) ;
471
469
if let Ok ( channel) = cipher. decrypt_vec ( & channel) {
472
470
dchannel = channel;
473
471
} else {
@@ -552,9 +550,9 @@ fn run_websocket_proxy(
552
550
let aeskey = GenericArray :: from_slice ( & aeschannel_ecb_locked. 0 ) ;
553
551
let aesecbkey = & aeschannel_ecb_locked. 1 ;
554
552
555
- let cipher = Aes128Ecb :: new_var ( & aesecbkey, Default :: default ( ) ) . unwrap ( ) ;
553
+ let cipher = Aes128Ecb :: new_from_slices ( & aesecbkey, Default :: default ( ) ) . unwrap ( ) ;
556
554
let cuid = cipher. encrypt_vec ( & uid. into_bytes ( ) ) ;
557
- let cipher = Aes128Ecb :: new_var ( & aesecbkey, Default :: default ( ) ) . unwrap ( ) ;
555
+ let cipher = Aes128Ecb :: new_from_slices ( & aesecbkey, Default :: default ( ) ) . unwrap ( ) ;
558
556
let cchannel = cipher. encrypt_vec ( & channel. clone ( ) . into_bytes ( ) ) ;
559
557
560
558
decoded_message = decoded_message. set_uid ( b64encode ( & cuid) ) ;
@@ -646,9 +644,9 @@ fn run_websocket_proxy(
646
644
aeschannel_ecb_locked. 1 = hasher_ecb_final. finalize ( ) . as_slice ( ) . to_vec ( ) ;
647
645
aeschannel_ecb_locked. 1 . truncate ( AES_NONCELEN ) ;
648
646
649
- let cipher = Aes128Ecb :: new_var ( & aeschannel_ecb_locked. 1 , Default :: default ( ) ) . unwrap ( ) ;
647
+ let cipher = Aes128Ecb :: new_from_slices ( & aeschannel_ecb_locked. 1 , Default :: default ( ) ) . unwrap ( ) ;
650
648
let cuid = cipher. encrypt_vec ( & uid_clone. into_bytes ( ) ) ;
651
- let cipher = Aes128Ecb :: new_var ( & aeschannel_ecb_locked. 1 , Default :: default ( ) ) . unwrap ( ) ;
649
+ let cipher = Aes128Ecb :: new_from_slices ( & aeschannel_ecb_locked. 1 , Default :: default ( ) ) . unwrap ( ) ;
652
650
let cchannel = cipher. encrypt_vec ( & channel_clone. into_bytes ( ) ) ;
653
651
654
652
//create hash for verification
@@ -667,9 +665,9 @@ fn run_websocket_proxy(
667
665
let aeskey = GenericArray :: from_slice ( & aeschannel_ecb_locked. 0 ) ;
668
666
let aesecbkey = & aeschannel_ecb_locked. 1 ;
669
667
670
- let cipher = Aes128Ecb :: new_var ( & aesecbkey, Default :: default ( ) ) . unwrap ( ) ;
668
+ let cipher = Aes128Ecb :: new_from_slices ( & aesecbkey, Default :: default ( ) ) . unwrap ( ) ;
671
669
let cuid = cipher. encrypt_vec ( & uid. clone ( ) . into_bytes ( ) ) ;
672
- let cipher = Aes128Ecb :: new_var ( & aesecbkey, Default :: default ( ) ) . unwrap ( ) ;
670
+ let cipher = Aes128Ecb :: new_from_slices ( & aesecbkey, Default :: default ( ) ) . unwrap ( ) ;
673
671
let cchannel = cipher. encrypt_vec ( & channel. clone ( ) . into_bytes ( ) ) ;
674
672
675
673
decoded_message = decoded_message. set_uid ( b64encode ( & cuid) ) ;
0 commit comments