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

em-websocket-server 0.1.2 → 0.13

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