From 17d57231c4d9d3dfb486b6d37b85d57e6065958e Mon Sep 17 00:00:00 2001
From: Eric Wong <e@80x24.org>
Date: Thu, 16 Mar 2017 03:16:51 +0000
Subject: tcp_info: support this struct under FreeBSD

Of course these fields are not portable between Linux and FreeBSD,
but they should remain ABI-compatible for future versions of each OS.

Tested on FreeBSD 10.3-RELEASE i386

TCP state names will be another problem...
---
 test/test_linux_tcp_info.rb | 68 ------------------------------------
 test/test_tcp_info.rb       | 85 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 85 insertions(+), 68 deletions(-)
 delete mode 100644 test/test_linux_tcp_info.rb
 create mode 100644 test/test_tcp_info.rb

(limited to 'test')

diff --git a/test/test_linux_tcp_info.rb b/test/test_linux_tcp_info.rb
deleted file mode 100644
index c947211..0000000
--- a/test/test_linux_tcp_info.rb
+++ /dev/null
@@ -1,68 +0,0 @@
-# -*- encoding: binary -*-
-require 'test/unit'
-require 'tempfile'
-require 'raindrops'
-require 'socket'
-require 'pp'
-$stderr.sync = $stdout.sync = true
-class TestLinuxTCP_Info < Test::Unit::TestCase
-
-  TEST_ADDR = ENV['UNICORN_TEST_ADDR'] || '127.0.0.1'
-
-  # Linux kernel commit 5ee3afba88f5a79d0bff07ddd87af45919259f91
-  TCP_INFO_useful_listenq = `uname -r`.strip >= '2.6.24'
-
-
-  def test_tcp_server
-    s = TCPServer.new(TEST_ADDR, 0)
-    rv = Raindrops::TCP_Info.new s
-    c = TCPSocket.new TEST_ADDR, s.addr[1]
-    tmp = Raindrops::TCP_Info.new s
-    TCP_INFO_useful_listenq and assert_equal 1, tmp.unacked
-
-    assert_equal 0, rv.unacked
-    a = s.accept
-    tmp = Raindrops::TCP_Info.new s
-    assert_equal 0, tmp.unacked
-    before = tmp.object_id
-
-    tmp.get!(s)
-    assert_equal before, tmp.object_id
-
-    ensure
-      c.close if c
-      a.close if a
-      s.close
-  end
-
-  def test_accessors
-    s = TCPServer.new TEST_ADDR, 0
-    tmp = Raindrops::TCP_Info.new s
-    tcp_info_methods = tmp.methods - Object.new.methods
-    assert tcp_info_methods.size >= 32
-    tcp_info_methods.each do |m|
-      next if m.to_sym == :get!
-      val = tmp.__send__ m
-      assert_kind_of Integer, val
-      assert val >= 0
-    end
-    ensure
-      s.close
-  end
-
-  def test_tcp_server_delayed
-    delay = 0.010
-    delay_ms = (delay * 1000).to_i
-    s = TCPServer.new(TEST_ADDR, 0)
-    c = TCPSocket.new TEST_ADDR, s.addr[1]
-    c.syswrite "."
-    sleep(delay * 1.2)
-    a = s.accept
-    i = Raindrops::TCP_Info.new(a)
-    assert i.last_data_recv >= delay_ms, "#{i.last_data_recv} < #{delay_ms}"
-    ensure
-      c.close if c
-      a.close if a
-      s.close
-  end
-end if RUBY_PLATFORM =~ /linux/
diff --git a/test/test_tcp_info.rb b/test/test_tcp_info.rb
new file mode 100644
index 0000000..15df087
--- /dev/null
+++ b/test/test_tcp_info.rb
@@ -0,0 +1,85 @@
+# -*- encoding: binary -*-
+require 'test/unit'
+require 'tempfile'
+require 'raindrops'
+require 'socket'
+require 'pp'
+$stderr.sync = $stdout.sync = true
+class TestTCP_Info < Test::Unit::TestCase
+
+  TEST_ADDR = ENV['UNICORN_TEST_ADDR'] || '127.0.0.1'
+
+  # Linux kernel commit 5ee3afba88f5a79d0bff07ddd87af45919259f91
+  TCP_INFO_useful_listenq = `uname -r`.strip >= '2.6.24'
+
+  def test_tcp_server_unacked
+    return if RUBY_PLATFORM !~ /linux/ # unacked not implemented on others...
+    s = TCPServer.new(TEST_ADDR, 0)
+    rv = Raindrops::TCP_Info.new s
+    c = TCPSocket.new TEST_ADDR, s.addr[1]
+    tmp = Raindrops::TCP_Info.new s
+    TCP_INFO_useful_listenq and assert_equal 1, tmp.unacked
+
+    assert_equal 0, rv.unacked
+    a = s.accept
+    tmp = Raindrops::TCP_Info.new s
+    assert_equal 0, tmp.unacked
+    before = tmp.object_id
+
+    tmp.get!(s)
+    assert_equal before, tmp.object_id
+
+  ensure
+    [ c, a, s ].compact.each(&:close)
+  end
+
+  def test_accessors
+    s = TCPServer.new TEST_ADDR, 0
+    tmp = Raindrops::TCP_Info.new s
+    tcp_info_methods = tmp.methods - Object.new.methods
+    assert tcp_info_methods.size >= 32
+    tcp_info_methods.each do |m|
+      next if m.to_sym == :get!
+      next if ! tmp.respond_to?(m)
+      val = tmp.__send__ m
+      assert_kind_of Integer, val
+      assert val >= 0
+    end
+    assert tmp.respond_to?(:state), 'every OS knows about TCP state, right?'
+  ensure
+    s.close
+  end
+
+  def test_tcp_server_delayed
+    delay = 0.010
+    delay_ms = (delay * 1000).to_i
+    s = TCPServer.new(TEST_ADDR, 0)
+    c = TCPSocket.new TEST_ADDR, s.addr[1]
+    c.syswrite "."
+    sleep(delay * 1.2)
+    a = s.accept
+    i = Raindrops::TCP_Info.new(a)
+    assert i.last_data_recv >= delay_ms, "#{i.last_data_recv} < #{delay_ms}"
+    ensure
+      c.close if c
+      a.close if a
+      s.close
+  end
+
+  def test_tcp_server_state_closed
+    s = TCPServer.new(TEST_ADDR, 0)
+    c = TCPSocket.new(TEST_ADDR, s.addr[1])
+    i = Raindrops::TCP_Info.allocate
+    a = s.accept
+    i.get!(a)
+    state = i.state
+    c = c.close
+    sleep(0.01) # wait for kernel to update state
+    i.get!(a)
+    assert_not_equal state, i.state
+  ensure
+    s.close if s
+    c.close if c
+    a.close if a
+  end
+end if defined? Raindrops::TCP_Info
-- 
cgit v1.2.3-24-ge0c7