Memory Usage output

lib/common/common_helper.rb

@@ -160,3 +160,8 @@ def remove_base64_images_from_html(html)
   imageregex = %r{data\s*:\s*image/[^\s;]+\s*;\s*base64\s*,\s*}
   html.gsub(/["']\s*#{imageregex}#{base64regex}\s*["']/, '""')
 end
+
+# @return [ Integer ] The memory of the current process in Bytes
+def get_memory_usage
+  `ps -o rss= -p #{Process.pid}`.to_i * 1024 # ps returns the value in KB
+end

lib/common/hacks.rb

@@ -97,3 +97,12 @@ module Terminal
     end
   end
 end
+
+class Numeric
+  def bytes_to_human
+    units = %w{B KB MB GB TB}
+    e = (Math.log(self)/Math.log(1024)).floor
+    s = "%.3f" % (to_f / 1024**e)
+    s.sub(/\.?0*$/, ' ' + units[e])
+  end
+end

wpscan.rb

@@ -103,7 +103,8 @@ def main
     end
 
     # Output runtime data
-    start_time = Time.now
+    start_time   = Time.now
+    start_memory = get_memory_usage
     puts "| URL: #{wp_target.url}"
     puts "| Started on #{start_time.asctime}"
     puts
@@ -332,10 +333,13 @@ def main
       end
     end
 
-    stop_time = Time.now
+    stop_time   = Time.now
+    elapsed     = stop_time - start_time
+    used_memory = get_memory_usage - start_memory
+    
     puts
     puts green("[+] Finished at #{stop_time.asctime}")
-    elapsed = stop_time - start_time
+    puts green("[+] Memory Used: #{used_memory.bytes_to_human}")
     puts green("[+] Elapsed time: #{Time.at(elapsed).utc.strftime('%H:%M:%S')}")
     exit(0) # must exit!