about summary refs log tree commit homepage
diff options
context:
space:
mode:
-rw-r--r--lib/dtas/player/client_handler.rb8
-rw-r--r--test/test_player_client_handler.rb12
2 files changed, 20 insertions, 0 deletions
diff --git a/lib/dtas/player/client_handler.rb b/lib/dtas/player/client_handler.rb
index 16714f0..7f1c72e 100644
--- a/lib/dtas/player/client_handler.rb
+++ b/lib/dtas/player/client_handler.rb
@@ -416,6 +416,14 @@ module DTAS::Player::ClientHandler # :nodoc:
   end
 
   def env_handler(io, msg)
+    if msg.empty?
+      # this may fail for large envs due to SEQPACKET size restrictions
+      # do we care?
+      env = ENV.map do |k,v|
+        "#{Shellwords.escape(k)}=#{Shellwords.escape(v)}"
+      end.join(' ')
+      return io.emit(env)
+    end
     msg.each do |kv|
       case kv
       when %r{\A([^=]+)=(.*)\z}
diff --git a/test/test_player_client_handler.rb b/test/test_player_client_handler.rb
index 21d8489..edcb35d 100644
--- a/test/test_player_client_handler.rb
+++ b/test/test_player_client_handler.rb
@@ -82,4 +82,16 @@ class TestPlayerClientHandler < Minitest::Unit::TestCase
     sink_handler(@io, %W(ls))
     assert_equal expect, Shellwords.split(@io[0])
   end
+
+  def test_env
+    env_handler(@io, [])
+    res = @io[0]
+    result = {}
+    Shellwords.split(res).each do |kv|
+      k, v = kv.split(/=/, 2)
+      result[k] = v
+    end
+    expect = ENV.to_hash
+    assert_equal expect, result
+  end
 end