uirusu 0.0.5 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: cd1b2173bea70412a787d51e6968bfefe2a5af9b
-  data.tar.gz: 9c00e08d2ccc82b620f7ba1670f4c0c2b30f35c6
+  metadata.gz: bc132e4e12a2eb719f8141899b88c13f78fbd7de
+  data.tar.gz: 76e35c93a4ba875b471fa5713a47dffa84fa4f26
 SHA512:
-  metadata.gz: 83754cc68a7631fded5d617588239bcb5928ccd7f536bfff6c0ec46644ec85c4b04d46e77a847aef182f0d2ee3fa2b503665e0e2a07550b700de315029dbdd93
-  data.tar.gz: ffbe2e9e701597b5b47384f076c1b7b993a90c937a150f6278172627543ce8f70e56b3d2a61e4995c17938d8b315a51403d5b77a51711176c77444e805a9f350
+  metadata.gz: 0546700722bcfae8842c4b94784b179d8a3b44d314c93715830ac94a8639fea190f1d6468a134ac5c094b64d7361131bda4bfa8f3b5c4803ce47952b050fc2b9
+  data.tar.gz: 018ff9713f99141ff6240ff0d23ece1890176077d774cb0bb5cd23f06748eb9129a8a2e298952fa435ad81c2b06e9117ab6a021acad0731b085f863dae684b56

data/NEWS.markdown

@@ -1,5 +1,11 @@
 # News
 
+# 0.0.6 (September)
+- Added support for hashing a directory and submitting it to the hash scan[request from myne-us]
+	- -d DIRECTORY will invoke this, all files will be hashed and submitted to the hash array to be hashed
+- Minor tweaks
+	- Fixed the lack of a hash on 'file not found' results
+
 # 0.0.5 (June 14, 2013)
 - Merged Pull request from [jfx41]
 - Lots of cleanup from jfx41

data/README.markdown

@@ -2,14 +2,13 @@
 
 uirusu is an [Virustotal](http://www.virustotal.com) automation and convenience tool for hash, file and URL submission.
 
-The current version is 0.0.4.
+The current version is 0.0.6.
 
 ## Requirements
 
-* ruby
+* ruby 1.9+
 * json
 * rest-client
-
 * **public api key from [virustotal.com](http://www.virustotal.com)**
 
 ## Installation
@@ -17,6 +16,14 @@ The current version is 0.0.4.
 	% gem install uirusu
 	% uirusu [options]
 
+## Setup is fairly easy
+
+### Create your configuration file
+	% uirusu --create-config
+
+### Edit your configuration file with API key
+	% $EDITOR ~/.uirusu
+
 ## Usage
 
 ### Searching a file of hashes
@@ -39,6 +46,8 @@ The current version is 0.0.4.
 ### Saving results to a file
 	% uirusu -s "http://www.google.com" --yaml-output > file.yaml
 
+### Scan a directory and have them searched and save the results as json
+	% uirusu -d /bin/ --json-output > file.json
 
 ## API Usage
 ```ruby

data/Rakefile

@@ -37,7 +37,15 @@ task :build do
 	system "gem build #{Uirusu::APP_NAME}.gemspec"
 end
 
-task :release => :build do
+task :tag_and_bag do
+	system "git tag -a v#{Uirusu::VERSION} -m 'version #{Uirusu::VERSION}'"
+	system "git push --tags"
+	system "git checkout master"
+	system "git merge #{Uirusu::VERSION}"
+	system "git push"
+end
+
+task :release => [:tag_and_bag, :build] do
 	system "gem push #{Uirusu::APP_NAME}-#{Uirusu::VERSION}.gem"
 	puts "Just released #{Uirusu::APP_NAME} v#{Uirusu::VERSION}. #{Uirusu::APP_NAME} is rubygem for using the Virustotal web service! More information at http://arxopia.com/projects/uirusu/"
 end

data/TODO.markdown

@@ -2,5 +2,5 @@
 
 **Release dates are estimates, and features can be changed at any time.**
 
-- TESTS!!!!
 - Better docs
+- Add risu report generation support [feature request from myne-us]

data/lib/uirusu.rb

@@ -28,10 +28,10 @@
 
 module Uirusu
 	APP_NAME = "uirusu"
-	VERSION = "0.0.5"
-	CONFIG_FILE = "~/.uirusu"
+	VERSION = "0.0.6"
+	CONFIG_FILE = Dir.home + "/.uirusu"
 	VT_API = "https://www.virustotal.com/vtapi/v2"
-	RESULT_FIELDS = [ :hash, :scanner, :version, :detected, :result, :md5, :sha1, :sha256, :update, :permalink, ]
+	RESULT_FIELDS = [ :hash, :scanner, :version, :detected, :result, :md5, :sha1, :sha256, :update, :permalink]
 end
 
 require 'json'
@@ -43,4 +43,5 @@ require 'uirusu/vtfile'
 require 'uirusu/vturl'
 require 'uirusu/vtcomment'
 require 'uirusu/vtresult'
+require 'uirusu/scanner'
 require 'uirusu/cli/application'

data/lib/uirusu/cli/application.rb

@@ -50,6 +50,7 @@ module Uirusu
 					@options['verbose'] = false
 					@options['rescan']  = false
 					@options[:timeout]  = 25
+					@options[:directory] = nil
 
 					opt = OptionParser.new do |opt|
 						opt.banner = "#{APP_NAME} v#{VERSION}\nJacob Hammack\nhttp://www.arxopia.com\n\n"
@@ -127,6 +128,10 @@ module Uirusu
 							end
 						end
 
+						opt.on('-d DIRECTORY', '--directory', 'Scans a directory recursively for files and submits the hashes') do |directory|
+							@options[:directory] = directory
+						end
+
 						opt.on('-p PROXY', '--proxy-server', 'Uses a specified proxy server') do |proxy|
 							@options['proxy'] = proxy
 						end
@@ -274,6 +279,14 @@ module Uirusu
 					RestClient.proxy = @options['proxy']
 				end
 
+				if @options[:directory] != nil
+					hashes = Uirusu::Scanner.scan(@options[:directory])
+
+					hashes.each do |hash|
+						@hashes.push hash
+					end
+				end
+
 				if @files_of_hashes != nil
 					@files_of_hashes.each do |file|
 						f = File.open(file, 'r')

data/lib/uirusu/scanner.rb

@@ -0,0 +1,74 @@
+# Copyright (c) 2012-2013 Arxopia LLC.
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+# Redistributions of source code must retain the above copyright notice,
+# this list of conditions and the following disclaimer.
+#
+# Redistributions in binary form must reproduce the above copyright notice,
+# this list of conditions and the following disclaimer in the documentation
+# and/or other materials provided with the distribution.
+#
+# Neither the name of the project's author nor the names of its contributors
+# may be used to endorse or promote products derived from this software
+# without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 'AS IS' AND
+# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY
+# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+require 'pathname'
+require 'digest/md5'
+
+module Uirusu
+
+	module Scanner
+		@hash_list = Array.new
+
+		# Recursively lists all files in a directory
+		# calling process_file on each file
+		#
+		def Scanner.recurse (file_name)
+			Dir.new("#{file_name}").each do |file|
+				next if file.match(/^\.+/)
+				path = "#{file_name}/#{file}"
+
+				if  FileTest.directory?("#{path}")
+					recurse("#{path}")
+				else
+					process_file(path)
+				end
+			end
+		end
+
+		# Processes a file, hashing it with MD5
+		#
+		def Scanner.process_file (file)
+			begin
+				digest = Digest::MD5.hexdigest(File.read(file))
+				@hash_list << digest
+
+			rescue Exception => e
+				puts "[!] Cannot read #{file}"
+			end
+		end
+
+		# Enumerates a directory recursively then returns the hash list
+		#
+		# @return [Array] Hash List
+		def Scanner.scan directory
+			recurse(directory)
+
+			return @hash_list
+		end
+	end
+end

data/lib/uirusu/vtresult.rb

@@ -53,6 +53,7 @@ module Uirusu
 				if result['response_code'] == 0
 					res = Hash.new
 					RESULT_FIELDS.each{|field| res[field] = '-' }
+					res[:hash] = hash
 					res['result'] = result['verbose_msg']
 					@results.push res
 
@@ -64,7 +65,7 @@ module Uirusu
 					md5 = result['md5']
 					sha1 = result['sha1']
 					sha256 = result['sha256']
-	
+
 					result['scans'].each do |scanner, value|
 						if value != ''
 							res = Hash.new
@@ -75,16 +76,16 @@ module Uirusu
 							res[:scanner] = scanner
 							res[:detected] = value['detected']
 							res[:version] = value['version']
-	
+
 							if value['result'] == nil
 								res[:result] = "Nothing detected"
 							else
 								res[:result] = value['result']
 							end
-	
+
 							res[:update] = value['update']
 							res[:permalink] = permalink unless permalink == nil
-	
+
 							@results.push res
 						end
 					end
@@ -95,13 +96,15 @@ module Uirusu
 			if @results.size == 0
 				res = Hash.new
 				RESULT_FIELDS.each{|field| res[field] = '-' }
+				res[:hash] = hash
 				res['result'] = result['verbose_msg']
 				@results.push res
 			end
 		end
 
+		# Outputs the result to STDOUT
 		#
-		#
+		# @return [String] Pretty text printable representation of the result
 		def to_stdout
 			result_string = String.new
 			hashes = Array.new
@@ -117,20 +120,23 @@ module Uirusu
 			result_string
 		end
 
+		# Outputs the result to JSON
 		#
-		#
+		# @return [String] JSON representation of the result
 		def to_json
 			JSON::pretty_generate(@results.map{|entry| { :vtresult => entry } })
 		end
 
+		# Outputs the result to YAML
 		#
-		#
+		# @return [String] YAML representation of the result
 		def to_yaml
 			@results.map{|entry| { :vtresult => entry } }.to_yaml
 		end
 
+		# Outputs the result to XML
 		#
-		#
+		# @return [String] XML representation of the result
 		def to_xml
 			result_string = String.new
 			result_string << "<results>\n"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: uirusu
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors:
 - Jacob Hammack
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-06-14 00:00:00.000000000 Z
+date: 2013-09-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json
@@ -57,6 +57,7 @@ files:
 - lib/uirusu.rb
 - lib/uirusu/cli/application.rb
 - lib/uirusu/vtresult.rb
+- lib/uirusu/scanner.rb
 - lib/uirusu/vtcomment.rb
 - lib/uirusu/vtfile.rb
 - lib/uirusu/vturl.rb
@@ -82,7 +83,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: 1.8.16
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.0.0
+rubygems_version: 2.0.3
 signing_key: 
 specification_version: 4
 summary: uirusu