pwn 0.5.164 → 0.5.165

data/bin/pwn_nessus_cloud_vulnscan

@@ -8,7 +8,7 @@ require 'yaml'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-cYPATH', '--yaml-config=YPATH', '<Required - YAML Config Containing Access & Secret Keys for Authentication>') do |c|
@@ -29,7 +29,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 

data/bin/pwn_nexpose

@@ -7,7 +7,7 @@ require 'optparse'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   # Nexpose Flags
@@ -19,7 +19,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 

data/bin/pwn_nmap_discover_tcp_udp

@@ -9,7 +9,7 @@ require 'time'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-fFILE', '--target-file=FILE', '<Required if --target-range excluded - File containing one supported nmap target / line e.g. foo.bar, 10.1.1.1, 192.168.1.1-20, 192.168.1.0/24, etc>') do |f|
@@ -42,7 +42,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 

data/bin/pwn_openvas_vulnscan

@@ -8,7 +8,7 @@ require 'yaml'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-cYPATH', '--yaml-config=YPATH', '<Required - YAML Config Containing Username & Password for Authentication>') do |c|
@@ -29,7 +29,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 

data/bin/pwn_owasp_zap_active_scan

@@ -7,7 +7,7 @@ require 'optparse'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-aAPIKEY', '--api_key=APIKEY', '<Required - OWASP Zap API Key (Tools>Options>API)>') do |a|
@@ -44,7 +44,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 

data/bin/pwn_pastebin_sample_filter

@@ -7,7 +7,7 @@ require 'optparse'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-rPATTERN', '--regex=PATTERN', "<Required - Regex Pattern for Interesting Pastes ('.*' for All)>") do |r|
@@ -20,7 +20,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 

data/bin/pwn_phone

@@ -11,7 +11,7 @@ require 'waveform'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-tPATH', '--target-file=PATH', '<Required - File Containing List of Targets to Dial>') do |t|
@@ -64,7 +64,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 
@@ -105,7 +105,7 @@ begin
   )
 
   # Generate HTML Report
-  print "#{$PROGRAM_NAME} Generating Report..."
+  print "#{File.basename($PROGRAM_NAME)} Generating Report..."
   PWN::Reports::Phone.generate(
     dir_path: session_root,
     results_hash: results_hash

data/bin/pwn_sast

@@ -8,7 +8,7 @@ require 'htmlentities'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-uGITURI', '--uri-source-root=GITURI', '<Required - HTTP URI of Git Repo Scanned e.g. https://github.com/0dayInc/pwn/tree/master>') do |u|
@@ -41,7 +41,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 
@@ -155,7 +155,7 @@ begin
   end
 
   # Generate HTML Report
-  print "#{$PROGRAM_NAME} Generating Report..."
+  print "#{File.basename($PROGRAM_NAME)} Generating Report..."
   PWN::Reports::SAST.generate(
     dir_path: dir_path,
     results_hash: results_hash

data/bin/pwn_serial_check_voicemail

@@ -7,7 +7,7 @@ require 'optparse'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-vNUM', '--voicemail_num=num', '<Required - Number to Voicemail Service>') do |num|
@@ -19,7 +19,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 

data/bin/pwn_serial_msr206

@@ -8,7 +8,7 @@ require 'json'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-dDEV', '--block-dev=DEV', '<Optional - MSR206 block device path (defaults to /dev/ttyUSB0)>') do |d|
@@ -37,7 +37,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 

data/bin/pwn_serial_son_micro_sm132_rfid

@@ -8,7 +8,7 @@ require 'json'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-dDEV', '--block-dev=DEV', '<Optional - SonMicroRFID block device path (defaults to /dev/ttyUSB0)>') do |d|
@@ -37,7 +37,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 

data/bin/pwn_shodan_graphql_introspection

@@ -14,7 +14,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 

data/bin/pwn_shodan_search

@@ -9,7 +9,7 @@ require 'json'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-cYAML', '--config-yaml=YAML', '<Required - YAML config containing api_key from Shodan.io>') do |y|
@@ -30,7 +30,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 

data/bin/pwn_simple_http_server

@@ -6,7 +6,7 @@ require 'optparse'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
   options.on('-iIP', '--bind-ip=IP', '<Optional - HTTP Bind IP to Listen> (Defaults to Localhost)') do |i|
     opts[:bind_ip] = i

data/bin/pwn_web_cache_deception

@@ -7,7 +7,7 @@ require 'optparse'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-tURL', '--target-url=URL', '<Required - URL to Target>') do |t|
@@ -44,7 +44,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 
@@ -142,12 +142,12 @@ begin
       pwn_www_mod = PWN::WWW.const_get(pwn_www_mod_str)
       unless pwn_www_mod.respond_to?('open') && pwn_www_mod.respond_to?('login') && pwn_www_mod.respond_to?('logout')
         puts "#{@dark_red}Module PWN::WWW::#{pwn_www_mod_str} Missing #open #login and/or #logout Method(s)#{@end_of_color}"
-        puts `#{$PROGRAM_NAME} -l`
+        puts `#{File.basename($PROGRAM_NAME)} -l`
         exit
       end
     rescue NAME_ERROR => e
       puts "#{@dark_red}Invalid module PWN::WWW::#{pwn_www_mod_str}#{@end_of_color}"
-      puts `#{$PROGRAM_NAME} -l`
+      puts `#{File.basename($PROGRAM_NAME)} -l`
       exit
     end
 

data/bin/pwn_www_checkip

@@ -8,7 +8,7 @@ require 'optparse'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
   options.on('-tIP_HOST', '--target=IP_HOST', '<Optional - IP or Host to Check (Default - Your Public IP)>') do |t|
     opts[:target] = t
@@ -64,6 +64,8 @@ begin
     )
     puts JSON.pretty_generate(ip_info_obj)
   end
+rescue Interrupt
+  puts "\n#{File.basename($PROGRAM_NAME)} => Goodbye."
 rescue StandardError => e
   raise e
 ensure

data/bin/pwn_www_uri_buster

@@ -1,6 +1,7 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: false
 
+require 'time'
 require 'pwn'
 require 'optparse'
 require 'uri'
@@ -12,7 +13,7 @@ require 'json'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-uURL', '--target-url=URL', '<Required - Target URL)>') do |t|
@@ -27,6 +28,14 @@ OptionParser.new do |options|
     opts[:append_pattern] = a
   end
 
+  options.on('-TFLOAT', '--max-timeout=FLOAT', '<Optional - HTTP Request Timeout (Default 6.0 Seconds)>') do |t|
+    opts[:max_timeout] = t
+  end
+
+  options.on('-TFLOAT', '--max-retries=FLOAT', '<Optional - HTTP Request Retries (Default 3)>') do |r|
+    opts[:max_retries] = r
+  end
+
   options.on('-pPROXY', '--proxy=PROXY', '<Optional - Proxy SCHEME://ADDRESS:PORT || tor>') do |p|
     opts[:proxy] = p
   end
@@ -61,7 +70,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 
@@ -71,43 +80,60 @@ def request_path(opts = {})
   http_request_headers = opts[:http_request_headers]
   wordlist_line = opts[:wordlist_line]
   http_method = opts[:http_method]
+  max_timeout = opts[:max_timeout]
+  timeout = opts[:max_timeout] ||= max_timeout / 21.0
+  max_retries = opts[:max_retries]
 
   rest_client_resp_hash = {}
-  begin
-    print '.'
-    http_uri = "#{target_url}/#{wordlist_line}"
-    if proxy
-      browser_obj = PWN::Plugins::TransparentBrowser.open(
-        browser_type: :rest,
-        proxy: proxy
-      )
-    else
-      browser_obj = PWN::Plugins::TransparentBrowser.open(
-        browser_type: :rest
-      )
-    end
-    rest_client = browser_obj[:browser]::Request
 
-    headers = {
-      user_agent: Faker::Internet.user_agent
-    }
+  # request_count && timeout values to begin with
+  request_count = 1
 
-    if http_request_headers
-      headers = JSON.parse(
-        http_request_headers,
-        symbolize_names: true
-      )
-    end
+  if proxy
+    browser_obj = PWN::Plugins::TransparentBrowser.open(
+      browser_type: :rest,
+      proxy: proxy
+    )
+  else
+    browser_obj = PWN::Plugins::TransparentBrowser.open(
+      browser_type: :rest
+    )
+  end
+  rest_client = browser_obj[:browser]::Request
+
+  headers = {
+    user_agent: Faker::Internet.user_agent
+  }
+
+  http_uri = "#{target_url}/#{wordlist_line}"
+
+  if http_request_headers
+    headers = JSON.parse(
+      http_request_headers,
+      symbolize_names: true
+    )
+  end
+
+  timestamp_fmt = '%Y-%m-%d %H:%M:%S.%L'
 
+  begin
+    request_timestamp = Time.now.strftime(timestamp_fmt)
     response = rest_client.execute(
       method: http_method,
       url: http_uri,
       headers: headers,
-      verify_ssl: false
+      verify_ssl: false,
+      timeout: timeout
     )
+    response_timestamp = Time.now.strftime(timestamp_fmt)
+
+    # duration in milliseconds
+    duration = Time.parse(response_timestamp) - Time.parse(request_timestamp)
 
     rest_client_resp_hash = {
-      request_timestamp: Time.now.strftime('%Y-%m-%d_%H-%M-%S'),
+      request_timestamp: request_timestamp,
+      response_timestamp: response_timestamp,
+      duration: duration,
       http_uri: http_uri,
       http_method: http_method,
       http_resp_code: response.code,
@@ -115,9 +141,8 @@ def request_path(opts = {})
       http_resp_headers: JSON.pretty_generate(response.headers),
       http_resp: "#{response.body[0..300]}..."
     }
-  rescue Errno::ECONNREFUSED
-    raise 'ERROR: Connection(s) Refused. Try lowering the --max-threads value.'
-  rescue Errno::ECONNRESET,
+  rescue Errno::ECONNREFUSED,
+         Errno::ECONNRESET,
          NoMethodError,
          OpenSSL::SSL::SSLError,
          RestClient::Exceptions::ReadTimeout,
@@ -125,9 +150,16 @@ def request_path(opts = {})
          RestClient::ServerBrokeConnection,
          SOCKSError => e
 
+    timeout += 0.1
+    retry if timeout < max_timeout
+
     # May be best to switch Tor channel if SOCKSError is rescued
+    response_timestamp = Time.now.strftime(timestamp_fmt)
+    duration = Time.parse(response_timestamp) - Time.parse(request_timestamp)
     rest_client_resp_hash = {
-      request_timestamp: Time.now.strftime('%Y-%m-%d_%H-%M-%S'),
+      request_timestamp: request_timestamp,
+      response_timestamp: response_timestamp,
+      duration: duration,
       http_uri: http_uri,
       http_method: http_method,
       http_resp_code: e.class,
@@ -136,9 +168,14 @@ def request_path(opts = {})
       http_resp: "ERROR: #{e.message}"
     }
   rescue RestClient::ExceptionWithResponse => e
+    response_timestamp = Time.now.strftime(timestamp_fmt)
+    duration = Time.parse(response_timestamp) - Time.parse(request_timestamp)
+
     if e.respond_to?(:response)
       rest_client_resp_hash = {
-        request_timestamp: Time.now.strftime('%Y-%m-%d_%H-%M-%S'),
+        request_timestamp: request_timestamp,
+        response_timestamp: response_timestamp,
+        duration: duration,
         http_uri: http_uri,
         http_method: http_method,
         http_resp_code: e.response.code,
@@ -148,7 +185,9 @@ def request_path(opts = {})
       }
     else
       resp_client_resp_hash = {
-        request_timestamp: Time.now.strftime('%Y-%m-%d_%H-%M-%S'),
+        request_timestamp: request_timestamp,
+        response_timestamp: response_timestamp,
+        duration: duration,
         http_uri: http_uri,
         http_method: http_method,
         http_resp_code: 'N/A',
@@ -163,17 +202,20 @@ def request_path(opts = {})
       url_encoded_wordlist_arr.push(CGI.escape(path))
     end
     wordlist_line = url_encoded_wordlist_arr.join('/')
-
-    retry
   rescue RestClient::TooManyRequests
+    request_count += 1
     sleep 60
-  ensure
-    browser_obj = PWN::Plugins::TransparentBrowser.close(
-      browser_obj: browser_obj
-    )
+
+    retry if request_count < max_retries
+  rescue SystemExit, Interrupt
+    puts "\n#{File.basename($PROGRAM_NAME)}.#{__method__} Goodbye."
   end
 
   rest_client_resp_hash
+ensure
+  browser_obj = PWN::Plugins::TransparentBrowser.close(
+    browser_obj: browser_obj
+  )
 end
 
 begin
@@ -189,18 +231,31 @@ begin
   raise "ERROR: #{wordlist} Does Not Exist." unless File.exist?(wordlist)
 
   append_pattern = opts[:append_pattern]
-  proxy = opts[:proxy]
+  max_timeout = opts[:max_timeout] ||= 6
+  if max_timeout
+    max_timeout = max_timeout.to_f
+    raise 'ERROR: --max-timeout must be a positive float.' unless max_timeout.positive?
+  end
 
-  max_threads = opts[:max_threads]
-  max_threads ||= 25
+  max_retries = opts[:max_retries] ||= 3
+  if max_retries
+    max_retries = max_retries.to_i
+    raise 'ERROR: --max-retries must be a positive integer.' unless max_retries.positive?
+  end
 
-  http_request_headers = opts[:http_request_headers]
+  proxy = opts[:proxy]
+  max_threads = opts[:max_threads] ||= 25
+  if max_threads
+    max_threads = max_threads.to_i
+    raise 'ERROR: --max-threads must be a positive integer.' unless max_threads.positive?
+  end
 
+  http_request_headers = opts[:http_request_headers]
   include_http_response_codes = opts[:include_http_response_codes]
-  include_http_response_codes = include_http_response_codes.delete("\s").split(',') if include_http_response_codes
+  include_http_response_codes = include_http_response_codes.to_s.delete("\s").split(',') if include_http_response_codes
 
   exclude_http_response_codes = opts[:exclude_http_response_codes]
-  exclude_http_response_codes = exclude_http_response_codes.delete("\s").split(',') if exclude_http_response_codes
+  exclude_http_response_codes = exclude_http_response_codes.to_s.delete("\s").split(',') if exclude_http_response_codes
 
   raise 'ERROR: Flags --include-response-codes and --exclude-response-codes cannot be used together.' if include_http_response_codes && exclude_http_response_codes
 
@@ -230,33 +285,60 @@ begin
 
     next if wordlist_line.match?(/^#/)
 
-    http_methods = %i[DELETE GET HEAD OPTIONS PATCH POST PUT TRACE].shuffle
+    http_methods = %i[
+      DELETE
+      GET
+      HEAD
+      OPTIONS
+      PATCH
+      POST
+      PUT
+      TRACE
+    ].shuffle
+
+    timeout = nil
     http_methods.each do |http_method|
       # TODO: Implement HTTP response timeout
+      prev_rest_client_resp_hash = results_hash[:data].last
       rest_client_resp_hash = request_path(
         target_url: target_url,
         proxy: proxy,
         http_request_headers: http_request_headers,
         wordlist_line: "#{wordlist_line}#{append_pattern}",
-        http_method: http_method
+        http_method: http_method,
+        max_timeout: max_timeout,
+        timeout: timeout,
+        max_retries: max_retries
       )
 
       mutex.synchronize do
-        if include_http_response_codes
-          ret_http_resp_code = rest_client_resp_hash[:http_resp_code].to_s
-          results_hash[:data].push(rest_client_resp_hash) if include_http_response_codes.include?(ret_http_resp_code)
-        elsif exclude_http_response_codes
-          ret_http_resp_code = rest_client_resp_hash[:http_resp_code].to_s
-          results_hash[:data].push(rest_client_resp_hash) unless exclude_http_response_codes.include?(ret_http_resp_code)
+        ret_http_resp_code = rest_client_resp_hash[:http_resp_code].to_s
+        # Better waY to implement this?
+        if include_http_response_codes.is_a?(Array)
+          # If include_http_response_codes is an array, only include
+          # the response into the results_hash if the response code
+          # is in the include_http_response_codes array
+          results_hash[:data].push(rest_client_resp_hash) if include_http_response_codes.include?(ret_http_resp_code) &&
+                                                             prev_rest_client_resp_hash != rest_client_resp_hash &&
+                                                             rest_client_resp_hash.any?
         else
-          results_hash[:data].push(rest_client_resp_hash)
+          # If exclude_http_response_codes is an array, only include
+          # the response into the results_hash if the response code
+          # is not in the exclude_http_response_codes array
+          results_hash[:data].push(rest_client_resp_hash) unless exclude_http_response_codes.is_a?(Array) &&
+                                                                 exclude_http_response_codes.include?(ret_http_resp_code) &&
+                                                                 prev_rest_client_resp_hash != rest_client_resp_hash &&
+                                                                 rest_client_resp_hash.any?
         end
+        # Finally puts the last line of the results_hash to STDOUT
+        puts results_hash[:data].last if results_hash[:data].last.any?
+        timeout = results_hash[:data].last[:duration].to_f if results_hash[:data].last.any?
       end
     end
   end
 
   # Generate HTML Report
-  print "#{$PROGRAM_NAME} Generating Report..."
+  print "#{File.basename($PROGRAM_NAME)} Generating Report..."
   PWN::Reports::URIBuster.generate(
     dir_path: dir_path,
     results_hash: results_hash
@@ -283,8 +365,10 @@ begin
       listen_port
     )
   end
-rescue SystemExit, Interrupt
-  puts "\nGoodbye."
+# rescue SystemExit, Interrupt
+#   puts "\nGoodbye."
+rescue Interrupt
+  puts "\n#{File.basename($PROGRAM_NAME)} Goodbye."
 rescue StandardError => e
   puts e.backtrace
   raise e

data/bin/pwn_xss_dom_vectors

@@ -7,7 +7,7 @@ require 'optparse'
 opts = {}
 OptionParser.new do |options|
   options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
+    #{File.basename($PROGRAM_NAME)} [opts]
   "
 
   options.on('-tFQDN', '--target-fqdn=FQDN', '<Required - FQDN to Target>') do |t|
@@ -36,7 +36,7 @@ OptionParser.new do |options|
 end.parse!
 
 if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
+  puts `#{File.basename($PROGRAM_NAME)} --help`
   exit 1
 end
 

data/lib/pwn/plugins/authentication_helper.rb

@@ -30,7 +30,7 @@ module PWN
         pass = TTY::Prompt.new.mask("#{prompt}: ")
         pass.to_s.strip.chomp.scrub
       rescue Interrupt
-        puts 'CTRL+C detected...goodbye.'
+        puts "#{self}.#{__method__} => Goodbye."
       rescue StandardError => e
         raise e
       end
@@ -46,7 +46,7 @@ module PWN
         mfa = TTY::Prompt.new.ask("#{prompt}: ")
         mfa.to_s.strip.chomp.scrub
       rescue Interrupt
-        puts 'CTRL+C detected...goodbye.'
+        puts "#{self}.#{__method__} => Goodbye."
       rescue StandardError => e
         raise e
       end