Contents

module Shadowsocks
  module Local
    class ServerConnector < ::Shadowsocks::Tunnel
      def post_init
        p "connecting #{server.remote_addr[3..-1]} via #{server.config.server}"
        addr_to_send = server.addr_to_send.clone

        send_data encrypt(table[:encrypt_table], addr_to_send)
        server.cached_pieces.each { |piece| send_data encrypt(table[:encrypt_table], piece) }
        server.cached_pieces = []

        server.stage = 5
      end

      def receive_data data
        server.send_data encrypt(table[:decrypt_table], data)
      end
    end

    class LocalListener < ::Shadowsocks::Listener
      private

      def data_handler data
        case stage
        when 0
          send_data "\x05\x00"
          @stage = 1
        when 1
          fireup_tunnel data
        when 4
          cached_pieces.push data
        when 5
          connector.send_data(encrypt(table[:encrypt_table], data)) and return
        end
      end

      def fireup_tunnel(data)
        begin
          unless data[1] == "\x01"
            send_data "\x05\x07\x00\x01"
            connection_cleanup and return
          end

          @addr_to_send = data[3]

          resolve_addrtype data

          send_data "\x05\x00\x00\x01\x00\x00\x00\x00" + [config.server_port].pack('s>')

          @stage = 4
          @connector = EventMachine.connect config.server, config.server_port, ServerConnector, self, @table

          if data.size > header_length
            cached_pieces.push data[header_length, data.size]
          end
        rescue Exception => e
          warn e
          connection_cleanup
        end
      end

      def resolve_addrtype data
        @addrtype = data[3]
        super
      end

      def domain_address data
        @addr_len       = data[4].unpack('c')[0]
        @addr_to_send  += data[4..5 + @addr_len + 2]
        super
      end

      def ip_address data
        @addr_to_send  += data[4..9]
        super
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
shadowsocks-0.2	lib/shadowsocks/local.rb
shadowsocks-0.1	lib/shadowsocks/local.rb