Merge pull request #223 from ninp0/master

PWN::Plugins::PWNLogger - add logging verbosity level parameter in #c…

Author: ninp0

Gemfile

@@ -18,7 +18,7 @@ gem 'aws-sdk', '3.1.0'
 gem 'bettercap', '1.6.2'
 gem 'brakeman', '5.4.1'
 gem 'bson', '4.15.0'
-gem 'bundler', '>=2.4.8'
+gem 'bundler', '>=2.4.9'
 gem 'bundler-audit', '0.9.1'
 gem 'bunny', '2.20.3'
 gem 'colorize', '0.8.1'
@@ -82,7 +82,7 @@ gem 'sqlite3', '1.6.1'
 gem 'thin', '1.8.1'
 gem 'tty-prompt', '0.23.1'
 gem 'watir', '7.2.2'
-gem 'waveform', '0.1.2'
+gem 'waveform', '0.1.3'
 gem 'webrick', '1.8.1'
 gem 'whois', '5.1.0'
 gem 'whois-parser', '2.0.0'

README.md

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

lib/pwn/plugins/http_intercept_helper.rb

@@ -10,10 +10,26 @@ module PWNLogger
       # PWN::Plugins::PWNLogger.create(
       # )
 
-      public_class_method def self.create
+      public_class_method def self.create(opts = {})
         logger = Logger.new($stdout)
-        logger.level = Logger::INFO
-        logger.datetime_format = '%Y-%m-%d %H:%M:%S'
+        level = opts[:level]
+
+        case level.to_s.downcase.to_sym
+        when :debug
+          logger.level = Logger::DEBUG
+        when :error
+          logger.level = Logger::ERROR
+        when :fatal
+          logger.level = Logger::FATAL
+        when :unknown
+          logger.level = Logger::UNKNOWN
+        when :warn
+          logger.level = Logger::WARN
+        else
+          logger.level = Logger::INFO
+        end
+
+        logger.datetime_format = '%Y-%m-%d %H:%M:%S.%N'
 
         logger.formatter = proc do |severity, _datetime, _progname, msg|
           # TODO: Include datetime & progname vars
@@ -37,8 +53,10 @@ module PWNLogger
 
       public_class_method def self.help
         puts "USAGE:
-          logger = #{self}.create()
-         #{self}.authors
+          logger = #{self}.create(
+            level: 'optional - logging verbosity :debug|:error|:fatal|:info|:unknown|:warn (Defaults to :info)'
+          )
+          #{self}.authors
         "
       end
     end

lib/pwn/plugins/pwn_logger.rb

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

lib/pwn/version.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe PWN::Plugins::AnsibleVault do
+  it 'should display information for authors' do
+    authors_response = PWN::Plugins::AnsibleVault
+    expect(authors_response).to respond_to :authors
+  end
+
+  it 'should display information for existing help method' do
+    help_response = PWN::Plugins::AnsibleVault
+    expect(help_response).to respond_to :help
+  end
+end

spec/lib/pwn/plugins/ansible_vault_spec.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe PWN::Plugins::BareSIP do
+  it 'should display information for authors' do
+    authors_response = PWN::Plugins::BareSIP
+    expect(authors_response).to respond_to :authors
+  end
+
+  it 'should display information for existing help method' do
+    help_response = PWN::Plugins::BareSIP
+    expect(help_response).to respond_to :help
+  end
+end

spec/lib/pwn/plugins/baresip_spec.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe PWN::Plugins::OpenAI do
+  it 'should display information for authors' do
+    authors_response = PWN::Plugins::OpenAI
+    expect(authors_response).to respond_to :authors
+  end
+
+  it 'should display information for existing help method' do
+    help_response = PWN::Plugins::OpenAI
+    expect(help_response).to respond_to :help
+  end
+end

spec/lib/pwn/plugins/openai_spec.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe PWN::Plugins::PWNLogger do
+  it 'should display information for authors' do
+    authors_response = PWN::Plugins::PWNLogger
+    expect(authors_response).to respond_to :authors
+  end
+
+  it 'should display information for existing help method' do
+    help_response = PWN::Plugins::PWNLogger
+    expect(help_response).to respond_to :help
+  end
+end

spec/lib/pwn/plugins/pwn_logger_spec.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe PWN::Plugins::Voice do
+  it 'should display information for authors' do
+    authors_response = PWN::Plugins::Voice
+    expect(authors_response).to respond_to :authors
+  end
+
+  it 'should display information for existing help method' do
+    help_response = PWN::Plugins::Voice
+    expect(help_response).to respond_to :help
+  end
+end