Commit: 5c8ecd515f4a0271af95433c1fb13a059320c63e - shogi-server (git)

shogi-server source

Commit MetaInfo

• Revision: 5c8ecd515f4a0271af95433c1fb13a059320c63e (tree)
• Time: 2020-12-14 20:42:55
• Author: Mizar <mizar.jp@gmai...>
• Commiter: Mizar

Log Message

tcp_keepalive

Change Summary

• modified: bin/usiToCsa.rb
• modified: shogi-server
• modified: shogi_server.rb

Incremental Difference

--- a/bin/usiToCsa.rb
+++ b/bin/usiToCsa.rb
@@ -103,7 +103,11 @@ def parse_command_line
     ["--password",    GetoptLong::REQUIRED_ARGUMENT],
     ["--ponder",      GetoptLong::NO_ARGUMENT],
     ["--port",        GetoptLong::REQUIRED_ARGUMENT],
-    ["--floodgate",   GetoptLong::NO_ARGUMENT])
+    ["--floodgate",   GetoptLong::NO_ARGUMENT],
+    ["--tcpkeepalive",       GetoptLong::REQUIRED_ARGUMENT],
+    ["--tcpkeepalive-idle",  GetoptLong::REQUIRED_ARGUMENT],
+    ["--tcpkeepalive-intvl", GetoptLong::REQUIRED_ARGUMENT],
+    ["--tcpkeepalive-cnt",   GetoptLong::REQUIRED_ARGUMENT])
   parser.quiet = true
   begin
     parser.each_option do |name, arg|
@@ -131,6 +135,14 @@ def parse_command_line
   options[:port]        ||= ENV["PORT"] || 4081
   options[:port]        = options[:port].to_i
   options[:floodgate]   ||= ENV["FLOODGATE"] || false
+  options[:tcpkeepalive]       ||= ENV["TCPKEEPALIVE"] || 0
+  options[:tcpkeepalive]       = options[:tcpkeepalive].to_i
+  options[:tcpkeepalive_idle]  ||= ENV["TCPKEEPALIVE_IDLE"] || 50
+  options[:tcpkeepalive_idle]  = options[:tcpkeepalive_idle].to_i
+  options[:tcpkeepalive_intvl] ||= ENV["TCPKEEPALIVE_INTVL"] || 5
+  options[:tcpkeepalive_intvl] = options[:tcpkeepalive_intvl].to_i
+  options[:tcpkeepalive_cnt]   ||= ENV["TCPKEEPALIVE_CNT"] || 9
+  options[:tcpkeepalive_cnt]   = options[:tcpkeepalive_cnt].to_i
 
   return options
 end
@@ -623,6 +635,30 @@ def login
   begin
     $server = TCPSocket.open($options[:host], $options[:port])
     $server.sync = true
+    if $options[:tcpkeepalive] > 0 &&
+      Socket.const_defined?(:SOL_SOCKET) &&
+      Socket.const_defined?(:SO_KEEPALIVE) &&
+      Socket.const_defined?(:IPPROTO_TCP)
+      $server.setsockopt(Socket::SOL_SOCKET, Socket::SO_KEEPALIVE, true)
+      if Socket.const_defined?(:TCP_KEEPIDLE) &&
+        Socket.const_defined?(:TCP_KEEPINTVL) &&
+        Socket.const_defined?(:TCP_KEEPCNT)
+        $server.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_KEEPIDLE, $options[:tcpkeepalive_idle])
+        $server.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_KEEPINTVL, $options[:tcpkeepalive_intvl])
+        $server.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_KEEPCNT, $options[:tcpkeepalive_cnt])
+      elsif RUBY_PLATFORM.downcase =~ /mswin|mingw|cygwin|bccwin/
+        # Windows 10 (1709 or later) ws2ipdef.h
+        # https://docs.microsoft.com/en-us/windows/win32/winsock/ipproto-tcp-socket-options
+        $server.setsockopt(Socket::IPPROTO_TCP, 3, $options[:tcpkeepalive_idle])
+        $server.setsockopt(Socket::IPPROTO_TCP, 17, $options[:tcpkeepalive_intvl])
+        $server.setsockopt(Socket::IPPROTO_TCP, 16, $options[:tcpkeepalive_cnt])
+      elsif RUBY_PLATFORM.downcase =~ /darwin/
+        # macOS 10.12.6 (Sierra) /usr/include/netinet/tcp.h
+        $server.setsockopt(Socket::IPPROTO_TCP, 0x10, $options[:tcpkeepalive_idle])
+        $server.setsockopt(Socket::IPPROTO_TCP, 0x101, $options[:tcpkeepalive_intvl])
+        $server.setsockopt(Socket::IPPROTO_TCP, 0x102, $options[:tcpkeepalive_cnt])
+      end
+    end
   rescue
     log_error "Failed to connect to the server"
     $server = nil

--- a/shogi-server
+++ b/shogi-server
@@ -208,7 +208,10 @@ def parse_command_line
     ["--least-time-per-move", GetoptLong::REQUIRED_ARGUMENT],
     ["--max-moves",           GetoptLong::REQUIRED_ARGUMENT],
     ["--pid-file",            GetoptLong::REQUIRED_ARGUMENT],
-    ["--player-log-dir",      GetoptLong::REQUIRED_ARGUMENT])
+    ["--player-log-dir",      GetoptLong::REQUIRED_ARGUMENT],
+    ["--tcpkeepalive-idle",   GetoptLong::REQUIRED_ARGUMENT],
+    ["--tcpkeepalive-intvl",  GetoptLong::REQUIRED_ARGUMENT],
+    ["--tcpkeepalive-cnt",    GetoptLong::REQUIRED_ARGUMENT])
   parser.quiet = true
   begin
     parser.each_option do |name, arg|
@@ -283,6 +286,15 @@ def check_command_line
 
   $options["least-time-per-move"] ||= ShogiServer::Default_Least_Time_Per_Move
   $options["least-time-per-move"] = $options["least-time-per-move"].to_i
+
+  $options["tcpkeepalive-idle"] ||= ShogiServer::Default_TcpKeepAlive_Idle
+  $options["tcpkeepalive-idle"] = $options["tcpkeepalive-idle"].to_i
+
+  $options["tcpkeepalive-intvl"] ||= ShogiServer::Default_TcpKeepAlive_Intvl
+  $options["tcpkeepalive-intvl"] = $options["tcpkeepalive-intvl"].to_i
+
+  $options["tcpkeepalive-cnt"] ||= ShogiServer::Default_TcpKeepAlive_Cnt
+  $options["tcpkeepalive-cnt"] = $options["tcpkeepalive-cnt"].to_i
 end
 
 # See if a file can be created in the directory.
@@ -455,8 +467,30 @@ def main
   server.start do |client|
     begin
       # client.sync = true # this is already set in WEBrick 
-      client.setsockopt(Socket::SOL_SOCKET, Socket::SO_KEEPALIVE, true)
-        # Keepalive time can be set by /proc/sys/net/ipv4/tcp_keepalive_time
+      if Socket.const_defined?(:SOL_SOCKET) &&
+        Socket.const_defined?(:SO_KEEPALIVE) &&
+        Socket.const_defined?(:IPPROTO_TCP)
+        client.setsockopt(Socket::SOL_SOCKET, Socket::SO_KEEPALIVE, true)
+          # Keepalive time can be set by /proc/sys/net/ipv4/tcp_keepalive_time
+        if Socket.const_defined?(:TCP_KEEPIDLE) &&
+          Socket.const_defined?(:TCP_KEEPINTVL) &&
+          Socket.const_defined?(:TCP_KEEPCNT)
+          client.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_KEEPIDLE, $options["tcpkeepalive-idle"])
+          client.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_KEEPINTVL, $options["tcpkeepalive-intvl"])
+          client.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_KEEPCNT, $options["tcpkeepalive-cnt"])
+        elsif RUBY_PLATFORM.downcase =~ /mswin|mingw|cygwin|bccwin/
+          # Windows 10 (1709 or later) ws2ipdef.h
+          # https://docs.microsoft.com/en-us/windows/win32/winsock/ipproto-tcp-socket-options
+          client.setsockopt(Socket::IPPROTO_TCP, 3, $options["tcpkeepalive-idle"])
+          client.setsockopt(Socket::IPPROTO_TCP, 17, $options["tcpkeepalive-intvl"])
+          client.setsockopt(Socket::IPPROTO_TCP, 16, $options["tcpkeepalive-cnt"])
+        elsif RUBY_PLATFORM.downcase =~ /darwin/
+          # macOS 10.12.6 (Sierra) /usr/include/netinet/tcp.h
+          client.setsockopt(Socket::IPPROTO_TCP, 0x10, $options["tcpkeepalive-idle"])
+          client.setsockopt(Socket::IPPROTO_TCP, 0x101, $options["tcpkeepalive-intvl"])
+          client.setsockopt(Socket::IPPROTO_TCP, 0x102, $options["tcpkeepalive-cnt"])
+        end
+      end
       player, login = login_loop(client) # loop
       unless player
         log_error("Detected a timed out login attempt")

--- a/shogi_server.rb
+++ b/shogi_server.rb
@@ -50,6 +50,9 @@ Default_Timeout = 60            # for single socket operation
 Default_Game_Name = "default-1500-0"
 Default_Max_Moves = 256
 Default_Least_Time_Per_Move = 0
+Default_TcpKeepAlive_Idle = 50
+Default_TcpKeepAlive_Intvl = 5
+Default_TcpKeepAlive_Cnt = 9
 One_Time = 10
 Login_Time = 300                # time for LOGIN
 Revision = "20160409"