summaryrefslogtreecommitdiffstats
path: root/test/test_pkey.rb
blob: 79647c8f1d3d156c930e091eac4523b67679f1f7 (plain)
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
46
47
48
49
# frozen_string_literal: false
require_relative "utils"

if defined?(OpenSSL::TestUtils)

class OpenSSL::TestPKey < OpenSSL::PKeyTestCase
  PKEYS = {
    OpenSSL::PKey::RSA => {
      key: OpenSSL::TestUtils::TEST_KEY_RSA1024,
      digest: OpenSSL::Digest::SHA1,
    },
    OpenSSL::PKey::DSA => {
      key: OpenSSL::TestUtils::TEST_KEY_DSA512,
      digest: OpenSSL::TestUtils::DSA_SIGNATURE_DIGEST,
    },
  }
  if defined?(OpenSSL::PKey::EC)
    PKEYS[OpenSSL::PKey::EC] = {
      key: OpenSSL::TestUtils::TEST_KEY_EC_P256V1,
      digest: OpenSSL::Digest::SHA1,
    }
  end

  def test_sign_verify
    data = "Sign me!"
    invalid_data = "Sign me?"
    PKEYS.each do |klass, prop|
      key = prop[:key]
      pub_key = dup_public(prop[:key])
      digest = prop[:digest].new
      signature = key.sign(digest, data)
      assert_equal(true, pub_key.verify(digest, signature, data))
      assert_equal(false, pub_key.verify(digest, signature, invalid_data))
      # digest state is irrelevant
      digest << "unya"
      assert_equal(true, pub_key.verify(digest, signature, data))
      assert_equal(false, pub_key.verify(digest, signature, invalid_data))

      if OpenSSL::OPENSSL_VERSION_NUMBER > 0x10000000
        digest = OpenSSL::Digest::SHA256.new
        signature = key.sign(digest, data)
        assert_equal(true, pub_key.verify(digest, signature, data))
        assert_equal(false, pub_key.verify(digest, signature, invalid_data))
      end
    end
  end
end

end