_bushido-faye-websocket 0.4.4

data/examples/autobahn_client.rb

@@ -0,0 +1,44 @@
+require 'rubygems'
+require File.expand_path('../../lib/faye/websocket', __FILE__)
+require 'cgi'
+
+EM.run {
+  host   = 'ws://localhost:9001'
+  agent  = "Faye (Ruby #{RUBY_VERSION})"
+  cases  = 0
+  skip   = []
+  
+  socket = Faye::WebSocket::Client.new("#{host}/getCaseCount")
+  
+  socket.onmessage = lambda do |event|
+    puts "Total cases to run: #{event.data}"
+    cases = event.data.to_i
+  end
+  
+  socket.onclose = lambda do |event|
+    run_case = lambda do |n|
+      if n > cases
+        socket = Faye::WebSocket::Client.new("#{host}/updateReports?agent=#{CGI.escape agent}")
+        socket.onclose = lambda { |e| EM.stop }
+        
+      elsif skip.include?(n)
+        EM.next_tick { run_case.call(n+1) }
+        
+      else
+        puts "Running test case ##{n} ..."
+        socket = Faye::WebSocket::Client.new("#{host}/runCase?case=#{n}&agent=#{CGI.escape agent}")
+        
+        socket.onmessage = lambda do |event|
+          socket.send(event.data)
+        end
+        
+        socket.onclose = lambda do |event|
+          run_case.call(n + 1)
+        end
+      end
+    end
+    
+    run_case.call(1)
+  end
+}
+

data/examples/client.rb

@@ -0,0 +1,30 @@
+require 'rubygems'
+require File.expand_path('../../lib/faye/websocket', __FILE__)
+require 'eventmachine'
+
+port   = ARGV[0] || 7000
+secure = ARGV[1] == 'ssl'
+
+EM.run {
+  scheme = secure ? 'wss' : 'ws'
+  url    = "#{scheme}://localhost:#{port}/"
+  socket = Faye::WebSocket::Client.new(url)
+  
+  puts "Connecting to #{socket.url}"
+  
+  socket.onopen = lambda do |event|
+    p [:open]
+    socket.send("Hello, WebSocket!")
+  end
+  
+  socket.onmessage = lambda do |event|
+    p [:message, event.data]
+    # socket.close 1002, 'Going away'
+  end
+  
+  socket.onclose = lambda do |event|
+    p [:close, event.code, event.reason]
+    EM.stop
+  end
+}
+

data/examples/config.ru

@@ -0,0 +1,17 @@
+# Run using your favourite async server:
+# 
+#     thin start -R examples/config.ru -p 7000
+#     rainbows -c spec/rainbows.conf -E production examples/config.ru -p 7000
+# 
+# If you run using one of these commands, the webserver is loaded before this
+# file, so Faye::WebSocket can figure out which adapter to load. If instead you
+# run using `rackup`, you need the `load_adapter` line below.
+# 
+#     rackup -E production -s thin examples/config.ru -p 7000
+
+require 'rubygems'
+require File.expand_path('../app', __FILE__)
+# Faye::WebSocket.load_adapter('thin')
+
+run App
+

data/examples/haproxy.conf

@@ -0,0 +1,21 @@
+defaults
+   mode http
+   timeout client  5s
+   timeout connect 5s
+   timeout server  5s
+
+frontend all 0.0.0.0:3000
+   mode http
+   timeout client 120s
+
+   option forwardfor
+   option http-server-close
+   option http-pretend-keepalive
+
+   default_backend sockets
+
+backend sockets
+   balance uri depth 2
+   timeout server  120s
+   server socket1 127.0.0.1:7000
+

data/examples/server.rb

@@ -0,0 +1,44 @@
+require 'rubygems'
+require 'rack/content_length'
+require 'rack/chunked'
+
+port   = ARGV[0] || 7000
+secure = ARGV[1] == 'ssl'
+engine = ARGV[2] || 'thin'
+spec   = File.expand_path('../../spec', __FILE__)
+
+require File.expand_path('../app', __FILE__)
+Faye::WebSocket.load_adapter(engine)
+
+case engine
+
+when 'goliath'
+  class WebSocketServer < Goliath::API
+    def response(env)
+      App.call(env)
+    end
+  end
+
+when 'rainbows'
+  rackup = Unicorn::Configurator::RACKUP
+  rackup[:port] = port
+  rackup[:set_listener] = true
+  options = rackup[:options]
+  options[:config_file] = spec + '/rainbows.conf'
+  Rainbows::HttpServer.new(App, options).start.join
+
+when 'thin'
+  EM.run {
+    thin = Rack::Handler.get('thin')
+    thin.run(App, :Port => port) do |server|
+      if secure
+        server.ssl_options = {
+          :private_key_file => spec + '/server.key',
+          :cert_chain_file  => spec + '/server.crt'
+        }
+        server.ssl = true
+      end
+    end
+  }
+end
+

data/examples/sse.html

@@ -0,0 +1,39 @@
+<!doctype html>
+<html>
+  <head>
+    <meta http-equiv="Content-type" content="text/html; charset=utf-8">
+    <title>EventSource test</title>
+  </head>
+  <body>
+    
+    <h1>EventSource test</h1>
+    <ul></ul>
+    
+    <script type="text/javascript">
+      var logger = document.getElementsByTagName('ul')[0],
+          socket = new EventSource('/');
+      
+      var log = function(text) {
+        logger.innerHTML += '<li>' + text + '</li>';
+      };
+      
+      socket.onopen = function() {
+        log('OPEN');
+      };
+      
+      socket.onmessage = function(event) {
+        log('MESSAGE: ' + event.data);
+      };
+      
+      socket.addEventListener('update', function(event) {
+        log('UPDATE(' + event.lastEventId + '): ' + event.data);
+      });
+      
+      socket.onerror = function(event) {
+        log('ERROR: ' + event.message);
+      };
+    </script>
+    
+  </body>
+</html>
+

data/examples/ws.html

@@ -0,0 +1,44 @@
+<!doctype html>
+<html>
+  <head>
+    <meta http-equiv="Content-type" content="text/html; charset=utf-8">
+    <title>WebSocket test</title>
+  </head>
+  <body>
+    
+    <h1>WebSocket test</h1>
+    <ul></ul>
+    
+    <script type="text/javascript">
+      var logger = document.getElementsByTagName('ul')[0],
+          Socket = window.MozWebSocket || window.WebSocket,
+          protos = ['foo', 'bar', 'xmpp'],
+          socket = new Socket('ws://' + location.hostname + ':' + location.port + '/', protos),
+          index  = 0;
+      
+      var log = function(text) {
+        logger.innerHTML += '<li>' + text + '</li>';
+      };
+      
+      socket.addEventListener('open', function() {
+        log('OPEN: ' + socket.protocol);
+        socket.send('Hello, world');
+      });
+      
+      socket.onerror = function(event) {
+        log('ERROR: ' + event.message);
+      };
+      
+      socket.onmessage = function(event) {
+        log('MESSAGE: ' + event.data);
+        setTimeout(function() { socket.send(++index + ' ' + event.data) }, 2000);
+      };
+      
+      socket.onclose = function(event) {
+        log('CLOSE: ' + event.code + ', ' + event.reason);
+      };
+    </script>
+    
+  </body>
+</html>
+

data/ext/faye_websocket_mask/FayeWebsocketMaskService.java

@@ -0,0 +1,61 @@
+package com.jcoglan.faye;
+
+import java.lang.Long;
+import java.io.IOException;
+
+import org.jruby.Ruby;
+import org.jruby.RubyArray;
+import org.jruby.RubyClass;
+import org.jruby.RubyFixnum;
+import org.jruby.RubyModule;
+import org.jruby.RubyObject;
+import org.jruby.anno.JRubyMethod;
+import org.jruby.runtime.ObjectAllocator;
+import org.jruby.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+import org.jruby.runtime.load.BasicLibraryService;
+
+public class FayeWebsocketMaskService implements BasicLibraryService {
+  private Ruby runtime;
+  
+  public boolean basicLoad(Ruby runtime) throws IOException {
+    this.runtime = runtime;
+    RubyModule faye = runtime.defineModule("Faye");
+    
+    RubyClass webSocketMask = faye.defineClassUnder("WebSocketMask", runtime.getObject(), new ObjectAllocator() {
+      public IRubyObject allocate(Ruby runtime, RubyClass rubyClass) {
+        return new WebsocketMask(runtime, rubyClass);
+      }
+    });
+    
+    webSocketMask.defineAnnotatedMethods(WebsocketMask.class);
+    return true;
+  }
+  
+  public class WebsocketMask extends RubyObject {
+    public WebsocketMask(final Ruby runtime, RubyClass rubyClass) {
+      super(runtime, rubyClass);
+    }
+    
+    @JRubyMethod
+    public IRubyObject mask(ThreadContext context, IRubyObject payload, IRubyObject mask) {
+      int n = ((RubyArray)payload).getLength(), i;
+      long p, m;
+      RubyArray unmasked = RubyArray.newArray(runtime, n);
+      
+      long[] maskArray = {
+        (Long)((RubyArray)mask).get(0),
+        (Long)((RubyArray)mask).get(1),
+        (Long)((RubyArray)mask).get(2),
+        (Long)((RubyArray)mask).get(3)
+      };
+      
+      for (i = 0; i < n; i++) {
+        p = (Long)((RubyArray)payload).get(i);
+        m = maskArray[i % 4];
+        unmasked.set(i, p ^ m);
+      }
+      return unmasked;
+    }
+  }
+}

data/ext/faye_websocket_mask/extconf.rb

@@ -0,0 +1,5 @@
+require 'mkmf'
+extension_name = 'faye_websocket_mask'
+dir_config(extension_name)
+create_makefile(extension_name)
+

data/ext/faye_websocket_mask/faye_websocket_mask.c

@@ -0,0 +1,33 @@
+#include <ruby.h>
+
+VALUE Faye = Qnil;
+VALUE FayeWebSocketMask = Qnil;
+
+void Init_faye_websocket_mask();
+VALUE method_faye_websocket_mask(VALUE self, VALUE payload, VALUE mask);
+
+void Init_faye_websocket_mask() {
+  Faye = rb_define_module("Faye");
+  FayeWebSocketMask = rb_define_module_under(Faye, "WebSocketMask");
+  rb_define_singleton_method(FayeWebSocketMask, "mask", method_faye_websocket_mask, 2);
+}
+
+VALUE method_faye_websocket_mask(VALUE self, VALUE payload, VALUE mask) {
+  int n = RARRAY_LEN(payload), i, p, m;
+  VALUE unmasked = rb_ary_new2(n);
+  
+  int mask_array[] = {
+    NUM2INT(rb_ary_entry(mask, 0)),
+    NUM2INT(rb_ary_entry(mask, 1)),
+    NUM2INT(rb_ary_entry(mask, 2)),
+    NUM2INT(rb_ary_entry(mask, 3))
+  };
+  
+  for (i = 0; i < n; i++) {
+    p = NUM2INT(rb_ary_entry(payload, i));
+    m = mask_array[i % 4];
+    rb_ary_store(unmasked, i, INT2NUM(p ^ m));
+  }
+  return unmasked;
+}
+

data/lib/faye/adapters/goliath.rb

@@ -0,0 +1,47 @@
+class Goliath::Connection
+  attr_accessor :socket_stream
+  alias :goliath_receive_data :receive_data
+  
+  def receive_data(data)
+    if @serving == :websocket
+      socket_stream.receive(data) if socket_stream
+    else
+      goliath_receive_data(data)
+      socket_stream.receive(@parser.upgrade_data) if socket_stream
+      @serving = :websocket if @api.websocket?
+    end
+  end
+  
+  def unbind
+    super
+  ensure
+    socket_stream.fail if socket_stream
+  end
+end
+
+class Goliath::API
+  include Faye::WebSocket::Adapter
+end
+
+class Goliath::Request
+  alias :goliath_process :process
+  
+  def process
+    env['em.connection'] = conn
+    goliath_process
+  end
+end
+
+class Goliath::Response
+  alias :goliath_head :head
+  alias :goliath_headers_output :headers_output
+  
+  def head
+    (status == 101) ? '' : goliath_head
+  end
+  
+  def headers_output
+    (status == 101) ? '' : goliath_headers_output
+  end
+end
+

data/lib/faye/adapters/rainbows.rb

@@ -0,0 +1,32 @@
+# WebSocket extensions for Rainbows
+# Based on code from the Cramp project
+# http://github.com/lifo/cramp
+
+# Copyright (c) 2009-2011 Pratik Naik
+# 
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+# 
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+module Faye
+  class WebSocket
+    autoload :RainbowsClient, File.expand_path('../rainbows_client', __FILE__)
+  end
+end
+
+Rainbows::O[:em_client_class] = "Faye::WebSocket::RainbowsClient"

data/lib/faye/adapters/rainbows_client.rb

@@ -0,0 +1,70 @@
+# WebSocket extensions for Rainbows
+# Based on code from the Cramp project
+# http://github.com/lifo/cramp
+
+# Copyright (c) 2009-2011 Pratik Naik
+# 
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+# 
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+module Faye
+  class WebSocket
+    
+    class RainbowsClient < Rainbows::EventMachine::Client
+      include Faye::WebSocket::Adapter
+      attr_accessor :socket_stream
+      
+      def receive_data(data)
+        return super unless @state == :websocket
+        socket_stream.receive(data) if socket_stream
+      end
+      
+      def app_call(*args)
+        @env['em.connection'] = self
+        if args.first == NULL_IO and @hp.content_length == 0 and websocket?
+          prepare_request_body
+        else
+          super
+        end
+      end
+      
+      def on_read(data)
+        if @state == :body and websocket? and @hp.body_eof?
+          @state = :websocket
+          @input.rewind
+          app_call StringIO.new(@buf)
+        else
+          super
+        end
+      end
+      
+      def unbind
+        super
+      ensure
+        socket_stream.fail if socket_stream
+      end
+      
+      def write_headers(*args)
+        super unless async_connection?
+      end
+    end
+    
+  end
+end
+