dkim-query 0.2.1

data/spec/query_spec.rb

@@ -0,0 +1,60 @@
+require 'spec_helper'
+require 'dkim/query/query'
+
+describe DKIM::Query do
+  subject { described_class }
+
+  let(:domain) { 'yahoo.com' }
+
+  describe ".host_without_tld" do
+    let(:tld) { ".com" }
+
+    it "should strip the last component of the domain" do
+      expect(subject.host_without_tld(domain)).to be == domain.chomp(tld)
+    end
+  end
+
+  describe ".selectors_for" do
+    subject { described_class.selectors_for(domain) }
+
+    described_class::SELECTORS.each do |selector|
+      it "should include '#{selector}'" do
+        expect(subject).to include(selector)
+      end
+    end
+
+    it "should include the domain without the TLD" do
+      expect(subject).to include('yahoo')
+    end
+  end
+
+  describe ".query" do
+    it "should return all found DKIM keys" do
+      expect(subject.query(domain)).to be == {
+        's1024' => 'k=rsa;  p=MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDrEee0Ri4Juz+QfiWYui/E9UGSXau/2P8LjnTD8V4Unn+2FAZVGE3kL23bzeoULYv4PeleB3gfmJiDJOKU3Ns5L4KJAUUHjFwDebt0NP+sBK0VKeTATL2Yr/S3bT/xhy+1xtj4RkdV7fVxTn56Lb4udUnwuxK4V5b5PdOKj/+XcwIDAQAB; n=A 1024 bit key;'
+      }
+    end
+
+    context "with custom selectors" do
+      let(:selectors) { ['google', 's1024'] }
+
+      it "should query those selectors only" do
+        expect(subject.query(domain, selectors: selectors)).to have_key('s1024')
+      end
+    end
+
+    context "with no selectors" do
+      it "should not find any keys" do
+        expect(subject.query(domain, selectors: [])).to be_empty
+      end
+    end
+
+    context "when given an invalid domain" do
+      let(:domain) { 'foo.bar.com' }
+
+      it "should return {}" do
+        expect(subject.query(domain)).to be == {}
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,13 @@
+if ENV['CODECLIMATE_REPO_TOKEN']
+  require "codeclimate-test-reporter"
+  CodeClimate::TestReporter.start
+end
+
+require 'rspec'
+require 'dkim/query'
+
+include DKIM::Query
+
+RSpec.configure do |specs|
+  specs
+end

data/tasks/alexa.rb

@@ -0,0 +1,43 @@
+##
+# Adapted from gist: https://gist.github.com/zerothabhishek/3015666
+# Orginal author: zerothabhishek
+##
+
+require 'nokogiri'
+require 'open-uri'
+require 'resolv'
+require 'csv'
+
+namespace :alexa do
+  desc 'Scrapes the Alexa Top 500 and updates spec/data/alexa.csv'
+  task :scrape do
+    resolver = Resolv::DNS.new
+
+    CSV.open("spec/data/alexa.csv","w") do |csv|
+      (0..19).each do |i|
+        url = "http://www.alexa.com/topsites/global;#{i} "
+        doc = Nokogiri::HTML(open(url))
+
+        doc.css(".site-listing").each do |li|
+          begin
+            site_name = li.css(".desc-container .desc-paragraph a")[0].content
+            site_rank = li.css(".count")[0].content
+
+            puts "Resolving #{site_name} ..."
+            dmarc = begin
+                      resolver.getresource(
+                        "_dmarc.#{site_name.downcase}",
+                        Resolv::DNS::Resource::IN::TXT
+                      ).strings.join
+                    rescue Resolv::ResolvError
+                    end
+
+            csv << [site_rank, site_name, dmarc]
+          rescue => exception
+            warn exception.message
+          end
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,118 @@
+--- !ruby/object:Gem::Specification
+name: dkim-query
+version: !ruby/object:Gem::Version
+  version: 0.2.1
+platform: ruby
+authors:
+- nicktitle
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-07-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: parslet
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: Search and retrieve DKIM records for any number of hosts
+email:
+- nick.esposito@trailofbits.com
+executables:
+- dkim-query
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- ".yardopts"
+- ChangeLog.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/dkim-query
+- dkim-query.gemspec
+- lib/dkim/query.rb
+- lib/dkim/query/domain.rb
+- lib/dkim/query/exceptions.rb
+- lib/dkim/query/key.rb
+- lib/dkim/query/malformed_key.rb
+- lib/dkim/query/parser.rb
+- lib/dkim/query/query.rb
+- lib/dkim/query/version.rb
+- spec/domain_spec.rb
+- spec/key_spec.rb
+- spec/parser_spec.rb
+- spec/query_spec.rb
+- spec/spec_helper.rb
+- tasks/alexa.rb
+homepage: https://github.com/trailofbits/dkim-query#readme
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 1.9.1
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.7
+signing_key: 
+specification_version: 4
+summary: DKIM Retriever and Parser
+test_files:
+- spec/domain_spec.rb
+- spec/key_spec.rb
+- spec/parser_spec.rb
+- spec/query_spec.rb
+- spec/spec_helper.rb
+has_rdoc: