aboutsummaryrefslogtreecommitdiff
path: root/device/noise_test.go
diff options
context:
space:
mode:
authorJason A. Donenfeld <Jason@zx2c4.com>2023-02-16 15:51:30 +0100
committerJason A. Donenfeld <Jason@zx2c4.com>2023-02-16 16:33:14 +0100
commitc7b76d3d9ecdc2ffde80decadda88c0c7cdfeedf (patch)
tree801fe59cc2d9c203de1dd69bf5cf15bf5d097186 /device/noise_test.go
parent1e2c3e5a3c1463cb8c7ec92d74aa739587b6642f (diff)
downloadwireguard-go-c7b76d3d9ecdc2ffde80decadda88c0c7cdfeedf.tar.gz
wireguard-go-c7b76d3d9ecdc2ffde80decadda88c0c7cdfeedf.zip
device: uniformly check ECDH output for zeros
For some reason, this was omitted for response messages. Reported-by: z <dzm@unexpl0.red> Fixes: 8c34c4c ("First set of code review patches") Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>
Diffstat (limited to 'device/noise_test.go')
-rw-r--r--device/noise_test.go6
1 files changed, 3 insertions, 3 deletions
diff --git a/device/noise_test.go b/device/noise_test.go
index 587d1e5..2dd5324 100644
--- a/device/noise_test.go
+++ b/device/noise_test.go
@@ -24,10 +24,10 @@ func TestCurveWrappers(t *testing.T) {
pk1 := sk1.publicKey()
pk2 := sk2.publicKey()
- ss1 := sk1.sharedSecret(pk2)
- ss2 := sk2.sharedSecret(pk1)
+ ss1, err1 := sk1.sharedSecret(pk2)
+ ss2, err2 := sk2.sharedSecret(pk1)
- if ss1 != ss2 {
+ if ss1 != ss2 || err1 != nil || err2 != nil {
t.Fatal("Failed to compute shared secet")
}
}