init commit of peertracker with ruby extension

mysql/announce.php

@@ -97,10 +97,12 @@ $_GET['info_hash'] = peertracker::$api->escape_sql($_GET['info_hash']);
 $_GET['peer_id']   = peertracker::$api->escape_sql($_GET['peer_id']);
 
 // announce peers
-peertracker::peers();
+// need to pass as_code and country
+$json = file_get_contents("http://localhost:4567/as_info/123.123.123.123.json");
+peertracker::peers($as_code, $country);
 
 // track client
-peertracker::event();
+peertracker::event($as_code, $country);
 
 // garbage collection
 peertracker::clean();
@@ -108,4 +110,4 @@ peertracker::clean();
 // close database
 peertracker::close();
 
-?>
+?>

ruby/Gemfile

@@ -0,0 +1,7 @@
+source "http://rubygems.org"
+
+gem 'activerecord', :require => "active_record"
+gem 'sinatra'
+gem 'mysql2'
+gem "rspec"
+gem "json"

ruby/Gemfile.lock

@@ -0,0 +1,48 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    activemodel (3.2.8)
+      activesupport (= 3.2.8)
+      builder (~> 3.0.0)
+    activerecord (3.2.8)
+      activemodel (= 3.2.8)
+      activesupport (= 3.2.8)
+      arel (~> 3.0.2)
+      tzinfo (~> 0.3.29)
+    activesupport (3.2.8)
+      i18n (~> 0.6)
+      multi_json (~> 1.0)
+    arel (3.0.2)
+    builder (3.0.0)
+    diff-lcs (1.1.3)
+    i18n (0.6.0)
+    json (1.7.4)
+    multi_json (1.3.6)
+    mysql2 (0.3.11)
+    rack (1.4.1)
+    rack-protection (1.2.0)
+      rack
+    rspec (2.11.0)
+      rspec-core (~> 2.11.0)
+      rspec-expectations (~> 2.11.0)
+      rspec-mocks (~> 2.11.0)
+    rspec-core (2.11.1)
+    rspec-expectations (2.11.2)
+      diff-lcs (~> 1.1.3)
+    rspec-mocks (2.11.1)
+    sinatra (1.3.2)
+      rack (~> 1.3, >= 1.3.6)
+      rack-protection (~> 1.2)
+      tilt (~> 1.3, >= 1.3.3)
+    tilt (1.3.3)
+    tzinfo (0.3.33)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  activerecord
+  json
+  mysql2
+  rspec
+  sinatra

ruby/bin/get_as_info.rb

@@ -0,0 +1,4 @@
+#! /usr/bin/ruby
+require File.join(File.dirname(__FILE__),'../lib/as_info.rb')
+
+Peertracker::ASInfo.set_as_info

ruby/bin/web_interface.rb

@@ -0,0 +1,7 @@
+Bundler.require
+require File.join(File.dirname(__FILE__),'../lib/as_info.rb')
+
+get '/as_info/:ip_address.json' do |ip_address|
+  response = Peertracker::ASInfo.new.set_as_info_for_ip(ip_address)
+  {:as_code => response[0], :country => response[1]}.to_json
+end

ruby/config/database.rb

@@ -0,0 +1,7 @@
+ActiveRecord::Base.establish_connection(
+  :adapter  => "mysql2",
+  :host     => "localhost",
+  :username => "root",
+  :password => "howareyou",
+  :database => "peertracker"
+)

ruby/config/migration.rb

@@ -0,0 +1,16 @@
+Bundler.require
+require File.join(File.dirname(__FILE__), "database.rb")
+
+class AddASInfoToPeers < ActiveRecord::Migration
+  def up
+    add_column :pt_peers, :as_code, :string
+    add_column :pt_peers, :country, :string
+  end
+
+  def down
+    remove_column :pt_peers, :as_code
+    remove_column :pt_peers, :country
+  end
+end
+
+AddASInfoToPeers.new.up

ruby/lib/as_info.rb

@@ -0,0 +1,30 @@
+Bundler.require
+require File.join(File.dirname(__FILE__), "../config/database.rb")
+require File.join(File.dirname(__FILE__), "./model/peer.rb")
+require File.join(File.dirname(__FILE__), "./url_creator.rb")
+require File.join(File.dirname(__FILE__), "./http_request.rb")
+require File.join(File.dirname(__FILE__), "./as_parser.rb")
+
+module Peertracker
+  class ASInfo
+    def set_as_info_for_ip(ip_address)
+      request = HTTPRequest.new(URLCreator.new("ip.searchwww.com", ip_address).create_url)
+      ASParser.new.parse_response(request.perform_request)
+    end
+
+    class << self
+      def set_as_info
+        Peer.without_as_info.each do |peer|
+          begin
+            puts "Parsing #{peer.ip}"
+          peer.as_code, peer.country = self.new.set_as_info_for_ip(peer.ip)
+          peer.save
+          rescue => e
+            puts "Could not fetch AS for #{peer.ip}"
+            puts "#{e.to_s}"
+          end
+        end
+      end
+    end
+  end
+end

ruby/lib/as_parser.rb

@@ -0,0 +1,23 @@
+class ASParser
+  def parse_response(response)
+    [parse_as(response),parse_country(response)]
+  end
+
+  def parse_as(response)
+    begin
+      response.scan(/Provider:<\/b><p>(\w*)/)[0][0]
+    rescue => e
+      puts "Could not parse AS"
+      nil
+    end
+  end
+
+  def parse_country(response)
+    begin
+      response.scan(/Location:<\/b><p>(.*)<p>/)[0][0].split(",").last.strip
+    rescue => e
+      puts "Could not parse Country"
+      nil
+    end
+  end
+end

ruby/lib/http_request.rb

@@ -0,0 +1,14 @@
+module Peertracker
+  class HTTPRequest
+    def initialize(host, path = "")
+      @host = host
+      @path = path
+    end
+
+    def perform_request
+      uri = URI("http://#{@host}#{@path}/")
+      response = Net::HTTP.get_response(uri)
+      response.body
+    end
+  end
+end

ruby/lib/model/peer.rb

@@ -0,0 +1,5 @@
+class Peer < ActiveRecord::Base
+  self.table_name = "pt_peers"
+  self.primary_key = "peer_id"
+  scope :without_as_info,  lambda{ where("pt_peers.as_code IS NULL") }
+end

ruby/lib/url_creator.rb

@@ -0,0 +1,18 @@
+require "net/http"
+
+module Peertracker
+  class URLCreator
+    def initialize(url_host, ip_address)
+      @ip_address = ip_address
+      @url_host = url_host
+    end
+
+    def create_subdomain
+      "%02x%02x%02x%02x" % @ip_address.scan(/(.*)\.(.*)\.(.*)\.(.*)/)[0]
+    end
+
+    def create_url
+      "#{create_subdomain}.#{@url_host}"
+    end
+  end
+end

ruby/spec/functional/set_as_info_spec.rb

@@ -0,0 +1,20 @@
+require File.join(File.dirname(__FILE__), "../../spec/spec_helper.rb")
+
+describe "ASInfo" do 
+  context "when given IP address" do
+    describe "sets as info for ip address" do
+      let(:ip_address) {"12.12.12.12"}
+
+      it "stores as_info to database" do
+        ASInfo.new.set_as_info_for_ip(ip_address)
+        Peer.find_by_ip_address(ip_address).as.should_not be_null
+      end
+    end
+  end
+
+  describe ".set_as_info" do
+    it "stores as info for all addresses"
+      ASInfo.set_as_info
+      Peer.all.each {|peer| peer.as.should_not be_null}
+  end
+end

ruby/spec/spec_helper.rb

@@ -0,0 +1 @@
+require File.join(File.dirname(__FILE__), "../lib/as_info.rb")