pwn 0.5.425 → 0.5.427

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 30904f7cfcbec2b9dc3ae06c5cb6ef6cf1c59e711bccbc7a7302b26e21639eef
-  data.tar.gz: 7e4c812591317c30242e558bf924d2eacd96563dbfea75ebf07161435c7c1720
+  metadata.gz: ac10751175e251ad205dd5e3c4cc6b1eeb12f45717b7099a018cb71b3628fa7f
+  data.tar.gz: 2535096c15a6baf77f987e7c79b12655a8191b895bc0a1907995fd0cae98159a
 SHA512:
-  metadata.gz: 2fffe14cfa05fb8960ab0551de18171f02643b4162d3ee9e06ab60adb081345f71b6c8c9409555f010762d356f419445a7358e3ccf3e501b53131cac7579b94f
-  data.tar.gz: 6795f441e69e5ec1b7bd43831bf273aec2596565ecd8734618a293c82f0fd360a16a1378399c72d1468d89bcd42c276f4b5d217a960d5d46525cfa15a80859d9
+  metadata.gz: 0d749319613654cbbc475657a6901e4cdad824b72a90889d4612bc89e88df188277941cdb41df44078ef500d549f5bc5de014bdacf141d37c63b82c7c1e14d65
+  data.tar.gz: b7538e5a12b5512f48ebf0976eb2c14009d7c97730eefc8271b77a0b7e53cbbe1b777420318c4bae032d4094e8ff495debc6a00f27ae31799fa966a213737cd3

data/Gemfile

@@ -90,7 +90,7 @@ gem 'ruby-nmap', '1.0.3'
 gem 'ruby-saml', '1.18.1'
 gem 'rvm', '1.11.3.9'
 gem 'savon', '2.15.1'
-gem 'selenium-devtools', '0.139.0'
+gem 'selenium-devtools', '0.140.0'
 # gem 'serialport', '1.3.2'
 # gem 'sinatra', '4.0.0'
 gem 'slack-ruby-client', '3.0.0'

data/README.md

@@ -37,7 +37,7 @@ $ cd /opt/pwn
 $ ./install.sh
 $ ./install.sh ruby-gem
 $ pwn
-pwn[v0.5.425]:001 >>> PWN.help
+pwn[v0.5.427]: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.425]:001 >>> PWN.help
+pwn[v0.5.427]: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.425]:001 >>> PWN.help
+pwn[v0.5.427]: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

@@ -14,16 +14,8 @@ OptionParser.new do |options|
     opts[:yaml_config_path] = p
   end
 
-  options.on('-dPATH', '--decryptor=PATH', '<Optional - File Containing Decryption Key && IV>') do |d|
-    opts[:decryption_file] = d
-  end
-
-  options.on('-kKEY', '--decryption-key=KEY', '<Optional - Decryption Key>') do |k|
-    opts[:key] = k
-  end
-
-  options.on('-iIV', '--decryption-iv=PATH', '<Optional - Decryption IV>') do |i|
-    opts[:iv] = i
+  options.on('-dPATH', '--yaml-decryptor=PATH', '<Optional - Out-of-Band YAML File with :key && :iv>') do |d|
+    opts[:yaml_decryptor_path] = d
   end
 end.parse!
 

data/lib/pwn/ai.rb

@@ -9,7 +9,7 @@ module PWN
     autoload :Ollama, 'pwn/ai/ollama'
     autoload :OpenAI, 'pwn/ai/open_ai'
 
-    # Display a List of Every PWN::Plugins Module
+    # Display a List of Every PWN::AI Module
 
     public_class_method def self.help
       constants.sort

data/lib/pwn/blockchain/btc.rb

@@ -0,0 +1,160 @@
+# frozen_string_literal: true
+
+require 'json'
+require 'rest-client'
+require 'tty-spinner'
+
+module PWN
+  module Blockchain
+    # This plugin interacts with BitCoin's Blockchain API.
+    module BTC
+      # Supported Method Parameters::
+      # btc_rest_call(
+      #   http_method: 'optional HTTP method (defaults to GET)
+      #   rest_call: 'required rest call to make per the schema',
+      #   params: 'optional params passed in the URI or HTTP Headers',
+      #   http_body: 'optional HTTP body sent in HTTP methods that support it e.g. POST',
+      #   timeout: 'optional timeout in seconds (defaults to 180)',
+      #   spinner: 'optional - display spinner (defaults to false)'
+      # )
+
+      private_class_method def self.btc_rest_call(opts = {})
+        http_method = if opts[:http_method].nil?
+                        :get
+                      else
+                        opts[:http_method].to_s.scrub.to_sym
+                      end
+
+        base_uri = 'https://api.blockcypher.com/v1/btc/'
+        rest_call = opts[:rest_call].to_s.scrub
+        params = opts[:params]
+        headers = { content_type: 'application/json; charset=UTF-8' }
+
+        http_body = opts[:http_body]
+        http_body ||= {}
+
+        timeout = opts[:timeout]
+        timeout ||= 180
+
+        spinner = opts[:spinner] || false
+
+        browser_obj = PWN::Plugins::TransparentBrowser.open(browser_type: :rest)
+        rest_client = browser_obj[:browser]::Request
+
+        if spinner
+          spin = TTY::Spinner.new(format: :dots)
+          spin.auto_spin
+        end
+
+        retries = 0
+        case http_method
+        when :delete, :get
+          headers[:params] = params
+          response = rest_client.execute(
+            method: http_method,
+            url: "#{base_uri}#{rest_call}",
+            headers: headers,
+            verify_ssl: false,
+            timeout: timeout
+          )
+
+        when :post
+          if http_body.key?(:multipart)
+            headers[:content_type] = 'multipart/form-data'
+
+            response = rest_client.execute(
+              method: http_method,
+              url: "#{base_uri}#{rest_call}",
+              headers: headers,
+              payload: http_body,
+              verify_ssl: false,
+              timeout: timeout
+            )
+          else
+            response = rest_client.execute(
+              method: http_method,
+              url: "#{base_uri}#{rest_call}",
+              headers: headers,
+              payload: http_body.to_json,
+              verify_ssl: false,
+              timeout: timeout
+            )
+          end
+
+        else
+          raise "Unsupported HTTP Method #{http_method} for #{self} Plugin"
+        end
+
+        response
+      rescue RestClient::ExceptionWithResponse => e
+        case e.http_code
+        when 400, 404
+          "#{e.http_code} #{e.message}: #{e.response.body}"
+        else
+          raise e
+        end
+      rescue StandardError => e
+        raise e
+      ensure
+        spin.stop if spinner
+      end
+
+      # Supported Method Parameters::
+      # latest_block = PWN::Blockchain::BTC.get_latest_block(
+      #   token: 'optional - API token for higher rate limits'
+      # )
+
+      public_class_method def self.get_latest_block(opts = {})
+        params = {}
+        params[:token] = opts[:token] if opts[:token]
+
+        rest_call = 'main'
+        response = btc_rest_call(rest_call: rest_call, params: params)
+
+        JSON.parse(response.body, symbolize_names: true)
+      rescue StandardError => e
+        raise e
+      end
+
+      # Supported Method Parameters::
+      # PWN::Blockchain::BTC.get_block_details(
+      #   height: 'required - block height number',
+      #   token: 'optional - API token for higher rate limits'
+      # )
+      public_class_method def self.get_block_details(opts = {})
+        height = opts[:height]
+        params = {}
+        params[:token] = opts[:token] if opts[:token]
+
+        rest_call = "main/blocks/#{height}"
+        response = btc_rest_call(rest_call: rest_call, params: params)
+
+        JSON.parse(response.body, symbolize_names: true)
+      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:
+          latest_block = #{self}.get_latest_block
+
+          block_details = #{self}.get_block_details(
+            height: 'required - block height number'
+          )
+
+          #{self}.authors
+        "
+      end
+    end
+  end
+end

data/lib/pwn/blockchain/eth.rb

@@ -0,0 +1,160 @@
+# frozen_string_literal: true
+
+require 'json'
+require 'rest-client'
+require 'tty-spinner'
+
+module PWN
+  module Blockchain
+    # This plugin interacts with BitCoin's Blockchain API.
+    module ETH
+      # Supported Method Parameters::
+      # eth_rest_call(
+      #   http_method: 'optional HTTP method (defaults to GET)
+      #   rest_call: 'required rest call to make per the schema',
+      #   params: 'optional params passed in the URI or HTTP Headers',
+      #   http_body: 'optional HTTP body sent in HTTP methods that support it e.g. POST',
+      #   timeout: 'optional timeout in seconds (defaults to 180)',
+      #   spinner: 'optional - display spinner (defaults to false)'
+      # )
+
+      private_class_method def self.eth_rest_call(opts = {})
+        http_method = if opts[:http_method].nil?
+                        :get
+                      else
+                        opts[:http_method].to_s.scrub.to_sym
+                      end
+
+        base_uri = 'https://api.blockcypher.com/v1/eth/'
+        rest_call = opts[:rest_call].to_s.scrub
+        params = opts[:params]
+        headers = { content_type: 'application/json; charset=UTF-8' }
+
+        http_body = opts[:http_body]
+        http_body ||= {}
+
+        timeout = opts[:timeout]
+        timeout ||= 180
+
+        spinner = opts[:spinner] || false
+
+        browser_obj = PWN::Plugins::TransparentBrowser.open(browser_type: :rest)
+        rest_client = browser_obj[:browser]::Request
+
+        if spinner
+          spin = TTY::Spinner.new(format: :dots)
+          spin.auto_spin
+        end
+
+        retries = 0
+        case http_method
+        when :delete, :get
+          headers[:params] = params
+          response = rest_client.execute(
+            method: http_method,
+            url: "#{base_uri}#{rest_call}",
+            headers: headers,
+            verify_ssl: false,
+            timeout: timeout
+          )
+
+        when :post
+          if http_body.key?(:multipart)
+            headers[:content_type] = 'multipart/form-data'
+
+            response = rest_client.execute(
+              method: http_method,
+              url: "#{base_uri}#{rest_call}",
+              headers: headers,
+              payload: http_body,
+              verify_ssl: false,
+              timeout: timeout
+            )
+          else
+            response = rest_client.execute(
+              method: http_method,
+              url: "#{base_uri}#{rest_call}",
+              headers: headers,
+              payload: http_body.to_json,
+              verify_ssl: false,
+              timeout: timeout
+            )
+          end
+
+        else
+          raise "Unsupported HTTP Method #{http_method} for #{self} Plugin"
+        end
+
+        response
+      rescue RestClient::ExceptionWithResponse => e
+        case e.http_code
+        when 400, 404
+          "#{e.http_code} #{e.message}: #{e.response.body}"
+        else
+          raise e
+        end
+      rescue StandardError => e
+        raise e
+      ensure
+        spin.stop if spinner
+      end
+
+      # Supported Method Parameters::
+      # latest_block = PWN::Blockchain::ETH.get_latest_block(
+      #   token: 'optional - API token for higher rate limits'
+      # )
+
+      public_class_method def self.get_latest_block(opts = {})
+        params = {}
+        params[:token] = opts[:token] if opts[:token]
+
+        rest_call = 'main'
+        response = eth_rest_call(rest_call: rest_call, params: params)
+
+        JSON.parse(response.body, symbolize_names: true)
+      rescue StandardError => e
+        raise e
+      end
+
+      # Supported Method Parameters::
+      # PWN::Blockchain::ETH.get_block_details(
+      #   height: 'required - block height number',
+      #   token: 'optional - API token for higher rate limits'
+      # )
+      public_class_method def self.get_block_details(opts = {})
+        height = opts[:height]
+        params = {}
+        params[:token] = opts[:token] if opts[:token]
+
+        rest_call = "main/blocks/#{height}"
+        response = eth_rest_call(rest_call: rest_call, params: params)
+
+        JSON.parse(response.body, symbolize_names: true)
+      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:
+          latest_block = #{self}.get_latest_block
+
+          block_details = #{self}.get_block_details(
+            height: 'required - block height number'
+          )
+
+          #{self}.authors
+        "
+      end
+    end
+  end
+end

data/lib/pwn/blockchain.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module PWN
+  # This file, using the autoload directive loads SP plugins
+  # into memory only when they're needed. For more information, see:
+  # http://www.rubyinside.com/ruby-techniques-revealed-autoload-1652.html
+  module Blockchain
+    autoload :BTC, 'pwn/blockchain/btc'
+    autoload :ETH, 'pwn/blockchain/eth'
+
+    # Display a List of Every PWN::Blockchain Module
+
+    public_class_method def self.help
+      constants.sort
+    end
+  end
+end

data/lib/pwn/plugins/repl.rb

@@ -18,7 +18,11 @@ module PWN
         mode = opts[:mode]
 
         proc do |_target_self, _nest_level, pi|
-          PWN::Plugins::Vault.refresh_config_for_repl(opts) if Pry.config.refresh_config
+          if Pry.config.refresh
+            # puts "Refreshing PWN env via #{opts[:yaml_config_path]}"
+            opts[:pi] = pi
+            PWN::Plugins::Vault.refresh_config_for_repl(opts)
+          end
 
           pi.config.pwn_repl_line += 1
           line_pad = format(
@@ -34,8 +38,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 = pi.config.pwn[:asm][:arch] ||= PWN::Plugins::DetectOS.arch
+            endian = pi.config.pwn[: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 +48,10 @@ module PWN
           end
 
           if pi.config.pwn_ai
-            ai_engine = pi.config.pwn_ai_engine
-            model = pi.config.pwn_ai_model
-            system_role_content = pi.config.pwn_ai_system_role_content
-            temp = pi.config.pwn_ai_temp
+            ai_engine = pi.config.pwn[:ai_engine]
+            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]
             pname = "pwn.ai:#{ai_engine}"
             pname = "pwn.ai:#{ai_engine}/#{model}" if model
             pname = "pwn.ai:#{ai_engine}/#{model}.SPEAK" if pi.config.pwn_ai_speak
@@ -173,10 +177,10 @@ module PWN
 
             reply = nil
             response_history = nil
-            shared_chan = pi.config.pwn_irc[:shared_chan]
+            shared_chan = pi.config.pwn[: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 = pi.config.pwn[:irc][:ai_agent_nicks]
+            ai_agents_arr = pi.config.pwn[:irc][:ai_agent_nicks].keys
             total_ai_agents = ai_agents_arr.length
             mutex = Mutex.new
             PWN::Plugins::ThreadPool.fill(
@@ -302,15 +306,12 @@ module PWN
                         next unless dm_agent == nick
 
                         response_history = ai_agents[dm_agent.to_sym][:response_history]
-                        ai_engine = pi.config.pwn_ai_engine
-                        ai_base_uri = pi.config.pwn_ai_base_uri
-                        ai_key = pi.config.pwn_ai_key
-                        ai_key ||= ''
-                        ai_temp = pi.config.pwn_ai_temp
-
-                        model = pi.config.pwn_ai_model
-                        system_role_content = pi.config.pwn_ai_system_role_content
-                        temp = pi.config.pwn_ai_temp
+                        ai_engine = pi.config.pwn[:ai_engine]
+                        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]
 
                         users_in_chan = PWN::Plugins::IRC.names(
                           irc_obj: irc_obj,
@@ -344,10 +345,10 @@ module PWN
                         case ai_engine
                         when :grok
                           response = PWN::AI::Grok.chat(
-                            base_uri: ai_base_uri,
-                            token: ai_key,
+                            base_uri: base_uri,
+                            token: key,
                             model: model,
-                            temp: ai_temp,
+                            temp: temp,
                             system_role_content: system_role_content,
                             request: request,
                             response_history: response_history,
@@ -355,10 +356,10 @@ module PWN
                           )
                         when :ollama
                           response = PWN::AI::Ollama.chat(
-                            base_uri: ai_base_uri,
-                            token: ai_key,
+                            base_uri: base_uri,
+                            token: key,
                             model: model,
-                            temp: ai_temp,
+                            temp: temp,
                             system_role_content: system_role_content,
                             request: request,
                             response_history: response_history,
@@ -366,10 +367,10 @@ module PWN
                           )
                         when :openai
                           response = PWN::AI::OpenAI.chat(
-                            base_uri: ai_base_uri,
-                            token: ai_key,
+                            base_uri: base_uri,
+                            token: key,
                             model: model,
-                            temp: ai_temp,
+                            temp: temp,
                             system_role_content: system_role_content,
                             request: request,
                             response_history: response_history,
@@ -434,7 +435,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 = pi.config.pwn[:irc][:ui_nick]
             join_channels = ai_agents_arr.map { |ai_chan| "##{ai_chan}" }.join(',')
 
             cmd0 = "/server add pwn #{host}/#{port} -notls"
@@ -469,12 +470,13 @@ module PWN
               return
             end
 
-            decryption_file = pi.config.decryption_file ||= "#{Dir.home}/.pwn/pwn.decryptor.yaml"
-            unless File.exist?(decryption_file)
-              puts "ERROR: pwn.decryptor.yaml not found: #{decryption_file}"
+            yaml_decryptor_path = pi.config.yaml_decryptor_path ||= "#{Dir.home}/.pwn/pwn.decryptor.yaml"
+            unless File.exist?(yaml_decryptor_path)
+              puts "ERROR: pwn.decryptor.yaml not found: #{yaml_decryptor_path}"
               return
             end
-            decryptor = YAML.load_file(decryption_file, symbolize_names: true)
+
+            decryptor = YAML.load_file(yaml_decryptor_path, symbolize_names: true)
             key = decryptor[:key]
             iv = decryptor[:iv]
 
@@ -537,10 +539,8 @@ module PWN
 
         # Initialize pwn.yaml Configuration using :before_session Hook
         Pry.config.hooks.add_hook(:before_session, :init_opts) do |_output, _binding, pi|
+          # puts "Refreshing PWN env via #{opts[:yaml_config_path]}"
           opts[:pi] = pi
-          Pry.config.yaml_config_path = opts[:yaml_config_path]
-          Pry.config.decryption_file = opts[:decryption_file]
-
           PWN::Plugins::Vault.refresh_config_for_repl(opts)
         end
 
@@ -548,8 +548,8 @@ module PWN
           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 = pi.config.pwn[:asm][:arch]
+            endian = pi.config.pwn[:asm][:endian]
 
             # Analyze request to determine if it should be processed as opcodes or asm.
             straight_hex = /^[a-fA-F0-9\s]+$/
@@ -585,29 +585,27 @@ module PWN
           if pi.config.pwn_ai && !request.chomp.empty?
             request = pi.input.line_buffer.to_s
             debug = pi.config.pwn_ai_debug
-            ai_engine = pi.config.pwn_ai_engine.to_s.to_sym
-            ai_key = pi.config.pwn_ai_key
-            ai_key ||= ''
-            if ai_key.empty?
-              ai_key = PWN::Plugins::AuthenticationHelper.mask_password(
+            ai_engine = pi.config.pwn[:ai_engine]
+            base_uri = pi.config.pwn[ai_engine][:base_uri]
+            key = pi.config.pwn[ai_engine][:key] ||= ''
+            if key.empty?
+              key = PWN::Plugins::AuthenticationHelper.mask_password(
                 prompt: 'pwn-ai Key'
               )
-              pi.config.pwn_ai_key = ai_key
+              pi.config.pwn[ai_engine][:key] = key
             end
 
             response_history = pi.config.pwn_ai_response_history
             speak_answer = pi.config.pwn_ai_speak
-            model = pi.config.pwn_ai_model
-            system_role_content = pi.config.pwn_ai_system_role_content
-            temp = pi.config.pwn_ai_temp
-
-            ai_base_uri = pi.config.pwn_ai_base_uri
+            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]
 
             case ai_engine
             when :grok
               response = PWN::AI::Grok.chat(
-                base_uri: ai_base_uri,
-                token: ai_key,
+                base_uri: base_uri,
+                token: key,
                 model: model,
                 system_role_content: system_role_content,
                 temp: temp,
@@ -618,8 +616,8 @@ module PWN
               )
             when :ollama
               response = PWN::AI::Ollama.chat(
-                base_uri: ai_base_uri,
-                token: ai_key,
+                base_uri: base_uri,
+                token: key,
                 model: model,
                 system_role_content: system_role_content,
                 temp: temp,
@@ -630,8 +628,8 @@ module PWN
               )
             when :openai
               response = PWN::AI::OpenAI.chat(
-                base_uri: ai_base_uri,
-                token: ai_key,
+                base_uri: base_uri,
+                token: key,
                 model: model,
                 system_role_content: system_role_content,
                 temp: temp,
@@ -690,7 +688,7 @@ module PWN
         pwn_config_root = "#{Dir.home}/.pwn"
         FileUtils.mkdir_p(pwn_config_root)
         opts[:yaml_config_path] ||= "#{pwn_config_root}/pwn.yaml"
-        opts[:decryption_file] ||= "#{pwn_config_root}/pwn.decryptor.yaml"
+        opts[:yaml_decryptor_path] ||= "#{pwn_config_root}/pwn.decryptor.yaml"
 
         add_hooks(opts)
 

data/lib/pwn/plugins/vault.rb

@@ -173,7 +173,7 @@ module PWN
         system(relative_editor, file)
 
         # If the Pry object exists, set refresh_config to true
-        Pry.config.refresh_config = true if defined?(Pry)
+        Pry.config.refresh = true if defined?(Pry)
 
         encrypt(
           file: file,
@@ -238,23 +238,24 @@ module PWN
       # PWN::Plugins::Vault.refresh_config_for_repl(
       #    yaml_config_path: 'required - full path to pwn.yaml file',
       #    pi: 'optional - Pry instance (default: Pry)',
-      #    decryption_file: 'optional - full path to decryption YAML file'
+      #    yaml_decryptor_path: 'optional - full path to decryption YAML file'
       #  )
       public_class_method def self.refresh_config_for_repl(opts = {})
         yaml_config_path = opts[:yaml_config_path]
 
         return false unless File.exist?(yaml_config_path)
 
-        pi = opts[:pi] ||= Pry
+        pi = opts[:pi]
+        raise 'ERROR: Pry instance is required.' if pi.nil?
 
         is_encrypted = PWN::Plugins::Vault.file_encrypted?(file: yaml_config_path)
 
         if is_encrypted
           # TODO: Implement "something you know, something you have, && something you are?"
-          decryption_file = opts[:decryption_file] ||= "#{Dir.home}/pwn.decryptor.yaml"
-          raise "ERROR: #{decryption_file} does not exist." unless File.exist?(decryption_file)
+          yaml_decryptor_path = opts[:yaml_decryptor_path] ||= "#{Dir.home}/.pwn/pwn.decryptor.yaml"
+          raise "ERROR: #{yaml_decryptor_path} does not exist." unless File.exist?(yaml_decryptor_path)
 
-          yaml_decryptor = YAML.load_file(decryption_file, symbolize_names: true)
+          yaml_decryptor = YAML.load_file(yaml_decryptor_path, symbolize_names: true)
 
           key = opts[:key] ||= yaml_decryptor[:key] ||= ENV.fetch('PWN_DECRYPTOR_KEY')
           key = PWN::Plugins::AuthenticationHelper.mask_password(prompt: 'Decryption Key') if key.nil?
@@ -270,52 +271,41 @@ module PWN
         else
           yaml_config = YAML.load_file(yaml_config_path, symbolize_names: true)
         end
-        pi.config.p = yaml_config
-        Pry.config.p = yaml_config
 
         valid_ai_engines = %i[
           grok
           openai
           ollama
         ]
-        ai_engine = yaml_config[:ai_engine].to_s.downcase.to_sym
 
+        # Convert ai_engine to symbol and downcase to ensure stability
+        yaml_config[:ai_engine] = yaml_config[:ai_engine].to_s.downcase.to_sym
+        pi.config.pwn = yaml_config
+        ai_engine = pi.config.pwn[:ai_engine]
         raise "ERROR: Unsupported AI Engine: #{ai_engine} in #{yaml_config_path}.  Supported AI Engines:\n#{valid_ai_engines.inspect}" unless valid_ai_engines.include?(ai_engine)
 
-        pi.config.pwn_ai_engine = ai_engine
-        Pry.config.pwn_ai_engine = ai_engine
-
-        pi.config.pwn_ai_base_uri = pi.config.p[ai_engine][:base_uri]
-        Pry.config.pwn_ai_base_uri = pi.config.pwn_ai_base_uri
-
-        pi.config.pwn_ai_key = pi.config.p[ai_engine][:key]
-        Pry.config.pwn_ai_key = pi.config.pwn_ai_key
-
-        pi.config.pwn_ai_model = pi.config.p[ai_engine][:model]
-        Pry.config.pwn_ai_model = pi.config.pwn_ai_model
-
-        pi.config.pwn_ai_system_role_content = pi.config.p[ai_engine][:system_role_content]
-        Pry.config.pwn_ai_system_role_content = pi.config.pwn_ai_system_role_content
-
-        pi.config.pwn_ai_temp = pi.config.p[ai_engine][:temp]
-        Pry.config.pwn_ai_temp = pi.config.pwn_ai_temp
-
-        pi.config.pwn_asm_arch = pi.config.p[:asm][:arch]
-        Pry.config.pwn_asm_arch = pi.config.pwn_asm_arch
-
-        pi.config.pwn_asm_endian = pi.config.p[:asm][:endian]
-        Pry.config.pwn_asm_endian = pi.config.pwn_asm_endian
-
-        pi.config.pwn_irc = pi.config.p[:irc]
-        Pry.config.pwn_irc = pi.config.pwn_irc
-
-        pi.config.pwn_hunter = pi.config.p[:hunter][:api_key]
-        Pry.config.pwn_hunter = pi.config.pwn_hunter
-
-        pi.config.pwn_shodan = pi.config.p[:shodan][:api_key]
-        Pry.config.pwn_shodan = pi.config.pwn_shodan
-
-        Pry.config.refresh_config = false
+        model = pi.config.pwn[ai_engine][:model]
+        system_role_content = pi.config.pwn[ai_engine][:system_role_content]
+
+        # Reset the ai response history for new configurations
+        pi.config.pwn_ai_response_history = {
+          id: '',
+          object: '',
+          model: model,
+          usage: {},
+          choices: [
+            {
+              role: 'system',
+              content: system_role_content
+            }
+          ]
+        }
+
+        # These two lines should be immutable for the session
+        pi.config.pwn[:yaml_config_path] = yaml_config_path
+        pi.config.pwn[:yaml_decryptor_path] = yaml_decryptor_path if is_encrypted
+
+        Pry.config.refresh = false
 
         true
       rescue StandardError => e
@@ -377,7 +367,7 @@ module PWN
           #{self}.refresh_config_for_repl(
             yaml_config_path: 'required - full path to pwn.yaml file',
             pi: 'optional - Pry instance (default: Pry)',
-            decryption_file: 'optional - full path to decryption YAML file'
+            yaml_decryptor_path: 'optional - full path to decryption YAML file'
           )
 
           #{self}.authors

data/lib/pwn/version.rb

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

data/lib/pwn.rb

@@ -11,6 +11,7 @@ module PWN
   autoload :AI, 'pwn/ai'
   autoload :AWS, 'pwn/aws'
   autoload :Banner, 'pwn/banner'
+  autoload :Blockchain, 'pwn/blockchain'
   autoload :FFI, 'pwn/ffi'
   autoload :Plugins, 'pwn/plugins'
   autoload :Reports, 'pwn/reports'

data/spec/lib/pwn/blockchain/btc_spec.rb

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

data/spec/lib/pwn/blockchain/eth_spec.rb

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

data/spec/lib/pwn/blockchain_spec.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe PWN::Blockchain do
+  it 'should return data for help method' do
+    help_response = PWN::Blockchain.help
+    expect(help_response).not_to be_nil
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.5.425
+  version: 0.5.427
 platform: ruby
 authors:
 - 0day Inc.
@@ -1051,14 +1051,14 @@ dependencies:
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 0.139.0
+        version: 0.140.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 0.139.0
+        version: 0.140.0
 - !ruby/object:Gem::Dependency
   name: slack-ruby-client
   requirement: !ruby/object:Gem::Requirement
@@ -1829,6 +1829,9 @@ files:
 - lib/pwn/banner/radare2.rb
 - lib/pwn/banner/radare2_ai.rb
 - lib/pwn/banner/white_rabbit.rb
+- lib/pwn/blockchain.rb
+- lib/pwn/blockchain/btc.rb
+- lib/pwn/blockchain/eth.rb
 - lib/pwn/ffi.rb
 - lib/pwn/ffi/stdio.rb
 - lib/pwn/plugins.rb
@@ -2174,6 +2177,9 @@ files:
 - spec/lib/pwn/banner/radare2_spec.rb
 - spec/lib/pwn/banner/white_rabbit_spec.rb
 - spec/lib/pwn/banner_spec.rb
+- spec/lib/pwn/blockchain/btc_spec.rb
+- spec/lib/pwn/blockchain/eth_spec.rb
+- spec/lib/pwn/blockchain_spec.rb
 - spec/lib/pwn/ffi/stdio_spec.rb
 - spec/lib/pwn/ffi_spec.rb
 - spec/lib/pwn/plugins/android_spec.rb