Fixes #153 Disable error trace when it's from the main script
This commit is contained in:
@@ -15,14 +15,13 @@ def main
|
|||||||
|
|
||||||
unless wpscan_options.has_options?
|
unless wpscan_options.has_options?
|
||||||
usage()
|
usage()
|
||||||
puts red('No argument supplied')
|
raise('No argument supplied')
|
||||||
exit(0)
|
|
||||||
end
|
end
|
||||||
|
|
||||||
if wpscan_options.help
|
if wpscan_options.help
|
||||||
help()
|
help()
|
||||||
usage()
|
usage()
|
||||||
exit
|
exit(1)
|
||||||
end
|
end
|
||||||
|
|
||||||
# Check for updates
|
# Check for updates
|
||||||
@@ -321,10 +320,14 @@ def main
|
|||||||
puts green("[+] Elapsed time: #{Time.at(elapsed).utc.strftime('%H:%M:%S')}")
|
puts green("[+] Elapsed time: #{Time.at(elapsed).utc.strftime('%H:%M:%S')}")
|
||||||
exit() # must exit!
|
exit() # must exit!
|
||||||
rescue => e
|
rescue => e
|
||||||
|
if e.backtrace[0] =~ /main/
|
||||||
|
puts red(e.message)
|
||||||
|
else
|
||||||
puts red("[ERROR] #{e.message}")
|
puts red("[ERROR] #{e.message}")
|
||||||
puts red('Trace :')
|
puts red('Trace :')
|
||||||
puts red(e.backtrace.join("\n"))
|
puts red(e.backtrace.join("\n"))
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
end
|
||||||
|
|
||||||
main()
|
main()
|
||||||
|
|||||||
Reference in New Issue
Block a user