dcase 0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 3d798b7687187ae48822a7697be6f6481ec5147d
+  data.tar.gz: a216c5043444fa4cd375af0f16a1724d76eade1e
+SHA512:
+  metadata.gz: 75fb759cf33a02d4bf5a13b11a003d75b0462b4b212f4bab0bd757e3222002519dd9e5c3ded0a72f889f77577bac5d1b17c98dde02ee1ea0bc95cfc0d916cb16
+  data.tar.gz: e5950388fe64ac83e88a81d00168a42e64cc251b9541d6c5e6c07a416c3e1a777812e9bee040489fef37cd1fa92598144cfab2076abcd2044d1ce4ba15f369e6

data/.gitignore

@@ -0,0 +1,25 @@
+*.gem
+*.rbc
+.bundle
+.config
+coverage
+InstalledFiles
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+
+# YARD artifacts
+.yardoc
+_yardoc
+doc/
+
+*.so
+*.bundle
+*.o
+.DS_Store
+Gemfile.lock
+run

data/Gemfile

@@ -0,0 +1,7 @@
+source "https://rubygems.org"
+
+group :development, :test do
+  gem "minitest", '~> 5.2.2'
+end
+
+gemspec

data/LICENSE

@@ -0,0 +1,20 @@
+The MIT License (MIT)
+
+Copyright (c) 2014 Sen
+
+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,54 @@
+DCase
+=====
+
+Current version: 0.2
+
+DCase is a lightweight dns proxy which can help you get through firewalls.
+
+Usage
+-----------
+
+First, make sure you have Ruby 2.0.
+
+    $ ruby -v
+    ruby 2.0.0p353
+
+Install Shadowsocks.
+
+    gem install dcase
+
+Create a file named `config.yml`, with the following content.
+
+    side: 'local or server'
+    password: 'your password'
+    port: '8440'
+    # you don't need this line when it's on server side
+    server: 'remote server address'
+
+Explanation of the fields:
+
+    side      Local or Server side
+    server    Remote server address
+    port      Remote server port
+    password  Password, should be same in client and server sides
+    
+`cd` into the directory of `config.yml`. Run `dcase` on your server. To run it in the background, run
+`nohup dcase -c ./config.yml > log &`.
+
+On your client machine, `cd` into the directory of `config.yml` also, run `sudo dcase -c config.yml`.
+
+Command line args
+------------------
+
+You can use args to override settings from `config.json`.
+
+    sudo dcase -s local -r remote_server_ip_address -p remote_server_port -k your_password
+    sudo dcase -s server -p remote_server_port -k your_password
+
+License
+-------
+MIT
+
+Bugs and Issues
+----------------
+Please visit [issue tracker](https://github.com/Sen/DCase/issues?state=open)

data/Rakefile

@@ -0,0 +1,12 @@
+require 'bundler/gem_tasks'
+require 'rake'
+require 'rake/testtask'
+
+# Load custom tasks
+Dir['tasks/*.rake'].sort.each { |f| load f }
+
+Rake::TestTask.new do |task|
+  task.libs << 'test'
+end
+
+task :default => :test

data/bin/dcase

@@ -0,0 +1,20 @@
+#!/usr/bin/env ruby
+
+lib = File.expand_path(File.dirname(__FILE__) + '/../lib')
+$LOAD_PATH.unshift(lib) if File.directory?(lib) && !$LOAD_PATH.include?(lib)
+
+require 'dcase'
+require 'dcase/cli'
+require 'dcase/config'
+
+begin
+  config = DCase::Config.new ARGV
+  cli = DCase::Cli.new({config: config})
+  cli.run
+rescue => e
+  raise e if $DEBUG
+  STDERR.puts e.message
+  STDERR.puts e.backtrace.join("\n")
+  exit 1
+end
+

data/config.yml

@@ -0,0 +1,3 @@
+password: 'your password'
+port: '8440'
+server: '127.0.0.1'

data/dcase.gemspec

@@ -0,0 +1,25 @@
+require File.expand_path('../lib/dcase/version', __FILE__)
+require 'date'
+
+Gem::Specification.new do |s|
+  s.name          = 'dcase'
+  s.version       = DCase::VERSION
+  s.date          = Date.today
+  s.summary       = "a secure dns proxy"
+  s.description   = "DCase is a lightweight dns proxy which can help you get through firewalls."
+  s.authors       = ["Sen"]
+  s.email         = 'sen9ob@gmail.com'
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- test/{functional,unit}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.homepage      = 'https://github.com/Sen/DCase'
+  s.license       = 'MIT'
+  s.extensions    = %w[ext/encrypt/extconf.rb]
+
+  s.add_dependency "celluloid-io", "~> 0.15.0"
+  s.add_dependency "ffi", "~> 1.9.0"
+
+  s.add_development_dependency "rake-compiler", "~> 0.9.2"
+  s.add_development_dependency "mocha", "~> 1.0.0"
+  s.add_development_dependency "rake"
+end

data/ext/encrypt/encrypt.c

@@ -0,0 +1,11 @@
+char* encrypt(int table[], char buf[], int len)
+{
+    int j = 0;
+    char *end = buf + len;
+    while (buf < end) {
+        *buf = (char)table[(unsigned char)*buf];
+        buf++;
+    }
+
+    return buf - len;
+}

data/ext/encrypt/extconf.rb

@@ -0,0 +1,3 @@
+require 'mkmf'
+
+create_makefile('encrypt')

data/lib/dcase.rb

@@ -0,0 +1,9 @@
+require 'celluloid/io'
+
+module DCase
+  autoload :Crypto, 'dcase/crypto'
+  autoload :Table,  'dcase/table'
+  autoload :Server, 'dcase/server'
+  autoload :Local,  'dcase/local'
+end
+

data/lib/dcase/cli.rb

@@ -0,0 +1,25 @@
+module DCase
+  class Cli
+    def initialize(options = {})
+      @config = options[:config]
+      @crypto = DCase::Crypto.new(@config.password)
+    end
+
+    def run
+      supervisor = \
+        case @config.side
+        when 'local'
+          puts "*** Local side is up, remote server port:#{@config.port}"
+          DCase::Local.supervise('0.0.0.0', 53, @crypto, @config)
+        when 'server'
+          puts "*** Server side is up, port:#{@config.port}"
+          DCase::Server.supervise('0.0.0.0', @config.port, @crypto)
+        end
+
+      puts "*** Hit Ctrl+c to stop"
+      trap("INT")  { supervisor.terminate; exit }
+      trap("TERM") { supervisor.terminate; exit }
+      sleep
+    end
+  end
+end

data/lib/dcase/config.rb

@@ -0,0 +1,90 @@
+require 'optparse'
+require 'yaml'
+
+require File.expand_path('../version', __FILE__)
+
+module DCase
+  class Config
+    attr_accessor :args, :side, :server, :port, :password, :config_path
+
+    def initialize(args)
+      @args = args || []
+
+      parse_args
+      read_config
+
+      check_values
+    end
+
+    def check_values
+      if side.nil?
+        raise 'You need to define side of dcase in'
+      end
+
+      if side == 'local' && server.nil?
+        raise 'You need to define remote server address'
+      end
+
+      if port.nil?
+        raise 'You need to define remote server port'
+      end
+
+      if password.nil?
+        raise 'You need to define a password, should be same in server/local side'
+      end
+    end
+
+    private
+
+    def read_config
+      @config_path = File.expand_path('../..', File.dirname(__FILE__)) + '/config.yml' unless @config_path
+      config = YAML.load_file @config_path
+
+      @side      = config["side"]          if @side.nil?
+      @server    = config["server"]        if @server.nil?
+      @port      = config["port"].to_i     if @port.nil?
+      @password  = config["password"]      if @password.nil?
+    end
+
+    def parse_args
+      opt_parser = OptionParser.new do |opts|
+        opts.banner = "Usage: dcase [options]"
+
+        opts.separator ""
+        opts.separator "Specific options:"
+
+        opts.on("-s", "--side SIDE", "Local or Server side") do |c|
+          @side = c
+        end
+
+        opts.on("-c", "--config PATH", "Config file path") do |c|
+          @config_path = c
+        end
+
+        opts.on("-r", "--remote ADDRESS", "Remote server address") do |c|
+          @server = c
+        end
+
+        opts.on("-p", "--port PORT", Integer, "Remote server port") do |c|
+          @port = c
+        end
+
+        opts.on("-k", "--password PASSWORD", "Password, should be same in client and server sides") do |c|
+          @password = c
+        end
+
+        opts.on_tail("-v", "--version", "Show shadowsocks gem version") do
+          puts DCase::VERSION
+          exit
+        end
+
+        opts.on_tail("-h", "--help", "Show this message") do
+          puts opts
+          exit
+        end
+      end
+
+      opt_parser.parse!(args)
+    end
+  end
+end

data/lib/dcase/crypto.rb

@@ -0,0 +1,17 @@
+module DCase
+  class Crypto
+    include DCase::Table
+
+    def initialize(password)
+      @encrypt_table, @decrypt_table = get_table(password)
+    end
+
+    def encrypt(data)
+      translate(@encrypt_table, data)
+    end
+
+    def decrypt(data)
+      translate(@decrypt_table, data)
+    end
+  end
+end

data/lib/dcase/local.rb

@@ -0,0 +1,38 @@
+module DCase
+  class Local
+    MAX_PACKET_SIZE = 512
+
+    include Celluloid::IO
+
+    attr_accessor :socket, :crypto
+
+    def initialize(addr, port, crypto, config)
+      @socket = UDPSocket.new
+      @socket.bind(addr, port)
+      @crypto = crypto
+      @config = config
+
+      async.run
+    end
+
+    def run
+      loop do
+        data, (_, port, addr) = @socket.recvfrom(MAX_PACKET_SIZE)
+        handle_data(data, port, addr)
+      end
+    end
+
+    def handle_data(data, port, addr)
+      connector = UDPSocket.new
+      connector.send crypto.encrypt(data), 0, @config.server, @config.port
+
+      async.request(connector, port, addr)
+    end
+
+    def request(connector, port, addr)
+      data, (_, _port, _addr) = connector.recvfrom(16384)
+      @socket.send crypto.decrypt(data), 0, addr, port
+      connector.close unless connector.closed?
+    end
+  end
+end

data/lib/dcase/server.rb

@@ -0,0 +1,41 @@
+module DCase
+  class Server
+    include Celluloid::IO
+    finalizer :finalize
+
+    attr_accessor :crypto
+
+    def initialize(host, port, crypto)
+      @server = UDPSocket.new
+      @server.bind(host, port)
+
+      @crypto = crypto
+
+      async.run
+    end
+
+    def finalize
+      @server.close unless @server.closed?
+    end
+
+    def run
+      loop do
+        data, (_, port, addr) = @server.recvfrom(16384)
+        handle_data(data, port, addr)
+      end
+    end
+
+    def handle_data(data, port, addr)
+      request = UDPSocket.new
+      request.send crypto.decrypt(data), 0, '8.8.8.8', 53
+
+      async.start_connect(request, port, addr)
+    end
+
+    def start_connect(request, port, addr)
+      data, (_, _port, _addr) = request.recvfrom(16384)
+      @server.send crypto.encrypt(data), 0, addr, port
+      request.close unless request.closed?
+    end
+  end
+end

data/lib/dcase/table.rb

@@ -0,0 +1,84 @@
+require "digest"
+require 'ffi'
+
+module Ext
+  def self.binary_path
+    path = ''
+    %w( so bundle dll ).each do |ext|
+      path = File.expand_path('..', File.dirname(__FILE__)) + "/encrypt.#{ext}"
+      break if File.exists? path
+    end
+
+    path
+  end
+
+  extend FFI::Library
+  ffi_lib binary_path
+  attach_function "encrypt", [:pointer, :pointer, :int], :pointer
+end
+
+module DCase
+  module Table
+    include Ext
+
+    def get_table key
+      table = Array.new(256, 0)
+      decrypt_table = Array.new(256, 0)
+
+      a = Digest::MD5.digest(key).unpack('Q<')[0]
+      i = 0
+
+      while i < 256
+        table[i] = i
+        i += 1
+      end
+      i = 1
+
+      while i < 1024
+        table = merge_sort(table, lambda { |x, y|
+          a % (x + i) - a % (y + i)
+        })
+        i += 1
+      end
+      i = 0
+      while i < 256
+        decrypt_table[table[i]] = i
+        i += 1
+      end
+      [ table, decrypt_table ]
+    end
+
+    def translate(table, buf)
+      table_ptr = FFI::MemoryPointer.new(:int, table.length)
+      table_ptr.put_array_of_int32(0, table)
+
+      buf_ptr = FFI::MemoryPointer.new(:string, buf.length)
+      buf_ptr.put_bytes(0, buf)
+
+      r = Ext.encrypt(table_ptr, buf_ptr, buf.length).get_bytes(0, buf.length)
+      table_ptr.free
+      buf_ptr.free
+      r
+    end
+
+    def merge (left, right, comparison)
+      result = []
+      while (left.length > 0) and (right.length > 0)
+        if comparison.call(left[0], right[0]) <= 0
+          result.push left.shift()
+        else
+          result.push right.shift()
+        end
+      end
+      result.push left.shift() while left.length > 0
+      result.push right.shift() while right.length > 0
+      result
+    end
+
+    def merge_sort (array, comparison)
+      return array if array.size < 2
+      middle = (array.size / 2).ceil
+      merge(merge_sort(array.slice(0, middle), comparison), merge_sort(array.slice(middle .. array.size), comparison), comparison)
+    end
+  end
+end

data/lib/dcase/version.rb

@@ -0,0 +1,3 @@
+module DCase
+  VERSION = '0.2'
+end

data/tasks/compile.rake

@@ -0,0 +1,9 @@
+require 'rake/extensiontask'
+
+def gemspec
+  @clean_gemspec ||= eval(File.read(File.expand_path('../../dcase.gemspec', __FILE__)))
+end
+
+Rake::ExtensionTask.new('encrypt', gemspec) do |ext|
+  ext.ext_dir = 'ext/encrypt'
+end

data/tasks/dev.rake

@@ -0,0 +1,7 @@
+task :gem do
+  `gem build ./dcase.gemspec`
+end
+
+task :install do
+  `gem install ./dcase-#{DCase::VERSION}.gem`
+end

data/test/test_helper.rb

@@ -0,0 +1,3 @@
+require "bundler/setup"
+require "minitest/autorun"
+require "mocha/setup"

metadata

@@ -0,0 +1,136 @@
+--- !ruby/object:Gem::Specification
+name: dcase
+version: !ruby/object:Gem::Version
+  version: '0.2'
+platform: ruby
+authors:
+- Sen
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-01-31 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: celluloid-io
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.15.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.15.0
+- !ruby/object:Gem::Dependency
+  name: ffi
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.9.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.9.0
+- !ruby/object:Gem::Dependency
+  name: rake-compiler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+- !ruby/object:Gem::Dependency
+  name: rake
+  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: DCase is a lightweight dns proxy which can help you get through firewalls.
+email: sen9ob@gmail.com
+executables:
+- dcase
+extensions:
+- ext/encrypt/extconf.rb
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- bin/dcase
+- config.yml
+- dcase.gemspec
+- ext/encrypt/encrypt.c
+- ext/encrypt/extconf.rb
+- lib/dcase.rb
+- lib/dcase/cli.rb
+- lib/dcase/config.rb
+- lib/dcase/crypto.rb
+- lib/dcase/local.rb
+- lib/dcase/server.rb
+- lib/dcase/table.rb
+- lib/dcase/version.rb
+- tasks/compile.rake
+- tasks/dev.rake
+- test/test_helper.rb
+homepage: https://github.com/Sen/DCase
+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.0.14
+signing_key: 
+specification_version: 4
+summary: a secure dns proxy
+test_files: []