123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380 |
- // Copyright © 2015 Daniel Fu <daniel820313@gmail.com>.
- // Copyright © 2019 Loki 'l0k18' Verloren <stalker.loki@protonmail.ch>.
- // Copyright © 2020 Gridfinity, LLC. <admin@gridfinity.com>.
- // Copyright © 2020 Jeffrey H. Johnson <jeff@gridfinity.com>.
- //
- // All rights reserved.
- //
- // All use of this code is governed by the MIT license.
- // The complete license is available in the LICENSE file.
- package lkcp9_test
- import (
- "bytes"
- "crypto/aes"
- "crypto/md5"
- "crypto/rand"
- "hash/crc32"
- "io"
- "testing"
- "go.gridfinity.dev/lkcp9"
- u "go.gridfinity.dev/leaktestfe"
- )
- func TestSM4(
- t *testing.T,
- ) {
- defer u.Leakplug(
- t,
- )
- bc, err := lkcp9.NewSM4BlockCrypt(
- pass[:16],
- )
- if err != nil {
- t.Fatal(
- err,
- )
- }
- cryptTest(
- t,
- bc,
- )
- }
- func TestAES(
- t *testing.T,
- ) {
- defer u.Leakplug(
- t,
- )
- bc, err := lkcp9.NewAESBlockCrypt(
- pass[:32],
- )
- if err != nil {
- t.Fatal(
- err,
- )
- }
- cryptTest(
- t,
- bc,
- )
- }
- func TestXOR(
- t *testing.T,
- ) {
- defer u.Leakplug(
- t,
- )
- bc, err := lkcp9.NewSimpleXORBlockCrypt(
- pass[:32],
- )
- if err != nil {
- t.Fatal(
- err,
- )
- }
- cryptTest(
- t,
- bc,
- )
- }
- func TestNone(
- t *testing.T,
- ) {
- defer u.Leakplug(
- t,
- )
- bc, err := lkcp9.NewNoneBlockCrypt(pass[:32])
- if err != nil {
- t.Fatal(err)
- }
- cryptTest(
- t,
- bc,
- )
- }
- func TestSalsa20(
- t *testing.T,
- ) {
- defer u.Leakplug(
- t,
- )
- bc, err := lkcp9.NewSalsa20BlockCrypt(
- pass[:32],
- )
- if err != nil {
- t.Fatal(
- err,
- )
- }
- cryptTest(
- t,
- bc,
- )
- }
- func cryptTest(
- t *testing.T,
- bc lkcp9.BlockCrypt,
- ) {
- defer u.Leakplug(
- t,
- )
- data := make(
- []byte,
- lkcp9.KcpMtuLimit,
- )
- io.ReadFull(
- rand.Reader,
- data,
- )
- dec := make(
- []byte,
- lkcp9.KcpMtuLimit,
- )
- enc := make(
- []byte,
- lkcp9.KcpMtuLimit,
- )
- bc.Encrypt(
- enc,
- data,
- )
- bc.Decrypt(
- dec,
- enc,
- )
- if !bytes.Equal(
- data,
- dec,
- ) {
- t.Fail()
- }
- }
- func BenchmarkSM4(b *testing.B) {
- bc, err := lkcp9.NewSM4BlockCrypt(pass[:16])
- if err != nil {
- b.Fatal(err)
- }
- benchCrypt(b, bc)
- }
- func BenchmarkAES128(b *testing.B) {
- bc, err := lkcp9.NewAESBlockCrypt(
- pass[:16],
- )
- if err != nil {
- b.Fatal(err)
- }
- benchCrypt(
- b,
- bc,
- )
- }
- func BenchmarkAES192(b *testing.B) {
- bc, err := lkcp9.NewAESBlockCrypt(
- pass[:24],
- )
- if err != nil {
- b.Fatal(
- err,
- )
- }
- benchCrypt(
- b,
- bc,
- )
- }
- func BenchmarkAES256(
- b *testing.B,
- ) {
- bc, err := lkcp9.NewAESBlockCrypt(
- pass[:32],
- )
- if err != nil {
- b.Fatal(
- err,
- )
- }
- benchCrypt(
- b,
- bc,
- )
- }
- func BenchmarkXOR(
- b *testing.B,
- ) {
- bc, err := lkcp9.NewSimpleXORBlockCrypt(
- pass[:32],
- )
- if err != nil {
- b.Fatal(
- err,
- )
- }
- benchCrypt(
- b,
- bc,
- )
- }
- func BenchmarkNone(
- b *testing.B,
- ) {
- bc, err := lkcp9.NewNoneBlockCrypt(
- pass[:32],
- )
- if err != nil {
- b.Fatal(err)
- }
- benchCrypt(
- b,
- bc,
- )
- }
- func BenchmarkSalsa20(b *testing.B) {
- bc, err := lkcp9.NewSalsa20BlockCrypt(pass[:32])
- if err != nil {
- b.Fatal(
- err,
- )
- }
- benchCrypt(
- b,
- bc,
- )
- }
- func benchCrypt(
- b *testing.B,
- bc lkcp9.BlockCrypt,
- ) {
- data := make(
- []byte,
- lkcp9.KcpMtuLimit,
- )
- io.ReadFull(
- rand.Reader,
- data,
- )
- dec := make(
- []byte,
- lkcp9.KcpMtuLimit,
- )
- enc := make(
- []byte,
- lkcp9.KcpMtuLimit,
- )
- b.ReportAllocs()
- b.SetBytes(
- int64(len(
- enc,
- ) * 2,
- ),
- )
- b.ResetTimer()
- for i := 0; i < b.N; i++ {
- bc.Encrypt(
- enc,
- data,
- )
- bc.Decrypt(
- dec,
- enc,
- )
- }
- }
- func BenchmarkCRC32(
- b *testing.B,
- ) {
- content := make(
- []byte,
- 1024,
- )
- b.SetBytes(
- int64(
- len(
- content,
- ),
- ),
- )
- for i := 0; i < b.N; i++ {
- crc32.ChecksumIEEE(
- content,
- )
- }
- }
- func BenchmarkCsprngSystem(
- b *testing.B,
- ) {
- data := make([]byte, md5.Size)
- b.SetBytes(int64(len(data)))
- for i := 0; i < b.N; i++ {
- io.ReadFull(rand.Reader, data)
- }
- }
- func BenchmarkCsprngMD5(b *testing.B) {
- var data [md5.Size]byte
- b.SetBytes(
- md5.Size,
- )
- for i := 0; i < b.N; i++ {
- data = md5.Sum(
- data[:],
- )
- }
- }
- func BenchmarkCsprngKcpNonceMD5(
- b *testing.B,
- ) {
- var ng lkcp9.KcpNonceMD5
- ng.Init()
- b.SetBytes(
- md5.Size,
- )
- data := make(
- []byte,
- md5.Size,
- )
- for i := 0; i < b.N; i++ {
- ng.Fill(
- data,
- )
- }
- }
- func BenchmarkCsprngNonceAES128(
- b *testing.B,
- ) {
- var ng lkcp9.KcpNonceAES128
- ng.Init()
- b.SetBytes(
- aes.BlockSize,
- )
- data := make(
- []byte,
- aes.BlockSize,
- )
- for i := 0; i < b.N; i++ {
- ng.Fill(
- data,
- )
- }
- }
|