RubyGems - em-websocket-server - Versions diffs - 0.1.2 → 0.13 - Mend

em-websocket-server 0.1.2 → 0.13

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (14) hide show

data/README.markdown +56 -63
data/em-websocket-server.gemspec +32 -25
data/examples/chat.html +37 -0
data/examples/chat.rb +70 -0
data/examples/tictactoe/tictactoe.html +51 -0
data/examples/tictactoe/tictactoe.js +130 -0
data/examples/tictactoe/tictactoe.rb +213 -0
data/examples/timesync.html +29 -0
data/examples/timesync.rb +37 -0
data/lib/web_socket.rb +12 -10
data/lib/web_socket/client.rb +70 -68
data/lib/web_socket/server.rb +201 -145
data/lib/web_socket/util.rb +32 -27
metadata +34 -10

data/examples/tictactoe/tictactoe.rb ADDED Viewed

@@ -0,0 +1,213 @@
+$:.unshift File.dirname(__FILE__) + '/../../lib'
+require 'rubygems'
+require 'web_socket'
+require 'json'
+require 'uuid'
+require 'pp'
+$games   = {}
+$waiting = nil
+$status  = nil
+class StatusChannel < EM::Channel
+	def initialize
+		super
+		@count = 0
+	end
+	def increment
+		@count += 1
+		push @count
+	end
+	def decrement
+		@count -= 1
+		push @count
+	end
+end
+class Game < EM::Channel
+	attr_accessor :id, :player1, :player2, :current, :grid
+	def initialize player1, player2
+		super()
+		@id 		= UUID.new
+		@player1 	= player1
+		@player2 	= player2
+		@grid 		= Matrix.diagonal(0,0,0)
+	end
+	def set_turn p
+		@current = p
+		@current.turn!
+	end
+	def start!
+		@current = nil
+		@player1.start!
+		@player2.start!
+		toggle
+	end
+	def move p, data
+		if @current == p
+			unless @matrix[data["x"].to_i][data["y"].to_i]
+				@matrix[data["x"].to_i][data["y"].to_i] = p.key
+				@player1.send_move(p.key, data)
+				@player2.send_move(p.key, data)
+				winner  = has_winner?
+				full	= full?
+				if winner || full
+					@player1.send_command("game_over")
+					@player2.send_command("game_over")
+					if winner
+						p.send_command("win")
+						opponent(p).send_command("loss")
+					else full?
+						@player1.send_command("draw")
+						@player2.send_command("draw")
+					end
+				else
+					toggle
+				end
+			end
+		end
+	end
+	def full?
+		@matrix.each do |row|
+			row.each do |col|
+				return false unless col
+			end
+		end
+		return true
+	end
+	def has_winner?
+		return true if @matrix[1][1] && (@matrix[0][0] == @matrix[1][1]) && (@matrix[1][1] == @matrix[2][2])
+		return true if @matrix[1][1] && (@matrix[0][2] == @matrix[1][1]) && (@matrix[1][1] == @matrix[2][0])
+		@matrix.each do |row|
+			return true if row[1] && (row[0] == row[1]) && (row[1] == row[2])
+		end
+		return false
+	end
+	def opponent p
+		@player1 == p ? @player2 : @player1
+	end
+	def toggle
+		set_turn(@current == @player1 ? @player2 : @player1)
+	end
+end
+class TickTackToeServer < WebSocket::Server
+	attr_accessor :game_id, :key, :status_key
+	def on_connect
+		@status_key = $status.subscribe do |c|
+			send_user_count c
+		end
+		$status.increment
+	end
+	def on_disconnect
+		$status.unsubscribe @status_key
+		$status.decrement
+		delete_game!
+	end
+	def on_receive data
+		begin
+			msg = JSON.parse(data)
+		rescue
+			send_command "error"
+			return
+		end
+		case msg["msg"]
+		when "join"
+			if $waiting.empty?
+				$waiting.push(self)
+				send_command "queued"
+			else
+				$waiting.pop do |opponent|
+					game = Game.new(self, opponent)
+					self.key = "X"
+					opponent.key = "O"
+					self.game_id = opponent.game_id = game.id
+					game.start!
+					$games[game_id] = game
+				end
+			end
+		when "move"
+			if game
+				game.move self, msg["data"]
+			else
+				log "Cannot move on a nil game!"
+			end
+		end
+	end
+	def delete_game!
+		if $games.key?(game_id)
+			$games.delete(game_id)
+		end
+	end
+	def game
+		$games[game_id]
+	end
+	def turn!
+		send_command "turn"
+	end
+	def game_over!
+		delete_game!
+		send_command "game_over"
+	end
+	def start!
+		send_command "start"
+	end
+	def send_move key, data
+		send_message({:msg => "move", :key => key, :data => data}.to_json)
+	end
+	def send_command cmd
+		send_message({:msg => cmd}.to_json)
+	end
+	def send_user_count count
+		send_message({:msg => :user_count, :data => count}.to_json)
+	end
+	def send_message msg
+		super msg
+		puts "Sent: #{msg}"
+	end
+end
+EM.epoll
+EM.set_descriptor_table_size(10240)
+EM.run do
+	$waiting  = EM::Queue.new
+	$status = StatusChannel.new
+	EM.start_server "0.0.0.0", 8000, TickTackToeServer
+end

data/examples/timesync.html ADDED Viewed

@@ -0,0 +1,29 @@
+<!doctype html>
+<html>
+<head>
+	<title>em-websocket-server test</title>
+</head>
+<body>
+<h1>em-websocket-server timesync demo</h1>
+<h2 id="time">opening socket</h2>
+<script>
+	var webSocket = new WebSocket('ws://localhost:8000/time');
+	webSocket.onopen = function(event){
+		document.getElementById('time').innerHTML = 'waiting for socket';
+	};
+	webSocket.onmessage = function(event){
+		var object = JSON.parse(event.data);
+		document.getElementById('time').innerHTML = object.time;
+	};
+	webSocket.onclose = function(event){
+		document.getElementById('time').innerHTML = 'socket closed';
+	};
+</script>
+</body>
+</html>

data/examples/timesync.rb ADDED Viewed

@@ -0,0 +1,37 @@
+$:.unshift File.dirname(__FILE__) + '/../lib'
+require 'rubygems'
+require 'web_socket'
+require 'json'
+class TimeServer < WebSocket::Server
+	def on_connect
+	  puts "Connection Accepted"
+		@timer = EM.add_periodic_timer(5, EM.Callback(self, :sync_time))
+	end
+	def on_disconnect
+	  puts "Connection released"
+		EM.cancel_timer @timer
+	end
+	def on_receive msg
+	end
+	def sync_time
+	  puts "Hi"
+		send_message({ :time => Time.now }.to_json)
+	end
+end
+EM.epoll  = true if EM.epoll?
+EM.kqueue = true if EM.kqueue?
+EM.set_descriptor_table_size(10240)
+EM.run do
+	EM.start_server "0.0.0.0", 8000, TimeServer
+end

data/lib/web_socket.rb CHANGED Viewed

@@ -1,11 +1,13 @@
-require 'rubygems'
-require 'eventmachine'
-require 'pp'
-module WebSocket
-end
-require 'web_socket/util.rb'
-require 'web_socket/frame.rb'
-require 'web_socket/server.rb'
+require 'rubygems'
+require 'eventmachine'
+require 'digest'
+require 'pp'
+module WebSocket
+  VERSION = 0.13
+end
+require 'web_socket/util.rb'
+require 'web_socket/frame.rb'
+require 'web_socket/server.rb'
 require 'web_socket/client.rb'

data/lib/web_socket/client.rb CHANGED Viewed

@@ -1,69 +1,71 @@
-module WebSocket
-	class Client < EM::Connection
-		def path
-			"/chat"
-		end
-		def host
-			"localhost:8000"
-		end
-		def origin
-			"localhost"
-		end
-		# em override
-		def post_init
-			@connected = false
-		end
-		def connection_completed
-			send_headers
-		end
-		# em override
-		def unbind
-			on_disconnect
-		end
-		def on_disconnect
-		end
-		def send_message msg
-			send_data Frame.encode(msg)
-		end
-		private
-		def receive_data data
-			unless @connected
-				handshake data
-			else
-				on_receive Frame.decode(data)
-			end
-		end
-		def handshake data
-			#convert the headers to a hash
-			@headers   = Util.parse_headers(data)
-			@connected = true
-			on_connect
-		end
-		def send_headers
-			result = "GET #{path} HTTP/1.1\r\n"
-			result << "Upgrade: WebSocket\r\n"
-			result << "Connection: Upgrade\r\n"
-			result << "Host: #{host}\r\n"
-			result << "Origin: #{origin}\r\n\r\n"
-			send_data result
-		end
-	end
+module WebSocket
+	class Client < EM::Connection
+		def path
+			"/chat"
+		end
+		def host
+			"localhost:8000"
+		end
+		def origin
+			"localhost"
+		end
+		# em override
+		def post_init
+			@connected = false
+		end
+		def connection_completed
+			send_headers
+		end
+		# em override
+		def unbind
+			on_disconnect
+		end
+		def on_disconnect
+		end
+		def send_message msg
+			send_data Frame.encode(msg)
+		end
+		private
+		def receive_data data
+			unless @connected
+				handshake data
+			else
+        while msg = data.slice!(/\000([^\377]*)\377/)
+          on_receive Frame.decode(msg)
+        end
+			end
+		end
+		def handshake data
+			#convert the headers to a hash
+			@headers   = Util.parse_headers(data)
+			@connected = true
+			on_connect
+		end
+		def send_headers
+			result = "GET #{path} HTTP/1.1\r\n"
+			result << "Upgrade: WebSocket\r\n"
+			result << "Connection: Upgrade\r\n"
+			result << "Host: #{host}\r\n"
+			result << "Origin: #{origin}\r\n\r\n"
+			send_data result
+		end
+	end
 end

data/lib/web_socket/server.rb CHANGED Viewed

@@ -1,146 +1,202 @@
-module WebSocket
-	class Server < EM::Connection
-		@@logger			= nil
-		@@num_connections	= 0
-		@@callbacks 		= {}
-		@@accepted_origins	= []
-		attr_accessor	:connected,
-						:headers
-		def initialize *args
-			super
-			@connected = false
-		end
-		def valid_origin?
-			@@accepted_origins.empty? || @@accepted_origins.include?(origin)
-		end
-		#not doing anything with this yet
-		def valid_path?
-			true
-		end
-		def valid_upgrade?
-			@headers[:upgrade] =~ /websocket/i
-		end
-		def origin
-			@headers[:origin]
-		end
-		def host
-			@headers[:host]
-		end
-		def path
-			@headers[:path]
-		end
-		def cookies
-			@headers[:cookie]
-		end
-		def self.path name, &block
-			@@callbacks[name] = block
-		end
-		#tcp connection established
-		def post_init
-			@@num_connections += 1
-		end
-		#must be public for em
-		def unbind
-			@@num_connections -= 1
-			on_disconnect
-		end
-		def send_message msg
-			send_data Frame.encode(msg)
-		end
-		protected
-		#override this method
-		def on_receive msg
-			log msg
-		end
-		#override this method
-		def on_connect
-			log "connected"
-		end
-		#override this method
-		def on_disconnect
-			log "disconnected"
-		end
-		def log msg
-			if @@logger
-				@@logger.info msg
-			else
-				puts msg
-			end
-		end
-		private
-		# when the connection receives data from the client
-		# we either handshake or handle the message at
-		# the app layer
-		def receive_data data
-			unless @connected
-				handshake data
-			else
-				on_receive Frame.decode(data)
-			end
-		end
-		# parse the headers, validate the origin and path
-		# and respond with appropiate headers for a
-		# healthy relationship with the client
-		def handshake data
-			#convert the headers to a hash
-			@headers = Util.parse_headers(data)
-			# close the connection if the connection
-			# originates from an invalid source
-			close_connection unless valid_origin?
-			# close the connection if a callback
-			# is not registered for the path
-			close_connection unless valid_path?
-			# don't respond to non-websocket HTTP requests
-			close_connection unless valid_upgrade?
-			#complete the handshake
-			send_headers
-			@connected = true
-			on_connect
-		end
-		# send the handshake response headers to
-		# complete the initial com
-		def send_headers
-			response = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n"
-			response << "Upgrade: WebSocket\r\n"
-			response << "Connection: Upgrade\r\n"
-			response << "WebSocket-Origin: #{origin}\r\n"
-			response << "WebSocket-Location: ws://#{host}#{path}\r\n\r\n"
-			send_data response
-		end
-	end
+module WebSocket
+  class Server < EM::Connection
+    @@logger      = nil
+    @@num_connections  = 0
+    @@callbacks     = {}
+    @@accepted_origins  = []
+    attr_accessor  :connected,
+            :headers
+    def initialize *args
+      super
+      @connected = false
+      @protocol_version = 75
+    end
+    def valid_origin?
+      @@accepted_origins.empty? || @@accepted_origins.include?(origin)
+    end
+    #not doing anything with this yet
+    def valid_path?
+      true
+    end
+    def valid_upgrade?
+      @headers[:upgrade] =~ /websocket/i
+    end
+    def origin
+      @headers[:origin]
+    end
+    def host
+      @headers[:host]
+    end
+    def path
+      @headers[:path]
+    end
+    def cookies
+      @headers[:cookie]
+    end
+    def protocol
+      @headers[:protocol]
+    end
+    def self.path name, &block
+      @@callbacks[name] = block
+    end
+    #tcp connection established
+    def post_init
+      @@num_connections += 1
+    end
+    #must be public for em
+    def unbind
+      @@num_connections -= 1
+      on_disconnect
+    end
+    def send_message msg
+      send_data Frame.encode(msg)
+    end
+    protected
+    #override this method
+    def on_receive msg
+      log msg
+    end
+    #override this method
+    def on_connect
+      log "connected"
+    end
+    #override this method
+    def on_disconnect
+      log "disconnected"
+    end
+    def log msg
+      if @@logger
+        @@logger.info msg
+      else
+        puts msg
+      end
+    end
+    private
+    # when the connection receives data from the client
+    # we either handshake or handle the message at
+    # the app layer
+    def receive_data data
+      unless @connected
+        handshake data
+      else
+        while msg = data.slice!(/\000([^\377]*)\377/)
+          on_receive Frame.decode(msg)
+        end
+      end
+    end
+    # parse the headers, validate the origin and path
+    # and respond with appropiate headers for a
+    # healthy relationship with the client
+    def handshake data
+      #convert the headers to a hash
+      @headers, challenge = Util.parse_headers(data)
+      if challenge
+        @protocol_version = 76
+        key1 = @headers[:'sec-websocket-key1']
+        key2 = @headers[:'sec-websocket-key2']
+        part1 = number_from_key(key1)
+        part2 = number_from_key(key2)
+        unless part1 && part2
+          close_connection
+          return
+        end
+        buffer = []
+        buffer += big_endian_bytes(part1)
+        buffer += big_endian_bytes(part2)
+        buffer += challenge.bytes.to_a
+        md5 = Digest::MD5.new
+        hash = md5.digest(buffer.pack('c*'))
+      end
+      # close the connection if the connection
+      # originates from an invalid source
+      close_connection unless valid_origin?
+      # close the connection if a callback
+      # is not registered for the path
+      close_connection unless valid_path?
+      # don't respond to non-websocket HTTP requests
+      close_connection unless valid_upgrade?
+      #complete the handshake
+      send_headers(hash)
+      @connected = true
+      on_connect
+    end
+    # send the handshake response headers to
+    # complete the initial com
+    def send_headers(hash)
+      response = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n"
+      response << "Upgrade: WebSocket\r\n"
+      response << "Connection: Upgrade\r\n"
+      if @protocol_version > 75
+        response << "Sec-WebSocket-Origin: #{origin}\r\n"
+        response << "Sec-WebSocket-Location: ws://#{host}#{path}\r\n"
+        response << "Sec-WebSocket-protocol: ws://#{host}#{path}\r\n"
+      else
+        response << "WebSocket-Origin: #{origin}\r\n"
+        response << "WebSocket-Location: ws://#{host}#{path}\r\n"
+      end
+      response << "\r\n"
+      response << hash if @protocol_version > 75
+      send_data response
+    end
+    def number_from_key(key)
+      digits = key.scan(/\d+/).join.to_i
+      spaces = key.scan(/\s+/).join.length
+      if spaces > 0
+        return digits / spaces
+      else
+        return nil
+      end
+    end
+    def big_endian_bytes(num)
+      bytes = []
+      4.times do
+        bytes.unshift(num & 0xff)
+        num >>= 8
+      end
+      bytes
+    end
+  end
 end