fofa 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9bf29364f1e2529bff9a522f6b9d0bea1b83ecc2
-  data.tar.gz: 3a56e6bf2bd1cf675325250d40079bbf8f82ab3a
+  metadata.gz: d3fd620a7b7d85bd240669ffec7ebeaebdc19985
+  data.tar.gz: 03faad6a1352c771e81714ca41984a2c0b208445
 SHA512:
-  metadata.gz: dbf1563ffd7138481a82f8c473bf9632641da7ebecabbaf13e123b2918e4668fea57eef04481288a2a173051c3f192ded62e65fd8ce5fb64d5bcc239dab136df
-  data.tar.gz: 1c44cfbf1fabee47384ae686b57778d4a8d4e10ff625db37644beb70e29d6df08af659dceeefe4bcedf6260948c669698ded6164b5436191f3557a64d3c4b16e
+  metadata.gz: e1b428d99107084eb27a6809a80555c7051d0f32a8acb3e70b7f9d67732d516cc264492cf2d3b6ed59f43d1cc89a4464de8971424dbc501880eb7dbba082b5e0
+  data.tar.gz: 91b398f986f0933dd76a650f0b26a6d2ebbb9e9df1a4a02b9d6505cccea4d18f80396e95545f9e5cff226a41b52477c4ae5332674d48714bde257c224419c8ef

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    fofa (0.2.0)
+    fofa (0.3.0)
 
 GEM
   remote: https://rubygems.org/

data/lib/fofa.rb

@@ -12,15 +12,15 @@ module Fofa
   		@apikey = apikey
   	end
 
-  	# Search from fofa
+  	# Search page results from fofa
   	#
   	# Example:
-  	#   >> Fofa::API.new(email,apikey).search("host=baidu.com")
+  	#   >> Fofa::API.new(email,apikey).search("host==baidu.com")
   	#   => {size:1, results:['1.1.1.1:80']}
   	#
   	# Arguments:
   	#   query: (String) fofa query string
-  	#   options: (Hash) page
+  	#   options: (Hash) page(default 1, only fetch one page)
   	def search(query, options={})
         options = {page:1}.merge(options)
     		url = "#{@api_server}/api/v1/search/all?key=#{@apikey}&email=#{@email}&q=#{URI.escape(query)}&page=#{options[:page]}"
@@ -35,7 +35,31 @@ module Fofa
         JSON.parse(resp.body)
     rescue => e
     	{"error"=>"Error: #{e.to_s}"}
-  	end
+    end
+
+    # Search all results from fofa
+    #
+    # Example:
+    #   >> Fofa::API.new(email,apikey).search_all("domain==baidu.com") {|results, page| ... }
+    #   => {size:1, results:['1.1.1.1:80']}
+    #
+    # Arguments:
+    #   query: (String) fofa query string
+    def search_all(query)
+      all_res = []
+      page = 0
+      loop do
+        page += 1
+        res = search(query, {page:page})
+        if res['results'].size > 0
+          all_res += res['results']
+          yield(res['results'], page) if block_given?
+        else
+          break
+        end
+      end
+      all_res
+    end
 
     # Import asset into fofa
     #

data/lib/fofa/version.rb

@@ -1,3 +1,3 @@
 module Fofa
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: fofa
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - fofa