chore: add more rescue clauses

ninoseki · Sep 10, 2018 · 58d6bc8 · 58d6bc8
1 parent 77d475a
commit 58d6bc8
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/lib/ryo/plugin/shodan.rb b/lib/ryo/plugin/shodan.rb
@@ -19,6 +19,8 @@ def self.discover(ip)
         new.discover(ip)
       rescue ArgumentError => e
         { error: e.to_s }
+      rescue StandardError => e
+        { error: e.to_s }
       end
     end
   end

diff --git a/lib/ryo/plugin/subdomain/find_subdomains.rb b/lib/ryo/plugin/subdomain/find_subdomains.rb
@@ -11,6 +11,7 @@ def endpoint
         def parse
           table = doc.at_css("table#table-view")
           return [] if table.nil?
+
           table.css("tr")[1..-1].map do |row|
             cols = row.css("td")
             domain = cols.first.at_css("a")&.text&.strip

diff --git a/lib/ryo/plugin/tech.rb b/lib/ryo/plugin/tech.rb
@@ -12,6 +12,8 @@ def self.discover(uri)
         plugins.each do |name, plugin|
           result = plugin.execute(target)
           results[name] = result unless result.empty?
+        rescue StandardError => _
+          next
         end
         results
       end