chkdfront 1.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 56a52ddfb0d5fbbfd69cb1e231bd8e5545d9a7c0467a5a438d1c7783029db7ed
+  data.tar.gz: 4187aec0c81896522bcef3a528cdf9057fd5e747209a5ff80fedc367a94093b6
+SHA512:
+  metadata.gz: f3f6407453d3ddf42344038bcfa085031d6d6a4192989b6d08c7d19d86238ac73fad971fd7576c5ddbec238a19c3c35a0d3ac1376e3020bae76bea72b3b26eb6
+  data.tar.gz: c2e3c72f7efd6079542404911238cd46a6127865c7e3e3a261de699c5800b924b0fed1f74b1a2fd7ae8362019678989e9cc4ebab61fd32c6dc1b27118aa4dc23

data/.gitignore

@@ -0,0 +1,146 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+### macOS template
+# General
+.DS_Store
+.AppleDouble
+.LSOverride
+
+# Icon must end with two \r
+Icon
+
+# Thumbnails
+._*
+
+# Files that might appear in the root of a volume
+.DocumentRevisions-V100
+.fseventsd
+.Spotlight-V100
+.TemporaryItems
+.Trashes
+.VolumeIcon.icns
+.com.apple.timemachine.donotpresent
+
+# Directories potentially created on remote AFP share
+.AppleDB
+.AppleDesktop
+Network Trash Folder
+Temporary Items
+.apdisk
+### Ruby template
+*.gem
+*.rbc
+/.config
+/coverage/
+/InstalledFiles
+/pkg/
+/spec/reports/
+/spec/examples.txt
+/test/tmp/
+/test/version_tmp/
+/tmp/
+
+# Used by dotenv library to load environment variables.
+# .env
+
+## Specific to RubyMotion:
+.dat*
+.repl_history
+build/
+*.bridgesupport
+build-iPhoneOS/
+build-iPhoneSimulator/
+
+## Specific to RubyMotion (use of CocoaPods):
+#
+# We recommend against adding the Pods directory to your .gitignore. However
+# you should judge for yourself, the pros and cons are mentioned at:
+# https://guides.cocoapods.org/using/using-cocoapods.html#should-i-check-the-pods-directory-into-source-control
+#
+# vendor/Pods/
+
+## Documentation cache and generated files:
+/.yardoc/
+/_yardoc/
+/doc/
+/rdoc/
+
+## Environment normalization:
+/.bundle/
+/vendor/bundle
+/lib/bundler/man/
+
+# for a library or gem, you might want to ignore these files since the code is
+# intended to run in multiple environments; otherwise, check them in:
+# Gemfile.lock
+# .ruby-version
+# .ruby-gemset
+
+# unless supporting rvm < 1.11.0 or doing something fancy, ignore this:
+.rvmrc
+### JetBrains template
+# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and WebStorm
+# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
+
+# User-specific stuff
+.idea/**/workspace.xml
+.idea/**/tasks.xml
+.idea/**/dictionaries
+.idea/**/shelf
+
+# Sensitive or high-churn files
+.idea/**/dataSources/
+.idea/**/dataSources.ids
+.idea/**/dataSources.local.xml
+.idea/**/sqlDataSources.xml
+.idea/**/dynamic.xml
+.idea/**/uiDesigner.xml
+.idea/**/dbnavigator.xml
+
+# Gradle
+.idea/**/gradle.xml
+.idea/**/libraries
+
+# CMake
+cmake-build-debug/
+cmake-build-release/
+
+# Mongo Explorer plugin
+.idea/**/mongoSettings.xml
+
+# File-based project format
+*.iws
+
+# IntelliJ
+out/
+
+# mpeltonen/sbt-idea plugin
+.idea_modules/
+
+# JIRA plugin
+atlassian-ide-plugin.xml
+
+# Cursive Clojure plugin
+.idea/replstate.xml
+
+# Crashlytics plugin (for Android Studio and IntelliJ)
+com_crashlytics_export_strings.xml
+crashlytics.properties
+crashlytics-build.properties
+fabric.properties
+
+# Editor-based Rest Client
+.idea/httpRequests
+### VisualStudioCode template
+.vscode/*
+!.vscode/settings.json
+!.vscode/tasks.json
+!.vscode/launch.json
+!.vscode/extensions.json
+

data/Gemfile

@@ -0,0 +1,4 @@
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in chkdfront.gemspec
+gemspec

data/README.md

@@ -0,0 +1,67 @@
+# Chkdfront
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/chkdfront`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+TODO: Delete this and the text above, and describe your gem
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'chkdfront'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install chkdfront
+
+## Usage
+
+```
+MMMMMMMMMMMMMWNK0kdolc;,'...  .;KMMMMMMMMMMMMMMMMMMMMMMMMMMM
+MMMMMMMMMN0xl;'.                lNMMMMMMMMMMMMMMMMMMMMMMMMMM
+MMMMMWKxc'.                     .kWMMMMMMMMMMMMMMMMMMMMMMMMM
+MMN0o;.                          ;XMMMMMMMMMMMMMMMMMMMMMMMMM
+Ol,.                             .xWMMMMMMMMMMMMMMMMMMMMMMMM
+'                        .        ;XMMMMMMMMMMMMMMMMMMMMMMMM
+l                    .;dO00ko;'.  .kMMMMMMMMMMMMMMMMMMMMMMMM
+X:                   .dKNWWXOl,.   oWMMMMMMMMMMMMMMMMMMMMMMM
+M0,        'ldkx;      .',,..,;:cc;dNMMMMMMMMMMMMMMMMMMMMMMM
+MWk.     .dNMMWk.          ;ONWWWWWWMMMMMMMMMMMMMMMMMMMMMMMM
+MMWx.    lXXKkc.          :XMXOo;;coxOKNWMMMMMMMMMMMMMMMMMMM
+MMMWx.   ....            ;KMXc.       .';cdkKNMMMMMMMMMMMMMM
+MMMMWk.            .:oxxkXMNl               .,cx0NMMMMMMMMMM
+MMMMMWO,         .oXWWXNMMNo.                   .,lONMMMMMMM
+MMMMMMMXl.      .kNOo,;OWWx.                        'lONMMMM
+MMMMMMMMWO;    .lx;.  lNMO'                           .,dKWM
+MMMMMMMMMMNx;. ..    ,0MX:   .,:ll:,.                    .:x
+MMMMMMMMMMMMNOc.    .dWWx.   .'lXWMNOc.                    .
+MMMMMMMMMMMMMMWXOo:':KMX:       .cxO0k;                   .o
+MMMMMMMMMMMMMMMMMMWNNWMk.                 ,odol,.        .dN
+MMMMMMMMMMMMMMMMMMMMMMWo                  .dXWMWO:      .dNM
+MMMMMMMMMMMMMMMMMMMMMMNc    ;,              'cdxko.    .kWMM
+MMMMMMMMMMMMMMMMMMMMMMNc    lKOl.                     ;0WMMM
+MMMMMMMMMMMMMMMMMMMMMMWo    ,KMW0l.                 .lXMMMMM
+MMMMMMMMMMMMMMMMMMMMMMMk.    :kKMWXOdolcc:,.       ,kWMMMMMM
+MMMMMMMMMMMMMMMMMMMMMMMNc     .'o0XNNNKOo;.      'dXMMMMMMMM
+MMMMMMMMMMMMMMMMMMMMMMMMK;       ..''..        'dXWMMMMMMMMM
+MMMMMMMMMMMMMMMMMMMMMMMMMKc                 .:xXMMMMMMMMMMMM
+MMMMMMMMMMMMMMMMMMMMMMMMMMNk;.          .'cxKWMMMMMMMMMMMMMM
+MMMMMMMMMMMMMMMMMMMMMMMMMMMMNxl:'....':oOXWMMMMMMMMMMMMMMMMM
+
+```
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/chkdfront.

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+task :default => :spec

data/bin/chkdfront

@@ -0,0 +1,173 @@
+#!/usr/bin/env ruby
+#
+# @Author: KING SABRI - @KINGSABRI
+#
+require 'chkdfront'
+
+logo = %Q{
+MMMMMMMMMMMMMWNK0kdolc;,'      ;KMMMMMMMMMMMMMMMMMMMMMMMMMMM
+MMMMMMMMMN0xl;'.                lNMMMMMMMMMMMMMMMMMMMMMMMMMM
+MMMMMWKxc'.                     .kWMMMMMMMMMMMMMMMMMMMMMMMMM
+MMN0o;.                          ;XMMMMMMMMMMMMMMMMMMMMMMMMM
+Ol,.                             .xWMMMMMMMMMMMMMMMMMMMMMMMM
+'                        .        ;XMMMMMMMMMMMMMMMMMMMMMMMM
+l                    .;dO00ko;'.  .kMMMMMMMMMMMMMMMMMMMMMMMM
+X:                   .dKNWWXOl,.   oWMMMMMMMMMMMMMMMMMMMMMMM
+M0,        'ldkx;      .',,..,;:cc;dNMMMMMMMMMMMMMMMMMMMMMMM
+MWk.     .doMaiN.          ;ONWWWWWWMMMMMMMMMMMMMMMMMMMMMMMM
+MMWx.    lXXKkc.          :XMXOo;;coxOKNWMMMMMMMMMMMMMMMMMMM
+MMMWx.   ....            ;KMXc.       .';cdkKNMMMMMMMMMMMMMM
+MMMMWk.            .:oxxkXMNl               .,cx0NMMMMMMMMMM
+MMMMMWO,         .oXWWXNMMNo.                   .,lONMMMMMMM
+MMMMMMMXl.      .kNOo,;OWWx.                        'lONMMMM
+MMMMMMMMWO;    .lx;.  lNMO'                           .,dKWM
+MMMMMMMMMMNx;. ..    ,0MX:   .,:ll:,.                    .:x
+MMMMMMMMMMMMNOc.    .dWWx.   .'lXWMNOc.                    .
+MMMMMMMMMMMMMMWXOo:':KMX:       .cxO0k;                   .o
+MMMMMMMMMMMMMMMMMMWNNWMk.                 ,domain.        .N
+MMMMMMMMMMMMMMMMMMMMMMWo                  .fronting     .dNM
+MMMMMMMMMMMMMMMMMMMMMMNc    ;,              'cdxko.    .kWMM
+MMMMMMMMMMMMMMMMMMMMMMNc    lKOl.                     ;0WMMM
+MMMMMMMMMMMMMMMMMMMMMMWo    ,KMW0l.                 .lXMMMMM
+MMMMMMMMMMMMMMMMMMMMMMMk.    :kKMWXOdolcc:,.       ,kWMMMMMM
+MMMMMMMMMMMMMMMMMMMMMMMNc     .'o0XNNNKOo;.      'dXMMMMMMMM
+MMMMMMMMMMMMMMMMMMMMMMMMK;       ..''..        'dXWMMMMMMMMM
+MM By: #{'@KINGSABRI'.bold} MMMMMMKc                  .:xXMMMMMMMMMMMM
+MM #{'chkdfront'.bold} v#{ChkDFront::VERSION} MMMMNk;.            .'cxKWMMMMMMMMMMMMMM
+MMMMMMMMMMMMMMMMMMMMMMMMMMMMNxl:'....':oOXWMMMMMMMMMMMMMMMMM
+}
+
+CLI::UI::StdoutRouter.enable
+
+options = OpenStruct.new(
+    front_target: nil, domain_front: nil, provider: :auto,
+    expect: nil,  proxy: nil, debug: false, tshoot: false
+)
+option_parser = OptionParser.new do |opts|
+  opts.banner = "Check DomainFront (chkdfront) - A tool verifies domain fronting."
+  opts.separator ""
+  opts.separator "Help menu:"
+
+  opts.on("-f", "--front-target URL", "Fronted target domain or URL.", "\te.g. images.businessweek.com") do |o|
+    options.front_target = o
+  end
+
+  opts.on("-d", "--domain-front DOMAIN", "DomainFront domain.", "\te.g. df36z1umwj2fze.cloudfront.net") do |o|
+    options.domain_front = o
+  end
+
+  opts.on("-e", "--expect STRING", "Expect a given string that indicates success. (case-sensitive)", "\te.g. It works") do |o|
+    options.expect = o
+  end
+
+  opts.on("-p", "--provider NUM",
+          "Choose CDN / Domain Front Provider:", Integer,
+          "\t[0] Auto    (default - auto tune request. Extra request to detect)",
+          "\t[1] Amazon  (tune request for Amazon domain fronting)",
+          "\t[2] Azure   (tune request for Azure domain fronting)",
+          "\t[3] Alibaba (tune request for Alibaba domain fronting)") do |provider|
+    case provider.to_i
+    when 0
+      options.provider = :auto
+    when 1
+      options.provider = :amazon
+    when 2
+      options.provider = :azure
+    when 3
+      options.provider = :alibaba
+    else
+      puts "[!] Unknown Provider: switching to auto.."
+      options.provider = :auto
+    end
+  end
+
+  opts.on("-t", "--troubleshoot [DOMAIN]", "Force troubleshooting procedures.",
+          "execute troubleshooting procedures(ping, http, nslookup) for all parties",
+          "(optional: original domain where CDN forwards, to include in the checks)",
+          "\te.g. c2.mydomain.com") do |o|
+    options.tshoot = o
+  end
+
+  opts.on("--proxy USER:PASS@HOST:PORT", "Use proxy settings if you're behind proxy.",
+          "\te.g. user1:Pass123@localhost:8080") do |o|
+    options.proxy = o
+  end
+
+  opts.on("--debug", "Force debugging.",
+          "show response's body and low-level request and response debug trace.",
+          "(default enabled when test fails.)"
+          ) do |o|
+    options.debug = o
+  end
+
+  opts.on("-h", "--help", "Show this message.") do
+    puts logo
+    puts opts
+    exit!
+  end
+
+  opts.on_tail "\nUsage:\n" + "  #{$PROGRAM_NAME} <OPTIONS>"
+  opts.on_tail "Example:"
+  opts.on_tail %{  #{$PROGRAM_NAME} -f images.businessweek.com -d df36z1umwj2fze.cloudfront.net}
+  opts.on_tail %{  #{$PROGRAM_NAME} -f images.businessweek.com -d df36z1umwj2fze.cloudfront.net --debug -t c2.mysite.com\n\n}
+end
+
+begin
+  option_parser.parse!(ARGV)
+  if (options.front_target || options.domain_front).nil?
+    puts option_parser
+  end
+
+  # list of hosts to troubleshoot later
+  hosts = [options.front_target, options.domain_front, options.tshoot].compact
+  cli = ChkDFront::CliOperations.new
+
+  if options.front_target && options.domain_front
+    spnr_opts = {output: $stdout, clear: true, format: :arrow_pulse, hide_cursor: true}
+    spinner = TTY::Spinner.new("[:spinner] please wait!", spnr_opts)
+    spinner.auto_spin
+
+    http = ChkDFront::Request.new(options.front_target, options.domain_front, options.proxy)
+    http.send_to(cli.find_provider(options.provider, options.domain_front))
+    cli.http     = http
+    cli.request  = http.request
+    cli.response = http.response
+
+    res  = http.response
+    # if the request succeeded
+    if res.code.to_i == 200
+      CLI::UI::Frame.open('Request', color: :bold, timing: false) do
+        CLI::UI::Frame.open("Response | #{res.message} (#{res.code})", color: :green, timing: false) do
+          cli.show_success
+          cli.show_expected(options.expect) if options.expect
+          cli.show_body; cli.show_debug     if options.debug
+        end
+      end
+      # if the user want to troubleshoot even request has succeeded
+      if options.tshoot || options.tshoot.nil?
+        CLI::UI::Frame.open('Troubleshooting'.bold, color: :magenta, timing: false) do
+          cli.troubleshoot(hosts)
+        end
+      end
+    else  # If request failed, troubleshoot
+      CLI::UI::Frame.open('Request', color: :bold, timing: false) do
+        CLI::UI::Frame.open("Response | #{res.message} (#{res.code})", color: :red, timing: false) do
+          cli.show_checks
+          cli.show_body
+          cli.show_debug
+        end
+      end
+      CLI::UI::Frame.open('Troubleshooting', color: :magenta, timing: false) { cli.troubleshoot(hosts) }
+    end
+  end
+rescue OptionParser::MissingArgument => e
+  puts option_parser
+  e.args.each {|arg| puts '[!] '+ "#{e.reason.capitalize} for '#{arg}' option."}
+rescue OptionParser::InvalidOption => e
+  puts option_parser
+  pp e
+rescue Exception => e
+  puts "#{$PROGRAM_NAME} Exception".error
+  puts e.backtrace_locations
+  puts e
+end

data/chkdfront.gemspec

@@ -0,0 +1,38 @@
+
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "chkdfront/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = 'chkdfront'
+  spec.version       = ChkDFront::VERSION
+  spec.authors       = ['@KINGSABRI']
+  spec.email         = ['king.sabri@gmail.com']
+
+  spec.summary       = %q{Check Domain Fronting (chkdfront) - It checks if domain fronting is working.}
+  spec.description   = %q{Check Domain Fronting (chkdfront) - It checks if domain fronting implementation is working.}
+  spec.homepage      = 'https://github.com/KINGSABRI/chkdfront'
+  spec.licenses      = ['MIT']
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files         = Dir.chdir(File.expand_path('..', __FILE__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+
+  spec.executables   = ['chkdfront']
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency 'cli-ui'
+  spec.add_dependency 'tty-spinner'
+  spec.add_dependency 'word_wrap'
+  spec.add_dependency 'net-ping'
+  spec.add_dependency 'net-dns'
+  spec.add_dependency 'adomain'
+
+  spec.add_development_dependency "bundler", "~> 2.0"
+  spec.add_development_dependency "rake", "~> 10.0"
+
+  spec.metadata['source_code_uri'] = 'https://github.com/KINGSABRI/chkdfront'
+  spec.metadata['bug_tracker_uri'] = 'https://github.com/KINGSABRI/chkdfront/issues'
+end

data/lib/chkdfront/cli_operations.rb

@@ -0,0 +1,126 @@
+class String
+  def red; colorize(self, "\e[1m\e[31m"); end
+  def green; colorize(self, "\e[1m\e[32m"); end
+  def dark_green; colorize(self, "\e[32m"); end
+  def yellow; colorize(self, "\e[1m\e[33m"); end
+  def blue; colorize(self, "\e[1m\e[34m"); end
+  def dark_blue; colorize(self, "\e[34m"); end
+  def purple; colorize(self, "\e[35m"); end
+  def dark_purple; colorize(self, "\e[1;35m"); end
+  def cyan; colorize(self, "\e[1;36m"); end
+  def dark_cyan; colorize(self, "\e[36m"); end
+  def pure; colorize(self, "\e[0m\e[28m"); end
+  def bold; colorize(self, "\e[1m"); end
+  def error; colorize(self, "\n[" + " ✖ ".red + "] "); end
+  def colorize(text, color_code) "#{color_code}#{text}\e[0m" end
+end
+
+module ChkDFront
+  class CliOperations
+    include ChkDFront::Providers
+    attr_accessor :http, :request, :response
+    def initialize
+      @http     = nil
+      @request  = nil
+      @response = nil
+
+      @regx_domain = /^(?:https?:\/\/)?(?:[^@\n]+@)?([^:\/\n?]+)/ # extract domain from url
+      @options = { output: $stdout, format: :pulse_2, success_mark: " ✔ ".green, error_mark: " ✖ ".red, hide_cursor: true }
+    end
+
+    def show_success
+      puts CLI::UI.fmt "{{v}} Front target:    " + "#{@http.front_target.host}"
+      puts CLI::UI.fmt "{{v}} Provider header: " + "#{@response['Via']}"
+      puts CLI::UI.fmt "{{v}} Domain front:    " + "#{@http.domain_front}"
+    end
+
+    def show_expected(string)
+      CLI::UI::Frame.divider('Expected String')
+      if @response.body.include?(string)
+        puts CLI::UI.fmt "{{v}} Found: " + "#{string}"
+      else
+        puts CLI::UI.fmt "{{x}} Not Found: " + "#{string}"
+      end
+    end
+
+    def show_checks
+      puts @http.res_err[:checks]
+    end
+
+    def show_body
+      CLI::UI::Frame.divider('Response Body')
+      puts @response.body
+    end
+    def show_debug
+      CLI::UI::Frame.divider('Debugging')
+      puts WordWrap.ww(@http.debug_output, CLI::UI::Terminal.width)
+    end
+
+    # Find provider if not given by the user
+    # @param [Symbol] provider_name
+    #   given by 'options.provider' option.
+    #   Default is :auto, accepted is :amazon, :azure, :alibaba
+    # @param [String] domain_front
+    #   given by 'options.domain_front' option.
+    def find_provider(provider_name, domain_front)
+      @options[:format] = :dots_9
+      spinner = TTY::Spinner.new("[:spinner] Auto-detecting", @options)
+      case provider_name
+      when :auto
+        domain = Adomain.domain(domain_front)
+        [amazon, azure, alibaba].map do |provider|
+          if provider[:dfront].include?(domain)
+            # spinner.update(title: "Provider found: #{provider[:name].bold}")
+            # spinner.reset
+            spinner.success(" | Provider found: #{provider[:name].bold}")
+            return provider[:name].downcase.to_sym
+          end
+        end
+      when :amazon || :azure || :alibaba
+        provider_name
+      else
+        spinner.error("Failed to auto detect provider: please use '-p' and choose from: 1, 2 or 3")
+      end
+    end
+
+    def troubleshoot(hosts = [])
+      icmp_ping hosts
+      http_ping hosts
+      dns_ping  hosts
+    end
+
+    def icmp_ping(hosts)
+      CLI::UI::Frame.divider('ICMP Ping', color: :reset)
+      hosts.map do |host|
+        host = host.scan(@regx_domain).join
+        spinner = TTY::Spinner.new("[:spinner] pinging #{host}", @options)
+        icmp = ChkDFront::Troubleshoot.icmp_ping(host)
+        icmp.ping? ? spinner.success : spinner.error
+      end
+    end
+
+    def http_ping(hosts)
+      CLI::UI::Frame.divider('HTTP Ping', color: :reset)
+      hosts.map do |host|
+        host = host.scan(@regx_domain).join
+        spinner = TTY::Spinner.new("[:spinner] pinging #{host}", @options)
+        http = ChkDFront::Troubleshoot.http_ping(host)
+        http.ping? ? spinner.success : spinner.error
+        puts http.exception unless http.ping?
+      end
+    end
+
+    def dns_ping(hosts)
+      CLI::UI::Frame.divider('NSlookup (CNAME)', color: :reset)
+      hosts.map do |host|
+        host = host.scan(@regx_domain).join
+        spinner = TTY::Spinner.new("[:spinner] nslookup #{host}", @options)
+        dns = ChkDFront::Troubleshoot.dns_ping(host)
+        dns.empty? ? spinner.error : spinner.success
+        puts dns
+      end
+    end
+  end
+end
+
+

data/lib/chkdfront/providers.rb

@@ -0,0 +1,68 @@
+module ChkDFront
+  # This module contains CDN provider specific settings headers.
+  module Providers
+    def get_provider(provider, domain_front)
+      if domain_front.port
+        host = "#{domain_front.host}:#{domain_front.port}"
+      else
+        host = domain_front.host
+      end
+      case provider
+      when :amazon  then  amazon(host)
+      when :azure   then  azure(host)
+      when :alibaba then  alibaba(host)
+      else
+        puts "Unknown Provider!!!!: #{provider}"
+        return false
+      end
+    end
+
+    # Amazon vendor specific settings and headers
+    # @param [String] host
+    # @return [Hash]
+    #   Returns a hash of { dfront: [domain1, domain2], headers: {}}
+    def amazon(host='')
+      {
+          name:    'Amazon',
+          dfront:  ['cloudfront.net'],
+          headers: {
+              'Host'       => host,
+              'User-Agent' => "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.13; rv:65.0) Gecko/20100101 Firefox/65.0",
+              'Connection' => 'close'
+          }
+      }
+    end
+
+    # Azure vendor specific settings and headers
+    # @param [String] host
+    # @return [Hash]
+    #   Returns a hash of { dfront: [domain1, domain2], headers: {}}
+    def azure(host='')
+      {
+          name:    'Azure',
+          dfront:  ['azureedge.net'],
+          headers: {
+            'Host'       => host,
+            'User-Agent' => "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.13; rv:65.0) Gecko/20100101 Firefox/65.0",
+            'Connection' => 'close'
+          }
+      }
+    end
+
+    # Alibaba vendor specific settings and headers
+    # @param [String] host
+    # @return [Hash]
+    #   Returns a hash of { dfront: [domain1, domain2], headers: {}}
+    def alibaba(host='')
+      {
+          name:    'Alibaba',
+          dfront:  ['kunlungr.com'],
+          headers: {
+              'Host'       => host,
+              'User-Agent' => "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.13; rv:65.0) Gecko/20100101 Firefox/65.0",
+              'Connection' => 'close'
+          }
+      }
+    end
+  end
+end

data/lib/chkdfront/request.rb

@@ -0,0 +1,179 @@
+module ChkDFront
+  # Contains all HTTP request related functionalities
+  class Request
+    include ChkDFront::Providers
+
+    # @attribute [r] front_target
+    #   @return [String] front_target
+    attr_reader :front_target
+    # @!attribute [r] domain_front
+    #   @return [String]
+    attr_reader :domain_front
+    # @!attribute [r] proxy
+    #   @return [String]
+    attr_reader :proxy
+    # @!attribute [r]
+    #   @return [String]
+    attr_reader :debug_output
+    # @!attribute [r] request
+    #   @return [Net::HTTP::Request]
+    attr_reader :request
+    # @!attribute [r] request
+    #   @return [Net::HTTP::Response]
+    attr_reader :response
+    # @!attribute [r] res_err
+    #   @return [Hash]
+    attr_reader :res_err
+
+    def initialize(front_target, domain_front, proxy)
+      @front_target = format_front_target(front_target)
+      @domain_front = format_domain_front(domain_front)
+      @proxy        = format_proxy(proxy)
+      @http         = setup_http
+    end
+
+    # Send GET/POST request to the targeted domain
+    #
+    # @param [Symbol] provider_name
+    #   Available providers symbols:
+    #     - :amazon
+    #     - :azure
+    #     - :alibaba
+    # @param [String] http_method
+    # @return [Object]
+    def send_to(provider_name, http_method='get')
+      provider = get_provider(provider_name, @domain_front)
+      @request = Net::HTTP.const_get(http_method.capitalize).new(
+          @front_target.request_uri, provider[:headers]
+      )
+      @response = @http.request(@request)
+      check_http(@response) # if response failed, give the user some suggestion
+    rescue SocketError => e
+      puts "#{self.class}##{__method__}:".error
+      puts e.message
+    rescue Net::ReadTimeout => e
+      puts "#{self.class}##{__method__}:".error
+      puts e.message
+      puts "Remote port is closed.".error
+      exit!
+    rescue Exception => e
+      puts "#{self.class}##{__method__}:".error
+      puts e.full_message
+    end
+
+    private
+
+    # @param [String] front_target
+    #   The domain of fronted domain (e.g. images.businessweek.com)
+    #     This domain should be on the same CDN that user already uses
+    # @return [OpenStruct]
+    def format_front_target(front_target)
+      front_target = "http://#{front_target}" unless front_target.match(/^(http|https):\/\//i)
+      uri = URI.parse(front_target)
+      OpenStruct.new(host: uri.host, port: uri.port, scheme: uri.scheme, request_uri: uri.request_uri)
+    end
+
+    # @param [String] domain_front
+    #   The CDN's domain front (e.g. df36z1umwj2fze.cloudfront.net)
+    #     this domain is what the users create by their provider.
+    # @return [String]
+    def format_domain_front(domain_front)
+      # if the user gave a URL instead a FQDN:PORT, parse it
+      #   Otherwise just use what you get
+      uri = URI.parse(domain_front)
+      if uri.scheme
+        OpenStruct.new(host: uri.host, port: uri.port)
+      else
+        host, port = domain_front.split(':')
+        OpenStruct.new(host: host, port: port)
+      end
+    end
+
+    #
+    # @param [String] proxy
+    #   The accepted proxy format is:
+    #     - Username and Password separated by column
+    #     - Hostname and Port separated by column
+    #     - Both above parts separated by @ sign
+    #   @example
+    #     user1:Pass123@localhost:8080
+    #
+    # @return [OpenStruct]
+    def format_proxy(proxy)
+      if proxy
+        p_url, p_creds = proxy.reverse&.split('@', 2).map(&:reverse)
+        p_host, p_port = p_url.split(':')      if p_url
+        p_user, p_pass = p_creds.split(':', 2) if p_creds
+      end
+      OpenStruct.new(host: p_host, port: p_port, user: p_user, pass: p_pass)
+    end
+
+    # @private
+    # Setups the initial http connection settings.
+    #   The @!method #send_to is going to use it later
+    # @return [Net::HTTP]
+    def setup_http
+      http = Net::HTTP.new(@front_target.host, @front_target.port,
+                           @proxy.host, @proxy.port,
+                           @proxy.user, @proxy.pass)
+      http.use_ssl = true if @front_target.scheme =~ /https/i
+      http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+      http.open_timeout = 5
+      http.read_timeout = 5
+      @debug_output = ""
+      http.set_debug_output(@debug_output)
+      return http
+    rescue Exception => e
+      puts "#{self.class}##{__method__}:".error
+      puts e.full_message
+    end
+
+    # @param [] response
+    #   HTTP response error-based suggestions.
+    #     If request fail, it checks the response type and suggest checks based on the error.
+    #     These suggestion are based on our experience to help users to troubleshoot.
+    #     If you've faced different cases and responses, please report it us to update the list
+    # @todo: move this to a yaml or json file instead
+    def check_http(response)
+      case response
+      when Net::HTTPOK
+        response
+      when Net::HTTPGatewayTimeOut
+        @res_err = {
+            orig:  response,
+            checks: [
+                "Check your original server:",
+                "  is server up?",
+                "  is service up?",
+                "  is service port correct?",
+            ]
+        }
+      when Net::HTTPBadGateway
+        @res_err = {
+            orig:  response,
+            checks: [
+                "- Check your original server:",
+                "  - is server up?",
+                "  - is service up?",
+                "  - is service port correct?",
+                "- Check your provider:",
+                "  - is front domain name correct? (#{@domain_front})",
+                "  - is destination port correct?.",
+            ]
+        }
+      when Net::HTTPForbidden
+        @res_err = {
+            orig:  response,
+            checks: [
+                "- Check your provider:",
+                "  - is front domain name correct? (#{@domain_front})",
+                "- Check your target:",
+                "  - is target domain correct? (#{@front_target.host})",
+                "  - is target port correct? (#{@front_target.port})"
+            ]
+        }
+      end
+    end
+  end
+end
+

data/lib/chkdfront/troubleshoot.rb

@@ -0,0 +1,22 @@
+module ChkDFront
+  # Contains all troubleshooting functionalities.
+  # Currently, ping, http, nslookup
+  class Troubleshoot
+    def self.icmp_ping(host)
+      Net::Ping::External.new(host)
+    end
+
+    def self.http_ping(host)
+      host = "http://#{host}" unless host.match(/^(http|https):\/\//i)
+      Net::Ping::HTTP.new(host)
+    end
+
+    def self.dns_ping(host)
+      dns = Net::DNS::Resolver.start(host)
+      dns.answer.select {|r| r.type == "CNAME"}&.map(&:cname)
+    rescue Exception => e
+      puts e.message
+      return []
+    end
+  end
+end

data/lib/chkdfront/version.rb

@@ -0,0 +1,3 @@
+module ChkDFront
+  VERSION = "1.0.1"
+end

data/lib/chkdfront.rb

@@ -0,0 +1,24 @@
+# Standard libraries
+require 'optparse'
+require 'ostruct'
+require 'net/http'
+require 'openssl'
+
+# Chkdfront
+require 'chkdfront/version'
+require 'chkdfront/providers'
+require 'chkdfront/request'
+require 'chkdfront/troubleshoot'
+require 'chkdfront/cli_operations'
+
+# External libraries
+require 'cli/ui'
+require 'tty-spinner'
+require 'word_wrap'
+require 'net/ping'
+require 'net/dns'
+require 'adomain'
+
+module ChkDFront
+  class Error < StandardError; end
+end

metadata

@@ -0,0 +1,171 @@
+--- !ruby/object:Gem::Specification
+name: chkdfront
+version: !ruby/object:Gem::Version
+  version: 1.0.1
+platform: ruby
+authors:
+- "@KINGSABRI"
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-04-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: cli-ui
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: tty-spinner
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: word_wrap
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-ping
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-dns
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: adomain
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !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: Check Domain Fronting (chkdfront) - It checks if domain fronting implementation
+  is working.
+email:
+- king.sabri@gmail.com
+executables:
+- chkdfront
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- README.md
+- Rakefile
+- bin/chkdfront
+- chkdfront.gemspec
+- lib/chkdfront.rb
+- lib/chkdfront/cli_operations.rb
+- lib/chkdfront/providers.rb
+- lib/chkdfront/request.rb
+- lib/chkdfront/troubleshoot.rb
+- lib/chkdfront/version.rb
+homepage: https://github.com/KINGSABRI/chkdfront
+licenses:
+- MIT
+metadata:
+  source_code_uri: https://github.com/KINGSABRI/chkdfront
+  bug_tracker_uri: https://github.com/KINGSABRI/chkdfront/issues
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.2
+signing_key: 
+specification_version: 4
+summary: Check Domain Fronting (chkdfront) - It checks if domain fronting is working.
+test_files: []