loyal_ipinfo 0.0.1

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright 2013 YOURNAME
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,25 @@
+= LoyalIpinfo
+
+get ip info
+
+eg:
+
+```ruby
+# config
+
+LoyalIpinfo.configure do |config|
+  # config.default_adapter = :qqwry
+  # config.default_library_file_path = ''  # library_file_path
+end
+
+# usage:
+
+worker = ::LoyalIpinfo::Worker.new
+
+result = worker.find('127.0.0.1')
+
+result.city # '本机地址'
+result.area # ''
+
+```
+

data/Rakefile

@@ -0,0 +1,8 @@
+begin
+  require 'bundler/setup'
+rescue LoadError
+  puts 'You must `gem install bundler` and `bundle install` to run rake tasks'
+end
+
+Bundler::GemHelper.install_tasks
+

data/lib/loyal_ipinfo/array_util.rb

@@ -0,0 +1,16 @@
+# -*- encoding : utf-8 -*-
+module LoyalIpinfo
+  class ArrayUtil
+    def self.extract_options!(arr)
+      if arr.last.is_a?(Hash)
+        arr.pop
+      else
+        {}
+      end
+    end
+
+    def self.init args
+      args.is_a?(Array) ? args : (args.nil? ? [] : [args])
+    end
+  end
+end

data/lib/loyal_ipinfo/config.rb

@@ -0,0 +1,33 @@
+# -*- encoding : utf-8 -*-
+module LoyalIpinfo
+  class << self
+    attr_writer :config
+
+    def config
+      @config ||= Config.new
+    end
+
+    def configure
+      yield self.config ||= Config.new
+    end
+  end
+
+  class Config
+    def default_adapter
+      @default_adapter ||= :qqwry  # 纯真qq
+    end
+
+    def default_adapter= _adapter
+      @default_adapter ||= _adapter
+    end
+
+    def default_library_file_path
+      @default_library_file_path ||= "#{File.dirname(__FILE__)}/../../resources/qqwry.dat"
+    end
+
+    def default_library_file_path= path
+      @default_library_file_path ||= path
+    end
+  end
+end
+

data/lib/loyal_ipinfo/version.rb

@@ -0,0 +1,4 @@
+# -*- encoding : utf-8 -*-
+module LoyalIpinfo
+  VERSION = "0.0.1"
+end

data/lib/loyal_ipinfo/worker.rb

@@ -0,0 +1,226 @@
+# -*- encoding : utf-8 -*-
+require 'iconv' unless RUBY_VERSION.respond_to?(:encode)
+
+module LoyalIpinfo
+
+  class Worker
+    attr_reader :adapter, :library_file_path
+
+    def initialize attrs={}
+      @adapter           ||= (attrs[:adapter] || ::LoyalIpinfo.config.default_adapter)
+      @library_file_path ||= (attrs[:library_file_path] || ::LoyalIpinfo.config.default_library_file_path)
+
+      case self.adapter
+      when :qqwry
+        self.extend ::LoyalIpinfo::QQwryAdapter
+      end
+    end
+  end
+
+  module QQwryAdapter
+    def find *args, &block
+      options = ::LoyalIpinfo::ArrayUtil.extract_options!(args)
+
+      @qqwry_db ||= QQWry::QQWryFile.new(self.library_file_path)
+
+      result = args.inject({}) do |_result, ip|
+        _result[ip] = @qqwry_db.find(ip)
+        _result
+      end
+
+      if block_given?
+        block.call result
+      end
+
+      if args.any?
+        if args.one?
+          result[args.first]
+        else
+          result
+        end
+      else
+        nil
+      end
+    end
+    module QQWry
+      module QQWryIO
+        def get_le32
+          arr = read(4).unpack("C4")
+          arr[0] | arr[1]<<8 | arr[2]<<16 | arr[3]<<24
+        end
+
+        def get_le24
+          arr = read(3).unpack("C3")
+          arr[0] | arr[1]<<8 | arr[2]<<16
+        end
+
+        def get_string
+          str = ""
+          while ((ch = read(1).unpack("a")[0]) != "\x00")
+            str += ch
+          end
+          str
+        end
+
+        def get_u8
+          read(1).unpack("C")[0]
+        end
+      end
+
+      module QQWryIpStr
+        def ip_str
+          [@ip].pack("L").unpack("C4").reverse.join(".")
+        end
+      end
+
+      class QQWryHeader
+        def initialize(file)
+          file.seek(0)
+          @first = file.get_le32
+          @last = file.get_le32
+        end
+        attr_reader :first, :last
+      end
+
+      class QQWryRecord
+        include QQWryIpStr
+
+        attr_reader :city, :area
+
+        def initialize(file, pos)
+          file.seek(pos)
+          @ip = file.get_le32
+          case file.get_u8
+          when 0x1
+            file.seek(pos = file.get_le24)
+            case file.get_u8
+            when 0x2
+              file.seek(file.get_le24)
+              @city = file.get_string
+              file.seek(pos + 4)
+            else
+              file.seek(-1, IO::SEEK_CUR)
+              @city = file.get_string
+            end
+            parse_area(file)
+          when 0x2
+            pos = file.get_le24
+            parse_area(file)
+            file.seek(pos)
+            @city = file.get_string
+          else
+            file.seek(-1, IO::SEEK_CUR)
+            @city = file.get_string
+            parse_area(file)
+          end
+
+          iconv_city
+        end
+
+        attr_reader :ip
+
+        def _conv(str)
+          begin
+            if str.respond_to?(:encode)
+              str.force_encoding('GBK').encode('UTF-8')
+            else
+              Iconv::conv('UTF-8', 'GBK', str)
+            end
+          rescue
+            str = str[0, str.length - 1]
+            retry
+          end
+        end
+
+        def iconv_city
+          @city = _conv @city.to_s
+          @city = '' if @city=~ /\s*CZ88\.NET\s*/
+        end
+
+        def parse_area(file)
+          case file.get_u8
+          when 0x2
+            if ((pos = file.get_le24) != 0)
+              file.seek(pos)
+              @area = file.get_string
+            else
+              @area = ""
+            end
+          else
+            file.seek(-1, IO::SEEK_CUR)
+            @area = file.get_string
+          end
+
+          @area = _conv(@area.to_s)
+
+          @area = '' if @area =~ /\s*CZ88\.NET\s*/
+        end
+
+        private :parse_area
+      end
+
+      class QQWryIndex
+        include QQWryIpStr
+
+        def initialize(file, pos)
+          file.seek(pos)
+          @ip = file.get_le32
+          @pos = file.get_le24
+        end
+        attr_reader :ip, :pos
+      end
+
+      class QQWryFile
+        def initialize(filename = "qqwry.dat")
+          @filename = filename
+        end
+
+        def each
+          File.open(@filename) do |file|
+            file.extend QQWryIO
+            header = QQWryHeader.new(file)
+            pos = header.first
+            while pos <= header.last
+              index = QQWryIndex.new(file, pos)
+              record = QQWryRecord.new(file, index.pos)
+              if block_given?
+                yield index, record
+              else
+                puts "#{index.ip_str}-#{record.ip_str} #{record.city}" +
+                " #{record.area}"
+              end
+              pos += 7
+            end
+          end
+        end
+
+        def find(ip)
+          if ip.class == String
+            ip = ip.split(".").collect{|x| x.to_i}.pack("C4").unpack("N")[0]
+          end
+
+          File.open(@filename) do |file|
+            file.extend QQWryIO
+            header = QQWryHeader.new(file)
+            first = header.first
+            left = 0
+            right = (header.last - first) / 7
+            while left <= right
+              middle = (left + right) / 2
+              middle_index = QQWryIndex.new(file, first + middle * 7)
+              if (ip > middle_index.ip)
+                left = middle + 1
+              elsif (ip < middle_index.ip)
+                right = middle - 1
+              else
+                return QQWryRecord.new(file, middle_index.pos)
+              end
+            end
+            index = QQWryIndex.new(file, first + right * 7)
+            return QQWryRecord.new(file, index.pos)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/loyal_ipinfo.rb

@@ -0,0 +1,9 @@
+# -*- encoding : utf-8 -*-
+require 'loyal_ipinfo/version'
+require 'loyal_ipinfo/config'
+require 'loyal_ipinfo/array_util'
+require 'loyal_ipinfo/worker'
+
+module LoyalIpinfo
+
+end

data/lib/tasks/loyal_ipinfo_tasks.rake

@@ -0,0 +1,4 @@
+# desc "Explaining what the task does"
+# task :loyal_ipinfo do
+#   # Task goes here
+# end

metadata

@@ -0,0 +1,77 @@
+--- !ruby/object:Gem::Specification
+name: loyal_ipinfo
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- happy
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-08-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bindata
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Description of LoyalIpinfo.
+email:
+- andywang7259@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- resources/qqwry.dat
+- lib/loyal_ipinfo/config.rb
+- lib/loyal_ipinfo/version.rb
+- lib/loyal_ipinfo/array_util.rb
+- lib/loyal_ipinfo/worker.rb
+- lib/loyal_ipinfo.rb
+- lib/tasks/loyal_ipinfo_tasks.rake
+- MIT-LICENSE
+- Rakefile
+- README.md
+homepage: http://develop.xiuxian123.com/
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 1415171691332789189
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 1415171691332789189
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: Summary of LoyalIpinfo.
+test_files: []