rbsrt 0.1.0

data/Rakefile

@@ -0,0 +1,29 @@
+require 'bundler'
+require 'rake/extensiontask'
+
+Bundler::GemHelper.install_tasks
+
+task :default => :test
+
+# desc "Generate Makefile"
+# file "Makefile" do
+#   cd File.expand_path("../ext/rbsrt/extconf", __FILE__)
+#   require File.expand_path("../ext/rbsrt/extconf", __FILE__)
+# end
+#
+# desc "Compile extension"
+# task :compile => ["Makefile"] do
+#   sh "make"
+# end
+
+require 'rake/extensiontask'
+
+Rake::ExtensionTask.new 'rbsrt' do |ext|
+  ext.lib_dir = "lib/rbsrt"
+end
+
+desc "Run tests"
+task :test => [ :compile ] do
+    require './test/test_helper'
+    Dir.glob( './test/**/*_test.rb').each { |file| require file }
+end

data/bin/rbsrt-client

@@ -0,0 +1,113 @@
+#!/usr/bin/env ruby
+
+$:.push File.expand_path("../../lib", __FILE__)
+
+# parse options
+
+require 'optparse'
+
+options = {
+  :direction => "out",
+  :exec => nil,
+  :file => nil,
+  :url => ARGV.last,
+  :passphrase => nil,
+  :streamid => {
+    :mode => :request,
+    :type => :stream
+  }
+}
+
+OptionParser.new do |opts|
+  opts.banner = "Usage: rbsrt-client [options] srt://host:port"
+  
+  opts.on("-e CMD", "--exec=CMD", "command to execute an use as input or output") do |cmd|
+    options[:exec] = cmd
+  end
+  
+  opts.on("-f PATH", "--file=PATH", "file to send to the server") do |path|
+    options[:path] = path
+    options[:type] = :file
+  end
+  
+  opts.on("-r NAME", "--resource-name=NAME", "name of the resource to transmit or receive") do |resource_name|
+    options[:streamid][:resource_name] = resource_name
+  end
+  
+  opts.on("-u USER_NAME", "--user-name=USER_NAME", "user name") do |user_name|
+    options[:streamid][:user_name] = user_name
+  end
+  
+  opts.on("-m MODE", "--mode=MODE", "connection mode. Can be one of 'request', 'publish' or 'bidirectional'") do |mode|
+    options[:streamid][:mode] = mode
+  end
+  
+  opts.on("-t TYPE", "--type=TYPE", "connection type. Can be one of 'stream', 'file' or 'auth'") do |type|
+    options[:streamid][:type] = type
+  end
+  
+  opts.on("-s SESSIONID", "--sessionid=SESSIONID", "a sessionid") do |sessionid|
+    options[:streamid][:sessionid] = sessionid
+  end
+  
+  opts.on("-S PASSPHRASE", "--passphrase=PASSPHRASE", "passphrase used to derive key for encryption") do |passphrase|
+    options[:passphrase] = passphrase
+  end
+end.parse!
+
+
+# start server
+
+require 'rbsrt'
+require 'uri'
+
+puts "create srt client"
+
+streamid_info = SRT::StreamIDComponents.new options[:streamid]
+
+puts streamid_info
+
+client = SRT::Client.new
+
+client.transmission_mode = streamid_info.type == :file ? :file : :live
+client.streamid = streamid_info.to_s
+client.passphrase = options[:passphrase]
+
+puts "connect srt client"
+
+puts "before connect: #{client.connected?}"
+
+at_exit { client.close }
+
+begin
+  uri = URI.parse(options[:url])
+  puts "connect to #{uri.host.to_s} on port #{uri.port.to_s}"
+  client.connect uri.host.to_s, uri.port.to_s
+rescue StandardError => e
+  puts "Connection failed. #{e.message}"
+  exit 1
+end
+
+
+if streamid_info.mode == :request
+  while chunk = client.read
+    print chunk
+  end
+end
+
+# transmit file contents
+if file = options[:file]
+  puts "file transport is not implemented yet"
+# transmist cmd stdout
+elsif cmd = options[:exec]
+  require 'open3'
+  Open3.popen3(cmd) do |stdin, stdout, stderr, wait_thr|
+    stdout.each do |buf|
+      puts "data: #{buf.length}"
+      client.sendmsg buf
+    end
+  end
+else
+  
+end
+  

data/bin/rbsrt-server

@@ -0,0 +1,68 @@
+#!/usr/bin/env ruby
+
+$:.push File.expand_path("../../lib", __FILE__)
+
+# parse options
+
+require 'optparse'
+
+options = {
+  :port => "5554",
+  :address => "0.0.0.0",
+  :recording_path => Dir.pwd,
+  :passphrase => nil
+}
+
+OptionParser.new do |opts|
+  opts.banner = "Usage: rbsrt-server [options]"
+
+  opts.on("-p PORT", "--port=PORT", "server port (default: #{options[:port]})") do |port|
+    options[:port] = port.to_s
+  end
+  
+  opts.on("-a ADDRESS", "--address=ADDRESS", "server address (default: #{options[:address]})") do |address|
+    options[:address] = address.to_s
+  end
+
+  opts.on("-r PATH", "--recording-path=PATH", "directory to save recordings (default: #{options[:recording_path]})") do |path|
+    options[:recording_path] = File.expand_path(path, Dir.pwd)
+  end
+
+  opts.on("-S PASSPHRASE", "--passphrase=PASSPHRASE", "passphrase use to derive key for encryption") do |passphrase|
+    options[:passphrase] = passphrase
+  end
+end.parse!
+
+
+
+
+# start the server
+
+require 'rbsrt'
+
+puts "start srt server on port #{options[:port]}, address #{options[:address]}"
+
+server = SRT::Server.new options[:address], options[:port]
+
+# currently we can only set one passphrase for every connection
+# this will change in the future.
+server.passphrase = options[:passphrase] if options[:passphrase]
+
+puts "starting server"
+
+server.start do |connection|
+  puts "new connection: connections=#{connection_count}, id=#{connection.id}, streamid=#{connection.streamid}"
+
+  output_file = File.new "recording-#{Process.pid}-#{connection.id}.ts", "w+"
+  output_file.close_on_exec = true
+
+  connection.at_data do |chunk|
+    puts "data"
+    output_file.write(chunk)
+  end
+
+  connection.at_close do
+    puts "closed connection"
+    output_file.close
+  end
+end

data/bin/rbsrt-sockets

@@ -0,0 +1,82 @@
+#!/usr/bin/env ruby
+
+$:.push File.expand_path("../../lib", __FILE__)
+
+require "rbsrt"
+
+pids = []
+
+puts "Running rbsrt Sockets examples"
+puts "SRT version: #{SRT::SRT_VERSION}"
+
+
+# server
+
+pids << fork do
+  server = SRT::Socket.new
+  server.transmission_mode = :file
+
+  puts "server state=#{server.state}, id=#{server.id}"
+
+  server.bind "127.0.0.1", "5556"
+
+  puts "server #{server.state}"
+
+  server.listen 3
+
+  puts "server #{server.state}"
+
+  client = server.accept
+
+  puts "server accepted:  streamid=\"#{client.streamid}\", id=#{client.id}, tsbpdmode=#{client.tsbpdmode?}"
+
+  loop do
+    break unless client.connected?
+    break unless data = client.recvmsg
+    puts "server received: #{data}"
+  rescue SRT::Error::CONNLOST => e
+    puts "connection lost: #{e}"
+    break
+  rescue SRT::Error => e
+    puts "server error: #{e.class}"
+  end
+
+  puts "client closed #{client.closed?}"
+
+  server.close
+
+  puts "server done"
+end
+
+# client
+
+pids << fork do
+
+  client = SRT::Socket.new
+
+  client.streamid = "a ruby srt socket"
+  client.transmission_mode = :file
+
+  client.connect "127.0.0.1", "5556"
+
+  10.times do |i|
+    payload = "Hello from ruby client #{i}"
+    puts "will send: \"#{payload}\""
+    client.sendmsg "Hello from ruby client #{i}"
+    puts "did send: \"#{payload}\""
+    # sleep 0.1
+  rescue SRT::Error => e
+    puts "client error: #{e}"
+  end
+
+  sleep 0.1
+
+  client.close
+
+  puts "client done"
+end
+
+
+# wait for completion
+
+pids.each { |pid| Process.wait pid }

data/ext/rbsrt/extconf.rb

@@ -0,0 +1,37 @@
+require 'mkmf'
+
+extension_name = 'rbsrt'
+
+RbConfig::MAKEFILE_CONFIG['CC'] = ENV['CC'] if ENV['CC']
+
+LIBDIR      = CONFIG['libdir']
+INCLUDEDIR  = CONFIG['includedir']
+
+HEADER_DIRS = []
+HEADER_DIRS << 'opt/local/include' if File.directory?('opt/local/include')
+HEADER_DIRS << '/usr/local/include' if File.directory?('/usr/local/include')
+HEADER_DIRS << INCLUDEDIR
+HEADER_DIRS << '/usr/include' if File.directory?('/usr/include')
+
+LIB_DIRS = []
+LIB_DIRS << '/opt/local/lib' if File.directory?('/opt/local/lib')
+LIB_DIRS << '/usr/local/lib' if File.directory?('/usr/local/lib')
+LIB_DIRS << LIBDIR
+LIB_DIRS << '/usr/lib' if File.directory?('/usr/lib')
+
+dir_config('srt', HEADER_DIRS, LIB_DIRS)
+# dir_config('c++', HEADER_DIRS, LIB_DIRS)
+
+# srt dependencies
+
+unless find_header('srt/srt.h')
+  abort "libsrt is missing.  please install libsrt: https://github.com/Haivision/srt"
+end
+
+unless find_library('srt', 'srt_create_socket')
+  abort "libsrt is missing.  please install libsrt: https://github.com/Haivision/srt"
+end
+
+dir_config(extension_name)
+
+create_makefile(extension_name)