pwn 0.4.641 → 0.4.643

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5c3b134c9581e5a978bd2c8c8ce3fe09d277ece0818a97e89f04a69357709a69
-  data.tar.gz: f683e628f3cd920bb11e94a3de8c0137f6b94e6a4b55bfb8081fd350f01cc9ff
+  metadata.gz: bbf3fa66db2d9b1b7d8be45081cf1d2e733fb6b2caa0b3fe7c092790c2a4b43a
+  data.tar.gz: 3b70ba7c9096251aec921f1ed72d109c50ccaaaedea256bb0d700a69ea8fbd4b
 SHA512:
-  metadata.gz: 6a934a8a8f2a9a6c0e6cd598fe3f93c5ead21c654166325331c5aed46afdee3eae10b1020ac5a4c87f49b166a4f74fa534730af3795953fddf1a6778b6cdb00f
-  data.tar.gz: b0f2371b0f80cfcdd9ba31e554d1907db8fe6a9b8d5b75c717244944e37838c4baaa5c7d180529d4eacbadf1b462176c1396aec3b155f1db6408154333ab3fae
+  metadata.gz: 0a738aa1e3c60bd364b50df32e25cbd34dcdc23e446fbf0f91f53fb2e310ea88ca0f36e986bbe2a51bb091ccf325e16cb2581e4ed75e8befaf42ae2671687ba0
+  data.tar.gz: 8e30cbbcb5dcc2a109df22e979ff6e013f578a7140d70ee9e17af0d9c0e9a4f869a7dc5da065543b2cd710751accbd5e11d172fca426a5b24f75052f62f5921d

data/Gemfile

@@ -66,13 +66,13 @@ gem 'rspec', '3.12.0'
 gem 'rtesseract', '3.1.2'
 gem 'rubocop', '1.50.2'
 gem 'rubocop-rake', '0.6.0'
-gem 'rubocop-rspec', '2.19.0'
+gem 'rubocop-rspec', '2.20.0'
 gem 'ruby-audio', '1.6.1'
 gem 'ruby-nmap', '1.0.1'
 gem 'ruby-saml', '1.15.0'
 gem 'rvm', '1.11.3.9'
 gem 'savon', '2.14.0'
-gem 'selenium-devtools', '0.111.0'
+gem 'selenium-devtools', '0.112.0'
 gem 'serialport', '1.3.2'
 gem 'sinatra', '3.0.6'
 gem 'slack-ruby-client', '2.1.0'

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

data/bin/pwn_chat

@@ -23,6 +23,10 @@ OptionParser.new do |options|
   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('-v', '--voice', '<Options - Voice Activate Responses (Defaults to false)>') do |v|
+    opts[:voice] = v
+  end
 end.parse!
 
 if opts.empty?
@@ -89,6 +93,8 @@ begin
 
   system_role_content = opts[:system_role_content]
 
+  @voice = true if opts[:voice]
+
   # Define Custom REPL Commands
   Pry::Commands.create_command 'welcome-banner' do
     description 'Display the random welcome banner, including basic usage.'
@@ -113,30 +119,68 @@ begin
   end
 
   @response_history = nil
+  @keep_in_memory = 0
   Pry.config.hooks.add_hook(:after_eval, :open_ai_hook) do |request, _pry|
-    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
-    )
-    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].slice(-6..)
-    @response_history[:choices] ||= response[:choices]
+    if request.instance_of?(String)
+      unless @response_history.nil?
+        choices_len = @response_history[:choices].length
+        @keep_in_memory = choices_len * -1
+      end
+
+      begin
+        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
+        )
+        puts "\n\n\n#{response[:choices].last[:content]}\n\n\n"
+
+        if @voice
+          text_path = "/tmp/#{File.basename($PROGRAM_NAME)}.response"
+          File.write(text_path, response[:choices].last[:content])
+          PWN::Plugins::Voice.text_to_speech(text_path: text_path)
+        end
+
+        @response_history = {
+          id: response[:id],
+          object: response[:object],
+          model: response[:model],
+          usage: response[:usage]
+        }
+        @response_history[:choices] ||= response[:choices]
+      rescue JSON::ParserError
+        max_tokens = 4097
+        if request.length > max_tokens
+          puts "Request Length Too Long: #{request.length}\n"
+        else
+          puts 'Token Length too long, trimming memory by 1 and retrying...'
+          @keep_in_memory += 1
+          @response_history[:choices] = @response_history[:choices].slice(@keep_in_memory..)
+
+          response = PWN::Plugins::OpenAI.chat(
+            token: token,
+            system_role_content: system_role_content,
+            request: "summarize what we've already discussed",
+            temp: 1,
+            max_tokens: 0,
+            response_history: @response_history
+          )
+          @response_history[:choices] = response[:choices].slice(@keep_in_memory..)
+
+          retry
+        end
+      end
+    end
   end
 
   if debug
     Pry.config.hooks.add_hook(:after_eval, :open_ai_hook_resp) do |_request, _pry|
-      puts @response_history
+      puts 'DEBUG: @response_history = '
+      pp @response_history
+      puts "@response_history[:choices] Length: #{@response_history[:choices].length}\n" unless @response_history.nil?
     end
   end
 

data/lib/pwn/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.4.641
+  version: 0.4.643
 platform: ruby
 authors:
 - 0day Inc.
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-04-17 00:00:00.000000000 Z
+date: 2023-04-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -786,14 +786,14 @@ dependencies:
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 2.19.0
+        version: 2.20.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 2.19.0
+        version: 2.20.0
 - !ruby/object:Gem::Dependency
   name: ruby-audio
   requirement: !ruby/object:Gem::Requirement
@@ -870,14 +870,14 @@ dependencies:
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 0.111.0
+        version: 0.112.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 0.111.0
+        version: 0.112.0
 - !ruby/object:Gem::Dependency
   name: serialport
   requirement: !ruby/object:Gem::Requirement