hive_geoip2 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 8db3b40343a6db6c1c0ffc20f8f31fd8a62d82e5
+  data.tar.gz: 7627dbb58b18b2b34410993bebe29127f7ef36e6
+SHA512:
+  metadata.gz: 6ba971888c29661ff3fddf3d46d00f67d0d734c82af1ef3fb1967f3eaa0d4b1fb5ae34f47b4e83f79563ae5be1a35c15a730aa42ca019dd0ec11c0327a321f37
+  data.tar.gz: 6c22963486413a1f7448d8a3483b8de9b4d97937b1cf1fd2556a0edc8defcf5eb33507db0e5660ad62d642336cb6449c9931ee1a930c35f638102be40f157107

data/ext/hive_geoip2/extconf.rb

@@ -0,0 +1,9 @@
+require 'mkmf'
+
+dir_config('maxminddb')
+
+unless have_header('maxminddb.h') && have_library('maxminddb')
+  abort 'Missing libmaxminddb'
+end
+
+create_makefile('hive_geoip2')

data/ext/hive_geoip2/hive_geoip2.c

@@ -0,0 +1,392 @@
+#define RSTRING_NOT_MODIFIED
+
+#include <ruby/ruby.h>
+#include <ruby/encoding.h>
+
+#include "maxminddb.h"
+
+#define WORDS_64 (64 / CHAR_BIT / SIZEOF_LONG)
+#define WORDS_128 (128/ CHAR_BIT / SIZEOF_LONG)
+
+VALUE rb_mHive;
+VALUE rb_cGeoIP2;
+
+static void guard_parse_data_list(VALUE arg);
+static MMDB_entry_data_list_s * parse_data_list(
+  MMDB_entry_data_list_s *data_list, VALUE *ret_obj
+);
+static void mmdb_try_open(char *db_path, MMDB_s *mmdb);
+static inline bool mmdb_is_closed(MMDB_s *mmdb);
+static inline void mmdb_close(MMDB_s *mmdb);
+static VALUE mmdb_lookup(MMDB_s *mmdb, char *ip_addr, bool cleanup);
+
+static VALUE rb_hive_geo_lookup(VALUE self, VALUE ip_arg);
+static VALUE rb_hive_geo_lookup2(VALUE self, VALUE ip_arg, VALUE db_arg);
+static VALUE rb_hive_geo_is_closed(VALUE self);
+static VALUE rb_hive_geo_close(VALUE self);
+static void rb_hive_geo_free(MMDB_s *mmdb);
+static VALUE rb_hive_geo_alloc(VALUE self);
+static VALUE rb_hive_geo_init(VALUE self, VALUE db_arg);
+
+struct args_parse_data_list {
+  MMDB_entry_data_list_s *data_list;
+  VALUE *ret_obj;
+};
+
+static void guard_parse_data_list(VALUE arg) {
+  struct args_parse_data_list *args = (struct args_parse_data_list *)arg;
+  
+  parse_data_list(args->data_list, args->ret_obj);
+}
+
+static MMDB_entry_data_list_s *
+parse_data_list(MMDB_entry_data_list_s *data_list, VALUE *ret_obj) {
+  switch (data_list->entry_data.type) {
+    case MMDB_DATA_TYPE_MAP:
+    {
+      uint32_t size = data_list->entry_data.data_size;
+      
+      VALUE val;
+      
+      VALUE hash = rb_hash_new();
+      
+      for (data_list = data_list->next; size && data_list; size--) {
+        VALUE key = rb_enc_str_new(
+          data_list->entry_data.utf8_string,
+          data_list->entry_data.data_size,
+          rb_utf8_encoding()
+        );
+        
+        data_list = data_list->next;
+        data_list = parse_data_list(data_list, &val);
+        
+        rb_hash_aset(hash, key, val);
+        
+      }
+      
+      *ret_obj = hash;
+      
+      break;
+    }
+    
+    case MMDB_DATA_TYPE_ARRAY:
+    {
+      uint32_t size = data_list->entry_data.data_size;
+      
+      VALUE val;
+      
+      VALUE ary = rb_ary_new();
+      
+      for (data_list = data_list->next; size && data_list; size--) {
+        data_list = parse_data_list(data_list, &val);
+        rb_ary_push(ary, val);
+      }
+      
+      *ret_obj = ary;
+      
+      break;
+    }
+    
+    case MMDB_DATA_TYPE_UTF8_STRING:
+    {
+      *ret_obj = rb_enc_str_new(
+        data_list->entry_data.utf8_string,
+        data_list->entry_data.data_size,
+        rb_utf8_encoding()
+      );
+      
+      data_list = data_list->next;
+      
+      break;
+    }
+    
+    case MMDB_DATA_TYPE_BYTES:
+    {
+      *ret_obj = rb_enc_str_new(
+        data_list->entry_data.bytes,
+        data_list->entry_data.data_size,
+        rb_ascii8bit_encoding()
+      );
+      
+      data_list = data_list->next;
+      
+      break;
+    }
+    
+    case MMDB_DATA_TYPE_DOUBLE:
+    {
+      *ret_obj = rb_float_new(data_list->entry_data.double_value);
+      
+      data_list = data_list->next;
+      
+      break;
+    }
+    
+    case MMDB_DATA_TYPE_FLOAT:
+    {
+      *ret_obj = rb_float_new(data_list->entry_data.float_value);
+      
+      data_list = data_list->next;
+      
+      break;
+    }
+    
+    case MMDB_DATA_TYPE_UINT16:
+    {
+      *ret_obj = UINT2NUM(data_list->entry_data.uint16);
+      
+      data_list = data_list->next;
+      
+      break;
+    }
+    
+    case MMDB_DATA_TYPE_UINT32:
+    {
+      *ret_obj = UINT2NUM(data_list->entry_data.uint32);
+      
+      data_list = data_list->next;
+      
+      break;
+    }
+    
+    case MMDB_DATA_TYPE_BOOLEAN:
+    {
+      *ret_obj = data_list->entry_data.boolean ? Qtrue : Qfalse;
+      
+      data_list = data_list->next;
+      
+      break;
+    }
+    
+    case MMDB_DATA_TYPE_UINT64:
+    {
+      *ret_obj = rb_big_unpack(&data_list->entry_data.uint64, WORDS_64);
+      
+      data_list = data_list->next;
+      
+      break;
+    }
+    
+    case MMDB_DATA_TYPE_UINT128:
+    {
+#if MMDB_UINT128_IS_BYTE_ARRAY
+      char buf[16];
+      uint8_t i;
+      uint8_t idx = 15;
+      uint8_t *data = data_list->entry_data.uint128;
+      
+      for (i = 0; i < 16; ++i) {
+        buf[i] = data[idx];
+        idx--;
+      }
+      
+      *ret_obj = rb_big_unpack(&buf, WORDS_128);
+#else
+      *ret_obj = rb_big_unpack(&data_list->entry_data.uint128, WORDS_128);
+#endif
+      
+      data_list = data_list->next;
+      
+      break;
+    }
+    
+    case MMDB_DATA_TYPE_INT32:
+    {
+      *ret_obj = INT2NUM(data_list->entry_data.int32);
+      
+      data_list = data_list->next;
+      
+      break;
+    }
+    
+    default:
+      rb_raise(rb_eRuntimeError,
+        "GeoIP2 - %s", MMDB_strerror(MMDB_INVALID_DATA_ERROR)
+      );
+  }
+  
+  return data_list;
+}
+
+static void mmdb_try_open(char *db_path, MMDB_s *mmdb) {
+  int status = MMDB_open(db_path, MMDB_MODE_MMAP, mmdb);
+  
+  if (status != MMDB_SUCCESS) {
+    rb_raise(rb_eIOError, "GeoIP2 - %s: %s",
+      MMDB_strerror(status), db_path
+    );
+  }
+}
+
+static inline bool mmdb_is_closed(MMDB_s *mmdb) {
+  return mmdb->file_content == NULL;
+}
+
+static inline void mmdb_close(MMDB_s *mmdb) {
+  MMDB_close(mmdb);
+  mmdb->file_content = NULL;
+}
+
+static VALUE mmdb_lookup(MMDB_s *mmdb, char *ip_addr, bool cleanup) {
+  int status, gai_error, mmdb_error;
+  
+  MMDB_lookup_result_s result =
+    MMDB_lookup_string(mmdb, ip_addr, &gai_error, &mmdb_error);
+  
+  if (mmdb_error != MMDB_SUCCESS) {
+    if (cleanup) {
+      mmdb_close(mmdb);
+    }
+    
+    rb_raise(rb_eRuntimeError,
+      "GeoIP2 - lookup failed: %s", MMDB_strerror(mmdb_error)
+    );
+  }
+  
+  if (gai_error != 0) {
+    if (cleanup) {
+      mmdb_close(mmdb);
+    }
+    
+    rb_raise(rb_eRuntimeError,
+      "GeoIP2 - getaddrinfo failed: %s", gai_strerror(gai_error)
+    );
+  }
+  
+  if (result.found_entry) {
+    MMDB_entry_data_list_s *data_list, *first;
+    
+    status = MMDB_get_entry_data_list(&result.entry, &data_list);
+    
+    if (status != MMDB_SUCCESS) {
+      MMDB_free_entry_data_list(data_list);
+      
+      if (cleanup) {
+        mmdb_close(mmdb);
+      }
+      
+      rb_raise(rb_eRuntimeError,
+        "GeoIP2 - couldn\'t fetch results: %s", MMDB_strerror(status)
+      );
+    }
+    
+    first = data_list;
+    
+    int exception = 0;
+    
+    VALUE ret_obj;
+    
+    struct args_parse_data_list args;
+    args.data_list = data_list;
+    args.ret_obj = &ret_obj;
+    
+    rb_protect(guard_parse_data_list, (VALUE)&args, &exception);
+    
+    MMDB_free_entry_data_list(first);
+    
+    if (cleanup) {
+      mmdb_close(mmdb);
+    }
+    
+    if (exception) {
+      rb_jump_tag(exception);
+    }
+    
+    return ret_obj;
+  }
+  else {
+    return Qnil;
+  }
+}
+
+static VALUE rb_hive_geo_lookup(VALUE self, VALUE ip_arg) {
+  Check_Type(ip_arg, T_STRING);
+  
+  char *ip_addr = StringValuePtr(ip_arg);
+  
+  MMDB_s *mmdb;
+  
+  Data_Get_Struct(self, MMDB_s, mmdb);
+  
+  if (mmdb_is_closed(mmdb)) {
+    rb_raise(rb_eIOError, "GeoIP2 - closed database");
+  }
+  
+  return mmdb_lookup(mmdb, ip_addr, false);
+}
+
+static VALUE rb_hive_geo_lookup2(VALUE self, VALUE ip_arg, VALUE db_arg) {
+  Check_Type(ip_arg, T_STRING);
+  Check_Type(db_arg, T_STRING);
+  
+  char *ip_addr = StringValuePtr(ip_arg);
+  char *db_path = StringValuePtr(db_arg);
+  
+  MMDB_s mmdb;
+  
+  mmdb_try_open(db_path, &mmdb);
+  
+  return mmdb_lookup(&mmdb, ip_addr, true);
+}
+
+static VALUE rb_hive_geo_is_closed(VALUE self) {
+  MMDB_s *mmdb;
+  
+  Data_Get_Struct(self, MMDB_s, mmdb);
+  
+  return mmdb_is_closed(mmdb) ? Qtrue : Qfalse;
+}
+
+static VALUE rb_hive_geo_close(VALUE self) {
+  MMDB_s *mmdb;
+  
+  Data_Get_Struct(self, MMDB_s, mmdb);
+  
+  if (!mmdb_is_closed(mmdb)) {
+    mmdb_close(mmdb);
+  }
+  
+  return Qnil;
+}
+
+static void rb_hive_geo_free(MMDB_s *mmdb) {
+  if (!mmdb_is_closed(mmdb)) {
+    mmdb_close(mmdb);
+  }
+  
+  xfree(mmdb);
+}
+
+static VALUE rb_hive_geo_alloc(VALUE self) {
+  MMDB_s *mmdb = ALLOC(MMDB_s);
+  
+  return Data_Wrap_Struct(self, NULL, rb_hive_geo_free, mmdb);
+}
+
+static VALUE rb_hive_geo_init(VALUE self, VALUE db_arg) {
+  Check_Type(db_arg, T_STRING);
+  
+  char *db_path = StringValuePtr(db_arg);
+  
+  MMDB_s *mmdb;
+  
+  Data_Get_Struct(self, MMDB_s, mmdb);
+  
+  mmdb_try_open(db_path, mmdb);
+  
+  return Qnil;
+}
+
+void Init_hive_geoip2() {
+  rb_mHive = rb_define_module("Hive");
+  rb_cGeoIP2 = rb_define_class_under(rb_mHive, "GeoIP2", rb_cObject);
+  
+  rb_define_alloc_func(rb_cGeoIP2, rb_hive_geo_alloc);
+  
+  rb_define_singleton_method(rb_cGeoIP2, "lookup", rb_hive_geo_lookup2, 2);
+  
+  rb_define_method(rb_cGeoIP2, "initialize", rb_hive_geo_init, 1);
+  rb_define_method(rb_cGeoIP2, "close", rb_hive_geo_close, 0);
+  rb_define_method(rb_cGeoIP2, "closed?", rb_hive_geo_is_closed, 0);
+  rb_define_method(rb_cGeoIP2, "lookup", rb_hive_geo_lookup, 1);
+}

data/hive_geoip2.gemspec

@@ -0,0 +1,17 @@
+Gem::Specification.new do |s|
+  s.name = 'hive_geoip2'
+  s.version = '0.1.0'
+  s.summary = 'libmaxminddb GeoIP2 Ruby bindings'
+  s.author = 'Maxime Youdine'
+  s.license = 'MIT'
+  s.homepage = 'https://github.com/desuwa/hive_geoip2'
+  s.required_ruby_version = '>= 1.9.2'
+  s.files = %w[
+    hive_geoip2.gemspec
+    lib/hive_geoip2/hive_geoip2.rb
+    ext/hive_geoip2/extconf.rb
+    ext/hive_geoip2/hive_geoip2.c
+  ]
+  s.extensions = ['ext/hive_geoip2/extconf.rb']
+  s.require_paths = ['lib']
+end

data/lib/hive_geoip2/hive_geoip2.rb

@@ -0,0 +1 @@
+require 'hive-geoip2.so'

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: hive_geoip2
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Maxime Youdine
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-09-22 00:00:00.000000000 Z
+dependencies: []
+description: 
+email: 
+executables: []
+extensions:
+- ext/hive_geoip2/extconf.rb
+extra_rdoc_files: []
+files:
+- ext/hive_geoip2/extconf.rb
+- ext/hive_geoip2/hive_geoip2.c
+- hive_geoip2.gemspec
+- lib/hive_geoip2/hive_geoip2.rb
+homepage: https://github.com/desuwa/hive_geoip2
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 1.9.2
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.1
+signing_key: 
+specification_version: 4
+summary: libmaxminddb GeoIP2 Ruby bindings
+test_files: []