pwn_shodan_search - intial working commit #bugfix

Author: ninp0

README.md

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

bin/pwn_shodan_search

@@ -20,7 +20,7 @@ OptionParser.new do |options|
     opts[:query_file] = q
   end
 
-  options.on('-oFILE', '--output-results-file=FILE', '<Optional - Defaults to /tmp/shodan-results-Time.now.strftime("%y-%m-%d.%H:%M:%S")>') do |o|
+  options.on('-oFILE', '--output-results-file=FILE', '<Optional - Defaults to /tmp/shodan-results-Time.now.strftime("%y-%m-%d.%H:%M:%S").txt>') do |o|
     opts[:output_results_file] = o
   end
 end.parse!
@@ -31,23 +31,24 @@ if opts.empty?
 end
 
 begin
-  yaml_file = opts[:yaml]
+  yaml_file = opts[:yaml].to_s
   raise "ERROR: #{yaml_file} does not exist." unless File.exist?(yaml_file)
 
   yaml = YAML.load_file(yaml_file, symbolize_names: true)
 
   api_key = yaml[:api_key]
 
-  query_file = opts[:query_file]
+  query_file = opts[:query_file].to_s
   raise "ERROR: #{query_file} does not exist." unless File.exist?(query_file)
 
   queries = File.readlines(query_file)
 
   timestamp = Time.now.strftime('%Y-%m-%d.%H:%M:%S')
   query_results_file = opts[:output_results_file]
   query_results_file ||= "/tmp/shodan-results-#{timestamp}.txt"
+  query_results_dir = File.dirname(query_results_file)
 
-  raw_query_results_file = "/tmp/shodan-results-#{timestamp}-RAW.json"
+  raw_query_results_file = "#{query_results_dir}/shodan-results-#{timestamp}-RAW.json"
   File.open(raw_query_results_file, 'w') do |r|
     File.open(query_results_file, 'w') do |f|
       queries.each do |query_line|

lib/pwn/version.rb

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