summaryrefslogtreecommitdiff
path: root/lib/net/ssh/transport/hmac/abstract.rb
blob: 22ad9a6e341c02fe00daca2753214b7acb4924bb (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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
require 'openssl'
require 'openssl/digest'

module Net
  module SSH
    module Transport
      module HMAC

        # The base class of all OpenSSL-based HMAC algorithm wrappers.
        class Abstract
          class <<self
            def key_length(*v)
              @key_length = nil if !defined?(@key_length)
              if v.empty?
                @key_length = superclass.key_length if @key_length.nil? && superclass.respond_to?(:key_length)
                return @key_length
              elsif v.length == 1
                @key_length = v.first
              else
                raise ArgumentError, "wrong number of arguments (#{v.length} for 1)"
              end
            end

            def mac_length(*v)
              @mac_length = nil if !defined?(@mac_length)
              if v.empty?
                @mac_length = superclass.mac_length if @mac_length.nil? && superclass.respond_to?(:mac_length)
                return @mac_length
              elsif v.length == 1
                @mac_length = v.first
              else
                raise ArgumentError, "wrong number of arguments (#{v.length} for 1)"
              end
            end

            def digest_class(*v)
              @digest_class = nil if !defined?(@digest_class)
              if v.empty?
                @digest_class = superclass.digest_class if @digest_class.nil? && superclass.respond_to?(:digest_class)
                return @digest_class
              elsif v.length == 1
                @digest_class = v.first
              else
                raise ArgumentError, "wrong number of arguments (#{v.length} for 1)"
              end
            end
          end

          def key_length
            self.class.key_length
          end

          def mac_length
            self.class.mac_length
          end

          def digest_class
            self.class.digest_class
          end

          # The key in use for this instance.
          attr_reader :key

          def initialize(key=nil)
            self.key = key
          end

          # Sets the key to the given value, truncating it so that it is the correct
          # length.
          def key=(value)
            @key = value ? value.to_s[0,key_length] : nil
          end

          # Compute the HMAC digest for the given data string.
          def digest(data)
            OpenSSL::HMAC.digest(digest_class.new, key, data)[0,mac_length]
          end
        end
      end
    end
  end
end