Add support for shell sessions to post/windows/gather/enum_hostfile

modules/post/windows/gather/enum_hostfile.rb

@@ -4,6 +4,7 @@
 ##
 
 class MetasploitModule < Msf::Post
+  include Msf::Post::File
 
   def initialize(info={})
     super(update_info(info,
@@ -20,35 +21,23 @@ def initialize(info={})
 
   def run
     # read in the hosts in the hosts file.
-    fd = session.fs.file.new("C:\\WINDOWS\\System32\\drivers\\etc\\hosts", "rb")
-
-    # Load up the original hosts file
-    buf = ''
-    until fd.eof?
-      buf << fd.read
-    end
-
-    # Finished loading the hosts file, close fd
-    fd.close
+    hosts = read_file "C:\\WINDOWS\\System32\\drivers\\etc\\hosts"
 
     # Store the original hosts file
     p = store_loot(
       'hosts.confige',
       'text/plain',
       session,
-      buf,
+      hosts,
       'hosts_file.txt',
       'Windows Hosts File'
     )
 
-    # Split lines
-    lines = buf.split("\n")
-
     # Print out each line that doesn't start w/ a comment
     entries = []
-    lines.each do |line|
+    hosts.each_line do |line|
       next if line =~ /^[\r|\n|#]/
-      entries << line
+      entries << line.strip
     end
 
     # Show results