rbsrt 0.1.0

data/ext/rbsrt/rbstats.h

@@ -0,0 +1,24 @@
+/*
+ * Ruby SRT - Ruby binding for Secure, Reliable, Transport
+ * Copyright (c) 2019 Klaas Speller, Recce.
+ * 
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ * 
+ */
+
+/**
+ * A Ruby wrapper for SRT (Secure Reliable Transport)
+ * 
+ * @author: Klaas Speller <klaas@recce.nl>
+ */
+
+#ifndef RBSRT_STATS_H
+#define RBSRT_STATS_H
+
+#include <ruby/ruby.h>
+
+void RBSRT_stat_init(VALUE srt_module);
+
+#endif /* RBSRT_STATS_H */

data/lib/rbsrt/streamid_components.rb

@@ -0,0 +1,113 @@
+# srt streamid parser based on https://github.com/Haivision/srt/blob/master/docs/AccessControl.md
+
+require "uri"
+
+module SRT
+  class StreamIDComponents
+    Types = ["stream", "file", "auth"]
+
+    Modes = ["request", "publish", "bidirectional"]
+
+    Header = "#!::"
+    
+    attr_accessor :resource_name
+    attr_accessor :user_name
+    attr_accessor :mode
+    attr_accessor :host_name
+    attr_accessor :type
+    attr_accessor :sessionid
+    attr_reader :extra
+
+    def initialize(*args)
+
+      @mode = :request
+      @type = :stream
+      @extra = {}
+
+      # no args
+      return unless args.first
+
+      # hash args
+      if args.first.class == ::Hash
+        merge_hash!(args.first)
+      elsif streamid = args.first.to_s
+        if streamid.index(Header) == 0
+          merge_str!(streamid)
+        elsif streamid =~ URI::regexp
+          merge_uri!(URI(streamid))
+        else
+          @resource_name = streamid
+        end
+      end
+    end
+
+    def to_s
+      streamid = Header.dup
+
+      vars = []
+
+      vars << "r=#{resource_name}" if resource_name
+      vars << "u=#{user_name}" if user_name
+      vars << "m=#{mode}" if mode
+      vars << "h=#{host_name}" if host_name
+      vars << "t=#{type}" if type
+      vars << "s=#{sessionid}" if sessionid
+      
+      extra.each { |key,val| vars << "#{key}=#{val}" }
+
+      streamid + vars.join(",")
+    end
+
+    def method_missing(name, *args)
+      return extra[name.to_s] if extra.key? name.to_s
+      match = name.to_s.match(/^(.*?)(=?)$/)
+      case match[2]
+      when "="
+        @extra[match[1]] = args.first
+      else
+        return nil
+      end
+    end
+
+    def set(key, val)
+      case key.to_s
+      when 'resource_name', 'r' then @resource_name = val
+      when 'user_name', 'u' then @user_name = val
+      when 'mode', 'm' then @mode = val.to_sym if Modes.include?(val.to_s)
+      when 'host_name', 'h' 
+        if sep = val.index("/")
+          @host_name = val[0..(sep - 1)]
+          (@resource_name = val[(sep + 1)..-1] unless @resource_name)
+        else
+          @host_name = val
+        end
+      when 'type', 't' then (@type = val.to_sym if Types.include?(val.to_s))
+      when 'sessionid', 's' then @sessionid = val
+      else @extra[key] = val
+      end
+    end
+
+    def merge_uri!(streamid)
+      uri = URI(streamid)
+
+      @host_name = uri.hostname
+      @resource_name = uri.path[1..-1] if uri.path
+
+      if uri.query && form = URI.decode_www_form(uri.query)
+        merge_hash!(Hash[form])
+      end
+    end
+
+    def merge_str!(streamid)
+      streamid.scan(/([a-z0-9_-]+?)=(.+?)(,|$)/i).each do |key,val,_|
+        set(key, val)
+      end
+    end
+
+    def merge_hash!(hash)
+      hash.each do |key, val|
+        set(key, val)
+      end
+    end
+  end
+end

data/lib/rbsrt/version.rb

@@ -0,0 +1,3 @@
+module SRT
+  VERSION = "0.1.0"
+end

data/lib/rbsrt.rb

@@ -0,0 +1,5 @@
+$:.push File.expand_path("../", __FILE__)
+
+require_relative "rbsrt/version"
+require_relative "rbsrt/streamid_components"
+require_relative "rbsrt/rbsrt"

data/rbsrt.gemspec

@@ -0,0 +1,22 @@
+$:.push File.expand_path("../lib", __FILE__)
+
+require "rbsrt/version"
+
+Gem::Specification.new do |s|
+  s.name        = 'rbsrt'
+  s.version     = SRT::VERSION
+  s.licenses    = ['MPLv2.0', 'MIT']
+  s.summary     = "Ruby API for SRT (Secure Reliable Transport)"
+  s.description = "Build SRT (Secure Reliable Transport) servers and clients with Ruby"
+  s.authors     = ["Klaas Speller"]
+  s.email       = ["klaas@recce.nl"]
+  s.files       = Dir['README*.md', 'lib/**/*', 'ext/**/*', 'bin/**/*'] + ['Rakefile', 'rbsrt.gemspec']
+  s.test_files  = Dir['test/**/*_test.rb']
+  s.bindir      = "bin"
+  s.executables = ["rbsrt-server", "rbsrt-client"]  
+  s.homepage    = 'https://github.com/spllr/rbsrt'
+  s.metadata    = { "source_code_uri" => "https://github.com/spllr/rbsrt" }
+  s.extensions  = %w[ext/rbsrt/extconf.rb]
+  s.required_ruby_version = ">= 2.2.0"
+  s.add_development_dependency "rake-compiler", "~> 1.0"
+end

data/test/rbsrt/poll_test.rb

@@ -0,0 +1,153 @@
+require 'minitest/spec'
+
+require "rbsrt"
+require "thread"
+
+describe SRT::Poll do
+  it "can be created" do
+    assert_instance_of(SRT::Poll, SRT::Poll.new)
+  end
+
+  
+
+  describe "monitor server sockets" do
+    before do
+      # puts "server start"
+      @server = SRT::Socket.new
+      @server.bind "127.0.0.1", "6789"
+      @server.listen 2
+    end
+  
+    after do
+      # puts "server stop"
+      @client.close if @client
+      @server.close if @server
+    end
+
+    it "will trigger when an new connection is available" do
+      poll = SRT::Poll.new
+
+      poll.add @server, :in
+
+      Thread.new do
+        # puts "connect to server"
+        @client = SRT::Socket.new
+        @client.connect "127.0.0.1", "6789"
+      end
+
+
+      # block/ yield style
+
+      did_call = false
+
+      poll.wait do |readable, writable, errors|  
+        did_call = true
+        refute_empty readable
+        assert_empty writable
+        assert_empty errors
+
+        assert_equal @server, readable.first
+
+        @accepted_sock = readable.first.accept
+
+        poll.add @accepted_sock, :in, :err
+      end
+
+      assert_equal true, did_call, "wait block must yield"
+
+      # return style
+
+      Thread.new do
+        # puts "write data to server"
+        @client.sendmsg "data from client"
+      end
+
+      readable, writable, errors = poll.wait 1000
+
+      refute_empty readable
+      assert_empty writable
+      assert_empty errors
+
+      assert_equal @accepted_sock, readable.first
+
+      assert_equal "data from client", readable.first.recvmsg
+
+      # removing sockets
+
+      poll.remove @client
+
+      Thread.new do
+        # puts "write data to server"
+        @client.sendmsg "data from client"
+      end
+
+      readable, writable, errors = poll.wait 100
+
+      assert_empty readable
+      assert_empty writable
+      assert_empty errors
+    end
+  end
+
+
+  describe "adding and updating sockets" do
+    before do
+      # puts "server start"
+      @server = SRT::Socket.new
+      @server.bind "127.0.0.1", "6789"
+      @server.listen 2
+    end
+  
+    after do
+      # puts "server stop"
+      @client.close if @client
+      @server.close if @server
+    end
+
+    it "can update a socket" do
+      poll = SRT::Poll.new
+
+      poll.add @server, :in
+
+      Thread.new do
+        @client = SRT::Socket.new
+        @client.connect "127.0.0.1", "6789"
+      end
+
+      readable, _, _ = poll.wait 100
+
+      remote_client = readable.first.accept
+
+      # puts "++ IN OUT"
+      poll.add remote_client, :err
+      # puts "-- IN OUT"
+
+      Thread.new do
+        @client.write "foobar 1"
+      end
+
+      readable, writable, errors = poll.wait 100
+
+      assert_empty readable
+      assert_empty writable
+      assert_empty errors
+
+      _ = remote_client.read
+      
+      # upddate
+
+      poll.update remote_client, :in
+
+      Thread.new do
+        @client.write "foobar 2"
+      end
+
+      readable, writable, errors = poll.wait 100
+
+      # pending
+      # refute_empty readable
+      # assert_empty writable
+      # assert_empty errors
+    end
+  end
+end