pwn 0.5.422 → 0.5.424

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5fda4003c5c2b625faae80ac66aa29bd12a7a450e3dc9259cb2d5d82e1d71716
-  data.tar.gz: 758acc86bc80e1dc4038ee14314025d163fa63d1682bd1cd9c9c2a184332d059
+  metadata.gz: 6132f715f6f2e55294823df435cfc144f67137d1ba0757aa9c21585313719e07
+  data.tar.gz: fef173b311c91b9cc2de754edb1719d68721fe4963c7007dacdc5928e12c717d
 SHA512:
-  metadata.gz: 8db71ed909971e84ce91cc04d3922de9a194c3b2179be7eb56f53ff47ae3e0c1624731cc6ca430ecad984f262968b89fd81146a0ac01e14f20e0b2e6a2b7f802
-  data.tar.gz: cb1af569edffd8ec983ddb9118b64cf0fc63a0e814de86adc35cfd8d5e39d38427b49228a9cb313c8dab2f393d9d15da6efae94cbc0b6eecdd043fd2f2600386
+  metadata.gz: ca41b4491a62fc9c54d1f4e12d9ee06787494ae8810a471419ddda102cff8a0f81f38ff03f5e3153f903f7cf8c9736edc2ad05c3d1375ed894688bf45fab1564
+  data.tar.gz: ef2fa3d5fd8f576b1c96932d243c9a3e692e36026832caf3f9736c8e56157cff833deed42eff973932dfebe353ab2bf2d2c5d0b72319eb18901e65f58188ff9b

data/README.md

@@ -37,7 +37,7 @@ $ cd /opt/pwn
 $ ./install.sh
 $ ./install.sh ruby-gem
 $ pwn
-pwn[v0.5.422]:001 >>> PWN.help
+pwn[v0.5.424]: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.422]:001 >>> PWN.help
+pwn[v0.5.424]: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.422]:001 >>> PWN.help
+pwn[v0.5.424]: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/lib/pwn/plugins/repl.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require 'fileutils'
 require 'pry'
 require 'tty-prompt'
 require 'yaml'
@@ -457,6 +458,36 @@ module PWN
           end
         end
 
+        Pry::Commands.create_command 'pwn-vault-edit' do
+          description 'Edit the pwn.yaml configuration file.'
+
+          def process
+            pi = pry_instance
+            yaml_config_path = pi.config.yaml_config_path ||= "#{Dir.home}/pwn.yaml"
+            unless File.exist?(yaml_config_path)
+              puts "ERROR: pwn.yaml not found: #{yaml_config_path}"
+              return
+            end
+
+            decryption_file = pi.config.decryption_file ||= "#{Dir.home}/pwn.decryptor.yaml"
+            unless File.exist?(decryption_file)
+              puts "ERROR: pwn.decryptor.yaml not found: #{decryption_file}"
+              return
+            end
+            decryptor = YAML.load_file(decryption_file, symbolize_names: true)
+            key = decryptor[:key]
+            iv = decryptor[:iv]
+
+            PWN::Plugins::Vault.edit(
+              file: yaml_config_path,
+              key: key,
+              iv: iv
+            )
+          rescue StandardError => e
+            raise e
+          end
+        end
+
         Pry::Commands.create_command 'toggle-pwn-ai-debug' do
           description 'Display the response_history object while using pwn.ai'
 
@@ -507,6 +538,9 @@ module PWN
         # Initialize pwn.yaml Configuration using :before_session Hook
         Pry.config.hooks.add_hook(:before_session, :init_opts) do |_output, _binding, pi|
           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
 
@@ -652,6 +686,12 @@ module PWN
         # Monkey Patch Pry, add commands, && hooks
         PWN::Plugins::MonkeyPatch.pry
         add_commands
+
+        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"
+
         add_hooks(opts)
 
         # Define PS1 Prompt

data/lib/pwn/plugins/vault.rb

@@ -243,10 +243,9 @@ module PWN
       public_class_method def self.refresh_config_for_repl(opts = {})
         yaml_config_path = opts[:yaml_config_path]
 
-        return false unless yaml_config_path
+        return false unless File.exist?(yaml_config_path)
 
         pi = opts[:pi] ||= Pry
-        raise "ERROR: #{yaml_config_path} does not exist." unless File.exist?(yaml_config_path)
 
         is_encrypted = PWN::Plugins::Vault.file_encrypted?(file: yaml_config_path)
 

data/lib/pwn/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.5.422
+  version: 0.5.424
 platform: ruby
 authors:
 - 0day Inc.