63 lines
1.2 KiB
Ruby
Executable File
63 lines
1.2 KiB
Ruby
Executable File
#!/usr/bin/env ruby
|
|
|
|
# A WebSocket server that echos back whatever it receives from the client.
|
|
# Copyright 2011 Joel Martin
|
|
# Licensed under LGPL version 3 (see docs/LICENSE.LGPL-3)
|
|
|
|
require 'socket'
|
|
$: << "other"
|
|
$: << "../other"
|
|
require 'websocket'
|
|
|
|
class WebSocketEcho < WebSocketServer
|
|
|
|
# Echo back whatever is received
|
|
def new_client(client)
|
|
|
|
cqueue = []
|
|
c_pend = 0
|
|
rlist = [client]
|
|
|
|
loop do
|
|
wlist = []
|
|
|
|
if cqueue.length > 0 or c_pend
|
|
wlist << client
|
|
end
|
|
|
|
ins, outs, excepts = IO.select(rlist, wlist, nil, 1)
|
|
if excepts.length > 0
|
|
raise Exception, "Socket exception"
|
|
end
|
|
|
|
if outs.include?(client)
|
|
# Send queued data to the client
|
|
c_pend = send_frames(cqueue)
|
|
cqueue = []
|
|
end
|
|
|
|
if ins.include?(client)
|
|
# Receive client data, decode it, and send it back
|
|
frames, closed = recv_frames
|
|
cqueue += frames
|
|
|
|
if closed
|
|
raise EClose, closed
|
|
end
|
|
end
|
|
|
|
end # loop
|
|
end
|
|
end
|
|
|
|
port = ARGV[0].to_i
|
|
puts "Starting server on port #{port}"
|
|
|
|
server = WebSocketEcho.new('listen_port' => port, 'verbose' => true)
|
|
server.start
|
|
server.join
|
|
|
|
puts "Server has been terminated"
|
|
|
|
# vim: sw=2
|