pwn 0.4.642 → 0.4.643

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e64877191025eecb6ff055c5b5ef451cb52f01502aa50b0ee6ef6e8f3fef9a2a
-  data.tar.gz: 557826a99958976edcd69803aef9fb73e88ccba55c919bf0854755b705b1a4cb
+  metadata.gz: bbf3fa66db2d9b1b7d8be45081cf1d2e733fb6b2caa0b3fe7c092790c2a4b43a
+  data.tar.gz: 3b70ba7c9096251aec921f1ed72d109c50ccaaaedea256bb0d700a69ea8fbd4b
 SHA512:
-  metadata.gz: ec85372b9cebb045568c79593facdd7d1182dc700c0586acc32b2bc617f6c0dd18eeea6118032e8e63159601db36d97d103fd32e123380552558ebb909a05c88
-  data.tar.gz: 66f5e4ca76b2ab383e85b501312139a0c53af1ba2f3218b8e1b0bbb73d416e9e56442ce3275e24b1045d922cbc3e2df902bbd355eb1c9639784c5410134a654f
+  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.642]: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.642]:001 >>> PWN.help
+pwn[v0.4.643]:001 >>> PWN.help
 ```
 
 

data/bin/pwn_chat

@@ -119,36 +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"
-
-    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
+    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].slice(-6..)
-    @response_history[:choices] ||= response[:choices]
+        @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.642'
+  VERSION = '0.4.643'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.4.642
+  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