File: target_info.rb

package info (click to toggle)
ruby-ntlm 0.6.3-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 408 kB
  • sloc: ruby: 2,663; makefile: 6
file content (89 lines) | stat: -rw-r--r-- 2,564 bytes parent folder | download
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
83
84
85
86
87
88
89
module Net
  module NTLM

    # Represents a list of AV_PAIR structures
    # @see https://msdn.microsoft.com/en-us/library/cc236646.aspx
    class TargetInfo

      # Allowed AvId values for an AV_PAIR
      MSV_AV_EOL               = "\x00\x00".freeze
      MSV_AV_NB_COMPUTER_NAME  = "\x01\x00".freeze
      MSV_AV_NB_DOMAIN_NAME    = "\x02\x00".freeze
      MSV_AV_DNS_COMPUTER_NAME = "\x03\x00".freeze
      MSV_AV_DNS_DOMAIN_NAME   = "\x04\x00".freeze
      MSV_AV_DNS_TREE_NAME     = "\x05\x00".freeze
      MSV_AV_FLAGS             = "\x06\x00".freeze
      MSV_AV_TIMESTAMP         = "\x07\x00".freeze
      MSV_AV_SINGLE_HOST       = "\x08\x00".freeze
      MSV_AV_TARGET_NAME       = "\x09\x00".freeze
      MSV_AV_CHANNEL_BINDINGS  = "\x0A\x00".freeze

      # @param av_pair_sequence [String] AV_PAIR list from challenge message
      def initialize(av_pair_sequence)
        @av_pairs = read_pairs(av_pair_sequence)
      end

      attr_reader :av_pairs

      def to_s
        result = ''
        av_pairs.each do |k,v|
          result << k
          result << [v.length].pack('S')
          result << v
        end
        result << Net::NTLM::TargetInfo::MSV_AV_EOL
        result << [0].pack('S')
        result.force_encoding(Encoding::ASCII_8BIT)
      end

      private

      VALID_PAIR_ID = [
          MSV_AV_EOL,
          MSV_AV_NB_COMPUTER_NAME,
          MSV_AV_NB_DOMAIN_NAME,
          MSV_AV_DNS_COMPUTER_NAME,
          MSV_AV_DNS_DOMAIN_NAME,
          MSV_AV_DNS_TREE_NAME,
          MSV_AV_FLAGS,
          MSV_AV_TIMESTAMP,
          MSV_AV_SINGLE_HOST,
          MSV_AV_TARGET_NAME,
          MSV_AV_CHANNEL_BINDINGS
      ].freeze

      def read_pairs(av_pair_sequence)
        offset = 0
        result = {}
        return result if av_pair_sequence.nil?

        until offset >= av_pair_sequence.length
          id = av_pair_sequence[offset..offset+1]

          unless VALID_PAIR_ID.include?(id)
            raise Net::NTLM::InvalidTargetDataError.new( 
              "Invalid AvId #{to_hex(id)} in AV_PAIR structure",
              av_pair_sequence
            )
          end

          length = av_pair_sequence[offset+2..offset+3].unpack('S')[0].to_i
          if length > 0
            value = av_pair_sequence[offset+4..offset+4+length-1]
            result[id] = value
          end

          offset += 4 + length
        end

        result
      end

      def to_hex(str)
        return nil if str.nil?
        str.bytes.map {|b| '0x' + b.to_s(16).rjust(2,'0').upcase}.join('-')
      end
    end
  end
end