pwn 0.4.645 → 0.4.646

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 175a84d917a62dd16839be4dad757fa30c0db1ec4b8be44ced24412e34061cfd
-  data.tar.gz: 56ab72f479c87b3206310be28c1bff3e1ecc71332b82ffc921de7fb612b15139
+  metadata.gz: a3c91157de7e5f636150debf731605340089f894541296f328201b94e5d02d79
+  data.tar.gz: 2f4c1d07bd0e1b56c03589cd500864e5b39fdd886ca67bc1302d47827047507b
 SHA512:
-  metadata.gz: b166ba9ac93ff8cd2908be9078540ad7fd4e0076e2579e1dfa648fb14e6c0caf6d1b598d27f2d39c26d245e7768f85c9b4e02f1e530dd3b08fa541ad162792bb
-  data.tar.gz: 9c9677f7900576017382915e3a2928d0feeedea1e0e7ad4fccca3dcc2903c9d99a7f76dc3888d5529af471b1d018c96fd1d49ebad1df9776ebb6b1fb3f09d178
+  metadata.gz: a8b51af5abbf70b01832ad5b88ea8981cc4224afbbfa823f21e1cdc02ad51891625cd57de522940c37e71b141b29cc8096419f4166ed56323187cc24e0f4188e
+  data.tar.gz: 2b0db372b007bbef8dd18c7acc273cf2595d47bd9f1285b22c2146d4a589fa54e1ac0fefcd75d44f0bcf2930ec2c770dffd0fe2dd2c21ebae6a4c1b403ebd684

data/README.md

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

data/bin/pwn

@@ -1,12 +1,25 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: true
 
+require 'optparse'
 require 'pwn'
 require 'pry'
-require 'colorize'
+require 'yaml'
+# require 'colorize'
 # require 'tty-prompt'
 # require 'tty-reader'
 
+opts = {}
+OptionParser.new do |options|
+  options.banner = "USAGE:
+    #{$PROGRAM_NAME} [opts]
+  "
+
+  options.on('-cPATH', '--yaml-config=PATH', '<Optional - OpenAI YAML File>') do |p|
+    opts[:yaml_config_path] = p
+  end
+end.parse!
+
 begin
   def gen_ps1_proc(opts = {})
     delim = opts[:delim]
@@ -73,11 +86,89 @@ begin
     end
   end
 
+  Pry::Commands.create_command 'toggle-chatGPT-debug' do
+    description "Display the response_history object while using OpenAI's ChatGPT."
+
+    def process
+      pi = pry_instance
+      pi.config.chat_gpt_debug ? pi.config.chat_gpt_debug = false : pi.config.chat_gpt_debug = true
+      puts 'ChatGPT Debug Enabled.' if pi.config.chat_gpt_debug
+      puts 'ChatGPT Debug Disabled.' unless pi.config.chat_gpt_debug
+    end
+  end
+
+  Pry::Commands.create_command 'toggle-chatGPT-speaks' do
+    description 'Use speech capabilities within PWN to speak OpenAI ChatGPT answers.'
+
+    def process
+      pi = pry_instance
+      pi.config.chat_gpt_speak ? pi.config.chat_gpt_speak = false : pi.config.chat_gpt_speak = true
+      puts 'ChatGPT Speech Enabled.' if pi.config.chat_gpt_speak
+      puts 'ChatGPT Speech Disabled.' unless pi.config.chat_gpt_speak
+    end
+  end
+
+  Pry::Commands.create_command 'toggle-chatGPT' do
+    description "Interact w/ OpenAI's ChatGPT"
+
+    def process
+      pi = pry_instance
+      pi.config.chat_gpt ? pi.config.chat_gpt = false : pi.config.chat_gpt = true
+      puts 'ChatGPT Enabled.' if pi.config.chat_gpt
+      puts 'ChatGPT Disabled.' unless pi.config.chat_gpt
+    end
+  end
+
   # Define REPL Hooks
+  # Welcome Banner Hook
   Pry.config.hooks.add_hook(:before_session, :welcome) do |output, _binding, _pry|
     output.puts PWN::Banner.welcome
   end
 
+  # ChatGPT Hooks
+  Pry.config.hooks.add_hook(:before_session, :init_opts) do |_output, _binding, pi|
+    pi.config.chat_gpt_token = ''
+    if opts[:yaml_config_path] && File.exist?(opts[:yaml_config_path])
+      yaml_config_path = opts[:yaml_config_path]
+      yaml_config = YAML.load_file(yaml_config_path, symbolize_names: true)
+      pi.config.chat_gpt_token = yaml_config[:bearer_token]
+    end
+  end
+
+  Pry.config.hooks.add_hook(:after_eval, :open_ai_hook) do |request, pi|
+    if request.instance_of?(String) && pi.config.chat_gpt
+      debug = pi.config.chat_gpt_debug
+      token = pi.config.chat_gpt_token
+      token = PWN::Plugins::AuthenticationHelper.mask_password(prompt: 'OpenAI API Key') if token.empty?
+      response_history = pi.config.chat_gpt_response_history
+      speak_answer = pi.config.chat_gpt_speak
+      response = PWN::Plugins::OpenAI.chat(
+        token: token,
+        request: request.to_s,
+        temp: 1,
+        max_tokens: 0,
+        response_history: response_history,
+        speak_answer: speak_answer
+      )
+      puts "\n\n\n#{response[:choices].last[:content]}\n\n\n"
+
+      response_history = {
+        id: response[:id],
+        object: response[:object],
+        model: response[:model],
+        usage: response[:usage]
+      }
+      response_history[:choices] ||= response[:choices]
+
+      if debug
+        puts 'DEBUG: response_history = '
+        pp response_history
+        puts "response_history[:choices] Length: #{response_history[:choices].length}\n" unless response_history.nil?
+      end
+      pi.config.chat_gpt_response_history = response_history
+    end
+  end
+
   # Define PS1 Prompt
   Pry.config.pwn_repl_line = 0
   arrow_ps1_proc = gen_ps1_proc

data/lib/pwn/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.4.645
+  version: 0.4.646
 platform: ruby
 authors:
 - 0day Inc.
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-04-18 00:00:00.000000000 Z
+date: 2023-04-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -1091,7 +1091,6 @@ executables:
 - pwn_char_html_entity_encoding
 - pwn_char_unicode_escaped_encoding
 - pwn_char_url_encoding
-- pwn_chat
 - pwn_defectdojo_engagement_create
 - pwn_defectdojo_importscan
 - pwn_defectdojo_reimportscan
@@ -1158,7 +1157,6 @@ files:
 - bin/pwn_char_html_entity_encoding
 - bin/pwn_char_unicode_escaped_encoding
 - bin/pwn_char_url_encoding
-- bin/pwn_chat
 - bin/pwn_defectdojo_engagement_create
 - bin/pwn_defectdojo_importscan
 - bin/pwn_defectdojo_reimportscan

data/bin/pwn_chat

@@ -1,173 +0,0 @@
-#!/usr/bin/env ruby
-# frozen_string_literal: true
-
-require 'optparse'
-require 'pwn'
-require 'pry'
-require 'yaml'
-
-opts = {}
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{$PROGRAM_NAME} [opts]
-  "
-
-  options.on('-cPATH', '--yaml-config=PATH', '<Required - OpenAI YAML File>') do |p|
-    opts[:yaml_config_path] = p
-  end
-
-  options.on('-d', '--[no-]debug', '<Options - Display response_history Object During Session>') do |d|
-    opts[:debug] = d
-  end
-
-  options.on('-sSTAGE', '--system-role-content=STAGE', '<Optional - system Role Content Value to Define Behavior of assistant responses (Defaults to value in PWN::Plugins::OpenAI.chat method)>') do |s|
-    opts[:system_role_content] = s
-  end
-
-  options.on('-S', '--speak-answer', '<Options - Speak Answers (Defaults to false)>') do |v|
-    opts[:speak_answer] = v
-  end
-end.parse!
-
-if opts.empty?
-  puts `#{$PROGRAM_NAME} --help`
-  exit 1
-end
-
-begin
-  def gen_ps1_proc(opts = {})
-    delim = opts[:delim]
-
-    # title = 'pwn'.red.bold
-    title = "\001\e[1m\002\001\e[31m\002#{File.basename($PROGRAM_NAME)}\001\e[0m\002"
-    # version = PWN::VERSION.cyan
-    version = "\001\e[36m\002v#{PWN::VERSION}\001\e[0m\002"
-    # dchars = '>>>'.green
-    dchars = "\001\e[32m\002>>>\001\e[0m\002"
-    # dchars = '***'.yellow if delim == :splat
-    dchars = "\001\e[33m\002***\001\e[0m\002" if delim == :splat
-
-    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} #{dchars} ".to_s.scrub
-    end
-  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
-    class History
-      def push(line)
-        return line if line.empty? || invalid_readline_line?(line)
-
-        begin
-          last_line = @history[-1]
-        rescue IndexError
-          last_line = nil
-        end
-
-        @history << line
-        @history_line_count += 1
-        @saver.call(line) if !should_ignore?(line) &&
-                             Pry.config.history_save
-
-        line
-      end
-      alias << push
-    end
-  end
-
-  # Get OptParse Cli Parameters
-  yaml_config_path = opts[:yaml_config_path]
-  raise "ERROR: YAML Config => #{yaml_config_path} not found." unless File.exist?(yaml_config_path)
-
-  yaml_config = YAML.load_file(yaml_config_path, symbolize_names: true)
-  token = yaml_config[:bearer_token]
-
-  debug = opts[:debug]
-
-  system_role_content = opts[:system_role_content]
-
-  @speak_answer = true if opts[:speak_answer]
-
-  # Define Custom REPL Commands
-  Pry::Commands.create_command 'welcome-banner' do
-    description 'Display the random welcome banner, including basic usage.'
-
-    def process
-      puts PWN::Banner.welcome
-    end
-  end
-
-  Pry::Commands.create_command 'toggle-pager' do
-    description 'Toggle less on returned objects surpassing the terminal.'
-
-    def process
-      pi = pry_instance
-      pi.config.pager ? pi.config.pager = false : pi.config.pager = true
-    end
-  end
-
-  # Define REPL Hooks
-  Pry.config.hooks.add_hook(:before_session, :welcome) do |output, _binding, _pry|
-    output.puts PWN::Banner.welcome
-  end
-
-  @response_history = nil
-  Pry.config.hooks.add_hook(:after_eval, :open_ai_hook) do |request, _pry|
-    if request.instance_of?(String)
-      response = PWN::Plugins::OpenAI.chat(
-        token: token,
-        system_role_content: system_role_content,
-        request: request.to_s,
-        temp: 1,
-        max_tokens: 0,
-        response_history: @response_history,
-        speak_answer: @speak_answer
-      )
-      puts "\n\n\n#{response[:choices].last[:content]}\n\n\n"
-
-      @response_history = {
-        id: response[:id],
-        object: response[:object],
-        model: response[:model],
-        usage: response[:usage]
-      }
-      @response_history[:choices] ||= response[:choices]
-    end
-  end
-
-  if debug
-    Pry.config.hooks.add_hook(:after_eval, :open_ai_hook_resp) do |_request, _pry|
-      puts 'DEBUG: @response_history = '
-      pp @response_history
-      puts "@response_history[:choices] Length: #{@response_history[:choices].length}\n" unless @response_history.nil?
-    end
-  end
-
-  # Define PS1 Prompt
-  Pry.config.pwn_repl_line = 0
-  arrow_ps1_proc = gen_ps1_proc
-  splat_ps1_proc = gen_ps1_proc(delim: :splat)
-  prompt_ps1 = [arrow_ps1_proc, splat_ps1_proc]
-
-  pwn_prompt = Pry::Prompt.new(
-    :pwn_chat,
-    'PWN Prototyping REPL w/ OpenAI Assistant',
-    prompt_ps1
-  )
-
-  # Start PWN REPL
-  Pry.config.prompt_name = :pwn_chat
-  Pry.start(
-    self,
-    prompt: pwn_prompt
-  )
-rescue StandardError => e
-  raise e
-end