pwn 0.5.432 → 0.5.434

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3d81ea28a631e0ee9113f3ddfeef28e8615744988fc4bbfff2a5a615a8929aac
-  data.tar.gz: f367343802e440d04dba25f75d56c8f71e8ff208bf4f776d70b600f31ad12b26
+  metadata.gz: 3cdf2e676f8720fc916611068b8c6011fec4a8f655848ba3fe7eaba5370e4f75
+  data.tar.gz: 6f1505c1d3eb735da84c2033577638bc74aafd122d8909e817dd50313f83ebb2
 SHA512:
-  metadata.gz: daa53bfee771930fc864e0971957515769e14ba7a2911658fa37b89826a491debb2b49b077f2966415b1a4e44f8dd8cb832bd71718d0defb77698cb43b528a24
-  data.tar.gz: ecd91103094b195aba6e12b94e8da3fa9b40422bbcc15089b6d0b193f77272ddcdc3e7fb43de2fe3fc5231be360f66781cfad24050b652007e966cc2bc189be9
+  metadata.gz: db901c38eddb0daa14ba19a0a370dcd174a809112bf24d76396af1cfedccb36886961176b813f32b2dcf4d839b283828aa7e7ddc91423d47e456d059808b85a4
+  data.tar.gz: 3e97e046c79229ff21390c782a2a3f9387c91d29682b047d8aeb0d354824b7e6c9662bc1481c80f6eb0e58b6fed5af274d7ffdce1f1651d74e09483abee8c408

data/Gemfile

@@ -75,7 +75,7 @@ gem 'pry-doc', '1.6.0'
 gem 'rake', '13.3.0'
 gem 'rb-readline', '0.5.5'
 gem 'rbvmomi2', '3.8.0'
-gem 'rdoc', '6.14.2'
+gem 'rdoc', '6.15.0'
 gem 'rest-client', '2.1.0'
 gem 'rex', '2.0.13'
 gem 'rmagick', '6.1.4'

data/README.md

@@ -37,7 +37,7 @@ $ cd /opt/pwn
 $ ./install.sh
 $ ./install.sh ruby-gem
 $ pwn
-pwn[v0.5.432]:001 >>> PWN.help
+pwn[v0.5.434]:001 >>> PWN.help
 ```
 
 [![Installing the pwn Security Automation Framework](https://raw.githubusercontent.com/0dayInc/pwn/master/documentation/pwn_install.png)](https://youtu.be/G7iLUY4FzsI)
@@ -52,7 +52,7 @@ $ rvm use ruby-3.4.4@pwn
 $ gem uninstall --all --executables pwn
 $ gem install --verbose pwn
 $ pwn
-pwn[v0.5.432]:001 >>> PWN.help
+pwn[v0.5.434]:001 >>> PWN.help
 ```
 
 If you're using a multi-user install of RVM do:
@@ -62,7 +62,7 @@ $ rvm use ruby-3.4.4@pwn
 $ rvmsudo gem uninstall --all --executables pwn
 $ rvmsudo gem install --verbose pwn
 $ pwn
-pwn[v0.5.432]:001 >>> PWN.help
+pwn[v0.5.434]:001 >>> PWN.help
 ```
 
 PWN periodically upgrades to the latest version of Ruby which is reflected in `/opt/pwn/.ruby-version`.  The easiest way to upgrade to the latest version of Ruby from a previous PWN installation is to run the following script:

data/bin/pwn

@@ -10,12 +10,12 @@ OptionParser.new do |options|
     #{File.basename($PROGRAM_NAME)} [opts]
   "
 
-  options.on('-YPATH', '--pwn-config=PATH', '<Optional - PWN YAML File>') do |p|
-    opts[:pwn_config_path] = p
+  options.on('-YPATH', '--pwn-env=PATH', '<Optional - PWN YAML File>') do |p|
+    opts[:pwn_env_path] = p
   end
 
-  options.on('-ZPATH', '--pwn-decryptor=PATH', '<Optional - Out-of-Band YAML File with :key && :iv>') do |d|
-    opts[:pwn_decryptor_path] = d
+  options.on('-ZPATH', '--pwn-dec=PATH', '<Optional - Out-of-Band YAML File with :key && :iv>') do |d|
+    opts[:pwn_dec_path] = d
   end
 end.parse!
 

data/bin/pwn_sast

@@ -12,12 +12,12 @@ OptionParser.new do |options|
   "
 
   # TODO: HOW TO LOAD THE pwn.yaml CONFIGURATION FILE FOR EVERYTHING UNLESS OVERRIDDEN???
-  options.on('-YPATH', '--pwn-config=PATH', '<Optional - PWN YAML File>') do |p|
-    opts[:pwn_config_path] = p
+  options.on('-YPATH', '--pwn-env=PATH', '<Optional - PWN YAML File>') do |p|
+    opts[:pwn_env_path] = p
   end
 
-  options.on('-ZPATH', '--pwn-decryptor=PATH', '<Optional - Out-of-Band YAML File with :key && :iv>') do |d|
-    opts[:pwn_decryptor_path] = d
+  options.on('-ZPATH', '--pwn-dec=PATH', '<Optional - Out-of-Band YAML File with :key && :iv>') do |d|
+    opts[:pwn_dec_path] = d
   end
 
   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|
@@ -57,12 +57,12 @@ end
 begin
   timestamp = Time.now.strftime('%Y-%m-%d_%H:%M:%S%z')
 
-  pwn_config_path = opts[:pwn_config_path]
-  pwn_decryptor_path = opts[:pwn_decryptor_path]
-  if pwn_config_path
+  pwn_env_path = opts[:pwn_env_path]
+  pwn_dec_path = opts[:pwn_dec_path]
+  if pwn_env_path
     PWN::Config.refresh(
-      pwn_config_path: pwn_config_path,
-      pwn_decryptor_path: pwn_decryptor_path
+      pwn_env_path: pwn_env_path,
+      pwn_dec_path: pwn_dec_path
     )
   end
 

data/lib/pwn/ai/grok.rb

@@ -25,13 +25,6 @@ module PWN
 
       private_class_method def self.grok_rest_call(opts = {})
         token = opts[:token]
-        if token.nil?
-          PWN::Plugins::AuthenticationHelper.mask_password(
-            prompt: 'Grok Key'
-          )
-          PWN::CONFIG[:ai][:grok][:key] = token
-        end
-
         http_method = if opts[:http_method].nil?
                         :get
                       else

data/lib/pwn/ai/introspection.rb

@@ -0,0 +1,80 @@
+# frozen_string_literal: true
+
+require 'json'
+require 'rest-client'
+require 'tty-spinner'
+
+module PWN
+  module AI
+    # This plugin interacts with PWN::Env's `ai` data structure
+    # when `PWN::Env[:ai][:introspection]` is set to `true`.
+    module Introspection
+      # Supported Method Parameters::
+      # response = PWN::AI::Introspection.reflect
+
+      public_class_method def self.reflect
+        engine = PWN::Env[:ai][:active].to_s.downcase.to_sym
+        base_uri = PWN::Env[:ai][engine][:base_uri]
+        model = PWN::Env[:ai][engine][:model]
+        key = PWN::Env[:ai][engine][:key]
+        system_role_content = PWN::Env[:ai][engine][:system_role_content]
+        temp = PWN::Env[:ai][engine][:temp]
+
+        case engine
+        when :grok
+          response = PWN::AI::Grok.chat(
+            base_uri: base_uri,
+            token: key,
+            model: model,
+            system_role_content: system_role_content,
+            temp: temp,
+            request: request.chomp,
+            spinner: false
+          )
+        when :ollama
+          response = PWN::AI::Ollama.chat(
+            base_uri: base_uri,
+            token: key,
+            model: model,
+            system_role_content: system_role_content,
+            temp: temp,
+            request: request.chomp,
+            spinner: false
+          )
+        when :openai
+          response = PWN::AI::OpenAI.chat(
+            base_uri: base_uri,
+            token: key,
+            model: model,
+            system_role_content: system_role_content,
+            temp: temp,
+            request: request.chomp,
+            spinner: false
+          )
+        end
+
+        response
+      rescue StandardError => e
+        raise e
+      end
+
+      # Author(s):: 0day Inc. <support@0dayinc.com>
+
+      public_class_method def self.authors
+        "AUTHOR(S):
+          0day Inc. <support@0dayinc.com>
+        "
+      end
+
+      # Display Usage for this Module
+
+      public_class_method def self.help
+        puts "USAGE:
+          #{self}.reflect
+
+          #{self}.authors
+        "
+      end
+    end
+  end
+end

data/lib/pwn/ai/ollama.rb

@@ -27,13 +27,6 @@ module PWN
       private_class_method def self.ollama_rest_call(opts = {})
         base_uri = opts[:base_uri]
         token = opts[:token]
-        if token.nil?
-          PWN::Plugins::AuthenticationHelper.mask_password(
-            prompt: 'Ollama Key'
-          )
-          PWN::CONFIG[:ai][:ollama][:key] = token
-        end
-
         http_method = if opts[:http_method].nil?
                         :get
                       else

data/lib/pwn/ai/open_ai.rb

@@ -26,13 +26,6 @@ module PWN
 
       private_class_method def self.open_ai_rest_call(opts = {})
         token = opts[:token]
-        if token.nil?
-          PWN::Plugins::AuthenticationHelper.mask_password(
-            prompt: 'OpenAI Key'
-          )
-          PWN::CONFIG[:ai][:openai][:key] = token
-        end
-
         http_method = if opts[:http_method].nil?
                         :get
                       else

data/lib/pwn/ai.rb

@@ -6,6 +6,7 @@ module PWN
   # http://www.rubyinside.com/ruby-techniques-revealed-autoload-1652.html
   module AI
     autoload :Grok, 'pwn/ai/grok'
+    autoload :Introspection, 'pwn/ai/introspection'
     autoload :Ollama, 'pwn/ai/ollama'
     autoload :OpenAI, 'pwn/ai/open_ai'
 

data/lib/pwn/config.rb

@@ -7,25 +7,25 @@ module PWN
   # Used to manage PWN configuration settings within PWN drivers.
   module Config
     # Supported Method Parameters::
-    # PWN::Config.refresh(
-    #   pwn_config_path: 'optional - Path to pwn.yaml file.  Defaults to ~/.pwn/pwn.yaml',
-    #   pwn_decryptor_path: 'optional - Path to pwn.decryptor.yaml file.  Defaults to ~/.pwn/pwn.decryptor.yaml'
+    # PWN::Config.refresh_env(
+    #   pwn_env_path: 'optional - Path to pwn.yaml file.  Defaults to ~/.pwn/pwn.yaml',
+    #   pwn_dec_path: 'optional - Path to pwn.decryptor.yaml file.  Defaults to ~/.pwn/pwn.decryptor.yaml'
     # )
 
-    public_class_method def self.refresh(opts = {})
-      pwn_config_root = "#{Dir.home}/.pwn"
-      FileUtils.mkdir_p(pwn_config_root)
+    public_class_method def self.refresh_env(opts = {})
+      pwn_env_root = "#{Dir.home}/.pwn"
+      FileUtils.mkdir_p(pwn_env_root)
 
-      pwn_config_path = opts[:pwn_config_path] ||= "#{pwn_config_root}/pwn.yaml"
-      return {} unless File.exist?(pwn_config_path)
+      pwn_env_path = opts[:pwn_env_path] ||= "#{pwn_env_root}/pwn.yaml"
+      return {} unless File.exist?(pwn_env_path)
 
-      is_encrypted = PWN::Plugins::Vault.file_encrypted?(file: pwn_config_path)
+      is_encrypted = PWN::Plugins::Vault.file_encrypted?(file: pwn_env_path)
 
       if is_encrypted
-        pwn_decryptor_path = opts[:pwn_decryptor_path] ||= "#{pwn_config_root}/pwn.decryptor.yaml"
-        raise "PWN Decryptor (#{pwn_decryptor_path}) does not exist!" unless File.exist?(pwn_decryptor_path)
+        pwn_dec_path = opts[:pwn_dec_path] ||= "#{pwn_env_root}/pwn.decryptor.yaml"
+        raise "PWN Decryptor (#{pwn_dec_path}) does not exist!" unless File.exist?(pwn_dec_path)
 
-        pwn_decryptor = YAML.load_file(pwn_decryptor_path, symbolize_names: true)
+        pwn_decryptor = YAML.load_file(pwn_dec_path, symbolize_names: true)
 
         key = opts[:key] ||= pwn_decryptor[:key] ||= ENV.fetch('PWN_DECRYPTOR_KEY')
         key = PWN::Plugins::AuthenticationHelper.mask_password(prompt: 'Decryption Key') if key.nil?
@@ -33,13 +33,13 @@ module PWN
         iv = opts[:iv] ||= pwn_decryptor[:iv] ||= ENV.fetch('PWN_DECRYPTOR_IV')
         iv = PWN::Plugins::AuthenticationHelper.mask_password(prompt: 'Decryption IV') if iv.nil?
 
-        config = PWN::Plugins::Vault.dump(
-          file: pwn_config_path,
+        env = PWN::Plugins::Vault.dump(
+          file: pwn_env_path,
           key: key,
           iv: iv
         )
       else
-        config = YAML.load_file(pwn_config_path, symbolize_names: true)
+        env = YAML.load_file(pwn_env_path, symbolize_names: true)
       end
 
       valid_ai_engines = %i[
@@ -48,14 +48,22 @@ module PWN
         ollama
       ]
 
-      engine = config[:ai][:active].to_s.downcase.to_sym
-      raise "ERROR: Unsupported AI Engine: #{engine} in #{pwn_config_path}.  Supported AI Engines:\n#{valid_ai_engines.inspect}" unless valid_ai_engines.include?(engine)
+      engine = env[:ai][:active].to_s.downcase.to_sym
+      raise "ERROR: Unsupported AI Engine: #{engine} in #{pwn_env_path}.  Supported AI Engines:\n#{valid_ai_engines.inspect}" unless valid_ai_engines.include?(engine)
 
-      model = config[:ai][engine][:model]
-      system_role_content = config[:ai][engine][:system_role_content]
+      key = env[:ai][engine][:key]
+      if key.nil?
+        key = PWN::Plugins::AuthenticationHelper.mask_password(
+          prompt: "#{engine} API Key"
+        )
+        env[:ai][engine][:key] = key
+      end
+
+      model = env[:ai][engine][:model]
+      system_role_content = env[:ai][engine][:system_role_content]
 
-      # Reset the ai response history on config refresh
-      config[:ai][engine][:response_history] = {
+      # Reset the ai response history on env refresh
+      env[:ai][engine][:response_history] = {
         id: '',
         object: '',
         model: model,
@@ -69,13 +77,36 @@ module PWN
       }
 
       # These two lines should be immutable for the session
-      config[:pwn_config_path] = pwn_config_path
-      config[:pwn_decryptor_path] = pwn_decryptor_path if is_encrypted
+      env[:pwn_env_path] = pwn_env_path
+      env[:pwn_dec_path] = pwn_dec_path if is_encrypted
 
       Pry.config.refresh = false if defined?(Pry)
 
-      PWN.send(:remove_const, :CONFIG) if PWN.const_defined?(:CONFIG)
-      PWN.const_set(:CONFIG, config)
+      PWN.send(:remove_const, :Env) if PWN.const_defined?(:Env)
+      PWN.const_set(:Env, env.freeze)
+    rescue StandardError => e
+      raise e
+    end
+
+    # Author(s):: 0day Inc. <support@0dayinc.com>
+
+    public_class_method def self.authors
+      "AUTHOR(S):
+        0day Inc. <support@0dayinc.com>
+      "
+    end
+
+    # Display Usage for this Module
+
+    public_class_method def self.help
+      puts "USAGE:
+        #{self}.refresh_env(
+          pwn_env_path: 'optional - Path to pwn.yaml file.  Defaults to ~/.pwn/pwn.yaml',
+          pwn_dec_path: 'optional - Path to pwn.decryptor.yaml file.  Defaults to ~/.pwn/pwn.decryptor.yaml'
+        )
+
+        #{self}.authors
+      "
     end
   end
 end

data/lib/pwn/plugins/repl.rb

@@ -18,7 +18,7 @@ module PWN
         mode = opts[:mode]
 
         proc do |_target_self, _nest_level, pi|
-          pi.config.pwn = PWN::Config.refresh(opts) if Pry.config.refresh
+          PWN::Config.refresh_env(opts) if Pry.config.refresh
 
           pi.config.pwn_repl_line += 1
           line_pad = format(
@@ -34,8 +34,8 @@ module PWN
           dchars = "\001\e[33m\002***\001\e[0m\002" if mode == :splat
 
           if pi.config.pwn_asm
-            arch = pi.config.pwn[:asm][:arch] ||= PWN::Plugins::DetectOS.arch
-            endian = pi.config.pwn[:asm][:endian] ||= PWN::Plugins::DetectOS.endian
+            arch = PWN::Env[:asm][:arch] ||= PWN::Plugins::DetectOS.arch
+            endian = PWN::Env[:asm][:endian] ||= PWN::Plugins::DetectOS.endian
 
             pi.config.prompt_name = "pwn.asm:#{arch}/#{endian}"
             name = "\001\e[1m\002\001\e[37m\002#{pi.config.prompt_name}\001\e[0m\002"
@@ -44,10 +44,10 @@ module PWN
           end
 
           if pi.config.pwn_ai
-            engine = pi.config.pwn[:ai][:active].to_s.downcase.to_sym
-            model = pi.config.pwn[:ai][engine][:model]
-            system_role_content = pi.config.pwn[:ai][engine][:system_role_content]
-            temp = pi.config.pwn[:ai][engine][:temp]
+            engine = PWN::Env[:ai][:active].to_s.downcase.to_sym
+            model = PWN::Env[:ai][engine][:model]
+            system_role_content = PWN::Env[:ai][engine][:system_role_content]
+            temp = PWN::Env[:ai][engine][:temp]
             pname = "pwn.ai:#{engine}"
             pname = "pwn.ai:#{engine}/#{model}" if model
             pname = "pwn.ai:#{engine}/#{model}.SPEAK" if pi.config.pwn_ai_speak
@@ -173,10 +173,10 @@ module PWN
 
             reply = nil
             response_history = nil
-            shared_chan = pi.config.pwn[:irc][:shared_chan]
+            shared_chan = PWN::Env[:irc][:shared_chan]
             mem_chan = '#mem'
-            ai_agents = pi.config.pwn[:irc][:ai_agent_nicks]
-            ai_agents_arr = pi.config.pwn[:irc][:ai_agent_nicks].keys
+            ai_agents = PWN::Env[:irc][:ai_agent_nicks]
+            ai_agents_arr = PWN::Env[:irc][:ai_agent_nicks].keys
             total_ai_agents = ai_agents_arr.length
             mutex = Mutex.new
             PWN::Plugins::ThreadPool.fill(
@@ -302,12 +302,12 @@ module PWN
                         next unless dm_agent == nick
 
                         response_history = ai_agents[dm_agent.to_sym][:response_history]
-                        engine = pi.config.pwn[:ai][:active].to_s.downcase.to_sym
-                        base_uri = pi.config.pwn[:ai][engine][:base_uri]
-                        key = pi.config.pwn[:ai][engine][:key] ||= ''
-                        temp = pi.config.pwn[:ai][engine][:temp]
-                        model = pi.config.pwn[:ai][engine][:model]
-                        system_role_content = pi.config.pwn[:ai][engine][:system_role_content]
+                        engine = PWN::Env[:ai][:active].to_s.downcase.to_sym
+                        base_uri = PWN::Env[:ai][engine][:base_uri]
+                        key = PWN::Env[:ai][engine][:key] ||= ''
+                        temp = PWN::Env[:ai][engine][:temp]
+                        model = PWN::Env[:ai][engine][:model]
+                        system_role_content = PWN::Env[:ai][engine][:system_role_content]
 
                         users_in_chan = PWN::Plugins::IRC.names(
                           irc_obj: irc_obj,
@@ -431,7 +431,7 @@ module PWN
 
             # TODO: Use TLS for IRC Connections
             # Use an IRC nCurses CLI Client
-            ui_nick = pi.config.pwn[:irc][:ui_nick]
+            ui_nick = PWN::Env[:irc][:ui_nick]
             join_channels = ai_agents_arr.map { |ai_chan| "##{ai_chan}" }.join(',')
 
             cmd0 = "/server add pwn #{host}/#{port} -notls"
@@ -460,24 +460,24 @@ module PWN
 
           def process
             pi = pry_instance
-            pwn_config_path = pi.config.pwn_config_path ||= "#{Dir.home}/.pwn/pwn.yaml"
-            unless File.exist?(pwn_config_path)
-              puts "ERROR: pwn.yaml not found: #{pwn_config_path}"
+            pwn_env_path = PWN::Env[:pwn_env_path] ||= "#{Dir.home}/.pwn/pwn.yaml"
+            unless File.exist?(pwn_env_path)
+              puts "ERROR: pwn environment file not found: #{pwn_env_path}"
               return
             end
 
-            pwn_decryptor_path = pi.config.pwn_decryptor_path ||= "#{Dir.home}/.pwn/pwn.decryptor.yaml"
-            unless File.exist?(pwn_decryptor_path)
-              puts "ERROR: pwn.decryptor.yaml not found: #{pwn_decryptor_path}"
+            pwn_dec_path = PWN::Env[:pwn_dec_path] ||= "#{Dir.home}/.pwn/pwn.decryptor.yaml"
+            unless File.exist?(pwn_dec_path)
+              puts "ERROR: pwn decryptor file not found: #{pwn_dec_path}"
               return
             end
 
-            decryptor = YAML.load_file(pwn_decryptor_path, symbolize_names: true)
+            decryptor = YAML.load_file(pwn_dec_path, symbolize_names: true)
             key = decryptor[:key]
             iv = decryptor[:iv]
 
             PWN::Plugins::Vault.edit(
-              file: pwn_config_path,
+              file: pwn_env_path,
               key: key,
               iv: iv
             )
@@ -530,20 +530,16 @@ module PWN
         # Define REPL Hooks
         # Welcome Banner Hook
         Pry.config.hooks.add_hook(:before_session, :welcome) do |output, _binding, _pi|
+          PWN::Config.refresh_env(opts)
           output.puts PWN::Banner.welcome
         end
 
-        # Initialize pwn.yaml Configuration using :before_session Hook
-        Pry.config.hooks.add_hook(:before_session, :init_opts) do |_output, _binding, pi|
-          pi.config.pwn = PWN::Config.refresh(opts)
-        end
-
         Pry.config.hooks.add_hook(:after_read, :pwn_asm_hook) do |request, pi|
           if pi.config.pwn_asm && !request.chomp.empty?
             request = pi.input.line_buffer
 
-            arch = pi.config.pwn[:asm][:arch]
-            endian = pi.config.pwn[:asm][:endian]
+            arch = PWN::Env[:asm][:arch]
+            endian = PWN::Env[:asm][:endian]
 
             # Analyze request to determine if it should be processed as opcodes or asm.
             straight_hex = /^[a-fA-F0-9\s]+$/
@@ -579,14 +575,14 @@ module PWN
           if pi.config.pwn_ai && !request.chomp.empty?
             request = pi.input.line_buffer.to_s
             debug = pi.config.pwn_ai_debug
-            engine = pi.config.pwn[:ai][:active].to_s.downcase.to_sym
-            base_uri = pi.config.pwn[:ai][engine][:base_uri]
-            key = pi.config.pwn[:ai][engine][:key] ||= ''
-            response_history = pi.config.pwn[:ai][engine][:response_history]
+            engine = PWN::Env[:ai][:active].to_s.downcase.to_sym
+            base_uri = PWN::Env[:ai][engine][:base_uri]
+            key = PWN::Env[:ai][engine][:key] ||= ''
+            response_history = PWN::Env[:ai][engine][:response_history]
             speak_answer = pi.config.pwn_ai_speak
-            model = pi.config.pwn[:ai][engine][:model]
-            system_role_content = pi.config.pwn[:ai][engine][:system_role_content]
-            temp = pi.config.pwn[:ai][engine][:temp]
+            model = PWN::Env[:ai][engine][:model]
+            system_role_content = PWN::Env[:ai][engine][:system_role_content]
+            temp = PWN::Env[:ai][engine][:temp]
 
             case engine
             when :grok
@@ -655,7 +651,7 @@ module PWN
               pp response_history
               puts "\nresponse_history[:choices] Length: #{response_history[:choices].length}\n" unless response_history.nil?
             end
-            pi.config.pwn[:ai][engine][:response_history] = response_history
+            PWN::Env[:ai][engine][:response_history] = response_history
           end
         end
       rescue StandardError => e
@@ -670,8 +666,8 @@ module PWN
       public_class_method def self.start(opts = {})
         # Monkey Patch Pry, add commands, && hooks
         PWN::Plugins::MonkeyPatch.pry
-        pwn_config_root = "#{Dir.home}/.pwn"
-        Pry.config.history_file = "#{pwn_config_root}/pwn_history"
+        pwn_env_root = "#{Dir.home}/.pwn"
+        Pry.config.history_file = "#{pwn_env_root}/pwn_history"
 
         add_commands
         add_hooks(opts)

data/lib/pwn/reports/sast.rb

@@ -25,17 +25,8 @@ module PWN
         }
         report_name = opts[:report_name] ||= File.basename(Dir.pwd)
 
-        ai_instrospection = PWN::CONFIG[:ai][:introspection]
-        if ai_instrospection
-          engine = PWN::CONFIG[:ai][:active].to_s.downcase.to_sym
-          base_uri = PWN::CONFIG[:ai][engine][:base_uri]
-          model = PWN::CONFIG[:ai][engine][:model]
-          key = PWN::CONFIG[:ai][engine][:key]
-          system_role_content = PWN::CONFIG[:ai][engine][:system_role_content]
-          temp = PWN::CONFIG[:ai][engine][:temp]
-
-          puts "Analyzing source code using AI engine: #{engine}\nModel: #{model}\nSystem Role Content: #{system_role_content}\nTemperature: #{temp}"
-        end
+        ai_instrospection = PWN::Env[:ai][:introspection]
+        puts "Analyzing source code using AI engine: #{engine}\nModel: #{model}\nSystem Role Content: #{system_role_content}\nTemperature: #{temp}" if ai_instrospection
 
         # Calculate percentage of AI analysis based on the number of entries
         total_entries = results_hash[:data].sum { |entry| entry[:line_no_and_contents].size }
@@ -63,44 +54,11 @@ module PWN
               line: line_no,
               source_code_snippet: source_code_snippet
             }.to_json
-            response = nil
             author = src_detail[:author].to_s.scrub.chomp.strip
 
-            if ai_instrospection
-              case engine
-              when :grok
-                response = PWN::AI::Grok.chat(
-                  base_uri: base_uri,
-                  token: key,
-                  model: model,
-                  system_role_content: system_role_content,
-                  temp: temp,
-                  request: request.chomp,
-                  spinner: false
-                )
-              when :ollama
-                response = PWN::AI::Ollama.chat(
-                  base_uri: base_uri,
-                  token: key,
-                  model: model,
-                  system_role_content: system_role_content,
-                  temp: temp,
-                  request: request.chomp,
-                  spinner: false
-                )
-              when :openai
-                response = PWN::AI::OpenAI.chat(
-                  base_uri: base_uri,
-                  token: key,
-                  model: model,
-                  system_role_content: system_role_content,
-                  temp: temp,
-                  request: request.chomp,
-                  spinner: false
-                )
-              end
-            end
-
+            # TODO: move PWN::AI::Introspection.reflect into each PWN::SAST::* module
+            # This will drastically speed up the overall SAST analysis process
+            response = PWN::AI::Introspection.reflect if ai_instrospection
             ai_analysis = nil
             if response.is_a?(Hash)
               ai_analysis = response[:choices].last[:text] if response[:choices].last.keys.include?(:text)

data/lib/pwn/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module PWN
-  VERSION = '0.5.432'
+  VERSION = '0.5.434'
 end

data/lib/pwn.rb

@@ -26,8 +26,8 @@ module PWN
   end
 
   # Initialize PWN configuration file
-  # PWN::CONFIG is the constant that stores the configuration data
-  PWN::Config.refresh
+  # PWN::Env is the constant that stores the configuration data
+  PWN::Config.refresh_env
 rescue StandardError => e
   puts e.backtrace
   raise e

data/spec/lib/pwn/ai/introspection_spec.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe PWN::AI::Introspection do
+  it 'should display information for authors' do
+    authors_response = PWN::AI::Introspection
+    expect(authors_response).to respond_to :authors
+  end
+
+  it 'should display information for existing help method' do
+    help_response = PWN::AI::Introspection
+    expect(help_response).to respond_to :help
+  end
+end

data/spec/lib/pwn/config_spec.rb

@@ -4,7 +4,17 @@ require 'spec_helper'
 
 describe PWN::Config do
   it 'should return data for refresh method' do
-    config_response = PWN::Config.refresh
+    config_response = PWN::Config.refresh_env
     expect(config_response).not_to be_nil
   end
+
+  it 'should display information for authors' do
+    authors_response = PWN::Config
+    expect(authors_response).to respond_to :authors
+  end
+
+  it 'should display information for existing help method' do
+    help_response = PWN::Config
+    expect(help_response).to respond_to :help
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.5.432
+  version: 0.5.434
 platform: ruby
 authors:
 - 0day Inc.
@@ -841,14 +841,14 @@ dependencies:
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 6.14.2
+        version: 6.15.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 6.14.2
+        version: 6.15.0
 - !ruby/object:Gem::Dependency
   name: rest-client
   requirement: !ruby/object:Gem::Requirement
@@ -1720,6 +1720,7 @@ files:
 - lib/pwn.rb
 - lib/pwn/ai.rb
 - lib/pwn/ai/grok.rb
+- lib/pwn/ai/introspection.rb
 - lib/pwn/ai/ollama.rb
 - lib/pwn/ai/open_ai.rb
 - lib/pwn/aws.rb
@@ -2068,6 +2069,7 @@ files:
 - pwn.gemspec
 - reinstall_pwn_gemset.sh
 - spec/lib/pwn/ai/grok_spec.rb
+- spec/lib/pwn/ai/introspection_spec.rb
 - spec/lib/pwn/ai/ollama_spec.rb
 - spec/lib/pwn/ai/open_ai_spec.rb
 - spec/lib/pwn/ai_spec.rb