censu 0.1.1

data/lib/censys/autonomous_system.rb

@@ -0,0 +1,43 @@
+module Censys
+  class AutonomousSystem
+    # @return [String]
+    attr_reader :name
+
+    # @return [String]
+    attr_reader :rir
+
+    # @return [String]
+    attr_reader :routed_prefix
+
+    # @return [String]
+    attr_reader :country_code
+
+    # @return [Array<Integer>]
+    attr_reader :path
+
+    # @return [String]
+    attr_reader :organization
+
+    # @return [Integer]
+    attr_reader :asn
+
+    # @return [String]
+    attr_reader :description
+
+    #
+    # Initializes the Autonomous System (AS) information.
+    #
+    # @param [Hash{String => Object}] attributes
+    #
+    def initialize(attributes)
+      @name          = attributes['name']
+      @rir           = attributes['rir']
+      @routed_prefix = attributes['routed_prefix']
+      @country_code  = attributes['country_code']
+      @path          = attributes['path']
+      @organization  = attributes['organization']
+      @asn           = attributes['asn']
+      @description   = attributes['description']
+    end
+  end
+end

data/lib/censys/certificate.rb

@@ -0,0 +1,31 @@
+require 'time'
+
+module Censys
+  class Certificate < Document
+    #
+    # @return [String]
+    #
+    def raw
+      @attributes['raw']
+    end
+
+    #
+    # @return [Hash]
+    #
+    def parsed
+      @attributes['parsed']
+    end
+
+    #
+    # @return [Boolean]
+    #
+    def valid_nss
+      @attributes['validation']['nss']['valid']
+    end
+
+    # @return [Time]
+    def validation_timestamp
+      @validation_timestamp ||= Time.parse(@attributes['validation_timestamp'])
+    end
+  end
+end

data/lib/censys/document.rb

@@ -0,0 +1,48 @@
+module Censys
+  class Document
+    #
+    # Initializes the document.
+    #
+    # @param [Hash{String => Object}] attributes
+    #
+    def initialize(attributes)
+      @attributes = attributes
+    end
+
+    #
+    # Dig attributes
+    #
+    # @return [Object]
+    #
+    def dig(*keys)
+      @attributes.dig(*keys)
+    end
+
+    #
+    # Tags.
+    #
+    # @return [Array<String>]
+    #
+    def tags
+      @attributes['tags']
+    end
+
+    #
+    # Time last updated at.
+    #
+    # @return [Time]
+    #
+    def updated_at
+      @updated_at ||= Time.parse(@attributes['updated_at'])
+    end
+
+    #
+    # Additional document metadata.
+    #
+    # @return [Hash{String => Object}]
+    #
+    def metadata
+      @attributes['metadata']
+    end
+  end
+end

data/lib/censys/document/has_asn.rb

@@ -0,0 +1,16 @@
+require 'censys/autonomous_system'
+
+module Censys
+  class Document
+    module HasASN
+      #
+      # Autonomous System (AS) information.
+      #
+      # @return [AutonomousSystem]
+      #
+      def autonomous_system
+        @autonomous_system ||= AutonomousSystem.new(@attributes['autonomous_system'])
+      end
+    end
+  end
+end

data/lib/censys/document/has_location.rb

@@ -0,0 +1,16 @@
+require 'censys/location'
+
+module Censys
+  class Document
+    module HasLocation
+      #
+      # Location information.
+      #
+      # @return [Location]
+      #
+      def location
+        @location ||= Location.new(@attributes['location'])
+      end
+    end
+  end
+end

data/lib/censys/document/has_services.rb

@@ -0,0 +1,21 @@
+module Censys
+  class Document
+    module HasServices
+      #
+      # Open ports.
+      #
+      # @return [Hash{String => Hash}]
+      #
+      def ports
+        @ports ||= Hash[@attributes.select { |key, _| key =~ /\A\d+\z/ }]
+      end
+
+      #
+      # @return [Array<String>]
+      #
+      def protocols
+        @attributes['protocols']
+      end
+    end
+  end
+end

data/lib/censys/exceptions.rb

@@ -0,0 +1,16 @@
+module Censys
+  class ResponseError < RuntimeError
+  end
+
+  class AuthenticationError < ResponseError
+  end
+
+  class NotFound < ResponseError
+  end
+
+  class RateLimited < ResponseError
+  end
+
+  class InternalServerError < ResponseError
+  end
+end

data/lib/censys/ipv4.rb

@@ -0,0 +1,26 @@
+require 'censys/document'
+require 'censys/document/has_services'
+require 'censys/document/has_location'
+require 'censys/document/has_asn'
+
+require 'time'
+
+module Censys
+  class IPv4 < Document
+    include HasServices
+    include HasLocation
+    include HasASN
+
+    def ip
+      @attributes['ip']
+    end
+
+    def protocols
+      @protocols ||= Array(@attributes['protocols'])
+    end
+
+    def to_s
+      ip
+    end
+  end
+end

data/lib/censys/location.rb

@@ -0,0 +1,54 @@
+module Censys
+  class Location
+    # @return [String]
+    attr_reader :postal_code
+
+    # @return [String]
+    attr_reader :city
+
+    # @return [String]
+    attr_reader :province
+
+    # @return [String]
+    attr_reader :country
+
+    # @return [String]
+    attr_reader :continent
+
+    # @return [String]
+    attr_reader :registered_country
+
+    # @return [String]
+    attr_reader :registered_country_code
+
+    # @return [String]
+    attr_reader :timezone
+
+    # @return [Float]
+    attr_reader :latitude
+
+    # @return [Float]
+    attr_reader :longitude
+
+    #
+    # Initializes the location information.
+    #
+    # @param [Hash{String => Object}] attributes
+    #
+    def initialize(attributes)
+      @postal_code = attributes['postal_code']
+      @city        = attributes['city']
+      @province    = attributes['province']
+      @country     = attributes['country']
+      @continent   = attributes['continent']
+
+      @registered_country      = attributes['registered_country']
+      @registered_country_code = attributes['registered_country_code']
+
+      @timezone = attributes['timezone']
+
+      @latitude  = attributes['latitude']
+      @longitude = attributes['longitude']
+    end
+  end
+end

data/lib/censys/report.rb

@@ -0,0 +1 @@
+require 'censys/report/response'

data/lib/censys/report/metadata.rb

@@ -0,0 +1,41 @@
+module Censys
+  module Report
+    class Metadata
+      # @return [Fixnum]
+      attr_reader :count
+
+      # @return [Fixnum]
+      attr_reader :backend_time
+
+      # @return [Fixnum]
+      attr_reader :non_null_count
+
+      # @return [Fixnum]
+      attr_reader :other_result_count
+
+      # @return [Fixnum]
+      attr_reader :buckets
+
+      # @return [Fixnum]
+      attr_reader :error_bound
+
+      # @return [String]
+      attr_reader :query
+
+      #
+      # Initializes the report metadata.
+      #
+      # @param [Hash{String => Object}] attributes
+      #
+      def initialize(attributes)
+        @count              = attributes['count']
+        @backend_time       = attributes['backend_time']
+        @non_null_count     = attributes['nonnull_count']
+        @other_result_count = attributes['other_result_count']
+        @buckets            = attributes['buckets']
+        @error_bound        = attributes['error_bound']
+        @query              = attributes['query']
+      end
+    end
+  end
+end

data/lib/censys/report/response.rb

@@ -0,0 +1,54 @@
+require 'censys/report/metadata'
+
+module Censys
+  module Report
+    class Response
+      include Enumerable
+
+      # Response status.
+      #
+      # @return [String]
+      attr_reader :status
+
+      # Response results.
+      #
+      # @return [Hash{String => Fixnum}]
+      attr_reader :results
+
+      # Response metadata.
+      #
+      # @return [Metadata]
+      attr_reader :metadata
+
+      def initialize(response)
+        @status  = response['status']
+        @results = Hash[response['results'].map { |result|
+          [result['key'], result['doc_count']]
+        }]
+        @metadata = Metadata.new(response['metadata'])
+      end
+
+      #
+      # Determines if the response was OK.
+      #
+      # @return [Boolean]
+      #
+      def ok?
+        @status == 'ok'
+      end
+
+      #
+      # Enumerate through all response results.
+      #
+      # @yield [key, doc_count]
+      #
+      # @yieldparam [String] key
+      #
+      # @yieldparam [Fixnum] doc_count
+      #
+      def each(&block)
+        @results.each(&block)
+      end
+    end
+  end
+end

data/lib/censys/search.rb

@@ -0,0 +1 @@
+require 'censys/search/response'

data/lib/censys/search/certificate.rb

@@ -0,0 +1,19 @@
+require 'censys/search/result'
+
+module Censys
+  module Search
+    class Certificate < Result
+      def fingerprint_sha256
+        @attributes['parsed.fingerprint_sha256']
+      end
+
+      def subject_dn
+        @attributes['parsed.subject_dn']
+      end
+
+      def issuer_dn
+        @attributes['parsed.issuer_dn']
+      end
+    end
+  end
+end

data/lib/censys/search/ipv4.rb

@@ -0,0 +1,18 @@
+require 'censys/search/result'
+
+module Censys
+  module Search
+    class IPv4 < Result
+      def ip
+        @attributes['ip']
+      end
+
+      def protocols
+        @attributes['protocols']
+      end
+
+      alias to_s ip
+      alias to_str ip
+    end
+  end
+end

data/lib/censys/search/metadata.rb

@@ -0,0 +1,33 @@
+module Censys
+  module Search
+    class Metadata
+      # @return [Integer]
+      attr_reader :count
+
+      # @return [String]
+      attr_reader :query
+
+      # @return [Integer]
+      attr_reader :backend_time
+
+      # @return [Integer]
+      attr_reader :page
+
+      # @return [Integer]
+      attr_reader :pages
+
+      #
+      # Initializes the search metadata.
+      #
+      # @param [Hash{String => Object}] metadata
+      #
+      def initialize(metadata)
+        @count = metadata['count']
+        @query = metadata['query']
+        @backend_time = metadata['backend_time']
+        @page  = metadata['page']
+        @pages = metadata['pages']
+      end
+    end
+  end
+end