diff options
| -rwxr-xr-x | Cargo.lock | 1 | ||||
| -rw-r--r-- | azalea-crypto/Cargo.toml | 7 | ||||
| -rw-r--r-- | azalea-crypto/benches/my_benchmark.rs | 18 | ||||
| -rw-r--r-- | azalea-crypto/src/lib.rs | 52 |
4 files changed, 78 insertions, 0 deletions
@@ -163,6 +163,7 @@ dependencies = [ "aes", "azalea-buf", "cfb8", + "criterion", "num-bigint", "rand", "rsa_public_encrypt_pkcs1", diff --git a/azalea-crypto/Cargo.toml b/azalea-crypto/Cargo.toml index 7b77eb3b..3947c26f 100644 --- a/azalea-crypto/Cargo.toml +++ b/azalea-crypto/Cargo.toml @@ -14,3 +14,10 @@ rand = {version = "^0.8.4", features = ["getrandom"]} rsa_public_encrypt_pkcs1 = "0.4.0" sha-1 = "^0.10.0" uuid = "^1.1.2" + +[dev-dependencies] +criterion = {version = "^0.3.5", features = ["html_reports"]} + +[[bench]] +harness = false +name = "my_benchmark" diff --git a/azalea-crypto/benches/my_benchmark.rs b/azalea-crypto/benches/my_benchmark.rs new file mode 100644 index 00000000..2d255d9c --- /dev/null +++ b/azalea-crypto/benches/my_benchmark.rs @@ -0,0 +1,18 @@ +use azalea_crypto::{create_cipher, encrypt_packet}; +use criterion::{criterion_group, criterion_main, Criterion, Throughput}; + +fn bench(c: &mut Criterion) { + let (enc, dec) = create_cipher(b"0123456789abcdef"); + + let mut packet = [0u8; 65536]; + for i in 0..packet.len() { + packet[i] = i as u8; + } + + c.bench_function("Encrypt", |b| { + b.iter(|| encrypt_packet(&mut enc.clone(), &mut packet.clone())) + }); +} + +criterion_group!(benches, bench); +criterion_main!(benches); diff --git a/azalea-crypto/src/lib.rs b/azalea-crypto/src/lib.rs index 9a17a472..d1a86ec7 100644 --- a/azalea-crypto/src/lib.rs +++ b/azalea-crypto/src/lib.rs @@ -100,4 +100,56 @@ mod tests { let digest = hex_digest(&digest_data(b"simon", &[], &[])); assert_eq!(digest, "88e16a1019277b15d58faf0541e11910eb756f6"); } + + #[test] + fn encode_packet_twice() { + let mut packet = vec![0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09]; + let (mut enc_cipher, _dec_cipher) = create_cipher(b"1234567890123456"); + encrypt_packet(&mut enc_cipher, &mut packet); + assert_eq!(packet, vec![117, 151, 183, 45, 229, 232, 43, 181, 121, 16]); + let mut packet = vec![0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f, 0x10, 0x11, 0x12, 0x13]; + encrypt_packet(&mut enc_cipher, &mut packet); + assert_eq!( + packet, + vec![185, 223, 129, 153, 173, 140, 133, 239, 59, 168] + ); + } + + #[test] + fn encode_packet_long() { + let mut packet = (0..=255).collect::<Vec<u8>>(); + let (mut enc_cipher, _dec_cipher) = create_cipher(b"1234567890123456"); + encrypt_packet(&mut enc_cipher, &mut packet); + assert_eq!( + packet, + vec![ + 117, 151, 183, 45, 229, 232, 43, 181, 121, 16, 185, 223, 129, 153, 173, 140, 133, + 239, 59, 168, 148, 39, 97, 19, 22, 219, 78, 70, 116, 143, 21, 223, 155, 99, 201, + 62, 133, 77, 244, 152, 254, 36, 135, 147, 45, 25, 66, 236, 2, 12, 101, 39, 140, 62, + 7, 57, 19, 101, 217, 91, 142, 243, 0, 3, 100, 142, 160, 21, 219, 145, 151, 37, 11, + 30, 190, 176, 26, 90, 143, 63, 255, 188, 254, 40, 41, 92, 163, 197, 8, 8, 111, 175, + 49, 234, 82, 34, 5, 96, 162, 7, 217, 42, 77, 38, 127, 213, 207, 251, 34, 173, 34, + 132, 23, 12, 118, 59, 51, 216, 173, 199, 137, 95, 132, 222, 243, 195, 81, 60, 205, + 52, 65, 209, 125, 137, 5, 52, 219, 165, 248, 35, 173, 57, 200, 182, 162, 148, 70, + 62, 102, 21, 220, 158, 71, 98, 47, 231, 196, 58, 8, 70, 160, 177, 159, 50, 20, 187, + 31, 249, 68, 26, 142, 171, 239, 193, 10, 174, 14, 80, 238, 114, 124, 185, 253, 246, + 47, 67, 37, 69, 70, 9, 69, 135, 13, 195, 253, 8, 241, 175, 170, 75, 231, 7, 92, 18, + 38, 132, 65, 146, 202, 130, 238, 224, 30, 113, 168, 241, 159, 131, 238, 67, 1, 244, + 74, 172, 86, 95, 192, 236, 198, 188, 81, 67, 49, 230, 166, 52, 224, 238, 11, 252, + 0, 179, 56, 209, 231, 62, 146, 106, 18, 217, 138, 89, 110, 240, 255, 192 + ] + ); + } + + #[test] + fn encode_decode_packet() { + let mut packet = vec![0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09]; + let (mut enc_cipher, mut dec_cipher) = create_cipher(b"1234567890123456"); + encrypt_packet(&mut enc_cipher, &mut packet); + decrypt_packet(&mut dec_cipher, &mut packet); + assert_eq!( + packet, + vec![0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09] + ); + } } |