websocket-client 0.1.1

data/examples/simple.rb

@@ -0,0 +1,26 @@
+#!/usr/bin/env ruby
+
+$: << File.dirname(__FILE__) + '/../lib'
+
+require 'websocket_client'
+
+## Create a disconnected client
+client = WebSocketClient.create( 'ws://localhost:8081/websockets/' ) 
+
+  ## Set up the message handler before connecting
+client.on_message do |msg|
+  puts "received #{msg}"
+end
+
+## Connect
+client.connect()
+
+## Use the connected client
+puts "sending"
+client.send( "HOWDY-1" )
+client.send( "HOWDY-2" )
+client.send( "HOWDY-3" )
+sleep(1)
+
+## Explicit disconncet
+client.disconnect

data/examples/with_block.rb

@@ -0,0 +1,27 @@
+#!/usr/bin/env ruby
+
+$: << File.dirname(__FILE__) + '/../lib'
+
+require 'websocket_client'
+
+## Create a disconnected client
+WebSocketClient.create( 'ws://localhost:8081/websockets/' ) do |client|
+
+  ## Set up the message handler before connecting
+  client.on_message do |msg|
+    puts "received #{msg}"
+  end
+
+  ## Connect
+  client.connect() do |client|
+
+    ## Use the connected client
+    puts "sending"
+    client.send( "HOWDY-1" )
+    client.send( "HOWDY-2" )
+    client.send( "HOWDY-3" )
+    sleep(1)
+  end
+
+  ## Disconnected
+end

data/lib/websocket_client/client.rb

@@ -0,0 +1,111 @@
+require 'socket'
+require 'uri'
+require 'websocket_client/ietf_00'
+
+
+module WebSocketClient
+
+  DEFAULT_HANDSHAKE = Ietf00
+
+  class DebugSocket
+    def initialize(socket)
+      @socket = socket
+    end
+
+    def method_missing(method, *args)
+      puts "> #{method}(#{args.join( ', ')})"
+      @socket.__send__( method, *args )
+    end
+  end
+
+  def self.create(uri,handshake_class=WebSocketClient::DEFAULT_HANDSHAKE,&block)
+    Client.new(uri, handshake_class, &block)
+  end
+
+  class Client
+    attr_reader :uri
+
+    def initialize(uri,handshake_class=WebSocketClient::DEFAULT_HANDSHAKE, &block)
+      @handshake_class = handshake_class
+      @socket = nil
+      @on_message_handler = nil
+      @handler_thread = nil
+      @close_state = nil
+      case ( uri )
+        when String
+          @uri = URI.parse( uri )
+        else
+          @uri = uri
+      end
+      block.call( self ) if block
+    end
+  
+    def connect(&block)
+      @socket = TCPSocket.open(uri.host, uri.port)
+      @handshake = @handshake_class.new( uri, @socket )
+      
+      start_handler
+  
+      if ( block )
+        begin
+          block.call( self ) 
+        ensure
+          disconnect()
+        end
+      end
+    end
+  
+    def disconnect
+      @socket.puts 0xFF
+      @socket.putc 0x00
+      @close_state = :requested
+    end
+  
+    def send(msg)
+      @socket.print @handshake.encode_text_message( msg )
+      @socket.flush
+    end
+
+    def wait_forever()
+      @handler_thread.join
+    end
+
+    def on_message(msg=nil,&block)
+      if ( block ) 
+        @on_message_handler = block
+      else
+        @on_message_handler.call( msg ) if @on_message_handler
+      end
+    end
+  
+    def start_handler
+      @handler_thread = Thread.new(@socket) do |socket|
+        msg = ''
+        while ( ! socket.eof? )
+          c = socket.getc
+          if ( c == 0x00 ) 
+            if ( @close_state == :half_closed )
+              socket.close
+              break;
+            else
+              msg = ''
+            end
+          elsif ( c == 0xFF ) 
+            if ( @close_state == :requested )
+              @close_state = :half_closed
+            else
+              on_message( msg ) 
+            end
+          else
+            if ( @close_state != nil )
+              @close_state = nil
+            end
+            msg << c
+          end
+        end
+      end
+    end
+  
+  end
+end
+

data/lib/websocket_client/ietf_00.rb

@@ -0,0 +1,90 @@
+require 'socket'
+require 'uri'
+require 'digest/md5'
+
+module WebSocketClient
+
+  class Ietf00
+
+    def initialize(uri, socket)
+      @socket = socket
+      key1, key2, key3, solution = generate_keys()
+
+      @socket.puts "GET #{uri.path} HTTP/1.1"
+      @socket.puts "Host: #{uri.host}"
+      @socket.puts "Connection: upgrade"
+      @socket.puts "Upgrade: websocket"
+      @socket.puts "Origin: http://#{uri.host}/"
+      @socket.puts "Sec-WebSocket-Version: 8"
+      @socket.puts "Sec-WebSocket-Key1: #{key1}"
+      @socket.puts "Sec-WebSocket-Key2: #{key2}"
+      @socket.puts ""
+      @socket.print key3
+      @socket.puts ""
+      @socket.flush
+      
+      while ( ! @socket.eof? ) 
+        line = @socket.gets
+        break if ( line.strip == '' ) 
+      end
+      challenge = @socket.read( 16 ) 
+      if ( challenge == solution )
+        puts "success!"
+      end
+      #dump 'solution', solution
+      #dump 'challenge', challenge
+    end
+
+    def encode_text_message(msg)
+      @socket.putc 0x00
+      @socket.print msg
+      @socket.putc 0xFF
+      @socket.flush
+    end
+
+    def generate_keys()
+      key1 = generate_header_key
+      key2 = generate_header_key
+      key3 = generate_content_key
+      [ key1, key2, key3, solve( key1, key2, key3 ) ]
+    end
+
+    def solve(key1, key2, key3)
+      int1 = solve_header_key( key1 )
+      int2 = solve_header_key( key2 )
+      input = int1.to_s + int2.to_s + key3
+      Digest::MD5.digest( input )
+    end
+
+    def solve_header_key(key)
+      key_digits = key.strip.gsub( /[^0-9]/, '').to_i
+      key_spaces = key.strip.gsub( /[^ ]/, '').size
+      solution = key_digits / key_spaces
+      solution
+    end
+
+    def generate_header_key
+      key = '' 
+      1.upto(32) do 
+        key << rand(90) + 32
+      end
+      1.upto( rand(10) + 2 ) do
+        key[rand(key.size-1)+1,1] = ' '
+      end
+      1.upto( rand(10) + 2 ) do
+        key[rand(key.size-1)+1,1] = rand(9).to_s
+      end
+      key
+    end
+
+    def generate_content_key
+      key = []
+      'tacobob1'.each_byte do |byte|
+        key << byte
+      end
+      key.pack('cccccccc')
+    end
+
+  end
+
+end

data/lib/websocket_client.rb

@@ -0,0 +1,2 @@
+
+require 'websocket_client/client'

metadata

@@ -0,0 +1,69 @@
+--- !ruby/object:Gem::Specification 
+name: websocket-client
+version: !ruby/object:Gem::Version 
+  hash: 25
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 1
+  version: 0.1.1
+platform: ruby
+authors: 
+- The TorqueBox Team
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-06-10 00:00:00 Z
+dependencies: []
+
+description: 
+email: team@torquebox.org
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/websocket_client/client.rb
+- lib/websocket_client/ietf_00.rb
+- lib/websocket_client.rb
+- examples/simple.rb
+- examples/with_block.rb
+homepage: 
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.7.1
+signing_key: 
+specification_version: 3
+summary: Pure ruby WebSocket client
+test_files: []
+