geoip2_compat 0.0.3

data/ext/geoip2_compat/maxminddb.h

@@ -0,0 +1,223 @@
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#ifndef MAXMINDDB_H
+#define MAXMINDDB_H
+
+#ifndef _POSIX_C_SOURCE
+#define _POSIX_C_SOURCE 200112L
+#endif
+
+#include <maxminddb_config.h>
+#include <stdarg.h>
+#include <stdbool.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <sys/types.h>
+
+#ifdef _WIN32
+#include <winsock2.h>
+#include <ws2tcpip.h>
+/* libmaxminddb package version from configure */
+#define PACKAGE_VERSION "1.0.3"
+
+typedef ADDRESS_FAMILY sa_family_t;
+
+#if defined(_MSC_VER)
+/* MSVC doesn't define signed size_t, copy it from configure */
+#define ssize_t int
+
+/* MSVC doesn't support restricted pointers */
+#define restrict
+#endif
+#else
+#include <netdb.h>
+#include <netinet/in.h>
+#include <sys/socket.h>
+#endif
+
+#define MMDB_DATA_TYPE_EXTENDED (0)
+#define MMDB_DATA_TYPE_POINTER (1)
+#define MMDB_DATA_TYPE_UTF8_STRING (2)
+#define MMDB_DATA_TYPE_DOUBLE (3)
+#define MMDB_DATA_TYPE_BYTES (4)
+#define MMDB_DATA_TYPE_UINT16 (5)
+#define MMDB_DATA_TYPE_UINT32 (6)
+#define MMDB_DATA_TYPE_MAP (7)
+#define MMDB_DATA_TYPE_INT32 (8)
+#define MMDB_DATA_TYPE_UINT64 (9)
+#define MMDB_DATA_TYPE_UINT128 (10)
+#define MMDB_DATA_TYPE_ARRAY (11)
+#define MMDB_DATA_TYPE_CONTAINER (12)
+#define MMDB_DATA_TYPE_END_MARKER (13)
+#define MMDB_DATA_TYPE_BOOLEAN (14)
+#define MMDB_DATA_TYPE_FLOAT (15)
+
+/* flags for open */
+#define MMDB_MODE_MMAP (1)
+#define MMDB_MODE_MASK (7)
+
+/* error codes */
+#define MMDB_SUCCESS (0)
+#define MMDB_FILE_OPEN_ERROR (1)
+#define MMDB_CORRUPT_SEARCH_TREE_ERROR (2)
+#define MMDB_INVALID_METADATA_ERROR (3)
+#define MMDB_IO_ERROR (4)
+#define MMDB_OUT_OF_MEMORY_ERROR (5)
+#define MMDB_UNKNOWN_DATABASE_FORMAT_ERROR (6)
+#define MMDB_INVALID_DATA_ERROR (7)
+#define MMDB_INVALID_LOOKUP_PATH_ERROR (8)
+#define MMDB_LOOKUP_PATH_DOES_NOT_MATCH_DATA_ERROR (9)
+#define MMDB_INVALID_NODE_NUMBER_ERROR (10)
+#define MMDB_IPV6_LOOKUP_IN_IPV4_DATABASE_ERROR (11)
+
+#if !(MMDB_UINT128_IS_BYTE_ARRAY)
+#if MMDB_UINT128_USING_MODE
+typedef unsigned int mmdb_uint128_t __attribute__ ((__mode__(TI)));
+#else
+typedef unsigned __int128 mmdb_uint128_t;
+#endif
+#endif
+
+/* This is a pointer into the data section for a given IP address lookup */
+typedef struct MMDB_entry_s {
+    struct MMDB_s *mmdb;
+    uint32_t offset;
+} MMDB_entry_s;
+
+typedef struct MMDB_lookup_result_s {
+    bool found_entry;
+    MMDB_entry_s entry;
+    uint16_t netmask;
+} MMDB_lookup_result_s;
+
+typedef struct MMDB_entry_data_s {
+    bool has_data;
+    union {
+        uint32_t pointer;
+        const char *utf8_string;
+        double double_value;
+        const uint8_t *bytes;
+        uint16_t uint16;
+        uint32_t uint32;
+        int32_t int32;
+        uint64_t uint64;
+#if MMDB_UINT128_IS_BYTE_ARRAY
+        uint8_t uint128[16];
+#else
+        mmdb_uint128_t uint128;
+#endif
+        bool boolean;
+        float float_value;
+    };
+    /* This is a 0 if a given entry cannot be found. This can only happen
+     * when a call to MMDB_(v)get_value() asks for hash keys or array
+     * indices that don't exist. */
+    uint32_t offset;
+    /* This is the next entry in the data section, but it's really only
+     * relevant for entries that part of a larger map or array
+     * struct. There's no good reason for an end user to look at this
+     * directly. */
+    uint32_t offset_to_next;
+    /* This is only valid for strings, utf8_strings or binary data */
+    uint32_t data_size;
+    /* This is an MMDB_DATA_TYPE_* constant */
+    uint32_t type;
+} MMDB_entry_data_s;
+
+/* This is the return type when someone asks for all the entry data in a map or array */
+typedef struct MMDB_entry_data_list_s {
+    MMDB_entry_data_s entry_data;
+    struct MMDB_entry_data_list_s *next;
+} MMDB_entry_data_list_s;
+
+typedef struct MMDB_description_s {
+    const char *language;
+    const char *description;
+} MMDB_description_s;
+
+typedef struct MMDB_metadata_s {
+    uint32_t node_count;
+    uint16_t record_size;
+    uint16_t ip_version;
+    const char *database_type;
+    struct {
+        size_t count;
+        const char **names;
+    } languages;
+    uint16_t binary_format_major_version;
+    uint16_t binary_format_minor_version;
+    uint64_t build_epoch;
+    struct {
+        size_t count;
+        MMDB_description_s **descriptions;
+    } description;
+} MMDB_metadata_s;
+
+typedef struct MMDB_ipv4_start_node_s {
+    uint16_t netmask;
+    uint32_t node_value;
+} MMDB_ipv4_start_node_s;
+
+typedef struct MMDB_s {
+    uint32_t flags;
+    const char *filename;
+    ssize_t file_size;
+    const uint8_t *file_content;
+    const uint8_t *data_section;
+    uint32_t data_section_size;
+    const uint8_t *metadata_section;
+    uint32_t metadata_section_size;
+    uint16_t full_record_byte_size;
+    uint16_t depth;
+    MMDB_ipv4_start_node_s ipv4_start_node;
+    MMDB_metadata_s metadata;
+} MMDB_s;
+
+typedef struct MMDB_search_node_s {
+    uint64_t left_record;
+    uint64_t right_record;
+} MMDB_search_node_s;
+
+    /* *INDENT-OFF* */
+    /* --prototypes automatically generated by dev-bin/regen-prototypes.pl - don't remove this comment */
+    extern int MMDB_open(const char *const filename, uint32_t flags, MMDB_s *const mmdb);
+    extern MMDB_lookup_result_s MMDB_lookup_string(MMDB_s *const mmdb,
+                                                   const char *const ipstr,
+                                                   int *const gai_error,
+                                                   int *const mmdb_error);
+    extern MMDB_lookup_result_s MMDB_lookup_sockaddr(
+               MMDB_s *const mmdb,
+               const struct sockaddr *const sockaddr,
+               int *const mmdb_error);
+    extern int MMDB_read_node(MMDB_s *const mmdb, uint32_t node_number,
+                              MMDB_search_node_s *const node);
+    extern int MMDB_get_value(MMDB_entry_s *const start,
+                              MMDB_entry_data_s *const entry_data,
+                              ...);
+    extern int MMDB_vget_value(MMDB_entry_s *const start,
+                               MMDB_entry_data_s *const entry_data,
+                               va_list va_path);
+    extern int MMDB_aget_value(MMDB_entry_s *const start,
+                               MMDB_entry_data_s *const entry_data,
+                               const char *const *const path);
+    extern int MMDB_get_metadata_as_entry_data_list(
+               MMDB_s *const mmdb, MMDB_entry_data_list_s **const entry_data_list);
+    extern int MMDB_get_entry_data_list(
+               MMDB_entry_s *start, MMDB_entry_data_list_s **const entry_data_list);
+    extern void MMDB_free_entry_data_list(MMDB_entry_data_list_s *const entry_data_list);
+    extern void MMDB_close(MMDB_s *const mmdb);
+    extern const char *MMDB_lib_version(void);
+    extern int MMDB_dump_entry_data_list(FILE *const stream,
+                                         MMDB_entry_data_list_s *const entry_data_list,
+                                         int indent);
+    extern const char *MMDB_strerror(int error_code);
+    /* --prototypes end - don't remove this comment-- */
+    /* *INDENT-ON* */
+
+#endif                          /* MAXMINDDB_H */
+
+#ifdef __cplusplus
+}
+#endif

data/ext/geoip2_compat/maxminddb_config.h

@@ -0,0 +1,14 @@
+#ifndef MAXMINDDB_CONFIG_H
+#define MAXMINDDB_CONFIG_H
+
+#ifndef MMDB_UINT128_USING_MODE
+/* Define as 1 if we we use unsigned int __atribute__ ((__mode__(TI))) for uint128 values */
+#define MMDB_UINT128_USING_MODE 0
+#endif
+
+#ifndef MMDB_UINT128_IS_BYTE_ARRAY
+/* Define as 1 if we don't have an unsigned __int128 type */
+#undef MMDB_UINT128_IS_BYTE_ARRAY
+#endif
+
+#endif                          /* MAXMINDDB_CONFIG_H */

data/geoip2_compat.gemspec

@@ -0,0 +1,21 @@
+Gem::Specification.new do |s|
+  s.name         = "geoip2_compat"
+  s.version      = "0.0.3"
+
+  s.licenses     = ["MIT"]
+
+  s.authors      = ["Dirkjan Bussink"]
+  s.email        = ["d.bussink@gmail.com"]
+
+  s.summary      = "Bindings for libmaxminddb to access GeoIP2 database files"
+  s.description  = "Lookup IPv4 and IPv6 addresses in Maxmind GeoIP2 database files"
+  s.homepage     = "http://github.com/dbussink/geoip2_compat"
+
+  s.files        = `git ls-files`.split("\n")
+  s.extensions   = ['ext/geoip2_compat/extconf.rb']
+  s.require_paths = ["lib"]
+
+  s.add_development_dependency 'rake', '~>10.0'
+  s.add_development_dependency 'rake-compiler', '~> 0'
+  s.add_development_dependency 'test-unit', '~> 0'
+end

data/lib/geoip2_compat.rb

@@ -0,0 +1 @@
+require "geoip2_compat/geoip2_compat"

data/test/test_geoip2_compat.rb

@@ -0,0 +1,49 @@
+require 'geoip2_compat'
+require 'test/unit'
+
+class GeoIP2Compat
+  class TestGeoIP2Compat < Test::Unit::TestCase
+
+    def setup
+      @db = GeoIP2Compat.new("test/GeoLite2-City.mmdb")
+    end
+
+    def test_initialize_success
+      assert_equal "test/GeoLite2-City.mmdb", @db.path
+    end
+
+    def test_initialize_failure
+      assert_raise ArgumentError do
+        GeoIP2Compat.new {}
+      end
+    end
+
+    def test_lookup_ipv4
+      res = {
+        :city=>"San Francisco",
+        :country_code=>"US",
+        :country_name=>"United States",
+        :latitude=>37.7697,
+        :longitude=>-122.3933,
+        :postal_code=>"94107",
+        :region=>"CA",
+        :region_name=>"California",
+      }
+      assert_equal res, @db.lookup("192.30.252.128")
+    end
+
+    def test_lookup_ipv6
+      res = {
+        :city=>"Grave",
+        :country_code=>"NL",
+        :country_name=>"Netherlands",
+        :latitude=>51.759,
+        :longitude=>5.7388,
+        :postal_code=>"5361",
+        :region=>"NB",
+        :region_name=>"North Brabant",
+      }
+      assert_equal res, @db.lookup("2001:888:0:1::888")
+    end
+  end
+end

metadata

@@ -0,0 +1,102 @@
+--- !ruby/object:Gem::Specification
+name: geoip2_compat
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+platform: ruby
+authors:
+- Dirkjan Bussink
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-12-29 00:00:00.000000000 Z
+dependencies:
+- !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'
+- !ruby/object:Gem::Dependency
+  name: rake-compiler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: test-unit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Lookup IPv4 and IPv6 addresses in Maxmind GeoIP2 database files
+email:
+- d.bussink@gmail.com
+executables: []
+extensions:
+- ext/geoip2_compat/extconf.rb
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".travis.yml"
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- ext/geoip2_compat/extconf.rb
+- ext/geoip2_compat/geoip2_compat.c
+- ext/geoip2_compat/maxminddb-compat-util.h
+- ext/geoip2_compat/maxminddb.c
+- ext/geoip2_compat/maxminddb.h
+- ext/geoip2_compat/maxminddb_config.h
+- geoip2_compat.gemspec
+- lib/geoip2_compat.rb
+- test/test_geoip2_compat.rb
+homepage: http://github.com/dbussink/geoip2_compat
+licenses:
+- MIT
+metadata: {}
+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: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Bindings for libmaxminddb to access GeoIP2 database files
+test_files: []