iptoasn 1.0.0 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9443333fd1ae8894b46d9e5373140f8735ef01baf007558bb4c9aefaa13c80a9
-  data.tar.gz: 7efe3470ae264bfff4af3693414339c7013467c8186ba63cc69a086480c65ec3
+  metadata.gz: 35a255d0f44e5d8c30e8b16f43a6827d755efd5d82309cda9097952791acf07b
+  data.tar.gz: b9e5b9bf9d8a908a34fee0f1f8ba67831e170c60ebe85e46a2eb60eed65851a3
 SHA512:
-  metadata.gz: 90669d169e556629bb2c8fffeaa7d9365d835562eae7581353068c98a1c66ee8acc07e3a93b7a844734a79fe58ff455cba57dcc45238ad2c53360586a6ea627a
-  data.tar.gz: dada5a7be40ab8fa831ba02e812414c472127b92963a0171a1b3b8ca0cc6844c0005414944bd7ad3c0e9223b2157e9c18d6b8f581b8fbb263e14bb532358a9d8
+  metadata.gz: dd0f1ec6eb277b812dcac18db162314481ebd5d9ca49d43ce8ea470152649035478e632267bc3baa96b16df1196930445234d92f178039bca4fd2162a19b23d1
+  data.tar.gz: 1e6e339f59892c5c70b33882d40bbe443b452410a185f4bf51e6a9713fad25b0ad2978a07fa26845657ac1d06aaf554309c0046a746f687046c776fcd082508d

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,90 @@
+# frozen_string_literal: true
+
+module IpToAsn
+  class Lookup
+    def initialize(db_ip2asn: nil, db_country: nil, db_names: nil) # rubocop:disable Metrics/AbcSize
+      if db_ip2asn.nil? || db_country.nil? || db_names.nil?
+        require 'iptoasn/data'
+        db_ip2asn = IpToAsn::Data.ip2asn
+        db_country = IpToAsn::Data.countries
+        db_names = IpToAsn::Data.asnames
+      end
+
+      @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: 1.0.0
+  version: 1.2.0
 platform: ruby
 authors:
 - OMAR
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-01-28 00:00:00.000000000 Z
+date: 2025-02-08 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