eventmachine 0.12.0-i386-mswin32

data/lib/protocols/stomp.rb

@@ -0,0 +1,127 @@
+# $Id: stomp.rb 668 2008-01-04 23:00:34Z blackhedd $
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 15 November 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+# 
+# 
+
+
+
+module EventMachine
+	module Protocols
+
+		# Implements Stomp (http://docs.codehaus.org/display/STOMP/Protocol).
+		#
+		module Stomp
+			include LineText2
+
+			class Message
+				attr_accessor :command, :header, :body
+				def initialize
+					@header = {}
+					@state = :precommand
+					@content_length = nil
+				end
+				def consume_line line
+					if @state == :precommand
+						unless line =~ /\A\s*\Z/
+							@command = line
+							@state = :headers
+						end
+					elsif @state == :headers
+						if line == ""
+							if @content_length
+								yield( [:sized_text, @content_length+1] )
+							else
+								@state = :body
+								yield( [:unsized_text] )
+							end
+						elsif line =~ /\A([^:]+):(.+)\Z/
+							k = $1.dup.strip
+							v = $2.dup.strip
+							@header[k] = v
+							if k == "content-length"
+								@content_length = v.to_i
+							end
+						else
+							# This is a protocol error. How to signal it?
+						end
+					elsif @state == :body
+						@body = line
+						yield( [:dispatch] )
+					end
+				end
+			end
+
+			def send_frame verb, headers={}, body=""
+				ary = [verb, "\n"]
+				headers.each {|k,v| ary << "#{k}:#{v}\n" }
+				ary << "content-length: #{body.to_s.length}\n"
+				ary << "content-type: text/plain; charset=UTF-8\n"
+				ary << "\n"
+				ary << body.to_s
+				ary << "\0"
+				send_data ary.join
+			end
+
+			def receive_line line
+				@stomp_initialized || init_message_reader
+				@stomp_message.consume_line(line) {|outcome|
+					if outcome.first == :sized_text
+						set_text_mode outcome[1]
+					elsif outcome.first == :unsized_text
+						set_delimiter "\0"
+					elsif outcome.first == :dispatch
+						receive_msg(@stomp_message) if respond_to?(:receive_msg)
+						init_message_reader
+					end
+				}
+			end
+			def receive_binary_data data
+				@stomp_message.body = data[0..-2]
+				receive_msg(@stomp_message) if respond_to?(:receive_msg)
+				init_message_reader
+			end
+
+			def init_message_reader
+				@stomp_initialized = true
+				set_delimiter "\n"
+				set_line_mode
+				@stomp_message = Message.new
+			end
+
+
+			# STOMP verbs
+			def connect parms={}
+				send_frame "CONNECT", parms
+			end
+			def send destination, body, parms={}
+				send_frame "SEND", parms.merge( :destination=>destination ), body.to_s
+			end
+			def subscribe dest, ack=false
+				send_frame "SUBSCRIBE", {:destination=>dest, :ack=>(ack ? "client" : "auto")}
+			end
+
+		end
+	end
+end
+

data/lib/protocols/tcptest.rb

@@ -0,0 +1,57 @@
+# $Id: tcptest.rb 668 2008-01-04 23:00:34Z blackhedd $
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 16 July 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+#
+# 
+
+
+
+module EventMachine
+module Protocols
+
+class TcpConnectTester < Connection
+  include EventMachine::Deferrable
+
+  def self.test( host, port )
+    EventMachine.connect( host, port, self )
+  end
+
+  def post_init
+    @start_time = Time.now
+  end
+
+  def connection_completed
+    @completed = true
+    set_deferred_status :succeeded, (Time.now - @start_time)
+    close_connection
+  end
+
+  def unbind
+    set_deferred_status :failed, (Time.now - @start_time)  unless @completed
+  end
+end
+
+
+end
+end
+

data/tests/test_basic.rb

@@ -0,0 +1,142 @@
+# $Id: test_basic.rb 668 2008-01-04 23:00:34Z blackhedd $
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 8 April 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+#
+# 
+
+$:.unshift "../lib"
+require 'eventmachine'
+require 'test/unit'
+
+class TestBasic < Test::Unit::TestCase
+
+  def setup
+  end
+
+  def teardown
+  end
+
+  #-------------------------------------
+
+  def test_libtype
+    lt = EventMachine.library_type
+    case (ENV["EVENTMACHINE_LIBRARY"] || $eventmachine_library || :xxx).to_sym
+    when :pure_ruby
+      assert_equal( :pure_ruby, lt )
+    when :extension
+      assert_equal( :extension, lt )
+    when :java
+      assert_equal( :java, lt )
+    else
+      assert_equal( :extension, lt )
+    end
+  end
+
+  #-------------------------------------
+
+
+  def test_em
+    EventMachine.run {
+      EventMachine.add_timer 0 do
+        EventMachine.stop
+      end
+    }
+  end
+
+  #-------------------------------------
+
+  def test_timer
+    n = 0
+    EventMachine.run {
+      EventMachine.add_periodic_timer(1) {
+        n += 1
+        EventMachine.stop if n == 2
+      }
+    }
+  end
+
+  #-------------------------------------
+
+  # This test once threw an already-running exception.
+  module Trivial
+    def post_init
+      EventMachine.stop
+    end
+  end
+
+  def test_server
+    EventMachine.run {
+      EventMachine.start_server "localhost", 9000, Trivial
+      EventMachine.connect "localhost", 9000
+    }
+    assert( true ) # make sure it halts
+  end
+
+  #--------------------------------------
+
+  # EventMachine#run_block starts the reactor loop, runs the supplied block, and then STOPS
+  # the loop automatically. Contrast with EventMachine#run, which keeps running the reactor
+  # even after the supplied block completes.
+  def test_run_block
+	  a = nil
+	  EM.run_block { a = "Worked" }
+	  assert a
+  end
+
+
+  #--------------------------------------
+
+  # TODO! This is an unfinished edge case.
+  # EM mishandles uncaught Ruby exceptions that fire from within #unbind handlers.
+  # A uncaught Ruby exception results in a call to EM::release_machine (which is in an ensure
+  # block in EM::run). But if EM is processing an unbind request, the release_machine call
+  # will cause a segmentation fault.
+  #
+
+  TestHost = "127.0.0.1"
+  TestPort = 9070
+
+  class UnbindError < EM::Connection
+	  def initialize *args
+		  super
+	  end
+	  def connection_completed
+		  close_connection_after_writing
+	  end
+	  def unbind
+		  raise "Blooey"
+	  end
+  end
+
+  def xxx_test_unbind_error
+	  assert_raise( RuntimeError ) {
+		  EM.run {
+			  EM.start_server TestHost, TestPort
+			  EM.connect TestHost, TestPort, UnbindError
+		  }
+	  }
+  end
+
+end
+
+

data/tests/test_defer.rb

@@ -0,0 +1,63 @@
+# $Id: test_defer.rb 668 2008-01-04 23:00:34Z blackhedd $
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 8 April 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+#
+#
+
+$:.unshift "../lib"
+require 'eventmachine'
+require 'test/unit'
+
+class TestDeferUsage < Test::Unit::TestCase
+
+	def setup
+	end
+
+	def teardown
+	end
+
+	def run_em_with_defers
+		n = 0
+		n_times = 20
+		EM.run {
+			n_times.times {
+				EM.defer proc {
+					sleep 0.1
+				}, proc {
+					n += 1
+					EM.stop if n == n_times
+				}
+			}
+		}
+		assert_equal( n, n_times )
+	end
+	def test_defers
+		10.times {
+			run_em_with_defers {|n,ntimes|
+				assert_equal( n, ntimes )
+			}
+		}
+	end
+
+end
+

data/tests/test_epoll.rb

@@ -0,0 +1,168 @@
+# $Id: test_epoll.rb 668 2008-01-04 23:00:34Z blackhedd $
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 8 April 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+#
+#
+# TODO, and I know this doesn't belong here, but if a datagram calls
+# send_data outside of a receive_data, there is no return address, and
+# the result is a very confusing error message.
+#
+
+$:.unshift "../lib"
+require 'eventmachine'
+require 'test/unit'
+
+
+class TestEpoll < Test::Unit::TestCase
+
+	def setup
+	end
+
+	def teardown
+	end
+
+
+	module TestEchoServer
+		def receive_data data
+			send_data data
+			close_connection_after_writing
+		end
+	end
+
+	module TestEchoClient
+		def connection_completed
+			send_data "ABCDE"
+			$max += 1
+		end
+		def receive_data data
+			raise "bad response" unless data == "ABCDE"
+		end
+		def unbind
+			$n -= 1
+			EM.stop if $n == 0
+		end
+	end
+
+
+	# We can set the rlimit/nofile of a process but we can only set it
+	# higher if we're running as root.
+	# On most systems, the default value is 1024.
+	# Java doesn't (currently) implement this.
+	def test_rlimit
+		unless RUBY_PLATFORM =~ /java/
+			a = EM.set_descriptor_table_size
+			assert( a <= 1024 )
+			a = EM.set_descriptor_table_size( 1024 )
+			assert( a == 1024 )
+		end
+	end
+
+	# Run a high-volume version of this test by kicking the number of connections
+	# up past 512. (Each connection uses two sockets, a client and a server.)
+	# (Will require running the test as root)
+	# This test exercises TCP clients and servers.
+	#
+	def test_descriptors
+		EM.epoll
+		s = EM.set_descriptor_table_size 60000
+		EM.run {
+			EM.start_server "127.0.0.1", 9800, TestEchoServer
+			$n = 0
+			$max = 0
+			100.times {
+				EM.connect("127.0.0.1", 9800, TestEchoClient) {$n += 1}
+			}
+		}
+		assert_equal(0, $n)
+		assert_equal(100, $max)
+	end
+
+	def test_defer
+		$n = 0
+		EM.epoll
+		EM.run {
+			sleep_proc = proc {sleep 1}
+			return_proc = proc {$n += 1; EM.stop}
+			EM.defer sleep_proc, return_proc
+		}
+		assert_equal( 1, $n )
+	end
+
+
+	module TestDatagramServer
+		def receive_data dgm
+			$in = dgm
+			send_data "abcdefghij"
+		end
+	end
+	module TestDatagramClient
+		def post_init
+			send_datagram "1234567890", "127.0.0.1", 9500
+		end
+		def receive_data dgm
+			$out = dgm
+			EM.stop
+		end
+	end
+
+	def test_datagrams
+		$in = $out = ""
+		EM.epoll
+		EM.run {
+			EM.open_datagram_socket "127.0.0.1", 9500, TestDatagramServer
+			EM.open_datagram_socket "127.0.0.1", 0, TestDatagramClient
+		}
+		assert_equal( "1234567890", $in )
+		assert_equal( "abcdefghij", $out )
+	end
+
+
+
+
+	def test_unix_domain
+		EM.epoll
+		s = EM.set_descriptor_table_size 60000
+		EM.run {
+			# The pure-Ruby version won't let us open the socket if the node already exists.
+			# Not sure, that actually may be correct and the compiled version is wrong.
+			# Pure Ruby also oddly won't let us make that many connections. This test used
+			# to run 100 times. Not sure where that lower connection-limit is coming from in
+			# pure Ruby.
+			# Let's not sweat the Unix-ness of the filename, since this test can't possibly
+			# work on Windows anyway.
+			#
+			fn = "/tmp/xxx.chain"
+			File.unlink(fn) if File.exist?(fn)
+			EM.start_unix_domain_server fn, TestEchoServer
+			$n = 0
+			$max = 0
+			50.times {
+				EM.connect_unix_domain(fn, TestEchoClient) {$n += 1}
+			}
+		}
+		assert_equal(0, $n)
+		assert_equal(50, $max)
+	end
+
+end
+