Brent Cook
154 lines
2.8 KiB
Ruby
154 lines
2.8 KiB
Ruby
# -*- coding: binary -*-
|
|
|
|
require 'rex/post/file_stat'
|
|
require 'rex/post/meterpreter/extensions/stdapi/stdapi'
|
|
|
|
module Rex
|
|
module Post
|
|
module Meterpreter
|
|
module Extensions
|
|
module Stdapi
|
|
module Fs
|
|
|
|
###
|
|
#
|
|
# This class wrappers gathering information about a given file and implements
|
|
# the Rex::Post::FileStat interface in terms of data acquisition.
|
|
#
|
|
###
|
|
class FileStat < Rex::Post::FileStat
|
|
|
|
class << self
|
|
attr_accessor :client
|
|
end
|
|
|
|
@@struct_stat = [
|
|
'st_dev', 4, # 0
|
|
'st_mode', 4, # 4
|
|
'st_nlink', 4, # 8
|
|
'st_uid', 4, # 12
|
|
'st_gid', 4, # 16
|
|
'st_rdev', 4, # 20
|
|
'st_ino', 8, # 24
|
|
'st_size', 8, # 32
|
|
'st_atime', 8, # 40
|
|
'st_mtime', 8, # 48
|
|
'st_ctime', 8, # 56
|
|
]
|
|
|
|
@@struct_stat32 = [
|
|
'st_dev', 4, # 0
|
|
'st_ino', 2, # 4
|
|
'st_mode', 2, # 6
|
|
'st_nlink', 2, # 8
|
|
'st_uid', 2, # 10
|
|
'st_gid', 2, # 12
|
|
'pad1', 2, # 14
|
|
'st_rdev', 4, # 16
|
|
'st_size', 4, # 20
|
|
'st_atime', 8, # 24
|
|
'st_mtime', 8, # 32
|
|
'st_ctime', 8, # 40
|
|
]
|
|
|
|
##
|
|
#
|
|
# Constructor
|
|
#
|
|
##
|
|
|
|
#
|
|
# Returns an instance of a FileStat object.
|
|
#
|
|
def initialize(file)
|
|
self.stathash = stat(file) if (file)
|
|
end
|
|
|
|
#
|
|
# Swaps in a new stat hash.
|
|
#
|
|
def update(stat_buf)
|
|
elem = @@struct_stat
|
|
hash = {}
|
|
offset = 0
|
|
index = 0
|
|
|
|
while (index < elem.length)
|
|
size = elem[index + 1]
|
|
format = 'V'
|
|
case size
|
|
when 2
|
|
format = 'v'
|
|
when 8
|
|
format = 'Q'
|
|
end
|
|
|
|
value = stat_buf[offset, size].unpack(format)[0]
|
|
offset += size
|
|
|
|
hash[elem[index]] = value
|
|
|
|
index += 2
|
|
end
|
|
|
|
return (self.stathash = hash)
|
|
end
|
|
|
|
#
|
|
# Swaps in a new old style stat hash.
|
|
#
|
|
def update32(stat_buf)
|
|
elem = @@struct_stat32
|
|
hash = {}
|
|
offset = 0
|
|
index = 0
|
|
|
|
while (index < elem.length)
|
|
size = elem[index + 1]
|
|
|
|
value = stat_buf[offset, size].unpack(size == 2 ? 'v' : 'V')[0]
|
|
offset += size
|
|
|
|
hash[elem[index]] = value
|
|
|
|
index += 2
|
|
end
|
|
|
|
return (self.stathash = hash)
|
|
end
|
|
|
|
protected
|
|
|
|
##
|
|
#
|
|
# Initializer
|
|
#
|
|
##
|
|
|
|
#
|
|
# Gets information about the supplied file and returns a populated
|
|
# hash to the requestor.
|
|
#
|
|
def stat(file)
|
|
request = Packet.create_request('stdapi_fs_stat')
|
|
|
|
request.add_tlv(TLV_TYPE_FILE_PATH, self.class.client.unicode_filter_decode( file ))
|
|
|
|
response = self.class.client.send_request(request)
|
|
stat_buf = response.get_tlv(TLV_TYPE_STAT_BUF)
|
|
|
|
unless stat_buf
|
|
stat_buf = response.get_tlv(TLV_TYPE_STAT_BUF32)
|
|
return update32(stat_buf.value)
|
|
end
|
|
|
|
# Next, we go through the returned stat_buf and fix up the values
|
|
# and insert them into a hash
|
|
return update(stat_buf.value)
|
|
end
|
|
|
|
end
|
|
|
|
end; end; end; end; end; end
|
|
|