test/openssl/test_digest.rb

   1 # frozen_string_literal: true
   2 require_relative 'utils'
   3
   4 if defined?(OpenSSL)
   5
   6 class OpenSSL::TestDigest < OpenSSL::TestCase
   7   def setup
   8     super
   9     @d1 = OpenSSL::Digest.new("MD5")
  10     @d2 = OpenSSL::Digest::MD5.new
  11   end
  12
  13   def test_digest
  14     null_hex = "d41d8cd98f00b204e9800998ecf8427e"
  15     null_bin = [null_hex].pack("H*")
  16     data = "DATA"
  17     hex = "e44f9e348e41cb272efa87387728571b"
  18     bin = [hex].pack("H*")
  19     assert_equal(null_bin, @d1.digest)
  20     assert_equal(null_hex, @d1.hexdigest)
  21     @d1 << data
  22     assert_equal(bin, @d1.digest)
  23     assert_equal(hex, @d1.hexdigest)
  24     assert_equal(bin, OpenSSL::Digest.digest('MD5', data))
  25     assert_equal(hex, OpenSSL::Digest.hexdigest('MD5', data))
  26   end
  27
  28   def test_eql
  29     assert(@d1 == @d2, "==")
  30     d = @d1.clone
  31     assert(d == @d1, "clone")
  32   end
  33
  34   def test_info
  35     assert_equal("MD5", @d1.name, "name")
  36     assert_equal("MD5", @d2.name, "name")
  37     assert_equal(16, @d1.size, "size")
  38   end
  39
  40   def test_dup
  41     @d1.update("DATA")
  42     assert_equal(@d1.name, @d1.dup.name, "dup")
  43     assert_equal(@d1.name, @d1.clone.name, "clone")
  44     assert_equal(@d1.digest, @d1.clone.digest, "clone .digest")
  45   end
  46
  47   def test_reset
  48     @d1.update("DATA")
  49     dig1 = @d1.digest
  50     @d1.reset
  51     @d1.update("DATA")
  52     dig2 = @d1.digest
  53     assert_equal(dig1, dig2, "reset")
  54   end
  55
  56   def test_digest_constants
  57     %w{MD5 SHA1 SHA224 SHA256 SHA384 SHA512}.each do |name|
  58       assert_not_nil(OpenSSL::Digest.new(name))
  59       klass = OpenSSL::Digest.const_get(name.tr('-', '_'))
  60       assert_not_nil(klass.new)
  61     end
  62   end
  63
  64   def test_digest_by_oid_and_name
  65     check_digest(OpenSSL::ASN1::ObjectId.new("MD5"))
  66     check_digest(OpenSSL::ASN1::ObjectId.new("SHA1"))
  67   end
  68
  69   def encode16(str)
  70     str.unpack("H*").first
  71   end
  72
  73   def test_sha2
  74     sha224_a = "abd37534c7d9a2efb9465de931cd7055ffdb8879563ae98078d6d6d5"
  75     sha256_a = "ca978112ca1bbdcafac231b39a23dc4da786eff8147c4e72b9807785afee48bb"
  76     sha384_a = "54a59b9f22b0b80880d8427e548b7c23abd873486e1f035dce9cd697e85175033caa88e6d57bc35efae0b5afd3145f31"
  77     sha512_a = "1f40fc92da241694750979ee6cf582f2d5d7d28e18335de05abc54d0560e0f5302860c652bf08d560252aa5e74210546f369fbbbce8c12cfc7957b2652fe9a75"
  78
  79     assert_equal(sha224_a, OpenSSL::Digest.hexdigest('SHA224', "a"))
  80     assert_equal(sha256_a, OpenSSL::Digest.hexdigest('SHA256', "a"))
  81     assert_equal(sha384_a, OpenSSL::Digest.hexdigest('SHA384', "a"))
  82     assert_equal(sha512_a, OpenSSL::Digest.hexdigest('SHA512', "a"))
  83
  84     assert_equal(sha224_a, encode16(OpenSSL::Digest.digest('SHA224', "a")))
  85     assert_equal(sha256_a, encode16(OpenSSL::Digest.digest('SHA256', "a")))
  86     assert_equal(sha384_a, encode16(OpenSSL::Digest.digest('SHA384', "a")))
  87     assert_equal(sha512_a, encode16(OpenSSL::Digest.digest('SHA512', "a")))
  88   end
  89
  90   def test_sha512_truncate
  91     pend "SHA512_224 is not implemented" unless digest_available?('SHA512-224')
  92     sha512_224_a = "d5cdb9ccc769a5121d4175f2bfdd13d6310e0d3d361ea75d82108327"
  93     sha512_256_a = "455e518824bc0601f9fb858ff5c37d417d67c2f8e0df2babe4808858aea830f8"
  94
  95     assert_equal(sha512_224_a, OpenSSL::Digest.hexdigest('SHA512-224', "a"))
  96     assert_equal(sha512_256_a, OpenSSL::Digest.hexdigest('SHA512-256', "a"))
  97
  98     assert_equal(sha512_224_a, encode16(OpenSSL::Digest.digest('SHA512-224', "a")))
  99     assert_equal(sha512_256_a, encode16(OpenSSL::Digest.digest('SHA512-256', "a")))
 100   end
 101
 102   def test_sha3
 103     pend "SHA3 is not implemented" unless digest_available?('SHA3-224')
 104     s224 = '6b4e03423667dbb73b6e15454f0eb1abd4597f9a1b078e3f5b5a6bc7'
 105     s256 = 'a7ffc6f8bf1ed76651c14756a061d662f580ff4de43b49fa82d80a4b80f8434a'
 106     s384 = '0c63a75b845e4f7d01107d852e4c2485c51a50aaaa94fc61995e71bbee983a2ac3713831264adb47fb6bd1e058d5f004'
 107     s512 = 'a69f73cca23a9ac5c8b567dc185a756e97c982164fe25859e0d1dcc1475c80a615b2123af1f5f94c11e3e9402c3ac558f500199d95b6d3e301758586281dcd26'
 108     assert_equal(OpenSSL::Digest.hexdigest('SHA3-224', ""), s224)
 109     assert_equal(OpenSSL::Digest.hexdigest('SHA3-256', ""), s256)
 110     assert_equal(OpenSSL::Digest.hexdigest('SHA3-384', ""), s384)
 111     assert_equal(OpenSSL::Digest.hexdigest('SHA3-512', ""), s512)
 112   end
 113
 114   def test_digest_by_oid_and_name_sha2
 115     check_digest(OpenSSL::ASN1::ObjectId.new("SHA224"))
 116     check_digest(OpenSSL::ASN1::ObjectId.new("SHA256"))
 117     check_digest(OpenSSL::ASN1::ObjectId.new("SHA384"))
 118     check_digest(OpenSSL::ASN1::ObjectId.new("SHA512"))
 119   end
 120
 121   def test_openssl_digest
 122     assert_equal OpenSSL::Digest::MD5, OpenSSL::Digest("MD5")
 123
 124     assert_raise NameError do
 125       OpenSSL::Digest("no such digest")
 126     end
 127   end
 128
 129   private
 130
 131   def check_digest(oid)
 132     d = OpenSSL::Digest.new(oid.sn)
 133     assert_not_nil(d)
 134     d = OpenSSL::Digest.new(oid.ln)
 135     assert_not_nil(d)
 136     d = OpenSSL::Digest.new(oid.oid)
 137     assert_not_nil(d)
 138   end
 139
 140   def digest_available?(name)
 141     begin
 142       OpenSSL::Digest.new(name)
 143     rescue RuntimeError
 144       false
 145     end
 146   end
 147 end
 148
 149 end