Land #8601, make session logging more useful, don't lose characters

rapid7 · Jun 23, 2017 · c3090a4 · c3090a4
2 parents fdd62ab + 03b3675
commit c3090a4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/lib/msf/core/session.rb b/lib/msf/core/session.rb
@@ -214,8 +214,9 @@ def log_file_name
 
     dstr  = sprintf("%.4d%.2d%.2d", dt.year, dt.mon, dt.mday)
     rhost = session_host.gsub(':', '_')
+    sname = name.to_s.gsub(/\W+/,'_')
 
-    "#{dstr}_#{rhost}_#{type}"
+    "#{dstr}_#{sname}_#{rhost}_#{type}"
   end
 
   #

diff --git a/lib/rex/logging/sinks/timestamp_flatfile.rb b/lib/rex/logging/sinks/timestamp_flatfile.rb
@@ -13,7 +13,7 @@ class TimestampFlatfile < Flatfile
 
   def log(sev, src, level, msg, from) # :nodoc:
     return unless msg.present?
-    msg = msg.chop.gsub(/\x1b\[[0-9;]*[mG]/,'').gsub(/[\x01-\x02]/, " ")
+    msg = msg.gsub(/\x1b\[[0-9;]*[mG]/,'').gsub(/[\x01-\x02]/, ' ').gsub(/\s+$/,'')
     fd.write("[#{get_current_timestamp}] #{msg}\n")
     fd.flush
   end