Class: Rex::Post::Meterpreter::Extensions::Stdapi::Railgun::Library

Inherits:
Object
  • Object
show all
Includes:
LibraryHelper
Defined in:
lib/rex/post/meterpreter/extensions/stdapi/railgun/library.rb

Overview

Represents a library, e.g. kernel32.dll

Constant Summary collapse

@@datatype_map =
{
  'HANDLE'   => 'LPVOID',
  # really should be PVOID* but LPVOID is handled specially with the 'L' prefix to *not* treat it as a pointer, and
  # for railgun's purposes LPVOID == ULONG_PTR
  'PHANDLE'  => 'PULONG_PTR',
  'SIZE_T'   => 'ULONG_PTR',
  'PSIZE_T'  => 'PULONG_PTR',
  'PLPVOID'  => 'PULONG_PTR',
  'ULONG'    => 'DWORD',
  'PULONG'   => 'PDWORD',
  'NTSTATUS' => 'DWORD'
}.freeze

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from LibraryHelper

#asciiz_to_str, #assemble_buffer, #param_to_number, #str_to_ascii_z, #str_to_uni_z, #uniz_to_str

Constructor Details

#initialize(library_path, consts_mgr) ⇒ Library

Returns a new instance of Library.



62
63
64
65
66
67
68
69
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/library.rb', line 62

def initialize(library_path, consts_mgr)
  @library_path = library_path

  # needed by LibraryHelper
  @consts_mgr = consts_mgr

  self.functions = {}
end

Instance Attribute Details

#functionsObject

Returns the value of attribute functions.



59
60
61
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/library.rb', line 59

def functions
  @functions
end

#library_pathObject (readonly)

Returns the value of attribute library_path.



60
61
62
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/library.rb', line 60

def library_path
  @library_path
end

Instance Method Details

#add_function(name, return_type, params, remote_name = nil, calling_conv = 'stdcall') ⇒ Object

Define a function for this library.

Every function argument is described by a tuple (type,name,direction)

Example:

add_function("MessageBoxW",   # name
  "DWORD",                    # return value
  [                           # params
 ["DWORD","hWnd","in"],
   ["PWCHAR","lpText","in"],
   ["PWCHAR","lpCaption","in"],
   ["DWORD","uType","in"],
  ])

Use remote_name when the actual library name is different from the ruby variable. You might need to do this for example when the actual func name is myFunc@4 or when you want to create an alternative version of an existing function.

When the new function is called it will return a list containing the return value and all inout params. See #call_function.



125
126
127
128
129
130
131
132
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/library.rb', line 125

def add_function(name, return_type, params, remote_name=nil, calling_conv='stdcall')
  return_type = reduce_type(return_type)
  params = reduce_parameter_types(params)
  if remote_name == nil
    remote_name = name
  end
  @functions[name] = LibraryFunction.new(return_type, params, remote_name, calling_conv)
end

#build_packet_and_layouts(packet, function, args, arch) ⇒ Object



134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/library.rb', line 134

def build_packet_and_layouts(packet, function, args, arch)
  case arch
  when ARCH_X64
    native = 'Q<'
  when ARCH_X86
    native = 'V'
  else
    raise NotImplementedError, 'Unsupported architecture (must be ARCH_X86 or ARCH_X64)'
  end

  # We transmit the immediate stack and three heap-buffers:
  # in, inout and out. The reason behind the separation is bandwidth.
  # We don't want to transmit uninitialized data in or no-longer-needed data out.

  # out-only-buffers that are ONLY transmitted on the way BACK
  out_only_layout = {} # paramName => BufferItem
  out_only_size_bytes = 0
  #puts " assembling out-only buffer"
  function.params.each_with_index do |param_desc, param_idx|
    #puts " processing #{param_desc[1]}"

    # Special case:
    # The user can choose to supply a Null pointer instead of a buffer
    # in this case we don't need space in any heap buffer
    if param_desc[0][0,1] == 'P' # type is a pointer (except LPVOID where the L negates this)
      if args[param_idx] == nil
        next
      end
    end

    # we care only about out-only buffers
    if param_desc[2] == 'out'
      if !args[param_idx].kind_of? Integer
        raise "error in param #{param_desc[1]}: Out-only buffers must be described by a number indicating their size in bytes"
      end
      buffer_size = args[param_idx]
      if param_desc[0] == 'PULONG_PTR'
        # bump up the size for an x64 pointer
        if arch == ARCH_X64 && buffer_size == 4
          buffer_size = args[param_idx] = 8
        end

        if arch == ARCH_X64
          if buffer_size != 8
            raise "Please pass 8 for 'out' PULONG_PTR, since they require a buffer of size 8"
          end
        elsif arch == ARCH_X86
          if buffer_size != 4
            raise "Please pass 4 for 'out' PULONG_PTR, since they require a buffer of size 4"
          end
        end
      end

      out_only_layout[param_desc[1]] = BufferItem.new(param_idx, out_only_size_bytes, buffer_size, param_desc[0])
      out_only_size_bytes += buffer_size
    end
  end

  in_only_layout, in_only_buffer = assemble_buffer('in', function, args, arch)
  inout_layout, inout_buffer = assemble_buffer('inout', function, args, arch)

  # now we build the stack
  # every stack dword will be described by two dwords:
  # first dword describes second dword:
  #	0 - literal,
  #	1 = relative to in-only buffer
  #	2 = relative to out-only buffer
  #	3 = relative to inout buffer

  # (literal numbers and pointers to buffers we have created)
  literal_pairs_blob = ""
  #puts " assembling literal stack"
  function.params.each_with_index do |param_desc, param_idx|
    #puts "  processing (#{param_desc[0]}, #{param_desc[1]}, #{param_desc[2]})"
    buffer = nil
    # is it a pointer to a buffer on our stack
    if ['PULONG_PTR', 'PDWORD', 'PWCHAR', 'PCHAR', 'PBLOB'].include?(param_desc[0])
      if ['PWCHAR', 'PCHAR', 'PBLOB'].include?(param_desc[0]) && param_desc[2] == 'in' && args[param_idx].is_a?(Integer)
        # allow PWCHAR, PCHAR and PBLOB to also be passed as a pointer instead of a buffer
        buffer  = [0].pack(native)
        num     = param_to_number(args[param_idx])
        buffer += [num].pack(native)
      elsif args[param_idx] == nil # null pointer?
        buffer  = [0].pack(native) # type: LPVOID  (so the library does not rebase it)
        buffer += [0].pack(native) # value: 0
      elsif param_desc[2] == 'in'
        buffer  = [1].pack(native)
        buffer += [in_only_layout[param_desc[1]].addr].pack(native)
      elsif param_desc[2] == 'out'
        buffer  = [2].pack(native)
        buffer += [out_only_layout[param_desc[1]].addr].pack(native)
      elsif param_desc[2] == 'inout'
        buffer  = [3].pack(native)
        buffer += [inout_layout[param_desc[1]].addr].pack(native)
      else
        raise 'unexpected direction'
      end
    else
      #puts "   not a pointer"
      # it's not a pointer (LPVOID is a pointer but is not backed by railgun memory, ala PBLOB)
      buffer = [0].pack(native)
      case param_desc[0]
        when 'LPVOID', 'ULONG_PTR'
          num     = param_to_number(args[param_idx])
          buffer += [num].pack(native)
        when 'DWORD'
          num     = param_to_number(args[param_idx])
          buffer += [num & 0xffffffff].pack(native)
        when 'WORD'
          num     = param_to_number(args[param_idx])
          buffer += [num & 0xffff].pack(native)
        when 'BYTE'
          num     = param_to_number(args[param_idx])
          buffer += [num & 0xff].pack(native)
        when 'BOOL'
          case args[param_idx]
            when true
              buffer += [1].pack(native)
            when false
              buffer += [0].pack(native)
            else
              raise "param #{param_desc[1]}: true or false expected"
          end
        else
          raise "unexpected type for param #{param_desc[1]}"
      end
    end

    #puts "   adding pair to blob"
    literal_pairs_blob += buffer
    #puts "   buffer size %X" % buffer.length
    #puts "   blob size so far: %X" % literal_pairs_blob.length
  end

  layouts = {in: in_only_layout, inout: inout_layout, out: out_only_layout}

  packet.add_tlv(TLV_TYPE_RAILGUN_SIZE_OUT, out_only_size_bytes)
  packet.add_tlv(TLV_TYPE_RAILGUN_STACKBLOB, literal_pairs_blob)
  packet.add_tlv(TLV_TYPE_RAILGUN_BUFFERBLOB_IN, in_only_buffer)
  packet.add_tlv(TLV_TYPE_RAILGUN_BUFFERBLOB_INOUT, inout_buffer)

  packet.add_tlv(TLV_TYPE_RAILGUN_LIBNAME, @library_path)
  packet.add_tlv(TLV_TYPE_RAILGUN_FUNCNAME, function.remote_name)
  packet.add_tlv(TLV_TYPE_RAILGUN_CALLCONV, function.calling_conv)
  [packet, layouts]
end

#build_response(packet, function, layouts, client) ⇒ Object



281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/library.rb', line 281

def build_response(packet, function, layouts, client)
  case client.native_arch
  when ARCH_X64
    native = 'Q<'
  when ARCH_X86
    native = 'V'
  else
    raise NotImplementedError, 'Unsupported architecture (must be ARCH_X86 or ARCH_X64)'
  end

  rec_inout_buffers = packet.get_tlv_value(TLV_TYPE_RAILGUN_BACK_BUFFERBLOB_INOUT)
  rec_out_only_buffers = packet.get_tlv_value(TLV_TYPE_RAILGUN_BACK_BUFFERBLOB_OUT)
  rec_return_value = packet.get_tlv_value(TLV_TYPE_RAILGUN_BACK_RET)
  rec_last_error = packet.get_tlv_value(TLV_TYPE_RAILGUN_BACK_ERR)
  rec_err_msg = packet.get_tlv_value(TLV_TYPE_RAILGUN_BACK_MSG)

  # Error messages come back with trailing CRLF, so strip it out if we do get a message.
  rec_err_msg.strip! unless rec_err_msg.nil?

  # the hash the function returns
  return_hash = {
    'GetLastError' => rec_last_error,
    'ErrorMessage' => rec_err_msg
  }

  # process return value
  case function.return_type
    when 'LPVOID', 'ULONG_PTR'
      if client.native_arch == ARCH_X64
        return_hash['return'] = rec_return_value
      else
        return_hash['return'] = rec_return_value & 0xffffffff
      end
    when 'DWORD'
      return_hash['return'] = rec_return_value & 0xffffffff
    when 'WORD'
      return_hash['return'] = rec_return_value & 0xffff
    when 'BYTE'
      return_hash['return'] = rec_return_value & 0xff
    when 'BOOL'
      return_hash['return'] = (rec_return_value != 0)
    when 'VOID'
      return_hash['return'] = nil
    when 'PCHAR'
      return_hash['return'] = rec_return_value == 0 ? nil : client.railgun.util.read_string(rec_return_value)
      return_hash['&return'] = rec_return_value
    when 'PWCHAR'
      return_hash['return'] = rec_return_value == 0 ? nil : client.railgun.util.read_wstring(rec_return_value)
      return_hash['&return'] = rec_return_value
    when 'PULONG_PTR'
      if client.native_arch == ARCH_X64
        return_hash['return'] = rec_return_value == 0 ? nil : client.railgun.util.memread(rec_return_value, 8)&.unpack1('Q<')
        return_hash['&return'] = rec_return_value
      else
        return_hash['return'] = rec_return_value == 0 ? nil : client.railgun.util.memread(rec_return_value, 4)&.unpack1('V')
        return_hash['&return'] = rec_return_value
      end
    else
      raise "unexpected return type: #{function.return_type}"
  end

  # process out-only buffers
  layouts[:out].each_pair do |param_name, buffer_item|
    buffer = rec_out_only_buffers[buffer_item.addr, buffer_item.length_in_bytes]
    case buffer_item.datatype
      when 'PULONG_PTR'
        return_hash[param_name] = buffer.unpack(native).first
      when 'PDWORD'
        return_hash[param_name] = buffer.unpack('V').first
      when 'PCHAR'
        return_hash[param_name] = asciiz_to_str(buffer)
      when 'PWCHAR'
        return_hash[param_name] = uniz_to_str(buffer)
      when 'PBLOB'
        return_hash[param_name] = buffer
      else
        raise "unexpected type in out-only buffer of #{param_name}: #{buffer_item.datatype}"
    end
  end

  # process in-out buffers
  layouts[:inout].each_pair do |param_name, buffer_item|
    buffer = rec_inout_buffers[buffer_item.addr, buffer_item.length_in_bytes]
    case buffer_item.datatype
      when 'PULONG_PTR'
        return_hash[param_name] = buffer.unpack(native).first
      when 'PDWORD'
        return_hash[param_name] = buffer.unpack('V').first
      when 'PCHAR'
        return_hash[param_name] = asciiz_to_str(buffer)
      when 'PWCHAR'
        return_hash[param_name] = uniz_to_str(buffer)
      when 'PBLOB'
        return_hash[param_name] = buffer
      else
        raise "unexpected type in in-out-buffer of #{param_name}: #{buffer_item.datatype}"
    end
  end

  return_hash
end

#call_function(function, args, client) ⇒ Object

Perform a function call in this library on the remote system.

Returns a Hash containing the return value, the result of GetLastError(), and any inout parameters.

Raises an exception if function is not a known function in this library, i.e., it hasn't been defined in a Def.



88
89
90
91
92
93
94
95
96
97
98
99
100
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/library.rb', line 88

def call_function(function, args, client)
  unless function.instance_of? LibraryFunction
    func_name = function.to_s

    unless known_function_names.include? func_name
      raise "Library-function #{func_name} not found. Known functions: #{PP.pp(known_function_names, '')}"
    end

    function = get_function(func_name)
  end

  return process_function_call(function, args, client)
end

#get_function(name) ⇒ Object



75
76
77
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/library.rb', line 75

def get_function(name)
  return functions[name]
end

#known_function_namesObject



71
72
73
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/library.rb', line 71

def known_function_names
  return functions.keys
end