Land #2266, @wchen-r7's patch forn [SeeRM #8345] and [SeeRM #8344]

rapid7 · Aug 26, 2013 · a58750f · a58750f
2 parents 6b15a07 + 6b8feaf
commit a58750f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/modules/auxiliary/gather/corpwatch_lookup_id.rb b/modules/auxiliary/gather/corpwatch_lookup_id.rb
@@ -30,7 +30,7 @@ def initialize(info = {})
 		register_options(
 			[
 				OptString.new('CW_ID', [ true, "The CorpWatch ID of the company", ""]),
-				OptString.new('YEAR', [ false, "Year to look up", ""]),
+				OptInt.new('YEAR', [ false, "Year to look up"]),
 				OptBool.new('GET_LOCATIONS', [ false, "Get locations for company", true]),
 				OptBool.new('GET_NAMES', [ false, "Get all registered names ofr the company", true]),
 				OptBool.new('GET_FILINGS', [ false, "Get all filings", false ]),
@@ -58,7 +58,7 @@ def run
 
 		loot = ""
 		uri = "/"
-		uri << (datastore['YEAR']) if datastore['YEAR'] != ""
+		uri << (datastore['YEAR']).to_s if datastore['YEAR'].to_s != ""
 		uri << ("/companies/" + datastore['CW_ID'])
 
 		res = send_request_cgi({

diff --git a/modules/auxiliary/gather/corpwatch_lookup_name.rb b/modules/auxiliary/gather/corpwatch_lookup_name.rb
@@ -33,7 +33,7 @@ def initialize(info = {})
 		register_options(
 			[
 				OptString.new('COMPANY_NAME', [ true, "Search for companies with this name", ""]),
-				OptString.new('YEAR', [ false, "Limit results to a specific year", ""]),
+				OptInt.new('YEAR', [ false, "Limit results to a specific year"]),
 				OptString.new('LIMIT', [ true, "Limit the number of results returned", "5"]),
 				OptString.new('CORPWATCH_APIKEY', [ false, "Use this API key when getting the data", ""]),
 			], self.class)
@@ -56,7 +56,7 @@ def run
 		datastore['RPORT'] = 80
 
 		uri = "/"
-		uri << (datastore['YEAR'] + "/") if datastore['YEAR'] != ""
+		uri << (datastore['YEAR'].to_s + "/") if datastore['YEAR'].to_s != ""
 		uri << "companies.xml"
 
 		res = send_request_cgi(