ronin-db-activerecord 0.1.0 → 0.1.1

data/db/migrate/{0029_create_ronin_http_header_names_table.rb → 0031_create_ronin_http_header_names_table.rb}

@@ -1,3 +1,4 @@
+# frozen_string_literal: true
 #
 # ronin-db-activerecord - ActiveRecord backend for the Ronin Database.
 #
@@ -19,6 +20,9 @@
 # along with ronin-db-activerecord.  If not, see <https://www.gnu.org/licenses/>.
 #
 
+#
+# Creates the `ronin_http_header_names` table.
+#
 class CreateRoninHttpHeaderNamesTable < ActiveRecord::Migration[7.0]
 
   def change

data/db/migrate/{0030_create_ronin_http_request_headers_table.rb → 0032_create_ronin_http_request_headers_table.rb}

@@ -1,3 +1,4 @@
+# frozen_string_literal: true
 #
 # ronin-db-activerecord - ActiveRecord backend for the Ronin Database.
 #
@@ -19,6 +20,9 @@
 # along with ronin-db-activerecord.  If not, see <https://www.gnu.org/licenses/>.
 #
 
+#
+# Creates the `ronin_http_request_headers` table.
+#
 class CreateRoninHttpRequestHeadersTable < ActiveRecord::Migration[7.0]
 
   def change

data/db/migrate/{0031_create_ronin_http_response_headers_table.rb → 0033_create_ronin_http_response_headers_table.rb}

@@ -1,3 +1,4 @@
+# frozen_string_literal: true
 #
 # ronin-db-activerecord - ActiveRecord backend for the Ronin Database.
 #
@@ -19,6 +20,9 @@
 # along with ronin-db-activerecord.  If not, see <https://www.gnu.org/licenses/>.
 #
 
+#
+# Creates the `ronin_http_response_headers` table.
+#
 class CreateRoninHttpResponseHeadersTable < ActiveRecord::Migration[7.0]
 
   def change

data/db/migrate/0034_create_ronin_service_credentials_table.rb

@@ -1,3 +1,4 @@
+# frozen_string_literal: true
 #
 # ronin-db-activerecord - ActiveRecord backend for the Ronin Database.
 #
@@ -19,6 +20,9 @@
 # along with ronin-db-activerecord.  If not, see <https://www.gnu.org/licenses/>.
 #
 
+#
+# Creates the `ronin_service_credentials` table.
+#
 class CreateRoninServiceCredentialsTable < ActiveRecord::Migration[7.0]
 
   def change

data/db/migrate/0035_create_ronin_web_credentials_table.rb

@@ -1,3 +1,4 @@
+# frozen_string_literal: true
 #
 # ronin-db-activerecord - ActiveRecord backend for the Ronin Database.
 #
@@ -19,6 +20,9 @@
 # along with ronin-db-activerecord.  If not, see <https://www.gnu.org/licenses/>.
 #
 
+#
+# Creates the `ronin_web_credentials` table.
+#
 class CreateRoninWebCredentialsTable < ActiveRecord::Migration[7.0]
 
   def change

data/gemspec.yml

@@ -1,5 +1,5 @@
 name: ronin-db-activerecord
-version: 0.1.0
+version: 0.1.1
 summary: ActiveRecord backend for the Ronin Database
 description:
   ronin-db-activerecord contains ActiveRecord models and migrations for the

data/lib/ronin/db/advisory.rb

@@ -56,9 +56,9 @@ module Ronin
       attribute :year, :integer
       validates :year, allow_nil:  true,
                        comparison: {
-                                     greater_than: 1990,
-                                     less_than_or_equal_to: Date.today.year
-                                   }
+                         greater_than: 1990,
+                         less_than_or_equal_to: Date.today.year
+                       }
 
       # @!attribute [rw] identifier
       #   The advisory identifier
@@ -84,7 +84,7 @@ module Ronin
         #   The ID does not appear to be a valid security ID.
         #
         def self.parse(string)
-          if (match = string.match(/\A([A-Z]+)-(\d{4})[:-]([0-9][0-9-]+)\z/))
+          if    (match = string.match(/\A([A-Z]+)-(\d{4})[:-]([0-9][0-9-]+)\z/))
             {
               id:         match[0],
               prefix:     match[1],

data/lib/ronin/db/asn.rb

@@ -142,7 +142,7 @@ module Ronin
       #
       # Searches for all ASNs with the matching name.
       #
-      # @param [String] name 
+      # @param [String] name
       #   The name to search for.
       #
       # @return [Array<ASN>]
@@ -159,11 +159,11 @@ module Ronin
       # @return [ASN, nil]
       #
       def self.containing_ip(ip)
-        ip = IPAddr.new(ip) unless ip.kind_of?(IPAddr)
+        ip      = IPAddr.new(ip) unless ip.kind_of?(IPAddr)
         ip_hton = ip.hton
 
         range_start_hton = self.arel_table[:range_start_hton]
-        range_end_hton  = self.arel_table[:range_end_hton]
+        range_end_hton   = self.arel_table[:range_end_hton]
 
         where(range_start_hton.lteq(ip_hton).and(range_end_hton.gteq(ip_hton))).first
       end

data/lib/ronin/db/ip_address.rb

@@ -262,6 +262,7 @@ module Ronin
                       begin
                         IPAddr.new(self.address)
                       rescue IPAddr::InvalidAddressError
+                        nil
                       end
                     end
       end
@@ -345,7 +346,6 @@ require 'ronin/db/host_name_ip_address'
 require 'ronin/db/host_name'
 require 'ronin/db/open_port'
 require 'ronin/db/port'
-require 'ronin/db/host_name'
 require 'ronin/db/os_guess'
 require 'ronin/db/os'
 require 'ronin/db/asn'

data/lib/ronin/db/mac_address.rb

@@ -78,8 +78,8 @@ module Ronin
       # @api public
       #
       def to_i
-        self.address.split(':').inject(0) do |bits,char|
-          bits = ((bits << 8) | char.hex)
+        self.address.split(':').reduce(0) do |bits,char|
+          ((bits << 8) | char.hex)
         end
       end
 

data/lib/ronin/db/migrations.rb

@@ -25,6 +25,9 @@ require 'active_record/migration'
 
 module Ronin
   module DB
+    #
+    # Handles migrating the database.
+    #
     module Migrations
       #
       # The current migration version of the database.
@@ -56,7 +59,7 @@ module Ronin
         context.migrate(target_version)
       end
 
-      # 
+      #
       # Explicitly migrates up the database to the target version.
       #
       # @param [Integer, nil] target_version
@@ -68,9 +71,9 @@ module Ronin
         context.up(target_version,&block)
       end
 
-      # 
+      #
       # Explicitly migrates down the database to the target version.
-      # 
+      #
       # @param [Integer, nil] target_version
       #   The desired target version.
       #
@@ -80,7 +83,7 @@ module Ronin
         context.down(target_version,&block)
       end
 
-      # 
+      #
       # Rollbacks the last number of migrations.
       #
       # @param [Integer] steps
@@ -92,7 +95,7 @@ module Ronin
         context.rollback(steps)
       end
 
-      # 
+      #
       # Applies the next number of migrations.
       #
       # @param [Integer] steps
@@ -104,8 +107,6 @@ module Ronin
         context.foreward(steps)
       end
 
-      private
-
       # Path to the `db/migrate/` directory in `ronin-db-activerecord`.
       DIR = File.expand_path('../../../db/migrate',__dir__)
 
@@ -129,8 +130,10 @@ module Ronin
       #
       # @return [MigrationContext]
       #
+      # @api private
+      #
       def self.context
-        @migrations ||= MigrationContext.new
+        @context ||= MigrationContext.new
       end
     end
   end

data/lib/ronin/db/model/has_unique_name.rb

@@ -44,11 +44,9 @@ module Ronin
         # @api semipublic
         #
         def self.included(base)
-          base.send :include, Model,
-                              HasName::InstanceMethods
+          base.send :include, Model, HasName::InstanceMethods
 
-          base.send :extend,  HasName::ClassMethods,
-                              HasUniqueName::ClassMethods
+          base.send :extend, HasName::ClassMethods, HasUniqueName::ClassMethods
 
           base.module_eval do
             # The name of the model

data/lib/ronin/db/model/importable.rb

@@ -37,6 +37,9 @@ module Ronin
           model.extend ClassMethods
         end
 
+        #
+        # Class-methods which will be added to the model.
+        #
         module ClassMethods
           #
           # Looks up a record with the given value.

data/lib/ronin/db/model/last_scanned_at.rb

@@ -21,6 +21,9 @@
 module Ronin
   module DB
     module Model
+      #
+      # Mixin which adds a `last_scanned_at` attribute to a model.
+      #
       module LastScannedAt
         #
         # @!attribute [rw] last_scanned_at

data/lib/ronin/db/model.rb

@@ -20,6 +20,9 @@
 
 module Ronin
   module DB
+    #
+    # Mixin for all `ronin-db-activerecord` models.
+    #
     module Model
       #
       # Sets the models `table_name_prefix` to `ronin_`.

data/lib/ronin/db/models.rb

@@ -56,6 +56,9 @@ require 'ronin/db/vulnerability'
 
 module Ronin
   module DB
+    #
+    # Manages all models defined in `ronin-db-activerecord`.
+    #
     module Models
       ALL = [
         Address,

data/lib/ronin/db/password.rb

@@ -109,13 +109,13 @@ module Ronin
       #
       # @example
       #   pass = Password.new(plain_text: 'secret')
-      #   
+      #
       #   pass.digest(:sha1)
       #   # => "e5e9fa1ba31ecd1ae84f75caaa474f3a663f05f4"
-      #   
+      #
       #   pass.digest(:sha1, prepend_salt: "A\x90\x00")
       #   # => "e2817656a48c49f24839ccf9295b389d8f985904"
-      #   
+      #
       #   pass.digest(:sha1, append_salt: "BBBB")
       #   # => "aa6ca21e446d425fc044bbb26e950a788444a5b8"
       #

data/lib/ronin/db/url.rb

@@ -87,7 +87,7 @@ module Ronin
 
       # @!attribute [rw] fragment
       #   The fragment of the URL.
-      #   
+      #
       #   @return [String, nil]
       attribute :fragment, :string
 
@@ -362,33 +362,30 @@ module Ronin
                         number:   uri.port
                       )
                     end
+        path      = normalized_path(uri)
+        query     = uri.query
+        fragment  = uri.fragment
 
-        path     = normalized_path(uri)
-        query    = uri.query
-        fragment = uri.fragment
-
-        query_params = []
-        
-        if uri.respond_to?(:query_params)
-          # find or create the URL query params
-          uri.query_params.each do |name,value|
-            query_params << URLQueryParam.new(
-              name:  URLQueryParamName.find_or_create_by(name: name),
-              value: value
-            )
-          end
-        end
-
-        # find or create the URL
+        # try to query a pre-existing URI then fallback to creating the URL
+        # with query params.
         return find_or_create_by(
           scheme:       scheme,
           host_name:    host_name,
           port:         port,
           path:         path,
           query:        query,
-          fragment:     fragment,
-          query_params: query_params
-        )
+          fragment:     fragment
+        ) do |new_url|
+          if uri.respond_to?(:query_params)
+            # find or create the URL query params
+            uri.query_params.each do |name,value|
+              new_url.query_params << URLQueryParam.new(
+                name:  URLQueryParamName.find_or_create_by(name: name),
+                value: value
+              )
+            end
+          end
+        end
       end
 
       #
@@ -430,9 +427,11 @@ module Ronin
         scheme = if self.scheme
                    self.scheme.name
                  end
+
         host = if self.host_name
                  self.host_name.name
                end
+
         port = if self.port
                  self.port.number
                end
@@ -460,8 +459,6 @@ module Ronin
         self.to_uri.to_s
       end
 
-      protected
-
       #
       # Normalizes the path of a URI.
       #

data/lib/ronin/db/url_scheme.rb

@@ -34,7 +34,7 @@ module Ronin
       include Model::HasUniqueName
 
       self.table_name = 'ronin_url_schemes'
-      
+
       # @!attribute [rw] id
       #   The primary key of the URL scheme.
       #

data/lib/ronin/db/vulnerability.rb

@@ -28,6 +28,10 @@ require 'active_record'
 
 module Ronin
   module DB
+    #
+    # Represents a discovered vulnerability associated with a URL, open port,
+    # host name, IP, or MAC address.
+    #
     class Vulnerability < ActiveRecord::Base
 
       include Model

data/ronin-db-activerecord.gemspec

@@ -1,4 +1,4 @@
-# encoding: utf-8
+# frozen_string_literal: true
 
 require 'yaml'
 
@@ -22,7 +22,7 @@ Gem::Specification.new do |gem|
   gem.homepage    = gemspec['homepage']
   gem.metadata    = gemspec['metadata'] if gemspec['metadata']
 
-  glob = lambda { |patterns| gem.files & Dir[*patterns] }
+  glob = ->(patterns) { gem.files & Dir[*patterns] }
 
   gem.files  = `git ls-files`.split($/)
   gem.files  = glob[gemspec['files']] if gemspec['files']
@@ -46,7 +46,7 @@ Gem::Specification.new do |gem|
   gem.required_rubygems_version = gemspec['required_rubygems_version']
   gem.post_install_message      = gemspec['post_install_message']
 
-  split = lambda { |string| string.split(/,\s*/) }
+  split = ->(string) { string.split(/,\s*/) }
 
   if gemspec['dependencies']
     gemspec['dependencies'].each do |name,versions|

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ronin-db-activerecord
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Postmodern
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-02-01 00:00:00.000000000 Z
+date: 2023-04-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: uri-query_params
@@ -66,6 +66,7 @@ files:
 - ".github/workflows/ruby.yml"
 - ".gitignore"
 - ".rspec"
+- ".rubocop.yml"
 - ".ruby-version"
 - ".yardopts"
 - COPYING.txt
@@ -73,39 +74,39 @@ files:
 - Gemfile
 - README.md
 - Rakefile
-- db/migrate/0001_create_ronin_ip_address_mac_addresses_table.rb
-- db/migrate/0002_create_ronin_vulnerabilities_table.rb
-- db/migrate/0003_create_ronin_url_schemes_table.rb
-- db/migrate/0004_create_ronin_url_query_param_names_table.rb
-- db/migrate/0005_create_ronin_user_names_table.rb
-- db/migrate/0006_create_ronin_software_vendors_table.rb
-- db/migrate/0007_create_ronin_advisories_table.rb
-- db/migrate/0008_create_ronin_host_name_ip_addresses_table.rb
-- db/migrate/0009_create_ronin_host_names_table.rb
-- db/migrate/0010_create_ronin_arches_table.rb
-- db/migrate/0011_create_ronin_email_addresses_table.rb
-- db/migrate/0012_create_ronin_oses_table.rb
-- db/migrate/0013_create_ronin_organizations_table.rb
-- db/migrate/0014_create_ronin_ip_addresses_table.rb
-- db/migrate/0015_create_ronin_os_guesses_table.rb
-- db/migrate/0016_create_ronin_url_query_params_table.rb
-- db/migrate/0017_create_ronin_passwords_table.rb
-- db/migrate/0018_create_ronin_open_ports_table.rb
-- db/migrate/0019_create_ronin_urls_table.rb
-- db/migrate/0020_create_ronin_softwares_table.rb
-- db/migrate/0021_create_ronin_mac_addresses_table.rb
-- db/migrate/0022_create_ronin_countries_table.rb
-- db/migrate/0023_create_ronin_services_table.rb
-- db/migrate/0024_create_ronin_credentials_table.rb
-- db/migrate/0025_create_ronin_ports_table.rb
+- db/migrate/0001_create_ronin_ip_addresses_table.rb
+- db/migrate/0002_create_ronin_mac_addresses_table.rb
+- db/migrate/0003_create_ronin_ip_address_mac_addresses_table.rb
+- db/migrate/0004_create_ronin_host_names_table.rb
+- db/migrate/0005_create_ronin_ports_table.rb
+- db/migrate/0006_create_ronin_services_table.rb
+- db/migrate/0007_create_ronin_open_ports_table.rb
+- db/migrate/0008_create_ronin_url_schemes_table.rb
+- db/migrate/0009_create_ronin_urls_table.rb
+- db/migrate/0010_create_ronin_advisories_table.rb
+- db/migrate/0011_create_ronin_vulnerabilities_table.rb
+- db/migrate/0012_create_ronin_url_query_param_names_table.rb
+- db/migrate/0013_create_ronin_user_names_table.rb
+- db/migrate/0014_create_ronin_software_vendors_table.rb
+- db/migrate/0015_create_ronin_host_name_ip_addresses_table.rb
+- db/migrate/0016_create_ronin_arches_table.rb
+- db/migrate/0017_create_ronin_email_addresses_table.rb
+- db/migrate/0018_create_ronin_oses_table.rb
+- db/migrate/0019_create_ronin_organizations_table.rb
+- db/migrate/0020_create_ronin_os_guesses_table.rb
+- db/migrate/0021_create_ronin_url_query_params_table.rb
+- db/migrate/0022_create_ronin_passwords_table.rb
+- db/migrate/0023_create_ronin_softwares_table.rb
+- db/migrate/0024_create_ronin_countries_table.rb
+- db/migrate/0025_create_ronin_credentials_table.rb
 - db/migrate/0026_create_ronin_asns_table.rb
 - db/migrate/0027_create_ronin_http_query_param_names_table.rb
-- db/migrate/0028_create_ronin_http_query_params_table.rb
-- db/migrate/0029_create_ronin_http_header_names_table.rb
-- db/migrate/0030_create_ronin_http_request_headers_table.rb
-- db/migrate/0031_create_ronin_http_response_headers_table.rb
-- db/migrate/0032_create_ronin_http_requests_table.rb
-- db/migrate/0033_create_ronin_http_responses_table.rb
+- db/migrate/0028_create_ronin_http_responses_table.rb
+- db/migrate/0029_create_ronin_http_requests_table.rb
+- db/migrate/0030_create_ronin_http_query_params_table.rb
+- db/migrate/0031_create_ronin_http_header_names_table.rb
+- db/migrate/0032_create_ronin_http_request_headers_table.rb
+- db/migrate/0033_create_ronin_http_response_headers_table.rb
 - db/migrate/0034_create_ronin_service_credentials_table.rb
 - db/migrate/0035_create_ronin_web_credentials_table.rb
 - gemspec.yml

data/db/migrate/0007_create_ronin_advisories_table.rb

@@ -1,42 +0,0 @@
-#
-# ronin-exploits - A Ruby library for ronin-rb that provides exploitation and
-# payload crafting functionality.
-#
-# Copyright (c) 2007-2022-2023 Hal Brodigan (postmodern.mod3 at gmail.com)
-#
-# This file is part of ronin-exploits.
-#
-# ronin-exploits is free software: you can redistribute it and/or modify
-# it under the terms of the GNU General Public License as published by
-# the Free Software Foundation, either version 3 of the License, or
-# (at your option) any later version.
-#
-# ronin-exploits is distributed in the hope that it will be useful,
-# but WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-# GNU General Public License for more details.
-#
-# You should have received a copy of the GNU General Public License
-# along with ronin-exploits.  If not, see <https://www.gnu.org/licenses/>
-#
-
-#
-# @api private
-#
-class CreateRoninAdvisoriesTable < ActiveRecord::Migration[7.0]
-
-  def change
-    create_table :ronin_advisories, id: false, if_not_exists: true do |t|
-      t.string :id, primary_key: true, null: false
-
-      t.string :prefix, null: false
-      t.integer :year, null: true
-      t.string :identifier, null: false
-
-      t.index :id, unique: true
-      t.index :publisher
-      t.index :year
-    end
-  end
-
-end