pwn 0.4.391 → 0.4.393

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 20f6117dd5a357f49c9cb2319b43b5189eca333b73a2607d0c05124cc685e4c7
-  data.tar.gz: e631631795c7b0d4ecc28afb7d544c5b7ad82868a564a6dba4c914a60e477c12
+  metadata.gz: 9638834cb2c2444a04cd23b17e5c0e187c396a712677dd2f02a0be933b7aea3e
+  data.tar.gz: 49240e2aec06df7358db080036c326969112bad268d0fa08a5b36456f1a6559f
 SHA512:
-  metadata.gz: 82d47c1278b2d7dabb9740beb4dec9bbc20b472dcf368d5dca5e3552ad201cc70d0bad0ac4bf36ac196342e3ccf0ee353e63f4250a31bcaab797e6bf4275a4eb
-  data.tar.gz: a60120db35b707cd6e2678c88c89d702117a4473764708507106e9b42bd41e79090a6b533e977610363fae0e8b943c7f4483fe75492dcfbab70799354512ea9a
+  metadata.gz: b9375a72e7962e7945f4f8eff2ca994331617e4f4bb392c620510cf5c303795956c27f0a214712d6bb866478bb14d16fb5de52e434fdbee70094b6c6b99fe791
+  data.tar.gz: 33ea8a6147132f1aa1ccea87c94c25718a865628056bb3517d80453ae70a3b7ba80c5eefd2e8765b7edb9a67f93316ba3e71dae7f016295d6bffc9618ec5c694

data/README.md

@@ -37,7 +37,7 @@ $ rvm use ruby-3.1.1@pwn
 $ rvm list gemsets
 $ gem install --verbose pwn
 $ pwn
-pwn[v0.4.391]:001 >>> PWN.help
+pwn[v0.4.393]: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.1.1@pwn
 $ gem uninstall --all --executables pwn
 $ gem install --verbose pwn
 $ pwn
-pwn[v0.4.391]:001 >>> PWN.help
+pwn[v0.4.393]:001 >>> PWN.help
 ```
 
 

data/bin/pwn

@@ -7,10 +7,6 @@ require 'tty-prompt'
 require 'tty-reader'
 
 begin
-  class LineCount
-    attr_accessor :total
-  end
-
   class Pry
     # Overwrite Pry::History.push method in History class to get duplicate history entries
     # in order to properly replay automation in this prototyping driver
@@ -24,8 +20,6 @@ begin
           last_line = nil
         end
 
-        # return line if line == last_line
-
         @history << line
         @history_line_count += 1
         @saver.call(line) if !should_ignore?(line) &&
@@ -37,24 +31,20 @@ begin
     end
   end
 
-  prompt = TTY::Prompt.new
-  reader = TTY::Reader.new
-
   cli = Pry
+  cli.config.prompt_name = :pwn
+  cli.config.pwn_repl_line = 0
+  cli.config.tty_prompt = TTY::Prompt.new
+  cli.config.tty_reader = TTY::Reader.new
+
   cli.config.hooks.add_hook(:before_session, :welcome) do |output, _binding, _pry|
     output.puts 'Use the #help method for more options.'
+    output.puts 'e.g help'
     output.puts 'e.g PWN.help'
     output.puts 'e.g PWN::Plugins.help'
     output.puts 'e.g PWN::Plugins::TransparentBrowser.help'
   end
 
-  cli.config.hooks.add_hook(:before_eval, :autocomplete) do |output, _binding, _pry|
-    reader.on(:keypress) do |key_event|
-      output.puts "Key: #{key_event.inspect}"
-      prompt.select('', key_event)
-    end
-  end
-
   cli::Commands.create_command 'toggle-pager' do
     description 'Toggle less on returned objects surpassing the terminal.'
 
@@ -66,23 +56,26 @@ begin
   # Custom Main & Wait (Multi-Line) Prompts
   title = "\001\e[1m\002\001\e[31m\002pwn\001\e[0m\002" # Bold Red
   version = "\001\e[36m\002v#{PWN::VERSION}\001\e[0m\002" # Cyan
-  @line_count = LineCount.new
-  @line_count.total = 0
+  # @line = cli.config.custom_attrs[:line_number]
   arrow = "\001\e[32m\002>>>\001\e[0m\002" # Green
   splat = "\001\e[33m\002***\001\e[0m\002" # Yellow
 
-  # arrow_ps1_proc = proc do |_target_self, _nest_level, _pry|
-  arrow_ps1_proc = proc do
-    @line_count.total += 1
-    line_pad = format('%0.3d', @line_count.total)
+  arrow_ps1_proc = proc do |_target_self, _nest_level, pry|
+    pry.config.pwn_repl_line += 1
+    line_pad = format(
+      '%0.3d',
+      pry.config.pwn_repl_line
+    )
     line_count = "\001\e[34m\002#{line_pad}\001\e[0m\002" # Blue
     "#{title}[#{version}]:#{line_count} #{arrow} ".to_s.scrub
   end
 
-  # splat_ps1_proc = proc do |_target_self, _nest_level, _pry|
-  splat_ps1_proc = proc do
-    @line_count.total += 1
-    line_pad = format('%0.3d', @line_count.total)
+  splat_ps1_proc = proc do |_target_self, _nest_level, pry|
+    pry.config.pwn_repl_line += 1
+    line_pad = format(
+      '%0.3d',
+      pry.config.pwn_repl_line
+    )
     line_count = "\001\e[34m\002#{line_pad}\001\e[0m\002" # Blue
     "#{title}[#{version}]:#{line_count} #{splat} ".to_s.scrub
   end

data/lib/pwn/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.4.391
+  version: 0.4.393
 platform: ruby
 authors:
 - 0day Inc.
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-04-01 00:00:00.000000000 Z
+date: 2022-04-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport