pwn 0.4.389 → 0.4.390

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8be720b98ca6c0e032826960b6635f881cec7eddc7880f39d3fe203c8f3f0bca
-  data.tar.gz: 51890215c46c839580f6b204d563a13ef71c95cfc979152e5e282c6c781b3426
+  metadata.gz: 1980f14b63483ec6b681ecbe3c97b18acfe587b73a3bb94cc08a6ff6cc9a940e
+  data.tar.gz: b8aedccf9888c441752213c20cf93281755c4e01a0e3902c7b00a5d1c57eeaf1
 SHA512:
-  metadata.gz: 94187e08a130c04bc15ff89ddffb84b1bcd5c493b19c3b45fd18954953feb9c0ab6a6ac59b71ddf01b50a2cf1c2b9c2433c2db2870c65d8cbe5a6c218207e143
-  data.tar.gz: 3a737824290a898e8655308d3d5f0278bc3f3faa3abf078ea331e995aed380871033d1219c9907b2055c5611a1815496373a98bf7ccc15abc69a8e9427b482d1
+  metadata.gz: ba09da802442ca21edf0ce271dbc68a81e30fa410e01c8d85f6c54e9dcc068702314d69215d048f87b26ba20d0cd886d8268c42f384dbaac3c6084c66932bd57
+  data.tar.gz: eb03adcd9751694fca6fe5023c3000ed1ec88158bf8236dc0a271a8e8bbe0deb3c95448b88c9babdddb916dcb5ee16111c689ccb065381030709e5b0dddcd48a

data/Gemfile

@@ -16,7 +16,6 @@ gem 'anemone', '0.7.2'
 gem 'authy', '3.0.0'
 gem 'aws-sdk', '3.1.0'
 gem 'bettercap', '1.6.2'
-gem 'bond', '0.5.1'
 gem 'brakeman', '5.2.1'
 gem 'bson', '4.14.1'
 gem 'bundler', '>=2.3.10'
@@ -48,7 +47,6 @@ gem 'packetfu', '1.1.13'
 gem 'pdf-reader', '2.9.2'
 gem 'pg', '1.3.5'
 gem 'pry', '0.14.1'
-gem 'pry-bond', '0.0.1'
 gem 'pry-doc', '1.3.0'
 gem 'rake', '13.0.6'
 gem 'rb-readline', '0.5.5'

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.389]:001 >>> PWN.help
+pwn[v0.4.390]: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.389]:001 >>> PWN.help
+pwn[v0.4.390]:001 >>> PWN.help
 ```
 
 

data/bin/pwn

@@ -3,13 +3,14 @@
 
 require 'pwn'
 require 'pry'
-require 'bond'
-require 'pry/bond'
 require 'tty-prompt'
 require 'tty-reader'
 
 begin
-  # TODO: Remove once this feature is available in mainline
+  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
@@ -36,12 +37,15 @@ begin
     end
   end
 
-  cli = Pry
+  prompt = TTY::Prompt.new
+  reader = TTY::Reader.new
 
-  cli.config.hooks.add_hook(:before_eval, :autocomplete) do
-    prompt = TTY::Prompt.new
-    reader = TTY::Reader.new
-    reader.on(:keypress) { |key_event| prompt(key_event) }
+  cli = Pry
+  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
@@ -55,22 +59,39 @@ 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
   arrow = "\001\e[32m\002>>>\001\e[0m\002" # Green
   splat = "\001\e[33m\002***\001\e[0m\002" # Yellow
 
-  prompt = [
-    proc do |_target_self, _nest_level, pry|
-      line_pad = format('%0.3d', pry.input_ring.size)
-      line_count = "\001\e[34m\002#{line_pad}\001\e[0m\002" # Blue
-      "#{title}[#{version}]:#{line_count} #{arrow} ".to_s.scrub
-    end,
-    proc do |_target_self, _nest_level, pry|
-      line_pad = format('%0.3d', pry.input_ring.size)
-      line_count = "\001\e[34m\002#{line_pad}\001\e[0m\002" # Blue
-      "#{title}[#{version}]:#{line_count} #{splat} ".to_s.scrub
-    end
-  ]
-  cli.start(self, prompt: Pry::Prompt.new(:pwn, 'PWN_PROTOTYPING_DRIVER', prompt))
+  # 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)
+    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)
+    line_count = "\001\e[34m\002#{line_pad}\001\e[0m\002" # Blue
+    "#{title}[#{version}]:#{line_count} #{splat} ".to_s.scrub
+  end
+
+  prompt_ps1 = [arrow_ps1_proc, splat_ps1_proc]
+
+  pwn_prompt = Pry::Prompt.new(
+    :pwn,
+    'PWN_PROTOTYPING_DRIVER',
+    prompt_ps1
+  )
+
+  cli.start(
+    self,
+    prompt: pwn_prompt
+  )
 rescue StandardError => e
   raise e
 end

data/lib/pwn/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.4.389
+  version: 0.4.390
 platform: ruby
 authors:
 - 0day Inc.
@@ -80,20 +80,6 @@ dependencies:
     - - '='
       - !ruby/object:Gem::Version
         version: 1.6.2
-- !ruby/object:Gem::Dependency
-  name: bond
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - '='
-      - !ruby/object:Gem::Version
-        version: 0.5.1
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - '='
-      - !ruby/object:Gem::Version
-        version: 0.5.1
 - !ruby/object:Gem::Dependency
   name: brakeman
   requirement: !ruby/object:Gem::Requirement
@@ -500,20 +486,6 @@ dependencies:
     - - '='
       - !ruby/object:Gem::Version
         version: 0.14.1
-- !ruby/object:Gem::Dependency
-  name: pry-bond
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - '='
-      - !ruby/object:Gem::Version
-        version: 0.0.1
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - '='
-      - !ruby/object:Gem::Version
-        version: 0.0.1
 - !ruby/object:Gem::Dependency
   name: pry-doc
   requirement: !ruby/object:Gem::Requirement