1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
# frozen_string_literal: true
require_relative "utils"
require 'benchmark'
if defined?(OpenSSL)
class OpenSSL::OSSL < OpenSSL::SSLTestCase
def test_secure_compare
assert_raises(ArgumentError) { OpenSSL.secure_compare("aaa", "a") }
assert_raises(ArgumentError) { OpenSSL.secure_compare("aaa", "aa") }
assert OpenSSL.secure_compare("aaa", "aaa")
assert OpenSSL.secure_compare(
OpenSSL::Digest::SHA256.digest("aaa"), OpenSSL::Digest::SHA256.digest("aaa")
)
assert_raises(ArgumentError) { OpenSSL.secure_compare("aaa", "aaaa") }
refute OpenSSL.secure_compare("aaa", "baa")
refute OpenSSL.secure_compare("aaa", "aba")
refute OpenSSL.secure_compare("aaa", "aab")
assert_raises(ArgumentError) { OpenSSL.secure_compare("aaa", "aaab") }
assert_raises(ArgumentError) { OpenSSL.secure_compare("aaa", "b") }
assert_raises(ArgumentError) { OpenSSL.secure_compare("aaa", "bb") }
refute OpenSSL.secure_compare("aaa", "bbb")
assert_raises(ArgumentError) { OpenSSL.secure_compare("aaa", "bbbb") }
end
def test_memcmp_timing
# Ensure using secure_compare takes almost exactly the same amount of time to compare two different strings.
# Regular string comparison will short-circuit on the first non-matching character, failing this test.
# NOTE: this test may be susceptible to noise if the system running the tests is otherwise under load.
a = "x" * 512_000
b = "#{a}y"
c = "y#{a}"
a = "#{a}x"
n = 10_000
a_b_time = Benchmark.measure { n.times { OpenSSL.secure_compare(a, b) } }.real
a_c_time = Benchmark.measure { n.times { OpenSSL.secure_compare(a, c) } }.real
assert_in_delta(a_b_time, a_c_time, 1, "secure_compare timing test failed")
end
end
end
|