From 14087e2665f9cd2dd013d87441c16ec2b7d28321 Mon Sep 17 00:00:00 2001 From: Sun Yimin Date: Thu, 5 Dec 2024 08:14:21 +0800 Subject: [PATCH] zuc: refactor the eea test case issue #284 --- zuc/eea_test.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/zuc/eea_test.go b/zuc/eea_test.go index 64b0b55..78017cc 100644 --- a/zuc/eea_test.go +++ b/zuc/eea_test.go @@ -169,14 +169,15 @@ func TestIssue284(t *testing.T) { if err != nil { t.Error(err) } - src := make([]byte, 200) - expected := make([]byte, 200) - dst := make([]byte, 200) + src := make([]byte, RoundBytes*2) + expected := make([]byte, RoundBytes*2) + dst := make([]byte, RoundBytes*2) c.XORKeyStream(expected, src) - for i := 124; i <= 200; i++ { + for i := RoundBytes - 3; i < RoundBytes+5; i++ { c.XORKeyStreamAt(dst, src[:i], 0) - if !bytes.Equal(expected[:i], dst[:i]) { + c.XORKeyStream(dst[i:], src[i:]) + if !bytes.Equal(expected, dst) { t.Fatalf("failed for len %v", i) } }