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

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

Overview

The Railgun class to dynamically expose the Windows API.

Constant Summary collapse

BUILTIN_LIBRARIES =

Railgun::Library’s that have builtin definitions.

If you want to add additional library definitions to be preloaded create a definition class ‘rex/post/meterpreter/extensions/stdapi/railgun/def/$platform/’. Naming is important and should follow convention. For example, if your library’s name was “my_library” file name: def_my_library.rb class name: Def_my_library entry below: ‘my_library’

{
  'linux' => [
    'libc'
  ].freeze,
  'osx' => [
    'libc',
    'libobjc'
  ].freeze,
  'windows' => [
    'kernel32',
    'ntdll',
    'user32',
    'ws2_32',
    'iphlpapi',
    'advapi32',
    'shell32',
    'netapi32',
    'crypt32',
    'wlanapi',
    'wldap32',
    'version',
    'psapi',
    'dbghelp',
    'winspool',
    'spoolss',
    'secur32'
  ].freeze
}.freeze
@@cached_libraries =

These libraries are loaded lazily and then shared amongst all railgun instances. For safety reasons this variable should only be read/written within #get_library.

{}
@@cache_semaphore =

if you are going to touch @@cached_libraries, wear protection

Mutex.new

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(client) ⇒ Railgun

Returns a new instance of Railgun.



123
124
125
126
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 123

def initialize(client)
  self.client = client
  self.libraries = {}
end

Dynamic Method Handling

This class handles dynamic methods through the method_missing method

#method_missing(lib_symbol, *args) ⇒ Object

Fake having members like user32 and kernel32. reason is that

...user32.MessageBoxW()

is prettier than

...libraries["user32"].functions["MessageBoxW"]()


297
298
299
300
301
302
303
304
305
306
307
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 297

def method_missing(lib_symbol, *args)
  lib_name = lib_symbol.to_s

  unless known_library_names.include? lib_name
    raise "Library #{lib_name} not found. Known libraries: #{PP.pp(known_library_names, '')}"
  end

  lib = get_library(lib_name)

  return LibraryWrapper.new(lib, client)
end

Instance Attribute Details

#clientObject

Contains a reference to the client that corresponds to this instance of railgun



112
113
114
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 112

def client
  @client
end

#librariesObject

Returns a Hash containing libraries added to this instance with #add_library as well as references to any frozen cached libraries added directly in #get_library and copies of any frozen libraries (added directly with #add_function) that the user attempted to modify with #add_function.

Keys are friendly library names and values are the corresponding library instance



108
109
110
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 108

def libraries
  @libraries
end

Class Method Details

.builtin_librariesObject



128
129
130
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 128

def self.builtin_libraries
  BUILTIN_LIBRARIES[client.platform]
end

Instance Method Details

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

Adds a function to an existing library definition.

If the library definition is frozen (ideally this should be the case for all cached libraries) an unfrozen copy is created and used henceforth for this instance.



210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 210

def add_function(lib_name, function_name, return_type, params, remote_name=nil, calling_conv='stdcall')
  unless known_library_names.include?(lib_name)
    raise "Library #{lib_name} not found. Known libraries: #{PP.pp(known_library_names, '')}"
  end

  lib = get_library(lib_name)

  # For backwards compatibility, we ensure the library is thawed
  if lib.frozen?
    # Duplicate not only the library, but its functions as well, frozen status will be lost
    lib = Marshal.load(Marshal.dump(lib))

    # Update local libraries with the modifiable duplicate
    libraries[lib_name] = lib
  end

  lib.add_function(function_name, return_type, params, remote_name, calling_conv)
end

#add_library(lib_name, remote_name = lib_name) ⇒ Object Also known as: add_dll

Adds a library to this Railgun.

The remote_name is the name used on the remote system and should be set appropriately if you want to include a path or the library name contains non-ruby-approved characters.

Raises an exception if a library with the given name has already been defined.



239
240
241
242
243
244
245
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 239

def add_library(lib_name, remote_name=lib_name)
  if libraries.has_key? lib_name
    raise "A library of name #{lib_name} has already been loaded."
  end

  libraries[lib_name] = Library.new(remote_name, constant_manager)
end

#api_constantsObject

Return this Railgun’s platform specific ApiConstants class.



146
147
148
149
150
151
152
153
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 146

def api_constants
  if @api_constants.nil?
    require "rex/post/meterpreter/extensions/stdapi/railgun/def/#{client.platform}/api_constants"
    @api_constants = Def.const_get('DefApiConstants_' << client.platform)
  end

  return @api_constants
end

#const(str) ⇒ Object

Return a constant matching str.



312
313
314
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 312

def const(str)
  return constant_manager.parse(str)
end

#constant_managerObject

Return this Railgun’s ConstManager instance, initially populated with constants defined in ApiConstants.



159
160
161
162
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 159

def constant_manager
  # Loads lazily
  return api_constants.manager
end

#get_library(lib_name) ⇒ Object Also known as: get_dll

Attempts to provide a library instance of the given name. Handles lazy loading and caching. Note that if a library of the given name does not exist then nil is returned.



257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 257

def get_library(lib_name)
  # If the library is not local, we now either load it from cache or load it
  # lazily. In either case, a reference to the library is stored in the
  # collection "libraries". If the library can not be found/created, no
  # actions are taken.
  unless libraries.has_key? lib_name
    # use a platform-specific name for caching to avoid conflicts with
    # libraries that exist on multiple platforms, e.g. libc.
    cached_lib_name = "#{client.platform}.#{lib_name}"
    # We read and write to @@cached_libraries and rely on state consistency
    @@cache_semaphore.synchronize do
      if @@cached_libraries.has_key? cached_lib_name
        libraries[lib_name] = @@cached_libraries[cached_lib_name]
      elsif BUILTIN_LIBRARIES[client.platform].include? lib_name
        # I highly doubt this case will ever occur, but I am paranoid
        if lib_name !~ /^\w+$/
          raise "Library name #{lib_name} is bad. Correct Railgun::BUILTIN_LIBRARIES['#{client.platform}']"
        end

        require "rex/post/meterpreter/extensions/stdapi/railgun/def/#{client.platform}/def_#{lib_name}"
        lib = Def.const_get("Def_#{client.platform}_#{lib_name}").create_library(constant_manager).freeze

        @@cached_libraries[cached_lib_name] = lib
        libraries[lib_name] = lib
      end
    end

  end

  return libraries[lib_name]
end

#known_library_namesObject



248
249
250
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 248

def known_library_names
  return BUILTIN_LIBRARIES[client.platform] | libraries.keys
end

#memread(address, length) ⇒ Object

Read data from a memory address on the host (useful for working with LPVOID parameters)



168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 168

def memread(address, length)

  raise "Invalid parameters." if(not address or not length)

  request = Packet.create_request(COMMAND_ID_STDAPI_RAILGUN_MEMREAD)

  request.add_tlv(TLV_TYPE_RAILGUN_MEM_ADDRESS, address)
  request.add_tlv(TLV_TYPE_RAILGUN_MEM_LENGTH, length)

  response = client.send_request(request)
  if(response.result == 0)
    return response.get_tlv_value(TLV_TYPE_RAILGUN_MEM_DATA)
  end

  return nil
end

#memwrite(address, data, length = nil) ⇒ Object

Write data to a memory address on the host (useful for working with LPVOID parameters)



189
190
191
192
193
194
195
196
197
198
199
200
201
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 189

def memwrite(address, data, length=nil)
  data = data.to_binary_s if data.is_a?(BinData::Struct)
  length = data.length if length.nil?
  raise "Invalid parameters." if(not address or not data or not length)

  request = Packet.create_request(COMMAND_ID_STDAPI_RAILGUN_MEMWRITE)
  request.add_tlv(TLV_TYPE_RAILGUN_MEM_ADDRESS, address)
  request.add_tlv(TLV_TYPE_RAILGUN_MEM_DATA, data)
  request.add_tlv(TLV_TYPE_RAILGUN_MEM_LENGTH, length)

  response = client.send_request(request)
  return response.result == 0
end

#multi(functions) ⇒ Object

The multi-call shorthand ([“kernel32”, “ExitProcess”, [0]])



319
320
321
322
323
324
325
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 319

def multi(functions)
  if @multicaller.nil?
    @multicaller = MultiCaller.new(client, self, constant_manager)
  end

  return @multicaller.call(functions)
end

#utilObject

Return this Railgun’s Util instance.



135
136
137
138
139
140
141
# File 'lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb', line 135

def util
  if @util.nil?
    @util = Util.new(self, client.native_arch)
  end

  return @util
end