60 lines
1.2 KiB
Ruby
60 lines
1.2 KiB
Ruby
#!/usr/bin/env ruby
|
|
# -*- coding: binary -*-
|
|
|
|
require 'rex/image_source/image_source'
|
|
require 'rex/struct2'
|
|
|
|
module Rex
|
|
module ImageSource
|
|
class Disk < ImageSource
|
|
|
|
attr_accessor :file, :file_offset, :size
|
|
|
|
WINDOW_SIZE = 4096
|
|
WINDOW_OVERLAP = 64
|
|
|
|
def initialize(_file, _offset = 0, _len = nil)
|
|
_len = _file.stat.size if !_len
|
|
|
|
self.file = _file
|
|
self.file_offset = _offset
|
|
self.size = _len
|
|
end
|
|
|
|
def read(offset, len)
|
|
if offset < 0 || offset+len > size
|
|
raise RangeError, "Offset #{offset} outside of image source", caller
|
|
end
|
|
|
|
file.seek(file_offset + offset)
|
|
file.read(len)
|
|
end
|
|
|
|
def index(search, offset = 0)
|
|
# do a sliding window search across the disk
|
|
while offset < size
|
|
|
|
# get a full window size if we can, we
|
|
# don't want to read past our boundaries
|
|
wsize = size - offset
|
|
wsize = WINDOW_SIZE if wsize > WINDOW_SIZE
|
|
|
|
window = self.read(offset, wsize)
|
|
res = window.index(search)
|
|
return res + offset if res
|
|
offset += WINDOW_SIZE - WINDOW_OVERLAP
|
|
end
|
|
end
|
|
|
|
def subsource(offset, len)
|
|
self.class.new(file, file_offset+offset, len)
|
|
end
|
|
|
|
def close
|
|
file.close
|
|
end
|
|
end
|
|
|
|
end
|
|
end
|