add a test to check if the file exist

2015-01-31 06:00:02 +01:00 · 2015-01-31 06:00:02 +01:00 · 03fcfc496a
parent fbb85c0391
commit 03fcfc496a
1 changed files with 8 additions and 0 deletions
--- a/modules/post/windows/gather/file_in_raw_ntfs.rb
+++ b/modules/post/windows/gather/file_in_raw_ntfs.rb
@ -43,6 +43,14 @@ class Metasploit3 < Msf::Post
    end

    file_path = datastore['FILE_PATH']
+
+    r = client.railgun.kernel32.GetFileAttributesA(file_path)
+
+    if r['GetLastError'] != 0
+      print_error("The file does not exist, use file format C:\\\\Windows\\\\System32\\\\drivers\\\\etc\\\\hosts")
+      return nil
+    end
+
    drive = file_path[0, 2]

    r = client.railgun.kernel32.CreateFileA("\\\\.\\#{drive}", "GENERIC_READ", "FILE_SHARE_DELETE|FILE_SHARE_READ|FILE_SHARE_WRITE",