Added Network::URL#defang (closes #545).

Author: postmodern

lib/ronin/support/network/url.rb

@@ -17,6 +17,7 @@
 #
 
 require 'ronin/support/network/http'
+require 'ronin/support/network/defang'
 
 require 'addressable/uri'
 require 'uri/query_params/core_ext/addressable/uri'
@@ -37,6 +38,21 @@ module Network
       #
       class URL < Addressable::URI
 
+        #
+        # Defangs the URL.
+        #
+        # @return [String]
+        #   The defanged URL.
+        #
+        # @example
+        #   url = URL.new("https://www.example.com:8080/foo?q=1")
+        #   url.defang
+        #   # => "hxxps[://]www[.]example[.]com[:]8080/foo?q=1"
+        #
+        def defang
+          Defang.defang_url(self)
+        end
+
         #
         # Returns the Status Code of the HTTP Response for the URL.
         #

spec/network/url_spec.rb

@@ -12,7 +12,18 @@
     expect(described_class).to include(URI::QueryParams::Mixin)
   end
 
-  subject { described_class.parse('https://example.com/') }
+  let(:url) { 'https://example.com/' }
+
+  subject { described_class.parse(url) }
+
+  describe "#defang" do
+    let(:url)      { 'http://www.example.com/foo?q=1' }
+    let(:defanged) { 'hxxp[://]www[.]example[.]com/foo?q=1' }
+
+    it "must return the defanged URL" do
+      expect(subject.defang).to eq(defanged)
+    end
+  end
 
   describe "#status" do
     context "integration", :network do