deadfinder 1.2.1 → 1.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cccf1eac41e7cb13f5391bd8191931b6225f3b16dd23a2e4432816d93730be2e
-  data.tar.gz: dddb0d51bc6b00507ae8622945114d373e7366714a74ebc4ede95057db1224f2
+  metadata.gz: e1e747b78068c1590c88f49acc42ca917a3c26684564c4465449181d5fea4125
+  data.tar.gz: 19f6a77d77bb0b3e2daba04ec364b1626eebbe30ec9973753fb75c1cb78ffc2e
 SHA512:
-  metadata.gz: d273d3d94cce7a98185dfda1713b43ddf632b19b11bff3f93c1c68bdbf3306c65825411b95f2b4ab3a517bfb88ca3db2f888a4b4a84814058a030b7c62098a0c
-  data.tar.gz: ca7d125f5c7a89275672418546e6913af1765bdc0eaaa8ce324be94c24e4b84989df9ec4fae371d6a4a7a160c97be47db2633b8de04be0eb98dfdc4ea2009978
+  metadata.gz: ddc7a3974fcbf62d78ce739862f1686b070bfea6f26ae534c5fe18b49fb48432f7483567ab109ef4b97d40916460be1b08524dc8615d5f537564555df817c781
+  data.tar.gz: b676e1009517b67da7cff2dc10f26a50f6635d1abfeab6e036092d2b63eff55fa758c3cb0ee2b63ec34157901f00708a2d0f4282e99e5fc61057f70ad939e23f

data/lib/deadfinder/logger.rb

@@ -7,6 +7,10 @@ class Logger
     puts 'ℹ '.colorize(:blue) + text.to_s.colorize(:light_blue)
   end
 
+  def self.error(text)
+    puts '⚠︎ '.colorize(:red) + text.to_s
+  end
+
   def self.target(text)
     puts '► '.colorize(:green) + text.to_s.colorize(:light_green)
   end

data/lib/deadfinder/version.rb

@@ -1,3 +1,3 @@
 # frozen_string_literal: true
 
-VERSION = '1.2.1'
+VERSION = '1.2.2'

data/lib/deadfinder.rb

@@ -19,38 +19,42 @@ Output = {}
 
 class DeadFinderRunner
   def run(target, options)
-    page = Nokogiri::HTML(URI.open(target))
-
-    nodeset_a = page.css('a')
-    link_a = nodeset_a.map { |element| element['href'] }.compact
-    nodeset_script = page.css('script')
-    link_script = nodeset_script.map { |element| element['src'] }.compact
-    nodeset_link = page.css('link')
-    link_link = nodeset_link.map { |element| element['href'] }.compact
-
-    link_merged = []
-    link_merged.concat link_a, link_script, link_link
-
-    Logger.target target
-    Logger.sub_info "Found #{link_merged.length} point. [a:#{link_a.length}/s:#{link_script.length}/l:#{link_link.length}]"
-    Logger.sub_info 'Checking'
-    jobs    = Channel.new(buffer: :buffered, capacity: 1000)
-    results = Channel.new(buffer: :buffered, capacity: 1000)
-
-    (1..options['concurrency']).each do |w|
-      Channel.go { worker(w, jobs, results, target, options) }
-    end
+    begin
+      page = Nokogiri::HTML(URI.open(target))
+
+      nodeset_a = page.css('a')
+      link_a = nodeset_a.map { |element| element['href'] }.compact
+      nodeset_script = page.css('script')
+      link_script = nodeset_script.map { |element| element['src'] }.compact
+      nodeset_link = page.css('link')
+      link_link = nodeset_link.map { |element| element['href'] }.compact
+
+      link_merged = []
+      link_merged.concat link_a, link_script, link_link
+
+      Logger.target target
+      Logger.sub_info "Found #{link_merged.length} point. [a:#{link_a.length}/s:#{link_script.length}/l:#{link_link.length}]"
+      Logger.sub_info 'Checking'
+      jobs    = Channel.new(buffer: :buffered, capacity: 1000)
+      results = Channel.new(buffer: :buffered, capacity: 1000)
+
+      (1..options['concurrency']).each do |w|
+        Channel.go { worker(w, jobs, results, target, options) }
+      end
 
-    link_a.uniq.each do |node|
-      result = generate_url node, target
-      jobs << result
-    end
-    jobs.close
+      link_merged.uniq.each do |node|
+        result = generate_url node, target
+        jobs << result
+      end
+      jobs.close
 
-    (1..link_a.uniq.length).each do
-      ~results
+      (1..link_merged.uniq.length).each do
+        ~results
+      end
+      Logger.sub_done 'Done'
+    rescue => e 
+      Logger.error "[#{e}] #{target}"
     end
-    Logger.sub_done 'Done'
   end
 
   def worker(_id, jobs, results, target, options)
@@ -102,9 +106,11 @@ end
 
 def run_sitemap(sitemap_url, options)
   app = DeadFinderRunner.new
+  base_uri = URI(sitemap_url)
   sitemap = SitemapParser.new sitemap_url, { recurse: true }
   sitemap.to_a.each do |url|
-    app.run url, options
+    turl = generate_url url, base_uri
+    app.run turl, options
   end
   gen_output
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: deadfinder
 version: !ruby/object:Gem::Version
-  version: 1.2.1
+  version: 1.2.2
 platform: ruby
 authors:
 - hahwul