iptoasn 0.6.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e7d28863be637ca0ad4c3b7b7ac648828fb963f3fed53bc237e695c83876737f
-  data.tar.gz: ac59b82846fcacd8e5d584ca7d10804dfe9097cd43d889bd53339c503686c011
+  metadata.gz: d832e981a75509b4b3d425c946c9acf45b8805bef9efa92d1261ec1331203f71
+  data.tar.gz: 455040daee6e754da6ffbb49cc53577bb5a5a8bb95b01e099ee14086dc322212
 SHA512:
-  metadata.gz: 5bfdd00ff5e6d10bea42810af1861de3bcaf0eafebf933788ccac9b0056a5974bc9e457bd5cc3077b6b4d0952f9a46ed758512891beaa59c7fcec977a42f2ca1
-  data.tar.gz: b0f0ad77ee6e089c3dfa0143523f8829ddf5fe25d3c38063fc383038a9a1350c8de00847925e3cff7a2c2c9439e797f93ac021bbb67f1904d10f5c62dbd1f7d7
+  metadata.gz: a02163999424c204244d864e311faba72e523f0ba07a30b60a726cc3f88cd35c6e9a82bccd37634a42cb91b7f9568ef1cb7c3104799bed92b869e588a8941956
+  data.tar.gz: 9b7f3740a6a3e51b1f4f04c21f07dbfbe70210c36199ef6f217408563a6f1b7073bb280214da67d60c717d97f8ba5274c71c29cde0682d88f49d71f3aeb77f98

data/lib/iptoasn/asentry.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module IpToAsn
+  DB_MAIN = 'ip2asn.dat'
+  DB_COUNTRIES = 'countries.dat'
+  DB_ASNAMES = 'asnames.dat'
+end
+
+module IpToAsn
+  class AsEntry
+    attr_accessor :range_start, :range_end, :number, :country, :name_offset, :name_length
+
+    def initialize(range_start, range_end, number, country, name_offset, name_length) # rubocop:disable Metrics/ParameterLists
+      @range_start = ip_to_int(range_start)
+      @range_end = ip_to_int(range_end)
+      @number = number
+      @country = country
+      @name_offset = name_offset
+      @name_length = name_length
+    end
+
+    def serialize
+      [@range_start, @range_end, @number, @country, @name_offset, @name_length].pack('L L L C L C')
+    end
+
+    def self.deserialize(binary_data)
+      range_start, range_end, number, country, name_offset, name_length = binary_data.unpack('L L L C L C')
+      new(int_to_ip(range_start), int_to_ip(range_end), number, country, name_offset, name_length)
+    end
+
+    def self.size
+      18 # L + L + L + C + L + C = 18 bytes
+    end
+
+    def self.int_to_ip(int)
+      [24, 16, 8, 0].map { |b| (int >> b) & 0xFF }.join('.')
+    end
+
+    private
+
+    def ip_to_int(ip)
+      ip.split('.').map(&:to_i).inject(0) { |acc, num| (acc << 8) | num }
+    end
+  end
+end

data/lib/iptoasn/iptoasn.rb

@@ -0,0 +1,87 @@
+# frozen_string_literal: true
+
+module IpToAsn
+  class Lookup
+    def initialize(db_ip2asn: nil, db_country: nil, db_names: nil) # rubocop:disable Metrics/AbcSize
+      db_ip2asn ||= File.expand_path('../../data/ip2asn.dat', __dir__)
+      db_country ||= File.expand_path('../../data/countries.dat', __dir__)
+      db_names ||= File.expand_path('../../data/asnames.dat', __dir__)
+
+      @db_ip2asn = File.open(db_ip2asn, 'r')
+      @db_country = File.read(db_country)
+      @db_names = File.open(db_names, 'r')
+
+      @reserved_ranges = {
+        IPAddr.new('10.0.0.0/8') => 'RFC1918 Private',
+        IPAddr.new('172.16.0.0/12') => 'RFC1918 Private',
+        IPAddr.new('192.168.0.0/16') => 'RFC1918 Private',
+        IPAddr.new('0.0.0.0/8') => 'Current Network',
+        IPAddr.new('127.0.0.0/8') => 'Loopback',
+        IPAddr.new('169.254.0.0/16') => 'Link Local'
+      }
+
+      @entry_size = IpToAsn::AsEntry.size
+      @db_ip2asn.seek(0, IO::SEEK_END)
+      @db_total_entries = @db_ip2asn.size / @entry_size
+      @db_ip2asn.seek(0)
+    end
+
+    def ip_to_int(ip)
+      ip.split('.').map(&:to_i).inject(0) { |acc, num| (acc << 8) | num }
+    end
+
+    def as_name(name_offset, name_length)
+      @db_names.seek(name_offset, IO::SEEK_SET)
+      @db_names.read(name_length)
+    end
+
+    def country(country_index)
+      @db_country[country_index * 2, 2]
+    end
+
+    def lookup(ip)
+      reserved_range_name = find_range(
+        ranges: @reserved_ranges,
+        ip_address: ip
+      )
+
+      return { as_number: 0, country_code: 'XX', as_name: reserved_range_name } unless reserved_range_name.nil?
+
+      search_db(ip)
+    end
+
+    def search_db(ip) # rubocop:disable Metrics/AbcSize
+      ip_int = ip_to_int(ip)
+
+      left = 0
+      right = @db_total_entries - 1
+
+      while left <= right
+        mid = (left + right) / 2
+        @db_ip2asn.seek(mid * @entry_size, IO::SEEK_SET)
+        data = @db_ip2asn.read(@entry_size)
+        break if data.nil? || data.size < @entry_size
+
+        entry = IpToAsn::AsEntry.deserialize(data)
+
+        if entry.range_start.to_i <= ip_int && ip_int <= entry.range_end.to_i
+          return {
+            as_number: entry.number,
+            as_name: as_name(entry.name_offset, entry.name_length),
+            country_code: country(entry.country)
+          }
+        elsif ip_int < entry.range_start.to_i
+          right = mid - 1
+        else
+          left = mid + 1
+        end
+      end
+    end
+
+    def find_range(ranges:, ip_address:)
+      ranges.filter_map do |k, v|
+        v if k.include? ip_address
+      end.first
+    end
+  end
+end

data/lib/iptoasn.rb

@@ -1,3 +1,4 @@
 # frozen_string_literal: true
 
-require_relative 'main'
+require_relative 'iptoasn/iptoasn'
+require_relative 'iptoasn/asentry'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: iptoasn
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 1.1.0
 platform: ruby
 authors:
 - OMAR
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-01-12 00:00:00.000000000 Z
+date: 2025-02-06 00:00:00.000000000 Z
 dependencies: []
 description: This gem wraps the IP to ASN dataset. It uses lazy loading so it doesn't
   load the entire ~25M dataset all at once.
@@ -18,33 +18,9 @@ extensions: []
 extra_rdoc_files: []
 files:
 - README.md
-- lib/chunks/chunk_aa.rb
-- lib/chunks/chunk_ab.rb
-- lib/chunks/chunk_ac.rb
-- lib/chunks/chunk_ad.rb
-- lib/chunks/chunk_ae.rb
-- lib/chunks/chunk_af.rb
-- lib/chunks/chunk_ag.rb
-- lib/chunks/chunk_ah.rb
-- lib/chunks/chunk_ai.rb
-- lib/chunks/chunk_aj.rb
-- lib/chunks/chunk_ak.rb
-- lib/chunks/chunk_al.rb
-- lib/chunks/chunk_am.rb
-- lib/chunks/chunk_an.rb
-- lib/chunks/chunk_ao.rb
-- lib/chunks/chunk_ap.rb
-- lib/chunks/chunk_aq.rb
-- lib/chunks/chunk_ar.rb
-- lib/chunks/chunk_as.rb
-- lib/chunks/chunk_at.rb
-- lib/chunks/chunk_au.rb
-- lib/chunks/chunk_av.rb
-- lib/chunks/chunk_aw.rb
-- lib/chunks/chunk_ax.rb
-- lib/index.rb
 - lib/iptoasn.rb
-- lib/main.rb
+- lib/iptoasn/asentry.rb
+- lib/iptoasn/iptoasn.rb
 homepage: https://github.com/ancat/iptoasn
 licenses:
 - MIT